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 <TF1.h>
  14 #include "AliPID.h"
  15 #include "AliVParticle.h"
  16 #include "AliVTrack.h"
  17
  18 class AliTOFPIDParams;
  19 class TH1F;
  20 class TH1D;
  21
  22 class AliTOFPIDResponse : public TObject {
  23 public:
  24
  25   AliTOFPIDResponse();
  26   AliTOFPIDResponse(Double_t *param);
  27   ~AliTOFPIDResponse(){}
  28
  29   void     SetTimeResolution(Float_t res) { fSigma = res; }
  30   void     SetTimeZero(Double_t t0) { fTime0=t0; }
  31   Double_t GetTimeZero() const { return fTime0; }
  32
  33   void     SetMaxMismatchProbability(Double_t p) {fPmax=p;}
  34   Double_t GetMaxMismatchProbability() const {return fPmax;}
  35
  36   Double_t GetExpectedSigma(Float_t mom, Float_t tof, Float_t massZ) const;
  37   Double_t GetExpectedSigma(Float_t mom, Float_t tof, AliPID::EParticleType type) const;
  38   Double_t GetExpectedSignal(const AliVTrack *track, AliPID::EParticleType type) const;
  39
  40   Double_t GetMismatchProbability(Double_t time,Double_t eta) const;
  41
  42   static Double_t GetTailRandomValue(Float_t pt=1.0,Float_t eta=0.0,Float_t time=0.0,Float_t addmism=0.0); // generate a random value to add a tail to TOF time (for MC analyses), addmism = additional mismatch in percentile
  43   static Double_t GetMismatchRandomValue(Float_t eta); // generate a random value for mismatched tracks (for MC analyses)
  44
  45   void     SetT0event(Float_t *t0event){for(Int_t i=0;i < fNmomBins;i++) fT0event[i] = t0event[i];};
  46   void     SetT0resolution(Float_t *t0resolution){for(Int_t i=0;i < fNmomBins;i++) fT0resolution[i] = t0resolution[i];};
  47   void     ResetT0info(){ for(Int_t i=0;i < fNmomBins;i++){ fT0event[i] = 0.0; fT0resolution[i] = 0.0; fMaskT0[i] = 0;} };
  48   void     SetMomBoundary();
  49   Int_t    GetMomBin(Float_t p) const;
  50   Int_t    GetNmomBins(){return fNmomBins;};
  51   Float_t  GetMinMom(Int_t ibin) const {if(ibin >=0 && ibin < fNmomBins) return fPCutMin[ibin]; else return 0.0;}; // overrun static array - coverity
  52   Float_t  GetMaxMom(Int_t ibin) const {if(ibin >=0 && ibin < fNmomBins) return fPCutMin[ibin+1]; else return 0.0;}; // overrun static array - coverity
  53   void     SetT0bin(Int_t ibin,Float_t t0bin){if(ibin >=0 && ibin < fNmomBins) fT0event[ibin] = t0bin;}; // overrun static array - coverity
  54   void     SetT0binRes(Int_t ibin,Float_t t0binRes){if(ibin >=0 && ibin < fNmomBins) fT0resolution[ibin] = t0binRes;}; // overrun static array - coverity
  55   void     SetT0binMask(Int_t ibin,Int_t t0binMask){if(ibin >=0 && ibin < fNmomBins) fMaskT0[ibin] = t0binMask;}; // overrun static array - coverity
  56   Float_t  GetT0bin(Int_t ibin) const {if(ibin >=0 && ibin < fNmomBins) return fT0event[ibin]; else return 0.0;}; // overrun static array - coverity
  57   Float_t  GetT0binRes(Int_t ibin) const {if(ibin >=0 && ibin < fNmomBins) return fT0resolution[ibin]; else return 0.0;}; // overrun static array - coverity
  58   Int_t    GetT0binMask(Int_t ibin) const {if(ibin >=0 && ibin < fNmomBins) return fMaskT0[ibin]; else return 0;}; // overrun static array - coverity
  59
  60   // Get Start Time for a track
  61   Float_t  GetStartTime(Float_t mom) const;
  62   Float_t  GetStartTimeRes(Float_t mom) const;
  63   Int_t    GetStartTimeMask(Float_t mom) const;
  64
  65   // Tracking resolution for expected times
  66   void SetTrackParameter(Int_t ip,Float_t value){if(ip>=0 && ip < 4) fPar[ip] = value;};
  67   Float_t GetTrackParameter(Int_t ip){if(ip>=0 && ip < 4) return fPar[ip]; else return -1.0;};
  68   Int_t GetTOFchannel(AliVParticle *trk) const;
  69
  70   Float_t GetTOFtail() {if(fTOFtailResponse) return fTOFtailResponse->GetParameter(3);else return -1;};
  71   void    SetTOFtail(Float_t tail);
  72   void    SetTOFtailAllPara(Float_t mean,Float_t tail);
  73
  74  private:
  75   Double_t fSigma;        // intrinsic TOF resolution
  76
  77   // obsolete
  78   Double_t fPmax;         // "maximal" probability of mismathing (at ~0.5 GeV/c)
  79   Double_t fTime0;        // time zero
  80   //--------------
  81
  82   // About event time (t0) info
  83   static const Int_t fNmomBins = 10; // number of momentum bin
  84   Float_t fT0event[fNmomBins];    // t0 (best, T0, T0-TOF, ...) of the event as a function of p
  85   Float_t fT0resolution[fNmomBins]; // t0 (best, T0, T0-TOF, ...) resolution as a function of p
  86   Float_t fPCutMin[fNmomBins+1]; // min values for p bins
  87   Int_t fMaskT0[fNmomBins]; // mask withthe T0 used (0x1=T0-TOF,0x2=T0A,0x3=TOC) for p bins
  88   Float_t fPar[4]; // parameter for expected times resolution
  89
  90   static TF1 *fTOFtailResponse; // function to generate a TOF tail
  91   static TH1F *fHmismTOF; // TOF mismatch distribution
  92   static TH1D *fHchannelTOFdistr;// TOF channel distance distribution
  93
  94   ClassDef(AliTOFPIDResponse,5)   // TOF PID class
  95 };
  96
  97 #endif