src/SMESHGUI/SMESHGUI_CreatePatternDlg.cxx

   1 //  SMESH SMESHGUI : GUI for SMESH component\r
   2 //\r
   3 //  Copyright (C) 2003  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,\r
   4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS\r
   5 //\r
   6 //  This library is free software; you can redistribute it and/or\r
   7 //  modify it under the terms of the GNU Lesser General Public\r
   8 //  License as published by the Free Software Foundation; either\r
   9 //  version 2.1 of the License.\r
  10 //\r
  11 //  This library is distributed in the hope that it will be useful,\r
  12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of\r
  13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\r
  14 //  Lesser General Public License for more details.\r
  15 //\r
  16 //  You should have received a copy of the GNU Lesser General Public\r
  17 //  License along with this library; if not, write to the Free Software\r
  18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA\r
  19 //\r
  20 //  See http://www.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org\r
  21 //\r
  22 //\r
  23 //\r
  24 //  File   : SMESHGUI_CreatePatternDlg.cxx\r
  25 //  Author : Sergey LITONIN\r
  26 //  Module : SMESH\r
  27 \r
  28 #include "SMESHGUI_CreatePatternDlg.h"\r
  29 \r
  30 #include "SMESHGUI.h"\r
  31 #include "SMESHGUI_SpinBox.h"\r
  32 #include "SMESHGUI_PatternWidget.h"\r
  33 #include "SMESHGUI_Utils.h"\r
  34 #include "SMESHGUI_VTKUtils.h"\r
  35 #include "SMESHGUI_PatternUtils.h"\r
  36 #include "SMESHGUI_GEOMGenUtils.h"\r
  37 \r
  38 #include "SMESH_NumberFilter.hxx"\r
  39 \r
  40 #include "SUIT_ResourceMgr.h"\r
  41 #include "SUIT_Desktop.h"\r
  42 #include "SUIT_FileDlg.h"\r
  43 \r
  44 #include "SalomeApp_Study.h"\r
  45 #include "LightApp_DataOwner.h"\r
  46 #include "LightApp_SelectionMgr.h"\r
  47 #include "SalomeApp_Tools.h"\r
  48 \r
  49 #include "SALOMEDS_SObject.hxx"\r
  50 \r
  51 #include "SALOME_ListIO.hxx"\r
  52 #include "SVTK_ViewModel.h"\r
  53 #include "SVTK_ViewWindow.h"\r
  54 #include "SVTK_Selector.h"\r
  55 #include "SVTK_Selection.h"\r
  56 \r
  57 // OCCT Includes\r
  58 #include <TColStd_MapOfInteger.hxx>\r
  59 \r
  60 // QT Includes\r
  61 #include <qframe.h>\r
  62 #include <qlayout.h>\r
  63 #include <qlineedit.h>\r
  64 #include <qpushbutton.h>\r
  65 #include <qgroupbox.h>\r
  66 #include <qlabel.h>\r
  67 #include <qradiobutton.h>\r
  68 #include <qcheckbox.h>\r
  69 #include <qbuttongroup.h>\r
  70 #include <qmessagebox.h>\r
  71 #include <qapplication.h>\r
  72 \r
  73 #define SPACING 5\r
  74 #define MARGIN  10\r
  75 \r
  76 /*!\r
  77  *  Class       : SMESHGUI_CreatePatternDlg\r
  78  *  Description : Dialog to specify filters for VTK viewer\r
  79  */\r
  80 \r
  81 //=======================================================================\r
  82 // function : SMESHGUI_CreatePatternDlg()\r
  83 // purpose  : Constructor\r
  84 //=======================================================================\r
  85 SMESHGUI_CreatePatternDlg::SMESHGUI_CreatePatternDlg( SMESHGUI*   theModule,\r
  86                                                       const int   theType,\r
  87                                                       const char* theName)\r
  88      : QDialog( SMESH::GetDesktop( theModule ), theName, false,\r
  89                 WStyle_Customize | WStyle_NormalBorder | WStyle_Title | WStyle_SysMenu),\r
  90      mySMESHGUI( theModule ),\r
  91      mySelectionMgr( SMESH::GetSelectionMgr( theModule ) )\r
  92 {\r
  93   setCaption(tr("CAPTION"));\r
  94 \r
  95   QVBoxLayout* aDlgLay = new QVBoxLayout (this, MARGIN, SPACING);\r
  96 \r
  97   QFrame* aMainFrame = createMainFrame  (this);\r
  98   QFrame* aBtnFrame  = createButtonFrame(this);\r
  99 \r
 100   aDlgLay->addWidget(aMainFrame);\r
 101   aDlgLay->addWidget(aBtnFrame);\r
 102 \r
 103   aDlgLay->setStretchFactor(aMainFrame, 1);\r
 104 \r
 105   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))\r
 106     mySelector = aViewWindow->GetSelector();\r
 107 \r
 108   Init(theType);\r
 109 }\r
 110 \r
 111 //=======================================================================\r
 112 // function : createMainFrame()\r
 113 // purpose  : Create frame containing dialog's input fields\r
 114 //=======================================================================\r
 115 QFrame* SMESHGUI_CreatePatternDlg::createMainFrame (QWidget* theParent)\r
 116 {\r
 117   QPixmap iconSlct     (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_SELECT")));\r
 118   QPixmap icon2d       (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_PATTERN_2d")));\r
 119   QPixmap icon3d       (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_PATTERN_3d")));\r
 120   QPixmap iconSample2d (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_PATTERN_SAMPLE_2D")));\r
 121 \r
 122   QGroupBox* aMainGrp = new QGroupBox(1, Qt::Horizontal, theParent);\r
 123   aMainGrp->setFrameStyle(QFrame::NoFrame);\r
 124   aMainGrp->setInsideMargin(0);\r
 125 \r
 126   // Pattern type group\r
 127 \r
 128   myTypeGrp = new QButtonGroup (1, Qt::Vertical, tr("PATTERN_TYPE"), aMainGrp);\r
 129   mySwitch2d = new QRadioButton (myTypeGrp);\r
 130   mySwitch3d = new QRadioButton (myTypeGrp);\r
 131   mySwitch2d->setPixmap(icon2d);\r
 132   mySwitch3d->setPixmap(icon3d);\r
 133   myTypeGrp->insert(mySwitch2d, Type_2d);\r
 134   myTypeGrp->insert(mySwitch3d, Type_3d);\r
 135 \r
 136   QGroupBox* aPatternGrp = new QGroupBox(1, Qt::Horizontal, tr("PATTERN"), aMainGrp);\r
 137 \r
 138   // Mesh and pattern name group\r
 139 \r
 140   QGroupBox* aMeshGrp = new QGroupBox(1, Qt::Vertical, aPatternGrp);\r
 141   aMeshGrp->setFrameStyle(QFrame::NoFrame);\r
 142   aMeshGrp->setInsideMargin(0);\r
 143 \r
 144   new QLabel(tr("MESH_OR_SUBMESH"), aMeshGrp);\r
 145 \r
 146   QPushButton* aSelBtn = new QPushButton(aMeshGrp);\r
 147   aSelBtn->setPixmap(iconSlct);\r
 148   myMeshEdit = new QLineEdit(aMeshGrp);\r
 149   myMeshEdit->setReadOnly(true);\r
 150 \r
 151   QGroupBox* aPattGrp = new QGroupBox(1, Qt::Vertical, aPatternGrp);\r
 152   aPattGrp->setFrameStyle(QFrame::NoFrame);\r
 153   aPattGrp->setInsideMargin(0);\r
 154 \r
 155   new QLabel(tr("PATTERN_NAME"), aPattGrp);\r
 156   myName = new QLineEdit(aPattGrp);\r
 157 \r
 158   // Picture 2d\r
 159 \r
 160   myPicture2d = new SMESHGUI_PatternWidget(aPatternGrp),\r
 161   myPicture2d->setSizePolicy(QSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding));\r
 162 \r
 163   // Project check box\r
 164 \r
 165   myProjectChk = new QCheckBox(tr("PROJECT"), aPatternGrp);\r
 166 \r
 167   // Connect signals and slots\r
 168 \r
 169   connect(myTypeGrp,    SIGNAL(clicked(int) ), SLOT(onTypeChanged(int)));\r
 170   connect(myProjectChk, SIGNAL(toggled(bool)), SLOT(onProject(bool)   ));\r
 171   connect(aSelBtn,      SIGNAL(clicked()      ), SLOT(onSelBtnClicked()));\r
 172 \r
 173   return aMainGrp;\r
 174 }\r
 175 \r
 176 //=======================================================================\r
 177 // function : createButtonFrame()\r
 178 // purpose  : Create frame containing buttons\r
 179 //=======================================================================\r
 180 QFrame* SMESHGUI_CreatePatternDlg::createButtonFrame (QWidget* theParent)\r
 181 {\r
 182   QFrame* aFrame = new QFrame(theParent);\r
 183   aFrame->setFrameStyle(QFrame::Box | QFrame::Sunken);\r
 184 \r
 185   myOkBtn    = new QPushButton(tr("SMESH_BUT_OK"    ), aFrame);\r
 186   mySaveBtn  = new QPushButton(tr("SAVE"            ), aFrame);\r
 187   myCloseBtn = new QPushButton(tr("SMESH_BUT_CANCEL"), aFrame);\r
 188 \r
 189   QSpacerItem* aSpacer = new QSpacerItem(0, 0, QSizePolicy::Expanding, QSizePolicy::Minimum);\r
 190 \r
 191   QHBoxLayout* aLay = new QHBoxLayout(aFrame, MARGIN, SPACING);\r
 192 \r
 193   aLay->addWidget(myOkBtn);\r
 194   aLay->addWidget(mySaveBtn);\r
 195   aLay->addItem(aSpacer);\r
 196   aLay->addWidget(myCloseBtn);\r
 197 \r
 198   connect(myOkBtn,    SIGNAL(clicked()), SLOT(onOk()));\r
 199   connect(myCloseBtn, SIGNAL(clicked()), SLOT(onClose()));\r
 200   connect(mySaveBtn, SIGNAL(clicked()), SLOT(onSave()));\r
 201 \r
 202   return aFrame;\r
 203 }\r
 204 \r
 205 //=======================================================================\r
 206 // function : ~SMESHGUI_CreatePatternDlg()\r
 207 // purpose  : Destructor\r
 208 //=======================================================================\r
 209 SMESHGUI_CreatePatternDlg::~SMESHGUI_CreatePatternDlg()\r
 210 {\r
 211   // no need to delete child widgets, Qt does it all for us\r
 212 }\r
 213 \r
 214 //=======================================================================\r
 215 // function : onProject()\r
 216 // purpose  : SLOT. Called when state of "Project nodes on ther face"\r
 217 //            checkbox is changed\r
 218 //=======================================================================\r
 219 void SMESHGUI_CreatePatternDlg::onProject (bool)\r
 220 {\r
 221   loadFromObject(false);\r
 222   displayPreview();\r
 223 }\r
 224 \r
 225 //=======================================================================\r
 226 // function : Init()\r
 227 // purpose  : Init dialog fields, connect signals and slots, show dialog\r
 228 //=======================================================================\r
 229 void SMESHGUI_CreatePatternDlg::Init( const int theType )\r
 230 {\r
 231   myIsLoaded     = false;\r
 232   myType         = -1;\r
 233   myNbPoints     = -1;\r
 234   mySubMesh      = SMESH::SMESH_subMesh::_nil();\r
 235   myMesh         = SMESH::SMESH_Mesh::_nil();\r
 236   myGeomObj      = GEOM::GEOM_Object::_nil();\r
 237   myPattern      = SMESH::SMESH_Pattern::_nil();\r
 238 \r
 239   erasePreview();\r
 240 \r
 241   mySMESHGUI->SetActiveDialogBox((QDialog*)this);\r
 242 \r
 243   // selection and SMESHGUI\r
 244   connect(mySelectionMgr, SIGNAL(currentSelectionChanged()), SLOT(onSelectionDone()));\r
 245   connect(mySMESHGUI, SIGNAL(SignalDeactivateActiveDialog()), SLOT(onDeactivate()));\r
 246   connect(mySMESHGUI, SIGNAL(SignalCloseAllDialogs()), SLOT(onClose()));\r
 247 \r
 248   mySwitch2d->setEnabled(theType == Type_2d);\r
 249   mySwitch3d->setEnabled(theType == Type_3d);\r
 250 \r
 251   if (theType == Type_2d)\r
 252     myProjectChk->show();\r
 253   else\r
 254     myProjectChk->hide();\r
 255 \r
 256   myTypeGrp->setButton(theType);\r
 257   onTypeChanged(theType);\r
 258 \r
 259   myName->setText(getDefaultName());\r
 260   myMeshEdit->setText("");\r
 261 \r
 262   setSizePolicy(QSizePolicy(QSizePolicy::MinimumExpanding, QSizePolicy::MinimumExpanding));\r
 263   qApp->processEvents();\r
 264   updateGeometry();\r
 265   myPicture2d->updateGeometry();\r
 266   adjustSize();\r
 267   resize(minimumSize());\r
 268 \r
 269   activateSelection();\r
 270   onSelectionDone();\r
 271 \r
 272   this->show();\r
 273 }\r
 274 \r
 275 //=======================================================================\r
 276 // function : SetMesh()\r
 277 // purpose  : Set mesh to dialog\r
 278 //=======================================================================\r
 279 void SMESHGUI_CreatePatternDlg::SetMesh (SMESH::SMESH_Mesh_ptr thePtr)\r
 280 {\r
 281   myMesh = SMESH::SMESH_Mesh::_duplicate(thePtr);\r
 282   mySubMesh = SMESH::SMESH_subMesh::_nil();\r
 283 \r
 284   bool isValidMesh = false;\r
 285   if (!myMesh->_is_nil())\r
 286   {\r
 287     _PTR(SObject) aSobj = SMESH::FindSObject(myMesh.in());\r
 288     //Handle(SALOME_InteractiveObject) anIObj =\r
 289     //  new SALOME_InteractiveObject(aSobj->GetID().c_str(), "SMESH");\r
 290     SUIT_DataOwnerPtr anIObj (new LightApp_DataOwner(aSobj->GetID().c_str()));\r
 291 \r
 292     isValidMesh = mySelectionMgr->isOk(anIObj);\r
 293   }\r
 294 \r
 295   if (isValidMesh) {\r
 296     _PTR(SObject) aSO = SMESH::FindSObject(myMesh.in());\r
 297     myMeshEdit->setText(aSO->GetName().c_str());\r
 298     myGeomObj = SMESH::GetGeom(aSO);\r
 299   } else {\r
 300     myMeshEdit->setText("");\r
 301     myGeomObj = GEOM::GEOM_Object::_nil();\r
 302   }\r
 303 \r
 304   if (myType == Type_2d) {\r
 305     loadFromObject(false);\r
 306     displayPreview();\r
 307   }\r
 308 }\r
 309 \r
 310 //=======================================================================\r
 311 // function : isValid()\r
 312 // purpose  : Verify validity of entry data\r
 313 //=======================================================================\r
 314 bool SMESHGUI_CreatePatternDlg::isValid()\r
 315 {\r
 316   if (myGeomObj->_is_nil()) {\r
 317     QMessageBox::information(SMESH::GetDesktop( mySMESHGUI ),\r
 318                              tr("SMESH_INSUFFICIENT_DATA"),\r
 319                              tr("SMESHGUI_INVALID_PARAMETERS"),\r
 320                              QMessageBox::Ok);\r
 321     return false;\r
 322   }\r
 323   else\r
 324     return true;\r
 325 }\r
 326 \r
 327 //=======================================================================\r
 328 // function : getDefaultName()\r
 329 // purpose  : Get default pattern name\r
 330 //=======================================================================\r
 331 QString SMESHGUI_CreatePatternDlg::getDefaultName() const\r
 332 {\r
 333   return myType == Type_2d ? tr("DEFAULT_2D") : tr("DEFAULT_3D");\r
 334 }\r
 335 \r
 336 //=======================================================================\r
 337 // function : onSave()\r
 338 // purpose  : SLOT called when "Save" button pressed. Build pattern and\r
 339 //           save it to disk\r
 340 //=======================================================================\r
 341 void SMESHGUI_CreatePatternDlg::onSave()\r
 342 {\r
 343   try {\r
 344     if (!isValid())\r
 345       return;\r
 346 \r
 347     if (!myIsLoaded)\r
 348       loadFromObject(true);\r
 349 \r
 350     // Load pattern from object\r
 351     if (!myIsLoaded)\r
 352       return;\r
 353 \r
 354     ///////////////////////////////////////////////////////\r
 355     SUIT_FileDlg* aDlg = new SUIT_FileDlg (this, false);\r
 356     aDlg->setCaption(tr("SAVE_PATTERN"));\r
 357     aDlg->setMode(QFileDialog::AnyFile);\r
 358     aDlg->setFilters(tr("PATTERN_FILT"));\r
 359     if (myName->text() != "")\r
 360       aDlg->setSelection(myName->text());\r
 361 \r
 362     if (aDlg->exec() != Accepted)\r
 363       return;\r
 364 \r
 365     QString fName = aDlg->selectedFile();\r
 366     if (fName.isEmpty())\r
 367       return;\r
 368 \r
 369     if (QFileInfo(fName).extension().isEmpty())\r
 370       fName = autoExtension(fName);\r
 371 \r
 372     fName = QDir::convertSeparators(fName);\r
 373 \r
 374     QString aData (myPattern->GetString());\r
 375     long aLen = aData.length();\r
 376 \r
 377     QFile aFile (fName);\r
 378     aFile.open(IO_WriteOnly);\r
 379     long aWritten = aFile.writeBlock(aData, aLen);\r
 380     aFile.close();\r
 381 \r
 382     if (aWritten != aLen) {\r
 383       QMessageBox::information(SMESH::GetDesktop( mySMESHGUI ), tr("SMESH_ERROR"),\r
 384                                tr("ERROR_OF_SAVING"), QMessageBox::Ok);\r
 385     } else {\r
 386       //SUIT_Application::getDesktop()->setSelectionModes(ActorSelection);\r
 387       if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))\r
 388         aViewWindow->SetSelectionMode(ActorSelection);\r
 389       disconnect(mySelectionMgr, 0, this, 0);\r
 390       disconnect(mySMESHGUI, 0, this, 0);\r
 391       mySMESHGUI->ResetState();\r
 392       accept();\r
 393       emit NewPattern();\r
 394     }\r
 395   } catch (const SALOME::SALOME_Exception& S_ex) {\r
 396     SalomeApp_Tools::QtCatchCorbaException(S_ex);\r
 397   } catch (...) {\r
 398   }\r
 399 }\r
 400 \r
 401 //=======================================================================\r
 402 // function : GetPatternName()\r
 403 // purpose  : Get name of pattern\r
 404 //=======================================================================\r
 405 QString SMESHGUI_CreatePatternDlg::GetPatternName() const\r
 406 {\r
 407   return myName->text();\r
 408 }\r
 409 \r
 410 //=======================================================================\r
 411 // function : GetPattern()\r
 412 // purpose  : Get result pattern\r
 413 //=======================================================================\r
 414 SMESH::SMESH_Pattern_ptr SMESHGUI_CreatePatternDlg::GetPattern()\r
 415 {\r
 416   return myPattern.in();\r
 417 }\r
 418 \r
 419 //=======================================================================\r
 420 // function : onOk()\r
 421 // purpose  : SLOT called when "Ok" button pressed.\r
 422 //=======================================================================\r
 423 void SMESHGUI_CreatePatternDlg::onOk()\r
 424 {\r
 425   try {\r
 426     if (!isValid())\r
 427       return;\r
 428 \r
 429     if (!myIsLoaded)\r
 430       loadFromObject(true);\r
 431 \r
 432     // Load pattern from object\r
 433     if (!myIsLoaded) {\r
 434       return;\r
 435     } else {\r
 436       //SUIT_Application::getDesktop()->setSelectionModes(ActorSelection);\r
 437       if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))\r
 438         aViewWindow->SetSelectionMode(ActorSelection);\r
 439       disconnect(mySelectionMgr, 0, this, 0);\r
 440       disconnect(mySMESHGUI, 0, this, 0);\r
 441       mySMESHGUI->ResetState();\r
 442       accept();\r
 443       emit NewPattern();\r
 444     }\r
 445   } catch (const SALOME::SALOME_Exception& S_ex) {\r
 446     SalomeApp_Tools::QtCatchCorbaException(S_ex);\r
 447   } catch (...) {\r
 448   }\r
 449 }\r
 450 \r
 451 //=======================================================================\r
 452 // function : onClose()\r
 453 // purpose  : SLOT called when "Close" button pressed. Close dialog\r
 454 //=======================================================================\r
 455 void SMESHGUI_CreatePatternDlg::onClose()\r
 456 {\r
 457   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))\r
 458     aViewWindow->SetSelectionMode(ActorSelection);\r
 459   disconnect(mySelectionMgr, 0, this, 0);\r
 460   disconnect(mySMESHGUI, 0, this, 0);\r
 461   mySMESHGUI->ResetState();\r
 462   reject();\r
 463   emit Close();\r
 464 }\r
 465 \r
 466 //=======================================================================\r
 467 // function : loadFromObject()\r
 468 // purpose  : Load pattern from geom object corresponding to the mesh/submesh\r
 469 //=======================================================================\r
 470 bool SMESHGUI_CreatePatternDlg::loadFromObject (const bool theMess)\r
 471 {\r
 472   try {\r
 473     myIsLoaded = false;\r
 474 \r
 475     if (myPattern->_is_nil())\r
 476       myPattern = SMESH::GetPattern();\r
 477 \r
 478     if (myMesh->_is_nil() && mySubMesh->_is_nil() || myGeomObj->_is_nil())\r
 479       return false;\r
 480 \r
 481     SMESH::SMESH_Mesh_ptr aMesh = mySubMesh->_is_nil() ? myMesh.in() : mySubMesh->GetFather();\r
 482 \r
 483     myIsLoaded = myType == Type_2d\r
 484       ? myPattern->LoadFromFace(aMesh, myGeomObj, myProjectChk->isChecked())\r
 485       : myPattern->LoadFrom3DBlock(aMesh, myGeomObj);\r
 486 \r
 487     if (!myIsLoaded && theMess) {\r
 488       QString aMess;\r
 489       SMESH::SMESH_Pattern::ErrorCode aCode = myPattern->GetErrorCode();\r
 490 \r
 491       if      (aCode == SMESH::SMESH_Pattern::ERR_LOAD_EMPTY_SUBMESH  ) aMess = tr("ERR_LOAD_EMPTY_SUBMESH");\r
 492       else if (aCode == SMESH::SMESH_Pattern::ERR_LOADF_NARROW_FACE   ) aMess = tr("ERR_LOADF_NARROW_FACE");\r
 493       else if (aCode == SMESH::SMESH_Pattern::ERR_LOADF_CLOSED_FACE   ) aMess = tr("ERR_LOADF_CLOSED_FACE");\r
 494       else if (aCode == SMESH::SMESH_Pattern::ERR_LOADV_BAD_SHAPE     ) aMess = tr("ERR_LOADV_BAD_SHAPE");\r
 495       else if (aCode == SMESH::SMESH_Pattern::ERR_LOADV_COMPUTE_PARAMS) aMess = tr("ERR_LOADV_COMPUTE_PARAMS");\r
 496       else                                                              aMess = tr("ERROR_OF_CREATION");\r
 497 \r
 498       QMessageBox::information(SMESH::GetDesktop( mySMESHGUI ),\r
 499                                tr("SMESH_ERROR"), aMess, QMessageBox::Ok);\r
 500     }\r
 501   } catch (const SALOME::SALOME_Exception& S_ex) {\r
 502     SalomeApp_Tools::QtCatchCorbaException(S_ex);\r
 503   }\r
 504 \r
 505   return myIsLoaded;\r
 506 }\r
 507 \r
 508 //=======================================================================\r
 509 // function : onSelectionDone()\r
 510 // purpose  : SLOT called when selection changed\r
 511 //=======================================================================\r
 512 void SMESHGUI_CreatePatternDlg::onSelectionDone()\r
 513 {\r
 514   try {\r
 515     SALOME_ListIO aList;\r
 516     mySelectionMgr->selectedObjects(aList, SVTK_Viewer::Type());\r
 517     if (aList.Extent() != 1)\r
 518       return;\r
 519 \r
 520     // Get mesh or sub-mesh from selection\r
 521     Handle(SALOME_InteractiveObject) anIO = aList.First();\r
 522     SMESH::SMESH_Mesh_var aMesh = SMESH::IObjectToInterface<SMESH::SMESH_Mesh>(anIO);\r
 523     SMESH::SMESH_subMesh_var aSubMesh = SMESH::IObjectToInterface<SMESH::SMESH_subMesh>(anIO);\r
 524     if (aMesh->_is_nil() && aSubMesh->_is_nil())\r
 525       return;\r
 526 \r
 527     // Get geom object corresponding to the mesh\r
 528     _PTR(SObject) aSO;\r
 529     if (!aMesh->_is_nil())\r
 530       aSO = SMESH::FindSObject(aMesh.in());\r
 531     else\r
 532       aSO = SMESH::FindSObject(aSubMesh.in());\r
 533 \r
 534     GEOM::GEOM_Object_var aGeomObj = SMESH::GetGeom(aSO);\r
 535     if (aGeomObj->_is_nil())\r
 536       return;\r
 537 \r
 538     myGeomObj = aGeomObj;\r
 539 \r
 540     // init class fields\r
 541     if (!aMesh->_is_nil()) {\r
 542       myMesh = aMesh;\r
 543       mySubMesh = SMESH::SMESH_subMesh::_nil();\r
 544     } else {\r
 545       mySubMesh = aSubMesh;\r
 546       myMesh = SMESH::SMESH_Mesh::_nil();\r
 547     }\r
 548 \r
 549     QString aName;\r
 550     SMESH::GetNameOfSelectedIObjects(mySelectionMgr, aName);\r
 551     myMeshEdit->setText(aName);\r
 552 \r
 553     if (myType == Type_2d) {\r
 554       loadFromObject(true);\r
 555       displayPreview();\r
 556     }\r
 557   } catch (...) {\r
 558     myMesh = SMESH::SMESH_Mesh::_nil();\r
 559     mySubMesh = SMESH::SMESH_subMesh::_nil();\r
 560     myGeomObj = GEOM::GEOM_Object::_nil();\r
 561     erasePreview();\r
 562   }\r
 563 }\r
 564 \r
 565 //=======================================================================\r
 566 // function : onDeactivate()\r
 567 // purpose  : SLOT called when dialog must be deativated\r
 568 //=======================================================================\r
 569 void SMESHGUI_CreatePatternDlg::onDeactivate()\r
 570 {\r
 571   disconnect(mySelectionMgr, 0, this, 0);\r
 572   setEnabled(false);\r
 573 }\r
 574 \r
 575 //=======================================================================\r
 576 // function : enterEvent()\r
 577 // purpose  : Event filter\r
 578 //=======================================================================\r
 579 void SMESHGUI_CreatePatternDlg::enterEvent (QEvent*)\r
 580 {\r
 581   // there is a stange problem that enterEvent() comes after onSave()\r
 582   if ( isVisible () ) {\r
 583     mySMESHGUI->EmitSignalDeactivateDialog();\r
 584     setEnabled(true);\r
 585     activateSelection();\r
 586     connect(mySelectionMgr, SIGNAL(currentSelectionChanged()), SLOT(onSelectionDone()));\r
 587   }\r
 588 }\r
 589 \r
 590 //=================================================================================\r
 591 // function : closeEvent()\r
 592 // purpose  : Close dialog box\r
 593 //=================================================================================\r
 594 void SMESHGUI_CreatePatternDlg::closeEvent (QCloseEvent*)\r
 595 {\r
 596   onClose();\r
 597 }\r
 598 \r
 599 //=======================================================================\r
 600 // function : onSelBtnClicked()\r
 601 // purpose  : SLOT. Called when -> button clicked.\r
 602 //=======================================================================\r
 603 void SMESHGUI_CreatePatternDlg::onSelBtnClicked()\r
 604 {\r
 605   onSelectionDone();\r
 606 }\r
 607 \r
 608 //================================================================\r
 609 // function : autoExtension()\r
 610 // purpose  : Append extension to the file name\r
 611 //================================================================\r
 612 QString SMESHGUI_CreatePatternDlg::autoExtension (const QString& theFileName) const\r
 613 {\r
 614   QString anExt = theFileName.section('.', -1);\r
 615   return anExt != "smp" && anExt != "SMP" ? theFileName + ".smp" : theFileName;\r
 616 }\r
 617 \r
 618 //=======================================================================\r
 619 // function : displayPreview()\r
 620 // purpose  : Display preview\r
 621 //=======================================================================\r
 622 void SMESHGUI_CreatePatternDlg::displayPreview()\r
 623 {\r
 624   // Redisplay preview in dialog\r
 625   try {\r
 626     if (!myIsLoaded) {\r
 627       erasePreview();\r
 628     } else {\r
 629       SMESH::point_array_var pnts = myPattern->GetPoints();\r
 630       SMESH::long_array_var keyPoints = myPattern->GetKeyPoints();\r
 631       SMESH::array_of_long_array_var elemPoints = myPattern->GetElementPoints(false);\r
 632 \r
 633       if (pnts->length()       == 0 ||\r
 634           keyPoints->length()  == 0 ||\r
 635           elemPoints->length() == 0) {\r
 636         myIsLoaded = false;\r
 637         erasePreview();\r
 638         return;\r
 639       }\r
 640 \r
 641       PointVector aPoints (pnts->length());\r
 642       QValueVector<int> aKeyPoints (keyPoints->length());\r
 643       ConnectivityVector anElemPoints (elemPoints->length());\r
 644 \r
 645       for (int i = 0, n = pnts->length(); i < n; i++)\r
 646         aPoints[ i ] = pnts[ i ];\r
 647 \r
 648       for (int i2 = 0, n2 = keyPoints->length(); i2 < n2; i2++)\r
 649         aKeyPoints[ i2 ] = keyPoints[ i2 ];\r
 650 \r
 651       for (int i3 = 0, n3 = elemPoints->length(); i3 < n3; i3++) {\r
 652         QValueVector<int> aVec (elemPoints[ i3 ].length());\r
 653         for (int i4 = 0, n4 = elemPoints[ i3 ].length(); i4 < n4; i4++)\r
 654           aVec[ i4 ] = elemPoints[ i3 ][ i4 ];\r
 655 \r
 656         anElemPoints[ i3 ] = aVec;\r
 657       }\r
 658 \r
 659       myPicture2d->SetPoints(aPoints, aKeyPoints, anElemPoints);\r
 660     }\r
 661 \r
 662     return;\r
 663 \r
 664   } catch (const SALOME::SALOME_Exception& S_ex) {\r
 665     SalomeApp_Tools::QtCatchCorbaException(S_ex);\r
 666   } catch (...) {\r
 667   }\r
 668   erasePreview();\r
 669 }\r
 670 \r
 671 //=======================================================================\r
 672 // function : erasePreview()\r
 673 // purpose  : Erase preview\r
 674 //=======================================================================\r
 675 void SMESHGUI_CreatePatternDlg::erasePreview()\r
 676 {\r
 677   // Erase preview in 2D viewer\r
 678   myPicture2d->SetPoints(PointVector(), QValueVector<int>(), ConnectivityVector());\r
 679 }\r
 680 \r
 681 //=======================================================================\r
 682 // function : activateSelection()\r
 683 // purpose  : Activate selection in accordance with current pattern type\r
 684 //=======================================================================\r
 685 void SMESHGUI_CreatePatternDlg::activateSelection()\r
 686 {\r
 687   mySelectionMgr->clearFilters();\r
 688   //SUIT_Application::getDesktop()->setSelectionModes(ActorSelection);\r
 689   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))\r
 690     aViewWindow->SetSelectionMode(ActorSelection);\r
 691 \r
 692   if (myType == Type_2d) {\r
 693     mySelectionMgr->installFilter(new SMESH_NumberFilter\r
 694       ("SMESH", TopAbs_SHAPE, -1, TopAbs_FACE));\r
 695   } else {\r
 696     TColStd_MapOfInteger aTypes;\r
 697     aTypes.Add(TopAbs_SHELL);\r
 698     aTypes.Add(TopAbs_SOLID);\r
 699     mySelectionMgr->installFilter(new SMESH_NumberFilter\r
 700       ("SMESH", TopAbs_FACE, 6, aTypes, GEOM::GEOM_Object::_nil(), true));\r
 701   }\r
 702 }\r
 703 \r
 704 //=======================================================================\r
 705 // function : onTypeChanged()\r
 706 // purpose  : SLOT. Called when pattern type changed.\r
 707 //            Change dialog's look and feel\r
 708 //=======================================================================\r
 709 void SMESHGUI_CreatePatternDlg::onTypeChanged (int theType)\r
 710 {\r
 711   if (myType == theType)\r
 712     return;\r
 713 \r
 714   myType = theType;\r
 715 \r
 716   if (theType == Type_2d)\r
 717     myPicture2d->show();\r
 718   else\r
 719     myPicture2d->hide();\r
 720 }\r