1 // File: SketchPlugin_ConstraintLength.cpp
2 // Created: 30 May 2014
3 // Author: Artem ZHIDKOV
5 #include "SketchPlugin_ConstraintLength.h"
6 #include <SketchPlugin_Line.h>
8 #include <GeomDataAPI_Point2D.h>
10 #include <ModelAPI_AttributeDouble.h>
11 #include <ModelAPI_Data.h>
12 #include <ModelAPI_Result.h>
14 #include <GeomDataAPI_Point2D.h>
16 #include <GeomAPI_Lin2d.h>
17 #include <GeomAPI_Pnt2d.h>
19 #include <Config_PropManager.h>
22 SketchPlugin_ConstraintLength::SketchPlugin_ConstraintLength()
26 void SketchPlugin_ConstraintLength::initAttributes()
28 data()->addAttribute(SketchPlugin_Constraint::VALUE(), ModelAPI_AttributeDouble::type());
29 data()->addAttribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT(), GeomDataAPI_Point2D::type());
30 data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::type());
33 void SketchPlugin_ConstraintLength::execute()
35 if (data()->attribute(SketchPlugin_Constraint::ENTITY_A())->isInitialized() &&
36 !data()->attribute(SketchPlugin_Constraint::VALUE())->isInitialized()) {
38 boost::shared_ptr<ModelAPI_AttributeRefAttr> aRef =
39 boost::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(data()->attribute(SketchPlugin_Constraint::ENTITY_A()));
40 FeaturePtr aFeature = ModelAPI_Feature::feature(aRef->object());
43 boost::shared_ptr<GeomDataAPI_Point2D> aPoint1 =
44 boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aFeature->data()->attribute(SketchPlugin_Line::START_ID()));
45 boost::shared_ptr<GeomDataAPI_Point2D> aPoint2 =
46 boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aFeature->data()->attribute(SketchPlugin_Line::END_ID()));
48 double aLenght = aPoint1->pnt()->distance(aPoint2->pnt());
50 boost::shared_ptr<ModelAPI_AttributeDouble> aValueAttr =
51 boost::dynamic_pointer_cast<ModelAPI_AttributeDouble>(data()->attribute(SketchPlugin_Constraint::VALUE()));
52 aValueAttr->setValue(aLenght);
57 boost::shared_ptr<GeomAPI_AISObject> SketchPlugin_ConstraintLength::getAISObject(
58 boost::shared_ptr<GeomAPI_AISObject> thePrevious)
63 boost::shared_ptr<GeomAPI_Pln> aPlane = sketch()->plane();
65 boost::shared_ptr<ModelAPI_AttributeRefAttr> anAttr =
66 boost::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(data()->attribute(SketchPlugin_Constraint::ENTITY_A()));
69 FeaturePtr aFeature = ModelAPI_Feature::feature(anAttr->object());
70 if (!aFeature || aFeature->getKind() != SketchPlugin_Line::ID())
73 boost::shared_ptr<GeomDataAPI_Point2D> aFlyOutAttr =
74 boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(data()->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
76 DataPtr aData = aFeature->data();
77 boost::shared_ptr<GeomDataAPI_Point2D> aStartPoint =
78 boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Line::START_ID()));
79 boost::shared_ptr<GeomDataAPI_Point2D> anEndPoint =
80 boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Line::END_ID()));
82 boost::shared_ptr<GeomAPI_Pnt> aPoint1 = sketch()->to3D(aStartPoint->x(), aStartPoint->y());
83 boost::shared_ptr<GeomAPI_Pnt> aPoint2 = sketch()->to3D(anEndPoint->x(), anEndPoint->y());
84 boost::shared_ptr<GeomAPI_Pnt> aFlyoutPnt = aFlyOutAttr->isInitialized() ?
85 sketch()->to3D(aFlyOutAttr->x(), aFlyOutAttr->y()) :
86 boost::shared_ptr<GeomAPI_Pnt>();
89 boost::shared_ptr<ModelAPI_AttributeDouble> aValueAttr =
90 boost::dynamic_pointer_cast<ModelAPI_AttributeDouble>(data()->attribute(SketchPlugin_Constraint::VALUE()));
91 double aValue = aValueAttr->value();
93 boost::shared_ptr<GeomAPI_AISObject> anAIS = thePrevious;
95 anAIS = boost::shared_ptr<GeomAPI_AISObject>(new GeomAPI_AISObject);
96 anAIS->createDistance(aPoint1, aPoint2, aFlyoutPnt, aPlane, aValue);
98 // Set color from preferences
99 std::vector<int> aRGB = Config_PropManager::color("Visualization", "length_color", LENGTH_COLOR);
100 anAIS->setColor(aRGB[0], aRGB[1], aRGB[2]);
104 void SketchPlugin_ConstraintLength::move(double theDeltaX, double theDeltaY)
106 boost::shared_ptr<ModelAPI_Data> aData = data();
107 if (!aData->isValid())
110 boost::shared_ptr<GeomDataAPI_Point2D> aPoint1 =
111 boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
112 aPoint1->setValue(aPoint1->x() + theDeltaX, aPoint1->y() + theDeltaY);