1 // Copyright (C) 2007-2013 CEA/DEN, EDF R&D, OPEN CASCADE
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.
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.
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
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 // File : StdMeshersGUI_QuadrangleParamWdg.cxx
20 // Author : Open CASCADE S.A.S. (jfa)
23 #include "StdMeshersGUI_QuadrangleParamWdg.h"
26 #include "SMESHGUI_SpinBox.h"
27 #include "StdMeshersGUI_SubShapeSelectorWdg.h"
30 #include <LightApp_SelectionMgr.h>
31 #include <SALOME_ListIO.hxx>
32 #include <SALOME_ListIteratorOfListIO.hxx>
33 #include <SUIT_ResourceMgr.h>
36 #include <QButtonGroup>
38 #include <QGridLayout>
42 #include <QListWidget>
43 #include <QListWidgetItem>
44 #include <QPushButton>
45 #include <QRadioButton>
46 #include <QTreeWidget>
47 #include <QVBoxLayout>
50 #include <SALOMEconfig.h>
51 #include CORBA_CLIENT_HEADER(SMESH_BasicHypothesis)
52 #include CORBA_CLIENT_HEADER(GEOM_Gen)
58 enum { TAB_TRANSITION, TAB_VERTEX, TAB_ENF_POINTS };
60 //================================================================================
61 // function : Constructor
63 //================================================================================
65 StdMeshersGUI_QuadrangleParamCreator::StdMeshersGUI_QuadrangleParamCreator(const QString& aHypType)
66 : StdMeshersGUI_StdHypothesisCreator( aHypType )
70 //=======================================================================
73 //=======================================================================
75 QString StdMeshersGUI_QuadrangleParamCreator::helpPage() const
77 return "a2d_meshing_hypo_page.html#hypo_quad_params_anchor";
80 //=======================================================================
81 //function : buildFrame
83 //=======================================================================
85 QFrame* StdMeshersGUI_QuadrangleParamCreator::buildFrame()
87 QFrame* fr = new QFrame();
89 QGridLayout* lay = new QGridLayout( fr );
90 lay->setMargin( MARGIN );
91 lay->setSpacing( SPACING );
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 );
106 myTypeWdg = new StdMeshersGUI_QuadrangleParamWdg( fr );
110 myVertexSelWdg = new StdMeshersGUI_SubShapeSelectorWdg( fr, TopAbs_VERTEX );
111 myVertexSelWdg->layout()->setMargin( MARGIN );
115 QWidget* pointsFrame = new QWidget( fr );
116 QVBoxLayout* pointsLay = new QVBoxLayout( pointsFrame );
117 pointsLay->setMargin(MARGIN);
118 pointsLay->setSpacing(SPACING);
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 );
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 );
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 );
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 );
153 pointsLay->addWidget( shapesGroup );
154 pointsLay->addWidget( coordsGroup );
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"));
162 lay->addWidget( myTabs, row, 0, 2, 3 );
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)));
172 LightApp_SelectionMgr* selMgr = SMESHGUI::GetSMESHGUI()->selectionMgr();
173 connect( selMgr, SIGNAL(currentSelectionChanged()), SLOT( onSelectionChanged()));
178 //=======================================================================
179 //function : retrieveParams
181 //=======================================================================
183 void StdMeshersGUI_QuadrangleParamCreator::retrieveParams() const
185 StdMeshers::StdMeshers_QuadrangleParams_var h =
186 StdMeshers::StdMeshers_QuadrangleParams::_narrow( initParamsHypothesis() );
190 myName->setText( hypName() );
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);
199 myVertexSelWdg->SetMainShapeEntry(aMainEntry);
204 myTypeWdg->SetType(int(h->GetQuadType()));
207 int vertID = h->GetTriaVertex();
209 SMESH::long_array_var aVec = new SMESH::long_array;
212 myVertexSelWdg->SetListOfIDs(aVec);
216 GEOM::ListOfGO_var shapes;
217 SMESH::nodes_array_var points;
218 h->GetEnforcedNodes( shapes, points );
219 for ( int i = 0; i < shapes->length(); ++i )
221 CORBA::String_var name = shapes[i]->GetName();
222 CORBA::String_var entry = shapes[i]->GetStudyEntry();
223 QListWidgetItem* item = new QListWidgetItem( name.in() );
224 item->setData( Qt::UserRole, entry.in() );
225 myShapesList->addItem( item );
227 for ( int i = 0; i < points->length(); ++i )
229 QTreeWidgetItem* item = new QTreeWidgetItem
231 << QString::number( points[i].x )
232 << QString::number( points[i].y )
233 << QString::number( points[i].z ));
234 item->setFlags( item->flags() | Qt::ItemIsEditable );
235 myCoordsTreeWdg->addTopLevelItem( item );
238 ((StdMeshersGUI_QuadrangleParamCreator*) this)->onSelectionChanged();
241 //=======================================================================
242 //function : storeParams
244 //=======================================================================
246 QString StdMeshersGUI_QuadrangleParamCreator::storeParams() const
248 StdMeshers::StdMeshers_QuadrangleParams_var h =
249 StdMeshers::StdMeshers_QuadrangleParams::_narrow( hypothesis() );
253 SMESH::SetName( SMESH::FindSObject( h ), myName->text().toLatin1().constData() );
256 h->SetQuadType( StdMeshers::QuadType( myTypeWdg->GetType()) );
259 if ( myVertexSelWdg->GetListSize() > 0 )
261 h->SetTriaVertex( myVertexSelWdg->GetListOfIDs()[0] ); // getlist must be called once
262 h->SetObjectEntry( myVertexSelWdg->GetMainShapeEntry() );
266 h->SetTriaVertex( -1 );
271 GEOM::ListOfGO_var goList = new GEOM::ListOfGO;
273 goList->length( myShapesList->count() );
274 for ( int i = 0; i < myShapesList->count(); ++i )
276 QListWidgetItem* item = myShapesList->item(i);
277 QString entry = item->data( Qt::UserRole ).toString();
278 Handle(SALOME_InteractiveObject) io =
279 new SALOME_InteractiveObject( entry.toStdString().c_str(), "GEOM" );
280 GEOM::GEOM_Object_var go = GEOMBase::ConvertIOinGEOMObject( io );
281 if ( !go->_is_nil() )
282 goList[ nbShapes++ ] = go;
284 goList->length( nbShapes );
286 SMESH::nodes_array_var points = new SMESH::nodes_array;
287 points->length( myCoordsTreeWdg->topLevelItemCount() );
288 for ( int i = 0; i < myCoordsTreeWdg->topLevelItemCount(); ++i )
290 QTreeWidgetItem* item = myCoordsTreeWdg->topLevelItem( i );
291 points[i].x = item->text(0).toInt();
292 points[i].y = item->text(1).toInt();
293 points[i].z = item->text(2).toInt();
295 h->SetEnforcedNodes( goList, points );
300 //=======================================================================
301 //function : onTypeChanged
303 //=======================================================================
305 void StdMeshersGUI_QuadrangleParamCreator::onTypeChanged(int type)
307 myTabs->setTabEnabled( TAB_ENF_POINTS, ( type != StdMeshers::QUAD_REDUCED ));
310 //=======================================================================
311 //function : onAddShape
313 //=======================================================================
315 void StdMeshersGUI_QuadrangleParamCreator::onAddShape()
317 if ( !mySelectedShapeIO.IsNull() )
319 QListWidgetItem* item = new QListWidgetItem( mySelectedShapeIO->getName() );
320 item->setData( Qt::UserRole, mySelectedShapeIO->getEntry() );
321 myShapesList->addItem( item );
322 mySelectedShapeIO.Nullify();
323 myAddShapeBut->setEnabled( false );
327 //=======================================================================
328 //function : onRemoveShape
330 //=======================================================================
332 void StdMeshersGUI_QuadrangleParamCreator::onRemoveShape()
334 if ( QListWidgetItem * item = myShapesList->currentItem() )
336 onSelectionChanged();
339 //=======================================================================
340 //function : onAddPoint
342 //=======================================================================
344 void StdMeshersGUI_QuadrangleParamCreator::onAddPoint()
346 QTreeWidgetItem* item = new QTreeWidgetItem( QStringList() << "0" << "0" << "0" );
347 item->setFlags( item->flags() | Qt::ItemIsEditable );
348 myCoordsTreeWdg->addTopLevelItem( item );
351 //=======================================================================
352 //function : onRemovePoint
354 //=======================================================================
356 void StdMeshersGUI_QuadrangleParamCreator::onRemovePoint()
358 if ( myCoordsTreeWdg->topLevelItemCount() )
359 delete myCoordsTreeWdg->currentItem();
362 //=======================================================================
363 //function : onSelectionChanged
365 //=======================================================================
367 void StdMeshersGUI_QuadrangleParamCreator::onSelectionChanged()
369 mySelectedShapeIO.Nullify();
371 // find a sole selected geometry
372 LightApp_SelectionMgr* selMgr = SMESHGUI::GetSMESHGUI()->selectionMgr();
373 SALOME_ListIO selList;
374 selMgr->selectedObjects( selList );
375 SALOME_ListIteratorOfListIO selIt( selList );
376 for ( ; selIt.More(); selIt.Next() )
378 GEOM::GEOM_Object_var go = GEOMBase::ConvertIOinGEOMObject( selIt.Value() );
379 if ( !go->_is_nil() )
381 if ( !mySelectedShapeIO.IsNull() )
383 mySelectedShapeIO.Nullify();
386 mySelectedShapeIO = selIt.Value();
387 if ( !mySelectedShapeIO->getName() || !mySelectedShapeIO->getName()[0] )
388 mySelectedShapeIO.Nullify();
391 // check if a selected geometry is not already in myShapesList
392 if ( !mySelectedShapeIO.IsNull() )
394 for ( int i = 0; i < myShapesList->count(); ++i )
395 if ( myShapesList->item(i)->data( Qt::UserRole ) == mySelectedShapeIO->getEntry() )
397 mySelectedShapeIO.Nullify();
401 myAddShapeBut->setEnabled( !mySelectedShapeIO.IsNull() );
404 //=======================================================================
405 //function : onTabChanged
407 //=======================================================================
409 void StdMeshersGUI_QuadrangleParamCreator::onTabChanged(int i)
411 myVertexSelWdg->showPreview( i == TAB_VERTEX );
414 //================================================================================
415 // function : Constructor
417 //================================================================================
419 StdMeshersGUI_QuadrangleParamWdg::StdMeshersGUI_QuadrangleParamWdg (QWidget * parent)
420 : QWidget(parent), myType(0)
422 myType = new QButtonGroup (this);
424 QGridLayout* typeLay = new QGridLayout( this );
426 typeLay->setMargin(MARGIN);
427 typeLay->setSpacing(SPACING);
429 QString aTypeKey ("SMESH_QUAD_TYPE_%1");
430 QString aPictKey ("ICON_StdMeshers_Quadrangle_Params_%1");
433 for (; itype < int(StdMeshers::QUAD_NB_TYPES); itype++) {
434 QRadioButton* rbi = new QRadioButton (tr(aTypeKey.arg(itype).toLatin1()), this);
435 QPixmap pmi (SMESHGUI::resourceMgr()->loadPixmap("SMESH", tr(aPictKey.arg(itype).toLatin1())));
436 QLabel* pli = new QLabel (this);
438 typeLay->addWidget(rbi, itype, 0, 1, 1);
439 typeLay->addWidget(pli, itype, 1, 1, 1);
440 myType->addButton(rbi, itype);
442 myType->button(0)->setChecked(true);
445 setMinimumWidth(300);
447 connect( myType, SIGNAL( buttonClicked(int)), this, SIGNAL( typeChanged(int)));
450 //================================================================================
451 // function : Destructor
453 //================================================================================
454 StdMeshersGUI_QuadrangleParamWdg::~StdMeshersGUI_QuadrangleParamWdg()
458 //=================================================================================
459 // function : SetType
461 //=================================================================================
462 void StdMeshersGUI_QuadrangleParamWdg::SetType (int theType)
464 myType->button(theType)->setChecked(true);
467 //=================================================================================
468 // function : GetType
470 //=================================================================================
471 int StdMeshersGUI_QuadrangleParamWdg::GetType()
473 return myType->checkedId();
476 //================================================================================
480 StdMeshersGUI_QuadrangleParamCreator::
481 ItemDelegate::ItemDelegate( QObject* parent ) : QItemDelegate( parent )
484 //================================================================================
486 \brief Create item editor widget
488 QWidget* StdMeshersGUI_QuadrangleParamCreator::
489 ItemDelegate::createEditor( QWidget* parent,
490 const QStyleOptionViewItem& option,
491 const QModelIndex& index ) const
493 SMESHGUI_SpinBox* sb = new SMESHGUI_SpinBox( parent );
494 sb->RangeStepAndValidator( COORD_MIN, COORD_MAX, 10 );