VertexFile_Gen.hh

 
#ifndef __RAT_VertexFile_Gen_
#define __RAT_VertexFile_Gen_
 
#include <TTimeStamp.h>
 
#include <RAT/GLG4Gen.hh>
 
class TTree;
 
namespace RAT {
 
namespace DS {
class Root;
}
 
class VertexFile_Gen : public GLG4Gen {
 public:
  VertexFile_Gen()
      : fStateStr(""), fCurrentEvent(0), fNumEvents(0), fMaxEvent(0), fLastEventTime(0), fTimeGen(0), fPosGen(0){};
  virtual ~VertexFile_Gen() = default;
 
  void GenerateEvent(G4Event *event);
  bool IsRepeatable() const { return true; };
 
  void ResetTime(double offset = 0.0);
 
  void SetState(G4String state);
  G4String GetState() const;
 
  void SetTimeState(G4String state);
  G4String GetTimeState() const;
 
  void SetPosState(G4String state);
  G4String GetPosState() const;
 
 protected:
  G4String fStateStr;
  int fCurrentEvent;  
  int fNumEvents;     
  int fMaxEvent;      
 
  TTimeStamp fLastEventTime;  
 
  GLG4TimeGen *fTimeGen;
  GLG4PosGen *fPosGen;
 
  TTree *fTTree;
  DS::Root *fDS;
};
 
}  // namespace RAT
#endif