Skip to content

Commit 2adaa18

Browse files
creetz16creetz16
committed
Add daughter x at IU to output tables
1 parent 5b21f91 commit 2adaa18

3 files changed

Lines changed: 15 additions & 0 deletions

File tree

‎PWGLF/DataModel/Vtx3BodyTables.h‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,9 @@ DECLARE_SOA_COLUMN(PzTrackPi, pzTrackPi, float); //! track1 pz at min
5959
DECLARE_SOA_COLUMN(PxTrackDe, pxTrackDe, float); //! track2 px at min
6060
DECLARE_SOA_COLUMN(PyTrackDe, pyTrackDe, float); //! track2 py at min
6161
DECLARE_SOA_COLUMN(PzTrackDe, pzTrackDe, float); //! track2 pz at min
62+
DECLARE_SOA_COLUMN(XTrackPr, xTrackPr, float); //! track0 x at IU
63+
DECLARE_SOA_COLUMN(XTrackPi, xTrackPi, float); //! track1 x at IU
64+
DECLARE_SOA_COLUMN(XTrackDe, xTrackDe, float); //! track2 x at IU
6265

6366
// DCAs to PV
6467
DECLARE_SOA_COLUMN(DCAXYTrackPrToPV, dcaXYtrackPrToPv, float); //! DCAXY of proton to PV (computed with KFParticle)
@@ -215,6 +218,7 @@ DECLARE_SOA_TABLE(Vtx3BodyDatas, "AOD", "VTX3BODYDATA", //!
215218
vtx3body::PxTrackPr, vtx3body::PyTrackPr, vtx3body::PzTrackPr,
216219
vtx3body::PxTrackPi, vtx3body::PyTrackPi, vtx3body::PzTrackPi,
217220
vtx3body::PxTrackDe, vtx3body::PyTrackDe, vtx3body::PzTrackDe,
221+
vtx3body::XTrackPr, vtx3body::XTrackPi, vtx3body::XTrackDe,
218222
vtx3body::DCAXYTrackPrToPV, vtx3body::DCAXYTrackPiToPV, vtx3body::DCAXYTrackDeToPV,
219223
vtx3body::DCATrackPrToPV, vtx3body::DCATrackPiToPV, vtx3body::DCATrackDeToPV,
220224
vtx3body::DCAXYTrackPrToPVProp, vtx3body::DCAXYTrackPiToPVProp, vtx3body::DCAXYTrackDeToPVProp,
@@ -266,6 +270,7 @@ DECLARE_SOA_TABLE(McVtx3BodyDatas, "AOD", "MC3BODYDATA", //!
266270
vtx3body::PxTrackPr, vtx3body::PyTrackPr, vtx3body::PzTrackPr,
267271
vtx3body::PxTrackPi, vtx3body::PyTrackPi, vtx3body::PzTrackPi,
268272
vtx3body::PxTrackDe, vtx3body::PyTrackDe, vtx3body::PzTrackDe,
273+
vtx3body::XTrackPr, vtx3body::XTrackPi, vtx3body::XTrackDe,
269274
vtx3body::DCAXYTrackPrToPV, vtx3body::DCAXYTrackPiToPV, vtx3body::DCAXYTrackDeToPV,
270275
vtx3body::DCATrackPrToPV, vtx3body::DCATrackPiToPV, vtx3body::DCATrackDeToPV,
271276
vtx3body::DCAXYTrackPrToPVProp, vtx3body::DCAXYTrackPiToPVProp, vtx3body::DCAXYTrackDeToPVProp,

‎PWGLF/TableProducer/Nuspex/decay3bodybuilder.cxx‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -937,6 +937,7 @@ struct decay3bodyBuilder {
937937
-1., -1., -1., // momProton
938938
-1., -1., -1., // momPion
939939
-1., -1., -1., // momDeuteron
940+
-1., -1., -1., // xProton, xPion, xDeuteron
940941
-1., -1., -1., // trackDCAxyToPV: 0 - proton, 1 - pion, 2 - deuteron
941942
-1., -1., -1., // trackDCAToPV: 0 - proton, 1 - pion, 2 - deuteron
942943
-1., -1., -1., // trackDCAxyToPVprop: 0 - proton, 1 - pion, 2 - deuteron
@@ -1098,6 +1099,7 @@ struct decay3bodyBuilder {
10981099
helper.decay3body.momProton[0], helper.decay3body.momProton[1], helper.decay3body.momProton[2],
10991100
helper.decay3body.momPion[0], helper.decay3body.momPion[1], helper.decay3body.momPion[2],
11001101
helper.decay3body.momDeuteron[0], helper.decay3body.momDeuteron[1], helper.decay3body.momDeuteron[2],
1102+
helper.decay3body.xProton, helper.decay3body.xPion, helper.decay3body.xDeuteron,
11011103
helper.decay3body.trackDCAxyToPV[0], helper.decay3body.trackDCAxyToPV[1], helper.decay3body.trackDCAxyToPV[2], // 0 - proton, 1 - pion, 2 - deuteron
11021104
helper.decay3body.trackDCAToPV[0], helper.decay3body.trackDCAToPV[1], helper.decay3body.trackDCAToPV[2], // 0 - proton, 1 - pion, 2 - deuteron
11031105
helper.decay3body.trackDCAxyToPVprop[0], helper.decay3body.trackDCAxyToPVprop[1], helper.decay3body.trackDCAxyToPVprop[2], // 0 - proton, 1 - pion, 2 - deuteron
@@ -1129,6 +1131,7 @@ struct decay3bodyBuilder {
11291131
helper.decay3body.momProton[0], helper.decay3body.momProton[1], helper.decay3body.momProton[2],
11301132
helper.decay3body.momPion[0], helper.decay3body.momPion[1], helper.decay3body.momPion[2],
11311133
helper.decay3body.momDeuteron[0], helper.decay3body.momDeuteron[1], helper.decay3body.momDeuteron[2],
1134+
helper.decay3body.xProton, helper.decay3body.xPion, helper.decay3body.xDeuteron,
11321135
helper.decay3body.trackDCAxyToPV[0], helper.decay3body.trackDCAxyToPV[1], helper.decay3body.trackDCAxyToPV[2], // 0 - proton, 1 - pion, 2 - deuteron
11331136
helper.decay3body.trackDCAToPV[0], helper.decay3body.trackDCAToPV[1], helper.decay3body.trackDCAToPV[2], // 0 - proton, 1 - pion, 2 - deuteron
11341137
helper.decay3body.trackDCAxyToPVprop[0], helper.decay3body.trackDCAxyToPVprop[1], helper.decay3body.trackDCAxyToPVprop[2], // 0 - proton, 1 - pion, 2 - deuteron

‎PWGLF/Utils/decay3bodyBuilderHelper.h‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,9 @@ struct decay3bodyCandidate {
6060
std::array<float, 3> posProton = {0.0f, 0.0f, 0.0f};
6161
std::array<float, 3> posPion = {0.0f, 0.0f, 0.0f};
6262
std::array<float, 3> posDeuteron = {0.0f, 0.0f, 0.0f};
63+
float xProton = 0.0f;
64+
float xPion = 0.0f;
65+
float xDeuteron = 0.0f;
6366
std::array<float, 3> trackDCAxyToPV = {0.0f, 0.0f, 0.0f}; // 0 - proton, 1 - pion, 2 - deuteron
6467
std::array<float, 3> trackDCAToPV = {0.0f, 0.0f, 0.0f}; // 0 - proton, 1 - pion, 2 - deuteron
6568
std::array<float, 3> trackDCAxyToPVprop = {0.0f, 0.0f, 0.0f}; // 0 - proton, 1 - pion, 2 - deuteron
@@ -212,6 +215,10 @@ class decay3bodyBuilderHelper
212215
decay3body.pionID = trackPion.globalIndex();
213216
decay3body.deuteronID = trackDeuteron.globalIndex();
214217

218+
decay3body.xProton = trackProton.x();
219+
decay3body.xPion = trackPion.x();
220+
decay3body.xDeuteron = trackDeuteron.x();
221+
215222
//_______________________________________________________________________
216223
// track selections
217224
if (useSelections) {

0 commit comments

Comments
 (0)