STEER/AliAODpidUtil.h

   1 #ifndef ALIAODPIDUTIL_H
   2 #define ALIAODPIDUTIL_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id: AliAODpidUtil.h 38493 2010-01-26 16:33:03Z hristov $ */
   7
   8 //-------------------------------------------------------
   9 //                    Combined PID class
  10 //                    for the AOD class
  11 //   Origin: Rosa Romita, GSI, r.romita@gsi.de
  12 //-------------------------------------------------------
  13 #include <Rtypes.h>
  14 #include <TMatrixD.h>
  15 #include "AliAODTrack.h" // Needed for inline functions
  16 #include "AliAODPid.h" // Needed for inline functions
  17 #include "AliTPCPIDResponse.h"
  18 #include "AliITSPIDResponse.h"
  19 #include "AliTOFPIDResponse.h"
  20 #include "AliTRDPIDResponse.h"
  21 //#include "HMPID/AliHMPID.h"
  22
  23 class AliAODEvent;
  24
  25 class AliAODpidUtil {
  26 public:
  27   AliAODpidUtil(): fRange(5.), fTPCResponse(), fITSResponse(), fTOFResponse(){;}
  28   virtual ~AliAODpidUtil() {;}
  29
  30
  31   Int_t MakePID(AliAODTrack *track,Float_t TimeZeroTOF,Double_t *p) const;
  32   void MakeTPCPID(AliAODTrack *track,Double_t *p) const;
  33   void MakeITSPID(AliAODTrack *track,Double_t *p) const;
  34   void MakeTOFPID(AliAODTrack *track, Float_t TimeZeroTOF,Double_t *p) const;
  35   //  void MakeHMPIDPID(AliESDtrack *track);
  36   void MakeTRDPID(AliAODTrack *track,Double_t *p) const;
  37
  38   Float_t NumberOfSigmasTPC(const AliAODTrack *track, AliPID::EParticleType type) const;
  39   Float_t NumberOfSigmasTOF(const AliAODTrack *track, AliPID::EParticleType type, const Float_t TimeZeroTOF) const;
  40   Float_t NumberOfSigmasITS(const AliAODTrack *track, AliPID::EParticleType type) const;
  41
  42   AliITSPIDResponse &GetITSResponse() {return fITSResponse;}
  43   AliTPCPIDResponse &GetTPCResponse() {return fTPCResponse;}
  44   AliTOFPIDResponse &GetTOFResponse() {return fTOFResponse;}
  45
  46 private:
  47   Float_t           fRange;          // nSigma max in likelihood
  48   AliTPCPIDResponse fTPCResponse;
  49   AliITSPIDResponse fITSResponse;
  50   AliTOFPIDResponse fTOFResponse;
  51   // AliHMPIDPIDResponse fHMPIDResponse;
  52  // AliTRDPIDResponse fTRDResponse;
  53
  54   ClassDef(AliAODpidUtil,1)  // PID calculation class
  55 };
  56
  57 inline Float_t AliAODpidUtil::NumberOfSigmasTPC(const AliAODTrack *track, AliPID::EParticleType type) const {
  58
  59   Double_t mom = track->P();
  60   AliAODPid *pidObj = track->GetDetPid();
  61   if (pidObj)
  62     mom = pidObj->GetTPCmomentum();
  63   return fTPCResponse.GetNumberOfSigmas(mom,pidObj->GetTPCsignal(),0,type);
  64 }
  65
  66 inline Float_t AliAODpidUtil::NumberOfSigmasTOF(const AliAODTrack *track, AliPID::EParticleType type, const Float_t TimeZeroTOF) const {
  67   Double_t times[AliPID::kSPECIES];
  68   AliAODPid *pidObj = track->GetDetPid();
  69   pidObj->GetIntegratedTimes(times);
  70   return (pidObj->GetTOFsignal() - TimeZeroTOF - times[type])/fTOFResponse.GetExpectedSigma(track->P(),times[type],AliPID::ParticleMass(type));
  71 }
  72
  73 inline Float_t AliAODpidUtil::NumberOfSigmasITS(const AliAODTrack *track, AliPID::EParticleType type) const {
  74   AliAODPid *pidObj = track->GetDetPid();
  75   return fITSResponse.GetNumberOfSigmas(track->P(),pidObj->GetITSsignal(),type);
  76 }
  77 #endif
  78
  79