gtree_example.cc

Go to the documentation of this file.

 
// geant4
#include "G4VisExecutive.hh"
#include "G4RunManagerFactory.hh"
#include "QBBC.hh"
 
// gtree
#include "gtree.h"
#include "gtree_options.h"
 
// gemc
#include "gdetectorConstruction.h"
 
// qt
#include <QApplication>
#include <QMainWindow>
#include <QTimer>
 

int main(int argc, char* argv[]) {
    // Initialize options and logging
    auto gopts = std::make_shared<GOptions>(argc, argv, gtree::defineOptions());
    auto log = std::make_shared<GLogger>(gopts, SFUNCTION_NAME, GTREE_LOGGER);
    auto timeout = gopts->getRequiredScalarDouble("tt");
 
    log->info(0, "Starting gtree example...");
 
    // Optional GUI setup (only if --gui is passed)
    QApplication* app = nullptr;
    QMainWindow* window = nullptr;
 
    // Geant4: create a default run manager and register a reference physics list.
    auto runManager = G4RunManagerFactory::CreateRunManager(G4RunManagerType::Default);
    auto physicsList = new QBBC;
    runManager->SetUserInitialization(physicsList);
 
    // Geant4: visualization manager used by geometry UI commands invoked by the widget.
    auto visManager = new G4VisExecutive;
    visManager->Initialize();
 
    if (gopts->getSwitch("gui")) {
        log->info(0, "gtree", "Running in GUI mode...");
        app = new QApplication(argc, argv);
        window = new QMainWindow();
        window->setWindowTitle(QString::fromUtf8("gtree example"));
    }
 
    // Build detector and initialize the run manager so geometry exists.
    auto gdetector = new GDetectorConstruction(gopts);
    runManager->SetUserInitialization(gdetector);
    runManager->Initialize();
 
    // Retrieve volume map used by GTree to build the system/volume hierarchy.
    auto g4vmap = gdetector->get_g4volumes_map();
    auto gvmap = gdetector->get_gvolumes_flat_map();
    auto gmmap = gdetector->get_gmirrors_flat_map();
 
    log->info(0, "gtree", "Detector construction successful with ", g4vmap.size(), " volumes.");
 
    if (gopts->getSwitch("gui")) {
 
        auto gtree = new GTree(gopts, g4vmap, gvmap, window, gmmap);
 
        window->setCentralWidget(gtree);
        window->show();
 
        // Quit after the configured delay (milliseconds) to make the example self-terminating.
        QTimer::singleShot(timeout, app, &QCoreApplication::quit);
 
        int appResult = QApplication::exec();
 
        // Clean up GUI resources
        delete gtree;
        delete window;
        delete app;
 
        // Clean up Geant4 and custom logic
        delete visManager;
 
        return appResult;
    }
 
 
    // CLI mode
    log->info(0, "Running gtree in command line mode...");
 
    return EXIT_SUCCESS;
}