[u/mrichter/AliRoot.git] / ANALYSIS / AliEPSelectionTask.h

#ifndef ALIEPSELECTIONTASK_H
#define ALIEPSELECTIONTASK_H

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

//*****************************************************
//   Class AliEPSelectionTask
//   author: Alberica Toia, Johanna Gramling
//*****************************************************

#include "AliAnalysisTaskSE.h"

class TFile;
class TH1F;
class TH2F;
class TList;
class TString;
class TVector2;
class TObjArray;

class AliESDEvent;
class AliESDtrackCuts;
class AliESDtrack;
class AliEventplane;

class AliEPSelectionTask : public AliAnalysisTaskSE {

 public:

  AliEPSelectionTask();
  AliEPSelectionTask(const char *name);
  virtual ~AliEPSelectionTask();

  // Implementation of interface methods
  virtual void UserCreateOutputObjects();
  virtual void UserExec(Option_t *option);
  virtual void Terminate(Option_t *option);
  
  TVector2 GetQ(AliEventplane* EP, TObjArray* event);
  void GetQsub(TVector2& Qsub1, TVector2& Qsub2, TObjArray* event);
  Double_t GetWeight(AliESDtrack* track);
  Double_t GetPhiWeight(AliESDtrack* track);

  virtual void  SetDebugLevel(Int_t level) {fDebug = level;}
  void SetInput(const char* input)         {fAnalysisInput = input;}
  void SetUseMCRP()			   {fUseMCRP = kTRUE;}
  void SetUsePhiWeight()		   {fUsePhiWeight = kTRUE;}
  void SetUsePtWeight()			   {fUsePtWeight = kTRUE;}
  void SetSaveTrackContribution()	   {fSaveTrackContribution = kTRUE;}
  void SetESDtrackCuts(TString status);
  void SetPhiDistribution(char* filename, char* listname);
  
 private:
   
  AliEPSelectionTask(const AliEPSelectionTask& ep);
  AliEPSelectionTask& operator= (const AliEPSelectionTask& ep); 

  Int_t    fDebug;	   		// Debug flag
  TString  fAnalysisInput; 		// "ESD", "AOD"
  TString  fStatus;			// "GLOBAL", "TPC"
  Bool_t   fUseMCRP;			// i.e. usable for Therminator, when MC RP is provided
  Bool_t   fUsePhiWeight;		// use of phi weights
  Bool_t   fUsePtWeight;		// use of pT weights
  Bool_t   fSaveTrackContribution;	// storage of contribution of each track to Q-Vector
  
  AliESDtrackCuts* fESDtrackCuts;       //! track cuts
  
  TObjArray* ftracklist;		//! list of accepted tracks for Q-Vector
  TH1F*	 fPhiDist;			// Phi distribution used to calculate phi weights

  TVector2* fQVector;			//! Q-Vector of the event  
  Double_t* fQContributionX;		//! array of the tracks' contributions to X component of Q-Vector - index = track ID
  Double_t* fQContributionY;		//! array of the tracks' contributions to Y component of Q-Vector - index = track ID
  Double_t  fEventplaneQ; 		//! Event plane angle from Q-Vector
  TVector2* fQsub1;			//! Q-Vector of sub-event 1
  TVector2* fQsub2;			//! Q-Vector of sub-event 2
  Double_t  fQsubRes;			//! Difference of EP angles of subevents
  
  TList* fOutputList;  
  TH1F*  fHOutEventplaneQ;    		//! control histogram: Event Plane angle
  TH1F*  fHOutPhi;			//! control histogram: original phi distribution
  TH1F*	 fHOutPhiCorr;			//! control histogram: corrected phi distribution
  TH2F*  fHOutsub1sub2;			//! control histogram: correlation of EP from subevents
  TH2F*  fHOutNTEPRes;			//! control histogram: Difference of EP angles of subevents vs Nch
  TH2F*  fHOutPTPsi;			//! control histogram: Difference of EP angle and emission angle of track vs track pT
  TH2F*	 fHOutDiff;			//! control histogram: Difference of MC RP and EP - only filled if fUseMCRP is true!
  TH2F*  fHOutleadPTPsi;		//! control histogram: emission angle of leading pT track vs EP angle

  ClassDef(AliEPSelectionTask,2); 
};

#endif
Commit	Line	Data
ce7adfe9	1	#ifndef ALIEPSELECTIONTASK_H
	2	#define ALIEPSELECTIONTASK_H
	3
	4	/* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	//*****************************************************
	8	// Class AliEPSelectionTask
	9	// author: Alberica Toia, Johanna Gramling
	10	//*****************************************************
	11
	12	#include "AliAnalysisTaskSE.h"
	13
	14	class TFile;
	15	class TH1F;
	16	class TH2F;
	17	class TList;
	18	class TString;
	19	class TVector2;
	20	class TObjArray;
	21
	22	class AliESDEvent;
	23	class AliESDtrackCuts;
	24	class AliESDtrack;
	25	class AliEventplane;
	26
	27	class AliEPSelectionTask : public AliAnalysisTaskSE {
	28
	29	public:
	30
	31	AliEPSelectionTask();
	32	AliEPSelectionTask(const char *name);
	33	virtual ~AliEPSelectionTask();
	34
	35	// Implementation of interface methods
	36	virtual void UserCreateOutputObjects();
	37	virtual void UserExec(Option_t *option);
	38	virtual void Terminate(Option_t *option);
	39
	40	TVector2 GetQ(AliEventplane* EP, TObjArray* event);
	41	void GetQsub(TVector2& Qsub1, TVector2& Qsub2, TObjArray* event);
	42	Double_t GetWeight(AliESDtrack* track);
	43	Double_t GetPhiWeight(AliESDtrack* track);
	44
	45	virtual void SetDebugLevel(Int_t level) {fDebug = level;}
	46	void SetInput(const char* input) {fAnalysisInput = input;}
	47	void SetUseMCRP() {fUseMCRP = kTRUE;}
	48	void SetUsePhiWeight() {fUsePhiWeight = kTRUE;}
	49	void SetUsePtWeight() {fUsePtWeight = kTRUE;}
	50	void SetSaveTrackContribution() {fSaveTrackContribution = kTRUE;}
	51	void SetESDtrackCuts(TString status);
	52	void SetPhiDistribution(char* filename, char* listname);
	53
	54	private:
	55
	56	AliEPSelectionTask(const AliEPSelectionTask& ep);
	57	AliEPSelectionTask& operator= (const AliEPSelectionTask& ep);
	58
	59	Int_t fDebug; // Debug flag
	60	TString fAnalysisInput; // "ESD", "AOD"
	61	TString fStatus; // "GLOBAL", "TPC"
	62	Bool_t fUseMCRP; // i.e. usable for Therminator, when MC RP is provided
	63	Bool_t fUsePhiWeight; // use of phi weights
	64	Bool_t fUsePtWeight; // use of pT weights
65	Bool_t fSaveTrackContribution; // storage of contribution of each track to Q-Vector
66
08c34a44	67	AliESDtrackCuts* fESDtrackCuts; //! track cuts
ce7adfe9	68
08c34a44	69	TObjArray* ftracklist; //! list of accepted tracks for Q-Vector
ce7adfe9	70	TH1F* fPhiDist; // Phi distribution used to calculate phi weights
ce7adfe9	71
08c34a44	72	TVector2* fQVector; //! Q-Vector of the event
	73	Double_t* fQContributionX; //! array of the tracks' contributions to X component of Q-Vector - index = track ID
	74	Double_t* fQContributionY; //! array of the tracks' contributions to Y component of Q-Vector - index = track ID
	75	Double_t fEventplaneQ; //! Event plane angle from Q-Vector
	76	TVector2* fQsub1; //! Q-Vector of sub-event 1
	77	TVector2* fQsub2; //! Q-Vector of sub-event 2
	78	Double_t fQsubRes; //! Difference of EP angles of subevents
ce7adfe9	79
ce7adfe9	80	TList* fOutputList;
08c34a44	81	TH1F* fHOutEventplaneQ; //! control histogram: Event Plane angle
	82	TH1F* fHOutPhi; //! control histogram: original phi distribution
	83	TH1F* fHOutPhiCorr; //! control histogram: corrected phi distribution
	84	TH2F* fHOutsub1sub2; //! control histogram: correlation of EP from subevents
	85	TH2F* fHOutNTEPRes; //! control histogram: Difference of EP angles of subevents vs Nch
	86	TH2F* fHOutPTPsi; //! control histogram: Difference of EP angle and emission angle of track vs track pT
	87	TH2F* fHOutDiff; //! control histogram: Difference of MC RP and EP - only filled if fUseMCRP is true!
	88	TH2F* fHOutleadPTPsi; //! control histogram: emission angle of leading pT track vs EP angle
ce7adfe9	89
	90	ClassDef(AliEPSelectionTask,2);
	91	};
	92
	93	#endif
	94