Skip to content

Commit c4ac023

Browse files
dsekihatdsekihat
committed
[PWGEM/Dilepton] add v0,cascade type in taggingHFE
1 parent 8cb1f41 commit c4ac023

2 files changed

Lines changed: 18 additions & 15 deletions

File tree

‎PWGEM/Dilepton/DataModel/lmeeMLTables.h‎

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -246,6 +246,9 @@ DECLARE_SOA_COLUMN(CascCPA, casccpa, float); //! cosPA of Cascade
246246
DECLARE_SOA_COLUMN(CascCPAXY, casccpaXY, float); //! cosPA of Cascade in XY plane
247247
DECLARE_SOA_COLUMN(CascCPARZ, casccpaRZ, float); //! cosPA of Cascade in RZ plane
248248

249+
DECLARE_SOA_COLUMN(V0Type, v0Type, uint8_t); //! v0 type, 0 = K0S, 1 = Lambda, 2 = AntiLambda
250+
DECLARE_SOA_COLUMN(CascadeType, cascadeType, uint8_t); //! cascade type, 0 = XiMunus, 1 = OmegaMinus
251+
249252
// LH pair variables
250253
DECLARE_SOA_COLUMN(MassLH, massLH, float); //! invariant mass of LH assuming pion
251254
DECLARE_SOA_COLUMN(PtLH, ptLH, float); //! pT of LH pair
@@ -287,7 +290,7 @@ DECLARE_SOA_TABLE(EMMLLHPairs, "AOD", "EMMLLHPAIR", //!
287290
using EMMLLHPair = EMMLLHPairs::iterator;
288291

289292
DECLARE_SOA_TABLE(EMMLLV0Pairs, "AOD", "EMMLLV0PAIR", //!
290-
emmllhpair::EMMLLeptonId,
293+
emmllhpair::EMMLLeptonId, emmllhpair::V0Type,
291294
emmllhpair::PtH, emmllhpair::RapidityV0,
292295
emmllhpair::V0CPA, emmllhpair::V0CPAXY, emmllhpair::V0CPARZ,
293296
emmllhpair::ImpParXYH, emmllhpair::ImpParZH, emmllhpair::ImpParCYYH, emmllhpair::ImpParCZYH, emmllhpair::ImpParCZZH,
@@ -299,8 +302,8 @@ DECLARE_SOA_TABLE(EMMLLV0Pairs, "AOD", "EMMLLV0PAIR", //!
299302
using EMMLLV0Pair = EMMLLV0Pairs::iterator;
300303

301304
DECLARE_SOA_TABLE(EMMLLCascPairs, "AOD", "EMMLLCAPAIR", //!
302-
emmllhpair::EMMLLeptonId,
303-
emmllhpair::PtH, emmllhpair::RapidityC,
305+
emmllhpair::EMMLLeptonId, emmllhpair::CascadeType,
306+
emmllhpair::Signed1PtH, emmllhpair::RapidityC,
304307
emmllhpair::CascCPA, emmllhpair::CascCPAXY, emmllhpair::CascCPARZ,
305308
emmllhpair::ImpParXYH, emmllhpair::ImpParZH, emmllhpair::ImpParCYYH, emmllhpair::ImpParCZYH, emmllhpair::ImpParCZZH,
306309
emmllhpair::MassLH, emmllhpair::PtLH, emmllhpair::DcaLH, emmllhpair::CPA, emmllhpair::CPAXY, emmllhpair::CPARZ,

‎PWGEM/Dilepton/Tasks/taggingHFE.cxx‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1835,7 +1835,7 @@ struct taggingHFE {
18351835
continue;
18361836
}
18371837

1838-
emmllv0pair(leptonTable.lastIndex(),
1838+
emmllv0pair(leptonTable.lastIndex(), 0,
18391839
v0.pt(), v0.rapidity(0),
18401840
RecoDecay::cpa(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()}),
18411841
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()}),
@@ -1896,7 +1896,7 @@ struct taggingHFE {
18961896
continue;
18971897
}
18981898

1899-
emmllv0pair(leptonTable.lastIndex(),
1899+
emmllv0pair(leptonTable.lastIndex(), 1,
19001900
v0.pt(), v0.rapidity(1),
19011901
RecoDecay::cpa(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()}),
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()}),
@@ -1962,8 +1962,8 @@ struct taggingHFE {
19621962
continue;
19631963
}
19641964

1965-
emmllcascpair(leptonTable.lastIndex(),
1966-
cascade.pt(), cascade.rapidity(0),
1965+
emmllcascpair(leptonTable.lastIndex(), 0,
1966+
cascade.sign() / cascade.pt(), cascade.rapidity(0),
19671967
RecoDecay::cpa(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()}),
19681968
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()}),
19691969
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()}),
@@ -2028,8 +2028,8 @@ struct taggingHFE {
20282028
continue;
20292029
}
20302030

2031-
emmllcascpair(leptonTable.lastIndex(),
2032-
cascade.pt(), cascade.rapidity(2),
2031+
emmllcascpair(leptonTable.lastIndex(), 1,
2032+
cascade.sign() / cascade.pt(), cascade.rapidity(2),
20332033
RecoDecay::cpa(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::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()}),
20352035
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()}),
@@ -2260,7 +2260,7 @@ struct taggingHFE {
22602260
continue;
22612261
}
22622262

2263-
emmllv0pair(leptonTable.lastIndex(),
2263+
emmllv0pair(leptonTable.lastIndex(), 0,
22642264
v0.pt(), v0.rapidity(0),
22652265
RecoDecay::cpa(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()}),
22662266
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()}),
@@ -2321,7 +2321,7 @@ struct taggingHFE {
23212321
continue;
23222322
}
23232323

2324-
emmllv0pair(leptonTable.lastIndex(),
2324+
emmllv0pair(leptonTable.lastIndex(), 2,
23252325
v0.pt(), v0.rapidity(1),
23262326
RecoDecay::cpa(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()}),
23272327
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()}),
@@ -2387,8 +2387,8 @@ struct taggingHFE {
23872387
continue;
23882388
}
23892389

2390-
emmllcascpair(leptonTable.lastIndex(),
2391-
cascade.pt(), cascade.rapidity(0),
2390+
emmllcascpair(leptonTable.lastIndex(), 0,
2391+
cascade.sign() / cascade.pt(), cascade.rapidity(0),
23922392
RecoDecay::cpa(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()}),
23932393
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()}),
23942394
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()}),
@@ -2453,8 +2453,8 @@ struct taggingHFE {
24532453
continue;
24542454
}
24552455

2456-
emmllcascpair(leptonTable.lastIndex(),
2457-
cascade.pt(), cascade.rapidity(2),
2456+
emmllcascpair(leptonTable.lastIndex(), 1,
2457+
cascade.sign() / cascade.pt(), cascade.rapidity(2),
24582458
RecoDecay::cpa(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()}),
24592459
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()}),
24602460
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()}),

0 commit comments

Comments
 (0)