Update (Rossella)
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAODPidHF.h
CommitLineData
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
16class 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