src/SketchSolver/SketchSolver_ConstraintAngle.cpp

   1 #include <SketchSolver_ConstraintAngle.h>
   2
   3 #include <GeomAPI_Dir2d.h>
   4 #include <GeomAPI_Lin2d.h>
   5 #include <GeomAPI_Pnt2d.h>
   6 #include <GeomAPI_XY.h>
   7
   8 #include <cmath>
   9
  10 void SketchSolver_ConstraintAngle::getAttributes(
  11     double& theValue, std::vector<Slvs_hEntity>& theAttributes)
  12 {
  13   SketchSolver_Constraint::getAttributes(theValue, theAttributes);
  14
  15   myAngle = theValue;
  16 }
  17
  18
  19 void SketchSolver_ConstraintAngle::adjustConstraint()
  20 {
  21   static const double aTol = 1000. * tolerance;
  22   Slvs_Constraint aConstraint = myStorage->getConstraint(mySlvsConstraints.front());
  23
  24   bool isFixed[2][2];
  25   std::shared_ptr<GeomAPI_Pnt2d> aPoints[2][2]; // start and end points of lines
  26   Slvs_hConstraint aFixedConstraint;
  27   Slvs_hEntity anEnt[2] = {aConstraint.entityA, aConstraint.entityB};
  28   for (int i = 0; i < 2; i++) {
  29     const Slvs_Entity& aLine = myStorage->getEntity(anEnt[i]);
  30     double aCoef = -1.0;
  31     for (int j = 0; j < 2; j++, aCoef += 2.0) {
  32       const Slvs_Entity& aPoint = myStorage->getEntity(aLine.point[j]);
  33       double aCoords[2];
  34       for (int k = 0; k < 2; k++)
  35         aCoords[k] = myStorage->getParameter(aPoint.param[k]).val;
  36       isFixed[i][j] = myStorage->isPointFixed(aPoint.h, aFixedConstraint, true);
  37       aPoints[i][j] = std::shared_ptr<GeomAPI_Pnt2d>(new GeomAPI_Pnt2d(aCoords[0], aCoords[1]));
  38     }
  39   }
  40
  41   std::shared_ptr<GeomAPI_Lin2d> aLine[2] = {
  42     std::shared_ptr<GeomAPI_Lin2d>(new GeomAPI_Lin2d(aPoints[0][0], aPoints[0][1])),
  43     std::shared_ptr<GeomAPI_Lin2d>(new GeomAPI_Lin2d(aPoints[1][0], aPoints[1][1]))
  44   };
  45   std::shared_ptr<GeomAPI_Pnt2d> anIntersection = aLine[0]->intersect(aLine[1]);
  46   if (!anIntersection)
  47     return;
  48   double aDist[2][2];
  49   for (int i = 0; i < 2; i++)
  50     for (int j = 0; j < 2; j++)
  51       aDist[i][j] = anIntersection->distance(aPoints[i][j]);
  52   std::shared_ptr<GeomAPI_Dir2d> aDir[2];
  53   for (int i = 0; i < 2; i++)
  54     if (aDist[i][1] > aDist[i][0])
  55       aDir[i] = std::shared_ptr<GeomAPI_Dir2d>(new GeomAPI_Dir2d(
  56           aPoints[i][1]->xy()->decreased(anIntersection->xy())));
  57     else
  58       aDir[i] = std::shared_ptr<GeomAPI_Dir2d>(new GeomAPI_Dir2d(
  59           aPoints[i][0]->xy()->decreased(anIntersection->xy())));
  60
  61   aConstraint.other = false;
  62   for (int i = 0; i < 2; i++)
  63     if (aLine[i]->direction()->dot(aDir[i]) < 0.0)
  64       aConstraint.other = !aConstraint.other;
  65   myStorage->updateConstraint(aConstraint);
  66
  67   bool isChanged = fabs(myAngle - aConstraint.valA) > aTol;
  68   // myAngle should be updated even if the angle of constraint is changed too little
  69   myAngle = aConstraint.valA;
  70   if (!isChanged)
  71     return; // the angle was not changed, no need to recalculate positions of lines
  72
  73   if (isFixed[0][0] && isFixed[0][1] && isFixed[1][0] && isFixed[1][1])
  74     return; // both lines are fixed => no need to update them
  75
  76   // Recalculate positions of lines to avoid conflicting constraints
  77   // while changing angle value several times
  78   double cosA = cos(myAngle * PI / 180.0);
  79   double sinA = sin(myAngle * PI / 180.0);
  80   if (aDir[0]->cross(aDir[1]) < 0.0)
  81     sinA *= -1.0;
  82   int aLineToUpd = 1;
  83   if (isFixed[1][0] && isFixed[1][1]) {
  84     sinA *= -1.0;
  85     aLineToUpd = 0;
  86   }
  87   double x = aDir[1-aLineToUpd]->x() * cosA - aDir[1-aLineToUpd]->y() * sinA;
  88   double y = aDir[1-aLineToUpd]->x() * sinA + aDir[1-aLineToUpd]->y() * cosA;
  89
  90   std::shared_ptr<GeomAPI_Pnt2d> aNewPoints[2];
  91   for (int i = 0; i < 2; i++) {
  92     aNewPoints[i] = std::shared_ptr<GeomAPI_Pnt2d>(
  93         new GeomAPI_Pnt2d(anIntersection->x() + x * aDist[aLineToUpd][i],
  94                           anIntersection->y() + y * aDist[aLineToUpd][i]));
  95   }
  96
  97   std::shared_ptr<GeomAPI_XY> aDelta;
  98   if (isFixed[aLineToUpd][0] && !isFixed[aLineToUpd][1])
  99     aDelta = aPoints[aLineToUpd][0]->xy()->decreased(aNewPoints[0]->xy());
 100   else if (!isFixed[aLineToUpd][0] && isFixed[aLineToUpd][1])
 101     aDelta = aPoints[aLineToUpd][1]->xy()->decreased(aNewPoints[1]->xy());
 102   if (aDelta) {
 103     for (int i = 0; i < 2; i++) {
 104       aNewPoints[i]->setX(aNewPoints[i]->x() + aDelta->x());
 105       aNewPoints[i]->setY(aNewPoints[i]->y() + aDelta->y());
 106     }
 107   }
 108
 109   // Update positions of points
 110   const Slvs_Entity& anUpdLine = myStorage->getEntity(anEnt[aLineToUpd]);
 111   Slvs_Param aParam;
 112   for (int i = 0; i < 2; i++) {
 113     const Slvs_Entity& aPoint = myStorage->getEntity(anUpdLine.point[i]);
 114     aParam = myStorage->getParameter(aPoint.param[0]);
 115     aParam.val = aNewPoints[i]->x();
 116     myStorage->updateParameter(aParam);
 117     aParam = myStorage->getParameter(aPoint.param[1]);
 118     aParam.val = aNewPoints[i]->y();
 119     myStorage->updateParameter(aParam);
 120   }
 121 }