X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FConstructionPlugin%2FConstructionPlugin_Plane.cpp;h=dc305d29ee3a8c3889ce4c45e3004800d845fdfe;hb=aef9c1a87bc7ca357d8dfe47c56ac5edc9935d35;hp=37faed97091646b4f309eda8efca90f12f5ee3e2;hpb=3d7368386762b057fff327d24d0079b4a73c3004;p=modules%2Fshaper.git diff --git a/src/ConstructionPlugin/ConstructionPlugin_Plane.cpp b/src/ConstructionPlugin/ConstructionPlugin_Plane.cpp index 37faed970..dc305d29e 100644 --- a/src/ConstructionPlugin/ConstructionPlugin_Plane.cpp +++ b/src/ConstructionPlugin/ConstructionPlugin_Plane.cpp @@ -6,50 +6,440 @@ #include "ConstructionPlugin_Plane.h" -#include -#include -#include +#include + #include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include -#define PLANE_SIZE 300 +#include +#include +#include +#include +#include +#include +#include +#include +static GeomShapePtr faceByThreeVertices(const std::shared_ptr theV1, + const std::shared_ptr theV2, + const std::shared_ptr theV3); +static std::shared_ptr makeRectangularFace( + const std::shared_ptr theFace, + const std::shared_ptr thePln); + +//================================================================================================== ConstructionPlugin_Plane::ConstructionPlugin_Plane() { } +//================================================================================================== void ConstructionPlugin_Plane::initAttributes() { - data()->addAttribute(FACE_ATTR, ModelAPI_AttributeSelection::type()); - data()->addAttribute(DISTANCE_ATTR, ModelAPI_AttributeDouble::type()); + data()->addAttribute(ConstructionPlugin_Plane::CREATION_METHOD(), + ModelAPI_AttributeString::typeId()); + + data()->addAttribute(PLANE(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(DISTANCE(), ModelAPI_AttributeDouble::typeId()); + // By general equation. + data()->addAttribute(A(), ModelAPI_AttributeDouble::typeId()); + data()->addAttribute(B(), ModelAPI_AttributeDouble::typeId()); + data()->addAttribute(C(), ModelAPI_AttributeDouble::typeId()); + data()->addAttribute(D(), ModelAPI_AttributeDouble::typeId()); + ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), A()); + ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), B()); + ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), C()); + ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), D()); + + // By three points. + data()->addAttribute(POINT1(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(POINT2(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(POINT3(), ModelAPI_AttributeSelection::typeId()); + + // By line and point. + data()->addAttribute(LINE(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(POINT(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(PERPENDICULAR(), ModelAPI_AttributeBoolean::typeId()); + + // By other plane. + data()->addAttribute(CREATION_METHOD_BY_OTHER_PLANE_OPTION(), ModelAPI_AttributeString::typeId()); + data()->addAttribute(REVERSE(), ModelAPI_AttributeBoolean::typeId()); + data()->addAttribute(COINCIDENT_POINT(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(AXIS(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(ANGLE(), ModelAPI_AttributeDouble::typeId()); + + // By two parallel planes. + data()->addAttribute(PLANE1(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(PLANE2(), ModelAPI_AttributeSelection::typeId()); } +//================================================================================================== void ConstructionPlugin_Plane::execute() { - AttributeSelectionPtr aFaceAttr = data()->selection(FACE_ATTR); - AttributeDoublePtr aDistAttr = data()->real(DISTANCE_ATTR); - if ((aFaceAttr.get() != NULL) && (aDistAttr.get() != NULL) && - aFaceAttr->isInitialized() && aDistAttr->isInitialized()) { + GeomShapePtr aShape; + + std::string aCreationMethod = string(CREATION_METHOD())->value(); + if(aCreationMethod == CREATION_METHOD_BY_GENERAL_EQUATION() || + aCreationMethod == "PlaneByGeneralEquation") { + aShape = createByGeneralEquation(); + } else if(aCreationMethod == CREATION_METHOD_BY_THREE_POINTS()) { + aShape = createByThreePoints(); + } else if(aCreationMethod == CREATION_METHOD_BY_LINE_AND_POINT()) { + aShape = createByLineAndPoint(); + } else if(aCreationMethod == CREATION_METHOD_BY_OTHER_PLANE()) { + std::string aCreationMethodOption = string(CREATION_METHOD_BY_OTHER_PLANE_OPTION())->value(); + if(aCreationMethodOption == CREATION_METHOD_BY_DISTANCE_FROM_OTHER()) { + aShape = createByDistanceFromOther(); + } else if(aCreationMethodOption == CREATION_METHOD_BY_COINCIDENT_TO_POINT()) { + aShape = createByCoincidentPoint(); + } else if(aCreationMethodOption == CREATION_METHOD_BY_ROTATION()) { + aShape = createByRotation(); + } + } else if(aCreationMethod == CREATION_METHOD_BY_TWO_PARALLEL_PLANES()) { + aShape = createByTwoParallelPlanes(); + } else { + setError("Error: Plane creation method \"" + aCreationMethod + "\" not supported."); + return; + } + + if(!aShape.get()) { + setError("Error: Could not create a plane."); + return; + } + + ResultConstructionPtr aConstr = document()->createConstruction(data()); + aConstr->setInfinite(true); + aConstr->setShape(aShape); + setResult(aConstr); +} + +//================================================================================================== +bool ConstructionPlugin_Plane::customisePresentation(ResultPtr theResult, AISObjectPtr thePrs, + std::shared_ptr theDefaultPrs) +{ + std::vector aColor; + // get color from the attribute of the result + if (theResult.get() != NULL && + theResult->data()->attribute(ModelAPI_Result::COLOR_ID()).get() != NULL) { + AttributeIntArrayPtr aColorAttr = theResult->data()->intArray(ModelAPI_Result::COLOR_ID()); + if (aColorAttr.get() && aColorAttr->size()) { + aColor.push_back(aColorAttr->value(0)); + aColor.push_back(aColorAttr->value(1)); + aColor.push_back(aColorAttr->value(2)); + } + } + if (aColor.empty()) + aColor = Config_PropManager::color("Visualization", "construction_plane_color"); + + bool isCustomized = false; + if (aColor.size() == 3) + isCustomized = thePrs->setColor(aColor[0], aColor[1], aColor[2]); + + isCustomized = thePrs->setTransparensy(0.6) || isCustomized; + + return isCustomized; +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Plane::createByGeneralEquation() +{ + AttributeDoublePtr anAttrA = real(ConstructionPlugin_Plane::A()); + AttributeDoublePtr anAttrB = real(ConstructionPlugin_Plane::B()); + AttributeDoublePtr anAttrC = real(ConstructionPlugin_Plane::C()); + AttributeDoublePtr anAttrD = real(ConstructionPlugin_Plane::D()); + std::shared_ptr aPlaneFace; + if ((anAttrA.get() != NULL) && (anAttrB.get() != NULL) && + (anAttrC.get() != NULL) && (anAttrD.get() != NULL) && + anAttrA->isInitialized() && anAttrB->isInitialized() && + anAttrC->isInitialized() && anAttrD->isInitialized() ) { + double aA = anAttrA->value(), aB = anAttrB->value(), + aC = anAttrC->value(), aD = anAttrD->value(); + std::shared_ptr aPlane = + std::shared_ptr(new GeomAPI_Pln(aA, aB, aC, aD)); + double aSize = Config_PropManager::integer(SKETCH_TAB_NAME, "planes_size"); + aSize *= 4.; + aPlaneFace = GeomAlgoAPI_FaceBuilder::squareFace(aPlane, aSize); + } + return aPlaneFace; +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Plane::createByThreePoints() +{ + // Get first point. + AttributeSelectionPtr aPointSelection1 = selection(POINT1()); + GeomShapePtr aPointShape1 = aPointSelection1->value(); + if(!aPointShape1.get()) { + aPointShape1 = aPointSelection1->context()->shape(); + } + std::shared_ptr aVertex1(new GeomAPI_Vertex(aPointShape1)); + + // Get second point. + AttributeSelectionPtr aPointSelection2 = selection(POINT2()); + GeomShapePtr aPointShape2 = aPointSelection2->value(); + if(!aPointShape2.get()) { + aPointShape2 = aPointSelection2->context()->shape(); + } + std::shared_ptr aVertex2(new GeomAPI_Vertex(aPointShape2)); + + // Get third point. + AttributeSelectionPtr aPointSelection3 = selection(POINT3()); + GeomShapePtr aPointShape3 = aPointSelection3->value(); + if(!aPointShape3.get()) { + aPointShape3 = aPointSelection3->context()->shape(); + } + std::shared_ptr aVertex3(new GeomAPI_Vertex(aPointShape3)); + + GeomShapePtr aRes = faceByThreeVertices(aVertex1, aVertex2, aVertex3); + + return aRes; +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Plane::createByLineAndPoint() +{ + // Get edge. + AttributeSelectionPtr anEdgeSelection = selection(LINE()); + GeomShapePtr aLineShape = anEdgeSelection->value(); + if(!aLineShape.get()) { + aLineShape = anEdgeSelection->context()->shape(); + } + std::shared_ptr anEdge(new GeomAPI_Edge(aLineShape)); + + // Get point. + AttributeSelectionPtr aPointSelection = selection(POINT()); + GeomShapePtr aPointShape = aPointSelection->value(); + if(!aPointShape.get()) { + aPointShape = aPointSelection->context()->shape(); + } + std::shared_ptr aVertex(new GeomAPI_Vertex(aPointShape)); + + // Get perpendicular flag. + bool anIsPerpendicular= boolean(PERPENDICULAR())->value(); + + GeomShapePtr aRes; + if(anIsPerpendicular) { + std::shared_ptr aLin = anEdge->line(); + std::shared_ptr aPnt = aVertex->point(); + std::shared_ptr aNewPln(new GeomAPI_Pln(aPnt, aLin->direction())); + double aSize = aLin->distance(aPnt) * 2; + // point may belong to line, so for the face size use maximum distance between point and line + // and the line size (distance between the start and end point) + double aDistance = anEdge->firstPoint()->distance(anEdge->lastPoint()); + aRes = GeomAlgoAPI_FaceBuilder::squareFace(aNewPln, aSize > aDistance ? aSize : aDistance); + } else { + std::shared_ptr aV1, aV2; + GeomAlgoAPI_ShapeTools::findBounds(anEdge, aV1, aV2); + aRes = faceByThreeVertices(aV1, aV2, aVertex); + } + + return aRes; +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Plane::createByDistanceFromOther() +{ + AttributeSelectionPtr aFaceAttr = data()->selection(ConstructionPlugin_Plane::PLANE()); + AttributeDoublePtr aDistAttr = data()->real(ConstructionPlugin_Plane::DISTANCE()); + std::shared_ptr aPlane; + if ((aFaceAttr.get() != NULL) && + (aDistAttr.get() != NULL) && + aFaceAttr->isInitialized() && aDistAttr->isInitialized()) { double aDist = aDistAttr->value(); + bool anIsReverse = boolean(REVERSE())->value(); + if(anIsReverse) aDist = -aDist; GeomShapePtr aShape = aFaceAttr->value(); - if (aShape.get() != NULL) { - std::shared_ptr aPln = GeomAlgoAPI_FaceBuilder::plane(aShape); - std::shared_ptr aOrig = aPln->location(); - std::shared_ptr aDir = aPln->direction(); - - aOrig->translate(aDir, aDist); - std::shared_ptr aPlane = - GeomAlgoAPI_FaceBuilder::square(aOrig, aDir, PLANE_SIZE); - ResultConstructionPtr aConstr = document()->createConstruction(data()); - aConstr->setShape(aPlane); - setResult(aConstr); + if (!aShape.get()) { + aShape = aFaceAttr->context()->shape(); } + + if(!aShape.get()) { + return aPlane; + } + + std::shared_ptr aFace(new GeomAPI_Face(aShape)); + + std::shared_ptr aPln = aFace->getPlane(); + std::shared_ptr aOrig = aPln->location(); + std::shared_ptr aDir = aPln->direction(); + + aOrig->translate(aDir, aDist); + std::shared_ptr aNewPln(new GeomAPI_Pln(aOrig, aDir)); + + aPlane = makeRectangularFace(aFace, aNewPln); + } + return aPlane; +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Plane::createByCoincidentPoint() +{ + // Get face. + AttributeSelectionPtr aFaceSelection = selection(PLANE()); + GeomShapePtr aFaceShape = aFaceSelection->value(); + if(!aFaceShape.get()) { + aFaceShape = aFaceSelection->context()->shape(); } + std::shared_ptr aFace(new GeomAPI_Face(aFaceShape)); + + // Get point. + AttributeSelectionPtr aPointSelection = selection(COINCIDENT_POINT()); + GeomShapePtr aPointShape = aPointSelection->value(); + if(!aPointShape.get()) { + aPointShape = aPointSelection->context()->shape(); + } + std::shared_ptr aVertex(new GeomAPI_Vertex(aPointShape)); + + std::shared_ptr anOrig = aVertex->point(); + std::shared_ptr aPln = aFace->getPlane(); + std::shared_ptr aDir = aPln->direction(); + + std::shared_ptr aNewPln(new GeomAPI_Pln(anOrig, aDir)); + + return makeRectangularFace(aFace, aNewPln); } -void ConstructionPlugin_Plane::customisePresentation(AISObjectPtr thePrs) +//================================================================================================== +std::shared_ptr ConstructionPlugin_Plane::createByRotation() { - thePrs->setColor(50, 255, 50); - thePrs->setTransparensy(0.6); -} \ No newline at end of file + // Get face. + AttributeSelectionPtr aFaceSelection = selection(PLANE()); + GeomShapePtr aFaceShape = aFaceSelection->value(); + if(!aFaceShape.get()) { + aFaceShape = aFaceSelection->context()->shape(); + } + std::shared_ptr aFace(new GeomAPI_Face(aFaceShape)); + aFace = makeRectangularFace(aFace, aFace->getPlane()); + + // Get axis. + AttributeSelectionPtr anAxisSelection = selection(AXIS()); + GeomShapePtr anAxisShape = anAxisSelection->value(); + if(!anAxisShape.get()) { + anAxisShape = anAxisSelection->context()->shape(); + } + std::shared_ptr anEdge(new GeomAPI_Edge(anAxisShape)); + + std::shared_ptr anAxis = + std::shared_ptr(new GeomAPI_Ax1(anEdge->line()->location(), + anEdge->line()->direction())); + + // Getting angle. + double anAngle = real(ANGLE())->value(); + + GeomAlgoAPI_Rotation aRotationAlgo(aFace, anAxis, anAngle); + if (!aRotationAlgo.check()) { + setError(aRotationAlgo.getError()); + return GeomShapePtr(); + } + aRotationAlgo.build(); + if (!aRotationAlgo.isDone()) { + setError("Error: Failed to rotate plane"); + return GeomShapePtr(); + } + + std::shared_ptr aRes(new GeomAPI_Face(aRotationAlgo.shape())); + return aRes; +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Plane::createByTwoParallelPlanes() +{ + // Get plane 1. + AttributeSelectionPtr aFaceSelection1 = selection(PLANE1()); + GeomShapePtr aFaceShape1 = aFaceSelection1->value(); + if(!aFaceShape1.get()) { + aFaceShape1 = aFaceSelection1->context()->shape(); + } + std::shared_ptr aFace1(new GeomAPI_Face(aFaceShape1)); + std::shared_ptr aPln1 = aFace1->getPlane(); + + // Get plane 2. + AttributeSelectionPtr aFaceSelection2 = selection(PLANE2()); + GeomShapePtr aFaceShape2 = aFaceSelection2->value(); + if(!aFaceShape2.get()) { + aFaceShape2 = aFaceSelection2->context()->shape(); + } + std::shared_ptr aFace2(new GeomAPI_Face(aFaceShape2)); + std::shared_ptr aPln2 = aFace2->getPlane(); + + std::shared_ptr anOrig1 = aPln1->location(); + std::shared_ptr aPntOnPln2 = aPln2->project(anOrig1); + + std::shared_ptr aNewOrig(new GeomAPI_Pnt(anOrig1->xyz()->added( + aPntOnPln2->xyz())->multiplied(0.5))); + + std::shared_ptr aNewPln(new GeomAPI_Pln(aNewOrig, aPln1->direction())); + + std::shared_ptr aRes = makeRectangularFace(aFace1, aNewPln); + + return aRes; +} + +//================================================================================================== +GeomShapePtr faceByThreeVertices(const std::shared_ptr theV1, + const std::shared_ptr theV2, + const std::shared_ptr theV3) +{ + std::shared_ptr aFace = + GeomAlgoAPI_FaceBuilder::planarFaceByThreeVertices(theV1, theV2, theV3); + + ListOfShape anObjects; + anObjects.push_back(theV1); + anObjects.push_back(theV2); + anObjects.push_back(theV3); + std::list > aBoundingPoints = + GeomAlgoAPI_ShapeTools::getBoundingBox(anObjects, 1.0); + GeomShapePtr aRes = GeomAlgoAPI_ShapeTools::fitPlaneToBox(aFace, aBoundingPoints); + + return aRes; +} + +//================================================================================================== +std::shared_ptr makeRectangularFace(const std::shared_ptr theFace, + const std::shared_ptr thePln) +{ + // Create rectangular face close to the selected + double aXmin, aYmin, Zmin, aXmax, aYmax, Zmax; + theFace->computeSize(aXmin, aYmin, Zmin, aXmax, aYmax, Zmax); + + // use all 8 points of the bounding box to find the 2D bounds + bool isFirst = true; + double aMinX2d, aMaxX2d, aMinY2d, aMaxY2d; + for(int aXIsMin = 0; aXIsMin < 2; aXIsMin++) { + for(int aYIsMin = 0; aYIsMin < 2; aYIsMin++) { + for(int aZIsMin = 0; aZIsMin < 2; aZIsMin++) { + std::shared_ptr aPnt(new GeomAPI_Pnt( + aXIsMin ? aXmin : aXmax, aYIsMin ? aYmin : aYmax, aZIsMin ? Zmin : Zmax)); + std::shared_ptr aPnt2d = aPnt->to2D(thePln); + if (isFirst || aPnt2d->x() < aMinX2d) + aMinX2d = aPnt2d->x(); + if (isFirst || aPnt2d->y() < aMinY2d) + aMinY2d = aPnt2d->y(); + if (isFirst || aPnt2d->x() > aMaxX2d) + aMaxX2d = aPnt2d->x(); + if (isFirst || aPnt2d->y() > aMaxY2d) + aMaxY2d = aPnt2d->y(); + if (isFirst) + isFirst = !isFirst; + } + } + } + double aWgap = (aMaxX2d - aMinX2d) * 0.1; + double aHgap = (aMaxY2d - aMinY2d) * 0.1; + std::shared_ptr aResFace = GeomAlgoAPI_FaceBuilder::planarFace(thePln, + aMinX2d - aWgap, aMinY2d - aHgap, aMaxX2d - aMinX2d + 2. * aWgap, + aMaxY2d - aMinY2d + 2. * aHgap); + + return aResFace; +}