Salome HOME
Preferences improvement
[modules/shaper.git] / src / SketchPlugin / SketchPlugin_ConstraintPerpendicular.cpp
1 // File:    SketchPlugin_ConstraintPerpendicular.cpp
2 // Created: 26 May 2014
3 // Author:  Artem ZHIDKOV
4
5 #include "SketchPlugin_ConstraintPerpendicular.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 #include <Config_PropManager.h>
19
20 SketchPlugin_ConstraintPerpendicular::SketchPlugin_ConstraintPerpendicular()
21 {
22 }
23
24 void SketchPlugin_ConstraintPerpendicular::initAttributes()
25 {
26   data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::type());
27   data()->addAttribute(SketchPlugin_Constraint::ENTITY_B(), ModelAPI_AttributeRefAttr::type());
28   data()->addAttribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT(), GeomDataAPI_Point2D::type());
29 }
30
31 void SketchPlugin_ConstraintPerpendicular::execute()
32 {
33 }
34
35 boost::shared_ptr<GeomAPI_AISObject> SketchPlugin_ConstraintPerpendicular::getAISObject(
36                     boost::shared_ptr<GeomAPI_AISObject> thePrevious)
37 {
38   if (!sketch())
39     return thePrevious;
40
41   boost::shared_ptr<ModelAPI_Data> aData = data();
42   boost::shared_ptr<ModelAPI_AttributeRefAttr> anAttr1 = 
43     boost::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(aData->attribute(SketchPlugin_Constraint::ENTITY_A()));
44   boost::shared_ptr<ModelAPI_AttributeRefAttr> anAttr2 = 
45     boost::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(aData->attribute(SketchPlugin_Constraint::ENTITY_B()));
46   if (!anAttr1 || !anAttr1->isObject() || 
47       !anAttr2 || !anAttr2->isObject())
48     return thePrevious;
49
50   FeaturePtr aFeature = ModelAPI_Feature::feature(anAttr1->object());
51   if (!aFeature)
52     return thePrevious;
53   boost::shared_ptr<SketchPlugin_Line> aLine1Feature = 
54     boost::dynamic_pointer_cast<SketchPlugin_Line>(aFeature);
55
56   aFeature = ModelAPI_Feature::feature(anAttr2->object());
57   if (!aFeature)
58     return thePrevious;
59   boost::shared_ptr<SketchPlugin_Line> aLine2Feature = 
60     boost::dynamic_pointer_cast<SketchPlugin_Line>(aFeature);
61
62   if (!aLine1Feature || !aLine2Feature)
63     return thePrevious;
64
65   boost::shared_ptr<GeomAPI_Pln> aPlane = sketch()->plane();
66   boost::shared_ptr<GeomAPI_Shape> aLine1, aLine2;
67   boost::shared_ptr<ModelAPI_ResultConstruction> aConst1 = 
68     boost::dynamic_pointer_cast<ModelAPI_ResultConstruction>(anAttr1->object());
69   if (aConst1) aLine1 = aConst1->shape();
70   boost::shared_ptr<ModelAPI_ResultConstruction> aConst2 = 
71     boost::dynamic_pointer_cast<ModelAPI_ResultConstruction>(anAttr2->object());
72   if (aConst2) aLine2 = aConst2->shape();
73
74   boost::shared_ptr<GeomAPI_AISObject> anAIS = thePrevious;
75   if (!anAIS)
76     anAIS = boost::shared_ptr<GeomAPI_AISObject>(new GeomAPI_AISObject);
77   anAIS->createPerpendicular(aLine1, aLine2, aPlane);
78
79   // Set color from preferences
80   std::vector<int> aRGB = Config_PropManager::color("Visualization", "perpendicular_color", PERPENDICULAR_COLOR);
81   anAIS->setColor(aRGB[0], aRGB[1], aRGB[2]);
82   return anAIS;
83 }
84
85 void SketchPlugin_ConstraintPerpendicular::move(double theDeltaX, double theDeltaY)
86 {
87   boost::shared_ptr<ModelAPI_Data> aData = data();
88   if (!aData->isValid())
89     return;
90
91   boost::shared_ptr<GeomDataAPI_Point2D> aPoint =
92         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
93   aPoint->setValue(aPoint->x() + theDeltaX, aPoint->y() + theDeltaY);
94 }
95