Salome HOME
Mantis issue 0020508: EDF 1059 GEOM : Selection of a group in Suppress Faces.
[modules/geom.git] / src / RepairGUI / RepairGUI_CloseContourDlg.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_CloseContourDlg.cxx
24 // Author : Lucien PIGNOLONI, Open CASCADE S.A.S.
25 //
26 #include "RepairGUI_CloseContourDlg.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 <LightApp_SelectionMgr.h>
36
37 // OCCT Includes
38 #include <TopAbs.hxx>
39 #include <TColStd_MapOfInteger.hxx>
40 #include <TColStd_IndexedMapOfInteger.hxx>
41
42 #include <GEOMImpl_Types.hxx>
43
44 //=================================================================================
45 // class    : RepairGUI_CloseContourDlg()
46 // purpose  : Constructs a RepairGUI_CloseContourDlg  which is a child of 'parent', with the
47 //            name 'name' and widget flags set to 'f'.
48 //            The dialog will by default be modeless, unless you set 'modal' to
49 //            TRUE to construct a modal dialog.
50 //=================================================================================
51 RepairGUI_CloseContourDlg::RepairGUI_CloseContourDlg (GeometryGUI* theGeometryGUI, QWidget* parent,
52                                                       bool modal)
53   : GEOMBase_Skeleton(theGeometryGUI, parent, modal)
54 {
55   QPixmap image0 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_DLG_CLOSECONTOUR")));
56   QPixmap image1 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_SELECT")));
57
58   setWindowTitle(tr("GEOM_CLOSECONTOUR_TITLE"));
59
60   /***************************************************************/
61   mainFrame()->GroupConstructors->setTitle(tr("GEOM_CLOSECONTOUR_TITLE"));
62   mainFrame()->RadioButton1->setIcon(image0);
63   mainFrame()->RadioButton2->setAttribute(Qt::WA_DeleteOnClose);
64   mainFrame()->RadioButton2->close();
65   mainFrame()->RadioButton3->setAttribute(Qt::WA_DeleteOnClose);
66   mainFrame()->RadioButton3->close();
67
68   GroupPoints = new DlgRef_2SelExt(centralWidget());
69   GroupPoints->GroupBox1->setTitle(tr("Contour to close"));
70   GroupPoints->TextLabel1->setText(tr("GEOM_SELECTED_SHAPE"));
71   GroupPoints->PushButton1->setIcon(image1);
72   GroupPoints->LineEdit1->setReadOnly(true);
73
74   GroupPoints->TextLabel2->setText(tr("Contour to close"));
75   GroupPoints->PushButton2->setIcon(image1);
76   GroupPoints->LineEdit2->setReadOnly(true);
77
78   QRadioButton* rb1 = new QRadioButton(tr("Close by common vertex"), GroupPoints->Box);
79   QRadioButton* rb2 = new QRadioButton(tr("Close by new edge"),      GroupPoints->Box);
80
81   myIsVertexGr = new QButtonGroup(GroupPoints->Box);
82   myIsVertexGr->addButton(rb1, 0);
83   myIsVertexGr->addButton(rb2, 1);
84   rb1->setChecked(true);
85
86   QVBoxLayout* l = new QVBoxLayout(GroupPoints->Box);
87   l->setMargin(0); l->setSpacing(6);
88   l->addWidget(rb1);
89   l->addWidget(rb2);
90
91   QVBoxLayout* layout = new QVBoxLayout(centralWidget());
92   layout->setMargin(0); layout->setSpacing(6);
93   layout->addWidget(GroupPoints);
94   /***************************************************************/
95
96   setHelpFileName("close_contour_operation_page.html");
97
98   Init();
99 }
100
101 //=================================================================================
102 // function : ~RepairGUI_CloseContourDlg()
103 // purpose  : Destroys the object and frees any allocated resources
104 //=================================================================================
105 RepairGUI_CloseContourDlg::~RepairGUI_CloseContourDlg()
106 {
107 }
108
109 //=================================================================================
110 // function : Init()
111 // purpose  :
112 //=================================================================================
113 void RepairGUI_CloseContourDlg::Init()
114 {
115   // init variables
116   GroupPoints->LineEdit1->clear();
117   GroupPoints->LineEdit2->clear();
118   myObject = GEOM::GEOM_Object::_nil();
119   myWiresInd = new GEOM::short_array();
120   myWiresInd->length(0);
121
122   // signals and slots connections
123   connect(buttonOk(),    SIGNAL(clicked()), this, SLOT(ClickOnOk()));
124   connect(buttonApply(), SIGNAL(clicked()), this, SLOT(ClickOnApply()));
125
126   connect(GroupPoints->PushButton1, SIGNAL(clicked()),       this, SLOT(SetEditCurrentArgument()));
127   connect(GroupPoints->PushButton2, SIGNAL(clicked()),       this, SLOT(SetEditCurrentArgument()));
128
129   connect(GroupPoints->LineEdit1,   SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
130   connect(GroupPoints->LineEdit2,   SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
131
132   initName(tr("CLOSE_CONTOUR_NEW_OBJ_NAME"));
133
134   GroupPoints->PushButton1->click();
135   SelectionIntoArgument();
136   resize(100,100);
137 }
138
139 //=================================================================================
140 // function : ClickOnOk()
141 // purpose  : Same than click on apply but close this dialog.
142 //=================================================================================
143 void RepairGUI_CloseContourDlg::ClickOnOk()
144 {
145   if (ClickOnApply())
146     ClickOnCancel();
147 }
148
149 //=================================================================================
150 // function : ClickOnApply()
151 // purpose  :
152 //=================================================================================
153 bool RepairGUI_CloseContourDlg::ClickOnApply()
154 {
155   if (!onAccept())
156     return false;
157
158   initName();
159   // activate first line edit
160   GroupPoints->PushButton1->click();
161   return true;
162 }
163
164 //=================================================================================
165 // function : SelectionIntoArgument()
166 // purpose  : Called when selection is changed or on dialog initialization or activation
167 //=================================================================================
168 void RepairGUI_CloseContourDlg::SelectionIntoArgument()
169 {
170   myEditCurrentArgument->setText("");
171   // the second argument depends on the first one
172   GroupPoints->LineEdit2->setText("");
173   myWiresInd->length(0);
174
175   if (myEditCurrentArgument == GroupPoints->LineEdit1)
176     myObject = GEOM::GEOM_Object::_nil();
177
178   LightApp_SelectionMgr* aSelMgr = myGeomGUI->getApp()->selectionMgr();
179   SALOME_ListIO aSelList;
180   aSelMgr->selectedObjects(aSelList);
181
182   if (aSelList.Extent() == 1) {
183     Handle(SALOME_InteractiveObject) anIO = aSelList.First();
184
185     if (myEditCurrentArgument == GroupPoints->LineEdit1) { // face selection
186       Standard_Boolean aRes;
187       myObject = GEOMBase::ConvertIOinGEOMObject(anIO, aRes);
188       if (aRes && GEOMBase::IsShape(myObject)) {
189         myEditCurrentArgument->setText(GEOMBase::GetName(myObject));
190         TopoDS_Shape aShape;
191         if (GEOMBase::GetShape(myObject, aShape, TopAbs_WIRE))
192           GroupPoints->LineEdit2->setText(myEditCurrentArgument->text());
193
194         // clear selection
195         disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
196         myGeomGUI->getApp()->selectionMgr()->clearSelected();
197         connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
198                 this, SLOT(SelectionIntoArgument()));
199
200         GroupPoints->PushButton2->click();
201       }
202       else
203         myObject = GEOM::GEOM_Object::_nil();
204     }
205     else if (myEditCurrentArgument == GroupPoints->LineEdit2) {
206       TColStd_IndexedMapOfInteger aMap;
207       aSelMgr->GetIndexes(anIO, aMap);
208       const int n = aMap.Extent();
209       myWiresInd->length(n);
210       for (int i = 1; i <= n; i++)
211         myWiresInd[i-1] = aMap(i);
212       if (n)
213         myEditCurrentArgument->setText(QString::number(n) + "_" + tr("GEOM_WIRE") + tr("_S_"));
214     }
215   }
216 }
217
218 //=================================================================================
219 // function : SetEditCurrentArgument()
220 // purpose  :
221 //=================================================================================
222 void RepairGUI_CloseContourDlg::SetEditCurrentArgument()
223 {
224   QPushButton* send = (QPushButton*)sender();
225
226   bool isEffective = false;
227
228   if (send == GroupPoints->PushButton1) {
229     isEffective = true;
230     myEditCurrentArgument = GroupPoints->LineEdit1;
231
232     GroupPoints->PushButton2->setDown(false);
233     GroupPoints->LineEdit2->setEnabled(false);
234   }
235   else if (send == GroupPoints->PushButton2 && !myObject->_is_nil()) {
236     isEffective = true;
237     myEditCurrentArgument = GroupPoints->LineEdit2;
238
239     GroupPoints->PushButton1->setDown(false);
240     GroupPoints->LineEdit1->setEnabled(false);
241   }
242
243   if (isEffective) {
244     initSelection();
245
246     // enable line edit
247     myEditCurrentArgument->setEnabled(true);
248     myEditCurrentArgument->setFocus();
249     // after setFocus(), because it will be setDown(false) when loses focus
250     send->setDown(true);
251   }
252 }
253
254 //=================================================================================
255 // function : LineEditReturnPressed()
256 // purpose  :
257 //=================================================================================
258 void RepairGUI_CloseContourDlg::LineEditReturnPressed()
259 {
260   const QObject* send = sender();
261   if (send == GroupPoints->LineEdit1 || send == GroupPoints->LineEdit2) {
262     myEditCurrentArgument = (QLineEdit*)send;
263     GEOMBase_Skeleton::LineEditReturnPressed();
264   }
265 }
266
267 //=================================================================================
268 // function : ActivateThisDialog()
269 // purpose  :
270 //=================================================================================
271 void RepairGUI_CloseContourDlg::ActivateThisDialog()
272 {
273   GEOMBase_Skeleton::ActivateThisDialog();
274   connect( myGeomGUI->getApp()->selectionMgr(), SIGNAL( currentSelectionChanged() ),
275            this, SLOT( SelectionIntoArgument() ) );
276
277   myEditCurrentArgument = GroupPoints->LineEdit1;
278   myEditCurrentArgument->setText( "" );
279   GroupPoints->LineEdit2->setText( "" );
280   myObject = GEOM::GEOM_Object::_nil();
281   myWiresInd->length( 0 );
282
283   initSelection();
284 }
285
286 //=================================================================================
287 // function : enterEvent()
288 // purpose  : Mouse enter onto the dialog to activate it
289 //=================================================================================
290 void RepairGUI_CloseContourDlg::enterEvent (QEvent*)
291 {
292   if (!mainFrame()->GroupConstructors->isEnabled())
293     ActivateThisDialog();
294 }
295
296 //=================================================================================
297 // function : createOperation
298 // purpose  :
299 //=================================================================================
300 GEOM::GEOM_IOperations_ptr RepairGUI_CloseContourDlg::createOperation()
301 {
302   return getGeomEngine()->GetIHealingOperations(getStudyId());
303 }
304
305 //=================================================================================
306 // function : isValid
307 // purpose  :
308 //=================================================================================
309 bool RepairGUI_CloseContourDlg::isValid (QString&)
310 {
311   TopoDS_Shape aTmpShape;
312   return !myObject->_is_nil() && (myWiresInd->length() ||
313                                   GEOMBase::GetShape(myObject, aTmpShape, TopAbs_WIRE));
314 }
315
316 //=================================================================================
317 // function : execute
318 // purpose  :
319 //=================================================================================
320 bool RepairGUI_CloseContourDlg::execute (ObjectList& objects)
321 {
322   GEOM::GEOM_IHealingOperations_var anOper = GEOM::GEOM_IHealingOperations::_narrow(getOperation());
323   GEOM::GEOM_Object_var anObj = anOper->CloseContour(myObject, myWiresInd, getIsByVertex());
324
325   bool aResult = !anObj->_is_nil();
326   if (aResult)
327     objects.push_back(anObj._retn());
328
329   return aResult;
330 }
331
332 //=================================================================================
333 // function : getIsByVertex
334 // purpose  :
335 //=================================================================================
336 bool RepairGUI_CloseContourDlg::getIsByVertex() const
337 {
338   return myIsVertexGr->button(0)->isChecked();
339 }
340
341 //=================================================================================
342 // function : initSelection
343 // purpose  :
344 //=================================================================================
345 void RepairGUI_CloseContourDlg::initSelection()
346 {
347   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
348
349   if (myEditCurrentArgument == GroupPoints->LineEdit1) {
350     TColStd_MapOfInteger aTypes;
351     aTypes.Add(GEOM_COMPOUND);
352     aTypes.Add(GEOM_SOLID);
353     aTypes.Add(GEOM_SHELL);
354     aTypes.Add(GEOM_FACE);
355     aTypes.Add(GEOM_WIRE);
356
357     globalSelection(aTypes);
358   }
359   else if (myEditCurrentArgument == GroupPoints->LineEdit2) {
360     localSelection(myObject, TopAbs_EDGE);
361     localSelection(myObject, TopAbs_WIRE);
362   }
363
364   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
365           this, SLOT(SelectionIntoArgument()));
366 }