Files
create/src/Mod/Fem/femtaskpanels/task_constraint_magnetization.py
luzpaz 38a01939e0 Migrate domain name from freecadweb to freecad (#9352)
* Migrate domain name from freecadweb to freecad
* Migrate src/Mod/Material files
* Migrate Stylesheet related files
* Migrate *.svg files
* Migrate miscellaneous files
* Migrate some build files
* Migrate recently added TD AR_IRAM template files

Closes #6415
2023-04-24 15:19:20 -05:00

188 lines
8.0 KiB
Python

# ***************************************************************************
# * Copyright (c) 2023 Uwe Stöhr <uwestoehr@lyx.org> *
# * *
# * This file is part of the FreeCAD CAx development system. *
# * *
# * This program is free software; you can redistribute it and/or modify *
# * it under the terms of the GNU Lesser General Public License (LGPL) *
# * as published by the Free Software Foundation; either version 2 of *
# * the License, or (at your option) any later version. *
# * for detail see the LICENCE text file. *
# * *
# * This program 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 program; if not, write to the Free Software *
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
# * USA *
# * *
# ***************************************************************************
__title__ = "FreeCAD FEM constraint magnetization task panel for the document object"
__author__ = "Uwe Stöhr"
__url__ = "https://www.freecad.org"
## @package task_constraint_magnetization
# \ingroup FEM
# \brief task panel for constraint magnetization object
import FreeCAD
import FreeCADGui
from femguiutils import selection_widgets
from femtools import femutils
from femtools import membertools
class _TaskPanel(object):
def __init__(self, obj):
self._obj = obj
self._paramWidget = FreeCADGui.PySideUic.loadUi(
FreeCAD.getHomePath() + "Mod/Fem/Resources/ui/Magnetization.ui")
self._initParamWidget()
# geometry selection widget
# magnetization is always a body force for 3D, therefore only allow solid
self._selectionWidget = selection_widgets.GeometryElementsSelection(
obj.References,
["Solid", "Face"],
True,
False
)
# form made from param and selection widget
self.form = [self._paramWidget, self._selectionWidget]
analysis = obj.getParentGroup()
self._mesh = None
self._part = None
if analysis is not None:
self._mesh = membertools.get_single_member(analysis, "Fem::FemMeshObject")
if self._mesh is not None:
self._part = femutils.get_part_to_mesh(self._mesh)
self._partVisible = None
self._meshVisible = None
def open(self):
if self._mesh is not None and self._part is not None:
self._meshVisible = self._mesh.ViewObject.isVisible()
self._partVisible = self._part.ViewObject.isVisible()
self._mesh.ViewObject.hide()
self._part.ViewObject.show()
def reject(self):
self._restoreVisibility()
FreeCADGui.ActiveDocument.resetEdit()
return True
def accept(self):
if self._obj.References != self._selectionWidget.references:
self._obj.References = self._selectionWidget.references
self._applyWidgetChanges()
self._obj.Document.recompute()
FreeCADGui.ActiveDocument.resetEdit()
self._restoreVisibility()
return True
def _restoreVisibility(self):
if self._mesh is not None and self._part is not None:
if self._meshVisible:
self._mesh.ViewObject.show()
else:
self._mesh.ViewObject.hide()
if self._partVisible:
self._part.ViewObject.show()
else:
self._part.ViewObject.hide()
def _initParamWidget(self):
self._paramWidget.realXQSB.setProperty(
'value', self._obj.Magnetization_re_1)
FreeCADGui.ExpressionBinding(
self._paramWidget.realXQSB).bind(self._obj, "Magnetization_re_1")
self._paramWidget.realYQSB.setProperty(
'value', self._obj.Magnetization_re_2)
FreeCADGui.ExpressionBinding(
self._paramWidget.realYQSB).bind(self._obj, "Magnetization_re_2")
self._paramWidget.realZQSB.setProperty(
'value', self._obj.Magnetization_re_3)
FreeCADGui.ExpressionBinding(
self._paramWidget.realZQSB).bind(self._obj, "Magnetization_re_3")
self._paramWidget.imagXQSB.setProperty(
'value', self._obj.Magnetization_im_1)
FreeCADGui.ExpressionBinding(
self._paramWidget.imagXQSB).bind(self._obj, "Magnetization_im_1")
self._paramWidget.imagYQSB.setProperty(
'value', self._obj.Magnetization_im_2)
FreeCADGui.ExpressionBinding(
self._paramWidget.imagYQSB).bind(self._obj, "Magnetization_im_2")
self._paramWidget.imagZQSB.setProperty(
'value', self._obj.Magnetization_im_3)
FreeCADGui.ExpressionBinding(
self._paramWidget.imagZQSB).bind(self._obj, "Magnetization_im_3")
self._paramWidget.reXunspecBox.setChecked(
self._obj.Magnetization_re_1_Disabled)
self._paramWidget.reYunspecBox.setChecked(
self._obj.Magnetization_re_2_Disabled)
self._paramWidget.reZunspecBox.setChecked(
self._obj.Magnetization_re_3_Disabled)
self._paramWidget.imXunspecBox.setChecked(
self._obj.Magnetization_im_1_Disabled)
self._paramWidget.imYunspecBox.setChecked(
self._obj.Magnetization_im_2_Disabled)
self._paramWidget.imZunspecBox.setChecked(
self._obj.Magnetization_im_3_Disabled)
def _applyMagnetizationChanges(self, enabledBox, magnetizationQSB):
enabled = enabledBox.isChecked()
magnetization = None
try:
magnetization = magnetizationQSB.property('value')
except ValueError:
FreeCAD.Console.PrintMessage(
"Wrong input. Not recognised input: '{}' "
"Magnetization has not been set.\n".format(magnetizationQSB.text())
)
magnetization = '0.0 A/m'
return enabled, magnetization
def _applyWidgetChanges(self):
# apply the magnetizations and their enabled state
self._obj.Magnetization_re_1_Disabled, self._obj.Magnetization_re_1 = \
self._applyMagnetizationChanges(
self._paramWidget.reXunspecBox,
self._paramWidget.realXQSB
)
self._obj.Magnetization_re_2_Disabled, self._obj.Magnetization_re_2 = \
self._applyMagnetizationChanges(
self._paramWidget.reYunspecBox,
self._paramWidget.realYQSB
)
self._obj.Magnetization_re_3_Disabled, self._obj.Magnetization_re_3 = \
self._applyMagnetizationChanges(
self._paramWidget.reZunspecBox,
self._paramWidget.realZQSB
)
self._obj.Magnetization_im_1_Disabled, self._obj.Magnetization_im_1 = \
self._applyMagnetizationChanges(
self._paramWidget.imXunspecBox,
self._paramWidget.imagXQSB
)
self._obj.Magnetization_im_2_Disabled, self._obj.Magnetization_im_2 = \
self._applyMagnetizationChanges(
self._paramWidget.imYunspecBox,
self._paramWidget.imagYQSB
)
self._obj.Magnetization_im_3_Disabled, self._obj.Magnetization_im_3 = \
self._applyMagnetizationChanges(
self._paramWidget.imZunspecBox,
self._paramWidget.imagZQSB
)