Files
geant4/source/visualization/management/src/G4VisCommandsCompound.cc
T
2022-12-09 14:43:28 +01:00

524 lines
20 KiB
C++

//
// ********************************************************************
// * License and Disclaimer *
// * *
// * The Geant4 software is copyright of the Copyright Holders of *
// * the Geant4 Collaboration. It is provided under the terms and *
// * conditions of the Geant4 Software License, included in the file *
// * LICENSE and available at http://cern.ch/geant4/license . These *
// * include a list of copyright holders. *
// * *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work make any representation or warranty, express or implied, *
// * regarding this software system or assume any liability for its *
// * use. Please see the license in the file LICENSE and URL above *
// * for the full disclaimer and the limitation of liability. *
// * *
// * This code implementation is the result of the scientific and *
// * technical work of the GEANT4 collaboration. *
// * By using, copying, modifying or distributing the software (or *
// * any work based on the software) you agree to acknowledge its *
// * use in resulting scientific publications, and indicate your *
// * acceptance of all terms of the Geant4 Software license. *
// ********************************************************************
//
//
// Compound /vis/ commands - John Allison 15th May 2000
#include "G4VisCommandsCompound.hh"
#include "G4VisManager.hh"
#include "G4UImanager.hh"
#include "G4UIcommandTree.hh"
#include "G4UIcmdWithAString.hh"
#include <sstream>
#include <set>
#define G4warn G4cout
////////////// /vis/drawTree ///////////////////////////////////////
G4VisCommandDrawTree::G4VisCommandDrawTree() {
G4bool omitable;
fpCommand = new G4UIcommand("/vis/drawTree", this);
fpCommand->SetGuidance
("Produces a representation of the geometry hierarchy. Further"
"\nguidance is given on running the command. Or look at the guidance"
"\nfor \"/vis/ASCIITree/verbose\".");
fpCommand->SetGuidance("The pre-existing scene and view are preserved.");
G4UIparameter* parameter;
parameter = new G4UIparameter("physical-volume-name", 's', omitable = true);
parameter -> SetDefaultValue("world");
fpCommand -> SetParameter (parameter);
parameter = new G4UIparameter("system", 's', omitable = true);
parameter -> SetDefaultValue("ATree");
fpCommand -> SetParameter (parameter);
}
G4VisCommandDrawTree::~G4VisCommandDrawTree() {
delete fpCommand;
}
void G4VisCommandDrawTree::SetNewValue(G4UIcommand*, G4String newValue) {
G4String pvname, system;
std::istringstream is(newValue);
is >> pvname >> system;
// Note: The second parameter, "system", is intended to allow the user
// a choice of dedicated tree printing/displaying systems but at present
// the only such dedicated system is ASCIITree. It doesn't make sense to
// specify OGLSX, for example. So to avoid confusion we restrict this
// feature to systems that have "Tree" in the name or nickname.
// Of course, some other systems, such as OGLSQt, have a tree browser
// built-in. The HepRApp offline browser also has a tree browser
// built in.
if (!G4StrUtil::contains(system, "Tree")) {
system = "ATree";
}
G4VGraphicsSystem* keepSystem = fpVisManager->GetCurrentGraphicsSystem();
G4Scene* keepScene = fpVisManager->GetCurrentScene();
G4VSceneHandler* keepSceneHandler = fpVisManager->GetCurrentSceneHandler();
G4VViewer* keepViewer = fpVisManager->GetCurrentViewer();
G4VisManager::Verbosity keepVisVerbosity = fpVisManager->GetVerbosity();
G4bool keepAbleness = fpVisManager->GetConcreteInstance()? true: false;
G4UImanager* UImanager = G4UImanager::GetUIpointer();
G4int keepUIVerbose = UImanager->GetVerboseLevel();
G4int newVerbose(0);
if (keepUIVerbose >= 2 ||
fpVisManager->GetVerbosity() >= G4VisManager::confirmations)
newVerbose = 2;
UImanager->SetVerboseLevel(newVerbose);
auto errorCode = UImanager->ApplyCommand(G4String("/vis/open " + system));
if (errorCode == 0) {
if (!keepAbleness) { // Enable temporarily
fpVisManager->SetVerboseLevel("Quiet");
UImanager->ApplyCommand("/vis/enable");
fpVisManager->SetVerboseLevel(keepVisVerbosity);
}
UImanager->ApplyCommand("/vis/viewer/reset");
UImanager->ApplyCommand(G4String("/vis/drawVolume " + pvname));
UImanager->ApplyCommand("/vis/viewer/flush");
if (!keepAbleness) { // Disable again
fpVisManager->SetVerboseLevel("Quiet");
UImanager->ApplyCommand("/vis/disable");
fpVisManager->SetVerboseLevel(keepVisVerbosity);
}
if (keepViewer) {
if (fpVisManager->GetVerbosity() >= G4VisManager::warnings) {
G4warn << "Reverting to " << keepViewer->GetName() << G4endl;
}
fpVisManager->SetCurrentGraphicsSystem(keepSystem);
fpVisManager->SetCurrentScene(keepScene);
fpVisManager->SetCurrentSceneHandler(keepSceneHandler);
fpVisManager->SetCurrentViewer(keepViewer);
}
}
UImanager->SetVerboseLevel(keepUIVerbose);
}
////////////// /vis/drawView ///////////////////////////////////////
G4VisCommandDrawView::G4VisCommandDrawView() {
G4bool omitable;
fpCommand = new G4UIcommand("/vis/drawView", this);
fpCommand->SetGuidance
("Draw view from this angle, etc.");
G4UIparameter* parameter;
parameter = new G4UIparameter("theta-degrees", 'd', omitable = true);
parameter -> SetDefaultValue(0.);
fpCommand -> SetParameter (parameter);
parameter = new G4UIparameter("phi-degrees", 'd', omitable = true);
parameter -> SetDefaultValue(0.);
fpCommand -> SetParameter (parameter);
parameter = new G4UIparameter("pan-right", 'd', omitable = true);
parameter -> SetDefaultValue(0.);
fpCommand -> SetParameter (parameter);
parameter = new G4UIparameter("pan-up", 'd', omitable = true);
parameter -> SetDefaultValue(0.);
fpCommand -> SetParameter (parameter);
parameter = new G4UIparameter("pan-unit", 's', omitable = true);
parameter -> SetDefaultValue("cm");
fpCommand -> SetParameter (parameter);
parameter = new G4UIparameter("zoom-factor", 'd', omitable = true);
parameter -> SetDefaultValue(1.);
fpCommand -> SetParameter (parameter);
parameter = new G4UIparameter("dolly", 'd', omitable = true);
parameter -> SetDefaultValue(0.);
fpCommand -> SetParameter (parameter);
parameter = new G4UIparameter("dolly-unit", 's', omitable = true);
parameter -> SetDefaultValue("cm");
fpCommand -> SetParameter (parameter);
}
G4VisCommandDrawView::~G4VisCommandDrawView() {
delete fpCommand;
}
void G4VisCommandDrawView::SetNewValue(G4UIcommand*, G4String newValue) {
G4VisManager::Verbosity verbosity = fpVisManager->GetVerbosity();
G4VViewer* currentViewer = fpVisManager->GetCurrentViewer();
if (!currentViewer) {
if (verbosity >= G4VisManager::warnings) {
G4warn <<
"WARNING: G4VisCommandsDrawView::SetNewValue: no current viewer."
<< G4endl;
}
return;
}
G4String thetaDeg;
G4String phiDeg;
G4String panRight;
G4String panUp;
G4String panUnit;
G4String zoomFactor;
G4String dolly;
G4String dollyUnit;
std::istringstream is(newValue);
is >> thetaDeg >> phiDeg >> panRight >> panUp >> panUnit
>> zoomFactor >> dolly >> dollyUnit;
G4UImanager* UImanager = G4UImanager::GetUIpointer();
G4ViewParameters vp = currentViewer->GetViewParameters();
G4bool keepAutoRefresh = vp.IsAutoRefresh();
vp.SetAutoRefresh(false);
currentViewer->SetViewParameters(vp);
UImanager->ApplyCommand(
G4String("/vis/viewer/set/viewpointThetaPhi " + thetaDeg + " " + phiDeg + " deg"));
UImanager->ApplyCommand(
G4String("/vis/viewer/panTo " + panRight + " " + panUp + " " + panUnit));
UImanager->ApplyCommand(
G4String("/vis/viewer/zoomTo " + zoomFactor));
vp = currentViewer->GetViewParameters();
vp.SetAutoRefresh(keepAutoRefresh);
currentViewer->SetViewParameters(vp);
UImanager->ApplyCommand(
G4String("/vis/viewer/dollyTo " + dolly + " " + dollyUnit));
}
////////////// /vis/drawLogicalVolume ///////////////////////////////////////
G4VisCommandDrawLogicalVolume::G4VisCommandDrawLogicalVolume() {
fpCommand = new G4UIcommand("/vis/drawLogicalVolume", this);
fpCommand->SetGuidance
("Draws logical volume with additional components.");
fpCommand->SetGuidance
("Synonymous with \"/vis/specify\".");
fpCommand->SetGuidance
("Creates a scene consisting of this logical volume and asks the"
"\n current viewer to draw it. The scene becomes current.");
const G4UIcommandTree* tree = G4UImanager::GetUIpointer()->GetTree();
const G4UIcommand* addLogVolCmd = tree->FindPath("/vis/scene/add/logicalVolume");
// Pick up guidance from /vis/scene/add/logicalVolume
CopyGuidanceFrom(addLogVolCmd,fpCommand);
// Pick up parameters from /vis/scene/add/logicalVolume
CopyParametersFrom(addLogVolCmd,fpCommand);
}
G4VisCommandDrawLogicalVolume::~G4VisCommandDrawLogicalVolume() {
delete fpCommand;
}
void G4VisCommandDrawLogicalVolume::SetNewValue(G4UIcommand*, G4String newValue) {
G4VisManager::Verbosity verbosity = fpVisManager->GetVerbosity();
G4UImanager* UImanager = G4UImanager::GetUIpointer();
G4VViewer* currentViewer = fpVisManager->GetCurrentViewer();
const G4ViewParameters& currentViewParams = currentViewer->GetViewParameters();
G4bool keepAutoRefresh = currentViewParams.IsAutoRefresh();
if (keepAutoRefresh) UImanager->ApplyCommand("/vis/viewer/set/autoRefresh false");
UImanager->ApplyCommand("/vis/scene/create");
UImanager->ApplyCommand(G4String("/vis/scene/add/logicalVolume " + newValue));
UImanager->ApplyCommand("/vis/sceneHandler/attach");
G4ViewParameters::DrawingStyle keepDrawingStyle = currentViewParams.GetDrawingStyle();
if(keepDrawingStyle != G4ViewParameters::wireframe)
UImanager->ApplyCommand("/vis/viewer/set/style wireframe");
G4bool keepMarkerNotHidden = currentViewParams.IsMarkerNotHidden();
if (!keepMarkerNotHidden) UImanager->ApplyCommand("/vis/viewer/set/hiddenMarker false");
if (keepAutoRefresh) UImanager->ApplyCommand("/vis/viewer/set/autoRefresh true");
if (verbosity >= G4VisManager::warnings) {
if (keepDrawingStyle != currentViewParams.GetDrawingStyle()) {
G4warn
<< "Drawing style changed to wireframe. To restore previous style:";
G4String style, edge;
switch (keepDrawingStyle) {
case G4ViewParameters::wireframe:
style = "wireframe"; edge = "false"; break;
case G4ViewParameters::hlr:
style = "wireframe"; edge = "true"; break;
case G4ViewParameters::hsr:
style = "surface"; edge = "false"; break;
case G4ViewParameters::hlhsr:
style = "surface"; edge = "true"; break;
case G4ViewParameters::cloud:
style = "cloud"; edge = ""; break;
}
G4warn << "\n /vis/viewer/set/style " + style;
if (!edge.empty()) G4warn << "\n /vis/viewer/set/hiddenEdge " + edge;
G4warn << G4endl;
}
if (keepMarkerNotHidden != currentViewParams.IsMarkerNotHidden()) {
G4warn
<< "Markers changed to \"not hidden\". To restore previous condition:"
<< "\n /vis/viewer/set/hiddenmarker true"
<< G4endl;
}
}
static G4bool warned = false;
if (verbosity >= G4VisManager::confirmations && !warned) {
G4cout <<
"NOTE: For systems which are not \"auto-refresh\" you will need to"
"\n issue \"/vis/viewer/refresh\" or \"/vis/viewer/flush\"."
<< G4endl;
warned = true;
}
}
////////////// /vis/drawVolume ///////////////////////////////////////
G4VisCommandDrawVolume::G4VisCommandDrawVolume() {
fpCommand = new G4UIcommand("/vis/drawVolume", this);
fpCommand->SetGuidance
("Creates a scene containing this physical volume and asks the"
"\ncurrent viewer to draw it. The scene becomes current.");
const G4UIcommandTree* tree = G4UImanager::GetUIpointer()->GetTree();
const G4UIcommand* addVolCmd = tree->FindPath("/vis/scene/add/volume");
// Pick up guidance from /vis/scene/add/volume
CopyGuidanceFrom(addVolCmd,fpCommand);
// Pick up parameters from /vis/scene/add/volume
CopyParametersFrom(addVolCmd,fpCommand);
}
G4VisCommandDrawVolume::~G4VisCommandDrawVolume() {
delete fpCommand;
}
void G4VisCommandDrawVolume::SetNewValue(G4UIcommand*, G4String newValue) {
G4VisManager::Verbosity verbosity = fpVisManager->GetVerbosity();
G4UImanager* UImanager = G4UImanager::GetUIpointer();
UImanager->ApplyCommand("/vis/scene/create");
UImanager->ApplyCommand(G4String("/vis/scene/add/volume " + newValue));
UImanager->ApplyCommand("/vis/sceneHandler/attach");
static G4bool warned = false;
if (verbosity >= G4VisManager::confirmations && !warned) {
G4cout <<
"NOTE: For systems which are not \"auto-refresh\" you will need to"
"\n issue \"/vis/viewer/refresh\" or \"/vis/viewer/flush\"."
<< G4endl;
warned = true;
}
}
////////////// /vis/open ///////////////////////////////////////
G4VisCommandOpen::G4VisCommandOpen() {
G4bool omitable;
fpCommand = new G4UIcommand("/vis/open", this);
fpCommand->SetGuidance
("Creates a scene handler and viewer ready for drawing.");
fpCommand->SetGuidance
("The scene handler and viewer names are auto-generated.");
// Pick up guidance from /vis/viewer/create
const G4UIcommandTree* tree = G4UImanager::GetUIpointer()->GetTree();
const G4UIcommand* viewerCreateCmd = tree->FindPath("/vis/viewer/create");
CopyGuidanceFrom(viewerCreateCmd,fpCommand,2);
G4UIparameter* parameter;
parameter = new G4UIparameter("graphics-system-name", 's', omitable = true);
parameter->SetCurrentAsDefault(true);
fpCommand->SetParameter(parameter);
parameter = new G4UIparameter("window-size-hint", 's', omitable = true);
parameter->SetGuidance
("integer (pixels) for square window placed by window manager or"
" X-Windows-type geometry string, e.g. 600x600-100+100");
parameter->SetDefaultValue("none");
fpCommand->SetParameter(parameter);
}
G4VisCommandOpen::~G4VisCommandOpen() {
delete fpCommand;
}
G4String G4VisCommandOpen::GetCurrentValue(G4UIcommand*)
{
G4String graphicsSystemName;
auto graphicsSystem = fpVisManager->GetCurrentGraphicsSystem();
if (graphicsSystem) {
graphicsSystemName = graphicsSystem->GetName ();
}
else {
graphicsSystemName = "none";
}
return graphicsSystemName;
}
void G4VisCommandOpen::SetNewValue (G4UIcommand* command, G4String newValue)
{
G4String systemName, windowSizeHint;
std::istringstream is(newValue);
is >> systemName >> windowSizeHint;
G4UImanager* UImanager = G4UImanager::GetUIpointer();
auto errorCode = UImanager->ApplyCommand(G4String("/vis/sceneHandler/create " + systemName));
if (errorCode) {
G4ExceptionDescription ed;
ed << "sub-command \"/vis/sceneHandler/create\" failed.";
command->CommandFailed(errorCode,ed);
goto finish;
}
errorCode = UImanager->ApplyCommand(G4String("/vis/viewer/create ! ! " + windowSizeHint));
if (errorCode) {
G4ExceptionDescription ed;
ed << "sub-command \"/vis/viewer/create\" failed.";
command->CommandFailed(errorCode,ed);
goto finish;
}
finish:
if (errorCode) {
G4ExceptionDescription ed;
ed << "Invoked command has failed - see above. Available graphics systems are:\n ";
fpVisManager->PrintAvailableGraphicsSystems(G4VisManager::warnings,ed);
command->CommandFailed(errorCode,ed);
}
}
////////////// /vis/plot ///////////////////////////////////////
G4VisCommandPlot::G4VisCommandPlot ()
{
G4bool omitable;
G4UIparameter* parameter;
fpCommand = new G4UIcommand("/vis/plot", this);
fpCommand -> SetGuidance("Draws plots.");
parameter = new G4UIparameter ("type", 's', omitable = false);
parameter -> SetParameterCandidates("h1 h2");
fpCommand -> SetParameter (parameter);
parameter = new G4UIparameter ("id", 'i', omitable = false);
fpCommand -> SetParameter (parameter);
}
G4VisCommandPlot::~G4VisCommandPlot ()
{
delete fpCommand;
}
G4String G4VisCommandPlot::GetCurrentValue (G4UIcommand*)
{
return "";
}
void G4VisCommandPlot::SetNewValue (G4UIcommand*, G4String newValue)
{
auto currentViewer = fpVisManager->GetCurrentViewer();
if (currentViewer->GetName().find("TOOLSSG") == std::string::npos) {
G4warn <<
"WARNING: Current viewer not able to draw plots."
"\n Try \"/vis/open TSG\", then \"/vis/plot " << newValue << "\" again."
<< G4endl;
return;
}
G4String type, id;
std::istringstream is (newValue);
is >> type >> id;
auto keepEnable = fpVisManager->IsEnabled();
auto ui = G4UImanager::GetUIpointer();
ui->ApplyCommand("/vis/enable");
ui->ApplyCommand("/vis/viewer/resetCameraParameters");
ui->ApplyCommand("/vis/scene/create");
ui->ApplyCommand("/vis/scene/endOfEventAction accumulate 0"); // Don't keep events
static G4int plotterID = 0;
std::ostringstream ossPlotter;
ossPlotter << "plotter-" << plotterID++;
const G4String& plotterName = ossPlotter.str();
ui->ApplyCommand("/vis/plotter/create " + plotterName);
ui->ApplyCommand("/vis/scene/add/plotter " + plotterName);
ui->ApplyCommand("/vis/plotter/add/" + type + ' ' + id + ' ' + plotterName);
ui->ApplyCommand("/vis/sceneHandler/attach");
if (!keepEnable) {
fpVisManager->Disable();
G4warn <<
"WARNING: drawing was enabled for plotting but is now restored to disabled mode."
<< G4endl;
}
}
////////////// /vis/specify ///////////////////////////////////////
G4VisCommandSpecify::G4VisCommandSpecify() {
G4bool omitable;
fpCommand = new G4UIcommand("/vis/specify", this);
fpCommand->SetGuidance
("Draws logical volume with Boolean components, voxels and readout geometry.");
fpCommand->SetGuidance
("Synonymous with \"/vis/drawLogicalVolume\".");
fpCommand->SetGuidance
("Creates a scene consisting of this logical volume and asks the"
"\n current viewer to draw it to the specified depth of descent"
"\n showing boolean components (if any), voxels (if any),"
"\n readout geometry (if any), local axes and overlaps (if any),"
"\n under control of the appropriate flag.");
fpCommand->SetGuidance
("Note: voxels are not constructed until start of run - /run/beamOn."
"\n (For voxels without a run, \"/run/beamOn 0\".)");
fpCommand->SetGuidance("The scene becomes current.");
G4UIparameter* parameter;
parameter = new G4UIparameter("logical-volume-name", 's', omitable = false);
fpCommand->SetParameter(parameter);
parameter = new G4UIparameter("depth-of-descent", 'i', omitable = true);
parameter->SetDefaultValue(1);
fpCommand->SetParameter(parameter);
parameter = new G4UIparameter("booleans-flag", 'b', omitable = true);
parameter->SetDefaultValue(true);
fpCommand->SetParameter(parameter);
parameter = new G4UIparameter("voxels-flag", 'b', omitable = true);
parameter->SetDefaultValue(true);
fpCommand->SetParameter(parameter);
parameter = new G4UIparameter("readout-flag", 'b', omitable = true);
parameter->SetDefaultValue(true);
fpCommand->SetParameter(parameter);
parameter = new G4UIparameter("axes-flag", 'b', omitable = true);
parameter->SetDefaultValue(true);
parameter -> SetGuidance ("Set \"false\" to suppress axes.");
fpCommand->SetParameter(parameter);
parameter = new G4UIparameter("check-overlap-flag", 'b', omitable = true);
parameter->SetDefaultValue(true);
parameter -> SetGuidance ("Set \"false\" to suppress overlap check.");
fpCommand->SetParameter(parameter);
}
G4VisCommandSpecify::~G4VisCommandSpecify() {
delete fpCommand;
}
void G4VisCommandSpecify::SetNewValue(G4UIcommand*, G4String newValue) {
G4VisManager::Verbosity verbosity = fpVisManager->GetVerbosity();
G4UImanager* UImanager = G4UImanager::GetUIpointer();
// UImanager->ApplyCommand(G4String("/geometry/print " + newValue));
UImanager->ApplyCommand("/vis/scene/create");
UImanager->ApplyCommand(G4String("/vis/scene/add/logicalVolume " + newValue));
UImanager->ApplyCommand("/vis/sceneHandler/attach");
static G4bool warned = false;
if (verbosity >= G4VisManager::confirmations && !warned) {
G4cout <<
"NOTE: For systems which are not \"auto-refresh\" you will need to"
"\n issue \"/vis/viewer/refresh\" or \"/vis/viewer/flush\"."
<< G4endl;
warned = true;
}
}