g4SceneProperties.cc

Go to the documentation of this file.

 
// g4display
#include "g4SceneProperties.h"
#include "g4Text.h"
 
// gemc
#include "gutilities.h"
 
// geant4
#include "G4UImanager.hh"
 
 
 
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");
 
    // Projecting options onto G4View and G4Camera structs
    auto g4view   = g4display::getG4View(gopts);
    auto g4camera = g4display::getG4Camera(gopts);
    // auto g4dawn   = g4display::getG4Dawn(gopts);
 
    std::vector<std::string> commands;
 
    cmds.emplace_back("/vis/scene/create gemc");
 
    if (use_dawn) {
        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) {
        cmds.emplace_back("/vis/open " + g4view.driver + " " + g4view.dimension + g4view.position);
 
        // Scene texts
        for (const std::string& c : addSceneTexts(gopts)) { commands.emplace_back(c); }
 
        double toDegrees  = 180 / 3.1415;
        double thetaValue = gutilities::getG4Number(g4camera.theta) * toDegrees;
        double phiValue   = gutilities::getG4Number(g4camera.phi) * toDegrees;
 
        // 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/lineSegmentsPerCircle " + std::to_string(g4view.segsPerCircle));
        cmds.emplace_back("/vis/viewer/set/autoRefresh true");
    }
 
 
    return cmds;
}
 
 
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);
 
    // Iterate over each text object to generate the corresponding visualization commands
    for (const auto& text : text_to_add) {
        commands.emplace_back("/vis/set/textColour " + text.color);
        std::string position = std::to_string(text.x) + " " + std::to_string(text.y);
        std::string size     = " " + std::to_string(text.size) + " ! ! ";
        if (text.z != GNOT_SPECIFIED_SCENE_TEXT_Z) {
            position += " " + std::to_string(text.z);
            commands.emplace_back(std::string("/vis/scene/add/text2D ").append(position).append(size).append(text.text));
        }
        else { commands.emplace_back(std::string("/vis/scene/add/text ").append(position).append(size).append(text.text)); }
 
        commands.emplace_back("/vis/set/textColour");
    }
 
    return commands;
}