Salome HOME
Merge branch 'BR_internationalization'
[modules/shaper.git] / src / SketchPlugin / SketchPlugin_Arc.cpp
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
2
3 // File:        SketchPlugin_Arc.cpp
4 // Created:     26 Apr 2014
5 // Author:      Artem ZHIDKOV
6
7 #include "SketchPlugin_Arc.h"
8 #include "SketchPlugin_Sketch.h"
9 #include <SketchPlugin_ConstraintCoincidence.h>
10 #include <SketchPlugin_ConstraintTangent.h>
11
12 #include <Events_Loop.h>
13 #include <ModelAPI_Data.h>
14 #include <ModelAPI_ResultConstruction.h>
15 #include <ModelAPI_AttributeDouble.h>
16 #include <ModelAPI_AttributeRefAttr.h>
17 #include <ModelAPI_AttributeSelection.h>
18 #include <ModelAPI_AttributeString.h>
19 #include <ModelAPI_Events.h>
20 #include <ModelAPI_Validator.h>
21 #include <ModelAPI_Session.h>
22 #include <ModelAPI_Tools.h>
23
24 #include <GeomAPI_Ax2.h>
25 #include <GeomAPI_Circ2d.h>
26 #include <GeomAPI_Circ.h>
27 #include <GeomAPI_Dir2d.h>
28 #include <GeomAPI_Dir.h>
29 #include <GeomAPI_Lin2d.h>
30 #include <GeomAPI_Lin.h>
31 #include <GeomAPI_Pnt2d.h>
32 #include <GeomAPI_XY.h>
33 #include <GeomDataAPI_Point2D.h>
34 #include <GeomDataAPI_Dir.h>
35 #include <GeomAlgoAPI_PointBuilder.h>
36 #include <GeomAlgoAPI_EdgeBuilder.h>
37 #include <GeomAlgoAPI_CompoundBuilder.h>
38 // for sqrt on Linux
39 #include <math.h>
40
41 const double tolerance = 1e-7;
42 const double paramTolerance = 1.e-4;
43 const double PI =3.141592653589793238463;
44
45 namespace {
46   static const std::string& ARC_TYPE_CENTER_START_END()
47   {
48     static const std::string TYPE("CenterStartEnd");
49     return TYPE;
50   }
51   static const std::string& ARC_TYPE_THREE_POINTS()
52   {
53     static const std::string TYPE("ThreePoints");
54     return TYPE;
55   }
56
57   static const std::string& PASSED_POINT_ID()
58   {
59     static const std::string PASSED_PNT("ArcPassedPoint");
60     return PASSED_PNT;
61   }
62   static const std::string& RADIUS_ID()
63   {
64     static const std::string RADIUS("ArcRadius");
65     return RADIUS;
66   }
67   static const std::string& ANGLE_ID()
68   {
69     static const std::string ANGLE("ArcAngle");
70     return ANGLE;
71   }
72
73   static const std::string& POINT_ID(int theIndex)
74   {
75     switch (theIndex) {
76     case 1: return SketchPlugin_Arc::START_ID();
77     case 2: return SketchPlugin_Arc::END_ID();
78     case 3: return PASSED_POINT_ID();
79     }
80
81     static const std::string DUMMY;
82     return DUMMY;
83   }
84 }
85
86
87
88 SketchPlugin_Arc::SketchPlugin_Arc()
89     : SketchPlugin_SketchEntity()
90 {
91   myStartUpdate = false;
92   myEndUpdate = false;
93   // default values
94   myXEndBefore = 0;
95   myYEndBefore = 0;
96
97   myParamBefore = 0;
98 }
99
100 void SketchPlugin_Arc::initDerivedClassAttributes()
101 {
102   data()->addAttribute(CENTER_ID(), GeomDataAPI_Point2D::typeId());
103   data()->addAttribute(START_ID(), GeomDataAPI_Point2D::typeId());
104   std::shared_ptr<GeomDataAPI_Point2D> anEndAttr = std::dynamic_pointer_cast<
105     GeomDataAPI_Point2D>(data()->addAttribute(END_ID(), GeomDataAPI_Point2D::typeId()));
106   data()->addAttribute(EXTERNAL_ID(), ModelAPI_AttributeSelection::typeId());
107   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), EXTERNAL_ID());
108
109   AttributeBooleanPtr isInversed = std::dynamic_pointer_cast<ModelAPI_AttributeBoolean>(
110     data()->addAttribute(INVERSED_ID(), ModelAPI_AttributeBoolean::typeId()));
111
112   // get the initial values
113   if (anEndAttr->isInitialized()) {
114     myXEndBefore = anEndAttr->x();
115     myYEndBefore = anEndAttr->y();
116   }
117
118   AttributeStringPtr anArcType = std::dynamic_pointer_cast<ModelAPI_AttributeString>(
119     data()->addAttribute(ARC_TYPE(), ModelAPI_AttributeString::typeId()));
120
121   data()->addAttribute(PASSED_POINT_ID(), GeomDataAPI_Point2D::typeId());
122   data()->addAttribute(TANGENT_POINT_ID(), ModelAPI_AttributeRefAttr::typeId());
123   data()->addAttribute(RADIUS_ID(), ModelAPI_AttributeDouble::typeId());
124   data()->addAttribute(ANGLE_ID(), ModelAPI_AttributeDouble::typeId());
125
126   // set after all to avoid in attributeChanged reference to not existing attributes
127   if (!isInversed->isInitialized())
128     isInversed->setValue(false);
129   anArcType->setValue(ARC_TYPE_CENTER_START_END());
130 }
131
132 void SketchPlugin_Arc::execute()
133 {
134   SketchPlugin_Sketch* aSketch = sketch();
135   // result for the arc is set only when all obligatory attributes are initialized,
136   // otherwise AIS object is used to visualize the arc's preview
137   if (aSketch && isFeatureValid()) {
138     ResultPtr aLastResult = lastResult();
139     bool hasResult = aLastResult && aLastResult.get();
140
141     // compute a circle point in 3D view
142     std::shared_ptr<GeomDataAPI_Point2D> aCenterAttr = std::dynamic_pointer_cast<
143         GeomDataAPI_Point2D>(data()->attribute(CENTER_ID()));
144     // compute the arc start point
145     std::shared_ptr<GeomDataAPI_Point2D> aStartAttr = std::dynamic_pointer_cast<
146         GeomDataAPI_Point2D>(data()->attribute(START_ID()));
147
148     std::shared_ptr<GeomAPI_Pnt> aCenter(aSketch->to3D(aCenterAttr->x(), aCenterAttr->y()));
149     // make a visible point
150     std::shared_ptr<GeomAPI_Shape> aCenterPointShape = GeomAlgoAPI_PointBuilder::point(aCenter);
151     std::shared_ptr<ModelAPI_ResultConstruction> aConstr1 = document()->createConstruction(
152         data(), 0);
153     aConstr1->setShape(aCenterPointShape);
154     aConstr1->setIsInHistory(false);
155     setResult(aConstr1, 0);
156
157     // make a visible circle
158     std::shared_ptr<GeomDataAPI_Dir> aNDir = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
159         aSketch->data()->attribute(SketchPlugin_Sketch::NORM_ID()));
160     std::shared_ptr<GeomAPI_Dir> aNormal = aNDir->dir();
161     std::shared_ptr<GeomAPI_Pnt> aStartPoint(aSketch->to3D(aStartAttr->x(), aStartAttr->y()));
162
163     // compute and change the arc end point
164     std::shared_ptr<GeomDataAPI_Point2D> anEndAttr = std::dynamic_pointer_cast<
165         GeomDataAPI_Point2D>(data()->attribute(END_ID()));
166     std::shared_ptr<GeomAPI_Circ2d> aCircleForArc(
167         new GeomAPI_Circ2d(aCenterAttr->pnt(), aStartAttr->pnt()));
168     std::shared_ptr<GeomAPI_Pnt2d> aProjection = aCircleForArc->project(anEndAttr->pnt());
169     if (aProjection && anEndAttr->pnt()->distance(aProjection) > tolerance)
170       anEndAttr->setValue(aProjection);
171     std::shared_ptr<GeomAPI_Pnt> aEndPoint(aSketch->to3D(anEndAttr->x(), anEndAttr->y()));
172     AttributeBooleanPtr isInversed =
173         std::dynamic_pointer_cast<ModelAPI_AttributeBoolean>(attribute(INVERSED_ID()));
174
175     std::shared_ptr<GeomAPI_Dir> anXDir(new GeomAPI_Dir(aStartPoint->xyz()->decreased(aCenter->xyz())));
176     std::shared_ptr<GeomAPI_Ax2> anAx2(new GeomAPI_Ax2(aCenter, aNormal, anXDir));
177     std::shared_ptr<GeomAPI_Circ> aCirc(new GeomAPI_Circ(anAx2, aCenter->distance(aStartPoint)));
178     double aParameterNew = 0.0;
179     if(aCirc->parameter(aEndPoint, paramTolerance, aParameterNew)) {
180       if(0 <= myParamBefore && myParamBefore <= PI / 2.0
181         && PI * 1.5 <= aParameterNew && aParameterNew <= PI * 2.0) {
182           isInversed->setValue(true);
183       } else if(PI * 1.5 <= myParamBefore && myParamBefore <= PI * 2.0
184         && 0 <= aParameterNew && aParameterNew <= PI / 2.0) {
185           isInversed->setValue(false);
186       }
187     }
188     myParamBefore = aParameterNew;
189
190     std::shared_ptr<GeomAPI_Shape> aCircleShape;
191     if(!isInversed->value()) {
192       aCircleShape = GeomAlgoAPI_EdgeBuilder::lineCircleArc(aCenter, aStartPoint, aEndPoint, aNormal);
193     } else {
194       aCircleShape = GeomAlgoAPI_EdgeBuilder::lineCircleArc(aCenter, aEndPoint, aStartPoint, aNormal);
195     }
196
197     if (aCircleShape) {
198       std::shared_ptr<ModelAPI_ResultConstruction> aConstr2 = document()->createConstruction(
199           data(), 1);
200       aConstr2->setShape(aCircleShape);
201       aConstr2->setIsInHistory(false);
202       setResult(aConstr2, 1);
203     }
204
205     // update radius and angle
206     updateDependentAttributes();
207   }
208 }
209
210 AISObjectPtr SketchPlugin_Arc::getAISObject(AISObjectPtr thePrevious)
211 {
212   SketchPlugin_Sketch* aSketch = sketch();
213   if (aSketch) {
214     // if the feature is valid, the execute() method should be performed, AIS object is empty
215     if (!isFeatureValid()) {
216       // compute a circle point in 3D view
217       std::shared_ptr<GeomDataAPI_Point2D> aCenterAttr = std::dynamic_pointer_cast<
218           GeomDataAPI_Point2D>(data()->attribute(CENTER_ID()));
219
220       std::list<std::shared_ptr<GeomAPI_Shape> > aShapes;
221       if (aCenterAttr->isInitialized()) {
222         std::shared_ptr<GeomAPI_Pnt> aCenter(aSketch->to3D(aCenterAttr->x(), aCenterAttr->y()));
223         // make a visible point
224         std::shared_ptr<GeomAPI_Shape> aCenterPointShape = GeomAlgoAPI_PointBuilder::point(aCenter);
225         aShapes.push_back(aCenterPointShape);
226
227         std::shared_ptr<GeomDataAPI_Point2D> aStartAttr = std::dynamic_pointer_cast<
228             GeomDataAPI_Point2D>(data()->attribute(SketchPlugin_Arc::START_ID()));
229         std::shared_ptr<GeomDataAPI_Point2D> aEndAttr = std::dynamic_pointer_cast<
230             GeomDataAPI_Point2D>(data()->attribute(SketchPlugin_Arc::END_ID()));
231         AttributeStringPtr aTypeAttr = std::dynamic_pointer_cast<ModelAPI_AttributeString>(
232             data()->attribute(ARC_TYPE()));
233
234         if (aStartAttr->isInitialized()) {
235           // make a visible circle
236           std::shared_ptr<GeomDataAPI_Dir> aNDir = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
237               aSketch->data()->attribute(SketchPlugin_Sketch::NORM_ID()));
238           bool aHasPlane = aNDir && !(aNDir->x() == 0 && aNDir->y() == 0 && aNDir->z() == 0);
239           if (aHasPlane) {
240             std::shared_ptr<GeomAPI_Dir> aNormal = aNDir->dir();
241             std::shared_ptr<GeomAPI_Pnt> aStartPoint(aSketch->to3D(aStartAttr->x(), aStartAttr->y()));
242             std::shared_ptr<GeomAPI_Pnt> aEndPoint = aStartPoint;
243             if (aTypeAttr && aTypeAttr->isInitialized() &&
244                 aTypeAttr->value() == ARC_TYPE_THREE_POINTS() && aEndAttr->isInitialized())
245               aEndPoint = aSketch->to3D(aEndAttr->x(), aEndAttr->y());
246
247             std::shared_ptr<GeomAPI_Shape> aCircleShape = GeomAlgoAPI_EdgeBuilder::lineCircleArc(
248                                                             aCenter, aStartPoint, aEndPoint, aNormal);
249             if (aCircleShape)
250               aShapes.push_back(aCircleShape);
251           }
252         }
253       }
254       if (!aShapes.empty()) {
255         std::shared_ptr<GeomAPI_Shape> aCompound = GeomAlgoAPI_CompoundBuilder::compound(aShapes);
256         AISObjectPtr anAIS = thePrevious;
257         if (!anAIS)
258           anAIS = AISObjectPtr(new GeomAPI_AISObject);
259         anAIS->createShape(aCompound);
260         anAIS->setWidth(3);
261         return anAIS;
262       }
263     }
264   }
265   return AISObjectPtr();
266 }
267
268 void SketchPlugin_Arc::move(double theDeltaX, double theDeltaY)
269 {
270   std::shared_ptr<ModelAPI_Data> aData = data();
271   if (!aData->isValid())
272     return;
273
274   aData->blockSendAttributeUpdated(true);
275
276   myStartUpdate = true;
277   myEndUpdate = true;
278   std::shared_ptr<GeomDataAPI_Point2D> aPoint2 = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
279       aData->attribute(SketchPlugin_Arc::START_ID()));
280   aPoint2->move(theDeltaX, theDeltaY);
281
282   std::shared_ptr<GeomDataAPI_Point2D> aPoint3 = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
283       aData->attribute(SketchPlugin_Arc::END_ID()));
284   aPoint3->move(theDeltaX, theDeltaY);
285   myStartUpdate = false;
286   myEndUpdate = false;
287
288   std::shared_ptr<GeomDataAPI_Point2D> aPoint1 = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
289       aData->attribute(SketchPlugin_Arc::CENTER_ID()));
290   aPoint1->move(theDeltaX, theDeltaY);
291
292   std::shared_ptr<GeomDataAPI_Point2D> aPassedPoint =
293       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(PASSED_POINT_ID()));
294   aPassedPoint->move(theDeltaX, theDeltaY);
295   aData->blockSendAttributeUpdated(false);
296 }
297
298 bool SketchPlugin_Arc::isFixed() {
299   return data()->selection(EXTERNAL_ID())->context().get() != NULL;
300 }
301
302 bool SketchPlugin_Arc::isFeatureValid()
303 {
304   AttributeStringPtr anArcTypeAttr =
305       std::dynamic_pointer_cast<ModelAPI_AttributeString>(data()->attribute(ARC_TYPE()));
306   if (!anArcTypeAttr)
307     return false;
308   std::string anArcType = anArcTypeAttr->value();
309
310   std::shared_ptr<GeomDataAPI_Point2D> aCenterAttr = std::dynamic_pointer_cast<
311       GeomDataAPI_Point2D>(data()->attribute(SketchPlugin_Arc::CENTER_ID()));
312   std::shared_ptr<GeomDataAPI_Point2D> aStartAttr = std::dynamic_pointer_cast<
313       GeomDataAPI_Point2D>(data()->attribute(SketchPlugin_Arc::START_ID()));
314   std::shared_ptr<GeomDataAPI_Point2D> anEndAttr = std::dynamic_pointer_cast<
315       GeomDataAPI_Point2D>(data()->attribute(SketchPlugin_Arc::END_ID()));
316   std::shared_ptr<GeomDataAPI_Point2D> aPassedAttr = std::dynamic_pointer_cast<
317       GeomDataAPI_Point2D>(data()->attribute(PASSED_POINT_ID()));
318
319   bool isValid = false;
320   if (anArcType == ARC_TYPE_THREE_POINTS())
321     isValid = aStartAttr->isInitialized() && anEndAttr->isInitialized() && aPassedAttr->isInitialized();
322   else
323     isValid = aCenterAttr->isInitialized() && aStartAttr->isInitialized() && anEndAttr->isInitialized();
324
325   return isValid;
326 }
327
328 static inline void adjustPeriod(double& theParam)
329 {
330   static const double PERIOD = 2.0 * PI;
331   while (theParam < 0.0) theParam += PERIOD;
332   while (theParam >= PERIOD) theParam -= PERIOD;
333 }
334
335 static inline void calculateArcAngleRadius(
336     const std::shared_ptr<GeomAPI_Circ2d>& theCircle,
337     const std::shared_ptr<GeomAPI_Pnt2d>& theStartPoint,
338     const std::shared_ptr<GeomAPI_Pnt2d>& theEndPoint,
339     const std::shared_ptr<GeomAPI_Pnt2d>& thePassedPoint,
340     AttributeDoublePtr theAngleAttr,
341     AttributeDoublePtr theRadiusAttr)
342 {
343   double aStartParam, aEndParam, aPassedParam;
344   theCircle->parameter(theStartPoint, paramTolerance, aStartParam);
345   theCircle->parameter(theEndPoint, paramTolerance, aEndParam);
346   theCircle->parameter(thePassedPoint, paramTolerance, aPassedParam);
347   adjustPeriod(aStartParam);
348   adjustPeriod(aEndParam);
349   adjustPeriod(aPassedParam);
350
351   if (aPassedParam >= aStartParam && aPassedParam <= aEndParam)
352     theAngleAttr->setValue((aEndParam - aStartParam) * 180.0 / PI);
353   else
354     theAngleAttr->setValue((aEndParam - aStartParam - 2.0 * PI) * 180.0 / PI);
355   theRadiusAttr->setValue(theCircle->radius());
356 }
357
358 static inline bool calculatePassedPoint(
359     const std::shared_ptr<GeomAPI_Pnt2d>& theCenter,
360     const std::shared_ptr<GeomAPI_Pnt2d>& theStartPoint,
361     const std::shared_ptr<GeomAPI_Pnt2d>& theEndPoint,
362     bool theArcReversed,
363     std::shared_ptr<GeomDataAPI_Point2D> thePassedPoint)
364 {
365   if (theCenter->distance(theStartPoint) < tolerance ||
366       theCenter->distance(theEndPoint) < tolerance)
367     return false;
368
369   std::shared_ptr<GeomAPI_Dir2d> aStartDir(new GeomAPI_Dir2d(
370       theStartPoint->xy()->decreased(theCenter->xy())));
371   std::shared_ptr<GeomAPI_Dir2d> aEndDir(new GeomAPI_Dir2d(
372       theEndPoint->xy()->decreased(theCenter->xy())));
373   std::shared_ptr<GeomAPI_XY> aMidDirXY = aStartDir->xy()->added(aEndDir->xy());
374   if (aMidDirXY->dot(aMidDirXY) < tolerance * tolerance) {
375     // start and end directions are opposite, so middle direction will be orthogonal
376     aMidDirXY->setX(-aStartDir->y());
377     aMidDirXY->setY(aStartDir->x());
378   }
379   std::shared_ptr<GeomAPI_Dir2d> aMidDir(new GeomAPI_Dir2d(aMidDirXY));
380   if ((aStartDir->cross(aMidDir) > 0) ^ !theArcReversed)
381     aMidDir->reverse();
382
383   double aRadius = theCenter->distance(theStartPoint);
384   std::shared_ptr<GeomAPI_XY> aPassedPnt = theCenter->xy()->added( aMidDir->xy()->multiplied(aRadius) );
385   thePassedPoint->setValue(aPassedPnt->x(), aPassedPnt->y());
386   return true;
387 }
388
389 void SketchPlugin_Arc::updateDependentAttributes()
390 {
391   std::shared_ptr<GeomDataAPI_Point2D> aCenterAttr = std::dynamic_pointer_cast<
392       GeomDataAPI_Point2D>(data()->attribute(CENTER_ID()));
393   std::shared_ptr<GeomDataAPI_Point2D> aStartAttr = std::dynamic_pointer_cast<
394       GeomDataAPI_Point2D>(data()->attribute(START_ID()));
395   std::shared_ptr<GeomDataAPI_Point2D> anEndAttr = std::dynamic_pointer_cast<
396       GeomDataAPI_Point2D>(data()->attribute(END_ID()));
397   std::shared_ptr<GeomDataAPI_Point2D> aPassedPoint =
398       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(PASSED_POINT_ID()));
399   AttributeDoublePtr aRadiusAttr = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(
400       data()->attribute(RADIUS_ID()));
401   AttributeDoublePtr anAngleAttr = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(
402       data()->attribute(ANGLE_ID()));
403
404   if (!aPassedPoint)
405     return;
406
407   data()->blockSendAttributeUpdated(true);
408
409   bool isOk = calculatePassedPoint(aCenterAttr->pnt(), aStartAttr->pnt(), anEndAttr->pnt(),
410                        isReversed(), aPassedPoint);
411   if (isOk && aRadiusAttr && anAngleAttr) {
412     std::shared_ptr<GeomAPI_Circ2d> aCircle(
413         new GeomAPI_Circ2d(aStartAttr->pnt(), anEndAttr->pnt(), aPassedPoint->pnt()));
414     if (aCircle->implPtr<void*>())
415       calculateArcAngleRadius(aCircle, aStartAttr->pnt(), anEndAttr->pnt(), aPassedPoint->pnt(),
416                               anAngleAttr, aRadiusAttr);
417   }
418   data()->blockSendAttributeUpdated(false);
419 }
420
421
422 void SketchPlugin_Arc::attributeChanged(const std::string& theID)
423 {
424   std::shared_ptr<GeomDataAPI_Point2D> aCenterAttr = std::dynamic_pointer_cast<
425       GeomDataAPI_Point2D>(data()->attribute(CENTER_ID()));
426   std::shared_ptr<GeomDataAPI_Point2D> aStartAttr = std::dynamic_pointer_cast<
427       GeomDataAPI_Point2D>(data()->attribute(START_ID()));
428   std::shared_ptr<GeomDataAPI_Point2D> anEndAttr = std::dynamic_pointer_cast<
429       GeomDataAPI_Point2D>(data()->attribute(END_ID()));
430   // the second condition for unability to move external segments anywhere
431   if (theID == EXTERNAL_ID() || isFixed()) {
432     std::shared_ptr<GeomAPI_Shape> aSelection = data()->selection(EXTERNAL_ID())->value();
433     // update arguments due to the selection value
434     if (aSelection && !aSelection->isNull() && aSelection->isEdge()) {
435       std::shared_ptr<GeomAPI_Edge> anEdge( new GeomAPI_Edge(aSelection));
436       std::shared_ptr<GeomAPI_Circ> aCirc = anEdge->circle();
437       if (aCirc.get()) {
438         aStartAttr->setValue(sketch()->to2D(anEdge->firstPoint()));
439         anEndAttr->setValue(sketch()->to2D(anEdge->lastPoint()));
440         aCenterAttr->setValue(sketch()->to2D(aCirc->center()));
441       }
442     }
443     return;
444   }
445
446   AttributeDoublePtr aRadiusAttr = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(
447       data()->attribute(RADIUS_ID()));
448   AttributeDoublePtr anAngleAttr = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(
449       data()->attribute(ANGLE_ID()));
450
451   if (theID == RADIUS_ID()) {
452     if (!aStartAttr->isInitialized() || !anEndAttr->isInitialized())
453       return;
454     // move center and passed point
455     std::shared_ptr<GeomAPI_XY> aStartPnt = aStartAttr->pnt()->xy();
456     std::shared_ptr<GeomAPI_XY> aEndPnt = anEndAttr->pnt()->xy();
457     double aDist = aStartPnt->distance(aEndPnt);
458     if (fabs(aDist) < tolerance)
459       return;
460     std::shared_ptr<GeomAPI_Dir2d> aDir(new GeomAPI_Dir2d(aEndPnt->decreased(aStartPnt)));
461     std::shared_ptr<GeomAPI_Dir2d> aMidPerpDir(new GeomAPI_Dir2d(-aDir->y(), aDir->x()));
462     std::shared_ptr<GeomAPI_XY> aMidPnt = aStartPnt->added(aEndPnt)->multiplied(0.5);
463
464     double anAngle = anAngleAttr->value() * PI / 180.0;
465     adjustPeriod(anAngle);
466     if (anAngle > PI)
467       aMidPerpDir->reverse();
468
469     double aRadius = aRadiusAttr->value();
470     // The center is placed on a perpendicular bisector of a start-end points segment.
471     // If the radius is smaller that necessary, start and end points are moved too.
472     double aDist2 = aRadius * aRadius - aDist * aDist / 4.0;
473     aDist = aDist2 > 0.0 ? sqrt(aDist2) : 0.0;
474     // distance between middle point and start point (does not changed if the arc diameter is greater than start-end distance)
475     aDist2 = sqrt(aRadius * aRadius - aDist * aDist);
476
477     std::shared_ptr<GeomAPI_XY> aCenter = aMidPnt->added(aMidPerpDir->xy()->multiplied(aDist));
478     aStartPnt = aMidPnt->added(aDir->xy()->multiplied(-aDist2));
479     aEndPnt = aMidPnt->added(aDir->xy()->multiplied(aDist2));
480
481     data()->blockSendAttributeUpdated(true);
482     aCenterAttr->setValue(aCenter->x(), aCenter->y());
483     aStartAttr->setValue(aStartPnt->x(), aStartPnt->y());
484     anEndAttr->setValue(aEndPnt->x(), aEndPnt->y());
485     updateDependentAttributes();
486     data()->blockSendAttributeUpdated(false);
487     return;
488   }
489   if (theID == ANGLE_ID()) {
490     if (!aStartAttr->isInitialized() || !aCenterAttr->isInitialized())
491       return;
492     data()->blockSendAttributeUpdated(true);
493     // move end point and passed point
494     std::shared_ptr<GeomAPI_XY> aCenter = aCenterAttr->pnt()->xy();
495     double anAngle = anAngleAttr->value() * PI / 180.0;
496     double sinA = sin(anAngle);
497     double cosA = cos(anAngle);
498     std::shared_ptr<GeomAPI_XY> aStartDir = aStartAttr->pnt()->xy()->decreased(aCenter);
499     std::shared_ptr<GeomAPI_XY> aDir(new GeomAPI_XY(
500         aStartDir->x() * cosA - aStartDir->y() * sinA,
501         aStartDir->x() * sinA + aStartDir->y() * cosA));
502     anEndAttr->setValue(aCenter->x() + aDir->x(), aCenter->y() + aDir->y());
503
504     anAngle /= 2.0;
505     sinA = sin(anAngle);
506     cosA = cos(anAngle);
507     aDir = std::shared_ptr<GeomAPI_XY>(new GeomAPI_XY(
508         aStartDir->x() * cosA - aStartDir->y() * sinA,
509         aStartDir->x() * sinA + aStartDir->y() * cosA));
510     std::shared_ptr<GeomDataAPI_Point2D> aPassedPoint =
511         std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(PASSED_POINT_ID()));
512     aPassedPoint->setValue(aCenter->x() + aDir->x(), aCenter->y() + aDir->y());
513
514     std::shared_ptr<GeomAPI_Circ2d> aCircle(
515         new GeomAPI_Circ2d(aStartAttr->pnt(), anEndAttr->pnt(), aPassedPoint->pnt()));
516     calculateArcAngleRadius(aCircle, aStartAttr->pnt(), anEndAttr->pnt(), aPassedPoint->pnt(),
517                             anAngleAttr, aRadiusAttr);
518     data()->blockSendAttributeUpdated(false);
519     return;
520   }
521
522   if (theID == CENTER_ID()) {
523     if (isFeatureValid())
524       projectEndPoint();
525     return;
526   }
527
528   AttributeStringPtr aTypeAttr =
529       std::dynamic_pointer_cast<ModelAPI_AttributeString>(attribute(ARC_TYPE()));
530   if (!aTypeAttr)
531     return;
532   std::string anArcType = aTypeAttr->value();
533
534   // update the points in accordance to the changed point changes
535   if (anArcType == ARC_TYPE_CENTER_START_END()) {
536     if (!isFeatureValid())
537       return;
538     if (theID == END_ID() && isStable()) {
539       // The arc is under construction, so its end point projected
540       // on the circle formed by center and start points
541       projectEndPoint();
542     }
543     updateDependentAttributes();
544   }
545   else if (anArcType == ARC_TYPE_THREE_POINTS() &&
546           (theID == START_ID() || theID == END_ID() || theID == PASSED_POINT_ID())) {
547     data()->blockSendAttributeUpdated(true);
548
549     std::shared_ptr<GeomAPI_Pnt2d> aPoints[3];
550     int aNbInitialized = 0;
551     for (int i = 1; i <= 3; ++i) {
552       std::shared_ptr<GeomDataAPI_Point2D> aCurPnt =
553           std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(POINT_ID(i)));
554       if (aCurPnt->isInitialized())
555         aPoints[aNbInitialized++] = aCurPnt->pnt();
556     }
557
558     if (aNbInitialized == 1)
559       aCenterAttr->setValue(aPoints[0]->x(), aPoints[0]->y());
560     else if (aNbInitialized == 2) {
561       // calculate center point, which gives a quarter of circle for the given start and end points
562       std::shared_ptr<GeomAPI_Pnt2d> aStartPnt = aPoints[0];
563       std::shared_ptr<GeomAPI_Pnt2d> aEndPnt = aPoints[1];
564       std::shared_ptr<GeomAPI_XY> aDir = aEndPnt->xy()->decreased(aStartPnt->xy())->multiplied(0.5);
565       double x = aDir->x();
566       double y = aDir->y();
567       aDir->setX(x - y);
568       aDir->setY(y + x);
569       std::shared_ptr<GeomAPI_XY> aCenter = aStartPnt->xy()->added(aDir);
570       double aRadius = sqrt(aDir->dot(aDir));
571
572       aCenterAttr->setValue(aCenter->x(), aCenter->y());
573       aRadiusAttr->setValue(aRadius);
574       anAngleAttr->setValue(90.0);
575     }
576     else {
577       std::shared_ptr<GeomAPI_Circ2d> aCircle(
578           new GeomAPI_Circ2d(aPoints[0], aPoints[1], aPoints[2]));
579
580       std::shared_ptr<GeomAPI_Pnt2d> aCenter = aCircle->center();
581       if (aCenter) {
582         aCenterAttr->setValue(aCenter);
583         if (theID == START_ID() || theID == END_ID())
584           updateDependentAttributes();
585         else
586           calculateArcAngleRadius(aCircle, aPoints[0], aPoints[1], aPoints[2],
587                                   anAngleAttr, aRadiusAttr);
588       }
589     }
590
591     data()->blockSendAttributeUpdated(false);
592   }
593   else if (anArcType == ARC_TYPE_TANGENT() && (theID == TANGENT_POINT_ID() || theID == END_ID())) {
594     SketchPlugin_Sketch* aSketch = sketch();
595     AttributeRefAttrPtr aTangPtAttr = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
596         data()->attribute(TANGENT_POINT_ID()));
597
598     if (aTangPtAttr->isInitialized() && anEndAttr->isInitialized()) {
599       data()->blockSendAttributeUpdated(true);
600       // compute orthogonal direction
601       std::shared_ptr<GeomAPI_Dir2d> anOrthoDir;
602       std::shared_ptr<GeomDataAPI_Point2D> aTangentPoint =
603           std::dynamic_pointer_cast<GeomDataAPI_Point2D>(aTangPtAttr->attr());
604       std::shared_ptr<GeomAPI_Pnt2d> aTangPnt2d = aTangentPoint->pnt();
605       FeaturePtr aTangFeature = ModelAPI_Feature::feature(aTangentPoint->owner());
606       std::shared_ptr<GeomAPI_Edge> aTangEdge = std::dynamic_pointer_cast<GeomAPI_Edge>(
607           aTangFeature->lastResult()->shape());
608       if (aTangEdge->isLine()) {
609         std::shared_ptr<GeomAPI_Dir> aDir = aTangEdge->line()->direction();
610         std::shared_ptr<GeomAPI_Pnt> aPnt(new GeomAPI_Pnt(aDir->x(), aDir->y(), aDir->z()));
611         std::shared_ptr<GeomAPI_Pnt2d> aPnt2d = aSketch->to2D(aPnt);
612         anOrthoDir = std::shared_ptr<GeomAPI_Dir2d>(new GeomAPI_Dir2d(-aPnt2d->y(), aPnt2d->x()));
613       }
614       else if (aTangEdge->isArc()) {
615         std::shared_ptr<GeomAPI_Pnt> aCenter = aTangEdge->circle()->center();
616         std::shared_ptr<GeomAPI_Pnt2d> aCenter2d = aSketch->to2D(aCenter);
617         anOrthoDir = std::shared_ptr<GeomAPI_Dir2d>(
618             new GeomAPI_Dir2d(aTangPnt2d->xy()->decreased(aCenter2d->xy())));
619       }
620
621       // compute parameters of the middle perpendicular
622       std::shared_ptr<GeomAPI_XY> aEndPntCoord = anEndAttr->pnt()->xy();
623       std::shared_ptr<GeomAPI_XY> aTempDir = aEndPntCoord->decreased(aTangPnt2d->xy());
624       std::shared_ptr<GeomAPI_Dir2d> aMidDir(new GeomAPI_Dir2d(-aTempDir->y(), aTempDir->x()));
625       std::shared_ptr<GeomAPI_Pnt2d> aMidPnt(
626           new GeomAPI_Pnt2d(aEndPntCoord->added(aTangPnt2d->xy())->multiplied(0.5)));
627
628       // compute center of arc by calculating intersection of orthogonal line and middle perpendicular
629       std::shared_ptr<GeomAPI_Lin2d> anOrthoLine(new GeomAPI_Lin2d(aTangPnt2d, anOrthoDir));
630       std::shared_ptr<GeomAPI_Lin2d> aMiddleLine(new GeomAPI_Lin2d(aMidPnt, aMidDir));
631       std::shared_ptr<GeomAPI_Pnt2d> aCenter = anOrthoLine->intersect(aMiddleLine);
632       if (aCenter) {
633         aCenterAttr->setValue(aCenter);
634         aStartAttr->setValue(aTangPnt2d);
635         updateDependentAttributes();
636       }
637
638       data()->blockSendAttributeUpdated(false);
639       tangencyArcConstraints();
640     }
641   }
642 }
643
644 void SketchPlugin_Arc::setReversed(bool isReversed)
645 {
646   std::dynamic_pointer_cast<ModelAPI_AttributeBoolean>(attribute(INVERSED_ID()))->setValue(isReversed);
647   myParamBefore = 0.0;
648 }
649
650 bool SketchPlugin_Arc::isReversed()
651 {
652   return std::dynamic_pointer_cast<ModelAPI_AttributeBoolean>(attribute(INVERSED_ID()))->value();
653 }
654
655 void SketchPlugin_Arc::tangencyArcConstraints()
656 {
657   if (!lastResult())
658     return;
659
660   std::shared_ptr<GeomDataAPI_Point2D> aStartAttr =
661       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(START_ID()));
662   AttributeRefAttrPtr aTangPtAttr = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
663       attribute(TANGENT_POINT_ID()));
664   if (!aTangPtAttr->attr())
665     return;
666
667   FeaturePtr aFeature = ModelAPI_Feature::feature(aStartAttr->owner());
668   ObjectPtr aThisArc = aFeature->lastResult();
669   aFeature = ModelAPI_Feature::feature(aTangPtAttr->attr()->owner());
670   ObjectPtr aTangFeature = aFeature->lastResult();
671
672   // trying to find constraints to fix the tangency of the arc
673   std::set<FeaturePtr> aCoincidence;
674   std::set<FeaturePtr> aTangency;
675
676   AttributeRefAttrPtr aRefAttrA, aRefAttrB;
677   std::set<AttributePtr> aRefs = data()->refsToMe();
678   const std::set<AttributePtr>& aRefsToResult = lastResult()->data()->refsToMe();
679   aRefs.insert(aRefsToResult.begin(), aRefsToResult.end());
680   std::set<AttributePtr>::const_iterator aRefIt = aRefs.begin();
681   for (; aRefIt != aRefs.end(); ++aRefIt) {
682     FeaturePtr aConstrFeature = ModelAPI_Feature::feature((*aRefIt)->owner());
683     if (aConstrFeature->getKind() == SketchPlugin_ConstraintCoincidence::ID()) {
684       aRefAttrA = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
685           aConstrFeature->attribute(SketchPlugin_Constraint::ENTITY_A()));
686       aRefAttrB = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
687           aConstrFeature->attribute(SketchPlugin_Constraint::ENTITY_B()));
688       if ((aRefAttrA && aRefAttrA->attr() == aStartAttr) ||
689           (aRefAttrB && aRefAttrB->attr() == aStartAttr))
690         aCoincidence.insert(aConstrFeature);
691     }
692     else if (aConstrFeature->getKind() == SketchPlugin_ConstraintTangent::ID()) {
693       aRefAttrA = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
694           aConstrFeature->attribute(SketchPlugin_Constraint::ENTITY_A()));
695       aRefAttrB = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
696           aConstrFeature->attribute(SketchPlugin_Constraint::ENTITY_B()));
697       if ((aRefAttrA && aRefAttrA->object() == aThisArc) ||
698           (aRefAttrB && aRefAttrB->object() == aThisArc))
699         aTangency.insert(aConstrFeature);
700     }
701   }
702   // search applicable pair of constraints
703   bool isFound = false;
704   FeaturePtr aPrevCoincidence, aPrevTangency;
705   std::set<FeaturePtr>::const_iterator aCIt, aTIt;
706   for (aCIt = aCoincidence.begin(); aCIt != aCoincidence.end() && !isFound; ++aCIt) {
707     aRefAttrA = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
708         (*aCIt)->attribute(SketchPlugin_Constraint::ENTITY_A()));
709     aRefAttrB = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
710         (*aCIt)->attribute(SketchPlugin_Constraint::ENTITY_B()));
711     AttributePtr anOtherPoint =
712         aRefAttrA->attr() == aStartAttr ? aRefAttrB->attr() : aRefAttrA->attr();
713     for (aTIt = aTangency.begin(); aTIt != aTangency.end() && !isFound; ++aTIt) {
714       aRefAttrA = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
715           (*aTIt)->attribute(SketchPlugin_Constraint::ENTITY_A()));
716       aRefAttrB = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
717           (*aTIt)->attribute(SketchPlugin_Constraint::ENTITY_B()));
718       ObjectPtr anOtherObject = aRefAttrA->object() == aThisArc ?
719           aRefAttrB->object() : aRefAttrA->object();
720       if (anOtherPoint->owner() == anOtherObject) {
721         isFound = true;
722         aPrevCoincidence = *aCIt;
723         aPrevTangency = *aTIt;
724       }
725     }
726   }
727
728   if (isFound) {
729     // update previous constraints
730     aRefAttrA = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
731         aPrevCoincidence->attribute(SketchPlugin_Constraint::ENTITY_A()));
732     aRefAttrB = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
733         aPrevCoincidence->attribute(SketchPlugin_Constraint::ENTITY_B()));
734     if (aRefAttrA->attr() == aStartAttr)
735       aRefAttrB->setAttr(aTangPtAttr->attr());
736     else
737       aRefAttrA->setAttr(aTangPtAttr->attr());
738
739     aRefAttrA = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
740         aPrevTangency->attribute(SketchPlugin_Constraint::ENTITY_A()));
741     aRefAttrB = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
742         aPrevTangency->attribute(SketchPlugin_Constraint::ENTITY_B()));
743     if (aRefAttrA->object() == aThisArc)
744       aRefAttrB->setObject(aTangFeature);
745     else
746       aRefAttrA->setObject(aTangFeature);
747   } else {
748     // Wait all constraints being removed, then send update events
749     static Events_ID aDeleteEvent = Events_Loop::eventByName(EVENT_OBJECT_DELETED);
750     bool isDeleteFlushed = Events_Loop::loop()->isFlushed(aDeleteEvent);
751     if (isDeleteFlushed)
752       Events_Loop::loop()->setFlushed(aDeleteEvent, false);
753     // Remove all obtained constraints which use current arc, because
754     // there is no information which of them were used to build tangency arc.
755     DocumentPtr aDoc = sketch()->document();
756     std::set<FeaturePtr> aFeaturesToBeRemoved;
757     for (aCIt = aCoincidence.begin(); aCIt != aCoincidence.end(); ++aCIt)
758       aFeaturesToBeRemoved.insert(*aCIt);
759     for (aTIt = aTangency.begin(); aTIt != aTangency.end(); ++aTIt)
760       aFeaturesToBeRemoved.insert(*aTIt);
761     ModelAPI_Tools::removeFeaturesAndReferences(aFeaturesToBeRemoved);
762     // Send events to update the sub-features by the solver.
763     if (isDeleteFlushed)
764       Events_Loop::loop()->setFlushed(aDeleteEvent, true);
765     else
766       Events_Loop::loop()->flush(aDeleteEvent);
767
768     // Wait all constraints being created, then send update events
769     static Events_ID anUpdateEvent = Events_Loop::eventByName(EVENT_OBJECT_UPDATED);
770     bool isUpdateFlushed = Events_Loop::loop()->isFlushed(anUpdateEvent);
771     if (isUpdateFlushed)
772       Events_Loop::loop()->setFlushed(anUpdateEvent, false);
773
774     // Create new constraints
775     FeaturePtr aConstraint = sketch()->addFeature(SketchPlugin_ConstraintCoincidence::ID());
776     aRefAttrA = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
777         aConstraint->attribute(SketchPlugin_Constraint::ENTITY_A()));
778     aRefAttrB = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
779         aConstraint->attribute(SketchPlugin_Constraint::ENTITY_B()));
780     aRefAttrA->setAttr(aStartAttr);
781     aRefAttrB->setAttr(aTangPtAttr->attr());
782     aConstraint->execute();
783     ModelAPI_EventCreator::get()->sendUpdated(aConstraint, anUpdateEvent);
784
785     aConstraint = sketch()->addFeature(SketchPlugin_ConstraintTangent::ID());
786     aRefAttrA = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
787         aConstraint->attribute(SketchPlugin_Constraint::ENTITY_A()));
788     aRefAttrB = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
789         aConstraint->attribute(SketchPlugin_Constraint::ENTITY_B()));
790     aRefAttrA->setObject(aThisArc);
791     aRefAttrB->setObject(aTangFeature);
792     aConstraint->execute();
793     ModelAPI_EventCreator::get()->sendUpdated(aConstraint, anUpdateEvent);
794
795     // Send events to update the sub-features by the solver.
796     if(isUpdateFlushed)
797       Events_Loop::loop()->setFlushed(anUpdateEvent, true);
798   }
799 }
800
801 void SketchPlugin_Arc::projectEndPoint()
802 {
803   std::shared_ptr<GeomDataAPI_Point2D> aCenterAttr = std::dynamic_pointer_cast<
804       GeomDataAPI_Point2D>(data()->attribute(CENTER_ID()));
805   std::shared_ptr<GeomDataAPI_Point2D> aStartAttr = std::dynamic_pointer_cast<
806       GeomDataAPI_Point2D>(data()->attribute(START_ID()));
807   std::shared_ptr<GeomDataAPI_Point2D> anEndAttr = std::dynamic_pointer_cast<
808       GeomDataAPI_Point2D>(data()->attribute(END_ID()));
809
810   if (aCenterAttr->pnt()->distance(aStartAttr->pnt()) < tolerance)
811     return;
812   data()->blockSendAttributeUpdated(true);
813   // compute and change the arc end point
814   std::shared_ptr<GeomAPI_Circ2d> aCircleForArc(
815       new GeomAPI_Circ2d(aCenterAttr->pnt(), aStartAttr->pnt()));
816   std::shared_ptr<GeomAPI_Pnt2d> aProjection = aCircleForArc->project(anEndAttr->pnt());
817   if (aProjection && anEndAttr->pnt()->distance(aProjection) > tolerance)
818     anEndAttr->setValue(aProjection);
819   updateDependentAttributes();
820   data()->blockSendAttributeUpdated(false);
821 }