src/SketchPlugin/SketchPlugin_ConstraintLength.cpp

   1 // File:    SketchPlugin_ConstraintLength.cpp
   2 // Created: 30 May 2014
   3 // Author:  Artem ZHIDKOV
   4
   5 #include "SketchPlugin_ConstraintLength.h"
   6 #include <SketchPlugin_Line.h>
   7
   8 #include <GeomDataAPI_Point2D.h>
   9
  10 #include <ModelAPI_AttributeDouble.h>
  11 #include <ModelAPI_Data.h>
  12 #include <ModelAPI_Result.h>
  13
  14 #include <GeomDataAPI_Point2D.h>
  15
  16 #include <GeomAPI_Lin2d.h>
  17 #include <GeomAPI_Pnt2d.h>
  18
  19 #include <Config_PropManager.h>
  20
  21 SketchPlugin_ConstraintLength::SketchPlugin_ConstraintLength()
  22 {
  23 }
  24
  25 void SketchPlugin_ConstraintLength::initAttributes()
  26 {
  27   data()->addAttribute(SketchPlugin_Constraint::VALUE(), ModelAPI_AttributeDouble::type());
  28   data()->addAttribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT(), GeomDataAPI_Point2D::type());
  29   data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::type());
  30 }
  31
  32 void SketchPlugin_ConstraintLength::execute()
  33 {
  34   boost::shared_ptr<ModelAPI_AttributeRefAttr> aRef = boost::dynamic_pointer_cast<
  35       ModelAPI_AttributeRefAttr>(data()->attribute(SketchPlugin_Constraint::ENTITY_A()));
  36   FeaturePtr aFeature = ModelAPI_Feature::feature(aRef->object());
  37   if (aFeature) {
  38     // set length value
  39     boost::shared_ptr<GeomDataAPI_Point2D> aPoint1 = boost::dynamic_pointer_cast<
  40         GeomDataAPI_Point2D>(aFeature->data()->attribute(SketchPlugin_Line::START_ID()));
  41     boost::shared_ptr<GeomDataAPI_Point2D> aPoint2 = boost::dynamic_pointer_cast<
  42         GeomDataAPI_Point2D>(aFeature->data()->attribute(SketchPlugin_Line::END_ID()));
  43
  44     double aLenght = aPoint1->pnt()->distance(aPoint2->pnt());
  45
  46     boost::shared_ptr<ModelAPI_AttributeDouble> aValueAttr = boost::dynamic_pointer_cast<
  47         ModelAPI_AttributeDouble>(data()->attribute(SketchPlugin_Constraint::VALUE()));
  48     if(!aValueAttr->isInitialized()) {
  49       aValueAttr->setValue(aLenght);
  50     }
  51   }
  52 }
  53
  54 AISObjectPtr SketchPlugin_ConstraintLength::getAISObject(AISObjectPtr thePrevious)
  55 {
  56   if (!sketch())
  57     return thePrevious;
  58
  59   boost::shared_ptr<GeomAPI_Pln> aPlane = sketch()->plane();
  60
  61   boost::shared_ptr<ModelAPI_AttributeRefAttr> anAttr = boost::dynamic_pointer_cast<
  62       ModelAPI_AttributeRefAttr>(data()->attribute(SketchPlugin_Constraint::ENTITY_A()));
  63   if (!anAttr)
  64     return thePrevious;
  65   FeaturePtr aFeature = ModelAPI_Feature::feature(anAttr->object());
  66   if (!aFeature || aFeature->getKind() != SketchPlugin_Line::ID())
  67     return thePrevious;
  68
  69   boost::shared_ptr<GeomDataAPI_Point2D> aFlyOutAttr = boost::dynamic_pointer_cast<
  70       GeomDataAPI_Point2D>(data()->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
  71
  72   DataPtr aData = aFeature->data();
  73   boost::shared_ptr<GeomDataAPI_Point2D> aStartPoint = boost::dynamic_pointer_cast<
  74       GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Line::START_ID()));
  75   boost::shared_ptr<GeomDataAPI_Point2D> anEndPoint = boost::dynamic_pointer_cast<
  76       GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Line::END_ID()));
  77
  78   boost::shared_ptr<GeomAPI_Pnt> aPoint1 = sketch()->to3D(aStartPoint->x(), aStartPoint->y());
  79   boost::shared_ptr<GeomAPI_Pnt> aPoint2 = sketch()->to3D(anEndPoint->x(), anEndPoint->y());
  80   boost::shared_ptr<GeomAPI_Pnt> aFlyoutPnt =
  81       aFlyOutAttr->isInitialized() ?
  82           sketch()->to3D(aFlyOutAttr->x(), aFlyOutAttr->y()) : boost::shared_ptr<GeomAPI_Pnt>();
  83
  84   // value calculation
  85   boost::shared_ptr<ModelAPI_AttributeDouble> aValueAttr = boost::dynamic_pointer_cast<
  86       ModelAPI_AttributeDouble>(data()->attribute(SketchPlugin_Constraint::VALUE()));
  87   double aValue = aValueAttr->value();
  88
  89   AISObjectPtr anAIS = thePrevious;
  90   if (!anAIS)
  91     anAIS = AISObjectPtr(new GeomAPI_AISObject);
  92   anAIS->createDistance(aPoint1, aPoint2, aFlyoutPnt, aPlane, aValue);
  93
  94   // Set color from preferences
  95   std::vector<int> aRGB = Config_PropManager::color("Visualization", "length_color", LENGTH_COLOR);
  96   anAIS->setColor(aRGB[0], aRGB[1], aRGB[2]);
  97   return anAIS;
  98 }
  99
 100 void SketchPlugin_ConstraintLength::move(double theDeltaX, double theDeltaY)
 101 {
 102   boost::shared_ptr<ModelAPI_Data> aData = data();
 103   if (!aData->isValid())
 104     return;
 105
 106   boost::shared_ptr<GeomDataAPI_Point2D> aPoint1 = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 107       aData->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
 108   aPoint1->setValue(aPoint1->x() + theDeltaX, aPoint1->y() + theDeltaY);
 109 }
 110