smesh/src/StdMeshersGUI/StdMeshersGUI_BlockRenumberCreator.cxx
2023-06-19 22:11:37 +01:00

516 lines
16 KiB
C++

// Copyright (C) 2007-2023 CEA, EDF, 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, or (at your option) any later version.
//
// 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
//
// File : StdMeshersGUI_BlockRenumberCreator.cxx
// Author : Open CASCADE S.A.S.
#include "StdMeshersGUI_BlockRenumberCreator.h"
#include "SMESHGUI.h"
#include "SMESHGUI_Utils.h"
#include "SMESHGUI_VTKUtils.h"
#include "SMESHGUI_HypothesesUtils.h"
#include CORBA_SERVER_HEADER(SMESH_BasicHypothesis)
#include <GEOMBase.h>
#include <LightApp_SelectionMgr.h>
#include <SALOMEDSClient_SObject.hxx>
#include <SALOMEDSClient_Study.hxx>
#include <SALOME_InteractiveObject.hxx>
#include <SALOME_ListIO.hxx>
#include <SUIT_ResourceMgr.h>
#include <SalomeApp_Tools.h>
#include <QApplication>
#include <QButtonGroup>
#include <QGridLayout>
#include <QGridLayout>
#include <QGroupBox>
#include <QHeaderView>
#include <QLabel>
#include <QLineEdit>
#include <QPushButton>
#include <QString>
#include <QTreeWidget>
#include <QTreeWidgetItem>
#define SPACING 6
#define MARGIN 11
namespace {
//================================================================================
/*!
* \brief Textual representation of GEOM_Object in QTreeWidget
*/
//================================================================================
QString toText( const GEOM::GEOM_Object_var& theGO,
const QString & /*theLabel*/,
QString& theEntry )
{
QString txt;
if ( !theGO->_is_nil() )
{
CORBA::String_var name = theGO->GetName();
CORBA::String_var entry = theGO->GetStudyEntry();
theEntry = entry.in();
txt = name.in();
//txt = theLabel + name.in() + " [" + theEntry + "]";
//txt += " ";//" (";
//txt += entry.in();
//txt += ")";
}
return txt;
}
//================================================================================
/*!
* \brief Find GEOM_Object by study entry
*/
//================================================================================
GEOM::GEOM_Object_var toGeom( const QString& entry )
{
return SMESH::EntryToInterface<GEOM::GEOM_Object>( entry );
}
//================================================================================
/*!
* \brief Find GEOM_Object in the tree
*/
//================================================================================
QTreeWidgetItem* findSolidInTree( const GEOM::GEOM_Object_var& go, QTreeWidget* tree )
{
if ( go->_is_nil() || !tree )
return nullptr;
for ( int i = 0; i < tree->topLevelItemCount(); ++i )
{
QTreeWidgetItem* item = tree->topLevelItem( i );
if ( item->data( 1, Qt::UserRole ).toString() == SMESH::toQStr( go->GetStudyEntry() ))
return item;
}
return nullptr;
}
}
//================================================================================
/*!
* \brief StdMeshersGUI_BlockRenumberCreator constructor
*/
//================================================================================
StdMeshersGUI_BlockRenumberCreator::StdMeshersGUI_BlockRenumberCreator(const QString& aHypType)
: StdMeshersGUI_StdHypothesisCreator( aHypType ),
mySolidFilter( "", TopAbs_SOLID, 1, TopAbs_SOLID ),
myVertexFilter( "", TopAbs_VERTEX, 1, TopAbs_VERTEX )
{
}
//================================================================================
/*!
* \brief StdMeshersGUI_BlockRenumberCreator destructor
*/
//================================================================================
StdMeshersGUI_BlockRenumberCreator::~StdMeshersGUI_BlockRenumberCreator()
{
SMESHGUI::selectionMgr()->clearFilters();
}
//================================================================================
/*!
* \brief Create widgets
*/
//================================================================================
QFrame* StdMeshersGUI_BlockRenumberCreator::buildFrame()
{
QFrame* fr = new QFrame();
//fr->setMinimumWidth(460);
QGridLayout* frLayout = new QGridLayout( fr );
frLayout->setSpacing( SPACING );
frLayout->setMargin( MARGIN );
// name
myName = 0;
int row = 0;
if( isCreation() )
{
myName = new QLineEdit( fr );
frLayout->addWidget( new QLabel( tr( "SMESH_NAME" ), fr ), row, 0 );
frLayout->addWidget( myName, row, 1 );
row++;
}
QGroupBox* groupBox = new QGroupBox( tr( "BLOCK_CS_GROUPBOX" ), fr );
frLayout->addWidget( groupBox, row, 0, 1, 2 );
QGridLayout* layout = new QGridLayout( groupBox );
layout->setSpacing( SPACING );
layout->setMargin( MARGIN );
layout->setColumnStretch( 0, 1 );
// tree
row = 0;
myBlockTree = new QTreeWidget( groupBox );
myBlockTree->setColumnCount( 2 );
myBlockTree->header()->hide();
// myBlockTree->setHeaderLabels( QStringList()
// << tr("COLUMN_BLOCK") << tr("COLUMN_V000") << tr("COLUMN_V001"));
layout->addWidget( myBlockTree, row, 0, row + 7, 1 );
// selection widgets
myButGroup = new QButtonGroup( groupBox );
myButGroup->setExclusive( true );
QLabel* label[3] = { new QLabel( tr( "SOLID" ), groupBox ),
new QLabel( tr( "V000" ), groupBox ),
new QLabel( tr( "V001" ), groupBox ) };
QIcon icon( SMESHGUI::resourceMgr()->loadPixmap("SMESH", tr("ICON_SELECT")));
for ( int i = 0; i < 3; ++i )
{
myShapeSelectBut[i] = new QPushButton( icon, "", groupBox );
myLineEdit [i] = new QLineEdit( groupBox );
layout->addWidget( label[i], row, 1 );
layout->addWidget( myShapeSelectBut[i], row, 2 );
layout->addWidget( myLineEdit[i], row, 3 );
myShapeSelectBut[i]->setCheckable( true );
myButGroup->addButton( myShapeSelectBut[i], i );
row++;
}
// buttons
myAddBut = new QPushButton( tr("ADD"), groupBox );
myModifBut = new QPushButton( tr("MODIFY"), groupBox );
myRemoveBut = new QPushButton( tr("SMESH_REMOVE"), groupBox );
layout->addWidget( myAddBut, row, 1, 1, 3 );
row++;
layout->addWidget( myModifBut, row, 1, 1, 3 );
row++;
layout->addWidget( myRemoveBut, row, 1, 1, 3 );
row++;
layout->setRowStretch( row, 1 );
LightApp_SelectionMgr* selMgr = SMESH::GetSelectionMgr( SMESHGUI::GetSMESHGUI() );
connect( selMgr, SIGNAL( currentSelectionChanged()), SLOT( onSelectionChange()));
connect( myBlockTree, SIGNAL( itemSelectionChanged() ), SLOT( onTreeSelectionChange()));
connect( myShapeSelectBut[0], SIGNAL( clicked(bool)), SLOT( onSelectBtnClick() ));
connect( myShapeSelectBut[1], SIGNAL( clicked(bool)), SLOT( onSelectBtnClick() ));
connect( myShapeSelectBut[2], SIGNAL( clicked(bool)), SLOT( onSelectBtnClick() ));
connect( myAddBut, SIGNAL( clicked() ), SLOT( onAddBtnClick() ));
connect( myModifBut, SIGNAL( clicked() ), SLOT( onModifBtnClick() ));
connect( myRemoveBut, SIGNAL( clicked() ), SLOT( onRemoveBtnClick() ));
return fr;
}
//================================================================================
/*!
* \brief Set myGO's to the tree
*/
//================================================================================
void StdMeshersGUI_BlockRenumberCreator::setBlockToTree(QTreeWidgetItem* solidItem)
{
if ( !myGO[0]->_is_nil() && !myGO[1]->_is_nil() && !myGO[2]->_is_nil() )
{
if ( !solidItem ) solidItem = new QTreeWidgetItem( myBlockTree );
solidItem->setExpanded( true );
QTreeWidgetItem* item = solidItem;
QFont boldFont = item->font( 0 );
boldFont.setBold( true );
QString entry, label[3] = { tr("BLOCK_LABEL"), tr("V000_LABEL"), tr("V001_LABEL") };
for ( int i = 0; i < 3; ++i )
{
if ( i > 0 && ! (item = solidItem->child( i - 1 )))
item = new QTreeWidgetItem( solidItem );
item->setText( 0, label[i] );
item->setText( 1, toText( myGO[i], label[i], entry ));
item->setData( 1, Qt::UserRole, entry );
item->setFont( 1, boldFont );
item->setToolTip( 1, entry );
}
myBlockTree->resizeColumnToContents( 0 );
myBlockTree->resizeColumnToContents( 1 );
}
for ( int i = 0; i < 3; ++i )
{
myGO[i] = GEOM::GEOM_Object::_nil();
myLineEdit[i]->setText("");
}
}
//================================================================================
/*!
* \brief Transfer parameters from hypothesis to widgets
*/
//================================================================================
void StdMeshersGUI_BlockRenumberCreator::retrieveParams() const
{
StdMeshersGUI_BlockRenumberCreator* me = const_cast<StdMeshersGUI_BlockRenumberCreator*>( this );
StdMeshers::StdMeshers_BlockRenumber_var h =
StdMeshers::StdMeshers_BlockRenumber::_narrow( initParamsHypothesis() );
if( myName )
me->myName->setText( hypName() );
me->myBlockTree->clear();
StdMeshers::blockcs_array_var blkArray = h->GetBlocksOrientation();
for ( CORBA::ULong i = 0; i < blkArray->length(); ++i )
{
me->myGO[0] = GEOM::GEOM_Object::_duplicate( blkArray[i].solid.in() );
me->myGO[1] = GEOM::GEOM_Object::_duplicate( blkArray[i].vertex000.in() );
me->myGO[2] = GEOM::GEOM_Object::_duplicate( blkArray[i].vertex001.in() );
me->setBlockToTree();
}
me->myShapeSelectBut[0]->click();
me->updateButtons();
}
//================================================================================
/*!
* \brief Transfer parameters from widgets to hypothesis
*/
//================================================================================
QString StdMeshersGUI_BlockRenumberCreator::storeParams() const
{
StdMeshers::StdMeshers_BlockRenumber_var h =
StdMeshers::StdMeshers_BlockRenumber::_narrow( hypothesis() );
try
{
if( isCreation() )
SMESH::SetName( SMESH::FindSObject( h ), myName->text().toUtf8().constData() );
StdMeshers::blockcs_array_var array = new StdMeshers::blockcs_array();
array->length( myBlockTree->topLevelItemCount() );
for ( int i = 0; i < myBlockTree->topLevelItemCount(); ++i )
{
StdMeshers::BlockCS& bcs = array[i];
QTreeWidgetItem* item0 = myBlockTree->topLevelItem( i );
QTreeWidgetItem* item1 = item0->child( 0 );
QTreeWidgetItem* item2 = item0->child( 1 );
bcs.solid = toGeom( item0->data( 1, Qt::UserRole ).toString() )._retn();
bcs.vertex000 = toGeom( item1->data( 1, Qt::UserRole ).toString() )._retn();
bcs.vertex001 = toGeom( item2->data( 1, Qt::UserRole ).toString() )._retn();
}
h->SetBlocksOrientation( array );
}
catch(const SALOME::SALOME_Exception& ex)
{
SalomeApp_Tools::QtCatchCorbaException(ex);
}
return "";
}
//================================================================================
/*!
* \brief Take selected object
*/
//================================================================================
void StdMeshersGUI_BlockRenumberCreator::onSelectionChange()
{
SALOME_ListIO list;
SMESHGUI::GetSMESHGUI()->selectionMgr()->selectedObjects( list );
int shapeID = myButGroup->checkedId();
if ( shapeID < 0 || shapeID > 2 )
shapeID = 0;
myGO[ shapeID ] = GEOM::GEOM_Object::_nil();
myLineEdit[ shapeID ]->clear();
if ( list.IsEmpty() )
return;
Handle(SALOME_InteractiveObject) io = list.First();
if ( !io->hasEntry() )
return;
myGO[ shapeID ] = SMESH::IObjectToInterface<GEOM::GEOM_Object>( io );
if ( !myGO[ shapeID ]->_is_nil() )
myLineEdit[ shapeID ]->setText( SMESH::toQStr( myGO[ shapeID ]->GetName() ));
updateButtons();
}
//================================================================================
/*!
* \brief Display selected block CS in myLineEdit's
*/
//================================================================================
void StdMeshersGUI_BlockRenumberCreator::onTreeSelectionChange()
{
QList<QTreeWidgetItem *> items = myBlockTree->selectedItems();
for ( QTreeWidgetItem* item : items )
{
if ( item->parent() )
item = item->parent();
QTreeWidgetItem* items[3] = { item, item->child( 0 ), item->child( 1 ) };
if ( items[1] && items[2] )
for ( int i = 0; i < 3; ++i )
{
myGO[i] = toGeom( items[i]->data( 1, Qt::UserRole ).toString() );
myLineEdit[i]->setText( items[i]->text( 1 ));
}
break;
}
updateButtons();
}
//================================================================================
/*!
* \brief Activate/deactivate buttons
*/
//================================================================================
void StdMeshersGUI_BlockRenumberCreator::updateButtons()
{
bool isSolidInTree = findSolidInTree( myGO[0], myBlockTree );
myAddBut ->setEnabled( !isSolidInTree &&
!myGO[0]->_is_nil() && !myGO[1]->_is_nil() && !myGO[2]->_is_nil() );
myModifBut ->setEnabled( isSolidInTree );
myRemoveBut->setEnabled( !myBlockTree->selectedItems().isEmpty() ||
myBlockTree->topLevelItemCount() == 1 );
}
//================================================================================
/*!
* \brief Install filter upon activation of another object selection
*/
//================================================================================
void StdMeshersGUI_BlockRenumberCreator::onSelectBtnClick()
{
int shapeID = myButGroup->checkedId();
LightApp_SelectionMgr* selMgr = SMESHGUI::selectionMgr();
selMgr->clearFilters();
selMgr->installFilter( shapeID > 0 ? &myVertexFilter : &mySolidFilter );
}
//================================================================================
/*!
* \brief Add shapes to the tree
*/
//================================================================================
void StdMeshersGUI_BlockRenumberCreator::onAddBtnClick()
{
setBlockToTree();
updateButtons();
}
//================================================================================
/*!
* \brief Modify a current block
*/
//================================================================================
void StdMeshersGUI_BlockRenumberCreator::onModifBtnClick()
{
if ( QTreeWidgetItem* item = findSolidInTree( myGO[0], myBlockTree ))
{
setBlockToTree( item );
}
updateButtons();
}
//================================================================================
/*!
* \brief Remove selected block from the tree
*/
//================================================================================
void StdMeshersGUI_BlockRenumberCreator::onRemoveBtnClick()
{
QList<QTreeWidgetItem *> items = myBlockTree->selectedItems();
if ( items.isEmpty() && myBlockTree->topLevelItemCount() == 1 )
items.push_back( myBlockTree->topLevelItem( 0 ));
for ( QTreeWidgetItem* item : items )
{
if ( item->parent() )
item = item->parent();
delete item;
}
updateButtons();
}
//================================================================================
/*!
* \brief Validate parameters
*/
//================================================================================
bool StdMeshersGUI_BlockRenumberCreator::checkParams( QString& /*msg*/ ) const
{
return true;
}
//================================================================================
/*!
* \brief Returns a name of help page
*/
//================================================================================
QString StdMeshersGUI_BlockRenumberCreator::helpPage() const
{
return "3d_meshing_hypo.html#renumber-hypothesis";
}
//================================================================================
/*!
* \brief Type name
*/
//================================================================================
QString StdMeshersGUI_BlockRenumberCreator::hypTypeName( const QString& ) const
{
return "BLOCK_RENUMBER";
}