[u/mrichter/AliRoot.git] / PWG1 / AliPerformanceTPC.h

#ifndef ALIPERFORMANCETPC_H
#define ALIPERFORMANCETPC_H

//------------------------------------------------------------------------------
// Class to keep information from comparison of 
// reconstructed and MC particle tracks (TPC resolution).   
// 
// Author: J.Otwinowski 04/02/2008 
//------------------------------------------------------------------------------

class TString;
class TNamed;
class TCanvas;
class TH1F;
class TH2F;

class AliESDVertex;
class AliESDtrack;
class AliMCEvent;
class AliStack;
class AliESDEvent; 
class AliESDfriend; 
class AliMCInfoCuts;
class AliRecInfoCuts;

#include "THnSparse.h"
#include "AliPerformanceObject.h"

class AliPerformanceTPC : public AliPerformanceObject {
public :
  AliPerformanceTPC(); 
  AliPerformanceTPC(Char_t* name, Char_t* title, Int_t analysisMode, Bool_t hptGenerator);
  virtual ~AliPerformanceTPC();

  // Init data members
  virtual void  Init();

  // Execute analysis
  virtual void  Exec(AliMCEvent* const mcEvent, AliESDEvent *const esdEvent, AliESDfriend *const esdFriend, const Bool_t bUseMC, const Bool_t bUseESDfriend);
  // Merge output objects (needed by PROOF) 
  virtual Long64_t Merge(TCollection* const list);

  // Analyse output histograms
  virtual void Analyse();

  // Get analysis folder
  virtual TFolder* GetAnalysisFolder() const {return fAnalysisFolder;}

  // Process events
  void ProcessConstrained(AliStack* const stack, AliESDtrack *const esdTrack);
  void ProcessTPC(AliStack* const stack, AliESDtrack *const esdTrack);
  void ProcessTPCITS(AliStack* const stack, AliESDtrack *const esdTrack);

  // Create folder for analysed histograms
  TFolder *CreateFolder(TString folder = "folderTPC",TString title = "Analysed TPC performance histograms");

  // Export objects to folder
  TFolder *ExportToFolder(TObjArray * array=0);

  // Selection cuts
  void SetAliRecInfoCuts(AliRecInfoCuts* const cuts=0) {fCutsRC = cuts;}   
  void SetAliMCInfoCuts(AliMCInfoCuts* const cuts=0) {fCutsMC = cuts;}  
   
  AliRecInfoCuts*  GetAliRecInfoCuts() const {return fCutsRC;}  
  AliMCInfoCuts*   GetAliMCInfoCuts()  const {return fCutsMC;}  

  // getters
  //
  THnSparse *GetTPCHisto() const  { return fTPCHisto; }
private:

  // TPC histogram
  THnSparseF *fTPCHisto; //-> nClust:chi2PerClust:nClust/nFindableClust:eta:phi:pt

  // Global cuts objects
  AliRecInfoCuts* fCutsRC;  // selection cuts for reconstructed tracks
  AliMCInfoCuts*  fCutsMC;  // selection cuts for MC tracks

  // analysis folder 
  TFolder *fAnalysisFolder; // folder for analysed histograms

  AliPerformanceTPC(const AliPerformanceTPC&); // not implemented
  AliPerformanceTPC& operator=(const AliPerformanceTPC&); // not implemented

  ClassDef(AliPerformanceTPC,1);
};

#endif
Commit	Line	Data
777a0ba6	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	//------------------------------------------------------------------------------
	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 AliESDEvent;
25e0c215	22	class AliESDfriend;
777a0ba6	23	class AliMCInfoCuts;
	24	class AliRecInfoCuts;
	25
	26	#include "THnSparse.h"
	27	#include "AliPerformanceObject.h"
	28
	29	class AliPerformanceTPC : public AliPerformanceObject {
	30	public :
	31	AliPerformanceTPC();
	32	AliPerformanceTPC(Char_t* name, Char_t* title, Int_t analysisMode, Bool_t hptGenerator);
	33	virtual ~AliPerformanceTPC();
	34
	35	// Init data members
	36	virtual void Init();
	37
	38	// Execute analysis
25e0c215	39	virtual void Exec(AliMCEvent* const mcEvent, AliESDEvent const esdEvent, AliESDfriend const esdFriend, const Bool_t bUseMC, const Bool_t bUseESDfriend);
777a0ba6	40	// Merge output objects (needed by PROOF)
	41	virtual Long64_t Merge(TCollection* const list);
	42
	43	// Analyse output histograms
	44	virtual void Analyse();
	45
	46	// Get analysis folder
	47	virtual TFolder* GetAnalysisFolder() const {return fAnalysisFolder;}
	48
	49	// Process events
	50	void ProcessConstrained(AliStack* const stack, AliESDtrack *const esdTrack);
	51	void ProcessTPC(AliStack* const stack, AliESDtrack *const esdTrack);
	52	void ProcessTPCITS(AliStack* const stack, AliESDtrack *const esdTrack);
	53
	54	// Create folder for analysed histograms
	55	TFolder *CreateFolder(TString folder = "folderTPC",TString title = "Analysed TPC performance histograms");
	56
	57	// Export objects to folder
	58	TFolder ExportToFolder(TObjArray array=0);
	59
	60	// Selection cuts
	61	void SetAliRecInfoCuts(AliRecInfoCuts* const cuts=0) {fCutsRC = cuts;}
	62	void SetAliMCInfoCuts(AliMCInfoCuts* const cuts=0) {fCutsMC = cuts;}
	63
	64	AliRecInfoCuts* GetAliRecInfoCuts() const {return fCutsRC;}
	65	AliMCInfoCuts* GetAliMCInfoCuts() const {return fCutsMC;}
	66
	67	// getters
	68	//
	69	THnSparse *GetTPCHisto() const { return fTPCHisto; }
	70	private:
	71
	72	// TPC histogram
	73	THnSparseF *fTPCHisto; //-> nClust:chi2PerClust:nClust/nFindableClust:eta:phi:pt
	74
	75	// Global cuts objects
	76	AliRecInfoCuts* fCutsRC; // selection cuts for reconstructed tracks
	77	AliMCInfoCuts* fCutsMC; // selection cuts for MC tracks
	78
	79	// analysis folder
	80	TFolder *fAnalysisFolder; // folder for analysed histograms
	81
	82	AliPerformanceTPC(const AliPerformanceTPC&); // not implemented
	83	AliPerformanceTPC& operator=(const AliPerformanceTPC&); // not implemented
	84
	85	ClassDef(AliPerformanceTPC,1);
	86	};
	87
	88	#endif