RMGMasterGenerator.hh

// Copyright (C) 2022 Luigi Pertoldi <https://orcid.org/0000-0002-0467-2571>
//
// 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_MASTER_GENERATOR_HH_
#define _RMG_MASTER_GENERATOR_HH_
 
#include <memory>
 
#include "G4VUserPrimaryGeneratorAction.hh"
 
#include "RMGVGenerator.hh"
#include "RMGVVertexGenerator.hh"
 
class G4Event;
class G4GenericMessenger;
class RMGMasterGenerator : public G4VUserPrimaryGeneratorAction {
 
  public:

    enum class Confinement {
      kUnConfined, 
      kVolume,   
      kFromFile, 
      kFromPoint 
    };

    enum class Generator {
      kG4gun,            
      kGPS,              
      kBxDecay0,         
      kFromFile,         
      kCosmicMuons,      
      kMUSUNCosmicMuons, 
      kUserDefined,      
      kGeomBench,        
      kUndefined         
    };
 
    RMGMasterGenerator();
    ~RMGMasterGenerator() = default;
 
    RMGMasterGenerator(RMGMasterGenerator const&) = delete;
    RMGMasterGenerator& operator=(RMGMasterGenerator const&) = delete;
    RMGMasterGenerator(RMGMasterGenerator&&) = delete;
    RMGMasterGenerator& operator=(RMGMasterGenerator&&) = delete;

    void GeneratePrimaries(G4Event* event) override;

    RMGVGenerator* GetGenerator() { return fGeneratorObj.get(); }
    RMGVVertexGenerator* GetVertexGenerator() { return fVertexGeneratorObj.get(); }
    [[nodiscard]] Confinement GetConfinement() const { return fConfinement; }

    void SetConfinement(Confinement code);
    void SetConfinementString(std::string code);
    void SetUserGenerator(RMGVGenerator* gen);
    void SetGenerator(Generator gen);
    void SetGeneratorString(std::string gen);
 
  private:
 
    Confinement fConfinement{Confinement::kUnConfined};
    std::unique_ptr<RMGVVertexGenerator> fVertexGeneratorObj;
 
    Generator fGenerator{Generator::kUndefined};
    std::unique_ptr<RMGVGenerator> fGeneratorObj;
 
    std::unique_ptr<G4GenericMessenger> fMessenger;
    void DefineCommands();
};
 
#endif
 
// vim: tabstop=2 shiftwidth=2 expandtab