Salome HOME
Working with Construction results in SketchPlugin
[modules/shaper.git] / src / SketchPlugin / SketchPlugin_ConstraintParallel.cpp
1 // File:    SketchPlugin_ConstraintParallel.cpp
2 // Created: 26 May 2014
3 // Author:  Artem ZHIDKOV
4
5 #include "SketchPlugin_ConstraintParallel.h"
6
7 #include <ModelAPI_AttributeDouble.h>
8 #include <ModelAPI_Data.h>
9 #include <ModelAPI_ResultConstruction.h>
10
11 #include <SketchPlugin_Line.h>
12 #include <SketchPlugin_Sketch.h>
13
14 #include <GeomDataAPI_Point2D.h>
15 #include <GeomAPI_Pnt2d.h>
16 #include <GeomAPI_Pnt.h>
17
18 SketchPlugin_ConstraintParallel::SketchPlugin_ConstraintParallel()
19 {
20 }
21
22 void SketchPlugin_ConstraintParallel::initAttributes()
23 {
24   data()->addAttribute(CONSTRAINT_ATTR_ENTITY_A, ModelAPI_AttributeRefAttr::type());
25   data()->addAttribute(CONSTRAINT_ATTR_ENTITY_B, ModelAPI_AttributeRefAttr::type());
26   data()->addAttribute(CONSTRAINT_ATTR_FLYOUT_VALUE_PNT, GeomDataAPI_Point2D::type());
27 }
28
29 void SketchPlugin_ConstraintParallel::execute()
30 {
31 }
32
33 boost::shared_ptr<GeomAPI_AISObject> SketchPlugin_ConstraintParallel::getAISObject(
34                     boost::shared_ptr<GeomAPI_AISObject> thePrevious)
35 {
36   if (!sketch())
37     return thePrevious;
38
39   boost::shared_ptr<ModelAPI_Data> aData = data();
40   boost::shared_ptr<ModelAPI_AttributeRefAttr> anAttr1 = 
41     boost::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(aData->attribute(CONSTRAINT_ATTR_ENTITY_A));
42   boost::shared_ptr<ModelAPI_AttributeRefAttr> anAttr2 = 
43     boost::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(aData->attribute(CONSTRAINT_ATTR_ENTITY_B));
44   if (!anAttr1 || !anAttr1->isFeature() || 
45       !anAttr2 || !anAttr2->isFeature())
46     return thePrevious;
47   boost::shared_ptr<SketchPlugin_Line> aLine1Feature = 
48     boost::dynamic_pointer_cast<SketchPlugin_Line>(anAttr1->feature());
49   boost::shared_ptr<SketchPlugin_Line> aLine2Feature = 
50     boost::dynamic_pointer_cast<SketchPlugin_Line>(anAttr2->feature());
51   if (!aLine1Feature || !aLine2Feature)
52     return thePrevious;
53
54   boost::shared_ptr<GeomAPI_Pln> aPlane = sketch()->plane();
55   boost::shared_ptr<GeomAPI_Shape> aLine1, aLine2;
56   boost::shared_ptr<ModelAPI_ResultConstruction> aConst1 = 
57     boost::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aLine1Feature->firstResult());
58   if (aConst1) aLine1 = aConst1->shape();
59   boost::shared_ptr<ModelAPI_ResultConstruction> aConst2 = 
60     boost::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aLine1Feature->firstResult());
61   if (aConst2) aLine2 = aConst2->shape();
62
63   boost::shared_ptr<GeomDataAPI_Point2D> aFlyoutAttr = 
64     boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(CONSTRAINT_ATTR_FLYOUT_VALUE_PNT));
65   boost::shared_ptr<GeomAPI_Pnt> aFlyoutPnt = sketch()->to3D(aFlyoutAttr->x(), aFlyoutAttr->y());
66
67   boost::shared_ptr<GeomAPI_AISObject> anAIS = thePrevious;
68   if (!anAIS)
69     anAIS = boost::shared_ptr<GeomAPI_AISObject>(new GeomAPI_AISObject);
70   anAIS->createParallel(aLine1, aLine2, aFlyoutPnt, aPlane);
71   return anAIS;
72 }
73
74 void SketchPlugin_ConstraintParallel::move(double theDeltaX, double theDeltaY)
75 {
76   boost::shared_ptr<ModelAPI_Data> aData = data();
77   if (!aData->isValid())
78     return;
79
80   boost::shared_ptr<GeomDataAPI_Point2D> aPoint =
81         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(CONSTRAINT_ATTR_FLYOUT_VALUE_PNT));
82   aPoint->setValue(aPoint->x() + theDeltaX, aPoint->y() + theDeltaY);
83 }
84