New class for PID of HF candidates (R. Romita)
[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);
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