[u/mrichter/AliRoot.git] / JETAN / AliJetESDReader.h

#ifndef ALIJETESDREADER_H
#define ALIJETESDREADER_H
 
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
 
//---------------------------------------------------------------------
// Jet ESD Reader 
// ESD reader for jet analysis
// Author: Mercedes Lopez Noriega (mercedes.lopez.noriega@cern.ch)
//=========================================================================
// Modified in order to use a fUnitArray object instead of a fMomentumArray
// Includes EMCal Geometry, fUnitArray, grid objects and tools for Hadron correction
// Author : magali.estienne@ires.in2p3.fr
//---------------------------------------------------------------------

#include "AliJetReader.h"
#include "AliJetUnitArray.h"
#include "AliJetGrid.h"
class AliJetESDReaderHeader;
class AliEMCALGeometry;
class AliJetDummyGeo;
class AliJetHadronCorrection;
class AliJetUnitArray;
class AliJetReaderHeader;
class AliESD;

class AliJetESDReader : public AliJetReader
{
 public: 
  AliJetESDReader();
  virtual ~AliJetESDReader();
  // Setters
  Bool_t FillMomentumArray(Int_t event); 
  void   OpenInputFiles();
  void   InitUnitArray();
  void   ConnectTree(TTree* tree);
  virtual void SetTPCGrid(AliJetGrid *grid)   {fTpcGrid = grid;}
  virtual void SetEMCalGrid(AliJetGrid *grid) {fEmcalGrid = grid;}
  // Correction of hadronic energy
  virtual void SetHadronCorrection(Int_t flag = 1) {fHCorrection = flag;}
  virtual void SetHadronCorrector(AliJetHadronCorrectionv1* corr) {fHadCorr = corr;}
 private:
  void SetEMCALGeometry();
  void InitParameters();
 protected:
  AliJetDummyGeo             *fGeom;             //! EMCAL Geometry 
  TChain                     *fChain;            //! chain for reconstructed tracks
  AliESD                     *fESD;              //! pointer to esd
  AliJetHadronCorrectionv1   *fHadCorr;          //! Pointer to Hadron Correction Object 
  AliJetGrid                 *fTpcGrid;          //! Pointer to grid object
  AliJetGrid                 *fEmcalGrid;        //! Pointer to grid object
  Float_t                     fPtCut;            // Pt cut for tracks to minimise background contribution
  Int_t                       fHCorrection;      // Hadron correction flag
  Int_t                       fNumUnits;         // Number of units in the unit object array
                                                 // (same as num towers in EMCAL)
  Int_t                       fDebug;            // Debug option
  Int_t                       fNIn;              // Number of Array filled in UnitArray
  Int_t                       fOpt;              // Detector to be used for jet reconstruction
  Int_t                       fNeta;             // Number of bins in eta of tpc grid
  Int_t                       fNphi;             // Number of bins in phi of tpc grid
  Bool_t                      fArrayInitialised; // To check that array of units is initialised
  

  ClassDef(AliJetESDReader,1)
};
 
#endif
Commit	Line	Data
99e5fe42	1	#ifndef ALIJETESDREADER_H
	2	#define ALIJETESDREADER_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
83a444b1	7	//---------------------------------------------------------------------
99e5fe42	8	// Jet ESD Reader
	9	// ESD reader for jet analysis
	10	// Author: Mercedes Lopez Noriega (mercedes.lopez.noriega@cern.ch)
b45b0c92	11	//=========================================================================
	12	// Modified in order to use a fUnitArray object instead of a fMomentumArray
	13	// Includes EMCal Geometry, fUnitArray, grid objects and tools for Hadron correction
	14	// Author : magali.estienne@ires.in2p3.fr
83a444b1	15	//---------------------------------------------------------------------
99e5fe42	16
99e5fe42	17	#include "AliJetReader.h"
b45b0c92	18	#include "AliJetUnitArray.h"
b45b0c92	19	#include "AliJetGrid.h"
99e5fe42	20	class AliJetESDReaderHeader;
b45b0c92	21	class AliEMCALGeometry;
5b81d1ce	22	class AliJetDummyGeo;
b45b0c92	23	class AliJetHadronCorrection;
	24	class AliJetUnitArray;
	25	class AliJetReaderHeader;
	26	class AliESD;
99e5fe42	27
	28	class AliJetESDReader : public AliJetReader
	29	{
	30	public:
	31	AliJetESDReader();
	32	virtual ~AliJetESDReader();
99e5fe42	33	// Setters
7d0f353c	34	Bool_t FillMomentumArray(Int_t event);
7d0f353c	35	void OpenInputFiles();
b45b0c92	36	void InitUnitArray();
7d0f353c	37	void ConnectTree(TTree* tree);
b45b0c92	38	virtual void SetTPCGrid(AliJetGrid *grid) {fTpcGrid = grid;}
	39	virtual void SetEMCalGrid(AliJetGrid *grid) {fEmcalGrid = grid;}
	40	// Correction of hadronic energy
	41	virtual void SetHadronCorrection(Int_t flag = 1) {fHCorrection = flag;}
	42	virtual void SetHadronCorrector(AliJetHadronCorrectionv1* corr) {fHadCorr = corr;}
	43	private:
	44	void SetEMCALGeometry();
	45	void InitParameters();
99e5fe42	46	protected:
19e6695b	47	AliJetDummyGeo *fGeom; //! EMCAL Geometry
eaabc21f	48	TChain *fChain; //! chain for reconstructed tracks
eaabc21f	49	AliESD *fESD; //! pointer to esd
19e6695b	50	AliJetHadronCorrectionv1 *fHadCorr; //! Pointer to Hadron Correction Object
	51	AliJetGrid *fTpcGrid; //! Pointer to grid object
	52	AliJetGrid *fEmcalGrid; //! Pointer to grid object
b45b0c92	53	Float_t fPtCut; // Pt cut for tracks to minimise background contribution
	54	Int_t fHCorrection; // Hadron correction flag
	55	Int_t fNumUnits; // Number of units in the unit object array
	56	// (same as num towers in EMCAL)
	57	Int_t fDebug; // Debug option
	58	Int_t fNIn; // Number of Array filled in UnitArray
	59	Int_t fOpt; // Detector to be used for jet reconstruction
	60	Int_t fNeta; // Number of bins in eta of tpc grid
	61	Int_t fNphi; // Number of bins in phi of tpc grid
	62	Bool_t fArrayInitialised; // To check that array of units is initialised
	63
	64
99e5fe42	65
	66	ClassDef(AliJetESDReader,1)
	67	};
	68
	69	#endif