]> SALOME platform Git repositories - modules/shaper.git/blob - src/SketchSolver/SketchSolver_ConstraintAngle.cpp
Salome HOME
d255da0e34fbc92d449238ac7cfbdaae1b4d1c45
[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   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 }