Skip to content

Commit c0c6b4c

Browse files
ananthapadmanabhan18ananthapadmanabhan18
authored
[PWGUD] More Tables (#10432)
1 parent 9f0f4ca commit c0c6b4c

File tree

1 file changed

+211
-0
lines changed

1 file changed

+211
-0
lines changed

‎PWGUD/Tasks/exclusiveRhoTo4Pi.cxx‎

Lines changed: 211 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,17 @@ namespace branch
4545
{
4646
DECLARE_SOA_COLUMN(Dcaxy, dcaxy, std::vector<double>);
4747
DECLARE_SOA_COLUMN(Dcaz, dcaz, std::vector<double>);
48+
DECLARE_SOA_COLUMN(TpcNsigmaKa, tpcNsigmaKa, std::vector<double>);
49+
DECLARE_SOA_COLUMN(TpcNsigmaPr, tpcNsigmaPr, std::vector<double>);
50+
DECLARE_SOA_COLUMN(TpcNsigmaMu, tpcNsigmaMu, std::vector<double>);
51+
DECLARE_SOA_COLUMN(TpcNsigmaEl, tpcNsigmaEl, std::vector<double>);
52+
DECLARE_SOA_COLUMN(TofNsigmaKa, tofNsigmaKa, std::vector<double>);
53+
DECLARE_SOA_COLUMN(TofNsigmaPr, tofNsigmaPr, std::vector<double>);
54+
DECLARE_SOA_COLUMN(TofNsigmaMu, tofNsigmaMu, std::vector<double>);
55+
DECLARE_SOA_COLUMN(TofNsigmaEl, tofNsigmaEl, std::vector<double>);
56+
DECLARE_SOA_COLUMN(TpcChi2, tpcChi2, std::vector<double>);
57+
DECLARE_SOA_COLUMN(TpcNClsFindable, tpcNClsFindable, std::vector<double>);
58+
DECLARE_SOA_COLUMN(ItsChi2, itsChi2, std::vector<double>);
4859
DECLARE_SOA_COLUMN(PionPt, pionPt, std::vector<double>);
4960
DECLARE_SOA_COLUMN(PionEta, pionEta, std::vector<double>);
5061
DECLARE_SOA_COLUMN(PionRapidity, pionRapidity, std::vector<double>);
@@ -60,6 +71,17 @@ DECLARE_SOA_COLUMN(FourPionCosThetaPair2, fourPionCosThetaPair2, double);
6071
DECLARE_SOA_TABLE(UDTree0c, "AOD", "UD0Charge",
6172
branch::Dcaxy,
6273
branch::Dcaz,
74+
branch::TpcNsigmaKa,
75+
branch::TpcNsigmaPr,
76+
branch::TpcNsigmaMu,
77+
branch::TpcNsigmaEl,
78+
branch::TofNsigmaKa,
79+
branch::TofNsigmaPr,
80+
branch::TofNsigmaMu,
81+
branch::TofNsigmaEl,
82+
branch::TpcChi2,
83+
branch::TpcNClsFindable,
84+
branch::ItsChi2,
6385
branch::PionPt,
6486
branch::PionEta,
6587
branch::PionRapidity,
@@ -75,6 +97,17 @@ DECLARE_SOA_TABLE(UDTree0c, "AOD", "UD0Charge",
7597
DECLARE_SOA_TABLE(UDTreen0c, "AOD", "UDn0Charge",
7698
branch::Dcaxy,
7799
branch::Dcaz,
100+
branch::TpcNsigmaKa,
101+
branch::TpcNsigmaPr,
102+
branch::TpcNsigmaMu,
103+
branch::TpcNsigmaEl,
104+
branch::TofNsigmaKa,
105+
branch::TofNsigmaPr,
106+
branch::TofNsigmaMu,
107+
branch::TofNsigmaEl,
108+
branch::TpcChi2,
109+
branch::TpcNClsFindable,
110+
branch::ItsChi2,
78111
branch::PionPt,
79112
branch::PionEta,
80113
branch::PionRapidity,
@@ -99,6 +132,17 @@ DECLARE_SOA_TABLE(MCTree, "AOD", "MC0Charge",
99132
DECLARE_SOA_TABLE(MCUDTree, "AOD", "UDMC0Charge",
100133
branch::Dcaxy,
101134
branch::Dcaz,
135+
branch::TpcNsigmaKa,
136+
branch::TpcNsigmaPr,
137+
branch::TpcNsigmaMu,
138+
branch::TpcNsigmaEl,
139+
branch::TofNsigmaKa,
140+
branch::TofNsigmaPr,
141+
branch::TofNsigmaMu,
142+
branch::TofNsigmaEl,
143+
branch::TpcChi2,
144+
branch::TpcNClsFindable,
145+
branch::ItsChi2,
102146
branch::PionPt,
103147
branch::PionEta,
104148
branch::PionRapidity,
@@ -586,6 +630,20 @@ struct exclusiveRhoTo4Pi { // o2-linter: disable=name/workflow-file,name/struct
586630
std::vector<double> pidcaXY;
587631
std::vector<double> pidcaZ;
588632

633+
std::vector<double> tpcNsigKa;
634+
std::vector<double> tpcNsigPr;
635+
std::vector<double> tpcNsigEl;
636+
std::vector<double> tpcNsigMu;
637+
638+
std::vector<double> tofNsigKa;
639+
std::vector<double> tofNsigPr;
640+
std::vector<double> tofNsigEl;
641+
std::vector<double> tofNsigMu;
642+
643+
std::vector<double> tpcchi2;
644+
std::vector<double> itschi2;
645+
std::vector<double> tpcNFindableCls;
646+
589647
std::vector<double> piPt;
590648
std::vector<double> piEta;
591649
std::vector<double> piRapidity;
@@ -616,6 +674,61 @@ struct exclusiveRhoTo4Pi { // o2-linter: disable=name/workflow-file,name/struct
616674
pidcaZ.push_back(Pi_minus_tracks[0].dcaZ());
617675
pidcaZ.push_back(Pi_minus_tracks[1].dcaZ());
618676

677+
tpcNsigKa.push_back(Pi_plus_tracks[0].tpcNSigmaKa());
678+
tpcNsigKa.push_back(Pi_plus_tracks[1].tpcNSigmaKa());
679+
tpcNsigKa.push_back(Pi_minus_tracks[0].tpcNSigmaKa());
680+
tpcNsigKa.push_back(Pi_minus_tracks[1].tpcNSigmaKa());
681+
682+
tpcNsigPr.push_back(Pi_plus_tracks[0].tpcNSigmaPr());
683+
tpcNsigPr.push_back(Pi_plus_tracks[1].tpcNSigmaPr());
684+
tpcNsigPr.push_back(Pi_minus_tracks[0].tpcNSigmaPr());
685+
tpcNsigPr.push_back(Pi_minus_tracks[1].tpcNSigmaPr());
686+
687+
tpcNsigEl.push_back(Pi_plus_tracks[0].tpcNSigmaEl());
688+
tpcNsigEl.push_back(Pi_plus_tracks[1].tpcNSigmaEl());
689+
tpcNsigEl.push_back(Pi_minus_tracks[0].tpcNSigmaEl());
690+
tpcNsigEl.push_back(Pi_minus_tracks[1].tpcNSigmaEl());
691+
692+
tpcNsigMu.push_back(Pi_plus_tracks[0].tpcNSigmaMu());
693+
tpcNsigMu.push_back(Pi_plus_tracks[1].tpcNSigmaMu());
694+
tpcNsigMu.push_back(Pi_minus_tracks[0].tpcNSigmaMu());
695+
tpcNsigMu.push_back(Pi_minus_tracks[1].tpcNSigmaMu());
696+
697+
tofNsigKa.push_back(Pi_plus_tracks[0].tofNSigmaKa());
698+
tofNsigKa.push_back(Pi_plus_tracks[1].tofNSigmaKa());
699+
tofNsigKa.push_back(Pi_minus_tracks[0].tofNSigmaKa());
700+
tofNsigKa.push_back(Pi_minus_tracks[1].tofNSigmaKa());
701+
702+
tofNsigPr.push_back(Pi_plus_tracks[0].tofNSigmaPr());
703+
tofNsigPr.push_back(Pi_plus_tracks[1].tofNSigmaPr());
704+
tofNsigPr.push_back(Pi_minus_tracks[0].tofNSigmaPr());
705+
tofNsigPr.push_back(Pi_minus_tracks[1].tofNSigmaPr());
706+
707+
tofNsigEl.push_back(Pi_plus_tracks[0].tofNSigmaEl());
708+
tofNsigEl.push_back(Pi_plus_tracks[1].tofNSigmaEl());
709+
tofNsigEl.push_back(Pi_minus_tracks[0].tofNSigmaEl());
710+
tofNsigEl.push_back(Pi_minus_tracks[1].tofNSigmaEl());
711+
712+
tofNsigMu.push_back(Pi_plus_tracks[0].tofNSigmaMu());
713+
tofNsigMu.push_back(Pi_plus_tracks[1].tofNSigmaMu());
714+
tofNsigMu.push_back(Pi_minus_tracks[0].tofNSigmaMu());
715+
tofNsigMu.push_back(Pi_minus_tracks[1].tofNSigmaMu());
716+
717+
tpcchi2.push_back(Pi_plus_tracks[0].tpcChi2NCl());
718+
tpcchi2.push_back(Pi_plus_tracks[1].tpcChi2NCl());
719+
tpcchi2.push_back(Pi_minus_tracks[0].tpcChi2NCl());
720+
tpcchi2.push_back(Pi_minus_tracks[1].tpcChi2NCl());
721+
722+
tpcNFindableCls.push_back(Pi_plus_tracks[0].tpcNClsFindable());
723+
tpcNFindableCls.push_back(Pi_plus_tracks[1].tpcNClsFindable());
724+
tpcNFindableCls.push_back(Pi_minus_tracks[0].tpcNClsFindable());
725+
tpcNFindableCls.push_back(Pi_minus_tracks[1].tpcNClsFindable());
726+
727+
itschi2.push_back(Pi_plus_tracks[0].itsChi2NCl());
728+
itschi2.push_back(Pi_plus_tracks[1].itsChi2NCl());
729+
itschi2.push_back(Pi_minus_tracks[0].itsChi2NCl());
730+
itschi2.push_back(Pi_minus_tracks[1].itsChi2NCl());
731+
619732
piPt.push_back(p1.Pt());
620733
piPt.push_back(p2.Pt());
621734
piPt.push_back(p3.Pt());
@@ -661,6 +774,9 @@ struct exclusiveRhoTo4Pi { // o2-linter: disable=name/workflow-file,name/struct
661774

662775
zeroChargeEventsData(
663776
pidcaXY, pidcaZ,
777+
tpcNsigKa, tpcNsigPr, tpcNsigEl, tpcNsigMu,
778+
tofNsigKa, tofNsigPr, tofNsigEl, tofNsigMu,
779+
tpcchi2, tpcNFindableCls, itschi2,
664780
piPt, piEta, piRapidity,
665781
p1234.Pt(), p1234.Eta(), p1234.Rapidity(), p1234.M(),
666782
fourPiPhiPair1, fourPiPhiPair2, fourPiCosThetaPair1, fourPiCosThetaPair2);
@@ -701,6 +817,17 @@ struct exclusiveRhoTo4Pi { // o2-linter: disable=name/workflow-file,name/struct
701817

702818
std::vector<double> pidcaXY;
703819
std::vector<double> pidcaZ;
820+
std::vector<double> tpcNsigKa;
821+
std::vector<double> tpcNsigPr;
822+
std::vector<double> tpcNsigEl;
823+
std::vector<double> tpcNsigMu;
824+
std::vector<double> tofNsigKa;
825+
std::vector<double> tofNsigPr;
826+
std::vector<double> tofNsigEl;
827+
std::vector<double> tofNsigMu;
828+
std::vector<double> tpcchi2;
829+
std::vector<double> itschi2;
830+
std::vector<double> tpcNFindableCls;
704831
std::vector<double> piPt;
705832
std::vector<double> piEta;
706833
std::vector<double> piRapidity;
@@ -717,13 +844,27 @@ struct exclusiveRhoTo4Pi { // o2-linter: disable=name/workflow-file,name/struct
717844
tempVec.SetXYZM(WTS_PID_Pi_tracks[i].px(), WTS_PID_Pi_tracks[i].py(), WTS_PID_Pi_tracks[i].pz(), o2::constants::physics::MassPionCharged);
718845
pidcaXY.push_back(WTS_PID_Pi_tracks[i].dcaXY());
719846
pidcaZ.push_back(WTS_PID_Pi_tracks[i].dcaZ());
847+
tpcNsigKa.push_back(WTS_PID_Pi_tracks[i].tpcNSigmaKa());
848+
tpcNsigPr.push_back(WTS_PID_Pi_tracks[i].tpcNSigmaPr());
849+
tpcNsigEl.push_back(WTS_PID_Pi_tracks[i].tpcNSigmaEl());
850+
tpcNsigMu.push_back(WTS_PID_Pi_tracks[i].tpcNSigmaMu());
851+
tofNsigKa.push_back(WTS_PID_Pi_tracks[i].tofNSigmaKa());
852+
tofNsigPr.push_back(WTS_PID_Pi_tracks[i].tofNSigmaPr());
853+
tofNsigEl.push_back(WTS_PID_Pi_tracks[i].tofNSigmaEl());
854+
tofNsigMu.push_back(WTS_PID_Pi_tracks[i].tofNSigmaMu());
855+
tpcchi2.push_back(WTS_PID_Pi_tracks[i].tpcChi2NCl());
856+
itschi2.push_back(WTS_PID_Pi_tracks[i].itsChi2NCl());
857+
tpcNFindableCls.push_back(WTS_PID_Pi_tracks[i].tpcNClsFindable());
720858
piPt.push_back(tempVec.Pt());
721859
piEta.push_back(tempVec.Eta());
722860
piRapidity.push_back(tempVec.Rapidity());
723861
}
724862

725863
nonzeroChargeEventsData(
726864
pidcaXY, pidcaZ,
865+
tpcNsigKa, tpcNsigPr, tpcNsigMu, tpcNsigEl,
866+
tofNsigKa, tofNsigPr, tofNsigMu, tofNsigEl,
867+
tpcchi2, tpcNFindableCls, itschi2,
727868
piPt, piEta, piRapidity,
728869
p1234.Pt(), p1234.Eta(), p1234.Rapidity(), p1234.M());
729870

@@ -1033,6 +1174,18 @@ struct exclusiveRhoTo4Pi { // o2-linter: disable=name/workflow-file,name/struct
10331174
std::vector<double> dcaxy;
10341175
std::vector<double> dcaz;
10351176

1177+
std::vector<double> tpcNsigKa;
1178+
std::vector<double> tpcNsigPr;
1179+
std::vector<double> tpcNsigEl;
1180+
std::vector<double> tpcNsigMu;
1181+
std::vector<double> tofNsigKa;
1182+
std::vector<double> tofNsigPr;
1183+
std::vector<double> tofNsigEl;
1184+
std::vector<double> tofNsigMu;
1185+
std::vector<double> tpcchi2;
1186+
std::vector<double> itschi2;
1187+
std::vector<double> tpcNFindableCls;
1188+
10361189
std::vector<double> piPt;
10371190
std::vector<double> piEta;
10381191
std::vector<double> piRapidity;
@@ -1047,6 +1200,61 @@ struct exclusiveRhoTo4Pi { // o2-linter: disable=name/workflow-file,name/struct
10471200
dcaz.push_back(Pi_minus_tracks[0].dcaZ());
10481201
dcaz.push_back(Pi_minus_tracks[1].dcaZ());
10491202

1203+
tpcNsigKa.push_back(Pi_plus_tracks[0].tpcNSigmaKa());
1204+
tpcNsigKa.push_back(Pi_plus_tracks[1].tpcNSigmaKa());
1205+
tpcNsigKa.push_back(Pi_minus_tracks[0].tpcNSigmaKa());
1206+
tpcNsigKa.push_back(Pi_minus_tracks[1].tpcNSigmaKa());
1207+
1208+
tpcNsigPr.push_back(Pi_plus_tracks[0].tpcNSigmaPr());
1209+
tpcNsigPr.push_back(Pi_plus_tracks[1].tpcNSigmaPr());
1210+
tpcNsigPr.push_back(Pi_minus_tracks[0].tpcNSigmaPr());
1211+
tpcNsigPr.push_back(Pi_minus_tracks[1].tpcNSigmaPr());
1212+
1213+
tpcNsigEl.push_back(Pi_plus_tracks[0].tpcNSigmaEl());
1214+
tpcNsigEl.push_back(Pi_plus_tracks[1].tpcNSigmaEl());
1215+
tpcNsigEl.push_back(Pi_minus_tracks[0].tpcNSigmaEl());
1216+
tpcNsigEl.push_back(Pi_minus_tracks[1].tpcNSigmaEl());
1217+
1218+
tpcNsigMu.push_back(Pi_plus_tracks[0].tpcNSigmaMu());
1219+
tpcNsigMu.push_back(Pi_plus_tracks[1].tpcNSigmaMu());
1220+
tpcNsigMu.push_back(Pi_minus_tracks[0].tpcNSigmaMu());
1221+
tpcNsigMu.push_back(Pi_minus_tracks[1].tpcNSigmaMu());
1222+
1223+
tofNsigKa.push_back(Pi_plus_tracks[0].tofNSigmaKa());
1224+
tofNsigKa.push_back(Pi_plus_tracks[1].tofNSigmaKa());
1225+
tofNsigKa.push_back(Pi_minus_tracks[0].tofNSigmaKa());
1226+
tofNsigKa.push_back(Pi_minus_tracks[1].tofNSigmaKa());
1227+
1228+
tofNsigPr.push_back(Pi_plus_tracks[0].tofNSigmaPr());
1229+
tofNsigPr.push_back(Pi_plus_tracks[1].tofNSigmaPr());
1230+
tofNsigPr.push_back(Pi_minus_tracks[0].tofNSigmaPr());
1231+
tofNsigPr.push_back(Pi_minus_tracks[1].tofNSigmaPr());
1232+
1233+
tofNsigEl.push_back(Pi_plus_tracks[0].tofNSigmaEl());
1234+
tofNsigEl.push_back(Pi_plus_tracks[1].tofNSigmaEl());
1235+
tofNsigEl.push_back(Pi_minus_tracks[0].tofNSigmaEl());
1236+
tofNsigEl.push_back(Pi_minus_tracks[1].tofNSigmaEl());
1237+
1238+
tofNsigMu.push_back(Pi_plus_tracks[0].tofNSigmaMu());
1239+
tofNsigMu.push_back(Pi_plus_tracks[1].tofNSigmaMu());
1240+
tofNsigMu.push_back(Pi_minus_tracks[0].tofNSigmaMu());
1241+
tofNsigMu.push_back(Pi_minus_tracks[1].tofNSigmaMu());
1242+
1243+
tpcchi2.push_back(Pi_plus_tracks[0].tpcChi2NCl());
1244+
tpcchi2.push_back(Pi_plus_tracks[1].tpcChi2NCl());
1245+
tpcchi2.push_back(Pi_minus_tracks[0].tpcChi2NCl());
1246+
tpcchi2.push_back(Pi_minus_tracks[1].tpcChi2NCl());
1247+
1248+
tpcNFindableCls.push_back(Pi_plus_tracks[0].tpcNClsFindable());
1249+
tpcNFindableCls.push_back(Pi_plus_tracks[1].tpcNClsFindable());
1250+
tpcNFindableCls.push_back(Pi_minus_tracks[0].tpcNClsFindable());
1251+
tpcNFindableCls.push_back(Pi_minus_tracks[1].tpcNClsFindable());
1252+
1253+
itschi2.push_back(Pi_plus_tracks[0].itsChi2NCl());
1254+
itschi2.push_back(Pi_plus_tracks[1].itsChi2NCl());
1255+
itschi2.push_back(Pi_minus_tracks[0].itsChi2NCl());
1256+
itschi2.push_back(Pi_minus_tracks[1].itsChi2NCl());
1257+
10501258
piPt.push_back(p1.Pt());
10511259
piPt.push_back(p2.Pt());
10521260
piPt.push_back(p3.Pt());
@@ -1063,6 +1271,9 @@ struct exclusiveRhoTo4Pi { // o2-linter: disable=name/workflow-file,name/struct
10631271
piRapidity.push_back(p4.Rapidity());
10641272

10651273
zeroChargeEventsMCreco(dcaxy, dcaz,
1274+
tpcNsigKa, tpcNsigPr, tpcNsigMu, tpcNsigEl,
1275+
tofNsigKa, tofNsigPr, tofNsigMu, tofNsigEl,
1276+
tpcchi2, tpcNFindableCls, itschi2,
10661277
piPt, piEta, piRapidity,
10671278
p1234.Pt(), p1234.Eta(), p1234.Rapidity(), p1234.M(),
10681279
phiPair1, phiPair2, cosThetaPair1, cosThetaPair2);

0 commit comments

Comments
 (0)