-// Copyright (C) 2017-2019 CEA/DEN, EDF R&D
+// Copyright (C) 2017-2020 CEA/DEN, EDF R&D
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
//
#include "FeaturesPlugin_Fillet.h"
-#include "FeaturesPlugin_Tools.h"
-#include <ModelAPI_Data.h>
#include <ModelAPI_AttributeDouble.h>
#include <ModelAPI_AttributeSelectionList.h>
#include <ModelAPI_AttributeString.h>
-#include <ModelAPI_ResultBody.h>
#include <ModelAPI_Session.h>
-#include <ModelAPI_Tools.h>
#include <ModelAPI_Validator.h>
-#include <GeomAlgoAPI_CompoundBuilder.h>
#include <GeomAlgoAPI_Fillet.h>
-#include <GeomAlgoAPI_MakeShapeList.h>
+#include <GeomAlgoAPI_MakeShape.h>
#include <GeomAlgoAPI_Tools.h>
-#include <GeomAPI_DataMapOfShapeMapOfShapes.h>
#include <GeomAPI_ShapeExplorer.h>
-#include <GeomAPI_ShapeIterator.h>
-
-
-// Obtain all sub-shapes from the shape and append them to the list
-static void collectSubs(const GeomShapePtr& theShape,
- ListOfShape& theSubs,
- const GeomAPI_Shape::ShapeType theShapeType)
-{
- GeomAPI_ShapeExplorer anExp(theShape, theShapeType);
- for (; anExp.more(); anExp.next()) {
- GeomShapePtr aShape = anExp.current();
- // Store all shapes with FORWARD orientation to avoid duplication of shared edges/vertices
- aShape->setOrientation(GeomAPI_Shape::FORWARD);
- theSubs.push_back(aShape);
- }
-}
// Extract edges from the list
-static ListOfShape selectEdges(const ListOfShape& theShapes)
+static ListOfShape extractEdges(const ListOfShape& theShapes)
{
ListOfShape anEdges;
for (ListOfShape::const_iterator anIt = theShapes.begin(); anIt != theShapes.end(); ++anIt)
- if ((*anIt)->isEdge())
- anEdges.push_back(*anIt);
+ for (GeomAPI_ShapeExplorer anExp(*anIt, GeomAPI_Shape::EDGE); anExp.more(); anExp.next())
+ anEdges.push_back(anExp.current());
return anEdges;
}
-// If theShape is a compound of a single sub-shape, return this sub-shape
-static GeomShapePtr unwrapCompound(const GeomShapePtr& theShape)
-{
- GeomShapePtr aShape = theShape;
- if(aShape->shapeType() == GeomAPI_Shape::COMPOUND) {
- int aSubResultsNb = 0;
- GeomAPI_ShapeIterator anIt(aShape);
- for(; anIt.more(); anIt.next())
- ++aSubResultsNb;
-
- if(aSubResultsNb == 1) {
- anIt.init(aShape);
- aShape = anIt.current();
- }
- }
- return aShape;
-}
-
FeaturesPlugin_Fillet::FeaturesPlugin_Fillet()
{
void FeaturesPlugin_Fillet::initAttributes()
{
data()->addAttribute(CREATION_METHOD(), ModelAPI_AttributeString::typeId());
- data()->addAttribute(OBJECT_LIST_ID(), ModelAPI_AttributeSelectionList::typeId());
+ AttributePtr aSelectionList =
+ data()->addAttribute(OBJECT_LIST_ID(), ModelAPI_AttributeSelectionList::typeId());
data()->addAttribute(START_RADIUS_ID(), ModelAPI_AttributeDouble::typeId());
data()->addAttribute(END_RADIUS_ID(), ModelAPI_AttributeDouble::typeId());
ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), END_RADIUS_ID());
+
+ initVersion(aSelectionList);
}
+AttributePtr FeaturesPlugin_Fillet::objectsAttribute()
+{
+ return attribute(OBJECT_LIST_ID());
+}
+
+const std::string& FeaturesPlugin_Fillet::modifiedShapePrefix() const
+{
+ static const std::string& THE_PREFIX("Fillet");
+ return THE_PREFIX;
+}
-void FeaturesPlugin_Fillet::execute()
+GeomMakeShapePtr FeaturesPlugin_Fillet::performOperation(const GeomShapePtr& theSolid,
+ const ListOfShape& theEdges)
{
AttributeStringPtr aCreationMethod = string(CREATION_METHOD());
if (!aCreationMethod)
- return;
-
- std::list<std::pair<GeomShapePtr, ListOfShape> > aSolidsAndSubs;
-
- // getting objects and sort them according to parent solids
- AttributeSelectionListPtr anObjectsSelList = selectionList(OBJECT_LIST_ID());
- for (int anObjectsIndex = 0; anObjectsIndex < anObjectsSelList->size(); ++anObjectsIndex) {
- AttributeSelectionPtr anObjectAttr = anObjectsSelList->value(anObjectsIndex);
- GeomShapePtr anObject = anObjectAttr->value();
- if (!anObject)
- return;
-
- ResultPtr aContext = anObjectAttr->context();
- GeomShapePtr aParent;
- if (aContext.get()) {
- ResultBodyPtr aCtxOwner = ModelAPI_Tools::bodyOwner(aContext);
- aParent = aCtxOwner ? aCtxOwner->shape() : aContext->shape();
- } else { // get it from a feature
- FeaturePtr aFeature = anObjectAttr->contextFeature();
- if (aFeature.get()) {
- aParent = aFeature->firstResult()->shape();
- }
- }
- if (!aParent)
- return;
-
- // searching this parent is already in the list aSolidsAndSubs
- std::list<std::pair<GeomShapePtr, ListOfShape> >::iterator aSearch = aSolidsAndSubs.begin();
- ListOfShape* aFound;
- for(; aSearch != aSolidsAndSubs.end(); aSearch++) {
- if (aSearch->first->isSame(aParent)) {
- aFound = &(aSearch->second);
- break;
- }
- }
- if (aSearch == aSolidsAndSubs.end()) { // not found, so, add a new one
- aSolidsAndSubs.push_back(std::pair<GeomShapePtr, ListOfShape>(aParent, ListOfShape()));
- aFound = &(aSolidsAndSubs.back().second);
- }
-
- ListOfShape anEdgesAndVertices;
- collectSubs(anObject, anEdgesAndVertices, GeomAPI_Shape::EDGE);
- collectSubs(anObject, anEdgesAndVertices, GeomAPI_Shape::VERTEX);
- for (ListOfShape::iterator aEIt = anEdgesAndVertices.begin();
- aEIt != anEdgesAndVertices.end(); ++aEIt)
- aFound->push_back(*aEIt);
- }
+ return GeomMakeShapePtr();
- bool isFixedRadius = true;
+ bool isFixedRadius = aCreationMethod->value() == CREATION_METHOD_SINGLE_RADIUS();
double aRadius1 = 0.0, aRadius2 = 0.0;
- if (aCreationMethod->value() == CREATION_METHOD_SINGLE_RADIUS())
+ if (isFixedRadius)
aRadius1 = real(RADIUS_ID())->value();
else {
aRadius1 = real(START_RADIUS_ID())->value();
aRadius2 = real(END_RADIUS_ID())->value();
- isFixedRadius = false;
}
// Perform fillet operation
- GeomAlgoAPI_MakeShapeList aMakeShapeList;
std::shared_ptr<GeomAlgoAPI_Fillet> aFilletBuilder;
- int aResultIndex = 0;
std::string anError;
- std::vector<FeaturesPlugin_Tools::ResultBaseAlgo> aResultBaseAlgoList;
- ListOfShape anOriginalShapesList, aResultShapesList;
-
- std::list<std::pair<GeomShapePtr, ListOfShape> >::iterator anIt = aSolidsAndSubs.begin();
- for (; anIt != aSolidsAndSubs.end(); ++anIt) {
- GeomShapePtr aSolid = anIt->first;
- ListOfShape aFilletEdgesAndVertices = anIt->second;
-
- ListOfShape aFilletEdges = selectEdges(aFilletEdgesAndVertices);
- if (isFixedRadius)
- aFilletBuilder.reset(new GeomAlgoAPI_Fillet(aSolid, aFilletEdges, aRadius1));
- else
- aFilletBuilder.reset(new GeomAlgoAPI_Fillet(aSolid, aFilletEdges, aRadius1, aRadius2));
-
- if (GeomAlgoAPI_Tools::AlgoError::isAlgorithmFailed(aFilletBuilder, getKind(), anError)) {
- setError(anError);
- return;
- }
-
- GeomShapePtr aResult = unwrapCompound(aFilletBuilder->shape());
- std::shared_ptr<ModelAPI_ResultBody> aResultBody =
- document()->createBody(data(), aResultIndex);
-
- ListOfShape aBaseShapes;
- aBaseShapes.push_back(aSolid);
- FeaturesPlugin_Tools::loadModifiedShapes(aResultBody, aBaseShapes, ListOfShape(),
- aFilletBuilder, aResult, "Fillet");
-
- setResult(aResultBody, aResultIndex);
- aResultIndex++;
-
- FeaturesPlugin_Tools::ResultBaseAlgo aRBA;
- aRBA.resultBody = aResultBody;
- aRBA.baseShape = aSolid;
- aRBA.makeShape = aFilletBuilder;
- aResultBaseAlgoList.push_back(aRBA);
- aResultShapesList.push_back(aResult);
- anOriginalShapesList.push_back(aSolid);
-
- const std::string aFilletFaceName = "Fillet";
- ListOfShape::iterator aSelectedBase = aFilletEdges.begin();
- for(; aSelectedBase != aFilletEdges.end(); aSelectedBase++) {
- GeomShapePtr aBase = *aSelectedBase;
- // Store new faces generated from edges and vertices
- aResultBody->loadGeneratedShapes(
- aFilletBuilder, aBase, GeomAPI_Shape::EDGE, aFilletFaceName, true);
- }
- }
-
- // Store deleted shapes after all results has been proceeded. This is to avoid issue when in one
- // result shape has been deleted, but in another it was modified or stayed.
- GeomShapePtr aResultShapesCompound = GeomAlgoAPI_CompoundBuilder::compound(aResultShapesList);
- FeaturesPlugin_Tools::loadDeletedShapes(aResultBaseAlgoList,
- anOriginalShapesList, aResultShapesCompound);
+ ListOfShape aFilletEdges = extractEdges(theEdges);
+ if (isFixedRadius)
+ aFilletBuilder.reset(new GeomAlgoAPI_Fillet(theSolid, aFilletEdges, aRadius1));
+ else
+ aFilletBuilder.reset(new GeomAlgoAPI_Fillet(theSolid, aFilletEdges, aRadius1, aRadius2));
- removeResults(aResultIndex);
+ if (GeomAlgoAPI_Tools::AlgoError::isAlgorithmFailed(aFilletBuilder, getKind(), anError)) {
+ setError(anError);
+ return GeomMakeShapePtr();
+ }
+ return aFilletBuilder;
}