AliBaseAODTask.h

Go to the documentation of this file.

#ifndef ALIBASEAODTASK_H
#define ALIBASEAODTASK_H
#include <AliAnalysisTaskSE.h>
#include <TAxis.h>
class AliAODEvent;
class AliAODForwardMult;
class AliAODCentralMult;
class AliAODMultEventClass;
class TList;

class AliBaseAODTask : public AliAnalysisTaskSE
{
public:
  AliBaseAODTask(); 
  AliBaseAODTask(const char* name,
     const char* title);
  virtual ~AliBaseAODTask() {} 
  virtual Bool_t Configure(const char* macro="-default-");
  void SetIPzAxis(Int_t n, Double_t min, Double_t max)
  {
    SetAxis(fIPzAxis, n, min, max);
  }
  void SetIPzAxis(Int_t n, Double_t max)
  {
    SetAxis(fIPzAxis, n, max);
  }
  void SetIPzAxis(const TString& spec)
  {
    SetAxis(fIPzAxis, spec);
  }
  void SetCentralityAxis(UShort_t n, Short_t* bins);
  void SetCentralityAxis(UShort_t n, Double_t* bins);
  void SetCentralityAxis(Short_t low, Short_t high);
  void SetCentralityAxis(const char* bins);
  void SetAbsMinCent(Double_t x=-1) { fAbsMinCent = x; }
  void SetTriggerMask(UInt_t mask);
  void SetTriggerMask(const char* mask);
  void SetFilterMask(UInt_t mask);
  void SetFilterMask(const char* mask);
  /* @} */
  virtual Bool_t Connect(const char* sumFile=0, const char* resFile=0);
  virtual Bool_t Book() = 0;
  virtual Bool_t PreData() { return true; }
  virtual Bool_t PreEvent() { return true; } 
  virtual Bool_t Event(AliAODEvent& aod) = 0;
  virtual Bool_t PostEvent() { return true; } 
  virtual Bool_t Finalize() = 0;
  /* @} */
     
  void Print(Option_t* option="") const;
  /* @} */  
protected:
  AliBaseAODTask(const AliBaseAODTask& o); 
  AliBaseAODTask& operator=(const AliBaseAODTask& o); 
  virtual void   Init() {}
  virtual void UserCreateOutputObjects();
  virtual void UserExec(Option_t* option);
  virtual void Terminate(Option_t* option);
  /* @} */
  static void FixAxis(TAxis& axis, const char* title=0);
  static void SetAxis(TAxis& axis, Int_t n, Double_t* borders);
  static void SetAxis(TAxis& axis, const TString& spec, const char* sep=":,");
  static void SetAxis(TAxis& axis, Int_t n, Double_t l, Double_t h);
  static void SetAxis(TAxis& axis, Int_t n, Double_t m);
  static void PrintAxis(const TAxis& axis, Int_t nSig=2, const char* alt=0);
  /* @} */
  AliAODForwardMult* GetForward(const AliAODEvent& aod, 
        Bool_t mc=false, 
        Bool_t verb=true);
  AliAODMultEventClass* GetMultClass(const AliAODEvent& aod, 
              Bool_t verb=true);
  AliAODCentralMult* GetCentral(const AliAODEvent& aod, 
        Bool_t mc=false, 
        Bool_t verb=true);
  TH2D* GetPrimary(const AliAODEvent& aod);
  virtual void StoreInformation(AliAODForwardMult& forward);
  virtual Bool_t StoreTrainName(Int_t no);
  virtual Bool_t CheckEvent(const AliAODForwardMult& forward);
  Bool_t HasCentrality() const 
  { 
    return (fCentAxis.GetNbins() >= 1 && 
      fCentAxis.GetXbins() && 
      fCentAxis.GetXbins()->GetArray()); 
  }
  virtual Double_t GetCentrality(AliAODEvent&       event,
         AliAODForwardMult* forward,
         Int_t&             qual);
  virtual Double_t GetCentrality(AliAODEvent&       event,
         AliAODForwardMult* forward);
  virtual Double_t GetIpZ(AliAODEvent& event,
        AliAODForwardMult* forward);
  virtual Bool_t GetIpXY(AliAODEvent& aod, Double_t& x, Double_t& y);
  /* @} */
  virtual const char* DefaultConfig() const { return "dNdetaConfig.C"; }

  UInt_t   fTriggerMask;   
  UInt_t   fFilterMask;    
  TAxis    fCentAxis;      
  Double_t   fAbsMinCent;
  TAxis    fIPzAxis;       
  TH1I*    fTriggers;      
  TH1I*    fEventStatus;   
  TH1D*    fVertex;        
  TH1D*    fCent;          
  TH1D*    fAccVertex;     
  TH2D*    fAccVertexXY;   
  TH1D*    fAccCent;       
  Bool_t   fFirstEvent;    
  Bool_t   fCloneList;     
  TList*   fSums;          
  TList*   fResults;       

  ClassDef(AliBaseAODTask,3)
};
#endif
//
// Local Variables:
//  mode: C++
// End:
//