fix coding convention errors
[u/mrichter/AliRoot.git] / TRD / qaRec / AliTRDcheckPID.h
1 #ifndef ALITRDCHECKPID_H
2 #define ALITRDCHECKPID_H
3
4 //////////////////////////////////////////////////////
5 //
6 // PID performance checker of the TRD
7 //
8 // Author : Alex Wilk <wilka@uni-muenster.de>
9 //          Alex Bercuci <A.Bercuci@gsi.de>
10 //          Markus Fasel <M.Fasel@gsi.de>
11 //
12 ///////////////////////////////////////////////////////
13
14 #ifndef ROOT_TAxis
15 #include "TAxis.h"
16 #endif
17
18 #ifndef ALITRDRECOTASK_H
19 #include "AliTRDrecoTask.h"
20 #endif
21
22 class AliTRDReconstructor;
23 class AliTRDpidUtil;
24 class AliTRDcheckPID : public AliTRDrecoTask 
25 {
26 public:
27   // Plots registered for this task
28   enum{
29      kEfficiency     =  0     // pi Efficiency plot
30     ,kdEdx           =  1     // dE/dx spectra
31     ,kdEdxSlice      =  2     // dE/dx spectra
32     ,kPH             =  3     // pulse height spectra
33     ,kNClus          =  4     //  number of clusters per track
34     ,kMomentum       =  5     // momentum distribution
35     ,kMomentumBin    =  6     // momentum distribution
36     ,kNTracklets     =  7     // Number of tracklets per track
37     ,kNPlots         =  8     // Number of plots for this tasks 
38   };
39   AliTRDcheckPID();
40   virtual ~AliTRDcheckPID();
41   
42   virtual void    CreateOutputObjects();
43   virtual Bool_t  GetRefFigure(Int_t ifig);
44   virtual Bool_t  PostProcess();
45   virtual void    Terminate(Option_t *);
46
47   TH1 *PlotLQ(const AliTRDtrackV1 *track = 0x0);
48   TH1 *PlotNN(const AliTRDtrackV1 *track = 0x0);
49   TH1 *PlotESD(const AliTRDtrackV1 *track = 0x0);
50   TH1 *PlotdEdx(const AliTRDtrackV1 *track = 0x0);
51   TH1 *PlotdEdxSlice(const AliTRDtrackV1 *track = 0x0);
52   TH1 *PlotPH(const AliTRDtrackV1 *track = 0x0);
53   TH1 *PlotNClus(const AliTRDtrackV1 *track = 0x0);
54   TH1 *PlotNTracklets(const AliTRDtrackV1 *track = 0x0);
55   TH1 *PlotMom(const AliTRDtrackV1 *track = 0x0);
56   TH1 *PlotMomBin(const AliTRDtrackV1 *track = 0x0);
57
58   void SetRequireMinNTracklets(Int_t mintracklets) { fMinNTracklets = mintracklets; }
59   void SetRequireMaxNTracklets(Int_t maxtracklets) { fMaxNTracklets = maxtracklets; }
60
61   TObjArray *GetGraphs() const { return fGraph; };
62   //TObjArray *GetHistos() { return fContainer; };
63   virtual TObjArray *Histos();
64   void EvaluatePionEfficiency(TObjArray * const histoContainer, TObjArray *results, Float_t electronEfficiency);
65   inline void SetMomentumBinning(Int_t nBins, Double_t *bins);
66   inline Int_t FindBin(Int_t species, Double_t momentum);
67   inline Bool_t IsInRange(Double_t momentum);
68
69 private:
70   AliTRDcheckPID(const AliTRDcheckPID&);               // not implemented
71   AliTRDcheckPID& operator=(const AliTRDcheckPID&);    // not implemented
72
73   Int_t  CalcPDG(AliTRDtrackV1* track = 0x0);
74   Bool_t CheckTrackQuality(const AliTRDtrackV1* track = 0x0) const;
75   
76   AliTRDReconstructor *fReconstructor;     //! reconstructor needed for recalculation the PID
77   AliTRDpidUtil       *fUtil;              //! utility class for PID calculations
78   TObjArray           *fGraph;             //! array of graphs filled in PostProcess
79   TObjArray           *fEfficiency;        //! array of histograms with efficiency
80   TAxis               *fMomentumAxis;      //! helper mementum binning
81   Int_t                fMinNTracklets;     // minimum number of required Tracklets (for systematic studies)
82   Int_t                fMaxNTracklets;     // maximum number of required Tracklets (for systematic studies) 
83   ClassDef(AliTRDcheckPID, 1); // TRD PID checker
84 };
85
86 //________________________________________________________________________
87 inline void AliTRDcheckPID::SetMomentumBinning(Int_t nBins, Double_t *bins){
88   //
89   // Set the Momentum Bins
90   //
91   if(fMomentumAxis) delete fMomentumAxis;
92   fMomentumAxis = new TAxis(nBins, bins);
93 }
94
95 //________________________________________________________________________
96 inline Int_t AliTRDcheckPID::FindBin(Int_t species, Double_t momentum){
97   //
98   // Find the Bin in the 2D Histogram
99   //
100   return species * fMomentumAxis->GetNbins() + (fMomentumAxis->FindBin(momentum) -1);
101 }
102
103 //________________________________________________________________________
104 inline Bool_t AliTRDcheckPID::IsInRange(Double_t momentum){
105   //
106   // Check Whether momentum is in the defined Range
107   //
108   return (momentum >= fMomentumAxis->GetXmin() && momentum <= fMomentumAxis->GetXmax());
109 }
110
111 #endif