[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / UserTasks / AliAnalysisTaskEmcalJetHMEC.h

#ifndef AliAnalysisTaskEmcalJetHMEC_H
#define AliAnalysisTaskEmcalJetHMEC_H

// $Id$

class TList;
class TH1;
class TH2;
class THnSparse;
class AliEmcalJet;
class AliESDEvent;
class AliEventPoolManager;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskEmcalJetHMEC : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskEmcalJetHMEC();
  AliAnalysisTaskEmcalJetHMEC(const char *name);
  virtual ~AliAnalysisTaskEmcalJetHMEC() {}
  
  virtual void            UserCreateOutputObjects();
  virtual Double_t        RelativePhi(Double_t mphi, Double_t vphi);
  virtual void            UserExec(Option_t *option);
  virtual void            Terminate(Option_t *);
  virtual Int_t           AcceptJet(AliEmcalJet *jet);
  virtual THnSparse*      NewTHnSparseF(const char* name, UInt_t entries);
  virtual void            GetDimParams(Int_t iEntry,TString &label, Int_t &nbins, Double_t &xmin, Double_t &xmax);

  virtual void            SetTracksName(const char *n)             {fTracksName=n;}
  virtual void            SetJetsName(const char *jn)              {fJetsName=jn;}

  virtual void            SetAreaCut(Double_t a)                   { fAreacut    = a; }
  virtual void            SetTrkBias(Double_t b)                   { fTrkBias    = b; }  //require a track with pt > b in jet
  virtual void            SetClusBias(Double_t b)                  { fClusBias   = b; }  //require a cluster with pt > b in jet

  virtual void            SetTrkEta(Double_t e)                    { fTrkEta   = e; }  //eta range of the associated tracks

  virtual void            SetJetEta(Double_t emin, Double_t emax)  { fEtamin = emin; fEtamax = emax; }
  virtual void            SetJetPhi(Double_t pmin, Double_t pmax)  { fPhimin = pmin; fPhimax = pmax; }
  virtual void            SetEventMixing(Int_t yesno)              { fDoEventMixing=yesno;}
  virtual void            SetMixingTracks(Int_t tracks)            { fMixingTracks = tracks; }


 protected:
  virtual Int_t          GetCentBin(Double_t cent) const;
  virtual Int_t          GetEtaBin(Double_t eta) const;
  virtual Int_t          GetpTjetBin(Double_t pt) const;

  TString                fTracksName;              //name of tracks collection
  TString                fJetsName;                //name of Jet collection
  Double_t               fPhimin;                  // phi min of jet
  Double_t               fPhimax;                  // phi max of jet
  Double_t               fEtamin;                  // eta min of jet
  Double_t               fEtamax;                  // eta max of jet
  Double_t               fAreacut;                 // area cut of jet
  Double_t               fTrkBias;
  Double_t               fClusBias;
  Double_t               fTrkEta;                  // eta min/max of tracks
  Int_t                  fDoEventMixing;           // flag to do evt mixing
  Int_t  		 fMixingTracks;		   // size of track buffer for event mixing
  TObjArray*             CloneAndReduceTrackList(TObjArray* tracks);

  AliESDEvent           *fESD;    //! ESD object
  AliEventPoolManager   *fPoolMgr; //!
  TList                 *fOutputList; //! Output list
  TH1                   *fHistTrackPt; //! Pt spectrum
  TH1                   *fHistCentrality;//!
  TH2                   *fHistJetEtaPhi;//!
  TH2                   *fHistTrackEtaPhi[7];//!
  TH2                   *fHistJetHEtaPhi;//!

  TH1                   *fHistJetPt[6]; //!
  TH1                   *fHistJetPtBias[6];//!
  TH1                   *fHistLeadJetPt[6];//!
  TH1                   *fHistLeadJetPtBias[6];//!
  TH1                   *fHistJetPtTT[6];//!
  TH2                   *fHistJetH[6][5][3];//!
  TH2                   *fHistJetHBias[6][5][3];//!
  TH2                   *fHistJetHTT[6][5][3];//!
  THnSparse             *fhnMixedEvents;      //!mixed events matrix
  THnSparse             *fhnJH;      //!Fg events matrix

 private:
   
  AliAnalysisTaskEmcalJetHMEC(const AliAnalysisTaskEmcalJetHMEC&); // not implemented
  AliAnalysisTaskEmcalJetHMEC& operator=(const AliAnalysisTaskEmcalJetHMEC&); // not implemented
  
  ClassDef(AliAnalysisTaskEmcalJetHMEC, 10); 
};
#endif
Commit	Line	Data
5b38fc6d	1	#ifndef AliAnalysisTaskEmcalJetHMEC_H
5b38fc6d	2	#define AliAnalysisTaskEmcalJetHMEC_H
1101e468	3
7f71d351	4	// $Id$
1101e468	5
	6	class TList;
	7	class TH1;
	8	class TH2;
55f64c2f	9	class THnSparse;
c1076e58	10	class AliEmcalJet;
1101e468	11	class AliESDEvent;
55f64c2f	12	class AliEventPoolManager;
55f64c2f	13
1101e468	14	#include "AliAnalysisTaskSE.h"
	15
	16	class AliAnalysisTaskEmcalJetHMEC : public AliAnalysisTaskSE {
	17	public:
	18	AliAnalysisTaskEmcalJetHMEC();
	19	AliAnalysisTaskEmcalJetHMEC(const char *name);
	20	virtual ~AliAnalysisTaskEmcalJetHMEC() {}
	21
1af24178	22	virtual void UserCreateOutputObjects();
	23	virtual Double_t RelativePhi(Double_t mphi, Double_t vphi);
	24	virtual void UserExec(Option_t *option);
	25	virtual void Terminate(Option_t *);
c1076e58	26	virtual Int_t AcceptJet(AliEmcalJet *jet);
1af24178	27	virtual THnSparse* NewTHnSparseF(const char* name, UInt_t entries);
1af24178	28	virtual void GetDimParams(Int_t iEntry,TString &label, Int_t &nbins, Double_t &xmin, Double_t &xmax);
55f64c2f	29
965c985f	30	virtual void SetTracksName(const char *n) {fTracksName=n;}
965c985f	31	virtual void SetJetsName(const char *jn) {fJetsName=jn;}
1101e468	32
1af24178	33	virtual void SetAreaCut(Double_t a) { fAreacut = a; }
965c985f	34	virtual void SetTrkBias(Double_t b) { fTrkBias = b; } //require a track with pt > b in jet
	35	virtual void SetClusBias(Double_t b) { fClusBias = b; } //require a cluster with pt > b in jet
	36
22d3f9c2	37	virtual void SetTrkEta(Double_t e) { fTrkEta = e; } //eta range of the associated tracks
22d3f9c2	38
1af24178	39	virtual void SetJetEta(Double_t emin, Double_t emax) { fEtamin = emin; fEtamax = emax; }
1af24178	40	virtual void SetJetPhi(Double_t pmin, Double_t pmax) { fPhimin = pmin; fPhimax = pmax; }
965c985f	41	virtual void SetEventMixing(Int_t yesno) { fDoEventMixing=yesno;}
965c985f	42	virtual void SetMixingTracks(Int_t tracks) { fMixingTracks = tracks; }
55f64c2f	43
	44
	45
	46
1101e468	47
	48	protected:
	49	virtual Int_t GetCentBin(Double_t cent) const;
	50	virtual Int_t GetEtaBin(Double_t eta) const;
	51	virtual Int_t GetpTjetBin(Double_t pt) const;
	52
965c985f	53	TString fTracksName; //name of tracks collection
965c985f	54	TString fJetsName; //name of Jet collection
22d3f9c2	55	Double_t fPhimin; // phi min of jet
	56	Double_t fPhimax; // phi max of jet
	57	Double_t fEtamin; // eta min of jet
	58	Double_t fEtamax; // eta max of jet
	59	Double_t fAreacut; // area cut of jet
965c985f	60	Double_t fTrkBias;
965c985f	61	Double_t fClusBias;
22d3f9c2	62	Double_t fTrkEta; // eta min/max of tracks
	63	Int_t fDoEventMixing; // flag to do evt mixing
	64	Int_t fMixingTracks; // size of track buffer for event mixing
1af24178	65	TObjArray* CloneAndReduceTrackList(TObjArray* tracks);
55f64c2f	66
1af24178	67	AliESDEvent *fESD; //! ESD object
9e4a0350	68	AliEventPoolManager *fPoolMgr; //!
1af24178	69	TList *fOutputList; //! Output list
1af24178	70	TH1 *fHistTrackPt; //! Pt spectrum
9e4a0350	71	TH1 *fHistCentrality;//!
	72	TH2 *fHistJetEtaPhi;//!
	73	TH2 *fHistTrackEtaPhi[7];//!
	74	TH2 *fHistJetHEtaPhi;//!
	75
	76	TH1 *fHistJetPt[6]; //!
	77	TH1 *fHistJetPtBias[6];//!
	78	TH1 *fHistLeadJetPt[6];//!
	79	TH1 *fHistLeadJetPtBias[6];//!
	80	TH1 *fHistJetPtTT[6];//!
	81	TH2 *fHistJetH[6][5][3];//!
	82	TH2 *fHistJetHBias[6][5][3];//!
	83	TH2 *fHistJetHTT[6][5][3];//!
1af24178	84	THnSparse *fhnMixedEvents; //!mixed events matrix
9e4a0350	85	THnSparse *fhnJH; //!Fg events matrix
1101e468	86
5b38fc6d	87	private:
1101e468	88
	89	AliAnalysisTaskEmcalJetHMEC(const AliAnalysisTaskEmcalJetHMEC&); // not implemented
	90	AliAnalysisTaskEmcalJetHMEC& operator=(const AliAnalysisTaskEmcalJetHMEC&); // not implemented
	91
41f87d59	92	ClassDef(AliAnalysisTaskEmcalJetHMEC, 10);
55f64c2f	93	};
1101e468	94	#endif