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