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

FEM: Python Gui Base object without display modes

Browse Source
This commit is contained in:
Bernd Hahnebach
2020-03-22 11:25:20 +01:00
parent 3ff9d36e92
commit 01c2386a19
3 changed files with 127 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/Mod/Fem/CMakeLists.txt
View File

@@ -337,6 +337,7 @@ SET(FemGuiScripts_SRCS
femguiobjects/_ViewProviderFemResultMechanical.py
femguiobjects/_ViewProviderFemSolverCalculix.py
femguiobjects/FemSelectionWidgets.py
femguiobjects/ViewProviderBaseObject.py
femguiobjects/ViewProviderFemConstraint.py
)

122
src/Mod/Fem/femguiobjects/ViewProviderBaseObject.py Normal file
View File

@@ -0,0 +1,122 @@
# ***************************************************************************
# * Copyright (c) 2017 Markus Hovorka <m.hovorka@live.de> *
# * Copyright (c) 2018 Bernd Hahnebach <bernd@bimstatik.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 base constraint ViewProvider"
__author__ = "Markus Hovorka, Bernd Hahnebach"
__url__ = "http://www.freecadweb.org"
## @package _BaseViewProvider
# \ingroup FEM
# \brief FreeCAD _Base ViewProvider for FEM workbench
from six import string_types
import FreeCAD
import FreeCADGui
import FemGui # needed to display the icons in TreeView
False if FemGui.__name__ else True # flake8, dummy FemGui usage
class ViewProxy(object):
"""Proxy View Provider for Pythons base constraint."""
def __init__(self, vobj):
vobj.Proxy = self
# needs to be overwritten, if no standard icon name is used
# see constraint body heat source as an example
def getIcon(self):
"""after load from FCStd file, self.icon does not exist, return constant path instead"""
# https://forum.freecadweb.org/viewtopic.php?f=18&t=44009
if (
hasattr(self.Object.Proxy, "Type")
and isinstance(self.Object.Proxy.Type, string_types)
and self.Object.Proxy.Type.startswith("Fem::")
):
icon_path = "/icons/{}.svg".format(self.Object.Proxy.Type.replace("Fem::", "FEM_"))
FreeCAD.Console.PrintLog("{} --> {}\n".format(self.Object.Name, icon_path))
return ":/{}".format(icon_path)
else:
FreeCAD.Console.PrintError("No icon returned for {}\n".format(self.Object.Name))
FreeCAD.Console.PrintMessage("{}\n".format(self.Object.Proxy.Type))
return ""
def attach(self, vobj):
self.Object = vobj.Object # used on various places, claim childreens, get icon, etc.
# self.ViewObject = vobj # not used ATM
def setEdit(self, vobj, mode=0, TaskPanel=None, hide_mesh=True):
if TaskPanel is None:
# avoid edit mode by return False
# https://forum.freecadweb.org/viewtopic.php?t=12139&start=10#p161062
return False
if hide_mesh is True:
# hide all FEM meshes and VTK FemPost* objects
for o in vobj.Object.Document.Objects:
if (
o.isDerivedFrom("Fem::FemMeshObject")
or o.isDerivedFrom("Fem::FemPostPipeline")
or o.isDerivedFrom("Fem::FemPostClipFilter")
or o.isDerivedFrom("Fem::FemPostScalarClipFilter")
or o.isDerivedFrom("Fem::FemPostWarpVectorFilter")
or o.isDerivedFrom("Fem::FemPostDataAlongLineFilter")
or o.isDerivedFrom("Fem::FemPostDataAtPointFilter")
or o.isDerivedFrom("Fem::FemPostCutFilter")
or o.isDerivedFrom("Fem::FemPostDataAlongLineFilter")
or o.isDerivedFrom("Fem::FemPostPlaneFunction")
or o.isDerivedFrom("Fem::FemPostSphereFunction")
):
o.ViewObject.hide()
# show task panel
task = TaskPanel(vobj.Object)
FreeCADGui.Control.showDialog(task)
return True
def unsetEdit(self, vobj, mode=0):
FreeCADGui.Control.closeDialog()
return True
def doubleClicked(self, vobj):
guidoc = FreeCADGui.getDocument(vobj.Object.Document)
# check if another VP is in edit mode
# https://forum.freecadweb.org/viewtopic.php?t=13077#p104702
if not guidoc.getInEdit():
guidoc.setEdit(vobj.Object.Name)
else:
from PySide.QtGui import QMessageBox
message = "Active Task Dialog found! Please close this one before opening a new one!"
QMessageBox.critical(None, "Error in tree view", message)
FreeCAD.Console.PrintError(message + "\n")
return True
# they are needed, see:
# https://forum.freecadweb.org/viewtopic.php?f=18&t=44021
# https://forum.freecadweb.org/viewtopic.php?f=18&t=44009
def __getstate__(self):
return None
def __setstate__(self, state):
return None

87
src/Mod/Fem/femguiobjects/ViewProviderFemConstraint.py
View File

@@ -26,46 +26,18 @@ __title__ = "FreeCAD FEM base constraint ViewProvider"
__author__ = "Markus Hovorka, Bernd Hahnebach"
__url__ = "http://www.freecadweb.org"
## @package _BaseViewProvider
## @package _ConstraintViewProvider
# \ingroup FEM
# \brief FreeCAD _Base ViewProvider for FEM workbench
# \brief FreeCAD _Base Constraint ViewProvider for FEM workbench
from pivy import coin
from six import string_types
import FreeCAD
import FreeCADGui
import FemGui # needed to display the icons in TreeView
from . import ViewProviderBaseObject
False if FemGui.__name__ else True # flake8, dummy FemGui usage
class ViewProxy(object):
class ViewProxy(ViewProviderBaseObject.ViewProxy):
"""Proxy View Provider for Pythons base constraint."""
def __init__(self, vobj):
vobj.Proxy = self
# needs to be overwritten, if no standard icon name is used
# see constraint body heat source as an example
def getIcon(self):
"""after load from FCStd file, self.icon does not exist, return constant path instead"""
# https://forum.freecadweb.org/viewtopic.php?f=18&t=44009
if (
hasattr(self.Object.Proxy, "Type")
and isinstance(self.Object.Proxy.Type, string_types)
and self.Object.Proxy.Type.startswith("Fem::")
):
icon_path = "/icons/{}.svg".format(self.Object.Proxy.Type.replace("Fem::", "FEM_"))
FreeCAD.Console.PrintLog("{} --> {}\n".format(self.Object.Name, icon_path))
return ":/{}".format(icon_path)
else:
FreeCAD.Console.PrintError("No icon returned for {}\n".format(self.Object.Name))
FreeCAD.Console.PrintMessage("{}\n".format(self.Object.Proxy.Type))
return ""
def attach(self, vobj):
default = coin.SoGroup()
vobj.addDisplayMode(default, "Default")
@@ -82,54 +54,3 @@ class ViewProxy(object):
def setDisplayMode(self, mode):
return mode
def setEdit(self, vobj, mode=0, TaskPanel=None, hide_mesh=True):
if TaskPanel is None:
# avoid edit mode by return False
# https://forum.freecadweb.org/viewtopic.php?t=12139&start=10#p161062
return False
if hide_mesh is True:
# hide all FEM meshes and VTK FemPost* objects
for o in vobj.Object.Document.Objects:
if (
o.isDerivedFrom("Fem::FemMeshObject")
or o.isDerivedFrom("Fem::FemPostPipeline")
or o.isDerivedFrom("Fem::FemPostClipFilter")
or o.isDerivedFrom("Fem::FemPostScalarClipFilter")
or o.isDerivedFrom("Fem::FemPostWarpVectorFilter")
or o.isDerivedFrom("Fem::FemPostDataAlongLineFilter")
or o.isDerivedFrom("Fem::FemPostDataAtPointFilter")
or o.isDerivedFrom("Fem::FemPostCutFilter")
or o.isDerivedFrom("Fem::FemPostDataAlongLineFilter")
or o.isDerivedFrom("Fem::FemPostPlaneFunction")
or o.isDerivedFrom("Fem::FemPostSphereFunction")
):
o.ViewObject.hide()
# show task panel
task = TaskPanel(vobj.Object)
FreeCADGui.Control.showDialog(task)
return True
def unsetEdit(self, vobj, mode=0):
FreeCADGui.Control.closeDialog()
return True
def doubleClicked(self, vobj):
guidoc = FreeCADGui.getDocument(vobj.Object.Document)
# check if another VP is in edit mode
# https://forum.freecadweb.org/viewtopic.php?t=13077#p104702
if not guidoc.getInEdit():
guidoc.setEdit(vobj.Object.Name)
else:
from PySide.QtGui import QMessageBox
message = "Active Task Dialog found! Please close this one before opening a new one!"
QMessageBox.critical(None, "Error in tree view", message)
FreeCAD.Console.PrintError(message + "\n")
return True
# they are needed, see https://forum.freecadweb.org/viewtopic.php?f=18&t=44021
def __getstate__(self):
return None
def __setstate__(self, state):
return None