[u/mrichter/AliRoot.git] / PWGGA / EMCALJetTasks / AliAnalysisTaskRho.h

#ifndef ALIANALYSISTASKRHO_cxx
#define ALIANALYSISTASKRHO_cxx

// $Id$

class TList;
class TH1F;
class TH2F;
class TClonesArray;
class TString;
class TF1;

#include <TParameter.h>

#include "AliAnalysisTaskRhoBase.h"

class AliAnalysisTaskRho : public AliAnalysisTaskRhoBase {

 public:
  AliAnalysisTaskRho();
  AliAnalysisTaskRho(const char *name);
  AliAnalysisTaskRho(const char *name, Bool_t histo);
  virtual ~AliAnalysisTaskRho() {}
  
  virtual void           UserCreateOutputObjects();
  virtual void           UserExec(Option_t*);
  virtual void           Terminate(Option_t*);

  void                   SetAreaCut(Double_t a = 0.0)                          { fAreaCut       = a    ; }
  void                   SetJetEta(Double_t emin, Double_t emax)               { fEtaMin        = emin ; fEtaMax = emax  ; }
  void                   SetJetPhi(Double_t pmin, Double_t pmax)               { fPhiMin        = pmin ; fPhiMax = pmax  ; }
  void                   SetJetsName(const char *n)                            { fJetsName      = n    ; }
  void                   SetScaleFunction(TF1* sf)                             { fScaleFunction = sf   ; }
  void                   SetTracksName(const char *n)                          { fTracksName    = n    ; }
  void                   SetExcludeLeadJets(UInt_t n)                          { fNExclLeadJets = n    ; }
  
 protected:
  virtual Double_t       GetScaleFactor(Double_t cent);

  TString                fTracksName;                    // name of track collection
  TString                fJetsName;                      // name of jet collection
  TString                fRhoScaledName;                 // name of scaled rho object
  Double_t               fPhiMin;                        // minimum phi
  Double_t               fPhiMax;                        // maximum phi
  Double_t               fEtaMin;                        // minimum eta
  Double_t               fEtaMax;                        // maximum eta
  Double_t               fAreaCut;                       // cut on jet area
  UInt_t                 fNExclLeadJets;                 // number of leading jets to be excluded from the median calculation
  TF1                   *fScaleFunction;                 // pre-computed scale factor as a function of centrality
  Bool_t                 fCreateHisto;                   // whether or not create histograms
  TList                 *fOutputList;                    //!output list
  TH1F                  *fHistCentrality;                //!centrality distribution
  TH1F                  *fHistJetPt;                     //!jet pt distribution
  TH1F                  *fHistJetArea;                   //!jet area
  TH2F                  *fHistRhovsCent;                 //!rho vs. centrality
  TH2F                  *fHistDeltaRhovsCent;            //!delta rho vs. centrality
  TH2F                  *fHistDeltaRhoScalevsCent;       //!delta rhoscaled vs. centrality
  TH2F                  *fHistJetPtvsCent;               //!jet pt vs. centrality
  TH2F                  *fHistJetAreavsCent;             //!jet area vs. centrality
  TH2F                  *fHistNjetvsCent;                //!no. of jets vs. centrality
 
  TH2F                  *fHistRhovsNtrack;               //!rho vs. no. of tracks
  TH2F                  *fHistDeltaRhovsNtrack;          //!delta rho vs. no. of tracks
  TH2F                  *fHistDeltaRhoScalevsNtrack;     //!delta rho scaled vs. no. of tracks
  TH2F                  *fHistJetPtvsNtrack;             //!jet pt vs. no. of tracks
  TH2F                  *fHistJetAreavsNtrack;           //!jet area vs. no. of tracks
  TH2F                  *fHistNjetvsNtrack;              //!no. of jets vs. no. of tracks
  TParameter<Double_t>  *fRhoScaled;                     //!per event scaled rho

  AliAnalysisTaskRho(const AliAnalysisTaskRho&);             // not implemented
  AliAnalysisTaskRho& operator=(const AliAnalysisTaskRho&);  // not implemented
  
  ClassDef(AliAnalysisTaskRho, 4); // Rho task
};
#endif
Commit	Line	Data
020052e4	1	#ifndef ALIANALYSISTASKRHO_cxx
	2	#define ALIANALYSISTASKRHO_cxx
	3
5b5bad1d	4	// $Id$
192fc3f4	5
020052e4	6	class TList;
	7	class TH1F;
	8	class TH2F;
	9	class TClonesArray;
	10	class TString;
192fc3f4	11	class TF1;
	12
	13	#include <TParameter.h>
020052e4	14
192fc3f4	15	#include "AliAnalysisTaskRhoBase.h"
020052e4	16
192fc3f4	17	class AliAnalysisTaskRho : public AliAnalysisTaskRhoBase {
020052e4	18
	19	public:
	20	AliAnalysisTaskRho();
	21	AliAnalysisTaskRho(const char *name);
5d845887	22	AliAnalysisTaskRho(const char *name, Bool_t histo);
020052e4	23	virtual ~AliAnalysisTaskRho() {}
020052e4	24
192fc3f4	25	virtual void UserCreateOutputObjects();
	26	virtual void UserExec(Option_t*);
	27	virtual void Terminate(Option_t*);
020052e4	28
192fc3f4	29	void SetAreaCut(Double_t a = 0.0) { fAreaCut = a ; }
	30	void SetJetEta(Double_t emin, Double_t emax) { fEtaMin = emin ; fEtaMax = emax ; }
	31	void SetJetPhi(Double_t pmin, Double_t pmax) { fPhiMin = pmin ; fPhiMax = pmax ; }
	32	void SetJetsName(const char *n) { fJetsName = n ; }
	33	void SetScaleFunction(TF1* sf) { fScaleFunction = sf ; }
	34	void SetTracksName(const char *n) { fTracksName = n ; }
c60e0a21	35	void SetExcludeLeadJets(UInt_t n) { fNExclLeadJets = n ; }
020052e4	36
020052e4	37	protected:
c60e0a21	38	virtual Double_t GetScaleFactor(Double_t cent);
020052e4	39
020052e4	40	TString fTracksName; // name of track collection
020052e4	41	TString fJetsName; // name of jet collection
192fc3f4	42	TString fRhoScaledName; // name of scaled rho object
	43	Double_t fPhiMin; // minimum phi
	44	Double_t fPhiMax; // maximum phi
	45	Double_t fEtaMin; // minimum eta
	46	Double_t fEtaMax; // maximum eta
	47	Double_t fAreaCut; // cut on jet area
c60e0a21	48	UInt_t fNExclLeadJets; // number of leading jets to be excluded from the median calculation
192fc3f4	49	TF1 *fScaleFunction; // pre-computed scale factor as a function of centrality
c60e0a21	50	Bool_t fCreateHisto; // whether or not create histograms
192fc3f4	51	TList *fOutputList; //!output list
	52	TH1F *fHistCentrality; //!centrality distribution
	53	TH1F *fHistJetPt; //!jet pt distribution
	54	TH1F *fHistJetArea; //!jet area
	55	TH2F *fHistRhovsCent; //!rho vs. centrality
	56	TH2F *fHistDeltaRhovsCent; //!delta rho vs. centrality
	57	TH2F *fHistDeltaRhoScalevsCent; //!delta rhoscaled vs. centrality
	58	TH2F *fHistJetPtvsCent; //!jet pt vs. centrality
	59	TH2F *fHistJetAreavsCent; //!jet area vs. centrality
	60	TH2F *fHistNjetvsCent; //!no. of jets vs. centrality
	61
	62	TH2F *fHistRhovsNtrack; //!rho vs. no. of tracks
	63	TH2F *fHistDeltaRhovsNtrack; //!delta rho vs. no. of tracks
	64	TH2F *fHistDeltaRhoScalevsNtrack; //!delta rho scaled vs. no. of tracks
	65	TH2F *fHistJetPtvsNtrack; //!jet pt vs. no. of tracks
	66	TH2F *fHistJetAreavsNtrack; //!jet area vs. no. of tracks
	67	TH2F *fHistNjetvsNtrack; //!no. of jets vs. no. of tracks
	68	TParameter<Double_t> *fRhoScaled; //!per event scaled rho
020052e4	69
	70	AliAnalysisTaskRho(const AliAnalysisTaskRho&); // not implemented
	71	AliAnalysisTaskRho& operator=(const AliAnalysisTaskRho&); // not implemented
	72
5b5bad1d	73	ClassDef(AliAnalysisTaskRho, 4); // Rho task
020052e4	74	};
020052e4	75	#endif