TRD/AliTRDgtuSim.h

   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;
  21 class AliTRDfeeParam;
  22 class TTree;
  23 class TList;
  24
  25 class AliTRDgtuSim : public TObject {
  26  public:
  27   AliTRDgtuSim(AliRunLoader *rl = 0x0);
  28   ~AliTRDgtuSim();
  29
  30   Bool_t LoadTracklets(AliLoader * const loader);
  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; }
  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);
  40   Bool_t WriteTracksToLoader();
  41
  42  protected:
  43   AliRunLoader  *fRunLoader;    //!
  44   AliTRDfeeParam *fFeeParam;    //!
  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