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 }