366 lines
13 KiB
C++
366 lines
13 KiB
C++
/***************************************************************************
|
|
* Copyright (c) 2002 Jürgen Riegel <juergen.riegel@web.de> *
|
|
* *
|
|
* This file is part of the FreeCAD CAx development system. *
|
|
* *
|
|
* This library is free software; you can redistribute it and/or *
|
|
* modify it under the terms of the GNU Library General Public *
|
|
* License as published by the Free Software Foundation; either *
|
|
* version 2 of the License, or (at your option) any later version. *
|
|
* *
|
|
* This library is distributed in the hope that it will be useful, *
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
* GNU Library General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU Library General Public *
|
|
* License along with this library; see the file COPYING.LIB. If not, *
|
|
* write to the Free Software Foundation, Inc., 59 Temple Place, *
|
|
* Suite 330, Boston, MA 02111-1307, USA *
|
|
* *
|
|
***************************************************************************/
|
|
|
|
|
|
#include <App/DocumentObjectGroup.h>
|
|
#include <App/GroupExtension.h>
|
|
#include <App/Part.h>
|
|
#include "Application.h"
|
|
#include "Action.h"
|
|
#include "cet_lut.hpp"
|
|
#include "CommandT.h"
|
|
#include "DockWindowManager.h"
|
|
#include "Document.h"
|
|
#include "PythonConsole.h"
|
|
#include "Selection.h"
|
|
#include "SelectionObject.h"
|
|
#include "ViewProvider.h"
|
|
#include "ViewProviderDocumentObject.h"
|
|
#include "ViewProviderLink.h"
|
|
|
|
using namespace Gui;
|
|
|
|
|
|
|
|
//===========================================================================
|
|
// Std_Recompute
|
|
//===========================================================================
|
|
|
|
DEF_STD_CMD(StdCmdFeatRecompute)
|
|
|
|
StdCmdFeatRecompute::StdCmdFeatRecompute()
|
|
:Command("Std_Recompute")
|
|
{
|
|
// setting the
|
|
sGroup = "File";
|
|
sMenuText = QT_TR_NOOP("&Recompute");
|
|
sToolTipText = QT_TR_NOOP("Recomputes a feature or document");
|
|
sWhatsThis = "Std_Recompute";
|
|
sStatusTip = sToolTipText;
|
|
sPixmap = "view-refresh";
|
|
sAccel = "Ctrl+R";
|
|
}
|
|
|
|
void StdCmdFeatRecompute::activated(int iMsg)
|
|
{
|
|
Q_UNUSED(iMsg);
|
|
}
|
|
|
|
//===========================================================================
|
|
// Std_RandomColor
|
|
//===========================================================================
|
|
|
|
DEF_STD_CMD_A(StdCmdRandomColor)
|
|
|
|
StdCmdRandomColor::StdCmdRandomColor()
|
|
:Command("Std_RandomColor")
|
|
{
|
|
sGroup = "File";
|
|
sMenuText = QT_TR_NOOP("Random &Color");
|
|
sToolTipText = QT_TR_NOOP("Assigns random diffuse colors for the selected objects");
|
|
sWhatsThis = "Std_RandomColor";
|
|
sStatusTip = sToolTipText;
|
|
sPixmap = "Std_RandomColor";
|
|
}
|
|
|
|
void StdCmdRandomColor::activated(int iMsg)
|
|
{
|
|
Q_UNUSED(iMsg);
|
|
|
|
auto setRandomColor = [](ViewProvider* view) {
|
|
// NOLINTBEGIN
|
|
int colIndex = rand() % (CET::R1.size() / 3) * 3;
|
|
float fRed = CET::R1[colIndex] / 255.0F;
|
|
float fGrn = CET::R1[colIndex + 1] / 255.0F;
|
|
float fBlu = CET::R1[colIndex + 2] / 255.0F;
|
|
// NOLINTEND
|
|
auto objColor = Base::Color(fRed, fGrn, fBlu);
|
|
|
|
auto vpLink = freecad_cast<ViewProviderLink*>(view);
|
|
if (vpLink) {
|
|
if (!vpLink->OverrideMaterial.getValue()) {
|
|
vpLink->OverrideMaterial.setValue(true);
|
|
}
|
|
vpLink->ShapeMaterial.setDiffuseColor(objColor);
|
|
}
|
|
else if (view) {
|
|
// clang-format off
|
|
// get the view provider of the selected object and set the shape color
|
|
if (auto prop = dynamic_cast<App::PropertyMaterialList*>(view->getPropertyByName("ShapeAppearance"))) {
|
|
prop->setDiffuseColor(objColor);
|
|
}
|
|
else if (auto prop = dynamic_cast<App::PropertyMaterial*>(view->getPropertyByName("ShapeAppearance"))) {
|
|
prop->setDiffuseColor(objColor);
|
|
}
|
|
// clang-format on
|
|
}
|
|
};
|
|
|
|
auto allowToChangeColor = [](const App::DocumentObject* obj) {
|
|
return (obj->isDerivedFrom<App::Part>() || obj->isDerivedFrom<App::DocumentObjectGroup>());
|
|
};
|
|
|
|
// get the complete selection
|
|
std::vector<SelectionSingleton::SelObj> sel = Selection().getCompleteSelection();
|
|
|
|
Command::openCommand(QT_TRANSLATE_NOOP("Command", "Set Random Color"));
|
|
for (const auto & it : sel) {
|
|
ViewProvider* view = Application::Instance->getViewProvider(it.pObject);
|
|
setRandomColor(view);
|
|
|
|
if (auto grp = it.pObject->getExtension<App::GroupExtension>()) {
|
|
if (allowToChangeColor(it.pObject)) {
|
|
std::vector<App::DocumentObject*> objs = grp->getObjects();
|
|
for (auto obj : objs) {
|
|
ViewProvider* view = Application::Instance->getViewProvider(obj);
|
|
setRandomColor(view);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
Command::commitCommand();
|
|
}
|
|
|
|
bool StdCmdRandomColor::isActive()
|
|
{
|
|
return (Gui::Selection().size() != 0);
|
|
}
|
|
|
|
//===========================================================================
|
|
// Std_ToggleFreeze
|
|
//===========================================================================
|
|
DEF_STD_CMD_A(StdCmdToggleFreeze)
|
|
|
|
StdCmdToggleFreeze::StdCmdToggleFreeze()
|
|
: Command("Std_ToggleFreeze")
|
|
{
|
|
sGroup = "File";
|
|
sMenuText = QT_TR_NOOP("Toggle Freeze");
|
|
static std::string toolTip = std::string("<p>")
|
|
+ QT_TR_NOOP("Toggles freeze state of the selected objects. A frozen object is not recomputed when its parents change.")
|
|
+ "</p>";
|
|
sToolTipText = toolTip.c_str();
|
|
sStatusTip = sToolTipText;
|
|
sWhatsThis = "Std_ToggleFreeze";
|
|
sPixmap = "Std_ToggleFreeze";
|
|
sAccel = "";
|
|
eType = AlterDoc;
|
|
}
|
|
|
|
void StdCmdToggleFreeze::activated(int iMsg)
|
|
{
|
|
Q_UNUSED(iMsg);
|
|
|
|
std::vector<Gui::SelectionSingleton::SelObj> sels = Gui::Selection().getCompleteSelection();
|
|
|
|
Command::openCommand(QT_TRANSLATE_NOOP("Command", "Toggle freeze"));
|
|
for (Gui::SelectionSingleton::SelObj& sel : sels) {
|
|
App::DocumentObject* obj = sel.pObject;
|
|
if (!obj) {
|
|
continue;
|
|
}
|
|
|
|
if (obj->isFreezed()){
|
|
obj->unfreeze();
|
|
} else {
|
|
obj->freeze();
|
|
}
|
|
|
|
}
|
|
Command::commitCommand();
|
|
}
|
|
|
|
bool StdCmdToggleFreeze::isActive()
|
|
{
|
|
return (Gui::Selection().size() != 0);
|
|
}
|
|
|
|
|
|
|
|
|
|
//===========================================================================
|
|
// Std_SendToPythonConsole
|
|
//===========================================================================
|
|
|
|
DEF_STD_CMD_A(StdCmdSendToPythonConsole)
|
|
|
|
StdCmdSendToPythonConsole::StdCmdSendToPythonConsole()
|
|
:Command("Std_SendToPythonConsole")
|
|
{
|
|
// setting the
|
|
sGroup = "Edit";
|
|
sMenuText = QT_TR_NOOP("&Send to Python Console");
|
|
sToolTipText = QT_TR_NOOP("Sends the selected object to the Python console");
|
|
sWhatsThis = "Std_SendToPythonConsole";
|
|
sStatusTip = sToolTipText;
|
|
sPixmap = "applications-python";
|
|
sAccel = "Ctrl+Shift+P";
|
|
}
|
|
|
|
bool StdCmdSendToPythonConsole::isActive()
|
|
{
|
|
//active only if either 1 object is selected or multiple subobjects from the same object
|
|
return Gui::Selection().getSelectionEx().size() == 1;
|
|
}
|
|
|
|
void StdCmdSendToPythonConsole::activated(int iMsg)
|
|
{
|
|
Q_UNUSED(iMsg);
|
|
const std::vector<Gui::SelectionObject> &sels = Gui::Selection().getSelectionEx("*", App::DocumentObject::getClassTypeId(),
|
|
ResolveMode::OldStyleElement, false);
|
|
if (sels.empty())
|
|
return;
|
|
const App::DocumentObject *obj = sels[0].getObject();
|
|
if (!obj)
|
|
return;
|
|
QString docname = QString::fromLatin1(obj->getDocument()->getName());
|
|
QString objname = QString::fromLatin1(obj->getNameInDocument());
|
|
try {
|
|
// clear variables from previous run, if any
|
|
QString cmd = QLatin1String("try:\n del(doc,lnk,obj,shp,sub,subs)\nexcept Exception:\n pass\n");
|
|
Gui::Command::runCommand(Gui::Command::Gui,cmd.toLatin1());
|
|
cmd = QStringLiteral("doc = App.getDocument(\"%1\")").arg(docname);
|
|
Gui::Command::runCommand(Gui::Command::Gui,cmd.toLatin1());
|
|
//support links
|
|
if (obj->isDerivedFrom<App::Link>()) {
|
|
cmd = QStringLiteral("lnk = doc.getObject(\"%1\")").arg(objname);
|
|
Gui::Command::runCommand(Gui::Command::Gui,cmd.toLatin1());
|
|
cmd = QStringLiteral("obj = lnk.getLinkedObject()");
|
|
Gui::Command::runCommand(Gui::Command::Gui,cmd.toLatin1());
|
|
const auto link = static_cast<const App::Link*>(obj);
|
|
obj = link->getLinkedObject();
|
|
} else {
|
|
cmd = QStringLiteral("obj = doc.getObject(\"%1\")").arg(objname);
|
|
Gui::Command::runCommand(Gui::Command::Gui,cmd.toLatin1());
|
|
}
|
|
if (obj->isDerivedFrom<App::GeoFeature>()) {
|
|
const auto geoObj = static_cast<const App::GeoFeature*>(obj);
|
|
const App::PropertyGeometry* geo = geoObj->getPropertyOfGeometry();
|
|
if (geo){
|
|
cmd = QStringLiteral("shp = obj.") + QLatin1String(geo->getName()); //"Shape", "Mesh", "Points", etc.
|
|
Gui::Command::runCommand(Gui::Command::Gui, cmd.toLatin1());
|
|
if (sels[0].hasSubNames()) {
|
|
std::vector<std::string> subnames = sels[0].getSubNames();
|
|
QString subname = QString::fromLatin1(subnames[0].c_str());
|
|
cmd = QStringLiteral("sub = obj.getSubObject(\"%1\")").arg(subname);
|
|
Gui::Command::runCommand(Gui::Command::Gui,cmd.toLatin1());
|
|
if (subnames.size() > 1) {
|
|
std::ostringstream strm;
|
|
strm << "subs = [";
|
|
for (const auto & subname : subnames) {
|
|
strm << "obj.getSubObject(\"" << subname << "\"),";
|
|
}
|
|
strm << "]";
|
|
Gui::Command::runCommand(Gui::Command::Gui, strm.str().c_str());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
//show the python console if it's not already visible, and set the keyboard focus to it
|
|
QWidget* pc = DockWindowManager::instance()->getDockWindow("Python console");
|
|
auto pcPython = qobject_cast<PythonConsole*>(pc);
|
|
if (pcPython) {
|
|
DockWindowManager::instance()->activate(pcPython);
|
|
pcPython->setFocus();
|
|
}
|
|
}
|
|
catch (const Base::Exception& e) {
|
|
e.reportException();
|
|
}
|
|
|
|
}
|
|
|
|
//===========================================================================
|
|
// Std_ToggleSkipRecompute
|
|
//===========================================================================
|
|
|
|
DEF_STD_CMD_AC(StdCmdToggleSkipRecompute)
|
|
|
|
StdCmdToggleSkipRecompute::StdCmdToggleSkipRecompute()
|
|
: Command("Std_ToggleSkipRecompute")
|
|
{
|
|
sGroup = "File";
|
|
sMenuText = QT_TR_NOOP("Skip Recomputes");
|
|
|
|
static std::string toolTip = QT_TR_NOOP("Enables or disables the recomputations of the document");
|
|
|
|
sToolTipText = toolTip.c_str();
|
|
sStatusTip = sToolTipText;
|
|
sWhatsThis = "Std_ToggleSkipRecompute";
|
|
eType = AlterDoc;
|
|
}
|
|
|
|
Gui::Action* StdCmdToggleSkipRecompute::createAction()
|
|
{
|
|
Action* pcAction = Command::createAction();
|
|
pcAction->setCheckable(true);
|
|
pcAction->setIcon(QIcon());
|
|
_pcAction = pcAction;
|
|
isActive();
|
|
return pcAction;
|
|
}
|
|
|
|
void StdCmdToggleSkipRecompute::activated(int iMsg)
|
|
{
|
|
const auto doc = this->getDocument();
|
|
if (doc == nullptr) {
|
|
return;
|
|
}
|
|
|
|
Command::openCommand(QT_TRANSLATE_NOOP("Command", "Skip recomputes"));
|
|
doc->setStatus(App::Document::SkipRecompute, (bool) iMsg);
|
|
if (_pcAction) {
|
|
_pcAction->setChecked((bool) iMsg);
|
|
}
|
|
Command::commitCommand();
|
|
}
|
|
|
|
bool StdCmdToggleSkipRecompute::isActive()
|
|
{
|
|
const auto doc = this->getDocument();
|
|
if (doc == nullptr) {
|
|
return false;
|
|
}
|
|
|
|
const bool skipRecomputeStatus = doc->testStatus(App::Document::SkipRecompute);
|
|
if (_pcAction && _pcAction->isChecked() != skipRecomputeStatus) {
|
|
_pcAction->setChecked(skipRecomputeStatus);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
namespace Gui {
|
|
|
|
void CreateFeatCommands()
|
|
{
|
|
CommandManager &rcCmdMgr = Application::Instance->commandManager();
|
|
|
|
rcCmdMgr.addCommand(new StdCmdFeatRecompute());
|
|
rcCmdMgr.addCommand(new StdCmdToggleFreeze());
|
|
rcCmdMgr.addCommand(new StdCmdRandomColor());
|
|
rcCmdMgr.addCommand(new StdCmdSendToPythonConsole());
|
|
rcCmdMgr.addCommand(new StdCmdToggleSkipRecompute());
|
|
}
|
|
|
|
} // namespace Gui
|