Salome HOME
TC 5.1.5: SIGSEGV on closing study
[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
205   // activate selection and connect selection manager
206   ConstructorsClicked(getConstructorId());
207   return true;
208 }
209
210 //=================================================================================
211 // function : SelectionIntoArgument()
212 // purpose  : Called when selection is changed or on dialog initialization or activation
213 //=================================================================================
214 void TransformationGUI_MirrorDlg::SelectionIntoArgument()
215 {
216   erasePreview();
217   myEditCurrentArgument->setText("");
218
219   if (myEditCurrentArgument == GroupPoints->LineEdit1)
220     myObjects.length(0);
221   else if (myEditCurrentArgument == GroupPoints->LineEdit2)
222     myArgument = GEOM::GEOM_Object::_nil();
223
224   LightApp_SelectionMgr* aSelMgr = myGeomGUI->getApp()->selectionMgr();
225   SALOME_ListIO aSelList;
226   aSelMgr->selectedObjects(aSelList);
227
228   QString aName;
229
230   if (myEditCurrentArgument == GroupPoints->LineEdit1) {
231     int aNbSel = GEOMBase::GetNameOfSelectedIObjects(aSelList, aName);
232     if (aNbSel < 1)
233       return;
234
235     GEOMBase::ConvertListOfIOInListOfGO(aSelList, myObjects);
236     if (!myObjects.length())
237       return;
238     else
239       myEditCurrentArgument->setText(aName);
240   }
241   else if (myEditCurrentArgument == GroupPoints->LineEdit2) {
242     if (aSelList.Extent() != 1)
243       return;
244
245     // nbSel == 1
246     Standard_Boolean testResult = Standard_False;
247     myArgument = GEOMBase::ConvertIOinGEOMObject(aSelList.First(), testResult);
248     if (!testResult || CORBA::is_nil(myArgument))
249       return;
250
251     aName = GEOMBase::GetName(myArgument);
252
253     if (testResult && !myArgument->_is_nil()) {
254       TopoDS_Shape aShape;
255       if (GEOMBase::GetShape(myArgument, aShape, TopAbs_SHAPE) && !aShape.IsNull()) {
256         TopAbs_ShapeEnum aNeedType = TopAbs_VERTEX;
257         if (getConstructorId() == 1)
258           aNeedType = TopAbs_EDGE;
259         else if (getConstructorId() == 2)
260           aNeedType = TopAbs_FACE;
261
262         TColStd_IndexedMapOfInteger aMap;
263         aSelMgr->GetIndexes(aSelList.First(), aMap);
264         if (aMap.Extent() == 1) {
265           int anIndex = aMap(1);
266           if (aNeedType == TopAbs_VERTEX)
267             aName += QString(":vertex_%1").arg(anIndex);
268           else
269             aName += QString(":edge_%1").arg(anIndex);
270
271           //Find SubShape Object in Father
272           GEOM::GEOM_Object_var aFindedObject = findObjectInFather(myArgument, aName);
273
274           if (aFindedObject->_is_nil()) { // Object not found in study
275             GEOM::GEOM_IShapesOperations_var aShapesOp =
276               getGeomEngine()->GetIShapesOperations(getStudyId());
277             myArgument = aShapesOp->GetSubShape(myArgument, anIndex);
278           }
279           else {
280             myArgument = aFindedObject; // get Object from study
281           }
282         }
283         else {
284           if (aShape.ShapeType() != aNeedType) {
285             myArgument = GEOM::GEOM_Object::_nil();
286             aName = "";
287           }
288         }
289       }
290     }
291     myEditCurrentArgument->setText(aName);
292
293     if (!myArgument->_is_nil() && !myObjects.length())
294       GroupPoints->PushButton1->click();
295   }
296
297   // clear selection
298   /*  disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
299   myGeomGUI->getApp()->selectionMgr()->clearSelected();
300   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
301   this, SLOT(SelectionIntoArgument()));*/
302   // here commented, because multiple objects can be selected IPAL 21437
303
304   displayPreview();
305 }
306
307 //=================================================================================
308 // function : SetEditCurrentArgument()
309 // purpose  :
310 //=================================================================================
311 void TransformationGUI_MirrorDlg::SetEditCurrentArgument()
312 {
313   QPushButton* send = (QPushButton*)sender();
314
315   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
316   globalSelection();
317
318   if (send == GroupPoints->PushButton1) {
319     myEditCurrentArgument = GroupPoints->LineEdit1;
320
321     GroupPoints->PushButton2->setDown(false);
322     GroupPoints->LineEdit2->setEnabled(false);
323   }
324   else if (send == GroupPoints->PushButton2) {
325     myEditCurrentArgument = GroupPoints->LineEdit2;
326
327     switch (getConstructorId()) {
328     case 0:
329       localSelection(GEOM::GEOM_Object::_nil(), TopAbs_VERTEX);
330       break;
331     case 1:
332       localSelection(GEOM::GEOM_Object::_nil(), TopAbs_EDGE);
333       break;
334     case 2:
335       globalSelection(GEOM_PLANE);
336       break;
337     }
338
339     GroupPoints->PushButton1->setDown(false);
340     GroupPoints->LineEdit1->setEnabled(false);
341   }
342   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
343           this, SLOT(SelectionIntoArgument()));
344
345   // enable line edit
346   myEditCurrentArgument->setEnabled(true);
347   myEditCurrentArgument->setFocus();
348   // after setFocus(), because it will be setDown(false) when loses focus
349   send->setDown(true);
350
351   // seems we need it only to avoid preview disappearing, caused by selection mode change
352   displayPreview();
353 }
354
355 //=================================================================================
356 // function : LineEditReturnPressed()
357 // purpose  :
358 //=================================================================================
359 void TransformationGUI_MirrorDlg::LineEditReturnPressed()
360 {
361   QLineEdit* send = (QLineEdit*)sender();
362   if (send == GroupPoints->LineEdit1 ||
363       send == GroupPoints->LineEdit2) {
364     myEditCurrentArgument = send;
365     GEOMBase_Skeleton::LineEditReturnPressed();
366   }
367 }
368
369 //=================================================================================
370 // function : ActivateThisDialog()
371 // purpose  :
372 //=================================================================================
373 void TransformationGUI_MirrorDlg::ActivateThisDialog()
374 {
375   GEOMBase_Skeleton::ActivateThisDialog();
376   connect( myGeomGUI->getApp()->selectionMgr(), SIGNAL( currentSelectionChanged() ),
377            this, SLOT( SelectionIntoArgument() ) );
378
379   ConstructorsClicked( getConstructorId() );
380 }
381
382 //=================================================================================
383 // function : enterEvent()
384 // purpose  : when mouse enter onto the QWidget
385 //=================================================================================
386 void TransformationGUI_MirrorDlg::enterEvent (QEvent*)
387 {
388   if (!mainFrame()->GroupConstructors->isEnabled())
389     ActivateThisDialog();
390 }
391
392 //=================================================================================
393 // function : createOperation
394 // purpose  :
395 //=================================================================================
396 GEOM::GEOM_IOperations_ptr  TransformationGUI_MirrorDlg::createOperation()
397 {
398   return getGeomEngine()->GetITransformOperations(getStudyId());
399 }
400
401 //=================================================================================
402 // function : isValid
403 // purpose  :
404 //=================================================================================
405 bool TransformationGUI_MirrorDlg::isValid (QString& /*msg*/)
406 {
407   return !(myObjects.length() == 0 || myArgument->_is_nil());
408 }
409
410 //=================================================================================
411 // function : execute
412 // purpose  :
413 //=================================================================================
414 bool  TransformationGUI_MirrorDlg::execute (ObjectList& objects)
415 {
416   bool res = false;
417   bool toCreateCopy = IsPreview() || GroupPoints->CheckButton1->isChecked();
418
419   GEOM::GEOM_Object_var anObj;
420
421   GEOM::GEOM_ITransformOperations_var anOper = GEOM::GEOM_ITransformOperations::_narrow(getOperation());
422
423   switch (getConstructorId()) {
424   case 0:
425     {
426       if (toCreateCopy) {
427         for (int i = 0; i < myObjects.length(); i++) {
428           anObj = anOper->MirrorPointCopy(myObjects[i], myArgument);
429           if (!anObj->_is_nil())
430             objects.push_back(anObj._retn());
431         }
432       }
433       else {
434         for (int i = 0; i < myObjects.length(); i++) {
435           anObj = anOper->MirrorPoint(myObjects[i], myArgument);
436           if (!anObj->_is_nil())
437             objects.push_back(anObj._retn());
438         }
439       }
440       res = true;
441       break;
442     }
443   case 1:
444     {
445       if (toCreateCopy) {
446         for (int i = 0; i < myObjects.length(); i++) {
447           anObj = anOper->MirrorAxisCopy(myObjects[i], myArgument);
448           if (!anObj->_is_nil())
449             objects.push_back(anObj._retn());
450         }
451       }
452       else {
453         for (int i = 0; i < myObjects.length(); i++) {
454           anObj = anOper->MirrorAxis(myObjects[i], myArgument);
455           if (!anObj->_is_nil())
456             objects.push_back(anObj._retn());
457         }
458       }
459       res = true;
460       break;
461     }
462   case 2:
463     {
464       if (toCreateCopy) {
465         for (int i = 0; i < myObjects.length(); i++) {
466             anObj = anOper->MirrorPlaneCopy(myObjects[i], myArgument);
467             if (!anObj->_is_nil())
468               objects.push_back(anObj._retn());
469         }
470       }
471       else {
472         for (int i = 0; i < myObjects.length(); i++) {
473           anObj = anOper->MirrorPlane(myObjects[i], myArgument);
474           if (!anObj->_is_nil())
475             objects.push_back(anObj._retn());
476         }
477       }
478       res = true;
479       break;
480     }
481   }
482
483   return res;
484 }
485
486 //=================================================================================
487 // function : restoreSubShapes
488 // purpose  :
489 //=================================================================================
490 void TransformationGUI_MirrorDlg::restoreSubShapes (SALOMEDS::Study_ptr   theStudy,
491                                                     SALOMEDS::SObject_ptr theSObject)
492 {
493   if (mainFrame()->CheckBoxRestoreSS->isChecked()) {
494     // empty list of arguments means that all arguments should be restored
495     getGeomEngine()->RestoreSubShapesSO(theStudy, theSObject, GEOM::ListOfGO(),
496                                         /*theFindMethod=*/GEOM::FSM_Transformed,
497                                         /*theInheritFirstArg=*/true,
498                                         mainFrame()->CheckBoxAddPrefix->isChecked());
499   }
500 }
501
502 //=================================================================================
503 // function :  CreateCopyModeChanged()
504 // purpose  :
505 //=================================================================================
506 void TransformationGUI_MirrorDlg::CreateCopyModeChanged (bool isCreateCopy)
507 {
508   mainFrame()->GroupBoxName->setEnabled(isCreateCopy);
509 }
510
511 //=================================================================================
512 // function : addSubshapeToStudy
513 // purpose  : virtual method to add new SubObjects if local selection
514 //=================================================================================
515 void TransformationGUI_MirrorDlg::addSubshapesToStudy()
516 {
517   bool toCreateCopy = IsPreview() || GroupPoints->CheckButton1->isChecked();
518   if (toCreateCopy) {
519     QMap<QString, GEOM::GEOM_Object_var> objMap;
520
521     switch (getConstructorId()) {
522     case 0:
523       objMap[GroupPoints->LineEdit2->text()] = myArgument;
524       break;
525     case 1:
526       objMap[GroupPoints->LineEdit2->text()] = myArgument;
527       break;
528     case 2:
529       return;
530     }
531     addSubshapesToFather(objMap);
532     }
533 }