1 // Copyright (C) 2014-2017 CEA/DEN, EDF R&D
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Lesser General Public License for more details.
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 // See http://www.salome-platform.org/ or
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
21 #include "SketcherPrs_PositionMgr.h"
22 #include "SketcherPrs_Tools.h"
24 #include <GeomAPI_Edge.h>
25 #include <GeomAPI_Curve.h>
26 #include <GeomAPI_Vertex.h>
27 #include <GeomAPI_Dir.h>
28 #include <GeomAPI_Ax3.h>
29 #include <GeomAPI_Circ.h>
31 #include <GeomDataAPI_Point2D.h>
33 #include <SketchPlugin_Line.h>
34 #include <SketchPlugin_Circle.h>
35 #include <SketchPlugin_Arc.h>
37 #include <BRepExtrema_ExtPC.hxx>
38 #include <TopoDS_Vertex.hxx>
39 #include <Geom_Curve.hxx>
40 #include <TColGeom_SequenceOfCurve.hxx>
43 static SketcherPrs_PositionMgr* MyPosMgr = NULL;
45 // The class is implemented as a singlton
46 SketcherPrs_PositionMgr* SketcherPrs_PositionMgr::get()
49 MyPosMgr = new SketcherPrs_PositionMgr();
53 SketcherPrs_PositionMgr::SketcherPrs_PositionMgr()
58 int SketcherPrs_PositionMgr::getPositionIndex(ObjectPtr theLine,
59 const SketcherPrs_SymbolPrs* thePrs)
61 if (myShapes.count(theLine) == 1) {
62 // Find the map and add new [Presentation - Index] pair
63 PositionsMap& aPosMap = myShapes[theLine];
64 if (aPosMap.count(thePrs) == 1) {
65 // return existing index
66 return aPosMap[thePrs];
68 // Add a new [Presentation - Index] pair
69 int aInd = int(aPosMap.size());
70 aPosMap[thePrs] = aInd;
74 // Create a new map with initial index
77 myShapes[theLine] = aPosMap;
83 gp_Vec getVector(ObjectPtr theShape, GeomDirPtr theDir, gp_Pnt theP)
86 std::shared_ptr<GeomAPI_Shape> aShape = SketcherPrs_Tools::getShape(theShape);
87 if (aShape->isEdge()) {
88 std::shared_ptr<GeomAPI_Curve> aCurve =
89 std::shared_ptr<GeomAPI_Curve>(new GeomAPI_Curve(aShape));
91 if (aCurve->isCircle()) {
92 GeomEdgePtr aEdgePtr(new GeomAPI_Edge(aShape));
93 GeomVertexPtr aVertexPtr(new GeomAPI_Vertex(theP.X(), theP.Y(), theP.Z()));
94 BRepExtrema_ExtPC aExtrema(aVertexPtr->impl<TopoDS_Vertex>(),
95 aEdgePtr->impl<TopoDS_Edge>());
96 int aNb = aExtrema.NbExt();
98 for (int i = 1; i <= aNb; i++) {
99 if (aExtrema.IsMin(i)) {
100 double aParam = aExtrema.Parameter(i);
101 Handle(Geom_Curve) aCurv = aCurve->impl<Handle_Geom_Curve>();
103 aCurv->D1(aParam, aP, aVec);
109 double aMidParam = (aCurve->startParam() + aCurve->endParam()) / 2.;
110 GeomPointPtr aPnt1 = aCurve->getPoint((aMidParam + aCurve->endParam()) / 2.);
111 GeomPointPtr aPnt2 = aCurve->getPoint((aMidParam + aCurve->startParam()) / 2.);
113 gp_Pnt aPn2 = aPnt2->impl<gp_Pnt>();
114 if (aPn2.IsEqual(theP, Precision::Confusion()))
115 aVec = gp_Vec(aPn2, aPnt1->impl<gp_Pnt>());
117 aVec = gp_Vec(aPnt1->impl<gp_Pnt>(), aPn2);
120 aVec = gp_Vec(theDir->impl<gp_Dir>());
125 gp_Pnt SketcherPrs_PositionMgr::getPosition(ObjectPtr theShape,
126 const SketcherPrs_SymbolPrs* thePrs,
127 double theStep, GeomPointPtr thePnt)
129 std::shared_ptr<GeomAPI_Shape> aShape = SketcherPrs_Tools::getShape(theShape);
130 gp_Pnt aP; // Central point
133 return getPointPosition(theShape, thePrs, theStep, thePnt);
135 if (aShape->isEdge()) {
136 std::shared_ptr<GeomAPI_Curve> aCurve =
137 std::shared_ptr<GeomAPI_Curve>(new GeomAPI_Curve(aShape));
138 // this is a circle or arc
139 double aMidParam = (aCurve->startParam() + aCurve->endParam()) / 2.;
140 std::shared_ptr<GeomAPI_Pnt> aPnt = aCurve->getPoint(aMidParam);
141 aP = aPnt->impl<gp_Pnt>();
144 std::shared_ptr<GeomAPI_Vertex> aVertex =
145 std::shared_ptr<GeomAPI_Vertex>(new GeomAPI_Vertex(aShape));
146 std::shared_ptr<GeomAPI_Pnt> aPnt = aVertex->point();
147 aP = aPnt->impl<gp_Pnt>();
151 gp_Vec aVec1 = getVector(theShape, thePrs->plane()->dirX(), aP);
153 // Compute shifting vector for a one symbol
154 gp_Vec aShift = aVec1.Crossed(thePrs->plane()->normal()->impl<gp_Dir>());
156 aShift.Multiply(theStep * 0.8);
158 // Shift the position coordinate according to position index
159 int aPos = getPositionIndex(theShape, thePrs);
161 if ((aPos % 2) == 0) {
163 aP.Translate(aShift);
168 aM = -(aPos + 2) / 4;
172 aP.Translate(-aShift);
174 if ((aPos - 1) % 4 == 0)
177 aM = -(aPos + 1) / 4;
181 // Normalize vector along the line
183 aVec1.Multiply(theStep);
184 aP.Translate(aVec1.Multiplied(aM));
190 //*****************************************************************
191 //! Returns curves connected to the given point
192 std::list<ObjectPtr> getCurves(const GeomPointPtr& thePnt, const SketcherPrs_SymbolPrs* thePrs)
194 std::list<ObjectPtr> aList;
195 GeomAx3Ptr aAx3 = thePrs->plane();
196 ModelAPI_CompositeFeature* aOwner = thePrs->sketcher();
197 GeomPnt2dPtr aPnt2d = thePnt->to2D(aAx3->origin(), aAx3->dirX(), aAx3->dirY());
199 int aNbSubs = aOwner->numberOfSubs();
200 for (int i = 0; i < aNbSubs; i++) {
201 FeaturePtr aFeature = aOwner->subFeature(i);
202 if (aFeature->getKind() == SketchPlugin_Line::ID()) {
203 AttributePoint2DPtr aSPnt1 = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
204 aFeature->data()->attribute(SketchPlugin_Line::START_ID()));
205 AttributePoint2DPtr aSPnt2 = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
206 aFeature->data()->attribute(SketchPlugin_Line::END_ID()));
208 GeomPnt2dPtr aPnt1 = aSPnt1->pnt();
209 GeomPnt2dPtr aPnt2 = aSPnt2->pnt();
211 if (aPnt1->isEqual(aPnt2d) || aPnt2->isEqual(aPnt2d)) {
212 GeomShapePtr aShp = SketcherPrs_Tools::getShape(aFeature->firstResult());
213 GeomCurvePtr aCurv = std::shared_ptr<GeomAPI_Curve>(new GeomAPI_Curve(aShp));
214 aList.push_back(aFeature->firstResult());
216 } else if ((aFeature->getKind() == SketchPlugin_Circle::ID()) ||
217 (aFeature->getKind() == SketchPlugin_Arc::ID())) {
220 std::list<ResultPtr> aResults = aFeature->results();
221 std::list<ResultPtr>::const_iterator aIt;
222 for (aIt = aResults.cbegin(); aIt != aResults.cend(); aIt++) {
223 GeomShapePtr aShp = SketcherPrs_Tools::getShape((*aIt));
224 if (aShp->isEdge()) {
226 aCurve = std::shared_ptr<GeomAPI_Curve>(new GeomAPI_Curve(aShp));
231 double aStart = aCurve->startParam();
232 double aEnd = aCurve->endParam();
233 GeomCirclePtr aCircle = GeomCirclePtr(new GeomAPI_Circ(aCurve));
235 if (aCircle->parameter(thePnt, 1.e-4, aParam) && (aParam >= aStart) && (aParam <= aEnd))
236 aList.push_back(aResObj);
243 //*****************************************************************
244 gp_Pnt SketcherPrs_PositionMgr::getPointPosition(
245 ObjectPtr theLine, const SketcherPrs_SymbolPrs* thePrs,
246 double theStep, GeomPointPtr thePnt)
248 std::list<ObjectPtr> aCurves = getCurves(thePnt, thePrs);
250 gp_Pnt aP = thePnt->impl<gp_Pnt>();
251 //gp_Vec aVec1 = getVector(theLine, thePrs->plane()->dirX(), aP);
252 std::list<gp_Vec> aVectors;
253 std::list<ObjectPtr>::const_iterator aItCurv;
254 for (aItCurv = aCurves.cbegin(); aItCurv != aCurves.cend(); aItCurv++) {
255 ObjectPtr aObject = (*aItCurv);
256 gp_Vec aVec = getVector(aObject, thePrs->plane()->dirX(), aP);
257 aVectors.push_back(aVec);
260 gp_Vec aBase = getVector(theLine, thePrs->plane()->dirX(), aP);
261 std::list<double> aAngles;
262 std::list<gp_Vec>::const_iterator aItVec;
263 for (aItVec = aVectors.cbegin(); aItVec != aVectors.cend(); aItVec++) {
264 gp_Vec aVec = (*aItVec);
265 double aAngle = aBase.Angle(aVec);
266 aAngles.push_back(aAngle);
268 // Compute shifting vector for a one symbol
269 //gp_Vec aShift = aVec1.Crossed(thePrs->plane()->normal()->impl<gp_Dir>());
270 //aShift.Normalize();
271 //aShift.Multiply(theStep * 1.5);
272 //aP.Translate(aShift);
276 //*****************************************************************
277 void SketcherPrs_PositionMgr::deleteConstraint(const SketcherPrs_SymbolPrs* thePrs)
279 std::map<ObjectPtr, PositionsMap>::iterator aIt;
280 std::list<ObjectPtr> aToDel;
281 // Clear map for deleted presentation
282 for (aIt = myShapes.begin(); aIt != myShapes.end(); ++aIt) {
283 PositionsMap& aPosMap = aIt->second;
284 if (aPosMap.count(thePrs) > 0) {
286 aPosMap.erase(aPosMap.find(thePrs));
287 if (aPosMap.size() == 0)
289 aToDel.push_back(aIt->first);
291 // Reindex objects positions in order to avoid spaces
292 PositionsMap::iterator aIt;
294 for (aIt = aPosMap.begin(); aIt != aPosMap.end(); aIt++, i++)
299 std::list<ObjectPtr>::const_iterator aListIt;
300 for (aListIt = aToDel.cbegin(); aListIt != aToDel.cend(); ++aListIt) {
301 myShapes.erase(*aListIt);