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