src/SketchPlugin/SketchPlugin_Fillet.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 "SketchPlugin_Fillet.h"
  22
  23 #include "SketchPlugin_Arc.h"
  24 #include "SketchPlugin_Line.h"
  25 #include "SketchPlugin_Point.h"
  26 #include "SketchPlugin_Sketch.h"
  27 #include "SketchPlugin_ConstraintDistance.h"
  28 #include "SketchPlugin_ConstraintEqual.h"
  29 #include "SketchPlugin_ConstraintCoincidence.h"
  30 #include "SketchPlugin_ConstraintLength.h"
  31 #include "SketchPlugin_ConstraintMiddle.h"
  32 #include "SketchPlugin_ConstraintTangent.h"
  33 #include "SketchPlugin_ConstraintRadius.h"
  34 #include "SketchPlugin_Tools.h"
  35
  36 #include <ModelAPI_AttributeDouble.h>
  37 #include <ModelAPI_AttributeInteger.h>
  38 #include <ModelAPI_AttributeRefAttr.h>
  39 #include <ModelAPI_Data.h>
  40 #include <ModelAPI_Events.h>
  41 #include <ModelAPI_Session.h>
  42 #include <ModelAPI_Tools.h>
  43 #include <ModelAPI_Validator.h>
  44
  45 #include <GeomAlgoAPI_Circ2dBuilder.h>
  46 #include <GeomAlgoAPI_EdgeBuilder.h>
  47
  48 #include <GeomAPI_Circ2d.h>
  49 #include <GeomAPI_Dir2d.h>
  50 #include <GeomAPI_Lin2d.h>
  51 #include <GeomAPI_Pnt2d.h>
  52 #include <GeomAPI_XY.h>
  53
  54 #include <GeomDataAPI_Point2D.h>
  55
  56 #include <Events_Loop.h>
  57
  58 #include <math.h>
  59
  60 const double tolerance = 1.e-7;
  61 const double paramTolerance = 1.e-4;
  62 const double PI = 3.141592653589793238463;
  63
  64 /// \brief Attract specified point on theNewArc to the attribute of theFeature
  65 static void recalculateAttributes(FeaturePtr theNewArc, const std::string& theNewArcAttribute,
  66   FeaturePtr theFeature, const std::string& theFeatureAttribute);
  67
  68
  69 /// \brief Calculate radius of a fillet.
  70 ///        It should not be greater than 1/3 of shortest edge length.
  71 static double calculateFilletRadius(FeaturePtr theFilletFeatures[2]);
  72
  73 /// \brief Calculates center of fillet arc and coordinates of tangency points
  74 static void calculateFilletCenter(FeaturePtr theFilletFeatures[2],
  75                                   double theFilletRadius,
  76                                   const std::shared_ptr<GeomAPI_Ax3>& theSketchPlane,
  77                                   std::shared_ptr<GeomAPI_XY>& theCenter,
  78                                   std::shared_ptr<GeomAPI_XY>& theTangentA,
  79                                   std::shared_ptr<GeomAPI_XY>& theTangentB);
  80
  81 static std::set<FeaturePtr> findFeaturesToRemove(const FeaturePtr theFeature,
  82                                                  const AttributePtr theAttribute);
  83
  84 SketchPlugin_Fillet::SketchPlugin_Fillet()
  85 : myFilletCreated(false)
  86 {
  87 }
  88
  89 void SketchPlugin_Fillet::initAttributes()
  90 {
  91   data()->addAttribute(FILLET_POINT_ID(), ModelAPI_AttributeRefAttr::typeId());
  92 }
  93
  94 void SketchPlugin_Fillet::execute()
  95 {
  96   // Wait all constraints being created, then send update events
  97   static Events_ID anUpdateEvent = Events_Loop::eventByName(EVENT_OBJECT_UPDATED);
  98   bool isUpdateFlushed = Events_Loop::loop()->isFlushed(anUpdateEvent);
  99   if (isUpdateFlushed)
 100     Events_Loop::loop()->setFlushed(anUpdateEvent, false);
 101
 102   // set flag here to avoid building Fillet presentation if "Redisplay" event appears
 103   myFilletCreated = true;
 104
 105   // create feature for fillet arc
 106   FeaturePtr aFilletArc = createFilletArc();
 107
 108   // collect features referred to the edges participating in fillet
 109   AttributePoint2DPtr aFilletPoints[2];
 110   int aFeatInd[2];
 111   int anAttrInd[2];
 112   std::set<FeaturePtr> aFeaturesToBeRemoved;
 113   for (int i = 0; i < 2; ++i) {
 114     bool isFirstIndex = (i == 0);
 115     aFeatInd[i] = myIsReversed == isFirstIndex ? 1 : 0;
 116     anAttrInd[i] = (myIsReversed == isFirstIndex ? 2 : 0) + (myIsNotInversed[aFeatInd[i]] ? 0 : 1);
 117     aFilletPoints[i] = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 118         myBaseFeatures[aFeatInd[i]]->attribute(myFeatAttributes[anAttrInd[i]]));
 119     std::set<FeaturePtr> aRemove =
 120         findFeaturesToRemove(myBaseFeatures[aFeatInd[i]], aFilletPoints[i]);
 121     aFeaturesToBeRemoved.insert(aRemove.begin(), aRemove.end());
 122   }
 123
 124   // keep "distance" constraints and remove all other references
 125   removeReferencesButKeepDistances(aFeaturesToBeRemoved, aFilletPoints);
 126
 127   // Update fillet edges.
 128   recalculateAttributes(aFilletArc, SketchPlugin_Arc::START_ID(),
 129                         myBaseFeatures[aFeatInd[0]], myFeatAttributes[anAttrInd[0]]);
 130   recalculateAttributes(aFilletArc, SketchPlugin_Arc::END_ID(),
 131                         myBaseFeatures[aFeatInd[1]], myFeatAttributes[anAttrInd[1]]);
 132
 133   FeaturePtr aConstraint;
 134
 135   // Create coincidence features.
 136   aConstraint = SketchPlugin_Tools::createConstraint(sketch(),
 137                     SketchPlugin_ConstraintCoincidence::ID(),
 138                     aFilletArc->attribute(SketchPlugin_Arc::START_ID()),
 139                     myBaseFeatures[aFeatInd[0]]->attribute(myFeatAttributes[anAttrInd[0]]));
 140   ModelAPI_EventCreator::get()->sendUpdated(aConstraint, anUpdateEvent);
 141   aConstraint = SketchPlugin_Tools::createConstraint(sketch(),
 142                     SketchPlugin_ConstraintCoincidence::ID(),
 143                     aFilletArc->attribute(SketchPlugin_Arc::END_ID()),
 144                     myBaseFeatures[aFeatInd[1]]->attribute(myFeatAttributes[anAttrInd[1]]));
 145   ModelAPI_EventCreator::get()->sendUpdated(aConstraint, anUpdateEvent);
 146
 147   // Create tangent features.
 148   for (int i = 0; i < 2; i++) {
 149     aConstraint = SketchPlugin_Tools::createConstraint(sketch(),
 150                       SketchPlugin_ConstraintTangent::ID(),
 151                       aFilletArc->lastResult(),
 152                       myBaseFeatures[i]->lastResult());
 153     aConstraint->execute();
 154     ModelAPI_EventCreator::get()->sendUpdated(aConstraint, anUpdateEvent);
 155   }
 156
 157   // Send events to update the sub-features by the solver.
 158   if (isUpdateFlushed)
 159     Events_Loop::loop()->setFlushed(anUpdateEvent, true);
 160 }
 161
 162 AISObjectPtr SketchPlugin_Fillet::getAISObject(AISObjectPtr thePrevious)
 163 {
 164   if(myFilletCreated) {
 165     return AISObjectPtr();
 166   }
 167
 168   SketchPlugin_Sketch* aSketch = sketch();
 169   if(!aSketch) {
 170     return AISObjectPtr();
 171   }
 172
 173   if (!calculateFilletParameters())
 174     return AISObjectPtr();
 175
 176   // Create arc for presentation.
 177   std::shared_ptr<GeomAPI_Pnt> aCenterPnt(aSketch->to3D(myCenterXY->x(), myCenterXY->y()));
 178   std::shared_ptr<GeomAPI_Pnt> aTangentPnt1(aSketch->to3D(myTangentXY1->x(),
 179                                                           myTangentXY1->y()));
 180   std::shared_ptr<GeomAPI_Pnt> aTangentPnt2(aSketch->to3D(myTangentXY2->x(),
 181                                                           myTangentXY2->y()));
 182   std::shared_ptr<GeomDataAPI_Dir> aNDir = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
 183     aSketch->data()->attribute(SketchPlugin_Sketch::NORM_ID()));
 184   std::shared_ptr<GeomAPI_Shape> anArcShape =
 185       GeomAlgoAPI_EdgeBuilder::lineCircleArc(aCenterPnt, aTangentPnt1, aTangentPnt2, aNDir->dir());
 186
 187   AISObjectPtr anAISObject = thePrevious;
 188   if(!anAISObject.get()) {
 189     anAISObject = AISObjectPtr(new GeomAPI_AISObject);
 190   }
 191   anAISObject->createShape(anArcShape);
 192   return anAISObject;
 193 }
 194
 195 bool SketchPlugin_Fillet::calculateFilletParameters()
 196 {
 197   // Get fillet point.
 198   AttributeRefAttrPtr aPointRefAttr = refattr(FILLET_POINT_ID());
 199   if (!aPointRefAttr->isInitialized() || aPointRefAttr->isObject())
 200     return false;
 201   std::shared_ptr<GeomDataAPI_Point2D> aFilletPoint2D =
 202     std::dynamic_pointer_cast<GeomDataAPI_Point2D>(aPointRefAttr->attr());
 203   if (!aFilletPoint2D.get())
 204     return false;
 205
 206   std::set<FeaturePtr> aFilletFeatures =
 207       SketchPlugin_Tools::findFeaturesCoincidentToPoint(aFilletPoint2D);
 208   if (aFilletFeatures.size() != 2) {
 209     setError("Error: Selected point does not have two suitable edges for fillet.");
 210     return false;
 211   }
 212
 213   std::set<FeaturePtr>::iterator aFIt = aFilletFeatures.begin();
 214   myBaseFeatures[0] = *aFIt;
 215   myBaseFeatures[1] = *(++aFIt);
 216
 217   std::shared_ptr<GeomAPI_Pnt2d> aFilletPnt2d = aFilletPoint2D->pnt();
 218   double aRadius = calculateFilletRadius(myBaseFeatures);
 219
 220   // Calculate arc attributes.
 221   static const int aNbFeatures = 2;
 222   // First pair of points relate to first feature, second pair -  to second.
 223   std::shared_ptr<GeomAPI_Pnt2d> aStartEndPnt[aNbFeatures * 2];
 224   for (int i = 0; i < aNbFeatures; i++) {
 225     std::string aStartAttr, aEndAttr;
 226     if (myBaseFeatures[i]->getKind() == SketchPlugin_Line::ID()) {
 227       aStartAttr = SketchPlugin_Line::START_ID();
 228       aEndAttr = SketchPlugin_Line::END_ID();
 229     } else if (myBaseFeatures[i]->getKind() == SketchPlugin_Arc::ID()) {
 230       aStartAttr = SketchPlugin_Arc::START_ID();
 231       aEndAttr = SketchPlugin_Arc::END_ID();
 232     } else { // Wrong argument.
 233       setError("Error: One of the points has wrong coincide feature");
 234       return false;
 235     }
 236     myFeatAttributes[2*i] = aStartAttr;
 237     aStartEndPnt[2*i] = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 238       myBaseFeatures[i]->attribute(aStartAttr))->pnt();
 239     myFeatAttributes[2*i+1] = aEndAttr;
 240     aStartEndPnt[2*i+1] = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 241       myBaseFeatures[i]->attribute(aEndAttr))->pnt();
 242   }
 243   for (int aFeatInd = 0; aFeatInd < aNbFeatures; aFeatInd++) {
 244     for (int j = 0; j < 2; j++) // loop on start-end of each feature
 245       if (aStartEndPnt[aFeatInd * aNbFeatures + j]->distance(aFilletPnt2d) < 1.e-10) {
 246         myIsNotInversed[aFeatInd] = (j==0);
 247         break;
 248       }
 249   }
 250
 251   std::shared_ptr<GeomAPI_Ax3> aSketchPlane = SketchPlugin_Sketch::plane(sketch());
 252   calculateFilletCenter(myBaseFeatures, aRadius, aSketchPlane,
 253                         myCenterXY, myTangentXY1, myTangentXY2);
 254
 255   // Tangent directions of the features in coincident point.
 256   std::shared_ptr<GeomAPI_Dir2d> aTangentDir[aNbFeatures];
 257   for (int i = 0; i < aNbFeatures; i++) {
 258     std::shared_ptr<GeomAPI_XY> aDir;
 259     if (myBaseFeatures[i]->getKind() == SketchPlugin_Line::ID()) {
 260       aDir = aStartEndPnt[2*i+1]->xy()->decreased(aStartEndPnt[2*i]->xy());
 261       if (!myIsNotInversed[i])
 262         aDir = aDir->multiplied(-1.0);
 263     } else if (myBaseFeatures[i]->getKind() == SketchPlugin_Arc::ID()) {
 264       std::shared_ptr<GeomAPI_Pnt2d> aCenterPoint =
 265         std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 266         myBaseFeatures[i]->attribute(SketchPlugin_Arc::CENTER_ID()))->pnt();
 267       aDir = myIsNotInversed[i] ? aStartEndPnt[2*i]->xy() : aStartEndPnt[2*i+1]->xy();
 268       aDir = aDir->decreased(aCenterPoint->xy());
 269
 270       double x = aDir->x();
 271       double y = aDir->y();
 272       aDir->setX(-y);
 273       aDir->setY(x);
 274       if (myIsNotInversed[i] ==
 275           std::dynamic_pointer_cast<SketchPlugin_Arc>(myBaseFeatures[i])->isReversed())
 276         aDir = aDir->multiplied(-1.0);
 277     }
 278     aTangentDir[i] = std::shared_ptr<GeomAPI_Dir2d>(new GeomAPI_Dir2d(aDir));
 279   }
 280
 281   // By default, the start point of fillet arc is connected to FeatureA,
 282   // and the end point - to FeatureB. But when the angle between TangentDirA and
 283   // TangentDirB greater 180 degree, the sequaence of features need to be reversed.
 284   double cosBA = aTangentDir[0]->cross(aTangentDir[1]); // cos(B-A),
 285   // where A and B - angles between corresponding tanget direction and the X axis
 286   myIsReversed = cosBA > 0.0;
 287
 288   if(myIsReversed) {
 289     std::shared_ptr<GeomAPI_XY> aTmp = myTangentXY1;
 290     myTangentXY1 = myTangentXY2;
 291     myTangentXY2 = aTmp;
 292   }
 293   return true;
 294 }
 295
 296 FeaturePtr SketchPlugin_Fillet::createFilletArc()
 297 {
 298   // Calculate Fillet parameters if does not yet
 299   if (!myBaseFeatures[0] || !myBaseFeatures[1])
 300     calculateFilletParameters();
 301
 302   // Create arc feature.
 303   FeaturePtr aFilletArc = sketch()->addFeature(SketchPlugin_Arc::ID());
 304
 305   // Set arc attributes.
 306   bool aWasBlocked = aFilletArc->data()->blockSendAttributeUpdated(true);
 307   std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 308       aFilletArc->attribute(SketchPlugin_Arc::CENTER_ID()))->setValue(myCenterXY->x(),
 309                                                                       myCenterXY->y());
 310   std::shared_ptr<GeomDataAPI_Point2D> aStartPoint =
 311       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 312           aFilletArc->attribute(SketchPlugin_Arc::START_ID()));
 313   std::shared_ptr<GeomDataAPI_Point2D> aEndPoint =
 314       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 315           aFilletArc->attribute(SketchPlugin_Arc::END_ID()));
 316   if(aStartPoint->isInitialized() && aEndPoint->isInitialized()
 317       && (aStartPoint->pnt()->xy()->distance(myTangentXY1) > tolerance
 318       || aEndPoint->pnt()->xy()->distance(myTangentXY2) > tolerance)) {
 319     std::dynamic_pointer_cast<SketchPlugin_Arc>(aFilletArc)->setReversed(false);
 320   }
 321   aStartPoint->setValue(myTangentXY1->x(), myTangentXY1->y());
 322   aEndPoint->setValue(myTangentXY2->x(), myTangentXY2->y());
 323   aFilletArc->data()->blockSendAttributeUpdated(aWasBlocked);
 324   aFilletArc->execute();
 325
 326   return aFilletArc;
 327 }
 328
 329 FeaturePtr SketchPlugin_Fillet::createFilletApex(const GeomPnt2dPtr& theCoordinates)
 330 {
 331   FeaturePtr anApex = sketch()->addFeature(SketchPlugin_Point::ID());
 332   AttributePoint2DPtr aCoord = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 333       anApex->attribute(SketchPlugin_Point::COORD_ID()));
 334   aCoord->setValue(theCoordinates);
 335
 336   // additional coincidence constraints
 337   static Events_ID anUpdateEvent = Events_Loop::eventByName(EVENT_OBJECT_UPDATED);
 338   FeaturePtr aConstraint;
 339   for (int i = 0; i < 2; i++) {
 340     aConstraint = SketchPlugin_Tools::createConstraint(sketch(),
 341                       SketchPlugin_ConstraintCoincidence::ID(),
 342                       aCoord,
 343                       myBaseFeatures[i]->lastResult());
 344     aConstraint->execute();
 345     ModelAPI_EventCreator::get()->sendUpdated(aConstraint, anUpdateEvent);
 346   }
 347
 348   return anApex;
 349 }
 350
 351 void SketchPlugin_Fillet::removeReferencesButKeepDistances(
 352     std::set<FeaturePtr>& theFeaturesToRemove,
 353     const AttributePoint2DPtr theFilletPoints[2])
 354 {
 355   FeaturePtr aFilletApex;
 356   struct Length {
 357     AttributePtr myPoints[2];
 358     std::string myValueText;
 359     double myValueDouble;
 360     GeomPnt2dPtr myFlyoutPoint;
 361     int myLocationType;
 362   };
 363   std::list<Length> aLengthToDistance;
 364
 365   std::set<FeaturePtr>::iterator aFeat = theFeaturesToRemove.begin();
 366   while (aFeat != theFeaturesToRemove.end()) {
 367     std::shared_ptr<SketchPlugin_ConstraintDistance> aDistance =
 368         std::dynamic_pointer_cast<SketchPlugin_ConstraintDistance>(*aFeat);
 369     if (aDistance) {
 370       if (!aFilletApex)
 371         aFilletApex = createFilletApex(theFilletPoints[0]->pnt());
 372       // update attributes of distance constraints
 373       bool isUpdated = false;
 374       for (int attrInd = 0; attrInd < CONSTRAINT_ATTR_SIZE && !isUpdated; ++attrInd) {
 375         AttributeRefAttrPtr aRefAttr =
 376             aDistance->refattr(SketchPlugin_Constraint::ATTRIBUTE(attrInd));
 377         if (aRefAttr && !aRefAttr->isObject() &&
 378            (aRefAttr->attr() == theFilletPoints[0] || aRefAttr->attr() == theFilletPoints[1])) {
 379           aRefAttr->setAttr(aFilletApex->attribute(SketchPlugin_Point::COORD_ID()));
 380           isUpdated = true;
 381         }
 382       }
 383       // avoid distance from removing if it is updated
 384       std::set<FeaturePtr>::iterator aKeepIt = aFeat++;
 385       if (isUpdated)
 386         theFeaturesToRemove.erase(aKeepIt);
 387
 388     } else {
 389       std::shared_ptr<SketchPlugin_ConstraintLength> aLength =
 390           std::dynamic_pointer_cast<SketchPlugin_ConstraintLength>(*aFeat);
 391       if (aLength) {
 392         if (!aFilletApex)
 393           aFilletApex = createFilletApex(theFilletPoints[0]->pnt());
 394         // remove Length, but create new distance constraint
 395         AttributeRefAttrPtr aRefAttr =
 396           aLength->refattr(SketchPlugin_Constraint::ENTITY_A());
 397         FeaturePtr aLine = ModelAPI_Feature::feature(aRefAttr->object());
 398         if (aLine) {
 399           aLengthToDistance.push_back(Length());
 400           Length& aNewLength = aLengthToDistance.back();
 401           // main attrbutes
 402           for (int i = 0; i < 2; ++i) {
 403             AttributePtr anAttr = aLine->attribute(
 404                 i == 0 ? SketchPlugin_Line::START_ID() : SketchPlugin_Line::END_ID());
 405             if (anAttr == theFilletPoints[0] || anAttr == theFilletPoints[1])
 406               aNewLength.myPoints[i] = aFilletApex->attribute(SketchPlugin_Point::COORD_ID());
 407             else
 408               aNewLength.myPoints[i] = anAttr;
 409           }
 410           // value
 411           AttributeDoublePtr aValue = aLength->real(SketchPlugin_Constraint::VALUE());
 412           aNewLength.myValueDouble = aValue->value();
 413           aNewLength.myValueText = aValue->text();
 414           // auxiliary attributes
 415           AttributePoint2DPtr aFlyoutAttr = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 416               aLength->attribute(SketchPlugin_ConstraintLength::FLYOUT_VALUE_PNT()));
 417           if (aFlyoutAttr && aFlyoutAttr->isInitialized())
 418             aNewLength.myFlyoutPoint = aFlyoutAttr->pnt();
 419           AttributeIntegerPtr aLocationAttr =
 420               aLength->integer(SketchPlugin_ConstraintLength::LOCATION_TYPE_ID());
 421           if (aLocationAttr && aLocationAttr->isInitialized())
 422             aNewLength.myLocationType = aLocationAttr->value();
 423           else
 424             aNewLength.myLocationType = -1;
 425         }
 426       }
 427
 428       ++aFeat;
 429     }
 430   }
 431
 432   // remove references
 433   ModelAPI_Tools::removeFeaturesAndReferences(theFeaturesToRemove);
 434   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_DELETED));
 435
 436   // restore Length constraints as point-point distances
 437   FeaturePtr aConstraint;
 438   std::list<Length>::iterator anIt = aLengthToDistance.begin();
 439   for (; anIt != aLengthToDistance.end(); ++anIt) {
 440     aConstraint = SketchPlugin_Tools::createConstraint(sketch(),
 441         SketchPlugin_ConstraintDistance::ID(), anIt->myPoints[0], anIt->myPoints[1]);
 442     // set value
 443     AttributeDoublePtr aValue = aConstraint->real(SketchPlugin_Constraint::VALUE());
 444     if (anIt->myValueText.empty())
 445       aValue->setValue(anIt->myValueDouble);
 446     else
 447       aValue->setText(anIt->myValueText);
 448     // set flyout point if exists
 449     if (anIt->myFlyoutPoint) {
 450       AttributePoint2DPtr aFlyoutAttr = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 451           aConstraint->attribute(SketchPlugin_ConstraintDistance::FLYOUT_VALUE_PNT()));
 452       aFlyoutAttr->setValue(anIt->myFlyoutPoint);
 453     }
 454     // set location type if initialized
 455     if (anIt->myLocationType >= 0) {
 456       AttributeIntegerPtr aLocationType =
 457           aConstraint->integer(SketchPlugin_ConstraintDistance::LOCATION_TYPE_ID());
 458       aLocationType->setValue(anIt->myLocationType);
 459     }
 460     aConstraint->execute();
 461     ModelAPI_EventCreator::get()->sendUpdated(aConstraint,
 462         Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
 463   }
 464 }
 465
 466 // =========   Auxiliary functions   =================
 467 void recalculateAttributes(FeaturePtr theNewArc,  const std::string& theNewArcAttribute,
 468                            FeaturePtr theFeature, const std::string& theFeatureAttribute)
 469 {
 470   std::shared_ptr<GeomAPI_Pnt2d> anArcPoint = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 471       theNewArc->attribute(theNewArcAttribute))->pnt();
 472   std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 473       theFeature->attribute(theFeatureAttribute))->setValue(anArcPoint->x(), anArcPoint->y());
 474 }
 475
 476 static std::shared_ptr<GeomAPI_Pnt2d> toPoint(const AttributePtr& theAttribute)
 477 {
 478   std::shared_ptr<GeomAPI_Pnt2d> aPoint;
 479   AttributePoint2DPtr aPointAttr = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(theAttribute);
 480   if (aPointAttr)
 481     aPoint = aPointAttr->pnt();
 482   return aPoint;
 483 }
 484
 485 static std::shared_ptr<GeomAPI_Lin2d> toLine(const FeaturePtr& theFeature)
 486 {
 487   std::shared_ptr<GeomAPI_Lin2d> aLine;
 488   if (theFeature->getKind() == SketchPlugin_Line::ID()) {
 489     std::shared_ptr<GeomAPI_Pnt2d> aStart =
 490         toPoint( theFeature->attribute(SketchPlugin_Line::START_ID()) );
 491     std::shared_ptr<GeomAPI_Pnt2d> aEnd =
 492         toPoint( theFeature->attribute(SketchPlugin_Line::END_ID()) );
 493     aLine = std::shared_ptr<GeomAPI_Lin2d>(new GeomAPI_Lin2d(aStart, aEnd));
 494   }
 495   return aLine;
 496 }
 497
 498 static std::shared_ptr<GeomAPI_Circ2d> toCircle(const FeaturePtr& theFeature)
 499 {
 500   std::shared_ptr<GeomAPI_Circ2d> aCircle;
 501   if (theFeature->getKind() == SketchPlugin_Arc::ID()) {
 502     std::shared_ptr<GeomAPI_Pnt2d> aCenter =
 503         toPoint( theFeature->attribute(SketchPlugin_Arc::CENTER_ID()) );
 504     std::shared_ptr<GeomAPI_Pnt2d> aStart =
 505         toPoint( theFeature->attribute(SketchPlugin_Arc::START_ID()) );
 506     aCircle = std::shared_ptr<GeomAPI_Circ2d>(new GeomAPI_Circ2d(aCenter, aStart));
 507   }
 508   return aCircle;
 509 }
 510
 511
 512 void calculateFilletCenter(FeaturePtr theFilletFeatures[2],
 513                            double theFilletRadius,
 514                            const std::shared_ptr<GeomAPI_Ax3>& theSketchPlane,
 515                            std::shared_ptr<GeomAPI_XY>& theCenter,
 516                            std::shared_ptr<GeomAPI_XY>& theTangentA,
 517                            std::shared_ptr<GeomAPI_XY>& theTangentB)
 518 {
 519   GeomShapePtr aShapeA = theFilletFeatures[0]->lastResult()->shape();
 520   GeomShapePtr aShapeB = theFilletFeatures[1]->lastResult()->shape();
 521
 522   GeomAlgoAPI_Circ2dBuilder aCircBuilder(theSketchPlane);
 523   aCircBuilder.addTangentCurve(aShapeA);
 524   aCircBuilder.addTangentCurve(aShapeB);
 525   aCircBuilder.setRadius(theFilletRadius);
 526
 527   std::shared_ptr<GeomAPI_Circ2d> aFilletCircle = aCircBuilder.circle();
 528   if (!aFilletCircle)
 529     return;
 530
 531   theCenter = aFilletCircle->center()->xy();
 532   // tangent points
 533   std::shared_ptr<GeomAPI_Pnt2d> aTgPoints[2];
 534   for (int i = 0; i < 2; ++i) {
 535     std::shared_ptr<GeomAPI_Circ2d> aCircle = toCircle(theFilletFeatures[i]);
 536     if (aCircle)
 537       aTgPoints[i] = aCircle->project(aFilletCircle->center());
 538     else {
 539       std::shared_ptr<GeomAPI_Lin2d> aLine = toLine(theFilletFeatures[i]);
 540       if (aLine)
 541         aTgPoints[i] = aLine->project(aFilletCircle->center());
 542     }
 543   }
 544   theTangentA = aTgPoints[0]->xy();
 545   theTangentB = aTgPoints[1]->xy();
 546 }
 547
 548 double calculateFilletRadius(FeaturePtr theFilletFeatures[2])
 549 {
 550   double aLengths[2] = { 0, 0 };
 551   for (int i = 0; i < 2; ++i) {
 552     GeomShapePtr aShape = theFilletFeatures[i]->lastResult()->shape();
 553     std::shared_ptr<GeomAPI_Edge> anEdge = std::dynamic_pointer_cast<GeomAPI_Edge>(aShape);
 554     if (anEdge)
 555       aLengths[i] = anEdge->length();
 556   }
 557   return std::min(aLengths[0], aLengths[1]) / 6.0;
 558 }
 559
 560 std::set<FeaturePtr> findFeaturesToRemove(const FeaturePtr theFeature,
 561                                           const AttributePtr theAttribute) {
 562   std::set<FeaturePtr> aFeaturesToBeRemoved;
 563   std::set<AttributePtr> aRefs = theFeature->data()->refsToMe();
 564   std::list<ResultPtr> aResults = theFeature->results();
 565   for(std::list<ResultPtr>::const_iterator aResultsIt = aResults.cbegin();
 566       aResultsIt != aResults.cend();
 567       ++aResultsIt) {
 568     ResultPtr aResult = *aResultsIt;
 569     std::set<AttributePtr> aResultRefs = aResult->data()->refsToMe();
 570     aRefs.insert(aResultRefs.begin(), aResultRefs.end());
 571   }
 572   for(std::set<AttributePtr>::const_iterator anIt = aRefs.cbegin();
 573     anIt != aRefs.cend();
 574     ++anIt) {
 575     std::shared_ptr<ModelAPI_Attribute> anAttr = (*anIt);
 576     FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(anAttr->owner());
 577     if(aFeature->getKind() == SketchPlugin_Fillet::ID()) {
 578       continue;
 579     }
 580     if(aFeature->getKind() == SketchPlugin_ConstraintLength::ID()
 581         || aFeature->getKind() == SketchPlugin_ConstraintEqual::ID()
 582         || aFeature->getKind() == SketchPlugin_ConstraintMiddle::ID()) {
 583       aFeaturesToBeRemoved.insert(aFeature);
 584     } else {
 585       std::list<AttributePtr> anAttrs =
 586           aFeature->data()->attributes(ModelAPI_AttributeRefAttr::typeId());
 587       for(std::list<AttributePtr>::const_iterator aRefAttrsIt = anAttrs.cbegin();
 588           aRefAttrsIt != anAttrs.cend();
 589           ++aRefAttrsIt) {
 590         AttributeRefAttrPtr anAttrRefAttr =
 591           std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(*aRefAttrsIt);
 592         if(anAttrRefAttr.get() && anAttrRefAttr->attr() == theAttribute) {
 593           aFeaturesToBeRemoved.insert(aFeature);
 594         }
 595       }
 596     }
 597   }
 598   return aFeaturesToBeRemoved;
 599 }