Salome HOME
Mantis issue 0021071: Regression in GetInPlace. A fix by PKV on Partition error.
[modules/geom.git] / src / GenerationGUI / GenerationGUI_PipeDlg.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   : GenerationGUI_PipeDlg.cxx
25 // Author : Lucien PIGNOLONI, Open CASCADE S.A.S.
26 //
27 #include "GenerationGUI_PipeDlg.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 #include <TopoDS_Shape.hxx>
39 #include <TopoDS.hxx>
40 #include <TopExp.hxx>
41 #include <TColStd_IndexedMapOfInteger.hxx>
42 #include <TopTools_IndexedMapOfShape.hxx>
43 #include <TColStd_IndexedMapOfInteger.hxx>
44 #include <TColStd_MapOfInteger.hxx>
45
46 #include <GEOMImpl_Types.hxx>
47
48 //=================================================================================
49 // class    : GenerationGUI_PipeDlg()
50 // purpose  : Constructs a GenerationGUI_PipeDlg 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 GenerationGUI_PipeDlg::GenerationGUI_PipeDlg (GeometryGUI* theGeometryGUI, QWidget* parent,
56                                               bool modal, Qt::WindowFlags fl)
57   : GEOMBase_Skeleton(theGeometryGUI, parent, modal, fl)
58 {
59   QPixmap image0 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_DLG_PIPE")));
60   QPixmap image1 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_SELECT")));
61   QPixmap image2 (SUIT_Session::session()->resourceMgr()->loadPixmap("GEOM", tr("ICON_DLG_PIPE_BINORMAL")));
62
63   setWindowTitle(tr("GEOM_PIPE_TITLE"));
64
65   /***************************************************************/
66   mainFrame()->GroupConstructors->setTitle(tr("GEOM_PIPE"));
67   mainFrame()->RadioButton1->setIcon(image0);
68   mainFrame()->RadioButton2->setIcon(image2);
69   mainFrame()->RadioButton3->setAttribute(Qt::WA_DeleteOnClose);
70   mainFrame()->RadioButton3->close();
71
72   GroupPoints = new DlgRef_3Sel1Check(centralWidget());
73
74   GroupPoints->GroupBox1->setTitle(tr("GEOM_ARGUMENTS"));
75   GroupPoints->TextLabel1->setText(tr("GEOM_BASE_OBJECT"));
76   GroupPoints->TextLabel2->setText(tr("GEOM_PATH_OBJECT"));
77   GroupPoints->TextLabel3->setText(tr("GEOM_BINORMAL"));
78   GroupPoints->PushButton1->setIcon(image1);
79   GroupPoints->PushButton2->setIcon(image1);
80   GroupPoints->PushButton3->setIcon(image1);
81   GroupPoints->CheckButton1->setText(tr("GEOM_SELECT_UNPUBLISHED_EDGES"));
82
83   QVBoxLayout* layout = new QVBoxLayout(centralWidget());
84   layout->setMargin(0); layout->setSpacing(6);
85   layout->addWidget(GroupPoints);
86   /***************************************************************/
87
88   setHelpFileName("create_extrusion_alongpath_page.html");
89
90   // Initialisation
91   Init();
92 }
93
94 //=================================================================================
95 // function : ~GenerationGUI_PipeDlg()
96 // purpose  : Destroys the object and frees any allocated resources
97 //=================================================================================
98 GenerationGUI_PipeDlg::~GenerationGUI_PipeDlg()
99 {
100   // no need to delete child widgets, Qt does it all for us
101 }
102
103 //=================================================================================
104 // function : Init()
105 // purpose  :
106 //=================================================================================
107 void GenerationGUI_PipeDlg::Init()
108 {
109   // init variables
110   GroupPoints->LineEdit1->setReadOnly(true);
111   GroupPoints->LineEdit2->setReadOnly(true);
112   GroupPoints->LineEdit3->setReadOnly(true);
113
114   GroupPoints->LineEdit1->setText("");
115   GroupPoints->LineEdit2->setText("");
116   GroupPoints->LineEdit3->setText("");
117   myPath = myVec = GEOM::GEOM_Object::_nil();
118   myOkBase = myOkPath = myOkVec = false;
119
120   GroupPoints->CheckButton1->setEnabled(false);
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(this,          SIGNAL(constructorsClicked(int)), this, SLOT(ConstructorsClicked(int)));
127
128   connect(GroupPoints->PushButton1, SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
129   connect(GroupPoints->PushButton2, SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
130   connect(GroupPoints->PushButton3, SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
131
132   connect(GroupPoints->LineEdit1,   SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
133   connect(GroupPoints->LineEdit2,   SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
134   connect(GroupPoints->LineEdit3,   SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
135
136   connect(GroupPoints->CheckButton1,   SIGNAL(toggled(bool)), this, SLOT(SelectionTypeButtonClicked()));
137
138   initName(tr("GEOM_PIPE"));
139   resize(100,100);
140
141   GroupPoints->TextLabel3->hide();
142   GroupPoints->PushButton3->hide();
143   GroupPoints->LineEdit3->hide();
144   ConstructorsClicked(0);
145
146   GroupPoints->PushButton1->click();
147   SelectionIntoArgument();
148 }
149
150 //=================================================================================
151 // function : ConstructorsClicked()
152 // purpose  : Radio button management
153 //=================================================================================
154 void GenerationGUI_PipeDlg::ConstructorsClicked( int constructorId )
155 {
156   erasePreview();
157
158   switch (constructorId)
159   {
160     case 0:
161     {
162       GroupPoints->TextLabel3->hide();
163       GroupPoints->PushButton3->hide();
164       GroupPoints->LineEdit3->hide();
165       break;
166     }
167     case 1:
168     {
169       GroupPoints->TextLabel3->show();
170       GroupPoints->PushButton3->show();
171       GroupPoints->LineEdit3->show();
172       break;
173     }
174   }
175   qApp->processEvents();
176   updateGeometry();
177   resize(100,100);
178
179   displayPreview();
180 }
181
182 //=================================================================================
183 // function : SelectionBittonClicked()
184 // purpose  : Selection type Radio button management
185 //=================================================================================
186 void GenerationGUI_PipeDlg::SelectionTypeButtonClicked()
187 {
188   globalSelection();
189   if ( GroupPoints->CheckButton1->isChecked() ) {
190     localSelection( GEOM::GEOM_Object::_nil(), TopAbs_EDGE );
191   } else {
192     TColStd_MapOfInteger aMap;
193     aMap.Add(GEOM_COMPOUND);
194     aMap.Add(GEOM_WIRE);
195     aMap.Add(GEOM_LINE);
196     globalSelection(aMap);
197   }
198   if(myEditCurrentArgument == GroupPoints->LineEdit2) {
199     myEditCurrentArgument->setText("");
200     myOkPath = false;
201   }
202   displayPreview();
203 }
204
205 //=================================================================================
206 // function : ClickOnOk()
207 // purpose  :
208 //=================================================================================
209 void GenerationGUI_PipeDlg::ClickOnOk()
210 {
211   if (ClickOnApply())
212     ClickOnCancel();
213 }
214
215 //=================================================================================
216 // function : ClickOnApply()
217 // purpose  :
218 //=================================================================================
219 bool GenerationGUI_PipeDlg::ClickOnApply()
220 {
221   if (!onAccept())
222     return false;
223
224   initName();
225   if ( getConstructorId() != 1 )
226         ConstructorsClicked( getConstructorId() );
227   // activate selection and connect selection manager
228   GroupPoints->PushButton1->click();
229   return true;
230 }
231
232 //=================================================================================
233 // function : SelectionIntoArgument()
234 // purpose  : Called when selection is changed or on dialog initialization or activation
235 //=================================================================================
236 void GenerationGUI_PipeDlg::SelectionIntoArgument()
237 {
238   erasePreview();
239   myEditCurrentArgument->setText("");
240   if      (myEditCurrentArgument == GroupPoints->LineEdit1) myOkBase = false;
241   else if (myEditCurrentArgument == GroupPoints->LineEdit2) myOkPath = false;
242   else if (myEditCurrentArgument == GroupPoints->LineEdit3) myOkVec  = false;
243
244   LightApp_SelectionMgr* aSelMgr = myGeomGUI->getApp()->selectionMgr();
245   SALOME_ListIO aSelList;
246   aSelMgr->selectedObjects(aSelList);
247
248   if (aSelList.Extent() < 1)
249     return;
250
251   Standard_Boolean testResult = Standard_False;
252   GEOM::GEOM_Object_var aSelectedObject = GEOMBase::ConvertIOinGEOMObject(aSelList.First(), testResult);
253   QString aName = GEOMBase::GetName(aSelectedObject);
254
255   if (!testResult || aSelectedObject->_is_nil())
256     return;
257
258   TopoDS_Shape S;
259   if (!GEOMBase::GetShape(aSelectedObject, S))
260     return;
261
262   if (myEditCurrentArgument == GroupPoints->LineEdit1) {
263     myOkBase = false;
264     GEOMBase::ConvertListOfIOInListOfGO(aSelList, myBaseObjects, true);
265     if (aSelList.Extent() > 1)
266       aName = QString( "%1_objects").arg( aSelList.Extent() );
267
268     if ( aSelList.Extent() > 0 ) {
269       // check base shapes
270       for (int i=0; i < myBaseObjects.length(); i++) {
271         GEOMBase::GetShape(myBaseObjects[i], S);
272         if (S.ShapeType() == TopAbs_COMPSOLID ||
273             /*S.ShapeType() == TopAbs_COMPOUND ||*/
274             S.ShapeType() == TopAbs_SOLID ||
275             S.ShapeType() == TopAbs_SHAPE)
276           return;
277         if ( getConstructorId() == 1 && 
278              (S.ShapeType() == TopAbs_SHELL || 
279               S.ShapeType() == TopAbs_VERTEX))
280           return;
281       }
282     }
283
284     myEditCurrentArgument->setText(aName);
285     myOkBase = true;
286   } else if (myEditCurrentArgument == GroupPoints->LineEdit2 || 
287            myEditCurrentArgument == GroupPoints->LineEdit3) {
288     myEditCurrentArgument == GroupPoints->LineEdit2 ? myOkPath = false : myOkVec = false;
289     bool myOk = false;
290
291     if (myEditCurrentArgument == GroupPoints->LineEdit2 && !GroupPoints->CheckButton1->isChecked() ){
292       myPath = aSelectedObject;
293       myOkPath = true;
294       myEditCurrentArgument->setText(aName);
295     } else { 
296       TColStd_IndexedMapOfInteger aMap;
297       aSelMgr->GetIndexes(aSelList.First(), aMap);
298       if (aMap.Extent() == 1) {
299         int anIndex = aMap(1);
300         aName.append(":edge_" + QString::number(anIndex));
301         
302         //Find SubShape Object in Father
303         GEOM::GEOM_Object_var aFindedObject = GEOMBase_Helper::findObjectInFather(aSelectedObject, aName);
304
305         if (aFindedObject->_is_nil()) { // Object not found in study
306           GEOM::GEOM_IShapesOperations_var aShapesOp =
307             getGeomEngine()->GetIShapesOperations(getStudyId());
308           aSelectedObject = aShapesOp->GetSubShape(aSelectedObject, anIndex);
309           myOk = true;
310         } else { // get Object from study
311           aSelectedObject = aFindedObject;
312           myOk = true;
313         }
314       } else {
315         myOk = true;
316         if (S.ShapeType() != TopAbs_EDGE) {
317           aSelectedObject = GEOM::GEOM_Object::_nil();
318           aName = "";
319           myOk = false;
320         }
321       }
322       if (myEditCurrentArgument == GroupPoints->LineEdit2) {
323         myPath = aSelectedObject;
324         myOkPath = myOk;
325       } else if (myEditCurrentArgument == GroupPoints->LineEdit3) {
326         myVec = aSelectedObject;
327         myOkVec = myOk;
328       }
329       myEditCurrentArgument->setText(aName);
330       
331       if (myOkPath) {
332         if (!myOkBase)
333           GroupPoints->PushButton1->click();
334         else if (!myOkVec)
335           GroupPoints->PushButton3->click();
336       } else if (myOkVec) {
337         if (!myOkBase)
338           GroupPoints->PushButton1->click();
339         else if (!myOkPath)
340           GroupPoints->PushButton2->click();
341       }
342     }
343   }
344   displayPreview();
345 }
346
347 //=================================================================================
348 // function : SetEditCurrentArgument()
349 // purpose  :
350 //=================================================================================
351 void GenerationGUI_PipeDlg::SetEditCurrentArgument()
352 {
353   QPushButton* send = (QPushButton*)sender();
354
355   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
356   globalSelection(GEOM_ALLSHAPES);
357   GroupPoints->PushButton1->setDown(false);
358   GroupPoints->PushButton2->setDown(false);
359   GroupPoints->PushButton3->setDown(false);
360   GroupPoints->LineEdit1->setEnabled(false);
361   GroupPoints->LineEdit2->setEnabled(false);
362   GroupPoints->LineEdit3->setEnabled(false);
363   if (send == GroupPoints->PushButton1) {
364     myEditCurrentArgument = GroupPoints->LineEdit1;
365     GroupPoints->CheckButton1->setEnabled(false);
366   }
367   else if (send == GroupPoints->PushButton2) {
368     myEditCurrentArgument = GroupPoints->LineEdit2;
369
370     if ( GroupPoints->CheckButton1->isChecked() ) {
371       localSelection( GEOM::GEOM_Object::_nil(), TopAbs_EDGE );
372     } else {
373       TColStd_MapOfInteger aMap;
374       aMap.Add(GEOM_COMPOUND);
375       aMap.Add(GEOM_WIRE);
376       aMap.Add(GEOM_LINE);
377       globalSelection(aMap);
378     }
379     GroupPoints->CheckButton1->setEnabled(true);
380   }
381   else if(send == GroupPoints->PushButton3) {
382     myEditCurrentArgument = GroupPoints->LineEdit3;
383     GroupPoints->CheckButton1->setEnabled(false);
384     localSelection(GEOM::GEOM_Object::_nil(), TopAbs_EDGE);
385   }
386   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
387           this, SLOT(SelectionIntoArgument()));
388
389   // enable line edit
390   myEditCurrentArgument->setEnabled(true);
391   myEditCurrentArgument->setFocus();
392   // after setFocus(), because it will be setDown(false) when loses focus
393   send->setDown(true);
394
395   // seems we need it only to avoid preview disappearing, caused by selection mode change
396   displayPreview();
397 }
398
399 //=================================================================================
400 // function : LineEditReturnPressed()
401 // purpose  :
402 //=================================================================================
403 void GenerationGUI_PipeDlg::LineEditReturnPressed()
404 {
405   QLineEdit* send = (QLineEdit*)sender();
406   if (send == GroupPoints->LineEdit1 ||
407       send == GroupPoints->LineEdit2 ||
408       send == GroupPoints->LineEdit3) {
409     myEditCurrentArgument = send;
410     GEOMBase_Skeleton::LineEditReturnPressed();
411   }
412
413   if (send == GroupPoints->LineEdit2)
414     GroupPoints->CheckButton1->setEnabled(true);
415   else
416     GroupPoints->CheckButton1->setEnabled(false);
417 }
418
419 //=================================================================================
420 // function : ActivateThisDialog()
421 // purpose  :
422 //=================================================================================
423 void GenerationGUI_PipeDlg::ActivateThisDialog()
424 {
425   GEOMBase_Skeleton::ActivateThisDialog();
426   connect( myGeomGUI->getApp()->selectionMgr(), SIGNAL( currentSelectionChanged() ),
427            this, SLOT( SelectionIntoArgument() ) );
428
429   ConstructorsClicked(getConstructorId());
430 }
431
432 //=================================================================================
433 // function : enterEvent()
434 // purpose  : when mouse enter onto the QWidget
435 //=================================================================================
436 void GenerationGUI_PipeDlg::enterEvent (QEvent*)
437 {
438   if (!mainFrame()->GroupConstructors->isEnabled())
439     ActivateThisDialog();
440 }
441
442 //=================================================================================
443 // function : createOperation
444 // purpose  :
445 //=================================================================================
446 GEOM::GEOM_IOperations_ptr GenerationGUI_PipeDlg::createOperation()
447 {
448   return getGeomEngine()->GetI3DPrimOperations(getStudyId());
449 }
450
451 //=================================================================================
452 // function : isValid
453 // purpose  :
454 //=================================================================================
455 bool GenerationGUI_PipeDlg::isValid (QString&)
456 {
457   switch ( getConstructorId() ) {
458   case 0 :
459     return myOkBase && myOkPath;
460   case 1 :
461     return myOkBase && myOkPath && myOkVec;
462   }
463 }
464
465 //=================================================================================
466 // function : execute
467 // purpose  :
468 //=================================================================================
469 bool GenerationGUI_PipeDlg::execute (ObjectList& objects)
470 {
471   GEOM::GEOM_Object_var anObj, aBase;
472
473   GEOM::GEOM_I3DPrimOperations_var anOper = GEOM::GEOM_I3DPrimOperations::_narrow(getOperation());
474
475   for (int i=0; i < myBaseObjects.length(); i++) {
476     aBase = myBaseObjects[i];
477     switch ( getConstructorId() ) {
478     case 0 :
479       anObj = anOper->MakePipe(aBase, myPath);
480       break;
481     case 1 :
482       anObj = anOper->MakePipeBiNormalAlongVector(aBase, myPath, myVec);
483       break;
484     }
485     
486     if (!anObj->_is_nil())
487       objects.push_back(anObj._retn());
488   }
489   return true;
490 }
491
492 //=================================================================================
493 // function : addSubshapeToStudy
494 // purpose  : virtual method to add new SubObjects if local selection
495 //=================================================================================
496 void GenerationGUI_PipeDlg::addSubshapesToStudy()
497 {
498   QMap<QString, GEOM::GEOM_Object_var> objMap;
499
500   objMap[GroupPoints->LineEdit2->text()] = myPath;
501
502   addSubshapesToFather(objMap);
503 }
504
505 //=================================================================================
506 // function : extractPrefix
507 // purpose  : redefined from GEOMBase_Helper class
508 //=================================================================================
509 bool GenerationGUI_PipeDlg::extractPrefix() const
510 {
511   return myBaseObjects.length() > 1;
512 }