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

#ifndef AliAnalysisTaskEmcalJetHMEC_H
#define AliAnalysisTaskEmcalJetHMEC_H

// $Id$

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

#include "AliAnalysisTaskEmcalJet.h"

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