MyEventAction.hh

Go to the documentation of this file.

/**@file
   Provides user EventAction class specialized for ROOT file output.

   @author G. Horton-Smith (Tohoku, Caltech, Kansas State)
*/
   
#include "TROOT.h"     
#include "TFile.h"     
#include "TTree.h"     
#include "TBranch.h"   
#include "TStopwatch.h"

#include <time.h>
#define ELAPSEDT 300

class MyEventAction : public GLG4VEventAction {
private:
  TFile* fROOTOutputFile;
  TTree* fROOTOutputTree;
  TTree* fROOTRunTree;
  TStopwatch* timer;
  time_t Ptime;
  
public:
  MyEventAction(): GLG4VEventAction(),
                   fROOTOutputFile(0), fROOTOutputTree(0)
  {
    fFileCmd->SetGuidance("This will be a ROOT Format File;");
    timer= new TStopwatch();    
  }
    
  ~MyEventAction(){ CloseFile(); }
    
  virtual void BeginOfEventAction(const G4Event* evt)
  {
    GLG4VEventAction::BeginOfEventAction(evt);
    timer->Start();
  }
  virtual void EndOfEventAction(const G4Event* evt)
  {
    timer->Stop();
    GLG4VEventAction::EndOfEventAction(evt);
  }
  virtual void OpenFile(const G4String filename,G4bool outputMode);
  virtual void CloseFile();
  virtual void FillData(const G4Event*);
  virtual void Clear(){;}
  
  void CreateTree();

/*
  virtual G4String GetFileName(){ return fFileName;}
  // virtual void SetFileName(const G4String filename){ fFileName=filename;}
*/

  // Here is data area for ROOT tree.  The data is copied here to allow
  // the "simple" style of ROOT tree.  I'm sorry it is sooo ugly....
  enum { max_primary_particles= 16, max_hits_for_ROOT=4000 };
  Int_t eventID;
  Int_t runID;
  Float_t UT;
  Float_t delta_UT;
  Int_t eventType;

  struct vertex_info_s {
    Int_t n_particles;
    Float_t t0[max_primary_particles],x0[max_primary_particles],
      y0[max_primary_particles],z0[max_primary_particles];
    Int_t pdg_code[max_primary_particles];
    Float_t px[max_primary_particles];
    Float_t py[max_primary_particles];
    Float_t pz[max_primary_particles];
    Float_t polx[max_primary_particles];
    Float_t poly[max_primary_particles];
    Float_t polz[max_primary_particles];
    Float_t ke[max_primary_particles];
  } vertex_info;
  Double_t vketot;
  Float_t vcentroid_x, vcentroid_y, vcentroid_z;
  
  Int_t n_photon_hits;
  Double_t hit_time[max_hits_for_ROOT];
  Short_t hit_pmt[max_hits_for_ROOT];
  Short_t hit_wl[max_hits_for_ROOT];
  Float_t hit_x[max_hits_for_ROOT], hit_y[max_hits_for_ROOT],
    hit_z[max_hits_for_ROOT], hit_px[max_hits_for_ROOT],
    hit_py[max_hits_for_ROOT], hit_pz[max_hits_for_ROOT],
    hit_polx[max_hits_for_ROOT], hit_poly[max_hits_for_ROOT],
    hit_polz[max_hits_for_ROOT];
  Int_t hit_count[max_hits_for_ROOT];
  // could maybe add trajectories someday

  Float_t totScintEdep;
  Float_t totScintEdepQuenched;
  Float_t scint_centroid_x, scint_centroid_y, scint_centroid_z;
};

---