kindred/create
1
1
Fork 0
Code Issues 37 Pull Requests Actions Packages Projects 9 Releases 2 Wiki Activity

FEM: constraint electrostatic potential, rename class and module

Browse Source
This commit is contained in:
Bernd Hahnebach
2020-05-22 01:07:29 +02:00
parent 40b2484f62
commit 7f5df5cb42
6 changed files with 32 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

149
src/Mod/Fem/femguiobjects/_ViewProviderFemConstraintElectrostaticPotential.py
View File

@@ -1,149 +0,0 @@
# ***************************************************************************
# * Copyright (c) 2017 Markus Hovorka <m.hovorka@live.de> *
# * *
# * 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 electrostatic potential ViewProvider for the document object"
__author__ = "Markus Hovorka, Bernd Hahnebach"
__url__ = "http://www.freecadweb.org"
## @package ViewProviderFemConstraintElctrostaticPotential
# \ingroup FEM
# \brief FreeCAD FEM view provider for constraint electrostatic potential object
import FreeCAD
import FreeCADGui
from FreeCAD import Units
from femguiutils import selection_widgets
from . import ViewProviderFemConstraint
from femtools import femutils
from femtools import membertools
class ViewProxy(ViewProviderFemConstraint.ViewProxy):
def setEdit(self, vobj, mode=0):
ViewProviderFemConstraint.ViewProxy.setEdit(
self,
vobj,
mode,
_TaskPanel
)
class _TaskPanel(object):
def __init__(self, obj):
self._obj = obj
self._refWidget = selection_widgets.BoundarySelector()
self._refWidget.setReferences(obj.References)
self._paramWidget = FreeCADGui.PySideUic.loadUi(
FreeCAD.getHomePath() + "Mod/Fem/Resources/ui/ElectrostaticPotential.ui")
self._initParamWidget()
self.form = [self._refWidget, self._paramWidget]
analysis = obj.getParentGroup()
self._mesh = membertools.get_single_member(analysis, "Fem::FemMeshObject")
self._part = None
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._refWidget.references():
self._obj.References = self._refWidget.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):
unit = "V"
q = Units.Quantity("{} {}".format(self._obj.Potential, unit))
self._paramWidget.potentialTxt.setText(
q.UserString)
self._paramWidget.potentialBox.setChecked(
not self._obj.PotentialEnabled)
self._paramWidget.potentialConstantBox.setChecked(
self._obj.PotentialConstant)
self._paramWidget.electricInfinityBox.setChecked(
self._obj.ElectricInfinity)
self._paramWidget.capacitanceBodyBox.setChecked(
not self._obj.CapacitanceBodyEnabled)
self._paramWidget.capacitanceBody_spinBox.setValue(
self._obj.CapacitanceBody)
def _applyWidgetChanges(self):
unit = "V"
self._obj.PotentialEnabled = \
not self._paramWidget.potentialBox.isChecked()
if self._obj.PotentialEnabled:
# if the input widget shows not a green hook, but the user presses ok
# we could run into a syntax error on getting the quantity, try mV
quantity = None
try:
quantity = Units.Quantity(self._paramWidget.potentialTxt.text())
except ValueError:
FreeCAD.Console.PrintMessage(
"Wrong input. OK has been triggered without a green hook "
"in the input field. Not recognised input: '{}' "
"Potential has not been set.\n"
.format(self._paramWidget.potentialTxt.text())
)
if quantity is not None:
self._obj.Potential = float(quantity.getValueAs(unit))
self._obj.PotentialConstant = self._paramWidget.potentialConstantBox.isChecked()
self._obj.ElectricInfinity = self._paramWidget.electricInfinityBox.isChecked()
self._obj.CapacitanceBodyEnabled = \
not self._paramWidget.capacitanceBodyBox.isChecked()
if self._obj.CapacitanceBodyEnabled:
self._paramWidget.capacitanceBody_spinBox.setEnabled(True)
self._obj.CapacitanceBody = self._paramWidget.capacitanceBody_spinBox.value()