Salome HOME
0020598: EDF 1191 GEOM : Creation of hexa block from two faces
[modules/geom.git] / src / RepairGUI / RepairGUI_RemoveHolesDlg.cxx
1 //  Copyright (C) 2007-2008  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 // GEOM GEOMGUI : GUI for Geometry component
23 // File   : RepairGUI_RemoveHolesDlg.cxx
24 // Author : Lucien PIGNOLONI, Open CASCADE S.A.S.
25 //
26 #include "RepairGUI_RemoveHolesDlg.h"
27
28 #include <DlgRef.h>
29 #include <GeometryGUI.h>
30 #include <GEOMBase.h>
31
32 #include <SUIT_Session.h>
33 #include <SUIT_ResourceMgr.h>
34 #include <SUIT_MessageBox.h>
35 #include <SalomeApp_Application.h>
36 #include <LightApp_SelectionMgr.h>
37
38 // OCCT Includes
39 #include <TopAbs.hxx>
40 #include <TColStd_MapOfInteger.hxx>
41 #include <TColStd_IndexedMapOfInteger.hxx>
42
43 #include <GEOMImpl_Types.hxx>
44
45 //=================================================================================
46 // class    : RepairGUI_RemoveHolesDlg()
47 // purpose  : Constructs a RepairGUI_RemoveHolesDlg  which is a child of 'parent', with the
48 //            name 'name' and widget flags set to 'f'.
49 //            The dialog will by default be modeless, unless you set 'modal' to
50 //            TRUE to construct a modal dialog.
51 //=================================================================================
52 RepairGUI_RemoveHolesDlg::RepairGUI_RemoveHolesDlg (GeometryGUI* theGeometryGUI, QWidget* parent,
53                                                     bool modal)
54   : GEOMBase_Skeleton(theGeometryGUI, parent, modal)
55 {
56   QPixmap image0 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_DLG_SUPPRESS_HOLES")));
57   QPixmap image1 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_SELECT")));
58
59   setWindowTitle(tr("GEOM_REMOVE_HOLES_TITLE"));
60
61   /***************************************************************/
62   mainFrame()->GroupConstructors->setTitle(tr("GEOM_REMOVE_HOLES_TITLE"));
63   mainFrame()->RadioButton1->setIcon(image0);
64   mainFrame()->RadioButton2->setAttribute(Qt::WA_DeleteOnClose);
65   mainFrame()->RadioButton2->close();
66   mainFrame()->RadioButton3->setAttribute(Qt::WA_DeleteOnClose);
67   mainFrame()->RadioButton3->close();
68
69   GroupPoints = new DlgRef_1Sel1Check1Sel(centralWidget());
70   GroupPoints->GroupBox1->setTitle(tr("GEOM_HOLES"));
71   GroupPoints->TextLabel1->setText(tr("GEOM_SELECTED_SHAPE"));
72   GroupPoints->PushButton1->setIcon(image1);
73   GroupPoints->LineEdit1->setReadOnly(true);
74   GroupPoints->CheckButton1->setText(tr("GEOM_REMOVE_ALL_HOLES"));
75   GroupPoints->TextLabel2->setText(tr("GEOM_WIRES_TO_REMOVE"));
76   GroupPoints->PushButton2->setIcon(image1);
77   GroupPoints->LineEdit2->setReadOnly(true);
78
79   myFreeBoundBtn = new QPushButton(tr("GEOM_DETECT") + QString(" [%1]").arg(tr("GEOM_FREE_BOUNDARIES")),
80                                     GroupPoints->Box);
81   QVBoxLayout* l = new QVBoxLayout(GroupPoints->Box);
82   l->setMargin(0); l->setSpacing(0);
83   l->addWidget(myFreeBoundBtn);
84
85   QVBoxLayout* layout = new QVBoxLayout(centralWidget());
86   layout->setMargin(0); layout->setSpacing(6);
87   layout->addWidget(GroupPoints);
88   /***************************************************************/
89
90   setHelpFileName("suppress_holes_operation_page.html");
91
92   Init();
93 }
94
95 //=================================================================================
96 // function : ~RepairGUI_RemoveHolesDlg()
97 // purpose  : Destroys the object and frees any allocated resources
98 //=================================================================================
99 RepairGUI_RemoveHolesDlg::~RepairGUI_RemoveHolesDlg()
100 {
101 }
102
103 //=================================================================================
104 // function : Init()
105 // purpose  :
106 //=================================================================================
107 void RepairGUI_RemoveHolesDlg::Init()
108 {
109   // init variables
110   GroupPoints->LineEdit1->clear();
111   GroupPoints->LineEdit2->clear();
112   myObject = GEOM::GEOM_Object::_nil();
113   myWiresInd = new GEOM::short_array();
114   myWiresInd->length(0);
115
116   myClosed = -1;
117   myOpen = -1;
118
119   // signals and slots connections
120   connect(buttonOk(),    SIGNAL(clicked()), this, SLOT(ClickOnOk()));
121   connect(buttonApply(), SIGNAL(clicked()), this, SLOT(ClickOnApply()));
122
123   connect(GroupPoints->PushButton1, SIGNAL(clicked()),       this, SLOT(SetEditCurrentArgument()));
124   connect(GroupPoints->PushButton2, SIGNAL(clicked()),       this, SLOT(SetEditCurrentArgument()));
125
126   connect(GroupPoints->LineEdit1,   SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
127   connect(GroupPoints->LineEdit2,   SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
128
129   connect(GroupPoints->CheckButton1, SIGNAL(clicked()), this, SLOT(onRemoveAllClicked()));
130
131   connect(myFreeBoundBtn, SIGNAL(clicked()), this, SLOT(onDetect()));
132
133   initName(tr("REMOVE_HOLES_NEW_OBJ_NAME"));
134
135   GroupPoints->PushButton1->click();
136   SelectionIntoArgument();
137   resize(100,100);
138 }
139
140 //=================================================================================
141 // function : ClickOnOk()
142 // purpose  : Same than click on apply but close this dialog.
143 //=================================================================================
144 void RepairGUI_RemoveHolesDlg::ClickOnOk()
145 {
146   if (ClickOnApply())
147     ClickOnCancel();
148 }
149
150 //=================================================================================
151 // function : ClickOnApply()
152 // purpose  :
153 //=================================================================================
154 bool RepairGUI_RemoveHolesDlg::ClickOnApply()
155 {
156   if (!onAccept())
157     return false;
158
159   initName();
160   // activate first line edit
161   GroupPoints->PushButton1->click();
162   return true;
163 }
164
165 //=================================================================================
166 // function : SelectionIntoArgument()
167 // purpose  : Called when selection is changed or on dialog initialization or activation
168 //=================================================================================
169 void RepairGUI_RemoveHolesDlg::SelectionIntoArgument()
170 {
171   erasePreview();
172   myEditCurrentArgument->setText("");
173   // the second argument depends on the first one
174   GroupPoints->LineEdit2->setText("");
175   myWiresInd->length(0);
176
177   if (myEditCurrentArgument == GroupPoints->LineEdit1)
178     myObject = GEOM::GEOM_Object::_nil();
179
180   LightApp_SelectionMgr* aSelMgr = myGeomGUI->getApp()->selectionMgr();
181   SALOME_ListIO aSelList;
182   aSelMgr->selectedObjects(aSelList);
183
184   if (aSelList.Extent() == 1) {
185     Handle(SALOME_InteractiveObject) anIO = aSelList.First();
186
187     if (myEditCurrentArgument == GroupPoints->LineEdit1) { // face selection
188       Standard_Boolean aRes;
189       myObject = GEOMBase::ConvertIOinGEOMObject(anIO, aRes);
190       if (aRes && GEOMBase::IsShape(myObject)) {
191         myEditCurrentArgument->setText(GEOMBase::GetName(myObject));
192
193         // clear selection
194         disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
195         myGeomGUI->getApp()->selectionMgr()->clearSelected();
196         connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
197                 this, SLOT(SelectionIntoArgument()));
198
199         if (!GroupPoints->CheckButton1->isChecked())
200           GroupPoints->PushButton2->click();
201       }
202       else
203         myObject = GEOM::GEOM_Object::_nil();
204     }
205     else if (myEditCurrentArgument == GroupPoints->LineEdit2 &&
206              !GroupPoints->CheckButton1->isChecked()) {
207       TColStd_IndexedMapOfInteger aMap;
208       aSelMgr->GetIndexes(anIO, aMap);
209       const int n = aMap.Extent();
210       myWiresInd->length(n);
211       for (int i = 1; i <= n; i++)
212         myWiresInd[i-1] = aMap(i);
213       if (n)
214         myEditCurrentArgument->setText(QString::number(n) + "_" + tr("GEOM_WIRE") + tr("_S_"));
215     }
216   }
217 }
218
219 //=================================================================================
220 // function : SetEditCurrentArgument()
221 // purpose  :
222 //=================================================================================
223 void RepairGUI_RemoveHolesDlg::SetEditCurrentArgument()
224 {
225   QPushButton* send = (QPushButton*)sender();
226
227   bool isEffective = false;
228
229   if (send == GroupPoints->PushButton1) {
230     isEffective = true;
231     myEditCurrentArgument = GroupPoints->LineEdit1;
232
233     GroupPoints->PushButton2->setDown(false);
234     GroupPoints->LineEdit2->setEnabled(false);
235   }
236   else if (send == GroupPoints->PushButton2 && !myObject->_is_nil()) {
237     isEffective = true;
238     myEditCurrentArgument = GroupPoints->LineEdit2;
239
240     GroupPoints->PushButton1->setDown(false);
241     GroupPoints->LineEdit1->setEnabled(false);
242   }
243
244   if (isEffective) {
245     initSelection();
246
247     // enable line edit
248     myEditCurrentArgument->setEnabled(true);
249     myEditCurrentArgument->setFocus();
250     // after setFocus(), because it will be setDown(false) when loses focus
251     send->setDown(true);
252   }
253 }
254
255 //=================================================================================
256 // function : LineEditReturnPressed()
257 // purpose  :
258 //=================================================================================
259 void RepairGUI_RemoveHolesDlg::LineEditReturnPressed()
260 {
261   const QObject* send = sender();
262   if (send == GroupPoints->LineEdit1 || send == GroupPoints->LineEdit2) {
263     myEditCurrentArgument = (QLineEdit*)send;
264     GEOMBase_Skeleton::LineEditReturnPressed();
265   }
266 }
267
268 //=================================================================================
269 // function : ActivateThisDialog()
270 // purpose  :
271 //=================================================================================
272 void RepairGUI_RemoveHolesDlg::ActivateThisDialog()
273 {
274   GEOMBase_Skeleton::ActivateThisDialog();
275   connect( myGeomGUI->getApp()->selectionMgr(), SIGNAL( currentSelectionChanged() ),
276            this, SLOT( SelectionIntoArgument() ) );
277
278   myEditCurrentArgument = GroupPoints->LineEdit1;
279   myEditCurrentArgument->setText( "" );
280   GroupPoints->LineEdit2->setText( "" );
281   myObject = GEOM::GEOM_Object::_nil();
282   myWiresInd->length( 0 );
283
284   myClosed = -1;
285   myOpen = -1;
286
287   initSelection();
288 }
289
290 //=================================================================================
291 // function : enterEvent()
292 // purpose  : Mouse enter onto the dialog to activate it
293 //=================================================================================
294 void RepairGUI_RemoveHolesDlg::enterEvent (QEvent*)
295 {
296   if (!mainFrame()->GroupConstructors->isEnabled())
297     ActivateThisDialog();
298 }
299
300 //=================================================================================
301 // function : createOperation
302 // purpose  :
303 //=================================================================================
304 GEOM::GEOM_IOperations_ptr RepairGUI_RemoveHolesDlg::createOperation()
305 {
306   return getGeomEngine()->GetIHealingOperations(getStudyId());
307 }
308
309 //=================================================================================
310 // function : isValid
311 // purpose  :
312 //=================================================================================
313 bool RepairGUI_RemoveHolesDlg::isValid (QString&)
314 {
315   myClosed = -1;
316   return !myObject->_is_nil() && (IsPreview() || GroupPoints->CheckButton1->isChecked() || myWiresInd->length());
317 }
318
319 //=================================================================================
320 // function : execute
321 // purpose  :
322 //=================================================================================
323 bool RepairGUI_RemoveHolesDlg::execute (ObjectList& objects)
324 {
325   bool aResult = false;
326
327   if (IsPreview()) {
328     // called from onDetect(): detect free boundary edges,
329     // highlight them (add to objects), display message dialog
330     GEOM::ListOfGO_var aClosed, anOpen;
331
332     GEOM::GEOM_IHealingOperations_var anOper = GEOM::GEOM_IHealingOperations::_narrow(getOperation());
333     aResult = anOper->GetFreeBoundary(myObject, aClosed, anOpen);
334
335     if (aResult) {
336       myClosed = aClosed->length();
337       myOpen = anOpen->length();
338       int i;
339       for (i = 0; i < myClosed; i++)
340         objects.push_back(aClosed[i]._retn());
341       for (i = 0; i < myOpen; i++)
342         objects.push_back(anOpen[i]._retn());
343     }
344     else
345       myClosed = -1;
346   }
347   else {
348     GEOM::GEOM_IHealingOperations_var anOper = GEOM::GEOM_IHealingOperations::_narrow(getOperation());
349     GEOM::GEOM_Object_var anObj = anOper->FillHoles(myObject, myWiresInd);
350     aResult = !anObj->_is_nil();
351     if (aResult)
352       objects.push_back(anObj._retn());
353   }
354
355   return aResult;
356 }
357
358 //=================================================================================
359 // function : onRemoveAllClicked
360 // purpose  :
361 //=================================================================================
362 void RepairGUI_RemoveHolesDlg::onRemoveAllClicked()
363 {
364   bool b = GroupPoints->CheckButton1->isChecked();
365   GroupPoints->TextLabel2->setEnabled(!b);
366   GroupPoints->PushButton2->setEnabled(!b);
367   GroupPoints->LineEdit2->setEnabled(!b);
368   if (b) {
369     GroupPoints->LineEdit2->setText("");
370     myWiresInd->length(0);
371   }
372 }
373
374 //=================================================================================
375 // function : initSelection
376 // purpose  :
377 //=================================================================================
378 void RepairGUI_RemoveHolesDlg::initSelection()
379 {
380   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
381
382   if (myEditCurrentArgument == GroupPoints->LineEdit1) {
383     TColStd_MapOfInteger aTypes;
384     aTypes.Add(GEOM_COMPOUND);
385     aTypes.Add(GEOM_SOLID);
386     aTypes.Add(GEOM_SHELL);
387     aTypes.Add(GEOM_FACE);
388
389     globalSelection(aTypes);
390   }
391   else if (myEditCurrentArgument == GroupPoints->LineEdit2) {
392     localSelection(myObject, TopAbs_EDGE);
393     localSelection(myObject, TopAbs_WIRE);
394   }
395
396   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
397           this, SLOT(SelectionIntoArgument()));
398 }
399
400 //=================================================================================
401 // function : onDetect
402 // purpose  :
403 //=================================================================================
404 void RepairGUI_RemoveHolesDlg::onDetect()
405 {
406   displayPreview(false, true, true, 3);
407
408   // field myClosed,myOpen is initialized in execute() method, called by displayPreview().
409   QString msg;
410   if (myClosed != -1)
411     msg = tr("GEOM_FREE_BOUNDS_MSG").arg(myClosed + myOpen).arg(myClosed).arg(myOpen);
412   else
413     msg = tr("GEOM_FREE_BOUNDS_ERROR");
414   SUIT_MessageBox::information(this, tr("GEOM_FREE_BOUNDS_TLT"), msg);
415 }