Skip to content

Commit

Permalink
FEM: Python Gui Base object without display modes
Browse files Browse the repository at this point in the history
  • Loading branch information
@berndhahnebach
berndhahnebach committed Mar 22, 2020
1 parent 3ff9d36 commit 01c2386
Show file tree
Hide file tree
Showing 3 changed files with 127 additions and 83 deletions.
1 change: 1 addition & 0 deletions src/Mod/Fem/CMakeLists.txt
View file
Expand Up @@ -337,6 +337,7 @@ SET(FemGuiScripts_SRCS
femguiobjects/_ViewProviderFemResultMechanical.py
femguiobjects/_ViewProviderFemSolverCalculix.py
femguiobjects/FemSelectionWidgets.py
femguiobjects/ViewProviderBaseObject.py
femguiobjects/ViewProviderFemConstraint.py
)

Expand Down
122 changes: 122 additions & 0 deletions src/Mod/Fem/femguiobjects/ViewProviderBaseObject.py
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 changes: 4 additions & 83 deletions src/Mod/Fem/femguiobjects/ViewProviderFemConstraint.py
View file
Expand Up @@ -26,46 +26,18 @@
__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
from . import ViewProviderBaseObject

import FemGui # needed to display the icons in TreeView


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")
Expand All @@ -82,54 +54,3 @@ def getDefaultDisplayMode(self):

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

0 comments on commit 01c2386

Please sign in to comment.