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

#ifndef ALIJETFILLUNITARRAYTRACKS_H
#define ALIJETFILLUNITARRAYTRACKS_H
 
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
 
//---------------------------------------------------------------------
// Jet Fill Unit Array 
// Called by ESD Reader for jet analysis
// Author: Magali Estienne (magali.estienne@subatech.in2p3.fr)
//---------------------------------------------------------------------

#ifndef ROOT_TTask
#include "TTask.h"
#endif

#include <TMatrixD.h>
#include <TArrayD.h>

class AliEMCALGeometry;
class AliJetDummyGeo;
class AliJetHadronCorrection;
class AliJetReader;
class AliJetESDReader;
class TClonesArray;
class AliJetUnitArray;
//class AliJetReaderHeader;
class AliJetReader;
class AliJetGrid;

class AliJetFillUnitArrayTracks : public TTask
{
 public: 
  AliJetFillUnitArrayTracks();
  virtual ~AliJetFillUnitArrayTracks();
  
  // Setter
  void SetReaderHeader(AliJetReaderHeader *readerHeader) {fReaderHeader = readerHeader;}
  void SetMomentumArray(TClonesArray *momentumArray) {fMomentumArray = momentumArray;}
  void SetUnitArray(AliJetUnitArray *unitArray) {fUnitArray = unitArray;}
  void SetHadCorrection(Int_t flag = 1) {fHCorrection = flag;}
  void SetHadCorrector(AliJetHadronCorrectionv1* corr) {fHadCorr = corr;}
  void SetTPCGrid(AliJetGrid *grid) {fTPCGrid = grid;}
  void SetEMCalGrid(AliJetGrid *grid) {fEMCalGrid = grid;}
  void SetGrid(Double_t phiMin,Double_t phiMax,Double_t etaMin,Double_t etaMax);

  // Getter
  AliJetUnitArray* GetUnitArray() {return fUnitArray;}
  //  Int_t GetIndexFromEtaPhi(Double_t eta,Double_t phi) const;
  void  GetEtaPhiFromIndex(Int_t index,Float_t &eta,Float_t &phi);
  Int_t GetNeta() {return fNeta;}
  Int_t GetNphi() {return fNphi;}

  void Exec(Option_t*);

 protected:
  Int_t   fNumUnits;      // Number of units in the unit object array (same as num towers in EMCAL)
  Float_t fEtaMinCal;     // Define EMCal acceptance in Eta
  Float_t fEtaMaxCal;     // Define EMCal acceptance in Eta
  Float_t fPhiMinCal;     // Define EMCal acceptance in Phi
  Float_t fPhiMaxCal;     // Define EMCal acceptance in Phi
  AliJetHadronCorrectionv1   *fHadCorr;         // Pointer to Hadron Correction Object
  Int_t                       fHCorrection;     // Hadron correction flag
  Int_t                       fNIn;             // Number of Array filled in UnitArray
  Int_t                       fOpt;             // Detector to be used for jet reconstruction
  Int_t                       fDebug;           // Debug option

  AliJetReaderHeader          *fReaderHeader;   // ReaderHeader
  TClonesArray                *fMomentumArray;  // MomentumArray
  AliJetUnitArray             *fUnitArray;      // UnitArray
  AliJetGrid                  *fTPCGrid;        // Define filled grid
  AliJetGrid                  *fEMCalGrid;      // Define filled grid

  // geometry info
  AliJetDummyGeo              *fGeom;           //!

  Int_t     fNphi;                    // number of points in the grid:   phi
  Int_t     fNeta;                    //               "                 eta
  TArrayD*  fPhi2;                    // grid points in phi
  TArrayD*  fEta2;                    // grid points in eta
  TArrayD*  fPhi;                     // grid points in phi
  TArrayD*  fEta;                     // grid points in eta
  TMatrixD* fIndex;                   // grid points in (phi,eta) 
  TMatrixD* fParams;                  // matrix of parameters in the grid points  
  Int_t     fGrid;                    // Select the grid acceptance you want to fill
                                      // 0 = TPC acceptance, 1 = TPC-EMCal acceptance
  Float_t   fPhiMin;
  Float_t   fPhiMax;
  Float_t   fEtaMin;
  Float_t   fEtaMax;
  Int_t     fEtaBinInTPCAcc;
  Int_t     fPhiBinInTPCAcc;
  Int_t     fEtaBinInEMCalAcc;
  Int_t     fPhiBinInEMCalAcc;
  Int_t     fNbinPhi;

 private:
  void SetEMCALGeometry();
  void InitParameters();

  //  Int_t fEvent;
  
  ClassDef(AliJetFillUnitArrayTracks,1) // Fill Unit Array with tpc and/or emcal information
};

#endif
Commit	Line	Data
4a01bb2c	1	#ifndef ALIJETFILLUNITARRAYTRACKS_H
	2	#define ALIJETFILLUNITARRAYTRACKS_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	//---------------------------------------------------------------------
	8	// Jet Fill Unit Array
	9	// Called by ESD Reader for jet analysis
	10	// Author: Magali Estienne (magali.estienne@subatech.in2p3.fr)
	11	//---------------------------------------------------------------------
	12
	13	#ifndef ROOT_TTask
	14	#include "TTask.h"
	15	#endif
	16
	17	#include <TMatrixD.h>
	18	#include <TArrayD.h>
	19
	20	class AliEMCALGeometry;
5b81d1ce	21	class AliJetDummyGeo;
4a01bb2c	22	class AliJetHadronCorrection;
	23	class AliJetReader;
	24	class AliJetESDReader;
	25	class TClonesArray;
	26	class AliJetUnitArray;
	27	//class AliJetReaderHeader;
	28	class AliJetReader;
	29	class AliJetGrid;
	30
	31	class AliJetFillUnitArrayTracks : public TTask
	32	{
	33	public:
	34	AliJetFillUnitArrayTracks();
	35	virtual ~AliJetFillUnitArrayTracks();
	36
	37	// Setter
	38	void SetReaderHeader(AliJetReaderHeader *readerHeader) {fReaderHeader = readerHeader;}
	39	void SetMomentumArray(TClonesArray *momentumArray) {fMomentumArray = momentumArray;}
	40	void SetUnitArray(AliJetUnitArray *unitArray) {fUnitArray = unitArray;}
	41	void SetHadCorrection(Int_t flag = 1) {fHCorrection = flag;}
	42	void SetHadCorrector(AliJetHadronCorrectionv1* corr) {fHadCorr = corr;}
	43	void SetTPCGrid(AliJetGrid *grid) {fTPCGrid = grid;}
	44	void SetEMCalGrid(AliJetGrid *grid) {fEMCalGrid = grid;}
	45	void SetGrid(Double_t phiMin,Double_t phiMax,Double_t etaMin,Double_t etaMax);
	46
	47	// Getter
	48	AliJetUnitArray* GetUnitArray() {return fUnitArray;}
	49	// Int_t GetIndexFromEtaPhi(Double_t eta,Double_t phi) const;
	50	void GetEtaPhiFromIndex(Int_t index,Float_t &eta,Float_t &phi);
	51	Int_t GetNeta() {return fNeta;}
	52	Int_t GetNphi() {return fNphi;}
	53
	54	void Exec(Option_t*);
	55
	56	protected:
	57	Int_t fNumUnits; // Number of units in the unit object array (same as num towers in EMCAL)
	58	Float_t fEtaMinCal; // Define EMCal acceptance in Eta
	59	Float_t fEtaMaxCal; // Define EMCal acceptance in Eta
	60	Float_t fPhiMinCal; // Define EMCal acceptance in Phi
	61	Float_t fPhiMaxCal; // Define EMCal acceptance in Phi
	62	AliJetHadronCorrectionv1 *fHadCorr; // Pointer to Hadron Correction Object
b45b0c92	63	Int_t fHCorrection; // Hadron correction flag
4a01bb2c	64	Int_t fNIn; // Number of Array filled in UnitArray
	65	Int_t fOpt; // Detector to be used for jet reconstruction
	66	Int_t fDebug; // Debug option
	67
	68	AliJetReaderHeader *fReaderHeader; // ReaderHeader
	69	TClonesArray *fMomentumArray; // MomentumArray
	70	AliJetUnitArray *fUnitArray; // UnitArray
	71	AliJetGrid *fTPCGrid; // Define filled grid
	72	AliJetGrid *fEMCalGrid; // Define filled grid
	73
	74	// geometry info
5b81d1ce	75	AliJetDummyGeo *fGeom; //!
4a01bb2c	76
	77	Int_t fNphi; // number of points in the grid: phi
	78	Int_t fNeta; // " eta
	79	TArrayD* fPhi2; // grid points in phi
	80	TArrayD* fEta2; // grid points in eta
	81	TArrayD* fPhi; // grid points in phi
	82	TArrayD* fEta; // grid points in eta
	83	TMatrixD* fIndex; // grid points in (phi,eta)
	84	TMatrixD* fParams; // matrix of parameters in the grid points
	85	Int_t fGrid; // Select the grid acceptance you want to fill
	86	// 0 = TPC acceptance, 1 = TPC-EMCal acceptance
	87	Float_t fPhiMin;
	88	Float_t fPhiMax;
	89	Float_t fEtaMin;
	90	Float_t fEtaMax;
	91	Int_t fEtaBinInTPCAcc;
	92	Int_t fPhiBinInTPCAcc;
	93	Int_t fEtaBinInEMCalAcc;
	94	Int_t fPhiBinInEMCalAcc;
	95	Int_t fNbinPhi;
	96
	97	private:
	98	void SetEMCALGeometry();
	99	void InitParameters();
	100
	101	// Int_t fEvent;
	102
	103	ClassDef(AliJetFillUnitArrayTracks,1) // Fill Unit Array with tpc and/or emcal information
	104	};
	105
	106	#endif