Skip to content

Commit cf4965b

Browse files
ananthapadmanabhan18ananthapadmanabhan18
committed
Even more Tables!
1 parent 59c4bc8 commit cf4965b

File tree

1 file changed

+49
-14
lines changed

1 file changed

+49
-14
lines changed

‎PWGUD/Tasks/exclusiveRhoTo4Pi.cxx‎

Lines changed: 49 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
#include "Math/Vector4D.h"
3333
#include "Math/Vector3D.h"
3434
#include "Math/GenVector/Boost.h"
35+
#include "CommonConstants/PhysicsConstants.h"
3536

3637
using namespace std;
3738
using namespace o2;
@@ -43,32 +44,49 @@ namespace o2::aod
4344
{
4445
namespace branch
4546
{
47+
DECLARE_SOA_COLUMN(Fv0signal, fv0signal, double);
48+
DECLARE_SOA_COLUMN(Ft0asignal, ft0asignal, double);
49+
DECLARE_SOA_COLUMN(Ft0csignal, ft0csignal, double);
50+
DECLARE_SOA_COLUMN(Fddasignal, fddasignal, double);
51+
DECLARE_SOA_COLUMN(Fddcsignal, fddcsignal, double);
52+
4653
DECLARE_SOA_COLUMN(Dcaxy, dcaxy, std::vector<double>);
4754
DECLARE_SOA_COLUMN(Dcaz, dcaz, std::vector<double>);
55+
4856
DECLARE_SOA_COLUMN(TpcNsigmaKa, tpcNsigmaKa, std::vector<double>);
4957
DECLARE_SOA_COLUMN(TpcNsigmaPr, tpcNsigmaPr, std::vector<double>);
5058
DECLARE_SOA_COLUMN(TpcNsigmaMu, tpcNsigmaMu, std::vector<double>);
5159
DECLARE_SOA_COLUMN(TpcNsigmaEl, tpcNsigmaEl, std::vector<double>);
60+
5261
DECLARE_SOA_COLUMN(TofNsigmaKa, tofNsigmaKa, std::vector<double>);
5362
DECLARE_SOA_COLUMN(TofNsigmaPr, tofNsigmaPr, std::vector<double>);
5463
DECLARE_SOA_COLUMN(TofNsigmaMu, tofNsigmaMu, std::vector<double>);
5564
DECLARE_SOA_COLUMN(TofNsigmaEl, tofNsigmaEl, std::vector<double>);
65+
5666
DECLARE_SOA_COLUMN(TpcChi2, tpcChi2, std::vector<double>);
5767
DECLARE_SOA_COLUMN(TpcNClsFindable, tpcNClsFindable, std::vector<double>);
5868
DECLARE_SOA_COLUMN(ItsChi2, itsChi2, std::vector<double>);
69+
5970
DECLARE_SOA_COLUMN(PionPt, pionPt, std::vector<double>);
6071
DECLARE_SOA_COLUMN(PionEta, pionEta, std::vector<double>);
6172
DECLARE_SOA_COLUMN(PionRapidity, pionRapidity, std::vector<double>);
73+
6274
DECLARE_SOA_COLUMN(FourPionPt, fourPionPt, double);
6375
DECLARE_SOA_COLUMN(FourPionEta, fourPionEta, double);
6476
DECLARE_SOA_COLUMN(FourPionRapidity, fourPionRapidity, double);
6577
DECLARE_SOA_COLUMN(FourPionMass, fourPionMass, double);
78+
DECLARE_SOA_COLUMN(FourPionPhi, fourPionPhi, double);
6679
DECLARE_SOA_COLUMN(FourPionPhiPair1, fourPionPhiPair1, double);
6780
DECLARE_SOA_COLUMN(FourPionPhiPair2, fourPionPhiPair2, double);
6881
DECLARE_SOA_COLUMN(FourPionCosThetaPair1, fourPionCosThetaPair1, double);
6982
DECLARE_SOA_COLUMN(FourPionCosThetaPair2, fourPionCosThetaPair2, double);
7083
} // namespace branch
7184
DECLARE_SOA_TABLE(UDTree0c, "AOD", "UD0Charge",
85+
branch::Fv0signal,
86+
branch::Ft0asignal,
87+
branch::Ft0csignal,
88+
branch::Fddasignal,
89+
branch::Fddcsignal,
7290
branch::Dcaxy,
7391
branch::Dcaz,
7492
branch::TpcNsigmaKa,
@@ -89,12 +107,18 @@ DECLARE_SOA_TABLE(UDTree0c, "AOD", "UD0Charge",
89107
branch::FourPionEta,
90108
branch::FourPionRapidity,
91109
branch::FourPionMass,
110+
branch::FourPionPhi,
92111
branch::FourPionPhiPair1,
93112
branch::FourPionPhiPair2,
94113
branch::FourPionCosThetaPair1,
95114
branch::FourPionCosThetaPair2);
96115

97116
DECLARE_SOA_TABLE(UDTreen0c, "AOD", "UDn0Charge",
117+
branch::Fv0signal,
118+
branch::Ft0asignal,
119+
branch::Ft0csignal,
120+
branch::Fddasignal,
121+
branch::Fddcsignal,
98122
branch::Dcaxy,
99123
branch::Dcaz,
100124
branch::TpcNsigmaKa,
@@ -114,7 +138,8 @@ DECLARE_SOA_TABLE(UDTreen0c, "AOD", "UDn0Charge",
114138
branch::FourPionPt,
115139
branch::FourPionEta,
116140
branch::FourPionRapidity,
117-
branch::FourPionMass);
141+
branch::FourPionMass,
142+
branch::FourPionPhi);
118143

119144
DECLARE_SOA_TABLE(MCTree, "AOD", "MC0Charge",
120145
branch::PionPt,
@@ -130,6 +155,11 @@ DECLARE_SOA_TABLE(MCTree, "AOD", "MC0Charge",
130155
branch::FourPionCosThetaPair2);
131156

132157
DECLARE_SOA_TABLE(MCUDTree, "AOD", "UDMC0Charge",
158+
branch::Fv0signal,
159+
branch::Ft0asignal,
160+
branch::Ft0csignal,
161+
branch::Fddasignal,
162+
branch::Fddcsignal,
133163
branch::Dcaxy,
134164
branch::Dcaz,
135165
branch::TpcNsigmaKa,
@@ -150,14 +180,15 @@ DECLARE_SOA_TABLE(MCUDTree, "AOD", "UDMC0Charge",
150180
branch::FourPionEta,
151181
branch::FourPionRapidity,
152182
branch::FourPionMass,
183+
branch::FourPionPhi,
153184
branch::FourPionPhiPair1,
154185
branch::FourPionPhiPair2,
155186
branch::FourPionCosThetaPair1,
156187
branch::FourPionCosThetaPair2);
157188
} // namespace o2::aod
158189

159190
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
160-
struct exclusiveRhoTo4Pi { // o2-linter: disable=name/workflow-file,name/struct
191+
struct exclusiveRhoTo4Pi { // "o2-linter: disable=name/workflow-file" (unable), o2-linter: disable=name/struct (UNable)
161192
SGSelector sgSelector;
162193
Produces<aod::UDTree0c> zeroChargeEventsData;
163194
Produces<aod::UDTreen0c> nonzeroChargeEventsData;
@@ -773,12 +804,13 @@ struct exclusiveRhoTo4Pi { // o2-linter: disable=name/workflow-file,name/struct
773804
fourPiCosThetaPair2 = cosThetaCollinsSoperFrame(k14, k23, k1234);
774805

775806
zeroChargeEventsData(
807+
collision.totalFV0AmplitudeA(), collision.totalFT0AmplitudeA(), collision.totalFT0AmplitudeC(), collision.totalFDDAmplitudeA(), collision.totalFDDAmplitudeC(),
776808
pidcaXY, pidcaZ,
777809
tpcNsigKa, tpcNsigPr, tpcNsigEl, tpcNsigMu,
778810
tofNsigKa, tofNsigPr, tofNsigEl, tofNsigMu,
779811
tpcchi2, tpcNFindableCls, itschi2,
780812
piPt, piEta, piRapidity,
781-
p1234.Pt(), p1234.Eta(), p1234.Rapidity(), p1234.M(),
813+
p1234.Pt(), p1234.Eta(), p1234.Rapidity(), p1234.M(), p1234.Phi(),
782814
fourPiPhiPair1, fourPiPhiPair2, fourPiCosThetaPair1, fourPiCosThetaPair2);
783815

784816
if (std::fabs(p1234.Rapidity()) < 0.5) {
@@ -861,12 +893,13 @@ struct exclusiveRhoTo4Pi { // o2-linter: disable=name/workflow-file,name/struct
861893
}
862894

863895
nonzeroChargeEventsData(
896+
collision.totalFV0AmplitudeA(), collision.totalFT0AmplitudeA(), collision.totalFT0AmplitudeC(), collision.totalFDDAmplitudeA(), collision.totalFDDAmplitudeC(),
864897
pidcaXY, pidcaZ,
865898
tpcNsigKa, tpcNsigPr, tpcNsigMu, tpcNsigEl,
866899
tofNsigKa, tofNsigPr, tofNsigMu, tofNsigEl,
867900
tpcchi2, tpcNFindableCls, itschi2,
868901
piPt, piEta, piRapidity,
869-
p1234.Pt(), p1234.Eta(), p1234.Rapidity(), p1234.M());
902+
p1234.Pt(), p1234.Eta(), p1234.Rapidity(), p1234.M(), p1234.Phi());
870903

871904
if (std::fabs(p1234.Rapidity()) < 0.5) {
872905
histosData.fill(HIST("pT_event_non0charge_WTS_PID_Pi"), p1234.Pt());
@@ -913,23 +946,23 @@ struct exclusiveRhoTo4Pi { // o2-linter: disable=name/workflow-file,name/struct
913946
}
914947

915948
for (const auto& mother : particle.mothers_as<aod::UDMcParticles>()) {
916-
if (mother.pdgCode() == 30113) {
949+
if (mother.pdgCode() == 30313) { // "o2-linter: pdg/explicit-code" (unable)
917950
motherVector.SetXYZM(mother.px(), mother.py(), mother.pz(), o2::constants::physics::MassPionCharged);
918951
histosMCgen.fill(HIST("MCgen_rhoPrime_pT"), motherVector.Pt());
919952

920953
if (flag == false) {
921954
histosMCgen.fill(HIST("rhoPrimeCounts"), 5);
922955
}
923956
flag = true;
924-
if (particle.pdgCode() == 211) {
957+
if (particle.pdgCode() == 211) { // "o2-linter: pdg/explicit-code" (unable)
925958
histosMCgen.fill(HIST("MCgen_particle_pT"), tempVector.Pt());
926959
histosMCgen.fill(HIST("MCgen_particle_rapidity"), tempVector.Rapidity());
927960
piPlusvectors.push_back(tempVector);
928961
piPt.push_back(tempVector.Pt());
929962
piEta.push_back(tempVector.Eta());
930963
piRapidity.push_back(tempVector.Rapidity());
931964
}
932-
if (particle.pdgCode() == -211) {
965+
if (particle.pdgCode() == -211) { // "o2-linter: pdg/explicit-code" (unable)
933966
histosMCgen.fill(HIST("MCgen_particle_pT"), tempVector.Pt());
934967
histosMCgen.fill(HIST("MCgen_particle_rapidity"), tempVector.Rapidity());
935968
piMinusvectors.push_back(tempVector);
@@ -1270,13 +1303,15 @@ struct exclusiveRhoTo4Pi { // o2-linter: disable=name/workflow-file,name/struct
12701303
piRapidity.push_back(p3.Rapidity());
12711304
piRapidity.push_back(p4.Rapidity());
12721305

1273-
zeroChargeEventsMCreco(dcaxy, dcaz,
1274-
tpcNsigKa, tpcNsigPr, tpcNsigMu, tpcNsigEl,
1275-
tofNsigKa, tofNsigPr, tofNsigMu, tofNsigEl,
1276-
tpcchi2, tpcNFindableCls, itschi2,
1277-
piPt, piEta, piRapidity,
1278-
p1234.Pt(), p1234.Eta(), p1234.Rapidity(), p1234.M(),
1279-
phiPair1, phiPair2, cosThetaPair1, cosThetaPair2);
1306+
zeroChargeEventsMCreco(
1307+
collision.totalFV0AmplitudeA(), collision.totalFT0AmplitudeA(), collision.totalFT0AmplitudeC(), collision.totalFDDAmplitudeA(), collision.totalFDDAmplitudeC(),
1308+
dcaxy, dcaz,
1309+
tpcNsigKa, tpcNsigPr, tpcNsigMu, tpcNsigEl,
1310+
tofNsigKa, tofNsigPr, tofNsigMu, tofNsigEl,
1311+
tpcchi2, tpcNFindableCls, itschi2,
1312+
piPt, piEta, piRapidity,
1313+
p1234.Pt(), p1234.Eta(), p1234.Rapidity(), p1234.M(), p1234.Phi(),
1314+
phiPair1, phiPair2, cosThetaPair1, cosThetaPair2);
12801315

12811316
if (std::fabs(p1234.Rapidity()) < 0.5) {
12821317
histosMCreco.fill(HIST("pT_event_0charge_WTS_PID_Pi"), p1234.Pt());

0 commit comments

Comments
 (0)