AliAnalysisTaskCounter.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKCOUNTER_H
#define ALIANALYSISTASKCOUNTER_H

//_________________________________________________________________________
//_________________________________________________________________________


class TH1F;
class TList;
class AliESDtrackCuts;
class AliGenPythiaEventHeader;
//class AliTriggerAnalysis;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskCounter : public AliAnalysisTaskSE { 
  
 public:
  
  AliAnalysisTaskCounter();  
  
  AliAnalysisTaskCounter(const char *name);  
  
  virtual       ~AliAnalysisTaskCounter();
  
  virtual void   UserCreateOutputObjects();
  
  virtual void   UserExec(Option_t *option);
  
  virtual void   FinishTaskOutput();
  
  virtual Bool_t Notify();
  
  static  Bool_t PythiaInfoFromFile(TString currFile, Float_t & xsec, Float_t & trials) ;

  void           SetTrackMultiplicityEtaCut(Float_t eta) { fTrackMultEtaCut   = eta    ; }
  void           SetZVertexCut(Float_t vcut)             { fZVertexCut        = vcut   ; }
  
  void           AcceptFastCluster()                     { fAcceptFastCluster = kTRUE  ; }
  void           RejectFastCluster()                     { fAcceptFastCluster = kFALSE ; }
  Bool_t         IsFastClusterAccepted()       const     { return fAcceptFastCluster   ; }
  
  Bool_t         CheckForPrimaryVertex() ;

  void           SwitchOnMCCrossSectionCalculation()     { fCheckMCCrossSection = kTRUE  ; }
  void           SwitchOffMCCrossSectionCalculation()    { fCheckMCCrossSection = kFALSE ; }
  
  void           SwitchOnAliCentrality ()                { fUseAliCentrality    = kTRUE  ; }
  void           SwitchOffAliCentrality()                { fUseAliCentrality    = kFALSE ; }
  
  void           SetCentralityClass(TString name)        { fCentralityClass   = name     ; }
  TString        GetCentralityClass()              const { return fCentralityClass       ; }

 private:
  
  Bool_t               fAcceptFastCluster;   
  Float_t              fZVertexCut;          
  Float_t              fTrackMultEtaCut;     
  Float_t              fAvgTrials;           
  TList*               fOutputContainer;     
  AliESDtrackCuts    * fESDtrackCuts;        
//AliTriggerAnalysis * fTriggerAnalysis;     ///< Trigger algorithm.
  TString              fCurrFileName;        
  Bool_t               fCheckMCCrossSection; 
  Bool_t               fUseAliCentrality;    
  TString              fCentralityClass;     
  
  //
  // Histograms
  //
  TH1I *  fhNEvents;         
  TH1F *  fhXVertex;         
  TH1F *  fhYVertex;         
  TH1F *  fhZVertex;         
  TH1F *  fhXGoodVertex;     
  TH1F *  fhYGoodVertex;     
  TH1F *  fhZGoodVertex;     
  TH1F *  fhCentrality;      
  TH1F *  fhEventPlaneAngle; 

  TH1F *  fh1Xsec ;          
  TH1F *  fh1Trials ;        
  
  AliAnalysisTaskCounter(           const AliAnalysisTaskCounter&); 
  
  AliAnalysisTaskCounter& operator=(const AliAnalysisTaskCounter&); 
  
  ClassDef(AliAnalysisTaskCounter, 7) ;
  
};

#endif //ALIANALYSISTASKCOUNTER_H