Salome HOME
Merge branch 'master' of newgeom:newgeom
[modules/shaper.git] / src / SketchPlugin / SketchPlugin_Line.cpp
1 // File:        SketchPlugin_Line.cxx
2 // Created:     27 Mar 2014
3 // Author:      Mikhail PONIKAROV
4
5 #include "SketchPlugin_Line.h"
6 #include "SketchPlugin_Sketch.h"
7 #include <ModelAPI_Data.h>
8 #include <ModelAPI_ResultConstruction.h>
9 #include <ModelAPI_AttributeSelection.h>
10 #include <ModelAPI_Validator.h>
11
12 #include <GeomAPI_Pnt.h>
13 #include <GeomAPI_Lin2d.h>
14 #include <GeomAPI_Pnt2d.h>
15
16 #include <GeomAlgoAPI_EdgeBuilder.h>
17 #include <GeomDataAPI_Point2D.h>
18
19 using namespace std;
20
21 SketchPlugin_Line::SketchPlugin_Line()
22     : SketchPlugin_Feature()
23 {
24 }
25
26 void SketchPlugin_Line::initAttributes()
27 {
28   data()->addAttribute(SketchPlugin_Line::START_ID(), GeomDataAPI_Point2D::type());
29   data()->addAttribute(SketchPlugin_Line::END_ID(), GeomDataAPI_Point2D::type());
30   data()->addAttribute(EXTERNAL_ID(), ModelAPI_AttributeSelection::type());
31   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), EXTERNAL_ID());
32 }
33
34 void SketchPlugin_Line::execute()
35 {
36   SketchPlugin_Sketch* aSketch = sketch();
37   if (aSketch) {
38     // compute a start point in 3D view
39     boost::shared_ptr<GeomDataAPI_Point2D> aStartAttr = boost::dynamic_pointer_cast<
40         GeomDataAPI_Point2D>(data()->attribute(SketchPlugin_Line::START_ID()));
41     // compute an end point in 3D view
42     boost::shared_ptr<GeomDataAPI_Point2D> anEndAttr = boost::dynamic_pointer_cast<
43         GeomDataAPI_Point2D>(data()->attribute(SketchPlugin_Line::END_ID()));
44     if (aStartAttr->isInitialized() && anEndAttr->isInitialized()) {
45       boost::shared_ptr<GeomAPI_Pnt> aStart(aSketch->to3D(aStartAttr->x(), aStartAttr->y()));
46       boost::shared_ptr<GeomAPI_Pnt> anEnd(aSketch->to3D(anEndAttr->x(), anEndAttr->y()));
47       // make linear edge
48       boost::shared_ptr<GeomAPI_Shape> anEdge = GeomAlgoAPI_EdgeBuilder::line(aStart, anEnd);
49       // store the result
50       boost::shared_ptr<ModelAPI_ResultConstruction> aConstr = document()->createConstruction(
51           data());
52       aConstr->setShape(anEdge);
53       aConstr->setIsInHistory(false);
54       setResult(aConstr);
55     }
56   }
57 }
58
59 void SketchPlugin_Line::move(double theDeltaX, double theDeltaY)
60 {
61   boost::shared_ptr<ModelAPI_Data> aData = data();
62   if (!aData->isValid())
63     return;
64
65   boost::shared_ptr<GeomDataAPI_Point2D> aPoint1 = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(
66       aData->attribute(SketchPlugin_Line::START_ID()));
67   aPoint1->setValue(aPoint1->x() + theDeltaX, aPoint1->y() + theDeltaY);
68
69   boost::shared_ptr<GeomDataAPI_Point2D> aPoint2 = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(
70       aData->attribute(SketchPlugin_Line::END_ID()));
71   aPoint2->setValue(aPoint2->x() + theDeltaX, aPoint2->y() + theDeltaY);
72 }
73
74 double SketchPlugin_Line::distanceToPoint(const boost::shared_ptr<GeomAPI_Pnt2d>& thePoint)
75 {
76   double aDelta = 0;
77
78   boost::shared_ptr<ModelAPI_Data> aData = data();
79   boost::shared_ptr<GeomDataAPI_Point2D> aPoint1 = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(
80       aData->attribute(SketchPlugin_Line::START_ID()));
81   boost::shared_ptr<GeomDataAPI_Point2D> aPoint2 = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(
82       aData->attribute(SketchPlugin_Line::END_ID()));
83
84   GeomAPI_Lin2d aLin2d(aPoint1->x(), aPoint1->y(), aPoint2->x(), aPoint2->y());
85
86   if (false/*projection*/) {  // TODO: if it has not been necessary, remove this block
87     boost::shared_ptr<GeomAPI_Pnt2d> aResult = aLin2d.project(thePoint);
88     aDelta = aResult->distance(thePoint);
89   } else {  // distance
90     aDelta = aLin2d.distance(thePoint);
91   }
92
93   return aDelta;
94 }
95
96 bool SketchPlugin_Line::isFixed() {
97   return data()->selection(EXTERNAL_ID())->context();
98 }