From 6cbb2e62f7a3f41ba81e3c60239d245e7575fc57 Mon Sep 17 00:00:00 2001 From: Bernd Hahnebach Date: Tue, 10 Mar 2020 23:12:18 +0100 Subject: [PATCH] FEM: fluid material, use correct icon in tree view --- src/Mod/Fem/femguiobjects/_ViewProviderFemMaterial.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/src/Mod/Fem/femguiobjects/_ViewProviderFemMaterial.py b/src/Mod/Fem/femguiobjects/_ViewProviderFemMaterial.py index 2df2d182f1..6d62ec2d40 100644 --- 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(