RMGStagingScheme.hh

// Copyright (C) 2025 Manuel Huber <https://orcid.org/0009-0000-5212-2999>
//
// This program is free software: you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License as published by the Free
// Software Foundation, either version 3 of the License, or (at your option) any
// later version.
//
// This program is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
// FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
// details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
#ifndef _RMG_STAGING_SCHEME_HH_
#define _RMG_STAGING_SCHEME_HH_
 
#include <memory>
#include <optional>
#include <set>
#include <string>
#include <vector>
 
#include "G4GenericMessenger.hh"
 
#include "RMGVOutputScheme.hh"
 
class G4Step;

class RMGStagingScheme : public RMGVOutputScheme {
 
  public:
 
    RMGStagingScheme();

    std::optional<G4ClassificationOfNewTrack> StackingActionClassify(const G4Track*, int) override;

    void SteppingAction(const G4Step*) override;

    void SetElectronVolumeSafety(double safety) { fElectronVolumeSafety = safety; }

    void AddElectronVolumeName(std::string volume) { fElectronVolumeNames.insert(volume); }

    void SetElectronMaxEnergyThresholdForStacking(double energy) {
      fElectronMaxEnergyThresholdForStacking = energy;
    }

    void SetElectronMinEnergyThresholdForStacking(double energy) {
      fElectronMinEnergyThresholdForStacking = energy;
    }
 
  private:
 
    std::unique_ptr<G4GenericMessenger> fElectronStagingMessengers;
    std::unique_ptr<G4GenericMessenger> fOpticalPhotonStagingMessengers;
 
 
    void DefineCommands();
 
    std::optional<G4ClassificationOfNewTrack> Classify_OpticalPhoton(const G4Track* aTrack) const;
    std::optional<G4ClassificationOfNewTrack> Classify_ElectronLike(const G4Track* aTrack) const;
 
    bool fDeferOpticalPhotonsToWaitingStage = false;
 
    bool fDeferElectronsToWaitingStage = false;
    bool fDeferPositronsToWaitingStage = false;
    double fElectronMaxEnergyThresholdForStacking = -1;
    double fElectronMinEnergyThresholdForStacking = -1;
    bool fSuspendElectronsOnEnergyDrop = false;
    double fElectronVolumeSafety = 0;
 
    std::set<std::string> fElectronVolumeNames;
};
 
#endif
 
// vim: tabstop=2 shiftwidth=2 expandtab