Salome HOME
Eliminate useless GetSubShape() lines in the python dump
[modules/geom.git] / src / TransformationGUI / TransformationGUI_ScaleDlg.cxx
1 //  Copyright (C) 2007-2010  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
23 // GEOM GEOMGUI : GUI for Geometry component
24 // File   : TransformationGUI_ScaleDlg.cxx
25 // Author : Lucien PIGNOLONI, Open CASCADE S.A.S.
26 //
27 #include "TransformationGUI_ScaleDlg.h"
28
29 #include <DlgRef.h>
30 #include <GeometryGUI.h>
31 #include <GEOMBase.h>
32
33 #include <SalomeApp_DoubleSpinBox.h>
34 #include <SUIT_Session.h>
35 #include <SUIT_ResourceMgr.h>
36 #include <SalomeApp_Application.h>
37 #include <LightApp_SelectionMgr.h>
38
39 // OCCT Includes
40 #include <TopoDS_Shape.hxx>
41 #include <TopoDS.hxx>
42 #include <TopExp.hxx>
43 #include <TColStd_IndexedMapOfInteger.hxx>
44 #include <TopTools_IndexedMapOfShape.hxx>
45
46 #include <GEOMImpl_Types.hxx>
47
48 //=================================================================================
49 // class    : TransformationGUI_ScaleDlg()
50 // purpose  : Constructs a TransformationGUI_ScaleDlg which is a child of 'parent', with the
51 //            name 'name' and widget flags set to 'f'.
52 //            The dialog will by default be modeless, unless you set 'modal' to
53 //            TRUE to construct a modal dialog.
54 //=================================================================================
55 TransformationGUI_ScaleDlg::TransformationGUI_ScaleDlg (GeometryGUI* theGeometryGUI, QWidget* parent,
56                                                         bool modal, Qt::WindowFlags fl)
57   : GEOMBase_Skeleton(theGeometryGUI, parent, modal, fl),
58     myInitial(true)
59 {
60   SUIT_ResourceMgr* aResMgr = myGeomGUI->getApp()->resourceMgr();
61   QPixmap image1 (aResMgr->loadPixmap("GEOM", tr("ICON_DLG_SCALE")));
62   QPixmap image2 (aResMgr->loadPixmap("GEOM", tr("ICON_DLG_SCALE_ALONG_AXES")));
63   QPixmap image3 (aResMgr->loadPixmap("GEOM", tr("ICON_SELECT")));
64
65   setWindowTitle(tr("GEOM_SCALE_TITLE"));
66
67   // Constructors
68   mainFrame()->GroupConstructors->setTitle(tr("GEOM_SCALE"));
69   mainFrame()->RadioButton1->setIcon(image1);
70   mainFrame()->RadioButton2->setIcon(image2);
71   mainFrame()->RadioButton3->setAttribute(Qt::WA_DeleteOnClose);
72   mainFrame()->RadioButton3->close();
73
74   // Own widgets
75   GroupBox1 = new QGroupBox(tr("GEOM_ARGUMENTS"), this);
76   QGridLayout* OwnLayout = new QGridLayout(GroupBox1);
77   OwnLayout->setSpacing(6);
78   OwnLayout->setMargin(11);
79
80   TextLabel1 = new QLabel(tr("GEOM_OBJECTS"), GroupBox1);
81   PushButton1 = new QPushButton(GroupBox1);
82   PushButton1->setIcon(image3);
83   PushButton1->setSizePolicy(QSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed));
84   LineEdit1 = new QLineEdit(GroupBox1);
85   LineEdit1->setReadOnly(true);
86   LineEdit1->setSizePolicy(QSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed));
87
88   TextLabel2 = new QLabel(tr("GEOM_CENTRAL_POINT"), GroupBox1);
89   PushButton2 = new QPushButton(GroupBox1);
90   PushButton2->setIcon(image3);
91   PushButton2->setSizePolicy(QSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed));
92   LineEdit2 = new QLineEdit(GroupBox1);
93   LineEdit2->setReadOnly(true);
94   LineEdit2->setSizePolicy(QSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed));
95
96   TextLabel3 = new QLabel(tr("GEOM_SCALE_FACTOR"), GroupBox1);
97   SpinBox_FX = new SalomeApp_DoubleSpinBox(GroupBox1);
98
99   TextLabel4 = new QLabel(tr("GEOM_SCALE_FACTOR_Y"), GroupBox1);
100   SpinBox_FY = new SalomeApp_DoubleSpinBox(GroupBox1);
101
102   TextLabel5 = new QLabel(tr("GEOM_SCALE_FACTOR_Z"), GroupBox1);
103   SpinBox_FZ = new SalomeApp_DoubleSpinBox(GroupBox1);
104
105   CheckBoxCopy = new QCheckBox(tr("GEOM_CREATE_COPY"), GroupBox1);
106   CheckBoxCopy->setChecked(true);
107
108   // Layouting
109   OwnLayout->addWidget(TextLabel1,   0, 0);
110   OwnLayout->addWidget(PushButton1,  0, 1);
111   OwnLayout->addWidget(LineEdit1,    0, 2, 1, 2);
112   OwnLayout->addWidget(TextLabel2,   1, 0);
113   OwnLayout->addWidget(PushButton2,  1, 1);
114   OwnLayout->addWidget(LineEdit2,    1, 2, 1, 2);
115   OwnLayout->addWidget(TextLabel3,   2, 0);
116   OwnLayout->addWidget(SpinBox_FX,   2, 2);
117   OwnLayout->addWidget(TextLabel4,   3, 0);
118   OwnLayout->addWidget(SpinBox_FY,   3, 2);
119   OwnLayout->addWidget(TextLabel5,   4, 0);
120   OwnLayout->addWidget(SpinBox_FZ,   4, 2);
121   OwnLayout->addWidget(CheckBoxCopy, 5, 0, 1, 4);
122
123   QVBoxLayout* layout = new QVBoxLayout(centralWidget());
124   layout->setMargin(0); layout->setSpacing(6);
125   layout->addWidget(GroupBox1);
126
127   setHelpFileName("scale_operation_page.html");
128
129   // Modification of an existing object by offset is not allowed
130   CheckBoxCopy->hide();
131
132   // Activate Create a Copy mode
133   CreateCopyModeChanged();
134
135   // Allowed inheritance of children and visual properties by the scaling result
136   mainFrame()->GroupBoxPublish->show();
137
138   Init();
139 }
140
141 //=================================================================================
142 // function : ~TransformationGUI_ScaleDlg()
143 // purpose  : Destroys the object and frees any allocated resources
144 //=================================================================================
145 TransformationGUI_ScaleDlg::~TransformationGUI_ScaleDlg()
146 {
147 }
148
149 //=================================================================================
150 // function : Init()
151 // purpose  :
152 //=================================================================================
153 void TransformationGUI_ScaleDlg::Init()
154 {
155   // Min, max, step and decimals for spin boxes & initial values
156   double aFactor = 2.0;
157   double SpecificStep = 0.5;
158
159   initSpinBox(SpinBox_FX, COORD_MIN, COORD_MAX, SpecificStep, "parametric_precision" );
160   initSpinBox(SpinBox_FY, COORD_MIN, COORD_MAX, SpecificStep, "parametric_precision" );
161   initSpinBox(SpinBox_FZ, COORD_MIN, COORD_MAX, SpecificStep, "parametric_precision" );
162   SpinBox_FX->setValue(aFactor);
163   SpinBox_FY->setValue(aFactor);
164   SpinBox_FZ->setValue(aFactor);
165
166   // init variables
167   LineEdit1->setText("");
168   LineEdit2->setText("");
169
170   myObjects.clear();
171   myPoint.nullify();
172
173   // Signals and slots connections
174   connect(buttonOk(),        SIGNAL(clicked()), this, SLOT(ClickOnOk()));
175   connect(buttonApply(),     SIGNAL(clicked()), this, SLOT(ClickOnApply()));
176   connect(this, SIGNAL(constructorsClicked(int)), this, SLOT(ConstructorsClicked(int)));
177
178   connect(PushButton1, SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
179   connect(PushButton2, SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
180
181   connect(SpinBox_FX, SIGNAL(valueChanged(double)), this, SLOT(ValueChangedInSpinBox()));
182   connect(SpinBox_FY, SIGNAL(valueChanged(double)), this, SLOT(ValueChangedInSpinBox()));
183   connect(SpinBox_FZ, SIGNAL(valueChanged(double)), this, SLOT(ValueChangedInSpinBox()));
184   
185   // san : Commented so as not to override specific step settings
186   //connect(myGeomGUI, SIGNAL(SignalDefaultStepValueChanged(double)), this, SLOT(SetDoubleSpinBoxStep(double)));
187
188   connect(CheckBoxCopy, SIGNAL(toggled(bool)), this, SLOT(CreateCopyModeChanged()));
189
190   initName(tr("GEOM_SCALE"));
191
192   ConstructorsClicked(0);
193 }
194
195 //=================================================================================
196 // function : ConstructorsClicked()
197 // purpose  : Radio button management
198 //=================================================================================
199 void TransformationGUI_ScaleDlg::ConstructorsClicked (int constructorId)
200 {
201   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
202
203   myPoint.nullify();
204   LineEdit2->clear();
205   
206   switch (constructorId) {
207   case 0: // translation an object by dx, dy, dz
208     TextLabel3->setText(tr("GEOM_SCALE_FACTOR"));
209     TextLabel4->hide();
210     TextLabel5->hide();
211     SpinBox_FY->hide();
212     SpinBox_FZ->hide();
213
214     PushButton1->click();
215     break;
216   case 1: // translation an object by 2 points
217     TextLabel3->setText(tr("GEOM_SCALE_FACTOR_X"));
218     TextLabel4->show();
219     TextLabel5->show();
220     SpinBox_FY->show();
221     SpinBox_FZ->show();
222
223     PushButton1->click();
224     break;
225   default:
226     break;
227   }
228
229   qApp->processEvents();
230   updateGeometry();
231   resize(minimumSizeHint());
232
233   if (myInitial) {
234     myInitial = false;
235     SelectionIntoArgument();
236   }
237   else {
238     displayPreview();
239   }
240 }
241
242 //=================================================================================
243 // function : ClickOnOk()
244 // purpose  :
245 //=================================================================================
246 void TransformationGUI_ScaleDlg::ClickOnOk()
247 {
248   if (ClickOnApply())
249     ClickOnCancel();
250 }
251
252 //=================================================================================
253 // function : ClickOnApply()
254 // purpose  :
255 //=================================================================================
256 bool TransformationGUI_ScaleDlg::ClickOnApply()
257 {
258   if (!onAccept(CheckBoxCopy->isChecked()))
259     return false;
260
261   initName(tr("GEOM_SCALE"));
262
263   myObjects.clear();
264   myEditCurrentArgument = LineEdit1;
265   myEditCurrentArgument->setText("");
266   myGeomGUI->getApp()->selectionMgr()->clearSelected();
267
268   // activate selection and connect selection manager
269   ConstructorsClicked(getConstructorId());
270   return true;
271 }
272
273 //=================================================================================
274 // function : SelectionIntoArgument()
275 // purpose  : Called when selection is changed or on dialog initialization or activation
276 //=================================================================================
277 void TransformationGUI_ScaleDlg::SelectionIntoArgument()
278 {
279   erasePreview();
280
281   if ( myEditCurrentArgument == LineEdit1 ) {
282     myObjects = getSelected( TopAbs_SHAPE, -1 );
283     if ( !myObjects.isEmpty() ) {
284       QString aName = myObjects.count() > 1 ? QString( "%1_objects").arg( myObjects.count() ) : GEOMBase::GetName( myObjects[0].get() );
285       myEditCurrentArgument->setText( aName );
286     }
287     else {
288       myEditCurrentArgument->setText("");
289     }
290   }
291   else {
292     GEOM::GeomObjPtr aSelectedObject = getSelected( TopAbs_VERTEX );
293     TopoDS_Shape aShape;
294     if ( aSelectedObject && GEOMBase::GetShape( aSelectedObject.get(), aShape ) && !aShape.IsNull() ) {
295       QString aName = GEOMBase::GetName( aSelectedObject.get() );
296       myEditCurrentArgument->setText( aName );
297       myPoint = aSelectedObject;
298       if ( myObjects.isEmpty() )
299         PushButton1->click();
300     }
301     else {
302       myPoint.nullify();
303       myEditCurrentArgument->setText("");
304     }
305   }
306
307   displayPreview();
308 }
309
310 //=================================================================================
311 // function : SetEditCurrentArgument()
312 // purpose  :
313 //=================================================================================
314 void TransformationGUI_ScaleDlg::SetEditCurrentArgument()
315 {
316   QPushButton* send = (QPushButton*)sender();
317
318   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
319   globalSelection();
320
321   if (send == PushButton1) {
322     myEditCurrentArgument = LineEdit1;
323
324     PushButton2->setDown(false);
325     LineEdit2->setEnabled(false);
326   }
327   else if (send == PushButton2) {
328     myEditCurrentArgument = LineEdit2;
329
330     localSelection(GEOM::GEOM_Object::_nil(), TopAbs_VERTEX);
331
332     PushButton1->setDown(false);
333     LineEdit1->setEnabled(false);
334   }
335   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
336           this, SLOT(SelectionIntoArgument()));
337
338   // enable line edit
339   myEditCurrentArgument->setEnabled(true);
340   myEditCurrentArgument->setFocus();
341   // after setFocus(), because it will be setDown(false) when loses focus
342   send->setDown(true);
343
344   // seems we need it only to avoid preview disappearing, caused by selection mode change
345   displayPreview();
346 }
347
348 //=================================================================================
349 // function : ActivateThisDialog()
350 // purpose  :
351 //=================================================================================
352 void TransformationGUI_ScaleDlg::ActivateThisDialog()
353 {
354   GEOMBase_Skeleton::ActivateThisDialog();
355   connect( myGeomGUI->getApp()->selectionMgr(), SIGNAL( currentSelectionChanged() ),
356            this, SLOT( SelectionIntoArgument() ) );
357
358   ConstructorsClicked( getConstructorId() );
359 }
360
361 //=================================================================================
362 // function : enterEvent()
363 // purpose  :
364 //=================================================================================
365 void TransformationGUI_ScaleDlg::enterEvent (QEvent*)
366 {
367   if (!mainFrame()->GroupConstructors->isEnabled())
368     ActivateThisDialog();
369 }
370
371 //=================================================================================
372 // function : ValueChangedInSpinBox()
373 // purpose  :
374 //=================================================================================
375 void TransformationGUI_ScaleDlg::ValueChangedInSpinBox()
376 {
377   displayPreview();
378 }
379
380 //=================================================================================
381 // function : SetDoubleSpinBoxStep()
382 // purpose  : Double spin box management
383 //=================================================================================
384 void TransformationGUI_ScaleDlg::SetDoubleSpinBoxStep(double step)
385 {
386   SpinBox_FX->setSingleStep(step);
387   SpinBox_FY->setSingleStep(step);
388   SpinBox_FZ->setSingleStep(step);
389 }
390
391 //=================================================================================
392 // function : createOperation
393 // purpose  :
394 //=================================================================================
395 GEOM::GEOM_IOperations_ptr TransformationGUI_ScaleDlg::createOperation()
396 {
397   return getGeomEngine()->GetITransformOperations(getStudyId());
398 }
399
400 //=================================================================================
401 // function : isValid
402 // purpose  :
403 //=================================================================================
404 bool TransformationGUI_ScaleDlg::isValid (QString& msg)
405 {
406   bool ok = false;
407   if ( getConstructorId() == 0 ) {
408     ok = SpinBox_FX->isValid( msg, !IsPreview() ) &&
409          !myObjects.isEmpty() &&
410          qAbs( SpinBox_FX->value() ) > 0.00001;
411   }
412   else {
413     ok = SpinBox_FX->isValid( msg, !IsPreview() ) &&
414          SpinBox_FY->isValid( msg, !IsPreview() ) &&
415          SpinBox_FZ->isValid( msg, !IsPreview() ) &&
416          !myObjects.isEmpty() &&
417          qAbs( SpinBox_FX->value() ) > 0.00001 &&
418          qAbs( SpinBox_FY->value() ) > 0.00001 &&
419          qAbs( SpinBox_FZ->value() ) > 0.00001;
420   }
421   return ok;
422 }
423
424 //=================================================================================
425 // function : execute
426 // purpose  :
427 //=================================================================================
428 bool TransformationGUI_ScaleDlg::execute (ObjectList& objects)
429 {
430   bool toCreateCopy = IsPreview() || CheckBoxCopy->isChecked();
431
432   GEOM::GEOM_Object_var anObj;
433
434   GEOM::GEOM_ITransformOperations_var anOper = GEOM::GEOM_ITransformOperations::_narrow(getOperation());
435
436   switch (getConstructorId())
437   {
438   case 0:
439     {
440       if (toCreateCopy)
441       {
442         for (int i = 0; i < myObjects.count(); i++)
443         {
444           anObj = anOper->ScaleShapeCopy(myObjects[i].get(), myPoint.get(), SpinBox_FX->value());
445           if (!anObj->_is_nil()) {
446             if(!IsPreview()) 
447               anObj->SetParameters(SpinBox_FX->text().toLatin1().constData());
448             objects.push_back(anObj._retn());
449           }
450         }
451       }
452       else
453       {
454         for (int i = 0; i < myObjects.count(); i++)
455         {
456           anObj = anOper->ScaleShape(myObjects[i].get(), myPoint.get(), SpinBox_FX->value());
457           if (!anObj->_is_nil())
458             objects.push_back(anObj._retn());
459         }
460       }
461     }
462     break;
463   case 1:
464     {
465       if (toCreateCopy)
466       {
467         for (int i = 0; i < myObjects.count(); i++)
468         {
469           anObj = anOper->ScaleShapeAlongAxesCopy(myObjects[i].get(), myPoint.get(), SpinBox_FX->value(),
470                                                   SpinBox_FY->value(), SpinBox_FZ->value());
471           if (!anObj->_is_nil())
472             if(!IsPreview()) {
473               QStringList aParameters;
474               aParameters<<SpinBox_FX->text();
475               aParameters<<SpinBox_FY->text();
476               aParameters<<SpinBox_FZ->text();
477               anObj->SetParameters(aParameters.join(":").toLatin1().constData());
478             }
479             objects.push_back(anObj._retn());
480         }
481       }
482       else
483       {
484         for (int i = 0; i < myObjects.count(); i++)
485         {
486           anObj = anOper->ScaleShapeAlongAxes(myObjects[i].get(), myPoint.get(), SpinBox_FX->value(),
487                                               SpinBox_FY->value(), SpinBox_FZ->value());
488           if (!anObj->_is_nil())
489             objects.push_back(anObj._retn());
490         }
491       }
492     }
493     break;
494   default:
495     break;
496   }
497
498   return true;
499 }
500
501 //=================================================================================
502 // function : restoreSubShapes
503 // purpose  :
504 //=================================================================================
505 void TransformationGUI_ScaleDlg::restoreSubShapes (SALOMEDS::Study_ptr   theStudy,
506                                                    SALOMEDS::SObject_ptr theSObject)
507 {
508   if (mainFrame()->CheckBoxRestoreSS->isChecked()) {
509     // empty list of arguments means that all arguments should be restored
510     getGeomEngine()->RestoreSubShapesSO(theStudy, theSObject, GEOM::ListOfGO(),
511                                         /*theFindMethod=*/GEOM::FSM_Transformed,
512                                         /*theInheritFirstArg=*/true,
513                                         mainFrame()->CheckBoxAddPrefix->isChecked());
514   }
515 }
516
517 //=================================================================================
518 // function :  CreateCopyModeChanged()
519 // purpose  :
520 //=================================================================================
521 void TransformationGUI_ScaleDlg::CreateCopyModeChanged()
522 {
523   mainFrame()->GroupBoxName->setEnabled(CheckBoxCopy->isChecked());
524 }
525
526 //=================================================================================
527 // function : addSubshapesToStudy
528 // purpose  : virtual method to add new SubObjects if local selection
529 //=================================================================================
530 void TransformationGUI_ScaleDlg::addSubshapesToStudy()
531 {
532   bool toCreateCopy = IsPreview() || CheckBoxCopy->isChecked();
533   if (toCreateCopy)
534   {
535     GEOMBase::PublishSubObject( myPoint.get() );
536   }
537 }