PWG1/TPC/AliPerformanceRes.h

   1 #ifndef ALIPERFORMANCERES_H
   2 #define ALIPERFORMANCERES_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 //------------------------------------------------------------------------------
  10
  11 class TString;
  12 class TNamed;
  13 class TCanvas;
  14 class TH1F;
  15 class TH2F;
  16
  17 class AliESDVertex;
  18 class AliESDtrack;
  19 class AliMCEvent;
  20 class AliStack;
  21 class AliTrackReference;
  22 class AliESDEvent;
  23 class AliESDfriend;
  24 class AliESDfriendTrack;
  25 class AliMCEvent;
  26 class AliMCParticle;
  27 class AliMCInfoCuts;
  28 class AliRecInfoCuts;
  29
  30 #include "THnSparse.h"
  31 #include "AliPerformanceObject.h"
  32
  33 class AliPerformanceRes : public AliPerformanceObject {
  34 public :
  35   AliPerformanceRes();
  36   AliPerformanceRes(Char_t* name, Char_t* title, Int_t analysisMode, Bool_t hptGenerator);
  37   virtual ~AliPerformanceRes();
  38
  39   // Init data members
  40   virtual void  Init();
  41
  42   // Execute analysis
  43   virtual void  Exec(AliMCEvent* const mcEvent, AliESDEvent *const esdEvent, AliESDfriend *const esdFriend, const Bool_t bUseMC, const Bool_t bUseESDfriend);
  44
  45   // Merge output objects (needed by PROOF)
  46   virtual Long64_t Merge(TCollection* const list);
  47
  48   // Analyse output histograms
  49   virtual void Analyse();
  50
  51   // Get analysis folder
  52   virtual TFolder* GetAnalysisFolder() const {return fAnalysisFolder;}
  53
  54   // Process events
  55   void ProcessConstrained(AliStack* const stack, AliESDtrack *const esdTrack, AliESDEvent* const esdEvent );
  56   void ProcessTPC(AliStack* const stack, AliESDtrack *const esdTrack, AliESDEvent* const esdEvent);
  57   void ProcessTPCITS(AliStack* const stack, AliESDtrack *const esdTrack, AliESDEvent* const esdEvent);
  58   void ProcessInnerTPC(AliMCEvent *const mcEvent, AliESDtrack *const esdTrack, AliESDEvent* const esdEvent);
  59   void ProcessOuterTPC(AliMCEvent *const mcEvent, AliESDtrack *const esdTrack, AliESDfriendTrack *const friendTrack, AliESDEvent* const esdEvent);
  60
  61   AliTrackReference *GetFirstTPCTrackRef(AliMCParticle *mcParticle);
  62   AliTrackReference *GetLastTPCTrackRef(AliMCParticle *mcParticle);
  63
  64   // Create folder for analysed histograms
  65   TFolder *CreateFolder(TString folder = "folderRes",TString title = "Analysed Resolution histograms");
  66
  67   // Export objects to folder
  68   TFolder *ExportToFolder(TObjArray * array=0);
  69
  70   // Selection cuts
  71   void SetAliRecInfoCuts(AliRecInfoCuts* const cuts=0) {fCutsRC = cuts;}
  72   void SetAliMCInfoCuts(AliMCInfoCuts* const cuts=0) {fCutsMC = cuts;}
  73
  74   AliRecInfoCuts*  GetAliRecInfoCuts() const {return fCutsRC;}
  75   AliMCInfoCuts*   GetAliMCInfoCuts()  const {return fCutsMC;}
  76
  77   TH1F*  MakeResol(TH2F * his, Int_t integ=0, Bool_t type=kFALSE, Int_t cut=0);
  78
  79   // getters
  80   //
  81   THnSparse *GetResolHisto() const  { return fResolHisto; }
  82   THnSparse *GetPullHisto()  const  { return fPullHisto; }
  83 private:
  84   //
  85   // Control histograms
  86   // 5 track parameters (details in STEER/AliExternalTrackParam.h)
  87   //
  88
  89   // resolution histogram
  90   THnSparseF *fResolHisto; //-> res_y:res_z:res_phi:res_lambda:res_pt:y:z:phi:eta:pt
  91
  92   // pull histogram
  93   //THnSparseF *fPullHisto;  //-> pull_y:pull_z:pull_phi:pull_lambda:pull_1pt:y:z:eta:phi:pt
  94   THnSparseF *fPullHisto;  //-> pull_y:pull_z:pull_snp:pull_tgl:pull_1pt:y:z:snp:tgl:1pt
  95
  96   // Global cuts objects
  97   AliRecInfoCuts*  fCutsRC;      // selection cuts for reconstructed tracks
  98   AliMCInfoCuts*  fCutsMC;       // selection cuts for MC tracks
  99
 100   // analysis folder
 101   TFolder *fAnalysisFolder; // folder for analysed histograms
 102
 103   AliPerformanceRes(const AliPerformanceRes&); // not implemented
 104   AliPerformanceRes& operator=(const AliPerformanceRes&); // not implemented
 105
 106   ClassDef(AliPerformanceRes,1);
 107 };
 108
 109 #endif