New class for PID of HF candidates (R. Romita)
[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  void BayesianProbability(AliAODTrack *track,TString detectors,Double_t *pid);
35  void CombinedProbability(AliAODTrack *track,Bool_t *type); //0 = pion, 1 = kaon, 2 = proton
36
37  protected:
38
39  Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie);
40  Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie);
41  Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie);
42  void BayesianProbabilityITS(AliAODTrack *track,Double_t *prob);
43  void BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob);
44  void BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob);
45
46  private:
47  Double_t fSigma; // sigma for the raw signal PID 
48  Double_t *fPriors; // set of priors
49
50
51  ClassDef(AliAODPidHF,3)
52
53 };
54
55 #endif