Salome HOME
CCAR: GEOM_Client modification to optimize the method find
[modules/geom.git] / src / RepairGUI / RepairGUI_SuppressFacesDlg.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 //  GEOM GEOMGUI : GUI for Geometry component
23 //  File   : RepairGUI_SuppressFacesDlg.cxx
24 //  Author : Lucien PIGNOLONI, Open CASCADE S.A.S.
25
26 #include "RepairGUI_SuppressFacesDlg.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 <SalomeApp_Application.h>
35 #include <SalomeApp_Study.h>
36 #include <LightApp_SelectionMgr.h>
37 #include <SALOME_ListIteratorOfListIO.hxx>
38
39 // OCCT Includes
40 #include <TopAbs.hxx>
41 #include <TopExp.hxx>
42 #include <TopExp_Explorer.hxx>
43 #include <TopTools_IndexedMapOfShape.hxx>
44 #include <TColStd_MapOfInteger.hxx>
45 #include <TColStd_IndexedMapOfInteger.hxx>
46
47 #include <GEOMImpl_Types.hxx>
48
49 //=================================================================================
50 // class    : RepairGUI_SuppressFacesDlg()
51 // purpose  : Constructs a RepairGUI_SuppressFacesDlg  which is a child of 'parent', with the
52 //            name 'name' and widget flags set to 'f'.
53 //            The dialog will by default be modeless, unless you set 'modal' to
54 //            TRUE to construct a modal dialog.
55 //=================================================================================
56 RepairGUI_SuppressFacesDlg::RepairGUI_SuppressFacesDlg (GeometryGUI* theGeometryGUI, QWidget* parent,
57                                                         bool modal)
58   : GEOMBase_Skeleton(theGeometryGUI, parent, modal)
59 {
60   QPixmap image0 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_DLG_SUPRESS_FACE")));
61   QPixmap image1 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_SELECT")));
62
63   setWindowTitle(tr("GEOM_SUPRESSFACE_TITLE"));
64
65   /***************************************************************/
66   mainFrame()->GroupConstructors->setTitle(tr("GEOM_SUPRESSFACE"));
67   mainFrame()->RadioButton1->setIcon(image0);
68   mainFrame()->RadioButton2->setAttribute(Qt::WA_DeleteOnClose);
69   mainFrame()->RadioButton2->close();
70   mainFrame()->RadioButton3->setAttribute(Qt::WA_DeleteOnClose);
71   mainFrame()->RadioButton3->close();
72
73   GroupArgs = new DlgRef_2Sel(centralWidget());
74   GroupArgs->GroupBox1->setTitle(tr("Faces to remove"));
75   GroupArgs->TextLabel1->setText(tr("GEOM_SELECTED_SHAPE"));
76   GroupArgs->PushButton1->setIcon(image1);
77   GroupArgs->LineEdit1->setReadOnly(true);
78
79   GroupArgs->TextLabel2->setText(tr("Faces to remove"));
80   GroupArgs->PushButton2->setIcon(image1);
81   GroupArgs->LineEdit2->setReadOnly(true);
82
83   QVBoxLayout* layout = new QVBoxLayout(centralWidget());
84   layout->setMargin(0); layout->setSpacing(6);
85   layout->addWidget(GroupArgs);
86   /***************************************************************/
87
88   setHelpFileName("suppress_faces_operation_page.html");
89
90   Init();
91 }
92
93 //=================================================================================
94 // function : ~RepairGUI_SuppressFacesDlg()
95 // purpose  : Destroys the object and frees any allocated resources
96 //=================================================================================
97 RepairGUI_SuppressFacesDlg::~RepairGUI_SuppressFacesDlg()
98 {
99 }
100
101 //=================================================================================
102 // function : Init()
103 // purpose  :
104 //=================================================================================
105 void RepairGUI_SuppressFacesDlg::Init()
106 {
107   // init variables
108   GroupArgs->LineEdit1->clear();
109   GroupArgs->LineEdit2->clear();
110   myObject = GEOM::GEOM_Object::_nil();
111   myFacesInd = new GEOM::short_array();
112   myFacesInd->length(0);
113
114   mainFrame()->GroupBoxPublish->show();
115
116   // signals and slots connections
117   connect(buttonOk(),    SIGNAL(clicked()), this, SLOT(ClickOnOk()));
118   connect(buttonApply(), SIGNAL(clicked()), this, SLOT(ClickOnApply()));
119
120   connect(GroupArgs->PushButton1, SIGNAL(clicked()),       this, SLOT(SetEditCurrentArgument()));
121   connect(GroupArgs->PushButton2, SIGNAL(clicked()),       this, SLOT(SetEditCurrentArgument()));
122
123   connect(GroupArgs->LineEdit1,   SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
124   connect(GroupArgs->LineEdit2,   SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
125
126   initName(tr("SUPRESS_FACE_NEW_OBJ_NAME"));
127
128   GroupArgs->PushButton1->click();
129   SelectionIntoArgument();
130   resize(100,100);
131 }
132
133 //=================================================================================
134 // function : ClickOnOk()
135 // purpose  : Same than click on apply but close this dialog.
136 //=================================================================================
137 void RepairGUI_SuppressFacesDlg::ClickOnOk()
138 {
139   if (ClickOnApply())
140     ClickOnCancel();
141 }
142
143 //=================================================================================
144 // function : ClickOnApply()
145 // purpose  :
146 //=================================================================================
147 bool RepairGUI_SuppressFacesDlg::ClickOnApply()
148 {
149   if (!onAccept())
150     return false;
151
152   initName();
153   // activate first line edit
154   GroupArgs->PushButton1->click();
155   return true;
156 }
157
158 //=================================================================================
159 // function : SelectionIntoArgument()
160 // purpose  : Called when selection is changed or on dialog initialization or activation
161 //=================================================================================
162 void RepairGUI_SuppressFacesDlg::SelectionIntoArgument()
163 {
164   myEditCurrentArgument->setText("");
165   // the second argument depends on the first one
166   GroupArgs->LineEdit2->setText("");
167   myFacesInd->length(0);
168
169   if (myEditCurrentArgument == GroupArgs->LineEdit1)
170     myObject = GEOM::GEOM_Object::_nil();
171
172   LightApp_SelectionMgr* aSelMgr = myGeomGUI->getApp()->selectionMgr();
173   SALOME_ListIO aSelList;
174   aSelMgr->selectedObjects(aSelList);
175
176   if (myEditCurrentArgument == GroupArgs->LineEdit1) {
177     if (aSelList.Extent() == 1) {
178       Handle(SALOME_InteractiveObject) anIO = aSelList.First();
179
180       Standard_Boolean aRes;
181       myObject = GEOMBase::ConvertIOinGEOMObject(anIO, aRes);
182       if (aRes && GEOMBase::IsShape(myObject)) {
183         myEditCurrentArgument->setText(GEOMBase::GetName(myObject));
184         TopoDS_Shape aShape;
185         if (GEOMBase::GetShape(myObject, aShape, TopAbs_FACE))
186           GroupArgs->LineEdit2->setText(myEditCurrentArgument->text());
187
188         // clear selection
189         disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
190         myGeomGUI->getApp()->selectionMgr()->clearSelected();
191         connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
192                 this, SLOT(SelectionIntoArgument()));
193
194         GroupArgs->PushButton2->click();
195       }
196       else
197         myObject = GEOM::GEOM_Object::_nil();
198     }
199   }
200   else if (myEditCurrentArgument == GroupArgs->LineEdit2) {
201     TColStd_IndexedMapOfInteger aMap;
202
203     if (aSelList.Extent() == 1) {
204       Handle(SALOME_InteractiveObject) anIO = aSelList.First();
205       aSelMgr->GetIndexes(anIO, aMap);
206     }
207
208     if (aMap.IsEmpty() && aSelList.Extent() > 0) { // try to detect selected published sub-shape
209       TColStd_MapOfInteger anIds;
210
211       SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>(myGeomGUI->getApp()->activeStudy());
212       if (!appStudy) return;
213       _PTR(Study) aStudy = appStudy->studyDS();
214
215       TopTools_IndexedMapOfShape aMainMap;
216       TopoDS_Shape aMainShape = GEOM_Client::get_client().GetShape(GeometryGUI::GetGeomGen(), myObject);
217       TopExp::MapShapes(aMainShape, aMainMap);
218
219       SALOME_ListIteratorOfListIO anIter (aSelList);
220       for (int i = 0; anIter.More(); anIter.Next(), i++) {
221         Handle(SALOME_InteractiveObject) anIO = anIter.Value();
222         QString anEntry = anIO->getEntry();
223
224         _PTR(SObject) aSObj (aStudy->FindObjectID(anEntry.toLatin1().constData()));
225         GEOM::GEOM_Object_var aGeomObj =
226           GEOM::GEOM_Object::_narrow(GeometryGUI::ClientSObjectToObject(aSObj));
227         TopoDS_Shape aShape;
228         if (GEOMBase::GetShape(aGeomObj, aShape)) {
229           if (aGeomObj->GetType() == GEOM_GROUP || aShape.ShapeType() == TopAbs_FACE) {
230             TopExp_Explorer anExp (aShape, TopAbs_FACE);
231             for (; anExp.More(); anExp.Next()) {
232               TopoDS_Shape aSubShape = anExp.Current();
233               int anIndex = aMainMap.FindIndex(aSubShape);
234               if (anIndex >= 0) {
235                 aMap.Add(anIndex);
236                 anIds.Add(anIndex);
237               }
238             }
239           }
240         }
241       }
242       if (!aMap.IsEmpty()) {
243         // highlight local faces, correspondent to OB selection
244         disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
245
246         aSelMgr->clearSelected();
247
248         Standard_Boolean isOk;
249         QString objIOR = GEOMBase::GetIORFromObject(myObject);
250         Handle(GEOM_AISShape) aSh = GEOMBase::ConvertIORinGEOMAISShape(objIOR.toLatin1().constData(), isOk, true);
251         if (!isOk || aSh.IsNull())
252           return;
253
254         aSelMgr->AddOrRemoveIndex(aSh->getIO(), anIds, false);
255
256         connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
257                 this, SLOT(SelectionIntoArgument()));
258       }
259     }
260
261     const int n = aMap.Extent();
262     myFacesInd->length(n);
263     for (int i = 1; i <= n; i++)
264       myFacesInd[i-1] = aMap(i);
265     if (n)
266       myEditCurrentArgument->setText(QString::number(n) + "_" + tr("GEOM_FACE") + tr("_S_"));
267   }
268 }
269
270 //=================================================================================
271 // function : SetEditCurrentArgument()
272 // purpose  :
273 //=================================================================================
274 void RepairGUI_SuppressFacesDlg::SetEditCurrentArgument()
275 {
276   QPushButton* send = (QPushButton*)sender();
277
278   bool isEffective = false;
279
280   if (send == GroupArgs->PushButton1) {
281     isEffective = true;
282     myEditCurrentArgument = GroupArgs->LineEdit1;
283
284     GroupArgs->PushButton2->setDown(false);
285     GroupArgs->LineEdit2->setEnabled(false);
286   }
287   else if (send == GroupArgs->PushButton2 && !myObject->_is_nil()) {
288     isEffective = true;
289     myEditCurrentArgument = GroupArgs->LineEdit2;
290
291     GroupArgs->PushButton1->setDown(false);
292     GroupArgs->LineEdit1->setEnabled(false);
293   }
294
295   if (isEffective) {
296     initSelection();
297
298     // enable line edit
299     myEditCurrentArgument->setEnabled(true);
300     myEditCurrentArgument->setFocus();
301     // after setFocus(), because it will be setDown(false) when loses focus
302     send->setDown(true);
303   }
304 }
305
306 //=================================================================================
307 // function : LineEditReturnPressed()
308 // purpose  :
309 //=================================================================================
310 void RepairGUI_SuppressFacesDlg::LineEditReturnPressed()
311 {
312   const QObject* send = sender();
313   if (send == GroupArgs->LineEdit1 || send == GroupArgs->LineEdit2) {
314     myEditCurrentArgument = (QLineEdit*)send;
315     GEOMBase_Skeleton::LineEditReturnPressed();
316   }
317 }
318
319 //=================================================================================
320 // function : ActivateThisDialog()
321 // purpose  :
322 //=================================================================================
323 void RepairGUI_SuppressFacesDlg::ActivateThisDialog()
324 {
325   GEOMBase_Skeleton::ActivateThisDialog();
326   connect( myGeomGUI->getApp()->selectionMgr(), SIGNAL( currentSelectionChanged() ),
327            this, SLOT( SelectionIntoArgument() ) );
328
329   myEditCurrentArgument = GroupArgs->LineEdit1;
330   myEditCurrentArgument->setText( "" );
331   GroupArgs->LineEdit2->setText( "" );
332   myObject = GEOM::GEOM_Object::_nil();
333   myFacesInd->length( 0 );
334
335   initSelection();
336 }
337
338 //=================================================================================
339 // function : enterEvent()
340 // purpose  : Mouse enter onto the dialog to activate it
341 //=================================================================================
342 void RepairGUI_SuppressFacesDlg::enterEvent (QEvent*)
343 {
344   if (!mainFrame()->GroupConstructors->isEnabled())
345     ActivateThisDialog();
346 }
347
348 //=================================================================================
349 // function : createOperation
350 // purpose  :
351 //=================================================================================
352 GEOM::GEOM_IOperations_ptr RepairGUI_SuppressFacesDlg::createOperation()
353 {
354   return getGeomEngine()->GetIHealingOperations(getStudyId());
355 }
356
357 //=================================================================================
358 // function : isValid
359 // purpose  :
360 //=================================================================================
361 bool RepairGUI_SuppressFacesDlg::isValid (QString&)
362 {
363   TopoDS_Shape aTmpShape;
364   return !myObject->_is_nil() && (myFacesInd->length() ||
365                                   GEOMBase::GetShape(myObject, aTmpShape, TopAbs_WIRE));
366 }
367
368 //=================================================================================
369 // function : execute
370 // purpose  :
371 //=================================================================================
372 bool RepairGUI_SuppressFacesDlg::execute (ObjectList& objects)
373 {
374   GEOM::GEOM_IHealingOperations_var anOper = GEOM::GEOM_IHealingOperations::_narrow(getOperation());
375   GEOM::GEOM_Object_var anObj = anOper->SuppressFaces(myObject, myFacesInd);
376
377   bool aResult = !anObj->_is_nil();
378   if (aResult)
379     objects.push_back(anObj._retn());
380
381   return aResult;
382 }
383
384 //=================================================================================
385 // function : initSelection
386 // purpose  :
387 //=================================================================================
388 void RepairGUI_SuppressFacesDlg::initSelection()
389 {
390   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
391
392   if (myEditCurrentArgument == GroupArgs->LineEdit1) {
393     TColStd_MapOfInteger aTypes;
394     aTypes.Add(GEOM_COMPOUND);
395     aTypes.Add(GEOM_SOLID);
396     aTypes.Add(GEOM_SHELL);
397     aTypes.Add(GEOM_FACE);
398
399     globalSelection(aTypes);
400   }
401   else if (myEditCurrentArgument == GroupArgs->LineEdit2) {
402     localSelection(myObject, TopAbs_FACE);
403   }
404
405   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
406           this, SLOT(SelectionIntoArgument()));
407 }
408
409 //=================================================================================
410 // function : restoreSubShapes
411 // purpose  :
412 //=================================================================================
413 void RepairGUI_SuppressFacesDlg::restoreSubShapes (SALOMEDS::Study_ptr   theStudy,
414                                                    SALOMEDS::SObject_ptr theSObject)
415 {
416   if (mainFrame()->CheckBoxRestoreSS->isChecked()) {
417     // empty list of arguments means that all arguments should be restored
418     getGeomEngine()->RestoreSubShapesSO(theStudy, theSObject, GEOM::ListOfGO(),
419                                         /*theFindMethod=*/GEOM::FSM_GetInPlace,
420                                         /*theInheritFirstArg=*/true,
421                                         mainFrame()->CheckBoxAddPrefix->isChecked());
422   }
423 }