Salome HOME
Dump Python extension
[modules/geom.git] / src / GenerationGUI / GenerationGUI_RevolDlg.cxx
1 // GEOM GEOMGUI : GUI for Geometry component
2 //
3 // Copyright (C) 2003  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License.
10 //
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 // Lesser General Public License for more details.
15 //
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22 // File   : GenerationGUI_RevolDlg.cxx
23 // Author : Lucien PIGNOLONI, Open CASCADE S.A.S.
24 //
25
26 #include "GenerationGUI_RevolDlg.h"
27
28 #include <DlgRef.h>
29 #include <GeometryGUI.h>
30 #include <GEOMBase.h>
31
32 #include <SUIT_Session.h>
33 #include <SUIT_ResourceMgr.h>
34 #include <SalomeApp_Application.h>
35 #include <LightApp_SelectionMgr.h>
36
37 #include <TopoDS_Shape.hxx>
38 #include <TopoDS_Edge.hxx>
39 #include <TopoDS.hxx>
40 #include <TopExp.hxx>
41 #include <TColStd_IndexedMapOfInteger.hxx>
42 #include <TopTools_IndexedMapOfShape.hxx>
43 #include <TopExp_Explorer.hxx>
44
45 #include <GEOMImpl_Types.hxx>
46
47 //=================================================================================
48 // class    : GenerationGUI_RevolDlg()
49 // purpose  : Constructs a GenerationGUI_RevolDlg which is a child of 'parent', with the
50 //            name 'name' and widget flags set to 'f'.
51 //            The dialog will by default be modeless, unless you set 'modal' to
52 //            TRUE to construct a modal dialog.
53 //=================================================================================
54 GenerationGUI_RevolDlg::GenerationGUI_RevolDlg (GeometryGUI* theGeometryGUI, QWidget* parent,
55                                                 bool modal, Qt::WindowFlags fl)
56   : GEOMBase_Skeleton(theGeometryGUI, parent, modal, fl)
57 {
58   QPixmap image0 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_DLG_REVOL")));
59   QPixmap image1 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_SELECT")));
60
61   setWindowTitle(tr("GEOM_REVOLUTION_TITLE"));
62
63   /***************************************************************/
64   mainFrame()->GroupConstructors->setTitle(tr("GEOM_REVOLUTION"));
65   mainFrame()->RadioButton1->setIcon(image0);
66   mainFrame()->RadioButton2->setAttribute(Qt::WA_DeleteOnClose);
67   mainFrame()->RadioButton2->close();
68   mainFrame()->RadioButton3->setAttribute(Qt::WA_DeleteOnClose);
69   mainFrame()->RadioButton3->close();
70   myBothway = false;
71
72   GroupPoints = new DlgRef_2Sel1Spin2Check(centralWidget());
73   GroupPoints->GroupBox1->setTitle(tr("GEOM_ARGUMENTS"));
74   GroupPoints->TextLabel1->setText(tr("GEOM_OBJECT"));
75   GroupPoints->TextLabel2->setText(tr("GEOM_AXIS"));
76   GroupPoints->TextLabel3->setText(tr("GEOM_ANGLE"));
77   GroupPoints->PushButton1->setIcon(image1);
78   GroupPoints->PushButton2->setIcon(image1);
79   GroupPoints->LineEdit1->setReadOnly(true);
80   GroupPoints->LineEdit2->setReadOnly(true);
81   GroupPoints->CheckButton1->setText(tr("GEOM_BOTHWAY"));
82   GroupPoints->CheckButton2->setText(tr("GEOM_REVERSE"));
83
84   QVBoxLayout* layout = new QVBoxLayout(centralWidget());
85   layout->setMargin(0); layout->setSpacing(6);
86   layout->addWidget(GroupPoints);
87   /***************************************************************/
88
89   setHelpFileName("create_revolution_page.html");
90
91   // Initialisation
92   Init();
93 }
94
95 //=================================================================================
96 // function : ~GenerationGUI_RevolDlg()
97 // purpose  : Destroys the object and frees any allocated resources
98 //=================================================================================
99 GenerationGUI_RevolDlg::~GenerationGUI_RevolDlg()
100 {
101   // no need to delete child widgets, Qt does it all for us
102 }
103
104 //=================================================================================
105 // function : Init()
106 // purpose  :
107 //=================================================================================
108 void GenerationGUI_RevolDlg::Init()
109 {
110   // min, max, step and decimals for spin boxes & initial values
111   double SpecificStep = 5;
112   initSpinBox(GroupPoints->SpinBox_DX, 0.0, 360.0, SpecificStep, 3); // VSR: TODO: DBL_DIGITS_DISPLAY
113   GroupPoints->SpinBox_DX->setValue(45.0);
114
115   // init variables
116   GroupPoints->LineEdit1->setReadOnly(true);
117   GroupPoints->LineEdit2->setReadOnly(true);
118
119   GroupPoints->LineEdit1->setText("");
120   GroupPoints->LineEdit2->setText("");
121   myBase = myAxis = GEOM::GEOM_Object::_nil();
122   myOkBase = myOkAxis = false;
123
124   // signals and slots connections
125   connect(buttonOk(),    SIGNAL(clicked()), this, SLOT(ClickOnOk()));
126   connect(buttonApply(), SIGNAL(clicked()), this, SLOT(ClickOnApply()));
127
128   connect(GroupPoints->PushButton1,  SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
129   connect(GroupPoints->PushButton2,  SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
130
131   connect(GroupPoints->LineEdit1,    SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
132   connect(GroupPoints->LineEdit2,    SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
133
134   connect(GroupPoints->SpinBox_DX,   SIGNAL(valueChanged(double)), this, SLOT(ValueChangedInSpinBox()));
135   connect(GroupPoints->CheckButton1, SIGNAL(toggled(bool)),        this, SLOT(onBothway()));
136   connect(GroupPoints->CheckButton2, SIGNAL(toggled(bool)),        this, SLOT(onReverse()));
137
138   connect(myGeomGUI, SIGNAL(SignalDefaultStepValueChanged(double)), this, SLOT(SetDoubleSpinBoxStep(double)));
139
140   initName(tr("GEOM_REVOLUTION"));
141
142   GroupPoints->PushButton1->click();
143   SelectionIntoArgument();
144 }
145
146 //=================================================================================
147 // function : SetDoubleSpinBoxStep()
148 // purpose  : Double spin box management
149 //=================================================================================
150 void GenerationGUI_RevolDlg::SetDoubleSpinBoxStep (double step)
151 {
152   GroupPoints->SpinBox_DX->setSingleStep(step);
153 }
154
155 //=================================================================================
156 // function : ClickOnOk()
157 // purpose  :
158 //=================================================================================
159 void GenerationGUI_RevolDlg::ClickOnOk()
160 {
161   if (ClickOnApply())
162     ClickOnCancel();
163 }
164
165 //=================================================================================
166 // function : ClickOnApply()
167 // purpose  :
168 //=================================================================================
169 bool GenerationGUI_RevolDlg::ClickOnApply()
170 {
171   if (!onAccept())
172     return false;
173
174   initName();
175   // activate selection and connect selection manager
176   GroupPoints->PushButton1->click();
177   return true;
178 }
179
180 //=======================================================================
181 //function : isAcceptableBase
182 //purpose  : return true if theBase can be used as algo argument
183 //=======================================================================
184 static bool isAcceptableBase (const TopoDS_Shape& theBase)
185 {
186   switch (theBase.ShapeType()) {
187   case TopAbs_VERTEX:
188   case TopAbs_EDGE:
189   case TopAbs_WIRE:
190   case TopAbs_FACE:
191   case TopAbs_SHELL:
192     return true;
193   case TopAbs_SOLID:
194   case TopAbs_COMPSOLID:
195     return false;
196   case TopAbs_COMPOUND: {
197     TopExp_Explorer exp(theBase, TopAbs_SOLID);
198     return !exp.More();
199   }
200   default:
201     return false;
202   }
203   return false;
204 }
205
206 //=================================================================================
207 // function : SelectionIntoArgument()
208 // purpose  : Called when selection is changed or on dialog initialization or activation
209 //=================================================================================
210 void GenerationGUI_RevolDlg::SelectionIntoArgument()
211 {
212   erasePreview();
213   myEditCurrentArgument->setText("");
214   if      (myEditCurrentArgument == GroupPoints->LineEdit1) myOkBase = false;
215   else if (myEditCurrentArgument == GroupPoints->LineEdit2) myOkAxis = false;
216
217   LightApp_SelectionMgr* aSelMgr = myGeomGUI->getApp()->selectionMgr();
218   SALOME_ListIO aSelList;
219   aSelMgr->selectedObjects(aSelList);
220
221   if (aSelList.Extent() != 1)
222     return;
223
224   // nbSel == 1
225   Standard_Boolean testResult = Standard_False;
226   GEOM::GEOM_Object_ptr aSelectedObject = GEOMBase::ConvertIOinGEOMObject(aSelList.First(), testResult);
227
228   if (!testResult || aSelectedObject->_is_nil())
229     return;
230
231   TopoDS_Shape S;
232   if (!GEOMBase::GetShape(aSelectedObject, S) || S.IsNull())
233     return;
234
235   QString aName = GEOMBase::GetName(aSelectedObject);
236
237   if (myEditCurrentArgument == GroupPoints->LineEdit1) {
238     if (!isAcceptableBase(S))
239       return;
240
241     myBase = aSelectedObject;
242     myOkBase = true;
243     myEditCurrentArgument->setText(aName);
244     if (!myOkAxis)
245       GroupPoints->PushButton2->click();
246   }
247   else if (myEditCurrentArgument == GroupPoints->LineEdit2) {
248     TColStd_IndexedMapOfInteger aMap;
249     aSelMgr->GetIndexes(aSelList.First(), aMap);
250     if (aMap.Extent() == 1) {
251       int anIndex = aMap(1);
252       aName.append(":edge_" + QString::number(anIndex));
253
254       //Find SubShape Object in Father
255       GEOM::GEOM_Object_var aFindedObject = GEOMBase_Helper::findObjectInFather(aSelectedObject, aName);
256
257       if (aFindedObject == GEOM::GEOM_Object::_nil()) { // Object not found in study
258         GEOM::GEOM_IShapesOperations_var aShapesOp =
259           getGeomEngine()->GetIShapesOperations(getStudyId());
260         myAxis = aShapesOp->GetSubShape(aSelectedObject, anIndex);
261         myOkAxis = true;
262       }
263       else {
264         myAxis = aFindedObject;
265         myOkAxis = true;
266       }
267     }
268     else {
269       myOkAxis = true;
270       if (S.ShapeType() != TopAbs_EDGE) {
271         aSelectedObject = GEOM::GEOM_Object::_nil();
272         aName = "";
273         myOkAxis = false;
274       }
275       myAxis = aSelectedObject;
276     }
277     myEditCurrentArgument->setText(aName);
278     if (myOkAxis && !myOkBase)
279       GroupPoints->PushButton1->click();
280   }
281
282   // clear selection
283   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
284   myGeomGUI->getApp()->selectionMgr()->clearSelected();
285   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
286           this, SLOT(SelectionIntoArgument()));
287
288   displayPreview();
289 }
290
291 //=================================================================================
292 // function : SetEditCurrentArgument()
293 // purpose  :
294 //=================================================================================
295 void GenerationGUI_RevolDlg::SetEditCurrentArgument()
296 {
297   QPushButton* send = (QPushButton*)sender();
298
299   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
300   globalSelection(GEOM_ALLSHAPES);
301   if (send == GroupPoints->PushButton1) {
302     myEditCurrentArgument = GroupPoints->LineEdit1;
303     GroupPoints->PushButton2->setDown(false);
304     GroupPoints->LineEdit2->setEnabled(false);
305   }
306   else if (send == GroupPoints->PushButton2) {
307     myEditCurrentArgument = GroupPoints->LineEdit2;
308     GroupPoints->PushButton1->setDown(false);
309     GroupPoints->LineEdit1->setEnabled(false);
310     localSelection(GEOM::GEOM_Object::_nil(), TopAbs_EDGE);
311   }
312   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
313           this, SLOT(SelectionIntoArgument()));
314
315   // enable line edit
316   myEditCurrentArgument->setEnabled(true);
317   myEditCurrentArgument->setFocus();
318   // after setFocus(), because it will be setDown(false) when loses focus
319   send->setDown(true);
320
321   // seems we need it only to avoid preview disappearing, caused by selection mode change
322   displayPreview();
323 }
324
325 //=================================================================================
326 // function : LineEditReturnPressed()
327 // purpose  :
328 //=================================================================================
329 void GenerationGUI_RevolDlg::LineEditReturnPressed()
330 {
331   QLineEdit* send = (QLineEdit*)sender();
332   if (send == GroupPoints->LineEdit1 ||
333       send == GroupPoints->LineEdit2) {
334     myEditCurrentArgument = send;
335     GEOMBase_Skeleton::LineEditReturnPressed();
336   }
337 }
338
339 //=================================================================================
340 // function : ActivateThisDialog()
341 // purpose  :
342 //=================================================================================
343 void GenerationGUI_RevolDlg::ActivateThisDialog()
344 {
345   GEOMBase_Skeleton::ActivateThisDialog();
346
347   // reinit, because some selected objects could be removed
348   Init();
349 }
350
351 //=================================================================================
352 // function : enterEvent()
353 // purpose  :
354 //=================================================================================
355 void GenerationGUI_RevolDlg::enterEvent (QEvent*)
356 {
357   if (!mainFrame()->GroupConstructors->isEnabled())
358     ActivateThisDialog();
359 }
360
361 //=================================================================================
362 // function : ValueChangedInSpinBox()
363 // purpose  :
364 //=================================================================================
365 void GenerationGUI_RevolDlg::ValueChangedInSpinBox()
366 {
367   displayPreview();
368 }
369
370 //=================================================================================
371 // function : getAngle()
372 // purpose  :
373 //=================================================================================
374 double GenerationGUI_RevolDlg::getAngle() const
375 {
376   return GroupPoints->SpinBox_DX->value();
377 }
378
379 //=================================================================================
380 // function : createOperation
381 // purpose  :
382 //=================================================================================
383 GEOM::GEOM_IOperations_ptr GenerationGUI_RevolDlg::createOperation()
384 {
385   return getGeomEngine()->GetI3DPrimOperations(getStudyId());
386 }
387
388 //=================================================================================
389 // function : isValid
390 // purpose  :
391 //=================================================================================
392 bool GenerationGUI_RevolDlg::isValid (QString& msg)
393 {
394   bool ok = GroupPoints->SpinBox_DX->isValid( msg, !IsPreview() );
395   return myOkBase && myOkAxis && ok;
396 }
397
398 //=================================================================================
399 // function : execute
400 // purpose  :
401 //=================================================================================
402 bool GenerationGUI_RevolDlg::execute (ObjectList& objects)
403 {
404   GEOM::GEOM_Object_var anObj;
405
406   if (!myBothway) {
407     anObj = GEOM::GEOM_I3DPrimOperations::_narrow(getOperation())->
408       MakeRevolutionAxisAngle(myBase, myAxis, getAngle() * PI180);
409   }
410   else {
411     anObj = GEOM::GEOM_I3DPrimOperations::_narrow(getOperation())->
412       MakeRevolutionAxisAngle2Ways(myBase, myAxis, getAngle() * PI180);
413   }
414
415   if (!anObj->_is_nil())
416   {
417     QStringList aParameters;
418     aParameters << GroupPoints->SpinBox_DX->text();
419     anObj->SetParameters(GeometryGUI::JoinObjectParameters(aParameters));
420
421     objects.push_back(anObj._retn());
422   }
423
424   return true;
425 }
426
427 //=================================================================================
428 // function :  onReverse()
429 // purpose  :
430 //=================================================================================
431 void GenerationGUI_RevolDlg::onReverse()
432 {
433   double anOldValue = GroupPoints->SpinBox_DX->value();
434   GroupPoints->SpinBox_DX->setValue(-anOldValue);
435 }
436
437 //=================================================================================
438 // function :  onBothway()
439 // purpose  :
440 //=================================================================================
441 void GenerationGUI_RevolDlg::onBothway()
442 {
443   bool anOldValue = myBothway;
444   myBothway = !anOldValue;
445   GroupPoints->CheckButton2->setEnabled(!myBothway);
446   displayPreview();
447 }
448
449 //=================================================================================
450 // function : addSubshapeToStudy
451 // purpose  : virtual method to add new SubObjects if local selection
452 //=================================================================================
453 void GenerationGUI_RevolDlg::addSubshapesToStudy()
454 {
455   QMap<QString, GEOM::GEOM_Object_var> objMap;
456
457   objMap[GroupPoints->LineEdit2->text()] = myAxis;
458
459   addSubshapesToFather(objMap);
460 }