PartialEnergySorter.h

#ifndef BIASING_UTILITY_PARTIALENERGYSORTER_H
#define BIASING_UTILITY_PARTIALENERGYSORTER_H
 
/*~~~~~~~~~~~~~~~~*/
/*   C++ StdLib   */
/*~~~~~~~~~~~~~~~~*/
#include <string>
 
/*~~~~~~~~~~~~~*/
/*   SimCore   */
/*~~~~~~~~~~~~~*/
#include "SimCore/UserAction.h"
 
// Forward declarations
class G4Step;
 
namespace biasing {
namespace utility {
 
class PartialEnergySorter : public simcore::UserAction {
 public:
  PartialEnergySorter(const std::string& name,
                      framework::config::Parameters& parameters);
 
  virtual ~PartialEnergySorter() {}
 
  void BeginOfEventAction(const G4Event* event) override;
 
  G4ClassificationOfNewTrack ClassifyNewTrack(
      const G4Track* aTrack,
      const G4ClassificationOfNewTrack& currentTrackClass) override;
 
  void stepping(const G4Step* step) override;
 
  std::vector<simcore::TYPE> getTypes() override {
    return {simcore::TYPE::STEPPING, simcore::TYPE::STACKING,
            simcore::TYPE::EVENT};
  }
 
  void NewStage() override {
    below_threshold_ = true;
  }
 
 private:
  double threshold_;
 
  bool below_threshold_;
 
};  // PartialEnergySorter
 
}  // namespace utility
}  // namespace biasing
 
#endif  // BIASING_UTILITY_PARTIALENERGYSORTER_H