[u/mrichter/AliRoot.git] / PWG4 / CorrelationsDPhi / FourierDecomposition / AliDhcTask.h

// Dihadron correlations task - simple task to read ESD or AOD input,
// calculate same- and mixed-event correlations, and fill THnSparse
// output. -A. Adare, Apr 2011


#ifndef AliDhcTask_cxx
#define AliDhcTask_cxx

class TH1;
class TH2;
class THnSparse;
class TObject;
class TObjArray;
class AliESDEvent;
class AliAODEvent;
class AliESDtrackCuts;
class KiddiePoolManager;

#include "AliAnalysisTaskSE.h"
#include "KiddiePoolClasses.h"

class AliDhcTask : public AliAnalysisTaskSE {
 public:
  AliDhcTask() : 
    AliAnalysisTaskSE(), fVerbosity(0), fESD(0), fAOD(0), fOutputList(0), 
    fHistPt(0), fHEvt(0), fHTrk(0), fHS(0), fHM(0), fPoolMgr(0), fCentrality(99), 
    fZVertex(99), fZVtxMax(10), fPtMin(0), fPtMax(100), fInputHandler(0), 
    fEsdTrackCutsTPCOnly(0) {}
  AliDhcTask(const char *name);
  virtual ~AliDhcTask() {}
  
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *option);
  virtual void   Terminate(Option_t *);
  void SetVerbosity(const Int_t v) { fVerbosity = v; }

 protected:
  enum ePairHistAxes  {kDeta, kPtAssc, kPtTrig, kCent, kDphi,
		       kZvtx, kChargeComb};
  enum eEventHistAxes {kZvtxEvt, kCentV0M, kCentCL1};
  enum eTrackHistAxes {kPhiTrk, kEtaTrk};
  enum ePairingScheme {kSameEvt, kDiffEvt};
  enum eDataType {kESD, kAOD};

  void BookHistos();
  void InitEventMixer();
  MiniEvent* GetESDTrax() const;
  MiniEvent* GetAODTrax() const;
  Bool_t VertexOk(TObject* obj) const;
  Double_t DeltaPhi(Double_t phia, Double_t phib, 
		    Double_t rangeMin = -TMath::Pi()/2, 
		    Double_t rangeMax = 3*TMath::Pi()/2) const;
  Int_t Correlate(const MiniEvent &arr1, const MiniEvent &arr2, 
		  Int_t pairing = kSameEvt, Double_t weight = 1.);

 private:
  Int_t        fVerbosity;       // 0 = silence
  AliESDEvent *fESD;             //! ESD object
  AliAODEvent *fAOD;             //! AOD object
  TList       *fOutputList;      //! Output list
  TH1F        *fHistPt;          //! Pt spectrum
  TH2         *fHEvt;            //! Cent, vtx, etc.
  TH2         *fHTrk;            //! Phi, Eta, etc.
  THnSparse   *fHS;              //! Same-evt correlations
  THnSparse   *fHM;              //! Diff-evt correlations
  KiddiePoolManager* fPoolMgr;   //! Event mixer
  Double_t    fCentrality;       //! V0M for now
  Double_t    fZVertex;          //! Of current event
  Double_t    fZVtxMax;          //! Max |z| cut (cm)
  Double_t    fPtMin;            //! Min pt cut
  Double_t    fPtMax;            //! Max pt cut
  AliInputEventHandler*  fInputHandler;
  AliESDtrackCuts* fEsdTrackCutsTPCOnly;

  AliDhcTask(const AliDhcTask&);            // not implemented
  AliDhcTask &operator=(const AliDhcTask&); // not implemented

  ClassDef(AliDhcTask, 1);
};

#endif
Commit	Line	Data
48a61f36	1	// Dihadron correlations task - simple task to read ESD or AOD input,
	2	// calculate same- and mixed-event correlations, and fill THnSparse
	3	// output. -A. Adare, Apr 2011
	4
	5
	6	#ifndef AliDhcTask_cxx
	7	#define AliDhcTask_cxx
	8
	9	class TH1;
	10	class TH2;
	11	class THnSparse;
	12	class TObject;
	13	class TObjArray;
	14	class AliESDEvent;
	15	class AliAODEvent;
	16	class AliESDtrackCuts;
	17	class KiddiePoolManager;
	18
	19	#include "AliAnalysisTaskSE.h"
	20	#include "KiddiePoolClasses.h"
	21
	22	class AliDhcTask : public AliAnalysisTaskSE {
	23	public:
	24	AliDhcTask() :
	25	AliAnalysisTaskSE(), fVerbosity(0), fESD(0), fAOD(0), fOutputList(0),
	26	fHistPt(0), fHEvt(0), fHTrk(0), fHS(0), fHM(0), fPoolMgr(0), fCentrality(99),
	27	fZVertex(99), fZVtxMax(10), fPtMin(0), fPtMax(100), fInputHandler(0),
	28	fEsdTrackCutsTPCOnly(0) {}
	29	AliDhcTask(const char *name);
	30	virtual ~AliDhcTask() {}
	31
	32	virtual void UserCreateOutputObjects();
	33	virtual void UserExec(Option_t *option);
	34	virtual void Terminate(Option_t *);
	35	void SetVerbosity(const Int_t v) { fVerbosity = v; }
	36
	37	protected:
	38	enum ePairHistAxes {kDeta, kPtAssc, kPtTrig, kCent, kDphi,
	39	kZvtx, kChargeComb};
	40	enum eEventHistAxes {kZvtxEvt, kCentV0M, kCentCL1};
	41	enum eTrackHistAxes {kPhiTrk, kEtaTrk};
	42	enum ePairingScheme {kSameEvt, kDiffEvt};
	43	enum eDataType {kESD, kAOD};
	44
	45	void BookHistos();
	46	void InitEventMixer();
	47	MiniEvent* GetESDTrax() const;
	48	MiniEvent* GetAODTrax() const;
	49	Bool_t VertexOk(TObject* obj) const;
	50	Double_t DeltaPhi(Double_t phia, Double_t phib,
	51	Double_t rangeMin = -TMath::Pi()/2,
	52	Double_t rangeMax = 3*TMath::Pi()/2) const;
	53	Int_t Correlate(const MiniEvent &arr1, const MiniEvent &arr2,
	54	Int_t pairing = kSameEvt, Double_t weight = 1.);
	55
	56	private:
	57	Int_t fVerbosity; // 0 = silence
	58	AliESDEvent *fESD; //! ESD object
	59	AliAODEvent *fAOD; //! AOD object
	60	TList *fOutputList; //! Output list
	61	TH1F *fHistPt; //! Pt spectrum
	62	TH2 *fHEvt; //! Cent, vtx, etc.
	63	TH2 *fHTrk; //! Phi, Eta, etc.
	64	THnSparse *fHS; //! Same-evt correlations
65	THnSparse *fHM; //! Diff-evt correlations
66	KiddiePoolManager* fPoolMgr; //! Event mixer
67	Double_t fCentrality; //! V0M for now
68	Double_t fZVertex; //! Of current event
69	Double_t fZVtxMax; //! Max \|z\| cut (cm)
70	Double_t fPtMin; //! Min pt cut
71	Double_t fPtMax; //! Max pt cut
72	AliInputEventHandler* fInputHandler;
73	AliESDtrackCuts* fEsdTrackCutsTPCOnly;
74
75	AliDhcTask(const AliDhcTask&); // not implemented
76	AliDhcTask &operator=(const AliDhcTask&); // not implemented
77
78	ClassDef(AliDhcTask, 1);
79	};
80
81	#endif