Coverity fix
[u/mrichter/AliRoot.git] / TRD / TRDbase / AliTRDpidUtil.h
1 #ifndef ALITRDPIDUTIL_H
2 #define ALITRDPIDUTIL_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id: AliTRDpidUtil.h 42548 2010-07-27 08:10:51Z cblume $ */
7
8 //////////////////////////////////////////////////////
9 //
10 // Class to calculate PID performance of the TRD
11 //
12 // Author : Alex Wilk <wilka@uni-muenster.de>
13 //
14 ///////////////////////////////////////////////////////
15
16 class TH1;
17 class AliESDtrack;
18 class AliTRDpidUtil : public TObject {
19 public:
20   enum {
21     kBins = 10001
22   };
23   enum ETRDPIDMethod {
24      kLQ   = 0 // 2D likelihood method
25     ,kNN   = 1 // Neural network method
26     ,kESD  = 2 // ESD results - check offline
27   };
28   enum{
29     kNNslices = 8
30    ,kLQslices = 3
31   };
32
33   AliTRDpidUtil();
34   virtual ~AliTRDpidUtil(){;}
35
36   Bool_t       CalculatePionEffi(TH1* histo1, TH1* histo2);
37
38   static Float_t  ElectronEfficiency()   { return fgEleEffi;};
39   
40   static Bool_t   IsElectron(const AliESDtrack *track, ETRDPIDMethod method = kNN);
41   static Double_t GetSystematicError(const AliESDtrack *track, ETRDPIDMethod method = kNN);
42   static Int_t GetNdEdxSlices(ETRDPIDMethod m)   { return m == kNN ? kNNslices : kLQslices;}
43   Double_t     GetCalcElectronEfficiency() const { return fCalcEleEffi;};
44   Double_t     GetPionEfficiency() const { return fPionEffi;};
45   Double_t     GetError() const          { return fError;};
46   Double_t     GetThreshold() const      { return fThreshold;};
47
48   static Int_t GetMomentumBin(Double_t p);
49   static Int_t Pdg2Pid(Int_t pdg);
50   static Int_t Mass2Pid(Float_t m);
51   static void  SetElectronEfficiency(Float_t eleeffi) {fgEleEffi = eleeffi;};
52
53 private:
54   AliTRDpidUtil(const AliTRDpidUtil&);               // not implemented
55   AliTRDpidUtil& operator=(const AliTRDpidUtil&);    // not implemented
56
57   static Float_t fgEleEffi;               // electron efficiency
58
59   Double_t fCalcEleEffi;                 // electron efficiency after calculation
60   Double_t fPionEffi;                    // pion efficiency 
61   Double_t fError;                       // pion efficiency error
62   Double_t fThreshold;                   // threshold for calculated electron efficiency
63
64   ClassDef(AliTRDpidUtil, 1)  // TRD PID efficiency calculator
65
66 };
67
68 #endif