Commit 4a50d5d8 authored by Håkan Wennlöf's avatar Håkan Wennlöf
Browse files

Moved Geant4 exception handler initialisation to Worker constructors. 

Previously, when it was set in [GeometryBuilderGeant4] and [DepositionGeant4], the exception handler got overwritten with the default Geant4 one.
parent 85efd796

src/modules/GeometryBuilderGeant4/GeometryBuilderGeant4Module.cpp

+2 −11
Original line number Diff line number Diff line
@@ -17,7 +17,6 @@ 
#include <utility>



#include <G4GlobalConfig.hh>

#include <G4StateManager.hh>

#include <G4UImanager.hh>

#include <G4UIterminal.hh>

#include <G4Version.hh>
@@ -36,7 +35,6 @@ 
#include "core/config/exceptions.h"

#include "core/geometry/GeometryManager.hpp"

#include "core/utils/log.h"

#include "tools/geant4/G4ExceptionHandler.hpp"

#include "tools/geant4/G4LoggingDestination.hpp"

#include "tools/geant4/MTRunManager.hpp"

#include "tools/geant4/RunManager.hpp"
@@ -47,9 +45,6 @@ using namespace ROOT; 
GeometryBuilderGeant4Module::GeometryBuilderGeant4Module(Configuration& config, Messenger*, GeometryManager* geo_manager)

    : Module(config), geo_manager_(geo_manager), run_manager_g4_(nullptr) {



    // Set exception handler for Geant4 exceptions:

    G4StateManager::GetStateManager()->SetExceptionHandler(new G4ExceptionHandler());



// Enable multithreading for Geant4 if it has been built with support for it:

#ifdef G4MULTITHREADED

    LOG(INFO) << "Detected Geant4 multithreading capabilities, enabling multithreading support";
@@ -126,20 +121,16 @@ void GeometryBuilderGeant4Module::initialize() { 
    check_dataset_g4("G4NEUTRONXSDATA");

#endif



    // Suppress all stdout output due to a part in Geant4 where G4cout is not used

    SUPPRESS_STREAM(std::cout);



    // Create the G4 run manager. If multithreading was requested we use the custom run manager

    // that support calling BeamOn operations in parallel. Otherwise we use default manager.

    if(multithreadingEnabled()) {

        LOG(DEBUG) << "Making a multi-thread RunManager";

        run_manager_g4_ = std::make_unique<MTRunManager>();

    } else {

        LOG(DEBUG) << "Making a single-thread RunManager";

        run_manager_g4_ = std::make_unique<RunManager>();

    }



    // Release stdout again

    RELEASE_STREAM(std::cout);



    // Set the geometry construction to use

    run_manager_g4_->SetUserInitialization(geometry_construction_);

src/tools/geant4/G4ExceptionHandler.hpp

+15 −2
Original line number Diff line number Diff line
@@ -13,6 +13,7 @@ 
#define ALLPIX_GEANT4_EXCEPTIONS_H



#include "core/module/exceptions.h"

#include "core/utils/log.h"



#include <G4VExceptionHandler.hh>
@@ -27,8 +28,20 @@ namespace allpix { 
     */

    class G4ExceptionHandler : public G4VExceptionHandler {

    public:

        G4bool Notify(const char*, const char* exceptionCode, G4ExceptionSeverity, const char* description) override {

            throw ModuleError("Caught Geant4 exception " + std::string(exceptionCode) + ": " + std::string(description));

        G4bool Notify(const char*, const char* code, G4ExceptionSeverity severity, const char* description) override {

            std::string message = "Caught Geant4 exception " + std::string(code) + ": " + std::string(description);

            if(severity == G4ExceptionSeverity::JustWarning) {

                LOG(WARNING) << message;

            } else if(severity == G4ExceptionSeverity::EventMustBeAborted) {

                throw AbortEventException(message);

            } else if(severity == G4ExceptionSeverity::RunMustBeAborted) {

                throw EndOfRunException(message);

            } else {

                throw ModuleError(message);

            }



            // Continue program execution:

            return false;

        };

    };

src/tools/geant4/MTRunManager.cpp

+5 −0
Original line number Diff line number Diff line
@@ -13,8 +13,11 @@ 
#include "G4LoggingDestination.hpp"

#include "WorkerRunManager.hpp"



#include <G4StateManager.hh>

#include <G4UImanager.hh>



#include "tools/geant4/G4ExceptionHandler.hpp"



using namespace allpix;



G4ThreadLocal WorkerRunManager* MTRunManager::worker_run_manager_ = nullptr;
@@ -22,6 +25,8 @@ G4ThreadLocal WorkerRunManager* MTRunManager::worker_run_manager_ = nullptr; 
MTRunManager::MTRunManager() {

    G4UImanager* ui_g4 = G4UImanager::GetUIpointer();

    ui_g4->SetCoutDestination(G4LoggingDestination::getInstance());

    // Set exception handler for Geant4 exceptions:

    G4StateManager::GetStateManager()->SetExceptionHandler(new G4ExceptionHandler());

}



void MTRunManager::Run(G4int n_event, uint64_t seed1, uint64_t seed2) { // NOLINT

src/tools/geant4/RunManager.cpp

+6 −0
Original line number Diff line number Diff line
@@ -14,12 +14,18 @@ 


#include <array>



#include <G4StateManager.hh>

#include <G4UImanager.hh>



#include "tools/geant4/G4ExceptionHandler.hpp"



using namespace allpix;



RunManager::RunManager() {

    G4UImanager* ui_g4 = G4UImanager::GetUIpointer();

    ui_g4->SetCoutDestination(G4LoggingDestination::getInstance());

    // Set exception handler for Geant4 exceptions:

    G4StateManager::GetStateManager()->SetExceptionHandler(new G4ExceptionHandler());

}



void RunManager::Run(G4int n_event, uint64_t seed1, uint64_t seed2) { // NOLINT

src/tools/geant4/WorkerRunManager.cpp

+5 −0
Original line number Diff line number Diff line
@@ -18,6 +18,7 @@ 
#include <atomic>



#include <G4MTRunManager.hh>

#include <G4StateManager.hh>

#include <G4Threading.hh>

#include <G4TransportationManager.hh>

#include <G4UImanager.hh>
@@ -29,6 +30,8 @@ 
#include <G4WorkerRunManager.hh>

#include <G4WorkerThread.hh>



#include "tools/geant4/G4ExceptionHandler.hpp"



using namespace allpix;



namespace {
@@ -39,6 +42,8 @@ namespace { 
WorkerRunManager::WorkerRunManager() {

    G4UImanager* ui_g4 = G4UImanager::GetUIpointer();

    ui_g4->SetCoutDestination(G4LoggingDestination::getInstance());

    // Set exception handler for Geant4 exceptions:

    G4StateManager::GetStateManager()->SetExceptionHandler(new G4ExceptionHandler());

}



WorkerRunManager::~WorkerRunManager() {