src/SketchSolver/SketchSolver_ConstraintMovement.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 #include <SketchSolver_ConstraintMovement.h>
   4 #include <SketchSolver_Error.h>
   5 #include <SketchSolver_Manager.h>
   6
   7 #include <PlaneGCSSolver_EdgeWrapper.h>
   8 #include <PlaneGCSSolver_PointWrapper.h>
   9
  10 #include <SketchPlugin_Arc.h>
  11 #include <SketchPlugin_Circle.h>
  12 #include <SketchPlugin_Line.h>
  13 #include <SketchPlugin_Point.h>
  14
  15 #include <GeomDataAPI_Point2D.h>
  16
  17 #include <GeomAPI_Pnt2d.h>
  18
  19
  20 SketchSolver_ConstraintMovement::SketchSolver_ConstraintMovement(FeaturePtr theFeature)
  21   : SketchSolver_ConstraintFixed(ConstraintPtr()),
  22     myMovedFeature(theFeature),
  23     mySimpleMove(true)
  24 {
  25 }
  26
  27 SketchSolver_ConstraintMovement::SketchSolver_ConstraintMovement(AttributePtr thePoint)
  28   : SketchSolver_ConstraintFixed(ConstraintPtr()),
  29     myDraggedPoint(thePoint),
  30     mySimpleMove(true)
  31 {
  32   myMovedFeature = ModelAPI_Feature::feature(thePoint->owner());
  33 }
  34
  35 void SketchSolver_ConstraintMovement::blockEvents(bool isBlocked)
  36 {
  37   if (myMovedFeature)
  38     myMovedFeature->data()->blockSendAttributeUpdated(isBlocked);
  39 }
  40
  41 void SketchSolver_ConstraintMovement::process()
  42 {
  43   cleanErrorMsg();
  44   if (!myMovedFeature || !myStorage) {
  45     // Not enough parameters are initialized
  46     return;
  47   }
  48
  49   mySolverConstraint = initMovement();
  50   if (!myErrorMsg.empty() || !mySolverConstraint) {
  51     // Nothing to move, clear the feature to avoid changing its group
  52     // after removing the Movement constraint.
  53     myMovedFeature = FeaturePtr();
  54     return;
  55   }
  56   myStorage->addMovementConstraint(mySolverConstraint);
  57 }
  58
  59
  60 static bool isSimpleMove(FeaturePtr theMovedFeature, AttributePtr theDraggedPoint)
  61 {
  62   bool isSimple = true;
  63   if (theMovedFeature->getKind() == SketchPlugin_Circle::ID())
  64     isSimple = (theDraggedPoint.get() != 0);
  65   else if (theMovedFeature->getKind() == SketchPlugin_Arc::ID()) {
  66     isSimple = (theDraggedPoint.get() != 0 &&
  67                 theDraggedPoint->id() == SketchPlugin_Arc::CENTER_ID());
  68   }
  69   return isSimple;
  70 }
  71
  72 ConstraintWrapperPtr SketchSolver_ConstraintMovement::initMovement()
  73 {
  74   ConstraintWrapperPtr aConstraint;
  75
  76   // if the feature is copy, do not move it
  77   std::shared_ptr<SketchPlugin_Feature> aSketchFeature =
  78       std::dynamic_pointer_cast<SketchPlugin_Feature>(myMovedFeature);
  79   if (!aSketchFeature || aSketchFeature->isCopy()) {
  80     myStorage->setNeedToResolve(true);
  81     return aConstraint;
  82   }
  83
  84   EntityWrapperPtr anEntity =
  85       myDraggedPoint ? myStorage->entity(myDraggedPoint) : myStorage->entity(myMovedFeature);
  86   if (!anEntity) {
  87     myStorage->update(myMovedFeature, true);
  88     anEntity =
  89         myDraggedPoint ? myStorage->entity(myDraggedPoint) : myStorage->entity(myMovedFeature);
  90     if (!anEntity)
  91       return aConstraint;
  92   }
  93
  94   mySimpleMove = isSimpleMove(myMovedFeature, myDraggedPoint);
  95
  96   if (mySimpleMove)
  97     aConstraint = fixFeature(anEntity);
  98   else {
  99     if (myDraggedPoint) // start or end point of arc has been moved
 100       aConstraint = fixArcExtremity(anEntity);
 101     else // arc or circle has been moved
 102       aConstraint = fixPointOnCircle(anEntity);
 103   }
 104
 105   return aConstraint;
 106 }
 107
 108 ConstraintWrapperPtr SketchSolver_ConstraintMovement::fixArcExtremity(
 109     const EntityWrapperPtr& theArcExtremity)
 110 {
 111   static const int nbParams = 4;
 112   myFixedValues.reserve(nbParams); // moved point and center of arc
 113
 114   EdgeWrapperPtr aCircularEntity = std::dynamic_pointer_cast<PlaneGCSSolver_EdgeWrapper>(
 115       myStorage->entity(myMovedFeature));
 116   std::shared_ptr<GCS::Arc> anArc =
 117       std::dynamic_pointer_cast<GCS::Arc>(aCircularEntity->entity());
 118
 119   PointWrapperPtr aPoint =
 120       std::dynamic_pointer_cast<PlaneGCSSolver_PointWrapper>(theArcExtremity);
 121
 122   double* aParams[nbParams] = { aPoint->point()->x, aPoint->point()->y,
 123                                 anArc->center.x, anArc->center.y };
 124
 125   std::list<GCSConstraintPtr> aConstraints;
 126   for (int i = 0; i < nbParams; ++i) {
 127     myFixedValues.push_back(*aParams[i]);
 128     GCSConstraintPtr aNewConstraint(new GCS::ConstraintEqual(&myFixedValues[i], aParams[i]));
 129     aNewConstraint->rescale(0.01);
 130     aConstraints.push_back(aNewConstraint);
 131   }
 132
 133   return ConstraintWrapperPtr(
 134       new PlaneGCSSolver_ConstraintWrapper(aConstraints, getType()));
 135 }
 136
 137 ConstraintWrapperPtr SketchSolver_ConstraintMovement::fixPointOnCircle(
 138     const EntityWrapperPtr& theCircular)
 139 {
 140   static const double scale = 0.01;
 141   static const int nbParams = 4;
 142   myFixedValues.reserve(nbParams); // moved point and center of arc/circle
 143
 144   EdgeWrapperPtr aCircularEntity =
 145       std::dynamic_pointer_cast<PlaneGCSSolver_EdgeWrapper>(theCircular);
 146   std::shared_ptr<GCS::Circle> aCircular =
 147       std::dynamic_pointer_cast<GCS::Circle>(aCircularEntity->entity());
 148
 149   // initialize fixed values
 150   myFixedValues.push_back(*aCircular->center.x + *aCircular->rad);
 151   myFixedValues.push_back(*aCircular->center.y);
 152   myFixedValues.push_back(*aCircular->center.x);
 153   myFixedValues.push_back(*aCircular->center.y);
 154
 155   // create a moved point
 156   GCS::Point aPointOnCircle;
 157   aPointOnCircle.x = &myFixedValues[0];
 158   aPointOnCircle.y = &myFixedValues[1];
 159
 160   std::list<GCSConstraintPtr> aConstraints;
 161   // point-on-circle
 162   GCSConstraintPtr aNewConstraint(
 163       new GCS::ConstraintP2PDistance(aPointOnCircle, aCircular->center, aCircular->rad));
 164   aNewConstraint->rescale(scale);
 165   aConstraints.push_back(aNewConstraint);
 166   // fixed center (x)
 167   aNewConstraint = GCSConstraintPtr(
 168       new GCS::ConstraintEqual(&myFixedValues[2], aCircular->center.x));
 169   aNewConstraint->rescale(scale);
 170   aConstraints.push_back(aNewConstraint);
 171   // fixed center (y)
 172   aNewConstraint = GCSConstraintPtr(
 173       new GCS::ConstraintEqual(&myFixedValues[3], aCircular->center.y));
 174   aNewConstraint->rescale(scale);
 175   aConstraints.push_back(aNewConstraint);
 176
 177   return ConstraintWrapperPtr(
 178       new PlaneGCSSolver_ConstraintWrapper(aConstraints, getType()));
 179 }
 180
 181
 182 void SketchSolver_ConstraintMovement::startPoint(
 183     const std::shared_ptr<GeomAPI_Pnt2d>& theStartPoint)
 184 {
 185   myStartPoint = theStartPoint;
 186   if (!mySimpleMove) {
 187     myFixedValues[0] = myStartPoint->x();
 188     myFixedValues[1] = myStartPoint->y();
 189   }
 190 }
 191
 192 void SketchSolver_ConstraintMovement::moveTo(
 193     const std::shared_ptr<GeomAPI_Pnt2d>& theDestinationPoint)
 194 {
 195   double aDelta[2] = { theDestinationPoint->x() - myStartPoint->x(),
 196                        theDestinationPoint->y() - myStartPoint->y() };
 197
 198   int aMaxSize = mySimpleMove ? (int)myFixedValues.size() : 2;
 199   for (int i = 0; i < aMaxSize; ++i)
 200     myFixedValues[i] += aDelta[i % 2];
 201 }