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, Bool_t highMult=kFALSE); \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   // project to 1d,2d,3d\r
  46   // is called from FinishTaskOuput() in AliPerformanceTask\r
  47   virtual void Analyse() = 0;\r
  48 \r
  49   // Get output folder for analysed histograms\r
  50   virtual TFolder* GetAnalysisFolder() const = 0;\r
  51   \r
  52   // create a summary stored in a ttree \r
  53   // has to be implented\r
  54   virtual TTree* CreateSummary() { return 0; }\r
  55   \r
  56   // project to 1d,2d,3d\r
  57   // is called from Terminate() in AliPerformanceTask\r
  58   // final spectra calculation\r
  59   virtual void AnalyseFinal() { ; }\r
  60 \r
  61   // \r
  62   virtual void SetAliRecInfoCuts(AliRecInfoCuts* const cuts=0) = 0;\r
  63   virtual void SetAliMCInfoCuts(AliMCInfoCuts* const cuts=0) = 0; \r
  64 \r
  65   // set and get analysisMode\r
  66   void SetAnalysisMode(const Int_t analysisMode=0) {fAnalysisMode = analysisMode;} \r
  67   Int_t GetAnalysisMode() const {return fAnalysisMode;}\r
  68 \r
  69   // set and get hpt generator \r
  70   void SetHptGenerator(const Bool_t hptGenerator=kFALSE) {fHptGenerator = hptGenerator;}\r
  71   Bool_t IsHptGenerator() const {return fHptGenerator;}\r
  72 \r
  73   // draw all histograms from the folder\r
  74   void PrintHisto(Bool_t logz = kTRUE, Char_t * outFileName = "PerformanceQAHisto.ps"); \r
  75 \r
  76   // create log axis \r
  77   Double_t *CreateLogAxis(Int_t nbins, Double_t xmin, Double_t xmax); \r
  78 \r
  79   // trigger class selection\r
  80   void SetTriggerClass(const Char_t *triggerClass) { fTriggerClass = triggerClass; }\r
  81   const Char_t* GetTriggerClass() const { return fTriggerClass; }\r
  82 \r
  83   // use track vertex\r
  84   void SetUseTrackVertex(Bool_t trackVtx = kTRUE) { fUseTrackVertex = trackVtx; }\r
  85   Bool_t IsUseTrackVertex() { return fUseTrackVertex; }\r
  86   \r
  87   Bool_t IsHighMultiplicity() { return fHighMultiplicity; }  \r
  88   \r
  89   void SetRunNumber(Int_t run) { fRunNumber = run; }\r
  90   Int_t GetRunNumber() const { return fRunNumber; }\r
  91 \r
  92   // use kink daughters\r
  93   void SetUseKinkDaughters(Bool_t kinkDaughters = kTRUE) { fUseKinkDaughters = kinkDaughters; }\r
  94   Bool_t IsUseKinkDaughters() { return fUseKinkDaughters; }\r
  95 \r
  96   // Centrality bin to be used\r
  97   void  SetUseCentralityBin(Int_t bin) { fUseCentralityBin = bin; }\r
  98   Int_t GetUseCentralityBin()          { return fUseCentralityBin; }\r
  99   \r
 100 protected: \r
 101 \r
 102   void AddProjection(TObjArray* aFolderObj, TString nameSparse, THnSparse *hSparse, Int_t xDim, TString* selString = 0);\r
 103   void AddProjection(TObjArray* aFolderObj, TString nameSparse, THnSparse *hSparse, Int_t xDim, Int_t yDim, TString* selString = 0);\r
 104   void AddProjection(TObjArray* aFolderObj, TString nameSparse, THnSparse *hSparse, Int_t xDim, Int_t yDim, Int_t zDim, TString* selString = 0);\r
 105 \r
 106   // analysis mode\r
 107   Int_t fAnalysisMode;  // 0-TPC, 1-TPCITS, 2-Constrained, 3-TPC inner wall, 4-TPC outer wall\r
 108 \r
 109   Int_t fRunNumber;\r
 110 \r
 111   // hpt generator\r
 112   Bool_t fHptGenerator; // hpt event generator\r
 113 \r
 114   // trigger class\r
 115   const Char_t * fTriggerClass;\r
 116 \r
 117   // use track vertex\r
 118   Bool_t fUseTrackVertex; // use track vertex\r
 119   \r
 120   // PbPb mode?\r
 121   Bool_t fHighMultiplicity; // flag to switch between pp and PbPb  \r
 122 \r
 123   Bool_t fUseKinkDaughters; // use kink daughthers, default is yes\r
 124 \r
 125   Int_t  fUseCentralityBin;  // centrality bin to be used \r
 126 \r
 127   AliPerformanceObject(const AliPerformanceObject&); // not implemented\r
 128   AliPerformanceObject& operator=(const AliPerformanceObject&); // not implemented\r
 129 \r
 130   ClassDef(AliPerformanceObject,6);\r
 131 };\r
 132 \r
 133 #endif\r