BertiniEventTopologyProcess.h

#ifndef SIMCORE_BERTINI_EVENTTOPOLOGY_PROCESS_H
#define SIMCORE_BERTINI_EVENTTOPOLOGY_PROCESS_H
 
#include <G4CascadeInterface.hh>
#include <G4EventManager.hh>
#include <G4HadFinalState.hh>
#include <G4HadProjectile.hh>
#include <G4HadronicInteraction.hh>
#include <G4Nucleus.hh>
#include <iostream>
 
#include "SimCore/G4User/UserEventInformation.h"
#include "SimCore/PhotoNuclearModels/PhotoNuclearModel.h"
 
namespace simcore {
 
/*
** A wrapper interface around the Bertini cascade process (G4CascadeInterface)
** which reruns the event generator until a particular condition is met for the
** products. The decision of whether or not to rerun the event generator is
** handled by the acceptEvent virtual function.
**
** For example of a derived class, see
** SimCore/include/SimCore/PhotoNuclearModels/BertiniNothingHardModel.h
**
** Note: You almost certainly want to use these types of models together with a
** photonuclear topology filter.
**
** Note: When performing N attempts, this will increment the event weight in the
** UserEventInformation by 1/N. To change this behaviour, override the
** incrementEventWeight function.
*/
 
class BertiniEventTopologyProcess : public G4CascadeInterface {
 public:
  BertiniEventTopologyProcess(bool count_light_ions = true)
      : G4CascadeInterface{}, count_light_ions_{count_light_ions} {}
 
  /*
   * The primary function for derived classes to customize. After each call to
   * the Bertini cascade, this function will be called to see whether or not to
   * keep the event. The products can be accessed from the `theParticleChange`
   * member inherited from G4CascadeInterface (i.e. the Bertini cascade).
   */
  virtual bool acceptEvent() const = 0;
 
  /*
   * Is the projectile of interest?
   *
   * If false, the cascade will only run once. Example use includes only
   * applying repeated simulations for particular energy ranges. Is called
   * automatically during `ApplyYourself`.
   *
   **/
  virtual bool acceptProjectile(const G4HadProjectile& projectile) const = 0;
 
  /*
   * Is the target nucleus of interest?
   *
   * If false, the cascade will only run once. Example use includes only
   * applying repeated simulations for high Z materials. Is called
   * automatically during `ApplyYourself`.
   *
   **/
  virtual bool acceptTarget(const G4Nucleus& targetNucleus) const = 0;
 
  /*
   * Run the Bertini cascade until the condition given by `acceptEvent` is
   * matched by the reaction products. Increments the event weight by 1/n where
   * n is the number of attempts needed to produce a match.
   *
   **/
  G4HadFinalState* ApplyYourself(const G4HadProjectile& projectile,
                                 G4Nucleus& targetNucleus) override;
 
  /*
   * Geant4 assumes that secondaries produced from the bertini cascade are owned
   *  by some other part of the code. Since we are re-running the cascade until
   *  we get something matching our condition in `acceptEvent`, we have to make
   *  sure to clean up any secondaries from failed attempts.
   *
   */
  void cleanupSecondaries();
 
  constexpr bool isLightIon(const int pdgCode) const {
    //
    if (pdgCode > 1000000000) {
      // Check if the atomic number is less than or equal to 4
      return ((pdgCode / 10) % 1000) <= 4;
    }
    return false;
  }
 
  constexpr bool skipCountingParticle(const int pdgcode) const {
    return !(pdgcode < 10000 || (count_light_ions_ && isLightIon(pdgcode)));
  }
 
  /*
   *  Update the event weight depending on the number of attempts made.
   *
   *  @param N The number of attempts used for a successful topology
   *  production.
   *
   **/
 
  virtual void incrementEventWeight(int N) {
    auto event_info{static_cast<UserEventInformation*>(
        G4EventManager::GetEventManager()->GetUserInformation())};
    event_info->incWeight(1. / N);
  }
 
 private:
  bool count_light_ions_;
};
}  // namespace simcore
 
#endif /* SIMCORE_BERTINI_EVENTTOPOLOGY_PROCESS_H */