Skip to content

Commit 02ef6e8

Browse files
committed
[PWGLF] Add explicit ROOT leaflists for coalescence tree branches
1 parent 196d274 commit 02ef6e8

1 file changed

Lines changed: 41 additions & 40 deletions

File tree

PWGLF/TableProducer/Nuspex/coalescenceTreeProducer.cxx

Lines changed: 41 additions & 40 deletions
Original file line numberDiff line numberDiff line change
@@ -28,13 +28,13 @@
2828
///
2929
/// \author Alberto Calivà <alberto.caliva@cern.ch>
3030

31-
#include "Framework/AnalysisTask.h"
32-
#include "Framework/Configurable.h"
33-
#include "Framework/HistogramRegistry.h"
34-
#include "Framework/InitContext.h"
35-
#include "Framework/Logger.h"
36-
#include "Framework/OutputObjHeader.h"
37-
#include "Framework/runDataProcessing.h"
31+
#include <Framework/AnalysisTask.h>
32+
#include <Framework/Configurable.h>
33+
#include <Framework/HistogramRegistry.h>
34+
#include <Framework/InitContext.h>
35+
#include <Framework/Logger.h>
36+
#include <Framework/OutputObjHeader.h>
37+
#include <Framework/runDataProcessing.h>
3838

3939
#include <Math/Boost.h>
4040
#include <Math/Vector3D.h>
@@ -71,7 +71,7 @@ struct CoalescenceTreeProducer {
7171

7272
Preslice<aod::McParticles> mcParticlesPerMcCollision = aod::mcparticle::mcCollisionId;
7373

74-
OutputObj<TTree> treeBoundState{"treeBoundState"};
74+
OutputObj<TTree> treeBoundState{"treeBoundState","treeBoundState"};
7575

7676
int64_t eventID = 0; // Event ID
7777
int64_t idB1 = 0, idB2 = 0, idB3 = 0; // MC particle IDs
@@ -154,45 +154,46 @@ struct CoalescenceTreeProducer {
154154
// Tree for pairs/triplets.
155155
// For deuterons only the first two baryons are stored.
156156
// For three-body states, the third-baryon branches are also created.
157-
treeBoundState->Branch("eventID", &eventID);
157+
treeBoundState->Branch("eventID", &eventID, "eventID/L");
158158

159-
treeBoundState->Branch("idB1", &idB1);
160-
treeBoundState->Branch("idB2", &idB2);
159+
treeBoundState->Branch("idB1", &idB1, "idB1/L");
160+
treeBoundState->Branch("idB2", &idB2, "idB2/L");
161161

162-
treeBoundState->Branch("pdgB1", &pdgB1);
163-
treeBoundState->Branch("pdgB2", &pdgB2);
162+
treeBoundState->Branch("pdgB1", &pdgB1, "pdgB1/I");
163+
treeBoundState->Branch("pdgB2", &pdgB2, "pdgB2/I");
164164

165-
treeBoundState->Branch("chargeB1", &chargeB1);
166-
treeBoundState->Branch("chargeB2", &chargeB2);
165+
treeBoundState->Branch("chargeB1", &chargeB1, "chargeB1/I");
166+
treeBoundState->Branch("chargeB2", &chargeB2, "chargeB2/I");
167167

168-
treeBoundState->Branch("xB1", &xB1);
169-
treeBoundState->Branch("yB1", &yB1);
170-
treeBoundState->Branch("zB1", &zB1);
171-
treeBoundState->Branch("tB1", &tB1);
172-
treeBoundState->Branch("pxB1", &pxB1);
173-
treeBoundState->Branch("pyB1", &pyB1);
174-
treeBoundState->Branch("pzB1", &pzB1);
168+
treeBoundState->Branch("xB1", &xB1, "xB1/F");
169+
treeBoundState->Branch("yB1", &yB1, "yB1/F");
170+
treeBoundState->Branch("zB1", &zB1, "zB1/F");
171+
treeBoundState->Branch("tB1", &tB1, "tB1/F");
172+
treeBoundState->Branch("pxB1", &pxB1, "pxB1/F");
173+
treeBoundState->Branch("pyB1", &pyB1, "pyB1/F");
174+
treeBoundState->Branch("pzB1", &pzB1, "pzB1/F");
175175

176-
treeBoundState->Branch("xB2", &xB2);
177-
treeBoundState->Branch("yB2", &yB2);
178-
treeBoundState->Branch("zB2", &zB2);
179-
treeBoundState->Branch("tB2", &tB2);
180-
treeBoundState->Branch("pxB2", &pxB2);
181-
treeBoundState->Branch("pyB2", &pyB2);
182-
treeBoundState->Branch("pzB2", &pzB2);
176+
treeBoundState->Branch("xB2", &xB2, "xB2/F");
177+
treeBoundState->Branch("yB2", &yB2, "yB2/F");
178+
treeBoundState->Branch("zB2", &zB2, "zB2/F");
179+
treeBoundState->Branch("tB2", &tB2, "tB2/F");
180+
treeBoundState->Branch("pxB2", &pxB2, "pxB2/F");
181+
treeBoundState->Branch("pyB2", &pyB2, "pyB2/F");
182+
treeBoundState->Branch("pzB2", &pzB2, "pzB2/F");
183183

184184
if (static_cast<int>(boundStateSpecies) != kDeuteron) {
185-
treeBoundState->Branch("idB3", &idB3);
186-
treeBoundState->Branch("pdgB3", &pdgB3);
187-
treeBoundState->Branch("chargeB3", &chargeB3);
188-
189-
treeBoundState->Branch("xB3", &xB3);
190-
treeBoundState->Branch("yB3", &yB3);
191-
treeBoundState->Branch("zB3", &zB3);
192-
treeBoundState->Branch("tB3", &tB3);
193-
treeBoundState->Branch("pxB3", &pxB3);
194-
treeBoundState->Branch("pyB3", &pyB3);
195-
treeBoundState->Branch("pzB3", &pzB3);
185+
treeBoundState->Branch("idB3", &idB3, "idB3/L");
186+
187+
treeBoundState->Branch("pdgB3", &pdgB3, "pdgB3/I");
188+
treeBoundState->Branch("chargeB3", &chargeB3, "chargeB3/I");
189+
190+
treeBoundState->Branch("xB3", &xB3, "xB3/F");
191+
treeBoundState->Branch("yB3", &yB3, "yB3/F");
192+
treeBoundState->Branch("zB3", &zB3, "zB3/F");
193+
treeBoundState->Branch("tB3", &tB3, "tB3/F");
194+
treeBoundState->Branch("pxB3", &pxB3, "pxB3/F");
195+
treeBoundState->Branch("pyB3", &pyB3, "pyB3/F");
196+
treeBoundState->Branch("pzB3", &pzB3, "pzB3/F");
196197
}
197198
}
198199

0 commit comments

Comments
 (0)