[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 AliESDEvent;
class AliESDtrackCuts;
class AliESDtrack;
class AliEventplane;
class AliOADBContainer;
class AliVTrack;
class THnSparse;

class AliEPSelectionTask : public AliAnalysisTaskSE {

 public:
  
  enum ResoMethod{kRandom,kEta};

  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,AliEventplane* EP);
  Double_t GetWeight(TObject* track1);
  Double_t GetPhiWeight(TObject* track1);

  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 SetTrackType(TString tracktype);
  void SetPhiDist();
  void SetPersonalESDtrackCuts(AliESDtrackCuts* trackcuts);
  void SetPersonalAODtrackCuts(UInt_t filterbit = 1, Float_t etalow = -0.8, Float_t etaup = 0.8, Float_t ptlow = 0.5, Float_t ptup = 20.);
  void SetPersonalPhiDistribution(const char* filename, char* listname);
  void SetEtaGap(Float_t etagap)             {fEtaGap = etagap;}
  void SetSubeventsSplitMethod(Int_t method) {fSplitMethod = method;}
  
 private:
   
  AliEPSelectionTask(const AliEPSelectionTask& ep);
  AliEPSelectionTask& operator= (const AliEPSelectionTask& ep); 

  TObjArray* GetAODTracksAndMaxID(AliAODEvent* aod, Int_t& maxid);
  void SetOADBandPeriod();
  TH1F* SelectPhiDist(AliVTrack *track);
  TObjArray* GetTracksForLHC11h(AliESDEvent* esd);

  TString  fAnalysisInput; 		// "ESD", "AOD"
  TString  fTrackType;			// "GLOBAL", "TPC"
  TString  fPeriod;	                // "LHC11h","LHC10h"
  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
  Bool_t   fUserphidist;		// bool, if personal phi distribution should be used
  Bool_t   fUsercuts;			// bool, if personal cuts should be used
  Int_t    fRunNumber;			// runnumber
  UInt_t   fAODfilterbit;               // AOD filter bit for AOD track selection  
  Float_t  fEtaGap;                     // Eta Gap between Subevent A and B
  Int_t    fSplitMethod;                // Splitting Method for subevents
  

  AliESDtrackCuts* fESDtrackCuts;       // track cuts
  
  AliOADBContainer* fEPContainer;	//! OADB Container
  TH1F*	 fPhiDist[4];			// array of Phi distributions used to calculate phi weights
  THnSparse *fSparseDist;               //! THn for eta-charge phi-weighting
  TH1F *fHruns;                         // information about runwise statistics of 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;                   // Output histograms
  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,4); 
};

#endif
Commit	Line	Data
9663fa4b	1	#ifndef AliEPSELECTIONTASK_H
9663fa4b	2	#define AliEPSELECTIONTASK_H
ce7adfe9	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;
ce7adfe9	20
	21	class AliESDEvent;
	22	class AliESDtrackCuts;
	23	class AliESDtrack;
	24	class AliEventplane;
90267ca6	25	class AliOADBContainer;
9663fa4b	26	class AliVTrack;
9663fa4b	27	class THnSparse;
ce7adfe9	28
	29	class AliEPSelectionTask : public AliAnalysisTaskSE {
	30
	31	public:
e51055a0	32
e51055a0	33	enum ResoMethod{kRandom,kEta};
ce7adfe9	34
	35	AliEPSelectionTask();
	36	AliEPSelectionTask(const char *name);
	37	virtual ~AliEPSelectionTask();
	38
	39	// Implementation of interface methods
	40	virtual void UserCreateOutputObjects();
	41	virtual void UserExec(Option_t *option);
	42	virtual void Terminate(Option_t *option);
	43
	44	TVector2 GetQ(AliEventplane* EP, TObjArray* event);
e51055a0	45	void GetQsub(TVector2& Qsub1, TVector2& Qsub2, TObjArray* event,AliEventplane* EP);
	46	Double_t GetWeight(TObject* track1);
	47	Double_t GetPhiWeight(TObject* track1);
ce7adfe9	48
e51055a0	49	virtual void SetDebugLevel(Int_t level) {fDebug = level;}
	50	void SetInput(const char* input) {fAnalysisInput = input;}
	51	void SetUseMCRP() {fUseMCRP = kTRUE;}
	52	void SetUsePhiWeight() {fUsePhiWeight = kTRUE;}
	53	void SetUsePtWeight() {fUsePtWeight = kTRUE;}
	54	void SetSaveTrackContribution() {fSaveTrackContribution = kTRUE;}
90267ca6	55	void SetTrackType(TString tracktype);
	56	void SetPhiDist();
	57	void SetPersonalESDtrackCuts(AliESDtrackCuts* trackcuts);
e51055a0	58	void SetPersonalAODtrackCuts(UInt_t filterbit = 1, Float_t etalow = -0.8, Float_t etaup = 0.8, Float_t ptlow = 0.5, Float_t ptup = 20.);
71916547	59	void SetPersonalPhiDistribution(const char* filename, char* listname);
e51055a0	60	void SetEtaGap(Float_t etagap) {fEtaGap = etagap;}
e51055a0	61	void SetSubeventsSplitMethod(Int_t method) {fSplitMethod = method;}
ce7adfe9	62
	63	private:
	64
	65	AliEPSelectionTask(const AliEPSelectionTask& ep);
	66	AliEPSelectionTask& operator= (const AliEPSelectionTask& ep);
	67
e51055a0	68	TObjArray* GetAODTracksAndMaxID(AliAODEvent* aod, Int_t& maxid);
9663fa4b	69	void SetOADBandPeriod();
	70	TH1F* SelectPhiDist(AliVTrack *track);
	71	TObjArray* GetTracksForLHC11h(AliESDEvent* esd);
e51055a0	72
ce7adfe9	73	TString fAnalysisInput; // "ESD", "AOD"
90267ca6	74	TString fTrackType; // "GLOBAL", "TPC"
9663fa4b	75	TString fPeriod; // "LHC11h","LHC10h"
ce7adfe9	76	Bool_t fUseMCRP; // i.e. usable for Therminator, when MC RP is provided
	77	Bool_t fUsePhiWeight; // use of phi weights
	78	Bool_t fUsePtWeight; // use of pT weights
	79	Bool_t fSaveTrackContribution; // storage of contribution of each track to Q-Vector
1e552991	80	Bool_t fUserphidist; // bool, if personal phi distribution should be used
	81	Bool_t fUsercuts; // bool, if personal cuts should be used
	82	Int_t fRunNumber; // runnumber
e51055a0	83	UInt_t fAODfilterbit; // AOD filter bit for AOD track selection
	84	Float_t fEtaGap; // Eta Gap between Subevent A and B
	85	Int_t fSplitMethod; // Splitting Method for subevents
ce7adfe9	86
e51055a0	87
1e552991	88	AliESDtrackCuts* fESDtrackCuts; // track cuts
ce7adfe9	89
90267ca6	90	AliOADBContainer* fEPContainer; //! OADB Container
9663fa4b	91	TH1F* fPhiDist[4]; // array of Phi distributions used to calculate phi weights
	92	THnSparse *fSparseDist; //! THn for eta-charge phi-weighting
	93	TH1F *fHruns; // information about runwise statistics of phi-weights
ce7adfe9	94
08c34a44	95	TVector2* fQVector; //! Q-Vector of the event
	96	Double_t* fQContributionX; //! array of the tracks' contributions to X component of Q-Vector - index = track ID
	97	Double_t* fQContributionY; //! array of the tracks' contributions to Y component of Q-Vector - index = track ID
	98	Double_t fEventplaneQ; //! Event plane angle from Q-Vector
	99	TVector2* fQsub1; //! Q-Vector of sub-event 1
	100	TVector2* fQsub2; //! Q-Vector of sub-event 2
	101	Double_t fQsubRes; //! Difference of EP angles of subevents
ce7adfe9	102
71916547	103	TList* fOutputList; // Output histograms
08c34a44	104	TH1F* fHOutEventplaneQ; //! control histogram: Event Plane angle
	105	TH1F* fHOutPhi; //! control histogram: original phi distribution
	106	TH1F* fHOutPhiCorr; //! control histogram: corrected phi distribution
	107	TH2F* fHOutsub1sub2; //! control histogram: correlation of EP from subevents
	108	TH2F* fHOutNTEPRes; //! control histogram: Difference of EP angles of subevents vs Nch
	109	TH2F* fHOutPTPsi; //! control histogram: Difference of EP angle and emission angle of track vs track pT
	110	TH2F* fHOutDiff; //! control histogram: Difference of MC RP and EP - only filled if fUseMCRP is true!
	111	TH2F* fHOutleadPTPsi; //! control histogram: emission angle of leading pT track vs EP angle
ce7adfe9	112
1e552991	113	ClassDef(AliEPSelectionTask,4);
ce7adfe9	114	};
	115
	116	#endif
	117