[u/mrichter/AliRoot.git] / EMCAL / AliEMCALJetFinderInput.h

#ifndef ALIEMCALJETFINDERINPUT_H
#define ALIEMCALJETFINDERINPUT_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *  *  * See cxx source for full Copyright notice     */

/* $Id$ */

//_________________________________________________________________________
//  Initial input object for jetfinder
//
//*-- Author: Mark Horner (LBL/UCT)
//
//


#include "TObject.h"
#include "TParticle.h"
#include "AliEMCALParton.h"
#include "AliEMCALDigit.h"
#include "AliEMCALJetFinderTypes.h"

class AliEMCALJetFinderInput : public TObject
{
	public:
		AliEMCALJetFinderInput();
		~AliEMCALJetFinderInput();
		void Reset(AliEMCALJetFinderResetType_t resettype);
		void SetDebug(Int_t debug=0){fDebug = debug;}
		void AddEnergyToDigit(Int_t digitID,Int_t denergy); 
		void AddTrack(TParticle track);
		void AddParton(AliEMCALParton *parton);
		void AddParticle(TParticle *particle);
		AliEMCALDigit* GetDigit(Int_t digitID);
		Int_t GetNDigits() const {return fNDigits;}
		TParticle* GetTrack(Int_t trackID);
		Int_t GetNTracks() const {return fNTracks;}
		AliEMCALParton* GetParton(Int_t partonID);
		Int_t GetNPartons() const {return fNPartons;}
		TParticle* GetParticle(Int_t particleID);
		Int_t GetNParticles() const {return fNParticles;}

	private:
		void InitArrays();
		AliEMCALDigit	fDigitsArray[13824];	// This is the digits array for the EMCAL
		Int_t		fNDigits;     	// This is the number of digits
		Int_t		fNMaxDigits;  	// This is the max number of digits
		TParticle	fTracksArray[3000]; 	// This is the track array 
		Int_t		fNTracks;     	// This stores the number of tracks	
		Int_t		fNMaxTracks;	// This stores the maximum number of tracks
		AliEMCALParton	fPartonsArray[4];  //  This is the partons array
		Int_t		fNPartons;	// This stores the number of partons
		Int_t		fNMaxPartons;	  // This stores the maximum number of partons
		TParticle	fParticlesArray[2000];// This stores the particles	
		Int_t		fNParticles;	// This stores the number of particles
	       	Int_t		fNMaxParticles; // This stroes the maximum number of particles
		Int_t		fDebug;		// This is the debug value 
		Bool_t		fInitialised;	// Stores whether or not the arrays have been initialised 
		
	ClassDef(AliEMCALJetFinderInput,3)
};
#endif
Commit	Line	Data
f7d5860b	1	#ifndef ALIEMCALJETFINDERINPUT_H
	2	#define ALIEMCALJETFINDERINPUT_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* * * See cxx source for full Copyright notice */
	6
	7	/* $Id$ */
	8
	9	//_________________________________________________________________________
	10	// Initial input object for jetfinder
	11	//
	12	//*-- Author: Mark Horner (LBL/UCT)
	13	//
	14	//
	15
	16
	17	#include "TObject.h"
	18	#include "TParticle.h"
f7d5860b	19	#include "AliEMCALParton.h"
	20	#include "AliEMCALDigit.h"
	21	#include "AliEMCALJetFinderTypes.h"
	22
	23	class AliEMCALJetFinderInput : public TObject
	24	{
	25	public:
	26	AliEMCALJetFinderInput();
	27	~AliEMCALJetFinderInput();
	28	void Reset(AliEMCALJetFinderResetType_t resettype);
	29	void SetDebug(Int_t debug=0){fDebug = debug;}
	30	void AddEnergyToDigit(Int_t digitID,Int_t denergy);
	31	void AddTrack(TParticle track);
	32	void AddParton(AliEMCALParton *parton);
	33	void AddParticle(TParticle *particle);
	34	AliEMCALDigit* GetDigit(Int_t digitID);
3278aef8	35	Int_t GetNDigits() const {return fNDigits;}
f7d5860b	36	TParticle* GetTrack(Int_t trackID);
3278aef8	37	Int_t GetNTracks() const {return fNTracks;}
f7d5860b	38	AliEMCALParton* GetParton(Int_t partonID);
3278aef8	39	Int_t GetNPartons() const {return fNPartons;}
f7d5860b	40	TParticle* GetParticle(Int_t particleID);
3278aef8	41	Int_t GetNParticles() const {return fNParticles;}
f7d5860b	42
	43	private:
	44	void InitArrays();
ece5da38	45	AliEMCALDigit fDigitsArray[13824]; // This is the digits array for the EMCAL
f7d5860b	46	Int_t fNDigits; // This is the number of digits
f7d5860b	47	Int_t fNMaxDigits; // This is the max number of digits
ece5da38	48	TParticle fTracksArray[3000]; // This is the track array
f7d5860b	49	Int_t fNTracks; // This stores the number of tracks
f7d5860b	50	Int_t fNMaxTracks; // This stores the maximum number of tracks
ece5da38	51	AliEMCALParton fPartonsArray[4]; // This is the partons array
f7d5860b	52	Int_t fNPartons; // This stores the number of partons
ece5da38	53	Int_t fNMaxPartons; // This stores the maximum number of partons
ece5da38	54	TParticle fParticlesArray[2000];// This stores the particles
f7d5860b	55	Int_t fNParticles; // This stores the number of particles
	56	Int_t fNMaxParticles; // This stroes the maximum number of particles
	57	Int_t fDebug; // This is the debug value
	58	Bool_t fInitialised; // Stores whether or not the arrays have been initialised
	59
58ea668c	60	ClassDef(AliEMCALJetFinderInput,3)
f7d5860b	61	};
f7d5860b	62	#endif