[u/mrichter/AliRoot.git] / TRD / AliTRDgtuSim.h

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

/* $Id: AliTRDgtuSim.h 27496 2008-07-22 08:35:45Z cblume $ */

// --------------------------------------------------------
// 
// GTU simulation
//
// --------------------------------------------------------

#include "TObject.h"

class AliRunLoader;
class AliLoader;
class AliESDEvent;

class AliTRDgtuTMU;
class AliTRDfeeParam;
class TTree;
class TList;

class AliTRDgtuSim : public TObject {
 public:
  AliTRDgtuSim(AliRunLoader *rl = 0x0);
  ~AliTRDgtuSim();

  Bool_t LoadTracklets(AliLoader * const loader);

  Bool_t RunGTU(AliLoader *loader, AliESDEvent *esd = 0x0);
  Bool_t RunGTUFromTrackletFile(TString filename, Int_t event, Int_t noev = 1);

  TTree* GetTreeOfTracks() { return fTrackTree; }
  Bool_t WriteTracksToTree(TList *listOfTracks, Int_t event = 0); 
  Bool_t WriteTracksToDataFile(TList *listOfTracks, Int_t event);
  Bool_t WriteTreesToFile() const;
  Bool_t WriteTracksToESD(const TList *const listOfTracks, AliESDEvent *esd);
  Bool_t WriteTracksToLoader(const TList *const listOfTracks);

 protected:
  AliRunLoader 	*fRunLoader;  	//!
  AliTRDfeeParam *fFeeParam;    //!
  AliTRDgtuTMU 	*fTMU; 		// pointer to TMU simulation class
  TClonesArray 	*fTrackletArray;	// array of tracklets
  TTree 	*fTrackTree; 	// tree to hold the tracks of one event, used for writing in WriteTracksToFile()
  TTree         *fTrackletTree; // tree to hold the gtu tracklets

 private:
  AliTRDgtuSim& operator=(const AliTRDgtuSim &rhs); // not implemented
  AliTRDgtuSim(const AliTRDgtuSim &rhs); // not implemented

  ClassDef(AliTRDgtuSim, 1);
};

#endif
Commit	Line	Data
52c19022	1	#ifndef ALITRDGTUSIM_H
	2	#define ALITRDGTUSIM_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id: AliTRDgtuSim.h 27496 2008-07-22 08:35:45Z cblume $ */
	7
	8	// --------------------------------------------------------
	9	//
	10	// GTU simulation
	11	//
	12	// --------------------------------------------------------
	13
	14	#include "TObject.h"
	15
	16	class AliRunLoader;
	17	class AliLoader;
	18	class AliESDEvent;
	19
	20	class AliTRDgtuTMU;
c1ad4629	21	class AliTRDfeeParam;
52c19022	22	class TTree;
	23	class TList;
	24
	25	class AliTRDgtuSim : public TObject {
	26	public:
	27	AliTRDgtuSim(AliRunLoader *rl = 0x0);
	28	~AliTRDgtuSim();
	29
36dc3337	30	Bool_t LoadTracklets(AliLoader * const loader);
52c19022	31
	32	Bool_t RunGTU(AliLoader loader, AliESDEvent esd = 0x0);
	33	Bool_t RunGTUFromTrackletFile(TString filename, Int_t event, Int_t noev = 1);
	34
	35	TTree* GetTreeOfTracks() { return fTrackTree; }
36dc3337	36	Bool_t WriteTracksToTree(TList *listOfTracks, Int_t event = 0);
	37	Bool_t WriteTracksToDataFile(TList *listOfTracks, Int_t event);
	38	Bool_t WriteTreesToFile() const;
	39	Bool_t WriteTracksToESD(const TList const listOfTracks, AliESDEvent esd);
2cf67435	40	Bool_t WriteTracksToLoader(const TList *const listOfTracks);
52c19022	41
	42	protected:
	43	AliRunLoader *fRunLoader; //!
c1ad4629	44	AliTRDfeeParam *fFeeParam; //!
52c19022	45	AliTRDgtuTMU *fTMU; // pointer to TMU simulation class
	46	TClonesArray *fTrackletArray; // array of tracklets
	47	TTree *fTrackTree; // tree to hold the tracks of one event, used for writing in WriteTracksToFile()
	48	TTree *fTrackletTree; // tree to hold the gtu tracklets
	49
	50	private:
	51	AliTRDgtuSim& operator=(const AliTRDgtuSim &rhs); // not implemented
	52	AliTRDgtuSim(const AliTRDgtuSim &rhs); // not implemented
	53
	54	ClassDef(AliTRDgtuSim, 1);
	55	};
	56
	57	#endif