RooTracker_Gen.hh

 
#ifndef __RAT_RooTracker_Gen__
#define __RAT_RooTracker_Gen__
 
#include <TFile.h>
#include <TTimeStamp.h>
 
#include <RAT/GLG4Gen.hh>
 
class TTree;
 
namespace RAT {
 
namespace DS {
class Root;
}
 
const int kNPmax = 250;  // gNtpConv.cxx
 
// GENIE StdHep record, see GENIE manual arxiv:1510.05494, Sec. 7.6.2.
struct StdHepRecord {
  StdHepRecord(TTree* t) {
    t->SetBranchAddress("EvtNum", &EvtNum);
    t->SetBranchAddress("EvtXSec", &EvtXSec);
    t->SetBranchAddress("EvtDXSec", &EvtDXSec);
    t->SetBranchAddress("EvtWght", &EvtWght);
    t->SetBranchAddress("EvtProb", &EvtProb);
    t->SetBranchAddress("EvtVtx", EvtVtx);
    t->SetBranchAddress("StdHepN", &StdHepN);
    t->SetBranchAddress("StdHepPdg", StdHepPdg);
    t->SetBranchAddress("StdHepStatus", StdHepStatus);
    t->SetBranchAddress("StdHepRescat", StdHepRescat);
    t->SetBranchAddress("StdHepX4", StdHepX4);
    t->SetBranchAddress("StdHepP4", StdHepP4);
    t->SetBranchAddress("StdHepPolz", StdHepPolz);
    t->SetBranchAddress("StdHepFd", StdHepFd);
    t->SetBranchAddress("StdHepLd", StdHepLd);
    t->SetBranchAddress("StdHepFm", StdHepFm);
    t->SetBranchAddress("StdHepLm", StdHepLm);
  }
 
  int EvtNum;
  double EvtXSec;
  double EvtDXSec;
  double EvtWght;
  double EvtProb;
  double EvtVtx[4];
  int StdHepN;
  int StdHepPdg[kNPmax];
  int StdHepStatus[kNPmax];
  int StdHepRescat[kNPmax];
  double StdHepX4[kNPmax][4];
  double StdHepP4[kNPmax][4];
  double StdHepPolz[kNPmax][3];
  int StdHepFd[kNPmax];
  int StdHepLd[kNPmax];
  int StdHepFm[kNPmax];
  int StdHepLm[kNPmax];
};
 
class RooTracker_Gen : public GLG4Gen {
 public:
  RooTracker_Gen()
      : fStateStr(""), fCurrentEvent(0), fNumEvents(0), fMaxEvent(0), fLastEventTime(0), fTimeGen(0), fPosGen(0) {}
 
  virtual ~RooTracker_Gen() { fFile->Close(); }
 
  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;
 
  TFile* fFile;
  TTree* fTTree;
  StdHepRecord* fRec;
};
 
}  // namespace RAT
 
#endif