Extensions: GeoFeatureGroup only for GeoFeature

src/App/DocumentObjectExtension.cpp

@@ -37,7 +37,7 @@ EXTENSION_PROPERTY_SOURCE(App::DocumentObjectExtension, App::Extension)
 
 DocumentObjectExtension::DocumentObjectExtension() 
 {
-    initExtension(App::DocumentObjectExtension::getExtensionClassTypeId());
+    initExtensionType(App::DocumentObjectExtension::getExtensionClassTypeId());
 }
 
 DocumentObjectExtension::~DocumentObjectExtension()

src/App/Extension.cpp

@@ -71,7 +71,7 @@ Extension::~Extension()
     }
 }
 
-void Extension::initExtension(Base::Type type) {
+void Extension::initExtensionType(Base::Type type) {
 
     m_extensionType = type;
     if(m_extensionType.isBad())

src/App/Extension.h

@@ -215,7 +215,7 @@ public:
     Extension();
     virtual ~Extension();
 
-    void initExtension(App::ExtensionContainer* obj);
+    virtual void initExtension(App::ExtensionContainer* obj);
     
     App::ExtensionContainer*       getExtendedContainer() {return m_base;}
     const App::ExtensionContainer* getExtendedContainer() const {return m_base;}
@@ -272,7 +272,7 @@ protected:
     friend class App::ExtensionContainer;
 
 protected:     
-    void initExtension(Base::Type type);
+    void initExtensionType(Base::Type type);
     bool m_isPythonExtension = false;
     Py::Object ExtensionPythonObject;
   
@@ -309,7 +309,7 @@ public:
     
     ExtensionPythonT() {
         ExtensionT::m_isPythonExtension = true;
-        ExtensionT::initExtension(ExtensionPythonT::getExtensionClassTypeId());
+        ExtensionT::initExtensionType(ExtensionPythonT::getExtensionClassTypeId());
         
         EXTENSION_ADD_PROPERTY(ExtensionProxy,(Py::Object()));
     }

src/App/GeoFeatureGroupExtension.cpp

@@ -45,21 +45,36 @@ EXTENSION_PROPERTY_SOURCE(App::GeoFeatureGroupExtension, App::GroupExtension)
 
 GeoFeatureGroupExtension::GeoFeatureGroupExtension(void)
 {
-    initExtension(GeoFeatureGroupExtension::getExtensionClassTypeId());
+    initExtensionType(GeoFeatureGroupExtension::getExtensionClassTypeId());
-    
-    EXTENSION_ADD_PROPERTY(Placement,(Base::Placement()));
 }
 
 GeoFeatureGroupExtension::~GeoFeatureGroupExtension(void)
 {
 }
 
+void GeoFeatureGroupExtension::initExtension(ExtensionContainer* obj) {
+    
+    if(!obj->isDerivedFrom(App::GeoFeature::getClassTypeId()))
+        throw Base::Exception("GeoFeatureGroupExtension can only be applied to GeoFeatures");
+        
+    App::Extension::initExtension(obj);
+}
+
+PropertyPlacement& GeoFeatureGroupExtension::placement() {
+
+    if(!getExtendedContainer())
+        throw Base::Exception("GeoFeatureGroupExtension was not applied to GeoFeature");
+    
+    return static_cast<App::GeoFeature*>(getExtendedContainer())->Placement;
+}
+
+
 void GeoFeatureGroupExtension::transformPlacement(const Base::Placement &transform)
 {
     // NOTE: Keep in sync with APP::GeoFeature
-    Base::Placement plm = this->Placement.getValue();
+    Base::Placement plm = this->placement().getValue();
     plm = transform * plm;
-    this->Placement.setValue(plm);
+    this->placement().setValue(plm);
 }
 
 std::vector<App::DocumentObject*> GeoFeatureGroupExtension::getGeoSubObjects () const {

src/App/GeoFeatureGroupExtension.h

@@ -41,7 +41,9 @@ class AppExport GeoFeatureGroupExtension : public App::GroupExtension
     EXTENSION_PROPERTY_HEADER(App::GeoFeatureGroupExtension);
 
 public:
-    PropertyPlacement Placement;
+    PropertyPlacement& placement();
+    
+    virtual void initExtension(ExtensionContainer* obj);
 
     /**
      * @brief transformPlacement applies transform to placement of this shape.

src/App/GroupExtension.cpp

@@ -38,7 +38,7 @@ EXTENSION_PROPERTY_SOURCE(App::GroupExtension, App::DocumentObjectExtension)
 
 GroupExtension::GroupExtension()
 {
-    initExtension(GroupExtension::getExtensionClassTypeId());
+    initExtensionType(GroupExtension::getExtensionClassTypeId());
     
     EXTENSION_ADD_PROPERTY_TYPE(Group,(0),"Base",(App::PropertyType)(Prop_Output),"List of referenced objects");
 }

src/App/OriginGroupExtension.cpp

@@ -39,7 +39,7 @@ EXTENSION_PROPERTY_SOURCE(App::OriginGroupExtension, App::GeoFeatureGroupExtensi
 
 OriginGroupExtension::OriginGroupExtension () {
     
-    initExtension(OriginGroupExtension::getExtensionClassTypeId());
+    initExtensionType(OriginGroupExtension::getExtensionClassTypeId());
     
     EXTENSION_ADD_PROPERTY_TYPE ( Origin, (0), 0, App::Prop_Hidden, "Origin linked to the group" );
 }

src/App/Part.cpp

@@ -35,7 +35,7 @@
 using namespace App;
 
 
-PROPERTY_SOURCE_WITH_EXTENSIONS(App::Part, App::DocumentObject)
+PROPERTY_SOURCE_WITH_EXTENSIONS(App::Part, App::GeoFeature)
 
 
 //===========================================================================

src/App/Part.h

@@ -35,7 +35,7 @@ namespace App
 
 /** Base class of all geometric document objects.
  */
-class AppExport Part : public App::DocumentObject, public App::OriginGroupExtension
+class AppExport Part : public App::GeoFeature, public App::OriginGroupExtension
 {
     PROPERTY_HEADER_WITH_EXTENSIONS(App::Part);
 

src/Gui/ViewProviderExtension.cpp

@@ -37,7 +37,7 @@ EXTENSION_PROPERTY_SOURCE(Gui::ViewProviderExtension, App::Extension)
 
 ViewProviderExtension::ViewProviderExtension() 
 {
-    initExtension(Gui::ViewProviderExtension::getExtensionClassTypeId());
+    initExtensionType(Gui::ViewProviderExtension::getExtensionClassTypeId());
 }
 
 ViewProviderExtension::~ViewProviderExtension()

src/Gui/ViewProviderExtension.h

@@ -97,7 +97,7 @@ public:
     
     ViewProviderExtensionPythonT() {
         ExtensionT::m_isPythonExtension = true;
-        ExtensionT::initExtension(ViewProviderExtensionPythonT::getExtensionClassTypeId());
+        ExtensionT::initExtensionType(ViewProviderExtensionPythonT::getExtensionClassTypeId());
         
         EXTENSION_ADD_PROPERTY(ExtensionProxy,(Py::Object()));
     }

src/Gui/ViewProviderGeoFeatureGroupExtension.cpp

@@ -38,7 +38,7 @@ EXTENSION_PROPERTY_SOURCE(Gui::ViewProviderGeoFeatureGroupExtension, Gui::ViewPr
 
 ViewProviderGeoFeatureGroupExtension::ViewProviderGeoFeatureGroupExtension()
 {
-    initExtension(ViewProviderGeoFeatureGroupExtension::getExtensionClassTypeId());
+    initExtensionType(ViewProviderGeoFeatureGroupExtension::getExtensionClassTypeId());
     
     pcGroupChildren = new SoGroup();
     pcGroupChildren->ref();
@@ -84,8 +84,8 @@ std::vector<std::string> ViewProviderGeoFeatureGroupExtension::extensionGetDispl
 void ViewProviderGeoFeatureGroupExtension::extensionUpdateData(const App::Property* prop)
 {
     auto obj = getExtendedViewProvider()->getObject()->getExtensionByType<App::GeoFeatureGroupExtension>();
-    if (obj && prop == &obj->Placement) {
+    if (obj && prop == &obj->placement()) {
-        getExtendedViewProvider()->setTransformation ( obj->Placement.getValue().toMatrix() );
+        getExtendedViewProvider()->setTransformation ( obj->placement().getValue().toMatrix() );
     } else {
         ViewProviderGroupExtension::extensionUpdateData ( prop );
     }

src/Gui/ViewProviderGroupExtension.cpp

@@ -45,7 +45,7 @@ EXTENSION_PROPERTY_SOURCE(Gui::ViewProviderGroupExtension, Gui::ViewProviderExte
 
 ViewProviderGroupExtension::ViewProviderGroupExtension()  : visible(false)
 {
-    initExtension(ViewProviderGroupExtension::getExtensionClassTypeId());
+    initExtensionType(ViewProviderGroupExtension::getExtensionClassTypeId());
 }
 
 ViewProviderGroupExtension::~ViewProviderGroupExtension()

src/Gui/ViewProviderOriginGroupExtension.cpp

@@ -48,7 +48,7 @@ EXTENSION_PROPERTY_SOURCE(Gui::ViewProviderOriginGroupExtension, Gui::ViewProvid
 
 ViewProviderOriginGroupExtension::ViewProviderOriginGroupExtension()
 {
-    initExtension(ViewProviderOriginGroupExtension::getExtensionClassTypeId());
+    initExtensionType(ViewProviderOriginGroupExtension::getExtensionClassTypeId());
 }
 
 ViewProviderOriginGroupExtension::~ViewProviderOriginGroupExtension()

src/Mod/Part/App/AttachExtension.cpp

@@ -59,7 +59,7 @@ AttachExtension::AttachExtension()
     EXTENSION_ADD_PROPERTY_TYPE(superPlacement, (Base::Placement()), "Attachment", App::Prop_None, "Extra placement to apply in addition to attachment (in local coordinates)");
 
     setAttacher(new AttachEngine3D);//default attacher
-    initExtension(AttachExtension::getExtensionClassTypeId());
+    initExtensionType(AttachExtension::getExtensionClassTypeId());
 }
 
 AttachExtension::~AttachExtension()