[u/mrichter/AliRoot.git] / ANALYSIS / AliCentralitySelectionTask.h

#ifndef ALICENTRALITYSELECTIONTASK_H
#define ALICENTRALITYSELECTIONTASK_H

/* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//*****************************************************
//   Class AliCentralitySelectionTask
//   author: Alberica Toia
//*****************************************************

#include "AliAnalysisTaskSE.h"

class TFile;
class TH1F;
class TList;
class TString;

class AliESDEvent;
class AliESDtrackCuts;

class AliCentralitySelectionTask : public AliAnalysisTaskSE {

 public:

  AliCentralitySelectionTask();
  AliCentralitySelectionTask(const char *name);
  AliCentralitySelectionTask& operator= (const AliCentralitySelectionTask& ana);
  AliCentralitySelectionTask(const AliCentralitySelectionTask& c);
  virtual ~AliCentralitySelectionTask();

  // Implementation of interface methods
  virtual void UserCreateOutputObjects();
  virtual void UserExec(Option_t *option);
  virtual void Terminate(Option_t *option);

  virtual void  SetDebugLevel(Int_t level) {fDebug = level;}
  void SetInput(const char* input)         {fAnalysisInput = input;}
  void SetMCInput()                        {fIsMCInput = kTRUE;}
  
  void ReadCentralityHistos(TString filename);
  void ReadCentralityHistos2(TString filename);
  
  Float_t GetCorrV0(const AliESDEvent* esd, float &v0CorrResc, int run);
  Float_t GetCorrSPD2(Float_t spd2raw,Float_t zv);

 private:

  Int_t SetupRun(AliESDEvent* esd);

  Int_t    fDebug;	   	// Debug flag
  TString  fAnalysisInput; 	// "ESD", "AOD"
  Bool_t   fIsMCInput;          // true when input is MC
  TFile   *fFile;               // file that holds the centrality vs multiplicity 1d
  TFile   *fFile2;              // file that holds the centrality vs multiplicity 2d  
  Int_t    fCurrentRun;         // current run number
  Int_t    fRunNo;              // reference run number

  AliESDtrackCuts* fTrackCuts;             //! optional track cuts

  Float_t  fCentV0M;            // percentile centrality from V0
  Float_t  fCentFMD;            // percentile centrality from FMD
  Float_t  fCentTRK;            // percentile centrality from tracks
  Float_t  fCentTKL;            // percentile centrality from tracklets
  Float_t  fCentCL0;            // percentile centrality from clusters in layer 0
  Float_t  fCentCL1;            // percentile centrality from clusters in layer 0
  Float_t  fCentV0MvsFMD;       // percentile centrality from V0 vs FMD
  Float_t  fCentTKLvsV0M;       // percentile centrality from tracklets vs V0
  Float_t  fCentZEMvsZDC;       // percentile centrality from ZEM vs ZDC

  TH1F    *fHtempV0M;           // histogram with centrality vs multiplicity using V0
  TH1F    *fHtempFMD;           // histogram with centrality vs multiplicity using FMD
  TH1F    *fHtempTRK;           // histogram with centrality vs multiplicity using tracks
  TH1F    *fHtempTKL;           // histogram with centrality vs multiplicity using tracklets
  TH1F    *fHtempCL0;           // histogram with centrality vs multiplicity using clusters in layer 0
  TH1F    *fHtempCL1;           // histogram with centrality vs multiplicity using clusters in layer 0
  TH1F    *fHtempV0MvsFMD;           // histogram with centrality vs multiplicity using V0 vs FMD   
  TH1F    *fHtempTKLvsV0M;           // histogram with centrality vs multiplicity using tracklets vs V0
  TH1F    *fHtempZEMvsZDC;           // histogram with centrality vs multiplicity using ZEM vs ZDC 

  TList       *fOutputList; // output list
  
  TH1F *fHOutCentV0M     ;    //control histogram for centrality
  TH1F *fHOutCentFMD     ;    //control histogram for centrality
  TH1F *fHOutCentTRK     ;    //control histogram for centrality
  TH1F *fHOutCentTKL     ;    //control histogram for centrality
  TH1F *fHOutCentCL0     ;    //control histogram for centrality
  TH1F *fHOutCentCL1     ;    //control histogram for centrality
  TH1F *fHOutCentV0MvsFMD;    //control histogram for centrality
  TH1F *fHOutCentTKLvsV0M;    //control histogram for centrality
  TH1F *fHOutCentZEMvsZDC;    //control histogram for centrality

  ClassDef(AliCentralitySelectionTask,1); 

};

#endif
Commit	Line	Data
dcd68d00	1	#ifndef ALICENTRALITYSELECTIONTASK_H
	2	#define ALICENTRALITYSELECTIONTASK_H
	3
	4	/* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	//*****************************************************
	8	// Class AliCentralitySelectionTask
	9	// author: Alberica Toia
	10	//*****************************************************
	11
	12	#include "AliAnalysisTaskSE.h"
	13
	14	class TFile;
9f14d90a	15	class TH1F;
68e344b7	16	class TList;
9f14d90a	17	class TString;
68e344b7	18
68e344b7	19	class AliESDEvent;
05915818	20	class AliESDtrackCuts;
dcd68d00	21
	22	class AliCentralitySelectionTask : public AliAnalysisTaskSE {
	23
	24	public:
	25
	26	AliCentralitySelectionTask();
	27	AliCentralitySelectionTask(const char *name);
	28	AliCentralitySelectionTask& operator= (const AliCentralitySelectionTask& ana);
	29	AliCentralitySelectionTask(const AliCentralitySelectionTask& c);
	30	virtual ~AliCentralitySelectionTask();
	31
	32	// Implementation of interface methods
	33	virtual void UserCreateOutputObjects();
	34	virtual void UserExec(Option_t *option);
	35	virtual void Terminate(Option_t *option);
	36
	37	virtual void SetDebugLevel(Int_t level) {fDebug = level;}
	38	void SetInput(const char* input) {fAnalysisInput = input;}
	39	void SetMCInput() {fIsMCInput = kTRUE;}
	40
05915818	41	void ReadCentralityHistos(TString filename);
	42	void ReadCentralityHistos2(TString filename);
	43
	44	Float_t GetCorrV0(const AliESDEvent* esd, float &v0CorrResc, int run);
62164ba5	45	Float_t GetCorrSPD2(Float_t spd2raw,Float_t zv);
62164ba5	46
dcd68d00	47	private:
68e344b7	48
	49	Int_t SetupRun(AliESDEvent* esd);
	50
dcd68d00	51	Int_t fDebug; // Debug flag
	52	TString fAnalysisInput; // "ESD", "AOD"
	53	Bool_t fIsMCInput; // true when input is MC
d15bf53f	54	TFile *fFile; // file that holds the centrality vs multiplicity 1d
05915818	55	TFile *fFile2; // file that holds the centrality vs multiplicity 2d
68e344b7	56	Int_t fCurrentRun; // current run number
62164ba5	57	Int_t fRunNo; // reference run number
d15bf53f	58
05915818	59	AliESDtrackCuts* fTrackCuts; //! optional track cuts
05915818	60
d15bf53f	61	Float_t fCentV0M; // percentile centrality from V0
	62	Float_t fCentFMD; // percentile centrality from FMD
	63	Float_t fCentTRK; // percentile centrality from tracks
	64	Float_t fCentTKL; // percentile centrality from tracklets
	65	Float_t fCentCL0; // percentile centrality from clusters in layer 0
be0d4e9b	66	Float_t fCentCL1; // percentile centrality from clusters in layer 0
d15bf53f	67	Float_t fCentV0MvsFMD; // percentile centrality from V0 vs FMD
	68	Float_t fCentTKLvsV0M; // percentile centrality from tracklets vs V0
	69	Float_t fCentZEMvsZDC; // percentile centrality from ZEM vs ZDC
	70
9f14d90a	71	TH1F *fHtempV0M; // histogram with centrality vs multiplicity using V0
	72	TH1F *fHtempFMD; // histogram with centrality vs multiplicity using FMD
	73	TH1F *fHtempTRK; // histogram with centrality vs multiplicity using tracks
	74	TH1F *fHtempTKL; // histogram with centrality vs multiplicity using tracklets
	75	TH1F *fHtempCL0; // histogram with centrality vs multiplicity using clusters in layer 0
	76	TH1F *fHtempCL1; // histogram with centrality vs multiplicity using clusters in layer 0
	77	TH1F *fHtempV0MvsFMD; // histogram with centrality vs multiplicity using V0 vs FMD
	78	TH1F *fHtempTKLvsV0M; // histogram with centrality vs multiplicity using tracklets vs V0
	79	TH1F *fHtempZEMvsZDC; // histogram with centrality vs multiplicity using ZEM vs ZDC
d15bf53f	80
05915818	81	TList *fOutputList; // output list
	82
	83	TH1F *fHOutCentV0M ; //control histogram for centrality
	84	TH1F *fHOutCentFMD ; //control histogram for centrality
	85	TH1F *fHOutCentTRK ; //control histogram for centrality
	86	TH1F *fHOutCentTKL ; //control histogram for centrality
	87	TH1F *fHOutCentCL0 ; //control histogram for centrality
	88	TH1F *fHOutCentCL1 ; //control histogram for centrality
	89	TH1F *fHOutCentV0MvsFMD; //control histogram for centrality
	90	TH1F *fHOutCentTKLvsV0M; //control histogram for centrality
	91	TH1F *fHOutCentZEMvsZDC; //control histogram for centrality
	92
dcd68d00	93	ClassDef(AliCentralitySelectionTask,1);
	94
	95	};
	96
	97	#endif
	98