PWGPP/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 15/10/2010
  10 //------------------------------------------------------------------------------
  11
  12 class TString;
  13 class TNamed;
  14 class TCanvas;
  15 class TH1;
  16 class TH2;
  17 class TH3;
  18
  19 class AliESDVertex;
  20 class AliESDtrack;
  21 class AliMCEvent;
  22 class AliStack;
  23 class AliESDEvent;
  24 class AliESDfriend;
  25 class AliMCInfoCuts;
  26 class AliRecInfoCuts;
  27
  28 #include "THnSparse.h"
  29 #include "AliPerformanceObject.h"
  30
  31 class AliPerformanceTPC : public AliPerformanceObject {
  32 public :
  33   //AliPerformanceTPC();
  34   AliPerformanceTPC(const Char_t* name="AliPerformanceTPC", const Char_t* title="AliPerformanceTPC",Int_t analysisMode=0,Bool_t hptGenerator=kFALSE, Int_t run=-1, Bool_t highMult = kFALSE);
  35
  36   virtual ~AliPerformanceTPC();
  37
  38   // Init data members
  39   virtual void  Init();
  40
  41   // Execute analysis
  42   virtual void  Exec(AliMCEvent* const mcEvent, AliESDEvent *const esdEvent, AliESDfriend *const esdFriend, const Bool_t bUseMC, const Bool_t bUseESDfriend);
  43   // Merge output objects (needed by PROOF)
  44   virtual Long64_t Merge(TCollection* const list);
  45
  46   // Analyse output histograms
  47   virtual void Analyse();
  48
  49   // Get analysis folder
  50   virtual TFolder* GetAnalysisFolder() const {return fAnalysisFolder;}
  51
  52   // produce summary
  53   virtual TTree* CreateSummary();
  54
  55   // Process events
  56   void ProcessConstrained(AliStack* const stack, AliESDtrack *const esdTrack, AliESDEvent *const esdEvent);
  57   void ProcessTPC(AliStack* const stack, AliESDtrack *const esdTrack, AliESDEvent *const esdEvent, Bool_t vertStatus);
  58   void ProcessTPCITS(AliStack* const stack, AliESDtrack *const esdTrack, AliESDEvent *const esdEvent, Bool_t vertStatus);
  59
  60   // Create folder for analysed histograms
  61   TFolder *CreateFolder(TString folder = "folderTPC",TString title = "Analysed TPC performance histograms");
  62
  63   // Export objects to folder
  64   TFolder *ExportToFolder(TObjArray * array=0);
  65
  66   // Selection cuts
  67   void SetAliRecInfoCuts(AliRecInfoCuts* const cuts=0) {fCutsRC = cuts;}
  68   void SetAliMCInfoCuts(AliMCInfoCuts* const cuts=0) {fCutsMC = cuts;}
  69
  70   AliRecInfoCuts*  GetAliRecInfoCuts() const {return fCutsRC;}
  71   AliMCInfoCuts*   GetAliMCInfoCuts()  const {return fCutsMC;}
  72
  73   // getters
  74   //
  75   THnSparse *GetTPCClustHisto() const  { return fTPCClustHisto; }
  76   THnSparse *GetTPCEventHisto() const  { return fTPCEventHisto; }
  77   THnSparse *GetTPCTrackHisto() const  { return fTPCTrackHisto; }
  78
  79   TObjArray* GetHistos() const { return fFolderObj; }
  80
  81   static Bool_t GetMergeTHnSparse() { return fgMergeTHnSparse; }
  82   static void SetMergeTHnSparse(Bool_t mergeTHnSparse) {fgUseMergeTHnSparse = kTRUE; fgMergeTHnSparse = mergeTHnSparse; }
  83
  84   void SetUseHLT(Bool_t useHLT = kTRUE) {fUseHLT = useHLT;}
  85   Bool_t GetUseHLT() { return fUseHLT; }
  86
  87
  88 private:
  89
  90   static Bool_t fgMergeTHnSparse;
  91   static Bool_t fgUseMergeTHnSparse;
  92
  93   // TPC histogram
  94   THnSparseF *fTPCClustHisto; //-> padRow:phi:TPCside
  95   THnSparseF *fTPCEventHisto;  //-> Xv:Yv:Zv:mult:multP:multN:vertStatus
  96   THnSparseF *fTPCTrackHisto;  //-> nClust:chi2PerClust:nClust/nFindableClust:DCAr:DCAz:eta:phi:pt:charge:vertStatus
  97   TObjArray* fFolderObj; // array of analysed histograms
  98
  99   // Global cuts objects
 100   AliRecInfoCuts* fCutsRC;  // selection cuts for reconstructed tracks
 101   AliMCInfoCuts*  fCutsMC;  // selection cuts for MC tracks
 102
 103   // analysis folder
 104   TFolder *fAnalysisFolder; // folder for analysed histograms
 105
 106   Bool_t fUseHLT; // use HLT ESD
 107
 108   AliPerformanceTPC(const AliPerformanceTPC&); // not implemented
 109   AliPerformanceTPC& operator=(const AliPerformanceTPC&); // not implemented
 110
 111   ClassDef(AliPerformanceTPC,11);
 112 };
 113
 114 #endif