/*************************************************************************** * Copyright (c) Juergen Riegel (juergen.riegel@web.de) 2014 * * Copyright (c) Alexander Golubev (Fat-Zer) 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 #include "GeoFeatureGroupExtension.h" #include "OriginFeature.h" #include "Origin.h" //#include "GeoFeatureGroupPy.h" //#include "FeaturePythonPyImp.h" using namespace App; EXTENSION_PROPERTY_SOURCE(App::GeoFeatureGroupExtension, App::GroupExtension) //=========================================================================== // Feature //=========================================================================== GeoFeatureGroupExtension::GeoFeatureGroupExtension(void) { initExtensionType(GeoFeatureGroupExtension::getExtensionClassTypeId()); } GeoFeatureGroupExtension::~GeoFeatureGroupExtension(void) { } void GeoFeatureGroupExtension::initExtension(ExtensionContainer* obj) { if(!obj->isDerivedFrom(App::GeoFeature::getClassTypeId())) throw Base::RuntimeError("GeoFeatureGroupExtension can only be applied to GeoFeatures"); App::GroupExtension::initExtension(obj); } PropertyPlacement& GeoFeatureGroupExtension::placement() { if(!getExtendedContainer()) throw Base::RuntimeError("GeoFeatureGroupExtension was not applied to GeoFeature"); return static_cast(getExtendedContainer())->Placement; } 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); } DocumentObject* GeoFeatureGroupExtension::getGroupOfObject(const DocumentObject* obj) { //compared to GroupExtension we do return here all geofeaturegroups including all extensions erived from it //like origingroup. That is needed as we use this function to get all local coordinate systems. Also there //is no reason to distuinguish between geofeatuergroups, there is only between group/geofeaturegroup auto list = obj->getInList(); for (auto obj : list) { if(obj->hasExtension(App::GeoFeatureGroupExtension::getExtensionClassTypeId())) return obj; } return nullptr; } Base::Placement GeoFeatureGroupExtension::globalGroupPlacement() { return recursiveGroupPlacement(this); } Base::Placement GeoFeatureGroupExtension::recursiveGroupPlacement(GeoFeatureGroupExtension* group) { auto inList = group->getExtendedObject()->getInList(); for(auto* link : inList) { if(link->hasExtension(App::GeoFeatureGroupExtension::getExtensionClassTypeId())) return recursiveGroupPlacement(link->getExtensionByType()) * group->placement().getValue(); } return group->placement().getValue(); } void GeoFeatureGroupExtension::addObject(App::DocumentObject* object) { if(!allowObject(object)) return; //cross CoordinateSystem links are not allowed, so we need to move the whole link group auto links = getCSRelevantLinks(object); links.push_back(object); std::vector grp = Group.getValues(); for( auto obj : links) { //only one geofeaturegroup per object. auto *group = App::GeoFeatureGroupExtension::getGroupOfObject(obj); if(group && group != getExtendedObject()) group->getExtensionByType()->removeObject(obj); if (!hasObject(obj)) grp.push_back(obj); } Group.setValues(grp); } std::vector< DocumentObject* > GeoFeatureGroupExtension::getObjectsFromLinks(DocumentObject* obj) { //we get all linked objects. We can't use outList() as this includes the links from expressions std::vector< App::DocumentObject* > result; std::vector list; obj->getPropertyList(list); for(App::Property* prop : list) { if(prop->getTypeId().isDerivedFrom(App::PropertyLink::getClassTypeId())) result.push_back(static_cast(prop)->getValue()); else if(prop->getTypeId().isDerivedFrom(App::PropertyLinkList::getClassTypeId())) { auto vec = static_cast(prop)->getValues(); result.insert(result.end(), vec.begin(), vec.end()); } else if(prop->getTypeId().isDerivedFrom(App::PropertyLinkSub::getClassTypeId())) result.push_back(static_cast(prop)->getValue()); else if(prop->getTypeId().isDerivedFrom(App::PropertyLinkSubList::getClassTypeId())) { auto vec = static_cast(prop)->getValues(); result.insert(result.end(), vec.begin(), vec.end()); } } //clear all null objects and douplicates result.erase(std::remove(result.begin(), result.end(), nullptr), result.end()); std::sort(result.begin(), result.end()); result.erase(std::unique(result.begin(), result.end()), result.end()); return result; } std::vector< DocumentObject* > GeoFeatureGroupExtension::getCSOutList(App::DocumentObject* obj) { if(!obj) return std::vector< App::DocumentObject* >(); //if the object is a geofeaturegroup than all dependencies belong to that CS, we don't want them if(obj->hasExtension(App::GeoFeatureGroupExtension::getExtensionClassTypeId())) return std::vector< App::DocumentObject* >(); //we get all linked objects. We can't use outList() as this includes the links from expressions auto result = getObjectsFromLinks(obj); //we remove all links to origin features and origins, they belong to a CS too and can't be moved result.erase(std::remove_if(result.begin(), result.end(), [](App::DocumentObject* obj)->bool { return (obj->isDerivedFrom(App::OriginFeature::getClassTypeId()) || obj->isDerivedFrom(App::Origin::getClassTypeId())); }), result.end()); //collect all dependencies of those objects std::vector< App::DocumentObject* > links; for(App::DocumentObject *obj : result) { auto vec = getCSOutList(obj); links.insert(links.end(), vec.begin(), vec.end()); } if (!links.empty()) { result.insert(result.end(), links.begin(), links.end()); std::sort(result.begin(), result.end()); result.erase(std::unique(result.begin(), result.end()), result.end()); } return result; } std::vector< DocumentObject* > GeoFeatureGroupExtension::getCSInList(DocumentObject* obj) { if(!obj) return std::vector< App::DocumentObject* >(); //we get all objects that link to it std::vector< App::DocumentObject* > result; //search the inlist for objects that have non-expression links to us for(App::DocumentObject* parent : obj->getInList()) { //not interested in other groups if(parent->hasExtension(App::GeoFeatureGroupExtension::getExtensionClassTypeId())) continue; //check if the link is real or if it is a expression one (could also be both, so it is not //enough to check the expressions) auto res = getObjectsFromLinks(parent); if(std::find(res.begin(), res.end(), obj) != res.end()) result.push_back(parent); } //clear all douplicates std::sort(result.begin(), result.end()); result.erase(std::unique(result.begin(), result.end()), result.end()); //collect all links to those objects std::vector< App::DocumentObject* > links; for(App::DocumentObject *obj : result) { auto vec = getCSInList(obj); links.insert(links.end(), vec.begin(), vec.end()); } if (!links.empty()) { result.insert(result.end(), links.begin(), links.end()); std::sort(result.begin(), result.end()); result.erase(std::unique(result.begin(), result.end()), result.end()); } return result; } std::vector< DocumentObject* > GeoFeatureGroupExtension::getCSRelevantLinks(DocumentObject* obj) { auto vec1 = getCSInList(obj); auto vec2 = getCSOutList(obj); vec1.insert(vec1.end(), vec2.begin(), vec2.end()); return vec1; } // Python feature --------------------------------------------------------- namespace App { EXTENSION_PROPERTY_SOURCE_TEMPLATE(App::GeoFeatureGroupExtensionPython, App::GeoFeatureGroupExtension) // explicit template instantiation template class AppExport ExtensionPythonT>; }