[FEM] add new constraint for magnetization
This commit is contained in:
Uwe
187
src/Mod/Fem/femtaskpanels/task_constraint_magnetization.py
Normal file
187
src/Mod/Fem/femtaskpanels/task_constraint_magnetization.py
Normal file
@@ -0,0 +1,187 @@
|
||||
# ***************************************************************************
|
||||
# * 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.freecadweb.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"],
|
||||
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)
|
||||
self._paramWidget.reYunspecBox.setChecked(
|
||||
self._obj.Magnetization_re_2)
|
||||
self._paramWidget.reZunspecBox.setChecked(
|
||||
self._obj.Magnetization_re_3)
|
||||
self._paramWidget.imXunspecBox.setChecked(
|
||||
self._obj.Magnetization_im_1)
|
||||
self._paramWidget.imYunspecBox.setChecked(
|
||||
self._obj.Magnetization_im_2)
|
||||
self._paramWidget.imZunspecBox.setChecked(
|
||||
self._obj.Magnetization_im_3)
|
||||
|
||||
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
|
||||
)
|
||||
Reference in New Issue
Block a user