[PWGEM/Dilepton] update taggingHFE.cxx

PWGEM/Dilepton/DataModel/lmeeMLTables.h

@@ -46,11 +46,11 @@
 {
 DECLARE_SOA_COLUMN(CollisionId, collisionId, int);                   //!
 DECLARE_SOA_COLUMN(HadronicRate, hadronicRate, float);               //!
 DECLARE_SOA_COLUMN(PIDLabel, pidlabel, uint8_t);                     //!
 DECLARE_SOA_COLUMN(TrackType, tracktype, uint8_t);                   //!
 DECLARE_SOA_COLUMN(TPCNClsFound, tpcNClsFound, uint8_t);             //!
 DECLARE_SOA_COLUMN(TPCNClsCrossedRows, tpcNClsCrossedRows, uint8_t); //!
 DECLARE_SOA_COLUMN(TPCNClsPID, tpcNClsPID, uint8_t);                 //!
 DECLARE_SOA_COLUMN(IsForValidation, isForValidation, bool);          //!
 DECLARE_SOA_COLUMN(Sign, sign, short);                               //!
 DECLARE_SOA_COLUMN(P, p, float);                                     //!
@@ -61,7 +61,7 @@
 // DECLARE_SOA_DYNAMIC_COLUMN(P, p, [](float pt, float eta) -> float { return pt * std::cosh(eta); });
 DECLARE_SOA_DYNAMIC_COLUMN(MeanClusterSizeITS, meanClusterSizeITS, [](uint32_t itsClusterSizes) -> float {
   int total_cluster_size = 0, nl = 0;
   for (unsigned int layer = 0; layer < 7; layer++) {
     int cluster_size_per_layer = (itsClusterSizes >> (layer * 4)) & 0xf;
     if (cluster_size_per_layer > 0) {
       nl++;
@@ -76,7 +76,7 @@
 });
 DECLARE_SOA_DYNAMIC_COLUMN(MeanClusterSizeITSob, meanClusterSizeITSob, [](uint32_t itsClusterSizes) -> float {
   int total_cluster_size = 0, nl = 0;
   for (unsigned int layer = 3; layer < 7; layer++) {
     int cluster_size_per_layer = (itsClusterSizes >> (layer * 4)) & 0xf;
     if (cluster_size_per_layer > 0) {
       nl++;
@@ -162,8 +162,8 @@
 DECLARE_SOA_COLUMN(IsCorrectMatch, isCorrectMatch, bool);   //!
 DECLARE_SOA_COLUMN(PdgCodeMFT, pdgCodeMFT, int);            //!
 DECLARE_SOA_COLUMN(PdgCodeMCHMID, pdgCodeMCHMID, int);      //!
 DECLARE_SOA_COLUMN(MatchMCHTrackId, mchTrackId, int);       //!
 DECLARE_SOA_COLUMN(DFId, dfId, uint64_t);                   //!
 
 DECLARE_SOA_COLUMN(MultMFT, multMFT, uint16_t); //! number of MFTsa tracks per collision
 } // namespace emmlfwdtrack
@@ -195,13 +195,27 @@
 using EMFwdTrackErrForML = EMFwdTrackErrsForML::iterator;
 
 // for SemiCharmTag at midrapidity, only electrons
-namespace emmllhpair
+namespace emmlevent
+{
+DECLARE_SOA_COLUMN(SubGeneratorId, subGeneratorId, int); //! sub generator Id of mc collision
+} // namespace emmlevent
+namespace emmltrack
 {
-DECLARE_SOA_COLUMN(Signed1PtL, signed1PtL, float); //! signed1Pt of lepton
-DECLARE_SOA_COLUMN(EtaL, etaL, float);             //! eta of lepton
-DECLARE_SOA_COLUMN(DcaL, dcaL, float);             //! dca of lepton
-DECLARE_SOA_COLUMN(DcaLSigma, dcaLsigma, float);   //! dca of lepton
+DECLARE_SOA_COLUMN(IsMotherFromHF, isMotherFromHF, bool); //! is HF included in decay history
+DECLARE_SOA_COLUMN(PdgCodeMother, pdgCodeMother, int);    //! pdg code of mother of lepton
+} // namespace emmltrack
 
+DECLARE_SOA_TABLE(EMMLLeptons, "AOD", "EMMLLEPTON", //!
+                  o2::soa::Index<>, collision::NumContrib, evsel::NumTracksInTimeRange, evsel::SumAmpFT0CInTimeRange, emmlevent::SubGeneratorId,
+                  track::Signed1Pt, track::Eta,
+                  track::DcaXY, track::DcaZ, o2::aod::track::CYY, o2::aod::track::CZY, o2::aod::track::CZZ,
+                  emmltrack::IsMotherFromHF, emmltrack::PdgCodeMother);
+// iterators
+using EMMLLepton = EMMLLeptons::iterator;
+
+namespace emmllhpair
+{
+DECLARE_SOA_INDEX_COLUMN(EMMLLepton, emmllepton);  //! most propable emeventId
 DECLARE_SOA_COLUMN(Signed1PtH, signed1PtH, float); //! signed1Pt of hadron
 DECLARE_SOA_COLUMN(EtaH, etaH, float);             //! eta of hadron
 DECLARE_SOA_COLUMN(DcaH, dcaH, float);             //! dca of hadron
@@ -214,16 +228,17 @@
 DECLARE_SOA_COLUMN(Lxyz, lxyz, float);           //! decay length of LH pair
 DECLARE_SOA_COLUMN(LxyzSigma, lxyzSigma, float); //! decay length resolution of LH pair
 
-DECLARE_SOA_COLUMN(PdgCodeH, pdgCodeH, int);     //! pdg code of associated hadron
-DECLARE_SOA_COLUMN(PdgCodeHFH, pdgCodeHFH, int); //! pdg code of HF hadron
+DECLARE_SOA_COLUMN(PdgCodeH, pdgCodeH, int);                    //! pdg code of associated hadron
+DECLARE_SOA_COLUMN(PdgCodeHFH, pdgCodeHFH, int);                //! pdg code of HF hadron
+DECLARE_SOA_COLUMN(FoundCommonMother, foundCommonMother, bool); //! decay length resolution of LH pair
 } // namespace emmllhpair
 
 DECLARE_SOA_TABLE(EMMLLHPairs, "AOD", "EMMLLHPAIR", //!
-                  o2::soa::Index<>, collision::NumContrib, evsel::NumTracksInTimeRange, evsel::SumAmpFT0CInTimeRange,
-                  emmllhpair::Signed1PtL, emmllhpair::EtaL, emmllhpair::DcaL, emmllhpair::DcaLSigma,
-                  emmllhpair::Signed1PtH, emmllhpair::EtaH, emmllhpair::DcaH, emmllhpair::DcaHSigma, emmllhpair::NSigmaKa,
+                  emmllhpair::EMMLLeptonId,
+                  track::Signed1Pt, track::Eta,
+                  track::DcaXY, track::DcaZ, o2::aod::track::CYY, o2::aod::track::CZY, o2::aod::track::CZZ, pidtpc::TPCNSigmaKa, pidtof::TOFNSigmaKa,
                   emmllhpair::Mass, emmllhpair::DcaLH, emmllhpair::CosPA, emmllhpair::Lxyz, emmllhpair::LxyzSigma,
-                  emmllhpair::PdgCodeH, emmllhpair::PdgCodeHFH);
+                  emmllhpair::PdgCodeH, emmllhpair::FoundCommonMother);
 
 // iterators
 using EMMLLHPair = EMMLLHPairs::iterator;