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