FEM: base Python VP, extend to get the icon without hard coded name

FreeCAD · Mar 6, 2020 · 5829fc4 · 5829fc4
1 parent f4f5d7b
commit 5829fc4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/Mod/Fem/femguiobjects/ViewProviderFemConstraint.py b/src/Mod/Fem/femguiobjects/ViewProviderFemConstraint.py
@@ -45,6 +45,12 @@ class ViewProxy(object):
     def __init__(self, vobj):
         vobj.Proxy = self
 
+    # needs to be overwritten, if no standard icon name is used
+    def getIcon(self):
+        """after load from FCStd file, self.icon does not exist, return constant path instead"""
+        file_name = self.Object.Proxy.Type.replace("Fem::", "FEM_")
+        return ":/icons/{}.svg".format(file_name)
+
     def attach(self, vobj):
         default = coin.SoGroup()
         vobj.addDisplayMode(default, "Default")