X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=STEER%2FSTEERBase%2FAliPIDResponse.h;h=ff2438675b877b2651f7d10d6546311e4d269836;hb=42fcc729e9f737175e6000f85deb0aa75075ec0b;hp=78e7053802940fc4c9519bfb97cb1f943589221d;hpb=0b39f221476ed8cb72ab95bb3d3494eb7ab81ad4;p=u%2Fmrichter%2FAliRoot.git diff --git a/STEER/STEERBase/AliPIDResponse.h b/STEER/STEERBase/AliPIDResponse.h index 78e70538029..ff2438675b8 100644 --- a/STEER/STEERBase/AliPIDResponse.h +++ b/STEER/STEERBase/AliPIDResponse.h @@ -11,25 +11,45 @@ // Origin: Jens Wiechula, Uni Tuebingen, jens.wiechula@cern.ch // //---------------------------------------------------------------// +#include "AliVParticle.h" +#include "AliVTrack.h" + #include "AliITSPIDResponse.h" #include "AliTPCPIDResponse.h" #include "AliTRDPIDResponse.h" #include "AliTOFPIDResponse.h" +#include "AliHMPIDPIDResponse.h" #include "AliEMCALPIDResponse.h" - -#include "AliVParticle.h" -#include "AliVTrack.h" +#include "AliPID.h" #include "TNamed.h" -class AliVEvent; class TF1; +class TObjArray; +class TLinearFitter; + +class AliVEvent; +class AliTRDPIDResponseObject; +class AliTOFPIDParams; +class AliHMPIDPIDParams; +class AliOADBContainer; class AliPIDResponse : public TNamed { public: AliPIDResponse(Bool_t isMC=kFALSE); virtual ~AliPIDResponse(); + enum EDetector { + kITS=0, + kTPC=1, + kTRD=2, + kTOF=3, + kHMPID=4, + kEMCAL=5, + kPHOS=6, + kNdetectors=7 + }; + enum EDetCode { kDetITS = 0x1, kDetTPC = 0x2, @@ -40,6 +60,12 @@ public: kDetPHOS = 0x40 }; + enum EBeamType { + kPP = 0, + kPPB, + kPBPB + }; + enum EStartTimeType_t {kFILL_T0,kTOF_T0, kT0_T0, kBest_T0}; enum ITSPIDmethod { kITSTruncMean, kITSLikelihood }; @@ -47,42 +73,79 @@ public: enum EDetPidStatus { kDetNoSignal=0, kDetPidOk=1, - kDetMismatch=2 + kDetMismatch=2, + kDetNoParams=3 }; - + AliITSPIDResponse &GetITSResponse() {return fITSResponse;} AliTPCPIDResponse &GetTPCResponse() {return fTPCResponse;} AliTOFPIDResponse &GetTOFResponse() {return fTOFResponse;} AliTRDPIDResponse &GetTRDResponse() {return fTRDResponse;} AliEMCALPIDResponse &GetEMCALResponse() {return fEMCALResponse;} - Float_t NumberOfSigmas(EDetCode detCode, const AliVParticle *track, AliPID::EParticleType type) const; + // ----------------------------------------- + // buffered getters + // - virtual Float_t NumberOfSigmasITS(const AliVParticle *track, AliPID::EParticleType type) const; - virtual Float_t NumberOfSigmasTPC(const AliVParticle *track, AliPID::EParticleType type) const; - virtual Float_t NumberOfSigmasEMCAL(const AliVTrack *track, AliPID::EParticleType type) const; - virtual Float_t NumberOfSigmasTOF(const AliVParticle *track, AliPID::EParticleType type) const = 0; - virtual Bool_t IdentifiedAsElectronTRD(const AliVTrack *track, Double_t efficiencyLevel) const; - - EDetPidStatus ComputePIDProbability (EDetCode detCode, const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + // Number of sigmas + EDetPidStatus NumberOfSigmas(EDetector detCode, const AliVParticle *track, AliPID::EParticleType type, Double_t &val) const; + + Float_t NumberOfSigmas(EDetector detCode, const AliVParticle *track, AliPID::EParticleType type) const; - EDetPidStatus ComputeITSProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; - EDetPidStatus ComputeTPCProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; - EDetPidStatus ComputeTOFProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; - EDetPidStatus ComputeTRDProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; - EDetPidStatus ComputeEMCALProbability(const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; - EDetPidStatus ComputePHOSProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; - EDetPidStatus ComputeHMPIDProbability(const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + virtual Float_t NumberOfSigmasITS (const AliVParticle *track, AliPID::EParticleType type) const; + virtual Float_t NumberOfSigmasTPC (const AliVParticle *track, AliPID::EParticleType type) const; + virtual Float_t NumberOfSigmasTPC (const AliVParticle *track, AliPID::EParticleType type, AliTPCPIDResponse::ETPCdEdxSource dedxSource) const; + virtual Float_t NumberOfSigmasEMCAL(const AliVParticle *track, AliPID::EParticleType type, Double_t &eop, Double_t showershape[4]) const; + virtual Float_t NumberOfSigmasTOF (const AliVParticle *track, AliPID::EParticleType type) const; + virtual Float_t NumberOfSigmasTOF (const AliVParticle *track, AliPID::EParticleType type, Float_t /*timeZeroTOF*/) const { return NumberOfSigmasTOF(track,type); } + virtual Float_t NumberOfSigmasHMPID(const AliVParticle *track, AliPID::EParticleType type) const; + virtual Float_t NumberOfSigmasEMCAL(const AliVParticle *track, AliPID::EParticleType type) const; + + Bool_t IdentifiedAsElectronTRD(const AliVTrack *track, Double_t efficiencyLevel,Double_t centrality=-1,AliTRDPIDResponse::ETRDPIDMethod PIDmethod=AliTRDPIDResponse::kLQ1D) const; + + // Signal delta + EDetPidStatus GetSignalDelta(EDetector detCode, const AliVParticle *track, AliPID::EParticleType type, Double_t &val, Bool_t ratio=kFALSE) const; + Double_t GetSignalDelta(EDetector detCode, const AliVParticle *track, AliPID::EParticleType type, Bool_t ratio=kFALSE) const; + + // Probabilities + EDetPidStatus ComputePIDProbability (EDetCode detCode, const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + EDetPidStatus ComputePIDProbability (EDetector detCode, const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + + virtual EDetPidStatus ComputeITSProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + virtual EDetPidStatus ComputeTPCProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + virtual EDetPidStatus ComputeTOFProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + virtual EDetPidStatus ComputeTRDProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + virtual EDetPidStatus ComputeEMCALProbability(const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + virtual EDetPidStatus ComputePHOSProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + virtual EDetPidStatus ComputeHMPIDProbability(const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + + virtual EDetPidStatus ComputeTRDProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[],AliTRDPIDResponse::ETRDPIDMethod PIDmethod) const; + + // pid status + EDetPidStatus CheckPIDStatus(EDetector detCode, const AliVTrack *track) const; + AliTOFPIDParams *GetTOFPIDParams() const {return fTOFPIDParams;} + Float_t GetTOFMismatchProbability(const AliVTrack *track = NULL) const; // if empty argument return the value stored during TOF probability computation void SetITSPIDmethod(ITSPIDmethod pmeth) { fITSPIDmethod = pmeth; } - virtual void SetTOFResponse(AliVEvent */*event*/,EStartTimeType_t /*option*/) {;} - void SetTRDslicesForPID(UInt_t slice1, UInt_t slice2) {fTRDslicesForPID[0]=slice1;fTRDslicesForPID[1]=slice2;} void SetOADBPath(const char* path) {fOADBPath=path;} const char *GetOADBPath() const {return fOADBPath.Data();} - void InitialiseEvent(AliVEvent *event, Int_t pass); + + void SetCustomTPCpidResponse(const char* tpcpid) { fCustomTPCpidResponse = tpcpid; } + const char* GetCustomTPCpidResponse() const { return fCustomTPCpidResponse.Data(); } + + void InitialiseEvent(AliVEvent *event, Int_t pass, Int_t run=-1); void SetCurrentFile(const char* file) { fCurrentFile=file; } + + void SetCurrentAliRootRev(Int_t alirootRev) { fCurrentAliRootRev = alirootRev; } + Int_t GetCurrentAliRootRev() const { return fCurrentAliRootRev; } + + // cache PID in the track + void SetCachePID(Bool_t cache) { fCachePID=cache; } + Bool_t GetCachePID() const { return fCachePID; } + void FillTrackDetectorPID(const AliVTrack *track, EDetector detector) const; + void FillTrackDetectorPID(); AliVEvent * GetCurrentEvent() const {return fCurrentEvent;} @@ -92,55 +155,97 @@ public: // event info Float_t GetCurrentCentrality() const {return fCurrCentrality;}; - + void SetCurrentCentrality(Float_t centrality) {fCurrCentrality=centrality;fEMCALResponse.SetCentrality(fCurrCentrality);}; + // TPC setting + void SetUseTPCEtaCorrection(Bool_t useEtaCorrection = kTRUE) { fUseTPCEtaCorrection = useEtaCorrection; }; + Bool_t UseTPCEtaCorrection() const { return fUseTPCEtaCorrection; }; + + void SetUseTPCMultiplicityCorrection(Bool_t useMultiplicityCorrection = kTRUE) { fUseTPCMultiplicityCorrection = useMultiplicityCorrection; }; + Bool_t UseTPCMultiplicityCorrection() const { return fUseTPCMultiplicityCorrection; }; + // TOF setting - void SetTOFtail(Float_t tail=1.1){if(tail > 0) fTOFtail=tail; else printf("TOF tail should be greater than 0 (nothing done)\n");}; - + void SetTOFtail(Float_t tail=0.9){if(tail > 0) fTOFtail=tail; else printf("TOF tail should be greater than 0 (nothing done)\n");}; + void SetTOFResponse(AliVEvent *vevent,EStartTimeType_t option); + + virtual Float_t GetTPCsignalTunedOnData(const AliVTrack *t) const {return t->GetTPCsignal();}; + virtual Float_t GetTOFsignalTunedOnData(const AliVTrack *t) const {return t->GetTOFsignal();}; + Bool_t IsTunedOnData() const {return fTuneMConData;}; + void SetTunedOnData(Bool_t flag=kTRUE,Int_t recoPass=0){fTuneMConData = flag; if(recoPass>0) fRecoPassUser = recoPass;}; + Int_t GetTunedOnDataMask() const {return fTuneMConDataMask;}; + void SetTunedOnDataMask(Int_t detMask) {fTuneMConDataMask = detMask;} + + // Utilities + TString GetChecksum(const TObject* obj) const; + AliPIDResponse(const AliPIDResponse &other); AliPIDResponse& operator=(const AliPIDResponse &other); - + protected: - AliITSPIDResponse fITSResponse; //PID response function of the ITS - AliTPCPIDResponse fTPCResponse; //PID response function of the TPC - AliTRDPIDResponse fTRDResponse; //PID response function of the TRD - AliTOFPIDResponse fTOFResponse; //PID response function of the TOF + AliITSPIDResponse fITSResponse; //PID response function of the ITS + AliTPCPIDResponse fTPCResponse; //PID response function of the TPC + AliTRDPIDResponse fTRDResponse; //PID response function of the TRD + AliTOFPIDResponse fTOFResponse; //PID response function of the TOF + AliHMPIDPIDResponse fHMPIDResponse; //PID response function of the HMPID AliEMCALPIDResponse fEMCALResponse; //PID response function of the EMCAL Float_t fRange; // nSigma max in likelihood ITSPIDmethod fITSPIDmethod; // 0 = trunc mean; 1 = likelihood + //unbuffered PID calculation + virtual Float_t GetNumberOfSigmasTOFold (const AliVParticle */*track*/, AliPID::EParticleType /*type*/) const {return 0;} + virtual Float_t GetSignalDeltaTOFold(const AliVParticle */*track*/, AliPID::EParticleType /*type*/, Bool_t /*ratio*/=kFALSE) const {return -9999.;} + + EDetPidStatus GetComputeTRDProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[],AliTRDPIDResponse::ETRDPIDMethod PIDmethod=AliTRDPIDResponse::kLQ1D) const; + EDetPidStatus GetTOFPIDStatus(const AliVTrack *track) const; + + Bool_t fTuneMConData; // switch to force the MC to be similar to data + Int_t fTuneMConDataMask; // select for which detectors enable MC tuning on data + + private: + static Float_t fgTOFmismatchProb; // TOF mismatch probability (Bayesian) + Bool_t fIsMC; // If we run on MC data + Bool_t fCachePID; TString fOADBPath; // OADB path to use + TString fCustomTPCpidResponse; // Custom TPC Pid Response file for debugging purposes TString fBeamType; //! beam type (PP) or (PBPB) TString fLHCperiod; //! LHC period TString fMCperiodTPC; //! corresponding MC period to use for the TPC splines TString fMCperiodUser; // MC prodution requested by the user TString fCurrentFile; //! name of currently processed file + Int_t fCurrentAliRootRev; //! Aliroot rev. used to reconstruct the data Int_t fRecoPass; //! reconstruction pass Int_t fRecoPassUser; // reconstruction pass explicitly set by the user Int_t fRun; //! current run number Int_t fOldRun; //! current run number + Float_t fResT0A; //! T0A resolution in current run + Float_t fResT0C; //! T0C resolution in current run + Float_t fResT0AC; //! T0A.and.T0C resolution in current run - TObjArray *fArrPidResponseMaster; //! TPC pid splines - TF1 *fResolutionCorrection; //! TPC resolution correction + TObjArray *fArrPidResponseMaster; //! TPC pid splines + TF1 *fResolutionCorrection; //! TPC resolution correction + AliOADBContainer* fOADBvoltageMaps; //! container with the voltage maps + Bool_t fUseTPCEtaCorrection; // Use TPC eta correction + Bool_t fUseTPCMultiplicityCorrection; // Use TPC multiplicity correction - AliTRDPIDParams *fTRDPIDParams; //! TRD PID Params - AliTRDPIDReference *fTRDPIDReference; //! TRD PID References - UInt_t fTRDslicesForPID[2]; //! TRD PID slices + AliTRDPIDResponseObject *fTRDPIDResponseObject; //! TRD PID Response Object - Int_t fTOFTimeZeroType; //! default start time type for tof (ESD) - Float_t fTOFres; //! TOF resolution Float_t fTOFtail; //! TOF tail effect used in TOF probability + AliTOFPIDParams *fTOFPIDParams; //! TOF PID Params - period depending (OADB loaded) + + AliHMPIDPIDParams *fHMPIDPIDParams; //! HMPID PID Params (OADB loaded) TObjArray *fEMCALPIDParams; //! EMCAL PID Params AliVEvent *fCurrentEvent; //! event currently being processed Float_t fCurrCentrality; //! current centrality + + EBeamType fBeamTypeNum; //! beam type enum void ExecNewRun(); @@ -152,15 +257,28 @@ private: void SetITSParametrisation(); //TPC + void SetTPCEtaMaps(Double_t refineFactorMapX = 6.0, Double_t refineFactorMapY = 6.0, Double_t refineFactorSigmaMapX = 6.0, + Double_t refineFactorSigmaMapY = 6.0); void SetTPCPidResponseMaster(); void SetTPCParametrisation(); Double_t GetTPCMultiplicityBin(const AliVEvent * const event); + + // TPC helpers for the eta maps + void AddPointToHyperplane(TH2D* h, TLinearFitter* linExtrapolation, Int_t binX, Int_t binY); + TH2D* RefineHistoViaLinearInterpolation(TH2D* h, Double_t refineFactorX = 6.0, Double_t refineFactorY = 6.0); //TRD void SetTRDPidResponseMaster(); void InitializeTRDResponse(); + void SetTRDSlices(UInt_t TRDslicesForPID[2],AliTRDPIDResponse::ETRDPIDMethod method) const; //TOF + void SetTOFPidResponseMaster(); + void InitializeTOFResponse(); + + //HMPID + void SetHMPIDPidResponseMaster(); + void InitializeHMPIDResponse(); //EMCAL void SetEMCALPidResponseMaster(); @@ -168,36 +286,47 @@ private: // void SetRecoInfo(); + + //------------------------------------------------- + //unbuffered PID calculation + // - ClassDef(AliPIDResponse,5); //PID response handling + // Number of sigmas + Float_t GetNumberOfSigmas(EDetector detCode, const AliVParticle *track, AliPID::EParticleType type) const; + Float_t GetNumberOfSigmasITS (const AliVParticle *track, AliPID::EParticleType type) const; + Float_t GetNumberOfSigmasTPC (const AliVParticle *track, AliPID::EParticleType type) const; + Float_t GetNumberOfSigmasTOF (const AliVParticle *track, AliPID::EParticleType type) const; + Float_t GetNumberOfSigmasHMPID(const AliVParticle *track, AliPID::EParticleType type) const; + Float_t GetNumberOfSigmasEMCAL(const AliVParticle *track, AliPID::EParticleType type, Double_t &eop, Double_t showershape[4]) const; + Float_t GetNumberOfSigmasEMCAL(const AliVParticle *track, AliPID::EParticleType type) const; + + Float_t GetBufferedNumberOfSigmas(EDetector detCode, const AliVParticle *track, AliPID::EParticleType type) const; + + // Signal deltas + EDetPidStatus GetSignalDeltaITS(const AliVParticle *track, AliPID::EParticleType type, Double_t &val, Bool_t ratio=kFALSE) const; + EDetPidStatus GetSignalDeltaTPC(const AliVParticle *track, AliPID::EParticleType type, Double_t &val, Bool_t ratio=kFALSE) const; + EDetPidStatus GetSignalDeltaTOF(const AliVParticle *track, AliPID::EParticleType type, Double_t &val, Bool_t ratio=kFALSE) const; + EDetPidStatus GetSignalDeltaHMPID(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio=kFALSE) const; + + // Probabilities + EDetPidStatus GetComputePIDProbability (EDetector detCode, const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + EDetPidStatus GetComputeITSProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + EDetPidStatus GetComputeTPCProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + EDetPidStatus GetComputeTOFProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + EDetPidStatus GetComputeEMCALProbability(const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + EDetPidStatus GetComputePHOSProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + EDetPidStatus GetComputeHMPIDProbability(const AliVTrack *track, Int_t nSpecies, Double_t p[]) const; + + // pid status + EDetPidStatus GetPIDStatus(EDetector det, const AliVTrack *track) const; + EDetPidStatus GetITSPIDStatus(const AliVTrack *track) const; + EDetPidStatus GetTPCPIDStatus(const AliVTrack *track) const; + EDetPidStatus GetTRDPIDStatus(const AliVTrack *track) const; + EDetPidStatus GetHMPIDPIDStatus(const AliVTrack *track) const; + EDetPidStatus GetPHOSPIDStatus(const AliVTrack *track) const; + EDetPidStatus GetEMCALPIDStatus(const AliVTrack *track) const; + + ClassDef(AliPIDResponse, 13); //PID response handling }; -inline Float_t AliPIDResponse::NumberOfSigmasTPC(const AliVParticle *vtrack, AliPID::EParticleType type) const { - AliVTrack *track=(AliVTrack*)vtrack; - Double_t mom = track->GetTPCmomentum(); - Double_t sig = track->GetTPCsignal(); - UInt_t sigN = track->GetTPCsignalN(); - - Double_t nSigma = -999.; - if (sigN>0) nSigma=fTPCResponse.GetNumberOfSigmas(mom,sig,sigN,type); - - return nSigma; -} - -inline Float_t AliPIDResponse::NumberOfSigmasITS(const AliVParticle *vtrack, AliPID::EParticleType type) const { - AliVTrack *track=(AliVTrack*)vtrack; - Float_t dEdx=track->GetITSsignal(); - if (dEdx<=0) return -999.; - - UChar_t clumap=track->GetITSClusterMap(); - Int_t nPointsForPid=0; - for(Int_t i=2; i<6; i++){ - if(clumap&(1<P(); - Bool_t isSA=kTRUE; - if(track->GetTPCNcls()>0) isSA=kFALSE; - return fITSResponse.GetNumberOfSigmas(mom,dEdx,type,nPointsForPid,isSA); -} - #endif