RMGStackingAction.hh

// Copyright (C) 2024 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_STACKING_ACTION_HH_
#define _RMG_STACKING_ACTION_HH_
 
#include "G4UserStackingAction.hh"
 
class G4Track;
class RMGRunAction;
class RMGStackingAction : public G4UserStackingAction {
 
  public:
 
    RMGStackingAction(RMGRunAction*);
    ~RMGStackingAction() = default;
 
    RMGStackingAction(RMGStackingAction const&) = delete;
    RMGStackingAction& operator=(RMGStackingAction const&) = delete;
    RMGStackingAction(RMGStackingAction&&) = delete;
    RMGStackingAction& operator=(RMGStackingAction&&) = delete;

    G4ClassificationOfNewTrack ClassifyNewTrack(const G4Track* aTrack) override;
    void NewStage() override;
    void PrepareNewEvent() override;
 
    [[nodiscard]] int GetCurrentStage() const { return fStage; }
 
  private:
 
    int fStage = 0;
    RMGRunAction* fRunAction = nullptr;
};
 
#endif
 
// vim: tabstop=2 shiftwidth=2 expandtab