AliAnaWeights.h

Go to the documentation of this file.

#ifndef ALIANAWEIGHTS_H
#define ALIANAWEIGHTS_H

//_________________________________________________________________________
//_________________________________________________________________________

class TH1F;
class TList;
class AliGenPythiaEventHeader;
#include <TF1.h>

#include <TObject.h>

class AliAnaWeights : public TObject {
  
 public:
  
  AliAnaWeights();  
    
  virtual         ~AliAnaWeights() ;
  
  //
  // General 
  //  
  TList *          GetCreateOutputHistograms();
  
  Int_t            GetDebug()                        const { return fDebug                 ; }
  
  void             SetDebug(Int_t d)                       { fDebug = d                    ; }
  
  //
  // Centrality weights
  //
  void             InitCentralityWeightsHistogram(Int_t nbins = 100, Int_t minCen = 0, Int_t maxCen = 100) ;
    
  TH1F *           GetCentralityWeightsHistogram() ;
    
  void             SetCentralityWeightsHistogram(TH1F* h)  { if ( fhCentralityWeight ) delete fhCentralityWeight ; fhCentralityWeight = h ; }
    
  void             SetCentrality(Float_t cen)              { fCentrality = cen             ; }
   
  Bool_t           IsCentralityWeightOn()            const { return fUseCentralityWeight   ; }
    
  void             SwitchOnCentralityWeight()              { fUseCentralityWeight = kTRUE  ; InitCentralityWeightsHistogram() ;  }
    
  void             SwitchOffCentralityWeight()             { fUseCentralityWeight = kFALSE ; }
    
  //
  // Pythia pT hard weights
  //
  virtual Double_t GetPythiaCrossSection() ;

  static  Bool_t   GetPythiaInfoFromFile(TString currFile, Float_t & xsec, Float_t & trials) ;
    
  virtual Double_t GetWeight() ;
    
  Bool_t           IsWeightSettingOn()               const { return ( fUseCentralityWeight || fCheckMCCrossSection ) ; }
  
  Bool_t           IsMCCrossSectionCalculationOn()   const { return fCheckMCCrossSection   ; }
  Bool_t           IsMCCrossSectionJustHistoFillOn() const { return fJustFillCrossSecHist  ; }

  void             SetPythiaEventHeader(AliGenPythiaEventHeader* py) { fPyEventHeader = py    ; }
  
  void             SwitchOnMCCrossSectionCalculation()     { fCheckMCCrossSection = kTRUE  ; }
  void             SwitchOffMCCrossSectionCalculation()    { fCheckMCCrossSection = kFALSE ; }

  void             SwitchOnMCCrossSectionHistoFill()       { fCheckMCCrossSection = kTRUE  ;  fJustFillCrossSecHist = kTRUE ; }

  void             SwitchOnMCCrossSectionFromEventHeader() { fCheckPythiaEventHeader = kTRUE  ; }
  void             SwitchOffMCCrossSectionFromEventHeader(){ fCheckPythiaEventHeader = kFALSE ; }

  //
  // Pt weights
  //
  Double_t         GetParticlePtWeight ( Float_t pt, Int_t pdg, TString genName, Int_t igen ) const ;
  
  void             SetEtaFunction(TF1* fun)                { if ( fEtaFunction ) delete fEtaFunction ; fEtaFunction = fun ; }
  void             SetPi0Function(TF1* fun)                { if ( fPi0Function ) delete fPi0Function ; fPi0Function = fun ; }
  
  void             SwitchOnMCParticlePtWeights ()          { fDoMCParticlePtWeights = kTRUE  ; }
  void             SwitchOffMCParticlePtWeights()          { fDoMCParticlePtWeights = kFALSE ; }

 private:
    
  Int_t            fDebug ;               
  
  //
  // Centrality weights
  //

  TH1F *           fhCentralityWeight ;   
  
  Float_t          fCentrality ;          
    
  Bool_t           fUseCentralityWeight ; 
    
  //
  // MC weights for particles
  //
    
  Bool_t           fDoMCParticlePtWeights;
  
  TF1 *            fEtaFunction;          
  
  TF1 *            fPi0Function;          
  
  //
  // MC weights, pT hard pythia
  //
  
  Double_t         fMCWeight ;            

  TString          fCurrFileName ;        

  Bool_t           fCheckMCCrossSection ; 
  
  Bool_t           fJustFillCrossSecHist; 
  
  TH1F *           fhXsec   ;             
   
  TH1F *           fhTrials ;             
    
  AliGenPythiaEventHeader * fPyEventHeader ; 
  
  Bool_t           fCheckPythiaEventHeader ; 
  
  AliAnaWeights(           const AliAnaWeights&); 
  
  AliAnaWeights& operator=(const AliAnaWeights&); 
  
  ClassDef(AliAnaWeights, 3) ;
  
} ;

#endif //ALIANAWEIGHTS_H