XsecBiasingOperator.h

#ifndef SIMCORE_XSECBIASINGOPERATOR_H_
#define SIMCORE_XSECBIASINGOPERATOR_H_
 
#include "Framework/Configure/Parameters.h"
#include "Framework/Factory.h"
#include "Framework/Logger.h"
#include "Framework/RunHeader.h"
 
//------------//
//   Geant4   //
//------------//
#include "G4BOptnChangeCrossSection.hh"
#include "G4BiasingProcessInterface.hh"
#include "G4BiasingProcessSharedData.hh"
#include "G4ParticleDefinition.hh"
#include "G4ProcessManager.hh"
#include "G4Track.hh"
#include "G4VBiasingOperator.hh"
 
namespace simcore {
 
class XsecBiasingOperator : public G4VBiasingOperator {
 public:
  XsecBiasingOperator(std::string name,
                      const framework::config::Parameters& parameters);
 
  DECLARE_FACTORY_WITH_WAREHOUSE(XsecBiasingOperator,
                                 std::shared_ptr<XsecBiasingOperator>,
                                 std::string,
                                 const framework::config::Parameters&);
 
  virtual ~XsecBiasingOperator() { delete xsec_operation_; }
 
  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;
 
  // NOLINTNEXTLINE
  virtual void RecordConfig(ldmx::RunHeader& header) const = 0;
 
 protected:
  // NOLINTNEXTLINE
  G4VBiasingOperation* BiasedXsec(double biased_xsec) {
    xsec_operation_->SetBiasedCrossSection(biased_xsec);
    xsec_operation_->Sample();
    return xsec_operation_;
  }
 
  bool processIsBiased(std::string process);
 
  G4BOptnChangeCrossSection* xsec_operation_{nullptr};
 
  G4ProcessManager* process_manager_{nullptr};
 
  G4VBiasingOperation* ProposeFinalStateBiasingOperation(
      const G4Track*, const G4BiasingProcessInterface*) {
    return nullptr;
  }
 
  G4VBiasingOperation* ProposeNonPhysicsBiasingOperation(
      const G4Track*, const G4BiasingProcessInterface*) {
    return nullptr;
  }
  framework::logging::logger the_log_;
 
};  // XsecBiasingOperator
}  // namespace simcore
 
#define DECLARE_XSECBIASINGOPERATOR(CLASS) \
  FACTORY_REGISTRATION(simcore::XsecBiasingOperator, CLASS)
 
#endif  // SIMCORE_XSECBIASINGOPERATOR_H_