VertexGen_ES.hh

#ifndef __RAT_VertexGen_ES__
#define __RAT_VertexGen_ES__
 
#include <G4Event.hh>
#include <G4ThreeVector.hh>
#include <RAT/ESgen.hh>
#include <RAT/GLG4VertexGen.hh>
#include <globals.hh>
 
 
namespace RAT {
 
class VertexGen_ES : public GLG4VertexGen {
 public:
  // Note that the database named is "ibd" by default in the
  // constructor.  In other words, we assume the anti-neutrino flux
  // is the same for both inverse beta-decay (IBD) and elastic
  // scattering (ES)... at least for now.
 
  VertexGen_ES(const char *arg_dbname = "solar");
  virtual ~VertexGen_ES();
  virtual void GeneratePrimaryVertex(G4Event *argEvent, G4ThreeVector &dx, G4double dt);
  // generates a primary vertex with given particle type, direction, and energy.
  virtual void SetState(G4String newValues);
  // format: dir_x dir_y dir_z
  // If dir_x==dir_y==dir_z==0, the directions are isotropic.
  virtual G4String GetState();
  // returns current state formatted as above
 
  void SetFlux(const G4String flux);
 
  G4String GetFlux() { return fFlux; };
 
  void SetNuFlavor(const G4String flavor);
 
  G4String GetNuFlavor() { return fNuFlavor; };
 
  ESgen *GetHelper() { return fESgen; };
 
  const G4String GetDBName() const { return fDBName; }
 
  void SetDBName(const G4String name);
 
 private:
  G4ParticleDefinition *fElectron, *fNue, *fNumu;
 
  G4ThreeVector fNuDir;
 
  G4String fFlux;
 
  G4String fNuFlavor;
 
  ESgen *fESgen;
 
  // Electron mass
  double fElectronMass;
 
  G4String fDBName;
 
  bool fRandomDir;
};
 
}  // namespace RAT
 
#endif