X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FFeaturesPlugin%2FFeaturesPlugin_Rotation.cpp;h=0c5b49762ae9bdbe36e8193886ea70782d303d6c;hb=06e7f5859095193fc7f498bd89a7d28009794f53;hp=d0754a9dce1e7fedb974268acbd6ad8db4aeb954;hpb=d2c5ce1931ca0695869fbf036af275fe282ab913;p=modules%2Fshaper.git diff --git a/src/FeaturesPlugin/FeaturesPlugin_Rotation.cpp b/src/FeaturesPlugin/FeaturesPlugin_Rotation.cpp old mode 100755 new mode 100644 index d0754a9dc..0c5b49762 --- a/src/FeaturesPlugin/FeaturesPlugin_Rotation.cpp +++ b/src/FeaturesPlugin/FeaturesPlugin_Rotation.cpp @@ -1,4 +1,4 @@ -// Copyright (C) 2014-2017 CEA/DEN, EDF R&D +// Copyright (C) 2014-2023 CEA, EDF // // This library is free software; you can redistribute it and/or // modify it under the terms of the GNU Lesser General Public @@ -12,10 +12,9 @@ // // You should have received a copy of the GNU Lesser General Public // License along with this library; if not, write to the Free Software -// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA // -// See http://www.salome-platform.org/ or -// email : webmaster.salome@opencascade.com +// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com // #include @@ -25,17 +24,24 @@ #include #include #include +#include +#include #include +#include #include +#include #include #include #include +#include #include #include +static const std::string ROTATION_VERSION_1("v9.5"); + //================================================================================================= FeaturesPlugin_Rotation::FeaturesPlugin_Rotation() { @@ -61,6 +67,11 @@ void FeaturesPlugin_Rotation::initAttributes() ModelAPI_AttributeSelection::typeId()); data()->addAttribute(FeaturesPlugin_Rotation::END_POINT_ID(), ModelAPI_AttributeSelection::typeId()); + + if (!aSelection->isInitialized()) { + // new feature, not read from file + data()->setVersion(ROTATION_VERSION_1); + } } //================================================================================================= @@ -69,37 +80,18 @@ void FeaturesPlugin_Rotation::execute() AttributeStringPtr aMethodTypeAttr = string(FeaturesPlugin_Rotation::CREATION_METHOD()); std::string aMethodType = aMethodTypeAttr->value(); - if (aMethodType == CREATION_METHOD_BY_ANGLE()) { - performTranslationByAxisAndAngle(); - } + GeomTrsfPtr aTrsf; + if (aMethodType == CREATION_METHOD_BY_ANGLE()) + aTrsf = rotationByAxisAndAngle(); + else if (aMethodType == CREATION_METHOD_BY_THREE_POINTS()) + aTrsf = rotationByThreePoints(); - if (aMethodType == CREATION_METHOD_BY_THREE_POINTS()) { - performTranslationByThreePoints(); - } + performRotation(aTrsf); } //================================================================================================= -void FeaturesPlugin_Rotation::performTranslationByAxisAndAngle() +GeomTrsfPtr FeaturesPlugin_Rotation::rotationByAxisAndAngle() { - // Getting objects. - ListOfShape anObjects; - std::list aContextes; - AttributeSelectionListPtr anObjectsSelList = - selectionList(FeaturesPlugin_Rotation::OBJECTS_LIST_ID()); - if (anObjectsSelList->size() == 0) { - return; - } - for(int anObjectsIndex = 0; anObjectsIndex < anObjectsSelList->size(); anObjectsIndex++) { - std::shared_ptr anObjectAttr = - anObjectsSelList->value(anObjectsIndex); - std::shared_ptr anObject = anObjectAttr->value(); - if(!anObject.get()) { - return; - } - anObjects.push_back(anObject); - aContextes.push_back(anObjectAttr->context()); - } - //Getting axis. static const std::string aSelectionError = "Error: The axis shape selection is bad."; AttributeSelectionPtr anObjRef = selection(AXIS_OBJECT_ID()); @@ -111,7 +103,7 @@ void FeaturesPlugin_Rotation::performTranslationByAxisAndAngle() } if (!aShape.get()) { setError(aSelectionError); - return; + return GeomTrsfPtr(); } GeomEdgePtr anEdge; @@ -128,91 +120,21 @@ void FeaturesPlugin_Rotation::performTranslationByAxisAndAngle() if (!anEdge.get()) { setError(aSelectionError); - return; + return GeomTrsfPtr(); } std::shared_ptr anAxis (new GeomAPI_Ax1(anEdge->line()->location(), anEdge->line()->direction())); - - // Getting angle. double anAngle = real(FeaturesPlugin_Rotation::ANGLE_ID())->value(); - // Rotating each object. - std::string anError; - int aResultIndex = 0; - std::list::iterator aContext = aContextes.begin(); - for(ListOfShape::iterator anObjectsIt = anObjects.begin(); anObjectsIt != anObjects.end(); - anObjectsIt++, aContext++) { - std::shared_ptr aBaseShape = *anObjectsIt; - bool isPart = aContext->get() && (*aContext)->groupName() == ModelAPI_ResultPart::group(); - - // Setting result. - if (isPart) { - std::shared_ptr aTrsf(new GeomAPI_Trsf()); - aTrsf->setRotation(anAxis, anAngle); - ResultPartPtr anOrigin = std::dynamic_pointer_cast(*aContext); - ResultPartPtr aResultPart = document()->copyPart(anOrigin, data(), aResultIndex); - aResultPart->setTrsf(*aContext, aTrsf); - setResult(aResultPart, aResultIndex); - } else { - std::shared_ptr aRotationAlgo(new GeomAlgoAPI_Rotation(aBaseShape, - anAxis, - anAngle)); - - if (!aRotationAlgo->check()) { - setError(aRotationAlgo->getError()); - return; - } - - aRotationAlgo->build(); - - // Checking that the algorithm worked properly. - if (GeomAlgoAPI_Tools::AlgoError::isAlgorithmFailed(aRotationAlgo, getKind(), anError)) { - setError(anError); - break; - } - - ResultBodyPtr aResultBody = document()->createBody(data(), aResultIndex); - - ListOfShape aShapes; - aShapes.push_back(aBaseShape); - FeaturesPlugin_Tools::loadModifiedShapes(aResultBody, - aShapes, - ListOfShape(), - aRotationAlgo, - aRotationAlgo->shape(), - "Rotated"); - setResult(aResultBody, aResultIndex); - } - aResultIndex++; - } - - // Remove the rest results if there were produced in the previous pass. - removeResults(aResultIndex); + GeomTrsfPtr aTrsf(new GeomAPI_Trsf()); + aTrsf->setRotation(anAxis, anAngle); + return aTrsf; } //================================================================================================= -void FeaturesPlugin_Rotation::performTranslationByThreePoints() +GeomTrsfPtr FeaturesPlugin_Rotation::rotationByThreePoints() { - // Getting objects. - ListOfShape anObjects; - std::list aContextes; - AttributeSelectionListPtr anObjectsSelList = - selectionList(FeaturesPlugin_Rotation::OBJECTS_LIST_ID()); - if (anObjectsSelList->size() == 0) { - return; - } - for(int anObjectsIndex = 0; anObjectsIndex < anObjectsSelList->size(); anObjectsIndex++) { - std::shared_ptr anObjectAttr = - anObjectsSelList->value(anObjectsIndex); - std::shared_ptr anObject = anObjectAttr->value(); - if(!anObject.get()) { - return; - } - anObjects.push_back(anObject); - aContextes.push_back(anObjectAttr->context()); - } - // Getting the center point and two points (start and end) std::shared_ptr aCenterPoint; std::shared_ptr aStartPoint; @@ -240,56 +162,76 @@ void FeaturesPlugin_Rotation::performTranslationByThreePoints() aStartPoint = GeomAlgoAPI_PointBuilder::point(aStartShape); anEndPoint = GeomAlgoAPI_PointBuilder::point(anEndShape); } + + if (!aCenterPoint || !aStartPoint || !anEndPoint) + return GeomTrsfPtr(); + } + + GeomTrsfPtr aTrsf(new GeomAPI_Trsf()); + aTrsf->setRotation(aCenterPoint, aStartPoint, anEndPoint); + return aTrsf; +} + +//================================================================================================= +void FeaturesPlugin_Rotation::performRotation(const GeomTrsfPtr& theTrsf) +{ + if (!theTrsf) { + setError("Invalid transformation."); + return; } - // Rotating each object. + bool isKeepSubShapes = data()->version() == ROTATION_VERSION_1; + + // Getting objects. + GeomAPI_ShapeHierarchy anObjects; + std::list aParts; + ResultPtr aTextureSource; + AttributeSelectionListPtr anObjSelList = selectionList(OBJECTS_LIST_ID()); + if (!FeaturesPlugin_Tools::shapesFromSelectionList( + anObjSelList, isKeepSubShapes, anObjects, aParts, aTextureSource)) + return; + std::string anError; int aResultIndex = 0; - std::list::iterator aContext = aContextes.begin(); - for(ListOfShape::iterator anObjectsIt = anObjects.begin(); anObjectsIt != anObjects.end(); - anObjectsIt++, aContext++) { + // Rotating each part. + for (std::list::iterator aPRes = aParts.begin(); aPRes != aParts.end(); ++aPRes) { + ResultPartPtr anOriginal = std::dynamic_pointer_cast(*aPRes); + ResultPartPtr aResultPart = document()->copyPart(anOriginal, data(), aResultIndex); + aResultPart->setTrsf(anOriginal, theTrsf); + setResult(aResultPart, aResultIndex++); + } + + // Collect transformations for each object in a part. + std::shared_ptr aMakeShapeList(new GeomAlgoAPI_MakeShapeList); + for(GeomAPI_ShapeHierarchy::iterator anObjectsIt = anObjects.begin(); + anObjectsIt != anObjects.end(); ++anObjectsIt) { std::shared_ptr aBaseShape = *anObjectsIt; - bool isPart = (*aContext)->groupName() == ModelAPI_ResultPart::group(); - - // Setting result. - if (isPart) { - std::shared_ptr aTrsf(new GeomAPI_Trsf()); - aTrsf->setRotation(aCenterPoint, aStartPoint, anEndPoint); - ResultPartPtr anOrigin = std::dynamic_pointer_cast(*aContext); - ResultPartPtr aResultPart = document()->copyPart(anOrigin, data(), aResultIndex); - aResultPart->setTrsf(*aContext, aTrsf); - setResult(aResultPart, aResultIndex); - } else { - std::shared_ptr aRotationAlgo(new GeomAlgoAPI_Rotation(aBaseShape, - aCenterPoint, - aStartPoint, - anEndPoint)); - - if (!aRotationAlgo->check()) { - setError(aRotationAlgo->getError()); - return; - } - - aRotationAlgo->build(); - - // Checking that the algorithm worked properly. - if (GeomAlgoAPI_Tools::AlgoError::isAlgorithmFailed(aRotationAlgo, getKind(), anError)) { - setError(anError); - break; - } - - ResultBodyPtr aResultBody = document()->createBody(data(), aResultIndex); - - ListOfShape aShapes; - aShapes.push_back(aBaseShape); - FeaturesPlugin_Tools::loadModifiedShapes(aResultBody, - aShapes, - ListOfShape(), - aRotationAlgo, - aRotationAlgo->shape(), - "Rotated"); - setResult(aResultBody, aResultIndex); + std::shared_ptr aRotationAlgo( + new GeomAlgoAPI_Transform(aBaseShape, theTrsf)); + + // Checking that the algorithm worked properly. + if (GeomAlgoAPI_Tools::AlgoError::isAlgorithmFailed(aRotationAlgo, getKind(), anError)) { + setError(anError); + break; } - aResultIndex++; + + anObjects.markModified(aBaseShape, aRotationAlgo->shape()); + aMakeShapeList->appendAlgo(aRotationAlgo); + } + + // Build results of the rotation. + const ListOfShape& anOriginalShapes = anObjects.objects(); + ListOfShape aTopLevel; + anObjects.topLevelObjects(aTopLevel); + for (ListOfShape::iterator anIt = aTopLevel.begin(); anIt != aTopLevel.end(); ++anIt) { + ResultBodyPtr aResultBody = document()->createBody(data(), aResultIndex); + ModelAPI_Tools::loadModifiedShapes(aResultBody, anOriginalShapes, ListOfShape(), + aMakeShapeList, *anIt, "Rotated"); + // Copy image data, if any + ModelAPI_Tools::copyImageAttribute(aTextureSource, aResultBody); + setResult(aResultBody, aResultIndex++); } + + // Remove the rest results if there were produced in the previous pass. + removeResults(aResultIndex); }