[u/mrichter/AliRoot.git] / TRD / AliTRDpidLQ.h

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

/* $Id$ */


///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//   The TRD particle identification class                                   //
//                                                                           //
//   Its main purposes are:                                                  //
//      - Creation and bookkeeping of the propability distributions          //
//      - Assignment of a e/pi propability to a given track based on         //
//        the LQ method                                                      //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

#include "AliTRDpid.h"

class AliTRDpidLQ : public AliTRDpid {

 public:

  AliTRDpidLQ();
  AliTRDpidLQ(const char* name, const char* title);
  AliTRDpidLQ(const AliTRDpidLQ &p);
  virtual ~AliTRDpidLQ();
  AliTRDpidLQ &operator=(const AliTRDpidLQ &p);

  virtual void          Copy(TObject &p) const;
  virtual Bool_t        Init();
  virtual Bool_t        AssignLikelihood()            { return 0; };
  virtual Bool_t        AssignLikelihood(TObjArray *) { return 0; };
  virtual Bool_t        AssignLikelihood(AliTRDtrack *t);
  virtual Bool_t        CreateHistograms(Int_t nmom, Float_t minmom, Float_t maxmom);
  virtual Bool_t        FillSpectra()                 { return 0; };
  virtual Bool_t        FillSpectra(TObjArray*)       { return 0; };
  virtual Bool_t        FillSpectra(const AliTRDtrack *t);

          Int_t         GetIndex(const AliTRDtrack *t);
          Int_t         GetIndex(Int_t imom, Int_t ipid);
          Int_t         GetIndex(Float_t mom, Int_t ipid);

          TObjArray*    GetHist() const                     { return fHist;        };

          Float_t       GetChargeMin() const                { return fChargeMin;   };
          Int_t         GetNClusterMin() const              { return fNClusterMin; };

          Int_t         GetNLh() const                      { return fNLh;         };
          Float_t       GetMinLh() const                    { return fMinLh;       };
          Float_t       GetMaxLh() const                    { return fMaxLh;       };

          void          SetChargeMin(Float_t min)     { fChargeMin   = min;  };
          void          SetNClusterMin(Int_t min)     { fNClusterMin = min;  };

          void          SetNLh(Int_t n)               { fNLh         = n;    };
          void          SetMinLh(Float_t min)         { fMinLh       = min;  };
          void          SetMaxLh(Float_t max)         { fMaxLh       = max;  };

 protected:

  Int_t           fNMom;             //  Number of momentum bins
  Float_t         fMinMom;           //  Lower momentum
  Float_t         fMaxMom;           //  Upper momentum
  Float_t         fWidMom;           //  Width of the momentum bins
  TObjArray      *fHist;             //  Array of histograms

  Int_t           fNLh;              //  Number of bins of the likelihood spectra
  Float_t         fMinLh;            //  Lower range of the likelihood spectra
  Float_t         fMaxLh;            //  Upper range of the likelihood spectra

  Float_t         fChargeMin;        //  Minimum charge required in one plane
  Int_t           fNClusterMin;      //  Minimum number of clusters required in one plane

  ClassDef(AliTRDpidLQ,1)            //  Assigns e/pi propability to the tracks based on LQ method 

};
#endif
Commit	Line	Data
16bf9884	1	#ifndef ALITRDPIDLQ_H
	2	#define ALITRDPIDLQ_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
0a29d0f1	8
	9	///////////////////////////////////////////////////////////////////////////////
	10	// //
	11	// The TRD particle identification class //
	12	// //
	13	// Its main purposes are: //
	14	// - Creation and bookkeeping of the propability distributions //
	15	// - Assignment of a e/pi propability to a given track based on //
	16	// the LQ method //
	17	// //
	18	///////////////////////////////////////////////////////////////////////////////
	19
16bf9884	20	#include "AliTRDpid.h"
	21
	22	class AliTRDpidLQ : public AliTRDpid {
	23
	24	public:
	25
	26	AliTRDpidLQ();
	27	AliTRDpidLQ(const char* name, const char* title);
	28	AliTRDpidLQ(const AliTRDpidLQ &p);
	29	virtual ~AliTRDpidLQ();
	30	AliTRDpidLQ &operator=(const AliTRDpidLQ &p);
	31
e0d47c25	32	virtual void Copy(TObject &p) const;
16bf9884	33	virtual Bool_t Init();
e0d47c25	34	virtual Bool_t AssignLikelihood() { return 0; };
e0d47c25	35	virtual Bool_t AssignLikelihood(TObjArray *) { return 0; };
16bf9884	36	virtual Bool_t AssignLikelihood(AliTRDtrack *t);
afc51ac2	37	virtual Bool_t CreateHistograms(Int_t nmom, Float_t minmom, Float_t maxmom);
e0d47c25	38	virtual Bool_t FillSpectra() { return 0; };
e0d47c25	39	virtual Bool_t FillSpectra(TObjArray*) { return 0; };
16bf9884	40	virtual Bool_t FillSpectra(const AliTRDtrack *t);
16bf9884	41
e0d47c25	42	Int_t GetIndex(const AliTRDtrack *t);
	43	Int_t GetIndex(Int_t imom, Int_t ipid);
	44	Int_t GetIndex(Float_t mom, Int_t ipid);
16bf9884	45
	46	TObjArray* GetHist() const { return fHist; };
	47
	48	Float_t GetChargeMin() const { return fChargeMin; };
	49	Int_t GetNClusterMin() const { return fNClusterMin; };
	50
	51	Int_t GetNLh() const { return fNLh; };
	52	Float_t GetMinLh() const { return fMinLh; };
	53	Float_t GetMaxLh() const { return fMaxLh; };
	54
afc51ac2	55	void SetChargeMin(Float_t min) { fChargeMin = min; };
afc51ac2	56	void SetNClusterMin(Int_t min) { fNClusterMin = min; };
16bf9884	57
afc51ac2	58	void SetNLh(Int_t n) { fNLh = n; };
	59	void SetMinLh(Float_t min) { fMinLh = min; };
	60	void SetMaxLh(Float_t max) { fMaxLh = max; };
16bf9884	61
	62	protected:
	63
	64	Int_t fNMom; // Number of momentum bins
	65	Float_t fMinMom; // Lower momentum
	66	Float_t fMaxMom; // Upper momentum
	67	Float_t fWidMom; // Width of the momentum bins
	68	TObjArray *fHist; // Array of histograms
	69
	70	Int_t fNLh; // Number of bins of the likelihood spectra
	71	Float_t fMinLh; // Lower range of the likelihood spectra
	72	Float_t fMaxLh; // Upper range of the likelihood spectra
	73
	74	Float_t fChargeMin; // Minimum charge required in one plane
	75	Int_t fNClusterMin; // Minimum number of clusters required in one plane
	76
	77	ClassDef(AliTRDpidLQ,1) // Assigns e/pi propability to the tracks based on LQ method
	78
	79	};
	80	#endif