]> SALOME platform Git repositories - modules/geom.git/blob - src/TransformationGUI/TransformationGUI_MirrorDlg.cxx
Salome HOME
Code refactoring
[modules/geom.git] / src / TransformationGUI / TransformationGUI_MirrorDlg.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_MirrorDlg.cxx
25 // Author : Lucien PIGNOLONI, Open CASCADE S.A.S.
26 //
27 #include "TransformationGUI_MirrorDlg.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    : TransformationGUI_MirrorDlg()
50 // purpose  : Constructs a TransformationGUI_MirrorDlg 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_MirrorDlg::TransformationGUI_MirrorDlg (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 image0 (aResMgr->loadPixmap("GEOM", tr("ICON_DLG_MIRROR_POINT")));
62   QPixmap image1 (aResMgr->loadPixmap("GEOM", tr("ICON_DLG_MIRROR_AXE")));
63   QPixmap image2 (aResMgr->loadPixmap("GEOM", tr("ICON_DLG_MIRROR_PLANE")));
64   QPixmap image3 (aResMgr->loadPixmap("GEOM", tr("ICON_SELECT")));
65
66   setWindowTitle(tr("GEOM_MIRROR_TITLE"));
67
68   /***************************************************************/
69   mainFrame()->GroupConstructors->setTitle(tr("GEOM_MIRROR"));
70   mainFrame()->RadioButton1->setIcon(image0);
71   mainFrame()->RadioButton2->setIcon(image1);
72   mainFrame()->RadioButton3->setIcon(image2);
73
74   GroupPoints = new DlgRef_2Sel1Spin2Check(centralWidget());
75   GroupPoints->SpinBox_DX->hide();
76   GroupPoints->TextLabel3->hide();
77   GroupPoints->CheckButton2->hide();
78   GroupPoints->GroupBox1->setTitle(tr("GEOM_ARGUMENTS"));
79   GroupPoints->TextLabel1->setText(tr("GEOM_OBJECTS"));
80   GroupPoints->TextLabel2->setText(tr("GEOM_POINT_MIRROR"));
81   GroupPoints->TextLabel2->setFixedWidth(74);
82   GroupPoints->PushButton1->setIcon(image3);
83   GroupPoints->PushButton2->setIcon(image3);
84   GroupPoints->CheckButton1->setText(tr("GEOM_CREATE_COPY"));
85
86   QVBoxLayout* layout = new QVBoxLayout(centralWidget());
87   layout->setMargin(0); layout->setSpacing(6);
88   layout->addWidget(GroupPoints);
89   /***************************************************************/
90
91   setHelpFileName("mirror_operation_page.html");
92
93   // Activate Create a Copy mode
94   GroupPoints->CheckButton1->setChecked(true);
95   CreateCopyModeChanged(true);
96
97   Init();
98 }
99
100 //=================================================================================
101 // function : ~TransformationGUI_MirrorDlg()
102 // purpose  : Destroys the object and frees any allocated resources
103 //=================================================================================
104 TransformationGUI_MirrorDlg::~TransformationGUI_MirrorDlg()
105 {
106   // no need to delete child widgets, Qt does it all for us
107 }
108
109 //=================================================================================
110 // function : Init()
111 // purpose  :
112 //=================================================================================
113 void TransformationGUI_MirrorDlg::Init()
114 {
115   // init variables
116   GroupPoints->LineEdit1->setReadOnly(true);
117   GroupPoints->LineEdit2->setReadOnly(true);
118
119   GroupPoints->LineEdit1->setText("");
120   GroupPoints->LineEdit2->setText("");
121
122   myArgument = GEOM::GEOM_Object::_nil();
123
124   mainFrame()->GroupBoxPublish->show();
125
126   // signals and slots connections
127   connect(buttonOk(),    SIGNAL(clicked()), this, SLOT(ClickOnOk()));
128   connect(buttonApply(), SIGNAL(clicked()), this, SLOT(ClickOnApply()));
129
130   connect(this, SIGNAL(constructorsClicked(int)), this, SLOT(ConstructorsClicked(int)));
131
132   connect(GroupPoints->PushButton1, SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
133   connect(GroupPoints->PushButton2, SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
134
135   connect(GroupPoints->LineEdit1,   SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
136   connect(GroupPoints->LineEdit2,   SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
137
138   connect(GroupPoints->CheckButton1, SIGNAL(toggled(bool)), this, SLOT(CreateCopyModeChanged(bool)));
139
140   initName(tr("GEOM_MIRROR"));
141
142   ConstructorsClicked(0);
143   resize(minimumSizeHint());
144 }
145
146 //=================================================================================
147 // function : ConstructorsClicked()
148 // purpose  : Radio button management
149 //=================================================================================
150 void TransformationGUI_MirrorDlg::ConstructorsClicked (int constructorId)
151 {
152   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
153
154   GroupPoints->LineEdit2->clear();
155   myArgument = GEOM::GEOM_Object::_nil();
156
157   switch (constructorId) {
158   case 0: // mirror an object by point
159     GroupPoints->TextLabel2->setText(tr("GEOM_POINT_MIRROR"));
160     break;
161   case 1: // mirror an object by axe
162     GroupPoints->TextLabel2->setText(tr("GEOM_AXE_MIRROR"));
163     break;
164   case 2: // mirror an object by plane
165     GroupPoints->TextLabel2->setText(tr("GEOM_PLANE_MIRROR"));
166     break;
167   }
168
169   GroupPoints->PushButton1->click();
170
171   if (myInitial) {
172     myInitial = false;
173     SelectionIntoArgument();
174   }
175   else {
176     displayPreview();
177   }
178 }
179
180 //=================================================================================
181 // function : ClickOnOk()
182 // purpose  :
183 //=================================================================================
184 void TransformationGUI_MirrorDlg::ClickOnOk()
185 {
186   if (ClickOnApply())
187     ClickOnCancel();
188 }
189
190 //=================================================================================
191 // function : ClickOnApply()
192 // purpose  :
193 //=================================================================================
194 bool TransformationGUI_MirrorDlg::ClickOnApply()
195 {
196   if (!onAccept(GroupPoints->CheckButton1->isChecked()))
197     return false;
198
199   initName();
200
201   myObjects.length(0);
202   myEditCurrentArgument = GroupPoints->LineEdit1;
203   myEditCurrentArgument->setText("");
204   myGeomGUI->getApp()->selectionMgr()->clearSelected();
205
206   // activate selection and connect selection manager
207   ConstructorsClicked(getConstructorId());
208   return true;
209 }
210
211 //=================================================================================
212 // function : SelectionIntoArgument()
213 // purpose  : Called when selection is changed or on dialog initialization or activation
214 //=================================================================================
215 void TransformationGUI_MirrorDlg::SelectionIntoArgument()
216 {
217   erasePreview();
218   myEditCurrentArgument->setText("");
219
220   if (myEditCurrentArgument == GroupPoints->LineEdit1)
221     myObjects.length(0);
222   else if (myEditCurrentArgument == GroupPoints->LineEdit2)
223     myArgument = GEOM::GEOM_Object::_nil();
224
225   LightApp_SelectionMgr* aSelMgr = myGeomGUI->getApp()->selectionMgr();
226   SALOME_ListIO aSelList;
227   aSelMgr->selectedObjects(aSelList);
228
229   QString aName;
230
231   if (myEditCurrentArgument == GroupPoints->LineEdit1) {
232     int aNbSel = GEOMBase::GetNameOfSelectedIObjects(aSelList, aName);
233     if (aNbSel < 1)
234       return;
235
236     GEOMBase::ConvertListOfIOInListOfGO(aSelList, myObjects);
237     if (!myObjects.length())
238       return;
239     else
240       myEditCurrentArgument->setText(aName);
241   }
242   else if (myEditCurrentArgument == GroupPoints->LineEdit2) {
243     if (aSelList.Extent() != 1)
244       return;
245
246     // nbSel == 1
247     myArgument = GEOMBase::ConvertIOinGEOMObject( aSelList.First() );
248     if ( CORBA::is_nil(myArgument) )
249       return;
250
251     aName = GEOMBase::GetName(myArgument);
252
253     TopoDS_Shape aShape;
254     if (GEOMBase::GetShape(myArgument, aShape, TopAbs_SHAPE) && !aShape.IsNull()) {
255       TopAbs_ShapeEnum aNeedType = TopAbs_VERTEX;
256       if (getConstructorId() == 1)
257         aNeedType = TopAbs_EDGE;
258       else if (getConstructorId() == 2)
259         aNeedType = TopAbs_FACE;
260       
261       TColStd_IndexedMapOfInteger aMap;
262       aSelMgr->GetIndexes(aSelList.First(), aMap);
263       if (aMap.Extent() == 1) {
264         int anIndex = aMap(1);
265         if (aNeedType == TopAbs_VERTEX)
266           aName += QString(":vertex_%1").arg(anIndex);
267         else
268           aName += QString(":edge_%1").arg(anIndex);
269         
270         //Find SubShape Object in Father
271         GEOM::GEOM_Object_var aFindedObject = findObjectInFather(myArgument, aName);
272         
273         if (aFindedObject->_is_nil()) { // Object not found in study
274           GEOM::GEOM_IShapesOperations_var aShapesOp =
275             getGeomEngine()->GetIShapesOperations(getStudyId());
276           myArgument = aShapesOp->GetSubShape(myArgument, anIndex);
277         }
278         else {
279           myArgument = aFindedObject; // get Object from study
280         }
281       }
282       else {
283         if (aShape.ShapeType() != aNeedType) {
284           myArgument = GEOM::GEOM_Object::_nil();
285           aName = "";
286         }
287       }
288     }
289     myEditCurrentArgument->setText(aName);
290
291     if (!myArgument->_is_nil() && !myObjects.length())
292       GroupPoints->PushButton1->click();
293   }
294
295   // clear selection
296   /*  disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
297   myGeomGUI->getApp()->selectionMgr()->clearSelected();
298   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
299   this, SLOT(SelectionIntoArgument()));*/
300   // here commented, because multiple objects can be selected IPAL 21437
301
302   displayPreview();
303 }
304
305 //=================================================================================
306 // function : SetEditCurrentArgument()
307 // purpose  :
308 //=================================================================================
309 void TransformationGUI_MirrorDlg::SetEditCurrentArgument()
310 {
311   QPushButton* send = (QPushButton*)sender();
312
313   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
314   globalSelection();
315
316   if (send == GroupPoints->PushButton1) {
317     myEditCurrentArgument = GroupPoints->LineEdit1;
318
319     GroupPoints->PushButton2->setDown(false);
320     GroupPoints->LineEdit2->setEnabled(false);
321   }
322   else if (send == GroupPoints->PushButton2) {
323     myEditCurrentArgument = GroupPoints->LineEdit2;
324
325     switch (getConstructorId()) {
326     case 0:
327       localSelection(GEOM::GEOM_Object::_nil(), TopAbs_VERTEX);
328       break;
329     case 1:
330       localSelection(GEOM::GEOM_Object::_nil(), TopAbs_EDGE);
331       break;
332     case 2:
333       globalSelection(GEOM_PLANE);
334       break;
335     }
336
337     GroupPoints->PushButton1->setDown(false);
338     GroupPoints->LineEdit1->setEnabled(false);
339   }
340   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
341           this, SLOT(SelectionIntoArgument()));
342
343   // enable line edit
344   myEditCurrentArgument->setEnabled(true);
345   myEditCurrentArgument->setFocus();
346   // after setFocus(), because it will be setDown(false) when loses focus
347   send->setDown(true);
348
349   // seems we need it only to avoid preview disappearing, caused by selection mode change
350   displayPreview();
351 }
352
353 //=================================================================================
354 // function : LineEditReturnPressed()
355 // purpose  :
356 //=================================================================================
357 void TransformationGUI_MirrorDlg::LineEditReturnPressed()
358 {
359   QLineEdit* send = (QLineEdit*)sender();
360   if (send == GroupPoints->LineEdit1 ||
361       send == GroupPoints->LineEdit2) {
362     myEditCurrentArgument = send;
363     GEOMBase_Skeleton::LineEditReturnPressed();
364   }
365 }
366
367 //=================================================================================
368 // function : ActivateThisDialog()
369 // purpose  :
370 //=================================================================================
371 void TransformationGUI_MirrorDlg::ActivateThisDialog()
372 {
373   GEOMBase_Skeleton::ActivateThisDialog();
374   connect( myGeomGUI->getApp()->selectionMgr(), SIGNAL( currentSelectionChanged() ),
375            this, SLOT( SelectionIntoArgument() ) );
376
377   ConstructorsClicked( getConstructorId() );
378 }
379
380 //=================================================================================
381 // function : enterEvent()
382 // purpose  : when mouse enter onto the QWidget
383 //=================================================================================
384 void TransformationGUI_MirrorDlg::enterEvent (QEvent*)
385 {
386   if (!mainFrame()->GroupConstructors->isEnabled())
387     ActivateThisDialog();
388 }
389
390 //=================================================================================
391 // function : createOperation
392 // purpose  :
393 //=================================================================================
394 GEOM::GEOM_IOperations_ptr  TransformationGUI_MirrorDlg::createOperation()
395 {
396   return getGeomEngine()->GetITransformOperations(getStudyId());
397 }
398
399 //=================================================================================
400 // function : isValid
401 // purpose  :
402 //=================================================================================
403 bool TransformationGUI_MirrorDlg::isValid (QString& /*msg*/)
404 {
405   return !(myObjects.length() == 0 || myArgument->_is_nil());
406 }
407
408 //=================================================================================
409 // function : execute
410 // purpose  :
411 //=================================================================================
412 bool  TransformationGUI_MirrorDlg::execute (ObjectList& objects)
413 {
414   bool res = false;
415   bool toCreateCopy = IsPreview() || GroupPoints->CheckButton1->isChecked();
416
417   GEOM::GEOM_Object_var anObj;
418
419   GEOM::GEOM_ITransformOperations_var anOper = GEOM::GEOM_ITransformOperations::_narrow(getOperation());
420
421   switch (getConstructorId()) {
422   case 0:
423     {
424       if (toCreateCopy) {
425         for (int i = 0; i < myObjects.length(); i++) {
426           anObj = anOper->MirrorPointCopy(myObjects[i], myArgument);
427           if (!anObj->_is_nil())
428             objects.push_back(anObj._retn());
429         }
430       }
431       else {
432         for (int i = 0; i < myObjects.length(); i++) {
433           anObj = anOper->MirrorPoint(myObjects[i], myArgument);
434           if (!anObj->_is_nil())
435             objects.push_back(anObj._retn());
436         }
437       }
438       res = true;
439       break;
440     }
441   case 1:
442     {
443       if (toCreateCopy) {
444         for (int i = 0; i < myObjects.length(); i++) {
445           anObj = anOper->MirrorAxisCopy(myObjects[i], myArgument);
446           if (!anObj->_is_nil())
447             objects.push_back(anObj._retn());
448         }
449       }
450       else {
451         for (int i = 0; i < myObjects.length(); i++) {
452           anObj = anOper->MirrorAxis(myObjects[i], myArgument);
453           if (!anObj->_is_nil())
454             objects.push_back(anObj._retn());
455         }
456       }
457       res = true;
458       break;
459     }
460   case 2:
461     {
462       if (toCreateCopy) {
463         for (int i = 0; i < myObjects.length(); i++) {
464             anObj = anOper->MirrorPlaneCopy(myObjects[i], myArgument);
465             if (!anObj->_is_nil())
466               objects.push_back(anObj._retn());
467         }
468       }
469       else {
470         for (int i = 0; i < myObjects.length(); i++) {
471           anObj = anOper->MirrorPlane(myObjects[i], myArgument);
472           if (!anObj->_is_nil())
473             objects.push_back(anObj._retn());
474         }
475       }
476       res = true;
477       break;
478     }
479   }
480
481   return res;
482 }
483
484 //=================================================================================
485 // function : restoreSubShapes
486 // purpose  :
487 //=================================================================================
488 void TransformationGUI_MirrorDlg::restoreSubShapes (SALOMEDS::Study_ptr   theStudy,
489                                                     SALOMEDS::SObject_ptr theSObject)
490 {
491   if (mainFrame()->CheckBoxRestoreSS->isChecked()) {
492     // empty list of arguments means that all arguments should be restored
493     getGeomEngine()->RestoreSubShapesSO(theStudy, theSObject, GEOM::ListOfGO(),
494                                         /*theFindMethod=*/GEOM::FSM_Transformed,
495                                         /*theInheritFirstArg=*/true,
496                                         mainFrame()->CheckBoxAddPrefix->isChecked());
497   }
498 }
499
500 //=================================================================================
501 // function :  CreateCopyModeChanged()
502 // purpose  :
503 //=================================================================================
504 void TransformationGUI_MirrorDlg::CreateCopyModeChanged (bool isCreateCopy)
505 {
506   mainFrame()->GroupBoxName->setEnabled(isCreateCopy);
507 }
508
509 //=================================================================================
510 // function : addSubshapeToStudy
511 // purpose  : virtual method to add new SubObjects if local selection
512 //=================================================================================
513 void TransformationGUI_MirrorDlg::addSubshapesToStudy()
514 {
515   bool toCreateCopy = IsPreview() || GroupPoints->CheckButton1->isChecked();
516   if (toCreateCopy) {
517     QMap<QString, GEOM::GEOM_Object_var> objMap;
518
519     switch (getConstructorId()) {
520     case 0:
521       objMap[GroupPoints->LineEdit2->text()] = myArgument;
522       break;
523     case 1:
524       objMap[GroupPoints->LineEdit2->text()] = myArgument;
525       break;
526     case 2:
527       return;
528     }
529     addSubshapesToFather(objMap);
530     }
531 }