Salome HOME
bf551e43c27dc156663e6cbe472b9b7c0ecabc66
[modules/shaper.git] / src / SketchSolver / SketchSolver_ConstraintTangent.cpp
1 #include <SketchSolver_ConstraintTangent.h>
2 #include <SketchSolver_Error.h>
3 #include <SketchSolver_Manager.h>
4
5 #include <GeomAPI_Pnt2d.h>
6
7
8 /// \brief Check whether the entities has only one shared point
9 static bool hasSingleCoincidence(EntityWrapperPtr theEntity1, EntityWrapperPtr theEntity2)
10 {
11   BuilderPtr aBuilder = SketchSolver_Manager::instance()->builder();
12
13   const std::list<EntityWrapperPtr>& aPoints1 = theEntity1->subEntities();
14   const std::list<EntityWrapperPtr>& aPoints2 = theEntity2->subEntities();
15
16   std::list<EntityWrapperPtr>::const_iterator aStartIt1 = aPoints1.begin();
17   if (theEntity1->type() == ENTITY_ARC) ++aStartIt1; // skip center of arc
18   std::list<EntityWrapperPtr>::const_iterator aStartIt2 = aPoints2.begin();
19   if (theEntity2->type() == ENTITY_ARC) ++aStartIt2; // skip center of arc
20
21   int aNbCoinc = 0;
22   std::list<EntityWrapperPtr>::const_iterator anIt1, anIt2;
23   for (anIt1 = aStartIt1; anIt1 != aPoints1.end(); ++anIt1) {
24     std::shared_ptr<GeomAPI_Pnt2d> aPt1 = aBuilder->point(*anIt1);
25     for (anIt2 = aStartIt2; anIt2 != aPoints2.end(); ++anIt2) {
26       std::shared_ptr<GeomAPI_Pnt2d> aPt2 = aBuilder->point(*anIt2);
27       if (aPt1->distance(aPt2) < tolerance)
28         ++aNbCoinc;
29     }
30   }
31   return aNbCoinc == 1;
32 }
33
34
35 void SketchSolver_ConstraintTangent::getAttributes(
36     double& theValue,
37     std::vector<EntityWrapperPtr>& theAttributes)
38 {
39   SketchSolver_Constraint::getAttributes(theValue, theAttributes);
40   if (!myErrorMsg.empty() || !theAttributes[2] || !theAttributes[3]) {
41     theAttributes.clear();
42     return;
43   }
44
45   // Check the quantity of entities of each type and their order (arcs first)
46   int aNbLines = 0;
47   int aNbArcs = 0;
48   bool isSwap = false; // whether need to swap arguments (arc goes before line)
49   std::vector<EntityWrapperPtr>::iterator anEntIt = theAttributes.begin() + 2;
50   for (; anEntIt != theAttributes.end(); ++anEntIt) {
51     if ((*anEntIt)->type() == ENTITY_LINE)
52       ++aNbLines;
53     else if ((*anEntIt)->type() == ENTITY_ARC) {
54       ++aNbArcs;
55       isSwap = aNbLines > 0;
56     }
57   }
58
59   if (aNbArcs < 1) {
60     myErrorMsg = SketchSolver_Error::INCORRECT_TANGENCY_ATTRIBUTE();
61     return;
62   }
63   if (aNbLines == 1 && aNbArcs == 1)
64     myType = CONSTRAINT_TANGENT_ARC_LINE;
65   else if (aNbArcs == 2)
66     myType = CONSTRAINT_TANGENT_ARC_ARC;
67   else {
68     myErrorMsg = SketchSolver_Error::INCORRECT_ATTRIBUTE();
69     return;
70   }
71
72   if (!hasSingleCoincidence(theAttributes[2], theAttributes[3]))
73     myErrorMsg = SketchSolver_Error::INCORRECT_ATTRIBUTE();
74
75   if (isSwap) {
76     EntityWrapperPtr aTemp = theAttributes[2];
77     theAttributes[2] = theAttributes[3];
78     theAttributes[3] = aTemp;
79   }
80 }