Salome HOME
Update unit-tests for SketchPlugin (part III)
[modules/shaper.git] / src / SketchPlugin / SketchPlugin_Point.cpp
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
2
3 // File:    SketchPlugin_Point.cpp
4 // Created: 07 May 2014
5 // Author:  Artem ZHIDKOV
6
7 #include "SketchPlugin_Point.h"
8 #include "SketchPlugin_Sketch.h"
9
10 #include <ModelAPI_Data.h>
11 #include <ModelAPI_ResultConstruction.h>
12 #include <ModelAPI_AttributeSelection.h>
13 #include <ModelAPI_Validator.h>
14 #include <ModelAPI_Session.h>
15
16 #include <GeomAPI_Pnt2d.h>
17 #include <GeomAPI_Vertex.h>
18 #include <GeomDataAPI_Point2D.h>
19 #include <GeomAlgoAPI_PointBuilder.h>
20
21 using namespace std;
22
23 SketchPlugin_Point::SketchPlugin_Point()
24     : SketchPlugin_SketchEntity()
25 {
26 }
27
28 void SketchPlugin_Point::initDerivedClassAttributes()
29 {
30   data()->addAttribute(SketchPlugin_Point::COORD_ID(), GeomDataAPI_Point2D::typeId());
31   data()->addAttribute(EXTERNAL_ID(), ModelAPI_AttributeSelection::typeId());
32   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), EXTERNAL_ID());
33 }
34
35 void SketchPlugin_Point::execute()
36 {
37   SketchPlugin_Sketch* aSketch = sketch();
38   if (aSketch) {
39     // compute a point in 3D view
40     std::shared_ptr<GeomDataAPI_Point2D> aPoint =
41         std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
42             data()->attribute(SketchPlugin_Point::COORD_ID()));
43     std::shared_ptr<GeomAPI_Pnt> aPoint3D(aSketch->to3D(aPoint->x(), aPoint->y()));
44     // make a visible point
45     std::shared_ptr<GeomAPI_Shape> aPointShape = GeomAlgoAPI_PointBuilder::point(aPoint3D);
46     std::shared_ptr<ModelAPI_ResultConstruction> aConstr = document()->createConstruction(data());
47     aConstr->setShape(aPointShape);
48     aConstr->setIsInHistory(false);
49     setResult(aConstr);
50   }
51 }
52
53 void SketchPlugin_Point::move(double theDeltaX, double theDeltaY)
54 {
55   std::shared_ptr<ModelAPI_Data> aData = data();
56   if (!aData->isValid())
57     return;
58
59   std::shared_ptr<GeomDataAPI_Point2D> aPoint1 = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
60       aData->attribute(SketchPlugin_Point::COORD_ID()));
61   aPoint1->move(theDeltaX, theDeltaY);
62 }
63
64 bool SketchPlugin_Point::isFixed() {
65   return data()->selection(EXTERNAL_ID())->context().get() != NULL;
66 }
67
68 void SketchPlugin_Point::attributeChanged(const std::string& theID) {
69   // the second condition for unability to move external point anywhere
70   if (theID == EXTERNAL_ID() || isFixed()) {
71     std::shared_ptr<GeomAPI_Shape> aSelection = data()->selection(EXTERNAL_ID())->value();
72      // update arguments due to the selection value
73     if (aSelection && !aSelection->isNull() && aSelection->isVertex()) {
74       std::shared_ptr<GeomAPI_Vertex> aVertex(new GeomAPI_Vertex(aSelection));
75       std::shared_ptr<GeomDataAPI_Point2D> aCoordAttr = 
76         std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(COORD_ID()));
77       aCoordAttr->setValue(sketch()->to2D(aVertex->point()));
78     }
79   }
80 }