PWG1/AliComparisonTask.h

   1 #ifndef ALICOMPARISONRESTASK_H
   2 #define ALICOMPARISONRESTASK_H
   3
   4 //------------------------------------------------------------------------------
   5 // Class to compare properties of reconstructed and MC particle tracks.
   6 //
   7 // Author: J.Otwinowski 04/02/2008
   8 //------------------------------------------------------------------------------
   9
  10 class AliComparisonRes;
  11 class AliComparisonEff;
  12 class AliComparisonDEdx;
  13 class AliComparisonDCA;
  14 class AliMagFMaps;
  15 class TList;
  16
  17 #include "AliAnalysisTask.h"
  18
  19 class AliComparisonTask : public AliAnalysisTask {
  20  public:
  21   AliComparisonTask(const char *name = "AliComparisonTask");
  22   virtual ~AliComparisonTask();
  23
  24   virtual void   ConnectInputData(Option_t *);
  25   virtual void   CreateOutputObjects();
  26   virtual void   Exec(Option_t *option);
  27   virtual void   Terminate(Option_t *);
  28
  29   // Read TTree entry (event by event)
  30   Bool_t  ReadEntry(Int_t evt);
  31
  32   // Set comparison objects
  33   void SetAliComparisonRes(AliComparisonRes* comp) {fCompRes = comp;}
  34   void SetAliComparisonEff(AliComparisonEff* comp) {fCompEff = comp;}
  35   void SetAliComparisonDEdx(AliComparisonDEdx* comp) {fCompDEdx = comp;}
  36   void SetAliComparisonDCA(AliComparisonDCA* comp) {fCompDCA = comp;}
  37
  38   void SetMagField(Int_t mag = 2) {fMagField = mag;}
  39   void SetGeometry(char* geom = "/d/alice12/jacek/sim/v4-10-Release/pp/0/geometry.root")  {fGeom = geom;}
  40
  41  private:
  42   TTree* fTree;                   //! input tree
  43   AliMCInfo *fInfoMC;             //! AliMCInfo object
  44   AliESDRecInfo *fInfoRC;         //! AliESDRecInfo object
  45   AliComparisonRes* fCompRes;     // TPC resolution comparison object
  46   AliComparisonEff* fCompEff;     // TPC efficiency comparison object
  47   AliComparisonDEdx* fCompDEdx;   // TPC DEdx comparison object
  48   AliComparisonDCA* fCompDCA;     // TPC DCA comparison object
  49
  50   TList* fOutput;                 //! list send on output slot 0
  51   static Int_t evtNumber;         //! event number
  52   Int_t  fMagField;               //! mag. field (0 - 0.2 T, 1 - 0.4 T, 2 - 0.5 T)
  53   AliMagFMaps *fMagFMap;          //! mag. field map
  54   const char *fGeom;              //! ROOT file with detector geometry
  55
  56   AliComparisonTask(const AliComparisonTask&); // not implemented
  57   AliComparisonTask& operator=(const AliComparisonTask&); // not implemented
  58
  59   ClassDef(AliComparisonTask, 1); // example of analysis
  60 };
  61
  62 #endif