[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliAnalysisTaskSingleParticle.h

#ifndef ALIANALYSISTASKSINGLEPARTICLE_H
#define ALIANALYSISTASKSINGLEPARTICLE_H

//===============================================================
//
// Analysis task for constructing MC or data driven single particle efficiencies
//
// Ionut C. Arsene, EMMI-GSI, 2011/12/07 
//
//===============================================================

#include "TList.h"

#include "AliAnalysisTaskSE.h"

class TH1D;
class TList;
class AliAnalysisCuts;
class AliCFContainer;
class AliVEvent;
class AliDielectronHistos;
class AliESDv0Cuts;
class AliESDv0KineCuts;

class AliAnalysisTaskSingleParticle : public AliAnalysisTaskSE {
  
public:
  enum { kNMaxDimensions = 20 };
    
  AliAnalysisTaskSingleParticle();
  AliAnalysisTaskSingleParticle(const char *name);
  virtual ~AliAnalysisTaskSingleParticle();

  virtual void UserExec(Option_t *option);
  virtual void UserCreateOutputObjects();
  virtual void FinishTaskOutput();
    
  void SetFillTRDfriendPH(Bool_t fill=kTRUE) {fFillTRDfriendPH = fill;}
  void UsePhysicsSelection(Bool_t phy=kTRUE) {fSelectPhysics=phy;}
  void SetTriggerMask(UInt_t mask) {fTriggerMask=mask;}
  UInt_t GetTriggerMask() const { return fTriggerMask; }
  void SetRejectPileup(Bool_t pileup=kTRUE)     { fRejectPileup=pileup;     }
  
  void SetEventFilter(AliAnalysisCuts * const filter) {fEventFilter=filter;}
  void SetTrackFilter(AliAnalysisCuts * const filter) {fTrackFilter=filter;}
  void SetPairFilter(AliAnalysisCuts* const filter) {fPairFilter=filter;}
  void SetV0Cuts(AliESDv0Cuts* const cuts) {fV0Cuts = cuts;}
  void SetLambdaFilter(AliAnalysisCuts* const filter) {fLambdaFilter = filter;}
  void SetK0sFilter(AliAnalysisCuts* const filter) {fK0sFilter = filter;}
  void SetHistogramManager(AliDielectronHistos * const histos) { fHistos=histos; }
  void SetV0KineCuts(AliESDv0KineCuts* const v0cuts) {fV0KineCuts = v0cuts;}

  void AddCFVar(Int_t var, Int_t nbins, Double_t lowLim, Double_t highLim);
  void AddCFVar(Int_t var, const Char_t* bins);

protected:
//  enum {kAllEvents=0, kPhysicsSelectionEvents, kFilteredEvents, kEventStatBins};
//  enum {kEventVtxZ=0, kNTracklets, kCentrality, kPt, kPin, kPhi, kEta, kDeltaPt, kDeltaPhi, kDeltaEta, kTPCnCls, kTPCnSigEle, kTPCnSigPio, kTPCnSigPro, kNVariables}; 

  AliCFContainer*  fCfContainer;      //  CF container
  AliDielectronHistos *fHistos;       // Histogram manager
  TList            fHistogramList;    // histogram list from the manager

  Bool_t           fSelectPhysics;    // Whether to use physics selection
  UInt_t           fTriggerMask;      // Event trigger mask
  Bool_t           fRejectPileup;     // pileup rejection wanted
  Bool_t           fFillTRDfriendPH;  // use the task to fill a TRD tracklet PH container

  AliAnalysisCuts* fEventFilter;      // event filter
  AliAnalysisCuts* fTrackFilter;      // tracking filter
  AliAnalysisCuts* fPairFilter;       // pair filter
  AliESDv0Cuts*    fV0Cuts;           // v0 standard filter
  AliAnalysisCuts* fLambdaFilter;     // additional cuts for lambda v0 exclusion
  AliAnalysisCuts* fK0sFilter;        // additional cuts for K0s v0 exclusion
  AliESDv0KineCuts* fV0KineCuts;      // V0 kine cuts
  
  Int_t            fCFNVars;          // number of CF variables
  Int_t            fCFVarsEnabled[kNMaxDimensions];    // id for every dimension
  Int_t            fCFVarsNbins[kNMaxDimensions];      // number of bins for every CF dimension
  Double_t         fCFVarRanges[kNMaxDimensions][2];   // range for CF dimensions
  TString          fCFVarBins[kNMaxDimensions];        // bin limits for CF dimensions

  TH1D *fEventStat;                  //! Histogram with event statistics
  
  void FillContainer(Int_t step, const Double_t* values, Int_t trackId, Int_t pdg);
  
  AliAnalysisTaskSingleParticle(const AliAnalysisTaskSingleParticle &c);
  AliAnalysisTaskSingleParticle& operator= (const AliAnalysisTaskSingleParticle &c);
  
  ClassDef(AliAnalysisTaskSingleParticle, 2); //Analysis Task handling single particle cuts
};
#endif
Commit	Line	Data
5720c765	1	#ifndef ALIANALYSISTASKSINGLEPARTICLE_H
	2	#define ALIANALYSISTASKSINGLEPARTICLE_H
	3
	4	//===============================================================
	5	//
	6	// Analysis task for constructing MC or data driven single particle efficiencies
	7	//
	8	// Ionut C. Arsene, EMMI-GSI, 2011/12/07
	9	//
	10	//===============================================================
	11
	12	#include "TList.h"
	13
	14	#include "AliAnalysisTaskSE.h"
	15
	16	class TH1D;
	17	class TList;
	18	class AliAnalysisCuts;
	19	class AliCFContainer;
	20	class AliVEvent;
	21	class AliDielectronHistos;
	22	class AliESDv0Cuts;
	23	class AliESDv0KineCuts;
	24
	25	class AliAnalysisTaskSingleParticle : public AliAnalysisTaskSE {
	26
	27	public:
	28	enum { kNMaxDimensions = 20 };
	29
	30	AliAnalysisTaskSingleParticle();
	31	AliAnalysisTaskSingleParticle(const char *name);
	32	virtual ~AliAnalysisTaskSingleParticle();
	33
	34	virtual void UserExec(Option_t *option);
	35	virtual void UserCreateOutputObjects();
	36	virtual void FinishTaskOutput();
	37
	38	void SetFillTRDfriendPH(Bool_t fill=kTRUE) {fFillTRDfriendPH = fill;}
	39	void UsePhysicsSelection(Bool_t phy=kTRUE) {fSelectPhysics=phy;}
	40	void SetTriggerMask(UInt_t mask) {fTriggerMask=mask;}
	41	UInt_t GetTriggerMask() const { return fTriggerMask; }
	42	void SetRejectPileup(Bool_t pileup=kTRUE) { fRejectPileup=pileup; }
	43
	44	void SetEventFilter(AliAnalysisCuts * const filter) {fEventFilter=filter;}
	45	void SetTrackFilter(AliAnalysisCuts * const filter) {fTrackFilter=filter;}
	46	void SetPairFilter(AliAnalysisCuts* const filter) {fPairFilter=filter;}
	47	void SetV0Cuts(AliESDv0Cuts* const cuts) {fV0Cuts = cuts;}
	48	void SetLambdaFilter(AliAnalysisCuts* const filter) {fLambdaFilter = filter;}
	49	void SetK0sFilter(AliAnalysisCuts* const filter) {fK0sFilter = filter;}
	50	void SetHistogramManager(AliDielectronHistos * const histos) { fHistos=histos; }
	51	void SetV0KineCuts(AliESDv0KineCuts* const v0cuts) {fV0KineCuts = v0cuts;}
	52
	53	void AddCFVar(Int_t var, Int_t nbins, Double_t lowLim, Double_t highLim);
	54	void AddCFVar(Int_t var, const Char_t* bins);
	55
	56	protected:
	57	// enum {kAllEvents=0, kPhysicsSelectionEvents, kFilteredEvents, kEventStatBins};
	58	// enum {kEventVtxZ=0, kNTracklets, kCentrality, kPt, kPin, kPhi, kEta, kDeltaPt, kDeltaPhi, kDeltaEta, kTPCnCls, kTPCnSigEle, kTPCnSigPio, kTPCnSigPro, kNVariables};
	59
	60	AliCFContainer* fCfContainer; // CF container
	61	AliDielectronHistos *fHistos; // Histogram manager
	62	TList fHistogramList; // histogram list from the manager
	63
	64	Bool_t fSelectPhysics; // Whether to use physics selection
65	UInt_t fTriggerMask; // Event trigger mask
66	Bool_t fRejectPileup; // pileup rejection wanted
67	Bool_t fFillTRDfriendPH; // use the task to fill a TRD tracklet PH container
68
69	AliAnalysisCuts* fEventFilter; // event filter
70	AliAnalysisCuts* fTrackFilter; // tracking filter
71	AliAnalysisCuts* fPairFilter; // pair filter
72	AliESDv0Cuts* fV0Cuts; // v0 standard filter
73	AliAnalysisCuts* fLambdaFilter; // additional cuts for lambda v0 exclusion
74	AliAnalysisCuts* fK0sFilter; // additional cuts for K0s v0 exclusion
75	AliESDv0KineCuts* fV0KineCuts; // V0 kine cuts
76
77	Int_t fCFNVars; // number of CF variables
78	Int_t fCFVarsEnabled[kNMaxDimensions]; // id for every dimension
79	Int_t fCFVarsNbins[kNMaxDimensions]; // number of bins for every CF dimension
80	Double_t fCFVarRanges[kNMaxDimensions][2]; // range for CF dimensions
81	TString fCFVarBins[kNMaxDimensions]; // bin limits for CF dimensions
82
83	TH1D *fEventStat; //! Histogram with event statistics
84
85	void FillContainer(Int_t step, const Double_t* values, Int_t trackId, Int_t pdg);
86
87	AliAnalysisTaskSingleParticle(const AliAnalysisTaskSingleParticle &c);
88	AliAnalysisTaskSingleParticle& operator= (const AliAnalysisTaskSingleParticle &c);
89
90	ClassDef(AliAnalysisTaskSingleParticle, 2); //Analysis Task handling single particle cuts
91	};
92	#endif