/[PAMELA software]/PamelaLevel2/inc/PamLevel2.h
ViewVC logotype

Contents of /PamelaLevel2/inc/PamLevel2.h

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.28 - (show annotations) (download)
Wed Oct 24 08:41:23 2007 UTC (17 years, 1 month ago) by pam-fi
Branch: MAIN
Changes since 1.27: +1 -1 lines
File MIME type: text/plain
fixed bug in UpdateRunInfo(Long64_t) -- (run with 1 event were skipped)

1
2
3 /**
4 * \file PamLevel2.h
5 * \author Elena Vannuccini
6 */
7 #ifndef pamlevel2_h
8 #define pamlevel2_h
9
10 #include <TRefArray.h>
11
12 #include <TrkStruct.h>
13 #include <TrkParams.h>
14 #include <TrkLevel0.h>
15 #include <TrkLevel1.h>
16 #include <TrkLevel2.h>
17 #include <TrkHough.h>
18 #include <CaloLevel0.h>
19 #include <CaloLevel1.h>
20 #include <CaloLevel2.h>
21 #include <ToFLevel2.h>
22 #include <TrigLevel2.h>
23 #include <S4Level2.h>
24 #include <NDLevel2.h>
25 #include <AcLevel2.h>
26 #include <OrbitalInfo.h>
27 #include <RunInfo.h>
28 #include <GLTables.h>
29
30
31 #include <TTree.h>
32 #include <TFriendElement.h>
33 #include <TChain.h>
34 #include <TChainElement.h>
35 #include <TFile.h>
36 #include <TList.h>
37 #include <TKey.h>
38 #include <TSystemFile.h>
39 #include <TSystemDirectory.h>
40 #include <TSQLServer.h>
41
42 #include <stdlib.h>
43 #include <iostream>
44 #include <iomanip>
45 using namespace std;
46
47 #define NCLONES 10
48
49 using namespace std;
50
51 /**
52 * \brief Class to describe particle tracks in the apparatus, including tracker and track-related variables from other detectors (calorimeter and TOF).
53 *
54 * This class inherites from TrkTrack, CaloTrkVar ecc...
55 *
56 */
57 //class PamTrack : public TrkTrack, public CaloTrkVar, public ToFTrkVar {
58 class PamTrack : public TObject {
59
60 private:
61
62 TrkTrack* trk_track;
63 CaloTrkVar* calo_track;
64 ToFTrkVar* tof_track;
65 Int_t pscore;
66 Int_t iscore;
67 bool candeleteobj;
68
69 // TClonesArray *Cluster; ///< clusters (if stored)
70
71 public:
72
73 PamTrack();
74 PamTrack(TrkTrack* , CaloTrkVar* , ToFTrkVar* );
75 PamTrack(const PamTrack&);
76 ~PamTrack(){ Delete(); };
77
78 void Clear();
79 void Clear(Option_t *option){Clear();};
80 void Delete();
81
82 Bool_t IsSolved(){if ((pscore-iscore)>0) return true; return false;};
83 Int_t GetScore(){return (pscore-iscore);};
84 Int_t GetPScore(){return (pscore);};
85 Int_t GetIScore(){return (iscore);};
86 void SetPScore(Int_t p){pscore=p;};
87 void SetIScore(Int_t i){iscore=i;};
88
89 TrkTrack* GetTrkTrack(){return trk_track;};
90 CaloTrkVar* GetCaloTrack(){return calo_track;};
91 ToFTrkVar* GetToFTrack(){return tof_track;};
92
93 ClassDef(PamTrack,1);
94 };
95
96 /**
97 * \brief Class to describe PAMELA LEVEL2 data.
98 *
99 * This class inherites from TrkLevel2, CaloLevel2 ecc... and it overrides the TrkLevel2:GetTrack(int it) and TrkLevel2:GetTrackImage(int it) methods.
100 * This allows to solve the tracker ambiguity (that is, to choose between track images) by using also information from other detectors.
101 */
102
103 //class PamLevel2 : public CaloLevel2, public CaloLevel1, public ToFLevel2, public TrkLevel2, public TrkLevel1, public TrkHough, public TrigLevel2, public S4Level2, public NDLevel2, public AcLevel2, public OrbitalInfo {
104 class PamLevel2 : public TObject {
105
106 private:
107
108 // ------------------------------
109 // variables to retrieve run info
110 // ------------------------------
111 // Int_t irun; //current run entry, relative to the TChain
112 Long64_t irun; //current run entry, relative to the TChain
113 // ULong64_t runfirstentry; //first entry of run, relative to the TChain
114 // ULong64_t runlastentry; //last entry of run, relative to the TChain
115 // Int_t irunentry; //position of current entry, relative to the run
116 Long64_t runfirstentry; //first entry of run, relative to the TChain
117 Long64_t runlastentry; //last entry of run, relative to the TChain
118 Long64_t irunentry; //position of current entry, relative to the run
119 TChain* run_tree;
120 TTree* run_tree_clone;
121 TChain* sel_tree;
122 TTree* sel_tree_clone;
123
124 // -------------------------------------------
125 // variables to store total live and dead time
126 // -------------------------------------------
127 ULong64_t totdltime[3];// 0=live 1=dead 2=n.events
128
129 // ---------------------------------
130 // variables to retrieve level0 info
131 // ---------------------------------
132 TFile* l0_file;
133 TTree* l0_tree;
134 Int_t iroot;
135 Int_t prevshift; // to remember if any shift is needed to synchronize l0 and l2 data instead of calculating any time
136 // --------------------------------------------
137 // variables to store Level0 info (if required)
138 // --------------------------------------------
139 EventHeader* h0_obj;
140 TrkLevel0* trk0_obj;
141 CaloLevel0* calo0_obj;
142
143 // ---------------------------
144 // variables to connect the DB
145 // ---------------------------
146 TSQLServer* dbc;
147 TString host;
148 TString user;
149 TString psw;
150
151 // ---------------------------------------------
152 // variables to retrieve and clone level1-2 info
153 // ---------------------------------------------
154 TChain* pam_tree;
155 TTree* pam_tree_clone[NCLONES];
156
157 // --------------------------------------------
158 // variables to store Level0 info (if required)
159 // --------------------------------------------
160 TrkLevel1* trk1_obj;
161 TrkLevel2* trk2_obj;
162 TrkHough* trkh_obj;
163 CaloLevel1* calo1_obj;
164 CaloLevel2* calo2_obj;
165 ToFLevel2* tof_obj;
166 TrigLevel2* trig_obj;
167 S4Level2* s4_obj;
168 NDLevel2* nd_obj;
169 AcLevel2* ac_obj;
170 OrbitalInfo* orb_obj;
171 GL_RUN* run_obj;
172 SoftInfo* soft_obj; // Emiliano
173
174 // -----------------------------------
175 // variables to sort the track
176 // -----------------------------------
177 TClonesArray *tsorted;
178 TClonesArray *timage;
179 PamTrack *track; //store the last required
180 TString howtosort;
181 Float_t sortthr;
182
183 // -----
184 // flags
185 // -----
186 Bool_t CAL2; //level2 branch
187 Bool_t CAL1; //level1 branch
188 Bool_t CAL0; //yoda
189
190 Bool_t TRK2; //level2 branch
191 Bool_t TRK1; //level1 branch
192 Bool_t TRKh; //hough branch
193 Bool_t TRK0; //yoda
194
195 Bool_t TOF;
196 Bool_t TOF0;
197
198 Bool_t TRG;
199 Bool_t S4;
200 Bool_t ND;
201 Bool_t AC;
202 Bool_t ORB;
203
204 Bool_t RUN;
205
206 Bool_t ISGP; //GP simulation
207
208
209 Int_t SELLI; //trees of selected events (there is a selection list)
210
211 void Initialize();
212
213
214 public:
215
216 PamLevel2(TString ddir,TString list,TString detlist);
217 PamLevel2(TString ddir,TString list);
218 PamLevel2();
219 ~PamLevel2(){Delete();};
220
221 void Clear();
222 void Reset();
223 void Delete();
224
225
226 // =============================================
227 // methods needed to read/write Level0-1-2 trees
228 // =============================================
229 void SetWhichTrees(TString);
230 void GetWhichTrees(TFile*);
231
232 TTree* GetPamTree(TFile*,TString);
233 TTree* GetPamTree(TFile* f){return GetPamTree(f,"");};
234
235 TList* GetListOfLevel2Files(TString, TString);
236 Bool_t CheckLevel2File(TString);
237 TChain* GetPamTree(TList*,TString);
238 TChain* GetPamTree(TString ddir,TString list,TString detlist){ return GetPamTree(GetListOfLevel2Files(ddir,list),detlist); };
239 TChain* GetPamTree(TString ddir,TString list){ return GetPamTree(GetListOfLevel2Files(ddir,list),""); };
240 TChain* AddFriend(TString);
241
242 TTree* GetRunTree(TFile*);
243 TChain* GetRunTree(TList*);
244 TChain* GetRunTree(TString ddir,TString list){ return GetRunTree(GetListOfLevel2Files(ddir,list)); };
245 Bool_t UpdateRunInfo(Long64_t);//{ return UpdateRunInfo(run_tree,iev); };
246 Bool_t UpdateRunInfo(TTree *,Long64_t);
247 Bool_t UpdateRunInfo(TChain *,Long64_t);
248
249 void CreateCloneTrees0(TChain *fChain, TFile* file);
250 void CreateCloneTrees(TFile* file);
251 void CreateCloneTrees(){CreateCloneTrees(NULL);};
252 void FillCloneTrees();
253 TTree *GetCloneTree(TString);
254 void WriteCloneTrees();
255
256 void SetBranchAddress(TTree *);
257 void SetBranchAddress(TChain *);
258
259 // Int_t GetEntry(Int_t);
260 Int_t GetEntry(Long64_t);
261 Int_t GetYodaEntry();
262
263 Long64_t GetEntries(){return pam_tree->GetEntries();};
264 TChain* GetPamTree(){return pam_tree;};
265 TChain* GetRunTree(){return run_tree;};
266 TTree* GetYodaTree();
267
268 TFile* GetYodaFile(){return l0_file;}; // can be useful to know the name of level0 file we are using or similar things
269
270 // =============
271 // DB connection
272 // =============
273 void SetDBHost(TString str){host=str;};
274 void SetDBUser(TString str){user=str;};
275 void SetDBPsw(TString str){psw=str;};
276 TSQLServer* GetSQLServer(){return dbc;};
277 Bool_t SetDBConnection();
278
279 // ========================================================
280 // methods needed to retrieve subdetector general variables
281 // ========================================================
282 void* GetPointerTo(const char*);
283 EventHeader *GetEventHeader() { return h0_obj; };
284
285 /* TrkLevel0 *GetTrkLevel0() { return trk0_obj; }; */
286 /* CaloLevel0 *GetCaloLevel0() { return calo0_obj; }; */
287 TrkLevel0 *GetTrkLevel0();
288 CaloLevel0 *GetCaloLevel0();
289
290
291 TrkLevel1 *GetTrkLevel1() { return trk1_obj; };
292 TrkLevel2 *GetTrkLevel2() { return trk2_obj; };
293 TrkHough *GetTrkHough() { return trkh_obj; };
294 CaloLevel1 *GetCaloLevel1() { return calo1_obj;};
295 CaloLevel2 *GetCaloLevel2() { return calo2_obj;};
296 ToFLevel2 *GetToFLevel2() { return tof_obj; };
297 TrigLevel2 *GetTrigLevel2() { return trig_obj; };
298 S4Level2 *GetS4Level2() { return s4_obj; };
299 NDLevel2 *GetNDLevel2() { return nd_obj; };
300 AcLevel2 *GetAcLevel2() { return ac_obj; };
301 OrbitalInfo *GetOrbitalInfo(){ return orb_obj; };
302 GL_RUN *GetRunInfo() { return run_obj; };
303 SoftInfo *GetSoftInfo() { return soft_obj; }; // Emiliano
304
305 // ==============================================
306 // methods to retrieve subdetectors stored tracks
307 // ==============================================
308 TrkTrack* GetTrkStoredTrack(int seqno){return trk2_obj->TrkLevel2::GetStoredTrack(seqno); };
309 CaloTrkVar* GetCaloStoredTrack(int);
310 ToFTrkVar* GetToFStoredTrack(int);
311 //
312 PamTrack* GetPamTrackAlong(TrkTrack*);
313 // PamTrack* GetPamTrackAlong(CaloTrkVar*); //se serve
314 // PamTrack* GetPamTrackAlong(ToFTrkVar*); //se serve
315
316 // =================================
317 // methods to retrieve pamela tracks
318 // =================================
319 PamTrack* GetStoredTrack(Int_t);
320 //
321 void SortTracks(TString); //assign value to variable sorted_tracks
322 void SortTracks(); //assign value to variable sorted_tracks
323 // TRefArray* GetTracks(); //return sorted_tracks
324 TClonesArray* GetTracks(); //return sorted_tracks
325 PamTrack* GetTrack(int);
326 PamTrack* GetTrackImage(int);
327 void SetSortingMethod(TString how){ howtosort = how; return;};
328 TString GetSortingMethod(){return howtosort;};
329 void SetNucleiSortingThreshold(Float_t nucleithr){ sortthr = nucleithr; return;};
330 Float_t GetNucleiSortingThreshold(){ return sortthr;};
331
332 // =================================
333 // other methods
334 // =================================
335 // Int_t ntrk(){ return trk2_obj->TrkLevel2::ntrk(); };
336
337 // -----
338 // flags methods
339 // -----
340 Bool_t IsCAL2(){return CAL2;}; //level2 branch
341 Bool_t IsCAL1(){return CAL1;}; //level1 branch
342 Bool_t IsCAL0(){return CAL0;}; //yoda
343
344 Bool_t IsTRK2(){return TRK2;}; //level2 branch
345 Bool_t IsTRK1(){return TRK1;}; //level1 branch
346 Bool_t IsTRKh(){return TRKh;}; //hough branch
347 Bool_t IsTRK0(){return TRK0;}; //yoda
348
349 Bool_t IsTOF(){return TOF;};
350 Bool_t IsTOF0(){return TOF0;};
351
352 Bool_t IsTRG(){return TRG;};
353 Bool_t IsS4(){return S4;};
354 Bool_t IsND(){return ND;};
355 Bool_t IsAC(){return AC;};
356 Bool_t IsORB(){return ORB;};
357
358 Bool_t IsRUN(){return RUN;};
359
360 Bool_t IsGood();
361
362 Int_t IsSELLI(){return SELLI;}; //trees of selected events (there is a selection list)
363
364 Bool_t IsGP(){return ISGP;};
365
366 void SetGP(Bool_t tt){ISGP=tt;};
367
368
369 ClassDef(PamLevel2,1);
370 };
371
372 //===============================================================================================
373 //
374 //
375 //
376 //
377 //===============================================================================================
378
379
380 #endif
381

  ViewVC Help
Powered by ViewVC 1.1.23