1 #ifndef ALITRDCHECKPID_H
2 #define ALITRDCHECKPID_H
4 //////////////////////////////////////////////////////
6 // PID performance checker of the TRD
8 // Author : Alex Wilk <wilka@uni-muenster.de>
9 // Alex Bercuci <A.Bercuci@gsi.de>
10 // Markus Fasel <M.Fasel@gsi.de>
12 ///////////////////////////////////////////////////////
18 #ifndef ALITRDRECOTASK_H
19 #include "AliTRDrecoTask.h"
26 class AliTRDReconstructor;
28 class AliTRDcheckPID : public AliTRDrecoTask
31 // Plots registered for this task
33 kEfficiency = 0 // e Efficiency plot
34 ,kdEdx = 1 // dE/dx spectra
35 ,kdEdxSlice = 2 // dE/dx spectra
36 ,kPH = 3 // pulse height spectra
37 ,kNClus = 4 // number of clusters per track
38 ,kMomentum = 5 // momentum distribution
39 ,kMomentumBin = 6 // momentum distribution
40 ,kNTracklets = 7 // Number of tracklets per track
41 ,kEfficiencyMu = 8 // mu Efficiency plot
42 ,kEfficiencyPi = 9 // pi Efficiency plot
43 ,kEfficiencyKa = 10 // K Efficiency plot
44 ,kEfficiencyPr = 11 // pr Efficiency plot
45 ,kV0 = 12 // V0 performance
46 ,kNPlots = 13 // Number of plots for this tasks
49 AliTRDcheckPID(char* name);
50 virtual ~AliTRDcheckPID();
52 virtual void UserCreateOutputObjects();
53 virtual Bool_t GetRefFigure(Int_t ifig);
54 virtual void UserExec(Option_t *opt);
55 virtual Bool_t PostProcess();
57 TH1 *PlotLQ(const AliTRDtrackV1 *track = 0x0);
58 TH1 *PlotNN(const AliTRDtrackV1 *track = 0x0);
59 TH1 *PlotESD(const AliTRDtrackV1 *track = 0x0);
60 TH1 *PlotdEdx(const AliTRDtrackV1 *track = 0x0);
61 TH1 *PlotdEdxSlice(const AliTRDtrackV1 *track = 0x0);
62 TH1 *PlotPH(const AliTRDtrackV1 *track = 0x0);
63 TH1 *PlotNClus(const AliTRDtrackV1 *track = 0x0);
64 TH1 *PlotNTracklets(const AliTRDtrackV1 *track = 0x0);
65 TH1 *PlotMom(const AliTRDtrackV1 *track = 0x0);
66 TH1 *PlotMomBin(const AliTRDtrackV1 *track = 0x0);
67 TH1 *PlotV0(const AliTRDtrackV1 *track = 0x0);
69 void SetRequireMinNTracklets(Int_t mintracklets) { fMinNTracklets = mintracklets; }
70 void SetRequireMaxNTracklets(Int_t maxtracklets) { fMaxNTracklets = maxtracklets; }
72 TObjArray *GetGraphs() const { return fGraph; };
73 static Char_t const* MethodName(Int_t id) { return fgMethod[id]; };
74 //TObjArray *GetHistos() { return fContainer; };
75 virtual TObjArray *Histos();
76 void EvaluateEfficiency(TObjArray* const histoContainer, TObjArray *results, Int_t species, Float_t electronEfficiency);
77 inline void SetMomentumBinning(Int_t nBins, Double_t *bins);
78 inline Int_t FindBin(Int_t species, Double_t momentum);
79 inline Bool_t IsInRange(Double_t momentum);
82 AliTRDcheckPID(const AliTRDcheckPID&); // not implemented
83 AliTRDcheckPID& operator=(const AliTRDcheckPID&); // not implemented
85 Int_t CalcPDG(AliTRDtrackV1* track = 0x0);
86 Bool_t CheckTrackQuality(const AliTRDtrackV1* track = 0x0) const;
89 static Char_t const *fgMethod[3]; // PID method name
90 AliTRDReconstructor *fReconstructor; // reconstructor needed for recalculation the PID
91 AliTRDpidUtil *fUtil; // utility class for PID calculations
92 TObjArray *fGraph; //! array of graphs filled in PostProcess
93 TObjArray *fPID; //! array of PID info/track for calibration
94 TObjArray *fV0s; //! array of V0 info
95 TObjArray *fEfficiency[AliPID::kSPECIES]; //! array of histograms with efficiency
96 TAxis *fMomentumAxis; // helper mementum binning
97 Int_t fMinNTracklets; // minimum number of required Tracklets (for systematic studies)
98 Int_t fMaxNTracklets; // maximum number of required Tracklets (for systematic studies)
99 ClassDef(AliTRDcheckPID, 2); // TRD PID checker
102 //________________________________________________________________________
103 inline void AliTRDcheckPID::SetMomentumBinning(Int_t nBins, Double_t *bins){
105 // Set the Momentum Bins
107 if(fMomentumAxis) delete fMomentumAxis;
108 fMomentumAxis = new TAxis(nBins, bins);
111 //________________________________________________________________________
112 inline Int_t AliTRDcheckPID::FindBin(Int_t species, Double_t momentum){
114 // Find the Bin in the 2D Histogram
116 return species * fMomentumAxis->GetNbins() + (fMomentumAxis->FindBin(momentum) -1);
119 //________________________________________________________________________
120 inline Bool_t AliTRDcheckPID::IsInRange(Double_t momentum){
122 // Check Whether momentum is in the defined Range
124 return (momentum >= fMomentumAxis->GetXmin() && momentum <= fMomentumAxis->GetXmax());