Skip to content

Commit c962180

Browse files
dsekihatdsekihat
committed
[PWGEM/Dilepton] put pair.pt back in taggingHFE
1 parent c35d8b3 commit c962180

3 files changed

Lines changed: 21 additions & 16 deletions

File tree

‎PWGEM/Dilepton/DataModel/lmeeMLTables.h‎

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -259,7 +259,8 @@ DECLARE_SOA_COLUMN(V0Type, v0Type, uint8_t); //! v0 type, 0 = K0S, 1 =
259259
DECLARE_SOA_COLUMN(CascadeType, cascadeType, uint8_t); //! cascade type, 0 = XiMunus, 1 = OmegaMinus
260260

261261
// LH pair variables
262-
DECLARE_SOA_COLUMN(MassLH, massLH, float); //! invariant mass of LH assuming pion
262+
DECLARE_SOA_COLUMN(MassLH, massLH, float); //! invariant mass of LH assuming kaon
263+
DECLARE_SOA_COLUMN(PtLH, ptLH, float); //! pt of LH
263264

264265
DECLARE_SOA_COLUMN(PtSVL, ptSVL, float); //! pT of lepton at SV
265266
DECLARE_SOA_COLUMN(PlSVL, plSVL, float); //! pL of lepton at SV
@@ -304,7 +305,7 @@ DECLARE_SOA_TABLE(EMMLLTPairs, "AOD", "EMMLLTPAIR", //!
304305
// pidtpc::TPCNSigmaPi, pidtof::TOFNSigmaPi,
305306
pidtpc::TPCNSigmaKa, pidtof::TOFNSigmaKa,
306307
// pidtpc::TPCNSigmaPr, pidtof::TOFNSigmaPr,
307-
emmllhpair::MassLH, emmllhpair::PtSVL, emmllhpair::PlSVL, emmllhpair::PtSVH, emmllhpair::PlSVH,
308+
emmllhpair::MassLH, emmllhpair::PtLH, emmllhpair::PtSVL, emmllhpair::PlSVL, emmllhpair::PtSVH, emmllhpair::PlSVH,
308309
emmllhpair::PtFD, emmllhpair::PlFD, emmllhpair::PtFDL, emmllhpair::PlFDL, emmllhpair::PtFDH, emmllhpair::PlFDH,
309310
emmllhpair::DcaLH, emmllhpair::CPA, emmllhpair::CPAXY, emmllhpair::CPARZ,
310311
emmllhpair::Lxy, emmllhpair::Lz, emmllhpair::Lxyz, emmllhpair::LxyErr, emmllhpair::LzErr, emmllhpair::LxyzErr,
@@ -318,7 +319,7 @@ DECLARE_SOA_TABLE(EMMLLV0Pairs, "AOD", "EMMLLV0PAIR", //!
318319
emmllhpair::PtH, emmllhpair::RapidityV0,
319320
emmllhpair::V0CPA, emmllhpair::V0CPAXY, emmllhpair::V0CPARZ,
320321
emmllhpair::ImpParXYH, emmllhpair::ImpParZH, emmllhpair::ImpParCYYH, emmllhpair::ImpParCZYH, emmllhpair::ImpParCZZH,
321-
emmllhpair::MassLH, emmllhpair::PtSVL, emmllhpair::PlSVL, emmllhpair::PtSVH, emmllhpair::PlSVH,
322+
emmllhpair::MassLH, emmllhpair::PtLH, emmllhpair::PtSVL, emmllhpair::PlSVL, emmllhpair::PtSVH, emmllhpair::PlSVH,
322323
emmllhpair::PtFD, emmllhpair::PlFD, emmllhpair::PtFDL, emmllhpair::PlFDL, emmllhpair::PtFDH, emmllhpair::PlFDH,
323324
emmllhpair::DcaLH, emmllhpair::CPA, emmllhpair::CPAXY, emmllhpair::CPARZ,
324325
emmllhpair::Lxy, emmllhpair::Lz, emmllhpair::Lxyz, emmllhpair::LxyErr, emmllhpair::LzErr, emmllhpair::LxyzErr,
@@ -332,7 +333,7 @@ DECLARE_SOA_TABLE(EMMLLCascPairs, "AOD", "EMMLLCPAIR", //!
332333
emmllhpair::Signed1PtH, emmllhpair::RapidityC,
333334
emmllhpair::CascCPA, emmllhpair::CascCPAXY, emmllhpair::CascCPARZ,
334335
emmllhpair::ImpParXYH, emmllhpair::ImpParZH, emmllhpair::ImpParCYYH, emmllhpair::ImpParCZYH, emmllhpair::ImpParCZZH,
335-
emmllhpair::MassLH, emmllhpair::PtSVL, emmllhpair::PlSVL, emmllhpair::PtSVH, emmllhpair::PlSVH,
336+
emmllhpair::MassLH, emmllhpair::PtLH, emmllhpair::PtSVL, emmllhpair::PlSVL, emmllhpair::PtSVH, emmllhpair::PlSVH,
336337
emmllhpair::PtFD, emmllhpair::PlFD, emmllhpair::PtFDL, emmllhpair::PlFDL, emmllhpair::PtFDH, emmllhpair::PlFDH,
337338
emmllhpair::DcaLH, emmllhpair::CPA, emmllhpair::CPAXY, emmllhpair::CPARZ,
338339
emmllhpair::Lxy, emmllhpair::Lz, emmllhpair::Lxyz, emmllhpair::LxyErr, emmllhpair::LzErr, emmllhpair::LxyzErr,

‎PWGEM/Dilepton/Tasks/taggingHFE.cxx‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1762,7 +1762,7 @@ struct taggingHFE {
17621762
// kaon.tpcNSigmaPi(), tofNSigmaPi,
17631763
kaon.tpcNSigmaKa(), tofNSigmaKa,
17641764
// kaon.tpcNSigmaPr(), tofNSigmaPr,
1765-
eKpair.mass, eKpair.ptSVL, eKpair.plSVL, eKpair.ptSVH, eKpair.plSVH,
1765+
eKpair.mass, eKpair.pt, eKpair.ptSVL, eKpair.plSVL, eKpair.ptSVH, eKpair.plSVH,
17661766
eKpair.ptFD, eKpair.plFD, eKpair.ptFDL, eKpair.plFDL, eKpair.ptFDH, eKpair.plFDH,
17671767
eKpair.dca2legs, eKpair.cospa, eKpair.cospaXY, eKpair.cospaRZ,
17681768
eKpair.lxy, eKpair.lz, eKpair.lxyz, eKpair.lxyErr, eKpair.lzErr, eKpair.lxyzErr,
@@ -1823,7 +1823,7 @@ struct taggingHFE {
18231823
// kaon.tpcNSigmaPi(), tofNSigmaPi,
18241824
kaon.tpcNSigmaKa(), tofNSigmaKa,
18251825
// kaon.tpcNSigmaPr(), tofNSigmaPr,
1826-
eKpair.mass, eKpair.ptSVL, eKpair.plSVL, eKpair.ptSVH, eKpair.plSVH,
1826+
eKpair.mass, eKpair.pt, eKpair.ptSVL, eKpair.plSVL, eKpair.ptSVH, eKpair.plSVH,
18271827
eKpair.ptFD, eKpair.plFD, eKpair.ptFDL, eKpair.plFDL, eKpair.ptFDH, eKpair.plFDH,
18281828
eKpair.dca2legs, eKpair.cospa, eKpair.cospaXY, eKpair.cospaRZ,
18291829
eKpair.lxy, eKpair.lz, eKpair.lxyz, eKpair.lxyErr, eKpair.lzErr, eKpair.lxyzErr,
@@ -1902,7 +1902,7 @@ struct taggingHFE {
19021902
RecoDecay::cpaXY(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{v0.x(), v0.y(), v0.z()}, std::array<float, 3>{v0.px(), v0.py(), v0.pz()}),
19031903
RecoDecay::cpaRZ(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{v0.x(), v0.y(), v0.z()}, std::array<float, 3>{v0.px(), v0.py(), v0.pz()}),
19041904
impactParameterV0.getY(), impactParameterV0.getZ(), impactParameterV0.getSigmaY2(), impactParameterV0.getSigmaYZ(), impactParameterV0.getSigmaZ2(),
1905-
eV0pair.mass, eV0pair.ptSVL, eV0pair.plSVL, eV0pair.ptSVH, eV0pair.plSVH,
1905+
eV0pair.mass, eV0pair.pt, eV0pair.ptSVL, eV0pair.plSVL, eV0pair.ptSVH, eV0pair.plSVH,
19061906
eV0pair.ptFD, eV0pair.plFD, eV0pair.ptFDL, eV0pair.plFDL, eV0pair.ptFDH, eV0pair.plFDH,
19071907
eV0pair.dca2legs, eV0pair.cospa, eV0pair.cospaXY, eV0pair.cospaRZ,
19081908
eV0pair.lxy, eV0pair.lz, eV0pair.lxyz, eV0pair.lxyErr, eV0pair.lzErr, eV0pair.lxyzErr,
@@ -1965,7 +1965,7 @@ struct taggingHFE {
19651965
RecoDecay::cpaXY(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{v0.x(), v0.y(), v0.z()}, std::array<float, 3>{v0.px(), v0.py(), v0.pz()}),
19661966
RecoDecay::cpaRZ(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{v0.x(), v0.y(), v0.z()}, std::array<float, 3>{v0.px(), v0.py(), v0.pz()}),
19671967
impactParameterV0.getY(), impactParameterV0.getZ(), impactParameterV0.getSigmaY2(), impactParameterV0.getSigmaYZ(), impactParameterV0.getSigmaZ2(),
1968-
eV0pair.mass, eV0pair.ptSVL, eV0pair.plSVL, eV0pair.ptSVH, eV0pair.plSVH,
1968+
eV0pair.mass, eV0pair.pt, eV0pair.ptSVL, eV0pair.plSVL, eV0pair.ptSVH, eV0pair.plSVH,
19691969
eV0pair.ptFD, eV0pair.plFD, eV0pair.ptFDL, eV0pair.plFDL, eV0pair.ptFDH, eV0pair.plFDH,
19701970
eV0pair.dca2legs, eV0pair.cospa, eV0pair.cospaXY, eV0pair.cospaRZ,
19711971
eV0pair.lxy, eV0pair.lz, eV0pair.lxyz, eV0pair.lxyErr, eV0pair.lzErr, eV0pair.lxyzErr,
@@ -2033,7 +2033,7 @@ struct taggingHFE {
20332033
RecoDecay::cpaXY(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{cascade.x(), cascade.y(), cascade.z()}, std::array<float, 3>{cascade.px(), cascade.py(), cascade.pz()}),
20342034
RecoDecay::cpaRZ(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{cascade.x(), cascade.y(), cascade.z()}, std::array<float, 3>{cascade.px(), cascade.py(), cascade.pz()}),
20352035
impactParameterCasc.getY(), impactParameterCasc.getZ(), impactParameterCasc.getSigmaY2(), impactParameterCasc.getSigmaYZ(), impactParameterCasc.getSigmaZ2(),
2036-
eCpair.mass, eCpair.ptSVL, eCpair.plSVL, eCpair.ptSVH, eCpair.plSVH,
2036+
eCpair.mass, eCpair.pt, eCpair.ptSVL, eCpair.plSVL, eCpair.ptSVH, eCpair.plSVH,
20372037
eCpair.ptFD, eCpair.plFD, eCpair.ptFDL, eCpair.plFDL, eCpair.ptFDH, eCpair.plFDH,
20382038
eCpair.dca2legs, eCpair.cospa, eCpair.cospaXY, eCpair.cospaRZ,
20392039
eCpair.lxy, eCpair.lz, eCpair.lxyz, eCpair.lxyErr, eCpair.lzErr, eCpair.lxyzErr,
@@ -2101,7 +2101,7 @@ struct taggingHFE {
21012101
RecoDecay::cpaXY(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{cascade.x(), cascade.y(), cascade.z()}, std::array<float, 3>{cascade.px(), cascade.py(), cascade.pz()}),
21022102
RecoDecay::cpaRZ(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{cascade.x(), cascade.y(), cascade.z()}, std::array<float, 3>{cascade.px(), cascade.py(), cascade.pz()}),
21032103
impactParameterCasc.getY(), impactParameterCasc.getZ(), impactParameterCasc.getSigmaY2(), impactParameterCasc.getSigmaYZ(), impactParameterCasc.getSigmaZ2(),
2104-
eCpair.mass, eCpair.ptSVL, eCpair.plSVL, eCpair.ptSVH, eCpair.plSVH,
2104+
eCpair.mass, eCpair.pt, eCpair.ptSVL, eCpair.plSVL, eCpair.ptSVH, eCpair.plSVH,
21052105
eCpair.ptFD, eCpair.plFD, eCpair.ptFDL, eCpair.plFDL, eCpair.ptFDH, eCpair.plFDH,
21062106
eCpair.dca2legs, eCpair.cospa, eCpair.cospaXY, eCpair.cospaRZ,
21072107
eCpair.lxy, eCpair.lz, eCpair.lxyz, eCpair.lxyErr, eCpair.lzErr, eCpair.lxyzErr,
@@ -2198,7 +2198,7 @@ struct taggingHFE {
21982198
// kaon.tpcNSigmaPi(), tofNSigmaPi,
21992199
kaon.tpcNSigmaKa(), tofNSigmaKa,
22002200
// kaon.tpcNSigmaPr(), tofNSigmaPr,
2201-
eKpair.mass, eKpair.ptSVL, eKpair.plSVL, eKpair.ptSVH, eKpair.plSVH,
2201+
eKpair.mass, eKpair.pt, eKpair.ptSVL, eKpair.plSVL, eKpair.ptSVH, eKpair.plSVH,
22022202
eKpair.ptFD, eKpair.plFD, eKpair.ptFDL, eKpair.plFDL, eKpair.ptFDH, eKpair.plFDH,
22032203
eKpair.dca2legs, eKpair.cospa, eKpair.cospaXY, eKpair.cospaRZ,
22042204
eKpair.lxy, eKpair.lz, eKpair.lxyz, eKpair.lxyErr, eKpair.lzErr, eKpair.lxyzErr,
@@ -2260,7 +2260,7 @@ struct taggingHFE {
22602260
// kaon.tpcNSigmaPi(), tofNSigmaPi,
22612261
kaon.tpcNSigmaKa(), tofNSigmaKa,
22622262
// kaon.tpcNSigmaPr(), tofNSigmaPr,
2263-
eKpair.mass, eKpair.ptSVL, eKpair.plSVL, eKpair.ptSVH, eKpair.plSVH,
2263+
eKpair.mass, eKpair.pt, eKpair.ptSVL, eKpair.plSVL, eKpair.ptSVH, eKpair.plSVH,
22642264
eKpair.ptFD, eKpair.plFD, eKpair.ptFDL, eKpair.plFDL, eKpair.ptFDH, eKpair.plFDH,
22652265
eKpair.dca2legs, eKpair.cospa, eKpair.cospaXY, eKpair.cospaRZ,
22662266
eKpair.lxy, eKpair.lz, eKpair.lxyz, eKpair.lxyErr, eKpair.lzErr, eKpair.lxyzErr,
@@ -2338,7 +2338,7 @@ struct taggingHFE {
23382338
RecoDecay::cpaXY(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{v0.x(), v0.y(), v0.z()}, std::array<float, 3>{v0.px(), v0.py(), v0.pz()}),
23392339
RecoDecay::cpaRZ(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{v0.x(), v0.y(), v0.z()}, std::array<float, 3>{v0.px(), v0.py(), v0.pz()}),
23402340
impactParameterV0.getY(), impactParameterV0.getZ(), impactParameterV0.getSigmaY2(), impactParameterV0.getSigmaYZ(), impactParameterV0.getSigmaZ2(),
2341-
eV0pair.mass, eV0pair.ptSVL, eV0pair.plSVL, eV0pair.ptSVH, eV0pair.plSVH,
2341+
eV0pair.mass, eV0pair.pt, eV0pair.ptSVL, eV0pair.plSVL, eV0pair.ptSVH, eV0pair.plSVH,
23422342
eV0pair.ptFD, eV0pair.plFD, eV0pair.ptFDL, eV0pair.plFDL, eV0pair.ptFDH, eV0pair.plFDH,
23432343
eV0pair.dca2legs, eV0pair.cospa, eV0pair.cospaXY, eV0pair.cospaRZ,
23442344
eV0pair.lxy, eV0pair.lz, eV0pair.lxyz, eV0pair.lxyErr, eV0pair.lzErr, eV0pair.lxyzErr,
@@ -2401,7 +2401,7 @@ struct taggingHFE {
24012401
RecoDecay::cpaXY(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{v0.x(), v0.y(), v0.z()}, std::array<float, 3>{v0.px(), v0.py(), v0.pz()}),
24022402
RecoDecay::cpaRZ(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{v0.x(), v0.y(), v0.z()}, std::array<float, 3>{v0.px(), v0.py(), v0.pz()}),
24032403
impactParameterV0.getY(), impactParameterV0.getZ(), impactParameterV0.getSigmaY2(), impactParameterV0.getSigmaYZ(), impactParameterV0.getSigmaZ2(),
2404-
eV0pair.mass, eV0pair.ptSVL, eV0pair.plSVL, eV0pair.ptSVH, eV0pair.plSVH,
2404+
eV0pair.mass, eV0pair.pt, eV0pair.ptSVL, eV0pair.plSVL, eV0pair.ptSVH, eV0pair.plSVH,
24052405
eV0pair.ptFD, eV0pair.plFD, eV0pair.ptFDL, eV0pair.plFDL, eV0pair.ptFDH, eV0pair.plFDH,
24062406
eV0pair.dca2legs, eV0pair.cospa, eV0pair.cospaXY, eV0pair.cospaRZ,
24072407
eV0pair.lxy, eV0pair.lz, eV0pair.lxyz, eV0pair.lxyErr, eV0pair.lzErr, eV0pair.lxyzErr,
@@ -2469,7 +2469,7 @@ struct taggingHFE {
24692469
RecoDecay::cpaXY(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{cascade.x(), cascade.y(), cascade.z()}, std::array<float, 3>{cascade.px(), cascade.py(), cascade.pz()}),
24702470
RecoDecay::cpaRZ(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{cascade.x(), cascade.y(), cascade.z()}, std::array<float, 3>{cascade.px(), cascade.py(), cascade.pz()}),
24712471
impactParameterCasc.getY(), impactParameterCasc.getZ(), impactParameterCasc.getSigmaY2(), impactParameterCasc.getSigmaYZ(), impactParameterCasc.getSigmaZ2(),
2472-
eCpair.mass, eCpair.ptSVL, eCpair.plSVL, eCpair.ptSVH, eCpair.plSVH,
2472+
eCpair.mass, eCpair.pt, eCpair.ptSVL, eCpair.plSVL, eCpair.ptSVH, eCpair.plSVH,
24732473
eCpair.ptFD, eCpair.plFD, eCpair.ptFDL, eCpair.plFDL, eCpair.ptFDH, eCpair.plFDH,
24742474
eCpair.dca2legs, eCpair.cospa, eCpair.cospaXY, eCpair.cospaRZ,
24752475
eCpair.lxy, eCpair.lz, eCpair.lxyz, eCpair.lxyErr, eCpair.lzErr, eCpair.lxyzErr,
@@ -2537,7 +2537,7 @@ struct taggingHFE {
25372537
RecoDecay::cpaXY(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{cascade.x(), cascade.y(), cascade.z()}, std::array<float, 3>{cascade.px(), cascade.py(), cascade.pz()}),
25382538
RecoDecay::cpaRZ(std::array<float, 3>{collision.posX(), collision.posY(), collision.posZ()}, std::array<float, 3>{cascade.x(), cascade.y(), cascade.z()}, std::array<float, 3>{cascade.px(), cascade.py(), cascade.pz()}),
25392539
impactParameterCasc.getY(), impactParameterCasc.getZ(), impactParameterCasc.getSigmaY2(), impactParameterCasc.getSigmaYZ(), impactParameterCasc.getSigmaZ2(),
2540-
eCpair.mass, eCpair.ptSVL, eCpair.plSVL, eCpair.ptSVH, eCpair.plSVH,
2540+
eCpair.mass, eCpair.pt, eCpair.ptSVL, eCpair.plSVL, eCpair.ptSVH, eCpair.plSVH,
25412541
eCpair.ptFD, eCpair.plFD, eCpair.ptFDL, eCpair.plFDL, eCpair.ptFDH, eCpair.plFDH,
25422542
eCpair.dca2legs, eCpair.cospa, eCpair.cospaXY, eCpair.cospaRZ,
25432543
eCpair.lxy, eCpair.lz, eCpair.lxyz, eCpair.lxyErr, eCpair.lzErr, eCpair.lxyzErr,

‎PWGEM/Dilepton/Utils/SemiCharmTag.h‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ namespace o2::aod::pwgem::dilepton::utils
3535

3636
struct LHPair { // struct to store electron-hadron pair information
3737
float mass{-999.f};
38+
float pt{-999.f};
3839
float dca2legs{-999.f};
3940
float cospa{-999.f};
4041
float cospaXY{-999.f};
@@ -167,6 +168,7 @@ LHPair makePairLeptonTrack(TFitter& fitter, TCollision const& collision, TLepton
167168
ROOT::Math::PxPyPzMVector v2(pvec1[0], pvec1[1], pvec1[2], o2::constants::physics::MassKaonCharged);
168169
ROOT::Math::PxPyPzMVector v12 = v1 + v2;
169170
pair.mass = v12.M();
171+
pair.pt = v12.Pt();
170172

171173
// float tmp = v12.P() * std::sin(std::acos(pair.cospa));
172174
// LOGF(info, "pair.ptFD = %f, tmp = %f", pair.ptFD, tmp);
@@ -293,6 +295,7 @@ LHPair makePairLeptonV0(TFitter& fitter, TCollision const& collision, TLepton co
293295

294296
ROOT::Math::PxPyPzMVector v12 = v1 + v2;
295297
pair.mass = v12.M();
298+
pair.pt = v12.Pt();
296299
pair.isOK = true;
297300

298301
return pair;
@@ -420,6 +423,7 @@ LHPair makePairLeptonCascade(TFitter& fitter, TCollision const& collision, TLept
420423

421424
ROOT::Math::PxPyPzMVector v12 = v1 + v2;
422425
pair.mass = v12.M();
426+
pair.pt = v12.Pt();
423427
pair.isOK = true;
424428

425429
return pair;

0 commit comments

Comments
 (0)