PWG1/TPC/AliPerformanceObject.h

   1 #ifndef ALIPERFORMANCEOBJECT_H\r
   2 #define ALIPERFORMANCEOBJECT_H\r
   3 \r
   4 //------------------------------------------------------------------------------\r
   5 // Base class to keep information from comparison of \r
   6 // reconstructed and MC particle tracks.   \r
   7 // \r
   8 // Author: J.Otwinowski 04/14/2008 \r
   9 // Changes by M.Knichel 15/10/2010\r
  10 //------------------------------------------------------------------------------\r
  11 \r
  12 #include "TNamed.h"\r
  13 #include "TFolder.h"\r
  14 #include "THnSparse.h"\r
  15 \r
  16 class TTree;\r
  17 class AliMCEvent;\r
  18 class AliESDEvent;\r
  19 class AliRecInfoCuts;\r
  20 class AliMCInfoCuts;\r
  21 class AliESDfriend;\r
  22 class AliESDVertex;\r
  23 \r
  24 class AliPerformanceObject : public TNamed {\r
  25 public :\r
  26   AliPerformanceObject(); \r
  27   AliPerformanceObject(const char* name="AliPerformanceObject", const char* title="AliPerformanceObject", Int_t run=-1); \r
  28   virtual ~AliPerformanceObject();\r
  29 \r
  30   // Init data members\r
  31   // call once before event loop\r
  32   virtual void Init() = 0;\r
  33   \r
  34   // init for high multiplicity (PbPb) \r
  35   // to be called instead of Init()\r
  36   virtual void InitHighMult();\r
  37   \r
  38   // Execute analysis\r
  39   // call in the event loop \r
  40   virtual void Exec(AliMCEvent* const infoMC=0, AliESDEvent* const infoRC=0, AliESDfriend* const infoFriend=0, const Bool_t bUseMC=kFALSE, const Bool_t bUseESDfriend=kFALSE) = 0;\r
  41 \r
  42   // Merge output objects (needed by PROOF) \r
  43   virtual Long64_t Merge(TCollection* const list=0) = 0;\r
  44 \r
  45   // Analyse output histograms\r
  46   virtual void Analyse() = 0;\r
  47 \r
  48   // Get output folder for analysed histograms\r
  49   virtual TFolder* GetAnalysisFolder() const = 0;\r
  50   \r
  51   // create a summary stored in a ttree \r
  52   // has to be implented\r
  53   // virtual TTree* CreateSummary() = 0;\r
  54   virtual TTree* CreateSummary() { return 0; }\r
  55 \r
  56   // \r
  57   virtual void SetAliRecInfoCuts(AliRecInfoCuts* const cuts=0) = 0;\r
  58   virtual void SetAliMCInfoCuts(AliMCInfoCuts* const cuts=0) = 0; \r
  59 \r
  60   // set and get analysisMode\r
  61   void SetAnalysisMode(const Int_t analysisMode=0) {fAnalysisMode = analysisMode;} \r
  62   Int_t GetAnalysisMode() const {return fAnalysisMode;}\r
  63 \r
  64   // set and get hpt generator \r
  65   void SetHptGenerator(const Bool_t hptGenerator=kFALSE) {fHptGenerator = hptGenerator;}\r
  66   Bool_t IsHptGenerator() const {return fHptGenerator;}\r
  67 \r
  68   // draw all histograms from the folder\r
  69   void PrintHisto(Bool_t logz = kTRUE, Char_t * outFileName = "PerformanceQAHisto.ps"); \r
  70 \r
  71   // create log axis \r
  72   Double_t *CreateLogAxis(Int_t nbins, Double_t xmin, Double_t xmax); \r
  73 \r
  74   // trigger class selection\r
  75   void SetTriggerClass(const Char_t *triggerClass) { fTriggerClass = triggerClass; }\r
  76   const Char_t* GetTriggerClass() const { return fTriggerClass; }\r
  77 \r
  78   // use track vertex\r
  79   void SetUseTrackVertex(Bool_t trackVtx = kTRUE) { fUseTrackVertex = trackVtx; }\r
  80   Bool_t IsUseTrackVertex() { return fUseTrackVertex; }\r
  81   \r
  82   Bool_t IsHighMultiplicity() { return fHighMultiplicity; }  \r
  83   \r
  84   void SetRunNumber(Int_t run) { fRunNumber = run; }\r
  85   Int_t GetRunNumber() const { return fRunNumber; }\r
  86 \r
  87   // use kink daughters\r
  88   void SetUseKinkDaughters(Bool_t kinkDaughters = kTRUE) { fUseKinkDaughters = kinkDaughters; }\r
  89   Bool_t IsUseKinkDaughters() { return fUseKinkDaughters; }\r
  90 \r
  91 protected: \r
  92 \r
  93   void AddProjection(TObjArray* aFolderObj, TString nameSparse, THnSparse *hSparse, Int_t xDim, TString* selString = 0);\r
  94   void AddProjection(TObjArray* aFolderObj, TString nameSparse, THnSparse *hSparse, Int_t xDim, Int_t yDim, TString* selString = 0);\r
  95   void AddProjection(TObjArray* aFolderObj, TString nameSparse, THnSparse *hSparse, Int_t xDim, Int_t yDim, Int_t zDim, TString* selString = 0);\r
  96 \r
  97   // analysis mode\r
  98   Int_t fAnalysisMode;  // 0-TPC, 1-TPCITS, 2-Constrained, 3-TPC inner wall, 4-TPC outer wall\r
  99 \r
 100   Int_t fRunNumber;\r
 101 \r
 102   // hpt generator\r
 103   Bool_t fHptGenerator; // hpt event generator\r
 104 \r
 105   // trigger class\r
 106   const Char_t * fTriggerClass;\r
 107 \r
 108   // use track vertex\r
 109   Bool_t fUseTrackVertex; // use track vertex\r
 110   \r
 111   // PbPb mode?\r
 112   Bool_t fHighMultiplicity; // flag to switch between pp and PbPb  \r
 113 \r
 114   Bool_t fUseKinkDaughters; // use kink daughthers, default is yes\r
 115 \r
 116   AliPerformanceObject(const AliPerformanceObject&); // not implemented\r
 117   AliPerformanceObject& operator=(const AliPerformanceObject&); // not implemented\r
 118 \r
 119   ClassDef(AliPerformanceObject,4);\r
 120 };\r
 121 \r
 122 #endif\r