/[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.13 by pam-fi, Fri Feb 16 15:00:35 2007 UTC revision 1.27 by mocchiut, Wed Sep 12 09:25:21 2007 UTC
# Line 15  Line 15 
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 30  Line 31 
31  #include <TTree.h>  #include <TTree.h>
32  #include <TFriendElement.h>  #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>
# Line 37  Line 39 
39  #include <TSystemDirectory.h>  #include <TSystemDirectory.h>
40  #include <TSQLServer.h>  #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  #define NCLONES 10
48    
# Line 57  class PamTrack : public TObject { Line 62  class PamTrack : public TObject {
62      TrkTrack*   trk_track;      TrkTrack*   trk_track;
63      CaloTrkVar* calo_track;      CaloTrkVar* calo_track;
64      ToFTrkVar*  tof_track;      ToFTrkVar*  tof_track;
65        Int_t pscore;
66        Int_t iscore;
67      bool candeleteobj;      bool candeleteobj;
68    
69  //    TClonesArray *Cluster;        ///< clusters (if stored)  //    TClonesArray *Cluster;        ///< clusters (if stored)
# Line 73  class PamTrack : public TObject { Line 79  class PamTrack : public TObject {
79      void Clear(Option_t *option){Clear();};      void Clear(Option_t *option){Clear();};
80      void Delete();      void Delete();
81    
82        Bool_t IsSolved(){if ((pscore-iscore)>0) return true; return false;};
83        Int_t GetScore(){return (pscore-iscore);};
84        Int_t GetPScore(){return (pscore);};
85        Int_t GetIScore(){return (iscore);};
86        void SetPScore(Int_t p){pscore=p;};
87        void SetIScore(Int_t i){iscore=i;};
88    
89      TrkTrack*   GetTrkTrack(){return trk_track;};      TrkTrack*   GetTrkTrack(){return trk_track;};
90      CaloTrkVar* GetCaloTrack(){return calo_track;};      CaloTrkVar* GetCaloTrack(){return calo_track;};
91      ToFTrkVar*  GetToFTrack(){return tof_track;};      ToFTrkVar*  GetToFTrack(){return tof_track;};
# Line 95  class PamLevel2 : public TObject { Line 108  class PamLevel2 : public TObject {
108      // ------------------------------      // ------------------------------
109      // variables to retrieve run info      // variables to retrieve run info
110      // ------------------------------      // ------------------------------
111      Int_t     irun;          //current run entry, relative to the TChain  //    Int_t     irun;          //current run entry, relative to the TChain
112      ULong64_t runfirstentry; //first entry of run, relative to the TChain      Long64_t  irun;          //current run entry, relative to the TChain
113      ULong64_t runlastentry;  //last  entry of run, relative to the TChain  //    ULong64_t runfirstentry; //first entry of run, relative to the TChain
114      Int_t     irunentry;     //position of current entry, relative to the run  //    ULong64_t runlastentry;  //last  entry of run, relative to the TChain
115    //    Int_t     irunentry;     //position of current entry, relative to the run
116        Long64_t  runfirstentry; //first entry of run, relative to the TChain
117        Long64_t  runlastentry;  //last  entry of run, relative to the TChain
118        Long64_t  irunentry;     //position of current entry, relative to the run
119      TChain*   run_tree;      TChain*   run_tree;
120      TTree*    run_tree_clone;      TTree*    run_tree_clone;
121      TChain*   sel_tree;      TChain*   sel_tree;
122      TTree*    sel_tree_clone;      TTree*    sel_tree_clone;
123        
124        // -------------------------------------------
125        // variables to store total live and dead time
126        // -------------------------------------------
127        ULong64_t  totdltime[3];// 0=live 1=dead 2=n.events  
128    
129      // ---------------------------------      // ---------------------------------
130      // variables to retrieve level0 info      // variables to retrieve level0 info
131      // ---------------------------------      // ---------------------------------
132      TFile*      l0_file;      TFile*      l0_file;
133      TTree*      l0_tree;      TTree*      l0_tree;
134      Int_t       iroot;            Int_t       iroot;      
135        Int_t       prevshift; // to remember if any shift is needed to synchronize l0 and l2 data instead of calculating any time
136      // --------------------------------------------      // --------------------------------------------
137      // variables to store Level0 info (if required)      // variables to store Level0 info (if required)
138      // --------------------------------------------      // --------------------------------------------
139      EventHeader* h0_obj;      EventHeader* h0_obj;
140      TrkLevel0*   trk0_obj;      TrkLevel0*   trk0_obj;
141        CaloLevel0*  calo0_obj;
142    
143      // ---------------------------      // ---------------------------
144      // variables to connect the DB      // variables to connect the DB
# Line 129  class PamLevel2 : public TObject { Line 153  class PamLevel2 : public TObject {
153      // ---------------------------------------------      // ---------------------------------------------
154      TChain* pam_tree;      TChain* pam_tree;
155      TTree*  pam_tree_clone[NCLONES];      TTree*  pam_tree_clone[NCLONES];
156    
157      // --------------------------------------------      // --------------------------------------------
158      // variables to store Level0 info (if required)      // variables to store Level0 info (if required)
159      // --------------------------------------------      // --------------------------------------------
# Line 152  class PamLevel2 : public TObject { Line 177  class PamLevel2 : public TObject {
177      TClonesArray *tsorted;      TClonesArray *tsorted;
178      TClonesArray *timage;      TClonesArray *timage;
179      PamTrack    *track; //store the last required      PamTrack    *track; //store the last required
180        TString howtosort;
181        Float_t sortthr;
182    
183      // -----      // -----
184      // flags      // flags
# Line 177  class PamLevel2 : public TObject { Line 203  class PamLevel2 : public TObject {
203    
204      Bool_t RUN;      Bool_t RUN;
205    
206        Bool_t ISGP; //GP simulation
207    
208    
209      Int_t  SELLI; //trees of selected events (there is a selection list)      Int_t  SELLI; //trees of selected events (there is a selection list)
210    
211        void Initialize();
212    
213    
214   public:   public:
215                    
216        PamLevel2(TString ddir,TString list,TString detlist);
217        PamLevel2(TString ddir,TString list);
218      PamLevel2();      PamLevel2();
219      ~PamLevel2(){Delete();};      ~PamLevel2(){Delete();};
220                    
221      void Clear();      void Clear();
222        void Reset();
223      void Delete();      void Delete();
224    
225                    
# Line 196  class PamLevel2 : public TObject { Line 230  class PamLevel2 : public TObject {
230      void    GetWhichTrees(TFile*);      void    GetWhichTrees(TFile*);
231    
232      TTree*  GetPamTree(TFile*,TString);      TTree*  GetPamTree(TFile*,TString);
233      TTree*  GetPamTree(TFile* f){return GetPamTree(f,"");};      TTree*  GetPamTree(TFile* f){return GetPamTree(f,"");};    
234    
235      TList*  GetListOfLevel2Files(TString, TString);      TList*  GetListOfLevel2Files(TString, TString);
236      Bool_t  CheckLevel2File(TString);      Bool_t  CheckLevel2File(TString);
237      TChain* GetPamTree(TList*,TString);      TChain* GetPamTree(TList*,TString);
238      TChain* GetPamTree(TString ddir,TString list,TString detlist){ return GetPamTree(GetListOfLevel2Files(ddir,list),detlist); };      TChain* GetPamTree(TString ddir,TString list,TString detlist){ return GetPamTree(GetListOfLevel2Files(ddir,list),detlist); };
239      TChain* GetPamTree(TString ddir,TString list){ return GetPamTree(GetListOfLevel2Files(ddir,list),""); };      TChain* GetPamTree(TString ddir,TString list){ return GetPamTree(GetListOfLevel2Files(ddir,list),""); };
240        TChain* AddFriend(TString);
241    
242      TTree*  GetRunTree(TFile*);      TTree*  GetRunTree(TFile*);
243      TChain* GetRunTree(TList*);      TChain* GetRunTree(TList*);
244      TChain* GetRunTree(TString ddir,TString list){ return GetRunTree(GetListOfLevel2Files(ddir,list)); };      TChain* GetRunTree(TString ddir,TString list){ return GetRunTree(GetListOfLevel2Files(ddir,list)); };
245      Bool_t  UpdateRunInfo(TTree *,ULong64_t);      Bool_t  UpdateRunInfo(TTree *,Long64_t);
246      Bool_t  UpdateRunInfo(TChain *,ULong64_t);      Bool_t  UpdateRunInfo(TChain *,Long64_t);
247        Bool_t  UpdateRunInfo(Long64_t iev);//{ return UpdateRunInfo(run_tree,iev); };
248    
249      void    CreateCloneTrees0(TChain *fChain, TFile* file);      void    CreateCloneTrees0(TChain *fChain, TFile* file);
250      void    CreateCloneTrees(TFile* file);      void    CreateCloneTrees(TFile* file);
251        void    CreateCloneTrees(){CreateCloneTrees(NULL);};
252      void    FillCloneTrees();      void    FillCloneTrees();
253      TTree  *GetCloneTree(TString);      TTree  *GetCloneTree(TString);
254      void    WriteCloneTrees();      void    WriteCloneTrees();
# Line 219  class PamLevel2 : public TObject { Line 256  class PamLevel2 : public TObject {
256      void    SetBranchAddress(TTree *);      void    SetBranchAddress(TTree *);
257      void    SetBranchAddress(TChain *);      void    SetBranchAddress(TChain *);
258    
259      Int_t   GetEntry(Int_t);  //    Int_t   GetEntry(Int_t);
260        Int_t   GetEntry(Long64_t);
261      Int_t   GetYodaEntry();      Int_t   GetYodaEntry();
262    
263        Long64_t GetEntries(){return pam_tree->GetEntries();};
264      TChain* GetPamTree(){return pam_tree;};      TChain* GetPamTree(){return pam_tree;};
265      TChain* GetRunTree(){return run_tree;};      TChain* GetRunTree(){return run_tree;};
266      TTree*  GetYodaTree();      TTree*  GetYodaTree();
267    
268        TFile*  GetYodaFile(){return l0_file;}; // can be useful to know the name of level0 file we are using or similar things
269    
270  //  =============  //  =============
271  //  DB connection  //  DB connection
272  //  =============  //  =============
273      void SetDBHost(TString str){host=str;};      void SetDBHost(TString str){host=str;};
274      void SetDBUser(TString str){user=str;};      void SetDBUser(TString str){user=str;};
275      void SetDBPsw(TString str){psw=str;};      void SetDBPsw(TString str){psw=str;};
276        TSQLServer* GetSQLServer(){return dbc;};
277        Bool_t SetDBConnection();
278    
279  //  ========================================================  //  ========================================================
280  //  methods needed to retrieve subdetector general variables  //  methods needed to retrieve subdetector general variables
281  //  ========================================================  //  ========================================================
282      void*       GetPointerTo(const char*);      void*       GetPointerTo(const char*);
283      TrkLevel0    *GetTrkLevel0()  { return trk0_obj; };      EventHeader  *GetEventHeader()  { return h0_obj; };
284    
285    /*     TrkLevel0    *GetTrkLevel0()   { return trk0_obj; }; */
286    /*     CaloLevel0   *GetCaloLevel0()  { return calo0_obj; }; */
287        TrkLevel0    *GetTrkLevel0();
288        CaloLevel0   *GetCaloLevel0();
289    
290    
291      TrkLevel1    *GetTrkLevel1()  { return trk1_obj; };      TrkLevel1    *GetTrkLevel1()  { return trk1_obj; };
292      TrkLevel2    *GetTrkLevel2()  { return trk2_obj; };      TrkLevel2    *GetTrkLevel2()  { return trk2_obj; };
# Line 271  class PamLevel2 : public TObject { Line 319  class PamLevel2 : public TObject {
319      PamTrack*     GetStoredTrack(Int_t);      PamTrack*     GetStoredTrack(Int_t);
320      //      //
321      void          SortTracks(TString);          //assign value to variable sorted_tracks      void          SortTracks(TString);          //assign value to variable sorted_tracks
322        void          SortTracks();          //assign value to variable sorted_tracks
323  //    TRefArray*    GetTracks();                  //return sorted_tracks  //    TRefArray*    GetTracks();                  //return sorted_tracks
324      TClonesArray* GetTracks();                  //return sorted_tracks      TClonesArray* GetTracks();                  //return sorted_tracks
325      PamTrack*     GetTrack(int);      PamTrack*     GetTrack(int);
326      PamTrack*     GetTrackImage(int);      PamTrack*     GetTrackImage(int);
327        void          SetSortingMethod(TString how){ howtosort = how; return;};
328        TString       GetSortingMethod(){return howtosort;};
329        void          SetNucleiSortingThreshold(Float_t nucleithr){ sortthr = nucleithr; return;};
330        Float_t       GetNucleiSortingThreshold(){ return sortthr;};
331                    
332  //  =================================  //  =================================
333  //  other methods  //  other methods
334  //  =================================  //  =================================
335  //    Int_t ntrk(){ return trk2_obj->TrkLevel2::ntrk(); };  //    Int_t ntrk(){ return trk2_obj->TrkLevel2::ntrk(); };
336    
337        // -----
338        // flags methods
339        // -----
340        Bool_t IsCAL2(){return CAL2;}; //level2 branch
341        Bool_t IsCAL1(){return CAL1;}; //level1 branch
342        Bool_t IsCAL0(){return CAL0;}; //yoda
343    
344        Bool_t IsTRK2(){return TRK2;}; //level2 branch
345        Bool_t IsTRK1(){return TRK1;}; //level1 branch
346        Bool_t IsTRKh(){return TRKh;}; //hough branch
347        Bool_t IsTRK0(){return TRK0;}; //yoda
348    
349        Bool_t IsTOF(){return TOF;};
350        Bool_t IsTOF0(){return TOF0;};
351    
352        Bool_t IsTRG(){return TRG;};
353        Bool_t IsS4(){return S4;};
354        Bool_t IsND(){return ND;};
355        Bool_t IsAC(){return AC;};
356        Bool_t IsORB(){return ORB;};
357    
358        Bool_t IsRUN(){return RUN;};
359    
360        Bool_t IsGood();
361    
362        Int_t  IsSELLI(){return SELLI;}; //trees of selected events (there is a selection list)
363    
364        Bool_t IsGP(){return ISGP;};
365    
366        void   SetGP(Bool_t tt){ISGP=tt;};
367    
368    
369      ClassDef(PamLevel2,1);      ClassDef(PamLevel2,1);
370  };  };
Legend:
Removed from v.1.13  
changed lines
  Added in v.1.27
  ViewVC Help
Powered by ViewVC 1.1.23