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

#ifndef ALICOMPARISONRES_H
#define ALICOMPARISONRES_H

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

class TFile;
class AliMCInfo;
class AliESDRecInfo;
class AliESDEvent; 
class AliESD;
class AliESDfriend;
class AliMCInfoCuts;
class AliRecInfoCuts;
class TH1I;
class TH3F;
class TH3;
class TProfile;
class TProfile2D;
class TString;
class AliESDVertex;

#include "TNamed.h"
#include "AliComparisonObject.h"

class AliComparisonRes : public AliComparisonObject {
public :
  AliComparisonRes(); 
  virtual ~AliComparisonRes();

  // Init data members
  virtual void     Init();

  // Execute analysis
  virtual void      Exec(AliMCInfo* infoMC, AliESDRecInfo *infoRC);

  // Merge output objects (needed by PROOF) 
  virtual Long64_t Merge(TCollection* list);

  // Analyse output histograms
  virtual void Analyse();

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

  // Process events
  void      ProcessConstrained(AliMCInfo* infoMC, AliESDRecInfo *infoRC);
  void      Process(AliMCInfo* infoMC, AliESDRecInfo *infoRC);

  // Create folder for analysed histograms
  TFolder *CreateFolder(TString folder = "folderRes",TString title = "Analysed Resolution histograms");

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

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

  static TH1F*       MakeResol(TH2F * his, Int_t integ, Bool_t type); 


private:
  //
  // Control histograms
  //
  TH2F* fPtResolLPT;        //-> pt resolution - low pt
  TH2F* fPtResolHPT;        //-> pt resolution - high pt 
  TH2F* fPtPullLPT;         //-> pt resolution - low pt
  TH2F* fPtPullHPT;         //-> pt resolution - high pt 

  //
  // Resolution constrained param
  //
  TH2F   *fCPhiResolTan;   //-> angular resolution -  constrained
  TH2F   *fCTanResolTan;   //-> angular resolution -  constrained
  TH2F   *fCPtResolTan;    //-> pt resolution      -  constrained
  TH2F   *fCPhiPullTan;    //-> angular resolution -  constrained
  TH2F   *fCTanPullTan;    //-> angular resolution -  constrained
  TH2F   *fCPtPullTan;     //-> pt resolution      -  constrained

  //
  // Histograms for track resolution parameterisation
  //

  TH2F* f1Pt2ResolS1PtTPC;      //-> (1/mcpt-1/pt)/(1+1/pt)^2 vs sqrt(1/pt) (TPC)
  TH2F* f1Pt2ResolS1PtTPCITS;   //-> (1/mcpt-1/pt)/(1+1/pt)^2 vs sqrt(1/pt) (TPC+ITS)
  TH2F* fYResolS1PtTPC;         //-> (mcy-y)/(0.2+1/pt) vs sqrt(1/pt) (TPC) 
  TH2F* fYResolS1PtTPCITS;      //-> (mcy-y)/(0.2+1/pt) vs sqrt(1/pt) (TPC + ITS) 
  TH2F* fZResolS1PtTPC;         //-> (mcz-z)/(0.2+1/pt) vs sqrt(1/pt) (TPC)
  TH2F* fZResolS1PtTPCITS;      //-> (mcz-z)/(0.2+1/pt) vs sqrt(1/pt) (TPC+ITS)
  TH2F* fPhiResolS1PtTPC;       //-> (mcphi-phi)/(0.1+1/pt) vs sqrt(1/pt) (TPC)
  TH2F* fPhiResolS1PtTPCITS;    //-> (mcphi-phi)/(0.1+1/pt) vs sqrt(1/pt) (TPC+ITS)
  TH2F* fThetaResolS1PtTPC;     //-> (mctheta-theta)/(0.1+1/pt) vs sqrt(1/pt) (TPC)
  TH2F* fThetaResolS1PtTPCITS;  //-> (mctheta-theta)/(0.1+1/pt) vs sqrt(1/pt) (TPC+ITS)
  
  // constrained
  TH2F* fC1Pt2ResolS1PtTPC;      //-> (1/mcpt-1/pt)/(1+1/pt)^2 vs sqrt(1/pt) (TPC)
  TH2F* fC1Pt2ResolS1PtTPCITS;   //-> (1/mcpt-1/pt)/(1+1/pt)^2 vs sqrt(1/pt) (TPC+ITS)
  TH2F* fCYResolS1PtTPC;         //-> (mcy-y)/(0.2+1/pt) vs sqrt(1/pt) (TPC) 
  TH2F* fCYResolS1PtTPCITS;      //-> (mcy-y)/(0.2+1/pt) vs sqrt(1/pt) (TPC + ITS) 
  TH2F* fCZResolS1PtTPC;         //-> (mcz-z)/(0.2+1/pt) vs sqrt(1/pt) (TPC)
  TH2F* fCZResolS1PtTPCITS;      //-> (mcz-z)/(0.2+1/pt) vs sqrt(1/pt) (TPC+ITS)
  TH2F* fCPhiResolS1PtTPC;       //-> (mcphi-phi)/(0.1+1/pt) vs sqrt(1/pt) (TPC)
  TH2F* fCPhiResolS1PtTPCITS;    //-> (mcphi-phi)/(0.1+1/pt) vs sqrt(1/pt) (TPC+ITS)
  TH2F* fCThetaResolS1PtTPC;     //-> (mctheta-theta)/(0.1+1/pt) vs sqrt(1/pt) (TPC)
  TH2F* fCThetaResolS1PtTPCITS;  //-> (mctheta-theta)/(0.1+1/pt) vs sqrt(1/pt) (TPC+ITS)

  AliESDVertex *fVertex;  //! 

  // 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

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

  ClassDef(AliComparisonRes,1);
};

#endif
Commit	Line	Data
3baa4bfd	1	#ifndef ALICOMPARISONRES_H
	2	#define ALICOMPARISONRES_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 TFile;
	12	class AliMCInfo;
	13	class AliESDRecInfo;
	14	class AliESDEvent;
	15	class AliESD;
	16	class AliESDfriend;
	17	class AliMCInfoCuts;
	18	class AliRecInfoCuts;
	19	class TH1I;
	20	class TH3F;
	21	class TH3;
	22	class TProfile;
	23	class TProfile2D;
	24	class TString;
	25	class AliESDVertex;
	26
	27	#include "TNamed.h"
	28	#include "AliComparisonObject.h"
	29
	30	class AliComparisonRes : public AliComparisonObject {
	31	public :
	32	AliComparisonRes();
	33	virtual ~AliComparisonRes();
	34
	35	// Init data members
	36	virtual void Init();
	37
	38	// Execute analysis
	39	virtual void Exec(AliMCInfo* infoMC, AliESDRecInfo *infoRC);
	40
	41	// Merge output objects (needed by PROOF)
	42	virtual Long64_t Merge(TCollection* list);
	43
	44	// Analyse output histograms
	45	virtual void Analyse();
	46
	47	// Get analysis folder
	48	virtual TFolder* GetAnalysisFolder() {return fAnalysisFolder;}
	49
	50	// Process events
	51	void ProcessConstrained(AliMCInfo* infoMC, AliESDRecInfo *infoRC);
	52	void Process(AliMCInfo* infoMC, AliESDRecInfo *infoRC);
	53
	54	// Create folder for analysed histograms
	55	TFolder *CreateFolder(TString folder = "folderRes",TString title = "Analysed Resolution histograms");
	56
b4126c69	57	// Export objects to folder
	58	TFolder ExportToFolder(TObjArray array=0);
	59
3baa4bfd	60	// Selection cuts
	61	void SetAliRecInfoCuts(AliRecInfoCuts* cuts=0) {fCutsRC = cuts;}
	62	void SetAliMCInfoCuts(AliMCInfoCuts* cuts=0) {fCutsMC = cuts;}
	63
	64	AliRecInfoCuts* GetAliRecInfoCuts() const {return fCutsRC;}
	65	AliMCInfoCuts* GetAliMCInfoCuts() const {return fCutsMC;}
	66
	67	static TH1F* MakeResol(TH2F * his, Int_t integ, Bool_t type);
	68
	69
	70	private:
	71	//
	72	// Control histograms
	73	//
	74	TH2F* fPtResolLPT; //-> pt resolution - low pt
	75	TH2F* fPtResolHPT; //-> pt resolution - high pt
	76	TH2F* fPtPullLPT; //-> pt resolution - low pt
	77	TH2F* fPtPullHPT; //-> pt resolution - high pt
	78
	79	//
	80	// Resolution constrained param
	81	//
	82	TH2F *fCPhiResolTan; //-> angular resolution - constrained
	83	TH2F *fCTanResolTan; //-> angular resolution - constrained
	84	TH2F *fCPtResolTan; //-> pt resolution - constrained
	85	TH2F *fCPhiPullTan; //-> angular resolution - constrained
	86	TH2F *fCTanPullTan; //-> angular resolution - constrained
	87	TH2F *fCPtPullTan; //-> pt resolution - constrained
	88
	89	//
	90	// Histograms for track resolution parameterisation
	91	//
	92
	93	TH2F* f1Pt2ResolS1PtTPC; //-> (1/mcpt-1/pt)/(1+1/pt)^2 vs sqrt(1/pt) (TPC)
	94	TH2F* f1Pt2ResolS1PtTPCITS; //-> (1/mcpt-1/pt)/(1+1/pt)^2 vs sqrt(1/pt) (TPC+ITS)
	95	TH2F* fYResolS1PtTPC; //-> (mcy-y)/(0.2+1/pt) vs sqrt(1/pt) (TPC)
	96	TH2F* fYResolS1PtTPCITS; //-> (mcy-y)/(0.2+1/pt) vs sqrt(1/pt) (TPC + ITS)
	97	TH2F* fZResolS1PtTPC; //-> (mcz-z)/(0.2+1/pt) vs sqrt(1/pt) (TPC)
	98	TH2F* fZResolS1PtTPCITS; //-> (mcz-z)/(0.2+1/pt) vs sqrt(1/pt) (TPC+ITS)
	99	TH2F* fPhiResolS1PtTPC; //-> (mcphi-phi)/(0.1+1/pt) vs sqrt(1/pt) (TPC)
	100	TH2F* fPhiResolS1PtTPCITS; //-> (mcphi-phi)/(0.1+1/pt) vs sqrt(1/pt) (TPC+ITS)
	101	TH2F* fThetaResolS1PtTPC; //-> (mctheta-theta)/(0.1+1/pt) vs sqrt(1/pt) (TPC)
	102	TH2F* fThetaResolS1PtTPCITS; //-> (mctheta-theta)/(0.1+1/pt) vs sqrt(1/pt) (TPC+ITS)
	103
	104	// constrained
	105	TH2F* fC1Pt2ResolS1PtTPC; //-> (1/mcpt-1/pt)/(1+1/pt)^2 vs sqrt(1/pt) (TPC)
	106	TH2F* fC1Pt2ResolS1PtTPCITS; //-> (1/mcpt-1/pt)/(1+1/pt)^2 vs sqrt(1/pt) (TPC+ITS)
	107	TH2F* fCYResolS1PtTPC; //-> (mcy-y)/(0.2+1/pt) vs sqrt(1/pt) (TPC)
	108	TH2F* fCYResolS1PtTPCITS; //-> (mcy-y)/(0.2+1/pt) vs sqrt(1/pt) (TPC + ITS)
	109	TH2F* fCZResolS1PtTPC; //-> (mcz-z)/(0.2+1/pt) vs sqrt(1/pt) (TPC)
	110	TH2F* fCZResolS1PtTPCITS; //-> (mcz-z)/(0.2+1/pt) vs sqrt(1/pt) (TPC+ITS)
	111	TH2F* fCPhiResolS1PtTPC; //-> (mcphi-phi)/(0.1+1/pt) vs sqrt(1/pt) (TPC)
	112	TH2F* fCPhiResolS1PtTPCITS; //-> (mcphi-phi)/(0.1+1/pt) vs sqrt(1/pt) (TPC+ITS)
	113	TH2F* fCThetaResolS1PtTPC; //-> (mctheta-theta)/(0.1+1/pt) vs sqrt(1/pt) (TPC)
	114	TH2F* fCThetaResolS1PtTPCITS; //-> (mctheta-theta)/(0.1+1/pt) vs sqrt(1/pt) (TPC+ITS)
	115
	116	AliESDVertex *fVertex; //!
	117
	118	// Global cuts objects
	119	AliRecInfoCuts* fCutsRC; // selection cuts for reconstructed tracks
	120	AliMCInfoCuts* fCutsMC; // selection cuts for MC tracks
	121
	122	// analysis folder
	123	TFolder *fAnalysisFolder; // folder for analysed histograms
124
125	AliComparisonRes(const AliComparisonRes&); // not implemented
126	AliComparisonRes& operator=(const AliComparisonRes&); // not implemented
127
128	ClassDef(AliComparisonRes,1);
129	};
130
131	#endif