/[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.4 by pam-fi, Fri Sep 29 10:02:30 2006 UTC revision 1.20 by mocchiut, Wed Mar 28 15:29:06 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>
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 18  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    
30    
31  #include <TTree.h>  #include <TTree.h>
32    #include <TFriendElement.h>
33  #include <TChain.h>  #include <TChain.h>
34  #include <TFile.h>  #include <TFile.h>
35  #include <TList.h>  #include <TList.h>
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 40  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 48  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 64  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 TrkLevel1, 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 {
92    class PamLevel2 : public TObject {
93    
94   private:   private:
95    
96      TrkLevel1*   trk_l1_obj;      // ------------------------------
97          TrkLevel2*   trk_obj;      // variables to retrieve run info
98          CaloLevel2*  calo_obj;      // ------------------------------
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;
144        TrkLevel2*   trk2_obj;
145        TrkHough*    trkh_obj;
146        CaloLevel1*  calo1_obj;
147        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;
155          GL_RUN*      run_obj;      SoftInfo*    soft_obj; // Emiliano
156        
157          TRefArray*   sorted_tracks; // TEMPORANEO ... lo mettero` membro di TrkLevel2      // -----------------------------------
158                // variables to sort the track
159          TChain*      Tout;      // -----------------------------------
160                TClonesArray *tsorted;
161   public:      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
170        Bool_t CAL1; //level1 branch
171        Bool_t CAL0; //yoda
172    
173        Bool_t TRK2; //level2 branch
174        Bool_t TRK1; //level1 branch
175        Bool_t TRKh; //hough branch
176        Bool_t TRK0; //yoda
177    
178        Bool_t TOF;
179        Bool_t TOF0;
180    
181        Bool_t TRG;
182        Bool_t S4;
183        Bool_t ND;
184        Bool_t AC;
185        Bool_t ORB;
186    
187          Bool_t CAL;      Bool_t RUN;
188          Bool_t TRK;  
189          Bool_t TRG;      Int_t  SELLI; //trees of selected events (there is a selection list)
190          Bool_t TOF;  
191          Bool_t S4;      void Initialize();
         Bool_t ND;  
         Bool_t AC;  
         Bool_t ORB;  
           
         Bool_t TRK_L1;  
   
     PamLevel2();  
         ~PamLevel2();  
           
         void Clear();  
192    
193          TrkLevel1*   GetTrkLevel1()  { return trk_l1_obj;  };  
194          TrkLevel2*   GetTrkLevel2()  { return trk_obj;  };   public:
195          CaloLevel2*  GetCaloLevel2() { return calo_obj; };          
196      ToFLevel2*   GetToFLevel2()  { return tof_obj;  };      PamLevel2(TString ddir,TString list,TString detlist);
197      TrigLevel2*  GetTrigLevel2() { return trig_obj; };      PamLevel2(TString ddir,TString list);
198      S4Level2*    GetS4Level2()   { return s4_obj;   };      PamLevel2();
199      NDLevel2*    GetNDLevel2()   { return nd_obj;   };      ~PamLevel2(){Delete();};
     AcLevel2*    GetAcLevel2()   { return ac_obj;   };  
         OrbitalInfo* GetOrbitalInfo(){ return orb_obj;  };  
         GL_RUN*      GetRunInfo()    { return run_obj;  };  
200                    
201  //  =====================================      void Clear();
202  //  methods needed to read Level2 trees      void Reset();
203  //  =====================================      void Delete();
204          void    SetWhichTrees(TString);  
205          TList*  GetListOfLevel2Files(TString, TString);          
206          Bool_t  CheckLevel2File(TString);  //  =============================================
207  /*      TTree*  LoadPamTrees(TFile*);  //  methods needed to read/write Level0-1-2 trees
208          TTree*  LoadPamTrees(TFile*,TString);  //  =============================================
209          TChain* LoadPamTrees(TList*,TString);      void    SetWhichTrees(TString);
210          TTree*  LoadRunTree(TFile*);      void    GetWhichTrees(TFile*);
211          TChain* LoadRunTree(TList*);*/  
212          void    SetBranchAddress();      TTree*  GetPamTree(TFile*,TString);
213          TTree*  GetPamTree(TFile*,TString);      TTree*  GetPamTree(TFile* f){return GetPamTree(f,"");};
214          TTree*  GetPamTree(TFile* f){return GetPamTree(f,"+ALL");};  
215          TChain* GetPamTree(TList*,TString);      TList*  GetListOfLevel2Files(TString, TString);
216          TTree*  GetRunTree(TFile*);      Bool_t  CheckLevel2File(TString);
217          TChain* GetRunTree(TList*);      TChain* GetPamTree(TList*,TString);
218        TChain* GetPamTree(TString ddir,TString list,TString detlist){ return GetPamTree(GetListOfLevel2Files(ddir,list),detlist); };
219        TChain* GetPamTree(TString ddir,TString list){ return GetPamTree(GetListOfLevel2Files(ddir,list),""); };
220    
221        TTree*  GetRunTree(TFile*);
222        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);
229        void    CreateCloneTrees(TFile* file);
230        void    FillCloneTrees();
231        TTree  *GetCloneTree(TString);
232        void    WriteCloneTrees();
233    
234        void    SetBranchAddress(TTree *);
235        void    SetBranchAddress(TChain *);
236    
237    //    Int_t   GetEntry(Int_t);
238        Int_t   GetEntry(Long64_t);
239        Int_t   GetYodaEntry();
240            
241  //  =====================================      TChain* GetPamTree(){return pam_tree;};
242  //  methods needed to set brach addresses      TChain* GetRunTree(){return run_tree;};
243  //  =====================================      TTree*  GetYodaTree();
244          void*       GetPointerToTrk()  {return &trk_obj;};  
245          void*       GetPointerToTrk(int s){      TFile*  GetYodaFile(){return l0_file;}; // can be useful to know the name of level0 file we are using or similar things
246                  switch (s) {  
247                          case 1:  return &trk_l1_obj;  //  =============
248                          case 2:  return &trk_obj;  //  DB connection
249                          default: return 0;  //  =============
250                  };      void SetDBHost(TString str){host=str;};
251          };      void SetDBUser(TString str){user=str;};
252          void*       GetPointerToCalo() {return &calo_obj; };      void SetDBPsw(TString str){psw=str;};
253          void*       GetPointerToToF()  {return &tof_obj;  };  
254          void*       GetPointerToTrig() {return &trig_obj; };  
255          void*       GetPointerToS4()   {return &s4_obj;   };  //  ========================================================
256          void*       GetPointerToND()   {return &nd_obj;   };  //  methods needed to retrieve subdetector general variables
257          void*       GetPointerToAc()   {return &ac_obj;   };  //  ========================================================
258          void*       GetPointerToOrb()  {return &orb_obj;  };      void*       GetPointerTo(const char*);
259          void*       GetPointerToRun()  {return &run_obj;  };      EventHeader  *GetEventHeader()  { return h0_obj; };
260          void*       GetPointerTo(const char*);      TrkLevel0    *GetTrkLevel0()  { return trk0_obj; };
261        CaloLevel0   *GetCaloLevel0()  { return calo0_obj; };
262    
263        TrkLevel1    *GetTrkLevel1()  { return trk1_obj; };
264        TrkLevel2    *GetTrkLevel2()  { return trk2_obj; };
265        TrkHough     *GetTrkHough()   { return trkh_obj; };
266        CaloLevel1   *GetCaloLevel1() { return calo1_obj;};
267        CaloLevel2   *GetCaloLevel2() { return calo2_obj;};
268        ToFLevel2    *GetToFLevel2()  { return tof_obj;  };
269        TrigLevel2   *GetTrigLevel2() { return trig_obj; };
270        S4Level2     *GetS4Level2()   { return s4_obj;   };
271        NDLevel2     *GetNDLevel2()   { return nd_obj;   };
272        AcLevel2     *GetAcLevel2()   { return ac_obj;   };
273        OrbitalInfo  *GetOrbitalInfo(){ return orb_obj;  };
274        GL_RUN       *GetRunInfo()    { return run_obj;  };
275        SoftInfo     *GetSoftInfo()   { return soft_obj; }; // Emiliano
276    
277  //  ==============================================  //  ==============================================
278  //  methods to retrieve subdetectors stored tracks  //  methods to retrieve subdetectors stored tracks
279  //  ==============================================  //  ==============================================
280          TrkTrack*   GetTrkStoredTrack(int seqno){return TrkLevel2::GetStoredTrack(seqno); };      TrkTrack*   GetTrkStoredTrack(int seqno){return trk2_obj->TrkLevel2::GetStoredTrack(seqno); };
281          CaloTrkVar* GetCaloStoredTrack(int);      CaloTrkVar* GetCaloStoredTrack(int);
282          ToFTrkVar*  GetToFStoredTrack(int);      ToFTrkVar*  GetToFStoredTrack(int);
283          //      //
284          PamTrack*   GetPamTrackAlong(TrkTrack*);      PamTrack*   GetPamTrackAlong(TrkTrack*);
285  //      PamTrack*   GetPamTrackAlong(CaloTrkVar*);      //se serve  //      PamTrack*   GetPamTrackAlong(CaloTrkVar*);      //se serve
286  //      PamTrack*   GetPamTrackAlong(ToFTrkVar*);      //se serve  //      PamTrack*   GetPamTrackAlong(ToFTrkVar*);      //se serve
287                    
# Line 167  class PamLevel2 : public CaloLevel2, pub Line 289  class PamLevel2 : public CaloLevel2, pub
289  //  methods to retrieve pamela tracks  //  methods to retrieve pamela tracks
290  //  =================================  //  =================================
291      PamTrack*     GetStoredTrack(Int_t);      PamTrack*     GetStoredTrack(Int_t);
292          //      //
293          void          SortTracks(TString);          //assign value to variable sorted_tracks      void          SortTracks(TString);          //assign value to variable sorted_tracks
294          TRefArray*    GetTracks();                  //return sorted_tracks      void          SortTracks();          //assign value to variable sorted_tracks
295          PamTrack*     GetTrack(int);  //    TRefArray*    GetTracks();                  //return sorted_tracks
296        TClonesArray* GetTracks();                  //return sorted_tracks
297        PamTrack*     GetTrack(int);
298      PamTrack*     GetTrackImage(int);      PamTrack*     GetTrackImage(int);
299        void          SetSortingMethod(TString how){ howtosort = how; return;};
300        TString       GetSortingMethod(){return howtosort;};
301        void          SetNucleiSortingThreshold(Float_t nucleithr){ sortthr = nucleithr; return;};
302        Float_t       GetNucleiSortingThreshold(){ return sortthr;};
303                    
304    //  =================================
305    //  other methods
306    //  =================================
307    //    Int_t ntrk(){ return trk2_obj->TrkLevel2::ntrk(); };
308    
309        // -----
310        // flags methods
311        // -----
312        Bool_t IsCAL2(){return CAL2;}; //level2 branch
313        Bool_t IsCAL1(){return CAL1;}; //level1 branch
314        Bool_t IsCAL0(){return CAL0;}; //yoda
315    
316        Bool_t IsTRK2(){return TRK2;}; //level2 branch
317        Bool_t IsTRK1(){return TRK1;}; //level1 branch
318        Bool_t IsTRKh(){return TRKh;}; //hough branch
319        Bool_t IsTRK0(){return TRK0;}; //yoda
320    
321        Bool_t IsTOF(){return TOF;};
322        Bool_t IsTOF0(){return TOF0;};
323    
324        Bool_t IsTRG(){return TRG;};
325        Bool_t IsS4(){return S4;};
326        Bool_t IsND(){return ND;};
327        Bool_t IsAC(){return AC;};
328        Bool_t IsORB(){return ORB;};
329    
330        Bool_t IsRUN(){return RUN;};
331    
332        Int_t  IsSELLI(){return SELLI;}; //trees of selected events (there is a selection list)
333    
334      ClassDef(PamLevel2,1);      ClassDef(PamLevel2,1);
335  };  };
336    
Legend:
Removed from v.1.4  
changed lines
  Added in v.1.20
  ViewVC Help
Powered by ViewVC 1.1.23