PartDesign: add 'VisibleFeature' attribute to BodyPy

realthunder · Aug 19, 2019 · 530963d · 530963d
1 parent e2129f0
commit 530963d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/Mod/PartDesign/App/BodyPy.xml b/src/Mod/PartDesign/App/BodyPy.xml
@@ -28,5 +28,11 @@
                     </UserDocu>
             </Documentation>
     </Methode>
+    <Attribute Name="VisibleFeature" ReadOnly="true">
+        <Documentation>
+            <UserDocu>Return the the visible feature of this body</UserDocu>
+        </Documentation>
+		<Parameter Name="VisibleFeature" Type="Object" />
+    </Attribute>
   </PythonExport>
 </GenerateModel>
diff --git a/src/Mod/PartDesign/App/BodyPyImp.cpp b/src/Mod/PartDesign/App/BodyPyImp.cpp
@@ -30,6 +30,7 @@
 
 #include "Mod/Part/App/Part2DObject.h"
 #include "Mod/PartDesign/App/Body.h"
+#include "Mod/PartDesign/App/Feature.h"
 
 // inclusion of the generated files (generated out of ItemPy.xml)
 #include "BodyPy.h"
@@ -89,3 +90,12 @@ PyObject* BodyPy::insertObject(PyObject *args)
 
     Py_Return;
 }
+
+Py::Object BodyPy::getVisibleFeature() const {
+    for(auto obj : getBodyPtr()->Group.getValues()) {
+        if(obj->Visibility.getValue() && obj->isDerivedFrom(PartDesign::Feature::getClassTypeId()))
+            return Py::Object(obj->getPyObject(),true);
+    }
+    return Py::Object();
+}
+