Salome HOME
0016756: EDF 545 GEOM : Extrusion, give the coordinates of the vector as input
[modules/geom.git] / src / PrimitiveGUI / PrimitiveGUI_TorusDlg.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   : PrimitiveGUI_TorusDlg.cxx
23 // Author : Lucien PIGNOLONI, Open CASCADE S.A.S.
24 //
25
26 #include "PrimitiveGUI_TorusDlg.h"
27
28 #include <DlgRef.h>
29 #include <GeometryGUI.h>
30 #include <GEOMBase.h>
31
32 #include <SUIT_ResourceMgr.h>
33 #include <SUIT_Session.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
44 #include <GEOMImpl_Types.hxx>
45
46 //=================================================================================
47 // class    : PrimitiveGUI_TorusDlg()
48 // purpose  : Constructs a PrimitiveGUI_TorusDlg which is a child of 'parent', with the 
49 //            name 'name' and widget flags set to 'f'.
50 //            The dialog will by default be modeless, unless you set 'modal' to
51 //            TRUE to construct a modal dialog.
52 //=================================================================================
53 PrimitiveGUI_TorusDlg::PrimitiveGUI_TorusDlg( GeometryGUI* theGeometryGUI, QWidget* parent,
54                                               bool modal, Qt::WindowFlags fl )
55   : GEOMBase_Skeleton( theGeometryGUI, parent, modal, fl )
56 {
57   SUIT_ResourceMgr* aResMgr = SUIT_Session::session()->resourceMgr();
58   QPixmap image0( aResMgr->loadPixmap( "GEOM", tr( "ICON_DLG_TORUS_PV" ) ) );
59   QPixmap image1( aResMgr->loadPixmap( "GEOM", tr( "ICON_DLG_TORUS_DXYZ" ) ) );
60   QPixmap image2( aResMgr->loadPixmap( "GEOM", tr( "ICON_SELECT" ) ) );
61
62   setWindowTitle( tr( "GEOM_TORUS_TITLE" ) );
63
64   /***************************************************************/
65   mainFrame()->GroupConstructors->setTitle( tr( "GEOM_TORUS" ) );
66   mainFrame()->RadioButton1->setIcon( image0 );
67   mainFrame()->RadioButton2->setIcon( image1 );
68   mainFrame()->RadioButton3->setAttribute( Qt::WA_DeleteOnClose );
69   mainFrame()->RadioButton3->close();
70
71   GroupPoints = new DlgRef_2Sel2Spin( centralWidget() );
72   GroupPoints->GroupBox1->setTitle( tr( "GEOM_ARGUMENTS" ) );
73   GroupPoints->TextLabel1->setText( tr( "GEOM_BASE_POINT" ) );
74   GroupPoints->TextLabel2->setText( tr( "GEOM_VECTOR" ) );
75   GroupPoints->TextLabel3->setText( tr( "GEOM_RADIUS_I" ).arg( 1 ) );
76   GroupPoints->TextLabel4->setText( tr( "GEOM_RADIUS_I" ).arg( 2 ) );
77   GroupPoints->PushButton1->setIcon( image2 );
78   GroupPoints->PushButton2->setIcon( image2 );
79
80   GroupDimensions = new DlgRef_2Spin( centralWidget() );
81   GroupDimensions->GroupBox1->setTitle( tr( "GEOM_BOX_OBJ" ) );
82   GroupDimensions->TextLabel1->setText( tr( "GEOM_RADIUS_I" ).arg( 1 ) );
83   GroupDimensions->TextLabel2->setText( tr( "GEOM_RADIUS_I" ).arg( 2 ) );
84
85   QVBoxLayout* layout = new QVBoxLayout( centralWidget() );
86   layout->setMargin( 0 ); layout->setSpacing( 6 );
87   layout->addWidget( GroupPoints );
88   layout->addWidget( GroupDimensions );
89   /***************************************************************/
90
91   setHelpFileName( "create_torus_page.html" );
92   
93   Init();
94 }
95
96
97 //=================================================================================
98 // function : ~PrimitiveGUI_TorusDlg()
99 // purpose  : Destroys the object and frees any allocated resources
100 //=================================================================================
101 PrimitiveGUI_TorusDlg::~PrimitiveGUI_TorusDlg()
102 {
103   // no need to delete child widgets, Qt does it all for us
104 }
105
106
107 //=================================================================================
108 // function : Init()
109 // purpose  :
110 //=================================================================================
111 void PrimitiveGUI_TorusDlg::Init()
112 {
113   /* init variables */
114   myEditCurrentArgument = GroupPoints->LineEdit1;
115   GroupPoints->LineEdit1->setReadOnly( true );
116   GroupPoints->LineEdit2->setReadOnly( true );
117  
118   myPoint = myDir = GEOM::GEOM_Object::_nil();
119
120   /* Get setting of step value from file configuration */
121   SUIT_ResourceMgr* resMgr = SUIT_Session::session()->resourceMgr();
122   double step = resMgr->doubleValue( "Geometry", "SettingsGeomStep", 100 );
123
124   /* min, max, step and decimals for spin boxes & initial values */
125   initSpinBox( GroupPoints->SpinBox_DX, 0.001, COORD_MAX, step, 3 ); // VSR: TODO: DBL_DIGITS_DISPLAY
126   initSpinBox( GroupPoints->SpinBox_DY, 0.001, COORD_MAX, step, 3 ); // VSR: TODO: DBL_DIGITS_DISPLAY
127   initSpinBox( GroupDimensions->SpinBox_DX, 0.001, COORD_MAX, step, 3 ); // VSR: TODO: DBL_DIGITS_DISPLAY
128   initSpinBox( GroupDimensions->SpinBox_DY, 0.001, COORD_MAX, step, 3 ); // VSR: TODO: DBL_DIGITS_DISPLAY
129
130   GroupPoints->SpinBox_DX->setValue( 300.0 );
131   GroupPoints->SpinBox_DY->setValue( 100.0 );
132   GroupDimensions->SpinBox_DX->setValue( 300.0 );
133   GroupDimensions->SpinBox_DY->setValue( 100.0 );
134
135   /* signals and slots connections */
136   connect( buttonOk(),    SIGNAL( clicked() ), this, SLOT( ClickOnOk() ) );
137   connect( buttonApply(), SIGNAL( clicked() ), this, SLOT( ClickOnApply() ) );
138
139   connect( this,          SIGNAL( constructorsClicked( int ) ), this, SLOT( ConstructorsClicked( int ) ) );
140
141   connect( GroupPoints->PushButton1, SIGNAL( clicked() ), this, SLOT( SetEditCurrentArgument() ) );
142   connect( GroupPoints->PushButton2, SIGNAL( clicked() ), this, SLOT( SetEditCurrentArgument() ) );
143
144   connect( GroupPoints->LineEdit1, SIGNAL( returnPressed() ), this, SLOT( LineEditReturnPressed() ) );
145   connect( GroupPoints->LineEdit2, SIGNAL( returnPressed() ), this, SLOT( LineEditReturnPressed() ) );
146
147   connect( GroupPoints->SpinBox_DX,     SIGNAL( valueChanged( double ) ), this, SLOT( ValueChangedInSpinBox() ) );
148   connect( GroupPoints->SpinBox_DY,     SIGNAL( valueChanged( double ) ), this, SLOT( ValueChangedInSpinBox() ) );
149   connect( GroupDimensions->SpinBox_DX, SIGNAL( valueChanged( double ) ), this, SLOT( ValueChangedInSpinBox() ) );
150   connect( GroupDimensions->SpinBox_DY, SIGNAL( valueChanged( double ) ), this, SLOT( ValueChangedInSpinBox() ) );
151
152   connect( myGeomGUI, SIGNAL( SignalDefaultStepValueChanged( double ) ), this, SLOT( SetDoubleSpinBoxStep( double ) ) );
153   
154   connect( myGeomGUI->getApp()->selectionMgr(), SIGNAL( currentSelectionChanged() ),
155            this, SLOT( SelectionIntoArgument() ) );
156   
157   initName( tr( "GEOM_TORUS" ) );
158
159   ConstructorsClicked( 0 );
160 }
161
162 //=================================================================================
163 // function : SetDoubleSpinBoxStep()
164 // purpose  : Double spin box management
165 //=================================================================================
166 void PrimitiveGUI_TorusDlg::SetDoubleSpinBoxStep( double step )
167 {
168   GroupPoints->SpinBox_DX->setSingleStep(step);
169   GroupPoints->SpinBox_DY->setSingleStep(step);
170   GroupDimensions->SpinBox_DX->setSingleStep(step);
171   GroupDimensions->SpinBox_DY->setSingleStep(step);
172 }
173
174
175 //=================================================================================
176 // function : ConstructorsClicked()
177 // purpose  : Radio button management
178 //=================================================================================
179 void PrimitiveGUI_TorusDlg::ConstructorsClicked( int constructorId )
180 {
181   disconnect( myGeomGUI->getApp()->selectionMgr(), 0, this, 0 );
182   
183   switch( constructorId ) {
184   case 0 :
185     {
186       globalSelection(); // close local contexts, if any
187       localSelection( GEOM::GEOM_Object::_nil(), TopAbs_VERTEX );
188       
189       GroupDimensions->hide();
190       GroupPoints->show();
191       
192       myEditCurrentArgument = GroupPoints->LineEdit1;
193       GroupPoints->LineEdit1->setText( "" );
194       GroupPoints->LineEdit2->setText( "" );
195       myPoint = myDir = GEOM::GEOM_Object::_nil();
196       
197       connect( myGeomGUI->getApp()->selectionMgr(),
198                SIGNAL( currentSelectionChanged() ), this, SLOT( SelectionIntoArgument() ) );
199       SelectionIntoArgument();
200       break;
201     }
202   case 1:
203     {
204       GroupPoints->hide();
205       GroupDimensions->show();
206       globalSelection(); // close local contexts, if any
207       
208       break ;
209     }
210   }
211   
212   qApp->processEvents();
213   updateGeometry();
214   resize( minimumSize() );
215
216   displayPreview();
217 }
218
219
220 //=================================================================================
221 // function : ClickOnOk()
222 // purpose  :
223 //=================================================================================
224 void PrimitiveGUI_TorusDlg::ClickOnOk()
225 {
226   if ( ClickOnApply() )
227     ClickOnCancel();
228 }
229
230
231 //=================================================================================
232 // function : ClickOnApply()
233 // purpose  :
234 //=================================================================================
235 bool PrimitiveGUI_TorusDlg::ClickOnApply()
236 {
237   if ( !onAccept() )
238     return false;
239
240   initName();
241
242   return true;
243 }
244
245
246 //=================================================================================
247 // function : SelectionIntoArgument()
248 // purpose  : Called when selection as changed or other case
249 //=================================================================================
250 void PrimitiveGUI_TorusDlg::SelectionIntoArgument()
251 {
252   if ( getConstructorId() != 0 )
253     return;
254   
255   myEditCurrentArgument->setText( "" );
256   
257   if ( IObjectCount() != 1 ) {
258     if ( myEditCurrentArgument == GroupPoints->LineEdit1 )
259       myPoint = GEOM::GEOM_Object::_nil();
260     else if ( myEditCurrentArgument == GroupPoints->LineEdit2 )
261       myDir = GEOM::GEOM_Object::_nil();
262     return;
263   }
264   
265   /* nbSel == 1 */
266   Standard_Boolean testResult = Standard_False;
267   GEOM::GEOM_Object_var aSelectedObject = GEOMBase::ConvertIOinGEOMObject( firstIObject(), testResult );
268   
269   if ( !testResult || CORBA::is_nil( aSelectedObject ) )
270     return;
271   
272
273   QString aName = GEOMBase::GetName(aSelectedObject);
274   TopoDS_Shape aShape;
275   if ( GEOMBase::GetShape( aSelectedObject, aShape, TopAbs_SHAPE ) && !aShape.IsNull() ) {
276     TopAbs_ShapeEnum aNeedType = TopAbs_VERTEX;
277     if ( myEditCurrentArgument == GroupPoints->LineEdit2 )
278       aNeedType = TopAbs_EDGE;
279
280     LightApp_SelectionMgr* aSelMgr = myGeomGUI->getApp()->selectionMgr();
281     TColStd_IndexedMapOfInteger aMap;
282     aSelMgr->GetIndexes( firstIObject(), aMap );
283     if ( aMap.Extent() == 1 ) { // Local Selection
284       int anIndex = aMap( 1 );
285       if ( aNeedType == TopAbs_EDGE )
286         aName.append( ":edge_" + QString::number( anIndex ) );
287       else
288         aName.append( ":vertex_" + QString::number( anIndex ) );
289
290       //Find SubShape Object in Father
291       GEOM::GEOM_Object_var aFindedObject = GEOMBase_Helper::findObjectInFather( aSelectedObject, aName );
292
293       if ( aFindedObject == GEOM::GEOM_Object::_nil() ) { // Object not found in study
294         GEOM::GEOM_IShapesOperations_var aShapesOp =
295           getGeomEngine()->GetIShapesOperations( getStudyId() );
296         aSelectedObject = aShapesOp->GetSubShape( aSelectedObject, anIndex );
297       }
298       else
299         aSelectedObject = aFindedObject; // get Object from study
300     }
301     else { // Global Selection
302       if ( aShape.ShapeType() != aNeedType ) {
303         aSelectedObject = GEOM::GEOM_Object::_nil();
304         aName = "";
305       }
306     }
307   }
308
309   myEditCurrentArgument->setText(aName);
310
311   if ( myEditCurrentArgument == GroupPoints->LineEdit1 )
312     myPoint = aSelectedObject;
313   else if ( myEditCurrentArgument == GroupPoints->LineEdit2 )
314     myDir = aSelectedObject;
315     
316   displayPreview();
317 }
318
319
320 //=================================================================================
321 // function : SetEditCurrentArgument()
322 // purpose  :
323 //=================================================================================
324 void PrimitiveGUI_TorusDlg::SetEditCurrentArgument()
325 {
326   QPushButton* send = (QPushButton*)sender();
327   globalSelection( GEOM_POINT ); // to break previous local selection
328   
329   if ( send == GroupPoints->PushButton1 ) {
330     myEditCurrentArgument = GroupPoints->LineEdit1;
331     localSelection( GEOM::GEOM_Object::_nil(), TopAbs_VERTEX );
332   }
333   else if ( send == GroupPoints->PushButton2 ) {
334     myEditCurrentArgument = GroupPoints->LineEdit2;
335     GEOM::GEOM_Object_var anObj;
336     localSelection( anObj, TopAbs_EDGE );
337   }
338   
339   myEditCurrentArgument->setFocus();
340   SelectionIntoArgument();
341 }
342
343
344 //=================================================================================
345 // function : LineEditReturnPressed()
346 // purpose  :
347 //=================================================================================
348 void PrimitiveGUI_TorusDlg::LineEditReturnPressed()
349 {  
350   QLineEdit* send = (QLineEdit*)sender();
351   if ( send == GroupPoints->LineEdit1 ||
352        send == GroupPoints->LineEdit2 ) {
353     myEditCurrentArgument = send;
354     GEOMBase_Skeleton::LineEditReturnPressed();
355   }
356 }
357
358
359 //=================================================================================
360 // function : ActivateThisDialog()
361 // purpose  :
362 //=================================================================================
363 void PrimitiveGUI_TorusDlg::ActivateThisDialog()
364 {
365   GEOMBase_Skeleton::ActivateThisDialog();
366   connect( myGeomGUI->getApp()->selectionMgr(),
367            SIGNAL( currentSelectionChanged() ), this, SLOT( SelectionIntoArgument() ) );
368  
369   ConstructorsClicked( getConstructorId() );
370 }
371
372
373 //=================================================================================
374 // function : enterEvent()
375 // purpose  :
376 //=================================================================================
377 void PrimitiveGUI_TorusDlg::enterEvent( QEvent* )
378 {
379   if ( !mainFrame()->GroupConstructors->isEnabled() )
380     ActivateThisDialog();
381 }
382
383
384 //=================================================================================
385 // function : ValueChangedInSpinBox
386 // purpose  :
387 //=================================================================================
388 void PrimitiveGUI_TorusDlg::ValueChangedInSpinBox()
389 {
390   displayPreview();
391 }
392
393
394 //=================================================================================
395 // function : createOperation
396 // purpose  :
397 //=================================================================================
398 GEOM::GEOM_IOperations_ptr PrimitiveGUI_TorusDlg::createOperation()
399 {
400   return getGeomEngine()->GetI3DPrimOperations( getStudyId() );
401 }
402
403
404 //=================================================================================
405 // function : isValid
406 // purpose  :
407 //=================================================================================
408 bool PrimitiveGUI_TorusDlg::isValid( QString& msg )
409 {
410   return getConstructorId() == 0 ? !(myPoint->_is_nil() || myDir->_is_nil()) : true;
411 }
412
413
414 //=================================================================================
415 // function : execute
416 // purpose  :
417 //=================================================================================
418 bool PrimitiveGUI_TorusDlg::execute( ObjectList& objects )
419 {
420   bool res = false;
421   
422   GEOM::GEOM_Object_var anObj;
423
424   switch ( getConstructorId() ) {
425   case 0 :
426     if (!CORBA::is_nil( myPoint ) && !CORBA::is_nil( myDir )){
427       anObj = GEOM::GEOM_I3DPrimOperations::_narrow( getOperation() )->MakeTorusPntVecRR( myPoint, myDir, getRadius1(), getRadius2() );
428       res = true;
429     }
430     break;
431   case 1 :
432     anObj = GEOM::GEOM_I3DPrimOperations::_narrow( getOperation() )->MakeTorusRR( getRadius1(), getRadius2() );
433     res = true;
434     break;
435   }
436
437   if ( !anObj->_is_nil() )
438     objects.push_back( anObj._retn() );
439
440   return res;
441 }
442
443
444 //=================================================================================
445 // function : getRadius1()
446 // purpose  :
447 //=================================================================================
448 double PrimitiveGUI_TorusDlg::getRadius1() const
449 {
450   int aConstructorId = getConstructorId();
451   if ( aConstructorId == 0 )
452     return GroupPoints->SpinBox_DX->value();
453   else if ( aConstructorId == 1 )
454     return GroupDimensions->SpinBox_DX->value();
455   return 0;
456 }
457
458
459 //=================================================================================
460 // function : getRadius2()
461 // purpose  :
462 //=================================================================================
463 double PrimitiveGUI_TorusDlg::getRadius2() const
464 {
465   int aConstructorId = getConstructorId();
466   if ( aConstructorId == 0 )
467     return GroupPoints->SpinBox_DY->value();
468   else if ( aConstructorId == 1 )
469     return GroupDimensions->SpinBox_DY->value();
470   return 0;
471 }
472
473 //=================================================================================
474 // function : addSubshapeToStudy
475 // purpose  : virtual method to add new SubObjects if local selection
476 //=================================================================================
477 void PrimitiveGUI_TorusDlg::addSubshapesToStudy()
478 {
479   QMap<QString, GEOM::GEOM_Object_var> objMap;
480
481   switch ( getConstructorId() ) {
482   case 0:
483     objMap[GroupPoints->LineEdit1->text()] = myPoint;
484     objMap[GroupPoints->LineEdit2->text()] = myDir;
485     break;
486   case 1:
487     return;
488   }
489   addSubshapesToFather( objMap );
490 }