]> SALOME platform Git repositories - modules/geom.git/blob - src/PrimitiveGUI/PrimitiveGUI_CylinderDlg.cxx
Salome HOME
Update of the implementation of an option to create portion of cylinders in the cylin...
[modules/geom.git] / src / PrimitiveGUI / PrimitiveGUI_CylinderDlg.cxx
1 // Copyright (C) 2007-2014  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, or (at your option) any later version.
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   : PrimitiveGUI_CylinderDlg.cxx
25 // Author : Lucien PIGNOLONI, Open CASCADE S.A.S.
26 //
27 #include "PrimitiveGUI_CylinderDlg.h"
28
29 #include <DlgRef.h>
30 #include <GeometryGUI.h>
31 #include <GEOMBase.h>
32
33 #include <SUIT_Session.h>
34 #include <SUIT_ResourceMgr.h>
35 #include <SalomeApp_Application.h>
36 #include <LightApp_SelectionMgr.h>
37
38 // OCCT Includes
39 #include <TopoDS_Shape.hxx>
40 #include <TopoDS_Edge.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    : PrimitiveGUI_CylinderDlg()
50 // purpose  : Constructs a PrimitiveGUI_CylinderDlg 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 PrimitiveGUI_CylinderDlg::PrimitiveGUI_CylinderDlg (GeometryGUI* theGeometryGUI, QWidget* parent,
56                                                     bool modal, Qt::WindowFlags fl)
57   : GEOMBase_Skeleton(theGeometryGUI, parent, modal, fl),
58     myInitial(true)
59 {
60   QPixmap image0 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_DLG_CYLINDER_PV")));
61   QPixmap image1 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_DLG_CYLINDER_DXYZ")));
62   QPixmap image2 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_SELECT")));
63
64   setWindowTitle(tr("GEOM_CYLINDER_TITLE"));
65
66   /***************************************************************/
67   mainFrame()->GroupConstructors->setTitle(tr("GEOM_CYLINDER"));
68   mainFrame()->RadioButton1->setIcon(image0);
69   mainFrame()->RadioButton2->setIcon(image1);
70   mainFrame()->RadioButton3->setAttribute(Qt::WA_DeleteOnClose);
71   mainFrame()->RadioButton3->close();
72
73   GroupPoints = new DlgRef_2Sel3Spin1Check(centralWidget());
74   GroupPoints->GroupBox1->setTitle(tr("GEOM_ARGUMENTS"));
75   GroupPoints->TextLabel1->setText(tr("GEOM_BASE_POINT"));
76   GroupPoints->TextLabel2->setText(tr("GEOM_VECTOR"));
77   GroupPoints->TextLabel3->setText(tr("GEOM_RADIUS"));
78   GroupPoints->TextLabel4->setText(tr("GEOM_HEIGHT"));
79   GroupPoints->TextLabel5->setText(tr("GEOM_ANGLE"));
80   GroupPoints->checkBox->setText(tr(""));
81   GroupPoints->PushButton1->setIcon(image2);
82   GroupPoints->PushButton2->setIcon(image2);
83   GroupPoints->SpinBox_DZ->setDisabled(true);
84
85   GroupDimensions = new DlgRef_3Spin1CheckCyl(centralWidget());
86   GroupDimensions->GroupBox1->setTitle(tr("GEOM_BOX_OBJ"));
87   GroupDimensions->TextLabel1->setText(tr("GEOM_RADIUS"));
88   GroupDimensions->TextLabel2->setText(tr("GEOM_HEIGHT"));
89   GroupDimensions->TextLabel3->setText(tr("GEOM_ANGLE"));
90   GroupDimensions->checkBox->setText(tr(""));
91   GroupDimensions->SpinBox_DZ->setDisabled(true);
92
93   QVBoxLayout* layout = new QVBoxLayout(centralWidget());
94   layout->setMargin(0); layout->setSpacing(6);
95   layout->addWidget(GroupPoints);
96   layout->addWidget(GroupDimensions);
97   /***************************************************************/
98
99   setHelpFileName("create_cylinder_page.html");
100
101   Init();
102 }
103
104 //=================================================================================
105 // function : ~PrimitiveGUI_CylinderDlg()
106 // purpose  : Destroys the object and frees any allocated resources
107 //=================================================================================
108 PrimitiveGUI_CylinderDlg::~PrimitiveGUI_CylinderDlg()
109 {
110   // no need to delete child widgets, Qt does it all for us
111 }
112
113 //=================================================================================
114 // function : Init()
115 // purpose  :
116 //=================================================================================
117 void PrimitiveGUI_CylinderDlg::Init()
118 {
119   // Get setting of step value from file configuration
120   SUIT_ResourceMgr* resMgr = SUIT_Session::session()->resourceMgr();
121   double step = resMgr->doubleValue("Geometry", "SettingsGeomStep", 100);
122
123   // min, max, step and decimals for spin boxes & initial values
124   double SpecificStep = 5;
125   initSpinBox(GroupPoints->SpinBox_DX, 0.00001, COORD_MAX, step, "length_precision" );
126   initSpinBox(GroupPoints->SpinBox_DY, 0.00001, COORD_MAX, step, "length_precision" );
127   initSpinBox(GroupPoints->SpinBox_DZ, 0.00001, 359.99999, SpecificStep, "angle_precision" );
128
129   initSpinBox(GroupDimensions->SpinBox_DX, 0.00001, COORD_MAX, step, "length_precision" );
130   initSpinBox(GroupDimensions->SpinBox_DY, 0.00001, COORD_MAX, step, "length_precision" );
131   initSpinBox(GroupDimensions->SpinBox_DZ, 0.00001, 359.99999, SpecificStep, "angle_precision" );
132
133   // init variables
134   myEditCurrentArgument = GroupPoints->LineEdit1;
135   GroupPoints->LineEdit1->setReadOnly(true);
136   GroupPoints->LineEdit2->setReadOnly(true);
137
138   GroupPoints->LineEdit1->setText("");
139   GroupPoints->LineEdit2->setText("");
140   myPoint.nullify();
141   myDir.nullify();
142
143   double aRadius(100.0), aHeight(300.0), aAngle(270.);
144   GroupPoints->SpinBox_DX->setValue(aRadius);
145   GroupPoints->SpinBox_DY->setValue(aHeight);
146   GroupPoints->SpinBox_DZ->setValue(aAngle);
147   GroupDimensions->SpinBox_DX->setValue(aRadius);
148   GroupDimensions->SpinBox_DY->setValue(aHeight);
149   GroupDimensions->SpinBox_DZ->setValue(aAngle);
150
151   // signals and slots connections
152   connect(buttonOk(),    SIGNAL(clicked()), this, SLOT(ClickOnOk()));
153   connect(buttonApply(), SIGNAL(clicked()), this, SLOT(ClickOnApply()));
154
155   connect(this,          SIGNAL(constructorsClicked(int)), this, SLOT(ConstructorsClicked(int)));
156
157   connect(GroupPoints->PushButton1, SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
158   connect(GroupPoints->PushButton2, SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
159
160   connect(GroupPoints->SpinBox_DX,     SIGNAL(valueChanged(double)), this, SLOT(ValueChangedInSpinBox()));
161   connect(GroupPoints->SpinBox_DY,     SIGNAL(valueChanged(double)), this, SLOT(ValueChangedInSpinBox()));
162   connect(GroupPoints->SpinBox_DZ,     SIGNAL(valueChanged(double)), this, SLOT(ValueChangedInSpinBox()));
163   connect(GroupDimensions->SpinBox_DX, SIGNAL(valueChanged(double)), this, SLOT(ValueChangedInSpinBox()));
164   connect(GroupDimensions->SpinBox_DY, SIGNAL(valueChanged(double)), this, SLOT(ValueChangedInSpinBox()));
165   connect(GroupDimensions->SpinBox_DZ, SIGNAL(valueChanged(double)), this, SLOT(ValueChangedInSpinBox()));
166
167   connect(myGeomGUI, SIGNAL(SignalDefaultStepValueChanged(double)), this, SLOT(SetDoubleSpinBoxStep(double)));
168   
169   connect(GroupPoints->checkBox, SIGNAL(toggled(bool)), this, SLOT(ActivateAngle()));
170   connect(GroupDimensions->checkBox, SIGNAL(toggled(bool)), this, SLOT(ActivateAngle()));
171   
172   initName(tr("GEOM_CYLINDER"));
173
174   setConstructorId(1); // simplest constructor
175   ConstructorsClicked(1);
176 }
177
178 //=================================================================================
179 // function : SetDoubleSpinBoxStep()
180 // purpose  : Double spin box management
181 //=================================================================================
182 void PrimitiveGUI_CylinderDlg::SetDoubleSpinBoxStep (double step)
183 {
184   GroupPoints->SpinBox_DX->setSingleStep(step);
185   GroupPoints->SpinBox_DY->setSingleStep(step);
186   GroupDimensions->SpinBox_DX->setSingleStep(step);
187   GroupDimensions->SpinBox_DY->setSingleStep(step);
188 }
189
190 //=================================================================================
191 // function : ConstructorsClicked()
192 // purpose  : Radio button management
193 //=================================================================================
194 void PrimitiveGUI_CylinderDlg::ConstructorsClicked (int constructorId)
195 {
196   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
197
198   switch (constructorId) {
199   case 0:
200     {
201       GroupDimensions->hide();
202       GroupPoints->show();
203
204       GroupPoints->PushButton1->click();
205       break;
206     }
207   case 1:
208     {
209       GroupPoints->hide();
210       GroupDimensions->show();
211       disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
212       globalSelection(); // close local contexts, if any
213       break;
214     }
215   }
216
217   qApp->processEvents();
218   updateGeometry();
219   resize(minimumSizeHint());
220   SelectionIntoArgument();
221
222   displayPreview(true);
223 }
224
225 //=================================================================================
226 // function : ClickOnOk()
227 // purpose  :
228 //=================================================================================
229 void PrimitiveGUI_CylinderDlg::ClickOnOk()
230 {
231   setIsApplyAndClose( true );
232   if (ClickOnApply())
233     ClickOnCancel();
234 }
235
236 //=================================================================================
237 // function : ClickOnApply()
238 // purpose  :
239 //=================================================================================
240 bool PrimitiveGUI_CylinderDlg::ClickOnApply()
241 {
242   if (!onAccept())
243     return false;
244
245   initName();
246   // activate selection and connect selection manager
247   ConstructorsClicked(getConstructorId());
248   return true;
249 }
250
251 //=================================================================================
252 // function : SelectionIntoArgument()
253 // purpose  : Called when selection is changed or on dialog initialization or activation
254 //=================================================================================
255 void PrimitiveGUI_CylinderDlg::SelectionIntoArgument()
256 {
257   if (getConstructorId() != 0)
258     return;
259
260   erasePreview();
261   myEditCurrentArgument->setText("");
262
263   LightApp_SelectionMgr* aSelMgr = myGeomGUI->getApp()->selectionMgr();
264   SALOME_ListIO aSelList;
265   aSelMgr->selectedObjects(aSelList);
266
267   if (aSelList.Extent() != 1) {
268     if (myEditCurrentArgument == GroupPoints->LineEdit1)
269       myPoint.nullify();
270     else if (myEditCurrentArgument == GroupPoints->LineEdit2)
271       myDir.nullify();
272     return;
273   }
274
275   TopAbs_ShapeEnum aNeedType = myEditCurrentArgument == GroupPoints->LineEdit2 ? TopAbs_EDGE : TopAbs_VERTEX;
276   GEOM::GeomObjPtr aSelectedObject = getSelected( aNeedType );
277   TopoDS_Shape aShape;
278   if ( aSelectedObject && GEOMBase::GetShape( aSelectedObject.get(), aShape ) && !aShape.IsNull() ) {
279     QString aName = GEOMBase::GetName( aSelectedObject.get() );
280
281     myEditCurrentArgument->setText(aName);
282     
283     if (myEditCurrentArgument == GroupPoints->LineEdit1) {
284       myPoint = aSelectedObject;
285       if (myPoint && !myDir)
286         GroupPoints->PushButton2->click();
287     }
288     else if (myEditCurrentArgument == GroupPoints->LineEdit2) {
289       myDir = aSelectedObject;
290       if (myDir && !myPoint)
291         GroupPoints->PushButton1->click();
292     }
293     // clear selection
294     if ((myDir && !myPoint) || (myPoint && !myDir)) {
295       disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
296       myGeomGUI->getApp()->selectionMgr()->clearSelected();
297       connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
298               this, SLOT(SelectionIntoArgument()));
299     }
300   }
301
302   displayPreview(true);
303 }
304
305 //=================================================================================
306 // function : SetEditCurrentArgument()
307 // purpose  :
308 //=================================================================================
309 void PrimitiveGUI_CylinderDlg::SetEditCurrentArgument()
310 {
311   QPushButton* send = (QPushButton*)sender();
312
313   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
314   if (send == GroupPoints->PushButton1) {
315     myEditCurrentArgument = GroupPoints->LineEdit1;
316
317     GroupPoints->PushButton2->setDown(false);
318     GroupPoints->LineEdit2->setEnabled(false);
319
320     globalSelection(GEOM_POINT); // to break previous local selection
321     localSelection(GEOM::GEOM_Object::_nil(), TopAbs_VERTEX);
322   }
323   else if (send == GroupPoints->PushButton2) {
324     myEditCurrentArgument = GroupPoints->LineEdit2;
325
326     GroupPoints->PushButton1->setDown(false);
327     GroupPoints->LineEdit1->setEnabled(false);
328
329     globalSelection(GEOM_LINE);  // to break previous local selection
330     localSelection(GEOM::GEOM_Object::_nil(), TopAbs_EDGE);
331   }
332   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
333           this, SLOT(SelectionIntoArgument()));
334
335   // enable line edit
336   myEditCurrentArgument->setEnabled(true);
337   myEditCurrentArgument->setFocus();
338   // after setFocus(), because it will be setDown(false) when loses focus
339   send->setDown(true);
340
341   // seems we need it only to avoid preview disappearing, caused by selection mode change
342   displayPreview(true);
343 }
344
345 //=================================================================================
346 // function : ActivateThisDialog()
347 // purpose  :
348 //=================================================================================
349 void PrimitiveGUI_CylinderDlg::ActivateThisDialog()
350 {
351   GEOMBase_Skeleton::ActivateThisDialog();
352
353   connect( myGeomGUI->getApp()->selectionMgr(), SIGNAL( currentSelectionChanged() ),
354            this, SLOT( SelectionIntoArgument() ) );
355
356   ConstructorsClicked( getConstructorId() );
357 }
358
359 //=================================================================================
360 // function : enterEvent()
361 // purpose  :
362 //=================================================================================
363 void PrimitiveGUI_CylinderDlg::enterEvent (QEvent*)
364 {
365   if (!mainFrame()->GroupConstructors->isEnabled())
366     ActivateThisDialog();
367 }
368
369 //=================================================================================
370 // function : ValueChangedInSpinBox()
371 // purpose  :
372 //=================================================================================
373 void PrimitiveGUI_CylinderDlg::ValueChangedInSpinBox()
374 {
375   displayPreview(true);
376 }
377
378 //=================================================================================
379 // function : createOperation
380 // purpose  :
381 //=================================================================================
382 GEOM::GEOM_IOperations_ptr PrimitiveGUI_CylinderDlg::createOperation()
383 {
384   return getGeomEngine()->GetI3DPrimOperations(getStudyId());
385 }
386
387 //=================================================================================
388 // function : isValid
389 // purpose  :
390 //=================================================================================
391 bool PrimitiveGUI_CylinderDlg::isValid (QString& msg)
392 {
393   bool ok = false;
394   if( getConstructorId() == 0 )
395   {
396     ok = GroupPoints->SpinBox_DX->isValid( msg, !IsPreview() ) &&
397          GroupPoints->SpinBox_DY->isValid( msg, !IsPreview() ) &&
398          GroupPoints->SpinBox_DZ->isValid( msg, !IsPreview() ) &&
399          myPoint && myDir;
400   }
401   else if( getConstructorId() == 1 )
402   {
403     ok = GroupDimensions->SpinBox_DX->isValid( msg, !IsPreview() ) &&
404          GroupDimensions->SpinBox_DY->isValid( msg, !IsPreview() ) &&
405          GroupDimensions->SpinBox_DZ->isValid( msg, !IsPreview() );
406   }
407   ok = qAbs( getHeight() ) > Precision::Confusion() && ok;
408   ok = qAbs( getRadius() ) > Precision::Confusion() && ok;
409   return ok;
410 }
411
412 //=================================================================================
413 // function : execute
414 // purpose  :
415 //=================================================================================
416 bool PrimitiveGUI_CylinderDlg::execute (ObjectList& objects)
417 {
418   bool res = false;
419   bool BAngle = false;
420   
421   GEOM::GEOM_Object_var anObj;
422
423   GEOM::GEOM_I3DPrimOperations_var anOper = GEOM::GEOM_I3DPrimOperations::_narrow(getOperation());
424
425   switch (getConstructorId()) {
426   case 0:
427     BAngle = GroupPoints->checkBox->isChecked();
428     if ( myPoint && myDir ) {
429       if(!BAngle){
430         anObj = anOper->MakeCylinderPntVecRH(myPoint.get(), myDir.get(), getRadius(), getHeight());
431         if (!anObj->_is_nil() && !IsPreview())
432         {
433           QStringList aParameters;
434           aParameters << GroupPoints->SpinBox_DX->text();
435           aParameters << GroupPoints->SpinBox_DY->text();
436           anObj->SetParameters(aParameters.join(":").toLatin1().constData());
437         }
438         res = true;
439       }
440       else if(BAngle){
441         anObj = anOper->MakeCylinderPntVecRHA(myPoint.get(), myDir.get(), getRadius(), getHeight(), getAngle()*M_PI/180.);
442         if (!anObj->_is_nil() && !IsPreview())
443         {
444           QStringList aParameters;
445           aParameters << GroupPoints->SpinBox_DX->text();
446           aParameters << GroupPoints->SpinBox_DY->text();
447           aParameters << GroupPoints->SpinBox_DZ->text();
448           anObj->SetParameters(aParameters.join(":").toLatin1().constData());
449         }
450         res = true;
451       }
452     }
453     break;
454   case 1:
455     BAngle = GroupDimensions->checkBox->isChecked();
456     if(!BAngle){
457       anObj = anOper->MakeCylinderRH(getRadius(), getHeight());
458       if (!anObj->_is_nil() && !IsPreview())
459       {
460         QStringList aParameters;
461         aParameters << GroupDimensions->SpinBox_DX->text();
462         aParameters << GroupDimensions->SpinBox_DY->text();
463         anObj->SetParameters(aParameters.join(":").toLatin1().constData());
464       }
465       res = true;
466     }
467     else if(BAngle){
468       anObj = anOper->MakeCylinderRHA(getRadius(), getHeight(), getAngle()*M_PI/180.);
469       if (!anObj->_is_nil() && !IsPreview())
470       {
471         QStringList aParameters;
472         aParameters << GroupDimensions->SpinBox_DX->text();
473         aParameters << GroupDimensions->SpinBox_DY->text();
474         aParameters << GroupDimensions->SpinBox_DZ->text();
475         anObj->SetParameters(aParameters.join(":").toLatin1().constData());
476       }
477       res = true;
478     }
479     break;
480   }
481
482   if (!anObj->_is_nil())
483     objects.push_back(anObj._retn());
484
485   return res;
486 }
487
488 //=================================================================================
489 // function : getRadius()
490 // purpose  :
491 //=================================================================================
492 double PrimitiveGUI_CylinderDlg::getRadius() const
493 {
494   int aConstructorId = getConstructorId();
495   if (aConstructorId == 0)
496     return GroupPoints->SpinBox_DX->value();
497   else if (aConstructorId == 1)
498     return GroupDimensions->SpinBox_DX->value();
499   return 0;
500 }
501
502 //=================================================================================
503 // function : getHeight()
504 // purpose  :
505 //=================================================================================
506 double PrimitiveGUI_CylinderDlg::getHeight() const
507 {
508   int aConstructorId = getConstructorId();
509   if (aConstructorId == 0)
510     return GroupPoints->SpinBox_DY->value();
511   else if (aConstructorId == 1)
512     return GroupDimensions->SpinBox_DY->value();
513   return 0;
514 }
515
516 //=================================================================================
517 // function : getAngle()
518 // purpose  :
519 //=================================================================================
520 double PrimitiveGUI_CylinderDlg::getAngle() const
521 {
522   int aConstructorId = getConstructorId();
523   if (aConstructorId == 0)
524     return GroupPoints->SpinBox_DZ->value();
525   else if (aConstructorId == 1)
526     return GroupDimensions->SpinBox_DZ->value();
527   return 0;
528 }
529
530 //=================================================================================
531 // function : addSubshapeToStudy
532 // purpose  : virtual method to add new SubObjects if local selection
533 //=================================================================================
534 void PrimitiveGUI_CylinderDlg::addSubshapesToStudy()
535 {
536   if ( getConstructorId() == 0 ) {
537     GEOMBase::PublishSubObject( myPoint.get() );
538     GEOMBase::PublishSubObject( myDir.get() );
539   }
540 }
541
542 //=================================================================================
543 // function :  ActivateAngle()
544 // purpose  : enable the 
545 //=================================================================================
546 void PrimitiveGUI_CylinderDlg::ActivateAngle()
547 {
548   int aConstructorId = getConstructorId();
549   if (aConstructorId == 0){
550     GroupPoints->SpinBox_DZ->setEnabled( GroupPoints->checkBox->isChecked() );
551     processPreview();
552   }
553   else if (aConstructorId == 1){
554     GroupDimensions->SpinBox_DZ->setEnabled( GroupDimensions->checkBox->isChecked() );
555     processPreview();
556   }
557 }