src/SMESHGUI/SMESHGUI_MultiEditDlg.cxx

   1 //  SMESH SMESHGUI : GUI for SMESH component
   2 //
   3 //  Copyright (C) 2003  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.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org
  21 //
  22 //
  23 //
  24 //  File   : SMESHGUI_MultiEditDlg.cxx
  25 //  Author : Sergey LITONIN
  26 //  Module : SMESH
  27
  28 #include "SMESHGUI_MultiEditDlg.h"
  29 #include "SMESHGUI_FilterDlg.h"
  30 #include "SMESHGUI_Filter.h"
  31
  32 #include "SMESHGUI.h"
  33 #include "SMESHGUI_Utils.h"
  34 #include "SMESHGUI_VTKUtils.h"
  35 #include "SMESHGUI_MeshUtils.h"
  36
  37 #include "QAD_Desktop.h"
  38 #include "QAD_RightFrame.h"
  39
  40 #include "VTKViewer_ViewFrame.h"
  41
  42 #include "SMESH_Actor.h"
  43 #include "SMDS_Mesh.hxx"
  44 #include "SMDS_MeshElement.hxx"
  45
  46 #include "SALOME_Selection.h"
  47 #include "SALOME_ListIteratorOfListIO.hxx"
  48 #include "VTKViewer_InteractorStyleSALOME.h"
  49
  50 #include <vtkCell.h>
  51 #include <vtkIdList.h>
  52 #include <vtkIntArray.h>
  53 #include <vtkCellArray.h>
  54 #include <vtkUnsignedCharArray.h>
  55 #include <vtkUnstructuredGrid.h>
  56 #include <vtkDataSetMapper.h>
  57
  58 #include <TColStd_IndexedMapOfInteger.hxx>
  59 #include <TColStd_MapIteratorOfMapOfInteger.hxx>
  60 #include <Precision.hxx>
  61 #include <TColStd_DataMapOfIntegerInteger.hxx>
  62
  63 #include <qcheckbox.h>
  64 #include <qframe.h>
  65 #include <qgroupbox.h>
  66 #include <qlabel.h>
  67 #include <qlayout.h>
  68 #include <qlineedit.h>
  69 #include <qlistbox.h>
  70 #include <qpushbutton.h>
  71 #include <qapplication.h>
  72
  73 #define SPACING 5
  74 #define MARGIN  10
  75
  76 /*
  77   Class       : SMESHGUI_MultiEditDlg
  78   Description : Description : Inversion of the diagonal of a pseudo-quadrangle formed by
  79                 2 neighboring triangles with 1 common edge
  80 */
  81
  82 //=======================================================================
  83 // name    : SMESHGUI_MultiEditDlg::SMESHGUI_MultiEditDlg
  84 // Purpose : Constructor
  85 //=======================================================================
  86 SMESHGUI_MultiEditDlg::SMESHGUI_MultiEditDlg( QWidget*              theParent,
  87                                               SALOME_Selection*     theSelection,
  88                                               const int             theMode,
  89                                               const char*           theName )
  90 : QDialog( theParent, theName, false,
  91            WStyle_Customize | WStyle_NormalBorder | WStyle_Title | WStyle_SysMenu )
  92 {
  93   myFilterDlg = 0;
  94   mySubmeshFilter = new SMESH_TypeFilter( SUBMESH );
  95   myGroupFilter = new SMESH_TypeFilter( GROUP );
  96
  97   myFilterType = theMode;
  98   QVBoxLayout* aDlgLay = new QVBoxLayout( this, MARGIN, SPACING );
  99
 100   QFrame* aMainFrame = createMainFrame  ( this );
 101   QFrame* aBtnFrame  = createButtonFrame( this );
 102
 103   aDlgLay->addWidget( aMainFrame );
 104   aDlgLay->addWidget( aBtnFrame );
 105
 106   aDlgLay->setStretchFactor( aMainFrame, 1 );
 107   aDlgLay->setStretchFactor( aBtnFrame, 0 );
 108   Init( theSelection ) ;
 109 }
 110
 111 //=======================================================================
 112 // name    : SMESHGUI_MultiEditDlg::createMainFrame
 113 // Purpose : Create frame containing dialog's input fields
 114 //=======================================================================
 115 QFrame* SMESHGUI_MultiEditDlg::createMainFrame( QWidget* theParent )
 116 {
 117   QGroupBox* aMainGrp = new QGroupBox( 1, Qt::Horizontal, theParent );
 118   aMainGrp->setFrameStyle( QFrame::NoFrame );
 119   aMainGrp->setInsideMargin( 0 );
 120
 121   QPixmap aPix( QAD_Desktop::getResourceManager()->loadPixmap( "SMESH",tr( "ICON_SELECT" ) ) );
 122
 123   // "Selected cells" group
 124   mySelGrp = new QGroupBox( 1, Qt::Horizontal,  aMainGrp );
 125   QFrame* aFrame = new QFrame( mySelGrp );
 126
 127   myListBox = new QListBox( aFrame );
 128   myListBox->setSelectionMode( QListBox::Extended );
 129   myListBox->setSizePolicy( QSizePolicy( QSizePolicy::MinimumExpanding, QSizePolicy::MinimumExpanding) );
 130 //  myListBox->setColumnMode( QListBox::FitToHeight );
 131
 132   myFilterBtn = new QPushButton( tr( "FILTER" )   , aFrame );
 133   myAddBtn    = new QPushButton( tr( "ADD" )      , aFrame );
 134   myRemoveBtn = new QPushButton( tr( "REMOVE" )   , aFrame );
 135   mySortBtn   = new QPushButton( tr( "SORT_LIST" ), aFrame );
 136
 137   QGridLayout* aLay = new QGridLayout( aFrame, 5, 2, 0, 5 );
 138   aLay->addMultiCellWidget( myListBox, 0, 4, 0, 0 );
 139   aLay->addWidget( myFilterBtn, 0, 1 );
 140   aLay->addWidget( myAddBtn, 1, 1 );
 141   aLay->addWidget( myRemoveBtn, 2, 1 );
 142   aLay->addWidget( mySortBtn, 3, 1 );
 143
 144   QSpacerItem* aSpacer = new QSpacerItem( 0, 0, QSizePolicy::Minimum, QSizePolicy::Expanding );
 145   aLay->addItem( aSpacer, 4, 1 );
 146
 147   myToAllChk = new QCheckBox( tr( "TO_ALL" ), mySelGrp );
 148
 149   // "Select from" group
 150   QGroupBox* aGrp = new QGroupBox( 3, Qt::Horizontal, tr( "SELECT_FROM" ), aMainGrp );
 151
 152   mySubmeshChk = new QCheckBox( tr( "SMESH_SUBMESH" ), aGrp );
 153   mySubmeshBtn = new QPushButton( aGrp );
 154   mySubmesh = new QLineEdit( aGrp );
 155   mySubmeshBtn->setPixmap( aPix );
 156
 157   myGroupChk = new QCheckBox( tr( "GROUP" ), aGrp );
 158   myGroupBtn = new QPushButton( aGrp );
 159   myGroup = new QLineEdit( aGrp );
 160   myGroupBtn->setPixmap( aPix );
 161
 162   return aMainGrp;
 163 }
 164
 165 //=======================================================================
 166 // name    : SMESHGUI_MultiEditDlg::createButtonFrame
 167 // Purpose : Create frame containing buttons
 168 //=======================================================================
 169 QFrame* SMESHGUI_MultiEditDlg::createButtonFrame( QWidget* theParent )
 170 {
 171   QFrame* aFrame = new QFrame( theParent );
 172   aFrame->setFrameStyle( QFrame::Box | QFrame::Sunken );
 173
 174   myOkBtn     = new QPushButton( tr( "SMESH_BUT_OK"    ), aFrame );
 175   myApplyBtn  = new QPushButton( tr( "SMESH_BUT_APPLY" ), aFrame );
 176   myCloseBtn  = new QPushButton( tr( "SMESH_BUT_CLOSE" ), aFrame );
 177
 178   QSpacerItem* aSpacer = new QSpacerItem( 0, 0, QSizePolicy::Expanding, QSizePolicy::Minimum );
 179
 180   QHBoxLayout* aLay = new QHBoxLayout( aFrame, MARGIN, SPACING );
 181
 182   aLay->addWidget( myOkBtn );
 183   aLay->addWidget( myApplyBtn );
 184   aLay->addItem( aSpacer);
 185   aLay->addWidget( myCloseBtn );
 186
 187   return aFrame;
 188 }
 189
 190 //=======================================================================
 191 // name    : SMESHGUI_MultiEditDlg::isValid
 192 // Purpose : Verify validity of input data
 193 //=======================================================================
 194 bool SMESHGUI_MultiEditDlg::isValid( const bool /*theMess*/ ) const
 195 {
 196   return (!myMesh->_is_nil() &&
 197           (myListBox->count() > 0 || (myToAllChk->isChecked() && myActor)));
 198 }
 199
 200 //=======================================================================
 201 // name    : SMESHGUI_MultiEditDlg::~SMESHGUI_MultiEditDlg
 202 // Purpose : Destructor
 203 //=======================================================================
 204 SMESHGUI_MultiEditDlg::~SMESHGUI_MultiEditDlg()
 205 {
 206   if ( myFilterDlg != 0 )
 207   {
 208     myFilterDlg->reparent( 0, QPoint() );
 209     delete myFilterDlg;
 210   }
 211 }
 212
 213 //=======================================================================
 214 // name    : SMESHGUI_MultiEditDlg::Init
 215 // Purpose : Init dialog fields, connect signals and slots, show dialog
 216 //=======================================================================
 217 void SMESHGUI_MultiEditDlg::Init( SALOME_Selection* theSelection )
 218 {
 219   SMESHGUI* aSMESHGUI = SMESHGUI::GetSMESHGUI();
 220   mySelection = theSelection;
 221   aSMESHGUI->SetActiveDialogBox( ( QDialog* )this ) ;
 222   myListBox->clear();
 223   myIds.Clear();
 224   myBusy = false;
 225   myActor = 0;
 226   emit ListContensChanged();
 227
 228   // main buttons
 229   connect( myOkBtn,    SIGNAL( clicked() ), SLOT( onOk() ) );
 230   connect( myCloseBtn, SIGNAL( clicked() ), SLOT( onClose() ) ) ;
 231   connect( myApplyBtn, SIGNAL( clicked() ), SLOT( onApply() ) );
 232
 233   // selection and SMESHGUI
 234   connect( mySelection, SIGNAL( currentSelectionChanged() ), SLOT( onSelectionDone() ) );
 235   connect( aSMESHGUI, SIGNAL( SignalDeactivateActiveDialog() ), SLOT( onDeactivate() ) );
 236   connect( aSMESHGUI, SIGNAL( SignalCloseAllDialogs() ), SLOT( onClose() ) );
 237
 238   // dialog controls
 239   connect( myFilterBtn, SIGNAL( clicked() ), SLOT( onFilterBtn()   ) );
 240   connect( myAddBtn   , SIGNAL( clicked() ), SLOT( onAddBtn()      ) );
 241   connect( myRemoveBtn, SIGNAL( clicked() ), SLOT( onRemoveBtn()   ) );
 242   connect( mySortBtn  , SIGNAL( clicked() ), SLOT( onSortListBtn() ) );
 243
 244   connect( mySubmeshChk, SIGNAL( stateChanged( int ) ), SLOT( onSubmeshChk() ) );
 245   connect( myGroupChk  , SIGNAL( stateChanged( int ) ), SLOT( onGroupChk()   ) );
 246   connect( myToAllChk  , SIGNAL( stateChanged( int ) ), SLOT( onToAllChk()   ) );
 247
 248   connect( myListBox, SIGNAL( selectionChanged() ), SLOT( onListSelectionChanged() ) );
 249
 250   onSelectionDone();
 251
 252   // set selection mode
 253   setSelectionMode();
 254   updateButtons();
 255 }
 256
 257 //=======================================================================
 258 // name    : SMESHGUI_MultiEditDlg::onOk
 259 // Purpose : SLOT called when "Ok" button pressed.
 260 //           Assign filters VTK viewer and close dialog
 261 //=======================================================================
 262 void SMESHGUI_MultiEditDlg::onOk()
 263 {
 264   if ( onApply() )
 265     onClose();
 266 }
 267
 268 //=======================================================================
 269 // name    : SMESHGUI_MultiEditDlg::getIds
 270 // Purpose : Retrive identifiers from list box
 271 //=======================================================================
 272 SMESH::long_array_var SMESHGUI_MultiEditDlg::getIds()
 273 {
 274   SMESH::long_array_var anIds = new SMESH::long_array;
 275
 276   if ( myToAllChk->isChecked() )
 277   {
 278     myIds.Clear();
 279     if ( myActor != 0 )
 280     {
 281       TVisualObjPtr aVisualObj = myActor->GetObject();
 282       vtkUnstructuredGrid* aGrid = aVisualObj->GetUnstructuredGrid();
 283       if ( aGrid != 0 )
 284       {
 285         for ( int i = 0, n = aGrid->GetNumberOfCells(); i < n; i++ )
 286         {
 287           vtkCell* aCell = aGrid->GetCell( i );
 288           if ( aCell != 0 )
 289           {
 290             int nbNodes = aCell->GetNumberOfPoints();
 291             if ( nbNodes == 3 && myFilterType == SMESHGUI_TriaFilter  ||
 292                  nbNodes == 4 && myFilterType == SMESHGUI_QuadFilter ||
 293                 ( nbNodes == 4 || nbNodes ==  3 ) && myFilterType == SMESHGUI_UnknownFilter )
 294             {
 295               int anObjId = aVisualObj->GetElemObjId( i );
 296               myIds.Add( anObjId );
 297             }
 298           }
 299         }
 300       }
 301     }
 302   }
 303
 304   anIds->length( myIds.Extent() );
 305   TColStd_MapIteratorOfMapOfInteger anIter( myIds );
 306   for ( int i = 0; anIter.More(); anIter.Next()  )
 307   {
 308     anIds[ i++ ] = anIter.Key();
 309   }
 310   return anIds._retn();
 311 }
 312
 313 //=======================================================================
 314 // name    : SMESHGUI_MultiEditDlg::onClose
 315 // Purpose : SLOT called when "Close" button pressed. Close dialog
 316 //=======================================================================
 317 void SMESHGUI_MultiEditDlg::onClose()
 318 {
 319   QAD_Application::getDesktop()->SetSelectionMode( ActorSelection );
 320   disconnect( mySelection, 0, this, 0 );
 321   disconnect( SMESHGUI::GetSMESHGUI(), 0, this, 0 );
 322   SMESHGUI::GetSMESHGUI()->ResetState();
 323
 324   SMESH::RemoveFilter(SMESHGUI_EdgeFilter);
 325   SMESH::RemoveFilter(SMESHGUI_FaceFilter);
 326   SMESH::RemoveFilter(SMESHGUI_QuadFilter);
 327   SMESH::RemoveFilter(SMESHGUI_TriaFilter);
 328   SMESH::SetPickable();
 329
 330   mySelection->ClearIObjects();
 331   mySelection->ClearFilters();
 332
 333   reject();
 334 }
 335
 336 //=======================================================================
 337 // name    : SMESHGUI_MultiEditDlg::onSelectionDone
 338 // Purpose : SLOT called when selection changed
 339 //=======================================================================
 340 void SMESHGUI_MultiEditDlg::onSelectionDone()
 341 {
 342   if ( myBusy || !isEnabled() ) return;
 343   myBusy = true;
 344
 345   int nbSel = mySelection->IObjectCount();
 346   myListBox->clearSelection();
 347
 348   if ( mySubmeshChk->isChecked() || myGroupChk->isChecked() )
 349   {
 350     QLineEdit* aNameEdit = mySubmeshChk->isChecked() ? mySubmesh : myGroup;
 351     int nbSel = mySelection->IObjectCount();
 352     if ( nbSel == 1 )
 353     {
 354       Handle(SALOME_InteractiveObject) anIO = mySelection->firstIObject();
 355       anIO.IsNull() ? aNameEdit->clear() : aNameEdit->setText( anIO->getName() );
 356
 357       if ( mySubmeshChk->isChecked() )
 358       {
 359         SMESH::SMESH_subMesh_var aSubMesh =
 360           SMESH::IObjectToInterface<SMESH::SMESH_subMesh>( anIO );
 361         if ( !aSubMesh->_is_nil() )
 362           myMesh = aSubMesh->GetFather();
 363       }
 364       else
 365       {
 366         SMESH::SMESH_Group_var aGroup =
 367           SMESH::IObjectToInterface<SMESH::SMESH_Group>( anIO );
 368         if ( !aGroup->_is_nil() )
 369           myMesh = aGroup->GetMesh();
 370       }
 371     }
 372     else if ( nbSel > 1 )
 373     {
 374       QString aStr = mySubmeshChk->isChecked() ?
 375         tr( "SMESH_SUBMESH_SELECTED" ) : tr( "SMESH_GROUP_SELECTED" );
 376       aNameEdit->setText( aStr.arg( nbSel ) );
 377     }
 378     else
 379       aNameEdit->clear();
 380   }
 381   else if ( nbSel == 1 )
 382   {
 383     QString aListStr = "";
 384     int aNbItems = SMESH::GetNameOfSelectedElements(mySelection, aListStr);
 385     if ( aNbItems > 0 )
 386     {
 387       QStringList anElements = QStringList::split(" ", aListStr);
 388       QListBoxItem* anItem = 0;
 389       for ( QStringList::iterator it = anElements.begin(); it != anElements.end(); ++it)
 390       {
 391         anItem = myListBox->findItem( *it, Qt::ExactMatch );
 392         if (anItem) myListBox->setSelected( anItem, true );
 393       }
 394     }
 395
 396     myMesh = SMESH::GetMeshByIO( mySelection->firstIObject() );
 397   }
 398
 399   if ( nbSel == 1 ) {
 400     myActor = SMESH::FindActorByEntry(mySelection->firstIObject()->getEntry());
 401     //myActor = SMESH::FindActorByObject( myMesh );
 402     VTKViewer_InteractorStyleSALOME* aStyle = SMESH::GetInteractorStyle();
 403     Handle(VTKViewer_Filter) aFilter1 = aStyle->GetFilter( myFilterType );
 404     Handle(VTKViewer_Filter) aFilter2 = aStyle->GetFilter( SMESHGUI_FaceFilter );
 405     if ( !aFilter1.IsNull() )
 406       aFilter1->SetActor( myActor );
 407     if ( !aFilter2.IsNull() )
 408       aFilter2->SetActor( myActor );
 409     if ( myActor )
 410       SMESH::SetPickable(myActor);
 411   }
 412   myBusy = false;
 413
 414   updateButtons();
 415 }
 416
 417 //=======================================================================
 418 // name    : SMESHGUI_MultiEditDlg::onDeactivate
 419 // Purpose : SLOT called when dialog must be deativated
 420 //=======================================================================
 421 void SMESHGUI_MultiEditDlg::onDeactivate()
 422 {
 423   setEnabled( false );
 424 }
 425
 426 //=======================================================================
 427 // name    : SMESHGUI_MultiEditDlg::enterEvent
 428 // Purpose : Event filter
 429 //=======================================================================
 430 void SMESHGUI_MultiEditDlg::enterEvent( QEvent* )
 431 {
 432   if ( !isEnabled() ) {
 433     SMESHGUI::GetSMESHGUI()->EmitSignalDeactivateDialog();
 434     setEnabled( true );
 435     setSelectionMode();
 436   }
 437 }
 438
 439
 440 //=================================================================================
 441 // function : closeEvent()
 442 // purpose  :
 443 //=================================================================================
 444 void SMESHGUI_MultiEditDlg::closeEvent( QCloseEvent* e )
 445 {
 446   onClose() ;
 447 }
 448 //=======================================================================
 449 //function : hideEvent
 450 //purpose  : caused by ESC key
 451 //=======================================================================
 452
 453 void SMESHGUI_MultiEditDlg::hideEvent ( QHideEvent * e )
 454 {
 455   if ( !isMinimized() )
 456     onClose();
 457 }
 458
 459 //=======================================================================
 460 // name    : SMESHGUI_MultiEditDlg::onFilterBtn
 461 // Purpose : SLOT. Called when "Filter" button pressed.
 462 //           Start "Selection filters" dialog
 463 //=======================================================================
 464 void SMESHGUI_MultiEditDlg::onFilterBtn()
 465 {
 466   if ( myFilterDlg == 0 )
 467   {
 468     myFilterDlg = new SMESHGUI_FilterDlg( (QWidget*)parent(), SMESH::FACE );
 469     connect( myFilterDlg, SIGNAL( Accepted() ), SLOT( onFilterAccepted() ) );
 470   }
 471   else
 472     myFilterDlg->Init( SMESH::FACE );
 473
 474   myFilterDlg->SetSelection( mySelection );
 475   myFilterDlg->SetMesh( myMesh );
 476   myFilterDlg->SetSourceWg( myListBox );
 477
 478   myFilterDlg->show();
 479 }
 480
 481 //=================================================================================
 482 // function : onFilterAccepted()
 483 // purpose  : SLOT. Called when Filter dlg closed with OK button.
 484 //            Uncheck "Select submesh" and "Select group" checkboxes
 485 //=================================================================================
 486 void SMESHGUI_MultiEditDlg::onFilterAccepted()
 487 {
 488   myIds.Clear();
 489   for ( int i = 0, n = myListBox->count(); i < n; i++ )
 490     myIds.Add( myListBox->text( i ).toInt() );
 491
 492   emit ListContensChanged();
 493
 494   if ( mySubmeshChk->isChecked() || myGroupChk->isChecked() )
 495   {
 496     mySubmeshChk->blockSignals( true );
 497     myGroupChk->blockSignals( true );
 498     mySubmeshChk->setChecked( false );
 499     myGroupChk->setChecked( false );
 500     mySubmeshChk->blockSignals( false );
 501     myGroupChk->blockSignals( false );
 502   }
 503   updateButtons();
 504 }
 505
 506 //=======================================================================
 507 // name    : SMESHGUI_MultiEditDlg::onAddBtn
 508 // Purpose : Verify whether Id of element satisfies to filters from viewer
 509 //=======================================================================
 510 bool SMESHGUI_MultiEditDlg::isIdValid( const int theId ) const
 511 {
 512   VTKViewer_InteractorStyleSALOME* aStyle = SMESH::GetInteractorStyle();
 513   Handle(SMESHGUI_Filter) aFilter1 =
 514     Handle(SMESHGUI_Filter)::DownCast( aStyle->GetFilter( myFilterType ) );
 515   Handle(SMESHGUI_Filter) aFilter2 =
 516     Handle(SMESHGUI_Filter)::DownCast( aStyle->GetFilter( SMESHGUI_FaceFilter ) );
 517   return ( aFilter1.IsNull() || aFilter1->IsObjValid( theId ) ) &&
 518          ( aFilter2.IsNull() || aFilter2->IsObjValid( theId ) );
 519 }
 520
 521 //=======================================================================
 522 // name    : SMESHGUI_MultiEditDlg::onAddBtn
 523 // Purpose : SLOT. Called when "Add" button pressed.
 524 //           Add selected in viewer entities in list box
 525 //=======================================================================
 526 void SMESHGUI_MultiEditDlg::onAddBtn()
 527 {
 528   int nbSelected = mySelection->IObjectCount();
 529   if ( nbSelected == 0 )
 530     return;
 531
 532   TColStd_IndexedMapOfInteger toBeAdded;
 533
 534   if ( !mySubmeshChk->isChecked() && !myGroupChk->isChecked() )
 535   {
 536     if ( nbSelected == 1 )
 537       SMESH::GetSelected( mySelection, toBeAdded );
 538   }
 539   else if ( mySubmeshChk->isChecked() )
 540   {
 541     SALOME_ListIteratorOfListIO anIter( mySelection->StoredIObjects() );
 542     for ( ; anIter.More(); anIter.Next() )
 543     {
 544       SMESH::SMESH_subMesh_var aSubMesh = SMESH::IObjectToInterface<SMESH::SMESH_subMesh>( anIter.Value() );
 545       if ( !aSubMesh->_is_nil() )
 546       {
 547         if ( aSubMesh->GetFather()->GetId() == myMesh->GetId() )
 548         {
 549           SMESH::long_array_var anIds = aSubMesh->GetElementsId();
 550           for ( int i = 0, n = anIds->length(); i < n; i++ )
 551           {
 552             if ( isIdValid( anIds[ i ] ) )
 553               toBeAdded.Add( anIds[ i ] );
 554           }
 555         }
 556       }
 557     }
 558   }
 559   else if ( myGroupChk->isChecked() )
 560   {
 561     SALOME_ListIteratorOfListIO anIter( mySelection->StoredIObjects() );
 562     for ( ; anIter.More(); anIter.Next() )
 563     {
 564       SMESH::SMESH_Group_var aGroup = SMESH::IObjectToInterface<SMESH::SMESH_Group>( anIter.Value() );
 565       if ( !aGroup->_is_nil() && aGroup->GetType() == SMESH::FACE )
 566       {
 567         if ( aGroup->GetMesh()->GetId() == myMesh->GetId() )
 568         {
 569           SMESH::long_array_var anIds = aGroup->GetListOfID();
 570           for ( int i = 0, n = anIds->length(); i < n; i++ )
 571           {
 572             if ( isIdValid( anIds[ i ] ) )
 573               toBeAdded.Add( anIds[ i ] );
 574           }
 575         }
 576       }
 577     }
 578   }
 579
 580   myBusy = true;
 581   bool isGroupOrSubmesh = ( mySubmeshChk->isChecked() || myGroupChk->isChecked() );
 582   mySubmeshChk->setChecked( false );
 583   myGroupChk->setChecked( false );
 584   for( int i = 1; i <= toBeAdded.Extent(); i++ )
 585     if ( myIds.Add( toBeAdded(i) ) ) {
 586       QListBoxItem * item = new QListBoxText( QString( "%1" ).arg( toBeAdded(i) ));
 587       myListBox->insertItem( item );
 588       myListBox->setSelected( item, true );
 589     }
 590   myBusy = false;
 591
 592   emit ListContensChanged();
 593
 594   if ( isGroupOrSubmesh )
 595     onListSelectionChanged();
 596
 597   updateButtons();
 598 }
 599
 600 //=======================================================================
 601 // name    : SMESHGUI_MultiEditDlg::updateButtons
 602 // Purpose : Enable/disable buttons of dialog in accordance with current state
 603 //=======================================================================
 604 void SMESHGUI_MultiEditDlg::updateButtons()
 605 {
 606   bool isOk = isValid(false);
 607   myOkBtn->setEnabled( isOk );
 608   myApplyBtn->setEnabled( isOk );
 609
 610   bool isListBoxNonEmpty = myListBox->count() > 0;
 611   bool isToAll = myToAllChk->isChecked();
 612   myFilterBtn->setEnabled( !isToAll );
 613   myRemoveBtn->setEnabled( isListBoxNonEmpty && !isToAll );
 614   mySortBtn->setEnabled( isListBoxNonEmpty &&!isToAll );
 615
 616   if ( isToAll ||
 617        myMesh->_is_nil() ||
 618        mySelection->IObjectCount() != 1 ||
 619        (SMESH::IObjectToInterface<SMESH::SMESH_subMesh>( mySelection->firstIObject() )->_is_nil() &&
 620         SMESH::IObjectToInterface<SMESH::SMESH_Group>( mySelection->firstIObject() )->_is_nil() &&
 621         SMESH::IObjectToInterface<SMESH::SMESH_Mesh>( mySelection->firstIObject() )->_is_nil()) )
 622     myAddBtn->setEnabled( false );
 623   else
 624     myAddBtn->setEnabled( true );
 625
 626   mySubmeshChk->setEnabled( !isToAll );
 627   mySubmeshBtn->setEnabled( mySubmeshChk->isChecked() );
 628   mySubmesh->setEnabled( mySubmeshChk->isChecked() );
 629
 630   myGroupChk->setEnabled( !isToAll );
 631   myGroupBtn->setEnabled( myGroupChk->isChecked() );
 632   myGroup->setEnabled( myGroupChk->isChecked() );
 633
 634   if ( !mySubmeshChk->isChecked() )
 635     mySubmesh->clear();
 636   if ( !myGroupChk->isChecked() )
 637     myGroup->clear();
 638
 639 }
 640
 641 //=======================================================================
 642 // name    : SMESHGUI_MultiEditDlg::onRemoveBtn
 643 // Purpose : SLOT. Called when "Remove" button pressed.
 644 //           Remove selected in list box entities
 645 //=======================================================================
 646 void SMESHGUI_MultiEditDlg::onRemoveBtn()
 647 {
 648   myBusy = true;
 649
 650   for ( int i = 0, n = myListBox->count(); i < n; i++ )
 651   {
 652     for ( int i = myListBox->count(); i > 0; i--) {
 653       if ( myListBox->isSelected( i - 1 ) )
 654       {
 655         int anId = myListBox->text( i - 1 ).toInt();
 656         myIds.Remove( anId );
 657         myIds.Remove( anId );
 658               myListBox->removeItem( i-1 );
 659       }
 660     }
 661   }
 662   myBusy = false;
 663
 664   emit ListContensChanged();
 665   updateButtons();
 666 }
 667
 668 //=======================================================================
 669 // name    : SMESHGUI_MultiEditDlg::onSortListBtn
 670 // Purpose : SLOT. Called when "Sort list" button pressed.
 671 //           Sort entities of list box
 672 //=======================================================================
 673 void SMESHGUI_MultiEditDlg::onSortListBtn()
 674 {
 675   myBusy = true;
 676
 677   int i, k = myListBox->count();
 678   if ( k > 0 )
 679   {
 680     QStringList aSelected;
 681     std::vector<int> anArray( k );
 682     QListBoxItem* anItem;
 683     for ( anItem = myListBox->firstItem(), i = 0; anItem != 0; anItem = anItem->next(), i++)
 684     {
 685       anArray[ i ] = anItem->text().toInt();
 686       if ( anItem->isSelected() )
 687         aSelected.append( anItem->text() );
 688     }
 689
 690     std::sort( anArray.begin(), anArray.end() );
 691     myListBox->clear();
 692     for ( i = 0; i < k; i++ )
 693       myListBox->insertItem( QString::number( anArray[ i ] ) );
 694
 695     for ( QStringList::iterator it = aSelected.begin(); it != aSelected.end(); ++it )
 696     {
 697       anItem = myListBox->findItem( *it, Qt::ExactMatch );
 698       if ( anItem )
 699         myListBox->setSelected( anItem, true );
 700     }
 701   }
 702   myBusy = false;
 703 }
 704
 705 //=======================================================================
 706 // name    : SMESHGUI_MultiEditDlg::onListSelectionChanged
 707 // Purpose : SLOT. Called when selection in list box changed.
 708 //           Highlight in selected entities
 709 //=======================================================================
 710 void SMESHGUI_MultiEditDlg::onListSelectionChanged()
 711 {
 712   if ( myActor == 0 || myBusy )
 713     return;
 714
 715   if ( mySubmeshChk->isChecked() || myGroupChk->isChecked() )
 716     return;
 717
 718   SMESH_Actor * anActor = SMESH::FindActorByObject( myMesh );
 719   if ( !anActor )
 720     anActor = myActor;
 721   TVisualObjPtr anObj = anActor->GetObject();
 722
 723   TColStd_MapOfInteger anIndexes;
 724   for ( QListBoxItem* anItem = myListBox->firstItem(); anItem != 0; anItem = anItem->next() )
 725   {
 726     if ( anItem->isSelected() )
 727     {
 728       int anId = anItem->text().toInt();
 729       if ( anObj->GetElemVTKId( anId ) >= 0 ) // avoid exception in hilight
 730         anIndexes.Add(anId);
 731     }
 732   }
 733
 734   mySelection->ClearIObjects();
 735   mySelection->AddOrRemoveIndex( anActor->getIO(), anIndexes, false, false );
 736   mySelection->AddIObject( anActor->getIO() );
 737 }
 738
 739 //=======================================================================
 740 // name    : SMESHGUI_MultiEditDlg::onSubmeshChk
 741 // Purpose : SLOT. Called when state of "SubMesh" check box changed.
 742 //           Activate/deactivate selection of submeshes
 743 //=======================================================================
 744 void SMESHGUI_MultiEditDlg::onSubmeshChk()
 745 {
 746   bool isChecked = mySubmeshChk->isChecked();
 747   mySubmeshBtn->setEnabled( isChecked );
 748   mySubmesh->setEnabled( isChecked );
 749   if ( !isChecked )
 750     mySubmesh->clear();
 751   if ( isChecked && myGroupChk->isChecked() )
 752       myGroupChk->setChecked( false );
 753
 754   setSelectionMode();
 755 }
 756
 757 //=======================================================================
 758 // name    : SMESHGUI_MultiEditDlg::onGroupChk
 759 // Purpose : SLOT. Called when state of "Group" check box changed.
 760 //           Activate/deactivate selection of groupes
 761 //=======================================================================
 762 void SMESHGUI_MultiEditDlg::onGroupChk()
 763 {
 764   bool isChecked = myGroupChk->isChecked();
 765   myGroupBtn->setEnabled( isChecked );
 766   myGroup->setEnabled( isChecked );
 767   if ( !isChecked )
 768     myGroup->clear();
 769   if ( isChecked && mySubmeshChk->isChecked() )
 770       mySubmeshChk->setChecked( false );
 771
 772   setSelectionMode();
 773 }
 774
 775 //=======================================================================
 776 // name    : SMESHGUI_MultiEditDlg::onToAllChk
 777 // Purpose : SLOT. Called when state of "Apply to all" check box changed.
 778 //           Activate/deactivate selection
 779 //=======================================================================
 780 void SMESHGUI_MultiEditDlg::onToAllChk()
 781 {
 782   bool isChecked = myToAllChk->isChecked();
 783
 784   if ( isChecked )
 785     myListBox->clear();
 786
 787   myIds.Clear();
 788
 789   emit ListContensChanged();
 790
 791   updateButtons();
 792   setSelectionMode();
 793
 794   if ( myActor )
 795     mySelection->AddIObject( myActor->getIO(), true );
 796 }
 797
 798
 799 //=======================================================================
 800 // name    : SMESHGUI_MultiEditDlg::setSelectionMode
 801 // Purpose : Set selection mode
 802 //=======================================================================
 803 void SMESHGUI_MultiEditDlg::setSelectionMode()
 804 {
 805   mySelection->ClearIObjects();
 806   mySelection->ClearFilters();
 807
 808   if ( mySubmeshChk->isChecked() )
 809   {
 810     QAD_Application::getDesktop()->SetSelectionMode( ActorSelection, true );
 811     mySelection->AddFilter( mySubmeshFilter );
 812   }
 813   else if ( myGroupChk->isChecked() )
 814   {
 815     QAD_Application::getDesktop()->SetSelectionMode( ActorSelection, true );
 816     mySelection->AddFilter( myGroupFilter );
 817   }
 818   else
 819   {
 820     QAD_Application::getDesktop()->SetSelectionMode( FaceSelection, true );
 821     if ( myFilterType == SMESHGUI_TriaFilter )
 822       SMESH::SetFilter( new SMESHGUI_TriangleFilter() );
 823     else if ( myFilterType == SMESHGUI_QuadFilter )
 824       SMESH::SetFilter( new SMESHGUI_QuadrangleFilter() );
 825   }
 826 }
 827
 828 //=======================================================================
 829 // name    : SMESHGUI_MultiEditDlg::onApply
 830 // Purpose : SLOT. Called when "Apply" button clicked.
 831 //=======================================================================
 832 bool SMESHGUI_MultiEditDlg::onApply()
 833 {
 834   if ( SMESHGUI::GetSMESHGUI()->ActiveStudyLocked() )
 835     return false;
 836   if ( !isValid( true ) )
 837     return false;
 838
 839   SMESH::SMESH_MeshEditor_var aMeshEditor = myMesh->GetMeshEditor();
 840   if ( aMeshEditor->_is_nil()  )
 841     return false;
 842
 843   SMESH::long_array_var anIds = getIds();
 844
 845   bool aResult = process( aMeshEditor, anIds.inout() );
 846   if ( aResult )
 847   {
 848     if ( myActor )
 849     {
 850       mySelection->ClearIObjects();
 851       SMESH::UpdateView();
 852       mySelection->AddIObject( myActor->getIO(), false );
 853     }
 854
 855     myListBox->clear();
 856     myIds.Clear();
 857     emit ListContensChanged();
 858
 859     updateButtons();
 860   }
 861
 862   return aResult;
 863 }
 864
 865 /*
 866   Class       : SMESHGUI_ChangeOrientationDlg
 867   Description : Modification of orientation of faces
 868 */
 869
 870 SMESHGUI_ChangeOrientationDlg::SMESHGUI_ChangeOrientationDlg( QWidget*          theParent,
 871                                                               SALOME_Selection* theSelection,
 872                                                               const char*       theName )
 873 : SMESHGUI_MultiEditDlg( theParent, theSelection, SMESHGUI_UnknownFilter, theName )
 874 {
 875   setCaption( tr( "CAPTION" ) );
 876 }
 877
 878 SMESHGUI_ChangeOrientationDlg::~SMESHGUI_ChangeOrientationDlg()
 879 {
 880 }
 881
 882 bool SMESHGUI_ChangeOrientationDlg::process( SMESH::SMESH_MeshEditor_ptr theEditor,
 883                                              const SMESH::long_array&    theIds )
 884 {
 885   return theEditor->Reorient( theIds );
 886 }
 887
 888 /*
 889   Class       : SMESHGUI_UnionOfTrianglesDlg
 890   Description : Construction of quadrangles by automatic association of triangles
 891 */
 892
 893 SMESHGUI_UnionOfTrianglesDlg::SMESHGUI_UnionOfTrianglesDlg( QWidget*          theParent,
 894                                                             SALOME_Selection* theSelection,
 895                                                             const char*       theName )
 896 : SMESHGUI_MultiEditDlg( theParent, theSelection, SMESHGUI_TriaFilter, theName )
 897 {
 898   setCaption( tr( "CAPTION" ) );
 899 }
 900
 901 SMESHGUI_UnionOfTrianglesDlg::~SMESHGUI_UnionOfTrianglesDlg()
 902 {
 903 }
 904
 905 bool SMESHGUI_UnionOfTrianglesDlg::process( SMESH::SMESH_MeshEditor_ptr theEditor,
 906                                             const SMESH::long_array&    theIds )
 907 {
 908   return theEditor->TriToQuad(theIds, SMESH::NumericalFunctor::_nil(), 1. );
 909 }
 910
 911 /*
 912   Class       : SMESHGUI_CuttingOfQuadsDlg
 913   Description : Construction of quadrangles by automatic association of triangles
 914 */
 915
 916 SMESHGUI_CuttingOfQuadsDlg::SMESHGUI_CuttingOfQuadsDlg( QWidget*          theParent,
 917                                                         SALOME_Selection* theSelection,
 918                                                         const char*       theName )
 919 : SMESHGUI_MultiEditDlg( theParent, theSelection, SMESHGUI_QuadFilter, theName )
 920 {
 921
 922   setCaption( tr( "CAPTION" ) );
 923   myPreviewActor = 0;
 924
 925   myUseDiagChk = new QCheckBox( tr( "USE_DIAGONAL_2_4" ), mySelGrp );
 926   myPreviewChk = new QCheckBox( tr( "PREVIEW" ), mySelGrp );
 927
 928   connect( myPreviewChk, SIGNAL( stateChanged( int ) ), this, SLOT( onPreviewChk() ) );
 929   connect( myUseDiagChk, SIGNAL( stateChanged( int ) ), this, SLOT( onPreviewChk() ) );
 930   connect( this, SIGNAL( ListContensChanged() ), this, SLOT( onPreviewChk() ) );
 931 }
 932
 933 SMESHGUI_CuttingOfQuadsDlg::~SMESHGUI_CuttingOfQuadsDlg()
 934 {
 935 }
 936
 937 void SMESHGUI_CuttingOfQuadsDlg::onClose()
 938 {
 939   erasePreview();
 940   SMESHGUI_MultiEditDlg::onClose();
 941 }
 942
 943 bool SMESHGUI_CuttingOfQuadsDlg::process( SMESH::SMESH_MeshEditor_ptr theEditor,
 944                                           const SMESH::long_array&    theIds )
 945 {
 946   return theEditor->SplitQuad( theIds, !myUseDiagChk->isChecked() );
 947 }
 948
 949 void SMESHGUI_CuttingOfQuadsDlg::onPreviewChk()
 950 {
 951   myPreviewChk->isChecked() ? displayPreview() : erasePreview();
 952 }
 953
 954 void SMESHGUI_CuttingOfQuadsDlg::erasePreview()
 955 {
 956   if ( myPreviewActor == 0 )
 957     return;
 958
 959   if ( VTKViewer_ViewFrame* vf = SMESH::GetCurrentVtkView() )
 960   {
 961     vf->RemoveActor(myPreviewActor);
 962     vf->Repaint();
 963   }
 964   myPreviewActor->Delete();
 965   myPreviewActor = 0;
 966 }
 967
 968 void SMESHGUI_CuttingOfQuadsDlg::displayPreview()
 969 {
 970   if ( myActor == 0 )
 971     return;
 972
 973   if ( myPreviewActor != 0 )
 974     erasePreview();
 975
 976   // get Ids of elements
 977   SMESH::long_array_var anElemIds = getIds();
 978   if ( getIds()->length() == 0 )
 979     return;
 980
 981   SMDS_Mesh* aMesh = myActor->GetObject()->GetMesh();
 982   if ( aMesh == 0 )
 983     return;
 984
 985   bool isDiag24 = myUseDiagChk->isChecked();
 986
 987   //Create grid
 988   vtkUnstructuredGrid* aGrid = vtkUnstructuredGrid::New();
 989
 990   vtkIdType aNbCells = anElemIds->length() * 2;
 991   vtkIdType aCellsSize = 4 * aNbCells;
 992   vtkCellArray* aConnectivity = vtkCellArray::New();
 993   aConnectivity->Allocate( aCellsSize, 0 );
 994
 995   vtkPoints* aPoints = vtkPoints::New();
 996   aPoints->SetNumberOfPoints( anElemIds->length() * 4 );
 997
 998   vtkUnsignedCharArray* aCellTypesArray = vtkUnsignedCharArray::New();
 999   aCellTypesArray->SetNumberOfComponents( 1 );
1000   aCellTypesArray->Allocate( aNbCells * aCellTypesArray->GetNumberOfComponents() );
1001
1002   vtkIdList *anIdList = vtkIdList::New();
1003   anIdList->SetNumberOfIds( 3 );
1004
1005   TColStd_DataMapOfIntegerInteger anIdToVtk;
1006
1007   int aNodes[ 4 ];
1008   int nbPoints = -1;
1009   for ( int i = 0, n = anElemIds->length(); i < n; i++ )
1010   {
1011     const SMDS_MeshElement* anElem = aMesh->FindElement( anElemIds[ i ] );
1012     if ( anElem == 0 || anElem->NbNodes() != 4 )
1013       continue;
1014
1015     SMDS_ElemIteratorPtr anIter = anElem->nodesIterator();
1016     int k = 0;
1017     while( anIter->more() )
1018       if ( const SMDS_MeshNode* aNode = (SMDS_MeshNode*)anIter->next() )
1019       {
1020         if ( !anIdToVtk.IsBound( aNode->GetID() ) )
1021         {
1022           aPoints->SetPoint( ++nbPoints, aNode->X(), aNode->Y(), aNode->Z() );
1023           anIdToVtk.Bind( aNode->GetID(), nbPoints );
1024         }
1025
1026         aNodes[ k++ ] = aNode->GetID();
1027       }
1028
1029     if ( k != 4 )
1030       continue;
1031
1032     if ( !isDiag24 )
1033     {
1034       anIdList->SetId( 0, anIdToVtk( aNodes[ 0 ] ) );
1035       anIdList->SetId( 1, anIdToVtk( aNodes[ 1 ] ) );
1036       anIdList->SetId( 2, anIdToVtk( aNodes[ 2 ] ) );
1037       aConnectivity->InsertNextCell( anIdList );
1038       aCellTypesArray->InsertNextValue( VTK_TRIANGLE );
1039
1040       anIdList->SetId( 0, anIdToVtk( aNodes[ 2 ] ) );
1041       anIdList->SetId( 1, anIdToVtk( aNodes[ 3 ] ) );
1042       anIdList->SetId( 2, anIdToVtk( aNodes[ 0 ] ) );
1043       aConnectivity->InsertNextCell( anIdList );
1044       aCellTypesArray->InsertNextValue( VTK_TRIANGLE );
1045     }
1046     else
1047     {
1048       anIdList->SetId( 0, anIdToVtk( aNodes[ 1 ] ) );
1049       anIdList->SetId( 1, anIdToVtk( aNodes[ 2 ] ) );
1050       anIdList->SetId( 2, anIdToVtk( aNodes[ 3 ] ) );
1051       aConnectivity->InsertNextCell( anIdList );
1052       aCellTypesArray->InsertNextValue( VTK_TRIANGLE );
1053
1054       anIdList->SetId( 0, anIdToVtk( aNodes[ 3 ] ) );
1055       anIdList->SetId( 1, anIdToVtk( aNodes[ 0 ] ) );
1056       anIdList->SetId( 2, anIdToVtk( aNodes[ 1 ] ) );
1057       aConnectivity->InsertNextCell( anIdList );
1058       aCellTypesArray->InsertNextValue( VTK_TRIANGLE );
1059     }
1060   }
1061
1062   vtkIntArray* aCellLocationsArray = vtkIntArray::New();
1063   aCellLocationsArray->SetNumberOfComponents( 1 );
1064   aCellLocationsArray->SetNumberOfTuples( aNbCells );
1065
1066   aConnectivity->InitTraversal();
1067   for( vtkIdType idType = 0, *pts, npts; aConnectivity->GetNextCell( npts, pts ); idType++ )
1068     aCellLocationsArray->SetValue( idType, aConnectivity->GetTraversalLocation( npts ) );
1069
1070   aGrid->SetPoints( aPoints );
1071   aGrid->SetCells( aCellTypesArray, aCellLocationsArray,aConnectivity );
1072
1073   // Create and display actor
1074   vtkDataSetMapper* aMapper = vtkDataSetMapper::New();
1075   aMapper->SetInput( aGrid );
1076
1077   myPreviewActor = SALOME_Actor::New();
1078   myPreviewActor->PickableOff();
1079   myPreviewActor->SetMapper( aMapper );
1080
1081   vtkProperty* aProp = vtkProperty::New();
1082   aProp->SetRepresentationToWireframe();
1083   aProp->SetColor( 250, 0, 250 );
1084   aProp->SetLineWidth( myActor->GetLineWidth() + 1 );
1085   myPreviewActor->SetProperty( aProp );
1086
1087   SMESH::GetCurrentVtkView()->AddActor( myPreviewActor );
1088   SMESH::GetCurrentVtkView()->Repaint();
1089
1090   aProp->Delete();
1091   aPoints->Delete();
1092   aConnectivity->Delete();
1093   aGrid->Delete();
1094   aMapper->Delete();
1095   anIdList->Delete();
1096   aCellTypesArray->Delete();
1097   aCellLocationsArray->Delete();
1098 }
1099
1100
1101
1102
1103
1104
1105
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121