Skip to content

Commit 92796d1

Browse files
rolavickalibuild
rolavick
and
alibuild
authored
[PWGUD] Small improvements in personal tasks + new table maker (#11728)
Co-authored-by: ALICE Action Bot <alibuild@cern.ch>
1 parent 3363702 commit 92796d1

File tree

7 files changed

+1079
-24
lines changed

7 files changed

+1079
-24
lines changed

‎PWGUD/Core/UPCTauCentralBarrelHelperRL.h‎

Lines changed: 23 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,12 @@
88
// In applying this license CERN does not waive the privileges and immunities
99
// granted to it by virtue of its status as an Intergovernmental Organization
1010
// or submit itself to any jurisdiction.
11-
///
12-
/// \brief
11+
12+
/// \file UPCTauCentralBarrelHelperRL.h
13+
/// \brief Personal helper file to analyze tau events from UPC collisions
1314
/// \author Roman Lavicka, roman.lavicka@cern.ch
1415
/// \since 27.10.2022
16+
///
1517

1618
#ifndef PWGUD_CORE_UPCTAUCENTRALBARRELHELPERRL_H_
1719
#define PWGUD_CORE_UPCTAUCENTRALBARRELHELPERRL_H_
@@ -156,6 +158,25 @@ int enumMyParticle(int valuePDG)
156158
}
157159
}
158160

161+
int trackPDGfromEnum(int trackEnum)
162+
// reads pdg value and returns particle number as in enumMyParticle
163+
{
164+
if (trackEnum == P_ELECTRON) {
165+
return 11;
166+
} else if (trackEnum == P_MUON) {
167+
return 13;
168+
} else if (trackEnum == P_PION) {
169+
return 211;
170+
} else if (trackEnum == P_KAON) {
171+
return 321;
172+
} else if (trackEnum == P_PROTON) {
173+
return 2212;
174+
} else {
175+
printDebugMessage("PDG value not found in enumMyParticle. Returning -1.");
176+
return -1.;
177+
}
178+
}
179+
159180
float pt(float px, float py)
160181
// Just a simple function to return pt
161182
{

‎PWGUD/DataModel/TwoTracksEventTables.h‎

Lines changed: 226 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,226 @@
1+
// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
2+
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
3+
// All rights not expressly granted are reserved.
4+
//
5+
// This software is distributed under the terms of the GNU General Public
6+
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
7+
//
8+
// In applying this license CERN does not waive the privileges and immunities
9+
// granted to it by virtue of its status as an Intergovernmental Organization
10+
// or submit itself to any jurisdiction.
11+
12+
///
13+
/// \file TwoTracksEventTables.h
14+
/// \author Roman Lavička
15+
/// \since 2025-06-20
16+
/// \brief A table to store information about events preselected to have exactly two tracks.
17+
/// \brief Good for UPC gammagamma (->elel,mumu,tautau) and gammalead (vector mesons)
18+
/// \brief If MC, careful with filling the mother
19+
///
20+
21+
#ifndef PWGUD_DATAMODEL_TWOTRACKSEVENTTABLES_H_
22+
#define PWGUD_DATAMODEL_TWOTRACKSEVENTTABLES_H_
23+
24+
#include "Framework/AnalysisDataModel.h"
25+
26+
namespace o2::aod
27+
{
28+
namespace two_tracks_tree
29+
{
30+
// event info
31+
DECLARE_SOA_COLUMN(RunNumber, runNumber, int32_t);
32+
DECLARE_SOA_COLUMN(Bc, bc, int);
33+
DECLARE_SOA_COLUMN(TotalTracks, totalTracks, int);
34+
DECLARE_SOA_COLUMN(NumContrib, numContrib, int);
35+
DECLARE_SOA_COLUMN(GlobalNonPVtracks, globalNonPVtracks, int);
36+
DECLARE_SOA_COLUMN(PosX, posX, float);
37+
DECLARE_SOA_COLUMN(PosY, posY, float);
38+
DECLARE_SOA_COLUMN(PosZ, posZ, float);
39+
DECLARE_SOA_COLUMN(RecoMode, recoMode, int);
40+
DECLARE_SOA_COLUMN(OccupancyInTime, occupancyInTime, int);
41+
DECLARE_SOA_COLUMN(HadronicRate, hadronicRate, double);
42+
DECLARE_SOA_COLUMN(Trs, trs, int);
43+
DECLARE_SOA_COLUMN(Trofs, trofs, int);
44+
DECLARE_SOA_COLUMN(Hmpr, hmpr, int);
45+
DECLARE_SOA_COLUMN(Tfb, tfb, int);
46+
DECLARE_SOA_COLUMN(ItsRofb, itsRofb, int);
47+
DECLARE_SOA_COLUMN(Sbp, sbp, int);
48+
DECLARE_SOA_COLUMN(ZvtxFT0vsPv, zvtxFT0vsPv, int);
49+
DECLARE_SOA_COLUMN(VtxITSTPC, vtxITSTPC, int);
50+
// FIT info
51+
DECLARE_SOA_COLUMN(TotalFT0AmplitudeA, totalFT0AmplitudeA, float);
52+
DECLARE_SOA_COLUMN(TotalFT0AmplitudeC, totalFT0AmplitudeC, float);
53+
DECLARE_SOA_COLUMN(TotalFV0AmplitudeA, totalFV0AmplitudeA, float);
54+
DECLARE_SOA_COLUMN(EnergyCommonZNA, energyCommonZNA, float);
55+
DECLARE_SOA_COLUMN(EnergyCommonZNC, energyCommonZNC, float);
56+
DECLARE_SOA_COLUMN(TimeFT0A, timeFT0A, float);
57+
DECLARE_SOA_COLUMN(TimeFT0C, timeFT0C, float);
58+
DECLARE_SOA_COLUMN(TimeFV0A, timeFV0A, float);
59+
DECLARE_SOA_COLUMN(TimeZNA, timeZNA, float);
60+
DECLARE_SOA_COLUMN(TimeZNC, timeZNC, float);
61+
// tracks
62+
DECLARE_SOA_COLUMN(TrkPx, trkPx, float[2]);
63+
DECLARE_SOA_COLUMN(TrkPy, trkPy, float[2]);
64+
DECLARE_SOA_COLUMN(TrkPz, trkPz, float[2]);
65+
DECLARE_SOA_COLUMN(TrkSign, trkSign, int[2]);
66+
DECLARE_SOA_COLUMN(TrkDCAxy, trkDCAxy, float[2]);
67+
DECLARE_SOA_COLUMN(TrkDCAz, trkDCAz, float[2]);
68+
DECLARE_SOA_COLUMN(TrkTimeRes, trkTimeRes, float[2]);
69+
DECLARE_SOA_COLUMN(Trk1ITSclusterSizes, trk1ITSclusterSizes, uint32_t);
70+
DECLARE_SOA_COLUMN(Trk2ITSclusterSizes, trk2ITSclusterSizes, uint32_t);
71+
DECLARE_SOA_COLUMN(TrkTPCsignal, trkTPCsignal, float[2]);
72+
DECLARE_SOA_COLUMN(TrkTPCnSigmaEl, trkTPCnSigmaEl, float[2]);
73+
DECLARE_SOA_COLUMN(TrkTPCnSigmaMu, trkTPCnSigmaMu, float[2]);
74+
DECLARE_SOA_COLUMN(TrkTPCnSigmaPi, trkTPCnSigmaPi, float[2]);
75+
DECLARE_SOA_COLUMN(TrkTPCnSigmaKa, trkTPCnSigmaKa, float[2]);
76+
DECLARE_SOA_COLUMN(TrkTPCnSigmaPr, trkTPCnSigmaPr, float[2]);
77+
DECLARE_SOA_COLUMN(TrkTPCinnerParam, trkTPCinnerParam, float[2]);
78+
DECLARE_SOA_COLUMN(TrkTOFsignal, trkTOFsignal, float[2]);
79+
DECLARE_SOA_COLUMN(TrkTOFnSigmaEl, trkTOFnSigmaEl, float[2]);
80+
DECLARE_SOA_COLUMN(TrkTOFnSigmaMu, trkTOFnSigmaMu, float[2]);
81+
DECLARE_SOA_COLUMN(TrkTOFnSigmaPi, trkTOFnSigmaPi, float[2]);
82+
DECLARE_SOA_COLUMN(TrkTOFnSigmaKa, trkTOFnSigmaKa, float[2]);
83+
DECLARE_SOA_COLUMN(TrkTOFnSigmaPr, trkTOFnSigmaPr, float[2]);
84+
DECLARE_SOA_COLUMN(TrkTOFexpMom, trkTOFexpMom, float[2]);
85+
// truth event
86+
DECLARE_SOA_COLUMN(TrueChannel, trueChannel, int);
87+
DECLARE_SOA_COLUMN(TrueHasRecoColl, trueHasRecoColl, bool);
88+
DECLARE_SOA_COLUMN(TruePosX, truePosX, float);
89+
DECLARE_SOA_COLUMN(TruePosY, truePosY, float);
90+
DECLARE_SOA_COLUMN(TruePosZ, truePosZ, float);
91+
// truth particles
92+
DECLARE_SOA_COLUMN(TrueMotherPx, trueMotherPx, float[2]);
93+
DECLARE_SOA_COLUMN(TrueMotherPy, trueMotherPy, float[2]);
94+
DECLARE_SOA_COLUMN(TrueMotherPz, trueMotherPz, float[2]);
95+
DECLARE_SOA_COLUMN(TrueDaugPx, trueDaugPx, float[2]);
96+
DECLARE_SOA_COLUMN(TrueDaugPy, trueDaugPy, float[2]);
97+
DECLARE_SOA_COLUMN(TrueDaugPz, trueDaugPz, float[2]);
98+
DECLARE_SOA_COLUMN(TrueDaugPdgCode, trueDaugPdgCode, int[2]);
99+
// additional info
100+
DECLARE_SOA_COLUMN(ProblematicEvent, problematicEvent, bool);
101+
102+
} // namespace two_tracks_tree
103+
DECLARE_SOA_TABLE(TwoTracks, "AOD", "TWOTRACK",
104+
two_tracks_tree::RunNumber,
105+
two_tracks_tree::Bc,
106+
two_tracks_tree::TotalTracks,
107+
two_tracks_tree::NumContrib,
108+
two_tracks_tree::GlobalNonPVtracks,
109+
two_tracks_tree::PosX,
110+
two_tracks_tree::PosY,
111+
two_tracks_tree::PosZ,
112+
two_tracks_tree::RecoMode,
113+
two_tracks_tree::OccupancyInTime,
114+
two_tracks_tree::HadronicRate,
115+
two_tracks_tree::Trs,
116+
two_tracks_tree::Trofs,
117+
two_tracks_tree::Hmpr,
118+
two_tracks_tree::Tfb,
119+
two_tracks_tree::ItsRofb,
120+
two_tracks_tree::Sbp,
121+
two_tracks_tree::ZvtxFT0vsPv,
122+
two_tracks_tree::VtxITSTPC,
123+
two_tracks_tree::TotalFT0AmplitudeA,
124+
two_tracks_tree::TotalFT0AmplitudeC,
125+
two_tracks_tree::TotalFV0AmplitudeA,
126+
two_tracks_tree::EnergyCommonZNA,
127+
two_tracks_tree::EnergyCommonZNC,
128+
two_tracks_tree::TimeFT0A,
129+
two_tracks_tree::TimeFT0C,
130+
two_tracks_tree::TimeFV0A,
131+
two_tracks_tree::TimeZNA,
132+
two_tracks_tree::TimeZNC,
133+
two_tracks_tree::TrkPx,
134+
two_tracks_tree::TrkPy,
135+
two_tracks_tree::TrkPz,
136+
two_tracks_tree::TrkSign,
137+
two_tracks_tree::TrkDCAxy,
138+
two_tracks_tree::TrkDCAz,
139+
two_tracks_tree::TrkTimeRes,
140+
two_tracks_tree::Trk1ITSclusterSizes,
141+
two_tracks_tree::Trk2ITSclusterSizes,
142+
two_tracks_tree::TrkTPCsignal,
143+
two_tracks_tree::TrkTPCnSigmaEl,
144+
two_tracks_tree::TrkTPCnSigmaMu,
145+
two_tracks_tree::TrkTPCnSigmaPi,
146+
two_tracks_tree::TrkTPCnSigmaKa,
147+
two_tracks_tree::TrkTPCnSigmaPr,
148+
two_tracks_tree::TrkTPCinnerParam,
149+
two_tracks_tree::TrkTOFsignal,
150+
two_tracks_tree::TrkTOFnSigmaEl,
151+
two_tracks_tree::TrkTOFnSigmaMu,
152+
two_tracks_tree::TrkTOFnSigmaPi,
153+
two_tracks_tree::TrkTOFnSigmaKa,
154+
two_tracks_tree::TrkTOFnSigmaPr,
155+
two_tracks_tree::TrkTOFexpMom);
156+
157+
DECLARE_SOA_TABLE(TrueTwoTracks, "AOD", "TRUETWOTRACK",
158+
two_tracks_tree::RunNumber,
159+
two_tracks_tree::Bc,
160+
two_tracks_tree::TotalTracks,
161+
two_tracks_tree::NumContrib,
162+
two_tracks_tree::GlobalNonPVtracks,
163+
two_tracks_tree::PosX,
164+
two_tracks_tree::PosY,
165+
two_tracks_tree::PosZ,
166+
two_tracks_tree::RecoMode,
167+
two_tracks_tree::OccupancyInTime,
168+
two_tracks_tree::HadronicRate,
169+
two_tracks_tree::Trs,
170+
two_tracks_tree::Trofs,
171+
two_tracks_tree::Hmpr,
172+
two_tracks_tree::Tfb,
173+
two_tracks_tree::ItsRofb,
174+
two_tracks_tree::Sbp,
175+
two_tracks_tree::ZvtxFT0vsPv,
176+
two_tracks_tree::VtxITSTPC,
177+
two_tracks_tree::TotalFT0AmplitudeA,
178+
two_tracks_tree::TotalFT0AmplitudeC,
179+
two_tracks_tree::TotalFV0AmplitudeA,
180+
two_tracks_tree::EnergyCommonZNA,
181+
two_tracks_tree::EnergyCommonZNC,
182+
two_tracks_tree::TimeFT0A,
183+
two_tracks_tree::TimeFT0C,
184+
two_tracks_tree::TimeFV0A,
185+
two_tracks_tree::TimeZNA,
186+
two_tracks_tree::TimeZNC,
187+
two_tracks_tree::TrkPx,
188+
two_tracks_tree::TrkPy,
189+
two_tracks_tree::TrkPz,
190+
two_tracks_tree::TrkSign,
191+
two_tracks_tree::TrkDCAxy,
192+
two_tracks_tree::TrkDCAz,
193+
two_tracks_tree::TrkTimeRes,
194+
two_tracks_tree::Trk1ITSclusterSizes,
195+
two_tracks_tree::Trk2ITSclusterSizes,
196+
two_tracks_tree::TrkTPCsignal,
197+
two_tracks_tree::TrkTPCnSigmaEl,
198+
two_tracks_tree::TrkTPCnSigmaMu,
199+
two_tracks_tree::TrkTPCnSigmaPi,
200+
two_tracks_tree::TrkTPCnSigmaKa,
201+
two_tracks_tree::TrkTPCnSigmaPr,
202+
two_tracks_tree::TrkTPCinnerParam,
203+
two_tracks_tree::TrkTOFsignal,
204+
two_tracks_tree::TrkTOFnSigmaEl,
205+
two_tracks_tree::TrkTOFnSigmaMu,
206+
two_tracks_tree::TrkTOFnSigmaPi,
207+
two_tracks_tree::TrkTOFnSigmaKa,
208+
two_tracks_tree::TrkTOFnSigmaPr,
209+
two_tracks_tree::TrkTOFexpMom,
210+
two_tracks_tree::TrueChannel,
211+
two_tracks_tree::TrueHasRecoColl,
212+
two_tracks_tree::TruePosX,
213+
two_tracks_tree::TruePosY,
214+
two_tracks_tree::TruePosZ,
215+
two_tracks_tree::TrueMotherPx,
216+
two_tracks_tree::TrueMotherPy,
217+
two_tracks_tree::TrueMotherPz,
218+
two_tracks_tree::TrueDaugPx,
219+
two_tracks_tree::TrueDaugPy,
220+
two_tracks_tree::TrueDaugPz,
221+
two_tracks_tree::TrueDaugPdgCode,
222+
two_tracks_tree::ProblematicEvent);
223+
224+
} // namespace o2::aod
225+
226+
#endif // PWGUD_DATAMODEL_TWOTRACKSEVENTTABLES_H_

‎PWGUD/TableProducer/CMakeLists.txt‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,11 @@ o2physics_add_dpl_workflow(tau-event-table-producer
3636
PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore
3737
COMPONENT_NAME Analysis)
3838

39+
o2physics_add_dpl_workflow(two-tracks-event-table-producer
40+
SOURCES twoTracksEventTableProducer.cxx
41+
PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore
42+
COMPONENT_NAME Analysis)
43+
3944
o2physics_add_dpl_workflow(fwdtrack-propagation
4045
SOURCES fwdTrackPropagation.cxx
4146
PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore O2::GlobalTracking

‎PWGUD/TableProducer/tauEventTableProducer.cxx‎

Lines changed: 25 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -43,10 +43,6 @@
4343
#include "PWGUD/DataModel/TauEventTables.h"
4444
#include "PWGUD/Core/SGSelector.h"
4545

46-
// ROOT headers
47-
#include "TLorentzVector.h"
48-
#include "TPDGCode.h"
49-
5046
using namespace o2;
5147
using namespace o2::framework;
5248
using namespace o2::framework::expressions;
@@ -77,6 +73,11 @@ struct TauEventTableProducer {
7773
Configurable<float> cutTrueGapSideFT0C{"cutTrueGapSideFT0C", 50., "FT0C threshold for SG selector"};
7874
Configurable<float> cutTrueGapSideZDC{"cutTrueGapSideZDC", 10000., "ZDC threshold for SG selector. 0 is <1n, 4.2 is <2n, 6.7 is <3n, 9.5 is <4n, 12.5 is <5n"};
7975
Configurable<float> cutFITtime{"cutFITtime", 40., "Maximum FIT time allowed. Default is 40ns"};
76+
Configurable<bool> cutEvTFb{"cutEvTFb", true, {"Event selection bit kNoTimeFrameBorder"}};
77+
Configurable<bool> cutEvITSROFb{"cutEvITSROFb", true, {"Event selection bit kNoITSROFrameBorder"}};
78+
Configurable<bool> cutEvSbp{"cutEvSbp", true, {"Event selection bit kNoSameBunchPileup"}};
79+
Configurable<bool> cutEvZvtxFT0vPV{"cutEvZvtxFT0vPV", false, {"Event selection bit kIsGoodZvtxFT0vsPV"}};
80+
Configurable<bool> cutEvVtxITSTPC{"cutEvVtxITSTPC", true, {"Event selection bit kIsVertexITSTPC"}};
8081
Configurable<float> cutEvOccupancy{"cutEvOccupancy", 100000., "Maximum allowed occupancy"};
8182
Configurable<bool> cutEvTrs{"cutEvTrs", false, {"Event selection bit kNoCollInTimeRangeStandard"}};
8283
Configurable<bool> cutEvTrofs{"cutEvTrofs", false, {"Event selection bit kNoCollInRofStandard"}};
@@ -156,6 +157,26 @@ struct TauEventTableProducer {
156157
bool isGoodROFtime(C const& coll)
157158
{
158159

160+
// kNoTimeFrameBorder
161+
if (cutSample.cutEvTFb && !coll.tfb())
162+
return false;
163+
164+
// kNoITSROFrameBorder
165+
if (cutSample.cutEvITSROFb && !coll.itsROFb())
166+
return false;
167+
168+
// kNoSameBunchPileup
169+
if (cutSample.cutEvSbp && !coll.sbp())
170+
return false;
171+
172+
// kIsGoodZvtxFT0vsPV
173+
if (cutSample.cutEvZvtxFT0vPV && !coll.zVtxFT0vPV())
174+
return false;
175+
176+
// kIsVertexITSTPC
177+
if (cutSample.cutEvVtxITSTPC && !coll.vtxITSTPC())
178+
return false;
179+
159180
// Occupancy
160181
if (coll.occupancyInTime() > cutSample.cutEvOccupancy)
161182
return false;

0 commit comments

Comments
 (0)