Salome HOME
Add tools
[modules/shaper.git] / src / SketchSolver / SketchSolver_ConstraintAngle.cpp
index cfad93f19f9b119fc42e44d6d5294488570f55d1..895f99e62c224cf713eab09e6b4b6c7f73732835 100644 (file)
@@ -1,5 +1,10 @@
 #include <SketchSolver_ConstraintAngle.h>
 
+#include <GeomAPI_Dir2d.h>
+#include <GeomAPI_Lin2d.h>
+#include <GeomAPI_Pnt2d.h>
+#include <GeomAPI_XY.h>
+
 #include <cmath>
 
 void SketchSolver_ConstraintAngle::getAttributes(
@@ -13,24 +18,116 @@ void SketchSolver_ConstraintAngle::getAttributes(
 
 void SketchSolver_ConstraintAngle::adjustConstraint()
 {
+  static const double aTol = 1000. * tolerance;
   Slvs_Constraint aConstraint = myStorage->getConstraint(mySlvsConstraints.front());
 
-  double aLineDir[2][2] = { {0.0, 0.0}, {0.0, 0.0} };
+  bool isFixed[2][2];
+  std::shared_ptr<GeomAPI_Pnt2d> aPoints[2][2]; // start and end points of lines
+  Slvs_hConstraint aFixedConstraint;
   Slvs_hEntity anEnt[2] = {aConstraint.entityA, aConstraint.entityB};
   for (int i = 0; i < 2; i++) {
     const Slvs_Entity& aLine = myStorage->getEntity(anEnt[i]);
     double aCoef = -1.0;
     for (int j = 0; j < 2; j++, aCoef += 2.0) {
       const Slvs_Entity& aPoint = myStorage->getEntity(aLine.point[j]);
+      double aCoords[2];
       for (int k = 0; k < 2; k++)
-        aLineDir[i][k] += aCoef * myStorage->getParameter(aPoint.param[k]).val;
+        aCoords[k] = myStorage->getParameter(aPoint.param[k]).val;
+      isFixed[i][j] = myStorage->isPointFixed(aPoint.h, aFixedConstraint, true);
+      aPoints[i][j] = std::shared_ptr<GeomAPI_Pnt2d>(new GeomAPI_Pnt2d(aCoords[0], aCoords[1]));
     }
   }
-  double aDot = aLineDir[0][0] * aLineDir[1][0] + aLineDir[0][1] * aLineDir[1][1];
 
-  aConstraint.other = aDot * (90.0 - fabs(aConstraint.valA)) < 0.0;
-  if ((90.0 - fabs(aConstraint.valA)) * (90.0 - fabs(myAngle)) < 0.0)
-    aConstraint.other = !aConstraint.other;
-  myAngle = aConstraint.valA;
+  std::shared_ptr<GeomAPI_Lin2d> aLine[2] = {
+    std::shared_ptr<GeomAPI_Lin2d>(new GeomAPI_Lin2d(aPoints[0][0], aPoints[0][1])),
+    std::shared_ptr<GeomAPI_Lin2d>(new GeomAPI_Lin2d(aPoints[1][0], aPoints[1][1]))
+  };
+  std::shared_ptr<GeomAPI_Pnt2d> anIntersection = aLine[0]->intersect(aLine[1]);
+  if (!anIntersection)
+    return;
+  double aDist[2][2];
+  for (int i = 0; i < 2; i++) {
+    for (int j = 0; j < 2; j++)
+      aDist[i][j] = anIntersection->distance(aPoints[i][j]);
+    if (aDist[i][0] > tolerance && aDist[i][1] > tolerance &&
+        aDist[i][0] + aDist[i][1] < aPoints[i][0]->distance(aPoints[i][1]) + 2.0 * tolerance) {
+      // the intersection point is an inner point of the line,
+      // we change the sign of distance till start point to calculate correct coordinates
+      // after rotation
+      aDist[i][0] *= -1.0;
+    }
+  }
+  std::shared_ptr<GeomAPI_Dir2d> aDir[2];
+  for (int i = 0; i < 2; i++)
+    if (aDist[i][1] > fabs(aDist[i][0]))
+      aDir[i] = std::shared_ptr<GeomAPI_Dir2d>(new GeomAPI_Dir2d(
+          aPoints[i][1]->xy()->decreased(anIntersection->xy())));
+    else {
+      aDir[i] = std::shared_ptr<GeomAPI_Dir2d>(new GeomAPI_Dir2d(
+          aPoints[i][0]->xy()->decreased(anIntersection->xy())));
+      // main direction is opposite => change signs
+      aDist[i][0] *= -1.0;
+      aDist[i][1] *= -1.0;
+    }
+
+  aConstraint.other = false;
+  for (int i = 0; i < 2; i++)
+    if (aLine[i]->direction()->dot(aDir[i]) < 0.0)
+      aConstraint.other = !aConstraint.other;
   myStorage->updateConstraint(aConstraint);
+
+  bool isChanged = fabs(myAngle - aConstraint.valA) > aTol;
+  // myAngle should be updated even if the angle of constraint is changed too little
+  myAngle = aConstraint.valA;
+  if (!isChanged)
+    return; // the angle was not changed, no need to recalculate positions of lines
+
+  if (isFixed[0][0] && isFixed[0][1] && isFixed[1][0] && isFixed[1][1])
+    return; // both lines are fixed => no need to update them
+
+  // Recalculate positions of lines to avoid conflicting constraints
+  // while changing angle value several times
+  double cosA = cos(myAngle * PI / 180.0);
+  double sinA = sin(myAngle * PI / 180.0);
+  if (aDir[0]->cross(aDir[1]) < 0.0)
+    sinA *= -1.0;
+  int aLineToUpd = 1;
+  if (isFixed[1][0] && isFixed[1][1]) {
+    sinA *= -1.0;
+    aLineToUpd = 0;
+  }
+  double x = aDir[1-aLineToUpd]->x() * cosA - aDir[1-aLineToUpd]->y() * sinA;
+  double y = aDir[1-aLineToUpd]->x() * sinA + aDir[1-aLineToUpd]->y() * cosA;
+
+  std::shared_ptr<GeomAPI_Pnt2d> aNewPoints[2];
+  for (int i = 0; i < 2; i++) {
+    aNewPoints[i] = std::shared_ptr<GeomAPI_Pnt2d>(
+        new GeomAPI_Pnt2d(anIntersection->x() + x * aDist[aLineToUpd][i],
+                          anIntersection->y() + y * aDist[aLineToUpd][i]));
+  }
+
+  std::shared_ptr<GeomAPI_XY> aDelta;
+  if (isFixed[aLineToUpd][0] && !isFixed[aLineToUpd][1])
+    aDelta = aPoints[aLineToUpd][0]->xy()->decreased(aNewPoints[0]->xy());
+  else if (!isFixed[aLineToUpd][0] && isFixed[aLineToUpd][1])
+    aDelta = aPoints[aLineToUpd][1]->xy()->decreased(aNewPoints[1]->xy());
+  if (aDelta) {
+    for (int i = 0; i < 2; i++) {
+      aNewPoints[i]->setX(aNewPoints[i]->x() + aDelta->x());
+      aNewPoints[i]->setY(aNewPoints[i]->y() + aDelta->y());
+    }
+  }
+
+  // Update positions of points
+  const Slvs_Entity& anUpdLine = myStorage->getEntity(anEnt[aLineToUpd]);
+  Slvs_Param aParam;
+  for (int i = 0; i < 2; i++) {
+    const Slvs_Entity& aPoint = myStorage->getEntity(anUpdLine.point[i]);
+    aParam = myStorage->getParameter(aPoint.param[0]);
+    aParam.val = aNewPoints[i]->x();
+    myStorage->updateParameter(aParam);
+    aParam = myStorage->getParameter(aPoint.param[1]);
+    aParam.val = aNewPoints[i]->y();
+    myStorage->updateParameter(aParam);
+  }
 }