[u/mrichter/AliRoot.git] / PWG1 / TRD / AliTRDcheckPID.h

#ifndef ALITRDCHECKPID_H
#define ALITRDCHECKPID_H

//////////////////////////////////////////////////////
//
// PID performance checker of the TRD
//
// Author : Alex Wilk <wilka@uni-muenster.de>
//          Alex Bercuci <A.Bercuci@gsi.de>
//          Markus Fasel <M.Fasel@gsi.de>
//
///////////////////////////////////////////////////////

#ifndef ROOT_TAxis
#include "TAxis.h"
#endif

#ifndef ALITRDRECOTASK_H
#include "AliTRDrecoTask.h"
#endif

#ifndef ALIPID_H
#include "AliPID.h"
#endif

class AliTRDReconstructor;
class AliTRDpidUtil;
class AliTRDcheckPID : public AliTRDrecoTask 
{
public:
  // Plots registered for this task
  enum{
    kEfficiency        =  0     // e Efficiency plot
    ,kdEdx             =  1     // dE/dx spectra
    ,kdEdxSlice        =  2     // dE/dx spectra
    ,kPH               =  3     // pulse height spectra
    ,kNClus            =  4     //  number of clusters per track
    ,kMomentum         =  5     // momentum distribution
    ,kMomentumBin      =  6     // momentum distribution
    ,kNTracklets       =  7     // Number of tracklets per track
    ,kEfficiencyMu     =  8     // mu Efficiency plot
    ,kEfficiencyPi     =  9     // pi Efficiency plot
    ,kEfficiencyKa     =  10    // K Efficiency plot
    ,kEfficiencyPr     =  11    // pr Efficiency plot
    ,kV0               =  12    // V0 performance
    ,kNPlots           =  13    // Number of plots for this tasks 
  };
  AliTRDcheckPID();
  AliTRDcheckPID(char* name);
  virtual ~AliTRDcheckPID();
  
  virtual void    UserCreateOutputObjects();
  virtual Bool_t  GetRefFigure(Int_t ifig);
  virtual void    UserExec(Option_t *opt);
  virtual Bool_t  PostProcess();

  TH1 *PlotLQ(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotNN(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotESD(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotdEdx(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotdEdxSlice(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotPH(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotNClus(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotNTracklets(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotMom(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotMomBin(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotV0(const AliTRDtrackV1 *track = 0x0);

  void SetRequireMinNTracklets(Int_t mintracklets) { fMinNTracklets = mintracklets; }
  void SetRequireMaxNTracklets(Int_t maxtracklets) { fMaxNTracklets = maxtracklets; }

  TObjArray *GetGraphs() const { return fGraph; };
  static Char_t const* MethodName(Int_t id) { return fgMethod[id]; };
  //TObjArray *GetHistos() { return fContainer; };
  virtual TObjArray *Histos();
  void EvaluateEfficiency(TObjArray* const histoContainer, TObjArray *results, Int_t species, Float_t electronEfficiency);
  inline void SetMomentumBinning(Int_t nBins, Double_t *bins);
  inline Int_t FindBin(Int_t species, Double_t momentum);
  inline Bool_t IsInRange(Double_t momentum);

private:
  AliTRDcheckPID(const AliTRDcheckPID&);               // not implemented
  AliTRDcheckPID& operator=(const AliTRDcheckPID&);    // not implemented

  Int_t  CalcPDG(AliTRDtrackV1* track = 0x0);
  Bool_t CheckTrackQuality(const AliTRDtrackV1* track = 0x0) const;
  void   LocalInit();

  static Char_t const *fgMethod[3];        // PID method name
  AliTRDReconstructor *fReconstructor;     // reconstructor needed for recalculation the PID
  AliTRDpidUtil       *fUtil;              // utility class for PID calculations
  TObjArray           *fGraph;             //! array of graphs filled in PostProcess
  TObjArray           *fPID;               //! array of PID info/track for calibration
  TObjArray           *fV0s;               //! array of V0 info
  TObjArray           *fEfficiency[AliPID::kSPECIES];      //! array of histograms with efficiency
  TAxis               *fMomentumAxis;      // helper mementum binning
  Int_t                fMinNTracklets;     // minimum number of required Tracklets (for systematic studies)
  Int_t                fMaxNTracklets;     // maximum number of required Tracklets (for systematic studies) 
  ClassDef(AliTRDcheckPID, 2); // TRD PID checker
};

//________________________________________________________________________
inline void AliTRDcheckPID::SetMomentumBinning(Int_t nBins, Double_t *bins){
  //
  // Set the Momentum Bins
  //
  if(fMomentumAxis) delete fMomentumAxis;
  fMomentumAxis = new TAxis(nBins, bins);
}

//________________________________________________________________________
inline Int_t AliTRDcheckPID::FindBin(Int_t species, Double_t momentum){
  //
  // Find the Bin in the 2D Histogram
  //
  return species * fMomentumAxis->GetNbins() + (fMomentumAxis->FindBin(momentum) -1);
}

//________________________________________________________________________
inline Bool_t AliTRDcheckPID::IsInRange(Double_t momentum){
  //
  // Check Whether momentum is in the defined Range
  //
  return (momentum >= fMomentumAxis->GetXmin() && momentum <= fMomentumAxis->GetXmax());
}

#endif
Commit	Line	Data
1ee39b3a	1	#ifndef ALITRDCHECKPID_H
	2	#define ALITRDCHECKPID_H
	3
	4	//////////////////////////////////////////////////////
	5	//
	6	// PID performance checker of the TRD
	7	//
	8	// Author : Alex Wilk <wilka@uni-muenster.de>
	9	// Alex Bercuci <A.Bercuci@gsi.de>
	10	// Markus Fasel <M.Fasel@gsi.de>
	11	//
	12	///////////////////////////////////////////////////////
	13
	14	#ifndef ROOT_TAxis
	15	#include "TAxis.h"
	16	#endif
	17
	18	#ifndef ALITRDRECOTASK_H
	19	#include "AliTRDrecoTask.h"
	20	#endif
	21
2e6aa247	22	#ifndef ALIPID_H
	23	#include "AliPID.h"
	24	#endif
	25
1ee39b3a	26	class AliTRDReconstructor;
	27	class AliTRDpidUtil;
	28	class AliTRDcheckPID : public AliTRDrecoTask
	29	{
	30	public:
	31	// Plots registered for this task
	32	enum{
2e6aa247	33	kEfficiency = 0 // e Efficiency plot
	34	,kdEdx = 1 // dE/dx spectra
	35	,kdEdxSlice = 2 // dE/dx spectra
	36	,kPH = 3 // pulse height spectra
	37	,kNClus = 4 // number of clusters per track
	38	,kMomentum = 5 // momentum distribution
	39	,kMomentumBin = 6 // momentum distribution
	40	,kNTracklets = 7 // Number of tracklets per track
	41	,kEfficiencyMu = 8 // mu Efficiency plot
	42	,kEfficiencyPi = 9 // pi Efficiency plot
	43	,kEfficiencyKa = 10 // K Efficiency plot
	44	,kEfficiencyPr = 11 // pr Efficiency plot
b9ddd472	45	,kV0 = 12 // V0 performance
b9ddd472	46	,kNPlots = 13 // Number of plots for this tasks
1ee39b3a	47	};
1ee39b3a	48	AliTRDcheckPID();
f8f46e4d	49	AliTRDcheckPID(char* name);
1ee39b3a	50	virtual ~AliTRDcheckPID();
1ee39b3a	51
f8f46e4d	52	virtual void UserCreateOutputObjects();
1ee39b3a	53	virtual Bool_t GetRefFigure(Int_t ifig);
f8f46e4d	54	virtual void UserExec(Option_t *opt);
1ee39b3a	55	virtual Bool_t PostProcess();
	56
	57	TH1 PlotLQ(const AliTRDtrackV1 track = 0x0);
	58	TH1 PlotNN(const AliTRDtrackV1 track = 0x0);
	59	TH1 PlotESD(const AliTRDtrackV1 track = 0x0);
	60	TH1 PlotdEdx(const AliTRDtrackV1 track = 0x0);
	61	TH1 PlotdEdxSlice(const AliTRDtrackV1 track = 0x0);
	62	TH1 PlotPH(const AliTRDtrackV1 track = 0x0);
	63	TH1 PlotNClus(const AliTRDtrackV1 track = 0x0);
	64	TH1 PlotNTracklets(const AliTRDtrackV1 track = 0x0);
	65	TH1 PlotMom(const AliTRDtrackV1 track = 0x0);
	66	TH1 PlotMomBin(const AliTRDtrackV1 track = 0x0);
b9ddd472	67	TH1 PlotV0(const AliTRDtrackV1 track = 0x0);
1ee39b3a	68
	69	void SetRequireMinNTracklets(Int_t mintracklets) { fMinNTracklets = mintracklets; }
	70	void SetRequireMaxNTracklets(Int_t maxtracklets) { fMaxNTracklets = maxtracklets; }
	71
	72	TObjArray *GetGraphs() const { return fGraph; };
2e6aa247	73	static Char_t const* MethodName(Int_t id) { return fgMethod[id]; };
1ee39b3a	74	//TObjArray *GetHistos() { return fContainer; };
1ee39b3a	75	virtual TObjArray *Histos();
2e6aa247	76	void EvaluateEfficiency(TObjArray* const histoContainer, TObjArray *results, Int_t species, Float_t electronEfficiency);
1ee39b3a	77	inline void SetMomentumBinning(Int_t nBins, Double_t *bins);
	78	inline Int_t FindBin(Int_t species, Double_t momentum);
	79	inline Bool_t IsInRange(Double_t momentum);
	80
	81	private:
	82	AliTRDcheckPID(const AliTRDcheckPID&); // not implemented
	83	AliTRDcheckPID& operator=(const AliTRDcheckPID&); // not implemented
	84
	85	Int_t CalcPDG(AliTRDtrackV1* track = 0x0);
	86	Bool_t CheckTrackQuality(const AliTRDtrackV1* track = 0x0) const;
2c5222ab	87	void LocalInit();
2c5222ab	88
2e6aa247	89	static Char_t const *fgMethod[3]; // PID method name
5935a6da	90	AliTRDReconstructor *fReconstructor; // reconstructor needed for recalculation the PID
5935a6da	91	AliTRDpidUtil *fUtil; // utility class for PID calculations
1ee39b3a	92	TObjArray *fGraph; //! array of graphs filled in PostProcess
b91fdd71	93	TObjArray *fPID; //! array of PID info/track for calibration
3d19c1b0	94	TObjArray *fV0s; //! array of V0 info
2e6aa247	95	TObjArray *fEfficiency[AliPID::kSPECIES]; //! array of histograms with efficiency
5723c09a	96	TAxis *fMomentumAxis; // helper mementum binning
1ee39b3a	97	Int_t fMinNTracklets; // minimum number of required Tracklets (for systematic studies)
1ee39b3a	98	Int_t fMaxNTracklets; // maximum number of required Tracklets (for systematic studies)
2e6aa247	99	ClassDef(AliTRDcheckPID, 2); // TRD PID checker
1ee39b3a	100	};
	101
	102	//________________________________________________________________________
	103	inline void AliTRDcheckPID::SetMomentumBinning(Int_t nBins, Double_t *bins){
	104	//
	105	// Set the Momentum Bins
	106	//
	107	if(fMomentumAxis) delete fMomentumAxis;
	108	fMomentumAxis = new TAxis(nBins, bins);
	109	}
	110
	111	//________________________________________________________________________
	112	inline Int_t AliTRDcheckPID::FindBin(Int_t species, Double_t momentum){
	113	//
	114	// Find the Bin in the 2D Histogram
	115	//
	116	return species * fMomentumAxis->GetNbins() + (fMomentumAxis->FindBin(momentum) -1);
	117	}
	118
	119	//________________________________________________________________________
	120	inline Bool_t AliTRDcheckPID::IsInRange(Double_t momentum){
	121	//
	122	// Check Whether momentum is in the defined Range
	123	//
	124	return (momentum >= fMomentumAxis->GetXmin() && momentum <= fMomentumAxis->GetXmax());
	125	}
	126
	127	#endif