/[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.14 by pamela, Sun Feb 25 16:26:55 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 <CaloLevel1.h>
19  #include <CaloLevel2.h>  #include <CaloLevel2.h>
20  #include <ToFLevel2.h>  #include <ToFLevel2.h>
21  #include <TrigLevel2.h>  #include <TrigLevel2.h>
# Line 17  Line 23 
23  #include <NDLevel2.h>  #include <NDLevel2.h>
24  #include <AcLevel2.h>  #include <AcLevel2.h>
25  #include <OrbitalInfo.h>  #include <OrbitalInfo.h>
26    #include <RunInfo.h>
27    #include <GLTables.h>
28    
29    
30  #include <TTree.h>  #include <TTree.h>
31    #include <TFriendElement.h>
32  #include <TChain.h>  #include <TChain.h>
33  #include <TFile.h>  #include <TFile.h>
34  #include <TList.h>  #include <TList.h>
35  #include <TKey.h>  #include <TKey.h>
36  #include <TSystemFile.h>  #include <TSystemFile.h>
37  #include <TSystemDirectory.h>  #include <TSystemDirectory.h>
38    #include <TSQLServer.h>
39    
40  #include <iostream>  #include <iostream>
41    
42    #define NCLONES 10
43    
44  using namespace std;  using namespace std;
45    
46  /**  /**
# Line 35  using namespace std; Line 49  using namespace std;
49   * This class inherites from TrkTrack, CaloTrkVar ecc...   * This class inherites from TrkTrack, CaloTrkVar ecc...
50   *   *
51   */   */
52  class PamTrack : public TrkTrack, public CaloTrkVar, public ToFTrkVar {  //class PamTrack : public TrkTrack, public CaloTrkVar, public ToFTrkVar {
53    class PamTrack : public TObject {
54    
55   private:   private:
56    
# Line 43  class PamTrack : public TrkTrack, public Line 58  class PamTrack : public TrkTrack, public
58      CaloTrkVar* calo_track;      CaloTrkVar* calo_track;
59      ToFTrkVar*  tof_track;      ToFTrkVar*  tof_track;
60    
61        bool candeleteobj;
62    
63    //    TClonesArray *Cluster;        ///< clusters (if stored)
64    
65   public:   public:
66    
67      PamTrack();      PamTrack();
   
68      PamTrack(TrkTrack* , CaloTrkVar* , ToFTrkVar* );      PamTrack(TrkTrack* , CaloTrkVar* , ToFTrkVar* );
69        PamTrack(const PamTrack&);
70        ~PamTrack(){ Delete(); };
71    
72        void Clear();
73        void Clear(Option_t *option){Clear();};
74        void Delete();
75    
76        TrkTrack*   GetTrkTrack(){return trk_track;};
77        CaloTrkVar* GetCaloTrack(){return calo_track;};
78        ToFTrkVar*  GetToFTrack(){return tof_track;};
79    
80      ClassDef(PamTrack,1);      ClassDef(PamTrack,1);
81  };  };
# Line 59  class PamTrack : public TrkTrack, public Line 87  class PamTrack : public TrkTrack, public
87   * 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.
88   */   */
89                                    
90  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 {
91    class PamLevel2 : public TObject {
92    
93   private:   private:
94    
95      TrkLevel2*   trk_obj;      // ------------------------------
96      CaloLevel2*  calo_obj;      // variables to retrieve run info
97        // ------------------------------
98        Int_t     irun;          //current run entry, relative to the TChain
99        ULong64_t runfirstentry; //first entry of run, relative to the TChain
100        ULong64_t runlastentry;  //last  entry of run, relative to the TChain
101        Int_t     irunentry;     //position of current entry, relative to the run
102        TChain*   run_tree;
103        TTree*    run_tree_clone;
104        TChain*   sel_tree;
105        TTree*    sel_tree_clone;
106        
107        // ---------------------------------
108        // variables to retrieve level0 info
109        // ---------------------------------
110        TFile*      l0_file;
111        TTree*      l0_tree;
112        Int_t       iroot;      
113        // --------------------------------------------
114        // variables to store Level0 info (if required)
115        // --------------------------------------------
116        EventHeader* h0_obj;
117        TrkLevel0*   trk0_obj;
118    
119        // ---------------------------
120        // variables to connect the DB
121        // ---------------------------
122        TSQLServer* dbc;
123        TString host;
124        TString user;
125        TString psw;
126    
127        // ---------------------------------------------
128        // variables to retrieve and clone level1-2 info
129        // ---------------------------------------------
130        TChain* pam_tree;
131        TTree*  pam_tree_clone[NCLONES];
132        // --------------------------------------------
133        // variables to store Level0 info (if required)
134        // --------------------------------------------
135        TrkLevel1*   trk1_obj;
136        TrkLevel2*   trk2_obj;
137        TrkHough*    trkh_obj;
138        CaloLevel1*  calo1_obj;
139        CaloLevel2*  calo2_obj;
140      ToFLevel2*   tof_obj;      ToFLevel2*   tof_obj;
141      TrigLevel2*  trig_obj;      TrigLevel2*  trig_obj;
142      S4Level2*    s4_obj;      S4Level2*    s4_obj;
143      NDLevel2*    nd_obj;      NDLevel2*    nd_obj;
144      AcLevel2*    ac_obj;      AcLevel2*    ac_obj;
145          OrbitalInfo* orb_obj;      OrbitalInfo* orb_obj;
146            GL_RUN*      run_obj;
147          TRefArray*   sorted_tracks; // TEMPORANEO ... lo mettero` menbro di TrkLevel2      SoftInfo*    soft_obj; // Emiliano
148            
149   public:      // -----------------------------------
150        // variables to sort the track
151        // -----------------------------------
152        TClonesArray *tsorted;
153        TClonesArray *timage;
154        PamTrack    *track; //store the last required
155    
156    
157        // -----
158        // flags
159        // -----
160        Bool_t CAL2; //level2 branch
161        Bool_t CAL1; //level1 branch
162        Bool_t CAL0; //yoda
163    
164        Bool_t TRK2; //level2 branch
165        Bool_t TRK1; //level1 branch
166        Bool_t TRKh; //hough branch
167        Bool_t TRK0; //yoda
168    
169        Bool_t TOF;
170        Bool_t TOF0;
171    
172        Bool_t TRG;
173        Bool_t S4;
174        Bool_t ND;
175        Bool_t AC;
176        Bool_t ORB;
177    
178        Bool_t RUN;
179    
180          Bool_t CAL;      Int_t  SELLI; //trees of selected events (there is a selection list)
181          Bool_t TRK;  
182          Bool_t TRG;  
183          Bool_t TOF;   public:
184          Bool_t S4;          
         Bool_t ND;  
         Bool_t AC;  
         Bool_t ORB;  
   
185      PamLevel2();      PamLevel2();
186          ~PamLevel2();      ~PamLevel2(){Delete();};
187                    
188          void Clear();      void Clear();
189        void Delete();
190    
191      TrkLevel2*   GetTrkLevel2()  { return trk_obj;  };          
192      CaloLevel2*  GetCaloLevel2() { return calo_obj; };  //  =============================================
193      ToFLevel2*   GetToFLevel2()  { return tof_obj;  };  //  methods needed to read/write Level0-1-2 trees
194      TrigLevel2*  GetTrigLevel2() { return trig_obj; };  //  =============================================
195      S4Level2*    GetS4Level2()   { return s4_obj;   };      void    SetWhichTrees(TString);
196      NDLevel2*    GetNDLevel2()   { return nd_obj;   };      void    GetWhichTrees(TFile*);
197      AcLevel2*    GetAcLevel2()   { return ac_obj;   };  
198          OrbitalInfo* GetOrbitalInfo(){ return orb_obj;  };      TTree*  GetPamTree(TFile*,TString);
199        TTree*  GetPamTree(TFile* f){return GetPamTree(f,"");};
200  //  =====================================  
201  //  methods needed to read Level2 trees      TList*  GetListOfLevel2Files(TString, TString);
202  //  =====================================      Bool_t  CheckLevel2File(TString);
203          void    SetWhichTrees(TString);      TChain* GetPamTree(TList*,TString);
204          TList*  GetListOfLevel2Files(TString, TString);      TChain* GetPamTree(TString ddir,TString list,TString detlist){ return GetPamTree(GetListOfLevel2Files(ddir,list),detlist); };
205          Bool_t  CheckLevel2File(TString);      TChain* GetPamTree(TString ddir,TString list){ return GetPamTree(GetListOfLevel2Files(ddir,list),""); };
206          TTree*  LoadPamTrees(TFile*);  
207          TTree*  LoadPamTrees(TFile*,TString);      TTree*  GetRunTree(TFile*);
208          TChain* LoadPamTrees(TList*,TString);      TChain* GetRunTree(TList*);
209            TChain* GetRunTree(TString ddir,TString list){ return GetRunTree(GetListOfLevel2Files(ddir,list)); };
210  //  =====================================      Bool_t  UpdateRunInfo(TTree *,ULong64_t);
211  //  methods needed to set brach addresses      Bool_t  UpdateRunInfo(TChain *,ULong64_t);
212  //  =====================================  
213          void*       GetPointerToTrk()  {return &trk_obj;};      void    CreateCloneTrees0(TChain *fChain, TFile* file);
214          void*       GetPointerToCalo() {return &calo_obj;};      void    CreateCloneTrees(TFile* file);
215          void*       GetPointerToToF()  {return &tof_obj;};      void    FillCloneTrees();
216          void*       GetPointerToTrig() {return &trig_obj;};      TTree  *GetCloneTree(TString);
217          void*       GetPointerToS4()   {return &s4_obj;};      void    WriteCloneTrees();
218          void*       GetPointerToND()   {return &nd_obj;};  
219          void*       GetPointerToAc()   {return &ac_obj;};      void    SetBranchAddress(TTree *);
220          void*       GetPointerToOrb()  {return &orb_obj;};      void    SetBranchAddress(TChain *);
221        
222        Int_t   GetEntry(Int_t);
223        Int_t   GetYodaEntry();
224    
225        TChain* GetPamTree(){return pam_tree;};
226        TChain* GetRunTree(){return run_tree;};
227        TTree*  GetYodaTree();
228    
229    //  =============
230    //  DB connection
231    //  =============
232        void SetDBHost(TString str){host=str;};
233        void SetDBUser(TString str){user=str;};
234        void SetDBPsw(TString str){psw=str;};
235    
236    
237    //  ========================================================
238    //  methods needed to retrieve subdetector general variables
239    //  ========================================================
240        void*       GetPointerTo(const char*);
241        TrkLevel0    *GetTrkLevel0()  { return trk0_obj; };
242    
243        TrkLevel1    *GetTrkLevel1()  { return trk1_obj; };
244        TrkLevel2    *GetTrkLevel2()  { return trk2_obj; };
245        TrkHough     *GetTrkHough()   { return trkh_obj; };
246        CaloLevel1   *GetCaloLevel1() { return calo1_obj;};
247        CaloLevel2   *GetCaloLevel2() { return calo2_obj;};
248        ToFLevel2    *GetToFLevel2()  { return tof_obj;  };
249        TrigLevel2   *GetTrigLevel2() { return trig_obj; };
250        S4Level2     *GetS4Level2()   { return s4_obj;   };
251        NDLevel2     *GetNDLevel2()   { return nd_obj;   };
252        AcLevel2     *GetAcLevel2()   { return ac_obj;   };
253        OrbitalInfo  *GetOrbitalInfo(){ return orb_obj;  };
254        GL_RUN       *GetRunInfo()    { return run_obj;  };
255        SoftInfo     *GetSoftInfo()   { return soft_obj; }; // Emiliano
256    
257    //  ==============================================
258    //  methods to retrieve subdetectors stored tracks
259    //  ==============================================
260        TrkTrack*   GetTrkStoredTrack(int seqno){return trk2_obj->TrkLevel2::GetStoredTrack(seqno); };
261        CaloTrkVar* GetCaloStoredTrack(int);
262        ToFTrkVar*  GetToFStoredTrack(int);
263        //
264        PamTrack*   GetPamTrackAlong(TrkTrack*);
265    //      PamTrack*   GetPamTrackAlong(CaloTrkVar*);      //se serve
266    //      PamTrack*   GetPamTrackAlong(ToFTrkVar*);      //se serve
267            
268  //  =================================  //  =================================
269  //  methods to retrieve pamela tracks  //  methods to retrieve pamela tracks
270  //  =================================  //  =================================
271      PamTrack*     GetStoredTrack(Int_t itrk);      PamTrack*     GetStoredTrack(Int_t);
272          void          SortTracks();                 //assign value to variable sorted_tracks      //
273          TRefArray*    GetTracks();                  //return sorted_tracks      void          SortTracks(TString);          //assign value to variable sorted_tracks
274          PamTrack*     GetTrack(int);  //    TRefArray*    GetTracks();                  //return sorted_tracks
275        TClonesArray* GetTracks();                  //return sorted_tracks
276        PamTrack*     GetTrack(int);
277      PamTrack*     GetTrackImage(int);      PamTrack*     GetTrackImage(int);
278                    
279    //  =================================
280    //  other methods
281    //  =================================
282    //    Int_t ntrk(){ return trk2_obj->TrkLevel2::ntrk(); };
283    
284        // -----
285        // flags methods
286        // -----
287        Bool_t IsCAL2(){return CAL2;}; //level2 branch
288        Bool_t IsCAL1(){return CAL1;}; //level1 branch
289        Bool_t IsCAL0(){return CAL0;}; //yoda
290    
291        Bool_t IsTRK2(){return TRK2;}; //level2 branch
292        Bool_t IsTRK1(){return TRK1;}; //level1 branch
293        Bool_t IsTRKh(){return TRKh;}; //hough branch
294        Bool_t IsTRK0(){return TRK0;}; //yoda
295    
296        Bool_t IsTOF(){return TOF;};
297        Bool_t IsTOF0(){return TOF0;};
298    
299        Bool_t IsTRG(){return TRG;};
300        Bool_t IsS4(){return S4;};
301        Bool_t IsND(){return ND;};
302        Bool_t IsAC(){return AC;};
303        Bool_t IsORB(){return ORB;};
304    
305        Bool_t IsRUN(){return RUN;};
306    
307        Int_t  IsSELLI(){return SELLI;}; //trees of selected events (there is a selection list)
308    
309      ClassDef(PamLevel2,1);      ClassDef(PamLevel2,1);
310  };  };
311    
Legend:
Removed from v.1.2  
changed lines
  Added in v.1.14
  ViewVC Help
Powered by ViewVC 1.1.23