Merge branch 'TRDdev' into TPCdev

[u/mrichter/AliRoot.git] / TPC / Upgrade / AliToyMCEventGeneratorSimple.h

diff --git a/TPC/Upgrade/AliToyMCEventGeneratorSimple.h b/TPC/Upgrade/AliToyMCEventGeneratorSimple.h
index e80b0cf724f7677122e8641ada29241a407938b0..89f6d82fd7d535c8020c18e2b98212208b8623aa 100644 (file)
--- a/TPC/Upgrade/AliToyMCEventGeneratorSimple.h
+++ b/TPC/Upgrade/AliToyMCEventGeneratorSimple.h
@@ -7,8 +7,9 @@
 #include <TString.h>
 class AliToyMCEvent;
 class AliESDtrackCuts;
-
-
+class TTree;
+class TFile;
+class TH1F;
 class AliToyMCEventGeneratorSimple : public AliToyMCEventGenerator {
  public:
   AliToyMCEventGeneratorSimple();
@@ -18,11 +19,20 @@ class AliToyMCEventGeneratorSimple : public AliToyMCEventGenerator {
 
   AliToyMCEvent* Generate(Double_t time);
   AliToyMCEvent* GenerateESD(AliESDEvent& esdEvent, Double_t time);
-  void SetParametersSimple(Double_t vertexMean, Double_t vertexSigma);
+  AliToyMCEvent* GenerateESD2(Double_t time);
+  AliToyMCEvent* GenerateLaser(Double_t time);
+  
+  void SetParametersToyGen(const Char_t* parfilename="$ALICE_ROOT/TPC/Upgrade/files/params.root", Double_t vertexMean = 0., Double_t vertexSigma = 7.);
+  void RunSimulation(Int_t nevents=10, Int_t ntracks=20, Int_t rate=50);
+  void RunSimulationBunchTrain(Int_t nevents=10, Int_t ntracks=20);
+  void RunSimulationESD(Int_t nevents=10, Int_t ntracks=20);
+  void RunSimulationLaser(Int_t nevents=1);
   
-  void RunSimulationSimple(const Int_t nevents=10, const Int_t ntracks=20);
-  void RunSimulationESD(const Int_t nevents=10, const Int_t ntracks=20);
   void SetInputESD(const Char_t* filename) {fInputFileNameESD = filename;}
+  Int_t OpenInputAndGetMaxEvents(Int_t type, Int_t nevents);
+  void RunSimulation2(Bool_t equalspacing, Int_t type, Int_t nevents, Int_t ntracks);
+  void GetNGeneratedEventsAndSpacing(Bool_t equalSpacing, Int_t &ngen, Double_t &spacing);
+  Bool_t CloseInputFile();
 
  private:
   
@@ -33,7 +43,15 @@ class AliToyMCEventGeneratorSimple : public AliToyMCEventGenerator {
   TString fInputFileNameESD;
 
   AliESDtrackCuts *fESDCuts;
-  //AliESDEvent* fESDEvent;
+  Int_t fInputIndex;
+  AliESDEvent* fESDEvent;
+  TTree* fESDTree;
+  TFile* fInputFile;
+  TH1F* fHPt;
+  TH1F* fHEta;
+  TH1I* fHMult;
+  Bool_t fHistosSet;
+  TFile* fParamFile;
 
   ClassDef(AliToyMCEventGeneratorSimple, 1)