Add method to get Gui document from view provider

FreeCAD · Aug 21, 2016 · 82f7944 · 82f7944
1 parent 32aa72e
commit 82f7944
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/Gui/ViewProviderDocumentObject.cpp b/src/Gui/ViewProviderDocumentObject.cpp
@@ -181,6 +181,12 @@ void ViewProviderDocumentObject::attach(App::DocumentObject *pcObj)
         DisplayMode.setValue(defmode);
 }
 
+Gui::Document* ViewProviderDocumentObject::getDocument() const
+{
+    App::Document* pAppDoc = pcObject->getDocument();
+    return Gui::Application::Instance->getDocument(pAppDoc);
+}
+
 Gui::MDIView* ViewProviderDocumentObject::getActiveView() const
 {
     App::Document* pAppDoc = pcObject->getDocument();

diff --git a/src/Gui/ViewProviderDocumentObject.h b/src/Gui/ViewProviderDocumentObject.h
@@ -44,6 +44,7 @@ namespace App
 namespace Gui {
 
 class MDIView;
+class Document;
 
 class GuiExport ViewProviderDocumentObject : public ViewProvider
 {
@@ -81,6 +82,8 @@ class GuiExport ViewProviderDocumentObject : public ViewProvider
     virtual void updateData(const App::Property*){}
     /// Get the object of this ViewProvider object
     App::DocumentObject *getObject(void) const {return pcObject;}
+    /// Get the GUI document to this ViewProvider object
+    Gui::Document* getDocument() const;
     /// Get the python wrapper for that ViewProvider
     PyObject* getPyObject();