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 AliTRDrecoParam;
  29 class AliTRDseedV1;
  30 class AliTRDtrackInfo;
  31 class AliTrackPoint;
  32 class AliTRDresolution : public AliTRDrecoTask
  33 {
  34 public:
  35   enum ETRDresolutionPlot {
  36      kCharge     =  0 // charge resolution
  37     ,kCluster    =  1 // cluster - track
  38     ,kTrack      =  2 // tracklet - track residuals/pulls
  39     ,kTrackIn    =  3 // tracklet - track residuals/pulls at lower TRD entrance
  40     ,kTrackOut   =  4 // tracklet - track residuals/pulls at lower TRD entrance during refit
  41     ,kMCcluster  =  5 // cluster-mc resolution/pulls
  42     ,kMCtracklet =  6 // tracklet-mc resolution/pulls
  43     ,kMCtrackIn  =  7 // TPC track monitor
  44     ,kMCtrackOut =  8 // TOF/HMPID track monitor
  45     ,kMCtrack    =  9 // TRD track monitor
  46     ,kNviews     = 10 // total number of resolution views
  47     ,kNprojs     = 70 // total number of projections for all views
  48   };
  49   enum ETRDresolutionSteer {
  50      kVerbose    = BIT(18) // switch verbosity
  51     ,kVisual     = BIT(19) // show partial results during processing
  52     ,kTrackRefit = BIT(20) // steer track refit
  53     ,kTrackSelect= BIT(21) // steer track selection
  54   };
  55   enum ETRDresolutionOutSlots {
  56      kClToTrk    = 2
  57     ,kClToMC     = 3
  58     ,kTrkltToTrk = 4
  59     ,kTrkltToMC  = 5
  60     ,kNOutSlots  = 4
  61   };
  62
  63   AliTRDresolution();
  64   AliTRDresolution(char* name);
  65   virtual ~AliTRDresolution();
  66
  67   static Bool_t  FitTrack(const Int_t np, AliTrackPoint *points, Float_t params[10]);
  68   void    UserCreateOutputObjects();
  69   Float_t GetDyRange() const {return fDyRange;}
  70   Float_t GetPtThreshold() const {return fPtThreshold;}
  71   Float_t GetSegmentationLevel() const {return fSegmentLevel;}
  72   Bool_t  GetRefFigure(Int_t ifig);
  73   TObjArray*  Histos();
  74   Bool_t  Load(const Char_t *file = "AnalysisResults.root", const Char_t *dir="TRD_Performance");
  75   void    MakeSummary();
  76
  77   TObjArray*  Results(Int_t i=0) const {return i ? fGraphS : fGraphM;}
  78   void    UserExec(Option_t * opt);
  79   void    InitExchangeContainers();
  80   Bool_t  HasTrackRefit() const {return TestBit(kTrackRefit);}
  81   Bool_t  HasTrackSelection() const {return TestBit(kTrackSelect);}
  82   Bool_t  IsVerbose() const {return TestBit(kVerbose);}
  83   Bool_t  IsVisual() const {return TestBit(kVisual);}
  84   Bool_t  PostProcess();
  85
  86   TH1*    PlotCharge(const AliTRDtrackV1 *t=NULL);
  87   TH1*    PlotCluster(const AliTRDtrackV1 *t=NULL);
  88   TH1*    PlotTracklet(const AliTRDtrackV1 *t=NULL);
  89   TH1*    PlotTrackIn(const AliTRDtrackV1 *t=NULL);
  90   TH1*    PlotTrackOut(const AliTRDtrackV1 *t=NULL);
  91   TH1*    PlotMC(const AliTRDtrackV1 *t=NULL);
  92
  93   void    SetDyRange(Float_t dy) {fDyRange = dy;}
  94   void    SetSegmentationLevel(Int_t l=0);
  95   void    SetPtThreshold(Float_t pt) {fPtThreshold = pt;}
  96   void    SetVerbose(Bool_t v = kTRUE) {SetBit(kVerbose, v);}
  97   void    SetVisual(Bool_t v = kTRUE) {SetBit(kVisual, v);}
  98   void    SetTrackRefit(Bool_t v = kTRUE) {SetBit(kTrackRefit, v);}
  99   void    SetTrackSelection(Bool_t v = kTRUE) {SetBit(kTrackSelect, v);}
 100
 101   void    Terminate(Option_t * opt);
 102   Bool_t  GetGraph(Float_t *bb, ETRDresolutionPlot ip, Int_t idx=-1, Bool_t kLEG=kTRUE, const Char_t *explain=NULL);
 103   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);
 104   static Bool_t  UseTrack(const Int_t np, AliTrackPoint *points, Float_t params[10]);
 105
 106 private:
 107   AliTRDresolution(const AliTRDresolution&);
 108   AliTRDresolution& operator=(const AliTRDresolution&);
 109
 110   void    AdjustF1(TH1 *h, TF1 *f);
 111   TObjArray*  BuildMonitorContainerCluster(const char* name, Bool_t expand=kFALSE);
 112   TObjArray*  BuildMonitorContainerTracklet(const char* name, Bool_t expand=kFALSE);
 113   TObjArray*  BuildMonitorContainerTrack(const char* name);
 114   void    GetLandauMpvFwhm(TF1 * const f, Float_t &mpv, Float_t &xm, Float_t &xM);
 115   void    GetRange(TH2 *h2, Char_t mod, Float_t *range);
 116   void    MakeSummaryPlot(TObjArray *a, TH2 *h2);
 117   Bool_t  Process(TH2* const h2, TF1 *f, Float_t k, TGraphErrors **g);
 118   Bool_t  Process2D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1., Int_t gidx=-1);
 119   Bool_t  Process2Darray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
 120   Bool_t  Process3D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
 121   Bool_t  Process3Dlinked(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
 122   Bool_t  Process3DL(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
 123   Bool_t  Process3Darray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
 124   Bool_t  Process3DlinkedArray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
 125   Bool_t  Pulls(Double_t dyz[2], Double_t cc[3], Double_t tilt) const;
 126
 127   UChar_t             fSegmentLevel;    // segmentation level [sector/stack/chamber]
 128   UShort_t            fIdxPlot;         // plot counter (internal)
 129   UShort_t            fIdxFrame;        // frame counter (internal)
 130   UShort_t            fNcomp[kNprojs];  // number of projections per task
 131   Char_t              *fAxTitle[kNprojs][4]; //! Title for all ref histos
 132   Float_t             fPtThreshold;     // pt threshold for some performance plots
 133   Float_t             fDyRange;         // min/max dy
 134   static Char_t const *fgPerformanceName[kNviews]; //! name of performance plot
 135   static Char_t const *fgParticle[11];    //! latex name of particles/sign
 136   static UChar_t const fgNproj[kNviews]; //! number of projections per task
 137   static Int_t const  fgkNresYsegm[3];   //! number of segments for saving y resolution
 138   static Char_t const *fgkResYsegmName[3];//! name of segment for saving y resolution
 139   TDatabasePDG        *fDBPDG;          // PDG database
 140   TObjArray           *fGraphS;         //! result holder - sigma values
 141   TObjArray           *fGraphM;         //! result holder - mean values
 142
 143   // calibration containers
 144   TObjArray           *fCl;     //! cluster2track calib
 145   TObjArray           *fMCcl;   //! cluster2mc calib
 146 /*  TObjArray           *fTrklt;  //! tracklet2track calib
 147   TObjArray           *fMCtrklt;//! tracklet2mc calib*/
 148
 149   ClassDef(AliTRDresolution, 9) // TRD tracking resolution task
 150 };
 151 #endif