]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWG1/TRD/AliTRDresolution.h
fix constantness definition for all compilers
[u/mrichter/AliRoot.git] / PWG1 / TRD / AliTRDresolution.h
index 323380bec47cb9843432d9d3b72cae78fac85eb9..0e6789a6f992db3b1a081511b628883bac6495f6 100644 (file)
 #include "AliTRDrecoTask.h"
 #endif
 
+class TAxis;
 class TH1;
 class TH2;
+class TH3;
 class TF1;
 class TGraphErrors;
 class TObjArray;
@@ -32,113 +34,169 @@ class AliTrackPoint;
 class AliTRDresolution : public AliTRDrecoTask
 {
 public:
-  enum ETRDresolutionPlot {
-     kCharge     =  0 // charge resolution
-    ,kCluster    =  1 // cluster - track
-    ,kTrack      =  2 // tracklet - track residuals/pulls
-    ,kTrackIn    =  3 // tracklet - track residuals/pulls at lower TRD entrance 
-    ,kTrackOut   =  4 // tracklet - track residuals/pulls at lower TRD entrance during refit
-    ,kMCcluster  =  5 // cluster-mc resolution/pulls
-    ,kMCtracklet =  6 // tracklet-mc resolution/pulls
-    ,kMCtrackIn  =  7 // TPC track monitor
-    ,kMCtrackOut =  8 // TOF/HMPID track monitor
-    ,kMCtrack    =  9 // TRD track monitor
-    ,kNviews     = 10 // total number of resolution views
-    ,kNprojs     = 70 // total number of projections for all views
-  };
   enum ETRDresolutionSteer {
      kVerbose    = BIT(18) // switch verbosity
     ,kVisual     = BIT(19) // show partial results during processing
     ,kTrackRefit = BIT(20) // steer track refit
+    ,kTrackSelect= BIT(21) // steer track selection
+    ,kXchange    = BIT(22) // use exchange containers
   };
-  enum ETRDresolutionOutSlots {
+  enum ETRDresolutionSlots {
      kClToTrk    = 2
-    ,kClToMC     = 3
-    ,kTrkltToTrk = 4
-    ,kTrkltToMC  = 5
-    ,kNOutSlots  = 4
+    ,kClToMC
+    ,kTrkltToTrk
+    ,kTrkltToMC
+    ,kNSlots     = 4
+  };
+  enum ETRDresolutionClass {
+    kCluster=0        // cluster - track
+    ,kTracklet        // tracklet - track residuals/pulls
+    ,kTrackIn         // tracklet - track residuals/pulls at lower TRD entrance
+    ,kMCcluster       // cluster-mc resolution/pulls
+    ,kMCtracklet      // tracklet-mc resolution/pulls
+    ,kMCtrackIn       // TPC track monitor
+    ,kMCtrack         // TRD track monitor
+/*    ,kTrackOut        // tracklet - track residuals/pulls at lower TRD entrance during refit
+    ,kMCtrackOut      // TOF/HMPID track monitor*/
+    ,kNclasses        // total number of resolution classes
+  };
+  enum ETRDresolutionClassProjs {
+    kClNproj=48       // cluster projections
+    ,kTrkltNproj=72   // tracklet projections
+    ,kTrkInNproj=8    // trackIn projections
+    ,kTrkNproj=72     // track projections
+  };
+  enum ETRDresolutionProjs {
+    kBC    = 0 // bunch cross
+    ,kPhi
+    ,kEta
+    ,kYrez
+    ,kPrez
+    ,kZrez
+    ,kSpeciesChgRC
+    ,kPt
+    ,kNdim  // no of dimensions in the THnSparse
+  };
+  enum ETRDresolutionSegmentation {
+     kSector    = 0
+    ,kStack
+    ,kDetector
+    ,kNbunchCross = 3  // no of classes for bunch crossing
+    ,kNpt         = 20 // no of log bins in pt spectrum
+    ,kNcharge     = 2
   };
 
   AliTRDresolution();
-  AliTRDresolution(char* name);
+  AliTRDresolution(char* name, Bool_t xchange=kTRUE);
   virtual ~AliTRDresolution();
   
-  static Bool_t  FitTrack(const Int_t np, AliTrackPoint *points, Float_t params[10]);
-  void    UserCreateOutputObjects();
-  Float_t GetPtThreshold() const {return fPtThreshold;}
-  Float_t GetSegmentationLevel() const {return fSegmentLevel;}
-  Bool_t  GetRefFigure(Int_t ifig);
+  static Bool_t   FitTrack(const Int_t np, AliTrackPoint *points, Float_t params[10]);
+  static Bool_t   FitTracklet(const Int_t ly, const Int_t np, const AliTrackPoint *points, const Float_t trackPars[10], Float_t trackletPars[3]);
+  void            UserCreateOutputObjects();
+//  Float_t GetCorrectionX(Int_t det, Int_t tb) const {return fXcorr[det][tb];}
+  Float_t         GetDyRange() const {return fDyRange;}
+  static Float_t  GetMeanStat(TH1 *h, Float_t cut=0., Option_t *opt="");
+  Float_t         GetPtThreshold() const {return fPtThreshold;}
+  static Int_t    GetPtBin(Float_t pt);
+  Bool_t          GetRefFigure(Int_t ifig);
+
   TObjArray*  Histos(); 
-  Bool_t  Load(const Char_t *file = "AnalysisResults.root", const Char_t *dir="TRD_Performance");
-  void    MakeSummary();
-
-  TObjArray*  Results(Int_t i=0) const {return i ? fGraphS : fGraphM;} 
-  void    UserExec(Option_t * opt);
-  void    InitExchangeContainers();
-  Bool_t  HasTrackRefit() const {return TestBit(kTrackRefit);}
-  Bool_t  IsVerbose() const {return TestBit(kVerbose);}
-  Bool_t  IsVisual() const {return TestBit(kVisual);}
-  Bool_t  PostProcess();
-
-  TH1*    PlotCharge(const AliTRDtrackV1 *t=NULL);
-  TH1*    PlotCluster(const AliTRDtrackV1 *t=NULL);
-  TH1*    PlotTracklet(const AliTRDtrackV1 *t=NULL);
-  TH1*    PlotTrackIn(const AliTRDtrackV1 *t=NULL);
-  TH1*    PlotTrackOut(const AliTRDtrackV1 *t=NULL);
-  TH1*    PlotMC(const AliTRDtrackV1 *t=NULL);
-
-  void    SetSegmentationLevel(Int_t l=0);
-  void    SetPtThreshold(Float_t pt) {fPtThreshold = pt;}
-  void    SetVerbose(Bool_t v = kTRUE) {SetBit(kVerbose, v);}
-  void    SetVisual(Bool_t v = kTRUE) {SetBit(kVisual, v);}
-  void    SetTrackRefit(Bool_t v = kTRUE) {SetBit(kTrackRefit, v);}
-
-  void    Terminate(Option_t * opt);
-  Bool_t  GetGraph(Float_t *bb, ETRDresolutionPlot ip, Int_t idx=-1, Bool_t kLEG=kTRUE, const Char_t *explain=NULL);
-  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);
-  
+//  Bool_t  Load(const Char_t *file = "AnalysisResults.root", const Char_t *dir="TRD_Performance");
+//  Bool_t  LoadCorrection(const Char_t *file=NULL);
+  void            MakeSummary();
+  static void     MakePtSegmentation(Float_t pt0=0.5, Float_t dpt=0.002);
+
+  TObjArray*      Results(ETRDresolutionClass c) const {if(!fProj) return NULL; return (TObjArray*)fProj->At(c);}
+  void            UserExec(Option_t * opt);
+  void            InitExchangeContainers();
+//  Bool_t  HasLoadCorrection() const {return TestBit(kLoadCorr);}
+  Bool_t          HasTrackRefit() const {return TestBit(kTrackRefit);}
+  Bool_t          HasTrackSelection() const {return TestBit(kTrackSelect);}
+  Bool_t          IsVerbose() const {return TestBit(kVerbose);}
+  Bool_t          IsVisual() const {return TestBit(kVisual);}
+  Bool_t          UseExchangeContainers() const {return TestBit(kXchange);}
+  Bool_t          PostProcess();
+
+  TH1*            PlotCluster(const AliTRDtrackV1 *t=NULL);
+  TH1*            PlotTracklet(const AliTRDtrackV1 *t=NULL);
+  TH1*            PlotTrackIn(const AliTRDtrackV1 *t=NULL);
+//  TH1*            PlotTrackOut(const AliTRDtrackV1 *t=NULL);
+  TH1*            PlotMC(const AliTRDtrackV1 *t=NULL);
+
+  static Bool_t   Process(TH2* const h2, TGraphErrors **g, Int_t stat=100);
+  void            SetDyRange(Float_t dy) {fDyRange = dy;}
+//  void    SetSegmentationLevel(Int_t l=0);
+  void            SetPtThreshold(Float_t pt) {fPtThreshold = pt;}
+//  void    SetLoadCorrection(Bool_t v = kTRUE) {SetBit(kLoadCorr, v);}
+  void            SetVerbose(Bool_t v = kTRUE) {SetBit(kVerbose, v);}
+  void            SetVisual(Bool_t v = kTRUE) {SetBit(kVisual, v);}
+  void            SetTrackRefit(Bool_t v = kTRUE) {SetBit(kTrackRefit, v);}
+  void            SetTrackSelection(Bool_t v = kTRUE) {SetBit(kTrackSelect, v);}
+  void            SetUseExchangeContainers(Bool_t v = kTRUE) {SetBit(kXchange, v);}
+
+  void            Terminate(Option_t * opt);
+  static Bool_t   UseTrack(const Int_t np, const AliTrackPoint *points, Float_t params[10]);
+
 private:
+  class AliTRDresolutionProjection{
+    friend class AliTRDresolution;  // Friend class
+  public:
+    AliTRDresolutionProjection();
+    virtual ~AliTRDresolutionProjection();
+    void  Build(const Char_t *n, const Char_t *t, Int_t ix, Int_t iy, Int_t iz, TAxis *aa[]);
+    void  Increment(Int_t bin[], Double_t v);
+    TH2*  Projection2D(const Int_t nstat, const Int_t ncol, const Int_t mid=0);
+    void  SetRebinStrategy(Int_t n, Int_t rebx[], Int_t reby[]);
+    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;}
+  private:
+    AliTRDresolutionProjection(const AliTRDresolutionProjection&);
+    AliTRDresolutionProjection& operator=(const AliTRDresolutionProjection&);
+  protected:
+    TH3  *fH;          // data container
+    Int_t fAx[3];      // projection axes
+    Int_t fNrebin;     // no. of rebinning steps
+    Int_t *fRebinX;    //[fNrebin] rebinning of the X axis
+    Int_t *fRebinY;    //[fNrebin] rebinning of the Y axis
+    Float_t fRange[4]; //show range of the z processed
+  ClassDef(AliTRDresolutionProjection, 1)  // wrapper for a projection container THnSparse -> TH3
+  };
+
   AliTRDresolution(const AliTRDresolution&);
   AliTRDresolution& operator=(const AliTRDresolution&);
 
-  void    AdjustF1(TH1 *h, TF1 *f);
-  TObjArray*  BuildMonitorContainerCluster(const char* name, Bool_t expand=kFALSE);
+  void        AdjustF1(TH1 *h, TF1 *f);
+  TObjArray*  BuildMonitorContainerCluster(const char* name, Bool_t expand=kFALSE, Float_t range=-1.);
   TObjArray*  BuildMonitorContainerTracklet(const char* name, Bool_t expand=kFALSE);
   TObjArray*  BuildMonitorContainerTrack(const char* name);
-  void    GetLandauMpvFwhm(TF1 * const f, Float_t &mpv, Float_t &xm, Float_t &xM);
-  void    GetRange(TH2 *h2, Char_t mod, Float_t *range);
-  void    MakeSummaryPlot(TObjArray *a, TH2 *h2);
-  Bool_t  Process(TH2* const h2, TF1 *f, Float_t k, TGraphErrors **g);
-  Bool_t  Process2D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1., Int_t gidx=-1);
-  Bool_t  Process2Darray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
-  Bool_t  Process3D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
-  Bool_t  Process3Dlinked(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
-  Bool_t  Process3DL(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
-  Bool_t  Process3Darray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
-  Bool_t  Process3DlinkedArray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
-  Bool_t  Pulls(Double_t dyz[2], Double_t cc[3], Double_t tilt) const;
-
-  UChar_t             fSegmentLevel;    // segmentation level [sector/stack/chamber]
-  UShort_t            fIdxPlot;         // plot counter (internal)
-  UShort_t            fIdxFrame;        // frame counter (internal)
-  UShort_t            fNcomp[kNprojs];  // number of projections per task
-  Char_t              *fAxTitle[kNprojs][4]; //! Title for all ref histos
-  Float_t             fPtThreshold;     // pt threshold for some performance plots
-  static Char_t const *fgPerformanceName[kNviews]; //! name of performance plot
-  static Char_t const *fgParticle[11];    //! latex name of particles/sign 
-  static UChar_t const fgNproj[kNviews]; //! number of projections per task
-  static Int_t const  fgkNresYsegm[3];   //! number of segments for saving y resolution
-  static Char_t const *fgkResYsegmName[3];//! name of segment for saving y resolution
-  TDatabasePDG        *fDBPDG;          // PDG database
-  TObjArray           *fGraphS;         //! result holder - sigma values
-  TObjArray           *fGraphM;         //! result holder - mean values
+  void        DrawSigma(TH2 *h2, Float_t scale=1, Float_t m=0., Float_t M=-1.,  const Char_t *t=NULL);
+  void        GetLandauMpvFwhm(TF1 * const f, Float_t &mpv, Float_t &xm, Float_t &xM);
+  void        GetRange(TH2 *h2, Char_t mod, Float_t *range);
+
+protected:
+  Bool_t      MakeProjectionCluster(Bool_t mc=kFALSE);
+  Bool_t      MakeProjectionTracklet(Bool_t mc=kFALSE);
+  Bool_t      MakeProjectionTrackIn(Bool_t mc=kFALSE);
+  Bool_t      MakeProjectionTrack();
+  Bool_t      Process(TH2* const h2, TF1 *f, Float_t k, TGraphErrors **g);
+  Bool_t      Pulls(Double_t dyz[2], Double_t cc[3], Double_t tilt) const;
+
+  UShort_t              fIdxPlot;         // plot counter (internal)
+  UShort_t              fIdxFrame;        // frame counter (internal)
+  Float_t               fPtThreshold;     // pt threshold for some performance plots
+  Float_t               fDyRange;         // min/max dy
+  static Char_t const  *fgPerformanceName[kNclasses]; //! name of performance plot
+  static Int_t const    fgkNbins[kNdim];  //! no of bins/projection
+  static Double_t const fgkMin[kNdim];    //! low limits for projections
+  static Double_t const fgkMax[kNdim];    //! high limits for projections
+  static Char_t const  *fgkTitle[kNdim];  //! title of projection 
+  static Float_t        fgPtBin[kNpt+1];  //! pt segmentation
+  TObjArray            *fProj;            //! result holder - sigma values
+  TDatabasePDG         *fDBPDG;           //! PDG database
 
   // calibration containers
-  TObjArray           *fCl;     //! cluster2track calib
-  TObjArray           *fMCcl;   //! cluster2mc calib
-/*  TObjArray           *fTrklt;  //! tracklet2track calib
-  TObjArray           *fMCtrklt;//! tracklet2mc calib*/
+  TObjArray            *fCl;              //! cluster2track calib
+  TObjArray            *fMCcl;            //! cluster2mc calib
   
-  ClassDef(AliTRDresolution, 8) // TRD tracking resolution task
+  ClassDef(AliTRDresolution, 10) // TRD tracking resolution task
 };
 #endif