#include "QGSP_BERT.hh"
#include "G4MTRunManager.hh"
#include "G4VUserActionInitialization.hh"
#include "G4UImanager.hh"
#include "G4UIExecutive.hh"
#include "G4VisExecutive.hh"

#include "Geometry.h"
#include "ParticleSource.h"

class ActionInitialzation : public G4VUserActionInitialization {
public:
    virtual void Build() const{
        SetUserAction(new GeneratorAction);
    }
};

int main(int argc, char** argv) {
    G4UIExecutive* ui = nullptr;
    if (argc == 1)
        ui = new G4UIExecutive(argc, argv);

    auto runManager = new G4MTRunManager;
    auto visManager = new G4VisExecutive;
    auto UIManager = G4UImanager::GetUIpointer();

    runManager->SetUserInitialization(new DetectorConstruction());
    runManager->SetUserInitialization(new QGSP_BERT());
    runManager->SetUserInitialization(new ActionInitialzation());

    visManager->Initialize();
    if ( !ui ) {
        G4String command = "/control/execute";
        G4String fileName = argv[1];
        UIManager->ApplyCommand(command + fileName);
    } else {
        UIManager->ApplyCommand("/control/execute vis.mac");
        ui->SessionStart();
        delete ui;
    }

    delete runManager;
    delete visManager;
    return 0;
}