]>
Commit | Line | Data |
---|---|---|
7ce8802c | 1 | #ifndef ALIAODPIDHF_H |
2 | #define ALIAODPIDHF_H | |
3 | ||
4 | /* Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. * | |
5 | * * See cxx source for full Copyright notice */ | |
6 | ||
7 | //*********************************************************** | |
8 | //// Class AliAODPidHF | |
9 | //// class for PID with AliAODRecoDecayHF | |
228f3aba | 10 | //// Authors: D. Caffarri caffarri@pd.infn.it, A.Dainese andrea.dainese@pd.infn.it, S. Dash dash@to.infn.it, F. Prino prino@to.infn.it, R. Romita r.romita@gsi.de, Y. Wang yifei@pi0.physi.uni-heidelberg.de |
7ce8802c | 11 | ////*********************************************************** |
12 | ||
13 | #include "AliAODPid.h" | |
14 | #include "AliAODTrack.h" | |
15 | ||
228f3aba | 16 | class AliAODPidHF : public AliAODPid{ |
7ce8802c | 17 | |
18 | public: | |
19 | ||
20 | AliAODPidHF(); | |
21 | AliAODPidHF(const AliAODPidHF& pid); | |
22 | AliAODPidHF& operator=(const AliAODPidHF& pid); | |
23 | virtual ~AliAODPidHF(); | |
24 | ||
25 | //Setters | |
79f3c225 | 26 | void SetSigma(Double_t *sigma){ |
27 | for(Int_t i=0; i<fnNSigma; i++) fnSigma[i]=sigma[i]; | |
28 | } | |
228f3aba | 29 | void SetSigma(Int_t idet,Double_t sigma){fnSigma[idet]=sigma;return;} |
79f3c225 | 30 | void SetSigmaForTPC(Double_t *sigma){for(Int_t i=0;i<3;i++) fnSigma[i]=sigma[i];return;} |
31 | void SetSigmaForTOF(Double_t sigma){fnSigma[3]=sigma;return;} | |
32 | void SetSigmaForITS(Double_t sigma){fnSigma[4]=sigma;return;} | |
228f3aba | 33 | void SetTofSigma(Double_t sigma){fTOFSigma=sigma;return;} |
7ce8802c | 34 | void SetPriors(Double_t *priors){fPriors=priors;return;} |
79f3c225 | 35 | void SetPLimit(Double_t *plim){for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;} |
36 | void SetPLimit(Double_t *plim,Int_t npLim){fnPLimit=npLim;for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;} | |
228f3aba | 37 | void SetAsym(Bool_t asym){fAsym=asym;return;} |
cdac45d3 | 38 | void SetTPC(Bool_t tpc){fTPC=tpc;return;} |
39 | void SetTOF(Bool_t tof){fTOF=tof;return;} | |
40 | void SetITS(Bool_t its){fITS=its;return;} | |
41 | void SetTRD(Bool_t trd){fTRD=trd;return;} | |
79f3c225 | 42 | void SetMatch(Int_t match){fMatch=match;return;} |
cdac45d3 | 43 | void SetCompat(Bool_t comp){fCompat=comp;return;} |
7ce8802c | 44 | |
79f3c225 | 45 | //Getters |
46 | Double_t GetSigma(Int_t idet){return fnSigma[idet];} | |
47 | Double_t GetTofSigma(){return fTOFSigma;} | |
48 | void GetPriors(Double_t *priors){priors=fPriors;return;} | |
49 | void GetPLimit(Double_t *plim){plim=fPLimit;} | |
50 | Bool_t GetAsym(){return fAsym;} | |
51 | Bool_t GetTPC(){return fTPC;} | |
52 | Bool_t GetTOF(){return fTOF;} | |
53 | Bool_t GetITS(){return fITS;} | |
54 | Bool_t GetTRD(){return fTRD;} | |
55 | Int_t GetMatch(){return fMatch;} | |
56 | Bool_t GetCompat(){return fCompat;} | |
7ce8802c | 57 | |
228f3aba | 58 | Int_t RawSignalPID (AliAODTrack *track, TString detector) const; |
59 | Bool_t IsKaonRaw (AliAODTrack *track, TString detector) const; | |
60 | Bool_t IsPionRaw (AliAODTrack *track, TString detector) const; | |
61 | Bool_t IsProtonRaw (AliAODTrack *track, TString detector) const; | |
62 | Bool_t IsElectronRaw (AliAODTrack *track, TString detector) const; | |
cdac45d3 | 63 | void BayesianProbability(AliAODTrack *track,Double_t *pid) const; |
228f3aba | 64 | void CombinedProbability(AliAODTrack *track,Bool_t *type) const; //0 = pion, 1 = kaon, 2 = proton |
65 | Bool_t CheckStatus(AliAODTrack *track,TString detectors) const; | |
66 | ||
67 | Bool_t TPCRawAsym(AliAODTrack* track,Int_t specie) const; | |
68 | Int_t MatchTPCTOF(AliAODTrack *track,Int_t mode,Int_t specie,Bool_t compat); | |
69 | ||
cdac45d3 | 70 | Int_t MakeRawPid(AliAODTrack *track,Int_t specie); //general method to perform PID using raw signals |
71 | ||
228f3aba | 72 | |
7ce8802c | 73 | |
74 | protected: | |
75 | ||
228f3aba | 76 | Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie) const; |
77 | Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie) const; | |
78 | Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie) const; | |
79 | void BayesianProbabilityITS(AliAODTrack *track,Double_t *prob) const; | |
80 | void BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob) const; | |
81 | void BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob) const; | |
82 | void BayesianProbabilityTRD(AliAODTrack *track,Double_t *prob) const; | |
7ce8802c | 83 | |
84 | private: | |
cdac45d3 | 85 | Int_t fnNSigma; // number of sigmas |
86 | Double_t *fnSigma; // [fnNSigma], sigma for the raw signal PID: 0-2 for TPC, 3 for TOF, 4 for ITS | |
228f3aba | 87 | Double_t fTOFSigma; // TOF precision |
cdac45d3 | 88 | Int_t fnPriors; //number of priors |
89 | Double_t *fPriors; // [fnPriors], set of priors | |
90 | Int_t fnPLimit; //number of Plimit | |
91 | Double_t *fPLimit; // [fnPLimit], limit of p intervals for asimmetric PID: fPLimit<p[0], fPLimit[0]<p<fPLimit[1], p>fPLimit[1] | |
228f3aba | 92 | Bool_t fAsym; // asimmetric PID required |
cdac45d3 | 93 | Bool_t fTPC; //switch to include or exclude TPC |
94 | Bool_t fTOF; // switch to include or exclude TOF | |
95 | Bool_t fITS; //switch to include or exclude ITS | |
96 | Bool_t fTRD; // switch to include or exclude TRD | |
79f3c225 | 97 | Int_t fMatch; //switch to combine the info from more detectors: 1 = || , 2 = &, 3 = p region |
cdac45d3 | 98 | Bool_t fCompat; // compatibility region : useful only if fMatch=1 |
228f3aba | 99 | |
7ce8802c | 100 | |
101 | ||
79f3c225 | 102 | ClassDef(AliAODPidHF,5) // AliAODPid for heavy flavor PID |
7ce8802c | 103 | |
104 | }; | |
105 | ||
106 | #endif |