Skip to content

Commit bb7de41

Browse files
committed
[PWGEM/Dilepton] add dynamic columns in muon table
1 parent 5e4aa34 commit bb7de41

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

PWGEM/Dilepton/DataModel/dileptonTables.h

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1016,6 +1016,11 @@ DECLARE_SOA_DYNAMIC_COLUMN(P, p, [](float pt, float eta) -> float { return pt *
10161016
DECLARE_SOA_DYNAMIC_COLUMN(Px, px, [](float pt, float phi) -> float { return pt * std::cos(phi); });
10171017
DECLARE_SOA_DYNAMIC_COLUMN(Py, py, [](float pt, float phi) -> float { return pt * std::sin(phi); });
10181018
DECLARE_SOA_DYNAMIC_COLUMN(Pz, pz, [](float pt, float eta) -> float { return pt * std::sinh(eta); });
1019+
DECLARE_SOA_DYNAMIC_COLUMN(FwdDCAXY, fwdDcaXY, [](float dcaX, float dcaY) -> float { return std::hypot(dcaX, dcaY); });
1020+
DECLARE_SOA_DYNAMIC_COLUMN(Chi2IP, chi2IP, [](float dcaX, float dcaY, float cXX, float cXY, float cYY) -> float {
1021+
float det = cXX * cYY - cXY * cXY; // determinanat
1022+
return (det < 0.f) ? 1e+10 : (dcaX * dcaX * cYY + dcaY * dcaY * cXX - 2.f * dcaX * dcaY * cXY) / det;
1023+
});
10191024
DECLARE_SOA_DYNAMIC_COLUMN(NClustersMFT, nClustersMFT, //! Number of MFT clusters
10201025
[](uint64_t mftClusterSizesAndTrackFlags) -> uint8_t {
10211026
uint8_t nClusters = 0;
@@ -1152,7 +1157,9 @@ DECLARE_SOA_TABLE_VERSIONED(EMPrimaryMuons_004, "AOD", "EMPRIMARYMU", 4, //!
11521157
emprimarymuon::P<fwdtrack::Pt, fwdtrack::Eta>,
11531158
emprimarymuon::Px<fwdtrack::Pt, fwdtrack::Phi>,
11541159
emprimarymuon::Py<fwdtrack::Pt, fwdtrack::Phi>,
1155-
emprimarymuon::Pz<fwdtrack::Pt, fwdtrack::Eta>);
1160+
emprimarymuon::Pz<fwdtrack::Pt, fwdtrack::Eta>,
1161+
emprimarymuon::FwdDCAXY<fwdtrack::FwdDcaX, fwdtrack::FwdDcaY>,
1162+
emprimarymuon::Chi2IP<fwdtrack::FwdDcaX, fwdtrack::FwdDcaY, emprimarymuon::CXXatDCA, emprimarymuon::CXYatDCA, emprimarymuon::CYYatDCA>);
11561163

11571164
using EMPrimaryMuons = EMPrimaryMuons_004;
11581165
// iterators

0 commit comments

Comments
 (0)