src/SketchPlugin/SketchPlugin_ConstraintDistance.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
   2
   3 // File:    SketchPlugin_ConstraintDistance.cpp
   4 // Created: 23 May 2014
   5 // Author:  Artem ZHIDKOV
   6
   7 #include "SketchPlugin_ConstraintDistance.h"
   8 #include <SketchPlugin_Point.h>
   9 #include <SketchPlugin_Circle.h>
  10 #include <SketchPlugin_Line.h>
  11
  12 #include <SketcherPrs_Tools.h>
  13 #include <SketcherPrs_Factory.h>
  14
  15 #include <GeomAPI_Dir2d.h>
  16 #include <GeomAPI_Lin2d.h>
  17 #include <GeomAPI_Pnt2d.h>
  18 #include <GeomAPI_XY.h>
  19 #include <GeomDataAPI_Point2D.h>
  20
  21 #include <ModelAPI_AttributeDouble.h>
  22 #include <ModelAPI_Data.h>
  23
  24 #include <Config_PropManager.h>
  25
  26 #include <math.h>
  27
  28 const double tolerance = 1e-7;
  29
  30
  31 SketchPlugin_ConstraintDistance::SketchPlugin_ConstraintDistance()
  32 {
  33   myFlyoutUpdate = false;
  34 }
  35
  36 //*************************************************************************************
  37 void SketchPlugin_ConstraintDistance::initAttributes()
  38 {
  39   data()->addAttribute(SketchPlugin_Constraint::VALUE(), ModelAPI_AttributeDouble::typeId());
  40   data()->addAttribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT(), GeomDataAPI_Point2D::typeId());
  41   data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::typeId());
  42   data()->addAttribute(SketchPlugin_Constraint::ENTITY_B(), ModelAPI_AttributeRefAttr::typeId());
  43 }
  44
  45 void SketchPlugin_ConstraintDistance::colorConfigInfo(std::string& theSection, std::string& theName,
  46                                                       std::string& theDefault)
  47 {
  48   theSection = "Visualization";
  49   theName = "sketch_dimension_color";
  50   theDefault = SKETCH_DIMENSION_COLOR;
  51 }
  52
  53 //*************************************************************************************
  54 void SketchPlugin_ConstraintDistance::execute()
  55 {
  56   std::shared_ptr<ModelAPI_Data> aData = data();
  57   AttributeDoublePtr anAttrValue = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(
  58       aData->attribute(SketchPlugin_Constraint::VALUE()));
  59
  60   if(anAttrValue->isInitialized())
  61     return;
  62
  63   double aDistance = calculateCurrentDistance();
  64   if(aDistance >= 0) {
  65     anAttrValue->setValue(aDistance);
  66   }
  67
  68   // the value should to be computed here, not in the getAISObject in order to change the model value
  69   // inside the object transaction. This is important for creating a constraint by preselection.
  70   // The display of the presentation in this case happens after the transaction commit
  71   std::shared_ptr<GeomDataAPI_Point2D> aFlyOutAttr = std::dynamic_pointer_cast<
  72       GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
  73   if(!aFlyOutAttr->isInitialized())
  74     compute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT());
  75 }
  76
  77 bool SketchPlugin_ConstraintDistance::compute(const std::string& theAttributeId)
  78 {
  79   if (theAttributeId != SketchPlugin_Constraint::FLYOUT_VALUE_PNT())
  80     return false;
  81
  82   if (!sketch())
  83     return false;
  84
  85   std::shared_ptr<GeomDataAPI_Point2D> aFlyOutAttr = std::dynamic_pointer_cast<
  86                            GeomDataAPI_Point2D>(attribute(theAttributeId));
  87   if (fabs(aFlyOutAttr->x()) >= tolerance || fabs(aFlyOutAttr->y()) >= tolerance)
  88     return false;
  89
  90   DataPtr aData = data();
  91   std::shared_ptr<GeomAPI_Ax3> aPlane = SketchPlugin_Sketch::plane(sketch());
  92   std::shared_ptr<GeomDataAPI_Point2D> aPoint_A = SketcherPrs_Tools::getFeaturePoint(
  93       aData, SketchPlugin_Constraint::ENTITY_A(), aPlane);
  94   std::shared_ptr<GeomDataAPI_Point2D> aPoint_B = SketcherPrs_Tools::getFeaturePoint(
  95       aData, SketchPlugin_Constraint::ENTITY_B(), aPlane);
  96
  97   std::shared_ptr<GeomAPI_Pnt2d> aPnt_A;
  98   std::shared_ptr<GeomAPI_Pnt2d> aPnt_B;
  99
 100   if (aPoint_A && aPoint_B) {
 101     aPnt_A = aPoint_A->pnt();
 102     aPnt_B = aPoint_B->pnt();
 103   } else if (!aPoint_A && aPoint_B) {
 104     FeaturePtr aLine = SketcherPrs_Tools::getFeatureLine(
 105         aData, SketchPlugin_Constraint::ENTITY_A());
 106     if (aLine) {
 107       aPnt_B = aPoint_B->pnt();
 108       aPnt_A = SketcherPrs_Tools::getProjectionPoint(aLine, aPnt_B);
 109     }
 110   } else if (aPoint_A && !aPoint_B) {
 111     FeaturePtr aLine = SketcherPrs_Tools::getFeatureLine(
 112         aData, SketchPlugin_Constraint::ENTITY_B());
 113     if (aLine) {
 114       aPnt_A = aPoint_A->pnt();
 115       aPnt_B = SketcherPrs_Tools::getProjectionPoint(aLine, aPnt_A);
 116     }
 117   }
 118   if (!aPnt_A || !aPnt_B)
 119     return false;
 120
 121   std::shared_ptr<GeomAPI_Pnt> aPoint1 = sketch()->to3D(aPnt_A->x(), aPnt_A->y());
 122   std::shared_ptr<GeomAPI_Pnt> aPoint2 = sketch()->to3D(aPnt_B->x(), aPnt_B->y());
 123   // it is not possible to create lin2d on the points with equal position
 124   if (aPoint1->distance(aPoint2) < tolerance)
 125     return false;
 126
 127   std::shared_ptr<GeomAPI_Lin2d> aLine = std::shared_ptr<GeomAPI_Lin2d>(new GeomAPI_Lin2d(aPnt_A, aPnt_B));
 128   double aDist = aPoint1->distance(aPoint2)/5.;
 129   std::shared_ptr<GeomAPI_Pnt2d> aFPnt = aLine->shiftedLocation(aDist);
 130   aFlyOutAttr->setValue(aFPnt);
 131
 132   return true;
 133 }
 134
 135 //*************************************************************************************
 136 AISObjectPtr SketchPlugin_ConstraintDistance::getAISObject(AISObjectPtr thePrevious)
 137 {
 138   if (!sketch())
 139     return thePrevious;
 140
 141   AISObjectPtr anAIS = SketcherPrs_Factory::lengthDimensionConstraint(this, sketch()->coordinatePlane(),
 142                                                                       thePrevious);
 143   return anAIS;
 144 }
 145
 146 //*************************************************************************************
 147 void SketchPlugin_ConstraintDistance::move(double theDeltaX, double theDeltaY)
 148 {
 149   std::shared_ptr<ModelAPI_Data> aData = data();
 150   if (!aData->isValid())
 151     return;
 152
 153   // Recalculate a shift of flyout point in terms of local coordinates
 154   std::shared_ptr<GeomAPI_XY> aDir(new GeomAPI_XY(theDeltaX, theDeltaY));
 155   std::shared_ptr<GeomAPI_Ax3> aPlane = SketchPlugin_Sketch::plane(sketch());
 156   std::shared_ptr<GeomDataAPI_Point2D> aPointA = SketcherPrs_Tools::getFeaturePoint(
 157       data(), SketchPlugin_Constraint::ENTITY_A(), aPlane);
 158   std::shared_ptr<GeomDataAPI_Point2D> aPointB = SketcherPrs_Tools::getFeaturePoint(
 159       data(), SketchPlugin_Constraint::ENTITY_B(), aPlane);
 160
 161   std::shared_ptr<GeomAPI_XY> aStartPnt;
 162   std::shared_ptr<GeomAPI_XY> aEndPnt;
 163   if (aPointA && aPointB) {
 164     aStartPnt = aPointA->pnt()->xy();
 165     aEndPnt = aPointB->pnt()->xy();
 166   } else if (aPointA) {
 167     FeaturePtr aLine = SketcherPrs_Tools::getFeatureLine(data(),
 168         SketchPlugin_Constraint::ENTITY_B());
 169     if (!aLine)
 170       return;
 171     std::shared_ptr<GeomAPI_Pnt2d> aPoint = aPointA->pnt();
 172     aStartPnt = aPoint->xy();
 173     aEndPnt = SketcherPrs_Tools::getProjectionPoint(aLine, aPoint)->xy();
 174   } else if (aPointB) {
 175     FeaturePtr aLine = SketcherPrs_Tools::getFeatureLine(data(),
 176         SketchPlugin_Constraint::ENTITY_A());
 177     if (!aLine)
 178       return;
 179     std::shared_ptr<GeomAPI_Pnt2d> aPoint = aPointB->pnt();
 180     aStartPnt = SketcherPrs_Tools::getProjectionPoint(aLine, aPoint)->xy();
 181     aEndPnt = aPoint->xy();
 182   } else
 183     return;
 184
 185   std::shared_ptr<GeomAPI_Dir2d> aLineDir(new GeomAPI_Dir2d(aEndPnt->decreased(aStartPnt)));
 186   double dX = aDir->dot(aLineDir->xy());
 187   double dY = -aDir->cross(aLineDir->xy());
 188
 189   std::shared_ptr<GeomDataAPI_Point2D> aPoint = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 190       aData->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
 191   myFlyoutUpdate = true;
 192   aPoint->setValue(aPoint->x() + dX, aPoint->y() + dY);
 193   myFlyoutUpdate = false;
 194 }
 195
 196 double SketchPlugin_ConstraintDistance::calculateCurrentDistance()
 197 {
 198   double aDistance = -1.;
 199
 200   std::shared_ptr<ModelAPI_Data> aData = data();
 201   std::shared_ptr<GeomAPI_Ax3> aPlane = SketchPlugin_Sketch::plane(sketch());
 202   std::shared_ptr<GeomDataAPI_Point2D> aPointA =
 203     SketcherPrs_Tools::getFeaturePoint(aData, SketchPlugin_Constraint::ENTITY_A(), aPlane);
 204   std::shared_ptr<GeomDataAPI_Point2D> aPointB =
 205       SketcherPrs_Tools::getFeaturePoint(aData, SketchPlugin_Constraint::ENTITY_B(), aPlane);
 206
 207   if (aPointA.get() && aPointB.get()) {  // both points
 208     aDistance = aPointA->pnt()->distance(aPointB->pnt());
 209   } else {
 210     FeaturePtr aLineFeature;
 211     std::shared_ptr<SketchPlugin_Line> aLine;
 212     if (!aPointA.get() && aPointB.get()) {  //Line and point
 213       aLineFeature = SketcherPrs_Tools::getFeatureLine(aData, SketchPlugin_Constraint::ENTITY_A());
 214       aLine = std::dynamic_pointer_cast<SketchPlugin_Line>(aLineFeature);
 215       if (aLine.get()) {
 216         aDistance = aLine->distanceToPoint(aPointB->pnt());
 217       }
 218     } else if (aPointA.get() && !aPointB.get()) {  // Point and line
 219       aLineFeature = SketcherPrs_Tools::getFeatureLine(aData, SketchPlugin_Constraint::ENTITY_B());
 220       aLine = std::dynamic_pointer_cast<SketchPlugin_Line>(aLineFeature);
 221       if (aLine.get()) {
 222         aDistance = aLine->distanceToPoint(aPointA->pnt());
 223       }
 224     }
 225   }
 226   return aDistance;
 227 }
 228
 229 void SketchPlugin_ConstraintDistance::attributeChanged(const std::string& theID)
 230 {
 231   if (theID == SketchPlugin_Constraint::ENTITY_A() ||
 232       theID == SketchPlugin_Constraint::ENTITY_B())
 233   {
 234     std::shared_ptr<ModelAPI_AttributeDouble> aValueAttr = std::dynamic_pointer_cast<
 235         ModelAPI_AttributeDouble>(data()->attribute(SketchPlugin_Constraint::VALUE()));
 236     if (!aValueAttr->isInitialized()) { // only if it is not initialized, try to compute the current value
 237       double aDistance = calculateCurrentDistance();
 238       if (aDistance > 0) { // set as value the length of updated references
 239         aValueAttr->setValue(aDistance);
 240       }
 241     }
 242   } else if (theID == SketchPlugin_Constraint::FLYOUT_VALUE_PNT() && !myFlyoutUpdate) {
 243     myFlyoutUpdate = true;
 244     // Recalculate flyout point in local coordinates of the distance constraint:
 245     // the X coordinate is a length of projection of the flyout point on the line binding two distanced points
 246     //                  or a line of projection of the distanced point onto the distanced segment
 247     // the Y coordinate is a distance from the flyout point to the line
 248     std::shared_ptr<GeomDataAPI_Point2D> aFlyoutAttr =
 249         std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 250         attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
 251     std::shared_ptr<GeomAPI_Pnt2d> aFlyoutPnt = aFlyoutAttr->pnt();
 252
 253     std::shared_ptr<GeomAPI_Ax3> aPlane = SketchPlugin_Sketch::plane(sketch());
 254     std::shared_ptr<GeomDataAPI_Point2D> aPointA = SketcherPrs_Tools::getFeaturePoint(
 255         data(), SketchPlugin_Constraint::ENTITY_A(), aPlane);
 256     std::shared_ptr<GeomDataAPI_Point2D> aPointB = SketcherPrs_Tools::getFeaturePoint(
 257         data(), SketchPlugin_Constraint::ENTITY_B(), aPlane);
 258
 259     std::shared_ptr<GeomAPI_XY> aStartPnt;
 260     std::shared_ptr<GeomAPI_XY> aEndPnt;
 261     if (aPointA && aPointB) {
 262       aStartPnt = aPointA->pnt()->xy();
 263       aEndPnt = aPointB->pnt()->xy();
 264     } else if (aPointA) {
 265       FeaturePtr aLine = SketcherPrs_Tools::getFeatureLine(data(),
 266           SketchPlugin_Constraint::ENTITY_B());
 267       if (!aLine)
 268         return;
 269       std::shared_ptr<GeomAPI_Pnt2d> aPoint = aPointA->pnt();
 270       aStartPnt = aPoint->xy();
 271       aEndPnt = SketcherPrs_Tools::getProjectionPoint(aLine, aPoint)->xy();
 272     } else if (aPointB) {
 273       FeaturePtr aLine = SketcherPrs_Tools::getFeatureLine(data(),
 274           SketchPlugin_Constraint::ENTITY_A());
 275       if (!aLine)
 276         return;
 277       std::shared_ptr<GeomAPI_Pnt2d> aPoint = aPointB->pnt();
 278       aStartPnt = SketcherPrs_Tools::getProjectionPoint(aLine, aPoint)->xy();
 279       aEndPnt = aPoint->xy();
 280     } else
 281       return;
 282
 283     if (aEndPnt->distance(aStartPnt) < tolerance)
 284       return;
 285
 286     std::shared_ptr<GeomAPI_Dir2d> aLineDir(new GeomAPI_Dir2d(aEndPnt->decreased(aStartPnt)));
 287     std::shared_ptr<GeomAPI_XY> aFlyoutDir = aFlyoutPnt->xy()->decreased(aStartPnt);
 288
 289     double X = aFlyoutDir->dot(aLineDir->xy());
 290     double Y = -aFlyoutDir->cross(aLineDir->xy());
 291     aFlyoutAttr->setValue(X, Y);
 292     myFlyoutUpdate = false;
 293   }
 294 }
 295