g4SceneProperties.cc

Go to the documentation of this file.

 
// g4display
#include "g4SceneProperties.h"
#include "g4Text.h"
 
// gemc
#include "gutilities.h"
 
// geant4
#include "G4UImanager.hh"
 
namespace {
bool is3DTextKind(const std::string& kind) {
    return kind == "3D" || kind == "3d" || kind == "text" || kind == "Text";
}
 
std::string colourToRGB(const std::string& colour) {
    if (colour.find(' ') != std::string::npos) { return colour; }
    if (colour == "black")   { return "0 0 0"; }
    if (colour == "blue")    { return "0 0 1"; }
    if (colour == "cyan")    { return "0 1 1"; }
    if (colour == "green")   { return "0 1 0"; }
    if (colour == "grey" || colour == "gray") { return "0.9 0.9 0.9"; }
    if (colour == "red")     { return "1 0 0"; }
    if (colour == "yellow")  { return "1 1 0"; }
    return "1 1 1";
}
}
 
std::vector<std::string> G4SceneProperties::scene_commands(const std::shared_ptr<GOptions> &gopts) {
    std::vector<std::string> cmds;
 
    bool gui = gopts->getSwitch("gui");
    bool use_dawn = gopts->getSwitch("useDawn");
 
    // Project options onto simple structs for downstream use.
    auto g4view = g4display::getG4View(gopts);
    auto g4camera = g4display::getG4Camera(gopts);
    auto g4light = g4display::getG4Light(gopts);
 
    // Create a named scene. Caller is expected to apply these commands to the Geant4 UI manager.
    cmds.emplace_back("/vis/scene/create gemc");
 
    if (use_dawn) {
        // DAWNFILE workflow: open the DAWN viewer and adjust a minimal set of scene properties.
        cmds.emplace_back("/vis/open DAWNFILE");
        cmds.emplace_back("/vis/geometry/set/visibility World 0 false");
        cmds.emplace_back("/vis/viewer/set/style surface");
    }
    if (gui || g4view.driver == "TOOLSSG_OFFSCREEN") {
        // Open the configured viewer driver. Offscreen drivers ignore window position, so omit it.
        std::string openArg = g4view.driver + " " + g4view.dimension;
        if (g4view.driver != "TOOLSSG_OFFSCREEN") openArg += g4view.position;
        cmds.emplace_back("/vis/open " + openArg);
 
        // Convert configured camera angles to degrees for the Geant4 viewer command.
        const double toDegrees = 180.0 / M_PI;
        double thetaValue      = gutilities::getG4Number(g4camera.theta) * toDegrees;
        double phiValue        = gutilities::getG4Number(g4camera.phi)   * toDegrees;
        double lightThetaValue = gutilities::getG4Number(g4light.theta)  * toDegrees;
        double lightPhiValue   = gutilities::getG4Number(g4light.phi)    * toDegrees;
        if (lightThetaValue == 0.0 && lightPhiValue == 0.0) {
            lightThetaValue = thetaValue;
            lightPhiValue   = phiValue;
        }
 
        // Disable auto refresh and quieten vis messages whilst scene and trajectories are established.
        cmds.emplace_back("/vis/viewer/set/autoRefresh false");
 
        cmds.emplace_back("/vis/viewer/set/viewpointThetaPhi " + std::to_string(thetaValue) + " " + std::to_string(phiValue));
        cmds.emplace_back("/vis/viewer/set/lightsThetaPhi " + std::to_string(lightThetaValue) + " " + std::to_string(lightPhiValue));
        cmds.emplace_back("/vis/viewer/set/lineSegmentsPerCircle " + std::to_string(g4view.segsPerCircle));
        cmds.emplace_back("/vis/viewer/set/background " + g4view.background);
        cmds.emplace_back("/vis/viewer/set/numberOfCloudPoints " + std::to_string(g4view.cloudPoints));
 
        cmds.emplace_back("/vis/viewer/set/autoRefresh true");
    }
 
    return cmds;
}
 
std::vector<std::string> G4SceneProperties::addSceneDecorations(const std::shared_ptr<GOptions> &gopts) {
    std::vector<std::string> commands;
    const auto decorations = g4display::getG4Decorations(gopts);
 
    if (decorations.scale) {
        commands.emplace_back("/vis/scene/add/scale " +
                              std::to_string(decorations.scaleLength) + " " +
                              decorations.scaleUnit + " " +
                              decorations.scaleDirection + " " +
                              colourToRGB(decorations.scaleColor));
    }
    if (decorations.axes) { commands.emplace_back("/vis/scene/add/axes"); }
    if (decorations.date) {
        commands.emplace_back("/vis/scene/add/date " + std::to_string(decorations.dateSize));
    }
    if (decorations.logo2D) { commands.emplace_back("/vis/scene/add/logo2D"); }
    if (decorations.logo) { commands.emplace_back("/vis/scene/add/logo"); }
    if (decorations.frame) {
        commands.emplace_back("/vis/set/colour " + decorations.frameColor);
        commands.emplace_back("/vis/set/lineWidth " + std::to_string(decorations.frameLineWidth));
        commands.emplace_back("/vis/scene/add/frame");
        commands.emplace_back("/vis/set/colour");
        commands.emplace_back("/vis/set/lineWidth");
    }
 
    return commands;
}
 
std::vector<std::string> G4SceneProperties::addSceneTexts(const std::shared_ptr<GOptions> &gopts) {
    std::vector<std::string> commands;
 
    std::vector<g4display::G4SceneText> text_to_add = g4display::getSceneTexts(gopts);
 
    // Map each configured text item into Geant4 text commands.
    for (const auto &text: text_to_add) {
        commands.emplace_back("/vis/set/textColour " + text.color);
        if (!text.layout.empty()) {
            commands.emplace_back("/vis/set/textLayout " + text.layout);
        }
 
        if (is3DTextKind(text.kind)) {
            commands.emplace_back(
                "/vis/scene/add/text " +
                std::to_string(text.x) + " " +
                std::to_string(text.y) + " " +
                std::to_string(text.z) + " " +
                text.unit + " " +
                std::to_string(text.size) + " " +
                std::to_string(text.dx) + " " +
                std::to_string(text.dy) + " " +
                text.text);
        } else {
            commands.emplace_back(
                "/vis/scene/add/text2D " +
                std::to_string(text.x) + " " +
                std::to_string(text.y) + " " +
                std::to_string(text.size) + " ! ! " +
                text.text);
        }
 
        if (!text.layout.empty()) {
            commands.emplace_back("/vis/set/textLayout left");
        }
 
        // Restore default text color.
        commands.emplace_back("/vis/set/textColour");
    }
 
    return commands;
}