Changes related to PID mode and parameters configuration (Rossella, Francesco)
[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
228f3aba 10//// Authors: D. Caffarri caffarri@pd.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
7ce8802c 11////***********************************************************
12
13#include "AliAODPid.h"
14#include "AliAODTrack.h"
15
228f3aba 16class AliAODPidHF : public AliAODPid{
7ce8802c 17
18 public:
19
20 AliAODPidHF();
21 AliAODPidHF(const AliAODPidHF& pid);
22 AliAODPidHF& operator=(const AliAODPidHF& pid);
23 virtual ~AliAODPidHF();
24
25 //Setters
228f3aba 26 void SetSigma(Double_t *sigma){fnSigma=sigma;return;}
27 void SetSigma(Int_t idet,Double_t sigma){fnSigma[idet]=sigma;return;}
28 void SetTofSigma(Double_t sigma){fTOFSigma=sigma;return;}
7ce8802c 29 void SetPriors(Double_t *priors){fPriors=priors;return;}
cdac45d3 30// void SetPLimit(Double_t *plim){fPLimit=plim;return;}
31 void SetPLimit(Double_t *plim){for(Int_t i=0;i<2;i++){fPLimit[i]=plim[i];}return;}
228f3aba 32 void SetAsym(Bool_t asym){fAsym=asym;return;}
cdac45d3 33 void SetTPC(Bool_t tpc){fTPC=tpc;return;}
34 void SetTOF(Bool_t tof){fTOF=tof;return;}
35 void SetITS(Bool_t its){fITS=its;return;}
36 void SetTRD(Bool_t trd){fTRD=trd;return;}
37 void SetMatch(Bool_t match){fMatch=match;return;}
38 void SetCompat(Bool_t comp){fCompat=comp;return;}
7ce8802c 39
40
228f3aba 41 Int_t RawSignalPID (AliAODTrack *track, TString detector) const;
42 Bool_t IsKaonRaw (AliAODTrack *track, TString detector) const;
43 Bool_t IsPionRaw (AliAODTrack *track, TString detector) const;
44 Bool_t IsProtonRaw (AliAODTrack *track, TString detector) const;
45 Bool_t IsElectronRaw (AliAODTrack *track, TString detector) const;
cdac45d3 46 void BayesianProbability(AliAODTrack *track,Double_t *pid) const;
228f3aba 47 void CombinedProbability(AliAODTrack *track,Bool_t *type) const; //0 = pion, 1 = kaon, 2 = proton
48 Bool_t CheckStatus(AliAODTrack *track,TString detectors) const;
49
50 Bool_t TPCRawAsym(AliAODTrack* track,Int_t specie) const;
51 Int_t MatchTPCTOF(AliAODTrack *track,Int_t mode,Int_t specie,Bool_t compat);
52
cdac45d3 53 Int_t MakeRawPid(AliAODTrack *track,Int_t specie); //general method to perform PID using raw signals
54
228f3aba 55
7ce8802c 56
57 protected:
58
228f3aba 59 Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie) const;
60 Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie) const;
61 Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie) const;
62 void BayesianProbabilityITS(AliAODTrack *track,Double_t *prob) const;
63 void BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob) const;
64 void BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob) const;
65 void BayesianProbabilityTRD(AliAODTrack *track,Double_t *prob) const;
7ce8802c 66
67 private:
cdac45d3 68 Int_t fnNSigma; // number of sigmas
69 Double_t *fnSigma; // [fnNSigma], sigma for the raw signal PID: 0-2 for TPC, 3 for TOF, 4 for ITS
228f3aba 70 Double_t fTOFSigma; // TOF precision
cdac45d3 71 Int_t fnPriors; //number of priors
72 Double_t *fPriors; // [fnPriors], set of priors
73 Int_t fnPLimit; //number of Plimit
74 Double_t *fPLimit; // [fnPLimit], limit of p intervals for asimmetric PID: fPLimit<p[0], fPLimit[0]<p<fPLimit[1], p>fPLimit[1]
228f3aba 75 Bool_t fAsym; // asimmetric PID required
cdac45d3 76 Bool_t fTPC; //switch to include or exclude TPC
77 Bool_t fTOF; // switch to include or exclude TOF
78 Bool_t fITS; //switch to include or exclude ITS
79 Bool_t fTRD; // switch to include or exclude TRD
80 Bool_t fMatch; //switch to combine the info from more detectors: 1 = || , 2 = &, 3 = p region
81 Bool_t fCompat; // compatibility region : useful only if fMatch=1
228f3aba 82
7ce8802c 83
84
cdac45d3 85 ClassDef(AliAODPidHF,3) // AliAODPid for heavy flavor PID
7ce8802c 86
87};
88
89#endif