o add possibility to switch off Material budget correction
[u/mrichter/AliRoot.git] / TPC / Upgrade / AliToyMCEventGenerator.h
CommitLineData
de0014b7 1#ifndef AliToyMCEventGenerator_H
2#define AliToyMCEventGenerator_H
3
a1a695e5 4#include <TString.h>
5
6class TFile;
7class TTree;
8
9class AliTPCParam;
10class AliTPCSpaceCharge3D;
11
12class AliToyMCTrack;
13class AliToyMCEvent;
de0014b7 14
de0014b7 15class AliToyMCEventGenerator : public TObject {
16 public:
17 AliToyMCEventGenerator();
18 AliToyMCEventGenerator(const AliToyMCEventGenerator &gen);
19 virtual ~AliToyMCEventGenerator();
20
21 virtual AliToyMCEvent* Generate(Double_t time) = 0;
22
23 Bool_t DistortTrack(AliToyMCTrack &trackIn, Double_t t0);
24
a1a695e5 25 void SetOutputFileName(const char* file) { fOutputFileName=file; }
26 const char* GetOutputFileName() const { return fOutputFileName.Data(); }
de0014b7 27 protected:
28 AliTPCParam *fTPCParam;
a1a695e5 29 AliToyMCEvent *fEvent;
30
31 Bool_t ConnectOutputFile();
32 Bool_t CloseOutputFile();
33 void FillTree();
de0014b7 34
35 private:
a1a695e5 36 AliToyMCEventGenerator& operator= (const AliToyMCEventGenerator& );
37
de0014b7 38 AliTPCSpaceCharge3D *fSpaceCharge;
a1a695e5 39
40 TString fOutputFileName;
41 TFile *fOutFile;
42 TTree *fOutTree;
de0014b7 43
44 ClassDef(AliToyMCEventGenerator, 1)
45
46};
47
48#endif
49