src/SketchPlugin/SketchPlugin_MacroArc.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
   2
   3 // File:        SketchPlugin_MacroArc.cpp
   4 // Created:     26 Apr 2014
   5 // Author:      Artem ZHIDKOV
   6
   7 #include "SketchPlugin_MacroArc.h"
   8
   9 #include "SketchPlugin_Arc.h"
  10 #include "SketchPlugin_Sketch.h"
  11 #include "SketchPlugin_Tools.h"
  12
  13 #include <ModelAPI_AttributeDouble.h>
  14 #include <ModelAPI_AttributeRefAttr.h>
  15 #include <ModelAPI_AttributeString.h>
  16 #include <ModelAPI_Session.h>
  17 #include <ModelAPI_Validator.h>
  18
  19 #include <GeomAPI_Circ.h>
  20 #include <GeomAPI_Circ2d.h>
  21 #include <GeomAPI_Curve.h>
  22 #include <GeomAPI_Dir2d.h>
  23 #include <GeomAPI_Edge.h>
  24 #include <GeomAPI_Lin.h>
  25 #include <GeomAPI_Lin2d.h>
  26 #include <GeomAPI_Pnt2d.h>
  27 #include <GeomAPI_Vertex.h>
  28 #include <GeomAPI_XY.h>
  29
  30 #include <GeomDataAPI_Point2D.h>
  31 #include <GeomDataAPI_Dir.h>
  32 #include <GeomAlgoAPI_PointBuilder.h>
  33 #include <GeomAlgoAPI_EdgeBuilder.h>
  34 #include <GeomAlgoAPI_CompoundBuilder.h>
  35
  36 // for sqrt on Linux
  37 #include <math.h>
  38
  39 const double tolerance = 1e-7;
  40 const double paramTolerance = 1.e-4;
  41 const double PI = 3.141592653589793238463;
  42
  43 static void projectPointOnCircle(AttributePoint2DPtr& thePoint, const GeomAPI_Circ2d& theCircle)
  44 {
  45   std::shared_ptr<GeomAPI_Pnt2d> aProjection = theCircle.project(thePoint->pnt());
  46   if(aProjection.get())
  47     thePoint->setValue(aProjection);
  48 }
  49
  50
  51 SketchPlugin_MacroArc::SketchPlugin_MacroArc()
  52 : SketchPlugin_SketchEntity(),
  53   myParamBefore(0.0)
  54 {
  55 }
  56
  57 void SketchPlugin_MacroArc::initAttributes()
  58 {
  59   data()->addAttribute(ARC_TYPE(), ModelAPI_AttributeString::typeId());
  60
  61   data()->addAttribute(CENTER_POINT_ID(), GeomDataAPI_Point2D::typeId());
  62   data()->addAttribute(START_POINT_ID(), GeomDataAPI_Point2D::typeId());
  63   data()->addAttribute(END_POINT_ID(), GeomDataAPI_Point2D::typeId());
  64
  65   data()->addAttribute(PASSED_POINT_ID(), GeomDataAPI_Point2D::typeId());
  66
  67   data()->addAttribute(TANGENT_POINT_ID(), ModelAPI_AttributeRefAttr::typeId());
  68
  69   data()->addAttribute(REVERSED_ID(), ModelAPI_AttributeBoolean::typeId());
  70
  71   data()->addAttribute(RADIUS_ID(), ModelAPI_AttributeDouble::typeId());
  72   data()->addAttribute(ANGLE_ID(), ModelAPI_AttributeDouble::typeId());
  73
  74   data()->addAttribute(AUXILIARY_ID(), ModelAPI_AttributeBoolean::typeId());
  75
  76   data()->addAttribute(CENTER_POINT_REF_ID(), ModelAPI_AttributeRefAttr::typeId());
  77   data()->addAttribute(START_POINT_REF_ID(), ModelAPI_AttributeRefAttr::typeId());
  78   data()->addAttribute(END_POINT_REF_ID(), ModelAPI_AttributeRefAttr::typeId());
  79   data()->addAttribute(PASSED_POINT_REF_ID(), ModelAPI_AttributeRefAttr::typeId());
  80
  81   boolean(REVERSED_ID())->setValue(false);
  82
  83   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), CENTER_POINT_REF_ID());
  84   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), START_POINT_REF_ID());
  85   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), END_POINT_REF_ID());
  86   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), PASSED_POINT_REF_ID());
  87 }
  88
  89 void SketchPlugin_MacroArc::attributeChanged(const std::string& theID)
  90 {
  91   std::string anArcType = string(ARC_TYPE())->value();
  92
  93   // If arc type switched reset according attributes.
  94   if(theID == ARC_TYPE()) {
  95     SketchPlugin_Tools::resetAttribute(this, CENTER_POINT_ID());
  96     SketchPlugin_Tools::resetAttribute(this, CENTER_POINT_REF_ID());
  97     SketchPlugin_Tools::resetAttribute(this, START_POINT_ID());
  98     SketchPlugin_Tools::resetAttribute(this, START_POINT_REF_ID());
  99     SketchPlugin_Tools::resetAttribute(this, END_POINT_ID());
 100     SketchPlugin_Tools::resetAttribute(this, END_POINT_REF_ID());
 101     SketchPlugin_Tools::resetAttribute(this, PASSED_POINT_ID());
 102     SketchPlugin_Tools::resetAttribute(this, PASSED_POINT_REF_ID());
 103     SketchPlugin_Tools::resetAttribute(this, TANGENT_POINT_ID());
 104     SketchPlugin_Tools::resetAttribute(this, REVERSED_ID());
 105     SketchPlugin_Tools::resetAttribute(this, RADIUS_ID());
 106     SketchPlugin_Tools::resetAttribute(this, ANGLE_ID());
 107
 108     myCenter.reset();
 109     myStart.reset();
 110     myEnd.reset();
 111     boolean(REVERSED_ID())->setValue(false);
 112     myParamBefore = 0.0;
 113   } else if(anArcType == ARC_TYPE_BY_CENTER_AND_POINTS())
 114     fillByCenterAndTwoPassed();
 115   else if(anArcType == ARC_TYPE_BY_THREE_POINTS())
 116     fillByThreePassedPoints();
 117   else if(anArcType == ARC_TYPE_BY_TANGENT_EDGE())
 118     fillByTangentEdge();
 119
 120   double aRadius = 0;
 121   double anAngle = 0;
 122   if(myCenter.get() && myStart.get()) {
 123     aRadius = myCenter->distance(myStart);
 124     if(myEnd.get()) {
 125       if(myStart->isEqual(myEnd)) {
 126         anAngle = 360;
 127       } else {
 128         GeomAPI_Circ2d aCircleForArc(myCenter, myStart);
 129         double aStartParam, anEndParam;
 130         aCircleForArc.parameter(myStart, paramTolerance, aStartParam);
 131         aCircleForArc.parameter(myEnd, paramTolerance, anEndParam);
 132         anAngle = (anEndParam - aStartParam) / PI * 180.0;
 133         if(boolean(REVERSED_ID())->value()) anAngle = 360.0 - anAngle;
 134       }
 135     }
 136   }
 137
 138   bool aWasBlocked = data()->blockSendAttributeUpdated(true);
 139   real(RADIUS_ID())->setValue(aRadius);
 140   real(ANGLE_ID())->setValue(anAngle);
 141   data()->blockSendAttributeUpdated(aWasBlocked, false);
 142 }
 143
 144 AISObjectPtr SketchPlugin_MacroArc::getAISObject(AISObjectPtr thePrevious)
 145 {
 146   if(!myStart.get() || !myEnd.get() || !myCenter.get()) {
 147     return AISObjectPtr();
 148   }
 149
 150   SketchPlugin_Sketch* aSketch = sketch();
 151   if(!aSketch) {
 152     return AISObjectPtr();
 153   }
 154
 155   std::shared_ptr<GeomAPI_Pnt> aStart = aSketch->to3D(myStart->x(), myStart->y());
 156   std::shared_ptr<GeomAPI_Pnt> anEnd = aSketch->to3D(myEnd->x(), myEnd->y());
 157   std::shared_ptr<GeomAPI_Pnt> aCenter = aSketch->to3D(myCenter->x(), myCenter->y());;
 158
 159   std::shared_ptr<GeomDataAPI_Dir> aNDir =
 160       std::dynamic_pointer_cast<GeomDataAPI_Dir>(
 161           aSketch->data()->attribute(SketchPlugin_Sketch::NORM_ID()));
 162   std::shared_ptr<GeomAPI_Dir> aNormal = aNDir->dir();
 163   GeomShapePtr anArcShape = boolean(REVERSED_ID())->value() ?
 164       GeomAlgoAPI_EdgeBuilder::lineCircleArc(aCenter, anEnd, aStart, aNormal)
 165     : GeomAlgoAPI_EdgeBuilder::lineCircleArc(aCenter, aStart, anEnd, aNormal);
 166   GeomShapePtr aCenterPointShape = GeomAlgoAPI_PointBuilder::vertex(aCenter);
 167
 168   if(!anArcShape.get() || !aCenterPointShape.get()) {
 169     return AISObjectPtr();
 170   }
 171
 172   std::list<std::shared_ptr<GeomAPI_Shape> > aShapes;
 173   aShapes.push_back(anArcShape);
 174   aShapes.push_back(aCenterPointShape);
 175
 176   std::shared_ptr<GeomAPI_Shape> aCompound = GeomAlgoAPI_CompoundBuilder::compound(aShapes);
 177   AISObjectPtr anAIS = thePrevious;
 178   if(!anAIS.get()) {
 179     anAIS.reset(new GeomAPI_AISObject());
 180   }
 181   anAIS->createShape(aCompound);
 182   return anAIS;
 183 }
 184
 185 void SketchPlugin_MacroArc::execute()
 186 {
 187   FeaturePtr anArcFeature = createArcFeature();
 188
 189   myCenter.reset();
 190   myStart.reset();
 191   myEnd.reset();
 192
 193   // Create constraints.
 194   std::string anArcType = string(ARC_TYPE())->value();
 195   if(anArcType == ARC_TYPE_BY_CENTER_AND_POINTS()) {
 196     SketchPlugin_Tools::createConstraint(this,
 197                                          CENTER_POINT_REF_ID(),
 198                                          anArcFeature->attribute(SketchPlugin_Arc::CENTER_ID()),
 199                                          ObjectPtr(),
 200                                          false);
 201     SketchPlugin_Tools::createConstraint(this,
 202                                          START_POINT_REF_ID(),
 203                                          anArcFeature->attribute(SketchPlugin_Arc::START_ID()),
 204                                          ObjectPtr(),
 205                                          false);
 206     SketchPlugin_Tools::createConstraint(this,
 207                                          END_POINT_REF_ID(),
 208                                          anArcFeature->attribute(SketchPlugin_Arc::END_ID()),
 209                                          ObjectPtr(),
 210                                          false);
 211   } else if(anArcType == ARC_TYPE_BY_THREE_POINTS()) {
 212     SketchPlugin_Tools::createConstraint(this,
 213                                          START_POINT_REF_ID(),
 214                                          anArcFeature->attribute(SketchPlugin_Arc::START_ID()),
 215                                          ObjectPtr(),
 216                                          false);
 217     SketchPlugin_Tools::createConstraint(this,
 218                                          END_POINT_REF_ID(),
 219                                          anArcFeature->attribute(SketchPlugin_Arc::END_ID()),
 220                                          ObjectPtr(),
 221                                          false);
 222     SketchPlugin_Tools::createConstraint(this,
 223                                          PASSED_POINT_REF_ID(),
 224                                          AttributePtr(),
 225                                          anArcFeature->lastResult(),
 226                                          true);
 227   } else if(anArcType == ARC_TYPE_BY_TANGENT_EDGE()) {
 228     SketchPlugin_Tools::createConstraint(this,
 229                                          END_POINT_REF_ID(),
 230                                          anArcFeature->attribute(SketchPlugin_Arc::END_ID()),
 231                                          ObjectPtr(),
 232                                          false);
 233   }
 234 }
 235
 236 FeaturePtr SketchPlugin_MacroArc::createArcFeature()
 237 {
 238   FeaturePtr anArcFeature = sketch()->addFeature(SketchPlugin_Arc::ID());
 239   std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 240       anArcFeature->attribute(SketchPlugin_Arc::CENTER_ID()))->setValue(myCenter);
 241   std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 242       anArcFeature->attribute(SketchPlugin_Arc::START_ID()))->setValue(myStart);
 243   std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 244       anArcFeature->attribute(SketchPlugin_Arc::END_ID()))->setValue(myEnd);
 245   anArcFeature->boolean(SketchPlugin_Arc::REVERSED_ID())
 246                 ->setValue(boolean(REVERSED_ID())->value());
 247   anArcFeature->boolean(SketchPlugin_Arc::AUXILIARY_ID())
 248                 ->setValue(boolean(AUXILIARY_ID())->value());
 249   anArcFeature->execute();
 250
 251   return anArcFeature;
 252 }
 253
 254 void SketchPlugin_MacroArc::fillByCenterAndTwoPassed()
 255 {
 256   AttributePoint2DPtr aCenterPointAttr =
 257       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(CENTER_POINT_ID()));
 258   if (!aCenterPointAttr->isInitialized())
 259       return;
 260
 261   AttributePoint2DPtr aStartPointAttr =
 262       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(START_POINT_ID()));
 263   if (!aStartPointAttr->isInitialized())
 264     return;
 265
 266   myCenter = aCenterPointAttr->pnt();
 267   myStart = aStartPointAttr->pnt();
 268   myEnd = myStart;
 269
 270   AttributePoint2DPtr anEndPointAttr =
 271       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(END_POINT_ID()));
 272   if (!anEndPointAttr->isInitialized())
 273     return;
 274
 275   GeomAPI_Circ2d aCircleForArc(myCenter, myStart);
 276
 277   // End point should be a projection on circle.
 278   bool aWasBlocked = data()->blockSendAttributeUpdated(true);
 279   projectPointOnCircle(anEndPointAttr, aCircleForArc);
 280   data()->blockSendAttributeUpdated(aWasBlocked, false);
 281   myEnd = anEndPointAttr->pnt();
 282
 283   // update the REVERSED flag
 284   recalculateReversedFlagByEnd(aCircleForArc);
 285 }
 286
 287 void SketchPlugin_MacroArc::recalculateReversedFlagByEnd(const GeomAPI_Circ2d& theCurrentCircular)
 288 {
 289   double aParameterNew = 0.0;
 290   if(theCurrentCircular.parameter(myEnd, paramTolerance, aParameterNew)) {
 291     if(myParamBefore <= PI / 2.0 && aParameterNew >= PI * 1.5) {
 292       boolean(REVERSED_ID())->setValue(true);
 293     } else if(myParamBefore >= PI * 1.5 && aParameterNew <= PI / 2.0) {
 294       boolean(REVERSED_ID())->setValue(false);
 295     }
 296   }
 297   myParamBefore = aParameterNew;
 298 }
 299
 300 void SketchPlugin_MacroArc::fillByThreePassedPoints()
 301 {
 302   AttributePoint2DPtr aStartPointAttr =
 303       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(START_POINT_ID()));
 304   if (!aStartPointAttr->isInitialized())
 305     return;
 306
 307   AttributePoint2DPtr anEndPointAttr =
 308       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(END_POINT_ID()));
 309   if (!anEndPointAttr->isInitialized())
 310     return;
 311
 312   myStart = aStartPointAttr->pnt();
 313   myEnd = anEndPointAttr->pnt();
 314
 315   AttributePoint2DPtr aPassedPointAttr =
 316       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(PASSED_POINT_ID()));
 317   if (aPassedPointAttr->isInitialized()) {
 318     std::shared_ptr<GeomAPI_Pnt2d> aPassedPnt;
 319     std::shared_ptr<GeomAPI_Shape> aTangentCurve;
 320     SketchPlugin_Tools::convertRefAttrToPointOrTangentCurve(
 321         refattr(PASSED_POINT_REF_ID()), aPassedPointAttr, aTangentCurve, aPassedPnt);
 322
 323     std::shared_ptr<GeomAPI_Interface> aPassed;
 324     if (aTangentCurve)
 325       aPassed = aTangentCurve;
 326     else
 327       aPassed = aPassedPnt;
 328
 329     std::shared_ptr<GeomAPI_Ax3> anAxis = SketchPlugin_Sketch::plane(sketch());
 330     GeomAPI_Circ2d aCircle(myStart, myEnd, aPassed, anAxis);
 331     myCenter = aCircle.center();
 332
 333     recalculateReversedFlagByPassed(aCircle);
 334   } else
 335     myCenter.reset(new GeomAPI_Pnt2d(myStart->xy()->added(myEnd->xy())->multiplied(0.5)));
 336 }
 337
 338 void SketchPlugin_MacroArc::recalculateReversedFlagByPassed(
 339     const GeomAPI_Circ2d& theCurrentCircular)
 340 {
 341   AttributePoint2DPtr aPassedAttr =
 342       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(PASSED_POINT_ID()));
 343   std::shared_ptr<GeomAPI_Pnt2d> aPassed = theCurrentCircular.project(aPassedAttr->pnt());
 344
 345   double aEndParam, aPassedParam;
 346   theCurrentCircular.parameter(myEnd, paramTolerance, aEndParam);
 347   theCurrentCircular.parameter(aPassed, paramTolerance, aPassedParam);
 348
 349   if(aPassedParam > aEndParam)
 350     boolean(REVERSED_ID())->setValue(true);
 351   else
 352     boolean(REVERSED_ID())->setValue(false);
 353
 354   myParamBefore = aEndParam;
 355 }
 356
 357 void SketchPlugin_MacroArc::fillByTangentEdge()
 358 {
 359   AttributeRefAttrPtr aTangentAttr = refattr(TANGENT_POINT_ID());
 360   if (!aTangentAttr->isInitialized())
 361     return;
 362
 363   AttributePoint2DPtr aTangentPointAttr =
 364       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(aTangentAttr->attr());
 365   if (!aTangentPointAttr->isInitialized())
 366     return;
 367
 368   AttributePoint2DPtr anEndPointAttr =
 369       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(END_POINT_ID()));
 370   if (!anEndPointAttr->isInitialized())
 371     return;
 372
 373   myStart = aTangentPointAttr->pnt();
 374   myEnd = anEndPointAttr->pnt();
 375   if (myStart->isEqual(myEnd))
 376     return;
 377
 378   // obtain a shape the tangent point belongs to
 379   FeaturePtr aTangentFeature = ModelAPI_Feature::feature(aTangentPointAttr->owner());
 380   std::shared_ptr<GeomAPI_Shape> aTangentShape = aTangentFeature->lastResult()->shape();
 381
 382   std::shared_ptr<GeomAPI_Ax3> anAxis = SketchPlugin_Sketch::plane(sketch());
 383   GeomAPI_Circ2d aCircle(myStart, myEnd, aTangentShape, anAxis);
 384   myCenter = aCircle.center();
 385
 386   // rebuild circle to set start point equal to zero parameter
 387   aCircle = GeomAPI_Circ2d(myCenter, myStart);
 388   recalculateReversedFlagByEnd(aCircle);
 389 }