make FemResultObject extensible in python as FemSolverObject

FreeCAD · Sep 16, 2016 · 7337720 · 7337720
1 parent 279418f
commit 7337720
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/Mod/Fem/App/AppFem.cpp b/src/Mod/Fem/App/AppFem.cpp
@@ -161,6 +161,7 @@ PyMODINIT_FUNC initFem()
     Fem::ConstraintFluidBoundary              ::init();
 
     Fem::FemResultObject            ::init();
+    Fem::FemResultObjectPython      ::init();
     Fem::FemSolverObject            ::init();
     Fem::FemSolverObjectPython      ::init();
 

diff --git a/src/Mod/Fem/App/FemResultObject.cpp b/src/Mod/Fem/App/FemResultObject.cpp
@@ -27,6 +27,7 @@
 #endif
 
 #include "FemResultObject.h"
+#include <App/FeaturePythonPyImp.h>
 #include <App/DocumentObjectPy.h>
 
 using namespace Fem;
@@ -96,6 +97,14 @@ template<> const char* Fem::FemResultObjectPython::getViewProviderName(void) con
 }
 /// @endcond
 
+template<> PyObject* Fem::FemResultObjectPython::getPyObject(void) {
+    if (PythonObject.is(Py::_None())) {
+        // ref counter is set to 1
+        PythonObject = Py::Object(new App::FeaturePythonPyT<App::DocumentObjectPy>(this),true);
+    }
+    return Py::new_reference_to(PythonObject);
+}
+
 // explicit template instantiation
 template class AppFemExport FeaturePythonT<Fem::FemResultObject>;