Salome HOME
IPAL21294 Change Orientation for Vector with "Create a Copy" creates an Edge.
[modules/geom.git] / src / GenerationGUI / GenerationGUI_RevolDlg.cxx
1 //  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 //  Copyright (C) 2003-2007  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 // GEOM GEOMGUI : GUI for Geometry component
23 // File   : GenerationGUI_RevolDlg.cxx
24 // Author : Lucien PIGNOLONI, Open CASCADE S.A.S.
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, -360.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   resize(100,100);
142
143   GroupPoints->PushButton1->click();
144   SelectionIntoArgument();
145 }
146
147 //=================================================================================
148 // function : SetDoubleSpinBoxStep()
149 // purpose  : Double spin box management
150 //=================================================================================
151 void GenerationGUI_RevolDlg::SetDoubleSpinBoxStep (double step)
152 {
153   GroupPoints->SpinBox_DX->setSingleStep(step);
154 }
155
156 //=================================================================================
157 // function : ClickOnOk()
158 // purpose  :
159 //=================================================================================
160 void GenerationGUI_RevolDlg::ClickOnOk()
161 {
162   if (ClickOnApply())
163     ClickOnCancel();
164 }
165
166 //=================================================================================
167 // function : ClickOnApply()
168 // purpose  :
169 //=================================================================================
170 bool GenerationGUI_RevolDlg::ClickOnApply()
171 {
172   if (!onAccept())
173     return false;
174
175   initName();
176   // activate selection and connect selection manager
177   GroupPoints->PushButton1->click();
178   return true;
179 }
180
181 //=======================================================================
182 //function : isAcceptableBase
183 //purpose  : return true if theBase can be used as algo argument
184 //=======================================================================
185 static bool isAcceptableBase (const TopoDS_Shape& theBase)
186 {
187   switch (theBase.ShapeType()) {
188   case TopAbs_VERTEX:
189   case TopAbs_EDGE:
190   case TopAbs_WIRE:
191   case TopAbs_FACE:
192   case TopAbs_SHELL:
193     return true;
194   case TopAbs_SOLID:
195   case TopAbs_COMPSOLID:
196     return false;
197   case TopAbs_COMPOUND: {
198     TopExp_Explorer exp(theBase, TopAbs_SOLID);
199     return !exp.More();
200   }
201   default:
202     return false;
203   }
204   return false;
205 }
206
207 //=================================================================================
208 // function : SelectionIntoArgument()
209 // purpose  : Called when selection is changed or on dialog initialization or activation
210 //=================================================================================
211 void GenerationGUI_RevolDlg::SelectionIntoArgument()
212 {
213   erasePreview();
214   myEditCurrentArgument->setText("");
215   if      (myEditCurrentArgument == GroupPoints->LineEdit1) myOkBase = false;
216   else if (myEditCurrentArgument == GroupPoints->LineEdit2) myOkAxis = false;
217
218   LightApp_SelectionMgr* aSelMgr = myGeomGUI->getApp()->selectionMgr();
219   SALOME_ListIO aSelList;
220   aSelMgr->selectedObjects(aSelList);
221
222   if (aSelList.Extent() != 1)
223     return;
224
225   // nbSel == 1
226   Standard_Boolean testResult = Standard_False;
227   GEOM::GEOM_Object_ptr aSelectedObject = GEOMBase::ConvertIOinGEOMObject(aSelList.First(), testResult);
228
229   if (!testResult || aSelectedObject->_is_nil())
230     return;
231
232   TopoDS_Shape S;
233   if (!GEOMBase::GetShape(aSelectedObject, S) || S.IsNull())
234     return;
235
236   QString aName = GEOMBase::GetName(aSelectedObject);
237
238   if (myEditCurrentArgument == GroupPoints->LineEdit1) {
239     if (!isAcceptableBase(S))
240       return;
241
242     myBase = aSelectedObject;
243     myOkBase = true;
244     myEditCurrentArgument->setText(aName);
245     if (!myOkAxis)
246       GroupPoints->PushButton2->click();
247   }
248   else if (myEditCurrentArgument == GroupPoints->LineEdit2) {
249     TColStd_IndexedMapOfInteger aMap;
250     aSelMgr->GetIndexes(aSelList.First(), aMap);
251     if (aMap.Extent() == 1) {
252       int anIndex = aMap(1);
253       aName.append(":edge_" + QString::number(anIndex));
254
255       //Find SubShape Object in Father
256       GEOM::GEOM_Object_var aFindedObject = GEOMBase_Helper::findObjectInFather(aSelectedObject, aName);
257
258       if (aFindedObject == GEOM::GEOM_Object::_nil()) { // Object not found in study
259         GEOM::GEOM_IShapesOperations_var aShapesOp =
260           getGeomEngine()->GetIShapesOperations(getStudyId());
261         myAxis = aShapesOp->GetSubShape(aSelectedObject, anIndex);
262         myOkAxis = true;
263       }
264       else {
265         myAxis = aFindedObject;
266         myOkAxis = true;
267       }
268     }
269     else {
270       myOkAxis = true;
271       if (S.ShapeType() != TopAbs_EDGE) {
272         aSelectedObject = GEOM::GEOM_Object::_nil();
273         aName = "";
274         myOkAxis = false;
275       }
276       myAxis = aSelectedObject;
277     }
278     myEditCurrentArgument->setText(aName);
279     if (myOkAxis && !myOkBase)
280       GroupPoints->PushButton1->click();
281   }
282
283   // clear selection
284   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
285   myGeomGUI->getApp()->selectionMgr()->clearSelected();
286   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
287           this, SLOT(SelectionIntoArgument()));
288
289   displayPreview();
290 }
291
292 //=================================================================================
293 // function : SetEditCurrentArgument()
294 // purpose  :
295 //=================================================================================
296 void GenerationGUI_RevolDlg::SetEditCurrentArgument()
297 {
298   QPushButton* send = (QPushButton*)sender();
299
300   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
301   globalSelection(GEOM_ALLSHAPES);
302   if (send == GroupPoints->PushButton1) {
303     myEditCurrentArgument = GroupPoints->LineEdit1;
304     GroupPoints->PushButton2->setDown(false);
305     GroupPoints->LineEdit2->setEnabled(false);
306   }
307   else if (send == GroupPoints->PushButton2) {
308     myEditCurrentArgument = GroupPoints->LineEdit2;
309     GroupPoints->PushButton1->setDown(false);
310     GroupPoints->LineEdit1->setEnabled(false);
311     localSelection(GEOM::GEOM_Object::_nil(), TopAbs_EDGE);
312   }
313   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
314           this, SLOT(SelectionIntoArgument()));
315
316   // enable line edit
317   myEditCurrentArgument->setEnabled(true);
318   myEditCurrentArgument->setFocus();
319   // after setFocus(), because it will be setDown(false) when loses focus
320   send->setDown(true);
321
322   // seems we need it only to avoid preview disappearing, caused by selection mode change
323   displayPreview();
324 }
325
326 //=================================================================================
327 // function : LineEditReturnPressed()
328 // purpose  :
329 //=================================================================================
330 void GenerationGUI_RevolDlg::LineEditReturnPressed()
331 {
332   QLineEdit* send = (QLineEdit*)sender();
333   if (send == GroupPoints->LineEdit1 ||
334       send == GroupPoints->LineEdit2) {
335     myEditCurrentArgument = send;
336     GEOMBase_Skeleton::LineEditReturnPressed();
337   }
338 }
339
340 //=================================================================================
341 // function : ActivateThisDialog()
342 // purpose  :
343 //=================================================================================
344 void GenerationGUI_RevolDlg::ActivateThisDialog()
345 {
346   GEOMBase_Skeleton::ActivateThisDialog();
347   connect( myGeomGUI->getApp()->selectionMgr(), SIGNAL( currentSelectionChanged() ),
348            this, SLOT( SelectionIntoArgument() ) );
349
350   displayPreview();
351 }
352
353 //=================================================================================
354 // function : enterEvent()
355 // purpose  :
356 //=================================================================================
357 void GenerationGUI_RevolDlg::enterEvent (QEvent*)
358 {
359   if (!mainFrame()->GroupConstructors->isEnabled())
360     ActivateThisDialog();
361 }
362
363 //=================================================================================
364 // function : ValueChangedInSpinBox()
365 // purpose  :
366 //=================================================================================
367 void GenerationGUI_RevolDlg::ValueChangedInSpinBox()
368 {
369   displayPreview();
370 }
371
372 //=================================================================================
373 // function : getAngle()
374 // purpose  :
375 //=================================================================================
376 double GenerationGUI_RevolDlg::getAngle() const
377 {
378   return GroupPoints->SpinBox_DX->value();
379 }
380
381 //=================================================================================
382 // function : createOperation
383 // purpose  :
384 //=================================================================================
385 GEOM::GEOM_IOperations_ptr GenerationGUI_RevolDlg::createOperation()
386 {
387   return getGeomEngine()->GetI3DPrimOperations(getStudyId());
388 }
389
390 //=================================================================================
391 // function : isValid
392 // purpose  :
393 //=================================================================================
394 bool GenerationGUI_RevolDlg::isValid (QString& msg)
395 {
396   bool ok = GroupPoints->SpinBox_DX->isValid( msg, !IsPreview() );
397   return myOkBase && myOkAxis && ok;
398 }
399
400 //=================================================================================
401 // function : execute
402 // purpose  :
403 //=================================================================================
404 bool GenerationGUI_RevolDlg::execute (ObjectList& objects)
405 {
406   GEOM::GEOM_Object_var anObj;
407
408   if (!myBothway) {
409     anObj = GEOM::GEOM_I3DPrimOperations::_narrow(getOperation())->
410       MakeRevolutionAxisAngle(myBase, myAxis, getAngle() * PI180);
411   }
412   else {
413     anObj = GEOM::GEOM_I3DPrimOperations::_narrow(getOperation())->
414       MakeRevolutionAxisAngle2Ways(myBase, myAxis, getAngle() * PI180);
415   }
416
417   if (!anObj->_is_nil())
418   {
419     if (!IsPreview())
420     {
421       QStringList aParameters;
422       aParameters << GroupPoints->SpinBox_DX->text();
423       anObj->SetParameters(GeometryGUI::JoinObjectParameters(aParameters));
424     }
425     objects.push_back(anObj._retn());
426   }
427
428   return true;
429 }
430
431 //=================================================================================
432 // function :  onReverse()
433 // purpose  :
434 //=================================================================================
435 void GenerationGUI_RevolDlg::onReverse()
436 {
437   double anOldValue = GroupPoints->SpinBox_DX->value();
438   GroupPoints->SpinBox_DX->setValue(-anOldValue);
439 }
440
441 //=================================================================================
442 // function :  onBothway()
443 // purpose  :
444 //=================================================================================
445 void GenerationGUI_RevolDlg::onBothway()
446 {
447   bool anOldValue = myBothway;
448   myBothway = !anOldValue;
449   GroupPoints->CheckButton2->setEnabled(!myBothway);
450   displayPreview();
451 }
452
453 //=================================================================================
454 // function : addSubshapeToStudy
455 // purpose  : virtual method to add new SubObjects if local selection
456 //=================================================================================
457 void GenerationGUI_RevolDlg::addSubshapesToStudy()
458 {
459   QMap<QString, GEOM::GEOM_Object_var> objMap;
460
461   objMap[GroupPoints->LineEdit2->text()] = myAxis;
462
463   addSubshapesToFather(objMap);
464 }