using AliCDBConnect task in test train of TRD
[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 TAxis;
23 class TH1;
24 class TH2;
25 class TH3;
26 class TF1;
27 class TGraphErrors;
28 class TObjArray;
29 class TDatabasePDG;
30 class AliTRDrecoParam;
31 class AliTRDseedV1;
32 class AliTRDtrackInfo;
33 class AliTrackPoint;
34 class AliTRDresolution : public AliTRDrecoTask
35 {
36 public:
37   enum ETRDresolutionSteer {
38      kVerbose    = BIT(18) // switch verbosity
39     ,kVisual     = BIT(19) // show partial results during processing
40     ,kTrackRefit = BIT(20) // steer track refit
41     ,kTrackSelect= BIT(21) // steer track selection
42     ,kXchange    = BIT(22) // use exchange containers
43   };
44   enum ETRDresolutionSlots {
45      kClToTrk    = 2
46     ,kClToMC
47     ,kTrkltToTrk
48     ,kTrkltToMC
49     ,kNSlots     = 4
50   };
51   enum ETRDresolutionClass {
52     kCluster=0        // cluster - track
53     ,kTracklet        // tracklet - track residuals/pulls
54     ,kTrackIn         // tracklet - track residuals/pulls at lower TRD entrance
55     ,kMCcluster       // cluster-mc resolution/pulls
56     ,kMCtracklet      // tracklet-mc resolution/pulls
57     ,kMCtrackIn       // TPC track monitor
58     ,kMCtrack         // TRD track monitor
59 /*    ,kTrackOut        // tracklet - track residuals/pulls at lower TRD entrance during refit
60     ,kMCtrackOut      // TOF/HMPID track monitor*/
61     ,kNclasses        // total number of resolution classes
62   };
63   enum ETRDresolutionClassProjs {
64     kClNproj=48       // cluster projections
65     ,kTrkltNproj=72   // tracklet projections
66     ,kTrkInNproj=8    // trackIn projections
67     ,kTrkNproj=72     // track projections
68   };
69   enum ETRDresolutionProjs {
70     kBC    = 0 // bunch cross
71     ,kPhi
72     ,kEta
73     ,kYrez
74     ,kPrez
75     ,kZrez
76     ,kSpeciesChgRC
77     ,kPt
78     ,kNdim  // no of dimensions in the THnSparse
79   };
80   enum ETRDresolutionSegmentation {
81      kSector    = 0
82     ,kStack
83     ,kDetector
84     ,kNbunchCross = 3  // no of classes for bunch crossing
85     ,kNpt         = 20 // no of log bins in pt spectrum
86     ,kNcharge     = 2
87   };
88
89   AliTRDresolution();
90   AliTRDresolution(char* name, Bool_t xchange=kTRUE);
91   virtual ~AliTRDresolution();
92   
93   static Bool_t   FitTrack(const Int_t np, AliTrackPoint *points, Float_t params[10]);
94   static Bool_t   FitTracklet(const Int_t ly, const Int_t np, const AliTrackPoint *points, const Float_t trackPars[10], Float_t trackletPars[3]);
95   void            UserCreateOutputObjects();
96 //  Float_t GetCorrectionX(Int_t det, Int_t tb) const {return fXcorr[det][tb];}
97   Float_t         GetDyRange() const {return fDyRange;}
98   static Float_t  GetMeanStat(TH1 *h, Float_t cut=0., Option_t *opt="");
99   Float_t         GetPtThreshold() const {return fPtThreshold;}
100   static Int_t    GetPtBin(Float_t pt);
101   Bool_t          GetRefFigure(Int_t ifig);
102
103   TObjArray*  Histos(); 
104 //  Bool_t  Load(const Char_t *file = "AnalysisResults.root", const Char_t *dir="TRD_Performance");
105 //  Bool_t  LoadCorrection(const Char_t *file=NULL);
106   void            MakeSummary();
107   static void     MakePtSegmentation(Float_t pt0=0.5, Float_t dpt=0.002);
108
109   TObjArray*      Results(ETRDresolutionClass c) const {if(!fProj) return NULL; return (TObjArray*)fProj->At(c);}
110   void            UserExec(Option_t * opt);
111   void            InitExchangeContainers();
112 //  Bool_t  HasLoadCorrection() const {return TestBit(kLoadCorr);}
113   Bool_t          HasTrackRefit() const {return TestBit(kTrackRefit);}
114   Bool_t          HasTrackSelection() const {return TestBit(kTrackSelect);}
115   Bool_t          IsVerbose() const {return TestBit(kVerbose);}
116   Bool_t          IsVisual() const {return TestBit(kVisual);}
117   Bool_t          UseExchangeContainers() const {return TestBit(kXchange);}
118   Bool_t          PostProcess();
119
120   TH1*            PlotCluster(const AliTRDtrackV1 *t=NULL);
121   TH1*            PlotTracklet(const AliTRDtrackV1 *t=NULL);
122   TH1*            PlotTrackIn(const AliTRDtrackV1 *t=NULL);
123 //  TH1*            PlotTrackOut(const AliTRDtrackV1 *t=NULL);
124   TH1*            PlotMC(const AliTRDtrackV1 *t=NULL);
125
126   static Bool_t   Process(TH2* const h2, TGraphErrors **g, Int_t stat=100);
127   void            SetDyRange(Float_t dy) {fDyRange = dy;}
128 //  void    SetSegmentationLevel(Int_t l=0);
129   void            SetPtThreshold(Float_t pt) {fPtThreshold = pt;}
130 //  void    SetLoadCorrection(Bool_t v = kTRUE) {SetBit(kLoadCorr, v);}
131   void            SetVerbose(Bool_t v = kTRUE) {SetBit(kVerbose, v);}
132   void            SetVisual(Bool_t v = kTRUE) {SetBit(kVisual, v);}
133   void            SetTrackRefit(Bool_t v = kTRUE) {SetBit(kTrackRefit, v);}
134   void            SetTrackSelection(Bool_t v = kTRUE) {SetBit(kTrackSelect, v);}
135   void            SetUseExchangeContainers(Bool_t v = kTRUE) {SetBit(kXchange, v);}
136
137   void            Terminate(Option_t * opt);
138   static Bool_t   UseTrack(const Int_t np, const AliTrackPoint *points, Float_t params[10]);
139
140 private:
141   class AliTRDresolutionProjection{
142     friend class AliTRDresolution;  // Friend class
143   public:
144     AliTRDresolutionProjection();
145     virtual ~AliTRDresolutionProjection();
146     void  Build(const Char_t *n, const Char_t *t, Int_t ix, Int_t iy, Int_t iz, TAxis *aa[]);
147     void  Increment(Int_t bin[], Double_t v);
148     TH2*  Projection2D(const Int_t nstat, const Int_t ncol, const Int_t mid=0);
149     void  SetRebinStrategy(Int_t n, Int_t rebx[], Int_t reby[]);
150     void  SetShowRange(Float_t zm, Float_t zM, Float_t em=0., Float_t eM=0.) {fRange[0] = zm; fRange[1] = zM; fRange[2] = em; fRange[3] = eM;}
151   private:
152     AliTRDresolutionProjection(const AliTRDresolutionProjection&);
153     AliTRDresolutionProjection& operator=(const AliTRDresolutionProjection&);
154   protected:
155     TH3  *fH;          // data container
156     Int_t fAx[3];      // projection axes
157     Int_t fNrebin;     // no. of rebinning steps
158     Int_t *fRebinX;    //[fNrebin] rebinning of the X axis
159     Int_t *fRebinY;    //[fNrebin] rebinning of the Y axis
160     Float_t fRange[4]; //show range of the z processed
161   ClassDef(AliTRDresolutionProjection, 1)  // wrapper for a projection container THnSparse -> TH3
162   };
163
164   AliTRDresolution(const AliTRDresolution&);
165   AliTRDresolution& operator=(const AliTRDresolution&);
166
167   void        AdjustF1(TH1 *h, TF1 *f);
168   TObjArray*  BuildMonitorContainerCluster(const char* name, Bool_t expand=kFALSE, Float_t range=-1.);
169   TObjArray*  BuildMonitorContainerTracklet(const char* name, Bool_t expand=kFALSE);
170   TObjArray*  BuildMonitorContainerTrack(const char* name);
171   void        DrawSigma(TH2 *h2, Float_t scale=1, Float_t m=0., Float_t M=-1.,  const Char_t *t=NULL);
172   void        GetLandauMpvFwhm(TF1 * const f, Float_t &mpv, Float_t &xm, Float_t &xM);
173   void        GetRange(TH2 *h2, Char_t mod, Float_t *range);
174
175 protected:
176   Bool_t      MakeProjectionCluster(Bool_t mc=kFALSE);
177   Bool_t      MakeProjectionTracklet(Bool_t mc=kFALSE);
178   Bool_t      MakeProjectionTrackIn(Bool_t mc=kFALSE);
179   Bool_t      MakeProjectionTrack();
180   Bool_t      Process(TH2* const h2, TF1 *f, Float_t k, TGraphErrors **g);
181   Bool_t      Pulls(Double_t dyz[2], Double_t cc[3], Double_t tilt) const;
182
183   UShort_t              fIdxPlot;         // plot counter (internal)
184   UShort_t              fIdxFrame;        // frame counter (internal)
185   Float_t               fPtThreshold;     // pt threshold for some performance plots
186   Float_t               fDyRange;         // min/max dy
187   static Char_t const  *fgPerformanceName[kNclasses]; //! name of performance plot
188   static Int_t const    fgkNbins[kNdim];  //! no of bins/projection
189   static Double_t const fgkMin[kNdim];    //! low limits for projections
190   static Double_t const fgkMax[kNdim];    //! high limits for projections
191   static Char_t const  *fgkTitle[kNdim];  //! title of projection 
192   static Float_t        fgPtBin[kNpt+1];  //! pt segmentation
193   TObjArray            *fProj;            //! result holder - sigma values
194   TDatabasePDG         *fDBPDG;           //! PDG database
195
196   // calibration containers
197   TObjArray            *fCl;              //! cluster2track calib
198   TObjArray            *fMCcl;            //! cluster2mc calib
199   
200   ClassDef(AliTRDresolution, 10) // TRD tracking resolution task
201 };
202 #endif