src/StdMeshersGUI/StdMeshersGUI_QuadrangleParamWdg.cxx

   1 // Copyright (C) 2007-2015  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19 //  File   : StdMeshersGUI_QuadrangleParamWdg.cxx
  20 //  Author : Open CASCADE S.A.S. (jfa)
  21 //  SMESH includes
  22
  23 #include "StdMeshersGUI_QuadrangleParamWdg.h"
  24
  25 #include "SMESHGUI.h"
  26 #include "SMESHGUI_SpinBox.h"
  27 #include "SMESHGUI_Utils.h"
  28 #include "StdMeshersGUI_SubShapeSelectorWdg.h"
  29
  30 #include <GEOMBase.h>
  31 #include <LightApp_SelectionMgr.h>
  32 #include <SALOME_ListIO.hxx>
  33 #include <SUIT_ResourceMgr.h>
  34
  35 // Qt includes
  36 #include <QButtonGroup>
  37 #include <QFrame>
  38 #include <QGridLayout>
  39 #include <QGroupBox>
  40 #include <QLabel>
  41 #include <QLineEdit>
  42 #include <QListWidget>
  43 #include <QListWidgetItem>
  44 #include <QPushButton>
  45 #include <QRadioButton>
  46 #include <QTreeWidget>
  47 #include <QVBoxLayout>
  48
  49 // IDL includes
  50 #include <SALOMEconfig.h>
  51 #include CORBA_CLIENT_HEADER(SMESH_BasicHypothesis)
  52 #include CORBA_CLIENT_HEADER(GEOM_Gen)
  53
  54
  55 #define SPACING 6
  56 #define MARGIN 11
  57
  58 enum { TAB_TRANSITION, TAB_VERTEX, TAB_ENF_POINTS };
  59
  60 //================================================================================
  61 // function : Constructor
  62 // purpose  :
  63 //================================================================================
  64
  65 StdMeshersGUI_QuadrangleParamCreator::StdMeshersGUI_QuadrangleParamCreator(const QString& aHypType)
  66   : StdMeshersGUI_StdHypothesisCreator( aHypType )
  67 {
  68 }
  69
  70 //=======================================================================
  71 //function : helpPage
  72 //purpose  :
  73 //=======================================================================
  74
  75 QString  StdMeshersGUI_QuadrangleParamCreator::helpPage() const
  76 {
  77   return "a2d_meshing_hypo_page.html#hypo_quad_params_anchor";
  78 }
  79
  80 //=======================================================================
  81 //function : buildFrame
  82 //purpose  :
  83 //=======================================================================
  84
  85 QFrame*  StdMeshersGUI_QuadrangleParamCreator::buildFrame()
  86 {
  87   QFrame* fr = new QFrame();
  88
  89   QGridLayout* lay = new QGridLayout( fr );
  90   lay->setMargin( MARGIN );
  91   lay->setSpacing( SPACING );
  92   int row = 0;
  93
  94   myName = 0;
  95   if ( isCreation() )
  96   {
  97     myName = new QLineEdit( fr );
  98     QLabel* nameLab = new QLabel( tr("SMESH_NAME"));
  99     lay->addWidget( nameLab, row, 0 );
 100     lay->addWidget( myName,  row, 1 );
 101     ++row;
 102   }
 103
 104   // Transition type
 105
 106   myTypeWdg = new StdMeshersGUI_QuadrangleParamWdg( fr );
 107
 108   // Vertexes
 109
 110   myVertexSelWdg = new StdMeshersGUI_SubShapeSelectorWdg( fr, TopAbs_VERTEX );
 111   myVertexSelWdg->layout()->setMargin( MARGIN );
 112
 113   // Enforced Points
 114
 115   QWidget* pointsFrame = new QWidget( fr );
 116   QVBoxLayout* pointsLay = new QVBoxLayout( pointsFrame );
 117   pointsLay->setMargin(MARGIN);
 118   pointsLay->setSpacing(SPACING);
 119
 120   // shapes
 121   QGroupBox* shapesGroup = new QGroupBox( tr("SHAPES"), pointsFrame );
 122   myShapesList = new QListWidget( shapesGroup );
 123   myAddShapeBut = new QPushButton( tr("SMESH_BUT_ADD"), shapesGroup );
 124   QPushButton* remShapeBut = new QPushButton( tr("SMESH_BUT_REMOVE"), shapesGroup );
 125   //
 126   QGridLayout* shapesLay = new QGridLayout( shapesGroup );
 127   shapesLay->setMargin(MARGIN);
 128   shapesLay->setSpacing(SPACING);
 129   shapesLay->addWidget( myShapesList,  0, 0, 3, 2 );
 130   shapesLay->addWidget( myAddShapeBut, 0, 2 );
 131   shapesLay->addWidget( remShapeBut,   1, 2 );
 132   shapesLay->setColumnStretch( 0, 1 );
 133   shapesLay->setRowStretch   ( 2, 1 );
 134
 135   // coords
 136   QGroupBox* coordsGroup = new QGroupBox( tr("POINTS"), pointsFrame );
 137   myCoordsTreeWdg = new QTreeWidget( coordsGroup );
 138   myCoordsTreeWdg->setColumnCount ( 3 );
 139   myCoordsTreeWdg->setHeaderLabels( QStringList() << "X" << "Y" << "Z" );
 140   myCoordsTreeWdg->setItemDelegate( new ItemDelegate( myCoordsTreeWdg ));
 141   QPushButton* addCoordBut = new QPushButton( tr("SMESH_BUT_ADD"), coordsGroup );
 142   QPushButton* remCoordBut = new QPushButton( tr("SMESH_BUT_REMOVE"), coordsGroup );
 143   //
 144   QGridLayout* coordsLay = new QGridLayout( coordsGroup );
 145   coordsLay->setMargin(MARGIN);
 146   coordsLay->setSpacing(SPACING);
 147   coordsLay->addWidget( myCoordsTreeWdg, 0, 0, 3, 2 );
 148   coordsLay->addWidget( addCoordBut,     0, 2 );
 149   coordsLay->addWidget( remCoordBut,     1, 2 );
 150   coordsLay->setColumnStretch( 0, 1 );
 151   coordsLay->setRowStretch   ( 2, 1 );
 152
 153   pointsLay->addWidget( shapesGroup );
 154   pointsLay->addWidget( coordsGroup );
 155
 156   // Tabs
 157   myTabs = new QTabWidget( fr );
 158   myTabs->addTab( myTypeWdg,      tr("TRANSITION"));
 159   myTabs->addTab( myVertexSelWdg, tr("SMESH_BASE_VERTEX"));
 160   myTabs->addTab( pointsFrame,    tr("ENF_NODES"));
 161
 162   lay->addWidget( myTabs, row, 0, 2, 3 );
 163
 164   // signals
 165   connect( myTypeWdg,     SIGNAL( typeChanged(int)),   SLOT( onTypeChanged(int)));
 166   connect( myAddShapeBut, SIGNAL( clicked()),          SLOT( onAddShape() ));
 167   connect( remShapeBut,   SIGNAL( clicked()),          SLOT( onRemoveShape() ));
 168   connect( addCoordBut,   SIGNAL( clicked()),          SLOT( onAddPoint() ));
 169   connect( remCoordBut,   SIGNAL( clicked()),          SLOT( onRemovePoint() ));
 170   connect( myTabs,        SIGNAL( currentChanged(int)),SLOT( onTabChanged(int)));
 171
 172   LightApp_SelectionMgr* selMgr = SMESHGUI::GetSMESHGUI()->selectionMgr();
 173   connect( selMgr, SIGNAL(currentSelectionChanged()), SLOT( onSelectionChanged()));
 174
 175   return fr;
 176 }
 177
 178 //=======================================================================
 179 //function : retrieveParams
 180 //purpose  :
 181 //=======================================================================
 182
 183 void StdMeshersGUI_QuadrangleParamCreator::retrieveParams() const
 184 {
 185   StdMeshers::StdMeshers_QuadrangleParams_var h =
 186     StdMeshers::StdMeshers_QuadrangleParams::_narrow( initParamsHypothesis() );
 187
 188   // name
 189   if( myName )
 190     myName->setText( hypName() );
 191
 192   // main shape
 193   myVertexSelWdg->SetMaxSize(1);
 194   QString anEntry    = SMESHGUI_GenericHypothesisCreator::getShapeEntry();
 195   QString aMainEntry = SMESHGUI_GenericHypothesisCreator::getMainShapeEntry();
 196   if ( anEntry.isEmpty() )
 197     anEntry = h->GetObjectEntry();
 198   myVertexSelWdg->SetGeomShapeEntry(anEntry,aMainEntry);
 199
 200   if ( !isCreation())
 201   {
 202     // type
 203     myTypeWdg->SetType(int(h->GetQuadType()));
 204
 205     // vertex
 206     int vertID = h->GetTriaVertex();
 207     if (vertID > 0) {
 208       SMESH::long_array_var aVec = new SMESH::long_array;
 209       aVec->length(1);
 210       aVec[0] = vertID;
 211       myVertexSelWdg->SetListOfIDs(aVec);
 212     }
 213
 214     // enforced nodes
 215     GEOM::ListOfGO_var     shapes;
 216     SMESH::nodes_array_var points;
 217     h->GetEnforcedNodes( shapes, points );
 218     for ( size_t i = 0; i < shapes->length(); ++i )
 219     {
 220       CORBA::String_var name  = shapes[i]->GetName();
 221       CORBA::String_var entry = shapes[i]->GetStudyEntry();
 222       QListWidgetItem* item = new QListWidgetItem( name.in() );
 223       item->setData( Qt::UserRole, entry.in() );
 224       myShapesList->addItem( item );
 225     }
 226     for ( size_t i = 0; i < points->length(); ++i )
 227     {
 228       QTreeWidgetItem* item = new QTreeWidgetItem
 229         ( QStringList()
 230           << QString::number( points[i].x )
 231           << QString::number( points[i].y )
 232           << QString::number( points[i].z ));
 233       item->setFlags( item->flags() | Qt::ItemIsEditable );
 234       myCoordsTreeWdg->addTopLevelItem( item );
 235     }
 236   }
 237   ((StdMeshersGUI_QuadrangleParamCreator*) this)->onSelectionChanged();
 238 }
 239
 240 //=======================================================================
 241 //function : storeParams
 242 //purpose  :
 243 //=======================================================================
 244
 245 QString  StdMeshersGUI_QuadrangleParamCreator::storeParams() const
 246 {
 247   StdMeshers::StdMeshers_QuadrangleParams_var h =
 248     StdMeshers::StdMeshers_QuadrangleParams::_narrow( hypothesis() );
 249
 250   // name
 251   if( myName )
 252     SMESH::SetName( SMESH::FindSObject( h ), myName->text().toLatin1().constData() );
 253
 254   // transition
 255   h->SetQuadType( StdMeshers::QuadType( myTypeWdg->GetType()) );
 256
 257   // vertex
 258   if ( myVertexSelWdg->GetListSize() > 0 )
 259   {
 260     h->SetTriaVertex( myVertexSelWdg->GetListOfIDs()[0] ); // getlist must be called once
 261     h->SetObjectEntry( myVertexSelWdg->GetMainShapeEntry() );
 262   }
 263   else
 264   {
 265     h->SetTriaVertex( -1 );
 266   }
 267
 268   // enfored nodes
 269
 270   GEOM::ListOfGO_var goList = new GEOM::ListOfGO;
 271   int nbShapes = 0;
 272   goList->length( myShapesList->count() );
 273   for ( int i = 0; i < myShapesList->count(); ++i )
 274   {
 275     QListWidgetItem* item = myShapesList->item(i);
 276     QString         entry = item->data( Qt::UserRole ).toString();
 277     Handle(SALOME_InteractiveObject) io =
 278       new SALOME_InteractiveObject( entry.toStdString().c_str(), "GEOM" );
 279     GEOM::GEOM_Object_var go = GEOMBase::ConvertIOinGEOMObject( io );
 280     if ( !go->_is_nil() )
 281       goList[ nbShapes++ ] = go;
 282   }
 283   goList->length( nbShapes );
 284
 285   SMESH::nodes_array_var points = new SMESH::nodes_array;
 286   points->length( myCoordsTreeWdg->topLevelItemCount() );
 287   for ( int i = 0; i < myCoordsTreeWdg->topLevelItemCount(); ++i )
 288   {
 289     QTreeWidgetItem* item = myCoordsTreeWdg->topLevelItem( i );
 290     points[i].x = item->text(0).toInt();
 291     points[i].y = item->text(1).toInt();
 292     points[i].z = item->text(2).toInt();
 293   }
 294   h->SetEnforcedNodes( goList, points );
 295
 296   return "";
 297 }
 298
 299 //=======================================================================
 300 //function : onTypeChanged
 301 //purpose  :
 302 //=======================================================================
 303
 304 void StdMeshersGUI_QuadrangleParamCreator::onTypeChanged(int type)
 305 {
 306   myTabs->setTabEnabled( TAB_ENF_POINTS, ( type != StdMeshers::QUAD_REDUCED ));
 307 }
 308
 309 //=======================================================================
 310 //function : onAddShape
 311 //purpose  :
 312 //=======================================================================
 313
 314 void StdMeshersGUI_QuadrangleParamCreator::onAddShape()
 315 {
 316   if ( !mySelectedShapeIO.IsNull() )
 317   {
 318     QListWidgetItem* item = new QListWidgetItem( mySelectedShapeIO->getName() );
 319     item->setData( Qt::UserRole, mySelectedShapeIO->getEntry() );
 320     myShapesList->addItem( item );
 321     mySelectedShapeIO.Nullify();
 322     myAddShapeBut->setEnabled( false );
 323   }
 324 }
 325
 326 //=======================================================================
 327 //function : onRemoveShape
 328 //purpose  :
 329 //=======================================================================
 330
 331 void StdMeshersGUI_QuadrangleParamCreator::onRemoveShape()
 332 {
 333   if ( QListWidgetItem * item = myShapesList->currentItem() )
 334     delete item;
 335   onSelectionChanged();
 336 }
 337
 338 //=======================================================================
 339 //function : onAddPoint
 340 //purpose  :
 341 //=======================================================================
 342
 343 void StdMeshersGUI_QuadrangleParamCreator::onAddPoint()
 344 {
 345   QTreeWidgetItem* item = new QTreeWidgetItem( QStringList() << "0" << "0" << "0" );
 346   item->setFlags( item->flags() | Qt::ItemIsEditable );
 347   myCoordsTreeWdg->addTopLevelItem( item );
 348 }
 349
 350 //=======================================================================
 351 //function : onRemovePoint
 352 //purpose  :
 353 //=======================================================================
 354
 355 void StdMeshersGUI_QuadrangleParamCreator::onRemovePoint()
 356 {
 357   if ( myCoordsTreeWdg->topLevelItemCount() )
 358     delete myCoordsTreeWdg->currentItem();
 359 }
 360
 361 //=======================================================================
 362 //function : onSelectionChanged
 363 //purpose  :
 364 //=======================================================================
 365
 366 void StdMeshersGUI_QuadrangleParamCreator::onSelectionChanged()
 367 {
 368   mySelectedShapeIO.Nullify();
 369
 370   // find a sole selected geometry
 371   LightApp_SelectionMgr* selMgr = SMESHGUI::GetSMESHGUI()->selectionMgr();
 372   SALOME_ListIO          selList;
 373   selMgr->selectedObjects( selList );
 374   SALOME_ListIteratorOfListIO selIt( selList );
 375   for ( ; selIt.More(); selIt.Next() )
 376   {
 377     GEOM::GEOM_Object_var go = GEOMBase::ConvertIOinGEOMObject( selIt.Value() );
 378     if ( !go->_is_nil() )
 379     {
 380       if ( !mySelectedShapeIO.IsNull() )
 381       {
 382         mySelectedShapeIO.Nullify();
 383         break;
 384       }
 385       mySelectedShapeIO = selIt.Value();
 386       if ( !mySelectedShapeIO->getName() || !mySelectedShapeIO->getName()[0] )
 387         mySelectedShapeIO.Nullify();
 388     }
 389   }
 390   // check if a selected geometry is not already in myShapesList
 391   if ( !mySelectedShapeIO.IsNull() )
 392   {
 393     for ( int i = 0; i < myShapesList->count(); ++i )
 394       if ( myShapesList->item(i)->data( Qt::UserRole ) == mySelectedShapeIO->getEntry() )
 395       {
 396         mySelectedShapeIO.Nullify();
 397         break;
 398       }
 399   }
 400   myAddShapeBut->setEnabled( !mySelectedShapeIO.IsNull() );
 401 }
 402
 403 //=======================================================================
 404 //function : onTabChanged
 405 //purpose  :
 406 //=======================================================================
 407
 408 void StdMeshersGUI_QuadrangleParamCreator::onTabChanged(int i)
 409 {
 410   myVertexSelWdg->ShowPreview( i == TAB_VERTEX );
 411 }
 412
 413 //================================================================================
 414 // function : Constructor
 415 // purpose  :
 416 //================================================================================
 417
 418 StdMeshersGUI_QuadrangleParamWdg::StdMeshersGUI_QuadrangleParamWdg (QWidget * parent)
 419   : QWidget(parent), myType(0)
 420 {
 421   myType = new QButtonGroup (this);
 422
 423   QGridLayout* typeLay = new QGridLayout( this );
 424
 425   typeLay->setMargin(MARGIN);
 426   typeLay->setSpacing(SPACING);
 427
 428   QString aTypeKey ("SMESH_QUAD_TYPE_%1");
 429   QString aPictKey ("ICON_StdMeshers_Quadrangle_Params_%1");
 430
 431   int itype = 0;
 432   for (; itype < int(StdMeshers::QUAD_NB_TYPES); itype++) {
 433     QRadioButton* rbi = new QRadioButton (tr(aTypeKey.arg(itype).toLatin1()), this);
 434     QPixmap pmi (SMESHGUI::resourceMgr()->loadPixmap("SMESH", tr(aPictKey.arg(itype).toLatin1())));
 435     QLabel* pli = new QLabel (this);
 436     pli->setPixmap(pmi);
 437     typeLay->addWidget(rbi, itype, 0, 1, 1);
 438     typeLay->addWidget(pli, itype, 1, 1, 1);
 439     myType->addButton(rbi, itype);
 440   }
 441   myType->button(0)->setChecked(true);
 442
 443   setLayout(typeLay);
 444   setMinimumWidth(300);
 445
 446   connect( myType, SIGNAL( buttonClicked(int)), this, SIGNAL( typeChanged(int)));
 447 }
 448
 449 //================================================================================
 450 // function : Destructor
 451 // purpose  :
 452 //================================================================================
 453 StdMeshersGUI_QuadrangleParamWdg::~StdMeshersGUI_QuadrangleParamWdg()
 454 {
 455 }
 456
 457 //=================================================================================
 458 // function : SetType
 459 // purpose  :
 460 //=================================================================================
 461 void StdMeshersGUI_QuadrangleParamWdg::SetType (int theType)
 462 {
 463   myType->button(theType)->setChecked(true);
 464 }
 465
 466 //=================================================================================
 467 // function : GetType
 468 // purpose  :
 469 //=================================================================================
 470 int StdMeshersGUI_QuadrangleParamWdg::GetType()
 471 {
 472   return myType->checkedId();
 473 }
 474
 475 //================================================================================
 476 /*!
 477   \brief Constructor
 478 */
 479 StdMeshersGUI_QuadrangleParamCreator::
 480 ItemDelegate::ItemDelegate( QObject* parent ) : QItemDelegate( parent )
 481 {
 482 }
 483 //================================================================================
 484 /*!
 485   \brief Create item editor widget
 486 */
 487 QWidget* StdMeshersGUI_QuadrangleParamCreator::
 488 ItemDelegate::createEditor( QWidget*                    parent,
 489                             const QStyleOptionViewItem& option,
 490                             const QModelIndex&          index ) const
 491 {
 492   SMESHGUI_SpinBox* sb = new SMESHGUI_SpinBox( parent );
 493   sb->RangeStepAndValidator( COORD_MIN, COORD_MAX, 10 );
 494   return sb;
 495 }