Salome HOME
Issue #18905(tuleap): Restore old plane of sketcher if user cancels plane change
[modules/shaper.git] / src / PartSet / PartSet_WidgetSketchLabel.cpp
1 // Copyright (C) 2014-2019  CEA/DEN, EDF R&D
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 //
19
20 #include "PartSet_WidgetSketchLabel.h"
21 #include "PartSet_Tools.h"
22 #include "PartSet_Module.h"
23 #include "PartSet_PreviewPlanes.h"
24 #include "PartSet_SketcherReentrantMgr.h"
25
26 #include "SketchPlugin_SketchEntity.h"
27
28 #include <XGUI_ActionsMgr.h>
29 #include <XGUI_Displayer.h>
30 #include <XGUI_ModuleConnector.h>
31 #include <XGUI_SelectionActivate.h>
32 #include <XGUI_Selection.h>
33 #include <XGUI_SelectionMgr.h>
34 #include <XGUI_Tools.h>
35 #include <XGUI_ViewerProxy.h>
36 #include <XGUI_Workshop.h>
37
38 #include <ModelAPI_ResultBody.h>
39 #include <ModelAPI_Tools.h>
40 #include <ModelAPI_AttributeString.h>
41 #include <ModelAPI_Events.h>
42
43 #include <ModuleBase_Operation.h>
44 #include <ModuleBase_ViewerPrs.h>
45 #include <ModuleBase_Tools.h>
46 #include <ModuleBase_IModule.h>
47 #include <ModuleBase_IPropertyPanel.h>
48
49 #include <GeomAlgoAPI_FaceBuilder.h>
50 #include <GeomAlgoAPI_ShapeTools.h>
51 #include <GeomDataAPI_Point.h>
52 #include <GeomDataAPI_Dir.h>
53 #include <GeomAPI_XYZ.h>
54 #include <GeomAPI_Face.h>
55 #include <GeomAPI_Edge.h>
56 #include <GeomAPI_ShapeExplorer.h>
57
58 #include <SketchPlugin_Sketch.h>
59 #include <SketcherPrs_Tools.h>
60
61 #include <Precision.hxx>
62 #include <gp_Pln.hxx>
63 #include <gp_Pnt.hxx>
64 #include <gp_Dir.hxx>
65 #include <AIS_Shape.hxx>
66 #include <AIS_DimensionSelectionMode.hxx>
67 #include <Bnd_Box.hxx>
68
69 #include <Config_WidgetAPI.h>
70 #include <Config_PropManager.h>
71
72 #include <QLabel>
73 #include <QApplication>
74 #include <QVBoxLayout>
75 #include <QHBoxLayout>
76 #include <QCheckBox>
77 #include <QGroupBox>
78 #include <QPushButton>
79 #include <QLineEdit>
80 #include <QDoubleValidator>
81 #include <QDialog>
82 #include <QTimer>
83
84 #ifndef DBL_MAX
85 #define DBL_MAX 1.7976931348623158e+308
86 #endif
87
88 PartSet_WidgetSketchLabel::PartSet_WidgetSketchLabel(QWidget* theParent,
89                         ModuleBase_IWorkshop* theWorkshop,
90                         const Config_WidgetAPI* theData,
91                         const QMap<PartSet_Tools::ConstraintVisibleState, bool>& toShowConstraints)
92 : ModuleBase_WidgetValidated(theParent, theWorkshop, theData), myOpenTransaction(false),
93 myIsSelection(false)
94 {
95   QVBoxLayout* aLayout = new QVBoxLayout(this);
96   ModuleBase_Tools::zeroMargins(aLayout);
97
98   myStackWidget = new QStackedWidget(this);
99   myStackWidget->setContentsMargins(0,0,0,0);
100   aLayout->addWidget(myStackWidget);
101
102   // Define label for plane selection
103   QWidget* aFirstWgt = new QWidget(this);
104
105   // Size of the View control
106   mySizeOfViewWidget = new QWidget(aFirstWgt);
107   QHBoxLayout* aSizeLayout = new QHBoxLayout(mySizeOfViewWidget);
108   aSizeLayout->addWidget(new QLabel(tr("Size of the view"), mySizeOfViewWidget));
109   mySizeOfView = new QLineEdit(mySizeOfViewWidget);
110
111   QDoubleValidator* aValidator = new QDoubleValidator(0, DBL_MAX, 12, mySizeOfView);
112   aValidator->setLocale(ModuleBase_Tools::doubleLocale());
113   aValidator->setNotation(QDoubleValidator::StandardNotation);
114   mySizeOfView->setValidator(aValidator);
115   aSizeLayout->addWidget(mySizeOfView);
116
117   myPartSetMessage = new QDialog(this, Qt::ToolTip);
118   myPartSetMessage->setModal(false);
119   myPartSetMessage->setStyleSheet("background-color:lightyellow;");
120   QVBoxLayout* aMsgLay = new QVBoxLayout(myPartSetMessage);
121   QString aMsg = tr("The Sketch is created in PartSet.\n"
122     "It will be necessary to create a Part in order to use this sketch for body creation");
123   aMsgLay->addWidget(new QLabel(aMsg, myPartSetMessage));
124   myPartSetMessage->hide();
125
126   mySizeMessage = new QDialog(mySizeOfView, Qt::ToolTip);
127   mySizeMessage->setModal(false);
128   mySizeMessage->setStyleSheet("background-color:lightyellow;");
129   aMsgLay = new QVBoxLayout(mySizeMessage);
130   aMsg = tr("A size of Sketch view can be defined here.");
131   aMsgLay->addWidget(new QLabel(aMsg, mySizeMessage));
132   mySizeMessage->hide();
133
134   QString aText = translate(theData->getProperty("title"));
135   QLabel* aLabel = new QLabel(aText, aFirstWgt);
136   aLabel->setWordWrap(true);
137   QString aTooltip = translate(theData->getProperty("tooltip"));
138   aLabel->setToolTip(aTooltip);
139   aLabel->setIndent(5);
140
141   aLayout = new QVBoxLayout(aFirstWgt);
142   ModuleBase_Tools::zeroMargins(aLayout);
143   aLayout->addWidget(mySizeOfViewWidget);
144   aLayout->addWidget(aLabel);
145
146   myRemoveExternal = new QCheckBox(tr("Remove external dependencies"), aFirstWgt);
147   myRemoveExternal->setChecked(false);
148   aLayout->addWidget(myRemoveExternal);
149   myRemoveExternal->setVisible(false);
150
151   aLayout->addStretch(1);
152
153   myStackWidget->addWidget(aFirstWgt);
154
155   // Define widget for sketch manmagement
156   QWidget* aSecondWgt = new QWidget(this);
157   aLayout = new QVBoxLayout(aSecondWgt);
158   ModuleBase_Tools::zeroMargins(aLayout);
159
160   QGroupBox* aViewBox = new QGroupBox(tr("Sketcher plane"), this);
161   QVBoxLayout* aViewLayout = new QVBoxLayout(aViewBox);
162
163   myViewInverted = new QCheckBox(tr("Reversed"), aViewBox);
164   aViewLayout->addWidget(myViewInverted);
165
166   QPushButton* aSetViewBtn =
167     new QPushButton(QIcon(":icons/plane_view.png"), tr("Set plane view"), aViewBox);
168   connect(aSetViewBtn, SIGNAL(clicked(bool)), this, SLOT(onSetPlaneView()));
169   aViewLayout->addWidget(aSetViewBtn);
170
171   aLayout->addWidget(aViewBox);
172
173   QMap<PartSet_Tools::ConstraintVisibleState, QString> aStates;
174   aStates[PartSet_Tools::Geometrical] = tr("Show geometrical constraints");
175   aStates[PartSet_Tools::Dimensional] = tr("Show dimensional constraints");
176   aStates[PartSet_Tools::Expressions] = tr("Show existing expressions");
177
178   QMap<PartSet_Tools::ConstraintVisibleState, QString>::const_iterator anIt = aStates.begin(),
179                                                         aLast = aStates.end();
180   for (; anIt != aLast; anIt++) {
181     QCheckBox* aShowConstraints = new QCheckBox(anIt.value(), this);
182     connect(aShowConstraints, SIGNAL(toggled(bool)), this, SLOT(onShowConstraint(bool)));
183     aLayout->addWidget(aShowConstraints);
184
185     PartSet_Tools::ConstraintVisibleState aState = anIt.key();
186     myShowConstraints[aState] = aShowConstraints;
187
188     if (toShowConstraints.contains(aState))
189       aShowConstraints->setChecked(toShowConstraints[aState]);
190   }
191   myShowPoints = new QCheckBox(tr("Show free points"), this);
192   connect(myShowPoints, SIGNAL(toggled(bool)), this, SIGNAL(showFreePoints(bool)));
193   aLayout->addWidget(myShowPoints);
194
195   myAutoConstraints = new QCheckBox(tr("Automatic constraints"), this);
196   myAutoConstraints->setToolTip(tr("Automatic vertical and horizontal constraints"));
197   connect(myAutoConstraints, SIGNAL(toggled(bool)), this, SIGNAL(autoConstraints(bool)));
198   aLayout->addWidget(myAutoConstraints);
199
200   QPushButton* aPlaneBtn = new QPushButton(tr("Change sketch plane"), aSecondWgt);
201   connect(aPlaneBtn, SIGNAL(clicked(bool)), SLOT(onChangePlane()));
202   aLayout->addWidget(aPlaneBtn);
203
204   aLayout->addSpacing(15);
205
206   myDoFLabel = new QLabel("", aSecondWgt);
207   aLayout->addWidget(myDoFLabel);
208
209   myShowDOFBtn = new QPushButton(tr("Show remaining DoFs"), aSecondWgt);
210   aLayout->addWidget(myShowDOFBtn);
211   myShowDOFBtn->setEnabled(false);
212   connect(myShowDOFBtn, SIGNAL(clicked(bool)), SLOT(onShowDOF()));
213
214   myStackWidget->addWidget(aSecondWgt);
215   //setLayout(aLayout);
216
217   myPreviewPlanes = new PartSet_PreviewPlanes();
218 }
219
220 PartSet_WidgetSketchLabel::~PartSet_WidgetSketchLabel()
221 {
222 }
223
224 bool PartSet_WidgetSketchLabel::setSelection(QList<ModuleBase_ViewerPrsPtr>& theValues,
225                                              const bool theToValidate)
226 {
227   // do not use the given selection if the plane of the sketch has been already set.
228   // If this check is absent, a selected plane in the viewer can be set in the sketch
229   // even if the sketch is built on another plane.
230   if (plane().get())
231     return true;
232
233   ModuleBase_ViewerPrsPtr aPrs = theValues.first();
234   bool aDone = setSelectionInternal(theValues, theToValidate);
235   if (aDone)
236     updateByPlaneSelected(aPrs);
237   return aDone;
238 }
239
240 QList<QWidget*> PartSet_WidgetSketchLabel::getControls() const
241 {
242   QList<QWidget*> aResult;
243   aResult << myStackWidget;
244   return aResult;
245 }
246
247 bool PartSet_WidgetSketchLabel::processSelection()
248 {
249   std::shared_ptr<GeomAPI_Pln> aPlane = plane();
250   if (aPlane.get())
251     return false;
252
253   QList<ModuleBase_ViewerPrsPtr> aSelected = getFilteredSelected();
254
255   if (aSelected.empty())
256     return false;
257   ModuleBase_ViewerPrsPtr aPrs = aSelected.first();
258   bool aDone = setSelectionInternal(aSelected, false);
259   if (aDone) {
260     updateByPlaneSelected(aPrs);
261     updateObject(myFeature);
262   }
263
264   return aDone;
265 }
266
267 void PartSet_WidgetSketchLabel::onShowConstraint(bool theOn)
268 {
269   QCheckBox* aSenderCheckBox = qobject_cast<QCheckBox*>(sender());
270
271   QMap<PartSet_Tools::ConstraintVisibleState, QCheckBox*>::const_iterator
272                           anIt = myShowConstraints.begin(), aLast = myShowConstraints.end();
273
274   PartSet_Tools::ConstraintVisibleState aState = PartSet_Tools::Geometrical;
275   bool aFound = false;
276   for (; anIt != aLast && !aFound; anIt++) {
277     aFound = anIt.value() == aSenderCheckBox;
278     if (aFound)
279       aState = anIt.key();
280   }
281   if (aFound)
282     emit showConstraintToggled(aState, theOn);
283 }
284
285 void PartSet_WidgetSketchLabel::blockAttribute(const AttributePtr& theAttribute,
286                                                const bool& theToBlock, bool& isFlushesActived,
287                                                bool& isAttributeSetInitializedBlocked,
288                                                bool& isAttributeSendUpdatedBlocked)
289 {
290   ModuleBase_WidgetValidated::blockAttribute(theAttribute, theToBlock, isFlushesActived,
291                                              isAttributeSetInitializedBlocked,
292                                              isAttributeSendUpdatedBlocked);
293   // We do not restore the previous state of isAttributeSetInitializedBlocked for each of
294   // attributes. It it is necessary, these states should be append to the method attributes
295   // or stored in the widget
296
297   std::list<AttributePtr> anAttributes = myFeature->data()->attributes("");
298   std::list<AttributePtr>::const_iterator anIt = anAttributes.begin(), aLast = anAttributes.end();
299   QStringList aValues;
300   for(; anIt != aLast; anIt++) {
301     AttributePtr anAttribute = *anIt;
302     if (theToBlock)
303       anAttribute->blockSetInitialized(true);
304     else
305       anAttribute->blockSetInitialized(isAttributeSetInitializedBlocked);
306   }
307 }
308
309 bool PartSet_WidgetSketchLabel::setSelectionInternal(
310                                           const QList<ModuleBase_ViewerPrsPtr>& theValues,
311                                           const bool theToValidate)
312 {
313   bool aDone = false;
314   if (theValues.empty()) {
315     // In order to make reselection possible, set empty object and shape should be done
316     setSelectionCustom(std::shared_ptr<ModuleBase_ViewerPrs>(
317                               new ModuleBase_ViewerPrs(ObjectPtr(), GeomShapePtr(), NULL)));
318     aDone = false;
319   }
320   else {
321     // it removes the processed value from the parameters list
322     ModuleBase_ViewerPrsPtr aValue = theValues.first();//.takeFirst();
323     if (!theToValidate || isValidInFilters(aValue)) {
324       myIsSelection = true;
325       aDone = setSelectionCustom(aValue);
326       myIsSelection = false;
327     }
328   }
329
330   return aDone;
331 }
332
333 void PartSet_WidgetSketchLabel::updateByPlaneSelected(const ModuleBase_ViewerPrsPtr& thePrs)
334 {
335   // Nullify a temporary remembered plane
336   if (myTmpPlane.get())
337     myTmpPlane.reset();
338
339   GeomPlanePtr aPlane = plane();
340   if (!aPlane.get())
341     return;
342
343   myPartSetMessage->hide();
344   mySizeMessage->hide();
345
346   // 1. hide main planes if they have been displayed and display sketch preview plane
347   myPreviewPlanes->erasePreviewPlanes(myWorkshop);
348
349   PartSet_Module* aModule = dynamic_cast<PartSet_Module*>(myWorkshop->module());
350   if (aModule) {
351     CompositeFeaturePtr aSketch = std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(myFeature);
352     bool isSetSizeOfView = false;
353     double aSizeOfView = 0;
354     QString aSizeOfViewStr = mySizeOfView->text();
355     if (!aSizeOfViewStr.isEmpty()) {
356       aSizeOfView = aSizeOfViewStr.toDouble(&isSetSizeOfView);
357       if (isSetSizeOfView && aSizeOfView <= 0) {
358         isSetSizeOfView = false;
359       }
360     }
361     aModule->sketchMgr()->previewSketchPlane()->setSizeOfView(aSizeOfView, isSetSizeOfView);
362     aModule->sketchMgr()->previewSketchPlane()->createSketchPlane(aSketch, myWorkshop);
363   }
364   // 2. if the planes were displayed, change the view projection
365
366   std::shared_ptr<GeomAPI_Dir> aDir = aPlane->direction();
367   gp_XYZ aXYZ = aDir->impl<gp_Dir>().XYZ();
368   double aTwist = 0.0;
369
370   // Rotate view if the sketcher plane is selected only from preview planes
371   // Preview planes are created only if there is no any shape
372   bool aRotate = Config_PropManager::boolean(SKETCH_TAB_NAME, "rotate_to_plane");
373   if (aRotate) {
374     myWorkshop->viewer()->setViewProjection(aXYZ.X(), aXYZ.Y(), aXYZ.Z(), aTwist);
375   }
376   QString aSizeOfViewStr = mySizeOfView->text();
377   if (!aSizeOfViewStr.isEmpty()) {
378     bool isOk;
379     double aSizeOfView = aSizeOfViewStr.toDouble(&isOk);
380     if (isOk && aSizeOfView > 0) {
381       Handle(V3d_View) aView3d = myWorkshop->viewer()->activeView();
382       if (!aView3d.IsNull()) {
383         Bnd_Box aBndBox;
384         double aHalfSize = aSizeOfView/2.0;
385         aBndBox.Update(-aHalfSize, -aHalfSize, -aHalfSize, aHalfSize, aHalfSize, aHalfSize);
386         aView3d->FitAll(aBndBox, 0.01, false);
387       }
388     }
389   }
390   if (myOpenTransaction) {
391     SessionPtr aMgr = ModelAPI_Session::get();
392     aMgr->finishOperation();
393     myOpenTransaction = false;
394   }
395   // 3. Clear text in the label
396   myStackWidget->setCurrentIndex(1);
397   //myLabel->setText("");
398   //myLabel->setToolTip("");
399   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
400
401   // 5. Clear selection mode and define sketching mode
402   emit planeSelected(plane());
403   // after the plane is selected in the sketch, the sketch selection should be activated
404   // it can not be performed in the sketch label widget because, we don't need to switch off
405   // the selection by any label deactivation, but need to switch it off by stop the sketch
406   myWorkshop->selectionActivate()->updateSelectionFilters();
407   myWorkshop->selectionActivate()->updateSelectionModes();
408
409   if (aModule)
410     aModule->onViewTransformed();
411
412   // 6. Update sketcher actions
413   XGUI_ActionsMgr* anActMgr = aWorkshop->actionsMgr();
414
415   myWorkshop->updateCommandStatus();
416   aWorkshop->selector()->clearSelection();
417   myWorkshop->viewer()->update();
418
419   myRemoveExternal->setVisible(false);
420 }
421
422 std::shared_ptr<GeomAPI_Pln> PartSet_WidgetSketchLabel::plane() const
423 {
424   CompositeFeaturePtr aSketch = std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(myFeature);
425   return PartSet_Tools::sketchPlane(aSketch);
426 }
427
428 bool PartSet_WidgetSketchLabel::focusTo()
429 {
430   ModuleBase_Tools::setFocus(myStackWidget, "PartSet_WidgetSketchLabel::focusTo()");
431   return true;
432 }
433
434 void PartSet_WidgetSketchLabel::enableFocusProcessing()
435 {
436   myStackWidget->installEventFilter(this);
437 }
438
439 void PartSet_WidgetSketchLabel::storeAttributeValue(const AttributePtr& theAttribute)
440 {
441   ModuleBase_WidgetValidated::storeAttributeValue(theAttribute);
442 }
443
444 void PartSet_WidgetSketchLabel::restoreAttributeValue(const AttributePtr& theAttribute,
445                                                       const bool theValid)
446 {
447   ModuleBase_WidgetValidated::restoreAttributeValue(theAttribute, theValid);
448
449   // it is not necessary to save the previous plane value because the plane is chosen once
450   DataPtr aData = feature()->data();
451   AttributeSelectionPtr aSelAttr = std::dynamic_pointer_cast<ModelAPI_AttributeSelection>
452     (aData->attribute(SketchPlugin_SketchEntity::EXTERNAL_ID()));
453   if (aSelAttr) {
454     ResultPtr anEmptyResult;
455     GeomShapePtr anEmptyShape;
456     aSelAttr->setValue(anEmptyResult, anEmptyShape);
457   }
458   std::shared_ptr<GeomDataAPI_Point> anOrigin = std::dynamic_pointer_cast<GeomDataAPI_Point>(
459     aData->attribute(SketchPlugin_Sketch::ORIGIN_ID()));
460   anOrigin->reset();
461   std::shared_ptr<GeomDataAPI_Dir> aNormal = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
462     aData->attribute(SketchPlugin_Sketch::NORM_ID()));
463   aNormal->reset();
464   std::shared_ptr<GeomDataAPI_Dir> aDirX = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
465     aData->attribute(SketchPlugin_Sketch::DIRX_ID()));
466   aDirX->reset();
467 }
468
469 bool PartSet_WidgetSketchLabel::setSelectionCustom(const ModuleBase_ViewerPrsPtr& thePrs)
470 {
471   if (myIsSelection && myRemoveExternal->isVisible()) {
472     if (myRemoveExternal->isChecked()) {
473       myFeature->customAction(SketchPlugin_Sketch::ACTION_REMOVE_EXTERNAL());
474     }
475   }
476   return fillSketchPlaneBySelection(thePrs);
477 }
478
479 bool PartSet_WidgetSketchLabel::canFillSketch(const ModuleBase_ViewerPrsPtr& thePrs)
480 {
481   bool aCanFillSketch = true;
482   // avoid any selection on sketch object
483   ObjectPtr anObject = thePrs->object();
484   ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObject);
485   if (aResult.get()) {
486     FeaturePtr aFeature = ModelAPI_Feature::feature(aResult);
487     if (aFeature->getKind() == SketchPlugin_Sketch::ID())
488       aCanFillSketch = false;
489   }
490   // check plane or planar face of any non-sketch object
491   if (aCanFillSketch) {
492     std::shared_ptr<GeomAPI_Face> aGeomFace;
493
494     GeomShapePtr aGeomShape = thePrs->shape();
495     if ((!aGeomShape.get() || aGeomShape->isNull()) && aResult.get()) {
496       aGeomShape = aResult->shape();
497     }
498
499     if (aGeomShape.get() && aGeomShape->shapeType() == GeomAPI_Shape::FACE) {
500       std::shared_ptr<GeomAPI_Face> aGeomFace(new GeomAPI_Face(aGeomShape));
501       aCanFillSketch = aGeomFace.get() && aGeomFace->isPlanar();
502     }
503     else
504       aCanFillSketch = false;
505   }
506   return aCanFillSketch;
507 }
508
509 bool PartSet_WidgetSketchLabel::fillSketchPlaneBySelection(const ModuleBase_ViewerPrsPtr& thePrs)
510 {
511   bool isOwnerSet = false;
512
513   const GeomShapePtr& aShape = thePrs->shape();
514   std::shared_ptr<GeomAPI_Dir> aDir;
515
516   if (aShape.get() && !aShape->isNull()) {
517     const TopoDS_Shape& aTDShape = aShape->impl<TopoDS_Shape>();
518     aDir = setSketchPlane(aTDShape);
519     isOwnerSet = aDir.get();
520   }
521   if (thePrs->object() && (feature() != thePrs->object())) {
522     FeaturePtr aFeature = ModelAPI_Feature::feature(thePrs->object());
523     DataPtr aData = feature()->data();
524     AttributeSelectionPtr aSelAttr =
525       std::dynamic_pointer_cast<ModelAPI_AttributeSelection>
526       (aData->attribute(SketchPlugin_SketchEntity::EXTERNAL_ID()));
527     if (aSelAttr.get()) {
528       ResultPtr aRes = std::dynamic_pointer_cast<ModelAPI_Result>(thePrs->object());
529       if (aRes.get()) {
530         GeomShapePtr aShapePtr;
531         if (!aShape.get() || aShape->isNull()) {  // selection happens in the OCC viewer
532           aShapePtr = ModelAPI_Tools::shape(aRes);
533         }
534         else { // selection happens in OB browser
535           aShapePtr = aShape;
536         }
537         if (aShapePtr.get() && aShapePtr->isFace()) {
538           const TopoDS_Shape& aTDShape = aShapePtr->impl<TopoDS_Shape>();
539           setSketchPlane(aTDShape);
540           aSelAttr->setValue(aRes, aShapePtr);
541           isOwnerSet = true;
542         }
543       }
544       else {
545         aSelAttr->setValue(aFeature, GeomShapePtr());
546         GeomShapePtr aShape = aSelAttr->value();
547         if (!aShape.get() && aSelAttr->contextFeature().get() &&
548           aSelAttr->contextFeature()->firstResult().get()) {
549           aShape = aSelAttr->contextFeature()->firstResult()->shape();
550         }
551         if (aShape.get() && aShape->isPlanar()) {
552           const TopoDS_Shape& aTDShape = aShape->impl<TopoDS_Shape>();
553           setSketchPlane(aTDShape);
554           isOwnerSet = true;
555         }
556       }
557     }
558   }
559   return isOwnerSet;
560 }
561
562 void PartSet_WidgetSketchLabel::activateCustom()
563 {
564   PartSet_Module* aModule = dynamic_cast<PartSet_Module*>(myWorkshop->module());
565   if (aModule) {
566     bool isBlocked = myAutoConstraints->blockSignals(true);
567     myAutoConstraints->setChecked(aModule->sketchReentranceMgr()->isAutoConstraints());
568     myAutoConstraints->blockSignals(isBlocked);
569   }
570
571   std::shared_ptr<GeomAPI_Pln> aPlane = plane();
572   if (aPlane.get()) {
573     myStackWidget->setCurrentIndex(1);
574     return;
575   }
576
577   myStackWidget->setCurrentIndex(0);
578   bool aBodyIsVisualized = myPreviewPlanes->hasVisualizedBodies(myWorkshop);
579
580   // Clear previous selection mode It is necessary for correct activation of preview planes
581   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
582   XGUI_Displayer* aDisp = aWorkshop->displayer();
583   aWorkshop->selectionActivate()->activateObjects(QIntList(), aDisp->displayedObjects(), false);
584
585   if (!aBodyIsVisualized) {
586     // We have to select a plane before any operation
587     myPreviewPlanes->showPreviewPlanes(myWorkshop);
588     mySizeOfViewWidget->setVisible(true);
589   }
590   else
591     mySizeOfViewWidget->setVisible(false);
592 }
593
594 void PartSet_WidgetSketchLabel::showEvent(QShowEvent* theEvent)
595 {
596   ModuleBase_WidgetValidated::showEvent(theEvent);
597   QTimer::singleShot(10, this, SLOT(onShowPanel()));
598 }
599
600 void PartSet_WidgetSketchLabel::onShowPanel()
601 {
602   //if (mySizeOfViewWidget->isVisible()) {
603   if (myStackWidget->currentIndex() == 0) {
604     DocumentPtr aDoc = feature()->document();
605     DocumentPtr aModDoc = ModelAPI_Session::get()->moduleDocument();
606     if (aModDoc == aDoc) {
607       myPartSetMessage->move(mapToGlobal(geometry().bottomLeft()));
608       myPartSetMessage->show();
609     }
610     if (mySizeOfViewWidget->isVisible()) {
611       QPoint aPnt = mySizeOfView->mapToGlobal(mySizeOfView->geometry().center());
612       mySizeMessage->move(aPnt);
613       mySizeMessage->show();
614     }
615   }
616 }
617
618 void PartSet_WidgetSketchLabel::deactivate()
619 {
620   if (myTmpPlane.get()) {
621     setSketchPlane(myTmpPlane);
622     myTmpPlane.reset();
623   }
624
625   ModuleBase_WidgetValidated::deactivate();
626   bool aHidePreview = myPreviewPlanes->isPreviewDisplayed();
627   myPreviewPlanes->erasePreviewPlanes(myWorkshop);
628
629   if (aHidePreview)
630     myWorkshop->viewer()->update();
631
632   if (myOpenTransaction) {
633     SessionPtr aMgr = ModelAPI_Session::get();
634     aMgr->finishOperation();
635     myOpenTransaction = false;
636   }
637 }
638
639 void PartSet_WidgetSketchLabel::selectionModes(int& theModuleSelectionModes, QIntList& theModes)
640 {
641   theModuleSelectionModes = -1;
642   std::shared_ptr<GeomAPI_Pln> aPlane = plane();
643   if (!aPlane.get())
644     theModes << TopAbs_FACE;
645 }
646
647 void PartSet_WidgetSketchLabel::selectionFilters(QIntList& theModuleSelectionFilters,
648                                                  SelectMgr_ListOfFilter& theSelectionFilters)
649 {
650   std::shared_ptr<GeomAPI_Pln> aPlane = plane();
651   if (aPlane.get())
652     return;
653   return ModuleBase_WidgetValidated::selectionFilters(theModuleSelectionFilters,
654                                                       theSelectionFilters);
655 }
656
657 std::shared_ptr<GeomAPI_Dir>
658   PartSet_WidgetSketchLabel::setSketchPlane(const TopoDS_Shape& theShape)
659 {
660   if (theShape.IsNull())
661     return std::shared_ptr<GeomAPI_Dir>();
662
663   // get selected shape
664   std::shared_ptr<GeomAPI_Shape> aGShape(new GeomAPI_Shape);
665   aGShape->setImpl(new TopoDS_Shape(theShape));
666
667   // get plane parameters
668   std::shared_ptr<GeomAPI_Face> aFace(new GeomAPI_Face(aGShape));
669   std::shared_ptr<GeomAPI_Pln> aPlane = aFace->getPlane();
670   if (!aPlane.get())
671     return std::shared_ptr<GeomAPI_Dir>();
672   return setSketchPlane(aPlane);
673 }
674
675 std::shared_ptr<GeomAPI_Dir>
676   PartSet_WidgetSketchLabel::setSketchPlane(std::shared_ptr<GeomAPI_Pln> thePlane)
677 {
678   // set plane parameters to feature
679   std::shared_ptr<ModelAPI_Data> aData = feature()->data();
680   double anA, aB, aC, aD;
681   thePlane->coefficients(anA, aB, aC, aD);
682
683   // calculate attributes of the sketch
684   std::shared_ptr<GeomAPI_Dir> aNormDir(new GeomAPI_Dir(anA, aB, aC));
685   std::shared_ptr<GeomAPI_XYZ> aCoords = aNormDir->xyz();
686   std::shared_ptr<GeomAPI_XYZ> aZero(new GeomAPI_XYZ(0, 0, 0));
687   aCoords = aCoords->multiplied(-aD * aCoords->distance(aZero));
688   std::shared_ptr<GeomAPI_Pnt> anOrigPnt(new GeomAPI_Pnt(aCoords));
689   // X axis is preferable to be dirX on the sketch
690   const double tol = Precision::Confusion();
691   bool isX = fabs(fabs(anA) - 1.0) < tol && fabs(aB) < tol && fabs(aC) < tol;
692   std::shared_ptr<GeomAPI_Dir> aTempDir(
693       isX ? new GeomAPI_Dir(0, 1, 0) : new GeomAPI_Dir(1, 0, 0));
694   std::shared_ptr<GeomAPI_Dir> aYDir(new GeomAPI_Dir(aNormDir->cross(aTempDir)));
695   std::shared_ptr<GeomAPI_Dir> aXDir(new GeomAPI_Dir(aYDir->cross(aNormDir)));
696
697   std::shared_ptr<GeomDataAPI_Point> anOrigin = std::dynamic_pointer_cast<GeomDataAPI_Point>(
698       aData->attribute(SketchPlugin_Sketch::ORIGIN_ID()));
699   anOrigin->setValue(anOrigPnt);
700   std::shared_ptr<GeomDataAPI_Dir> aNormal = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
701       aData->attribute(SketchPlugin_Sketch::NORM_ID()));
702   aNormal->setValue(aNormDir);
703   std::shared_ptr<GeomDataAPI_Dir> aDirX = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
704       aData->attribute(SketchPlugin_Sketch::DIRX_ID()));
705   aDirX->setValue(aXDir);
706   std::shared_ptr<GeomAPI_Dir> aDir = thePlane->direction();
707   return aDir;
708 }
709
710 void PartSet_WidgetSketchLabel::onSetPlaneView()
711 {
712   std::shared_ptr<GeomAPI_Pln> aPlane = plane();
713   if (aPlane.get()) {
714     std::shared_ptr<GeomAPI_Dir> aDirection = aPlane->direction();
715     gp_Dir aDir = aDirection->impl<gp_Dir>();
716     if (myViewInverted->isChecked())
717       aDir.Reverse();
718     myWorkshop->viewer()->setViewProjection(aDir.X(), aDir.Y(), aDir.Z(), 0.);
719     PartSet_Module* aModule = dynamic_cast<PartSet_Module*>(myWorkshop->module());
720     if (aModule)
721       aModule->onViewTransformed();
722   }
723 }
724
725
726 //******************************************************
727 QList<std::shared_ptr<ModuleBase_ViewerPrs>> PartSet_WidgetSketchLabel::findCircularEdgesInPlane()
728 {
729   QList<std::shared_ptr<ModuleBase_ViewerPrs>> aResult;
730   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
731   XGUI_Displayer* aDisplayer = aWorkshop->displayer();
732   QObjectPtrList aDispObjects = aDisplayer->displayedObjects();
733
734   std::shared_ptr<GeomAPI_Pln> aPlane = plane();
735   foreach(ObjectPtr aObj, aDispObjects) {
736     ResultPtr aResObj = std::dynamic_pointer_cast<ModelAPI_Result>(aObj);
737     if (aResObj.get()) {
738       GeomShapePtr aShape = aResObj->shape();
739       if (aShape.get()) {
740         GeomAPI_ShapeExplorer aExplorer(aShape, GeomAPI_Shape::EDGE);
741         for(; aExplorer.more(); aExplorer.next()) {
742           GeomShapePtr aEdgeShape = aExplorer.current();
743           GeomAPI_Edge anEdge(aEdgeShape);
744           if ((anEdge.isCircle() || anEdge.isArc() || anEdge.isEllipse()) &&
745                anEdge.isInPlane(aPlane)) {
746             bool isContains = false;
747             // Check that edge is not used.
748             // It is possible that the same edge will be taken from different faces
749             foreach(std::shared_ptr<ModuleBase_ViewerPrs> aPrs, aResult) {
750               GeomAPI_Edge aUsedEdge(aPrs->shape());
751               if (aUsedEdge.isEqual(aEdgeShape)) {
752                 isContains = true;
753                 break;
754               }
755             }
756             if (!isContains) {
757               std::shared_ptr<ModuleBase_ViewerPrs>
758                 aPrs(new ModuleBase_ViewerPrs(aResObj, aEdgeShape));
759               aResult.append(aPrs);
760             }
761           }
762         }
763       }
764     }
765   }
766   return aResult;
767 }
768
769 //******************************************************
770 void PartSet_WidgetSketchLabel::onChangePlane()
771 {
772   PartSet_Module* aModule = dynamic_cast<PartSet_Module*>(myWorkshop->module());
773   if (aModule) {
774     mySizeOfViewWidget->setVisible(false);
775     myRemoveExternal->setVisible(true);
776     myStackWidget->setCurrentIndex(0);
777
778     bool aBodyIsVisualized = myPreviewPlanes->hasVisualizedBodies(myWorkshop);
779
780     if (!aBodyIsVisualized) {
781       // We have to select a plane before any operation
782       myPreviewPlanes->showPreviewPlanes(myWorkshop);
783     }
784
785     CompositeFeaturePtr aSketch = std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(myFeature);
786     myTmpPlane = PartSet_Tools::sketchPlane(aSketch);
787     PartSet_Tools::nullifySketchPlane(aSketch);
788
789     Handle(SelectMgr_Filter) aFilter = aModule->selectionFilter(SF_SketchPlaneFilter);
790     if (!aFilter.IsNull()) {
791       std::shared_ptr<GeomAPI_Pln> aPln;
792       Handle(ModuleBase_ShapeInPlaneFilter)::DownCast(aFilter)->setPlane(aPln);
793     }
794     XGUI_Workshop* aWorkshop = aModule->getWorkshop();
795
796     aWorkshop->selectionActivate()->updateSelectionFilters();
797     aWorkshop->selectionActivate()->updateSelectionModes();
798
799     SessionPtr aMgr = ModelAPI_Session::get();
800     aMgr->startOperation("Change Sketch plane");
801     myOpenTransaction = true;
802     myWorkshop->viewer()->update();
803   }
804 }
805
806 void PartSet_WidgetSketchLabel::setShowPointsState(bool theState)
807 {
808   bool aBlock = myShowPoints->blockSignals(true);
809   myShowPoints->setChecked(theState);
810   myShowPoints->blockSignals(aBlock);
811 }
812
813 bool PartSet_WidgetSketchLabel::restoreValueCustom()
814 {
815   if (myFeature.get()) {
816     CompositeFeaturePtr aSketch = std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(myFeature);
817     if (aSketch.get() && (aSketch->numberOfSubs() > 0)) {
818       AttributeStringPtr aDOFStr = aSketch->string("SolverDOF");
819       if (aDOFStr.get()) {
820         QString aVal(aDOFStr->value().c_str());
821         if (aVal.contains('=')) {
822           // to support old data
823           aVal = aVal.right(aVal.length() - aVal.lastIndexOf('='));
824         }
825         int aDoF = aVal.toInt();
826         if (aDoF == 0) {
827           myDoFLabel->setText(tr("Sketch is fully fixed (DoF = 0)"));
828           myShowDOFBtn->setEnabled(false);
829         } else {
830           myDoFLabel->setText(tr("DoF (degrees of freedom) = ") + aVal);
831           myShowDOFBtn->setEnabled(true);
832         }
833       }
834     }
835     else {
836       myDoFLabel->setText("");
837       myShowDOFBtn->setEnabled(false);
838     }
839   }
840   return true;
841 }
842
843
844 void PartSet_WidgetSketchLabel::onShowDOF()
845 {
846   CompositeFeaturePtr aCompFeature =
847     std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(myFeature);
848   if (aCompFeature.get()) {
849     static const Events_ID anEvent = Events_Loop::eventByName(EVENT_GET_DOF_OBJECTS);
850     ModelAPI_EventCreator::get()->sendUpdated(aCompFeature, anEvent);
851     Events_Loop::loop()->flush(anEvent);
852   }
853 }