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