[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 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 SetPercentileFile(TString filename);
  void SetPercentileFile2(TString filename);
  void ReadCentralityHistos();
  void ReadCentralityHistos2();

  void AddPercentileFileToList(TString filename) { fFileList->Add(new TObjString(filename)); }
  void AddPercentileFile2ToList(TString filename) { fFileList2->Add(new TObjString(filename)); }

  Float_t GetCorrV0(const AliESDEvent* esd, float &v0CorrResc);
  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
  TString  fCentfilename;       // name of this file 1d
  TString  fCentfilename2;      // name of this file 2d
  
  TList*   fFileList;           //! list of input files names
  TList*   fFileList2;          //! list of input files 2 names
  Int_t    fCurrentRun;         // current run number
  Int_t    fRunNo;              // reference run number

  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 

  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;
dcd68d00	20
	21	class AliCentralitySelectionTask : public AliAnalysisTaskSE {
	22
	23	public:
	24
	25	AliCentralitySelectionTask();
	26	AliCentralitySelectionTask(const char *name);
	27	AliCentralitySelectionTask& operator= (const AliCentralitySelectionTask& ana);
	28	AliCentralitySelectionTask(const AliCentralitySelectionTask& c);
	29	virtual ~AliCentralitySelectionTask();
	30
	31	// Implementation of interface methods
	32	virtual void UserCreateOutputObjects();
	33	virtual void UserExec(Option_t *option);
	34	virtual void Terminate(Option_t *option);
	35
	36	virtual void SetDebugLevel(Int_t level) {fDebug = level;}
	37	void SetInput(const char* input) {fAnalysisInput = input;}
	38	void SetMCInput() {fIsMCInput = kTRUE;}
	39
d15bf53f	40	void SetPercentileFile(TString filename);
	41	void SetPercentileFile2(TString filename);
	42	void ReadCentralityHistos();
	43	void ReadCentralityHistos2();
dcd68d00	44
68e344b7	45	void AddPercentileFileToList(TString filename) { fFileList->Add(new TObjString(filename)); }
	46	void AddPercentileFile2ToList(TString filename) { fFileList2->Add(new TObjString(filename)); }
	47
f75629b5	48	Float_t GetCorrV0(const AliESDEvent* esd, float &v0CorrResc);
62164ba5	49	Float_t GetCorrSPD2(Float_t spd2raw,Float_t zv);
62164ba5	50
dcd68d00	51	private:
68e344b7	52
	53	Int_t SetupRun(AliESDEvent* esd);
	54
dcd68d00	55	Int_t fDebug; // Debug flag
	56	TString fAnalysisInput; // "ESD", "AOD"
	57	Bool_t fIsMCInput; // true when input is MC
d15bf53f	58	TFile *fFile; // file that holds the centrality vs multiplicity 1d
68e344b7	59	TFile *fFile2; // file that holds the centrality vs multiplicity 2d
d15bf53f	60	TString fCentfilename; // name of this file 1d
68e344b7	61	TString fCentfilename2; // name of this file 2d
68e344b7	62
9f14d90a	63	TList* fFileList; //! list of input files names
9f14d90a	64	TList* fFileList2; //! list of input files 2 names
68e344b7	65	Int_t fCurrentRun; // current run number
62164ba5	66	Int_t fRunNo; // reference run number
d15bf53f	67
	68	Float_t fCentV0M; // percentile centrality from V0
	69	Float_t fCentFMD; // percentile centrality from FMD
	70	Float_t fCentTRK; // percentile centrality from tracks
	71	Float_t fCentTKL; // percentile centrality from tracklets
	72	Float_t fCentCL0; // percentile centrality from clusters in layer 0
be0d4e9b	73	Float_t fCentCL1; // percentile centrality from clusters in layer 0
d15bf53f	74	Float_t fCentV0MvsFMD; // percentile centrality from V0 vs FMD
	75	Float_t fCentTKLvsV0M; // percentile centrality from tracklets vs V0
	76	Float_t fCentZEMvsZDC; // percentile centrality from ZEM vs ZDC
	77
9f14d90a	78	TH1F *fHtempV0M; // histogram with centrality vs multiplicity using V0
	79	TH1F *fHtempFMD; // histogram with centrality vs multiplicity using FMD
	80	TH1F *fHtempTRK; // histogram with centrality vs multiplicity using tracks
	81	TH1F *fHtempTKL; // histogram with centrality vs multiplicity using tracklets
	82	TH1F *fHtempCL0; // histogram with centrality vs multiplicity using clusters in layer 0
	83	TH1F *fHtempCL1; // histogram with centrality vs multiplicity using clusters in layer 0
	84	TH1F *fHtempV0MvsFMD; // histogram with centrality vs multiplicity using V0 vs FMD
	85	TH1F *fHtempTKLvsV0M; // histogram with centrality vs multiplicity using tracklets vs V0
	86	TH1F *fHtempZEMvsZDC; // histogram with centrality vs multiplicity using ZEM vs ZDC
d15bf53f	87
dcd68d00	88	ClassDef(AliCentralitySelectionTask,1);
	89
	90	};
	91
	92	#endif
	93