AliAnalysisTaskRhoBaseDev.h

Go to the documentation of this file.

/* Copyright(c) 1998-2017, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

#ifndef ALIANALYSISTASKRHOBASEDEV_H
#define ALIANALYSISTASKRHOBASEDEV_H

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

#include <map>
#include <string>

#include "AliAnalysisTaskJetUE.h"


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

  void                   UserCreateOutputObjects();

  void                   SetOutRhoName(const char *name)                       { fOutRhoName           = name    ;
                                                                                 fOutRhoScaledName     = Form("%s_Scaled",name);     }
  void                   SetScaleFunction(TF1* sf)                             { fScaleFunction        = sf      ;                   }
  void                   SetRhoFunction(TF1* rf)                               { fRhoFunction          = rf      ;                   }
  TF1*                   LoadRhoFunction(const char* path, const char* name);
  void                   SetAttachToEvent(Bool_t a)                            { fAttachToEvent        = a       ;                   }

  const char*            GetOutRhoName() const                                 { return fOutRhoName.Data()       ;                   }
  const char*            GetOutRhoScaledName() const                           { return fOutRhoScaledName.Data() ;                   }

  static AliAnalysisTaskRhoBaseDev* AddTaskRhoBaseDev(
     TString        nTracks                        = "usedefault",
     TString        nClusters                      = "usedefault",
     TString        nRho                           = "Rho",
     Double_t       jetradius                      = 0.2,
     UInt_t         acceptance                     = AliEmcalJet::kTPCfid,
     AliJetContainer::EJetType_t jetType           = AliJetContainer::kChargedJet,
     AliJetContainer::ERecoScheme_t rscheme        = AliJetContainer::pt_scheme,
     Bool_t         histo                          = kTRUE,
     TString        suffix                         = ""
  );

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

  virtual Bool_t                      VerifyContainers() { return kTRUE; }

  virtual void                        CalculateRho();

  virtual Double_t                    GetRhoFactor(Double_t cent);
  virtual Double_t                    GetScaleFactor(Double_t cent);

  TString                             fOutRhoName;                    
  TString                             fOutRhoScaledName;              
  TF1                                *fRhoFunction;                   
  TF1                                *fScaleFunction;                 
  Bool_t                              fAttachToEvent;                 

  Bool_t                              fTaskConfigured;                

  // Exported background density
  AliRhoParameter                    *fOutRho;                        
  AliRhoParameter                    *fOutRhoScaled;                  

  // Histograms
  TH2                                *fHistRhoVsCent;                 

  std::map<std::string, TH2*>         fHistRhoVsLeadJetPt;            
  std::map<std::string, TH2*>         fHistLeadJetPtVsCent;           
  std::map<std::string, TH2*>         fHistLeadJetPtDensityVsCent;    
  std::map<std::string, TH2*>         fHistTotJetAreaVsCent;          
  std::map<std::string, TH2*>         fHistLeadJetNconstVsCent;       
  std::map<std::string, TH2**>        fHistLeadJetNconstVsPt;         
  std::map<std::string, TH2*>         fHistNjetVsCent;                
  std::map<std::string, TH2*>         fHistNjetVsNtrack;              

  TH2                                *fHistRhoVsLeadTrackPt;          
  TH2                                *fHistRhoVsNtrack;               
  TH2                                *fHistLeadTrackPtVsCent;         
  TH2                                *fHistNtrackVsCent;              

  TH2                                *fHistRhoVsLeadClusterE;         
  TH2                                *fHistRhoVsNcluster;             
  TH2                                *fHistLeadClusterEVsCent;        
  TH2                                *fHistNclusterVsCent;            

  TH2                                *fHistRhoScaledVsCent;           
  TH2                                *fHistRhoScaledVsNtrack;         
  TH2                                *fHistRhoScaledVsNcluster;       

  AliAnalysisTaskRhoBaseDev(const AliAnalysisTaskRhoBaseDev&);             // not implemented
  AliAnalysisTaskRhoBaseDev& operator=(const AliAnalysisTaskRhoBaseDev&);  // not implemented
  
  ClassDef(AliAnalysisTaskRhoBaseDev, 2);
};
#endif