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 |
10 | //// Authors: D. Caffarri caffarri@bo.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 |
11 | ////*********************************************************** |
12 | |
13 | #include "AliAODPid.h" |
14 | #include "AliAODTrack.h" |
15 | |
16 | class AliAODPidHF : public AliAODPid{ |
17 | |
18 | public: |
19 | |
20 | AliAODPidHF(); |
21 | AliAODPidHF(const AliAODPidHF& pid); |
22 | AliAODPidHF& operator=(const AliAODPidHF& pid); |
23 | virtual ~AliAODPidHF(); |
24 | |
25 | //Setters |
26 | void SetSigma(Double_t sigma){fSigma=sigma;return;} |
27 | void SetPriors(Double_t *priors){fPriors=priors;return;} |
28 | |
29 | |
30 | Int_t RawSignalPID (AliAODTrack *track, TString detector); |
31 | Bool_t IsKaonRaw (AliAODTrack *track, TString detector); |
32 | Bool_t IsPionRaw (AliAODTrack *track, TString detector); |
33 | Bool_t IsProtonRaw (AliAODTrack *track, TString detector); |
f9e3cf8e |
34 | Bool_t IsElectronRaw (AliAODTrack *track, TString detector); |
7ce8802c |
35 | void BayesianProbability(AliAODTrack *track,TString detectors,Double_t *pid); |
36 | void CombinedProbability(AliAODTrack *track,Bool_t *type); //0 = pion, 1 = kaon, 2 = proton |
f9e3cf8e |
37 | Bool_t CheckStatus(AliAODTrack *track,TString detectors); |
7ce8802c |
38 | |
39 | protected: |
40 | |
41 | Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie); |
42 | Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie); |
43 | Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie); |
44 | void BayesianProbabilityITS(AliAODTrack *track,Double_t *prob); |
45 | void BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob); |
46 | void BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob); |
f9e3cf8e |
47 | void BayesianProbabilityTRD(AliAODTrack *track,Double_t *prob); |
7ce8802c |
48 | |
49 | private: |
50 | Double_t fSigma; // sigma for the raw signal PID |
51 | Double_t *fPriors; // set of priors |
52 | |
53 | |
54 | ClassDef(AliAODPidHF,3) |
55 | |
56 | }; |
57 | |
58 | #endif |