new PID 2dLQ implementation to fix bug 54540
[u/mrichter/AliRoot.git] / TRD / Cal / AliTRDCalPID.h
1 #ifndef ALITRDCALPID_H
2 #define ALITRDCALPID_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 ///////////////////////////////////////////////////////////////////////////////
9 //                                                                           //
10 // Authors:                                                                  //
11 //                                                                           //
12 //  Alex Bercuci <A.Bercuci@gsi.de>                                          //
13 //  Alex Wilk <wilka@uni-muenster.de>                                        //
14 //  Prashant Shukla <shukla@pi0.physi.uni-heidelberg.de>                     //
15 //                                                                           //
16 ///////////////////////////////////////////////////////////////////////////////
17
18 #include <TNamed.h>
19
20 #include "AliPID.h"
21
22 class AliTRDCalPID : public TNamed
23 {
24
25  public:
26
27   enum {
28     kNMom   = 11,
29     kNSlicesLQ = 2,
30     kNSlicesNN = 8
31   };
32
33   AliTRDCalPID();
34   AliTRDCalPID(const Text_t *name, const Text_t *title);
35   virtual         ~AliTRDCalPID();
36
37   virtual Bool_t   LoadReferences(Char_t *refFile) = 0;
38   static  Double_t GetMomentum(Int_t ip) { 
39     return (ip<0 || ip>=kNMom) ? -1.0 : fgTrackMomentum[ip]; }
40   static  Double_t GetMomentumBinning(Int_t ip) { 
41     return (ip<0 || ip>=kNMom+1) ? -1.0 : fgTrackMomentumBinning[ip]; }
42   virtual TObject *GetModel(Int_t ip, Int_t iType, Int_t iPlane) const = 0;
43   //virtual static Int_t GetModelID(Int_t mom, Int_t spec, Int_t plane) = 0;
44   virtual Double_t GetProbability(Int_t spec, Float_t mom
45                                 , const Float_t * const dedx
46                                 , Float_t length, Int_t plane) const = 0;
47   static  Color_t  GetPartColor(Int_t i)                    { return fgPartColor[i]; }
48   static  Int_t    GetPartIndex(Int_t pdg);
49   static  const Char_t  *GetPartName(Int_t i)               { return fPartName[i];   }
50   static  const Char_t  *GetPartSymb(Int_t i)               { return fPartSymb[i];   }
51
52           void     SetPartName(Int_t i, const Char_t *name) { fPartName[i] = name;   }
53           void     SetPartSymb(Int_t i, const Char_t *symb) { fPartSymb[i] = symb;   }
54
55  protected:
56   virtual void     Init() = 0;
57
58   static const Char_t   *fPartName[AliPID::kSPECIES];     //! Names of particle species
59   static const Char_t   *fPartSymb[AliPID::kSPECIES];     //! Symbols of particle species
60   static       Color_t   fgPartColor[AliPID::kSPECIES];   //! Colors of particle species
61   static       Float_t   fgTrackMomentum[kNMom];          //  Track momenta for which response functions are available
62   static       Float_t   fgTrackMomentumBinning[kNMom+1]; //  Defines the start and the endpoints of the momentum bins
63   TObjArray             *fModel;                          //  Model for probability estimate
64
65  private:
66
67   AliTRDCalPID(const AliTRDCalPID& pd);
68   AliTRDCalPID    &operator=(const AliTRDCalPID &c);
69
70   ClassDef(AliTRDCalPID, 4)                               //  Base class for TRD PID methods
71
72 };
73 #endif