[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 TRefArray;
class AliJetGrid;
class AliEMCALGeometry;
class AliJetDummyGeo;
class AliESD;
class AliESDEvent;

class AliJetFillUnitArrayTracks : public TTask
{
 public: 
  AliJetFillUnitArrayTracks();
  AliJetFillUnitArrayTracks(AliESDEvent *fESD);
  virtual ~AliJetFillUnitArrayTracks();
  
  // Setter
  void          SetReaderHeader(AliJetReaderHeader *readerHeader) {fReaderHeader = readerHeader;}
  void          SetGeom(AliJetDummyGeo *geom) {fGeom = geom;}
  void          SetMomentumArray(TClonesArray *momentumArray) {fMomentumArray = momentumArray;}
  void          SetUnitArray(TClonesArray *unitArray) {fUnitArray = unitArray;}
  void          SetRefArray(TRefArray *refArray) {fRefArray = refArray;}
  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);
  void          SetESD(AliESDEvent *esd) {fESD = esd;}
  void          SetGrid0(AliJetGrid *grid0){fGrid0 = grid0;}
  void          SetGrid1(AliJetGrid *grid1){fGrid1 = grid1;}
  void          SetGrid2(AliJetGrid *grid2){fGrid2 = grid2;}
  void          SetGrid3(AliJetGrid *grid3){fGrid3 = grid3;}
  void          SetGrid4(AliJetGrid *grid4){fGrid4 = grid4;}
  void          SetProcId(Bool_t id) {fProcId = id;}
  Bool_t        GetProcId() {return fProcId;}

  // Getter
  TClonesArray* GetUnitArray() {return fUnitArray;}
  TRefArray*    GetRefArray() {return fRefArray;}
  void          GetEtaPhiFromIndex(Int_t index,Float_t &eta,Float_t &phi);
  Int_t         GetNeta() {return fNeta;}
  Int_t         GetNphi() {return fNphi;}
  Int_t         GetHadCorrection() {return fHCorrection;}
  Int_t         GetMult() {return fNTracks;}
  Int_t         GetMultCut() {return fNTracksCut;}
  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                       fNTracks;         // Number of tracks stored in UnitArray
  Int_t                       fNTracksCut;      // Number of tracks stored in UnitArray with a pt cut 
  Int_t                       fOpt;             // Detector to be used for jet reconstruction
  Bool_t                      fDZ;              // Use or not dead zones
  Int_t                       fDebug;           // Debug option

  AliJetReaderHeader          *fReaderHeader;   // ReaderHeader
  TClonesArray                *fMomentumArray;  // MomentumArray
  TClonesArray                *fUnitArray;      // UnitArray
  TRefArray                   *fRefArray;       // UnitArray
  Bool_t                      fProcId;          // Bool_t for TProcessID synchronization  
  AliJetGrid                  *fTPCGrid;        // Define filled grid
  AliJetGrid                  *fEMCalGrid;      // Define filled grid
  AliJetDummyGeo              *fGeom;           // Define EMCal geometry
  AliESDEvent                 *fESD;            // ESD
  AliJetGrid                  *fGrid0;          // Grid used for dead zones definition
  AliJetGrid                  *fGrid1;          // Grid used for dead zones definition
  AliJetGrid                  *fGrid2;          // Grid used for dead zones definition
  AliJetGrid                  *fGrid3;          // Grid used for dead zones definition
  AliJetGrid                  *fGrid4;          // Grid used for dead zones definition

  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:
  AliJetFillUnitArrayTracks(const AliJetFillUnitArrayTracks &det);
  AliJetFillUnitArrayTracks &operator=(const AliJetFillUnitArrayTracks &det);
  void InitParameters();

  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
8838ab7a	20	class AliEMCALGeometry;
8838ab7a	21	class AliJetDummyGeo;
4a01bb2c	22	class AliJetHadronCorrection;
	23	class AliJetReader;
	24	class AliJetESDReader;
	25	class TClonesArray;
ee7de0dd	26	class TRefArray;
4a01bb2c	27	class AliJetGrid;
8838ab7a	28	class AliEMCALGeometry;
ee7de0dd	29	class AliJetDummyGeo;
	30	class AliESD;
	31	class AliESDEvent;
4a01bb2c	32
	33	class AliJetFillUnitArrayTracks : public TTask
	34	{
	35	public:
	36	AliJetFillUnitArrayTracks();
ee7de0dd	37	AliJetFillUnitArrayTracks(AliESDEvent *fESD);
4a01bb2c	38	virtual ~AliJetFillUnitArrayTracks();
	39
	40	// Setter
8838ab7a	41	void SetReaderHeader(AliJetReaderHeader *readerHeader) {fReaderHeader = readerHeader;}
	42	void SetGeom(AliJetDummyGeo *geom) {fGeom = geom;}
	43	void SetMomentumArray(TClonesArray *momentumArray) {fMomentumArray = momentumArray;}
	44	void SetUnitArray(TClonesArray *unitArray) {fUnitArray = unitArray;}
	45	void SetRefArray(TRefArray *refArray) {fRefArray = refArray;}
	46	void SetHadCorrection(Int_t flag = 1) {fHCorrection = flag;}
	47	void SetHadCorrector(AliJetHadronCorrectionv1* corr) {fHadCorr = corr;}
	48	void SetTPCGrid(AliJetGrid *grid) {fTPCGrid = grid;}
	49	void SetEMCalGrid(AliJetGrid *grid) {fEMCalGrid = grid;}
	50	void SetGrid(Double_t phiMin,Double_t phiMax,Double_t etaMin,Double_t etaMax);
	51	void SetESD(AliESDEvent *esd) {fESD = esd;}
	52	void SetGrid0(AliJetGrid *grid0){fGrid0 = grid0;}
	53	void SetGrid1(AliJetGrid *grid1){fGrid1 = grid1;}
	54	void SetGrid2(AliJetGrid *grid2){fGrid2 = grid2;}
	55	void SetGrid3(AliJetGrid *grid3){fGrid3 = grid3;}
	56	void SetGrid4(AliJetGrid *grid4){fGrid4 = grid4;}
	57	void SetProcId(Bool_t id) {fProcId = id;}
	58	Bool_t GetProcId() {return fProcId;}
4a01bb2c	59
4a01bb2c	60	// Getter
ee7de0dd	61	TClonesArray* GetUnitArray() {return fUnitArray;}
ee7de0dd	62	TRefArray* GetRefArray() {return fRefArray;}
ee7de0dd	63	void GetEtaPhiFromIndex(Int_t index,Float_t &eta,Float_t &phi);
8838ab7a	64	Int_t GetNeta() {return fNeta;}
	65	Int_t GetNphi() {return fNphi;}
	66	Int_t GetHadCorrection() {return fHCorrection;}
	67	Int_t GetMult() {return fNTracks;}
	68	Int_t GetMultCut() {return fNTracksCut;}
ee7de0dd	69	void Exec(Option_t*);
4a01bb2c	70
4a01bb2c	71	protected:
8838ab7a	72	Int_t fNumUnits; // Number of units in the unit object array (same as num towers in EMCAL)
	73	Float_t fEtaMinCal; // Define EMCal acceptance in Eta
	74	Float_t fEtaMaxCal; // Define EMCal acceptance in Eta
	75	Float_t fPhiMinCal; // Define EMCal acceptance in Phi
	76	Float_t fPhiMaxCal; // Define EMCal acceptance in Phi
4a01bb2c	77	AliJetHadronCorrectionv1 *fHadCorr; // Pointer to Hadron Correction Object
8838ab7a	78	Int_t fHCorrection; // Hadron correction flag
ee7de0dd	79	Int_t fNTracks; // Number of tracks stored in UnitArray
ee7de0dd	80	Int_t fNTracksCut; // Number of tracks stored in UnitArray with a pt cut
4a01bb2c	81	Int_t fOpt; // Detector to be used for jet reconstruction
ee7de0dd	82	Bool_t fDZ; // Use or not dead zones
4a01bb2c	83	Int_t fDebug; // Debug option
	84
	85	AliJetReaderHeader *fReaderHeader; // ReaderHeader
	86	TClonesArray *fMomentumArray; // MomentumArray
ee7de0dd	87	TClonesArray *fUnitArray; // UnitArray
ee7de0dd	88	TRefArray *fRefArray; // UnitArray
8838ab7a	89	Bool_t fProcId; // Bool_t for TProcessID synchronization
4a01bb2c	90	AliJetGrid *fTPCGrid; // Define filled grid
4a01bb2c	91	AliJetGrid *fEMCalGrid; // Define filled grid
ee7de0dd	92	AliJetDummyGeo *fGeom; // Define EMCal geometry
ee7de0dd	93	AliESDEvent *fESD; // ESD
8838ab7a	94	AliJetGrid *fGrid0; // Grid used for dead zones definition
	95	AliJetGrid *fGrid1; // Grid used for dead zones definition
	96	AliJetGrid *fGrid2; // Grid used for dead zones definition
	97	AliJetGrid *fGrid3; // Grid used for dead zones definition
	98	AliJetGrid *fGrid4; // Grid used for dead zones definition
4a01bb2c	99
8838ab7a	100	Int_t fNphi; // number of points in the grid: phi
	101	Int_t fNeta; // " eta
	102	TArrayD *fPhi2; // grid points in phi
	103	TArrayD *fEta2; // grid points in eta
	104	TArrayD *fPhi; // grid points in phi
	105	TArrayD *fEta; // grid points in eta
	106	TMatrixD *fIndex; // grid points in (phi,eta)
	107	TMatrixD *fParams; // matrix of parameters in the grid points
	108	Int_t fGrid; // Select the grid acceptance you want to fill
	109	// 0 = TPC acceptance, 1 = TPC-EMCal acceptance
	110	Float_t fPhiMin;
	111	Float_t fPhiMax;
	112	Float_t fEtaMin;
	113	Float_t fEtaMax;
	114	Int_t fEtaBinInTPCAcc;
	115	Int_t fPhiBinInTPCAcc;
	116	Int_t fEtaBinInEMCalAcc;
	117	Int_t fPhiBinInEMCalAcc;
	118	Int_t fNbinPhi;
4a01bb2c	119
4a01bb2c	120	private:
a1fa3a77	121	AliJetFillUnitArrayTracks(const AliJetFillUnitArrayTracks &det);
a1fa3a77	122	AliJetFillUnitArrayTracks &operator=(const AliJetFillUnitArrayTracks &det);
4a01bb2c	123	void InitParameters();
4a01bb2c	124
4a01bb2c	125	ClassDef(AliJetFillUnitArrayTracks,1) // Fill Unit Array with tpc and/or emcal information
	126	};
	127
	128	#endif