1 // SMESH SMESHGUI : GUI for SMESH component
3 // Copyright (C) 2003 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
24 // File : SMESHGUI_MakeNodeAtPointDlg.cxx
25 // Author : Edward AGAPOV
28 #include "SMESHGUI_MakeNodeAtPointDlg.h"
31 #include "SMESHGUI_GEOMGenUtils.h"
32 #include "SMESHGUI_IdValidator.h"
33 #include "SMESHGUI_MeshUtils.h"
34 #include "SMESHGUI_Utils.h"
35 #include "SMESHGUI_VTKUtils.h"
36 #include "SMESHGUI_SpinBox.h"
37 #include "SMESHGUI_MeshEditPreview.h"
39 #include "SMDS_Mesh.hxx"
40 #include "SMDS_MeshNode.hxx"
41 #include "SMESH_Actor.h"
42 #include "SMESH_ActorUtils.h"
43 #include "SMESH_NumberFilter.hxx"
44 #include "SMESH_LogicalFilter.hxx"
47 #include "GeometryGUI.h"
49 #include "LightApp_DataOwner.h"
50 #include "LightApp_SelectionMgr.h"
51 #include "SALOMEDSClient_SObject.hxx"
52 #include "SALOME_ListIO.hxx"
53 #include "SUIT_Desktop.h"
54 #include "SVTK_Selector.h"
55 #include "SVTK_ViewWindow.h"
56 #include "SVTK_ViewModel.h"
57 #include "SalomeApp_Tools.h"
58 #include "SalomeApp_TypeFilter.h"
59 #include "SUIT_ResourceMgr.h"
60 #include "SUIT_OverrideCursor.h"
61 #include "SUIT_MessageBox.h"
64 #include <TColStd_MapOfInteger.hxx>
65 #include <TopoDS_Vertex.hxx>
66 #include <BRep_Tool.hxx>
71 #include <qlineedit.h>
72 #include <qpushbutton.h>
74 #include <qradiobutton.h>
75 #include <qbuttongroup.h>
76 #include <qapplication.h>
77 #include <qstringlist.h>
78 #include <qcheckbox.h>
79 #include <qmessagebox.h>
81 #include <vtkProperty.h>
84 #include <SALOMEconfig.h>
85 #include CORBA_SERVER_HEADER(SMESH_Mesh)
86 #include CORBA_SERVER_HEADER(SMESH_MeshEditor)
92 * \brief Dialog to publish a sub-shape of the mesh main shape
93 * by selecting mesh elements
95 SMESHGUI_MakeNodeAtPointDlg::SMESHGUI_MakeNodeAtPointDlg()
96 : SMESHGUI_Dialog( 0, false, true )
98 setCaption(tr("CAPTION"));
100 QVBoxLayout* aDlgLay = new QVBoxLayout (mainFrame(), MARGIN, SPACING);
102 QFrame* aMainFrame = createMainFrame (mainFrame());
104 aDlgLay->addWidget(aMainFrame);
106 aDlgLay->setStretchFactor(aMainFrame, 1);
109 //=======================================================================
110 // function : createMainFrame()
111 // purpose : Create frame containing dialog's input fields
112 //=======================================================================
113 QFrame* SMESHGUI_MakeNodeAtPointDlg::createMainFrame (QWidget* theParent)
115 QFrame* aFrame = new QFrame(theParent);
117 SUIT_ResourceMgr* rm = SMESH::GetResourceMgr( SMESHGUI::GetSMESHGUI() );
118 QPixmap iconMoveNode (rm->loadPixmap("SMESH", tr("ICON_DLG_MOVE_NODE")));
119 QPixmap iconSelect (rm->loadPixmap("SMESH", tr("ICON_SELECT")));
123 QButtonGroup* aPixGrp = new QButtonGroup(1, Qt::Vertical, tr("MESH_PASS_THROUGH_POINT"), aFrame);
124 aPixGrp->setExclusive(TRUE);
125 QRadioButton* aRBut = new QRadioButton(aPixGrp);
126 aRBut->setPixmap(iconMoveNode);
127 aRBut->setChecked(TRUE);
131 QGroupBox* aCoordGrp = new QGroupBox(1, Qt::Vertical, tr("SMESH_COORDINATES"), aFrame);
132 myCoordBtn = new QPushButton(aCoordGrp);
133 myCoordBtn->setPixmap(iconSelect);
134 myCoordBtn->setToggleButton(TRUE);
136 QLabel* aXLabel = new QLabel(tr("SMESH_X"), aCoordGrp);
137 aXLabel->setSizePolicy( QSizePolicy( QSizePolicy::Fixed, QSizePolicy::Fixed ) );
138 myX = new SMESHGUI_SpinBox(aCoordGrp);
140 QLabel* aYLabel = new QLabel(tr("SMESH_Y"), aCoordGrp);
141 aYLabel->setAlignment( Qt::AlignRight | Qt::AlignVCenter | Qt::ExpandTabs );
142 myY = new SMESHGUI_SpinBox(aCoordGrp);
144 QLabel* aZLabel = new QLabel(tr("SMESH_Z"), aCoordGrp);
145 aZLabel->setAlignment( Qt::AlignRight | Qt::AlignVCenter | Qt::ExpandTabs );
146 myZ = new SMESHGUI_SpinBox(aCoordGrp);
148 myX->RangeStepAndValidator(COORD_MIN, COORD_MAX, 10.0, 3);
149 myY->RangeStepAndValidator(COORD_MIN, COORD_MAX, 10.0, 3);
150 myZ->RangeStepAndValidator(COORD_MIN, COORD_MAX, 10.0, 3);
154 QButtonGroup* aMethodGrp = new QButtonGroup(1, Qt::Vertical, tr("METHOD"), aFrame);
155 aMethodGrp->setExclusive(TRUE);
156 myMoveRBtn = new QRadioButton(tr("MOVE_EXISTING_METHOD"), aMethodGrp);
157 myCreateRBtn = new QRadioButton(tr("CREATE_NEW_METHOD"), aMethodGrp);
161 myNodeToMoveGrp = new QGroupBox(tr("NODE_2MOVE"), aFrame, "anIdGrp");
163 QLabel* idLabel = new QLabel(tr("NODE_2MOVE_ID"), myNodeToMoveGrp, "idLabel");
164 myIdBtn = new QPushButton(myNodeToMoveGrp);
165 myIdBtn->setPixmap(iconSelect);
166 myIdBtn->setToggleButton(TRUE);
167 myId = new QLineEdit(myNodeToMoveGrp,"myId");
168 myId->setValidator(new SMESHGUI_IdValidator(this, "validator", 1));
169 myAutoSearchChkBox = new QCheckBox( tr("AUTO_SEARCH"), myNodeToMoveGrp, "myAutoSearchChkBox");
170 myPreviewChkBox = new QCheckBox( tr("PREVIEW"), myNodeToMoveGrp, "myPreviewChkBox");
172 myNodeToMoveGrp->setColumnLayout(0, Qt::Vertical);
173 myNodeToMoveGrp->layout()->setSpacing(0);
174 myNodeToMoveGrp->layout()->setMargin(0);
175 QGridLayout* myNodeToMoveGrpLayout = new QGridLayout(myNodeToMoveGrp->layout());
176 myNodeToMoveGrpLayout->setAlignment(Qt::AlignTop);
177 myNodeToMoveGrpLayout->setSpacing(SPACING);
178 myNodeToMoveGrpLayout->setMargin(MARGIN);
179 myNodeToMoveGrpLayout->addWidget( idLabel, 0, 0 );
180 myNodeToMoveGrpLayout->addWidget( myIdBtn, 0, 1 );
181 myNodeToMoveGrpLayout->addWidget( myId, 0, 2 );
182 myNodeToMoveGrpLayout->addMultiCellWidget( myAutoSearchChkBox, 1, 1, 0, 2 );
183 myNodeToMoveGrpLayout->addMultiCellWidget( myPreviewChkBox, 2, 2, 0, 2 );
185 QVBoxLayout* aLay = new QVBoxLayout(aFrame);
186 aLay->addWidget(aPixGrp);
187 aLay->addWidget(aCoordGrp);
188 aLay->addWidget(aMethodGrp);
189 aLay->addWidget(myNodeToMoveGrp);
191 connect(myCoordBtn, SIGNAL (toggled(bool)), this, SLOT(ButtonToggled(bool)));
192 connect(myMoveRBtn, SIGNAL (toggled(bool)), this, SLOT(ButtonToggled(bool)));
193 connect(myCreateRBtn, SIGNAL (toggled(bool)), this, SLOT(ButtonToggled(bool)));
194 connect(myIdBtn, SIGNAL (toggled(bool)), this, SLOT(ButtonToggled(bool)));
195 connect(myAutoSearchChkBox, SIGNAL (toggled(bool)), this, SLOT(ButtonToggled(bool)));
197 myMoveRBtn->setChecked(TRUE);
198 myIdBtn->setOn(TRUE);
199 myAutoSearchChkBox->setChecked(TRUE);
204 //================================================================================
206 * \brief SLOT called when any button is toggled
207 * \param bool - on or off
209 //================================================================================
211 void SMESHGUI_MakeNodeAtPointDlg::ButtonToggled (bool on)
213 const QObject* aSender = sender();
215 if ( aSender == myCoordBtn ) // button to set coord by node selection
217 if ( myIdBtn->isEnabled() )
218 myIdBtn->setOn( !on );
220 else if ( aSender == myIdBtn ) // button to select a node to move
222 myCoordBtn->setOn( !on );
224 else if ( aSender == myMoveRBtn ) // move node method
226 myNodeToMoveGrp->setEnabled( TRUE );
228 else if ( aSender == myCreateRBtn ) // create node method
230 myNodeToMoveGrp->setEnabled( FALSE );
231 myCoordBtn->setOn( TRUE );
234 if ( aSender == myAutoSearchChkBox ) // automatic node search
238 myId->setReadOnly ( TRUE );
239 myIdBtn->setOn( FALSE );
240 myIdBtn->setEnabled( FALSE );
241 myCoordBtn->setOn( TRUE );
244 myId->setReadOnly ( FALSE );
245 myIdBtn->setEnabled( TRUE );
250 //================================================================================
254 //================================================================================
256 SMESHGUI_MakeNodeAtPointOp::SMESHGUI_MakeNodeAtPointOp()
259 myDlg = new SMESHGUI_MakeNodeAtPointDlg;
262 // connect signals and slots
263 connect(myDlg->myX, SIGNAL (valueChanged(double)), this, SLOT(redisplayPreview()));
264 connect(myDlg->myY, SIGNAL (valueChanged(double)), this, SLOT(redisplayPreview()));
265 connect(myDlg->myZ, SIGNAL (valueChanged(double)), this, SLOT(redisplayPreview()));
266 connect(myDlg->myId,SIGNAL (textChanged(const QString&)),SLOT(redisplayPreview()));
267 connect(myDlg->myPreviewChkBox, SIGNAL (toggled(bool)),SLOT(redisplayPreview()));
268 connect(myDlg->myAutoSearchChkBox,SIGNAL (toggled(bool)),SLOT(redisplayPreview()));
269 connect(myDlg->myMoveRBtn, SIGNAL (toggled(bool)),SLOT(redisplayPreview()));
270 connect(myDlg->myCreateRBtn, SIGNAL (toggled(bool)),SLOT(redisplayPreview()));
273 //=======================================================================
274 // function : startOperation()
275 // purpose : Init dialog fields, connect signals and slots, show dialog
276 //=======================================================================
277 void SMESHGUI_MakeNodeAtPointOp::startOperation()
282 // init simulation with a current View
283 if ( mySimulation ) delete mySimulation;
284 mySimulation = new SMESHGUI_MeshEditPreview(SMESH::GetViewWindow( getSMESHGUI() ));
285 vtkProperty* aProp = vtkProperty::New();
286 aProp->SetRepresentationToWireframe();
287 aProp->SetColor(250, 0, 250);
288 aProp->SetPointSize(5);
289 aProp->SetLineWidth( SMESH::GetFloat("SMESH:element_width",1) + 1);
290 mySimulation->GetActor()->SetProperty(aProp);
293 SMESHGUI_SelectionOp::startOperation();
295 // SalomeApp_TypeFilter depends on a current study
296 if ( myFilter ) delete myFilter;
297 QPtrList<SUIT_SelectionFilter> filters;
298 filters.append( new SalomeApp_TypeFilter((SalomeApp_Study*)study(), "SMESH" ));
299 TColStd_MapOfInteger vertexType;
300 vertexType.Add( TopAbs_VERTEX );
301 filters.append( new SMESH_NumberFilter("GEOM", TopAbs_VERTEX, 1, vertexType ));
302 myFilter = new SMESH_LogicalFilter( filters, SMESH_LogicalFilter::LO_OR );
304 activateSelection(); // set filters
306 myDlg->myX->SetValue(0);
307 myDlg->myY->SetValue(0);
308 myDlg->myZ->SetValue(0);
309 myDlg->myId->setText("");
312 onSelectionDone(); // init myMeshActor
315 // myMeshOldDisplayMode = myMeshActor->GetRepresentation();
316 // myMeshActor->SetRepresentation( VTK_WIREFRAME );
317 myMeshActor->SetPointRepresentation(true);
318 SMESH::RepaintCurrentView();
323 //================================================================================
325 * \brief Stops operation
327 //================================================================================
329 void SMESHGUI_MakeNodeAtPointOp::stopOperation()
332 mySimulation->SetVisibility(false);
334 // myMeshActor->SetRepresentation( myMeshOldDisplayMode );
335 myMeshActor->SetPointRepresentation(false);
336 SMESH::RepaintCurrentView();
339 selectionMgr()->removeFilter( myFilter );
340 SMESHGUI_SelectionOp::stopOperation();
343 //================================================================================
345 * \brief perform it's intention action: move or create a node
347 //================================================================================
349 bool SMESHGUI_MakeNodeAtPointOp::onApply()
351 if ( !myMeshActor ) {
352 SUIT_MessageBox::warn1( dlg(), tr( "SMESH_WRN_WARNING" ),
353 tr("INVALID_MESH"), tr( "SMESH_BUT_OK" ) );
358 if ( !isValid() ) { // node id is invalid
359 SUIT_MessageBox::warn1( dlg(), tr( "SMESH_WRN_WARNING" ),
360 tr("INVALID_ID"), tr( "SMESH_BUT_OK" ) );
367 SMESH::SMESH_Mesh_var aMesh = SMESH::GetMeshByIO(myMeshActor->getIO());
368 if (aMesh->_is_nil()) {
369 QMessageBox::information(SMESHGUI::desktop(), tr("SMESH_ERROR"),
370 tr("SMESHG_NO_MESH"), QMessageBox::Ok);
373 SMESH::SMESH_MeshEditor_var aMeshEditor = aMesh->GetMeshEditor();
374 if (aMeshEditor->_is_nil())
378 if ( myDlg->myCreateRBtn->isOn() )
380 aResult = aMeshEditor->AddNode(myDlg->myX->GetValue(),
381 myDlg->myY->GetValue(),
382 myDlg->myZ->GetValue());
386 int anId = myDlg->myId->text().toInt();
387 aResult = aMeshEditor->MoveClosestNodeToPoint(myDlg->myX->GetValue(),
388 myDlg->myY->GetValue(),
389 myDlg->myZ->GetValue(),
394 myDlg->myId->setText("");
397 selectionMgr()->setSelectedObjects(aList,false);
398 aList.Append(myMeshActor->getIO());
399 selectionMgr()->setSelectedObjects(aList,false);
403 catch (const SALOME::SALOME_Exception& S_ex) {
404 SalomeApp_Tools::QtCatchCorbaException(S_ex);
412 //================================================================================
414 * \brief Check selected node id validity
416 //================================================================================
418 bool SMESHGUI_MakeNodeAtPointOp::isValid()
423 myDlg->myMoveRBtn->isOn() &&
424 !myDlg->myAutoSearchChkBox->isChecked() )
427 int id = myDlg->myId->text().toInt();
429 if (SMDS_Mesh* aMesh = myMeshActor->GetObject()->GetMesh())
430 ok = aMesh->FindNode( id );
435 //================================================================================
437 * \brief SLOT called when selection changed
439 //================================================================================
441 void SMESHGUI_MakeNodeAtPointOp::onSelectionDone()
443 if ( !myDlg->isShown() || !myDlg->isEnabled() )
447 selectionMgr()->selectedObjects(aList, SVTK_Viewer::Type());
448 if (aList.Extent() != 1)
450 Handle(SALOME_InteractiveObject) anIO = aList.First();
451 SMESH_Actor* aMeshActor = SMESH::FindActorByEntry(anIO->getEntry());
453 if (!aMeshActor) { // coord by geom
454 if ( myDlg->myCoordBtn->isOn() ) {
455 GEOM::GEOM_Object_var geom = SMESH::IObjectToInterface<GEOM::GEOM_Object>(anIO);
456 if ( !geom->_is_nil() ) {
457 TopoDS_Vertex aShape;
458 if ( GEOMBase::GetShape(geom, aShape) &&
459 aShape.ShapeType() == TopAbs_VERTEX ) {
460 gp_Pnt P = BRep_Tool::Pnt(aShape);
462 myDlg->myX->SetValue(P.X());
463 myDlg->myY->SetValue(P.Y());
464 myDlg->myZ->SetValue(P.Z());
474 myMeshActor = aMeshActor;
477 int nbElems = SMESH::GetNameOfSelectedElements(selector(),anIO, aString);
479 if (SMDS_Mesh* aMesh = aMeshActor->GetObject()->GetMesh()) {
480 if (const SMDS_MeshNode* aNode = aMesh->FindNode(aString.toInt())) {
482 if ( myDlg->myCoordBtn->isOn() ) { // set coord
483 myDlg->myX->SetValue(aNode->X());
484 myDlg->myY->SetValue(aNode->Y());
485 myDlg->myZ->SetValue(aNode->Z());
489 else if ( myDlg->myIdBtn->isOn() &&
490 myDlg->myIdBtn->isEnabled() ) { // set node to move
491 myDlg->myId->setText(aString);
502 //================================================================================
504 * \brief update preview
506 //================================================================================
508 void SMESHGUI_MakeNodeAtPointOp::redisplayPreview()
514 SMESH::MeshPreviewStruct_var aMeshPreviewStruct;
516 bool moveShown = false;
517 if ( myDlg->myMoveRBtn->isOn() && // Move method
520 const bool autoSearch = myDlg->myAutoSearchChkBox->isChecked();
521 const bool preview = myDlg->myPreviewChkBox->isChecked();
523 myDlg->myId->setText("");
524 if ( preview && ( autoSearch || isValid() ))
527 SMESH::SMESH_Mesh_var aMesh = SMESH::GetMeshByIO(myMeshActor->getIO());
528 if (!aMesh->_is_nil()) {
529 SMESH::SMESH_MeshEditor_var aPreviewer = aMesh->GetMeshEditPreviewer();
530 if (!aPreviewer->_is_nil())
532 SUIT_OverrideCursor aWaitCursor;
534 // find id and/or just compute preview
535 int anId = aPreviewer->MoveClosestNodeToPoint(myDlg->myX->GetValue(),
536 myDlg->myY->GetValue(),
537 myDlg->myZ->GetValue(),
538 myDlg->myId->text().toInt());
539 if ( autoSearch ) { // set found id
542 idTxt = idTxt.arg( anId );
545 myDlg->myId->setText( idTxt );
547 if ( preview ) { // fill preview data
548 aMeshPreviewStruct = aPreviewer->GetPreviewData();
549 moveShown = ( anId > 0 );
560 aMeshPreviewStruct = new SMESH::MeshPreviewStruct();
562 aMeshPreviewStruct->nodesXYZ.length(1);
563 aMeshPreviewStruct->nodesXYZ[0].x = myDlg->myX->GetValue();
564 aMeshPreviewStruct->nodesXYZ[0].y = myDlg->myY->GetValue();
565 aMeshPreviewStruct->nodesXYZ[0].z = myDlg->myZ->GetValue();
567 aMeshPreviewStruct->elementTypes.length(1);
568 aMeshPreviewStruct->elementTypes[0].SMDS_ElementType = SMESH::NODE;
569 aMeshPreviewStruct->elementTypes[0].isPoly = false;
570 aMeshPreviewStruct->elementTypes[0].nbNodesInElement = 1;
572 aMeshPreviewStruct->elementConnectivities.length(1);
573 aMeshPreviewStruct->elementConnectivities[0] = 0;
577 if ( aMeshPreviewStruct.operator->() )
579 mySimulation->SetData(aMeshPreviewStruct._retn());
583 mySimulation->SetVisibility(false);
589 //================================================================================
591 * \brief Activate Node selection
593 //================================================================================
595 void SMESHGUI_MakeNodeAtPointOp::activateSelection()
597 selectionMgr()->clearFilters();
598 SMESH::SetPointRepresentation(false);
599 selectionMgr()->installFilter( myFilter );
600 setSelectionMode( NodeSelection );
603 //================================================================================
607 //================================================================================
609 SMESHGUI_MakeNodeAtPointOp::~SMESHGUI_MakeNodeAtPointOp()
611 if ( myDlg ) delete myDlg;
612 if ( mySimulation ) delete mySimulation;
613 if ( myFilter ) delete myFilter;
616 //================================================================================
618 * \brief Gets dialog of this operation
619 * \retval LightApp_Dialog* - pointer to dialog of this operation
621 //================================================================================
623 LightApp_Dialog* SMESHGUI_MakeNodeAtPointOp::dlg() const