Salome HOME
0021189: [CEA 451] areCoordsInside implementation in GE
[modules/geom.git] / src / RepairGUI / RepairGUI_CloseContourDlg.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   : RepairGUI_CloseContourDlg.cxx
25 // Author : Lucien PIGNOLONI, Open CASCADE S.A.S.
26 //
27 #include "RepairGUI_CloseContourDlg.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 <TopAbs.hxx>
40 #include <TColStd_MapOfInteger.hxx>
41 #include <TColStd_IndexedMapOfInteger.hxx>
42
43 #include <GEOMImpl_Types.hxx>
44
45 //=================================================================================
46 // class    : RepairGUI_CloseContourDlg()
47 // purpose  : Constructs a RepairGUI_CloseContourDlg  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_CloseContourDlg::RepairGUI_CloseContourDlg (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_CLOSECONTOUR")));
57   QPixmap image1 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_SELECT")));
58
59   setWindowTitle(tr("GEOM_CLOSECONTOUR_TITLE"));
60
61   /***************************************************************/
62   mainFrame()->GroupConstructors->setTitle(tr("GEOM_CLOSECONTOUR_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_2SelExt(centralWidget());
70   GroupPoints->GroupBox1->setTitle(tr("Contour to close"));
71   GroupPoints->TextLabel1->setText(tr("GEOM_SELECTED_SHAPE"));
72   GroupPoints->PushButton1->setIcon(image1);
73   GroupPoints->LineEdit1->setReadOnly(true);
74
75   GroupPoints->TextLabel2->setText(tr("Contour to close"));
76   GroupPoints->PushButton2->setIcon(image1);
77   GroupPoints->LineEdit2->setReadOnly(true);
78
79   QRadioButton* rb1 = new QRadioButton(tr("Close by common vertex"), GroupPoints->Box);
80   QRadioButton* rb2 = new QRadioButton(tr("Close by new edge"),      GroupPoints->Box);
81
82   myIsVertexGr = new QButtonGroup(GroupPoints->Box);
83   myIsVertexGr->addButton(rb1, 0);
84   myIsVertexGr->addButton(rb2, 1);
85   rb1->setChecked(true);
86
87   QVBoxLayout* l = new QVBoxLayout(GroupPoints->Box);
88   l->setMargin(0); l->setSpacing(6);
89   l->addWidget(rb1);
90   l->addWidget(rb2);
91
92   QVBoxLayout* layout = new QVBoxLayout(centralWidget());
93   layout->setMargin(0); layout->setSpacing(6);
94   layout->addWidget(GroupPoints);
95   /***************************************************************/
96
97   setHelpFileName("close_contour_operation_page.html");
98
99   Init();
100 }
101
102 //=================================================================================
103 // function : ~RepairGUI_CloseContourDlg()
104 // purpose  : Destroys the object and frees any allocated resources
105 //=================================================================================
106 RepairGUI_CloseContourDlg::~RepairGUI_CloseContourDlg()
107 {
108 }
109
110 //=================================================================================
111 // function : Init()
112 // purpose  :
113 //=================================================================================
114 void RepairGUI_CloseContourDlg::Init()
115 {
116   // init variables
117   GroupPoints->LineEdit1->clear();
118   GroupPoints->LineEdit2->clear();
119   myObject = GEOM::GEOM_Object::_nil();
120   myWiresInd = new GEOM::short_array();
121   myWiresInd->length(0);
122
123   // signals and slots connections
124   connect(buttonOk(),    SIGNAL(clicked()), this, SLOT(ClickOnOk()));
125   connect(buttonApply(), SIGNAL(clicked()), this, SLOT(ClickOnApply()));
126
127   connect(GroupPoints->PushButton1, SIGNAL(clicked()),       this, SLOT(SetEditCurrentArgument()));
128   connect(GroupPoints->PushButton2, SIGNAL(clicked()),       this, SLOT(SetEditCurrentArgument()));
129
130   connect(GroupPoints->LineEdit1,   SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
131   connect(GroupPoints->LineEdit2,   SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
132
133   initName(tr("CLOSE_CONTOUR_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_CloseContourDlg::ClickOnOk()
145 {
146   if (ClickOnApply())
147     ClickOnCancel();
148 }
149
150 //=================================================================================
151 // function : ClickOnApply()
152 // purpose  :
153 //=================================================================================
154 bool RepairGUI_CloseContourDlg::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_CloseContourDlg::SelectionIntoArgument()
170 {
171   myEditCurrentArgument->setText("");
172   // the second argument depends on the first one
173   GroupPoints->LineEdit2->setText("");
174   myWiresInd->length(0);
175
176   if (myEditCurrentArgument == GroupPoints->LineEdit1)
177     myObject = GEOM::GEOM_Object::_nil();
178
179   LightApp_SelectionMgr* aSelMgr = myGeomGUI->getApp()->selectionMgr();
180   SALOME_ListIO aSelList;
181   aSelMgr->selectedObjects(aSelList);
182
183   if (aSelList.Extent() == 1) {
184     Handle(SALOME_InteractiveObject) anIO = aSelList.First();
185
186     if (myEditCurrentArgument == GroupPoints->LineEdit1) { // face selection
187       myObject = GEOMBase::ConvertIOinGEOMObject( anIO );
188       if ( 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 }