]> SALOME platform Git repositories - modules/shaper.git/blob - src/ConstructionPlugin/ConstructionPlugin_Axis.cpp
Salome HOME
Merge branch 'Dev_0.6.1' of newgeom:newgeom into Dev_0.6.1
[modules/shaper.git] / src / ConstructionPlugin / ConstructionPlugin_Axis.cpp
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
2
3 // File:        ConstructionPlugin_Axis.cpp
4 // Created:     12 Dec 2014
5 // Author:      Vitaly Smetannikov
6
7 #include "ConstructionPlugin_Axis.h"
8
9 #include <ModelAPI_AttributeSelection.h>
10 #include <ModelAPI_ResultConstruction.h>
11
12 #include <GeomAPI_Edge.h>
13 #include <GeomAlgoAPI_EdgeBuilder.h>
14 #include <GeomAlgoAPI_PointBuilder.h>
15
16 using namespace std;
17
18 static const double MINIMAL_LENGTH = 1.e-5;
19
20 ConstructionPlugin_Axis::ConstructionPlugin_Axis()
21 {
22 }
23
24 void ConstructionPlugin_Axis::initAttributes()
25 {
26   data()->addAttribute(POINT_ATTR_FIRST,  ModelAPI_AttributeSelection::type());
27   data()->addAttribute(POINT_ATTR_SECOND, ModelAPI_AttributeSelection::type());
28 }
29
30 void ConstructionPlugin_Axis::execute()
31 {
32   AttributeSelectionPtr aRef1 = data()->selection(POINT_ATTR_FIRST);
33   AttributeSelectionPtr aRef2 = data()->selection(POINT_ATTR_SECOND);
34   if ((aRef1.get() != NULL) && (aRef2.get() != NULL)) {
35     GeomShapePtr aShape1 = aRef1->value();
36     GeomShapePtr aShape2 = aRef2->value();
37     if (aShape1->isVertex() && aShape2->isVertex() && (!aShape1->isEqual(aShape2))) {
38       std::shared_ptr<GeomAPI_Pnt> aStart = GeomAlgoAPI_PointBuilder::point(aShape1);
39       std::shared_ptr<GeomAPI_Pnt> anEnd = GeomAlgoAPI_PointBuilder::point(aShape2);
40       if (aStart->distance(anEnd) > MINIMAL_LENGTH) {
41         std::shared_ptr<GeomAPI_Edge> anEdge = GeomAlgoAPI_EdgeBuilder::line(aStart, anEnd);
42
43         ResultConstructionPtr aConstr = document()->createConstruction(data());
44         aConstr->setShape(anEdge);
45         setResult(aConstr);
46       }
47     }
48   }
49 }
50
51 void ConstructionPlugin_Axis::customisePresentation(AISObjectPtr thePrs)
52 {
53   thePrs->setColor(0, 0, 0);
54   thePrs->setLineStyle(3);
55 }