Salome HOME
Merge remote-tracking branch 'remotes/origin/master' into SolveSpace
[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 <GeomAPI_Pnt.h>
9 #include <GeomAlgoAPI_EdgeBuilder.h>
10 #include <GeomDataAPI_Point2D.h>
11
12 using namespace std;
13
14 // face of the square-face displayed for selection of general plane
15 const double PLANE_SIZE = 200;
16
17 SketchPlugin_Line::SketchPlugin_Line()
18   : SketchPlugin_Feature()
19 {
20   setSketch(0);
21 }
22
23 void SketchPlugin_Line::initAttributes()
24 {
25   data()->addAttribute(LINE_ATTR_START, GeomDataAPI_Point2D::type());
26   data()->addAttribute(LINE_ATTR_END, GeomDataAPI_Point2D::type());
27 }
28
29 void SketchPlugin_Line::execute() 
30 {
31 }
32
33 const boost::shared_ptr<GeomAPI_Shape>& SketchPlugin_Line::preview()
34 {
35   SketchPlugin_Sketch* aSketch = sketch();
36   if (aSketch) {
37     // compute a start point in 3D view
38     boost::shared_ptr<GeomDataAPI_Point2D> aStartAttr = 
39       boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(data()->attribute(LINE_ATTR_START));
40     boost::shared_ptr<GeomAPI_Pnt> aStart(aSketch->to3D(aStartAttr->x(), aStartAttr->y()));
41     // compute an end point in 3D view
42     boost::shared_ptr<GeomDataAPI_Point2D> anEndAttr = 
43       boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(data()->attribute(LINE_ATTR_END));
44     boost::shared_ptr<GeomAPI_Pnt> anEnd(aSketch->to3D(anEndAttr->x(), anEndAttr->y()));
45     // make linear edge
46     boost::shared_ptr<GeomAPI_Shape> anEdge = GeomAlgoAPI_EdgeBuilder::line(aStart, anEnd);
47     setPreview(anEdge);
48   }
49   return getPreview();
50 }