second interation on implementing variable detector segmentation
[u/mrichter/AliRoot.git] / PWG1 / TRD / AliTRDresolution.h
1 #ifndef ALITRDRESOLUTION_H
2 #define ALITRDRESOLUTION_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id: AliTRDresolution.h 27496 2008-07-22 08:35:45Z cblume $ */
7
8 ////////////////////////////////////////////////////////////////////////////
9 //                                                                        //
10 //  TRD Resolution performance                                            //
11 //                                                                        //
12 //  Authors:                                                              //
13 //    Alexandru Bercuci <A.Bercuci@gsi.de>                                //
14 //    Markus Fasel <M.Fasel@gsi.de>                                       //
15 //                                                                        //
16 ////////////////////////////////////////////////////////////////////////////
17
18 #ifndef ALITRDRECOTASK_H
19 #include "AliTRDrecoTask.h"
20 #endif
21
22 class TH1;
23 class TH2;
24 class TF1;
25 class TGraphErrors;
26 class TObjArray;
27 class TDatabasePDG;
28 class AliTRDReconstructor;
29 class AliTRDgeometry;
30 class AliTRDrecoParam;
31 class AliTRDseedV1;
32 class AliTRDtrackInfo;
33 class AliTRDresolution : public AliTRDrecoTask
34 {
35 public:
36   enum ETRDresolutionPlot {
37      kCharge     =  0 // charge resolution
38     ,kCluster    =  1 // cluster - track
39     ,kTrack      =  2 // tracklet - track residuals/pulls
40     ,kTrackIn    =  3 // tracklet - track residuals/pulls at lower TRD entrance 
41     ,kTrackOut   =  4 // tracklet - track residuals/pulls at lower TRD entrance during refit
42     ,kMCcluster  =  5 // cluster-mc resolution/pulls
43     ,kMCtracklet =  6 // tracklet-mc resolution/pulls
44     ,kMCtrackIn  =  7 // TPC track monitor
45     ,kMCtrackOut =  8 // TOF/HMPID track monitor
46     ,kMCtrack    =  9 // TRD track monitor
47     ,kNviews     = 10 // total number of resolution views
48     ,kNprojs     = 70 // total number of projections for all views
49   };
50   enum ETRDresolutionSteer {
51     kVerbose  = 0
52     ,kVisual  = 1
53   };
54   enum ETRDresolutionOutSlots {
55      kClToTrk    = 2
56     ,kTrkltToTrk = 3
57     ,kClToMC     = 4
58     ,kTrkltToMC  = 5
59     ,kNOutSlots  = 4
60   };
61
62   AliTRDresolution();
63   AliTRDresolution(char* name);
64   virtual ~AliTRDresolution();
65   
66   void    UserCreateOutputObjects();
67   Float_t GetPtThreshold() const {return fPtThreshold;}
68   Float_t GetSegmentationLevel() {return fSegmentLevel;}
69   Bool_t  GetRefFigure(Int_t ifig);
70   TObjArray*  Histos(); 
71   Bool_t  Load(const Char_t *filename = "TRD.Performance.root");
72
73   TObjArray*  Results(Int_t i=0) const {return i ? fGraphS : fGraphM;} 
74   void    UserExec(Option_t * opt);
75   Bool_t  IsVerbose() const {return TESTBIT(fStatus, kVerbose);}
76   Bool_t  IsVisual() const {return TESTBIT(fStatus, kVisual);}
77   Bool_t  PostProcess();
78
79   TH1*    PlotCharge(const AliTRDtrackV1 *t=NULL);
80   TH1*    PlotCluster(const AliTRDtrackV1 *t=NULL);
81   TH1*    PlotTracklet(const AliTRDtrackV1 *t=NULL);
82   TH1*    PlotTrackIn(const AliTRDtrackV1 *t=NULL);
83   TH1*    PlotTrackOut(const AliTRDtrackV1 *t=NULL);
84   TH1*    PlotMC(const AliTRDtrackV1 *t=NULL);
85
86   void    SetSegmentationLevel(Int_t l=0);
87   void    SetPtThreshold(Float_t pt) {fPtThreshold = pt;}
88   void    SetRecoParam(AliTRDrecoParam *r);
89   void    SetVerbose(Bool_t v = kTRUE) {v ? SETBIT(fStatus ,kVerbose): CLRBIT(fStatus ,kVerbose);}
90   void    SetVisual(Bool_t v = kTRUE) {v ? SETBIT(fStatus, kVisual) : CLRBIT(fStatus, kVisual);}
91
92   void    Terminate(Option_t * opt);
93   Bool_t  GetGraph(Float_t *bb, ETRDresolutionPlot ip, Int_t idx=-1, Bool_t kLEG=kTRUE, const Char_t *explain=NULL);
94   Bool_t  GetGraphArray(Float_t *bb, ETRDresolutionPlot ip, Int_t idx, Bool_t kLEG=kTRUE, Int_t n=0, Int_t *sel=NULL, const Char_t *explain=NULL);
95   
96 private:
97   AliTRDresolution(const AliTRDresolution&);
98   AliTRDresolution& operator=(const AliTRDresolution&);
99
100   void    AdjustF1(TH1 *h, TF1 *f);
101   TObjArray*  BuildMonitorContainerCluster(const char* name, Bool_t expand=kFALSE);
102   TObjArray*  BuildMonitorContainerTracklet(const char* name, Bool_t expand=kFALSE);
103   TObjArray*  BuildMonitorContainerTrack(const char* name);
104   void    GetLandauMpvFwhm(TF1 * const f, Float_t &mpv, Float_t &xm, Float_t &xM);
105   Bool_t  Process(TH2* const h2, TF1 *f, Float_t k, TGraphErrors **g);
106   Bool_t  Process2D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1., Int_t gidx=-1);
107   Bool_t  Process2Darray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
108   Bool_t  Process3D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
109   Bool_t  Process3Dlinked(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
110   Bool_t  Process3DL(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
111   Bool_t  Process3Darray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
112   Bool_t  Process3DlinkedArray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
113   Bool_t  Pulls(Double_t dyz[2], Double_t cc[3], Double_t tilt);
114
115   UChar_t             fStatus;          // steer parameter of the task
116   UChar_t             fSegmentLevel;    // steer parameter of the task
117   UShort_t            fIdxPlot;         //! plot counter (internal)
118   UShort_t            fIdxFrame;        //! frame counter (internal)
119   UShort_t            fNcomp[kNprojs];  //! number of projections per task
120   Char_t              *fAxTitle[kNprojs][4]; //! Title for all ref histos
121   Float_t             fPtThreshold;     //! pt threshold for some performance plots
122   static Char_t const *fgPerformanceName[kNviews]; //! name of performance plot
123   static UChar_t const fgNproj[kNviews]; //! number of projections per task
124   static Int_t const  fgkNresYsegm[3];  //! number of segments for saving y resolution
125   static Char_t const *fgkResYsegmName[3];//! name of segment for saving y resolution
126   AliTRDReconstructor *fReconstructor;  //! local reconstructor
127   AliTRDgeometry      *fGeo;            //! TRD geometry
128   TDatabasePDG        *fDBPDG;          //! PDG database
129   TObjArray           *fGraphS;         //! result holder - sigma values
130   TObjArray           *fGraphM;         //! result holder - mean values
131
132   // calibration containers
133   TObjArray           *fCl;     //! cluster2track calib
134   TObjArray           *fTrklt;  //! tracklet2track calib
135   TObjArray           *fMCcl;   //! cluster2mc calib
136   TObjArray           *fMCtrklt;//! tracklet2mc calib
137   
138   ClassDef(AliTRDresolution, 6) // TRD tracking resolution task
139 };
140 #endif