ForkMaster/smesh
1
0
Fork 0
mirror of https://git.salome-platform.org/gitpub/modules/smesh.git synced 2024-12-28 18:30:35 +05:00
Code Issues Packages Projects Releases Wiki Activity
8479b646bc
smesh/src/SMESHGUI/SMESHGUI_MakeNodeAtPointDlg.cxx
abd 85c0f44772 Update copyright information
2008-11-27 13:08:54 +00:00

633 lines
20 KiB
C++
Raw Blame History

// Copyright (C) 2007-2008 CEA/DEN, EDF R&D, OPEN CASCADE
//
// Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
//
// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
//
// SMESH SMESHGUI : GUI for SMESH component
// File : SMESHGUI_MakeNodeAtPointDlg.cxx
// Author : Edward AGAPOV
// Module : SMESH
//
#include "SMESHGUI_MakeNodeAtPointDlg.h"
#include "SMESHGUI.h"
#include "SMESHGUI_GEOMGenUtils.h"
#include "SMESHGUI_IdValidator.h"
#include "SMESHGUI_MeshUtils.h"
#include "SMESHGUI_Utils.h"
#include "SMESHGUI_VTKUtils.h"
#include "SMESHGUI_SpinBox.h"
#include "SMESHGUI_MeshEditPreview.h"
#include "SMDS_Mesh.hxx"
#include "SMDS_MeshNode.hxx"
#include "SMESH_Actor.h"
#include "SMESH_ActorUtils.h"
#include "SMESH_NumberFilter.hxx"
#include "SMESH_LogicalFilter.hxx"
#include "GEOMBase.h"
#include "GeometryGUI.h"
#include "LightApp_DataOwner.h"
#include "LightApp_SelectionMgr.h"
#include "SALOMEDSClient_SObject.hxx"
#include "SALOME_ListIO.hxx"
#include "SUIT_Desktop.h"
#include "SVTK_Selector.h"
#include "SVTK_ViewWindow.h"
#include "SVTK_ViewModel.h"
#include "SalomeApp_Tools.h"
#include "SalomeApp_TypeFilter.h"
#include "SUIT_ResourceMgr.h"
#include "SUIT_OverrideCursor.h"
#include "SUIT_MessageBox.h"
// OCCT Includes
#include <TColStd_MapOfInteger.hxx>
#include <TopoDS_Vertex.hxx>
#include <BRep_Tool.hxx>
// QT Includes
#include <qframe.h>
#include <qlayout.h>
#include <qlineedit.h>
#include <qpushbutton.h>
#include <qlabel.h>
#include <qradiobutton.h>
#include <qbuttongroup.h>
#include <qapplication.h>
#include <qstringlist.h>
#include <qcheckbox.h>
#include <qmessagebox.h>
#include <vtkProperty.h>
// IDL Headers
#include <SALOMEconfig.h>
#include CORBA_SERVER_HEADER(SMESH_Mesh)
#include CORBA_SERVER_HEADER(SMESH_MeshEditor)
#define SPACING 5
#define MARGIN 10
/*!
* \brief Dialog to publish a sub-shape of the mesh main shape
* by selecting mesh elements
*/
SMESHGUI_MakeNodeAtPointDlg::SMESHGUI_MakeNodeAtPointDlg()
: SMESHGUI_Dialog( 0, false, true )
{
setCaption(tr("CAPTION"));
QVBoxLayout* aDlgLay = new QVBoxLayout (mainFrame(), MARGIN, SPACING);
QFrame* aMainFrame = createMainFrame (mainFrame());
aDlgLay->addWidget(aMainFrame);
aDlgLay->setStretchFactor(aMainFrame, 1);
}
//=======================================================================
// function : createMainFrame()
// purpose : Create frame containing dialog's input fields
//=======================================================================
QFrame* SMESHGUI_MakeNodeAtPointDlg::createMainFrame (QWidget* theParent)
{
QFrame* aFrame = new QFrame(theParent);
SUIT_ResourceMgr* rm = SMESH::GetResourceMgr( SMESHGUI::GetSMESHGUI() );
QPixmap iconMoveNode (rm->loadPixmap("SMESH", tr("ICON_DLG_MOVE_NODE")));
QPixmap iconSelect (rm->loadPixmap("SMESH", tr("ICON_SELECT")));
// constructor
QButtonGroup* aPixGrp = new QButtonGroup(1, Qt::Vertical, tr("MESH_PASS_THROUGH_POINT"), aFrame);
aPixGrp->setExclusive(TRUE);
QRadioButton* aRBut = new QRadioButton(aPixGrp);
aRBut->setPixmap(iconMoveNode);
aRBut->setChecked(TRUE);
// coordinates
QGroupBox* aCoordGrp = new QGroupBox(1, Qt::Vertical, tr("SMESH_COORDINATES"), aFrame);
myCoordBtn = new QPushButton(aCoordGrp);
myCoordBtn->setPixmap(iconSelect);
myCoordBtn->setToggleButton(TRUE);
QLabel* aXLabel = new QLabel(tr("SMESH_X"), aCoordGrp);
aXLabel->setSizePolicy( QSizePolicy( QSizePolicy::Fixed, QSizePolicy::Fixed ) );
myX = new SMESHGUI_SpinBox(aCoordGrp);
QLabel* aYLabel = new QLabel(tr("SMESH_Y"), aCoordGrp);
//aYLabel->setAlignment( Qt::AlignRight | Qt::AlignVCenter | Qt::ExpandTabs );
aYLabel->setSizePolicy( QSizePolicy( QSizePolicy::Fixed, QSizePolicy::Fixed ) );
myY = new SMESHGUI_SpinBox(aCoordGrp);
QLabel* aZLabel = new QLabel(tr("SMESH_Z"), aCoordGrp);
//aZLabel->setAlignment( Qt::AlignRight | Qt::AlignVCenter | Qt::ExpandTabs );
aZLabel->setSizePolicy( QSizePolicy( QSizePolicy::Fixed, QSizePolicy::Fixed ) );
myZ = new SMESHGUI_SpinBox(aCoordGrp);
myX->RangeStepAndValidator(COORD_MIN, COORD_MAX, 10.0, DBL_DIGITS_DISPLAY);
myY->RangeStepAndValidator(COORD_MIN, COORD_MAX, 10.0, DBL_DIGITS_DISPLAY);
myZ->RangeStepAndValidator(COORD_MIN, COORD_MAX, 10.0, DBL_DIGITS_DISPLAY);
// Method selection
QButtonGroup* aMethodGrp = new QButtonGroup(1, Qt::Vertical, tr("METHOD"), aFrame);
aMethodGrp->setExclusive(TRUE);
myMoveRBtn = new QRadioButton(tr("MOVE_EXISTING_METHOD"), aMethodGrp);
myCreateRBtn = new QRadioButton(tr("CREATE_NEW_METHOD"), aMethodGrp);
// node ID
myNodeToMoveGrp = new QGroupBox(tr("NODE_2MOVE"), aFrame, "anIdGrp");
QLabel* idLabel = new QLabel(tr("NODE_2MOVE_ID"), myNodeToMoveGrp, "idLabel");
myIdBtn = new QPushButton(myNodeToMoveGrp);
myIdBtn->setPixmap(iconSelect);
myIdBtn->setToggleButton(TRUE);
myId = new QLineEdit(myNodeToMoveGrp,"myId");
myId->setValidator(new SMESHGUI_IdValidator(this, "validator", 1));
myAutoSearchChkBox = new QCheckBox( tr("AUTO_SEARCH"), myNodeToMoveGrp, "myAutoSearchChkBox");
myPreviewChkBox = new QCheckBox( tr("PREVIEW"), myNodeToMoveGrp, "myPreviewChkBox");
myNodeToMoveGrp->setColumnLayout(0, Qt::Vertical);
myNodeToMoveGrp->layout()->setSpacing(0);
myNodeToMoveGrp->layout()->setMargin(0);
QGridLayout* myNodeToMoveGrpLayout = new QGridLayout(myNodeToMoveGrp->layout());
myNodeToMoveGrpLayout->setAlignment(Qt::AlignTop);
myNodeToMoveGrpLayout->setSpacing(SPACING);
myNodeToMoveGrpLayout->setMargin(MARGIN);
myNodeToMoveGrpLayout->addWidget( idLabel, 0, 0 );
myNodeToMoveGrpLayout->addWidget( myIdBtn, 0, 1 );
myNodeToMoveGrpLayout->addWidget( myId, 0, 2 );
myNodeToMoveGrpLayout->addMultiCellWidget( myAutoSearchChkBox, 1, 1, 0, 2 );
myNodeToMoveGrpLayout->addMultiCellWidget( myPreviewChkBox, 2, 2, 0, 2 );
QVBoxLayout* aLay = new QVBoxLayout(aFrame);
aLay->addWidget(aPixGrp);
aLay->addWidget(aCoordGrp);
aLay->addWidget(aMethodGrp);
aLay->addWidget(myNodeToMoveGrp);
connect(myCoordBtn, SIGNAL (toggled(bool)), this, SLOT(ButtonToggled(bool)));
connect(myMoveRBtn, SIGNAL (toggled(bool)), this, SLOT(ButtonToggled(bool)));
connect(myCreateRBtn, SIGNAL (toggled(bool)), this, SLOT(ButtonToggled(bool)));
connect(myIdBtn, SIGNAL (toggled(bool)), this, SLOT(ButtonToggled(bool)));
connect(myAutoSearchChkBox, SIGNAL (toggled(bool)), this, SLOT(ButtonToggled(bool)));
myMoveRBtn->setChecked(TRUE);
myIdBtn->setOn(TRUE);
myAutoSearchChkBox->setChecked(TRUE);
return aFrame;
}
//================================================================================
/*!
* \brief SLOT called when any button is toggled
* \param bool - on or off
*/
//================================================================================
void SMESHGUI_MakeNodeAtPointDlg::ButtonToggled (bool on)
{
const QObject* aSender = sender();
if ( on ) {
if ( aSender == myCoordBtn ) // button to set coord by node selection
{
if ( myIdBtn->isEnabled() )
myIdBtn->setOn( !on );
}
else if ( aSender == myIdBtn ) // button to select a node to move
{
myCoordBtn->setOn( !on );
}
else if ( aSender == myMoveRBtn ) // move node method
{
myNodeToMoveGrp->setEnabled( TRUE );
}
else if ( aSender == myCreateRBtn ) // create node method
{
myNodeToMoveGrp->setEnabled( FALSE );
myCoordBtn->setOn( TRUE );
}
}
if ( aSender == myAutoSearchChkBox ) // automatic node search
{
if ( on ) {
myId->setText("");
myId->setReadOnly ( TRUE );
myIdBtn->setOn( FALSE );
myIdBtn->setEnabled( FALSE );
myCoordBtn->setOn( TRUE );
}
else {
myId->setReadOnly ( FALSE );
myIdBtn->setEnabled( TRUE );
}
}
}
//================================================================================
/*!
* \brief Constructor
*/
//================================================================================
SMESHGUI_MakeNodeAtPointOp::SMESHGUI_MakeNodeAtPointOp()
{
mySimulation = 0;
myDlg = new SMESHGUI_MakeNodeAtPointDlg;
myFilter = 0;
myHelpFileName = "mesh_through_point_page.html";
// connect signals and slots
connect(myDlg->myX, SIGNAL (valueChanged(double)), this, SLOT(redisplayPreview()));
connect(myDlg->myY, SIGNAL (valueChanged(double)), this, SLOT(redisplayPreview()));
connect(myDlg->myZ, SIGNAL (valueChanged(double)), this, SLOT(redisplayPreview()));
connect(myDlg->myId,SIGNAL (textChanged(const QString&)),SLOT(redisplayPreview()));
connect(myDlg->myPreviewChkBox, SIGNAL (toggled(bool)),SLOT(redisplayPreview()));
connect(myDlg->myAutoSearchChkBox,SIGNAL (toggled(bool)),SLOT(redisplayPreview()));
connect(myDlg->myMoveRBtn, SIGNAL (toggled(bool)),SLOT(redisplayPreview()));
connect(myDlg->myCreateRBtn, SIGNAL (toggled(bool)),SLOT(redisplayPreview()));
}
//=======================================================================
// function : startOperation()
// purpose : Init dialog fields, connect signals and slots, show dialog
//=======================================================================
void SMESHGUI_MakeNodeAtPointOp::startOperation()
{
myNoPreview = false;
myMeshActor = 0;
// init simulation with a current View
if ( mySimulation ) delete mySimulation;
mySimulation = new SMESHGUI_MeshEditPreview(SMESH::GetViewWindow( getSMESHGUI() ));
vtkProperty* aProp = vtkProperty::New();
aProp->SetRepresentationToWireframe();
aProp->SetColor(250, 0, 250);
aProp->SetPointSize(5);
aProp->SetLineWidth( SMESH::GetFloat("SMESH:element_width",1) + 1);
mySimulation->GetActor()->SetProperty(aProp);
aProp->Delete();
SMESHGUI_SelectionOp::startOperation();
// SalomeApp_TypeFilter depends on a current study
if ( myFilter ) delete myFilter;
QPtrList<SUIT_SelectionFilter> filters;
filters.append( new SalomeApp_TypeFilter((SalomeApp_Study*)study(), "SMESH" ));
TColStd_MapOfInteger vertexType;
vertexType.Add( TopAbs_VERTEX );
filters.append( new SMESH_NumberFilter("GEOM", TopAbs_VERTEX, 1, vertexType ));
myFilter = new SMESH_LogicalFilter( filters, SMESH_LogicalFilter::LO_OR );
activateSelection(); // set filters
myDlg->myX->SetValue(0);
myDlg->myY->SetValue(0);
myDlg->myZ->SetValue(0);
myDlg->myId->setText("");
myDlg->show();
onSelectionDone(); // init myMeshActor
if ( myMeshActor ) {
// myMeshOldDisplayMode = myMeshActor->GetRepresentation();
// myMeshActor->SetRepresentation( VTK_WIREFRAME );
myMeshActor->SetPointRepresentation(true);
SMESH::RepaintCurrentView();
redisplayPreview();
}
}
//================================================================================
/*!
* \brief Stops operation
*/
//================================================================================
void SMESHGUI_MakeNodeAtPointOp::stopOperation()
{
myNoPreview = true;
mySimulation->SetVisibility(false);
if ( myMeshActor ) {
// myMeshActor->SetRepresentation( myMeshOldDisplayMode );
myMeshActor->SetPointRepresentation(false);
SMESH::RepaintCurrentView();
myMeshActor = 0;
}
selectionMgr()->removeFilter( myFilter );
SMESHGUI_SelectionOp::stopOperation();
}
//================================================================================
/*!
* \brief perform it's intention action: move or create a node
*/
//================================================================================
bool SMESHGUI_MakeNodeAtPointOp::onApply()
{
if( isStudyLocked() )
return false;
if ( !myMeshActor ) {
SUIT_MessageBox::warn1( dlg(), tr( "SMESH_WRN_WARNING" ),
tr("INVALID_MESH"), tr( "SMESH_BUT_OK" ) );
dlg()->show();
return false;
}
if ( !isValid() ) { // node id is invalid
SUIT_MessageBox::warn1( dlg(), tr( "SMESH_WRN_WARNING" ),
tr("INVALID_ID"), tr( "SMESH_BUT_OK" ) );
dlg()->show();
return false;
}
try {
SMESH::SMESH_Mesh_var aMesh = SMESH::GetMeshByIO(myMeshActor->getIO());
if (aMesh->_is_nil()) {
QMessageBox::information(SMESHGUI::desktop(), tr("SMESH_ERROR"),
tr("SMESHG_NO_MESH"), QMessageBox::Ok);
return true;
}
SMESH::SMESH_MeshEditor_var aMeshEditor = aMesh->GetMeshEditor();
if (aMeshEditor->_is_nil())
return true;
int aResult = 0;
if ( myDlg->myCreateRBtn->isOn() )
{
aResult = aMeshEditor->AddNode(myDlg->myX->GetValue(),
myDlg->myY->GetValue(),
myDlg->myZ->GetValue());
}
else
{
int anId = myDlg->myId->text().toInt();
aResult = aMeshEditor->MoveClosestNodeToPoint(myDlg->myX->GetValue(),
myDlg->myY->GetValue(),
myDlg->myZ->GetValue(),
anId);
}
if (aResult)
{
myDlg->myId->setText("");
SALOME_ListIO aList;
selectionMgr()->setSelectedObjects(aList,false);
aList.Append(myMeshActor->getIO());
selectionMgr()->setSelectedObjects(aList,false);
SMESH::UpdateView();
}
}
catch (const SALOME::SALOME_Exception& S_ex) {
SalomeApp_Tools::QtCatchCorbaException(S_ex);
}
catch (...) {
}
return true;
}
//================================================================================
/*!
* \brief Check selected node id validity
*/
//================================================================================
bool SMESHGUI_MakeNodeAtPointOp::isValid()
{
bool ok = true;
if ( myMeshActor &&
myDlg->myMoveRBtn->isOn() &&
!myDlg->myAutoSearchChkBox->isChecked() )
{
ok = false;
int id = myDlg->myId->text().toInt();
if ( id > 0 )
if (SMDS_Mesh* aMesh = myMeshActor->GetObject()->GetMesh())
ok = aMesh->FindNode( id );
}
return ok;
}
//================================================================================
/*!
* \brief SLOT called when selection changed
*/
//================================================================================
void SMESHGUI_MakeNodeAtPointOp::onSelectionDone()
{
if ( !myDlg->isShown() || !myDlg->isEnabled() )
return;
try {
SALOME_ListIO aList;
selectionMgr()->selectedObjects(aList, SVTK_Viewer::Type());
if (aList.Extent() != 1)
return;
Handle(SALOME_InteractiveObject) anIO = aList.First();
SMESH_Actor* aMeshActor = SMESH::FindActorByEntry(anIO->getEntry());
if (!aMeshActor) { // coord by geom
if ( myDlg->myCoordBtn->isOn() ) {
GEOM::GEOM_Object_var geom = SMESH::IObjectToInterface<GEOM::GEOM_Object>(anIO);
if ( !geom->_is_nil() ) {
TopoDS_Vertex aShape;
if ( GEOMBase::GetShape(geom, aShape) &&
aShape.ShapeType() == TopAbs_VERTEX ) {
gp_Pnt P = BRep_Tool::Pnt(aShape);
myNoPreview = true;
myDlg->myX->SetValue(P.X());
myDlg->myY->SetValue(P.Y());
myDlg->myZ->SetValue(P.Z());
myNoPreview = false;
redisplayPreview();
}
}
return;
}
}
if ( !myMeshActor )
myMeshActor = aMeshActor;
QString aString;
int nbElems = SMESH::GetNameOfSelectedElements(selector(),anIO, aString);
if (nbElems == 1) {
if (SMDS_Mesh* aMesh = aMeshActor->GetObject()->GetMesh()) {
if (const SMDS_MeshNode* aNode = aMesh->FindNode(aString.toInt())) {
myNoPreview = true;
if ( myDlg->myCoordBtn->isOn() ) { // set coord
myDlg->myX->SetValue(aNode->X());
myDlg->myY->SetValue(aNode->Y());
myDlg->myZ->SetValue(aNode->Z());
myNoPreview = false;
redisplayPreview();
}
else if ( myDlg->myIdBtn->isOn() &&
myDlg->myIdBtn->isEnabled() ) { // set node to move
myDlg->myId->setText(aString);
myNoPreview = false;
redisplayPreview();
}
}
}
}
} catch (...) {
}
}
//================================================================================
/*!
* \brief update preview
*/
//================================================================================
void SMESHGUI_MakeNodeAtPointOp::redisplayPreview()
{
if ( myNoPreview )
return;
myNoPreview = true;
SMESH::MeshPreviewStruct_var aMeshPreviewStruct;
bool moveShown = false;
if ( myDlg->myMoveRBtn->isOn() && // Move method
myMeshActor)
{
const bool autoSearch = myDlg->myAutoSearchChkBox->isChecked();
const bool preview = myDlg->myPreviewChkBox->isChecked();
if ( autoSearch )
myDlg->myId->setText("");
if ( preview && ( autoSearch || isValid() ))
{
try {
SMESH::SMESH_Mesh_var aMesh = SMESH::GetMeshByIO(myMeshActor->getIO());
if (!aMesh->_is_nil()) {
SMESH::SMESH_MeshEditor_var aPreviewer = aMesh->GetMeshEditPreviewer();
if (!aPreviewer->_is_nil())
{
SUIT_OverrideCursor aWaitCursor;
// find id and/or just compute preview
int anId = aPreviewer->MoveClosestNodeToPoint(myDlg->myX->GetValue(),
myDlg->myY->GetValue(),
myDlg->myZ->GetValue(),
myDlg->myId->text().toInt());
if ( autoSearch ) { // set found id
QString idTxt("%1");
if ( anId > 0 )
idTxt = idTxt.arg( anId );
else
idTxt = "";
myDlg->myId->setText( idTxt );
}
if ( preview ) { // fill preview data
aMeshPreviewStruct = aPreviewer->GetPreviewData();
moveShown = ( anId > 0 );
}
}
}
}catch (...) {
}
}
}
if ( !moveShown )
{
aMeshPreviewStruct = new SMESH::MeshPreviewStruct();
aMeshPreviewStruct->nodesXYZ.length(1);
aMeshPreviewStruct->nodesXYZ[0].x = myDlg->myX->GetValue();
aMeshPreviewStruct->nodesXYZ[0].y = myDlg->myY->GetValue();
aMeshPreviewStruct->nodesXYZ[0].z = myDlg->myZ->GetValue();
aMeshPreviewStruct->elementTypes.length(1);
aMeshPreviewStruct->elementTypes[0].SMDS_ElementType = SMESH::NODE;
aMeshPreviewStruct->elementTypes[0].isPoly = false;
aMeshPreviewStruct->elementTypes[0].nbNodesInElement = 1;
aMeshPreviewStruct->elementConnectivities.length(1);
aMeshPreviewStruct->elementConnectivities[0] = 0;
}
// display data
if ( aMeshPreviewStruct.operator->() )
{
mySimulation->SetData(aMeshPreviewStruct._retn());
}
else
{
mySimulation->SetVisibility(false);
}
myNoPreview = false;
}
//================================================================================
/*!
* \brief Activate Node selection
*/
//================================================================================
void SMESHGUI_MakeNodeAtPointOp::activateSelection()
{
selectionMgr()->clearFilters();
SMESH::SetPointRepresentation(false);
selectionMgr()->installFilter( myFilter );
setSelectionMode( NodeSelection );
}
//================================================================================
/*!
* \brief Destructor
*/
//================================================================================
SMESHGUI_MakeNodeAtPointOp::~SMESHGUI_MakeNodeAtPointOp()
{
if ( myDlg ) delete myDlg;
if ( mySimulation ) delete mySimulation;
if ( myFilter ) delete myFilter;
}
//================================================================================
/*!
* \brief Gets dialog of this operation
* \retval LightApp_Dialog* - pointer to dialog of this operation
*/
//================================================================================
LightApp_Dialog* SMESHGUI_MakeNodeAtPointOp::dlg() const
{
return myDlg;
}
Reference in New Issue View Git Blame Copy Permalink