/[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.9 by pam-fi, Wed Dec 13 09:57:28 2006 UTC revision 1.17 by pam-fi, Wed Mar 21 09:07:31 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>  #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>
# Line 20  Line 24 
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    
# Line 33  Line 36 
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  /**  /**
# Line 43  using namespace std; Line 50  using namespace std;
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    
# Line 51  class PamTrack : public TrkTrack, public Line 59  class PamTrack : public TrkTrack, public
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  };  };
# Line 71  class PamTrack : public TrkTrack, public Line 88  class PamTrack : public TrkTrack, public
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        // variables to store Level0 info (if required)
141        // --------------------------------------------
142      TrkLevel1*   trk1_obj;      TrkLevel1*   trk1_obj;
143      TrkLevel2*   trk2_obj;      TrkLevel2*   trk2_obj;
144      TrkHough*    trkh_obj;      TrkHough*    trkh_obj;
   
145      CaloLevel1*  calo1_obj;      CaloLevel1*  calo1_obj;
146      CaloLevel2*  calo2_obj;      CaloLevel2*  calo2_obj;
   
147      ToFLevel2*   tof_obj;      ToFLevel2*   tof_obj;
148      TrigLevel2*  trig_obj;      TrigLevel2*  trig_obj;
149      S4Level2*    s4_obj;      S4Level2*    s4_obj;
150      NDLevel2*    nd_obj;      NDLevel2*    nd_obj;
151      AcLevel2*    ac_obj;      AcLevel2*    ac_obj;
152      OrbitalInfo* orb_obj;      OrbitalInfo* orb_obj;
           
153      GL_RUN*      run_obj;      GL_RUN*      run_obj;
154            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 */  
155    
156      TTree*       tree_clone[8];      // -----------------------------------
157        // variables to sort the track
158        // -----------------------------------
159        TClonesArray *tsorted;
160        TClonesArray *timage;
161        PamTrack    *track; //store the last required
162    
163      TRefArray*   sorted_tracks;  
164                // -----
165        // flags
166        // -----
167      Bool_t CAL2; //level2 branch      Bool_t CAL2; //level2 branch
168      Bool_t CAL1; //level1 branch      Bool_t CAL1; //level1 branch
169        Bool_t CAL0; //yoda
170    
171      Bool_t TRK2; //level2 branch      Bool_t TRK2; //level2 branch
172      Bool_t TRK1; //level1 branch      Bool_t TRK1; //level1 branch
173      Bool_t TRKh; //hough branch      Bool_t TRKh; //hough branch
174        Bool_t TRK0; //yoda
175    
     Bool_t TRG;  
176      Bool_t TOF;      Bool_t TOF;
177        Bool_t TOF0;
178    
179        Bool_t TRG;
180      Bool_t S4;      Bool_t S4;
181      Bool_t ND;      Bool_t ND;
182      Bool_t AC;      Bool_t AC;
183      Bool_t ORB;      Bool_t ORB;
184    
185      Bool_t RUN;      Bool_t RUN;
186    
187        Int_t  SELLI; //trees of selected events (there is a selection list)
188    
189    
190     public:
191                    
192      PamLevel2();      PamLevel2();
193      ~PamLevel2();      ~PamLevel2(){Delete();};
194                    
195      void Clear();      void Clear();
196        void Delete();
197    
198  /*     TrkLevel1*   GetTrkLevel1()  { return trk1_obj;  }; */          
199  /*     TrkLevel2*   GetTrkLevel2()  { return trk2_obj;  }; */  //  =============================================
200  /*     TrkHough*    GetTrkHough()   { return trkh_obj;  }; */  //  methods needed to read/write Level0-1-2 trees
201  /*     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  
 //  =========================================  
202      void    SetWhichTrees(TString);      void    SetWhichTrees(TString);
203      void    GetWhichTrees(TFile*);      void    GetWhichTrees(TFile*);
204    
# Line 181  class PamLevel2 : public CaloLevel2, pub Line 213  class PamLevel2 : public CaloLevel2, pub
213    
214      TTree*  GetRunTree(TFile*);      TTree*  GetRunTree(TFile*);
215      TChain* GetRunTree(TList*);      TChain* GetRunTree(TList*);
216        TChain* GetRunTree(TString ddir,TString list){ return GetRunTree(GetListOfLevel2Files(ddir,list)); };
217        Bool_t  UpdateRunInfo(TTree *,Long64_t);
218        Bool_t  UpdateRunInfo(TChain *,Long64_t);
219        Bool_t  UpdateRunInfo(Long64_t iev){ return UpdateRunInfo(run_tree,iev); };
220    
221      void    CreateCloneTrees0(TChain *fChain, TFile* file);      void    CreateCloneTrees0(TChain *fChain, TFile* file);
222      void    CreateCloneTrees(TFile* file);      void    CreateCloneTrees(TFile* file);
# Line 191  class PamLevel2 : public CaloLevel2, pub Line 227  class PamLevel2 : public CaloLevel2, pub
227      void    SetBranchAddress(TTree *);      void    SetBranchAddress(TTree *);
228      void    SetBranchAddress(TChain *);      void    SetBranchAddress(TChain *);
229    
230  //  =====================================  //    Int_t   GetEntry(Int_t);
231  //  methods needed to set brach addresses      Int_t   GetEntry(Long64_t);
232  //  =====================================      Int_t   GetYodaEntry();
233  /*     void*       GetPointerToTrk(); */      
234  /*     void*       GetPointerToTrk(int s); */      TChain* GetPamTree(){return pam_tree;};
235  /*     void*       GetPointerToCalo(); */      TChain* GetRunTree(){return run_tree;};
236  /*     void*       GetPointerToCalo(int s); */      TTree*  GetYodaTree();
237  /*     void*       GetPointerToToF(); */  
238  /*     void*       GetPointerToTrig(); */      TFile*  GetYodaFile(){return l0_file;}; // can be useful to know the name of level0 file we are using or similar things
239  /*     void*       GetPointerToS4(); */  
240  /*     void*       GetPointerToND(); */  //  =============
241  /*     void*       GetPointerToAc(); */  //  DB connection
242  /*     void*       GetPointerToOrb(); */  //  =============
243  /*     void*       GetPointerToRun(); */      void SetDBHost(TString str){host=str;};
244        void SetDBUser(TString str){user=str;};
245        void SetDBPsw(TString str){psw=str;};
246    
247    
248    //  ========================================================
249    //  methods needed to retrieve subdetector general variables
250    //  ========================================================
251      void*       GetPointerTo(const char*);      void*       GetPointerTo(const char*);
252        EventHeader  *GetEventHeader()  { return h0_obj; };
253        TrkLevel0    *GetTrkLevel0()  { return trk0_obj; };
254        CaloLevel0   *GetCaloLevel0()  { return calo0_obj; };
255    
256        TrkLevel1    *GetTrkLevel1()  { return trk1_obj; };
257        TrkLevel2    *GetTrkLevel2()  { return trk2_obj; };
258        TrkHough     *GetTrkHough()   { return trkh_obj; };
259        CaloLevel1   *GetCaloLevel1() { return calo1_obj;};
260        CaloLevel2   *GetCaloLevel2() { return calo2_obj;};
261        ToFLevel2    *GetToFLevel2()  { return tof_obj;  };
262        TrigLevel2   *GetTrigLevel2() { return trig_obj; };
263        S4Level2     *GetS4Level2()   { return s4_obj;   };
264        NDLevel2     *GetNDLevel2()   { return nd_obj;   };
265        AcLevel2     *GetAcLevel2()   { return ac_obj;   };
266        OrbitalInfo  *GetOrbitalInfo(){ return orb_obj;  };
267        GL_RUN       *GetRunInfo()    { return run_obj;  };
268        SoftInfo     *GetSoftInfo()   { return soft_obj; }; // Emiliano
269    
270  //  ==============================================  //  ==============================================
271  //  methods to retrieve subdetectors stored tracks  //  methods to retrieve subdetectors stored tracks
272  //  ==============================================  //  ==============================================
273      TrkTrack*   GetTrkStoredTrack(int seqno){return TrkLevel2::GetStoredTrack(seqno); };      TrkTrack*   GetTrkStoredTrack(int seqno){return trk2_obj->TrkLevel2::GetStoredTrack(seqno); };
274      CaloTrkVar* GetCaloStoredTrack(int);      CaloTrkVar* GetCaloStoredTrack(int);
275      ToFTrkVar*  GetToFStoredTrack(int);      ToFTrkVar*  GetToFStoredTrack(int);
276      //      //
# Line 224  class PamLevel2 : public CaloLevel2, pub Line 284  class PamLevel2 : public CaloLevel2, pub
284      PamTrack*     GetStoredTrack(Int_t);      PamTrack*     GetStoredTrack(Int_t);
285      //      //
286      void          SortTracks(TString);          //assign value to variable sorted_tracks      void          SortTracks(TString);          //assign value to variable sorted_tracks
287      TRefArray*    GetTracks();                  //return sorted_tracks  //    TRefArray*    GetTracks();                  //return sorted_tracks
288        TClonesArray* GetTracks();                  //return sorted_tracks
289      PamTrack*     GetTrack(int);      PamTrack*     GetTrack(int);
290      PamTrack*     GetTrackImage(int);      PamTrack*     GetTrackImage(int);
291                    
292  //  =================================  //  =================================
293  //  other methods  //  other methods
294  //  =================================  //  =================================
295      Int_t ntrk(){ return TrkLevel2::ntrk(); };  //    Int_t ntrk(){ return trk2_obj->TrkLevel2::ntrk(); };
296    
297        // -----
298        // flags methods
299        // -----
300        Bool_t IsCAL2(){return CAL2;}; //level2 branch
301        Bool_t IsCAL1(){return CAL1;}; //level1 branch
302        Bool_t IsCAL0(){return CAL0;}; //yoda
303    
304        Bool_t IsTRK2(){return TRK2;}; //level2 branch
305        Bool_t IsTRK1(){return TRK1;}; //level1 branch
306        Bool_t IsTRKh(){return TRKh;}; //hough branch
307        Bool_t IsTRK0(){return TRK0;}; //yoda
308    
309        Bool_t IsTOF(){return TOF;};
310        Bool_t IsTOF0(){return TOF0;};
311    
312        Bool_t IsTRG(){return TRG;};
313        Bool_t IsS4(){return S4;};
314        Bool_t IsND(){return ND;};
315        Bool_t IsAC(){return AC;};
316        Bool_t IsORB(){return ORB;};
317    
318        Bool_t IsRUN(){return RUN;};
319    
320        Int_t  IsSELLI(){return SELLI;}; //trees of selected events (there is a selection list)
321    
322      ClassDef(PamLevel2,1);      ClassDef(PamLevel2,1);
323  };  };
Legend:
Removed from v.1.9  
changed lines
  Added in v.1.17
  ViewVC Help
Powered by ViewVC 1.1.23