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

Diff of /PamelaLevel2/inc/PamLevel2.h

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

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