src/PartSet/PartSet_WidgetPoint2d.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:        PartSet_WidgetPoint2D.cpp
   4 // Created:     25 Apr 2014
   5 // Author:      Natalia ERMOLAEVA
   6
   7 #include "PartSet_WidgetPoint2d.h"
   8 #include <PartSet_Tools.h>
   9 #include <PartSet_Module.h>
  10
  11 #include <ModuleBase_ParamSpinBox.h>
  12 #include <ModuleBase_Tools.h>
  13 #include <ModuleBase_IViewer.h>
  14 #include <ModuleBase_IViewWindow.h>
  15 #include <ModuleBase_ISelection.h>
  16
  17 #include <Config_Keywords.h>
  18 #include <Config_WidgetAPI.h>
  19
  20 #include <Events_Loop.h>
  21 #include <ModelAPI_Events.h>
  22
  23 #include <ModelAPI_Feature.h>
  24 #include <ModelAPI_Data.h>
  25 #include <ModelAPI_Object.h>
  26 #include <GeomDataAPI_Point2D.h>
  27 #include <GeomAPI_Pnt2d.h>
  28
  29 #include <SketchPlugin_Feature.h>
  30 #include <SketchPlugin_ConstraintCoincidence.h>
  31 #include <SketchPlugin_Line.h>
  32 #include <SketchPlugin_Arc.h>
  33 #include <SketchPlugin_Circle.h>
  34 #include <SketchPlugin_Point.h>
  35
  36 #include <QGroupBox>
  37 #include <QGridLayout>
  38 #include <QLabel>
  39 #include <QEvent>
  40 #include <QMouseEvent>
  41 #include <QApplication>
  42
  43 #include <TopoDS.hxx>
  44 #include <TopoDS_Vertex.hxx>
  45 #include <BRep_Tool.hxx>
  46
  47 #include <cfloat>
  48 #include <climits>
  49
  50 const double MaxCoordinate = 1e12;
  51
  52 static QStringList MyFeaturesForCoincedence;
  53
  54 PartSet_WidgetPoint2D::PartSet_WidgetPoint2D(QWidget* theParent,
  55                                              ModuleBase_IWorkshop* theWorkshop,
  56                                              const Config_WidgetAPI* theData,
  57                                              const std::string& theParentId)
  58  : ModuleBase_ModelWidget(theParent, theData, theParentId), myWorkshop(theWorkshop)
  59 {
  60   if (MyFeaturesForCoincedence.isEmpty()) {
  61     MyFeaturesForCoincedence << SketchPlugin_Line::ID().c_str()
  62       << SketchPlugin_Arc::ID().c_str()
  63       << SketchPlugin_Point::ID().c_str()
  64       << SketchPlugin_Circle::ID().c_str();
  65   }
  66
  67   // the control should accept the focus, so the boolen flag is corrected to be true
  68   myIsObligatory = true;
  69   //myOptionParam = theData->getProperty(PREVIOUS_FEATURE_PARAM);
  70   QString aPageName = QString::fromStdString(theData->getProperty(CONTAINER_PAGE_NAME));
  71   myGroupBox = new QGroupBox(aPageName, theParent);
  72   myGroupBox->setFlat(false);
  73
  74   QGridLayout* aGroupLay = new QGridLayout(myGroupBox);
  75   ModuleBase_Tools::adjustMargins(aGroupLay);
  76   aGroupLay->setSpacing(2);
  77   aGroupLay->setColumnStretch(1, 1);
  78   {
  79     QLabel* aLabel = new QLabel(myGroupBox);
  80     aLabel->setText(tr("X "));
  81     aGroupLay->addWidget(aLabel, 0, 0);
  82
  83     myXSpin = new ModuleBase_ParamSpinBox(myGroupBox);
  84     myXSpin->setMinimum(-DBL_MAX);
  85     myXSpin->setMaximum(DBL_MAX);
  86     myXSpin->setToolTip(tr("X"));
  87     aGroupLay->addWidget(myXSpin, 0, 1);
  88
  89     connect(myXSpin, SIGNAL(valueChanged(const QString&)), this, SIGNAL(valuesModified()));
  90   }
  91   {
  92     QLabel* aLabel = new QLabel(myGroupBox);
  93     aLabel->setText(tr("Y "));
  94     aGroupLay->addWidget(aLabel, 1, 0);
  95
  96     myYSpin = new ModuleBase_ParamSpinBox(myGroupBox);
  97     myYSpin->setMinimum(-DBL_MAX);
  98     myYSpin->setMaximum(DBL_MAX);
  99     myYSpin->setToolTip(tr("Y"));
 100     aGroupLay->addWidget(myYSpin, 1, 1);
 101
 102     connect(myYSpin, SIGNAL(valueChanged(const QString&)), this, SIGNAL(valuesModified()));
 103   }
 104   QVBoxLayout* aLayout = new QVBoxLayout(this);
 105   ModuleBase_Tools::zeroMargins(aLayout);
 106   aLayout->addWidget(myGroupBox);
 107   setLayout(aLayout);
 108 }
 109
 110 bool PartSet_WidgetPoint2D::resetCustom()
 111 {
 112   bool aDone = false;
 113   if (!isUseReset() || isComputedDefault() || myXSpin->hasVariable() || myYSpin->hasVariable()) {
 114     aDone = false;
 115   }
 116   else {
 117     bool isOk;
 118     double aDefValue = QString::fromStdString(getDefaultValue()).toDouble(&isOk);
 119     // it is important to block the spin box control in order to do not through out the
 120     // locking of the validating state.
 121     ModuleBase_Tools::setSpinValue(myXSpin, isOk ? aDefValue : 0.0);
 122     ModuleBase_Tools::setSpinValue(myYSpin, isOk ? aDefValue : 0.0);
 123     storeValueCustom();
 124     aDone = true;
 125   }
 126   return aDone;
 127 }
 128
 129 PartSet_WidgetPoint2D::~PartSet_WidgetPoint2D()
 130 {
 131 }
 132
 133 bool PartSet_WidgetPoint2D::setSelection(QList<ModuleBase_ViewerPrs>& theValues,
 134                                          const bool theToValidate)
 135 {
 136   bool isDone = false;
 137   if (theValues.empty())
 138     return isDone;
 139
 140   ModuleBase_ViewerPrs aValue = theValues.takeFirst();
 141   TopoDS_Shape aShape = aValue.shape();
 142   if (!aShape.IsNull()) {
 143     Handle(V3d_View) aView = myWorkshop->viewer()->activeView();
 144     double aX, aY;
 145     if (getPoint2d(aView, aShape, aX, aY)) {
 146       isDone = setPoint(aX, aY);
 147     }
 148   }
 149   else if (canBeActivatedByMove()) {
 150     if (feature()->getKind() == SketchPlugin_Line::ID()) {
 151       FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aValue.object());
 152       // Initialise new line with first point equal to end of previous
 153       if (aFeature.get()) {
 154         std::shared_ptr<ModelAPI_Data> aData = aFeature->data();
 155         std::shared_ptr<GeomDataAPI_Point2D> aPoint =
 156           std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 157                                        aData->attribute(SketchPlugin_Line::END_ID()));
 158         if (aPoint) {
 159           setPoint(aPoint->x(), aPoint->y());
 160           PartSet_Tools::setConstraints(mySketch, feature(), attributeID(), aPoint->x(),
 161                                         aPoint->y());
 162           isDone = true;
 163         }
 164       }
 165     }
 166   }
 167   return isDone;
 168 }
 169
 170 void PartSet_WidgetPoint2D::selectContent()
 171 {
 172   myXSpin->selectAll();
 173 }
 174
 175 bool PartSet_WidgetPoint2D::setPoint(double theX, double theY)
 176 {
 177   if (fabs(theX) >= MaxCoordinate)
 178     return false;
 179   if (fabs(theY) >= MaxCoordinate)
 180     return false;
 181
 182   ModuleBase_Tools::setSpinValue(myXSpin, theX);
 183   ModuleBase_Tools::setSpinValue(myYSpin, theY);
 184
 185   storeValue();
 186   return true;
 187 }
 188
 189 bool PartSet_WidgetPoint2D::storeValueCustom() const
 190 {
 191   std::shared_ptr<ModelAPI_Data> aData = myFeature->data();
 192   if (!aData) // can be on abort of sketcher element
 193     return false;
 194   std::shared_ptr<GeomDataAPI_Point2D> aPoint = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 195       aData->attribute(attributeID()));
 196
 197   PartSet_WidgetPoint2D* that = (PartSet_WidgetPoint2D*) this;
 198   bool isBlocked = that->blockSignals(true);
 199   bool isImmutable = aPoint->setImmutable(true);
 200
 201   // if text is not empty then setValue will be ignored
 202   // so we should set the text at first
 203   aPoint->setText(myXSpin->hasVariable() ? myXSpin->text().toStdString() : "",
 204                   myYSpin->hasVariable() ? myYSpin->text().toStdString() : "");
 205   aPoint->setValue(!myXSpin->hasVariable() ? myXSpin->value() : aPoint->x(),
 206                    !myYSpin->hasVariable() ? myYSpin->value() : aPoint->y());
 207
 208   // after movement the solver will call the update event: optimization
 209   moveObject(myFeature);
 210   aPoint->setImmutable(isImmutable);
 211   that->blockSignals(isBlocked);
 212
 213   return true;
 214 }
 215
 216 bool PartSet_WidgetPoint2D::restoreValueCustom()
 217 {
 218   std::shared_ptr<ModelAPI_Data> aData = myFeature->data();
 219   std::shared_ptr<GeomDataAPI_Point2D> aPoint = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 220       aData->attribute(attributeID()));
 221   QString aTextX = QString::fromStdString(aPoint->textX());
 222   QString aTextY = QString::fromStdString(aPoint->textY());
 223
 224   bool isDouble = false;
 225   double aVal = 0;
 226   if (aTextX.isEmpty()) {
 227     ModuleBase_Tools::setSpinValue(myXSpin, aPoint->x());
 228   } else {
 229     aVal = aTextX.toDouble(&isDouble);
 230     if (isDouble)
 231       ModuleBase_Tools::setSpinValue(myXSpin, aVal);
 232     else
 233       ModuleBase_Tools::setSpinText(myXSpin, aTextX);
 234   }
 235   if (aTextY.isEmpty()) {
 236     ModuleBase_Tools::setSpinValue(myYSpin, aPoint->y());
 237   } else {
 238     aVal = aTextY.toDouble(&isDouble);
 239     if (isDouble)
 240       ModuleBase_Tools::setSpinValue(myYSpin, aVal);
 241     else
 242       ModuleBase_Tools::setSpinText(myYSpin, aTextY);
 243   }
 244   //if (aTextX.empty() || aTextY.empty()) {
 245   //  ModuleBase_Tools::setSpinValue(myXSpin, aPoint->x());
 246   //  ModuleBase_Tools::setSpinValue(myYSpin, aPoint->y());
 247   //} else {
 248   //  ModuleBase_Tools::setSpinText(myXSpin, QString::fromStdString(aTextX));
 249   //  ModuleBase_Tools::setSpinText(myYSpin, QString::fromStdString(aTextY));
 250   //}
 251   return true;
 252 }
 253
 254 QList<QWidget*> PartSet_WidgetPoint2D::getControls() const
 255 {
 256   QList<QWidget*> aControls;
 257   aControls.append(myXSpin);
 258   aControls.append(myYSpin);
 259   return aControls;
 260 }
 261
 262
 263 void PartSet_WidgetPoint2D::activateCustom()
 264 {
 265   ModuleBase_IViewer* aViewer = myWorkshop->viewer();
 266   connect(aViewer, SIGNAL(mouseMove(ModuleBase_IViewWindow*, QMouseEvent*)),
 267           this, SLOT(onMouseMove(ModuleBase_IViewWindow*, QMouseEvent*)));
 268   connect(aViewer, SIGNAL(mouseRelease(ModuleBase_IViewWindow*, QMouseEvent*)),
 269           this, SLOT(onMouseRelease(ModuleBase_IViewWindow*, QMouseEvent*)));
 270
 271   QIntList aModes;
 272   aModes << TopAbs_VERTEX;
 273   aModes << TopAbs_EDGE;
 274   myWorkshop->activateSubShapesSelection(aModes);
 275 }
 276
 277 bool PartSet_WidgetPoint2D::canBeActivatedByMove()
 278 {
 279   bool aCanBeActivated = false;
 280   if (feature()->getKind() == SketchPlugin_Line::ID() &&
 281       attributeID() == SketchPlugin_Line::START_ID())
 282     aCanBeActivated = true;
 283
 284   return aCanBeActivated;
 285 }
 286
 287 void PartSet_WidgetPoint2D::deactivate()
 288 {
 289   ModuleBase_ModelWidget::deactivate();
 290   ModuleBase_IViewer* aViewer = myWorkshop->viewer();
 291   disconnect(aViewer, SIGNAL(mouseMove(ModuleBase_IViewWindow*, QMouseEvent*)),
 292              this, SLOT(onMouseMove(ModuleBase_IViewWindow*, QMouseEvent*)));
 293   disconnect(aViewer, SIGNAL(mouseRelease(ModuleBase_IViewWindow*, QMouseEvent*)),
 294              this, SLOT(onMouseRelease(ModuleBase_IViewWindow*, QMouseEvent*)));
 295
 296   myWorkshop->deactivateSubShapesSelection();
 297 }
 298
 299 bool PartSet_WidgetPoint2D::getPoint2d(const Handle(V3d_View)& theView,
 300                                        const TopoDS_Shape& theShape,
 301                                        double& theX, double& theY) const
 302 {
 303   if (!theShape.IsNull()) {
 304     if (theShape.ShapeType() == TopAbs_VERTEX) {
 305       const TopoDS_Vertex& aVertex = TopoDS::Vertex(theShape);
 306       if (!aVertex.IsNull()) {
 307         // A case when point is taken from existing vertex
 308         gp_Pnt aPoint = BRep_Tool::Pnt(aVertex);
 309         PartSet_Tools::convertTo2D(aPoint, mySketch, theView, theX, theY);
 310         return true;
 311       }
 312     }
 313   }
 314   return false;
 315 }
 316
 317 void PartSet_WidgetPoint2D::setConstraintWith(const ObjectPtr& theObject)
 318 {
 319   // Create point-edge coincedence
 320   FeaturePtr aFeature = mySketch->addFeature(SketchPlugin_ConstraintCoincidence::ID());
 321   std::shared_ptr<ModelAPI_Data> aData = aFeature->data();
 322
 323   std::shared_ptr<ModelAPI_AttributeRefAttr> aRef1 = std::dynamic_pointer_cast<
 324       ModelAPI_AttributeRefAttr>(aData->attribute(SketchPlugin_Constraint::ENTITY_A()));
 325   AttributePtr aThisAttr = feature()->data()->attribute(attributeID());
 326   std::shared_ptr<GeomDataAPI_Point2D> aThisPoint =
 327     std::dynamic_pointer_cast<GeomDataAPI_Point2D>(aThisAttr);
 328   aRef1->setAttr(aThisPoint);
 329
 330   std::shared_ptr<ModelAPI_AttributeRefAttr> aRef2 = std::dynamic_pointer_cast<
 331       ModelAPI_AttributeRefAttr>(aData->attribute(SketchPlugin_Constraint::ENTITY_B()));
 332   aRef2->setObject(theObject);
 333
 334   aFeature->execute();
 335 }
 336
 337 void PartSet_WidgetPoint2D::onMouseRelease(ModuleBase_IViewWindow* theWnd, QMouseEvent* theEvent)
 338 {
 339   // the contex menu release by the right button should not be processed by this widget
 340   if (theEvent->button() != Qt::LeftButton)
 341     return;
 342
 343   ModuleBase_ISelection* aSelection = myWorkshop->selection();
 344   Handle(V3d_View) aView = theWnd->v3dView();
 345   // TODO: This fragment doesn't work because bug in OCC Viewer. It can be used after fixing.
 346   NCollection_List<TopoDS_Shape> aShapes;
 347   std::list<ObjectPtr> aObjects;
 348   aSelection->selectedShapes(aShapes, aObjects);
 349   // if we have selection
 350   if (aShapes.Extent() > 0) {
 351     TopoDS_Shape aShape = aShapes.First();
 352     ObjectPtr aObject = aObjects.front();
 353     FeaturePtr aSelectedFeature = ModelAPI_Feature::feature(aObject);
 354     bool anExternal = false;
 355     if (aSelectedFeature.get() != NULL) {
 356       std::shared_ptr<SketchPlugin_Feature> aSPFeature =
 357               std::dynamic_pointer_cast<SketchPlugin_Feature>(aSelectedFeature);
 358       if ((!aSPFeature) && (!aShape.IsNull())) {
 359         anExternal = true;
 360         ResultPtr aFixedObject = PartSet_Tools::findFixedObjectByExternal(aShape, aObject, mySketch);
 361         if (!aFixedObject.get())
 362           aObject = PartSet_Tools::createFixedObjectByExternal(aShape, aObject, mySketch);
 363
 364         double aX, aY;
 365         if (getPoint2d(aView, aShape, aX, aY) && isFeatureContainsPoint(myFeature, aX, aY)) {
 366           // do not create a constraint to the point, which already used by the feature
 367           // if the feature contains the point, focus is not switched
 368           setPoint(aX, aY);
 369         }
 370         else {
 371           if (getPoint2d(aView, aShape, aX, aY))
 372             setPoint(aX, aY);
 373           setConstraintWith(aObject);
 374           emit vertexSelected();
 375           emit focusOutWidget(this);
 376         }
 377       }
 378     }
 379     if (!anExternal) {
 380       double aX, aY;
 381       bool isProcessed = false;
 382       if (getPoint2d(aView, aShape, aX, aY) && isFeatureContainsPoint(myFeature, aX, aY)) {
 383         // when the point is selected, the coordinates of the point should be set into the attribute
 384         // if the feature contains the point, focus is not switched
 385         setPoint(aX, aY);
 386       }
 387       else {
 388         // do not set a coincidence constraint in the attribute if the feature contains a point
 389         // with the same coordinates. It is important for line creation in order to do not set
 390         // the same constraints for the same points, oterwise the result line has zero length.
 391         if (getPoint2d(aView, aShape, aX, aY)) {
 392           setPoint(aX, aY);
 393           PartSet_Tools::setConstraints(mySketch, feature(), attributeID(), aX, aY);
 394         }
 395         else if (aShape.ShapeType() == TopAbs_EDGE) {
 396           if (MyFeaturesForCoincedence.contains(myFeature->getKind().c_str()))
 397             setConstraintWith(aObject);
 398         }
 399         // it is important to perform updateObject() in order to the current value is
 400         // processed by Sketch Solver. Test case: line is created from a previous point
 401         // to some distance, but in the area of the highlighting of the point. Constraint
 402         // coincidence is created, after the solver is performed, the distance between the
 403         // points of the line becomes less than the tolerance. Validator of the line returns
 404         // false, the line will be aborted, but sketch stays valid.
 405         updateObject(feature());
 406         emit vertexSelected();
 407         emit focusOutWidget(this);
 408       }
 409     }
 410   }
 411   // End of Bug dependent fragment
 412   else {
 413     // A case when point is taken from mouse event
 414     gp_Pnt aPoint = PartSet_Tools::convertClickToPoint(theEvent->pos(), theWnd->v3dView());
 415     double aX, anY;
 416     PartSet_Tools::convertTo2D(aPoint, mySketch, aView, aX, anY);
 417
 418     // if the feature contains the point, focus is not switched
 419     if (!setPoint(aX, anY) || isFeatureContainsPoint(myFeature, aX, anY))
 420       return;
 421
 422     /// Start alternative code
 423     //std::shared_ptr<GeomDataAPI_Point2D> aFeaturePoint = std::dynamic_pointer_cast<
 424     //    GeomDataAPI_Point2D>(feature()->data()->attribute(attributeID()));
 425     //QList<FeaturePtr> aIgnore;
 426     //aIgnore.append(feature());
 427
 428     //double aTolerance = aView->Convert(7);
 429     //std::shared_ptr<GeomDataAPI_Point2D> aAttrPnt =
 430     //  PartSet_Tools::findAttributePoint(mySketch, aX, anY, aTolerance, aIgnore);
 431     //if (aAttrPnt.get() != NULL) {
 432     //  aFeaturePoint->setValue(aAttrPnt->pnt());
 433     //  PartSet_Tools::createConstraint(mySketch, aAttrPnt, aFeaturePoint);
 434     //  emit vertexSelected();
 435     //}
 436     /// End alternative code
 437     emit focusOutWidget(this);
 438   }
 439 }
 440
 441
 442 void PartSet_WidgetPoint2D::onMouseMove(ModuleBase_IViewWindow* theWnd, QMouseEvent* theEvent)
 443 {
 444   if (isEditingMode())
 445     return;
 446
 447   gp_Pnt aPoint = PartSet_Tools::convertClickToPoint(theEvent->pos(), theWnd->v3dView());
 448
 449   double aX, anY;
 450   PartSet_Tools::convertTo2D(aPoint, mySketch, theWnd->v3dView(), aX, anY);
 451   // we need to block the value state change
 452   bool isBlocked = blockValueState(true);
 453   setPoint(aX, anY);
 454   blockValueState(isBlocked);
 455   setValueState(ModifiedInViewer);
 456 }
 457
 458 double PartSet_WidgetPoint2D::x() const
 459 {
 460   return myXSpin->value();
 461 }
 462
 463 double PartSet_WidgetPoint2D::y() const
 464 {
 465   return myYSpin->value();
 466 }
 467
 468
 469 bool PartSet_WidgetPoint2D::isFeatureContainsPoint(const FeaturePtr& theFeature,
 470                                                    double theX, double theY)
 471 {
 472   bool aPointIsFound = false;
 473   AttributePtr aWidgetAttribute = myFeature->attribute(attributeID());
 474
 475   std::shared_ptr<GeomAPI_Pnt2d> aPnt2d =
 476                                     std::shared_ptr<GeomAPI_Pnt2d>(new GeomAPI_Pnt2d(theX, theY));
 477   std::list<AttributePtr> anAttributes =
 478                                 myFeature->data()->attributes(GeomDataAPI_Point2D::typeId());
 479   std::list<AttributePtr>::iterator anIter = anAttributes.begin();
 480   for(; anIter != anAttributes.end() && !aPointIsFound; anIter++) {
 481     AttributePoint2DPtr aPoint2DAttribute =
 482       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(*anIter);
 483     if (aPoint2DAttribute == aWidgetAttribute)
 484       continue;
 485     if (aPoint2DAttribute.get() && aPoint2DAttribute->isInitialized()) {
 486       aPointIsFound = aPoint2DAttribute->pnt()->isEqual(aPnt2d);
 487     }
 488   }
 489   return aPointIsFound;
 490 }
 491
 492 /*void PartSet_WidgetPoint2D::onValuesChanged()
 493 {
 494   emit valuesChanged();
 495 }*/
 496
 497 bool PartSet_WidgetPoint2D::processEnter()
 498 {
 499   bool isModified = myXSpin->isModified() || myYSpin->isModified();
 500   if (isModified) {
 501     bool isXModified = myXSpin->isModified();
 502     emit valuesChanged();
 503     //onValuesChanged();
 504     myXSpin->clearModified();
 505     myYSpin->clearModified();
 506     if (isXModified)
 507       myXSpin->selectAll();
 508     else
 509       myYSpin->selectAll();
 510   }
 511   return isModified;
 512 }