/[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.1 by pam-fi, Fri Jun 16 16:43:55 2006 UTC revision 1.16 by mocchiut, Sun Mar 18 20:25:53 2007 UTC
# Line 7  Line 7 
7  #ifndef pamlevel2_h  #ifndef pamlevel2_h
8  #define pamlevel2_h  #define pamlevel2_h
9    
10  #include <TClonesArray.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>
23  #include <S4Level2.h>  #include <S4Level2.h>
24  #include <NDLevel2.h>  #include <NDLevel2.h>
25  #include <AcLevel2.h>  #include <AcLevel2.h>
26    #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>
34  #include <TFile.h>  #include <TFile.h>
35    #include <TList.h>
36    #include <TKey.h>
37    #include <TSystemFile.h>
38    #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 29  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 37  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;};
78        CaloTrkVar* GetCaloTrack(){return calo_track;};
79        ToFTrkVar*  GetToFTrack(){return tof_track;};
80    
81      ClassDef(PamTrack,1);      ClassDef(PamTrack,1);
82  };  };
# Line 53  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 TrkLevel2, public ToFLevel2, public TrigLevel2, public S4Level2, public NDLevel2, public AcLevel2 {  //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:
95    
96      TrkLevel2*  trk_obj;      // ------------------------------
97      CaloLevel2* calo_obj;      // variables to retrieve run info
98      ToFLevel2*  tof_obj;      // ------------------------------
99      TrigLevel2* trig_obj;      Int_t     irun;          //current run entry, relative to the TChain
100      S4Level2*   s4_obj;      ULong64_t runfirstentry; //first entry of run, relative to the TChain
101      NDLevel2*   nd_obj;      ULong64_t runlastentry;  //last  entry of run, relative to the TChain
102      AcLevel2*   ac_obj;      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   public:      // ---------------------------------
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;
139        TrkLevel2*   trk2_obj;
140        TrkHough*    trkh_obj;
141        CaloLevel1*  calo1_obj;
142        CaloLevel2*  calo2_obj;
143        ToFLevel2*   tof_obj;
144        TrigLevel2*  trig_obj;
145        S4Level2*    s4_obj;
146        NDLevel2*    nd_obj;
147        AcLevel2*    ac_obj;
148        OrbitalInfo* orb_obj;
149        GL_RUN*      run_obj;
150        SoftInfo*    soft_obj; // Emiliano
151    
152        // -----------------------------------
153        // variables to sort the track
154        // -----------------------------------
155        TClonesArray *tsorted;
156        TClonesArray *timage;
157        PamTrack    *track; //store the last required
158    
159    
160        // -----
161        // flags
162        // -----
163        Bool_t CAL2; //level2 branch
164        Bool_t CAL1; //level1 branch
165        Bool_t CAL0; //yoda
166    
167        Bool_t TRK2; //level2 branch
168        Bool_t TRK1; //level1 branch
169        Bool_t TRKh; //hough branch
170        Bool_t TRK0; //yoda
171    
172        Bool_t TOF;
173        Bool_t TOF0;
174    
175        Bool_t TRG;
176        Bool_t S4;
177        Bool_t ND;
178        Bool_t AC;
179        Bool_t ORB;
180    
181        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();};
190            
191        void Clear();
192        void Delete();
193    
194            
195    //  =============================================
196    //  methods needed to read/write Level0-1-2 trees
197    //  =============================================
198        void    SetWhichTrees(TString);
199        void    GetWhichTrees(TFile*);
200    
201        TTree*  GetPamTree(TFile*,TString);
202        TTree*  GetPamTree(TFile* f){return GetPamTree(f,"");};
203    
204        TList*  GetListOfLevel2Files(TString, TString);
205        Bool_t  CheckLevel2File(TString);
206        TChain* GetPamTree(TList*,TString);
207        TChain* GetPamTree(TString ddir,TString list,TString detlist){ return GetPamTree(GetListOfLevel2Files(ddir,list),detlist); };
208        TChain* GetPamTree(TString ddir,TString list){ return GetPamTree(GetListOfLevel2Files(ddir,list),""); };
209    
210        TTree*  GetRunTree(TFile*);
211        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);
217        void    CreateCloneTrees(TFile* file);
218        void    FillCloneTrees();
219        TTree  *GetCloneTree(TString);
220        void    WriteCloneTrees();
221    
222      TrkLevel2*  GetTrkLevel2()  {return trk_obj;};      void    SetBranchAddress(TTree *);
223      CaloLevel2* GetCaloLevel2() {return calo_obj;};      void    SetBranchAddress(TChain *);
     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;};  
       
 //  =================================  
 //  methods needed to ... well I am not shure they are needed  
 //  =================================  
     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;};  
224    
225      TTree* LoadPamTrees(TFile*);      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
244    //  ========================================================
245        void*       GetPointerTo(const char*);
246        EventHeader  *GetEventHeader()  { return h0_obj; };
247        TrkLevel0    *GetTrkLevel0()  { return trk0_obj; };
248        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;};
254        CaloLevel2   *GetCaloLevel2() { return calo2_obj;};
255        ToFLevel2    *GetToFLevel2()  { return tof_obj;  };
256        TrigLevel2   *GetTrigLevel2() { return trig_obj; };
257        S4Level2     *GetS4Level2()   { return s4_obj;   };
258        NDLevel2     *GetNDLevel2()   { return nd_obj;   };
259        AcLevel2     *GetAcLevel2()   { return ac_obj;   };
260        OrbitalInfo  *GetOrbitalInfo(){ return orb_obj;  };
261        GL_RUN       *GetRunInfo()    { return run_obj;  };
262        SoftInfo     *GetSoftInfo()   { return soft_obj; }; // Emiliano
263    
264    //  ==============================================
265    //  methods to retrieve subdetectors stored tracks
266    //  ==============================================
267        TrkTrack*   GetTrkStoredTrack(int seqno){return trk2_obj->TrkLevel2::GetStoredTrack(seqno); };
268        CaloTrkVar* GetCaloStoredTrack(int);
269        ToFTrkVar*  GetToFStoredTrack(int);
270        //
271        PamTrack*   GetPamTrackAlong(TrkTrack*);
272    //      PamTrack*   GetPamTrackAlong(CaloTrkVar*);      //se serve
273    //      PamTrack*   GetPamTrackAlong(ToFTrkVar*);      //se serve
274            
275  //  =================================  //  =================================
276  //  methods to retrieve pamela tracks  //  methods to retrieve pamela tracks
277  //  =================================  //  =================================
278      PamTrack*     GetStoredTrack(Int_t itrk);      PamTrack*     GetStoredTrack(Int_t);
279      TClonesArray* GetTracks();      //
280        void          SortTracks(TString);          //assign value to variable sorted_tracks
281    //    TRefArray*    GetTracks();                  //return sorted_tracks
282        TClonesArray* GetTracks();                  //return sorted_tracks
283      PamTrack*     GetTrack(int);      PamTrack*     GetTrack(int);
284      PamTrack*     GetTrackImage(int);      PamTrack*     GetTrackImage(int);
285            
286    //  =================================
287    //  other methods
288    //  =================================
289    //    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  };  };
318    
Legend:
Removed from v.1.1  
changed lines
  Added in v.1.16
  ViewVC Help
Powered by ViewVC 1.1.23