src/SMESHGUI/SMESHGUI_ShapeByMeshDlg.cxx

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