XsecBiasingOperator.h

#ifndef SIMCORE_XSECBIASINGOPERATOR_H_
#define SIMCORE_XSECBIASINGOPERATOR_H_
 
#include "Framework/Configure/Parameters.h"
#include "Framework/RunHeader.h"
#include "SimCore/Factory.h"
 
//------------//
//   Geant4   //
//------------//
#include "G4BOptnChangeCrossSection.hh"
#include "G4BiasingProcessInterface.hh"
#include "G4BiasingProcessSharedData.hh"
#include "G4Electron.hh"
#include "G4Gamma.hh"
#include "G4KaonZeroLong.hh"
#include "G4Neutron.hh"
#include "G4ParticleDefinition.hh"
#include "G4ParticleTable.hh"
#include "G4ProcessManager.hh"
#include "G4RunManager.hh"
#include "G4Track.hh"
#include "G4VBiasingOperator.hh"
 
namespace simcore {
 
class XsecBiasingOperator : public G4VBiasingOperator {
 public:
  XsecBiasingOperator(std::string name,
                      const framework::config::Parameters& parameters);
 
  using Factory =
      ::simcore::Factory<XsecBiasingOperator,
                         std::shared_ptr<XsecBiasingOperator>, std::string,
                         const framework::config::Parameters&>;
 
  virtual ~XsecBiasingOperator() = default;
 
  virtual G4VBiasingOperation* ProposeOccurenceBiasingOperation(
      const G4Track* track,
      const G4BiasingProcessInterface* callingProcess) = 0;
 
  void StartRun();
 
  virtual std::string getProcessToBias() const = 0;
 
  virtual std::string getParticleToBias() const = 0;
 
  virtual std::string getVolumeToBias() const = 0;
 
  virtual void RecordConfig(ldmx::RunHeader& header) const = 0;
 
 protected:
  G4VBiasingOperation* BiasedXsec(double biased_xsec) {
    xsecOperation_->SetBiasedCrossSection(biased_xsec);
    xsecOperation_->Sample();
    return xsecOperation_;
  }
 
  bool processIsBiased(std::string process);
 
  G4BOptnChangeCrossSection* xsecOperation_{nullptr};
 
  G4ProcessManager* processManager_{nullptr};
 
  G4VBiasingOperation* ProposeFinalStateBiasingOperation(
      const G4Track*, const G4BiasingProcessInterface*) {
    return nullptr;
  }
 
  G4VBiasingOperation* ProposeNonPhysicsBiasingOperation(
      const G4Track*, const G4BiasingProcessInterface*) {
    return nullptr;
  }
 
};  // XsecBiasingOperator
}  // namespace simcore
 
#define DECLARE_XSECBIASINGOPERATOR(CLASS)                                  \
  namespace {                                                               \
  auto v = ::simcore::XsecBiasingOperator::Factory::get().declare<CLASS>(); \
  }
 
#endif  // SIMCORE_XSECBIASINGOPERATOR_H_