@@ -216,6 +216,8 @@ DECLARE_SOA_COLUMN(CascChi2OverNdf, cascChi2OverNdf, float);
216216DECLARE_SOA_COLUMN (XicChi2OverNdf, xicChi2OverNdf, float );
217217DECLARE_SOA_COLUMN (MassV0Chi2OverNdf, massV0Chi2OverNdf, float );
218218DECLARE_SOA_COLUMN (MassCascChi2OverNdf, massCascChi2OverNdf, float );
219+ // MC
220+ DECLARE_SOA_COLUMN (PdgFromMc, pdgFromMc, int );
219221} // namespace full
220222
221223DECLARE_SOA_TABLE (HfToXiPiEvs, " AOD" , " HFTOXIPIEV" ,
@@ -308,12 +310,14 @@ DECLARE_SOA_TABLE(HfKfXicFulls, "AOD", "HFKFXICFULL",
308310 full::FlagMcMatchRec, full::DebugMcRec, full::OriginRec, full::CollisionMatched);
309311
310312DECLARE_SOA_TABLE (HfCandToXiPiGen, " AOD" , " HFCANDTOXIPIGEN" ,
313+ full::InvMassCharmBaryon,
311314 full::PtCharmBaryon,
312315 full::EtaCharmBaryon,
313316 full::PhiCharmBaryon,
314317 full::YCharmBaryon,
315318 full::FlagMcMatchRec,
316- full::OriginRec)
319+ full::OriginRec,
320+ full::PdgFromMc)
317321} // namespace o2::aod
318322
319323// / Writes the full information in an output TTree
@@ -821,12 +825,14 @@ struct HfTreeCreatorToXiPiQa {
821825 if (fillParticle) {
822826 rowCandidateParticles.reserve (mcParticles.size ());
823827 for (const auto & particle : mcParticles) {
824- rowCandidateParticles (particle.pt (),
828+ rowCandidateParticles (RecoDecay::m (std::array{particle.px (), particle.py (), particle.pz ()}, particle.e ()),
829+ particle.pt (),
825830 particle.eta (),
826831 particle.phi (),
827832 RecoDecay::y (particle.pVector (), o2::constants::physics::MassXiC0),
828833 particle.flagMcMatchGen (),
829- particle.originMcGen ());
834+ particle.originMcGen (),
835+ particle.pdgCode ());
830836 }
831837 }
832838 }
@@ -852,12 +858,14 @@ struct HfTreeCreatorToXiPiQa {
852858 if (fillParticle) {
853859 rowCandidateParticles.reserve (mcParticles.size ());
854860 for (const auto & particle : mcParticles) {
855- rowCandidateParticles (particle.pt (),
861+ rowCandidateParticles (RecoDecay::m (std::array{particle.px (), particle.py (), particle.pz ()}, particle.e ()),
862+ particle.pt (),
856863 particle.eta (),
857864 particle.phi (),
858- RecoDecay::y (particle.pVector (), o2::constants::physics::MassXiC0 ),
865+ RecoDecay::y (particle.pVector (), o2::constants::physics::MassOmegaC0 ),
859866 particle.flagMcMatchGen (),
860- particle.originMcGen ());
867+ particle.originMcGen (),
868+ particle.pdgCode ());
861869 }
862870 }
863871 }
@@ -883,12 +891,14 @@ struct HfTreeCreatorToXiPiQa {
883891 if (fillParticle) {
884892 rowCandidateParticles.reserve (mcParticles.size ());
885893 for (const auto & particle : mcParticles) {
886- rowCandidateParticles (particle.pt (),
894+ rowCandidateParticles (RecoDecay::m (std::array{particle.px (), particle.py (), particle.pz ()}, particle.e ()),
895+ particle.pt (),
887896 particle.eta (),
888897 particle.phi (),
889898 RecoDecay::y (particle.pVector (), o2::constants::physics::MassXiC0),
890899 particle.flagMcMatchGen (),
891- particle.originMcGen ());
900+ particle.originMcGen (),
901+ particle.pdgCode ());
892902 }
893903 }
894904 }
@@ -914,12 +924,14 @@ struct HfTreeCreatorToXiPiQa {
914924 if (fillParticle) {
915925 rowCandidateParticles.reserve (mcParticles.size ());
916926 for (const auto & particle : mcParticles) {
917- rowCandidateParticles (particle.pt (),
927+ rowCandidateParticles (RecoDecay::m (std::array{particle.px (), particle.py (), particle.pz ()}, particle.e ()),
928+ particle.pt (),
918929 particle.eta (),
919930 particle.phi (),
920931 RecoDecay::y (particle.pVector (), o2::constants::physics::MassXiC0),
921932 particle.flagMcMatchGen (),
922- particle.originMcGen ());
933+ particle.originMcGen (),
934+ particle.pdgCode ());
923935 }
924936 }
925937 }
@@ -945,12 +957,14 @@ struct HfTreeCreatorToXiPiQa {
945957 if (fillParticle) {
946958 rowCandidateParticles.reserve (mcParticles.size ());
947959 for (const auto & particle : mcParticles) {
948- rowCandidateParticles (particle.pt (),
960+ rowCandidateParticles (RecoDecay::m (std::array{particle.px (), particle.py (), particle.pz ()}, particle.e ()),
961+ particle.pt (),
949962 particle.eta (),
950963 particle.phi (),
951964 RecoDecay::y (particle.pVector (), o2::constants::physics::MassXiC0),
952965 particle.flagMcMatchGen (),
953- particle.originMcGen ());
966+ particle.originMcGen (),
967+ particle.pdgCode ());
954968 }
955969 }
956970 }
@@ -976,12 +990,14 @@ struct HfTreeCreatorToXiPiQa {
976990 if (fillParticle) {
977991 rowCandidateParticles.reserve (mcParticles.size ());
978992 for (const auto & particle : mcParticles) {
979- rowCandidateParticles (particle.pt (),
993+ rowCandidateParticles (RecoDecay::m (std::array{particle.px (), particle.py (), particle.pz ()}, particle.e ()),
994+ particle.pt (),
980995 particle.eta (),
981996 particle.phi (),
982997 RecoDecay::y (particle.pVector (), o2::constants::physics::MassXiC0),
983998 particle.flagMcMatchGen (),
984- particle.originMcGen ());
999+ particle.originMcGen (),
1000+ particle.pdgCode ());
9851001 }
9861002 }
9871003 }
@@ -1007,12 +1023,14 @@ struct HfTreeCreatorToXiPiQa {
10071023 if (fillParticle) {
10081024 rowCandidateParticles.reserve (mcParticles.size ());
10091025 for (const auto & particle : mcParticles) {
1010- rowCandidateParticles (particle.pt (),
1026+ rowCandidateParticles (RecoDecay::m (std::array{particle.px (), particle.py (), particle.pz ()}, particle.e ()),
1027+ particle.pt (),
10111028 particle.eta (),
10121029 particle.phi (),
1013- RecoDecay::y (particle.pVector (), o2::constants::physics::MassXiC0 ),
1030+ RecoDecay::y (particle.pVector (), o2::constants::physics::MassOmegaC0 ),
10141031 particle.flagMcMatchGen (),
1015- particle.originMcGen ());
1032+ particle.originMcGen (),
1033+ particle.pdgCode ());
10161034 }
10171035 }
10181036 }
@@ -1049,12 +1067,14 @@ struct HfTreeCreatorToXiPiQa {
10491067 if (fillParticle) {
10501068 rowCandidateParticles.reserve (mcParticles.size ());
10511069 for (const auto & particle : mcParticles) {
1052- rowCandidateParticles (particle.pt (),
1070+ rowCandidateParticles (RecoDecay::m (std::array{particle.px (), particle.py (), particle.pz ()}, particle.e ()),
1071+ particle.pt (),
10531072 particle.eta (),
10541073 particle.phi (),
10551074 RecoDecay::y (particle.pVector (), o2::constants::physics::MassXiC0),
10561075 particle.flagMcMatchGen (),
1057- particle.originMcGen ());
1076+ particle.originMcGen (),
1077+ particle.pdgCode ());
10581078 }
10591079 }
10601080 }
@@ -1080,12 +1100,14 @@ struct HfTreeCreatorToXiPiQa {
10801100 if (fillParticle) {
10811101 rowCandidateParticles.reserve (mcParticles.size ());
10821102 for (const auto & particle : mcParticles) {
1083- rowCandidateParticles (particle.pt (),
1103+ rowCandidateParticles (RecoDecay::m (std::array{particle.px (), particle.py (), particle.pz ()}, particle.e ()),
1104+ particle.pt (),
10841105 particle.eta (),
10851106 particle.phi (),
10861107 RecoDecay::y (particle.pVector (), o2::constants::physics::MassXiC0),
10871108 particle.flagMcMatchGen (),
1088- particle.originMcGen ());
1109+ particle.originMcGen (),
1110+ particle.pdgCode ());
10891111 }
10901112 }
10911113 }
@@ -1111,12 +1133,14 @@ struct HfTreeCreatorToXiPiQa {
11111133 if (fillParticle) {
11121134 rowCandidateParticles.reserve (mcParticles.size ());
11131135 for (const auto & particle : mcParticles) {
1114- rowCandidateParticles (particle.pt (),
1136+ rowCandidateParticles (RecoDecay::m (std::array{particle.px (), particle.py (), particle.pz ()}, particle.e ()),
1137+ particle.pt (),
11151138 particle.eta (),
11161139 particle.phi (),
11171140 RecoDecay::y (particle.pVector (), o2::constants::physics::MassXiC0),
11181141 particle.flagMcMatchGen (),
1119- particle.originMcGen ());
1142+ particle.originMcGen (),
1143+ particle.pdgCode ());
11201144 }
11211145 }
11221146 }
@@ -1142,12 +1166,14 @@ struct HfTreeCreatorToXiPiQa {
11421166 if (fillParticle) {
11431167 rowCandidateParticles.reserve (mcParticles.size ());
11441168 for (const auto & particle : mcParticles) {
1145- rowCandidateParticles (particle.pt (),
1169+ rowCandidateParticles (RecoDecay::m (std::array{particle.px (), particle.py (), particle.pz ()}, particle.e ()),
1170+ particle.pt (),
11461171 particle.eta (),
11471172 particle.phi (),
11481173 RecoDecay::y (particle.pVector (), o2::constants::physics::MassXiC0),
11491174 particle.flagMcMatchGen (),
1150- particle.originMcGen ());
1175+ particle.originMcGen (),
1176+ particle.pdgCode ());
11511177 }
11521178 }
11531179 }
0 commit comments