X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FConstructionPlugin%2FConstructionPlugin_Point.cpp;h=46c15698bbf5d5fb6cc99cc2f57a28a3accfa853;hb=1fd15bed1c62f994a3438bc4a12f44d0361f98c9;hp=9539004b7645ed5ff075cfa952fb4f46fc82753c;hpb=7bf19255421b34594c7b0a76d0ce28166d0ce895;p=modules%2Fshaper.git diff --git a/src/ConstructionPlugin/ConstructionPlugin_Point.cpp b/src/ConstructionPlugin/ConstructionPlugin_Point.cpp index 9539004b7..46c15698b 100644 --- a/src/ConstructionPlugin/ConstructionPlugin_Point.cpp +++ b/src/ConstructionPlugin/ConstructionPlugin_Point.cpp @@ -1,38 +1,412 @@ -// Copyright (C) 2014-20xx CEA/DEN, EDF R&D - -// File: ConstructionPlugin_Point.cxx -// Created: 27 Mar 2014 -// Author: Mikhail PONIKAROV +// Copyright (C) 2014-2021 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 +// License as published by the Free Software Foundation; either +// version 2.1 of the License, or (at your option) any later version. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// Lesser General Public License for more details. +// +// 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 +// +// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com +// #include "ConstructionPlugin_Point.h" -#include "ModelAPI_Session.h" -#include "ModelAPI_Document.h" -#include "ModelAPI_Data.h" -#include "ModelAPI_AttributeDouble.h" + +#include +#include +#include +#include +#include #include + #include +#include + +#include +#include + +#include #include +#include +#include +#include -using namespace std; +#include +//================================================================================================== ConstructionPlugin_Point::ConstructionPlugin_Point() { } +//================================================================================================== +const std::string& ConstructionPlugin_Point::getKind() +{ + static std::string MY_KIND = ConstructionPlugin_Point::ID(); + return MY_KIND; +} + +//================================================================================================== void ConstructionPlugin_Point::initAttributes() { - data()->addAttribute(POINT_ATTR_X, ModelAPI_AttributeDouble::type()); - data()->addAttribute(POINT_ATTR_Y, ModelAPI_AttributeDouble::type()); - data()->addAttribute(POINT_ATTR_Z, ModelAPI_AttributeDouble::type()); + data()->addAttribute(POINT3D(), GeomDataAPI_Point::typeId()); + + data()->addAttribute(CREATION_METHOD(), ModelAPI_AttributeString::typeId()); + + data()->addAttribute(INTERSECTION_LINE_1(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(INTERSECTION_LINE_2(), ModelAPI_AttributeSelection::typeId()); + + data()->addAttribute(INTERSECTION_LINE(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(INTERSECTION_PLANE(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(USE_OFFSET(), ModelAPI_AttributeString::typeId()); + data()->addAttribute(OFFSET(), ModelAPI_AttributeDouble::typeId()); + data()->addAttribute(REVERSE_OFFSET(), ModelAPI_AttributeBoolean::typeId()); + + data()->addAttribute(EDGE(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(OFFSET_TYPE(), ModelAPI_AttributeString::typeId()); + data()->addAttribute(DISTANCE(), ModelAPI_AttributeDouble::typeId()); + data()->addAttribute(RATIO(), ModelAPI_AttributeDouble::typeId()); + data()->addAttribute(REVERSE(), ModelAPI_AttributeBoolean::typeId()); + + data()->addAttribute(POINT_TO_PROJECT(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(PROJECTION_TYPE(), ModelAPI_AttributeString::typeId()); + data()->addAttribute(EDGE_FOR_POINT_PROJECTION(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(FACE_FOR_POINT_PROJECTION(), ModelAPI_AttributeSelection::typeId()); + + data()->addAttribute(INTERSECTION_TYPE(), ModelAPI_AttributeString::typeId()); + data()->addAttribute(INTERSECTION_PLANE_1(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(INTERSECTION_PLANE_2(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(INTERSECTION_PLANE_3(), ModelAPI_AttributeSelection::typeId()); + + data()->addAttribute(GEOMETRICAL_PROPERTY_TYPE(), ModelAPI_AttributeString::typeId()); + data()->addAttribute(OBJECT_FOR_CENTER_OF_GRAVITY(), ModelAPI_AttributeSelection::typeId()); + data()->addAttribute(OBJECT_FOR_CENTER_OF_CIRCLE(), ModelAPI_AttributeSelection::typeId()); } +//================================================================================================== void ConstructionPlugin_Point::execute() { - std::shared_ptr aPnt( - new GeomAPI_Pnt(data()->real(POINT_ATTR_X)->value(), data()->real(POINT_ATTR_Y)->value(), - data()->real(POINT_ATTR_Z)->value())); + GeomShapePtr aShape; + + // to support compatibility with old documents where aCreationMethod did not exist + std::string aCreationMethod = + string(CREATION_METHOD()).get() && !string(CREATION_METHOD())->value().empty() ? + string(CREATION_METHOD())->value() : CREATION_METHOD_BY_XYZ(); + if(aCreationMethod == CREATION_METHOD_BY_XYZ()) { + aShape = createByXYZ(); + } else if(aCreationMethod == CREATION_METHOD_BY_DISTANCE_ON_EDGE()) { + aShape = createByDistanceOnEdge(); + } else if(aCreationMethod == CREATION_METHOD_BY_PROJECTION()) { + if (string(PROJECTION_TYPE())->value() == PROJECTION_TYPE_ON_EDGE()) { + aShape = createByProjectionOnEdge(); + } else { + aShape = createByProjectionOnFace(); + } + } else if(aCreationMethod == CREATION_METHOD_BY_INTERSECTION()) { + std::string anIntersectionType = string(INTERSECTION_TYPE())->value(); + if (anIntersectionType == INTERSECTION_TYPE_BY_LINES()) { + aShape = createByLinesIntersection(); + } else if (anIntersectionType == INTERSECTION_TYPE_BY_LINE_AND_PLANE()) { + // this may produce several points + std::list > aPoints = createByLineAndPlaneIntersection(); + if (!aPoints.empty()) { // if no points found produce the standard error later + int anIndex = 0; + std::list >::iterator aPIter = aPoints.begin(); + for (; aPIter != aPoints.end(); aPIter++, anIndex++) { + std::shared_ptr aConstr = + document()->createConstruction(data(), anIndex); + aConstr->setShape(*aPIter); + setResult(aConstr, anIndex); + } + removeResults(anIndex); + return; + } + } else { + aShape = createByPlanesIntersection(); + } + } else if (aCreationMethod == CREATION_METHOD_BY_GEOMETRICAL_PROPERTY()) { + std::string aGeometricalPropertyType = string(GEOMETRICAL_PROPERTY_TYPE())->value(); + if (aGeometricalPropertyType == GEOMETRICAL_PROPERTY_TYPE_BY_CENTER_OF_GRAVITY()) { + aShape = createByCenterOfGravity(); + } else { + aShape = createByCenterOfCircle(); + } + } + if(!aShape.get()) { + setError("Error: intersection not found."); + return; + } + + removeResults(1); // for case the point type was switched from multi-results type std::shared_ptr aConstr = document()->createConstruction(data()); - aConstr->setShape(GeomAlgoAPI_PointBuilder::point(aPnt)); + aConstr->setInfinite(true); + aConstr->setShape(aShape); setResult(aConstr); } + +//================================================================================================== +bool ConstructionPlugin_Point::customisePresentation(ResultPtr theResult, + AISObjectPtr thePrs) +{ + 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", COLOR_NAME()); + + bool isCustomized = false; + if (aColor.size() == 3) + isCustomized = thePrs->setColor(aColor[0], aColor[1], aColor[2]); + //thePrs->setPointMarker(1, 1.); // Set point as a '+' symbol + return isCustomized; +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Point::createByXYZ() +{ + AttributePointPtr aPoint = + std::dynamic_pointer_cast(data()->attribute(POINT3D())); + return GeomAlgoAPI_PointBuilder::vertex(aPoint->x(), aPoint->y(), aPoint->z()); +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Point::createByDistanceOnEdge() +{ + // Get edge. + AttributeSelectionPtr anEdgeSelection = selection(EDGE()); + GeomShapePtr aShape = anEdgeSelection->value(); + if(!aShape.get()) { + aShape = anEdgeSelection->context()->shape(); + } + std::shared_ptr anEdge(new GeomAPI_Edge(aShape)); + + // Get distance value and percent flag. + double aValue; + bool anIsPercent = false; + if (string(OFFSET_TYPE())->value() == OFFSET_TYPE_BY_DISTANCE()) { + aValue = real(DISTANCE())->value(); + anIsPercent = false; + } else { + aValue = real(RATIO())->value() * 100.0; + anIsPercent = true; + } + + // Get reverse flag. + bool anIsReverse = boolean(REVERSE())->value(); + + return GeomAlgoAPI_PointBuilder::vertexOnEdge(anEdge, aValue, anIsPercent, anIsReverse); +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Point::createByProjectionOnEdge() +{ + // Get point. + AttributeSelectionPtr aPointSelection = selection(POINT_TO_PROJECT()); + GeomShapePtr aPointShape = aPointSelection->value(); + if (!aPointShape.get()) { + aPointShape = aPointSelection->context()->shape(); + } + std::shared_ptr aVertex(new GeomAPI_Vertex(aPointShape)); + + // Get edge. + AttributeSelectionPtr anEdgeSelection = selection(EDGE_FOR_POINT_PROJECTION()); + GeomShapePtr anEdgeShape = anEdgeSelection->value(); + if (!anEdgeShape.get()) { + anEdgeShape = anEdgeSelection->context()->shape(); + } + std::shared_ptr anEdge(new GeomAPI_Edge(anEdgeShape)); + + return GeomAlgoAPI_PointBuilder::vertexByProjection(aVertex, anEdge); +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Point::createByProjectionOnFace() +{ + // Get point. + AttributeSelectionPtr aPointSelection = selection(POINT_TO_PROJECT()); + GeomShapePtr aPointShape = aPointSelection->value(); + if(!aPointShape.get()) { + aPointShape = aPointSelection->context()->shape(); + } + std::shared_ptr aVertex(new GeomAPI_Vertex(aPointShape)); + + // Get plane. + AttributeSelectionPtr aPlaneSelection = selection(FACE_FOR_POINT_PROJECTION()); + GeomShapePtr aPlaneShape = aPlaneSelection->value(); + if(!aPlaneShape.get()) { + aPlaneShape = aPlaneSelection->context()->shape(); + } + std::shared_ptr aFace(new GeomAPI_Face(aPlaneShape)); + + return GeomAlgoAPI_PointBuilder::vertexByProjection(aVertex, aFace); +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Point::createByLinesIntersection() +{ + // Get first line. + AttributeSelectionPtr aFirstLineSelection= selection(INTERSECTION_LINE_1()); + GeomShapePtr aFirstLineShape = aFirstLineSelection->value(); + if(!aFirstLineShape.get()) { + aFirstLineShape = aFirstLineSelection->context()->shape(); + } + std::shared_ptr aFirstEdge(new GeomAPI_Edge(aFirstLineShape)); + + // Get second line. + AttributeSelectionPtr aSecondLineSelection= selection(INTERSECTION_LINE_2()); + GeomShapePtr aSecondLineShape = aSecondLineSelection->value(); + if(!aSecondLineShape.get()) { + aSecondLineShape = aSecondLineSelection->context()->shape(); + } + std::shared_ptr aSecondEdge(new GeomAPI_Edge(aSecondLineShape)); + + return GeomAlgoAPI_PointBuilder::vertexByIntersection(aFirstEdge, aSecondEdge); +} + +//================================================================================================== +std::list > + ConstructionPlugin_Point::createByLineAndPlaneIntersection() +{ + // Get line. + AttributeSelectionPtr aLineSelection = selection(INTERSECTION_LINE()); + GeomShapePtr aLineShape = aLineSelection->value(); + if(!aLineShape.get()) { + aLineShape = aLineSelection->context()->shape(); + } + GeomEdgePtr anEdge; + if (aLineShape->isEdge()) { + anEdge = aLineShape->edge(); + } + else if (aLineShape->isCompound()) { + GeomAPI_ShapeIterator anIt(aLineShape); + anEdge = anIt.current()->edge(); + } + + // Get plane. + AttributeSelectionPtr aPlaneSelection= selection(INTERSECTION_PLANE()); + GeomShapePtr aPlaneShape = aPlaneSelection->value(); + if(!aPlaneShape.get()) { + aPlaneShape = aPlaneSelection->context()->shape(); + } + GeomFacePtr aFace; + if (aPlaneShape->isFace()) { + aFace = aPlaneShape->face(); + } + else if (aPlaneShape->isCompound()) { + GeomAPI_ShapeIterator anIt(aPlaneShape); + aFace = anIt.current()->face(); + } + + if (!string(USE_OFFSET())->value().empty()) { + double anOffset = real(OFFSET())->value(); + if (boolean(REVERSE_OFFSET())->value()) + anOffset = -anOffset; + if (fabs(anOffset) > 1.e-9) { // move face + aFace->translate(aFace->getPlane()->direction(), anOffset); + } + } + + return GeomAlgoAPI_ShapeTools::intersect(anEdge, aFace); +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Point::createByPlanesIntersection() +{ + // Get plane. + AttributeSelectionPtr aPlaneSelection1 = selection(INTERSECTION_PLANE_1()); + GeomShapePtr aPlaneShape1 = aPlaneSelection1->value(); + if (!aPlaneShape1.get()) { + aPlaneShape1 = aPlaneSelection1->context()->shape(); + } + std::shared_ptr aFace1(new GeomAPI_Face(aPlaneShape1)); + std::shared_ptr aPln1 = aFace1->getPlane(); + + // Get plane. + AttributeSelectionPtr aPlaneSelection2 = selection(INTERSECTION_PLANE_2()); + GeomShapePtr aPlaneShape2 = aPlaneSelection2->value(); + if (!aPlaneShape2.get()) { + aPlaneShape2 = aPlaneSelection2->context()->shape(); + } + std::shared_ptr aFace2(new GeomAPI_Face(aPlaneShape2)); + std::shared_ptr aPln2 = aFace2->getPlane(); + + // Get plane. + AttributeSelectionPtr aPlaneSelection3 = selection(INTERSECTION_PLANE_3()); + GeomShapePtr aPlaneShape3 = aPlaneSelection3->value(); + if (!aPlaneShape3.get()) { + aPlaneShape3 = aPlaneSelection3->context()->shape(); + } + std::shared_ptr aFace3(new GeomAPI_Face(aPlaneShape3)); + std::shared_ptr aPln3 = aFace3->getPlane(); + + std::shared_ptr aVertex; + + std::shared_ptr anIntersectLine = aPln1->intersect(aPln2); + if (!anIntersectLine.get()) { + return aVertex; + } + + std::shared_ptr aPnt = aPln3->intersect(anIntersectLine); + if (aPnt.get()) { + aVertex.reset(new GeomAPI_Vertex(aPnt->x(), aPnt->y(), aPnt->z())); + } + + return aVertex; +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Point::createByCenterOfGravity() +{ + // Get shape. + AttributeSelectionPtr aShapeSelection = selection(OBJECT_FOR_CENTER_OF_GRAVITY()); + GeomShapePtr aShape = aShapeSelection->value(); + if (!aShape.get()) + { + aShape = aShapeSelection->context()->shape(); + } + + std::shared_ptr aVertex; + std::shared_ptr aPnt = GeomAlgoAPI_ShapeTools::centreOfMass(aShape); + if (aPnt.get()) + { + aVertex.reset(new GeomAPI_Vertex(aPnt->x(), aPnt->y(), aPnt->z())); + } + + return aVertex; +} + +//================================================================================================== +std::shared_ptr ConstructionPlugin_Point::createByCenterOfCircle() +{ + // Get shape. + AttributeSelectionPtr aShapeSelection = selection(OBJECT_FOR_CENTER_OF_CIRCLE()); + GeomShapePtr aShape = aShapeSelection->value(); + if (!aShape.get()) { + aShape = aShapeSelection->context()->shape(); + } + std::shared_ptr anEdge(new GeomAPI_Edge(aShape)); + std::shared_ptr aCirc = anEdge->circle(); + + std::shared_ptr aVertex; + std::shared_ptr aPnt = aCirc->center(); + if (aPnt.get()) { + aVertex.reset(new GeomAPI_Vertex(aPnt->x(), aPnt->y(), aPnt->z())); + } + + return aVertex; +}