PrimaryGenerator.h

Go to the documentation of this file.

 
#ifndef SIMCORE_PRIMARYGENERATOR_H
#define SIMCORE_PRIMARYGENERATOR_H
 
/*~~~~~~~~~~~~~~~~*/
/*   C++ StdLib   */
/*~~~~~~~~~~~~~~~~*/
#include <string>
 
/*~~~~~~~~~~~~*/
/*   Geant4   */
/*~~~~~~~~~~~~*/
#include "G4VPrimaryGenerator.hh"
 
/*~~~~~~~~~~~~~~~*/
/*   Framework   */
/*~~~~~~~~~~~~~~~*/
#include "Framework/Configure/Parameters.h"
#include "Framework/Event.h"
#include "Framework/Factory.h"
#include "Framework/RunHeader.h"
 
// Forward Declarations
class G4Event;
class G4PrimaryVertex;
 
namespace simcore {
 
class PrimaryGenerator : public G4VPrimaryGenerator {
 public:
  PrimaryGenerator(const std::string& name,
                   const framework::config::Parameters& parameters);
 
  DECLARE_FACTORY_WITH_WAREHOUSE(PrimaryGenerator,
                                 std::shared_ptr<PrimaryGenerator>,
                                 const std::string&,
                                 const framework::config::Parameters&);
 
  virtual ~PrimaryGenerator() = default;
 
  virtual void prepEvent(const framework::Event& event) {}
 
  virtual void GeneratePrimaryVertex(G4Event*) = 0;
 
  // NOLINTNEXTLINE
  virtual void RecordConfig(const std::string& id, ldmx::RunHeader& rh) = 0;
 
  std::string name() { return name_; }
 
  void smearBeamspot(G4PrimaryVertex* primary_vertex);
 
  bool useBeamspot() const { return use_beamspot_; }
 
 protected:
  std::string name_{""};
 
  bool use_beamspot_{false};
 
  double beamspot_x_size_{0};
 
  double beamspot_y_size_{0};
 
  double beamspot_z_size_{0};
};  // PrimaryGenerator
 
}  // namespace simcore
 
#define DECLARE_GENERATOR(CLASS) \
  FACTORY_REGISTRATION(simcore::PrimaryGenerator, CLASS)
 
#endif  // SIMCORE_PRIMARYGENERATOR_H