Salome HOME
895f99e62c224cf713eab09e6b4b6c7f73732835
[modules/shaper.git] / 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     if (aDist[i][0] > tolerance && aDist[i][1] > tolerance &&
53         aDist[i][0] + aDist[i][1] < aPoints[i][0]->distance(aPoints[i][1]) + 2.0 * tolerance) {
54       // the intersection point is an inner point of the line,
55       // we change the sign of distance till start point to calculate correct coordinates
56       // after rotation
57       aDist[i][0] *= -1.0;
58     }
59   }
60   std::shared_ptr<GeomAPI_Dir2d> aDir[2];
61   for (int i = 0; i < 2; i++)
62     if (aDist[i][1] > fabs(aDist[i][0]))
63       aDir[i] = std::shared_ptr<GeomAPI_Dir2d>(new GeomAPI_Dir2d(
64           aPoints[i][1]->xy()->decreased(anIntersection->xy())));
65     else {
66       aDir[i] = std::shared_ptr<GeomAPI_Dir2d>(new GeomAPI_Dir2d(
67           aPoints[i][0]->xy()->decreased(anIntersection->xy())));
68       // main direction is opposite => change signs
69       aDist[i][0] *= -1.0;
70       aDist[i][1] *= -1.0;
71     }
72
73   aConstraint.other = false;
74   for (int i = 0; i < 2; i++)
75     if (aLine[i]->direction()->dot(aDir[i]) < 0.0)
76       aConstraint.other = !aConstraint.other;
77   myStorage->updateConstraint(aConstraint);
78
79   bool isChanged = fabs(myAngle - aConstraint.valA) > aTol;
80   // myAngle should be updated even if the angle of constraint is changed too little
81   myAngle = aConstraint.valA;
82   if (!isChanged)
83     return; // the angle was not changed, no need to recalculate positions of lines
84
85   if (isFixed[0][0] && isFixed[0][1] && isFixed[1][0] && isFixed[1][1])
86     return; // both lines are fixed => no need to update them
87
88   // Recalculate positions of lines to avoid conflicting constraints
89   // while changing angle value several times
90   double cosA = cos(myAngle * PI / 180.0);
91   double sinA = sin(myAngle * PI / 180.0);
92   if (aDir[0]->cross(aDir[1]) < 0.0)
93     sinA *= -1.0;
94   int aLineToUpd = 1;
95   if (isFixed[1][0] && isFixed[1][1]) {
96     sinA *= -1.0;
97     aLineToUpd = 0;
98   }
99   double x = aDir[1-aLineToUpd]->x() * cosA - aDir[1-aLineToUpd]->y() * sinA;
100   double y = aDir[1-aLineToUpd]->x() * sinA + aDir[1-aLineToUpd]->y() * cosA;
101
102   std::shared_ptr<GeomAPI_Pnt2d> aNewPoints[2];
103   for (int i = 0; i < 2; i++) {
104     aNewPoints[i] = std::shared_ptr<GeomAPI_Pnt2d>(
105         new GeomAPI_Pnt2d(anIntersection->x() + x * aDist[aLineToUpd][i],
106                           anIntersection->y() + y * aDist[aLineToUpd][i]));
107   }
108
109   std::shared_ptr<GeomAPI_XY> aDelta;
110   if (isFixed[aLineToUpd][0] && !isFixed[aLineToUpd][1])
111     aDelta = aPoints[aLineToUpd][0]->xy()->decreased(aNewPoints[0]->xy());
112   else if (!isFixed[aLineToUpd][0] && isFixed[aLineToUpd][1])
113     aDelta = aPoints[aLineToUpd][1]->xy()->decreased(aNewPoints[1]->xy());
114   if (aDelta) {
115     for (int i = 0; i < 2; i++) {
116       aNewPoints[i]->setX(aNewPoints[i]->x() + aDelta->x());
117       aNewPoints[i]->setY(aNewPoints[i]->y() + aDelta->y());
118     }
119   }
120
121   // Update positions of points
122   const Slvs_Entity& anUpdLine = myStorage->getEntity(anEnt[aLineToUpd]);
123   Slvs_Param aParam;
124   for (int i = 0; i < 2; i++) {
125     const Slvs_Entity& aPoint = myStorage->getEntity(anUpdLine.point[i]);
126     aParam = myStorage->getParameter(aPoint.param[0]);
127     aParam.val = aNewPoints[i]->x();
128     myStorage->updateParameter(aParam);
129     aParam = myStorage->getParameter(aPoint.param[1]);
130     aParam.val = aNewPoints[i]->y();
131     myStorage->updateParameter(aParam);
132   }
133 }