[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 TH2F;
class TList;
class TObjString;

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)); }

 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

  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

  TH1D    *fHtempV0M;           // histogram with centrality vs multiplicity using V0
  TH1D    *fHtempFMD;           // histogram with centrality vs multiplicity using FMD
  TH1D    *fHtempTRK;           // histogram with centrality vs multiplicity using tracks
  TH1D    *fHtempTKL;           // histogram with centrality vs multiplicity using tracklets
  TH1D    *fHtempCL0;           // histogram with centrality vs multiplicity using clusters in layer 0
  TH1D    *fHtempCL1;           // histogram with centrality vs multiplicity using clusters in layer 0
  TH1D    *fHtempV0MvsFMD;           // histogram with centrality vs multiplicity using V0 vs FMD   
  TH1D    *fHtempTKLvsV0M;           // histogram with centrality vs multiplicity using tracklets vs V0
  TH1D    *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;
	15	class TH2F;
68e344b7	16	class TList;
	17	class TObjString;
	18
	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
dcd68d00	48	private:
68e344b7	49
	50	Int_t SetupRun(AliESDEvent* esd);
	51
dcd68d00	52	Int_t fDebug; // Debug flag
	53	TString fAnalysisInput; // "ESD", "AOD"
	54	Bool_t fIsMCInput; // true when input is MC
d15bf53f	55	TFile *fFile; // file that holds the centrality vs multiplicity 1d
68e344b7	56	TFile *fFile2; // file that holds the centrality vs multiplicity 2d
d15bf53f	57	TString fCentfilename; // name of this file 1d
68e344b7	58	TString fCentfilename2; // name of this file 2d
	59
	60	TList* fFileList; // list of input files names
	61	TList* fFileList2; // list of input files 2 names
	62	Int_t fCurrentRun; // current run number
d15bf53f	63
	64	Float_t fCentV0M; // percentile centrality from V0
	65	Float_t fCentFMD; // percentile centrality from FMD
	66	Float_t fCentTRK; // percentile centrality from tracks
	67	Float_t fCentTKL; // percentile centrality from tracklets
	68	Float_t fCentCL0; // percentile centrality from clusters in layer 0
be0d4e9b	69	Float_t fCentCL1; // percentile centrality from clusters in layer 0
d15bf53f	70	Float_t fCentV0MvsFMD; // percentile centrality from V0 vs FMD
	71	Float_t fCentTKLvsV0M; // percentile centrality from tracklets vs V0
	72	Float_t fCentZEMvsZDC; // percentile centrality from ZEM vs ZDC
	73
	74	TH1D *fHtempV0M; // histogram with centrality vs multiplicity using V0
	75	TH1D *fHtempFMD; // histogram with centrality vs multiplicity using FMD
	76	TH1D *fHtempTRK; // histogram with centrality vs multiplicity using tracks
	77	TH1D *fHtempTKL; // histogram with centrality vs multiplicity using tracklets
	78	TH1D *fHtempCL0; // histogram with centrality vs multiplicity using clusters in layer 0
be0d4e9b	79	TH1D *fHtempCL1; // histogram with centrality vs multiplicity using clusters in layer 0
d15bf53f	80	TH1D *fHtempV0MvsFMD; // histogram with centrality vs multiplicity using V0 vs FMD
	81	TH1D *fHtempTKLvsV0M; // histogram with centrality vs multiplicity using tracklets vs V0
	82	TH1D *fHtempZEMvsZDC; // histogram with centrality vs multiplicity using ZEM vs ZDC
	83
dcd68d00	84	ClassDef(AliCentralitySelectionTask,1);
	85
	86	};
	87
	88	#endif
	89