AliFMDEnergyFitterTask.h

Go to the documentation of this file.

// Histogram and fit the energy loss distributions for the FMD
// 
// Wraps AliFMDEnergyFitter 
#ifndef ALIFMDENERGYFITTERTASK_H
#define ALIFMDENERGYFITTERTASK_H

#include "AliBaseESDTask.h"
#include "AliFMDEventInspector.h"
#include "AliFMDEnergyFitter.h"
#include "AliFMDESDFixer.h"
class AliESDEvent;
class TH2D;
class TList;
class TTree;


class AliFMDEnergyFitterTask : public AliBaseESDTask
{
public:
  AliFMDEnergyFitterTask(const char* name);
  AliFMDEnergyFitterTask();
  virtual Bool_t Setup();
  virtual Bool_t Book();
  virtual void PreCorrections(const AliESDEvent* esd);
  virtual Bool_t PreData(const TAxis& vertex, const TAxis& eta);
  virtual Bool_t Event(AliESDEvent& esd);
  virtual Bool_t Finalize();
  void Print(Option_t* option="") const;
  AliFMDEventInspector& GetEventInspector() { return fEventInspector; }
  const AliFMDEventInspector& GetEventInspector() const{return fEventInspector;}
  AliFMDESDFixer& GetESDFixer() { return fESDFixer; }
  AliFMDEnergyFitter& GetEnergyFitter() { return fEnergyFitter; }
  void SetDebug(Int_t dbg);
  void SetOnlyMB(Bool_t onlyMB) { fOnlyMB = onlyMB; }
  /* @} */
  virtual TAxis* DefaultEtaAxis() const;
  virtual TAxis* DefaultVertexAxis() const;
  /* @} */
protected: 
  AliFMDEnergyFitterTask(const AliFMDEnergyFitterTask& o);
  AliFMDEnergyFitterTask& operator=(const AliFMDEnergyFitterTask& o);

  virtual const char* DefaultConfig() const { return "elossFitConfig.C"; }

  AliFMDEventInspector fEventInspector; // Algorithm
  AliFMDESDFixer       fESDFixer;       // Algorithm
  AliFMDEnergyFitter   fEnergyFitter;   // Algorithm
  Bool_t               fOnlyMB;         // Only MB flag

  ClassDef(AliFMDEnergyFitterTask,4) // Forward multiplicity class
};

#endif
// Local Variables:
//  mode: C++
// End: