Extensions: bring files in correct order

src/App/GeoFeatureGroupExtension.cpp

@@ -0,0 +1,152 @@
+/***************************************************************************
+ *   Copyright (c) Juergen Riegel          (juergen.riegel@web.de) 2014    *
+ *   Copyright (c) Alexander Golubev (Fat-Zer) <fatzer2@gmail.com> 2015    *
+ *                                                                         *
+ *   This file is part of the FreeCAD CAx development system.              *
+ *                                                                         *
+ *   This library is free software; you can redistribute it and/or         *
+ *   modify it under the terms of the GNU Library General Public           *
+ *   License as published by the Free Software Foundation; either          *
+ *   version 2 of the License, or (at your option) any later version.      *
+ *                                                                         *
+ *   This library  is distributed in the hope that it will be useful,      *
+ *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
+ *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
+ *   GNU Library General Public License for more details.                  *
+ *                                                                         *
+ *   You should have received a copy of the GNU Library General Public     *
+ *   License along with this library; see the file COPYING.LIB. If not,    *
+ *   write to the Free Software Foundation, Inc., 59 Temple Place,         *
+ *   Suite 330, Boston, MA  02111-1307, USA                                *
+ *                                                                         *
+ ***************************************************************************/
+
+
+#include "PreCompiled.h"
+
+#ifndef _PreComp_
+#endif
+
+#include <App/Document.h>
+
+#include "GeoFeatureGroupExtension.h"
+//#include "GeoFeatureGroupPy.h"
+//#include "FeaturePythonPyImp.h"
+
+using namespace App;
+
+
+PROPERTY_SOURCE(App::GeoFeatureGroupExtension, App::GroupExtension)
+
+
+//===========================================================================
+// Feature
+//===========================================================================
+
+GeoFeatureGroupExtension::GeoFeatureGroupExtension(void)
+{
+    initExtension(GeoFeatureGroupExtension::getClassTypeId());
+    
+    ADD_PROPERTY(Placement,(Base::Placement()));
+}
+
+GeoFeatureGroupExtension::~GeoFeatureGroupExtension(void)
+{
+}
+
+void GeoFeatureGroupExtension::transformPlacement(const Base::Placement &transform)
+{
+    // NOTE: Keep in sync with APP::GeoFeature
+    Base::Placement plm = this->Placement.getValue();
+    plm = transform * plm;
+    this->Placement.setValue(plm);
+}
+
+std::vector<App::DocumentObject*> GeoFeatureGroupExtension::getGeoSubObjects () const {
+    const auto & objs = Group.getValues();
+
+    std::set<const App::GroupExtension*> processedGroups;
+    std::set<App::DocumentObject*> rvSet;
+    std::set<App::DocumentObject*> curSearchSet (objs.begin(), objs.end());
+
+    processedGroups.insert ( this );
+
+    while ( !curSearchSet.empty() ) {
+        rvSet.insert ( curSearchSet.begin (), curSearchSet.end () );
+
+        std::set<App::DocumentObject*> nextSearchSet;
+        for ( auto obj: curSearchSet) {
+            if ( isNonGeoGroup (obj) ) {
+                const App::DocumentObjectGroup *grp = static_cast<const App::DocumentObjectGroup *> (obj);
+                // Check if we havent already processed the element may happen in case of nontree structure
+                // Note: if the condition is false this generally indicates malformed structure
+                if ( processedGroups.find (grp) == processedGroups.end() ) {
+                    processedGroups.insert ( grp );
+                    const auto & objs = grp->Group.getValues();
+                    nextSearchSet.insert (objs.begin(), objs.end());
+                }
+            }
+        }
+        nextSearchSet.swap (curSearchSet);
+    }
+
+    return std::vector<App::DocumentObject*> ( rvSet.begin(), rvSet.end() );
+}
+
+bool GeoFeatureGroupExtension::geoHasObject (const DocumentObject* obj) const {
+    const auto & objs = Group.getValues();
+
+    if (!obj) {
+        return false;
+    }
+
+    std::set<const App::GroupExtension*> processedGroups;
+    std::set<const App::DocumentObject*> curSearchSet (objs.begin(), objs.end());
+
+    processedGroups.insert ( this );
+
+    while ( !curSearchSet.empty() ) {
+        if ( curSearchSet.find (obj) != curSearchSet.end() ) {
+            return true;
+        }
+        std::set<const App::DocumentObject*> nextSearchSet;
+        for ( auto obj: curSearchSet) {
+            if ( isNonGeoGroup (obj) ) {
+                const App::DocumentObjectGroup *grp = static_cast<const App::DocumentObjectGroup *> (obj);
+                if ( processedGroups.find (grp) == processedGroups.end() ) {
+                    processedGroups.insert ( grp );
+                    const auto & objs = grp->Group.getValues();
+                    nextSearchSet.insert (objs.begin(), objs.end());
+                }
+            }
+        }
+        nextSearchSet.swap (curSearchSet);
+    }
+    return false;
+}
+
+DocumentObject* GeoFeatureGroupExtension::getGroupOfObject(const DocumentObject* obj, bool indirect)
+{
+    const Document* doc = obj->getDocument();
+    std::vector<DocumentObject*> grps = doc->getObjectsOfType(GeoFeatureGroupExtension::getClassTypeId());
+    for (std::vector<DocumentObject*>::const_iterator it = grps.begin(); it != grps.end(); ++it) {
+        GeoFeatureGroupExtension* grp = (GeoFeatureGroupExtension*)(*it);
+        if ( indirect ) {
+            if (grp->geoHasObject(obj)) {
+                return dynamic_cast<App::DocumentObject*>(grp);
+            }
+        } else {
+            if (grp->hasObject(obj)) {
+                return dynamic_cast<App::DocumentObject*>(grp);
+            }
+        }
+    }
+
+    return 0;
+}
+
+// Python feature ---------------------------------------------------------
+
+// explicit template instantiation
+//template class AppExport DocumentObjectExtensionPython<App::GeoFeatureGroupExtension>;
+