1f793616257511fb2f301672cfef8e154b806eac
[u/mrichter/AliRoot.git] / STEER / STEERBase / AliTOFPIDResponse.h
1 #ifndef ALITOFPIDRESPONSE_H
2 #define ALITOFPIDRESPONSE_H
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 //-------------------------------------------------------
8 //                    TOF PID class
9 //   Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch 
10 //-------------------------------------------------------
11
12 #include "TObject.h"
13 #include "AliPID.h"
14
15 class AliTOFPIDParams;
16
17 class AliTOFPIDResponse : public TObject {
18 public:
19
20   AliTOFPIDResponse();
21   AliTOFPIDResponse(Double_t *param);
22   ~AliTOFPIDResponse(){}
23
24   void     SetTimeResolution(Float_t res) { fSigma = res; }
25   void     SetTimeZero(Double_t t0) { fTime0=t0; }
26   Double_t GetTimeZero() const { return fTime0; }
27
28   void     SetMaxMismatchProbability(Double_t p) {fPmax=p;}
29   Double_t GetMaxMismatchProbability() const {return fPmax;}
30
31   Double_t GetExpectedSigma(Float_t mom, Float_t tof, Float_t mass) const;
32
33
34   Double_t GetMismatchProbability(Double_t p,Double_t mass) const;
35
36   void     SetT0event(Float_t *t0event){for(Int_t i=0;i < fNmomBins;i++) fT0event[i] = t0event[i];};
37   void     SetT0resolution(Float_t *t0resolution){for(Int_t i=0;i < fNmomBins;i++) fT0resolution[i] = t0resolution[i];};
38   void     ResetT0info(){ for(Int_t i=0;i < fNmomBins;i++){ fT0event[i] = 0.0; fT0resolution[i] = 0.0; fMaskT0[i] = 0;} };
39   void     SetMomBoundary();
40   Int_t    GetMomBin(Float_t p) const;
41   Int_t    GetNmomBins(){return fNmomBins;};
42   Float_t  GetMinMom(Int_t ibin) const {if(ibin >=0 && ibin < fNmomBins) return fPCutMin[ibin]; else return 0.0;}; // overrun static array - coverity
43   Float_t  GetMaxMom(Int_t ibin) const {if(ibin >=0 && ibin < fNmomBins) return fPCutMin[ibin+1]; else return 0.0;}; // overrun static array - coverity
44   void     SetT0bin(Int_t ibin,Float_t t0bin){if(ibin >=0 && ibin < fNmomBins) fT0event[ibin] = t0bin;}; // overrun static array - coverity
45   void     SetT0binRes(Int_t ibin,Float_t t0binRes){if(ibin >=0 && ibin < fNmomBins) fT0resolution[ibin] = t0binRes;}; // overrun static array - coverity
46   void     SetT0binMask(Int_t ibin,Int_t t0binMask){if(ibin >=0 && ibin < fNmomBins) fMaskT0[ibin] = t0binMask;}; // overrun static array - coverity
47   Float_t  GetT0bin(Int_t ibin) const {if(ibin >=0 && ibin < fNmomBins) return fT0event[ibin]; else return 0.0;}; // overrun static array - coverity
48   Float_t  GetT0binRes(Int_t ibin) const {if(ibin >=0 && ibin < fNmomBins) return fT0resolution[ibin]; else return 0.0;}; // overrun static array - coverity
49   Int_t    GetT0binMask(Int_t ibin) const {if(ibin >=0 && ibin < fNmomBins) return fMaskT0[ibin]; else return 0;}; // overrun static array - coverity
50
51   // Get Start Time for a track
52   Float_t  GetStartTime(Float_t mom) const;
53   Float_t  GetStartTimeRes(Float_t mom) const;
54   Int_t    GetStartTimeMask(Float_t mom) const;
55
56   // Tracking resolution for expected times
57   void SetTrackParameter(Int_t ip,Float_t value){if(ip>=0 && ip < 4) fPar[ip] = value;};
58   Float_t GetTrackParameter(Int_t ip){if(ip>=0 && ip < 4) return fPar[ip]; else return -1.0;};
59
60  private:
61   Double_t fSigma;        // intrinsic TOF resolution
62
63   // obsolete
64   Double_t fPmax;         // "maximal" probability of mismathing (at ~0.5 GeV/c)
65   Double_t fTime0;        // time zero
66   //--------------
67
68   // About event time (t0) info
69   static const Int_t fNmomBins = 10; // number of momentum bin 
70   Float_t fT0event[fNmomBins];    // t0 (best, T0, T0-TOF, ...) of the event as a function of p 
71   Float_t fT0resolution[fNmomBins]; // t0 (best, T0, T0-TOF, ...) resolution as a function of p 
72   Float_t fPCutMin[fNmomBins+1]; // min values for p bins
73   Int_t fMaskT0[fNmomBins]; // mask withthe T0 used (0x1=T0-TOF,0x2=T0A,0x3=TOC) for p bins
74   Float_t fPar[4]; // parameter for expected times resolution
75
76   ClassDef(AliTOFPIDResponse,4)   // TOF PID class
77 };
78
79 #endif