]>
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 | |
228f3aba | 26 | void SetSigma(Double_t *sigma){fnSigma=sigma;return;} |
27 | void SetSigma(Int_t idet,Double_t sigma){fnSigma[idet]=sigma;return;} | |
28 | void SetTofSigma(Double_t sigma){fTOFSigma=sigma;return;} | |
7ce8802c | 29 | void SetPriors(Double_t *priors){fPriors=priors;return;} |
228f3aba | 30 | void SetPLimit(Double_t *plim){fPLimit=plim;return;} |
31 | void SetAsym(Bool_t asym){fAsym=asym;return;} | |
7ce8802c | 32 | |
33 | ||
228f3aba | 34 | Int_t RawSignalPID (AliAODTrack *track, TString detector) const; |
35 | Bool_t IsKaonRaw (AliAODTrack *track, TString detector) const; | |
36 | Bool_t IsPionRaw (AliAODTrack *track, TString detector) const; | |
37 | Bool_t IsProtonRaw (AliAODTrack *track, TString detector) const; | |
38 | Bool_t IsElectronRaw (AliAODTrack *track, TString detector) const; | |
39 | void BayesianProbability(AliAODTrack *track,TString detectors,Double_t *pid) const; | |
40 | void CombinedProbability(AliAODTrack *track,Bool_t *type) const; //0 = pion, 1 = kaon, 2 = proton | |
41 | Bool_t CheckStatus(AliAODTrack *track,TString detectors) const; | |
42 | ||
43 | Bool_t TPCRawAsym(AliAODTrack* track,Int_t specie) const; | |
44 | Int_t MatchTPCTOF(AliAODTrack *track,Int_t mode,Int_t specie,Bool_t compat); | |
45 | ||
46 | ||
7ce8802c | 47 | |
48 | protected: | |
49 | ||
228f3aba | 50 | Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie) const; |
51 | Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie) const; | |
52 | Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie) const; | |
53 | void BayesianProbabilityITS(AliAODTrack *track,Double_t *prob) const; | |
54 | void BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob) const; | |
55 | void BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob) const; | |
56 | void BayesianProbabilityTRD(AliAODTrack *track,Double_t *prob) const; | |
7ce8802c | 57 | |
58 | private: | |
53c43440 | 59 | |
60 | Int_t fnNSigma; // size of the nsigma array | |
61 | Double_t *fnSigma; //[fnNSigma] sigma for the raw signal PID: 0-2 for TPC, 3 for TOF, 4 for ITS | |
228f3aba | 62 | Double_t fTOFSigma; // TOF precision |
53c43440 | 63 | Int_t fnPriors; // size of the priors array |
64 | Double_t *fPriors; //[fnPriors] set of priors | |
65 | Int_t fnPLimit; // size of the plimit array | |
66 | Double_t *fPLimit; //[fnPLimit] limit of p intervals for asimmetric PID: fPLimit<p[0], fPLimit[0]<p<fPLimit[1], p>fPLimit[1] | |
228f3aba | 67 | Bool_t fAsym; // asimmetric PID required |
68 | ||
7ce8802c | 69 | |
70 | ||
53c43440 | 71 | ClassDef(AliAODPidHF,4) // AliAODPid for heavy flavor PID |
7ce8802c | 72 | |
73 | }; | |
74 | ||
75 | #endif |