PWG1/TPC/AliPerformanceTPC.h

   1 #ifndef ALIPERFORMANCETPC_H
   2 #define ALIPERFORMANCETPC_H
   3
   4 //------------------------------------------------------------------------------
   5 // Class to keep information from comparison of
   6 // reconstructed and MC particle tracks (TPC resolution).
   7 //
   8 // Author: J.Otwinowski 04/02/2008
   9 // Changes by M.Knichel 27/07/2010
  10 //------------------------------------------------------------------------------
  11
  12 class TString;
  13 class TNamed;
  14 class TCanvas;
  15 class TH1F;
  16 class TH2F;
  17
  18 class AliESDVertex;
  19 class AliESDtrack;
  20 class AliMCEvent;
  21 class AliStack;
  22 class AliESDEvent;
  23 class AliESDfriend;
  24 class AliMCInfoCuts;
  25 class AliRecInfoCuts;
  26
  27 #include "THnSparse.h"
  28 #include "AliPerformanceObject.h"
  29
  30 class AliPerformanceTPC : public AliPerformanceObject {
  31 public :
  32   AliPerformanceTPC();
  33   AliPerformanceTPC(Char_t* name, Char_t* title, Int_t analysisMode, Bool_t hptGenerator);
  34   virtual ~AliPerformanceTPC();
  35
  36   // Init data members
  37   virtual void  Init();
  38
  39   // Execute analysis
  40   virtual void  Exec(AliMCEvent* const mcEvent, AliESDEvent *const esdEvent, AliESDfriend *const esdFriend, const Bool_t bUseMC, const Bool_t bUseESDfriend);
  41   // Merge output objects (needed by PROOF)
  42   virtual Long64_t Merge(TCollection* const list);
  43
  44   // Analyse output histograms
  45   virtual void Analyse();
  46
  47   // Get analysis folder
  48   virtual TFolder* GetAnalysisFolder() const {return fAnalysisFolder;}
  49
  50   // Process events
  51   void ProcessConstrained(AliStack* const stack, AliESDtrack *const esdTrack, AliESDEvent *const esdEvent);
  52   void ProcessTPC(AliStack* const stack, AliESDtrack *const esdTrack, AliESDEvent *const esdEvent, Bool_t vertStatus);
  53   void ProcessTPCITS(AliStack* const stack, AliESDtrack *const esdTrack, AliESDEvent *const esdEvent, Bool_t vertStatus);
  54
  55   // Create folder for analysed histograms
  56   TFolder *CreateFolder(TString folder = "folderTPC",TString title = "Analysed TPC performance histograms");
  57
  58   // Export objects to folder
  59   TFolder *ExportToFolder(TObjArray * array=0);
  60
  61   // Selection cuts
  62   void SetAliRecInfoCuts(AliRecInfoCuts* const cuts=0) {fCutsRC = cuts;}
  63   void SetAliMCInfoCuts(AliMCInfoCuts* const cuts=0) {fCutsMC = cuts;}
  64
  65   AliRecInfoCuts*  GetAliRecInfoCuts() const {return fCutsRC;}
  66   AliMCInfoCuts*   GetAliMCInfoCuts()  const {return fCutsMC;}
  67
  68   // getters
  69   //
  70   THnSparse *GetTPCClustHisto() const  { return fTPCClustHisto; }
  71   THnSparse *GetTPCEventHisto() const  { return fTPCEventHisto; }
  72   THnSparse *GetTPCTrackHisto() const  { return fTPCTrackHisto; }
  73   TObjArray* GetTPCHistos() const { return fFolderObj; }
  74   static Bool_t GetMergeTHnSparse() { return fgMergeTHnSparse; }
  75   static void SetMergeTHnSparse(Bool_t mergeTHnSparse) { fgMergeTHnSparse = mergeTHnSparse; }
  76
  77
  78 private:
  79
  80   static Bool_t fgMergeTHnSparse;
  81   void AddTrackHistos(TObjArray* aFolderObj, Char_t* selString);
  82
  83   // TPC histogram
  84   THnSparseF *fTPCClustHisto; //-> padRow:phi:TPCSide
  85   THnSparseF *fTPCEventHisto;  //-> Xv:Yv:Zv:mult:multP:multN:vertStatus
  86   THnSparseF *fTPCTrackHisto;  //-> nClust:chi2PerClust:nClust/nFindableClust:DCAr:DCAz:eta:phi:pt:charge:vertStatus
  87   TObjArray* fFolderObj; // array of analysed histograms
  88
  89   // Global cuts objects
  90   AliRecInfoCuts* fCutsRC;  // selection cuts for reconstructed tracks
  91   AliMCInfoCuts*  fCutsMC;  // selection cuts for MC tracks
  92
  93   // analysis folder
  94   TFolder *fAnalysisFolder; // folder for analysed histograms
  95
  96
  97   AliPerformanceTPC(const AliPerformanceTPC&); // not implemented
  98   AliPerformanceTPC& operator=(const AliPerformanceTPC&); // not implemented
  99
 100   ClassDef(AliPerformanceTPC,5);
 101 };
 102
 103 #endif