[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / Nuclei / B2 / AliLnID.h

#ifndef ALILNID_H
#define ALILNID_H

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

// class for light nuclei identification
// author: Eulogio Serradilla <eulogio.serradilla@cern.ch>

#include <TObject.h>
#include <AliPID.h>

class TParticle;
class AliITSPIDResponse;
class AliTPCPIDResponse;

class AliLnID: public TObject
{
  public:
	AliLnID();
	AliLnID(const AliLnID& other);
	AliLnID& operator=(const AliLnID& other);
	
	virtual ~AliLnID();
	
	Int_t GetPID(const TParticle* p) const;
	
	Int_t GetPID(Int_t partCode, Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t pTPC, Double_t dEdxTPC, Int_t nPointsTPC, Double_t pTOF, Double_t beta, Double_t nSigITS=3, Double_t nSigTPC=3, Double_t nSigTOF=3) const;
	
	Int_t GetTPCpid(Int_t partCode, Double_t pTPC, Double_t dEdx, Double_t nPoints, Double_t nSigma=3) const;
	
	Int_t GetITSTPCpid(Int_t partCode, Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t nSigmaITS, Double_t pTPC, Double_t dEdxTPC, Double_t nPointsTPC, Double_t nSigmaTPC) const;
	
	Int_t GetTPCTOFpid(Int_t partCode, Double_t pTPC, Double_t dEdx, Double_t nPoints, Double_t nSigmaTPC, Double_t pTOF, Double_t beta, Double_t nSigmaTOF) const;
	
	Int_t GetMaxLikelihoodPID(Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t pTPC, Double_t dEdxTPC, Int_t nPointsTPC, Double_t pTOF, Double_t beta) const;
	
	Int_t GetBayesPID(Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t pTPC, Double_t dEdxTPC, Int_t nPointsTPC, Double_t pTOF, Double_t beta) const;
	
	Bool_t GetITSlikelihood(Double_t p, Double_t dEdx, Int_t nPoints, Double_t* r) const;
	Bool_t GetTPClikelihood(Double_t p, Double_t dEdx, Int_t nPoints, Double_t* r) const;
	Bool_t GetTOFlikelihood(Double_t p, Double_t beta, Double_t* r) const;
	
	Double_t GetBetaExpectedSigma(Double_t p, Double_t mass) const;
	
	Bool_t GetITSmatch(Int_t pid, Double_t p, Double_t dEdx, Int_t nPoints, Double_t nSigma=3.) const;
	Bool_t GetTPCmatch(Int_t pid, Double_t pTPC, Double_t dEdx, Double_t nPoints, Double_t nSigma=3.) const;
	Bool_t GetTOFmatch(Int_t pid, Double_t pTOF, Double_t beta, Double_t nSigma=3.) const;
	
	Int_t GetPidProcedure() const { return fPidProcedure; }
	
	Bool_t IsITSTPCmismatch(Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t pTPC, Double_t dEdxTPC, Int_t nPointsTPC, Double_t nSigma=5.) const;
	Bool_t IsITSTOFmismatch(Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t pTOF, Double_t beta, Double_t nSigma=5.) const;
	Bool_t IsTPCTOFmismatch(Double_t pTPC, Double_t dEdxTPC, Int_t nPointsTPC, Double_t pTOF, Double_t beta, Double_t nSigma=5.) const;
	
	void SetPriorProbabilities(const Double_t* prob);
	void SetPriorProbabilities(Double_t e, Double_t mu, Double_t pi, Double_t k, Double_t p, Double_t d, Double_t t, Double_t he3, Double_t alpha);
	
	void SetPidProcedure(Int_t proc) { fPidProcedure = proc; }
	
	void SetITSBetheBlochParams(const Double_t* param, Double_t res=0.13);
	
	void SetTPCBetheBlochParams(const Double_t* param, Double_t mip=1., Double_t res=0.06);
	void SetTPCBetheBlochParams(Double_t c0, Double_t c1, Double_t c2, Double_t c3, Double_t c4, Double_t mip=1., Double_t res=0.06);
	
	void SetTPCChargeCorrection(Double_t zexp) { fZexp = zexp; }
	
	enum { kSPECIES = 9 };
	enum { kBayes=0, kMaxLikelihood, kTPC, kITSTPC, kTPCTOF };
	
  private:
 
	Double_t Beta(Double_t p, Double_t m) const;
	Bool_t GetLikelihood(Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t pTPC, Double_t dEdxTPC, Int_t nPointsTPC, Double_t pTOF, Double_t beta, Double_t* r) const;
	Int_t GetIndexOfMaxValue(const Double_t* w) const;
	
  private:
 
	Int_t fPidProcedure; // PID procedure code
	AliPID::EParticleType fSpecies[kSPECIES]; // particle species known by the pid
	
	Double_t fPrior[kSPECIES]; // prior probabilities
	Double_t fRange; // number of sigmas to the expected values
	Double_t fZexp; // TPC BB charge dependence
	
	AliITSPIDResponse* fITSpid; // ITS likelihood
	AliTPCPIDResponse* fTPCpid; // TPC likelihood
	
	ClassDef(AliLnID, 1)
};

#endif // ALILNID_H
Commit	Line	Data
6b91e8c0	1	#ifndef ALILNID_H
	2	#define ALILNID_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	// class for light nuclei identification
	8	// author: Eulogio Serradilla <eulogio.serradilla@cern.ch>
	9
	10	#include <TObject.h>
	11	#include <AliPID.h>
	12
f4d6dd11	13	class TParticle;
052278a6	14	class AliITSPIDResponse;
052278a6	15	class AliTPCPIDResponse;
f4d6dd11	16
6b91e8c0	17	class AliLnID: public TObject
	18	{
	19	public:
	20	AliLnID();
	21	AliLnID(const AliLnID& other);
	22	AliLnID& operator=(const AliLnID& other);
	23
	24	virtual ~AliLnID();
	25
f4d6dd11	26	Int_t GetPID(const TParticle* p) const;
6b91e8c0	27
	28	Int_t GetPID(Int_t partCode, Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t pTPC, Double_t dEdxTPC, Int_t nPointsTPC, Double_t pTOF, Double_t beta, Double_t nSigITS=3, Double_t nSigTPC=3, Double_t nSigTOF=3) const;
	29
	30	Int_t GetTPCpid(Int_t partCode, Double_t pTPC, Double_t dEdx, Double_t nPoints, Double_t nSigma=3) const;
	31
	32	Int_t GetITSTPCpid(Int_t partCode, Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t nSigmaITS, Double_t pTPC, Double_t dEdxTPC, Double_t nPointsTPC, Double_t nSigmaTPC) const;
	33
	34	Int_t GetTPCTOFpid(Int_t partCode, Double_t pTPC, Double_t dEdx, Double_t nPoints, Double_t nSigmaTPC, Double_t pTOF, Double_t beta, Double_t nSigmaTOF) const;
	35
	36	Int_t GetMaxLikelihoodPID(Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t pTPC, Double_t dEdxTPC, Int_t nPointsTPC, Double_t pTOF, Double_t beta) const;
	37
	38	Int_t GetBayesPID(Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t pTPC, Double_t dEdxTPC, Int_t nPointsTPC, Double_t pTOF, Double_t beta) const;
	39
	40	Bool_t GetITSlikelihood(Double_t p, Double_t dEdx, Int_t nPoints, Double_t* r) const;
	41	Bool_t GetTPClikelihood(Double_t p, Double_t dEdx, Int_t nPoints, Double_t* r) const;
	42	Bool_t GetTOFlikelihood(Double_t p, Double_t beta, Double_t* r) const;
	43
	44	Double_t GetBetaExpectedSigma(Double_t p, Double_t mass) const;
6b91e8c0	45
	46	Bool_t GetITSmatch(Int_t pid, Double_t p, Double_t dEdx, Int_t nPoints, Double_t nSigma=3.) const;
	47	Bool_t GetTPCmatch(Int_t pid, Double_t pTPC, Double_t dEdx, Double_t nPoints, Double_t nSigma=3.) const;
	48	Bool_t GetTOFmatch(Int_t pid, Double_t pTOF, Double_t beta, Double_t nSigma=3.) const;
6b91e8c0	49
	50	Int_t GetPidProcedure() const { return fPidProcedure; }
	51
	52	Bool_t IsITSTPCmismatch(Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t pTPC, Double_t dEdxTPC, Int_t nPointsTPC, Double_t nSigma=5.) const;
	53	Bool_t IsITSTOFmismatch(Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t pTOF, Double_t beta, Double_t nSigma=5.) const;
	54	Bool_t IsTPCTOFmismatch(Double_t pTPC, Double_t dEdxTPC, Int_t nPointsTPC, Double_t pTOF, Double_t beta, Double_t nSigma=5.) const;
	55
	56	void SetPriorProbabilities(const Double_t* prob);
	57	void SetPriorProbabilities(Double_t e, Double_t mu, Double_t pi, Double_t k, Double_t p, Double_t d, Double_t t, Double_t he3, Double_t alpha);
	58
	59	void SetPidProcedure(Int_t proc) { fPidProcedure = proc; }
	60
	61	void SetITSBetheBlochParams(const Double_t* param, Double_t res=0.13);
	62
	63	void SetTPCBetheBlochParams(const Double_t* param, Double_t mip=1., Double_t res=0.06);
	64	void SetTPCBetheBlochParams(Double_t c0, Double_t c1, Double_t c2, Double_t c3, Double_t c4, Double_t mip=1., Double_t res=0.06);
	65
78a09943	66	void SetTPCChargeCorrection(Double_t zexp) { fZexp = zexp; }
78a09943	67
6b91e8c0	68	enum { kSPECIES = 9 };
052278a6	69	enum { kBayes=0, kMaxLikelihood, kTPC, kITSTPC, kTPCTOF };
6b91e8c0	70
	71	private:
	72
	73	Double_t Beta(Double_t p, Double_t m) const;
	74	Bool_t GetLikelihood(Double_t pITS, Double_t dEdxITS, Int_t nPointsITS, Double_t pTPC, Double_t dEdxTPC, Int_t nPointsTPC, Double_t pTOF, Double_t beta, Double_t* r) const;
	75	Int_t GetIndexOfMaxValue(const Double_t* w) const;
	76
	77	private:
	78
	79	Int_t fPidProcedure; // PID procedure code
6b91e8c0	80	AliPID::EParticleType fSpecies[kSPECIES]; // particle species known by the pid
	81
	82	Double_t fPrior[kSPECIES]; // prior probabilities
	83	Double_t fRange; // number of sigmas to the expected values
78a09943	84	Double_t fZexp; // TPC BB charge dependence
6b91e8c0	85
052278a6	86	AliITSPIDResponse* fITSpid; // ITS likelihood
052278a6	87	AliTPCPIDResponse* fTPCpid; // TPC likelihood
6b91e8c0	88
	89	ClassDef(AliLnID, 1)
	90	};
	91
	92	#endif // ALILNID_H