[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnCutPID.h

//
// Class AliRsnCutRange
//
// General implementation of cuts which check a value inside a range.
// This range can be defined by two integers or two doubles.
// A user-friendly enumeration allows to define what is checked.
//
// authors: Martin Vala (martin.vala@cern.ch)
//          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
//

#ifndef ALIRSNCUTPID_H
#define ALIRSNCUTPID_H

#include "AliPID.h"
#include "AliRsnCut.h"

class AliRsnDaughter;

class AliRsnCutPID : public AliRsnCut {
public:

   enum EDetector {
      kITS,
      kTPC,
      kTRD,
      kTOF,
      kHMPID,
      kDetectors
   };

   AliRsnCutPID();
   AliRsnCutPID(const char *name, AliPID::EParticleType pid, Double_t probMin = 0.0, Bool_t perfectPID = kFALSE);
   virtual ~AliRsnCutPID() {;};

   void           SetDefault(Bool_t yn = kTRUE) {fUseDefault = yn;}
   void           SetPrior(Int_t i, Double_t value) {if (i >= 0 && i < AliPID::kSPECIES) fPrior[i] = value;}
   void           SetPerfect(Bool_t yn = kTRUE) {fPerfect = yn;}

   void           IncludeDetector(EDetector det, Double_t threshold = 0., Bool_t goAbove = kTRUE);
   void           ExcludeDetector(EDetector det) {if (CheckBounds(det)) fUseDetector[det] = kFALSE;}

   Bool_t         ComputeWeights(AliRsnDaughter *daughter);
   Int_t          RealisticPID(AliRsnDaughter * const daughter, Double_t &prob);
   Int_t          PerfectPID(AliRsnDaughter * const daughter);
   Double_t       GetWeight(Int_t i) {if (i >= 0 && i < AliPID::kSPECIES) return fWeight[i]; return 0.0;}

   virtual Bool_t IsSelected(TObject *object);

protected:

   Bool_t   CheckBounds(EDetector det) const {return (det >= kITS && det < kDetectors);}
   Bool_t   CheckThreshold(EDetector det, Double_t value);

   Double_t              fPrior[AliPID::kSPECIES];        // prior probability
   Double_t              fWeight[AliPID::kSPECIES];       // PID weights used for combinations

   Bool_t                fPerfect;                        // choice to use perfect PID
   Bool_t                fUseDefault;                     // choice to use default combined PID weights (or customized)
   Bool_t                fUseDetector[kDetectors];        // flag to include/exclude each single detector
   Double_t              fPtThreshold[kDetectors];        // pT threshold above/below which a detector is considered
   Double_t              fGoAboveThreshold[kDetectors];   // to choose if detector is used balow or above threshold

   ClassDef(AliRsnCutPID, 1)
};

#endif
Commit	Line	Data
2dab9030	1	//
	2	// Class AliRsnCutRange
	3	//
	4	// General implementation of cuts which check a value inside a range.
	5	// This range can be defined by two integers or two doubles.
	6	// A user-friendly enumeration allows to define what is checked.
	7	//
	8	// authors: Martin Vala (martin.vala@cern.ch)
	9	// Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
	10	//
	11
	12	#ifndef ALIRSNCUTPID_H
	13	#define ALIRSNCUTPID_H
	14
	15	#include "AliPID.h"
	16	#include "AliRsnCut.h"
	17
	18	class AliRsnDaughter;
	19
2a1c7696	20	class AliRsnCutPID : public AliRsnCut {
	21	public:
	22
	23	enum EDetector {
2dab9030	24	kITS,
	25	kTPC,
	26	kTRD,
	27	kTOF,
	28	kHMPID,
	29	kDetectors
2a1c7696	30	};
	31
	32	AliRsnCutPID();
	33	AliRsnCutPID(const char *name, AliPID::EParticleType pid, Double_t probMin = 0.0, Bool_t perfectPID = kFALSE);
	34	virtual ~AliRsnCutPID() {;};
	35
	36	void SetDefault(Bool_t yn = kTRUE) {fUseDefault = yn;}
	37	void SetPrior(Int_t i, Double_t value) {if (i >= 0 && i < AliPID::kSPECIES) fPrior[i] = value;}
	38	void SetPerfect(Bool_t yn = kTRUE) {fPerfect = yn;}
	39
	40	void IncludeDetector(EDetector det, Double_t threshold = 0., Bool_t goAbove = kTRUE);
	41	void ExcludeDetector(EDetector det) {if (CheckBounds(det)) fUseDetector[det] = kFALSE;}
	42
	43	Bool_t ComputeWeights(AliRsnDaughter *daughter);
	44	Int_t RealisticPID(AliRsnDaughter * const daughter, Double_t &prob);
	45	Int_t PerfectPID(AliRsnDaughter * const daughter);
	46	Double_t GetWeight(Int_t i) {if (i >= 0 && i < AliPID::kSPECIES) return fWeight[i]; return 0.0;}
	47
	48	virtual Bool_t IsSelected(TObject *object);
	49
	50	protected:
	51
	52	Bool_t CheckBounds(EDetector det) const {return (det >= kITS && det < kDetectors);}
	53	Bool_t CheckThreshold(EDetector det, Double_t value);
	54
	55	Double_t fPrior[AliPID::kSPECIES]; // prior probability
	56	Double_t fWeight[AliPID::kSPECIES]; // PID weights used for combinations
	57
	58	Bool_t fPerfect; // choice to use perfect PID
	59	Bool_t fUseDefault; // choice to use default combined PID weights (or customized)
	60	Bool_t fUseDetector[kDetectors]; // flag to include/exclude each single detector
	61	Double_t fPtThreshold[kDetectors]; // pT threshold above/below which a detector is considered
	62	Double_t fGoAboveThreshold[kDetectors]; // to choose if detector is used balow or above threshold
	63
	64	ClassDef(AliRsnCutPID, 1)
2dab9030	65	};
	66
	67	#endif