]> SALOME platform Git repositories - modules/shaper.git/blob - src/SketcherPrs/SketcherPrs_PositionMgr.cpp
Salome HOME
Issue #2208: Tangent constraint position implementation
[modules/shaper.git] / src / SketcherPrs / SketcherPrs_PositionMgr.cpp
1 // Copyright (C) 2014-2017  CEA/DEN, EDF R&D
2 //
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.
7 //
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.
12 //
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
16 //
17 // See http://www.salome-platform.org/ or
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
19 //
20
21 #include "SketcherPrs_PositionMgr.h"
22 #include "SketcherPrs_Tools.h"
23
24 #include <GeomAPI_Edge.h>
25 #include <GeomAPI_Curve.h>
26 #include <GeomAPI_Vertex.h>
27 #include <GeomAPI_Dir.h>
28
29 #include <BRepExtrema_ExtPC.hxx>
30 #include <TopoDS_Vertex.hxx>
31 #include <Geom_Curve.hxx>
32
33 static SketcherPrs_PositionMgr* MyPosMgr = NULL;
34
35 // The class is implemented as a singlton
36 SketcherPrs_PositionMgr* SketcherPrs_PositionMgr::get()
37 {
38   if (MyPosMgr == NULL)
39     MyPosMgr = new SketcherPrs_PositionMgr();
40   return MyPosMgr;
41 }
42
43 SketcherPrs_PositionMgr::SketcherPrs_PositionMgr()
44 {
45 }
46
47
48 int SketcherPrs_PositionMgr::getPositionIndex(ObjectPtr theLine,
49                                               const SketcherPrs_SymbolPrs* thePrs)
50 {
51   if (myShapes.count(theLine) == 1) {
52     // Find the map and add new [Presentation - Index] pair
53     PositionsMap& aPosMap = myShapes[theLine];
54     if (aPosMap.count(thePrs) == 1) {
55       // return existing index
56       return aPosMap[thePrs];
57     } else {
58       // Add a new [Presentation - Index] pair
59       int aInd = int(aPosMap.size());
60       aPosMap[thePrs] = aInd;
61       return aInd;
62     }
63   } else {
64     // Create a new map with initial index
65     PositionsMap aPosMap;
66     aPosMap[thePrs] = 0;
67     myShapes[theLine] = aPosMap;
68     return 0;
69   }
70 }
71
72
73 gp_Vec getVector(ObjectPtr theShape, GeomDirPtr theDir, gp_Pnt theP)
74 {
75   gp_Vec aVec;
76   std::shared_ptr<GeomAPI_Shape> aShape = SketcherPrs_Tools::getShape(theShape);
77   if (aShape->isEdge()) {
78     std::shared_ptr<GeomAPI_Curve> aCurve =
79       std::shared_ptr<GeomAPI_Curve>(new GeomAPI_Curve(aShape));
80
81     if (aCurve->isCircle()) {
82       GeomEdgePtr aEdgePtr(new GeomAPI_Edge(aShape));
83       GeomVertexPtr aVertexPtr(new GeomAPI_Vertex(theP.X(), theP.Y(), theP.Z()));
84       BRepExtrema_ExtPC aExtrema(aVertexPtr->impl<TopoDS_Vertex>(),
85                                  aEdgePtr->impl<TopoDS_Edge>());
86       int aNb = aExtrema.NbExt();
87       if (aNb > 0) {
88         for (int i = 1; i <= aNb; i++) {
89           if (aExtrema.IsMin(i)) {
90             double aParam = aExtrema.Parameter(i);
91             Handle(Geom_Curve) aCurv = aCurve->impl<Handle_Geom_Curve>();
92             gp_Pnt aP;
93             aCurv->D1(aParam, aP, aVec);
94             break;
95           }
96         }
97       }
98     } else {
99       double aMidParam = (aCurve->startParam() + aCurve->endParam()) / 2.;
100       GeomPointPtr aPnt1 = aCurve->getPoint((aMidParam + aCurve->endParam()) / 2.);
101       GeomPointPtr aPnt2 = aCurve->getPoint((aMidParam + aCurve->startParam()) / 2.);
102
103       aVec = gp_Vec(aPnt1->impl<gp_Pnt>(), aPnt2->impl<gp_Pnt>());
104     }
105   } else {
106     aVec = gp_Vec(theDir->impl<gp_Dir>());
107   }
108   return aVec;
109 }
110
111 gp_Pnt SketcherPrs_PositionMgr::getPosition(ObjectPtr theShape,
112                                             const SketcherPrs_SymbolPrs* thePrs,
113                                             double theStep, GeomPointPtr thePnt)
114 {
115   std::shared_ptr<GeomAPI_Shape> aShape = SketcherPrs_Tools::getShape(theShape);
116   gp_Pnt aP; // Central point
117
118   if (thePnt.get()) {
119     aP = thePnt->impl<gp_Pnt>();
120   } else {
121     if (aShape->isEdge()) {
122       std::shared_ptr<GeomAPI_Curve> aCurve =
123         std::shared_ptr<GeomAPI_Curve>(new GeomAPI_Curve(aShape));
124       // this is a circle or arc
125       double aMidParam = (aCurve->startParam() + aCurve->endParam()) / 2.;
126       std::shared_ptr<GeomAPI_Pnt> aPnt = aCurve->getPoint(aMidParam);
127       aP = aPnt->impl<gp_Pnt>();
128     } else {
129       // This is a point
130       std::shared_ptr<GeomAPI_Vertex> aVertex =
131         std::shared_ptr<GeomAPI_Vertex>(new GeomAPI_Vertex(aShape));
132       std::shared_ptr<GeomAPI_Pnt> aPnt = aVertex->point();
133       aP = aPnt->impl<gp_Pnt>();
134     }
135   }
136   // main vector
137   gp_Vec aVec1 = getVector(theShape, thePrs->plane()->dirX(), aP);
138
139   // Compute shifting vector for a one symbol
140   gp_Vec aShift = aVec1.Crossed(thePrs->plane()->normal()->impl<gp_Dir>());
141   aShift.Normalize();
142   // For point based symbols step = 1.2, for line based = 0.8
143   aShift.Multiply(theStep * (thePnt.get()? 1.2 : 0.8));
144
145   // Shift the position coordinate according to position index
146   int aPos = getPositionIndex(theShape, thePrs);
147   int aM = 1;
148   if ((aPos % 2) == 0) {
149     // Even position
150     aP.Translate(aShift);
151     if (aPos > 0) {
152       if (aPos % 4 == 0)
153         aM = aPos / 4;
154       else
155         aM = -(aPos + 2) / 4;
156     }
157   } else {
158     // Odd position
159     aP.Translate(-aShift);
160     if (aPos > 1) {
161       if ((aPos - 1) % 4 == 0)
162         aM = (aPos - 1) / 4;
163       else
164         aM = -(aPos + 1) / 4;
165     }
166   }
167   if (aPos > 1) {
168     // Normalize vector along the line
169     aVec1.Normalize();
170     aVec1.Multiply(theStep);
171     aP.Translate(aVec1.Multiplied(aM));
172   }
173   return aP;
174 }
175
176 void SketcherPrs_PositionMgr::deleteConstraint(const SketcherPrs_SymbolPrs* thePrs)
177 {
178   std::map<ObjectPtr, PositionsMap>::iterator aIt;
179   std::list<ObjectPtr> aToDel;
180   // Clear map for deleted presentation
181   for (aIt = myShapes.begin(); aIt != myShapes.end(); ++aIt) {
182     PositionsMap& aPosMap = aIt->second;
183     if (aPosMap.count(thePrs) > 0) {
184       // Erase index
185       aPosMap.erase(aPosMap.find(thePrs));
186       if (aPosMap.size() == 0)
187         // Delete the map
188         aToDel.push_back(aIt->first);
189       else {
190         // Reindex objects positions in order to avoid spaces
191         PositionsMap::iterator aIt;
192         int i = 0;
193         for (aIt = aPosMap.begin(); aIt != aPosMap.end(); aIt++, i++)
194           aIt->second = i;
195       }
196     }
197   }
198   std::list<ObjectPtr>::const_iterator aListIt;
199   for (aListIt = aToDel.cbegin(); aListIt != aToDel.cend(); ++aListIt) {
200     myShapes.erase(*aListIt);
201   }
202 }