[PWGEM/Dilepton] update taggingHFE.cxx (#16032)

Author: dsekihat

PWGEM/Dilepton/DataModel/lmeeMLTables.h

@@ -222,11 +222,16 @@ DECLARE_SOA_COLUMN(DcaH, dcaH, float);             //! dca of hadron
 DECLARE_SOA_COLUMN(DcaHSigma, dcaHsigma, float);   //! dca resolution of hadron
 DECLARE_SOA_COLUMN(NSigmaKa, nSigmaKa, float);     //! PID n sigma with respect to charged Kaon
 
-DECLARE_SOA_COLUMN(Mass, mass, float);           //! invariant mass of LH assuming kaon
-DECLARE_SOA_COLUMN(DcaLH, dcalh, float);         //! DCA between lepton and hadron
-DECLARE_SOA_COLUMN(CosPA, cospa, float);         //! flight direction of LH pair
+DECLARE_SOA_COLUMN(Mass, mass, float);   //! invariant mass of LH assuming kaon
+DECLARE_SOA_COLUMN(DcaLH, dcalh, float); //! DCA between lepton and hadron
+DECLARE_SOA_COLUMN(CosPA, cospa, float); //! flight direction of LH pair
+
 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(Lxy, lxy, float);             //! decay length of LH pair
+DECLARE_SOA_COLUMN(LxySigma, lxySigma, float);   //! decay length resolution of LH pair
+DECLARE_SOA_COLUMN(Lz, lz, float);               //! decay length of LH pair
+DECLARE_SOA_COLUMN(LzSigma, lzSigma, 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
@@ -237,7 +242,10 @@ DECLARE_SOA_TABLE(EMMLLHPairs, "AOD", "EMMLLHPAIR", //!
                   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::Mass, emmllhpair::DcaLH, emmllhpair::CosPA,
+                  emmllhpair::Lxyz, emmllhpair::LxyzSigma,
+                  emmllhpair::Lxy, emmllhpair::LxySigma,
+                  emmllhpair::Lz, emmllhpair::LzSigma,
                   emmllhpair::PdgCodeH, emmllhpair::FoundCommonMother);
 
 // iterators

PWGEM/Dilepton/Tasks/taggingHFE.cxx

@@ -1258,7 +1258,11 @@ struct taggingHFE {
           emmllhpair(leptonTable.lastIndex(),
                      trackParCov.getQ2Pt(), trackParCov.getEta(), dcaXY_kaon, dcaZ_kaon, trackParCov.getSigmaY2(), trackParCov.getSigmaZY(), trackParCov.getSigmaZ2(),
                      kaon.tpcNSigmaKa(), kaon.tofNSigmaKa(),
-                     eKpair.mass, eKpair.dca2legs, eKpair.cospa, eKpair.lxyz, eKpair.lxyzErr, mckaon.pdgCode(), foundCommonMother);
+                     eKpair.mass, eKpair.dca2legs, eKpair.cospa,
+                     eKpair.lxyz, eKpair.lxyzErr,
+                     eKpair.lxy, eKpair.lxyErr,
+                     eKpair.lz, eKpair.lzErr,
+                     mckaon.pdgCode(), foundCommonMother);
 
         } // end of kaon loop
 
@@ -1463,7 +1467,11 @@ struct taggingHFE {
           emmllhpair(leptonTable.lastIndex(),
                      trackParCov.getQ2Pt(), trackParCov.getEta(), dcaXY_kaon, dcaZ_kaon, trackParCov.getSigmaY2(), trackParCov.getSigmaZY(), trackParCov.getSigmaZ2(),
                      kaon.tpcNSigmaKa(), kaon.tofNSigmaKa(),
-                     eKpair.mass, eKpair.dca2legs, eKpair.cospa, eKpair.lxyz, eKpair.lxyzErr, mckaon.pdgCode(), foundCommonMother);
+                     eKpair.mass, eKpair.dca2legs, eKpair.cospa,
+                     eKpair.lxyz, eKpair.lxyzErr,
+                     eKpair.lxy, eKpair.lxyErr,
+                     eKpair.lz, eKpair.lzErr,
+                     mckaon.pdgCode(), foundCommonMother);
 
         } // end of kaon loop