]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWG1/TRD/AliTRDresolution.h
Updating DA to store TDC offsets
[u/mrichter/AliRoot.git] / PWG1 / TRD / AliTRDresolution.h
index 2152eb27668d3c704d4c5b403a8c254ffaf91264..e41c8692d32c8616b3f04e61f560d07d136ea0b2 100644 (file)
@@ -25,92 +25,142 @@ class TF1;
 class TGraphErrors;
 class TObjArray;
 class TDatabasePDG;
-class AliTRDReconstructor;
-class AliTRDgeometry;
 class AliTRDrecoParam;
 class AliTRDseedV1;
 class AliTRDtrackInfo;
+class AliTrackPoint;
 class AliTRDresolution : public AliTRDrecoTask
 {
 public:
-  enum ETRDresolutionPlot {
-     kCharge     =  0 // charge resolution
-    ,kCluster    =  1 // cluster - track
-    ,kTrackTRD   =  2 // tracklet - track residuals/pulls
-    ,kTrackTPC   =  3 // tracklet - track residuals/pulls at lower TRD entrance 
-    ,kMCcluster  =  4 // cluster-mc resolution/pulls
-    ,kMCtracklet =  5 // tracklet-mc resolution/pulls
-    ,kMCtrackTPC =  6 // TPC track monitor
-    ,kMCtrackTOF =  7 // TOF/HMPID track monitor
-    ,kMCtrackTRD =  8 // TRD track monitor
-    ,kNviews     =  9 // total number of resolution views
-    ,kNprojs     = 53 // total number of projections for all views
-  };
   enum ETRDresolutionSteer {
-    kVerbose  = 0
-    ,kVisual  = 1
+     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
+//    ,kLoadCorr   = BIT(23) // steer load corrections
+  };
+  enum ETRDresolutionSlots {
+     kClToTrk    = 2
+    ,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
+    ,kTrackOut        // tracklet - track residuals/pulls at lower TRD entrance during refit
+    ,kMCcluster       // cluster-mc resolution/pulls
+    ,kMCtracklet      // tracklet-mc resolution/pulls
+    ,kMCtrackIn       // TPC track monitor
+    ,kMCtrackOut      // TOF/HMPID track monitor
+    ,kMCtrack         // TRD track monitor
+    ,kNclasses        // total number of resolution classes
+  };
+  enum ETRDresolutionProjs {
+    kBC    = 0 // bunch cross
+    ,kPhi
+    ,kEta
+    ,kSpeciesChgRC
+    ,kYrez
+    ,kZrez
+    ,kPrez
+//    ,kPt
+    ,kNdim  // no of dimensions in the THnSparse
+  };
+  enum ETRDresolutionSegmentation {
+     kSector    = 0
+    ,kStack
+    ,kDetector
+    ,kNbunchCross = 3  // no of classes for bunch crossing
+    ,kNpt         = 24 // no of log bins in pt spectrum
+    ,kNcharge     = 2
   };
 
   AliTRDresolution();
   AliTRDresolution(char* name);
   virtual ~AliTRDresolution();
   
+  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;}
+  Float_t GetMeanWithBoundary(TH1 *h, Float_t zm, Float_t zM, Float_t dz) const;
+  Float_t GetPtThreshold() const {return fPtThreshold;}
+  static Int_t GetPtBin(Float_t pt);
   Bool_t  GetRefFigure(Int_t ifig);
+
   TObjArray*  Histos(); 
-  TObjArray*  Results(Int_t i=0) const {return i ? fGraphS : fGraphM;} 
+//  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.2, Float_t dpt=0.002);
+
+  TObjArray*  Results(ETRDresolutionClass c) const {if(!fProj) return NULL; return (TObjArray*)fProj->At(c);}
   void    UserExec(Option_t * opt);
-  Bool_t  IsVerbose() const {return TESTBIT(fStatus, kVerbose);}
-  Bool_t  IsVisual() const {return TESTBIT(fStatus, kVisual);}
+  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  PostProcess();
 
-  TH1*    PlotCharge(const AliTRDtrackV1 *t=0x0);
-  TH1*    PlotCluster(const AliTRDtrackV1 *t=0x0);
-  TH1*    PlotTracklet(const AliTRDtrackV1 *t=0x0);
-  TH1*    PlotTrackTPC(const AliTRDtrackV1 *t=0x0);
-  TH1*    PlotMC(const AliTRDtrackV1 *t=0x0);
+  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    SetRecoParam(AliTRDrecoParam *r);
-  void    SetVerbose(Bool_t v = kTRUE) {v ? SETBIT(fStatus ,kVerbose): CLRBIT(fStatus ,kVerbose);}
-  void    SetVisual(Bool_t v = kTRUE) {v ? SETBIT(fStatus, kVisual) : CLRBIT(fStatus, kVisual);}
+  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    Terminate(Option_t * opt);
-  Bool_t  GetGraphPlot(Float_t *bb, ETRDresolutionPlot ip, Int_t idx=-1);
-  Bool_t  GetGraphTrack(Float_t *bb, Int_t idx, Int_t ist, Int_t n=123456789, Bool_t kLEG=kFALSE);
-  Bool_t  GetGraphTrackTPC(Float_t *bb, Int_t idx, Int_t ist=0, Int_t n=123456789, Bool_t kLEG=kFALSE);
-  
+  static Bool_t  UseTrack(const Int_t np, const AliTrackPoint *points, Float_t params[10]);
+
 private:
   AliTRDresolution(const AliTRDresolution&);
   AliTRDresolution& operator=(const AliTRDresolution&);
+
   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);
+  Bool_t  MakeProjectionCluster();
+  Bool_t  MakeProjectionTracklet();
+  Bool_t  MakeProjectionTrackIn();
   Bool_t  Process(TH2* const h2, TF1 *f, Float_t k, TGraphErrors **g);
-  Bool_t  Process2D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=0x0,  Float_t scale=1., Int_t gidx=-1);
-  Bool_t  Process3D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=0x0,  Float_t scale=1.);
-  Bool_t  Process3Drange(ETRDresolutionPlot ip, Int_t idx=-1, Int_t gidx=-1, TF1 *f=0x0,  Float_t scale=1., Int_t zbin0=0, Int_t zbin1=0);
-  Bool_t  Process3DL(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=0x0,  Float_t scale=1.);
-  Bool_t  Process4D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=0x0,  Float_t scale=1., Int_t n=-1);
+  Bool_t  Pulls(Double_t dyz[2], Double_t cc[3], Double_t tilt) const;
 
-  UChar_t             fStatus;          // steer parameter of the task
-  UShort_t            fIdxPlot;         //! plot counter (internal)
-  UShort_t            fIdxFrame;        //! frame counter (internal)
-  static Char_t const *fgPerformanceName[kNviews]; // name of performance plot
-  static UChar_t const fgNhistos[kNviews]; // number of histos per task
-  static UChar_t const fgNproj[kNviews]; // number of projections per task
-  static UChar_t const fgNcomp[kNprojs]; // number of projections per task
-  static Char_t const *fgAxTitle[kNprojs][4]; // Title for all ref histos
-  AliTRDReconstructor *fReconstructor;  //! local reconstructor
-  AliTRDgeometry      *fGeo;            //! TRD geometry
-  TDatabasePDG        *fDBPDG;          //! PDG database
-  TObjArray           *fGraphS;         //! result holder - sigma values
-  TObjArray           *fGraphM;         //! result holder - mean values
+  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 UChar_t const  fgNproj[kNclasses];//! number of projections per task
+  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           *fTrklt;  //! tracklet2track calib
-  TObjArray           *fMCcl;   //! cluster2mc calib
-  TObjArray           *fMCtrklt;//! tracklet2mc calib
+  TObjArray           *fCl;               //! cluster2track calib
+  TObjArray           *fMCcl;             //! cluster2mc calib
   
-  ClassDef(AliTRDresolution, 4) // TRD tracking resolution task
+  ClassDef(AliTRDresolution, 10) // TRD tracking resolution task
 };
 #endif