]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESHGUI/SMESHGUI_ShapeByMeshDlg.cxx
Salome HOME
untabify
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_ShapeByMeshDlg.cxx
1 // Copyright (C) 2007-2011  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 // File   : SMESHGUI_ShapeByMeshDlg.cxx
24 // Author : Edward AGAPOV, Open CASCADE S.A.S.
25 // SMESH includes
26 //
27 #include "SMESHGUI_ShapeByMeshDlg.h"
28
29 #include "SMESHGUI.h"
30 #include "SMESHGUI_GEOMGenUtils.h"
31 #include "SMESHGUI_IdValidator.h"
32 #include "SMESHGUI_MeshUtils.h"
33 #include "SMESHGUI_Utils.h"
34 #include "SMESHGUI_VTKUtils.h"
35
36 #include <SMDS_Mesh.hxx>
37 #include <SMDS_MeshNode.hxx>
38 #include <SMESH_Actor.h>
39
40 // SALOME GEOM includes
41 #include <GEOMBase.h>
42 #include <GeometryGUI.h>
43
44 // SALOME GUI includes
45 #include <LightApp_DataOwner.h>
46 #include <LightApp_SelectionMgr.h>
47 #include <SALOME_ListIO.hxx>
48 #include <SUIT_Desktop.h>
49 #include <SVTK_Selector.h>
50 #include <SVTK_ViewWindow.h>
51 #include <SVTK_ViewModel.h>
52 #include <SalomeApp_Tools.h>
53
54 // SALOME KERNEL includes
55 #include <SALOMEDSClient_SObject.hxx>
56
57 // OCCT includes
58 #include <TColStd_MapOfInteger.hxx>
59 #include <TopoDS_Shape.hxx>
60 #include <TopExp_Explorer.hxx>
61
62 // Qt includes
63 #include <QFrame>
64 #include <QVBoxLayout>
65 #include <QHBoxLayout>
66 #include <QGridLayout>
67 #include <QLineEdit>
68 #include <QLabel>
69 #include <QRadioButton>
70 #include <QButtonGroup>
71 #include <QGroupBox>
72 #include <QStringList>
73
74 #define SPACING 6
75 #define MARGIN  11
76
77 enum { EDGE = 0, FACE, VOLUME };
78
79 /*!
80  * \brief Dialog to publish a sub-shape of the mesh main shape
81  *        by selecting mesh elements
82  */
83 SMESHGUI_ShapeByMeshDlg::SMESHGUI_ShapeByMeshDlg()
84   : SMESHGUI_Dialog( 0, false, true, OK | Close )
85 {
86   setWindowTitle(tr("CAPTION"));
87
88   QVBoxLayout* aDlgLay = new QVBoxLayout (mainFrame());
89   aDlgLay->setMargin(MARGIN);
90   aDlgLay->setSpacing(SPACING);
91
92   QFrame* aMainFrame = createMainFrame  (mainFrame());
93
94   aDlgLay->addWidget(aMainFrame);
95
96   aDlgLay->setStretchFactor(aMainFrame, 1);
97 }
98
99 //=======================================================================
100 // function : createMainFrame()
101 // purpose  : Create frame containing dialog's input fields
102 //=======================================================================
103 QFrame* SMESHGUI_ShapeByMeshDlg::createMainFrame (QWidget* theParent)
104 {
105   QFrame* aMainGrp = new QFrame(theParent);
106   QGridLayout* aLayout = new QGridLayout(aMainGrp);
107   aLayout->setMargin(0);
108   aLayout->setSpacing(SPACING);
109
110   // elem type
111   myElemTypeBox = new QGroupBox(tr("SMESH_ELEMENT_TYPE"), aMainGrp);
112   myElemTypeGroup = new QButtonGroup(aMainGrp);
113   QHBoxLayout* myElemTypeBoxLayout = new QHBoxLayout(myElemTypeBox);
114   myElemTypeBoxLayout->setMargin(MARGIN);
115   myElemTypeBoxLayout->setSpacing(SPACING);
116
117   QRadioButton* aEdgeRb   = new QRadioButton( tr("SMESH_EDGE"),   myElemTypeBox);
118   QRadioButton* aFaceRb   = new QRadioButton( tr("SMESH_FACE"),   myElemTypeBox);
119   QRadioButton* aVolumeRb = new QRadioButton( tr("SMESH_VOLUME"), myElemTypeBox);
120   
121   myElemTypeBoxLayout->addWidget(aEdgeRb);
122   myElemTypeBoxLayout->addWidget(aFaceRb);
123   myElemTypeBoxLayout->addWidget(aVolumeRb);
124   myElemTypeGroup->addButton(aEdgeRb, 0);
125   myElemTypeGroup->addButton(aFaceRb, 1);
126   myElemTypeGroup->addButton(aVolumeRb, 2);
127   aEdgeRb->setChecked(true);
128   
129   // element id
130   QLabel* anIdLabel = new QLabel( tr("ELEMENT_ID"), aMainGrp );
131   myElementId = new QLineEdit( aMainGrp );
132   myElementId->setValidator( new SMESHGUI_IdValidator( theParent, 
133                                                        !myIsMultipleAllowed ? 1 : 0 ) ); // 0 for any number of entities
134
135   // shape name
136   QLabel* aNameLabel = new QLabel( tr("GEOMETRY_NAME"), aMainGrp );
137   myGeomName = new QLineEdit( aMainGrp );
138
139   aLayout->addWidget(myElemTypeBox, 0, 0, 1, 2);
140   aLayout->addWidget(anIdLabel,   1, 0);
141   aLayout->addWidget(myElementId, 1, 1);
142   aLayout->addWidget(aNameLabel,  2, 0);
143   aLayout->addWidget(myGeomName,  2, 1);
144
145   return aMainGrp;
146 }
147
148 //=======================================================================
149 // function : ~SMESHGUI_ShapeByMeshDlg()
150 // purpose  : Destructor
151 //=======================================================================
152 SMESHGUI_ShapeByMeshDlg::~SMESHGUI_ShapeByMeshDlg()
153 {
154 }
155
156 void SMESHGUI_ShapeByMeshDlg:: setMultipleAllowed( bool isAllowed )
157 {
158   myIsMultipleAllowed = isAllowed;
159 }
160
161 //================================================================================
162 /*!
163  * \brief Constructor
164  */
165 //================================================================================
166 SMESHGUI_ShapeByMeshOp::SMESHGUI_ShapeByMeshOp(bool isMultipleAllowed):
167   myIsMultipleAllowed(isMultipleAllowed)
168 {
169   if ( GeometryGUI::GetGeomGen()->_is_nil() )// check that GEOM_Gen exists
170     GeometryGUI::InitGeomGen();
171
172   myDlg = new SMESHGUI_ShapeByMeshDlg;
173   myDlg->setMultipleAllowed(myIsMultipleAllowed);
174
175   connect(myDlg->myElemTypeGroup, SIGNAL(buttonClicked(int)), SLOT(onTypeChanged(int)));
176   connect(myDlg->myElementId, SIGNAL(textChanged(const QString&)), SLOT(onElemIdChanged(const QString&)));
177 }
178
179
180 //=======================================================================
181 // function : startOperation()
182 // purpose  : Init dialog fields, connect signals and slots, show dialog
183 //=======================================================================
184 void SMESHGUI_ShapeByMeshOp::startOperation()
185 {
186   //SetMesh( SMESH::SMESH_Mesh::_nil() );
187   myIsManualIdEnter = false;
188
189   SMESHGUI_SelectionOp::startOperation();
190
191   //activateSelection(); // set filters
192   onSelectionDone(); // desable/enable [ OK ]
193
194   myDlg->show();
195 }
196
197 //================================================================================
198 /*!
199  * \brief Destructor
200  */
201 //================================================================================
202 SMESHGUI_ShapeByMeshOp::~SMESHGUI_ShapeByMeshOp()
203 {
204   if ( myDlg )
205     delete myDlg;
206 }
207
208 //================================================================================
209 /*!
210  * \brief Gets dialog of this operation
211  * \retval LightApp_Dialog* - pointer to dialog of this operation
212  */
213 //================================================================================
214 LightApp_Dialog* SMESHGUI_ShapeByMeshOp::dlg() const
215 {
216   return myDlg;
217 }
218
219 SMESH::SMESH_Mesh_ptr SMESHGUI_ShapeByMeshOp::GetMesh()
220 {
221   return myMesh;
222 }
223
224 //=======================================================================
225 // function : GetShape()
226 // purpose  : Get published sub-shape
227 //=======================================================================
228 GEOM::GEOM_Object_ptr SMESHGUI_ShapeByMeshOp::GetShape()
229 {
230   return myGeomObj.in();
231 }
232
233 //=======================================================================
234 // function : SetMesh()
235 // purpose  : Set mesh to dialog
236 //=======================================================================
237
238 void SMESHGUI_ShapeByMeshOp::SetMesh (SMESH::SMESH_Mesh_ptr thePtr)
239 {
240   myMesh    = SMESH::SMESH_Mesh::_duplicate(thePtr);
241   myGeomObj = GEOM::GEOM_Object::_nil();
242   myHasSolids = false;
243
244   std::vector< bool > hasElement (myDlg->myElemTypeGroup->buttons().count(), false);
245   if (!myMesh->_is_nil() )
246     {
247       //     _PTR(SObject) aSobj = SMESH::FindSObject(myMesh.in());
248       //     SUIT_DataOwnerPtr anIObj (new LightApp_DataOwner(aSobj->GetID().c_str()));
249
250       std::vector< int > nbShapes( TopAbs_SHAPE, 0 );
251       int shapeDim = 0; // max dim with several shapes
252       //if ( /*mySelectionMgr*/ selectionMgr()->isOk(anIObj) ) // check that the mesh has a valid shape
253       {
254         _PTR(SObject) aSO = SMESH::FindSObject(myMesh.in());
255         GEOM::GEOM_Object_var mainShape = SMESH::GetGeom(aSO);
256         if ( !mainShape->_is_nil() ) 
257           {
258             TopoDS_Shape aShape;
259             if ( GEOMBase::GetShape(mainShape, aShape))
260               {
261                 TopAbs_ShapeEnum types[4] = { TopAbs_EDGE, TopAbs_FACE, TopAbs_SHELL, TopAbs_SOLID };
262                 for ( int dim = 4; dim > 0; --dim ) {
263                   TopAbs_ShapeEnum type = types[ dim - 1 ];
264                   TopAbs_ShapeEnum avoid = ( type == TopAbs_SHELL ) ? TopAbs_SOLID : TopAbs_SHAPE;
265                   TopExp_Explorer exp( aShape, type, avoid );
266                   for ( ; nbShapes[ type ] < 2 && exp.More(); exp.Next() )
267                     ++nbShapes[ type ];
268                   if ( nbShapes[ type ] > 1 ) {
269                     shapeDim = dim;
270                     break;
271                   }
272                 }
273               }
274           }
275       }
276       if (shapeDim > 0)
277         {
278           if ( nbShapes[ TopAbs_SHELL ] + nbShapes[ TopAbs_SOLID ] > 1 )
279             shapeDim = 3;
280           hasElement[ EDGE ]   = shapeDim > 0 && myMesh->NbEdges();
281           hasElement[ FACE ]   = shapeDim > 1 && myMesh->NbFaces();
282           hasElement[ VOLUME ] = shapeDim > 2 && myMesh->NbVolumes();
283         }
284       myHasSolids = nbShapes[ TopAbs_SOLID ];
285     }
286
287   // disable inexistant elem types
288   for ( int i = 0; i < myDlg->myElemTypeGroup->buttons().count(); ++i ) {
289     if ( QAbstractButton* button = myDlg->myElemTypeGroup->button( i ) )
290       button->setEnabled( hasElement[ i ] );
291   }
292   myDlg->myElementId->setEnabled( hasElement[ EDGE ] );
293   myDlg->myGeomName-> setEnabled( hasElement[ EDGE ] );
294
295   setElementID("");
296 }
297
298 //=======================================================================
299 // function : commitOperation()
300 // purpose  : called when "Ok" button pressed.
301 //=======================================================================
302
303 void SMESHGUI_ShapeByMeshOp::commitOperation()
304 {
305   SMESHGUI_SelectionOp::commitOperation();
306   try {
307     QStringList aListId = myDlg->myElementId->text().split( " ", QString::SkipEmptyParts);
308     if (aListId.count() == 1)
309       {
310         int elemID = (aListId.first()).toInt();
311         myGeomObj = GEOM::GEOM_Object::_duplicate(
312             SMESHGUI::GetSMESHGen()->GetGeometryByMeshElement
313           ( myMesh.in(), elemID, myDlg->myGeomName->text().toLatin1().constData()) );
314       }
315     else
316       {
317         GEOM::GEOM_Gen_var geomGen = SMESH::GetGEOMGen();
318         _PTR(Study) aStudy = SMESH::GetActiveStudyDocument();
319         
320         if (geomGen->_is_nil() || !aStudy)
321           return;
322         
323         GEOM::GEOM_IShapesOperations_var aShapesOp =
324           geomGen->GetIShapesOperations(aStudy->StudyId());
325         if (aShapesOp->_is_nil() )
326           return;
327         
328         TopAbs_ShapeEnum aGroupType = TopAbs_SHAPE;
329         
330         std::map<double, GEOM::GEOM_Object_var> aGeomObjectsMap;
331         GEOM::GEOM_Object_var aGeomObject;
332
333         GEOM::GEOM_Object_var aMeshShape = myMesh->GetShapeToMesh();
334         
335         for ( int i = 0; i < aListId.count(); i++ )
336           {
337             aGeomObject =
338               SMESHGUI::GetSMESHGen()->FindGeometryByMeshElement(myMesh.in(), aListId[i].toInt());
339
340             if (aGeomObject->_is_nil()) continue;
341             
342             double anId = aShapesOp->GetSubShapeIndex(aMeshShape, aGeomObject);
343             if (aShapesOp->IsDone() && aGeomObjectsMap.find(anId) == aGeomObjectsMap.end())
344               {
345                 aGeomObjectsMap[anId] = aGeomObject;
346
347                 TopAbs_ShapeEnum aSubShapeType = (TopAbs_ShapeEnum)aGeomObject->GetShapeType();
348                 if (i == 0)
349                   aGroupType = aSubShapeType;
350                 else if (aSubShapeType != aGroupType)
351                   aGroupType = TopAbs_SHAPE;
352               }
353           }
354         
355         int aNumberOfGO = aGeomObjectsMap.size();
356         if (aNumberOfGO == 1)
357           myGeomObj = (*aGeomObjectsMap.begin()).second;
358         else if (aNumberOfGO > 1)
359           {
360             GEOM::GEOM_IGroupOperations_var aGroupOp =
361               geomGen->GetIGroupOperations(aStudy->StudyId());
362             if(aGroupOp->_is_nil())
363               return;
364             
365             GEOM::ListOfGO_var aGeomObjects = new GEOM::ListOfGO();
366             aGeomObjects->length( aNumberOfGO );
367
368             int i = 0;
369             std::map<double, GEOM::GEOM_Object_var>::iterator anIter;
370             for (anIter = aGeomObjectsMap.begin(); anIter!=aGeomObjectsMap.end(); anIter++)
371               aGeomObjects[i++] = (*anIter).second;
372           
373             //create geometry group
374             myGeomObj = aGroupOp->CreateGroup(aMeshShape, aGroupType);
375             aGroupOp->UnionList(myGeomObj, aGeomObjects);
376
377             if (!aGroupOp->IsDone())
378               return;
379           }
380         
381         // publish the GEOM object in study
382         QString aNewGeomGroupName ( myDlg->myGeomName->text() );
383           
384         SALOMEDS::SObject_var aNewGroupSO =
385           geomGen->AddInStudy(SMESHGUI::GetSMESHGen()->GetCurrentStudy(), myGeomObj, 
386                               aNewGeomGroupName.toLatin1().data(), aMeshShape);
387       }
388   }
389   catch (const SALOME::SALOME_Exception& S_ex) {
390     SalomeApp_Tools::QtCatchCorbaException(S_ex);
391   }
392   catch (...) {
393   }
394
395 }
396
397 bool SMESHGUI_ShapeByMeshOp::onApply()
398 {
399   return true;
400 }
401
402 //=======================================================================
403 // function : onSelectionDone()
404 // purpose  : SLOT called when selection changed. Enable/desable [ OK ]
405 //=======================================================================
406 void SMESHGUI_ShapeByMeshOp::onSelectionDone()
407 {
408   myDlg->setButtonEnabled( false, QtxDialog::OK );
409   setElementID("");
410
411   try {
412     SALOME_ListIO aList;
413     selectionMgr()->selectedObjects(aList, SVTK_Viewer::Type());
414     if (!myIsMultipleAllowed && aList.Extent() != 1)
415       return;
416
417     SMESH::SMESH_Mesh_var aMesh = SMESH::GetMeshByIO(aList.First());
418     if (aMesh->_is_nil() || myMesh->_is_nil() || aMesh->GetId() != myMesh->GetId() )
419       return;
420
421     QString aString;
422     int nbElems = SMESH::GetNameOfSelectedElements(selector(),//myViewWindow->GetSelector(),
423                                                    aList.First(), aString);
424     if (nbElems > 0) {
425       if (!myIsMultipleAllowed && nbElems != 1 )
426         return;
427       setElementID( aString );
428       myDlg->setButtonEnabled( true, QtxDialog::OK );
429     }
430   } catch (...) {
431   }
432 }
433
434 //=======================================================================
435 // function : activateSelection()
436 // purpose  : Activate selection in accordance with current pattern type
437 //=======================================================================
438 void SMESHGUI_ShapeByMeshOp::activateSelection()
439 {
440   selectionMgr()->clearFilters();
441   //SMESH::SetPointRepresentation(false);
442
443   myDlg->myGeomName->setText("");
444
445   QString geomName;
446   Selection_Mode mode = EdgeSelection;
447   switch ( myDlg->myElemTypeGroup->checkedId() ) {
448   case EDGE  :
449     mode = EdgeSelection;   geomName = tr("GEOM_EDGE"); break;
450   case FACE  :
451     mode = FaceSelection;   geomName = tr("GEOM_FACE"); break;
452   case VOLUME:
453     mode = VolumeSelection; geomName = tr(myHasSolids ? "GEOM_SOLID" : "GEOM_SHELL"); break;
454   default: return;
455   }
456   if ( selectionMode() != mode )
457     setSelectionMode( mode );
458
459   myDlg->myGeomName->setText( GEOMBase::GetDefaultName( geomName ));
460 }
461
462 //=======================================================================
463 //function : onTypeChanged
464 //purpose  : SLOT. Called when element type changed.
465 //=======================================================================
466
467 void SMESHGUI_ShapeByMeshOp::onTypeChanged (int theType)
468 {
469   setElementID("");
470   activateSelection();
471 }
472
473 //=======================================================================
474 //function : onTypeChanged
475 //purpose  : SLOT. Called when element id is entered
476 //           Highlight the element whose Ids the user entered manually
477 //=======================================================================
478
479 void SMESHGUI_ShapeByMeshOp::onElemIdChanged(const QString& theNewText)
480 {
481   myDlg->setButtonEnabled( false, QtxDialog::OK );
482
483   if ( myIsManualIdEnter && !myMesh->_is_nil() )
484   {
485     if ( SMESH_Actor* actor = SMESH::FindActorByObject(myMesh) )
486     {
487       if ( SMDS_Mesh* aMesh = actor->GetObject()->GetMesh() )
488       {
489         SMDSAbs_ElementType type = SMDSAbs_Edge;
490         switch ( myDlg->myElemTypeGroup->checkedId() ) {
491         case EDGE  : type = SMDSAbs_Edge;   break;
492         case FACE  : type = SMDSAbs_Face;   break;
493         case VOLUME: type = SMDSAbs_Volume; break;
494         default: return;
495         }
496         TColStd_MapOfInteger newIndices;
497         QStringList aListId = theNewText.split( " ", QString::SkipEmptyParts);
498         for ( int i = 0; i < aListId.count(); i++ ) {
499           if ( const SMDS_MeshElement * e = aMesh->FindElement( aListId[ i ].toInt() ))
500             if ( e->GetType() == type )
501               newIndices.Add( e->GetID() );
502         }
503         
504         if ( !newIndices.IsEmpty() )
505         {
506           if (!myIsMultipleAllowed && newIndices.Extent() != 1)
507             return;
508           if ( SVTK_Selector* s = selector() ) {
509             s->AddOrRemoveIndex( actor->getIO(), newIndices, false );
510             viewWindow()->highlight( actor->getIO(), true, true );
511             myDlg->setButtonEnabled( true, QtxDialog::OK );
512           }
513         }
514       }
515     }
516   }
517 }
518
519 //=======================================================================
520 //function : setElementID
521 //purpose  : programmatically set element id
522 //=======================================================================
523
524 void SMESHGUI_ShapeByMeshOp::setElementID(const QString& theText)
525 {
526   myIsManualIdEnter = false;
527   myDlg->myElementId->setText(theText);
528   myIsManualIdEnter = true;
529 }