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

#ifndef ALIANALYSISTASKRHOMASSBASE_H
#define ALIANALYSISTASKRHOMASSBASE_H

// $Id$

class TString;
class TF1;
class TH1F;
class TH2F;
class AliRhoParameter;

#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskRhoMassBase : public AliAnalysisTaskEmcalJet {
 public:
  AliAnalysisTaskRhoMassBase();
  AliAnalysisTaskRhoMassBase(const char *name, Bool_t histo=kFALSE);
  virtual ~AliAnalysisTaskRhoMassBase() {}

  void                   UserCreateOutputObjects();

  void                   SetOutRhoMassName(const char *name)                   { fOutRhoMassName           = name ; 
                                                                                 fOutRhoMassScaledName     = Form("%s_Scaled",name) ; }
  void                   SetCompareRhoMassName(const char *name)               { fCompareRhoMassName       = name ;                   }
  void                   SetCompareRhoMassScaledName(const char *name)         { fCompareRhoMassScaledName = name ;                   }
  void                   SetScaleFunction(TF1* sf)                             { fScaleFunction        = sf   ;                       }
  void                   SetRhoMassFunction(TF1* rf)                           { fRhoMassFunction      = rf   ;                       }
  void                   SetAttachToEvent(Bool_t a)                            { fAttachToEvent        = a    ;                       }

  const TString&         GetOutRhoMassName() const                             { return fOutRhoMassName;                              }
  const TString&         GetOutRhoMassScaledName() const                       { return fOutRhoMassScaledName;                        } 

 protected:
  void                   ExecOnce();
  Bool_t                 Run();
  Bool_t                 FillHistograms();

  virtual Double_t       GetRhoMassFactor(Double_t cent);
  virtual Double_t       GetScaleFactor(Double_t cent);

  TString                fOutRhoMassName;                // name of output rho mass object
  TString                fOutRhoMassScaledName;          // name of output scaled rho mass object
  TString                fCompareRhoMassName;            // name of rho mass object to compare
  TString                fCompareRhoMassScaledName;      // name of scaled rho mass object to compare
  TF1                   *fRhoMassFunction;               // pre-computed rho mass as a function of centrality
  TF1                   *fScaleFunction;                 // pre-computed scale factor as a function of centrality
  Bool_t                 fAttachToEvent;                 // whether or not attach rho mass to the event objects list

  AliRhoParameter       *fOutRhoMass;                    //!output rho object
  AliRhoParameter       *fOutRhoMassScaled;              //!output scaled rho object
  AliRhoParameter       *fCompareRhoMass;                //!rho object to compare
  AliRhoParameter       *fCompareRhoMassScaled;          //!scaled rho object to compare

  TH2F                  *fHistJetMassvsCent;             //!jet mass vs. centrality
  TH2F                  *fHistRhoMassvsCent;             //!rho mass vs. centrality
  TH2F                  *fHistRhoMassScaledvsCent;       //!rho mass scaled vs. centrality
  TH2F                  *fHistDeltaRhoMassvsCent;        //!delta rho mass vs. centrality
  TH2F                  *fHistDeltaRhoMassScalevsCent;   //!delta rho mass scaled vs. centrality

  TH2F                  *fHistRhoMassvsNtrack;           //!rho mass vs. no. of tracks
  TH2F                  *fHistRhoMassScaledvsNtrack;     //!rho mass scaled vs. no. of tracks
  TH2F                  *fHistDeltaRhoMassvsNtrack;      //!delta rho mass vs. no. of tracks
  TH2F                  *fHistDeltaRhoMassScalevsNtrack; //!delta rho mass scaled vs. no. of tracks
 
  TH2F                  *fHistRhoMassvsNcluster;         //!rho mass vs. no. of clusters
  TH2F                  *fHistRhoMassScaledvsNcluster;   //!rho mass scaled vs. no. of clusters

  TH2F                  *fHistGammaVsNtrack;             //!Gamma(<E>/<M>) vs Ntrack

  AliAnalysisTaskRhoMassBase(const AliAnalysisTaskRhoMassBase&);             // not implemented
  AliAnalysisTaskRhoMassBase& operator=(const AliAnalysisTaskRhoMassBase&);  // not implemented
  
  ClassDef(AliAnalysisTaskRhoMassBase, 2); // Rho mass base task
};
#endif
Commit	Line	Data
4d3b366f	1	#ifndef ALIANALYSISTASKRHOMASSBASE_H
	2	#define ALIANALYSISTASKRHOMASSBASE_H
	3
	4	// $Id$
	5
	6	class TString;
	7	class TF1;
	8	class TH1F;
	9	class TH2F;
	10	class AliRhoParameter;
	11
	12	#include "AliAnalysisTaskEmcalJet.h"
	13
	14	class AliAnalysisTaskRhoMassBase : public AliAnalysisTaskEmcalJet {
	15	public:
	16	AliAnalysisTaskRhoMassBase();
	17	AliAnalysisTaskRhoMassBase(const char *name, Bool_t histo=kFALSE);
	18	virtual ~AliAnalysisTaskRhoMassBase() {}
	19
	20	void UserCreateOutputObjects();
	21
	22	void SetOutRhoMassName(const char *name) { fOutRhoMassName = name ;
	23	fOutRhoMassScaledName = Form("%s_Scaled",name) ; }
	24	void SetCompareRhoMassName(const char *name) { fCompareRhoMassName = name ; }
	25	void SetCompareRhoMassScaledName(const char *name) { fCompareRhoMassScaledName = name ; }
	26	void SetScaleFunction(TF1* sf) { fScaleFunction = sf ; }
	27	void SetRhoMassFunction(TF1* rf) { fRhoMassFunction = rf ; }
	28	void SetAttachToEvent(Bool_t a) { fAttachToEvent = a ; }
	29
	30	const TString& GetOutRhoMassName() const { return fOutRhoMassName; }
	31	const TString& GetOutRhoMassScaledName() const { return fOutRhoMassScaledName; }
	32
	33	protected:
	34	void ExecOnce();
	35	Bool_t Run();
	36	Bool_t FillHistograms();
	37
	38	virtual Double_t GetRhoMassFactor(Double_t cent);
	39	virtual Double_t GetScaleFactor(Double_t cent);
	40
	41	TString fOutRhoMassName; // name of output rho mass object
	42	TString fOutRhoMassScaledName; // name of output scaled rho mass object
	43	TString fCompareRhoMassName; // name of rho mass object to compare
	44	TString fCompareRhoMassScaledName; // name of scaled rho mass object to compare
	45	TF1 *fRhoMassFunction; // pre-computed rho mass as a function of centrality
	46	TF1 *fScaleFunction; // pre-computed scale factor as a function of centrality
	47	Bool_t fAttachToEvent; // whether or not attach rho mass to the event objects list
	48
	49	AliRhoParameter *fOutRhoMass; //!output rho object
	50	AliRhoParameter *fOutRhoMassScaled; //!output scaled rho object
	51	AliRhoParameter *fCompareRhoMass; //!rho object to compare
	52	AliRhoParameter *fCompareRhoMassScaled; //!scaled rho object to compare
	53
	54	TH2F *fHistJetMassvsCent; //!jet mass vs. centrality
4d3b366f	55	TH2F *fHistRhoMassvsCent; //!rho mass vs. centrality
	56	TH2F *fHistRhoMassScaledvsCent; //!rho mass scaled vs. centrality
	57	TH2F *fHistDeltaRhoMassvsCent; //!delta rho mass vs. centrality
	58	TH2F *fHistDeltaRhoMassScalevsCent; //!delta rho mass scaled vs. centrality
	59
	60	TH2F *fHistRhoMassvsNtrack; //!rho mass vs. no. of tracks
	61	TH2F *fHistRhoMassScaledvsNtrack; //!rho mass scaled vs. no. of tracks
	62	TH2F *fHistDeltaRhoMassvsNtrack; //!delta rho mass vs. no. of tracks
	63	TH2F *fHistDeltaRhoMassScalevsNtrack; //!delta rho mass scaled vs. no. of tracks
	64
	65	TH2F *fHistRhoMassvsNcluster; //!rho mass vs. no. of clusters
	66	TH2F *fHistRhoMassScaledvsNcluster; //!rho mass scaled vs. no. of clusters
	67
	68	TH2F *fHistGammaVsNtrack; //!Gamma(<E>/<M>) vs Ntrack
	69
	70	AliAnalysisTaskRhoMassBase(const AliAnalysisTaskRhoMassBase&); // not implemented
	71	AliAnalysisTaskRhoMassBase& operator=(const AliAnalysisTaskRhoMassBase&); // not implemented
	72
713c5683	73	ClassDef(AliAnalysisTaskRhoMassBase, 2); // Rho mass base task
4d3b366f	74	};
4d3b366f	75	#endif