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

#ifndef ALIANALYSISTASKRHO_cxx
#define ALIANALYSISTASKRHO_cxx

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

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskRho : public AliAnalysisTaskSE {

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

  void                  SetTracksName(const char *n)                          { fTracksName  = n    ; }
  void                  SetJetsName(const char *n)                            { fJetsName    = n    ; }
  void                  SetRhosName(const char *rn)                           { fRhosName    = rn   ; }
  void                  SetJetPhi(Double_t pmin = 1.8, Double_t pmax = 2.74)  { fPhiMin      = pmin ; fPhiMax = pmax  ; }
  void                  SetJetEta(Double_t emin = -0.3, Double_t emax = 0.3)  { fEtaMin      = emin ; fEtaMax = emax  ; }
  void                  SetAreaCut(Double_t a = 0.0)                          { fAreaCut     = a    ; }
  
 protected:
  virtual void          Sort(vector<Double_t>& v)            ;
  virtual Double_t      GetMedian(vector<Double_t> v, int c) ;
  virtual Double_t      GetScaleFactor(Double_t cent)        ;
  virtual Double_t      GetRhoFactor(Double_t cent)          ;

 private: 
  TString                fTracksName;                    // name of track collection
  TString                fJetsName;                      // name of jet collection
  TString                fRhosName;                      // name of Rho array output
  TString                fClustersName;                  // name of clusters collection

  TList                 *fOutputList;                    //! Output list
  TH1F                  *fHistCentrality;
  TH1F                  *fHistJetPt;
  TH1F                  *fHistJetArea; 
  TH2F                  *fHistRhovsCent;
  TH2F                  *fHistDeltaRhovsCent;  
  TH2F                  *fHistDeltaRhoScalevsCent;  
  TH2F                  *fHistJetPtvsCent;  
  TH2F                  *fHistJetAreavsCent;  
  TH2F                  *fHistNjetvsCent;

  TH2F                 *fHistRhovsNtrack;
  TH2F                 *fHistDeltaRhovsNtrack;   
  TH2F                 *fHistDeltaRhoScalevsNtrack;   
  TH2F                 *fHistJetPtvsNtrack;
  TH2F                 *fHistJetAreavsNtrack;
  TH2F                 *fHistNjetvsNtrack;

  TClonesArray         *fArrRhos;

  Double_t              fPhiMin;
  Double_t              fPhiMax;
  Double_t              fEtaMin;
  Double_t              fEtaMax;
  Double_t              fAreaCut;
  Int_t                 fCswitch;

  AliAnalysisTaskRho(const AliAnalysisTaskRho&);             // not implemented
  AliAnalysisTaskRho& operator=(const AliAnalysisTaskRho&);  // not implemented
  
  ClassDef(AliAnalysisTaskRho, 1); // Rho task
};

#endif
Commit	Line	Data
020052e4	1	#ifndef ALIANALYSISTASKRHO_cxx
	2	#define ALIANALYSISTASKRHO_cxx
	3
	4	class TList;
	5	class TH1F;
	6	class TH2F;
	7	class TClonesArray;
	8	class TString;
	9
	10	#include "AliAnalysisTaskSE.h"
	11
	12	class AliAnalysisTaskRho : public AliAnalysisTaskSE {
	13
	14	public:
	15	AliAnalysisTaskRho();
	16	AliAnalysisTaskRho(const char *name);
	17	virtual ~AliAnalysisTaskRho() {}
	18
	19	virtual void UserCreateOutputObjects();
	20	virtual void UserExec(Option_t*);
	21	virtual void Terminate(Option_t*);
	22
	23	void SetTracksName(const char *n) { fTracksName = n ; }
	24	void SetJetsName(const char *n) { fJetsName = n ; }
	25	void SetRhosName(const char *rn) { fRhosName = rn ; }
	26	void SetJetPhi(Double_t pmin = 1.8, Double_t pmax = 2.74) { fPhiMin = pmin ; fPhiMax = pmax ; }
	27	void SetJetEta(Double_t emin = -0.3, Double_t emax = 0.3) { fEtaMin = emin ; fEtaMax = emax ; }
	28	void SetAreaCut(Double_t a = 0.0) { fAreaCut = a ; }
	29
	30	protected:
	31	virtual void Sort(vector<Double_t>& v) ;
	32	virtual Double_t GetMedian(vector<Double_t> v, int c) ;
	33	virtual Double_t GetScaleFactor(Double_t cent) ;
	34	virtual Double_t GetRhoFactor(Double_t cent) ;
	35
	36	private:
	37	TString fTracksName; // name of track collection
	38	TString fJetsName; // name of jet collection
	39	TString fRhosName; // name of Rho array output
	40	TString fClustersName; // name of clusters collection
	41
	42	TList *fOutputList; //! Output list
	43	TH1F *fHistCentrality;
	44	TH1F *fHistJetPt;
	45	TH1F *fHistJetArea;
	46	TH2F *fHistRhovsCent;
	47	TH2F *fHistDeltaRhovsCent;
	48	TH2F *fHistDeltaRhoScalevsCent;
	49	TH2F *fHistJetPtvsCent;
	50	TH2F *fHistJetAreavsCent;
	51	TH2F *fHistNjetvsCent;
	52
	53	TH2F *fHistRhovsNtrack;
	54	TH2F *fHistDeltaRhovsNtrack;
	55	TH2F *fHistDeltaRhoScalevsNtrack;
	56	TH2F *fHistJetPtvsNtrack;
	57	TH2F *fHistJetAreavsNtrack;
	58	TH2F *fHistNjetvsNtrack;
	59
	60	TClonesArray *fArrRhos;
	61
	62	Double_t fPhiMin;
	63	Double_t fPhiMax;
	64	Double_t fEtaMin;
65	Double_t fEtaMax;
66	Double_t fAreaCut;
67	Int_t fCswitch;
68
69	AliAnalysisTaskRho(const AliAnalysisTaskRho&); // not implemented
70	AliAnalysisTaskRho& operator=(const AliAnalysisTaskRho&); // not implemented
71
72	ClassDef(AliAnalysisTaskRho, 1); // Rho task
73	};
74
75	#endif