X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FFeaturesPlugin%2FFeaturesPlugin_Extrusion.cpp;h=1decb449c2ebb77ed1bdd4a67ffa1312a507d4b1;hb=48a4bd38c2b9d1b6b51ebca992a5eab30e85d99b;hp=b49f5885488d020c7618e28923def08d36433ae3;hpb=d9db5498b576854aae37ddf8c0bb6fceb264712e;p=modules%2Fshaper.git diff --git a/src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp b/src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp index b49f58854..1decb449c 100644 --- a/src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp +++ b/src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp @@ -1,153 +1,212 @@ +// 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 +#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 +#define EDGE 6 + 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()); + AttributeSelectionListPtr aSelection = + std::dynamic_pointer_cast(data()->addAttribute( + FeaturesPlugin_Extrusion::LIST_ID(), ModelAPI_AttributeSelectionList::typeId())); + // extrusion works with faces always + aSelection->setSelectionType("FACE"); + + data()->addAttribute(FeaturesPlugin_Extrusion::TO_SIZE_ID(), ModelAPI_AttributeDouble::typeId()); + data()->addAttribute(FeaturesPlugin_Extrusion::FROM_SIZE_ID(), ModelAPI_AttributeDouble::typeId()); + + //data()->addAttribute(FeaturesPlugin_Extrusion::AXIS_OBJECT_ID(), ModelAPI_AttributeReference::typeId()); + + data()->addAttribute(FeaturesPlugin_Extrusion::FROM_OBJECT_ID(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(FeaturesPlugin_Extrusion::TO_OBJECT_ID(), ModelAPI_AttributeSelection::typeId()); + + ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), FeaturesPlugin_Extrusion::FROM_OBJECT_ID()); + ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), FeaturesPlugin_Extrusion::TO_OBJECT_ID()); } void FeaturesPlugin_Extrusion::execute() { - boost::shared_ptr aFaceRef = boost::dynamic_pointer_cast< - ModelAPI_AttributeSelection>(data()->attribute(FeaturesPlugin_Extrusion::FACE_ID())); - if (!aFaceRef) - return; - - boost::shared_ptr aFace = - boost::dynamic_pointer_cast(aFaceRef->value()); - if (!aFace) - return; - - boost::shared_ptr aContext = - boost::dynamic_pointer_cast(aFaceRef->context()); - - double aSize = data()->real(FeaturesPlugin_Extrusion::SIZE_ID())->value(); - if (data()->boolean(FeaturesPlugin_Extrusion::REVERSE_ID())->value()) - aSize = -aSize; - - 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; - } + AttributeSelectionListPtr aFaceRefs = selectionList(FeaturesPlugin_Extrusion::LIST_ID()); + + std::shared_ptr aFromShape; + std::shared_ptr aToShape; - // 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; + // Getting bounding planes. + std::shared_ptr anObjRef = selection(FeaturesPlugin_Extrusion::FROM_OBJECT_ID()); + if (anObjRef) { + aFromShape = std::dynamic_pointer_cast(anObjRef->value()); + } + anObjRef = selection(FeaturesPlugin_Extrusion::TO_OBJECT_ID()); + if (anObjRef) { + aToShape = std::dynamic_pointer_cast(anObjRef->value()); } - 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 aFromSize = real(FeaturesPlugin_Extrusion::FROM_SIZE_ID())->value(); + double aToSize = real(FeaturesPlugin_Extrusion::TO_SIZE_ID())->value(); + + // for each selected face generate a result + int anIndex = 0, aResultIndex = 0; + for(; anIndex < aFaceRefs->size(); anIndex++) { + std::shared_ptr aFaceRef = aFaceRefs->value(anIndex); + ResultPtr aContextRes = aFaceRef->context(); + std::shared_ptr aContext = aContextRes->shape(); + if (!aContext.get()) { + static const std::string aContextError = "The selection context is bad"; + setError(aContextError); + break; + } + + std::shared_ptr aValueFace = aFaceRef->value(); + int aFacesNum = -1; // this mean that "aFace" is used + ResultConstructionPtr aConstruction = + std::dynamic_pointer_cast(aContextRes); + if (!aValueFace.get()) { // this may be the whole sketch result selected, check and get faces + if (aConstruction.get()) { + aFacesNum = aConstruction->facesNum(); + } else { + static const std::string aFaceError = "Can not find basis for extrusion"; + setError(aFaceError); + break; + } + } + + for(int aFaceIndex = 0; aFaceIndex < aFacesNum || aFacesNum == -1; aFaceIndex++) { + ResultBodyPtr aResultBody = document()->createBody(data(), aResultIndex); + std::shared_ptr aBaseShape; + if (aFacesNum == -1) { + aBaseShape = aValueFace; + } else { + aBaseShape = std::dynamic_pointer_cast(aConstruction->face(aFaceIndex)); } - if (!aRoot.IsSame (aNewShape)) { - boost::shared_ptr aNew(new GeomAPI_Shape()); - aNew->setImpl((void *)&aNewShape); - theResultBody->generated(aRootG, aNew,_LATERAL_TAG); - } - } - } + // If bounding faces was not set creating them. + std::shared_ptr aBaseFace(new GeomAPI_Face(aBaseShape)); + std::shared_ptr aBasePln = aBaseFace->getPlane(); + std::shared_ptr aBaseDir = aBasePln->direction(); + std::shared_ptr aBaseLoc = aBasePln->location(); + std::shared_ptr aBasePlane = GeomAlgoAPI_FaceBuilder::planarFace(aBaseLoc, aBaseDir); + + std::shared_ptr aBoundingFromShape = aFromShape ? aFromShape : aBasePlane; + std::shared_ptr aBoundingToShape = aToShape ? aToShape : aBasePlane; + + // Moving bounding faces according to "from" and "to" sizes. + std::shared_ptr aFromFace(new GeomAPI_Face(aBoundingFromShape)); + std::shared_ptr aFromPln = aFromFace->getPlane(); + std::shared_ptr aFromLoc = aFromPln->location(); + std::shared_ptr aFromDir = aFromPln->direction(); + + std::shared_ptr aToFace(new GeomAPI_Face(aBoundingToShape)); + std::shared_ptr aToPln = aToFace->getPlane(); + std::shared_ptr aToLoc = aToPln->location(); + std::shared_ptr aToDir = aToPln->direction(); + + bool aSign = aFromLoc->xyz()->dot(aBaseDir->xyz()) > aToLoc->xyz()->dot(aBaseDir->xyz()); + + std::shared_ptr aFromPnt(new GeomAPI_Pnt(aFromLoc->xyz()->added(aBaseDir->xyz()->multiplied(aSign ? aFromSize : -aFromSize)))); + aBoundingFromShape = GeomAlgoAPI_FaceBuilder::planarFace(aFromPnt, aFromDir); + + std::shared_ptr aToPnt(new GeomAPI_Pnt(aToLoc->xyz()->added(aBaseDir->xyz()->multiplied(aSign ? -aToSize : aToSize)))); + aBoundingToShape = GeomAlgoAPI_FaceBuilder::planarFace(aToPnt, aToDir); + + //GeomAlgoAPI_Extrusion aFeature(aFace, aFromSize); + GeomAlgoAPI_Prism aFeature(aBaseShape, aBoundingFromShape, aBoundingToShape); + if(!aFeature.isDone()) { + static const std::string aFeatureError = "Extrusion algorithm failed"; + setError(aFeatureError); + break; + } - //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); + // Check if shape is valid + if(aFeature.shape()->isNull()) { + static const std::string aShapeError = "Resulting shape is Null"; + setError(aShapeError); + break; + } + if(!aFeature.isValid()) { + std::string aFeatureError = "Warning: resulting shape is not valid"; + setError(aFeatureError); + break; + } + //LoadNamingDS + LoadNamingDS(aFeature, aResultBody, aBaseShape, aContext); + + setResult(aResultBody, aResultIndex); + aResultIndex++; + + if (aFacesNum == -1) + break; + } } + // remove the rest results if there were produced in the previous pass + removeResults(aResultIndex); +} + +//============================================================================ +void FeaturesPlugin_Extrusion::LoadNamingDS(GeomAlgoAPI_Prism& theFeature, + std::shared_ptr theResultBody, + std::shared_ptr theBasis, + std::shared_ptr theContext) +{ + //load result + if(theBasis->isEqual(theContext)) + theResultBody->store(theFeature.shape()); + else + theResultBody->storeGenerated(theBasis, theFeature.shape()); - + GeomAPI_DataMapOfShapeShape* aSubShapes = new GeomAPI_DataMapOfShapeShape(); + theFeature.mapOfShapes(*aSubShapes); - //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); + //Insert lateral face : Face from Edge + std::string aLatName = "LateralFace"; + theResultBody->loadAndOrientGeneratedShapes(theFeature.makeShape(), theBasis, EDGE,_LATERAL_TAG, aLatName, *aSubShapes); + + //Insert bottom face + std::string aBotName = "BottomFace"; + std::shared_ptr aBottomFace = theFeature.firstShape(); + if(!aBottomFace->isNull()) { + if(aSubShapes->isBound(aBottomFace)) { + aBottomFace = aSubShapes->find(aBottomFace); + } + theResultBody->generated(aBottomFace, aBotName, _FIRST_TAG); } + //Insert top face + std::string aTopName = "TopFace"; + std::shared_ptr aTopFace = theFeature.lastShape(); + if (!aTopFace->isNull()) { + if (aSubShapes->isBound(aTopFace)) { + aTopFace = aSubShapes->find(aTopFace); + } + theResultBody->generated(aTopFace, aTopName, _LAST_TAG); + } }