9 |
|
|
10 |
#include <TRefArray.h> |
#include <TRefArray.h> |
11 |
|
|
12 |
|
#include <TrkStruct.h> |
13 |
|
#include <TrkParams.h> |
14 |
|
#include <TrkLevel0.h> |
15 |
#include <TrkLevel1.h> |
#include <TrkLevel1.h> |
16 |
#include <TrkLevel2.h> |
#include <TrkLevel2.h> |
17 |
#include <TrkHough.h> |
#include <TrkHough.h> |
18 |
|
#include <CaloLevel0.h> |
19 |
#include <CaloLevel1.h> |
#include <CaloLevel1.h> |
20 |
#include <CaloLevel2.h> |
#include <CaloLevel2.h> |
21 |
#include <ToFLevel2.h> |
#include <ToFLevel2.h> |
36 |
#include <TKey.h> |
#include <TKey.h> |
37 |
#include <TSystemFile.h> |
#include <TSystemFile.h> |
38 |
#include <TSystemDirectory.h> |
#include <TSystemDirectory.h> |
39 |
|
#include <TSQLServer.h> |
40 |
|
|
41 |
#include <iostream> |
#include <iostream> |
42 |
|
|
43 |
|
#define NCLONES 10 |
44 |
|
|
45 |
using namespace std; |
using namespace std; |
46 |
|
|
47 |
/** |
/** |
92 |
class PamLevel2 : public TObject { |
class PamLevel2 : public TObject { |
93 |
|
|
94 |
private: |
private: |
|
public: |
|
95 |
|
|
96 |
|
// ------------------------------ |
97 |
|
// variables to retrieve run info |
98 |
|
// ------------------------------ |
99 |
|
Int_t irun; //current run entry, relative to the TChain |
100 |
|
ULong64_t runfirstentry; //first entry of run, relative to the TChain |
101 |
|
ULong64_t runlastentry; //last entry of run, relative to the TChain |
102 |
|
Int_t irunentry; //position of current entry, relative to the run |
103 |
|
TChain* run_tree; |
104 |
|
TTree* run_tree_clone; |
105 |
|
TChain* sel_tree; |
106 |
|
TTree* sel_tree_clone; |
107 |
|
|
108 |
|
// --------------------------------- |
109 |
|
// variables to retrieve level0 info |
110 |
|
// --------------------------------- |
111 |
|
TFile* l0_file; |
112 |
|
TTree* l0_tree; |
113 |
|
Int_t iroot; |
114 |
|
Int_t prevshift; // to remember if any shift is needed to synchronize l0 and l2 data instead of calculating any time |
115 |
|
// -------------------------------------------- |
116 |
|
// variables to store Level0 info (if required) |
117 |
|
// -------------------------------------------- |
118 |
|
EventHeader* h0_obj; |
119 |
|
TrkLevel0* trk0_obj; |
120 |
|
CaloLevel0* calo0_obj; |
121 |
|
|
122 |
|
// --------------------------- |
123 |
|
// variables to connect the DB |
124 |
|
// --------------------------- |
125 |
|
TSQLServer* dbc; |
126 |
|
TString host; |
127 |
|
TString user; |
128 |
|
TString psw; |
129 |
|
|
130 |
|
// --------------------------------------------- |
131 |
|
// variables to retrieve and clone level1-2 info |
132 |
|
// --------------------------------------------- |
133 |
|
TChain* pam_tree; |
134 |
|
TTree* pam_tree_clone[NCLONES]; |
135 |
|
// -------------------------------------------- |
136 |
|
// variables to store Level0 info (if required) |
137 |
|
// -------------------------------------------- |
138 |
TrkLevel1* trk1_obj; |
TrkLevel1* trk1_obj; |
139 |
TrkLevel2* trk2_obj; |
TrkLevel2* trk2_obj; |
140 |
TrkHough* trkh_obj; |
TrkHough* trkh_obj; |
|
|
|
141 |
CaloLevel1* calo1_obj; |
CaloLevel1* calo1_obj; |
142 |
CaloLevel2* calo2_obj; |
CaloLevel2* calo2_obj; |
|
|
|
143 |
ToFLevel2* tof_obj; |
ToFLevel2* tof_obj; |
144 |
TrigLevel2* trig_obj; |
TrigLevel2* trig_obj; |
145 |
S4Level2* s4_obj; |
S4Level2* s4_obj; |
146 |
NDLevel2* nd_obj; |
NDLevel2* nd_obj; |
147 |
AcLevel2* ac_obj; |
AcLevel2* ac_obj; |
148 |
OrbitalInfo* orb_obj; |
OrbitalInfo* orb_obj; |
|
|
|
149 |
GL_RUN* run_obj; |
GL_RUN* run_obj; |
150 |
SoftInfo* soft_obj; // Emiliano |
SoftInfo* soft_obj; // Emiliano |
|
|
|
|
TTree* tree_clone[8]; |
|
|
|
|
|
// TRefArray* sorted_tracks; |
|
|
// TRefArray sorted_tracks; |
|
151 |
|
|
152 |
|
// ----------------------------------- |
153 |
|
// variables to sort the track |
154 |
|
// ----------------------------------- |
155 |
TClonesArray *tsorted; |
TClonesArray *tsorted; |
156 |
TClonesArray *timage; |
TClonesArray *timage; |
|
|
|
|
|
|
157 |
PamTrack *track; //store the last required |
PamTrack *track; //store the last required |
158 |
|
|
159 |
|
|
160 |
|
// ----- |
161 |
|
// flags |
162 |
|
// ----- |
163 |
Bool_t CAL2; //level2 branch |
Bool_t CAL2; //level2 branch |
164 |
Bool_t CAL1; //level1 branch |
Bool_t CAL1; //level1 branch |
165 |
|
Bool_t CAL0; //yoda |
166 |
|
|
167 |
Bool_t TRK2; //level2 branch |
Bool_t TRK2; //level2 branch |
168 |
Bool_t TRK1; //level1 branch |
Bool_t TRK1; //level1 branch |
169 |
Bool_t TRKh; //hough branch |
Bool_t TRKh; //hough branch |
170 |
|
Bool_t TRK0; //yoda |
171 |
|
|
|
Bool_t TRG; |
|
172 |
Bool_t TOF; |
Bool_t TOF; |
173 |
|
Bool_t TOF0; |
174 |
|
|
175 |
|
Bool_t TRG; |
176 |
Bool_t S4; |
Bool_t S4; |
177 |
Bool_t ND; |
Bool_t ND; |
178 |
Bool_t AC; |
Bool_t AC; |
179 |
Bool_t ORB; |
Bool_t ORB; |
180 |
|
|
181 |
Bool_t RUN; |
Bool_t RUN; |
182 |
|
|
183 |
|
Int_t SELLI; //trees of selected events (there is a selection list) |
184 |
|
|
185 |
|
|
186 |
|
public: |
187 |
|
|
188 |
PamLevel2(); |
PamLevel2(); |
189 |
~PamLevel2(){Delete();}; |
~PamLevel2(){Delete();}; |
191 |
void Clear(); |
void Clear(); |
192 |
void Delete(); |
void Delete(); |
193 |
|
|
194 |
|
|
195 |
// ========================================= |
// ============================================= |
196 |
// methods needed to read/write Level2 trees |
// methods needed to read/write Level0-1-2 trees |
197 |
// ========================================= |
// ============================================= |
198 |
void SetWhichTrees(TString); |
void SetWhichTrees(TString); |
199 |
void GetWhichTrees(TFile*); |
void GetWhichTrees(TFile*); |
200 |
|
|
209 |
|
|
210 |
TTree* GetRunTree(TFile*); |
TTree* GetRunTree(TFile*); |
211 |
TChain* GetRunTree(TList*); |
TChain* GetRunTree(TList*); |
212 |
|
TChain* GetRunTree(TString ddir,TString list){ return GetRunTree(GetListOfLevel2Files(ddir,list)); }; |
213 |
|
Bool_t UpdateRunInfo(TTree *,ULong64_t); |
214 |
|
Bool_t UpdateRunInfo(TChain *,ULong64_t); |
215 |
|
|
216 |
void CreateCloneTrees0(TChain *fChain, TFile* file); |
void CreateCloneTrees0(TChain *fChain, TFile* file); |
217 |
void CreateCloneTrees(TFile* file); |
void CreateCloneTrees(TFile* file); |
222 |
void SetBranchAddress(TTree *); |
void SetBranchAddress(TTree *); |
223 |
void SetBranchAddress(TChain *); |
void SetBranchAddress(TChain *); |
224 |
|
|
225 |
|
Int_t GetEntry(Int_t); |
226 |
|
Int_t GetYodaEntry(); |
227 |
|
|
228 |
|
TChain* GetPamTree(){return pam_tree;}; |
229 |
|
TChain* GetRunTree(){return run_tree;}; |
230 |
|
TTree* GetYodaTree(); |
231 |
|
|
232 |
|
TFile* GetYodaFile(){return l0_file;}; // can be useful to know the name of level0 file we are using or similar things |
233 |
|
|
234 |
|
// ============= |
235 |
|
// DB connection |
236 |
|
// ============= |
237 |
|
void SetDBHost(TString str){host=str;}; |
238 |
|
void SetDBUser(TString str){user=str;}; |
239 |
|
void SetDBPsw(TString str){psw=str;}; |
240 |
|
|
241 |
|
|
242 |
// ======================================================== |
// ======================================================== |
243 |
// methods needed to retrieve subdetector general variables |
// methods needed to retrieve subdetector general variables |
244 |
// ======================================================== |
// ======================================================== |
245 |
void* GetPointerTo(const char*); |
void* GetPointerTo(const char*); |
246 |
TrkLevel1 *GetTrkLevel1() { return trk1_obj;}; |
EventHeader *GetEventHeader() { return h0_obj; }; |
247 |
TrkLevel2 *GetTrkLevel2() { return trk2_obj;}; |
TrkLevel0 *GetTrkLevel0() { return trk0_obj; }; |
248 |
TrkHough *GetTrkHough() { return trkh_obj;}; |
CaloLevel0 *GetCaloLevel0() { return calo0_obj; }; |
249 |
|
|
250 |
|
TrkLevel1 *GetTrkLevel1() { return trk1_obj; }; |
251 |
|
TrkLevel2 *GetTrkLevel2() { return trk2_obj; }; |
252 |
|
TrkHough *GetTrkHough() { return trkh_obj; }; |
253 |
CaloLevel1 *GetCaloLevel1() { return calo1_obj;}; |
CaloLevel1 *GetCaloLevel1() { return calo1_obj;}; |
254 |
CaloLevel2 *GetCaloLevel2() { return calo2_obj;}; |
CaloLevel2 *GetCaloLevel2() { return calo2_obj;}; |
255 |
ToFLevel2 *GetToFLevel2() { return tof_obj;}; |
ToFLevel2 *GetToFLevel2() { return tof_obj; }; |
256 |
TrigLevel2 *GetTrigLevel2() { return trig_obj;}; |
TrigLevel2 *GetTrigLevel2() { return trig_obj; }; |
257 |
S4Level2 *GetS4Level2() { return s4_obj;}; |
S4Level2 *GetS4Level2() { return s4_obj; }; |
258 |
NDLevel2 *GetNDLevel2() { return nd_obj;}; |
NDLevel2 *GetNDLevel2() { return nd_obj; }; |
259 |
AcLevel2 *GetAcLevel2() { return ac_obj;}; |
AcLevel2 *GetAcLevel2() { return ac_obj; }; |
260 |
OrbitalInfo *GetOrbitalInfo(){ return orb_obj;}; |
OrbitalInfo *GetOrbitalInfo(){ return orb_obj; }; |
261 |
GL_RUN *GetRunInfo() { return run_obj; }; |
GL_RUN *GetRunInfo() { return run_obj; }; |
262 |
SoftInfo *GetSoftInfo() { return soft_obj; }; // Emiliano |
SoftInfo *GetSoftInfo() { return soft_obj; }; // Emiliano |
263 |
|
|
264 |
// ============================================== |
// ============================================== |
288 |
// ================================= |
// ================================= |
289 |
// Int_t ntrk(){ return trk2_obj->TrkLevel2::ntrk(); }; |
// Int_t ntrk(){ return trk2_obj->TrkLevel2::ntrk(); }; |
290 |
|
|
291 |
|
// ----- |
292 |
|
// flags methods |
293 |
|
// ----- |
294 |
|
Bool_t IsCAL2(){return CAL2;}; //level2 branch |
295 |
|
Bool_t IsCAL1(){return CAL1;}; //level1 branch |
296 |
|
Bool_t IsCAL0(){return CAL0;}; //yoda |
297 |
|
|
298 |
|
Bool_t IsTRK2(){return TRK2;}; //level2 branch |
299 |
|
Bool_t IsTRK1(){return TRK1;}; //level1 branch |
300 |
|
Bool_t IsTRKh(){return TRKh;}; //hough branch |
301 |
|
Bool_t IsTRK0(){return TRK0;}; //yoda |
302 |
|
|
303 |
|
Bool_t IsTOF(){return TOF;}; |
304 |
|
Bool_t IsTOF0(){return TOF0;}; |
305 |
|
|
306 |
|
Bool_t IsTRG(){return TRG;}; |
307 |
|
Bool_t IsS4(){return S4;}; |
308 |
|
Bool_t IsND(){return ND;}; |
309 |
|
Bool_t IsAC(){return AC;}; |
310 |
|
Bool_t IsORB(){return ORB;}; |
311 |
|
|
312 |
|
Bool_t IsRUN(){return RUN;}; |
313 |
|
|
314 |
|
Int_t IsSELLI(){return SELLI;}; //trees of selected events (there is a selection list) |
315 |
|
|
316 |
ClassDef(PamLevel2,1); |
ClassDef(PamLevel2,1); |
317 |
}; |
}; |