[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAODPidHF.h

#ifndef ALIAODPIDHF_H
#define ALIAODPIDHF_H

/* Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
 *  * See cxx source for full Copyright notice                               */

//***********************************************************
//// Class AliAODPidHF
//// class for PID with AliAODRecoDecayHF 
//// 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
////***********************************************************

#include "AliAODPid.h"
#include "AliAODTrack.h"

class AliAODPidHF : public AliAODPid{

 public:

 AliAODPidHF();
 AliAODPidHF(const AliAODPidHF& pid);
 AliAODPidHF& operator=(const AliAODPidHF& pid);
 virtual ~AliAODPidHF();

 //Setters
 void SetSigma(Double_t sigma){fSigma=sigma;return;}
 void SetPriors(Double_t *priors){fPriors=priors;return;}
 

 Int_t RawSignalPID (AliAODTrack *track, TString detector);
 Bool_t IsKaonRaw (AliAODTrack *track, TString detector);
 Bool_t IsPionRaw (AliAODTrack *track, TString detector);
 Bool_t IsProtonRaw (AliAODTrack *track, TString detector);
 Bool_t IsElectronRaw (AliAODTrack *track, TString detector);
 void BayesianProbability(AliAODTrack *track,TString detectors,Double_t *pid);
 void CombinedProbability(AliAODTrack *track,Bool_t *type); //0 = pion, 1 = kaon, 2 = proton
 Bool_t CheckStatus(AliAODTrack *track,TString detectors);

 protected:

 Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie);
 Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie);
 Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie);
 void BayesianProbabilityITS(AliAODTrack *track,Double_t *prob);
 void BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob);
 void BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob);
 void BayesianProbabilityTRD(AliAODTrack *track,Double_t *prob);

 private:
 Double_t fSigma; // sigma for the raw signal PID 
 Double_t *fPriors; // set of priors


 ClassDef(AliAODPidHF,3)

};

#endif
Commit	Line	Data
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
	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);
f9e3cf8e	34	Bool_t IsElectronRaw (AliAODTrack *track, TString detector);
7ce8802c	35	void BayesianProbability(AliAODTrack track,TString detectors,Double_t pid);
7ce8802c	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