[u/mrichter/AliRoot.git] / EVGEN / AliGenReaderTreeK.h

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

/* $Id$ */
//
// Realisation of AliGenReader to be used with AliGenExtFile
// It reads events from a kinematics TreeK.
// Author: andreas.morsch@cern.ch
//
#include "AliGenReader.h"
#include "AliStack.h"

class TFile;
class AliRunLoader;
class AliStack;
class TString;
class TObjArray;

class AliGenReaderTreeK : public AliGenReader
{
 public:
    AliGenReaderTreeK();
    AliGenReaderTreeK(const AliGenReaderTreeK &reader);
    virtual ~AliGenReaderTreeK();
    // Initialise 
    virtual void Init();
    // Read
    virtual Int_t NextEvent();
    virtual TParticle*  NextParticle();
    virtual void RewindEvent();
    virtual void SetOnlyPrimaries(Bool_t flag){fOnlyPrimaries = flag;}
    AliGenReaderTreeK & operator=(const AliGenReaderTreeK & rhs);
    void SetDirs(TObjArray* dirs){fDirs = dirs;} //sets array directories names
    void AddDir(const char* dirname);
    AliRunLoader * GetRunLoader() const {return fInRunLoader;}

 protected:
    Int_t             fNcurrent;          // points to the next entry
    Int_t             fNparticle;         // Next particle in list
    Int_t             fNp;                // number of particles
    AliRunLoader     *fInRunLoader;       //!Run Loader of the input event
    TFile            *fBaseFile;          //! pointer to base file
    AliStack         *fStack;             //! Particle stack
    Bool_t            fOnlyPrimaries;     // Flag indicating wether only primaries are read
    TObjArray        *fDirs;              //arry with directories to read data from
    Int_t             fCurrentDir;        //Number of current directory
    static const TString fgkEventFolderName;//!name of folder where event to read is mounted
    
    TString&   GetDirName(Int_t entry);
    TParticle* GetParticle(Int_t i);
 private:
    void Copy(TObject&) const;    
    ClassDef(AliGenReaderTreeK,1) // Read particles from TreeK
};

inline 
TParticle* AliGenReaderTreeK::GetParticle(Int_t i)
 {
  if (fStack && i<fNp) return fStack->Particle(i);
  return 0x0;
 }

#endif
Commit	Line	Data
	1	#ifndef ALIGENREADERTreeK_H
	2	#define ALIGENREADERTREEK_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7	//
	8	// Realisation of AliGenReader to be used with AliGenExtFile
	9	// It reads events from a kinematics TreeK.
	10	// Author: andreas.morsch@cern.ch
	11	//
	12	#include "AliGenReader.h"
	13	#include "AliStack.h"
	14
	15	class TFile;
	16	class AliRunLoader;
	17	class AliStack;
	18	class TString;
	19	class TObjArray;
	20
	21	class AliGenReaderTreeK : public AliGenReader
	22	{
	23	public:
	24	AliGenReaderTreeK();
	25	AliGenReaderTreeK(const AliGenReaderTreeK &reader);
	26	virtual ~AliGenReaderTreeK();
	27	// Initialise
	28	virtual void Init();
	29	// Read
	30	virtual Int_t NextEvent();
	31	virtual TParticle* NextParticle();
	32	virtual void RewindEvent();
	33	virtual void SetOnlyPrimaries(Bool_t flag){fOnlyPrimaries = flag;}
	34	AliGenReaderTreeK & operator=(const AliGenReaderTreeK & rhs);
	35	void SetDirs(TObjArray* dirs){fDirs = dirs;} //sets array directories names
	36	void AddDir(const char* dirname);
	37	AliRunLoader * GetRunLoader() const {return fInRunLoader;}
	38
	39	protected:
	40	Int_t fNcurrent; // points to the next entry
	41	Int_t fNparticle; // Next particle in list
	42	Int_t fNp; // number of particles
	43	AliRunLoader *fInRunLoader; //!Run Loader of the input event
	44	TFile *fBaseFile; //! pointer to base file
	45	AliStack *fStack; //! Particle stack
	46	Bool_t fOnlyPrimaries; // Flag indicating wether only primaries are read
	47	TObjArray *fDirs; //arry with directories to read data from
	48	Int_t fCurrentDir; //Number of current directory
	49	static const TString fgkEventFolderName;//!name of folder where event to read is mounted
	50
	51	TString& GetDirName(Int_t entry);
	52	TParticle* GetParticle(Int_t i);
	53	private:
	54	void Copy(TObject&) const;
	55	ClassDef(AliGenReaderTreeK,1) // Read particles from TreeK
	56	};
	57
	58	inline
	59	TParticle* AliGenReaderTreeK::GetParticle(Int_t i)
	60	{
	61	if (fStack && i<fNp) return fStack->Particle(i);
	62	return 0x0;
	63	}
	64
	65	#endif
	66
	67
	68
	69
	70
	71