src/SMESHGUI/SMESHGUI_ShapeByMeshDlg.cxx

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