FEM: Migrate _CommandFemFromShape to FemCommands and add new is_activ…

…e type

Signed-off-by: Przemo Firszt <przemo@firszt.eu>

src/Mod/Fem/FemCommands.py

@@ -54,6 +54,8 @@ def IsActive(self):
                 active = FreeCADGui.ActiveDocument is not None and FemGui.getActiveAnalysis() is not None
             elif self.is_active == 'with_results':
                 active = FreeCADGui.ActiveDocument is not None and FemGui.getActiveAnalysis() is not None and self.results_present()
+            elif self.is_active == 'with_part_feature':
+                active = FreeCADGui.ActiveDocument is not None and FemGui.getActiveAnalysis() is not None and self.part_feature_selected()
             return active
 
         def results_present(self):
@@ -63,3 +65,10 @@ def results_present(self):
                 if o.isDerivedFrom('Fem::FemResultObject'):
                     results = True
             return results
+
+        def part_feature_selected(self):
+            sel = FreeCADGui.Selection.getSelection()
+            if len(sel) == 1 and sel[0].isDerivedFrom("Part::Feature"):
+                return True
+            else:
+                return False

src/Mod/Fem/_CommandFemFromShape.py

@@ -25,17 +25,20 @@
 __url__ = "http://www.freecadweb.org"
 
 import FreeCAD
+from FemCommands import FemCommands
 
 if FreeCAD.GuiUp:
     import FreeCADGui
     from PySide import QtCore
 
 
-class _CommandFemFromShape:
-    def GetResources(self):
-        return {'Pixmap': 'fem-fem-mesh-from-shape',
-                'MenuText': QtCore.QT_TRANSLATE_NOOP("Fem_CreateFromShape", "Create FEM mesh"),
-                'ToolTip': QtCore.QT_TRANSLATE_NOOP("Fem_CreateFromShape", "Create FEM mesh from shape")}
+class _CommandFemFromShape(FemCommands):
+    def __init__(self):
+        super(_CommandFemFromShape, self).__init__()
+        self.resources = {'Pixmap': 'fem-fem-mesh-from-shape',
+                          'MenuText': QtCore.QT_TRANSLATE_NOOP("Fem_CreateFromShape", "Create FEM mesh"),
+                          'ToolTip': QtCore.QT_TRANSLATE_NOOP("Fem_CreateFromShape", "Create FEM mesh from shape")}
+        self.is_active = 'with_part_feature'
 
     def Activated(self):
         FreeCAD.ActiveDocument.openTransaction("Create FEM mesh")
@@ -50,11 +53,6 @@ def Activated(self):
 
         FreeCADGui.Selection.clearSelection()
 
-    def IsActive(self):
-        sel = FreeCADGui.Selection.getSelection()
-        if len(sel) == 1:
-            return sel[0].isDerivedFrom("Part::Feature")
-        return False
 
 if FreeCAD.GuiUp:
     FreeCADGui.addCommand('Fem_CreateFromShape', _CommandFemFromShape())