8779d877e1089829ee75f6c0a9647d2a0582ec2d
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAODPidHF.h
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);
34  Bool_t IsElectronRaw (AliAODTrack *track, TString detector);
35  void BayesianProbability(AliAODTrack *track,TString detectors,Double_t *pid);
36  void CombinedProbability(AliAODTrack *track,Bool_t *type); //0 = pion, 1 = kaon, 2 = proton
37  Bool_t CheckStatus(AliAODTrack *track,TString detectors);
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);
47  void BayesianProbabilityTRD(AliAODTrack *track,Double_t *prob);
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