AliAnalysisTaskRhoMassBase.h

Go to the documentation of this file.

#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    ;                       }
  void                   SetSmallSystem(Bool_t setter = kTRUE)                 {fIsPbPb = !setter; }


  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
  Bool_t                 fIsPbPb;                        // different histogram ranges for pp/pPb and PbPb
  
  AliRhoParameter       *fOutRhoMass;                    
  AliRhoParameter       *fOutRhoMassScaled;              
  AliRhoParameter       *fCompareRhoMass;                
  AliRhoParameter       *fCompareRhoMassScaled;          

  TH2F                  *fHistJetMassvsCent;             
  TH2F                  *fHistRhoMassvsCent;             
  TH2F                  *fHistRhoMassScaledvsCent;       
  TH2F                  *fHistDeltaRhoMassvsCent;        
  TH2F                  *fHistDeltaRhoMassScalevsCent;   

  TH2F                  *fHistRhoMassvsNtrack;           
  TH2F                  *fHistRhoMassScaledvsNtrack;     
  TH2F                  *fHistDeltaRhoMassvsNtrack;      
  TH2F                  *fHistDeltaRhoMassScalevsNtrack; 
 
  TH2F                  *fHistRhoMassvsNcluster;         
  TH2F                  *fHistRhoMassScaledvsNcluster;   

  TH2F                  *fHistGammaVsNtrack;             

  AliAnalysisTaskRhoMassBase(const AliAnalysisTaskRhoMassBase&);             // not implemented
  AliAnalysisTaskRhoMassBase& operator=(const AliAnalysisTaskRhoMassBase&);  // not implemented
  
  ClassDef(AliAnalysisTaskRhoMassBase, 2); // Rho mass base task
};
#endif