FEM: fluid material, use correct icon in tree view

2020-03-10 23:12:18 +01:00
parent 550b5a0965
commit 6cbb2e62f7
1 changed files with 10 additions and 1 deletions
--- a/src/Mod/Fem/femguiobjects/_ViewProviderFemMaterial.py
+++ b/src/Mod/Fem/femguiobjects/_ViewProviderFemMaterial.py
@@ -53,7 +53,16 @@ class _ViewProviderFemMaterial(ViewProviderFemConstraint.ViewProxy):
    """

    def getIcon(self):
-        return ":/icons/FEM_MaterialSolid.svg"
+        if hasattr(self.Object, "Category"):
+            if self.Object.Category == "Solid":
+                return ":/icons/FEM_MaterialSolid.svg"
+            elif self.Object.Category == "Fluid":
+                return ":/icons/FEM_MaterialFluid.svg"
+            else:
+                return ""
+        else:
+            FreeCAD.Console.Error("Document object does not have Category property")
+            return ""

    def setEdit(self, vobj, mode=0):
        ViewProviderFemConstraint.ViewProxy.setEdit(