Adding abstract class for comparison components

[u/mrichter/AliRoot.git] / PWG1 / AliComparisonTask.h

#ifndef ALICOMPARISONRESTASK_H
#define ALICOMPARISONRESTASK_H

//------------------------------------------------------------------------------
// Class to compare properties of reconstructed and MC particle tracks. 
// 
// Author: J.Otwinowski 04/02/2008 
//------------------------------------------------------------------------------

class AliComparisonObject;
class AliMagFMaps;
class TList;

#include "AliAnalysisTask.h"

class AliComparisonTask : public AliAnalysisTask {
 public:
  AliComparisonTask(const char *name = "AliComparisonTask");
  virtual ~AliComparisonTask();
  
  virtual void   ConnectInputData(Option_t *);
  virtual void   CreateOutputObjects();
  virtual void   Exec(Option_t *option);
  virtual void   Terminate(Option_t *);

  // Read TTree entry (event by event)
  Bool_t  ReadEntry(Int_t evt);

  // Set comparison objects
  Bool_t AddComparisonObject(AliComparisonObject* comp);
  void SetMagField(Int_t mag = 2) {fMagField = mag;}

 private:
  TTree* fTree;                   //! input tree
  AliMCInfo *fInfoMC;             //! AliMCInfo object
  AliESDRecInfo *fInfoRC;         //! AliESDRecInfo object

  TList* fOutput;                 //! list send on output slot 0
  static Int_t evtNumber;         //! event number
  Int_t  fMagField;               //! mag. field (0 - 0.2 T, 1 - 0.4 T, 2 - 0.5 T) 
  AliMagFMaps *fMagFMap;          //! mag. field map 
  TIterator *pitList;             //! iterator over the output objetcs  
  TList *fCompList;               // list of comparison objects

  AliComparisonTask(const AliComparisonTask&); // not implemented
  AliComparisonTask& operator=(const AliComparisonTask&); // not implemented
  
  ClassDef(AliComparisonTask, 1); // example of analysis
};

#endif