src/SMESHGUI/SMESHGUI_MultiEditDlg.cxx

   1 //  Copyright (C) 2007-2008  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 // File   : SMESHGUI_MultiEditDlg.cxx
  23 // Author : Sergey LITONIN, Open CASCADE S.A.S.
  24 // SMESH includes
  25 //
  26 #include "SMESHGUI_MultiEditDlg.h"
  27
  28 #include "SMESHGUI.h"
  29 #include "SMESHGUI_Filter.h"
  30 #include "SMESHGUI_FilterDlg.h"
  31 #include "SMESHGUI_Utils.h"
  32 #include "SMESHGUI_VTKUtils.h"
  33 #include "SMESHGUI_MeshUtils.h"
  34 #include "SMESHGUI_FilterUtils.h"
  35 #include "SMESHGUI_SpinBox.h"
  36
  37 #include <SMESH_Actor.h>
  38 #include <SMESH_TypeFilter.hxx>
  39 #include <SMDS_Mesh.hxx>
  40
  41 // SALOME GUI includes
  42 #include <SUIT_ResourceMgr.h>
  43 #include <SUIT_Desktop.h>
  44 #include <SUIT_Session.h>
  45 #include <SUIT_MessageBox.h>
  46
  47 #include <LightApp_SelectionMgr.h>
  48 #include <LightApp_Application.h>
  49 #include <SALOME_ListIO.hxx>
  50 #include <SALOME_ListIteratorOfListIO.hxx>
  51
  52 #include <SVTK_Selector.h>
  53 #include <SVTK_ViewWindow.h>
  54 #include <VTKViewer_CellLocationsArray.h>
  55
  56 // OCCT includes
  57 #include <TColStd_IndexedMapOfInteger.hxx>
  58 #include <TColStd_DataMapOfIntegerInteger.hxx>
  59 #include <TColStd_MapIteratorOfMapOfInteger.hxx>
  60
  61 // VTK includes
  62 #include <vtkIdList.h>
  63 #include <vtkCellArray.h>
  64 #include <vtkUnsignedCharArray.h>
  65 #include <vtkUnstructuredGrid.h>
  66 #include <vtkDataSetMapper.h>
  67 #include <vtkProperty.h>
  68
  69 // Qt includes
  70 #include <QLabel>
  71 #include <QVBoxLayout>
  72 #include <QHBoxLayout>
  73 #include <QGridLayout>
  74 #include <QListWidget>
  75 #include <QCheckBox>
  76 #include <QComboBox>
  77 #include <QGroupBox>
  78 #include <QLineEdit>
  79 #include <QPushButton>
  80 #include <QButtonGroup>
  81 #include <QRadioButton>
  82 #include <QKeyEvent>
  83
  84 // IDL includes
  85 #include <SALOMEconfig.h>
  86 #include CORBA_SERVER_HEADER(SMESH_Group)
  87
  88 #define SPACING 6
  89 #define MARGIN  11
  90
  91 /*!
  92  *  Class       : SMESHGUI_MultiEditDlg
  93  *  Description : Description : Inversion of the diagonal of a pseudo-quadrangle formed by
  94  *                2 neighboring triangles with 1 common edge
  95  */
  96
  97 //=======================================================================
  98 // name    : SMESHGUI_MultiEditDlg::SMESHGUI_MultiEditDlg
  99 // Purpose : Constructor
 100 //=======================================================================
 101 SMESHGUI_MultiEditDlg
 102 ::SMESHGUI_MultiEditDlg(SMESHGUI* theModule,
 103                         const int theMode,
 104                         const bool the3d2d):
 105   QDialog(SMESH::GetDesktop(theModule)),
 106   mySelector(SMESH::GetViewWindow(theModule)->GetSelector()),
 107   mySelectionMgr(SMESH::GetSelectionMgr(theModule)),
 108   mySMESHGUI(theModule)
 109 {
 110   setModal(false);
 111   setAttribute(Qt::WA_DeleteOnClose, true);
 112
 113   myFilterDlg = 0;
 114   myEntityType = 0;
 115
 116   myFilterType = theMode;
 117   QVBoxLayout* aDlgLay = new QVBoxLayout(this);
 118   aDlgLay->setMargin(MARGIN);
 119   aDlgLay->setSpacing(SPACING);
 120
 121   QWidget* aMainFrame = createMainFrame  (this, the3d2d);
 122   QWidget* aBtnFrame  = createButtonFrame(this);
 123
 124   aDlgLay->addWidget(aMainFrame);
 125   aDlgLay->addWidget(aBtnFrame);
 126
 127   Init();
 128 }
 129
 130 //=======================================================================
 131 // name    : SMESHGUI_MultiEditDlg::createMainFrame
 132 // Purpose : Create frame containing dialog's input fields
 133 //=======================================================================
 134 QWidget* SMESHGUI_MultiEditDlg::createMainFrame (QWidget* theParent, const bool the3d2d)
 135 {
 136   QWidget* aMainGrp = new QWidget(theParent);
 137   QVBoxLayout* aMainGrpLayout = new QVBoxLayout(aMainGrp);
 138   aMainGrpLayout->setMargin(0);
 139   aMainGrpLayout->setSpacing(SPACING);
 140
 141   QPixmap aPix (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_SELECT")));
 142
 143   // "Selected cells" group
 144   mySelGrp = new QGroupBox(aMainGrp);
 145
 146   QGridLayout* mySelGrpLayout = new QGridLayout(mySelGrp);
 147   mySelGrpLayout->setMargin(MARGIN);
 148   mySelGrpLayout->setSpacing(SPACING);
 149
 150   myEntityTypeGrp = 0;
 151   if (the3d2d) {
 152     QGroupBox* aEntityTypeGrp = new QGroupBox(tr("SMESH_ELEMENTS_TYPE"), mySelGrp);
 153     myEntityTypeGrp = new QButtonGroup(mySelGrp);
 154     QHBoxLayout* aEntityLayout = new QHBoxLayout(aEntityTypeGrp);
 155     aEntityLayout->setMargin(MARGIN);
 156     aEntityLayout->setSpacing(SPACING);
 157
 158     QRadioButton* aFaceRb = new QRadioButton(tr("SMESH_FACE"), aEntityTypeGrp);
 159     QRadioButton* aVolumeRb = new QRadioButton(tr("SMESH_VOLUME"), aEntityTypeGrp);
 160
 161     aEntityLayout->addWidget(aFaceRb);
 162     aEntityLayout->addWidget(aVolumeRb);
 163
 164     myEntityTypeGrp->addButton(aFaceRb, 0);
 165     myEntityTypeGrp->addButton(aVolumeRb, 1);
 166     aFaceRb->setChecked(true);
 167     myEntityType = myEntityTypeGrp->checkedId();
 168
 169     mySelGrpLayout->addWidget(aEntityTypeGrp, 0, 0, 1, 2);
 170   }
 171
 172   myListBox = new QListWidget(mySelGrp);
 173   myListBox->setSelectionMode(QListWidget::ExtendedSelection);
 174   myListBox->setSizePolicy(QSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding));
 175   myListBox->installEventFilter(this);
 176
 177   myFilterBtn = new QPushButton(tr("FILTER"),    mySelGrp);
 178   myAddBtn    = new QPushButton(tr("ADD"),       mySelGrp);
 179   myRemoveBtn = new QPushButton(tr("REMOVE"),    mySelGrp);
 180   mySortBtn   = new QPushButton(tr("SORT_LIST"), mySelGrp);
 181
 182   int row = mySelGrpLayout->rowCount();
 183   mySelGrpLayout->addWidget(myListBox,   row,   0, 6, 1);
 184   mySelGrpLayout->addWidget(myFilterBtn, row,   1);
 185   mySelGrpLayout->addWidget(myAddBtn,    row+2, 1);
 186   mySelGrpLayout->addWidget(myRemoveBtn, row+3, 1);
 187   mySelGrpLayout->addWidget(mySortBtn,   row+5, 1);
 188   mySelGrpLayout->setRowMinimumHeight(row+1, 10);
 189   mySelGrpLayout->setRowMinimumHeight(row+4, 10);
 190   mySelGrpLayout->setRowStretch(row+1, 5);
 191   mySelGrpLayout->setRowStretch(row+4, 5);
 192
 193   myToAllChk = new QCheckBox(tr("TO_ALL"), mySelGrp);
 194   mySelGrpLayout->addWidget(myToAllChk, mySelGrpLayout->rowCount(), 0,
 195                             1, mySelGrpLayout->columnCount());
 196
 197   // Split/Join criterion group
 198   myCriterionGrp = new QGroupBox(tr("SPLIT_JOIN_CRITERION"), aMainGrp);
 199   QVBoxLayout* aCriterionLayout = new QVBoxLayout(myCriterionGrp);
 200   aCriterionLayout->setMargin(MARGIN);
 201   aCriterionLayout->setSpacing(SPACING);
 202
 203   myChoiceWidget = new QWidget(myCriterionGrp);
 204   myGroupChoice = new QButtonGroup(myChoiceWidget);
 205   QVBoxLayout* aGroupChoiceLayout = new QVBoxLayout(myChoiceWidget);
 206   aGroupChoiceLayout->setMargin(0);
 207   aGroupChoiceLayout->setSpacing(SPACING);
 208
 209   QRadioButton* aDiag13RB  = new QRadioButton(tr("USE_DIAGONAL_1_3"), myChoiceWidget);
 210   QRadioButton* aDiag24RB  = new QRadioButton(tr("USE_DIAGONAL_2_4"), myChoiceWidget);
 211   QRadioButton* aNumFuncRB = new QRadioButton(tr("USE_NUMERIC_FUNC"), myChoiceWidget);
 212
 213   aGroupChoiceLayout->addWidget(aDiag13RB);
 214   aGroupChoiceLayout->addWidget(aDiag24RB);
 215   aGroupChoiceLayout->addWidget(aNumFuncRB);
 216   myGroupChoice->addButton(aDiag13RB,  0);
 217   myGroupChoice->addButton(aDiag24RB,  1);
 218   myGroupChoice->addButton(aNumFuncRB, 2);
 219   aDiag13RB->setChecked(true);
 220
 221   myComboBoxFunctor = new QComboBox(myCriterionGrp);
 222   myComboBoxFunctor->addItem(tr("ASPECTRATIO_ELEMENTS"));
 223   myComboBoxFunctor->addItem(tr("MINIMUMANGLE_ELEMENTS"));
 224   myComboBoxFunctor->addItem(tr("SKEW_ELEMENTS"));
 225   myComboBoxFunctor->addItem(tr("AREA_ELEMENTS"));
 226   //myComboBoxFunctor->addItem(tr("LENGTH2D_EDGES")); // for existing elements only
 227   //myComboBoxFunctor->addItem(tr("MULTI2D_BORDERS")); // for existing elements only
 228   myComboBoxFunctor->setCurrentIndex(0);
 229
 230   aCriterionLayout->addWidget(myChoiceWidget);
 231   aCriterionLayout->addWidget(myComboBoxFunctor);
 232
 233   myCriterionGrp->hide();
 234   myChoiceWidget->hide();
 235   myComboBoxFunctor->setEnabled(false);
 236
 237   // "Select from" group
 238   QGroupBox* aGrp = new QGroupBox(tr("SELECT_FROM"), aMainGrp);
 239   QGridLayout* aGrpLayout = new QGridLayout(aGrp);
 240   aGrpLayout->setMargin(MARGIN);
 241   aGrpLayout->setSpacing(SPACING);
 242
 243   mySubmeshChk = new QCheckBox(tr("SMESH_SUBMESH"), aGrp);
 244   mySubmeshBtn = new QPushButton(aGrp);
 245   mySubmesh = new QLineEdit(aGrp);
 246   mySubmesh->setReadOnly(true);
 247   mySubmeshBtn->setIcon(aPix);
 248
 249   myGroupChk = new QCheckBox(tr("SMESH_GROUP"), aGrp);
 250   myGroupBtn = new QPushButton(aGrp);
 251   myGroup = new QLineEdit(aGrp);
 252   myGroup->setReadOnly(true);
 253   myGroupBtn->setIcon(aPix);
 254
 255   aGrpLayout->addWidget(mySubmeshChk,0,0);
 256   aGrpLayout->addWidget(mySubmeshBtn,0,1);
 257   aGrpLayout->addWidget(mySubmesh,0,2);
 258   aGrpLayout->addWidget(myGroupChk,1,0);
 259   aGrpLayout->addWidget(myGroupBtn,1,1);
 260   aGrpLayout->addWidget(myGroup,1,2);
 261
 262   aMainGrpLayout->addWidget(mySelGrp);
 263   aMainGrpLayout->addWidget(myCriterionGrp);
 264   aMainGrpLayout->addWidget(aGrp);
 265
 266   return aMainGrp;
 267 }
 268
 269 //=======================================================================
 270 // name    : SMESHGUI_MultiEditDlg::createButtonFrame
 271 // Purpose : Create frame containing buttons
 272 //=======================================================================
 273 QWidget* SMESHGUI_MultiEditDlg::createButtonFrame (QWidget* theParent)
 274 {
 275   QGroupBox* aFrame = new QGroupBox(theParent);
 276
 277   myOkBtn     = new QPushButton (tr("SMESH_BUT_APPLY_AND_CLOSE"), aFrame);
 278   myApplyBtn  = new QPushButton (tr("SMESH_BUT_APPLY"), aFrame);
 279   myCloseBtn  = new QPushButton (tr("SMESH_BUT_CLOSE"), aFrame);
 280   myHelpBtn   = new QPushButton (tr("SMESH_BUT_HELP"),  aFrame);
 281
 282   QHBoxLayout* aLay = new QHBoxLayout (aFrame);
 283   aLay->setMargin(MARGIN);
 284   aLay->setSpacing(SPACING);
 285
 286   aLay->addWidget(myOkBtn);
 287   aLay->addSpacing(10);
 288   aLay->addWidget(myApplyBtn);
 289   aLay->addSpacing(10);
 290   aLay->addStretch();
 291   aLay->addWidget(myCloseBtn);
 292   aLay->addWidget(myHelpBtn);
 293
 294   return aFrame;
 295 }
 296
 297 //=======================================================================
 298 // name    : SMESHGUI_MultiEditDlg::isValid
 299 // Purpose : Verify validity of input data
 300 //=======================================================================
 301 bool SMESHGUI_MultiEditDlg::isValid (const bool /*theMess*/)
 302 {
 303   return (!myMesh->_is_nil() &&
 304           (myListBox->count() > 0 || (myToAllChk->isChecked() && myActor)));
 305 }
 306
 307 //=======================================================================
 308 // name    : SMESHGUI_MultiEditDlg::~SMESHGUI_MultiEditDlg
 309 // Purpose : Destructor
 310 //=======================================================================
 311 SMESHGUI_MultiEditDlg::~SMESHGUI_MultiEditDlg()
 312 {
 313   if (myFilterDlg != 0)
 314   {
 315     myFilterDlg->setParent(0);
 316     delete myFilterDlg;
 317   }
 318 }
 319
 320 //=======================================================================
 321 // name    : SMESHGUI_MultiEditDlg::eventFilter
 322 // Purpose : event filter
 323 //=======================================================================
 324 bool SMESHGUI_MultiEditDlg::eventFilter (QObject* object, QEvent* event)
 325 {
 326   if (object == myListBox && event->type() == QEvent::KeyPress) {
 327     QKeyEvent* ke = (QKeyEvent*)event;
 328     if (ke->key() == Qt::Key_Delete)
 329       onRemoveBtn();
 330   }
 331   return QDialog::eventFilter(object, event);
 332 }
 333
 334 //=======================================================================
 335 // name    : SMESHGUI_MultiEditDlg::getNumericalFunctor
 336 // Purpose :
 337 //=======================================================================
 338 SMESH::NumericalFunctor_ptr SMESHGUI_MultiEditDlg::getNumericalFunctor()
 339 {
 340   SMESH::NumericalFunctor_var aNF = SMESH::NumericalFunctor::_nil();
 341
 342   SMESH::FilterManager_var aFilterMgr = SMESH::GetFilterManager();
 343   if (aFilterMgr->_is_nil())
 344     return aNF._retn();
 345
 346   if (myComboBoxFunctor->currentText() == tr("ASPECTRATIO_ELEMENTS"))
 347     aNF = aFilterMgr->CreateAspectRatio();
 348   else if (myComboBoxFunctor->currentText() == tr("WARP_ELEMENTS"))
 349     aNF = aFilterMgr->CreateWarping();
 350   else if (myComboBoxFunctor->currentText() == tr("MINIMUMANGLE_ELEMENTS"))
 351     aNF = aFilterMgr->CreateMinimumAngle();
 352   else if (myComboBoxFunctor->currentText() == tr("TAPER_ELEMENTS"))
 353     aNF = aFilterMgr->CreateTaper();
 354   else if (myComboBoxFunctor->currentText() == tr("SKEW_ELEMENTS"))
 355     aNF = aFilterMgr->CreateSkew();
 356   else if (myComboBoxFunctor->currentText() == tr("AREA_ELEMENTS"))
 357     aNF = aFilterMgr->CreateArea();
 358   else if (myComboBoxFunctor->currentText() == tr("LENGTH2D_EDGES"))
 359     aNF = aFilterMgr->CreateLength2D();
 360   else if (myComboBoxFunctor->currentText() == tr("MULTI2D_BORDERS"))
 361     aNF = aFilterMgr->CreateMultiConnection2D();
 362   else;
 363
 364   return aNF._retn();
 365 }
 366
 367 //=======================================================================
 368 // name    : SMESHGUI_MultiEditDlg::Init
 369 // Purpose : Init dialog fields, connect signals and slots, show dialog
 370 //=======================================================================
 371 void SMESHGUI_MultiEditDlg::Init()
 372 {
 373   mySMESHGUI->SetActiveDialogBox((QDialog*)this);
 374   myListBox->clear();
 375   myIds.Clear();
 376   myBusy = false;
 377   myActor = 0;
 378   emit ListContensChanged();
 379
 380   // main buttons
 381   connect(myOkBtn,    SIGNAL(clicked()), SLOT(onOk()));
 382   connect(myCloseBtn, SIGNAL(clicked()), SLOT(onClose()));
 383   connect(myApplyBtn, SIGNAL(clicked()), SLOT(onApply()));
 384   connect(myHelpBtn,  SIGNAL(clicked()), SLOT(onHelp()));
 385
 386   // selection and SMESHGUI
 387   connect(mySelectionMgr, SIGNAL(currentSelectionChanged()), SLOT(onSelectionDone()));
 388   connect(mySMESHGUI, SIGNAL(SignalDeactivateActiveDialog()), SLOT(onDeactivate()));
 389   connect(mySMESHGUI, SIGNAL(SignalCloseAllDialogs()), SLOT(onClose()));
 390
 391   // dialog controls
 392   connect(myFilterBtn, SIGNAL(clicked()), SLOT(onFilterBtn()  ));
 393   connect(myAddBtn,    SIGNAL(clicked()), SLOT(onAddBtn()     ));
 394   connect(myRemoveBtn, SIGNAL(clicked()), SLOT(onRemoveBtn()  ));
 395   connect(mySortBtn,   SIGNAL(clicked()), SLOT(onSortListBtn()));
 396
 397   connect(mySubmeshChk, SIGNAL(stateChanged(int)), SLOT(onSubmeshChk()));
 398   connect(myGroupChk,   SIGNAL(stateChanged(int)), SLOT(onGroupChk()  ));
 399   connect(myToAllChk,   SIGNAL(stateChanged(int)), SLOT(onToAllChk()  ));
 400
 401   if (myEntityTypeGrp)
 402     connect(myEntityTypeGrp, SIGNAL(buttonClicked(int)), SLOT(on3d2dChanged(int)));
 403
 404   connect(myListBox, SIGNAL(itemSelectionChanged()), SLOT(onListSelectionChanged()));
 405
 406   onSelectionDone();
 407
 408   // set selection mode
 409   setSelectionMode();
 410   updateButtons();
 411 }
 412
 413 //=======================================================================
 414 // name    : SMESHGUI_MultiEditDlg::onOk
 415 // Purpose : SLOT called when "Ok" button pressed.
 416 //           Assign filters VTK viewer and close dialog
 417 //=======================================================================
 418 void SMESHGUI_MultiEditDlg::onOk()
 419 {
 420   if (onApply())
 421     onClose();
 422 }
 423
 424 //=======================================================================
 425 // name    : SMESHGUI_MultiEditDlg::getIds
 426 // Purpose : Retrive identifiers from list box
 427 //=======================================================================
 428 SMESH::long_array_var SMESHGUI_MultiEditDlg::getIds()
 429 {
 430   SMESH::long_array_var anIds = new SMESH::long_array;
 431
 432   if (myToAllChk->isChecked())
 433   {
 434     myIds.Clear();
 435     SMESH_Actor * anActor = SMESH::FindActorByObject(myMesh);
 436     if (!anActor)
 437       anActor = myActor;
 438     if (anActor != 0)
 439     {
 440       // skl 07.02.2006
 441       SMDS_Mesh* aMesh = myActor->GetObject()->GetMesh();
 442       if( myFilterType == SMESH::TriaFilter ||
 443           myFilterType == SMESH::QuadFilter ||
 444           myFilterType == SMESH::FaceFilter ) {
 445         SMDS_FaceIteratorPtr it = aMesh->facesIterator();
 446         while(it->more()) {
 447           const SMDS_MeshFace* f = it->next();
 448           if(myFilterType == SMESH::FaceFilter) {
 449             myIds.Add(f->GetID());
 450           }
 451           else if( myFilterType==SMESH::TriaFilter &&
 452                    ( f->NbNodes()==3 || f->NbNodes()==6 ) ) {
 453             myIds.Add(f->GetID());
 454           }
 455           else if( myFilterType==SMESH::QuadFilter &&
 456                    ( f->NbNodes()==4 || f->NbNodes()==8 ) ) {
 457             myIds.Add(f->GetID());
 458           }
 459         }
 460       }
 461       else if(myFilterType == SMESH::VolumeFilter) {
 462         SMDS_VolumeIteratorPtr it = aMesh->volumesIterator();
 463         while(it->more()) {
 464           const SMDS_MeshVolume* f = it->next();
 465           myIds.Add(f->GetID());
 466         }
 467       }
 468       /* commented by skl 07.02.2006
 469       TVisualObjPtr aVisualObj = anActor->GetObject();
 470       vtkUnstructuredGrid* aGrid = aVisualObj->GetUnstructuredGrid();
 471       if (aGrid != 0) {
 472         for (int i = 0, n = aGrid->GetNumberOfCells(); i < n; i++) {
 473           vtkCell* aCell = aGrid->GetCell(i);
 474           if (aCell != 0) {
 475             vtkTriangle* aTri = vtkTriangle::SafeDownCast(aCell);
 476             vtkQuad*     aQua = vtkQuad::SafeDownCast(aCell);
 477             vtkPolygon*  aPG  = vtkPolygon::SafeDownCast(aCell);
 478
 479             vtkCell3D*   a3d  = vtkCell3D::SafeDownCast(aCell);
 480             vtkConvexPointSet* aPH = vtkConvexPointSet::SafeDownCast(aCell);
 481
 482             if (aTri && myFilterType == SMESHGUI_TriaFilter ||
 483                 aQua && myFilterType == SMESHGUI_QuadFilter ||
 484                 (aTri || aQua || aPG) && myFilterType == SMESHGUI_FaceFilter ||
 485                 (a3d || aPH) && myFilterType == SMESHGUI_VolumeFilter) {
 486               int anObjId = aVisualObj->GetElemObjId(i);
 487               myIds.Add(anObjId);
 488             }
 489           }
 490         }
 491       }
 492       */
 493     }
 494   }
 495
 496   anIds->length(myIds.Extent());
 497   TColStd_MapIteratorOfMapOfInteger anIter(myIds);
 498   for (int i = 0; anIter.More(); anIter.Next() )
 499   {
 500     anIds[ i++ ] = anIter.Key();
 501   }
 502   return anIds._retn();
 503 }
 504
 505 //=======================================================================
 506 // name    : SMESHGUI_MultiEditDlg::onClose
 507 // Purpose : SLOT called when "Close" button pressed. Close dialog
 508 //=======================================================================
 509 void SMESHGUI_MultiEditDlg::onClose()
 510 {
 511   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
 512     aViewWindow->SetSelectionMode(ActorSelection);
 513   disconnect(mySelectionMgr, 0, this, 0);
 514   disconnect(mySMESHGUI, 0, this, 0);
 515   mySMESHGUI->ResetState();
 516
 517   SMESH::RemoveFilters();
 518   SMESH::SetPickable();
 519
 520   //mySelectionMgr->clearSelected();
 521   mySelectionMgr->clearFilters();
 522
 523   reject();
 524 }
 525
 526 //=================================================================================
 527 // function : onHelp()
 528 // purpose  :
 529 //=================================================================================
 530 void SMESHGUI_MultiEditDlg::onHelp()
 531 {
 532   LightApp_Application* app = (LightApp_Application*)(SUIT_Session::session()->activeApplication());
 533   if (app)
 534     app->onHelpContextModule(mySMESHGUI ? app->moduleName(mySMESHGUI->moduleName()) : QString(""), myHelpFileName);
 535   else {
 536     QString platform;
 537 #ifdef WIN32
 538     platform = "winapplication";
 539 #else
 540     platform = "application";
 541 #endif
 542     SUIT_MessageBox::warning(this, tr("WRN_WARNING"),
 543                              tr("EXTERNAL_BROWSER_CANNOT_SHOW_PAGE").
 544                              arg(app->resourceMgr()->stringValue("ExternalBrowser",
 545                                                                  platform)).
 546                              arg(myHelpFileName));
 547   }
 548 }
 549
 550 //=======================================================================
 551 // name    : SMESHGUI_MultiEditDlg::onSelectionDone
 552 // Purpose : SLOT called when selection changed
 553 //=======================================================================
 554 void SMESHGUI_MultiEditDlg::onSelectionDone()
 555 {
 556   if (myBusy || !isEnabled()) return;
 557   myBusy = true;
 558
 559   const SALOME_ListIO& aList = mySelector->StoredIObjects();
 560
 561   int nbSel = aList.Extent();
 562   myListBox->clearSelection();
 563
 564   if (mySubmeshChk->isChecked() || myGroupChk->isChecked()) {
 565     QLineEdit* aNameEdit = mySubmeshChk->isChecked() ? mySubmesh : myGroup;
 566     if (nbSel == 1) {
 567       Handle(SALOME_InteractiveObject) anIO = aList.First();
 568       QString aName = "";
 569       SMESH::GetNameOfSelectedIObjects(mySelectionMgr, aName);
 570       anIO.IsNull() ? aNameEdit->clear() : aNameEdit->setText(aName);
 571
 572       if (mySubmeshChk->isChecked()) {
 573         SMESH::SMESH_subMesh_var aSubMesh =
 574           SMESH::IObjectToInterface<SMESH::SMESH_subMesh>(anIO);
 575         if (!aSubMesh->_is_nil())
 576           myMesh = aSubMesh->GetFather();
 577       } else {
 578         SMESH::SMESH_GroupBase_var aGroup =
 579           SMESH::IObjectToInterface<SMESH::SMESH_GroupBase>(anIO);
 580         if (!aGroup->_is_nil())
 581           myMesh = aGroup->GetMesh();
 582       }
 583     } else if (nbSel > 1) {
 584       QString aStr = mySubmeshChk->isChecked() ?
 585         tr("SMESH_SUBMESH_SELECTED") : tr("SMESH_GROUP_SELECTED");
 586       aNameEdit->setText(aStr.arg(nbSel));
 587     } else {
 588       aNameEdit->clear();
 589     }
 590   } else if (nbSel > 0) {
 591     QString aListStr = "";
 592     Handle(SALOME_InteractiveObject) anIO = aList.First();
 593     int aNbItems = SMESH::GetNameOfSelectedElements(mySelector,anIO,aListStr);
 594     if (aNbItems > 0) {
 595       QStringList anElements = aListStr.split(" ", QString::SkipEmptyParts);
 596       for (QStringList::iterator it = anElements.begin(); it != anElements.end(); ++it) {
 597         QList<QListWidgetItem*> items = myListBox->findItems(*it, Qt::MatchExactly);
 598         QListWidgetItem* anItem;
 599         foreach(anItem, items)
 600           anItem->setSelected(true);
 601       }
 602     }
 603     myMesh = SMESH::GetMeshByIO(anIO);
 604   }
 605
 606   if (nbSel > 0) {
 607     myActor = SMESH::FindActorByEntry(aList.First()->getEntry());
 608     if (!myActor)
 609       myActor = SMESH::FindActorByObject(myMesh);
 610     SVTK_Selector* aSelector = SMESH::GetSelector();
 611     Handle(VTKViewer_Filter) aFilter = aSelector->GetFilter(myFilterType);
 612     if (!aFilter.IsNull())
 613       aFilter->SetActor(myActor);
 614   }
 615   myBusy = false;
 616
 617   updateButtons();
 618 }
 619
 620 //=======================================================================
 621 // name    : SMESHGUI_MultiEditDlg::onDeactivate
 622 // Purpose : SLOT called when dialog must be deativated
 623 //=======================================================================
 624 void SMESHGUI_MultiEditDlg::onDeactivate()
 625 {
 626   setEnabled(false);
 627 }
 628
 629 //=======================================================================
 630 // name    : SMESHGUI_MultiEditDlg::enterEvent
 631 // Purpose : Event filter
 632 //=======================================================================
 633 void SMESHGUI_MultiEditDlg::enterEvent (QEvent*)
 634 {
 635   if (!isEnabled()) {
 636     mySMESHGUI->EmitSignalDeactivateDialog();
 637     setEnabled(true);
 638     setSelectionMode();
 639   }
 640 }
 641
 642 //=======================================================================
 643 // name    : SMESHGUI_MultiEditDlg::closeEvent
 644 // Purpose :
 645 //=======================================================================
 646 void SMESHGUI_MultiEditDlg::closeEvent (QCloseEvent*)
 647 {
 648   onClose();
 649 }
 650 //=======================================================================
 651 // name    : SMESHGUI_MultiEditDlg::hideEvent
 652 // Purpose : caused by ESC key
 653 //=======================================================================
 654 void SMESHGUI_MultiEditDlg::hideEvent (QHideEvent*)
 655 {
 656   if (!isMinimized())
 657     onClose();
 658 }
 659
 660 //=======================================================================
 661 // name    : SMESHGUI_MultiEditDlg::onFilterBtn
 662 // Purpose : SLOT. Called when "Filter" button pressed.
 663 //           Start "Selection filters" dialog
 664 //=======================================================================
 665 void SMESHGUI_MultiEditDlg::onFilterBtn()
 666 {
 667   if (myFilterDlg == 0) {
 668     myFilterDlg = new SMESHGUI_FilterDlg( mySMESHGUI, entityType() ? SMESH::VOLUME : SMESH::FACE);
 669     connect(myFilterDlg, SIGNAL(Accepted()), SLOT(onFilterAccepted()));
 670   } else {
 671     myFilterDlg->Init(entityType() ? SMESH::VOLUME : SMESH::FACE);
 672   }
 673
 674   myFilterDlg->SetSelection();
 675   myFilterDlg->SetMesh(myMesh);
 676   myFilterDlg->SetSourceWg(myListBox);
 677
 678   myFilterDlg->show();
 679 }
 680
 681 //=======================================================================
 682 // name    : onFilterAccepted()
 683 // Purpose : SLOT. Called when Filter dlg closed with OK button.
 684 //            Uncheck "Select submesh" and "Select group" checkboxes
 685 //=======================================================================
 686 void SMESHGUI_MultiEditDlg::onFilterAccepted()
 687 {
 688   myIds.Clear();
 689   for (int i = 0, n = myListBox->count(); i < n; i++)
 690     myIds.Add(myListBox->item(i)->text().toInt());
 691
 692   emit ListContensChanged();
 693
 694   if (mySubmeshChk->isChecked() || myGroupChk->isChecked()) {
 695     mySubmeshChk->blockSignals(true);
 696     myGroupChk->blockSignals(true);
 697     mySubmeshChk->setChecked(false);
 698     myGroupChk->setChecked(false);
 699     mySubmeshChk->blockSignals(false);
 700     myGroupChk->blockSignals(false);
 701   }
 702   updateButtons();
 703 }
 704
 705 //=======================================================================
 706 // name    : SMESHGUI_MultiEditDlg::isIdValid
 707 // Purpose : Verify whether Id of element satisfies to filters from viewer
 708 //=======================================================================
 709 bool SMESHGUI_MultiEditDlg::isIdValid (const int theId) const
 710 {
 711   SVTK_Selector* aSelector = SMESH::GetSelector();
 712   Handle(SMESHGUI_Filter) aFilter =
 713     Handle(SMESHGUI_Filter)::DownCast(aSelector->GetFilter(myFilterType));
 714
 715   return (!aFilter.IsNull() && aFilter->IsObjValid(theId));
 716 }
 717
 718 //=======================================================================
 719 // name    : SMESHGUI_MultiEditDlg::onAddBtn
 720 // Purpose : SLOT. Called when "Add" button pressed.
 721 //           Add selected in viewer entities in list box
 722 //=======================================================================
 723 void SMESHGUI_MultiEditDlg::onAddBtn()
 724 {
 725   const SALOME_ListIO& aList = mySelector->StoredIObjects();
 726
 727   int nbSelected = aList.Extent();
 728   if (nbSelected == 0)
 729     return;
 730
 731   TColStd_IndexedMapOfInteger toBeAdded;
 732
 733   if (!mySubmeshChk->isChecked() && !myGroupChk->isChecked()) {
 734     if (nbSelected > 0)
 735       mySelector->GetIndex(aList.First(),toBeAdded);
 736   } else if (mySubmeshChk->isChecked()) {
 737     SALOME_ListIteratorOfListIO anIter(aList);
 738     for ( ; anIter.More(); anIter.Next()) {
 739       SMESH::SMESH_subMesh_var aSubMesh =
 740         SMESH::IObjectToInterface<SMESH::SMESH_subMesh>(anIter.Value());
 741       if (!aSubMesh->_is_nil()) {
 742         if (aSubMesh->GetFather()->GetId() == myMesh->GetId()) {
 743           SMESH::long_array_var anIds = aSubMesh->GetElementsId();
 744           for (int i = 0, n = anIds->length(); i < n; i++) {
 745             if (isIdValid(anIds[ i ]))
 746               toBeAdded.Add(anIds[ i ]);
 747           }
 748         }
 749       }
 750     }
 751   } else if (myGroupChk->isChecked()) {
 752     SALOME_ListIteratorOfListIO anIter(aList);
 753     for ( ; anIter.More(); anIter.Next()) {
 754       SMESH::SMESH_GroupBase_var aGroup =
 755         SMESH::IObjectToInterface<SMESH::SMESH_GroupBase>(anIter.Value());
 756       if (!aGroup->_is_nil() && (aGroup->GetType() == SMESH::FACE &&
 757                                  entityType() == 0 || aGroup->GetType() == SMESH::VOLUME &&
 758                                  entityType() == 1)) {
 759         if (aGroup->GetMesh()->GetId() == myMesh->GetId()) {
 760           SMESH::long_array_var anIds = aGroup->GetListOfID();
 761           for (int i = 0, n = anIds->length(); i < n; i++) {
 762             if (isIdValid(anIds[ i ]))
 763               toBeAdded.Add(anIds[ i ]);
 764           }
 765         }
 766       }
 767     }
 768   } else {
 769   }
 770
 771   myBusy = true;
 772   bool isGroupOrSubmesh = (mySubmeshChk->isChecked() || myGroupChk->isChecked());
 773   mySubmeshChk->setChecked(false);
 774   myGroupChk->setChecked(false);
 775   for(int i = 1; i <= toBeAdded.Extent(); i++)
 776     if (myIds.Add(toBeAdded(i))) {
 777       QListWidgetItem* item = new QListWidgetItem(QString("%1").arg(toBeAdded(i)));
 778       myListBox->addItem(item);
 779       item->setSelected(true);
 780     }
 781   myBusy = false;
 782
 783   emit ListContensChanged();
 784
 785   if (isGroupOrSubmesh)
 786     onListSelectionChanged();
 787
 788   updateButtons();
 789 }
 790
 791 //=======================================================================
 792 // name    : SMESHGUI_MultiEditDlg::updateButtons
 793 // Purpose : Enable/disable buttons of dialog in accordance with current state
 794 //=======================================================================
 795 void SMESHGUI_MultiEditDlg::updateButtons()
 796 {
 797   bool isOk = isValid(false);
 798   myOkBtn->setEnabled(isOk);
 799   myApplyBtn->setEnabled(isOk);
 800
 801   bool isListBoxNonEmpty = myListBox->count() > 0;
 802   bool isToAll = myToAllChk->isChecked();
 803   myFilterBtn->setEnabled(!isToAll);
 804   myRemoveBtn->setEnabled(isListBoxNonEmpty && !isToAll);
 805   mySortBtn->setEnabled(isListBoxNonEmpty &&!isToAll);
 806
 807   const SALOME_ListIO& aList = mySelector->StoredIObjects();
 808
 809   if (isToAll ||
 810       myMesh->_is_nil() ||
 811       aList.Extent() < 1 ||
 812       (SMESH::IObjectToInterface<SMESH::SMESH_subMesh>(aList.First())->_is_nil() &&
 813        SMESH::IObjectToInterface<SMESH::SMESH_GroupBase>(aList.First())->_is_nil() &&
 814        SMESH::IObjectToInterface<SMESH::SMESH_Mesh>(aList.First())->_is_nil()))
 815     myAddBtn->setEnabled(false);
 816   else
 817     myAddBtn->setEnabled(true);
 818
 819   mySubmeshChk->setEnabled(!isToAll);
 820   mySubmeshBtn->setEnabled(mySubmeshChk->isChecked());
 821   mySubmesh->setEnabled(mySubmeshChk->isChecked());
 822
 823   myGroupChk->setEnabled(!isToAll);
 824   myGroupBtn->setEnabled(myGroupChk->isChecked());
 825   myGroup->setEnabled(myGroupChk->isChecked());
 826
 827   if (!mySubmeshChk->isChecked())
 828     mySubmesh->clear();
 829   if (!myGroupChk->isChecked())
 830     myGroup->clear();
 831
 832 }
 833
 834 //=======================================================================
 835 // name    : SMESHGUI_MultiEditDlg::onRemoveBtn
 836 // Purpose : SLOT. Called when "Remove" button pressed.
 837 //           Remove selected in list box entities
 838 //=======================================================================
 839 void SMESHGUI_MultiEditDlg::onRemoveBtn()
 840 {
 841   myBusy = true;
 842
 843   QList<QListWidgetItem*> selItems = myListBox->selectedItems();
 844   QListWidgetItem* item;
 845   foreach(item, selItems)
 846   {
 847     myIds.Remove(item->text().toInt());
 848     delete item;
 849   }
 850
 851   myBusy = false;
 852
 853   emit ListContensChanged();
 854   updateButtons();
 855 }
 856
 857 //=======================================================================
 858 // name    : SMESHGUI_MultiEditDlg::onSortListBtn
 859 // Purpose : SLOT. Called when "Sort list" button pressed.
 860 //           Sort entities of list box
 861 //=======================================================================
 862 void SMESHGUI_MultiEditDlg::onSortListBtn()
 863 {
 864   myBusy = true;
 865
 866   int i, k = myListBox->count();
 867   if (k > 0)
 868   {
 869     QList<int> aSelected;
 870     std::vector<int> anArray(k);
 871     for (i = 0; i < k; i++)
 872     {
 873       int id = myListBox->item(i)->text().toInt();
 874       anArray[ i ] = id;
 875       if (myListBox->item(i)->isSelected())
 876         aSelected.append(id);
 877     }
 878
 879     std::sort(anArray.begin(), anArray.end());
 880
 881     myListBox->clear();
 882     for (i = 0; i < k; i++) {
 883       QListWidgetItem* item = new QListWidgetItem(QString::number(anArray[i]));
 884       myListBox->addItem(item);
 885       item->setSelected(aSelected.contains(anArray[i]));
 886     }
 887   }
 888   myBusy = false;
 889 }
 890
 891 //=======================================================================
 892 // name    : SMESHGUI_MultiEditDlg::onListSelectionChanged
 893 // Purpose : SLOT. Called when selection in list box changed.
 894 //           Highlight in selected entities
 895 //=======================================================================
 896 void SMESHGUI_MultiEditDlg::onListSelectionChanged()
 897 {
 898   if (myActor == 0 || myBusy)
 899     return;
 900
 901   if (mySubmeshChk->isChecked() || myGroupChk->isChecked())
 902     return;
 903
 904   SMESH_Actor * anActor = SMESH::FindActorByObject(myMesh);
 905   if (!anActor)
 906     anActor = myActor;
 907   TVisualObjPtr anObj = anActor->GetObject();
 908
 909   TColStd_MapOfInteger anIndexes;
 910   int total = myListBox->count();
 911   for (int i = 0; i < total; i++)
 912   {
 913     if (myListBox->item(i)->isSelected())
 914     {
 915       int anId = myListBox->item(i)->text().toInt();
 916       if (anObj->GetElemVTKId(anId) >= 0) // avoid exception in hilight
 917         anIndexes.Add(anId);
 918     }
 919   }
 920
 921   mySelector->AddOrRemoveIndex(anActor->getIO(),anIndexes,false);
 922   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
 923     aViewWindow->highlight(anActor->getIO(),true,true);
 924 }
 925
 926 //=======================================================================
 927 // name    : SMESHGUI_MultiEditDlg::onSubmeshChk
 928 // Purpose : SLOT. Called when state of "SubMesh" check box changed.
 929 //           Activate/deactivate selection of submeshes
 930 //=======================================================================
 931 void SMESHGUI_MultiEditDlg::onSubmeshChk()
 932 {
 933   bool isChecked = mySubmeshChk->isChecked();
 934   mySubmeshBtn->setEnabled(isChecked);
 935   mySubmesh->setEnabled(isChecked);
 936   if (!isChecked)
 937     mySubmesh->clear();
 938   if (isChecked && myGroupChk->isChecked())
 939       myGroupChk->setChecked(false);
 940
 941   setSelectionMode();
 942 }
 943
 944 //=======================================================================
 945 // name    : SMESHGUI_MultiEditDlg::onGroupChk
 946 // Purpose : SLOT. Called when state of "Group" check box changed.
 947 //           Activate/deactivate selection of groupes
 948 //=======================================================================
 949 void SMESHGUI_MultiEditDlg::onGroupChk()
 950 {
 951   bool isChecked = myGroupChk->isChecked();
 952   myGroupBtn->setEnabled(isChecked);
 953   myGroup->setEnabled(isChecked);
 954   if (!isChecked)
 955     myGroup->clear();
 956   if (isChecked && mySubmeshChk->isChecked())
 957       mySubmeshChk->setChecked(false);
 958
 959   setSelectionMode();
 960 }
 961
 962 //=======================================================================
 963 // name    : SMESHGUI_MultiEditDlg::onToAllChk
 964 // Purpose : SLOT. Called when state of "Apply to all" check box changed.
 965 //           Activate/deactivate selection
 966 //=======================================================================
 967 void SMESHGUI_MultiEditDlg::onToAllChk()
 968 {
 969   bool isChecked = myToAllChk->isChecked();
 970
 971   if (isChecked)
 972     myListBox->clear();
 973
 974   myIds.Clear();
 975
 976   emit ListContensChanged();
 977
 978   updateButtons();
 979   setSelectionMode();
 980 }
 981
 982 //=======================================================================
 983 // name    : SMESHGUI_MultiEditDlg::setSelectionMode
 984 // Purpose : Set selection mode
 985 //=======================================================================
 986 void SMESHGUI_MultiEditDlg::setSelectionMode()
 987 {
 988   SMESH::RemoveFilters();
 989
 990   mySelectionMgr->clearSelected();
 991   mySelectionMgr->clearFilters();
 992
 993   if (mySubmeshChk->isChecked()) {
 994     if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
 995       aViewWindow->SetSelectionMode(ActorSelection);
 996     mySelectionMgr->installFilter(new SMESH_TypeFilter(SUBMESH));
 997   }
 998   else if (myGroupChk->isChecked()) {
 999     if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
1000       aViewWindow->SetSelectionMode(ActorSelection);
1001     mySelectionMgr->installFilter(new SMESH_TypeFilter(GROUP));
1002   }
1003
1004   if (entityType()) {
1005     if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
1006       aViewWindow->SetSelectionMode(VolumeSelection);
1007     SMESH::SetFilter(new SMESHGUI_VolumesFilter());
1008   } else {
1009     if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
1010       aViewWindow->SetSelectionMode(FaceSelection);
1011     if (myFilterType == SMESH::TriaFilter)
1012       SMESH::SetFilter(new SMESHGUI_TriangleFilter());
1013     else if (myFilterType == SMESH::QuadFilter)
1014       SMESH::SetFilter(new SMESHGUI_QuadrangleFilter());
1015     else
1016       SMESH::SetFilter(new SMESHGUI_FacesFilter());
1017   }
1018 }
1019
1020 //=======================================================================
1021 // name    : SMESHGUI_MultiEditDlg::onApply
1022 // Purpose : SLOT. Called when "Apply" button clicked.
1023 //=======================================================================
1024 bool SMESHGUI_MultiEditDlg::onApply()
1025 {
1026   if (mySMESHGUI->isActiveStudyLocked())
1027     return false;
1028   if (!isValid(true))
1029     return false;
1030
1031   SMESH::SMESH_MeshEditor_var aMeshEditor = myMesh->GetMeshEditor();
1032   if (aMeshEditor->_is_nil())
1033     return false;
1034
1035   myBusy = true;
1036
1037   SMESH::long_array_var anIds = getIds();
1038
1039   bool aResult = process(aMeshEditor, anIds.inout());
1040   if (aResult) {
1041     if (myActor) {
1042       SALOME_ListIO sel;
1043       mySelectionMgr->selectedObjects( sel );
1044       mySelector->ClearIndex();
1045       mySelectionMgr->setSelectedObjects( sel );
1046       SMESH::UpdateView();
1047     }
1048
1049     myListBox->clear();
1050     myIds.Clear();
1051     emit ListContensChanged();
1052
1053     updateButtons();
1054   }
1055
1056   myBusy = false;
1057   return aResult;
1058 }
1059
1060 //=======================================================================
1061 // name    : SMESHGUI_MultiEditDlg::on3d2dChanged
1062 // Purpose :
1063 //=======================================================================
1064 void SMESHGUI_MultiEditDlg::on3d2dChanged (int type)
1065 {
1066   if (myEntityType != type) {
1067     myEntityType = type;
1068
1069     myListBox->clear();
1070     myIds.Clear();
1071
1072     emit ListContensChanged();
1073
1074     if (type)
1075       myFilterType = SMESH::VolumeFilter;
1076     else
1077       myFilterType = SMESH::FaceFilter;
1078
1079     updateButtons();
1080     setSelectionMode();
1081   }
1082 }
1083
1084 //=======================================================================
1085 // name    : SMESHGUI_MultiEditDlg::entityType
1086 // Purpose :
1087 //=======================================================================
1088 int SMESHGUI_MultiEditDlg::entityType()
1089 {
1090   return myEntityType;
1091 }
1092
1093 //=================================================================================
1094 // function : keyPressEvent()
1095 // purpose  :
1096 //=================================================================================
1097 void SMESHGUI_MultiEditDlg::keyPressEvent( QKeyEvent* e )
1098 {
1099   QDialog::keyPressEvent( e );
1100   if ( e->isAccepted() )
1101     return;
1102
1103   if ( e->key() == Qt::Key_F1 ) {
1104     e->accept();
1105     onHelp();
1106   }
1107 }
1108
1109 /*!
1110  *  Class       : SMESHGUI_ChangeOrientationDlg
1111  *  Description : Modification of orientation of faces
1112  */
1113
1114 SMESHGUI_ChangeOrientationDlg
1115 ::SMESHGUI_ChangeOrientationDlg(SMESHGUI* theModule):
1116   SMESHGUI_MultiEditDlg(theModule, SMESH::FaceFilter, true)
1117 {
1118   setWindowTitle(tr("CAPTION"));
1119   myHelpFileName = "changing_orientation_of_elements_page.html";
1120 }
1121
1122 SMESHGUI_ChangeOrientationDlg::~SMESHGUI_ChangeOrientationDlg()
1123 {
1124 }
1125
1126 bool SMESHGUI_ChangeOrientationDlg::process (SMESH::SMESH_MeshEditor_ptr theEditor,
1127                                              const SMESH::long_array&    theIds)
1128 {
1129   return theEditor->Reorient(theIds);
1130 }
1131
1132 /*!
1133  *  Class       : SMESHGUI_UnionOfTrianglesDlg
1134  *  Description : Construction of quadrangles by automatic association of triangles
1135  */
1136
1137 SMESHGUI_UnionOfTrianglesDlg
1138 ::SMESHGUI_UnionOfTrianglesDlg(SMESHGUI* theModule):
1139   SMESHGUI_MultiEditDlg(theModule, SMESH::TriaFilter, false)
1140 {
1141   setWindowTitle(tr("CAPTION"));
1142
1143   myComboBoxFunctor->setEnabled(true);
1144   myComboBoxFunctor->addItem(tr("WARP_ELEMENTS")); // for quadrangles only
1145   myComboBoxFunctor->addItem(tr("TAPER_ELEMENTS")); // for quadrangles only
1146
1147   // Maximum angle
1148   QWidget* aMaxAngleGrp = new QWidget(myCriterionGrp);
1149   QHBoxLayout* aMaxAngleGrpLayout = new QHBoxLayout(aMaxAngleGrp);
1150   aMaxAngleGrpLayout->setMargin(0);
1151   aMaxAngleGrpLayout->setSpacing(SPACING);
1152
1153   QLabel* aLab = new QLabel (tr("MAXIMUM_ANGLE"), aMaxAngleGrp);
1154   myMaxAngleSpin = new SMESHGUI_SpinBox (aMaxAngleGrp);
1155   myMaxAngleSpin->RangeStepAndValidator(0, 180.0, 1.0, 3);
1156   myMaxAngleSpin->SetValue(30.0);
1157
1158   aMaxAngleGrpLayout->addWidget(aLab);
1159   aMaxAngleGrpLayout->addWidget(myMaxAngleSpin);
1160
1161   ((QVBoxLayout*)(myCriterionGrp->layout()))->addWidget(aMaxAngleGrp);
1162   myCriterionGrp->show();
1163
1164   myHelpFileName = "uniting_set_of_triangles_page.html";
1165 }
1166
1167 SMESHGUI_UnionOfTrianglesDlg::~SMESHGUI_UnionOfTrianglesDlg()
1168 {
1169 }
1170
1171 bool SMESHGUI_UnionOfTrianglesDlg::isValid (const bool theMess)
1172 {
1173   bool ok = SMESHGUI_MultiEditDlg::isValid( theMess );
1174   if( !ok )
1175     return false;
1176
1177   QString msg;
1178   ok = myMaxAngleSpin->isValid( msg, theMess );
1179   if( !ok ) {
1180     if( theMess ) {
1181       QString str( tr( "SMESH_INCORRECT_INPUT" ) );
1182       if ( !msg.isEmpty() )
1183         str += "\n" + msg;
1184       SUIT_MessageBox::critical( this, tr( "SMESH_ERROR" ), str );
1185     }
1186     return false;
1187   }
1188
1189   return ok;
1190 }
1191
1192 bool SMESHGUI_UnionOfTrianglesDlg::process (SMESH::SMESH_MeshEditor_ptr theEditor,
1193                                             const SMESH::long_array&    theIds)
1194 {
1195   SMESH::NumericalFunctor_var aCriterion = getNumericalFunctor();
1196   double aMaxAngle = myMaxAngleSpin->GetValue() * PI / 180.0;
1197   bool ok = theEditor->TriToQuad(theIds, aCriterion, aMaxAngle);
1198   if( ok ) {
1199     QStringList aParameters;
1200     aParameters << myMaxAngleSpin->text();
1201     myMesh->SetParameters( SMESHGUI::JoinObjectParameters(aParameters) );
1202   }
1203   return ok;
1204 }
1205
1206
1207 /*!
1208  *  Class       : SMESHGUI_CuttingOfQuadsDlg
1209  *  Description : Automatic splitting of quadrangles into triangles
1210  */
1211
1212 SMESHGUI_CuttingOfQuadsDlg
1213 ::SMESHGUI_CuttingOfQuadsDlg(SMESHGUI* theModule):
1214   SMESHGUI_MultiEditDlg(theModule, SMESH::QuadFilter, false)
1215 {
1216   setWindowTitle(tr("CAPTION"));
1217   myPreviewActor = 0;
1218
1219   myPreviewChk = new QCheckBox (tr("PREVIEW"), mySelGrp);
1220   QGridLayout* aLay = (QGridLayout*)(mySelGrp->layout());
1221   aLay->addWidget(myPreviewChk, aLay->rowCount(), 0, 1, aLay->columnCount());
1222
1223   myCriterionGrp->show();
1224   myChoiceWidget->show();
1225   myComboBoxFunctor->setEnabled(false);
1226
1227   connect(myPreviewChk,      SIGNAL(stateChanged(int)),    this, SLOT(onPreviewChk()));
1228   connect(myGroupChoice,     SIGNAL(buttonClicked(int)),   this, SLOT(onCriterionRB()));
1229   connect(myComboBoxFunctor, SIGNAL(activated(int)),       this, SLOT(onPreviewChk()));
1230   connect(this,              SIGNAL(ListContensChanged()), this, SLOT(onPreviewChk()));
1231
1232   myHelpFileName = "cutting_quadrangles_page.html";
1233 }
1234
1235 SMESHGUI_CuttingOfQuadsDlg::~SMESHGUI_CuttingOfQuadsDlg()
1236 {
1237 }
1238
1239 void SMESHGUI_CuttingOfQuadsDlg::onClose()
1240 {
1241   erasePreview();
1242   SMESHGUI_MultiEditDlg::onClose();
1243 }
1244
1245 bool SMESHGUI_CuttingOfQuadsDlg::process (SMESH::SMESH_MeshEditor_ptr theEditor,
1246                                           const SMESH::long_array&    theIds)
1247 {
1248   switch (myGroupChoice->checkedId()) {
1249   case 0: // use diagonal 1-3
1250     return theEditor->SplitQuad(theIds, true);
1251   case 1: // use diagonal 2-4
1252     return theEditor->SplitQuad(theIds, false);
1253   default: // use numeric functor
1254     break;
1255   }
1256
1257   SMESH::NumericalFunctor_var aCriterion = getNumericalFunctor();
1258   return theEditor->QuadToTri(theIds, aCriterion);
1259 }
1260
1261 void SMESHGUI_CuttingOfQuadsDlg::onCriterionRB()
1262 {
1263   if (myGroupChoice->checkedId() == 2) // Use numeric functor
1264     myComboBoxFunctor->setEnabled(true);
1265   else
1266     myComboBoxFunctor->setEnabled(false);
1267
1268   onPreviewChk();
1269 }
1270
1271 void SMESHGUI_CuttingOfQuadsDlg::onPreviewChk()
1272 {
1273   myPreviewChk->isChecked() ? displayPreview() : erasePreview();
1274 }
1275
1276 void SMESHGUI_CuttingOfQuadsDlg::erasePreview()
1277 {
1278   if (myPreviewActor == 0)
1279     return;
1280
1281   if (SVTK_ViewWindow* vf = SMESH::GetCurrentVtkView()) {
1282     vf->RemoveActor(myPreviewActor);
1283     vf->Repaint();
1284   }
1285   myPreviewActor->Delete();
1286   myPreviewActor = 0;
1287 }
1288
1289 void SMESHGUI_CuttingOfQuadsDlg::displayPreview()
1290 {
1291   if (myActor == 0)
1292     return;
1293
1294   if (myPreviewActor != 0)
1295     erasePreview();
1296
1297   // get Ids of elements
1298   SMESH::long_array_var anElemIds = getIds();
1299   if (getIds()->length() == 0)
1300     return;
1301
1302   SMDS_Mesh* aMesh = myActor->GetObject()->GetMesh();
1303   if (aMesh == 0)
1304     return;
1305
1306   // 0 - use diagonal 1-3, 1 - use diagonal 2-4, 2 - use numerical functor
1307   int aChoice = myGroupChoice->checkedId();
1308   SMESH::NumericalFunctor_var aCriterion  = SMESH::NumericalFunctor::_nil();
1309   SMESH::SMESH_MeshEditor_var aMeshEditor = SMESH::SMESH_MeshEditor::_nil();
1310   if (aChoice == 2) {
1311     aCriterion  = getNumericalFunctor();
1312     aMeshEditor = myMesh->GetMeshEditor();
1313     if (aMeshEditor->_is_nil())
1314       return;
1315   }
1316
1317   //Create grid
1318   vtkUnstructuredGrid* aGrid = vtkUnstructuredGrid::New();
1319
1320   vtkIdType aNbCells = anElemIds->length() * 2;
1321   vtkIdType aCellsSize = 4 * aNbCells;
1322   vtkCellArray* aConnectivity = vtkCellArray::New();
1323   aConnectivity->Allocate(aCellsSize, 0);
1324
1325   vtkPoints* aPoints = vtkPoints::New();
1326   aPoints->SetNumberOfPoints(anElemIds->length() * 4);
1327
1328   vtkUnsignedCharArray* aCellTypesArray = vtkUnsignedCharArray::New();
1329   aCellTypesArray->SetNumberOfComponents(1);
1330   aCellTypesArray->Allocate(aNbCells * aCellTypesArray->GetNumberOfComponents());
1331
1332   vtkIdList *anIdList = vtkIdList::New();
1333   anIdList->SetNumberOfIds(3);
1334
1335   TColStd_DataMapOfIntegerInteger anIdToVtk;
1336
1337   int aNodes[ 4 ];
1338   int nbPoints = -1;
1339   for (int i = 0, n = anElemIds->length(); i < n; i++)
1340   {
1341     const SMDS_MeshElement* anElem = aMesh->FindElement(anElemIds[ i ]);
1342     if (anElem == 0 || anElem->NbNodes() != 4)
1343       continue;
1344
1345     SMDS_ElemIteratorPtr anIter = anElem->nodesIterator();
1346     int k = 0;
1347     while (anIter->more()) {
1348       const SMDS_MeshNode* aNode = static_cast<const SMDS_MeshNode*>(anIter->next());
1349       if (aNode)
1350       {
1351         if (!anIdToVtk.IsBound(aNode->GetID()))
1352         {
1353           aPoints->SetPoint(++nbPoints, aNode->X(), aNode->Y(), aNode->Z());
1354           anIdToVtk.Bind(aNode->GetID(), nbPoints);
1355         }
1356
1357         aNodes[ k++ ] = aNode->GetID();
1358       }
1359     }
1360
1361     if (k != 4)
1362       continue;
1363
1364     bool isDiag13 = true;
1365     if (aChoice == 0) // use diagonal 1-3
1366     {
1367       isDiag13 = true;
1368     }
1369     else if (aChoice == 1) // use diagonal 2-4
1370     {
1371       isDiag13 = false;
1372     }
1373     else // use numerical functor
1374     {
1375       // compare two sets of possible triangles
1376       int diag = aMeshEditor->BestSplit(anElemIds[i], aCriterion);
1377       if (diag == 1) // 1-3
1378         isDiag13 = true;
1379       else if (diag == 2) // 2-4
1380         isDiag13 = false;
1381       else // error
1382         continue;
1383     }
1384
1385     if (isDiag13)
1386     {
1387       anIdList->SetId(0, anIdToVtk(aNodes[ 0 ]));
1388       anIdList->SetId(1, anIdToVtk(aNodes[ 1 ]));
1389       anIdList->SetId(2, anIdToVtk(aNodes[ 2 ]));
1390       aConnectivity->InsertNextCell(anIdList);
1391       aCellTypesArray->InsertNextValue(VTK_TRIANGLE);
1392
1393       anIdList->SetId(0, anIdToVtk(aNodes[ 2 ]));
1394       anIdList->SetId(1, anIdToVtk(aNodes[ 3 ]));
1395       anIdList->SetId(2, anIdToVtk(aNodes[ 0 ]));
1396       aConnectivity->InsertNextCell(anIdList);
1397       aCellTypesArray->InsertNextValue(VTK_TRIANGLE);
1398     }
1399     else
1400     {
1401       anIdList->SetId(0, anIdToVtk(aNodes[ 1 ]));
1402       anIdList->SetId(1, anIdToVtk(aNodes[ 2 ]));
1403       anIdList->SetId(2, anIdToVtk(aNodes[ 3 ]));
1404       aConnectivity->InsertNextCell(anIdList);
1405       aCellTypesArray->InsertNextValue(VTK_TRIANGLE);
1406
1407       anIdList->SetId(0, anIdToVtk(aNodes[ 3 ]));
1408       anIdList->SetId(1, anIdToVtk(aNodes[ 0 ]));
1409       anIdList->SetId(2, anIdToVtk(aNodes[ 1 ]));
1410       aConnectivity->InsertNextCell(anIdList);
1411       aCellTypesArray->InsertNextValue(VTK_TRIANGLE);
1412     }
1413   }
1414
1415   VTKViewer_CellLocationsArray* aCellLocationsArray = VTKViewer_CellLocationsArray::New();
1416   aCellLocationsArray->SetNumberOfComponents(1);
1417   aCellLocationsArray->SetNumberOfTuples(aNbCells);
1418
1419   aConnectivity->InitTraversal();
1420   for(vtkIdType idType = 0, *pts, npts; aConnectivity->GetNextCell(npts, pts); idType++)
1421     aCellLocationsArray->SetValue(idType, aConnectivity->GetTraversalLocation(npts));
1422
1423   aGrid->SetPoints(aPoints);
1424   aGrid->SetCells(aCellTypesArray, aCellLocationsArray,aConnectivity);
1425
1426   // Create and display actor
1427   vtkDataSetMapper* aMapper = vtkDataSetMapper::New();
1428   aMapper->SetInput(aGrid);
1429
1430   myPreviewActor = SALOME_Actor::New();
1431   myPreviewActor->PickableOff();
1432   myPreviewActor->SetMapper(aMapper);
1433
1434   vtkProperty* aProp = vtkProperty::New();
1435   aProp->SetRepresentationToWireframe();
1436   aProp->SetColor(250, 0, 250);
1437   aProp->SetLineWidth(myActor->GetLineWidth() + 1);
1438   myPreviewActor->SetProperty(aProp);
1439
1440   SMESH::GetCurrentVtkView()->AddActor(myPreviewActor);
1441   SMESH::GetCurrentVtkView()->Repaint();
1442
1443   aProp->Delete();
1444   aPoints->Delete();
1445   aConnectivity->Delete();
1446   aGrid->Delete();
1447   aMapper->Delete();
1448   anIdList->Delete();
1449   aCellTypesArray->Delete();
1450   aCellLocationsArray->Delete();
1451 }