1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
3 // File: SketchPlugin_MacroArc.cpp
4 // Created: 26 Apr 2014
5 // Author: Artem ZHIDKOV
7 #include "SketchPlugin_MacroArc.h"
9 #include "SketchPlugin_Arc.h"
10 #include "SketchPlugin_ConstraintTangent.h"
11 #include "SketchPlugin_Sketch.h"
12 #include "SketchPlugin_Tools.h"
14 #include <ModelAPI_AttributeDouble.h>
15 #include <ModelAPI_AttributeRefAttr.h>
16 #include <ModelAPI_AttributeString.h>
17 #include <ModelAPI_Session.h>
18 #include <ModelAPI_Validator.h>
20 #include <GeomAPI_Circ.h>
21 #include <GeomAPI_Circ2d.h>
22 #include <GeomAPI_Curve.h>
23 #include <GeomAPI_Dir2d.h>
24 #include <GeomAPI_Edge.h>
25 #include <GeomAPI_Lin.h>
26 #include <GeomAPI_Lin2d.h>
27 #include <GeomAPI_Pnt2d.h>
28 #include <GeomAPI_Vertex.h>
29 #include <GeomAPI_XY.h>
31 #include <GeomDataAPI_Point2D.h>
32 #include <GeomDataAPI_Dir.h>
33 #include <GeomAlgoAPI_PointBuilder.h>
34 #include <GeomAlgoAPI_EdgeBuilder.h>
35 #include <GeomAlgoAPI_CompoundBuilder.h>
40 const double tolerance = 1e-7;
41 const double paramTolerance = 1.e-4;
42 const double PI = 3.141592653589793238463;
44 static void projectPointOnCircle(AttributePoint2DPtr& thePoint, const GeomAPI_Circ2d& theCircle)
46 std::shared_ptr<GeomAPI_Pnt2d> aProjection = theCircle.project(thePoint->pnt());
48 thePoint->setValue(aProjection);
52 SketchPlugin_MacroArc::SketchPlugin_MacroArc()
53 : SketchPlugin_SketchEntity(),
58 void SketchPlugin_MacroArc::initAttributes()
60 data()->addAttribute(ARC_TYPE(), ModelAPI_AttributeString::typeId());
62 data()->addAttribute(CENTER_POINT_ID(), GeomDataAPI_Point2D::typeId());
63 data()->addAttribute(START_POINT_1_ID(), GeomDataAPI_Point2D::typeId());
64 data()->addAttribute(END_POINT_1_ID(), GeomDataAPI_Point2D::typeId());
66 data()->addAttribute(START_POINT_2_ID(), GeomDataAPI_Point2D::typeId());
67 data()->addAttribute(END_POINT_2_ID(), GeomDataAPI_Point2D::typeId());
68 data()->addAttribute(PASSED_POINT_ID(), GeomDataAPI_Point2D::typeId());
70 data()->addAttribute(TANGENT_POINT_ID(), ModelAPI_AttributeRefAttr::typeId());
71 data()->addAttribute(END_POINT_3_ID(), GeomDataAPI_Point2D::typeId());
73 data()->addAttribute(REVERSED_ID(), ModelAPI_AttributeBoolean::typeId());
75 data()->addAttribute(RADIUS_ID(), ModelAPI_AttributeDouble::typeId());
76 data()->addAttribute(ANGLE_ID(), ModelAPI_AttributeDouble::typeId());
78 data()->addAttribute(AUXILIARY_ID(), ModelAPI_AttributeBoolean::typeId());
80 data()->addAttribute(CENTER_POINT_REF_ID(), ModelAPI_AttributeRefAttr::typeId());
81 data()->addAttribute(START_POINT_REF_ID(), ModelAPI_AttributeRefAttr::typeId());
82 data()->addAttribute(END_POINT_REF_ID(), ModelAPI_AttributeRefAttr::typeId());
83 data()->addAttribute(PASSED_POINT_REF_ID(), ModelAPI_AttributeRefAttr::typeId());
85 boolean(REVERSED_ID())->setValue(false);
87 ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), CENTER_POINT_REF_ID());
88 ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), START_POINT_REF_ID());
89 ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), END_POINT_REF_ID());
90 ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), PASSED_POINT_REF_ID());
93 void SketchPlugin_MacroArc::attributeChanged(const std::string& theID)
95 std::string anArcType = string(ARC_TYPE())->value();
97 // If arc type switched reset according attributes.
98 if(theID == ARC_TYPE()) {
99 SketchPlugin_Tools::resetAttribute(this, CENTER_POINT_ID());
100 SketchPlugin_Tools::resetAttribute(this, CENTER_POINT_REF_ID());
101 SketchPlugin_Tools::resetAttribute(this, START_POINT_1_ID());
102 SketchPlugin_Tools::resetAttribute(this, START_POINT_REF_ID());
103 SketchPlugin_Tools::resetAttribute(this, END_POINT_1_ID());
104 SketchPlugin_Tools::resetAttribute(this, END_POINT_REF_ID());
105 SketchPlugin_Tools::resetAttribute(this, START_POINT_2_ID());
106 SketchPlugin_Tools::resetAttribute(this, END_POINT_2_ID());
107 SketchPlugin_Tools::resetAttribute(this, PASSED_POINT_ID());
108 SketchPlugin_Tools::resetAttribute(this, PASSED_POINT_REF_ID());
109 SketchPlugin_Tools::resetAttribute(this, TANGENT_POINT_ID());
110 SketchPlugin_Tools::resetAttribute(this, END_POINT_3_ID());
111 SketchPlugin_Tools::resetAttribute(this, REVERSED_ID());
112 SketchPlugin_Tools::resetAttribute(this, RADIUS_ID());
113 SketchPlugin_Tools::resetAttribute(this, ANGLE_ID());
118 boolean(REVERSED_ID())->setValue(false);
120 } else if(anArcType == ARC_TYPE_BY_CENTER_AND_POINTS())
121 fillByCenterAndTwoPassed();
122 else if(anArcType == ARC_TYPE_BY_THREE_POINTS())
123 fillByThreePassedPoints();
124 else if(anArcType == ARC_TYPE_BY_TANGENT_EDGE())
129 if(myCenter.get() && myStart.get()) {
130 aRadius = myCenter->distance(myStart);
132 if(myStart->isEqual(myEnd)) {
135 GeomAPI_Circ2d aCircleForArc(myCenter, myStart);
136 double aStartParam, anEndParam;
137 aCircleForArc.parameter(myStart, paramTolerance, aStartParam);
138 aCircleForArc.parameter(myEnd, paramTolerance, anEndParam);
139 anAngle = (anEndParam - aStartParam) / PI * 180.0;
140 if(boolean(REVERSED_ID())->value()) anAngle = 360.0 - anAngle;
145 bool aWasBlocked = data()->blockSendAttributeUpdated(true);
146 real(RADIUS_ID())->setValue(aRadius);
147 real(ANGLE_ID())->setValue(anAngle);
148 data()->blockSendAttributeUpdated(aWasBlocked, false);
151 GeomShapePtr SketchPlugin_MacroArc::getArcShape()
153 if(!myStart.get() || !myEnd.get() || !myCenter.get()) {
154 return GeomShapePtr();
157 SketchPlugin_Sketch* aSketch = sketch();
159 return GeomShapePtr();
162 std::shared_ptr<GeomAPI_Pnt> aCenter(aSketch->to3D(myCenter->x(), myCenter->y()));
163 std::shared_ptr<GeomAPI_Pnt> aStart(aSketch->to3D(myStart->x(), myStart->y()));
164 std::shared_ptr<GeomAPI_Pnt> anEnd(aSketch->to3D(myEnd->x(), myEnd->y()));
165 std::shared_ptr<GeomDataAPI_Dir> aNDir =
166 std::dynamic_pointer_cast<GeomDataAPI_Dir>(aSketch->attribute(SketchPlugin_Sketch::NORM_ID()));
167 std::shared_ptr<GeomAPI_Dir> aNormal(new GeomAPI_Dir(aNDir->x(), aNDir->y(), aNDir->z()));
169 GeomShapePtr anArcShape = boolean(REVERSED_ID())->value() ?
170 GeomAlgoAPI_EdgeBuilder::lineCircleArc(aCenter, anEnd, aStart, aNormal)
171 : GeomAlgoAPI_EdgeBuilder::lineCircleArc(aCenter, aStart, anEnd, aNormal);
176 AISObjectPtr SketchPlugin_MacroArc::getAISObject(AISObjectPtr thePrevious)
178 if(!myStart.get() || !myEnd.get() || !myCenter.get()) {
179 return AISObjectPtr();
182 SketchPlugin_Sketch* aSketch = sketch();
184 return AISObjectPtr();
187 GeomShapePtr anArcShape = getArcShape();
188 std::shared_ptr<GeomAPI_Pnt> aCenter = aSketch->to3D(myCenter->x(), myCenter->y());;
189 GeomShapePtr aCenterPointShape = GeomAlgoAPI_PointBuilder::vertex(aCenter);
191 if(!anArcShape.get() || !aCenterPointShape.get()) {
192 return AISObjectPtr();
195 std::list<std::shared_ptr<GeomAPI_Shape> > aShapes;
196 aShapes.push_back(anArcShape);
197 aShapes.push_back(aCenterPointShape);
199 std::shared_ptr<GeomAPI_Shape> aCompound = GeomAlgoAPI_CompoundBuilder::compound(aShapes);
200 AISObjectPtr anAIS = thePrevious;
202 anAIS.reset(new GeomAPI_AISObject());
204 anAIS->createShape(aCompound);
208 void SketchPlugin_MacroArc::execute()
210 FeaturePtr anArcFeature = createArcFeature();
216 // Create constraints.
217 std::string anArcType = string(ARC_TYPE())->value();
218 if(anArcType == ARC_TYPE_BY_CENTER_AND_POINTS()) {
219 SketchPlugin_Tools::createConstraint(this,
220 CENTER_POINT_REF_ID(),
221 anArcFeature->attribute(SketchPlugin_Arc::CENTER_ID()),
224 SketchPlugin_Tools::createConstraint(this,
225 START_POINT_REF_ID(),
226 anArcFeature->attribute(SketchPlugin_Arc::START_ID()),
229 SketchPlugin_Tools::createConstraint(this,
231 anArcFeature->attribute(SketchPlugin_Arc::END_ID()),
234 } else if(anArcType == ARC_TYPE_BY_THREE_POINTS()) {
235 SketchPlugin_Tools::createConstraint(this,
236 START_POINT_REF_ID(),
237 anArcFeature->attribute(SketchPlugin_Arc::START_ID()),
240 SketchPlugin_Tools::createConstraint(this,
242 anArcFeature->attribute(SketchPlugin_Arc::END_ID()),
245 SketchPlugin_Tools::createConstraint(this,
246 PASSED_POINT_REF_ID(),
248 anArcFeature->lastResult(),
250 } else if(anArcType == ARC_TYPE_BY_TANGENT_EDGE()) {
251 // constraints for tangent arc
252 SketchPlugin_Tools::createConstraint(this,
254 anArcFeature->attribute(SketchPlugin_Arc::START_ID()),
257 FeaturePtr aTangent = sketch()->addFeature(SketchPlugin_ConstraintTangent::ID());
258 AttributeRefAttrPtr aRefAttrA = aTangent->refattr(SketchPlugin_Constraint::ENTITY_A());
259 AttributeRefAttrPtr aTgPntRefAttr = refattr(TANGENT_POINT_ID());
260 FeaturePtr aTgFeature = ModelAPI_Feature::feature(aTgPntRefAttr->attr()->owner());
261 aRefAttrA->setObject(aTgFeature->lastResult());
262 AttributeRefAttrPtr aRefAttrB = aTangent->refattr(SketchPlugin_Constraint::ENTITY_B());
263 aRefAttrB->setObject(anArcFeature->lastResult());
264 // constraint for end point
265 SketchPlugin_Tools::createConstraint(this,
267 anArcFeature->attribute(SketchPlugin_Arc::END_ID()),
273 FeaturePtr SketchPlugin_MacroArc::createArcFeature()
275 FeaturePtr anArcFeature = sketch()->addFeature(SketchPlugin_Arc::ID());
276 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
277 anArcFeature->attribute(SketchPlugin_Arc::CENTER_ID()))->setValue(myCenter);
278 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
279 anArcFeature->attribute(SketchPlugin_Arc::START_ID()))->setValue(myStart);
280 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
281 anArcFeature->attribute(SketchPlugin_Arc::END_ID()))->setValue(myEnd);
282 anArcFeature->boolean(SketchPlugin_Arc::REVERSED_ID())
283 ->setValue(boolean(REVERSED_ID())->value());
284 anArcFeature->boolean(SketchPlugin_Arc::AUXILIARY_ID())
285 ->setValue(boolean(AUXILIARY_ID())->value());
286 anArcFeature->execute();
291 void SketchPlugin_MacroArc::fillByCenterAndTwoPassed()
293 AttributePoint2DPtr aCenterPointAttr =
294 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(CENTER_POINT_ID()));
295 if (!aCenterPointAttr->isInitialized())
298 AttributePoint2DPtr aStartPointAttr =
299 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(START_POINT_1_ID()));
300 if (!aStartPointAttr->isInitialized())
303 myCenter = aCenterPointAttr->pnt();
304 myStart = aStartPointAttr->pnt();
307 AttributePoint2DPtr anEndPointAttr =
308 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(END_POINT_1_ID()));
309 if (!anEndPointAttr->isInitialized())
312 GeomAPI_Circ2d aCircleForArc(myCenter, myStart);
314 // End point should be a projection on circle.
315 bool aWasBlocked = data()->blockSendAttributeUpdated(true);
316 projectPointOnCircle(anEndPointAttr, aCircleForArc);
317 data()->blockSendAttributeUpdated(aWasBlocked, false);
318 myEnd = anEndPointAttr->pnt();
320 // update the REVERSED flag
321 recalculateReversedFlagByEnd(aCircleForArc);
324 void SketchPlugin_MacroArc::recalculateReversedFlagByEnd(const GeomAPI_Circ2d& theCurrentCircular)
326 double aParameterNew = 0.0;
327 if(theCurrentCircular.parameter(myEnd, paramTolerance, aParameterNew)) {
328 if(myParamBefore <= PI / 2.0 && aParameterNew >= PI * 1.5) {
329 boolean(REVERSED_ID())->setValue(true);
330 } else if(myParamBefore >= PI * 1.5 && aParameterNew <= PI / 2.0) {
331 boolean(REVERSED_ID())->setValue(false);
334 myParamBefore = aParameterNew;
337 void SketchPlugin_MacroArc::fillByThreePassedPoints()
339 AttributePoint2DPtr aStartPointAttr =
340 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(START_POINT_2_ID()));
341 if (!aStartPointAttr->isInitialized())
344 AttributePoint2DPtr anEndPointAttr =
345 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(END_POINT_2_ID()));
346 if (!anEndPointAttr->isInitialized())
349 myStart = aStartPointAttr->pnt();
350 myEnd = anEndPointAttr->pnt();
352 AttributePoint2DPtr aPassedPointAttr =
353 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(PASSED_POINT_ID()));
354 if (aPassedPointAttr->isInitialized()) {
355 std::shared_ptr<GeomAPI_Pnt2d> aPassedPnt;
356 std::shared_ptr<GeomAPI_Shape> aTangentCurve;
357 SketchPlugin_Tools::convertRefAttrToPointOrTangentCurve(
358 refattr(PASSED_POINT_REF_ID()), aPassedPointAttr, aTangentCurve, aPassedPnt);
360 std::shared_ptr<GeomAPI_Interface> aPassed;
362 aPassed = aTangentCurve;
364 aPassed = aPassedPnt;
366 std::shared_ptr<GeomAPI_Ax3> anAxis = SketchPlugin_Sketch::plane(sketch());
367 GeomAPI_Circ2d aCircle(myStart, myEnd, aPassed, anAxis);
368 myCenter = aCircle.center();
370 recalculateReversedFlagByPassed(aCircle);
372 myCenter.reset(new GeomAPI_Pnt2d(myStart->xy()->added(myEnd->xy())->multiplied(0.5)));
375 void SketchPlugin_MacroArc::recalculateReversedFlagByPassed(
376 const GeomAPI_Circ2d& theCurrentCircular)
378 AttributePoint2DPtr aPassedAttr =
379 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(PASSED_POINT_ID()));
380 std::shared_ptr<GeomAPI_Pnt2d> aPassed = theCurrentCircular.project(aPassedAttr->pnt());
382 double aEndParam, aPassedParam;
383 theCurrentCircular.parameter(myEnd, paramTolerance, aEndParam);
384 theCurrentCircular.parameter(aPassed, paramTolerance, aPassedParam);
386 if(aPassedParam > aEndParam)
387 boolean(REVERSED_ID())->setValue(true);
389 boolean(REVERSED_ID())->setValue(false);
391 myParamBefore = aEndParam;
394 void SketchPlugin_MacroArc::fillByTangentEdge()
396 AttributeRefAttrPtr aTangentAttr = refattr(TANGENT_POINT_ID());
397 if (!aTangentAttr->isInitialized())
400 AttributePoint2DPtr aTangentPointAttr =
401 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(aTangentAttr->attr());
402 if (!aTangentPointAttr->isInitialized())
405 AttributePoint2DPtr anEndPointAttr =
406 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(END_POINT_3_ID()));
407 if (!anEndPointAttr->isInitialized())
410 myStart = aTangentPointAttr->pnt();
411 myEnd = anEndPointAttr->pnt();
412 if (myStart->isEqual(myEnd))
415 // obtain a shape the tangent point belongs to
416 FeaturePtr aTangentFeature = ModelAPI_Feature::feature(aTangentPointAttr->owner());
417 std::shared_ptr<GeomAPI_Shape> aTangentShape = aTangentFeature->lastResult()->shape();
419 std::shared_ptr<GeomAPI_Ax3> anAxis = SketchPlugin_Sketch::plane(sketch());
420 GeomAPI_Circ2d aCircle(myStart, myEnd, aTangentShape, anAxis);
421 myCenter = aCircle.center();
423 // rebuild circle to set start point equal to zero parameter
424 aCircle = GeomAPI_Circ2d(myCenter, myStart);
425 recalculateReversedFlagByEnd(aCircle);