X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FFeaturesPlugin%2FFeaturesPlugin_Extrusion.cpp;h=225e3cf481e78ac7398aab43b1fa18bcad0c055c;hb=85b662214060cf733b44ac7822e3e4db49e2da4b;hp=b49f5885488d020c7618e28923def08d36433ae3;hpb=4fcd5da2d972334e887716499b0ea75d9d6c51c2;p=modules%2Fshaper.git diff --git a/src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp b/src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp index b49f58854..225e3cf48 100644 --- a/src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp +++ b/src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp @@ -1,153 +1,147 @@ +// Copyright (C) 2014-20xx CEA/DEN, EDF R&D --> + // File: FeaturesPlugin_Extrusion.cpp // Created: 30 May 2014 // Author: Vitaly SMETANNIKOV #include "FeaturesPlugin_Extrusion.h" -#include -#include -#include -#include -#include + #include #include -#include -#include -#include -#include -#include -#include -#include -using namespace std; -#define _LATERAL_TAG 1 -#define _FIRST_TAG 2 -#define _LAST_TAG 3 -#ifdef _DEBUG -#include -#include -#endif +#include +#include +#include + +#include + +#include +#include +#include + +//================================================================================================= FeaturesPlugin_Extrusion::FeaturesPlugin_Extrusion() { } +//================================================================================================= void FeaturesPlugin_Extrusion::initAttributes() { - data()->addAttribute(FeaturesPlugin_Extrusion::FACE_ID(), ModelAPI_AttributeSelection::type()); - data()->addAttribute(FeaturesPlugin_Extrusion::SIZE_ID(), ModelAPI_AttributeDouble::type()); - data()->addAttribute(FeaturesPlugin_Extrusion::REVERSE_ID(), ModelAPI_AttributeBoolean::type()); + initCompositeSketchAttribtues(InitBaseObjectsList); + + data()->addAttribute(CREATION_METHOD(), ModelAPI_AttributeString::typeId()); + + data()->addAttribute(TO_SIZE_ID(), ModelAPI_AttributeDouble::typeId()); + data()->addAttribute(FROM_SIZE_ID(), ModelAPI_AttributeDouble::typeId()); + + data()->addAttribute(TO_OBJECT_ID(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(TO_OFFSET_ID(), ModelAPI_AttributeDouble::typeId()); + + data()->addAttribute(FROM_OBJECT_ID(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(FROM_OFFSET_ID(), ModelAPI_AttributeDouble::typeId()); + + data()->addAttribute(DIRECTION_OBJECT_ID(), ModelAPI_AttributeSelection::typeId()); + + ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), TO_OBJECT_ID()); + ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), FROM_OBJECT_ID()); + ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), DIRECTION_OBJECT_ID()); + + initCompositeSketchAttribtues(InitSketchLauncher); } +//================================================================================================= void FeaturesPlugin_Extrusion::execute() { - boost::shared_ptr aFaceRef = boost::dynamic_pointer_cast< - ModelAPI_AttributeSelection>(data()->attribute(FeaturesPlugin_Extrusion::FACE_ID())); - if (!aFaceRef) - return; + ListOfShape aBaseShapesList; + ListOfMakeShape aMakeShapesList; - boost::shared_ptr aFace = - boost::dynamic_pointer_cast(aFaceRef->value()); - if (!aFace) + // Make extrusions. + if(!makeExtrusions(aBaseShapesList, aMakeShapesList)) { return; + } - boost::shared_ptr aContext = - boost::dynamic_pointer_cast(aFaceRef->context()); + // Store results. + int aResultIndex = 0; + ListOfShape::const_iterator aBaseIt = aBaseShapesList.cbegin(); + ListOfMakeShape::const_iterator anAlgoIt = aMakeShapesList.cbegin(); + for(; aBaseIt != aBaseShapesList.cend() && anAlgoIt != aMakeShapesList.cend(); ++aBaseIt, ++anAlgoIt) { + storeResult(*aBaseIt, *anAlgoIt, aResultIndex++); + } - double aSize = data()->real(FeaturesPlugin_Extrusion::SIZE_ID())->value(); - if (data()->boolean(FeaturesPlugin_Extrusion::REVERSE_ID())->value()) - aSize = -aSize; + removeResults(aResultIndex); +} - boost::shared_ptr aResultBody = document()->createBody(data()); - //TCollection_AsciiString anError; - GeomAlgoAPI_Extrusion aFeature(aFace, aSize); - if(!aFeature.isDone()) { - std::string aFeatureError = "Extrusion algorithm failed"; - Events_Error::send(aFeatureError, this); - return; +//================================================================================================= +bool FeaturesPlugin_Extrusion::makeExtrusions(ListOfShape& theBaseShapes, + ListOfMakeShape& theMakeShapes) +{ + theMakeShapes.clear(); + + // Getting base shapes. + getBaseShapes(theBaseShapes); + + //Getting direction. + std::shared_ptr aDir; + std::shared_ptr anEdge; + AttributeSelectionPtr aSelection = selection(DIRECTION_OBJECT_ID()); + if(aSelection.get() && aSelection->value().get() && aSelection->value()->isEdge()) { + anEdge = std::shared_ptr(new GeomAPI_Edge(aSelection->value())); + } else if(aSelection->context().get() && + aSelection->context()->shape().get() && + aSelection->context()->shape()->isEdge()) { + anEdge = std::shared_ptr(new GeomAPI_Edge(aSelection->context()->shape())); } - - // Check if shape is valid - if (!aFeature.shape()->impl().IsNull()) { - std::string aShapeError = "Resulting shape is Null"; - Events_Error::send(aShapeError, this); - #ifdef _DEBUG - std::cerr << aShapeError << std::endl; - #endif - return; + if(anEdge.get()) { + if(anEdge->isLine()) { + aDir = anEdge->line()->direction(); + } } - if(!aFeature.isValid()) { - std::string aFeatureError = "Warning: resulting shape is not valid"; - Events_Error::send(aFeatureError, this); - return; - } - //LoadNamingDS - LoadNamingDS(aFeature, aResultBody, aFace, aContext); - - setResult(aResultBody); -} - //============================================================================ -void FeaturesPlugin_Extrusion::LoadNamingDS(GeomAlgoAPI_Extrusion& theFeature, - boost::shared_ptr theResultBody, - boost::shared_ptr theBasis, - boost::shared_ptr theContext) -{ - - - //load result - if(theBasis->impl().IsEqual(theContext->impl())) - theResultBody->store(theFeature.shape()); - else - theResultBody->storeGenerated(theContext, theFeature.shape()); - - TopTools_DataMapOfShapeShape aSubShapes; - for (TopExp_Explorer Exp(theFeature.shape()->impl(),TopAbs_FACE); Exp.More(); Exp.Next()) { - aSubShapes.Bind(Exp.Current(),Exp.Current()); - } - - //Insert lateral face : Face from Edge - //GeomAlgoAPI_DFLoader::loadAndOrientGeneratedShapes(*myBuilder, myBasis, TopAbs_EDGE, aLateralFaceBuilder, aSubShapes); - - - TopTools_MapOfShape aView; - TopExp_Explorer aShapeExplorer (theFeature.shape()->impl(), TopAbs_EDGE); - for (; aShapeExplorer.More(); aShapeExplorer.Next ()) { - const TopoDS_Shape& aRoot = aShapeExplorer.Current (); - if (!aView.Add(aRoot)) continue; - boost::shared_ptr aRootG(new GeomAPI_Shape()); - aRootG->setImpl((void *)&aRoot); - const ListOfShape& aShapes = theFeature.generated(aRootG); - std::list >::const_iterator anIt = aShapes.begin(), aLast = aShapes.end(); - for (; anIt != aLast; anIt++) { - TopoDS_Shape aNewShape = (*anIt)->impl(); - if (aSubShapes.IsBound(aNewShape)) { - aNewShape.Orientation((aSubShapes(aNewShape)).Orientation()); - } + // Getting sizes. + double aToSize = 0.0; + double aFromSize = 0.0; - if (!aRoot.IsSame (aNewShape)) { - boost::shared_ptr aNew(new GeomAPI_Shape()); - aNew->setImpl((void *)&aNewShape); - theResultBody->generated(aRootG, aNew,_LATERAL_TAG); - } - } + if(string(CREATION_METHOD())->value() == "BySizes") { + aToSize = real(TO_SIZE_ID())->value(); + aFromSize = real(FROM_SIZE_ID())->value(); + } else { + aToSize = real(TO_OFFSET_ID())->value(); + aFromSize = real(FROM_OFFSET_ID())->value(); } - //Insert bottom face - const boost::shared_ptr& aBottomFace = theFeature.firstShape(); - if (!aBottomFace->impl().IsNull()) { - if (aSubShapes.IsBound(aBottomFace->impl())) { - aBottomFace->setImpl((void *)&aSubShapes(aBottomFace->impl())); - } - theResultBody->generated(aBottomFace, _FIRST_TAG); + // Getting bounding planes. + GeomShapePtr aToShape; + GeomShapePtr aFromShape; + + if(string(CREATION_METHOD())->value() == "ByPlanesAndOffsets") { + aSelection = selection(TO_OBJECT_ID()); + if(aSelection.get()) { + aToShape = std::dynamic_pointer_cast(aSelection->value()); + if(!aToShape.get() && aSelection->context().get()) { + aToShape = aSelection->context()->shape(); + } + } + aSelection = selection(FROM_OBJECT_ID()); + if(aSelection.get()) { + aFromShape = std::dynamic_pointer_cast(aSelection->value()); + if(!aFromShape.get() && aSelection->context().get()) { + aFromShape = aSelection->context()->shape(); + } + } } - + // Generating result for each base shape. + for(ListOfShape::const_iterator anIter = theBaseShapes.cbegin(); anIter != theBaseShapes.cend(); anIter++) { + std::shared_ptr aBaseShape = *anIter; + + std::shared_ptr aPrismAlgo(new GeomAlgoAPI_Prism(aBaseShape, aDir, + aToShape, aToSize, + aFromShape, aFromSize)); + if(!isMakeShapeValid(aPrismAlgo)) { + return false; + } - //Insert top face - boost::shared_ptr aTopFace = theFeature.lastShape(); - if (!aTopFace->impl().IsNull()) { - if (aSubShapes.IsBound(aTopFace->impl())) { - aTopFace->setImpl((void *)&aSubShapes(aTopFace->impl())); - } - theResultBody->generated(aTopFace, _FIRST_TAG); + theMakeShapes.push_back(aPrismAlgo); } + return true; }