AliBaseESDTask.h

Go to the documentation of this file.

//
// Base class for FMD ESD input - sub classes can use the services to
// easily setup a task to process the FMD ESD data 
//
#ifndef ALIFMDESDTASK_H
#define ALIFMDESDTASK_H

#include <AliAnalysisTaskSE.h>
class AliFMDEventInspector;
class AliCorrectionManagerBase;
class AliAODHandler;
class AliESDEvent;
class TAxis;
class TList;

class AliBaseESDTask : public AliAnalysisTaskSE
{
public:
  AliBaseESDTask();
  AliBaseESDTask(const char* name, const char* title,
     AliCorrectionManagerBase* manager);
  virtual Bool_t Connect(const char* sumFile=0, 
       const char* resFile=0)
  {
    return Connect(sumFile, resFile, false);
  }
  virtual Bool_t Connect(const char* sumFile, 
       const char* resFile,
       Bool_t      old);
  virtual Bool_t Setup() { return true; }
  virtual Bool_t Book() = 0;
  virtual void PreCorrections(const AliESDEvent* esd);
  virtual Bool_t PreData(const TAxis& vertex, const TAxis& eta);
  virtual Bool_t PreEvent() { return true; }
  virtual Bool_t Event(AliESDEvent& esd) = 0;
  virtual Bool_t PostEvent() { return true; }
  virtual Bool_t Finalize() { return true; }
  virtual void Print(Option_t* option="") const;
  virtual void SetDebug(Int_t dbg);
  virtual void SetDebugLevel(Int_t dbg) 
  { 
    AliAnalysisTaskSE::SetDebugLevel(dbg); 
    SetDebug(dbg);
  }
  void SetIPzMethod(const char* str);
  /* @} */
  // --- Configuration etc -------------------------------------------
  virtual Bool_t Configure(const char* macro="-default-");
  virtual AliFMDEventInspector& GetEventInspector() = 0;
  virtual const AliFMDEventInspector& GetEventInspector() const = 0;
  /* @} */
protected:
  AliBaseESDTask(const AliBaseESDTask& o);
  AliBaseESDTask& operator=(const AliBaseESDTask& o);
  // --- Customize ---------------------------------------------------
  virtual Bool_t IsMC() const { return false; }
  virtual TAxis* DefaultEtaAxis() const;
  virtual TAxis* DefaultVertexAxis() const;
  virtual AliCorrectionManagerBase* GetManager() const { return fCorrManager; }
  virtual AliCorrectionManagerBase* GetManager() { return fCorrManager; }

  // --- Task methods ------------------------------------------------
  void LocalInit();
  void UserCreateOutputObjects();
  void UserExec(Option_t* option);
  void Terminate(Option_t* option);
  // --- Services for derived classes --------------------------------
  virtual void CreateBranches(AliAODHandler*) {}
  virtual void MarkEventForStore() const;
  virtual Bool_t CheckCorrections(UInt_t what) const;
  virtual Bool_t ReadCorrections(const TAxis*& pe, 
         const TAxis*& pv,
         Bool_t mc=false,
         Bool_t sat=false);
  virtual AliESDEvent* GetESDEvent();
  virtual Bool_t StoreTrainName(Int_t no);
  virtual const char* DefaultConfig() const 
  {
    return "ForwardAODConfig.C";
  }

  // --- Members -----------------------------------------------------
  Bool_t fFirstEvent;        // Wheter we're waiting for the first event
  TList* fList;              // Output list 
  TList* fResults;           // Results list 
  UInt_t fNeededCorrections; // Set this to bit-mask of corrections we need
  UInt_t fExtraCorrections;  // Set this to bit-mask of corrections we'd like
  Bool_t fCloneList;         // Result list is a clone of sum list
private:
  AliCorrectionManagerBase* fCorrManager; // Pointer to corrections manager

  ClassDef(AliBaseESDTask,1);
};
inline void AliBaseESDTask::PreCorrections(const AliESDEvent*) {}
#endif
// Local Variables:
//   mode: C++
// End: