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> |
24 |
#include <NDLevel2.h> |
#include <NDLevel2.h> |
25 |
#include <AcLevel2.h> |
#include <AcLevel2.h> |
26 |
#include <OrbitalInfo.h> |
#include <OrbitalInfo.h> |
|
#include <OrbitalInfo.h> |
|
27 |
#include <RunInfo.h> |
#include <RunInfo.h> |
28 |
#include <GLTables.h> |
#include <GLTables.h> |
29 |
|
|
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 |
/** |
/** |
50 |
* This class inherites from TrkTrack, CaloTrkVar ecc... |
* This class inherites from TrkTrack, CaloTrkVar ecc... |
51 |
* |
* |
52 |
*/ |
*/ |
53 |
class PamTrack : public TrkTrack, public CaloTrkVar, public ToFTrkVar { |
//class PamTrack : public TrkTrack, public CaloTrkVar, public ToFTrkVar { |
54 |
|
class PamTrack : public TObject { |
55 |
|
|
56 |
private: |
private: |
57 |
|
|
59 |
CaloTrkVar* calo_track; |
CaloTrkVar* calo_track; |
60 |
ToFTrkVar* tof_track; |
ToFTrkVar* tof_track; |
61 |
|
|
62 |
|
bool candeleteobj; |
63 |
|
|
64 |
|
// TClonesArray *Cluster; ///< clusters (if stored) |
65 |
|
|
66 |
public: |
public: |
67 |
|
|
68 |
PamTrack(); |
PamTrack(); |
|
|
|
69 |
PamTrack(TrkTrack* , CaloTrkVar* , ToFTrkVar* ); |
PamTrack(TrkTrack* , CaloTrkVar* , ToFTrkVar* ); |
70 |
|
PamTrack(const PamTrack&); |
71 |
|
~PamTrack(){ Delete(); }; |
72 |
|
|
73 |
|
void Clear(); |
74 |
|
void Clear(Option_t *option){Clear();}; |
75 |
|
void Delete(); |
76 |
|
|
77 |
TrkTrack* GetTrkTrack(){return trk_track;}; |
TrkTrack* GetTrkTrack(){return trk_track;}; |
78 |
CaloTrkVar* GetCaloTrack(){return calo_track;}; |
CaloTrkVar* GetCaloTrack(){return calo_track;}; |
79 |
ToFTrkVar* GetToFTrack(){return tof_track;}; |
ToFTrkVar* GetToFTrack(){return tof_track;}; |
80 |
|
|
81 |
ClassDef(PamTrack,1); |
ClassDef(PamTrack,1); |
82 |
}; |
}; |
88 |
* This allows to solve the tracker ambiguity (that is, to choose between track images) by using also information from other detectors. |
* This allows to solve the tracker ambiguity (that is, to choose between track images) by using also information from other detectors. |
89 |
*/ |
*/ |
90 |
|
|
91 |
class PamLevel2 : public CaloLevel2, public CaloLevel1, public ToFLevel2, public TrkLevel2, public TrkLevel1, public TrkHough, public TrigLevel2, public S4Level2, public NDLevel2, public AcLevel2, public OrbitalInfo { |
//class PamLevel2 : public CaloLevel2, public CaloLevel1, public ToFLevel2, public TrkLevel2, public TrkLevel1, public TrkHough, public TrigLevel2, public S4Level2, public NDLevel2, public AcLevel2, public OrbitalInfo { |
92 |
|
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 |
|
Long64_t irun; //current run entry, relative to the TChain |
101 |
|
// ULong64_t runfirstentry; //first entry of run, relative to the TChain |
102 |
|
// ULong64_t runlastentry; //last entry of run, relative to the TChain |
103 |
|
// Int_t irunentry; //position of current entry, relative to the run |
104 |
|
Long64_t runfirstentry; //first entry of run, relative to the TChain |
105 |
|
Long64_t runlastentry; //last entry of run, relative to the TChain |
106 |
|
Long64_t irunentry; //position of current entry, relative to the run |
107 |
|
TChain* run_tree; |
108 |
|
TTree* run_tree_clone; |
109 |
|
TChain* sel_tree; |
110 |
|
TTree* sel_tree_clone; |
111 |
|
|
112 |
|
// --------------------------------- |
113 |
|
// variables to retrieve level0 info |
114 |
|
// --------------------------------- |
115 |
|
TFile* l0_file; |
116 |
|
TTree* l0_tree; |
117 |
|
Int_t iroot; |
118 |
|
Int_t prevshift; // to remember if any shift is needed to synchronize l0 and l2 data instead of calculating any time |
119 |
|
// -------------------------------------------- |
120 |
|
// variables to store Level0 info (if required) |
121 |
|
// -------------------------------------------- |
122 |
|
EventHeader* h0_obj; |
123 |
|
TrkLevel0* trk0_obj; |
124 |
|
CaloLevel0* calo0_obj; |
125 |
|
|
126 |
|
// --------------------------- |
127 |
|
// variables to connect the DB |
128 |
|
// --------------------------- |
129 |
|
TSQLServer* dbc; |
130 |
|
TString host; |
131 |
|
TString user; |
132 |
|
TString psw; |
133 |
|
|
134 |
|
// --------------------------------------------- |
135 |
|
// variables to retrieve and clone level1-2 info |
136 |
|
// --------------------------------------------- |
137 |
|
TChain* pam_tree; |
138 |
|
TTree* pam_tree_clone[NCLONES]; |
139 |
|
|
140 |
|
// -------------------------------------------- |
141 |
|
// variables to store Level0 info (if required) |
142 |
|
// -------------------------------------------- |
143 |
TrkLevel1* trk1_obj; |
TrkLevel1* trk1_obj; |
144 |
TrkLevel2* trk2_obj; |
TrkLevel2* trk2_obj; |
145 |
TrkHough* trkh_obj; |
TrkHough* trkh_obj; |
|
|
|
146 |
CaloLevel1* calo1_obj; |
CaloLevel1* calo1_obj; |
147 |
CaloLevel2* calo2_obj; |
CaloLevel2* calo2_obj; |
|
|
|
148 |
ToFLevel2* tof_obj; |
ToFLevel2* tof_obj; |
149 |
TrigLevel2* trig_obj; |
TrigLevel2* trig_obj; |
150 |
S4Level2* s4_obj; |
S4Level2* s4_obj; |
151 |
NDLevel2* nd_obj; |
NDLevel2* nd_obj; |
152 |
AcLevel2* ac_obj; |
AcLevel2* ac_obj; |
153 |
OrbitalInfo* orb_obj; |
OrbitalInfo* orb_obj; |
|
|
|
154 |
GL_RUN* run_obj; |
GL_RUN* run_obj; |
155 |
|
SoftInfo* soft_obj; // Emiliano |
|
/* TrkLevel1* trk1_clone; */ |
|
|
/* TrkLevel2* trk2_clone; */ |
|
|
/* TrkHough* trkh_clone; */ |
|
|
|
|
|
/* CaloLevel1* calo1_clone; */ |
|
|
/* CaloLevel2* calo2_clone; */ |
|
|
|
|
|
/* ToFLevel2* tof_clone; */ |
|
|
/* TrigLevel2* trig_clone; */ |
|
|
/* S4Level2* s4_clone; */ |
|
|
/* NDLevel2* nd_clone; */ |
|
|
/* AcLevel2* ac_clone; */ |
|
|
/* OrbitalInfo* orb_clone; */ |
|
|
|
|
|
/* TTree* T_clone; //tracker */ |
|
|
/* TTree* C_clone; //calo */ |
|
|
/* TTree* O_clone; //tof */ |
|
|
/* TTree* R_clone; //trigger */ |
|
|
/* TTree* S_clone; //s4 */ |
|
|
/* TTree* A_clone; //anti */ |
|
|
/* TTree* N_clone; //nd */ |
|
|
/* TTree* B_clone; //orbital */ |
|
|
|
|
|
TTree* tree_clone[8]; |
|
156 |
|
|
157 |
TRefArray* sorted_tracks; |
// ----------------------------------- |
158 |
|
// variables to sort the track |
159 |
|
// ----------------------------------- |
160 |
|
TClonesArray *tsorted; |
161 |
|
TClonesArray *timage; |
162 |
|
PamTrack *track; //store the last required |
163 |
|
TString howtosort; |
164 |
|
Float_t sortthr; |
165 |
|
|
166 |
|
// ----- |
167 |
|
// flags |
168 |
|
// ----- |
169 |
Bool_t CAL2; //level2 branch |
Bool_t CAL2; //level2 branch |
170 |
Bool_t CAL1; //level1 branch |
Bool_t CAL1; //level1 branch |
171 |
|
Bool_t CAL0; //yoda |
172 |
|
|
173 |
Bool_t TRK2; //level2 branch |
Bool_t TRK2; //level2 branch |
174 |
Bool_t TRK1; //level1 branch |
Bool_t TRK1; //level1 branch |
175 |
Bool_t TRKh; //hough branch |
Bool_t TRKh; //hough branch |
176 |
|
Bool_t TRK0; //yoda |
177 |
|
|
|
Bool_t TRG; |
|
178 |
Bool_t TOF; |
Bool_t TOF; |
179 |
|
Bool_t TOF0; |
180 |
|
|
181 |
|
Bool_t TRG; |
182 |
Bool_t S4; |
Bool_t S4; |
183 |
Bool_t ND; |
Bool_t ND; |
184 |
Bool_t AC; |
Bool_t AC; |
185 |
Bool_t ORB; |
Bool_t ORB; |
186 |
|
|
187 |
Bool_t RUN; |
Bool_t RUN; |
188 |
|
|
189 |
|
Int_t SELLI; //trees of selected events (there is a selection list) |
190 |
|
|
191 |
|
void Initialize(); |
192 |
|
|
193 |
|
|
194 |
|
public: |
195 |
|
|
196 |
|
PamLevel2(TString ddir,TString list,TString detlist); |
197 |
|
PamLevel2(TString ddir,TString list); |
198 |
PamLevel2(); |
PamLevel2(); |
199 |
~PamLevel2(); |
~PamLevel2(){Delete();}; |
200 |
|
|
201 |
void Clear(); |
void Clear(); |
202 |
|
void Reset(); |
203 |
|
void Delete(); |
204 |
|
|
205 |
/* TrkLevel1* GetTrkLevel1() { return trk1_obj; }; */ |
|
206 |
/* TrkLevel2* GetTrkLevel2() { return trk2_obj; }; */ |
// ============================================= |
207 |
/* TrkHough* GetTrkHough() { return trkh_obj; }; */ |
// methods needed to read/write Level0-1-2 trees |
208 |
/* CaloLevel1* GetCaloLevel1() { return calo1_obj; }; */ |
// ============================================= |
|
/* CaloLevel2* GetCaloLevel2() { return calo2_obj; }; */ |
|
|
/* ToFLevel2* GetToFLevel2() { return tof_obj; }; */ |
|
|
/* TrigLevel2* GetTrigLevel2() { return trig_obj; }; */ |
|
|
/* S4Level2* GetS4Level2() { return s4_obj; }; */ |
|
|
/* NDLevel2* GetNDLevel2() { return nd_obj; }; */ |
|
|
/* AcLevel2* GetAcLevel2() { return ac_obj; }; */ |
|
|
/* OrbitalInfo* GetOrbitalInfo(){ return orb_obj; }; */ |
|
|
/* TrkLevel1* GetTrkLevel1() { return ; }; */ |
|
|
/* TrkLevel2* GetTrkLevel2() { return TrkLevel2::GetTrkLevel2(); }; */ |
|
|
/* TrkHough* GetTrkHough() { return TrkHough::GetTrkHough(); }; */ |
|
|
/* CaloLevel1* GetCaloLevel1() { return CaloLevel1::GetCaloLevel1(); }; */ |
|
|
/* CaloLevel2* GetCaloLevel2() { return CaloLevel2::GetCaloLevel2(); }; */ |
|
|
/* ToFLevel2* GetToFLevel2() { return ToFLevel2::GetToFLevel2(); }; */ |
|
|
/* TrigLevel2* GetTrigLevel2() { return TrigLevel2::GetTrigLevel2(); }; */ |
|
|
/* S4Level2* GetS4Level2() { return S4Level2::GetS4Level2(); }; */ |
|
|
/* NDLevel2* GetNDLevel2() { return NDLevel2::GetNDLevel2(); }; */ |
|
|
/* AcLevel2* GetAcLevel2() { return AcLevel2::GetAcLevel2(); }; */ |
|
|
/* OrbitalInfo* GetOrbitalInfo(){ return OrbitalInfo::GetOrbitalInfo(); }; */ |
|
|
GL_RUN* GetRunInfo() { return run_obj; }; |
|
|
|
|
|
// ========================================= |
|
|
// methods needed to read/write Level2 trees |
|
|
// ========================================= |
|
209 |
void SetWhichTrees(TString); |
void SetWhichTrees(TString); |
210 |
void GetWhichTrees(TFile*); |
void GetWhichTrees(TFile*); |
211 |
|
|
220 |
|
|
221 |
TTree* GetRunTree(TFile*); |
TTree* GetRunTree(TFile*); |
222 |
TChain* GetRunTree(TList*); |
TChain* GetRunTree(TList*); |
223 |
|
TChain* GetRunTree(TString ddir,TString list){ return GetRunTree(GetListOfLevel2Files(ddir,list)); }; |
224 |
|
Bool_t UpdateRunInfo(TTree *,Long64_t); |
225 |
|
Bool_t UpdateRunInfo(TChain *,Long64_t); |
226 |
|
Bool_t UpdateRunInfo(Long64_t iev);//{ return UpdateRunInfo(run_tree,iev); }; |
227 |
|
|
228 |
void CreateCloneTrees0(TChain *fChain, TFile* file); |
void CreateCloneTrees0(TChain *fChain, TFile* file); |
229 |
void CreateCloneTrees(TFile* file); |
void CreateCloneTrees(TFile* file); |
234 |
void SetBranchAddress(TTree *); |
void SetBranchAddress(TTree *); |
235 |
void SetBranchAddress(TChain *); |
void SetBranchAddress(TChain *); |
236 |
|
|
237 |
// ===================================== |
// Int_t GetEntry(Int_t); |
238 |
// methods needed to set brach addresses |
Int_t GetEntry(Long64_t); |
239 |
// ===================================== |
Int_t GetYodaEntry(); |
240 |
/* void* GetPointerToTrk(); */ |
|
241 |
/* void* GetPointerToTrk(int s); */ |
Long64_t GetEntries(){return pam_tree->GetEntries();}; |
242 |
/* void* GetPointerToCalo(); */ |
TChain* GetPamTree(){return pam_tree;}; |
243 |
/* void* GetPointerToCalo(int s); */ |
TChain* GetRunTree(){return run_tree;}; |
244 |
/* void* GetPointerToToF(); */ |
TTree* GetYodaTree(); |
245 |
/* void* GetPointerToTrig(); */ |
|
246 |
/* void* GetPointerToS4(); */ |
TFile* GetYodaFile(){return l0_file;}; // can be useful to know the name of level0 file we are using or similar things |
247 |
/* void* GetPointerToND(); */ |
|
248 |
/* void* GetPointerToAc(); */ |
// ============= |
249 |
/* void* GetPointerToOrb(); */ |
// DB connection |
250 |
/* void* GetPointerToRun(); */ |
// ============= |
251 |
|
void SetDBHost(TString str){host=str;}; |
252 |
|
void SetDBUser(TString str){user=str;}; |
253 |
|
void SetDBPsw(TString str){psw=str;}; |
254 |
|
|
255 |
|
|
256 |
|
// ======================================================== |
257 |
|
// methods needed to retrieve subdetector general variables |
258 |
|
// ======================================================== |
259 |
void* GetPointerTo(const char*); |
void* GetPointerTo(const char*); |
260 |
|
EventHeader *GetEventHeader() { return h0_obj; }; |
261 |
|
TrkLevel0 *GetTrkLevel0() { return trk0_obj; }; |
262 |
|
CaloLevel0 *GetCaloLevel0() { return calo0_obj; }; |
263 |
|
|
264 |
|
TrkLevel1 *GetTrkLevel1() { return trk1_obj; }; |
265 |
|
TrkLevel2 *GetTrkLevel2() { return trk2_obj; }; |
266 |
|
TrkHough *GetTrkHough() { return trkh_obj; }; |
267 |
|
CaloLevel1 *GetCaloLevel1() { return calo1_obj;}; |
268 |
|
CaloLevel2 *GetCaloLevel2() { return calo2_obj;}; |
269 |
|
ToFLevel2 *GetToFLevel2() { return tof_obj; }; |
270 |
|
TrigLevel2 *GetTrigLevel2() { return trig_obj; }; |
271 |
|
S4Level2 *GetS4Level2() { return s4_obj; }; |
272 |
|
NDLevel2 *GetNDLevel2() { return nd_obj; }; |
273 |
|
AcLevel2 *GetAcLevel2() { return ac_obj; }; |
274 |
|
OrbitalInfo *GetOrbitalInfo(){ return orb_obj; }; |
275 |
|
GL_RUN *GetRunInfo() { return run_obj; }; |
276 |
|
SoftInfo *GetSoftInfo() { return soft_obj; }; // Emiliano |
277 |
|
|
278 |
// ============================================== |
// ============================================== |
279 |
// methods to retrieve subdetectors stored tracks |
// methods to retrieve subdetectors stored tracks |
280 |
// ============================================== |
// ============================================== |
281 |
TrkTrack* GetTrkStoredTrack(int seqno){return TrkLevel2::GetStoredTrack(seqno); }; |
TrkTrack* GetTrkStoredTrack(int seqno){return trk2_obj->TrkLevel2::GetStoredTrack(seqno); }; |
282 |
CaloTrkVar* GetCaloStoredTrack(int); |
CaloTrkVar* GetCaloStoredTrack(int); |
283 |
ToFTrkVar* GetToFStoredTrack(int); |
ToFTrkVar* GetToFStoredTrack(int); |
284 |
// |
// |
292 |
PamTrack* GetStoredTrack(Int_t); |
PamTrack* GetStoredTrack(Int_t); |
293 |
// |
// |
294 |
void SortTracks(TString); //assign value to variable sorted_tracks |
void SortTracks(TString); //assign value to variable sorted_tracks |
295 |
TRefArray* GetTracks(); //return sorted_tracks |
void SortTracks(); //assign value to variable sorted_tracks |
296 |
|
// TRefArray* GetTracks(); //return sorted_tracks |
297 |
|
TClonesArray* GetTracks(); //return sorted_tracks |
298 |
PamTrack* GetTrack(int); |
PamTrack* GetTrack(int); |
299 |
PamTrack* GetTrackImage(int); |
PamTrack* GetTrackImage(int); |
300 |
|
void SetSortingMethod(TString how){ howtosort = how; return;}; |
301 |
|
TString GetSortingMethod(){return howtosort;}; |
302 |
|
void SetNucleiSortingThreshold(Float_t nucleithr){ sortthr = nucleithr; return;}; |
303 |
|
Float_t GetNucleiSortingThreshold(){ return sortthr;}; |
304 |
|
|
305 |
// ================================= |
// ================================= |
306 |
// other methods |
// other methods |
307 |
// ================================= |
// ================================= |
308 |
Int_t ntrk(){ return TrkLevel2::ntrk(); }; |
// Int_t ntrk(){ return trk2_obj->TrkLevel2::ntrk(); }; |
309 |
|
|
310 |
|
// ----- |
311 |
|
// flags methods |
312 |
|
// ----- |
313 |
|
Bool_t IsCAL2(){return CAL2;}; //level2 branch |
314 |
|
Bool_t IsCAL1(){return CAL1;}; //level1 branch |
315 |
|
Bool_t IsCAL0(){return CAL0;}; //yoda |
316 |
|
|
317 |
|
Bool_t IsTRK2(){return TRK2;}; //level2 branch |
318 |
|
Bool_t IsTRK1(){return TRK1;}; //level1 branch |
319 |
|
Bool_t IsTRKh(){return TRKh;}; //hough branch |
320 |
|
Bool_t IsTRK0(){return TRK0;}; //yoda |
321 |
|
|
322 |
|
Bool_t IsTOF(){return TOF;}; |
323 |
|
Bool_t IsTOF0(){return TOF0;}; |
324 |
|
|
325 |
|
Bool_t IsTRG(){return TRG;}; |
326 |
|
Bool_t IsS4(){return S4;}; |
327 |
|
Bool_t IsND(){return ND;}; |
328 |
|
Bool_t IsAC(){return AC;}; |
329 |
|
Bool_t IsORB(){return ORB;}; |
330 |
|
|
331 |
|
Bool_t IsRUN(){return RUN;}; |
332 |
|
|
333 |
|
Bool_t IsGood(); |
334 |
|
|
335 |
|
Int_t IsSELLI(){return SELLI;}; //trees of selected events (there is a selection list) |
336 |
|
|
337 |
ClassDef(PamLevel2,1); |
ClassDef(PamLevel2,1); |
338 |
}; |
}; |