Salome HOME
53aa6befb1d4f36d48572ed5a140214a3607ae4a
[modules/shaper.git] / src / PartSet / PartSet_FeatureLinePrs.cpp
1 // File:        PartSet_FeaturePrs.h
2 // Created:     04 Jun 2014
3 // Author:      Natalia ERMOLAEVA
4
5 #include <PartSet_FeatureLinePrs.h>
6 #include <PartSet_Tools.h>
7
8 #include <SketchPlugin_Feature.h>
9 #include <SketchPlugin_Sketch.h>
10 #include <SketchPlugin_ConstraintCoincidence.h>
11 #include <SketchPlugin_Line.h>
12 #include <SketchPlugin_Constraint.h>
13
14 #include <GeomDataAPI_Point2D.h>
15
16 #include <ModelAPI_Data.h>
17 #include <ModelAPI_Document.h>
18 #include <ModelAPI_AttributeRefAttr.h>
19 #include <ModelAPI_AttributeRefList.h>
20
21 #include <Precision.hxx>
22 #include <V3d_View.hxx>
23
24 using namespace std;
25
26 PartSet_FeatureLinePrs::PartSet_FeatureLinePrs(FeaturePtr theSketch)
27 : PartSet_FeaturePrs(theSketch)
28 {
29 }
30
31 std::string PartSet_FeatureLinePrs::getKind()
32 {
33   return SKETCH_LINE_KIND;
34 }
35
36 void PartSet_FeatureLinePrs::initFeature(FeaturePtr theFeature)
37 {
38   if (feature() && theFeature)
39   {
40     // use the last point of the previous feature as the first of the new one
41     boost::shared_ptr<ModelAPI_Data> aData = theFeature->data();
42     boost::shared_ptr<GeomDataAPI_Point2D> anInitPoint = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>
43                                                                   (aData->attribute(LINE_ATTR_END));
44     PartSet_Tools::setFeaturePoint(feature(), anInitPoint->x(), anInitPoint->y(), LINE_ATTR_START);
45     PartSet_Tools::setFeaturePoint(feature(), anInitPoint->x(), anInitPoint->y(), LINE_ATTR_END);
46
47     aData = feature()->data();
48     boost::shared_ptr<GeomDataAPI_Point2D> aPoint = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>
49                                                                  (aData->attribute(LINE_ATTR_START));
50     PartSet_Tools::createConstraint(sketch(), anInitPoint, aPoint);
51   }
52 }
53
54 PartSet_SelectionMode PartSet_FeatureLinePrs::setPoint(double theX, double theY,
55                                                        const PartSet_SelectionMode& theMode)
56 {
57   PartSet_SelectionMode aMode = theMode;
58   switch (theMode)
59   {
60     case SM_FirstPoint: {
61       PartSet_Tools::setFeaturePoint(feature(), theX, theY, LINE_ATTR_START);
62       PartSet_Tools::setFeaturePoint(feature(), theX, theY, LINE_ATTR_END);
63       aMode = SM_SecondPoint;
64     }
65     break;
66     case SM_SecondPoint: {
67       PartSet_Tools::setFeaturePoint(feature(), theX, theY, LINE_ATTR_END);
68       aMode = SM_DonePoint;
69    }
70     break;
71     default:
72       break;
73   }
74   return aMode;
75 }
76
77 std::string PartSet_FeatureLinePrs::getAttribute(const PartSet_SelectionMode& theMode) const
78 {
79   std::string aAttribute;
80   switch (theMode)
81   {
82     case SM_FirstPoint:
83       aAttribute = LINE_ATTR_START;
84     break;
85     case SM_SecondPoint:
86       aAttribute = LINE_ATTR_END;
87     break;
88     default:
89     break;
90   }
91   return aAttribute;
92 }
93
94 PartSet_SelectionMode PartSet_FeatureLinePrs::getNextMode(const std::string& theAttribute) const
95 {
96   PartSet_SelectionMode aMode;
97
98   if (theAttribute == LINE_ATTR_START)
99     aMode = SM_SecondPoint;
100   else if (theAttribute == LINE_ATTR_END)
101     aMode = SM_DonePoint;
102   return aMode;
103 }
104
105 void PartSet_FeatureLinePrs::projectPointOnLine(FeaturePtr theFeature,
106                                                 const PartSet_SelectionMode& theMode,
107                                                 const gp_Pnt& thePoint, Handle(V3d_View) theView,
108                                                 double& theX, double& theY)
109 {
110   if (theFeature) {
111     double X0, X1, X2, X3;
112     double Y0, Y1, Y2, Y3;
113     getLinePoint(theFeature, LINE_ATTR_START, X2, Y2);
114     getLinePoint(theFeature, LINE_ATTR_END, X3, Y3);
115     PartSet_Tools::convertTo2D(thePoint, sketch(), theView, X1, Y1);
116
117     switch (theMode) {
118       case SM_FirstPoint:
119         PartSet_Tools::projectPointOnLine(X2, Y2, X3, Y3, X1, Y1, theX, theY);
120       break;
121       case SM_SecondPoint: {
122         getLinePoint(feature(), LINE_ATTR_START, X0, Y0);
123         PartSet_Tools::intersectLines(X0, Y0, X1, Y1, X2, Y2, X3, Y3, theX, theY);
124       }
125       break;
126       default:
127       break;
128     }
129   }
130 }
131
132 double PartSet_FeatureLinePrs::distanceToPoint(FeaturePtr theFeature,
133                                       double theX, double theY)
134 {
135   double aDelta = 0;
136   if (!theFeature || theFeature->getKind() != SKETCH_LINE_KIND)
137     return aDelta;
138
139   boost::shared_ptr<ModelAPI_Data> aData = theFeature->data();
140   boost::shared_ptr<GeomDataAPI_Point2D> aPoint1 =
141         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(LINE_ATTR_START));
142   boost::shared_ptr<GeomDataAPI_Point2D> aPoint2 =
143         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(LINE_ATTR_END));
144
145   double aX, anY;
146   PartSet_Tools::projectPointOnLine(aPoint1->x(), aPoint1->y(), aPoint2->x(), aPoint2->y(), theX, theY, aX, anY);
147   aDelta = gp_Pnt(theX, theY, 0).Distance(gp_Pnt(aX, anY, 0));
148
149   return aDelta;
150 }
151
152 boost::shared_ptr<GeomDataAPI_Point2D> PartSet_FeatureLinePrs::findPoint(FeaturePtr theFeature,
153                                                                          double theX, double theY)
154 {
155   boost::shared_ptr<GeomDataAPI_Point2D> aPoint2D;
156   if (!theFeature || theFeature->getKind() != SKETCH_LINE_KIND)
157     return aPoint2D;
158
159   boost::shared_ptr<ModelAPI_Data> aData = theFeature->data();
160   aPoint2D = PartSet_FeatureLinePrs::findPoint(theFeature, theX, theY);
161
162   boost::shared_ptr<GeomDataAPI_Point2D> aPoint =
163         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(LINE_ATTR_START));
164   if (fabs(aPoint->x() - theX) < Precision::Confusion() &&
165       fabs(aPoint->y() - theY) < Precision::Confusion())
166     aPoint2D = aPoint;
167   else {
168     aPoint = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(LINE_ATTR_END));
169     if (fabs(aPoint->x() - theX) < Precision::Confusion() &&
170         fabs(aPoint->y() - theY) < Precision::Confusion())
171       aPoint2D = aPoint;
172   }
173   return aPoint2D;
174 }
175
176 boost::shared_ptr<GeomDataAPI_Point2D> PartSet_FeatureLinePrs::featurePoint
177                                                      (const PartSet_SelectionMode& theMode)
178 {
179   std::string aPointArg;
180   switch (theMode)
181   {
182     case SM_FirstPoint:
183       aPointArg = LINE_ATTR_START;
184       break;
185     case SM_SecondPoint:
186       aPointArg = LINE_ATTR_END;
187       break;
188     default:
189       break;
190   }
191   boost::shared_ptr<ModelAPI_Data> aData = feature()->data();
192   boost::shared_ptr<GeomDataAPI_Point2D> aPoint = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>
193                                                               (aData->attribute(aPointArg));
194   return aPoint;
195 }
196
197 void PartSet_FeatureLinePrs::getLinePoint(FeaturePtr theFeature, const std::string& theAttribute,
198                                           double& theX, double& theY)
199 {
200   if (!theFeature || theFeature->getKind() != PartSet_FeatureLinePrs::getKind())
201     return;
202   boost::shared_ptr<ModelAPI_Data> aData = theFeature->data();
203   boost::shared_ptr<GeomDataAPI_Point2D> aPoint =
204         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(theAttribute));
205   theX = aPoint->x();
206   theY = aPoint->y();
207 }