Extensions: Add ViewProvider extension python types

FreeCAD · Oct 8, 2016 · ed32752 · ed32752
1 parent 5163242
commit ed32752
Show file tree

Hide file tree

Showing 9 changed files with 65 additions and 1 deletion.
diff --git a/src/Gui/Application.cpp b/src/Gui/Application.cpp
@@ -1472,9 +1472,13 @@ void Application::initTypes(void)
     // View Provider
     Gui::ViewProvider                           ::init();
     Gui::ViewProviderExtension                  ::init();
+    Gui::ViewProviderExtensionPython            ::init();
     Gui::ViewProviderGroupExtension             ::init();
+    Gui::ViewProviderGroupExtensionPython       ::init();
     Gui::ViewProviderGeoFeatureGroupExtension   ::init();
+    Gui::ViewProviderGeoFeatureGroupExtensionPython::init();    
     Gui::ViewProviderOriginGroupExtension       ::init();
+    Gui::ViewProviderOriginGroupExtensionPython ::init();
     Gui::ViewProviderExtern                     ::init();
     Gui::ViewProviderDocumentObject             ::init();
     Gui::ViewProviderFeature                    ::init();

diff --git a/src/Gui/ViewProviderExtension.cpp b/src/Gui/ViewProviderExtension.cpp
@@ -61,3 +61,9 @@ void ViewProviderExtension::extensionUpdateData(const App::Property*) {
 
 }
 
+namespace Gui {
+PROPERTY_SOURCE_TEMPLATE(Gui::ViewProviderExtensionPython, Gui::ViewProviderExtension)
+
+// explicit template instantiation
+template class GuiExport ViewProviderExtensionPythonT<ViewProviderExtension>;
+}
diff --git a/src/Gui/ViewProviderExtension.h b/src/Gui/ViewProviderExtension.h
@@ -85,6 +85,31 @@ class GuiExport ViewProviderExtension : public App::Extension
     Gui::ViewProviderDocumentObject* m_viewBase = nullptr;
 };
 
-} //App
+/**
+ * Generic Python extension class which allows to behave every extension
+ * derived class as Python extension -- simply by subclassing.
+ */
+template <class ExtensionT>
+class ViewProviderExtensionPythonT : public ExtensionT
+{
+    PROPERTY_HEADER(Gui::ViewProviderExtensionPythonT<ExtensionT>);
+
+public:
+    typedef ExtensionT Inherited;
+
+    ViewProviderExtensionPythonT() {
+        ExtensionT::m_isPythonExtension = true;
+
+        ADD_PROPERTY(Proxy,(Py::Object()));
+    }
+    virtual ~ViewProviderExtensionPythonT() {
+    }
+
+    App::PropertyPythonObject Proxy;
+};
+
+typedef ViewProviderExtensionPythonT<Gui::ViewProviderExtension> ViewProviderExtensionPython;
+
+} //Gui
 
 #endif // GUI_VIEWPROVIDEREXTENSION_H
diff --git a/src/Gui/ViewProviderGeoFeatureGroupExtension.cpp b/src/Gui/ViewProviderGeoFeatureGroupExtension.cpp
@@ -89,4 +89,12 @@ void ViewProviderGeoFeatureGroupExtension::extensionUpdateData(const App::Proper
     } else {
         ViewProviderGroupExtension::extensionUpdateData ( prop );
     }
+}
+
+
+namespace Gui {
+PROPERTY_SOURCE_TEMPLATE(Gui::ViewProviderGeoFeatureGroupExtensionPython, Gui::ViewProviderGeoFeatureGroupExtension)
+
+// explicit template instantiation
+template class GuiExport ViewProviderExtensionPythonT<ViewProviderGeoFeatureGroupExtension>;
 }
diff --git a/src/Gui/ViewProviderGeoFeatureGroupExtension.h b/src/Gui/ViewProviderGeoFeatureGroupExtension.h
@@ -62,6 +62,8 @@ class GuiExport ViewProviderGeoFeatureGroupExtension : public ViewProviderGroupE
     SoGroup *pcGroupChildren;
 };
 
+typedef ViewProviderExtensionPythonT<Gui::ViewProviderGeoFeatureGroupExtension> ViewProviderGeoFeatureGroupExtensionPython;
+
 } //namespace Gui
 
 #endif // GUI_VIEWPROVIDERGEOFEATUREGROUPEXTENSION_H
diff --git a/src/Gui/ViewProviderGroupExtension.cpp b/src/Gui/ViewProviderGroupExtension.cpp
@@ -204,3 +204,10 @@ void ViewProviderGroupExtension::extensionRestore(Base::XMLReader& reader) {
     getExtendedViewProvider()->Visibility.setStatus(App::Property::User1, false); // unset
 }
 
+
+namespace Gui {
+PROPERTY_SOURCE_TEMPLATE(Gui::ViewProviderGroupExtensionPython, Gui::ViewProviderGroupExtension)
+
+// explicit template instantiation
+template class GuiExport ViewProviderExtensionPythonT<ViewProviderGroupExtension>;
+}
diff --git a/src/Gui/ViewProviderGroupExtension.h b/src/Gui/ViewProviderGroupExtension.h
@@ -58,6 +58,8 @@ class GuiExport ViewProviderGroupExtension : public ViewProviderExtension
     std::vector<ViewProvider*> nodes;
 };
 
+typedef ViewProviderExtensionPythonT<Gui::ViewProviderGroupExtension> ViewProviderGroupExtensionPython;
+
 } //namespace Gui
 
 #endif // GUI_VIEWPROVIDERGROUPEXTENSION_H
diff --git a/src/Gui/ViewProviderOriginGroupExtension.cpp b/src/Gui/ViewProviderOriginGroupExtension.cpp
@@ -190,3 +190,11 @@ void ViewProviderOriginGroupExtension::updateOriginSize () {
 
     vpOrigin->Size.setValue ( size * 1.3 );
 }
+
+
+namespace Gui {
+PROPERTY_SOURCE_TEMPLATE(Gui::ViewProviderOriginGroupExtensionPython, Gui::ViewProviderOriginGroupExtension)
+
+// explicit template instantiation
+template class GuiExport ViewProviderExtensionPythonT<ViewProviderOriginGroupExtension>;
+}
diff --git a/src/Gui/ViewProviderOriginGroupExtension.h b/src/Gui/ViewProviderOriginGroupExtension.h
@@ -60,6 +60,8 @@ class GuiExport ViewProviderOriginGroupExtension : public ViewProviderGeoFeature
     boost::signals::connection connectChangedObjectGui;
 };
 
+typedef ViewProviderExtensionPythonT<Gui::ViewProviderOriginGroupExtension> ViewProviderOriginGroupExtensionPython;
+
 } //namespace Gui
 
 #endif // GUI_VIEWPROVIDERORIGINGROUPEXTENSION_H