PWGGA/EMCALJetTasks/AliAnalysisTaskRho.h

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