Files
create/src/Mod/Fem/Gui/ViewProviderFemPostFunction.cpp

1290 lines
42 KiB
C++

/***************************************************************************
* Copyright (c) 2015 Stefan Tröger <stefantroeger@gmx.net> *
* *
* 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 "PreCompiled.h"
#ifndef _PreComp_
#include <cmath>
#include <Inventor/actions/SoSearchAction.h>
#include <Inventor/draggers/SoHandleBoxDragger.h>
#include <Inventor/draggers/SoJackDragger.h>
#include <Inventor/manips/SoCenterballManip.h>
#include <Inventor/manips/SoHandleBoxManip.h>
#include <Inventor/manips/SoJackManip.h>
#include <Inventor/manips/SoTransformManip.h>
#include <Inventor/nodes/SoCoordinate3.h>
#include <Inventor/nodes/SoLineSet.h>
#include <Inventor/nodes/SoMaterial.h>
#include <Inventor/nodes/SoSeparator.h>
#include <Inventor/nodes/SoTransform.h>
#include <QApplication>
#include <QMessageBox>
#include <QTextStream>
#endif
#include <App/Document.h>
#include <Base/UnitsApi.h>
#include <Gui/Application.h>
#include <Gui/Control.h>
#include <Gui/Document.h>
#include <Gui/MainWindow.h>
#include <Gui/TaskView/TaskDialog.h>
#include <Gui/View3DInventor.h>
#include <Gui/View3DInventorViewer.h>
#include "FemSettings.h"
#include "TaskPostBoxes.h"
#include "ViewProviderAnalysis.h"
#include "ViewProviderFemPostFunction.h"
#include "ui_BoxWidget.h"
#include "ui_CylinderWidget.h"
#include "ui_PlaneWidget.h"
#include "ui_SphereWidget.h"
using namespace FemGui;
namespace sp = std::placeholders;
void FunctionWidget::setViewProvider(ViewProviderFemPostFunction* view)
{
// NOLINTBEGIN
m_view = view;
m_object = view->getObject<Fem::FemPostFunction>();
m_connection = m_object->getDocument()->signalChangedObject.connect(
std::bind(&FunctionWidget::onObjectsChanged, this, sp::_1, sp::_2));
// NOLINTEND
}
void FunctionWidget::onObjectsChanged(const App::DocumentObject& obj, const App::Property& p)
{
if (&obj == m_object) {
onChange(p);
}
}
PROPERTY_SOURCE(FemGui::ViewProviderFemPostFunctionProvider, Gui::ViewProviderDocumentObject)
ViewProviderFemPostFunctionProvider::ViewProviderFemPostFunctionProvider() = default;
ViewProviderFemPostFunctionProvider::~ViewProviderFemPostFunctionProvider() = default;
void ViewProviderFemPostFunctionProvider::onChanged(const App::Property* prop)
{
Gui::ViewProviderDocumentObject::onChanged(prop);
updateSize();
}
void ViewProviderFemPostFunctionProvider::updateData(const App::Property* prop)
{
Gui::ViewProviderDocumentObject::updateData(prop);
Fem::FemPostFunctionProvider* obj = getObject<Fem::FemPostFunctionProvider>();
if (prop == &obj->Group) {
updateSize();
}
}
void ViewProviderFemPostFunctionProvider::updateSize()
{
std::vector<App::DocumentObject*> vec = claimChildren();
for (auto it : vec) {
if (!it->isDerivedFrom<Fem::FemPostFunction>()) {
continue;
}
ViewProviderFemPostFunction* vp = static_cast<FemGui::ViewProviderFemPostFunction*>(
Gui::Application::Instance->getViewProvider(it));
vp->AutoScaleFactorX.setValue(SizeX.getValue());
vp->AutoScaleFactorY.setValue(SizeY.getValue());
vp->AutoScaleFactorZ.setValue(SizeZ.getValue());
}
}
bool ViewProviderFemPostFunctionProvider::onDelete(const std::vector<std::string>&)
{
// warn the user if the object has unselected children
auto objs = claimChildren();
return ViewProviderFemAnalysis::checkSelectedChildren(objs,
this->getDocument(),
"functions list");
}
bool ViewProviderFemPostFunctionProvider::canDelete(App::DocumentObject* obj) const
{
// deletions of objects from a FemFunction don't necessarily destroy anything
// thus we can pass this action
// we can warn the user if necessary in the object's ViewProvider in the onDelete() function
Q_UNUSED(obj)
return true;
}
// ***************************************************************************
PROPERTY_SOURCE(FemGui::ViewProviderFemPostFunction, Gui::ViewProviderDocumentObject)
ViewProviderFemPostFunction::ViewProviderFemPostFunction()
: m_manip(nullptr)
, m_autoscale(false)
, m_isDragging(false)
, m_autoRecompute(false)
{
ADD_PROPERTY_TYPE(AutoScaleFactorX,
(1),
"AutoScale",
App::Prop_Hidden,
"Automatic scaling factor");
ADD_PROPERTY_TYPE(AutoScaleFactorY,
(1),
"AutoScale",
App::Prop_Hidden,
"Automatic scaling factor");
ADD_PROPERTY_TYPE(AutoScaleFactorZ,
(1),
"AutoScale",
App::Prop_Hidden,
"Automatic scaling factor");
m_geometrySeperator = new SoSeparator();
m_geometrySeperator->ref();
m_scale = new SoScale();
m_scale->ref();
m_scale->scaleFactor = SbVec3f(1, 1, 1);
}
ViewProviderFemPostFunction::~ViewProviderFemPostFunction()
{
m_geometrySeperator->unref();
m_manip->unref();
m_scale->unref();
}
void ViewProviderFemPostFunction::attach(App::DocumentObject* pcObj)
{
ViewProviderDocumentObject::attach(pcObj);
// setup the graph for editing the function unit geometry
SoMaterial* color = new SoMaterial();
color->diffuseColor.setValue(0, 0, 1);
color->transparency.setValue(0.5);
SoTransform* transform = new SoTransform();
m_manip = setupManipulator();
m_manip->ref();
SoSeparator* pcEditNode = new SoSeparator();
pcEditNode->ref();
pcEditNode->addChild(color);
pcEditNode->addChild(transform);
pcEditNode->addChild(m_geometrySeperator);
m_geometrySeperator->insertChild(m_scale, 0);
// Now we replace the SoTransform node by a manipulator
// Note: Even SoCenterballManip inherits from SoTransform
// we cannot use it directly (in above code) because the
// translation and center fields are overridden.
SoSearchAction sa;
sa.setInterest(SoSearchAction::FIRST);
sa.setSearchingAll(FALSE);
sa.setNode(transform);
sa.apply(pcEditNode);
SoPath* path = sa.getPath();
if (path) {
m_manip->replaceNode(path);
SoDragger* dragger = m_manip->getDragger();
dragger->addStartCallback(dragStartCallback, this);
dragger->addFinishCallback(dragFinishCallback, this);
dragger->addMotionCallback(dragMotionCallback, this);
}
addDisplayMaskMode(pcEditNode, "Default");
setDisplayMaskMode("Default");
pcEditNode->unref();
}
SbBox3f ViewProviderFemPostFunction::getBoundingsOfView() const
{
SbBox3f box;
Gui::Document* doc = this->getDocument();
Gui::View3DInventor* view =
qobject_cast<Gui::View3DInventor*>(doc->getViewOfViewProvider(this));
if (view) {
Gui::View3DInventorViewer* viewer = view->getViewer();
box = viewer->getBoundingBox();
}
return box;
}
bool ViewProviderFemPostFunction::findScaleFactor(double& scale) const
{
SbBox3f bbox = getBoundingsOfView();
if (bbox.hasVolume()) {
float dx, dy, dz;
bbox.getSize(dx, dy, dz);
// we want the manipulator to have 20 % of the max size of the object
scale = 0.2 * std::max(std::max(dx, dy), dz);
return true;
}
return false;
}
bool ViewProviderFemPostFunction::doubleClicked()
{
Gui::Application::Instance->activeDocument()->setEdit(this, (int)ViewProvider::Default);
return true;
}
SoTransformManip* ViewProviderFemPostFunction::setupManipulator()
{
return new SoCenterballManip;
}
std::vector<std::string> ViewProviderFemPostFunction::getDisplayModes() const
{
std::vector<std::string> StrList;
StrList.emplace_back("Default");
return StrList;
}
void ViewProviderFemPostFunction::dragStartCallback(void* data, SoDragger*)
{
// This is called when a manipulator is about to manipulating
Gui::Application::Instance->activeDocument()->openCommand(
QT_TRANSLATE_NOOP("Command", "Edit Mirror"));
static_cast<ViewProviderFemPostFunction*>(data)->m_isDragging = true;
ViewProviderFemPostFunction* that = static_cast<ViewProviderFemPostFunction*>(data);
that->m_autoRecompute = FemSettings().getPostAutoRecompute();
}
void ViewProviderFemPostFunction::dragFinishCallback(void* data, SoDragger*)
{
// This is called when a manipulator has done manipulating
Gui::Application::Instance->activeDocument()->commitCommand();
ViewProviderFemPostFunction* that = static_cast<ViewProviderFemPostFunction*>(data);
if (that->m_autoRecompute) {
that->getObject()->getDocument()->recompute();
}
static_cast<ViewProviderFemPostFunction*>(data)->m_isDragging = false;
}
void ViewProviderFemPostFunction::dragMotionCallback(void* data, SoDragger* drag)
{
ViewProviderFemPostFunction* that = static_cast<ViewProviderFemPostFunction*>(data);
that->draggerUpdate(drag);
if (that->m_autoRecompute) {
that->getObject()->getDocument()->recompute();
}
}
bool ViewProviderFemPostFunction::setEdit(int ModNum)
{
if (ModNum == ViewProvider::Default || ModNum == 1) {
Gui::TaskView::TaskDialog* dlg = Gui::Control().activeDialog();
TaskDlgPost* postDlg = qobject_cast<TaskDlgPost*>(dlg);
if (postDlg && postDlg->getView() != this) {
postDlg = nullptr; // another pad left open its task panel
}
if (dlg && !postDlg) {
QMessageBox msgBox(Gui::getMainWindow());
msgBox.setText(QObject::tr("A dialog is already open in the task panel"));
msgBox.setInformativeText(QObject::tr("Do you want to close this dialog?"));
msgBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No);
msgBox.setDefaultButton(QMessageBox::Yes);
int ret = msgBox.exec();
if (ret == QMessageBox::Yes) {
Gui::Control().reject();
}
else {
return false;
}
}
// start the edit dialog
if (postDlg) {
Gui::Control().showDialog(postDlg);
}
else {
postDlg = new TaskDlgPost(this);
auto panel = new TaskPostFunction(this);
postDlg->addTaskBox(panel->windowIcon().pixmap(32), panel);
Gui::Control().showDialog(postDlg);
}
return true;
}
else {
return ViewProviderDocumentObject::setEdit(ModNum);
}
}
void ViewProviderFemPostFunction::unsetEdit(int ModNum)
{
if (ModNum == ViewProvider::Default) {
// when pressing ESC make sure to close the dialog
Gui::Control().closeDialog();
}
else {
ViewProviderDocumentObject::unsetEdit(ModNum);
}
}
void ViewProviderFemPostFunction::onChanged(const App::Property* prop)
{
Gui::ViewProviderDocumentObject::onChanged(prop);
if (m_autoscale) {
m_scale->scaleFactor = SbVec3f(AutoScaleFactorX.getValue(),
AutoScaleFactorY.getValue(),
AutoScaleFactorZ.getValue());
}
}
// ***************************************************************************
PROPERTY_SOURCE(FemGui::ViewProviderFemPostBoxFunction, FemGui::ViewProviderFemPostFunction)
ViewProviderFemPostBoxFunction::ViewProviderFemPostBoxFunction()
{
sPixmap = "fem-post-geo-box";
setAutoScale(false);
// setup the visualisation geometry
getGeometryNode()->addChild(ShapeNodes::postBox());
}
ViewProviderFemPostBoxFunction::~ViewProviderFemPostBoxFunction() = default;
void ViewProviderFemPostBoxFunction::draggerUpdate(SoDragger* m)
{
Fem::FemPostBoxFunction* func = getObject<Fem::FemPostBoxFunction>();
SoHandleBoxDragger* dragger = static_cast<SoHandleBoxDragger*>(m);
const SbVec3f& center = dragger->translation.getValue();
SbVec3f scale = dragger->scaleFactor.getValue();
func->Center.setValue(center[0], center[1], center[2]);
func->Length.setValue(scale[0]);
func->Width.setValue(scale[1]);
func->Height.setValue(scale[2]);
}
void ViewProviderFemPostBoxFunction::updateData(const App::Property* p)
{
Fem::FemPostBoxFunction* func = getObject<Fem::FemPostBoxFunction>();
if (!isDragging()
&& (p == &func->Center || p == &func->Length || p == &func->Width || p == &func->Height)) {
const Base::Vector3d& center = func->Center.getValue();
float l = func->Length.getValue();
float w = func->Width.getValue();
float h = func->Height.getValue();
SbMatrix s, t;
s.setScale(SbVec3f(l, w, h));
t.setTranslate(SbVec3f(center.x, center.y, center.z));
s.multRight(t);
getManipulator()->setMatrix(s);
}
Gui::ViewProviderDocumentObject::updateData(p);
}
SoTransformManip* ViewProviderFemPostBoxFunction::setupManipulator()
{
return new SoHandleBoxManip;
}
FunctionWidget* ViewProviderFemPostBoxFunction::createControlWidget()
{
return new BoxWidget();
}
BoxWidget::BoxWidget()
: ui(new Ui_BoxWidget)
{
ui->setupUi(this);
QSize size = ui->centerX->sizeForText(QStringLiteral("000000000000"));
ui->centerX->setMinimumWidth(size.width());
ui->centerY->setMinimumWidth(size.width());
ui->centerZ->setMinimumWidth(size.width());
ui->length->setMinimumWidth(size.width());
ui->width->setMinimumWidth(size.width());
ui->height->setMinimumWidth(size.width());
int UserDecimals = Base::UnitsApi::getDecimals();
ui->centerX->setDecimals(UserDecimals);
ui->centerY->setDecimals(UserDecimals);
ui->centerZ->setDecimals(UserDecimals);
ui->length->setDecimals(UserDecimals);
ui->width->setDecimals(UserDecimals);
ui->height->setDecimals(UserDecimals);
connect(ui->centerX,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&BoxWidget::centerChanged);
connect(ui->centerY,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&BoxWidget::centerChanged);
connect(ui->centerZ,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&BoxWidget::centerChanged);
connect(ui->length,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&BoxWidget::lengthChanged);
connect(ui->width,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&BoxWidget::widthChanged);
connect(ui->height,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&BoxWidget::heightChanged);
}
BoxWidget::~BoxWidget() = default;
void BoxWidget::applyPythonCode()
{}
void BoxWidget::setViewProvider(ViewProviderFemPostFunction* view)
{
FemGui::FunctionWidget::setViewProvider(view);
setBlockObjectUpdates(true);
Fem::FemPostBoxFunction* func = getObject<Fem::FemPostBoxFunction>();
Base::Unit unit = func->Center.getUnit();
ui->centerX->setUnit(unit);
ui->centerY->setUnit(unit);
ui->centerZ->setUnit(unit);
unit = func->Length.getUnit();
ui->length->setUnit(unit);
unit = func->Width.getUnit();
ui->width->setUnit(unit);
unit = func->Height.getUnit();
ui->height->setUnit(unit);
setBlockObjectUpdates(false);
onChange(func->Center);
onChange(func->Length);
onChange(func->Width);
onChange(func->Height);
}
void BoxWidget::onChange(const App::Property& p)
{
setBlockObjectUpdates(true);
Fem::FemPostBoxFunction* func = getObject<Fem::FemPostBoxFunction>();
if (&p == &func->Center) {
const Base::Vector3d& vec = static_cast<const App::PropertyVector*>(&p)->getValue();
ui->centerX->setValue(vec.x);
ui->centerY->setValue(vec.y);
ui->centerZ->setValue(vec.z);
}
else if (&p == &func->Length) {
double l = static_cast<const App::PropertyDistance*>(&p)->getValue();
ui->length->setValue(l);
}
else if (&p == &func->Width) {
double w = static_cast<const App::PropertyDistance*>(&p)->getValue();
ui->width->setValue(w);
}
else if (&p == &func->Height) {
double h = static_cast<const App::PropertyDistance*>(&p)->getValue();
ui->height->setValue(h);
}
setBlockObjectUpdates(false);
}
void BoxWidget::centerChanged(double)
{
if (!blockObjectUpdates()) {
Base::Vector3d vec(ui->centerX->value().getValue(),
ui->centerY->value().getValue(),
ui->centerZ->value().getValue());
getObject<Fem::FemPostBoxFunction>()->Center.setValue(vec);
}
}
void BoxWidget::lengthChanged(double)
{
if (!blockObjectUpdates()) {
double l = ui->length->value().getValue();
getObject<Fem::FemPostBoxFunction>()->Length.setValue(l);
}
}
void BoxWidget::widthChanged(double)
{
if (!blockObjectUpdates()) {
double w = ui->width->value().getValue();
getObject<Fem::FemPostBoxFunction>()->Width.setValue(w);
}
}
void BoxWidget::heightChanged(double)
{
if (!blockObjectUpdates()) {
double h = ui->height->value().getValue();
getObject<Fem::FemPostBoxFunction>()->Height.setValue(h);
}
}
// ***************************************************************************
PROPERTY_SOURCE(FemGui::ViewProviderFemPostCylinderFunction, FemGui::ViewProviderFemPostFunction)
ViewProviderFemPostCylinderFunction::ViewProviderFemPostCylinderFunction()
{
sPixmap = "fem-post-geo-cylinder";
setAutoScale(false);
// setup the visualisation geometry
getGeometryNode()->addChild(ShapeNodes::postCylinder());
}
ViewProviderFemPostCylinderFunction::~ViewProviderFemPostCylinderFunction() = default;
void ViewProviderFemPostCylinderFunction::draggerUpdate(SoDragger* m)
{
Fem::FemPostCylinderFunction* func = getObject<Fem::FemPostCylinderFunction>();
SoJackDragger* dragger = static_cast<SoJackDragger*>(m);
const SbVec3f& center = dragger->translation.getValue();
SbVec3f norm(0, 0, 1);
dragger->rotation.getValue().multVec(norm, norm);
func->Center.setValue(center[0], center[1], center[2]);
func->Radius.setValue(dragger->scaleFactor.getValue()[0]);
func->Axis.setValue(norm[0], norm[1], norm[2]);
}
void ViewProviderFemPostCylinderFunction::updateData(const App::Property* p)
{
Fem::FemPostCylinderFunction* func = getObject<Fem::FemPostCylinderFunction>();
if (!isDragging() && (p == &func->Center || p == &func->Radius || p == &func->Axis)) {
Base::Vector3d trans = func->Center.getValue();
Base::Vector3d axis = func->Axis.getValue();
double radius = func->Radius.getValue();
SbMatrix translate;
SbRotation rot(SbVec3f(0.0, 0.0, 1.0), SbVec3f(axis.x, axis.y, axis.z));
translate.setTransform(SbVec3f(trans.x, trans.y, trans.z),
rot,
SbVec3f(radius, radius, radius));
getManipulator()->setMatrix(translate);
}
Gui::ViewProviderDocumentObject::updateData(p);
}
SoTransformManip* ViewProviderFemPostCylinderFunction::setupManipulator()
{
return new SoJackManip;
}
FunctionWidget* ViewProviderFemPostCylinderFunction::createControlWidget()
{
return new CylinderWidget();
}
CylinderWidget::CylinderWidget()
: ui(new Ui_CylinderWidget)
{
ui->setupUi(this);
QSize size = ui->centerX->sizeForText(QStringLiteral("000000000000"));
ui->centerX->setMinimumWidth(size.width());
ui->centerY->setMinimumWidth(size.width());
ui->centerZ->setMinimumWidth(size.width());
ui->axisX->setMinimumWidth(size.width());
ui->axisY->setMinimumWidth(size.width());
ui->axisZ->setMinimumWidth(size.width());
ui->radius->setMinimumWidth(size.width());
int UserDecimals = Base::UnitsApi::getDecimals();
ui->centerX->setDecimals(UserDecimals);
ui->centerY->setDecimals(UserDecimals);
ui->centerZ->setDecimals(UserDecimals);
ui->axisX->setDecimals(UserDecimals);
ui->axisY->setDecimals(UserDecimals);
ui->axisZ->setDecimals(UserDecimals);
connect(ui->centerX,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&CylinderWidget::centerChanged);
connect(ui->centerY,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&CylinderWidget::centerChanged);
connect(ui->centerZ,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&CylinderWidget::centerChanged);
connect(ui->axisX,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&CylinderWidget::axisChanged);
connect(ui->axisY,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&CylinderWidget::axisChanged);
connect(ui->axisZ,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&CylinderWidget::axisChanged);
connect(ui->radius,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&CylinderWidget::radiusChanged);
}
CylinderWidget::~CylinderWidget() = default;
void CylinderWidget::applyPythonCode()
{}
void CylinderWidget::setViewProvider(ViewProviderFemPostFunction* view)
{
FemGui::FunctionWidget::setViewProvider(view);
setBlockObjectUpdates(true);
Fem::FemPostCylinderFunction* func = getObject<Fem::FemPostCylinderFunction>();
Base::Unit unit = func->Center.getUnit();
ui->centerX->setUnit(unit);
ui->centerY->setUnit(unit);
ui->centerZ->setUnit(unit);
unit = func->Radius.getUnit();
ui->radius->setUnit(unit);
setBlockObjectUpdates(false);
onChange(func->Center);
onChange(func->Radius);
onChange(func->Axis);
}
void CylinderWidget::onChange(const App::Property& p)
{
setBlockObjectUpdates(true);
Fem::FemPostCylinderFunction* func = getObject<Fem::FemPostCylinderFunction>();
if (&p == &func->Axis) {
const Base::Vector3d& vec = static_cast<const App::PropertyVector*>(&p)->getValue();
ui->axisX->setValue(vec.x);
ui->axisY->setValue(vec.y);
ui->axisZ->setValue(vec.z);
}
else if (&p == &func->Center) {
const Base::Vector3d& vec = static_cast<const App::PropertyVectorDistance*>(&p)->getValue();
ui->centerX->setValue(vec.x);
ui->centerY->setValue(vec.y);
ui->centerZ->setValue(vec.z);
}
else if (&p == &func->Radius) {
double val = static_cast<const App::PropertyDistance*>(&p)->getValue();
ui->radius->setValue(val);
}
setBlockObjectUpdates(false);
}
void CylinderWidget::centerChanged(double)
{
if (!blockObjectUpdates()) {
Base::Vector3d vec(ui->centerX->value().getValue(),
ui->centerY->value().getValue(),
ui->centerZ->value().getValue());
getObject<Fem::FemPostCylinderFunction>()->Center.setValue(vec);
}
}
void CylinderWidget::axisChanged(double)
{
if (!blockObjectUpdates()) {
Base::Vector3d vec(ui->axisX->value().getValue(),
ui->axisY->value().getValue(),
ui->axisZ->value().getValue());
getObject<Fem::FemPostCylinderFunction>()->Axis.setValue(vec);
}
}
void CylinderWidget::radiusChanged(double)
{
if (!blockObjectUpdates()) {
getObject<Fem::FemPostCylinderFunction>()->Radius.setValue(ui->radius->value().getValue());
}
}
// ***************************************************************************
PROPERTY_SOURCE(FemGui::ViewProviderFemPostPlaneFunction, FemGui::ViewProviderFemPostFunction)
// NOTE: The technical lower limit is at 1e-4 that the Coin3D manipulator can handle
static const App::PropertyFloatConstraint::Constraints scaleConstraint = {
1e-4,
std::numeric_limits<double>::max(),
1.0};
ViewProviderFemPostPlaneFunction::ViewProviderFemPostPlaneFunction()
: m_detectscale(false)
{
ADD_PROPERTY_TYPE(Scale,
(1000.0),
"Manipulator",
App::Prop_None,
"Scaling factor for the manipulator");
Scale.setConstraints(&scaleConstraint);
sPixmap = "fem-post-geo-plane";
setAutoScale(true);
// setup the visualisation geometry
getGeometryNode()->addChild(ShapeNodes::postPlane());
}
ViewProviderFemPostPlaneFunction::~ViewProviderFemPostPlaneFunction() = default;
void ViewProviderFemPostPlaneFunction::draggerUpdate(SoDragger* m)
{
Fem::FemPostPlaneFunction* func = getObject<Fem::FemPostPlaneFunction>();
SoJackDragger* dragger = static_cast<SoJackDragger*>(m);
// the new axis of the plane
const SbVec3f& base = dragger->translation.getValue();
const SbVec3f& scale = dragger->scaleFactor.getValue();
SbVec3f norm(0.0, 1.0, 0.0);
dragger->rotation.getValue().multVec(norm, norm);
func->Origin.setValue(base[0], base[1], base[2]);
func->Normal.setValue(norm[0], norm[1], norm[2]);
this->Scale.setValue(scale[0]);
}
void ViewProviderFemPostPlaneFunction::onChanged(const App::Property* prop)
{
if (prop == &Scale) {
// When loading the Scale property from a project then keep that
if (Scale.getConstraints()) {
m_detectscale = true;
}
if (!isDragging()) {
// get current matrix
SbVec3f t, s;
SbRotation r, so;
SbMatrix matrix = getManipulator()
->getDragger()
->getMotionMatrix(); // clazy:exclude=rule-of-two-soft
matrix.getTransform(t, r, s, so);
float scale = static_cast<float>(Scale.getValue());
s.setValue(scale, scale, scale);
matrix.setTransform(t, r, s, so);
getManipulator()->setMatrix(matrix);
}
}
ViewProviderFemPostFunction::onChanged(prop);
}
void ViewProviderFemPostPlaneFunction::updateData(const App::Property* p)
{
Fem::FemPostPlaneFunction* func = getObject<Fem::FemPostPlaneFunction>();
if (!isDragging() && (p == &func->Origin || p == &func->Normal)) {
if (!m_detectscale) {
double s;
if (findScaleFactor(s)) {
m_detectscale = true;
this->Scale.setValue(s);
}
}
Base::Vector3d trans = func->Origin.getValue();
Base::Vector3d norm = func->Normal.getValue();
norm.Normalize();
SbRotation rot(SbVec3f(0.0, 1.0, 0.0), SbVec3f(norm.x, norm.y, norm.z));
float scale = static_cast<float>(Scale.getValue());
SbMatrix mat;
mat.setTransform(SbVec3f(trans.x, trans.y, trans.z), rot, SbVec3f(scale, scale, scale));
getManipulator()->setMatrix(mat);
}
Gui::ViewProviderDocumentObject::updateData(p);
}
SoTransformManip* ViewProviderFemPostPlaneFunction::setupManipulator()
{
return new SoJackManip;
}
FunctionWidget* ViewProviderFemPostPlaneFunction::createControlWidget()
{
return new PlaneWidget();
}
PlaneWidget::PlaneWidget()
: ui(new Ui_PlaneWidget)
{
ui->setupUi(this);
QSize size = ui->originX->sizeForText(QStringLiteral("000000000000"));
ui->originX->setMinimumWidth(size.width());
ui->originY->setMinimumWidth(size.width());
ui->originZ->setMinimumWidth(size.width());
ui->normalX->setMinimumWidth(size.width());
ui->originY->setMinimumWidth(size.width());
ui->originZ->setMinimumWidth(size.width());
int UserDecimals = Base::UnitsApi::getDecimals();
ui->originX->setDecimals(UserDecimals);
ui->originY->setDecimals(UserDecimals);
ui->originZ->setDecimals(UserDecimals);
ui->normalX->setDecimals(UserDecimals);
ui->normalY->setDecimals(UserDecimals);
ui->normalZ->setDecimals(UserDecimals);
connect(ui->originX,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&PlaneWidget::originChanged);
connect(ui->originY,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&PlaneWidget::originChanged);
connect(ui->originZ,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&PlaneWidget::originChanged);
connect(ui->normalX,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&PlaneWidget::normalChanged);
connect(ui->normalY,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&PlaneWidget::normalChanged);
connect(ui->normalZ,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&PlaneWidget::normalChanged);
}
PlaneWidget::~PlaneWidget() = default;
void PlaneWidget::applyPythonCode()
{}
void PlaneWidget::setViewProvider(ViewProviderFemPostFunction* view)
{
FemGui::FunctionWidget::setViewProvider(view);
Fem::FemPostPlaneFunction* func = getObject<Fem::FemPostPlaneFunction>();
const Base::Unit unit = func->Origin.getUnit();
setBlockObjectUpdates(true);
ui->originX->setUnit(unit);
ui->originY->setUnit(unit);
ui->originZ->setUnit(unit);
setBlockObjectUpdates(false);
// The normal vector is unitless. It uses nevertheless Gui::PrefQuantitySpinBox to keep dialog
// uniform.
onChange(func->Normal);
onChange(func->Origin);
}
void PlaneWidget::onChange(const App::Property& p)
{
setBlockObjectUpdates(true);
Fem::FemPostPlaneFunction* func = getObject<Fem::FemPostPlaneFunction>();
if (&p == &func->Normal) {
const Base::Vector3d& vec = static_cast<const App::PropertyVector*>(&p)->getValue();
ui->normalX->setValue(vec.x);
ui->normalY->setValue(vec.y);
ui->normalZ->setValue(vec.z);
}
else if (&p == &func->Origin) {
const Base::Vector3d& vec = static_cast<const App::PropertyVectorDistance*>(&p)->getValue();
ui->originX->setValue(vec.x);
ui->originY->setValue(vec.y);
ui->originZ->setValue(vec.z);
}
setBlockObjectUpdates(false);
}
void PlaneWidget::normalChanged(double)
{
if (!blockObjectUpdates()) {
Base::Vector3d vec(ui->normalX->value().getValue(),
ui->normalY->value().getValue(),
ui->normalZ->value().getValue());
getObject<Fem::FemPostPlaneFunction>()->Normal.setValue(vec);
}
}
void PlaneWidget::originChanged(double)
{
if (!blockObjectUpdates()) {
Base::Vector3d vec(ui->originX->value().getValue(),
ui->originY->value().getValue(),
ui->originZ->value().getValue());
getObject<Fem::FemPostPlaneFunction>()->Origin.setValue(vec);
}
}
// ***************************************************************************
PROPERTY_SOURCE(FemGui::ViewProviderFemPostSphereFunction, FemGui::ViewProviderFemPostFunction)
ViewProviderFemPostSphereFunction::ViewProviderFemPostSphereFunction()
{
sPixmap = "fem-post-geo-sphere";
setAutoScale(false);
// setup the visualisation geometry
getGeometryNode()->addChild(ShapeNodes::postSphere());
}
ViewProviderFemPostSphereFunction::~ViewProviderFemPostSphereFunction() = default;
SoTransformManip* ViewProviderFemPostSphereFunction::setupManipulator()
{
SoHandleBoxManip* manip = new SoHandleBoxManip();
manip->getDragger()->setPart("extruder1", new SoSeparator);
manip->getDragger()->setPart("extruder2", new SoSeparator);
manip->getDragger()->setPart("extruder3", new SoSeparator);
manip->getDragger()->setPart("extruder4", new SoSeparator);
manip->getDragger()->setPart("extruder5", new SoSeparator);
manip->getDragger()->setPart("extruder6", new SoSeparator);
manip->getDragger()->setPart("extruder1Active", new SoSeparator);
manip->getDragger()->setPart("extruder2Active", new SoSeparator);
manip->getDragger()->setPart("extruder3Active", new SoSeparator);
manip->getDragger()->setPart("extruder4Active", new SoSeparator);
manip->getDragger()->setPart("extruder5Active", new SoSeparator);
manip->getDragger()->setPart("extruder6Active", new SoSeparator);
return manip;
}
void ViewProviderFemPostSphereFunction::draggerUpdate(SoDragger* m)
{
Fem::FemPostSphereFunction* func = getObject<Fem::FemPostSphereFunction>();
SoHandleBoxDragger* dragger = static_cast<SoHandleBoxDragger*>(m);
// the new axis of the plane
SbRotation rot, scaleDir;
const SbVec3f& center = dragger->translation.getValue();
SbVec3f norm(0, 0, 1);
func->Center.setValue(center[0], center[1], center[2]);
func->Radius.setValue(dragger->scaleFactor.getValue()[0]);
}
void ViewProviderFemPostSphereFunction::updateData(const App::Property* p)
{
Fem::FemPostSphereFunction* func = getObject<Fem::FemPostSphereFunction>();
if (!isDragging() && (p == &func->Center || p == &func->Radius)) {
Base::Vector3d trans = func->Center.getValue();
double radius = func->Radius.getValue();
SbMatrix t, translate;
t.setScale(radius);
translate.setTranslate(SbVec3f(trans.x, trans.y, trans.z));
t.multRight(translate);
getManipulator()->setMatrix(t);
}
Gui::ViewProviderDocumentObject::updateData(p);
}
FunctionWidget* ViewProviderFemPostSphereFunction::createControlWidget()
{
return new SphereWidget();
}
SphereWidget::SphereWidget()
: ui(new Ui_SphereWidget)
{
ui->setupUi(this);
QSize size = ui->centerX->sizeForText(QStringLiteral("000000000000"));
ui->centerX->setMinimumWidth(size.width());
ui->centerY->setMinimumWidth(size.width());
ui->centerZ->setMinimumWidth(size.width());
ui->radius->setMinimumWidth(size.width());
int UserDecimals = Base::UnitsApi::getDecimals();
ui->centerX->setDecimals(UserDecimals);
ui->centerY->setDecimals(UserDecimals);
ui->centerZ->setDecimals(UserDecimals);
connect(ui->centerX,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&SphereWidget::centerChanged);
connect(ui->centerY,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&SphereWidget::centerChanged);
connect(ui->centerZ,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&SphereWidget::centerChanged);
connect(ui->radius,
qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
this,
&SphereWidget::radiusChanged);
}
SphereWidget::~SphereWidget() = default;
void SphereWidget::applyPythonCode()
{}
void SphereWidget::setViewProvider(ViewProviderFemPostFunction* view)
{
FemGui::FunctionWidget::setViewProvider(view);
setBlockObjectUpdates(true);
Fem::FemPostSphereFunction* func = getObject<Fem::FemPostSphereFunction>();
Base::Unit unit = func->Center.getUnit();
ui->centerX->setUnit(unit);
ui->centerY->setUnit(unit);
ui->centerZ->setUnit(unit);
unit = func->Radius.getUnit();
ui->radius->setUnit(unit);
setBlockObjectUpdates(false);
onChange(func->Center);
onChange(func->Radius);
}
void SphereWidget::onChange(const App::Property& p)
{
setBlockObjectUpdates(true);
Fem::FemPostSphereFunction* func = getObject<Fem::FemPostSphereFunction>();
if (&p == &func->Radius) {
double val = static_cast<const App::PropertyDistance*>(&p)->getValue();
ui->radius->setValue(val);
}
else if (&p == &func->Center) {
const Base::Vector3d& vec = static_cast<const App::PropertyVectorDistance*>(&p)->getValue();
ui->centerX->setValue(vec.x);
ui->centerY->setValue(vec.y);
ui->centerZ->setValue(vec.z);
}
setBlockObjectUpdates(false);
}
void SphereWidget::centerChanged(double)
{
if (!blockObjectUpdates()) {
Base::Vector3d vec(ui->centerX->value().getValue(),
ui->centerY->value().getValue(),
ui->centerZ->value().getValue());
getObject<Fem::FemPostSphereFunction>()->Center.setValue(vec);
}
}
void SphereWidget::radiusChanged(double)
{
if (!blockObjectUpdates()) {
getObject<Fem::FemPostSphereFunction>()->Radius.setValue(ui->radius->value().getValue());
}
}
namespace FemGui
{
namespace ShapeNodes
{
SoGroup* postBox()
{
SoCoordinate3* points = new SoCoordinate3();
points->point.setNum(18);
points->point.set1Value(0, -0.5, -0.5, -0.5);
points->point.set1Value(1, 0.5, -0.5, -0.5);
points->point.set1Value(2, 0.5, 0.5, -0.5);
points->point.set1Value(3, -0.5, 0.5, -0.5);
points->point.set1Value(4, -0.5, -0.5, -0.5);
points->point.set1Value(5, -0.5, -0.5, 0.5);
points->point.set1Value(6, 0.5, -0.5, 0.5);
points->point.set1Value(7, 0.5, 0.5, 0.5);
points->point.set1Value(8, -0.5, 0.5, 0.5);
points->point.set1Value(9, -0.5, -0.5, 0.5);
points->point.set1Value(10, -0.5, -0.5, -0.5);
points->point.set1Value(11, -0.5, -0.5, 0.5);
points->point.set1Value(12, 0.5, -0.5, -0.5);
points->point.set1Value(13, 0.5, -0.5, 0.5);
points->point.set1Value(14, 0.5, 0.5, -0.5);
points->point.set1Value(15, 0.5, 0.5, 0.5);
points->point.set1Value(16, -0.5, 0.5, -0.5);
points->point.set1Value(17, -0.5, 0.5, 0.5);
int vert[6];
vert[0] = 5;
vert[1] = 5;
vert[2] = 2;
vert[3] = 2;
vert[4] = 2;
vert[5] = 2;
SoGroup* group = new SoGroup();
SoLineSet* line = new SoLineSet();
line->numVertices.setValues(0, 6, vert);
group->addChild(points);
group->addChild(line);
return group;
}
SoGroup* postCylinder()
{
using std::numbers::pi;
SoCoordinate3* points = new SoCoordinate3();
int nCirc = 20;
const int nSide = 8;
float h = 3.0;
points->point.setNum(2 * (nCirc + 1 + nSide));
int idx = 0;
// top and bottom
for (int i = 0; i < 2; ++i) {
for (int j = 0; j < nCirc + 1; ++j) {
points->point.set1Value(idx,
SbVec3f(std::cos(2 * pi / nCirc * j),
std::sin(2 * pi / nCirc * j),
-h / 2. + h * i));
++idx;
}
}
// sides
for (int i = 0; i < nSide; ++i) {
for (int j = 0; j < 2; ++j) {
points->point.set1Value(idx,
SbVec3f(std::cos(2 * pi / nSide * i),
std::sin(2 * pi / nSide * i),
-h / 2. + h * j));
++idx;
}
}
// numVertices
int vert[nSide + 2] {};
vert[0] = nCirc + 1;
vert[1] = nCirc + 1;
for (int i = 0; i < nSide; ++i) {
vert[i + 2] = 2;
}
SoLineSet* line = new SoLineSet();
SoGroup* group = new SoGroup();
line->numVertices.setValues(0, nSide + 2, vert);
group->addChild(points);
group->addChild(line);
return group;
}
SoGroup* postPlane()
{
SoCoordinate3* points = new SoCoordinate3();
points->point.setNum(4);
points->point.set1Value(0, -0.5, -0.5, 0.0);
points->point.set1Value(1, -0.5, 0.5, 0.0);
points->point.set1Value(2, 0.5, 0.5, 0.0);
points->point.set1Value(3, 0.5, -0.5, 0.0);
points->point.set1Value(4, -0.5, -0.5, 0.0);
SoGroup* group = new SoGroup();
SoLineSet* line = new SoLineSet();
group->addChild(points);
group->addChild(line);
return group;
}
SoGroup* postSphere()
{
using std::numbers::pi;
SoCoordinate3* points = new SoCoordinate3();
points->point.setNum(2 * 84);
int idx = 0;
for (int i = 0; i < 4; i++) {
for (int j = 0; j < 21; j++) {
points->point.set1Value(idx,
SbVec3f(std::sin(2 * pi / 20 * j) * std::cos(pi / 4 * i),
std::sin(2 * pi / 20 * j) * std::sin(pi / 4 * i),
std::cos(2 * pi / 20 * j)));
++idx;
}
}
for (int i = 0; i < 4; i++) {
for (int j = 0; j < 21; j++) {
points->point.set1Value(idx,
SbVec3f(std::sin(pi / 4 * i) * std::cos(2 * pi / 20 * j),
std::sin(pi / 4 * i) * std::sin(2 * pi / 20 * j),
std::cos(pi / 4 * i)));
++idx;
}
}
SoGroup* group = new SoGroup();
SoLineSet* line = new SoLineSet();
group->addChild(points);
group->addChild(line);
return group;
}
} // namespace ShapeNodes
} // namespace FemGui
#include "moc_ViewProviderFemPostFunction.cpp"