ForkMaster/smesh
1
0
Fork 0
mirror of https://git.salome-platform.org/gitpub/modules/smesh.git synced 2024-11-16 02:28:34 +05:00
Code Issues Packages Projects Releases Wiki Activity
96313c9c5d
smesh/src/SMESHGUI/SMESHGUI_MultiEditDlg.cxx
vsr 0635c9fc80 Merge from BR_V5_DEV 16Feb09
2009-02-17 05:27:49 +00:00

1452 lines
46 KiB
C++
Executable File
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
//
// File : SMESHGUI_MultiEditDlg.cxx
// Author : Sergey LITONIN, Open CASCADE S.A.S.
// SMESH includes
//
#include "SMESHGUI_MultiEditDlg.h"
#include "SMESHGUI.h"
#include "SMESHGUI_Filter.h"
#include "SMESHGUI_FilterDlg.h"
#include "SMESHGUI_Utils.h"
#include "SMESHGUI_VTKUtils.h"
#include "SMESHGUI_MeshUtils.h"
#include "SMESHGUI_FilterUtils.h"
#include "SMESHGUI_SpinBox.h"
#include <SMESH_Actor.h>
#include <SMESH_TypeFilter.hxx>
#include <SMDS_Mesh.hxx>
// SALOME GUI includes
#include <SUIT_ResourceMgr.h>
#include <SUIT_Desktop.h>
#include <SUIT_Session.h>
#include <SUIT_MessageBox.h>
#include <LightApp_SelectionMgr.h>
#include <LightApp_Application.h>
#include <SALOME_ListIO.hxx>
#include <SALOME_ListIteratorOfListIO.hxx>
#include <SVTK_Selector.h>
#include <SVTK_ViewWindow.h>
#include <VTKViewer_CellLocationsArray.h>
// OCCT includes
#include <TColStd_IndexedMapOfInteger.hxx>
#include <TColStd_DataMapOfIntegerInteger.hxx>
#include <TColStd_MapIteratorOfMapOfInteger.hxx>
// VTK includes
#include <vtkIdList.h>
#include <vtkCellArray.h>
#include <vtkUnsignedCharArray.h>
#include <vtkUnstructuredGrid.h>
#include <vtkDataSetMapper.h>
#include <vtkProperty.h>
// Qt includes
#include <QLabel>
#include <QVBoxLayout>
#include <QHBoxLayout>
#include <QGridLayout>
#include <QListWidget>
#include <QCheckBox>
#include <QComboBox>
#include <QGroupBox>
#include <QLineEdit>
#include <QPushButton>
#include <QButtonGroup>
#include <QRadioButton>
#include <QKeyEvent>
// IDL includes
#include <SALOMEconfig.h>
#include CORBA_SERVER_HEADER(SMESH_Group)
#define SPACING 6
#define MARGIN 11
/*!
* Class : SMESHGUI_MultiEditDlg
* Description : Description : Inversion of the diagonal of a pseudo-quadrangle formed by
* 2 neighboring triangles with 1 common edge
*/
//=======================================================================
// name : SMESHGUI_MultiEditDlg::SMESHGUI_MultiEditDlg
// Purpose : Constructor
//=======================================================================
SMESHGUI_MultiEditDlg
::SMESHGUI_MultiEditDlg(SMESHGUI* theModule,
const int theMode,
const bool the3d2d):
QDialog(SMESH::GetDesktop(theModule)),
mySelector(SMESH::GetViewWindow(theModule)->GetSelector()),
mySelectionMgr(SMESH::GetSelectionMgr(theModule)),
mySMESHGUI(theModule)
{
setModal(false);
setAttribute(Qt::WA_DeleteOnClose, true);
myFilterDlg = 0;
myEntityType = 0;
myFilterType = theMode;
QVBoxLayout* aDlgLay = new QVBoxLayout(this);
aDlgLay->setMargin(MARGIN);
aDlgLay->setSpacing(SPACING);
QWidget* aMainFrame = createMainFrame (this, the3d2d);
QWidget* aBtnFrame = createButtonFrame(this);
aDlgLay->addWidget(aMainFrame);
aDlgLay->addWidget(aBtnFrame);
Init();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::createMainFrame
// Purpose : Create frame containing dialog's input fields
//=======================================================================
QWidget* SMESHGUI_MultiEditDlg::createMainFrame (QWidget* theParent, const bool the3d2d)
{
QWidget* aMainGrp = new QWidget(theParent);
QVBoxLayout* aMainGrpLayout = new QVBoxLayout(aMainGrp);
aMainGrpLayout->setMargin(0);
aMainGrpLayout->setSpacing(SPACING);
QPixmap aPix (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_SELECT")));
// "Selected cells" group
mySelGrp = new QGroupBox(aMainGrp);
QGridLayout* mySelGrpLayout = new QGridLayout(mySelGrp);
mySelGrpLayout->setMargin(MARGIN);
mySelGrpLayout->setSpacing(SPACING);
myEntityTypeGrp = 0;
if (the3d2d) {
QGroupBox* aEntityTypeGrp = new QGroupBox(tr("SMESH_ELEMENTS_TYPE"), mySelGrp);
myEntityTypeGrp = new QButtonGroup(mySelGrp);
QHBoxLayout* aEntityLayout = new QHBoxLayout(aEntityTypeGrp);
aEntityLayout->setMargin(MARGIN);
aEntityLayout->setSpacing(SPACING);
QRadioButton* aFaceRb = new QRadioButton(tr("SMESH_FACE"), aEntityTypeGrp);
QRadioButton* aVolumeRb = new QRadioButton(tr("SMESH_VOLUME"), aEntityTypeGrp);
aEntityLayout->addWidget(aFaceRb);
aEntityLayout->addWidget(aVolumeRb);
myEntityTypeGrp->addButton(aFaceRb, 0);
myEntityTypeGrp->addButton(aVolumeRb, 1);
aFaceRb->setChecked(true);
myEntityType = myEntityTypeGrp->checkedId();
mySelGrpLayout->addWidget(aEntityTypeGrp, 0, 0, 1, 2);
}
myListBox = new QListWidget(mySelGrp);
myListBox->setSelectionMode(QListWidget::ExtendedSelection);
myListBox->setSizePolicy(QSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding));
myListBox->installEventFilter(this);
myFilterBtn = new QPushButton(tr("FILTER"), mySelGrp);
myAddBtn = new QPushButton(tr("ADD"), mySelGrp);
myRemoveBtn = new QPushButton(tr("REMOVE"), mySelGrp);
mySortBtn = new QPushButton(tr("SORT_LIST"), mySelGrp);
int row = mySelGrpLayout->rowCount();
mySelGrpLayout->addWidget(myListBox, row, 0, 6, 1);
mySelGrpLayout->addWidget(myFilterBtn, row, 1);
mySelGrpLayout->addWidget(myAddBtn, row+2, 1);
mySelGrpLayout->addWidget(myRemoveBtn, row+3, 1);
mySelGrpLayout->addWidget(mySortBtn, row+5, 1);
mySelGrpLayout->setRowMinimumHeight(row+1, 10);
mySelGrpLayout->setRowMinimumHeight(row+4, 10);
mySelGrpLayout->setRowStretch(row+1, 5);
mySelGrpLayout->setRowStretch(row+4, 5);
myToAllChk = new QCheckBox(tr("TO_ALL"), mySelGrp);
mySelGrpLayout->addWidget(myToAllChk, mySelGrpLayout->rowCount(), 0,
1, mySelGrpLayout->columnCount());
// Split/Join criterion group
myCriterionGrp = new QGroupBox(tr("SPLIT_JOIN_CRITERION"), aMainGrp);
QVBoxLayout* aCriterionLayout = new QVBoxLayout(myCriterionGrp);
aCriterionLayout->setMargin(MARGIN);
aCriterionLayout->setSpacing(SPACING);
myChoiceWidget = new QWidget(myCriterionGrp);
myGroupChoice = new QButtonGroup(myChoiceWidget);
QVBoxLayout* aGroupChoiceLayout = new QVBoxLayout(myChoiceWidget);
aGroupChoiceLayout->setMargin(0);
aGroupChoiceLayout->setSpacing(SPACING);
QRadioButton* aDiag13RB = new QRadioButton(tr("USE_DIAGONAL_1_3"), myChoiceWidget);
QRadioButton* aDiag24RB = new QRadioButton(tr("USE_DIAGONAL_2_4"), myChoiceWidget);
QRadioButton* aNumFuncRB = new QRadioButton(tr("USE_NUMERIC_FUNC"), myChoiceWidget);
aGroupChoiceLayout->addWidget(aDiag13RB);
aGroupChoiceLayout->addWidget(aDiag24RB);
aGroupChoiceLayout->addWidget(aNumFuncRB);
myGroupChoice->addButton(aDiag13RB, 0);
myGroupChoice->addButton(aDiag24RB, 1);
myGroupChoice->addButton(aNumFuncRB, 2);
aDiag13RB->setChecked(true);
myComboBoxFunctor = new QComboBox(myCriterionGrp);
myComboBoxFunctor->addItem(tr("ASPECTRATIO_ELEMENTS"));
myComboBoxFunctor->addItem(tr("MINIMUMANGLE_ELEMENTS"));
myComboBoxFunctor->addItem(tr("SKEW_ELEMENTS"));
myComboBoxFunctor->addItem(tr("AREA_ELEMENTS"));
//myComboBoxFunctor->addItem(tr("LENGTH2D_EDGES")); // for existing elements only
//myComboBoxFunctor->addItem(tr("MULTI2D_BORDERS")); // for existing elements only
myComboBoxFunctor->setCurrentIndex(0);
aCriterionLayout->addWidget(myChoiceWidget);
aCriterionLayout->addWidget(myComboBoxFunctor);
myCriterionGrp->hide();
myChoiceWidget->hide();
myComboBoxFunctor->setEnabled(false);
// "Select from" group
QGroupBox* aGrp = new QGroupBox(tr("SELECT_FROM"), aMainGrp);
QGridLayout* aGrpLayout = new QGridLayout(aGrp);
aGrpLayout->setMargin(MARGIN);
aGrpLayout->setSpacing(SPACING);
mySubmeshChk = new QCheckBox(tr("SMESH_SUBMESH"), aGrp);
mySubmeshBtn = new QPushButton(aGrp);
mySubmesh = new QLineEdit(aGrp);
mySubmesh->setReadOnly(true);
mySubmeshBtn->setIcon(aPix);
myGroupChk = new QCheckBox(tr("SMESH_GROUP"), aGrp);
myGroupBtn = new QPushButton(aGrp);
myGroup = new QLineEdit(aGrp);
myGroup->setReadOnly(true);
myGroupBtn->setIcon(aPix);
aGrpLayout->addWidget(mySubmeshChk,0,0);
aGrpLayout->addWidget(mySubmeshBtn,0,1);
aGrpLayout->addWidget(mySubmesh,0,2);
aGrpLayout->addWidget(myGroupChk,1,0);
aGrpLayout->addWidget(myGroupBtn,1,1);
aGrpLayout->addWidget(myGroup,1,2);
aMainGrpLayout->addWidget(mySelGrp);
aMainGrpLayout->addWidget(myCriterionGrp);
aMainGrpLayout->addWidget(aGrp);
return aMainGrp;
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::createButtonFrame
// Purpose : Create frame containing buttons
//=======================================================================
QWidget* SMESHGUI_MultiEditDlg::createButtonFrame (QWidget* theParent)
{
QGroupBox* aFrame = new QGroupBox(theParent);
myOkBtn = new QPushButton (tr("SMESH_BUT_APPLY_AND_CLOSE"), aFrame);
myApplyBtn = new QPushButton (tr("SMESH_BUT_APPLY"), aFrame);
myCloseBtn = new QPushButton (tr("SMESH_BUT_CLOSE"), aFrame);
myHelpBtn = new QPushButton (tr("SMESH_BUT_HELP"), aFrame);
QHBoxLayout* aLay = new QHBoxLayout (aFrame);
aLay->setMargin(MARGIN);
aLay->setSpacing(SPACING);
aLay->addWidget(myOkBtn);
aLay->addSpacing(10);
aLay->addWidget(myApplyBtn);
aLay->addSpacing(10);
aLay->addStretch();
aLay->addWidget(myCloseBtn);
aLay->addWidget(myHelpBtn);
return aFrame;
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::isValid
// Purpose : Verify validity of input data
//=======================================================================
bool SMESHGUI_MultiEditDlg::isValid (const bool /*theMess*/)
{
return (!myMesh->_is_nil() &&
(myListBox->count() > 0 || (myToAllChk->isChecked() && myActor)));
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::~SMESHGUI_MultiEditDlg
// Purpose : Destructor
//=======================================================================
SMESHGUI_MultiEditDlg::~SMESHGUI_MultiEditDlg()
{
if (myFilterDlg != 0)
{
myFilterDlg->setParent(0);
delete myFilterDlg;
}
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::eventFilter
// Purpose : event filter
//=======================================================================
bool SMESHGUI_MultiEditDlg::eventFilter (QObject* object, QEvent* event)
{
if (object == myListBox && event->type() == QEvent::KeyPress) {
QKeyEvent* ke = (QKeyEvent*)event;
if (ke->key() == Qt::Key_Delete)
onRemoveBtn();
}
return QDialog::eventFilter(object, event);
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::getNumericalFunctor
// Purpose :
//=======================================================================
SMESH::NumericalFunctor_ptr SMESHGUI_MultiEditDlg::getNumericalFunctor()
{
SMESH::NumericalFunctor_var aNF = SMESH::NumericalFunctor::_nil();
SMESH::FilterManager_var aFilterMgr = SMESH::GetFilterManager();
if (aFilterMgr->_is_nil())
return aNF._retn();
if (myComboBoxFunctor->currentText() == tr("ASPECTRATIO_ELEMENTS"))
aNF = aFilterMgr->CreateAspectRatio();
else if (myComboBoxFunctor->currentText() == tr("WARP_ELEMENTS"))
aNF = aFilterMgr->CreateWarping();
else if (myComboBoxFunctor->currentText() == tr("MINIMUMANGLE_ELEMENTS"))
aNF = aFilterMgr->CreateMinimumAngle();
else if (myComboBoxFunctor->currentText() == tr("TAPER_ELEMENTS"))
aNF = aFilterMgr->CreateTaper();
else if (myComboBoxFunctor->currentText() == tr("SKEW_ELEMENTS"))
aNF = aFilterMgr->CreateSkew();
else if (myComboBoxFunctor->currentText() == tr("AREA_ELEMENTS"))
aNF = aFilterMgr->CreateArea();
else if (myComboBoxFunctor->currentText() == tr("LENGTH2D_EDGES"))
aNF = aFilterMgr->CreateLength2D();
else if (myComboBoxFunctor->currentText() == tr("MULTI2D_BORDERS"))
aNF = aFilterMgr->CreateMultiConnection2D();
else;
return aNF._retn();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::Init
// Purpose : Init dialog fields, connect signals and slots, show dialog
//=======================================================================
void SMESHGUI_MultiEditDlg::Init()
{
mySMESHGUI->SetActiveDialogBox((QDialog*)this);
myListBox->clear();
myIds.Clear();
myBusy = false;
myActor = 0;
emit ListContensChanged();
// main buttons
connect(myOkBtn, SIGNAL(clicked()), SLOT(onOk()));
connect(myCloseBtn, SIGNAL(clicked()), SLOT(onClose()));
connect(myApplyBtn, SIGNAL(clicked()), SLOT(onApply()));
connect(myHelpBtn, SIGNAL(clicked()), SLOT(onHelp()));
// selection and SMESHGUI
connect(mySelectionMgr, SIGNAL(currentSelectionChanged()), SLOT(onSelectionDone()));
connect(mySMESHGUI, SIGNAL(SignalDeactivateActiveDialog()), SLOT(onDeactivate()));
connect(mySMESHGUI, SIGNAL(SignalCloseAllDialogs()), SLOT(onClose()));
// dialog controls
connect(myFilterBtn, SIGNAL(clicked()), SLOT(onFilterBtn() ));
connect(myAddBtn, SIGNAL(clicked()), SLOT(onAddBtn() ));
connect(myRemoveBtn, SIGNAL(clicked()), SLOT(onRemoveBtn() ));
connect(mySortBtn, SIGNAL(clicked()), SLOT(onSortListBtn()));
connect(mySubmeshChk, SIGNAL(stateChanged(int)), SLOT(onSubmeshChk()));
connect(myGroupChk, SIGNAL(stateChanged(int)), SLOT(onGroupChk() ));
connect(myToAllChk, SIGNAL(stateChanged(int)), SLOT(onToAllChk() ));
if (myEntityTypeGrp)
connect(myEntityTypeGrp, SIGNAL(buttonClicked(int)), SLOT(on3d2dChanged(int)));
connect(myListBox, SIGNAL(itemSelectionChanged()), SLOT(onListSelectionChanged()));
onSelectionDone();
// set selection mode
setSelectionMode();
updateButtons();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::onOk
// Purpose : SLOT called when "Ok" button pressed.
// Assign filters VTK viewer and close dialog
//=======================================================================
void SMESHGUI_MultiEditDlg::onOk()
{
if (onApply())
onClose();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::getIds
// Purpose : Retrive identifiers from list box
//=======================================================================
SMESH::long_array_var SMESHGUI_MultiEditDlg::getIds()
{
SMESH::long_array_var anIds = new SMESH::long_array;
if (myToAllChk->isChecked())
{
myIds.Clear();
SMESH_Actor * anActor = SMESH::FindActorByObject(myMesh);
if (!anActor)
anActor = myActor;
if (anActor != 0)
{
// skl 07.02.2006
SMDS_Mesh* aMesh = myActor->GetObject()->GetMesh();
if( myFilterType == SMESH::TriaFilter ||
myFilterType == SMESH::QuadFilter ||
myFilterType == SMESH::FaceFilter ) {
SMDS_FaceIteratorPtr it = aMesh->facesIterator();
while(it->more()) {
const SMDS_MeshFace* f = it->next();
if(myFilterType == SMESH::FaceFilter) {
myIds.Add(f->GetID());
}
else if( myFilterType==SMESH::TriaFilter &&
( f->NbNodes()==3 || f->NbNodes()==6 ) ) {
myIds.Add(f->GetID());
}
else if( myFilterType==SMESH::QuadFilter &&
( f->NbNodes()==4 || f->NbNodes()==8 ) ) {
myIds.Add(f->GetID());
}
}
}
else if(myFilterType == SMESH::VolumeFilter) {
SMDS_VolumeIteratorPtr it = aMesh->volumesIterator();
while(it->more()) {
const SMDS_MeshVolume* f = it->next();
myIds.Add(f->GetID());
}
}
/* commented by skl 07.02.2006
TVisualObjPtr aVisualObj = anActor->GetObject();
vtkUnstructuredGrid* aGrid = aVisualObj->GetUnstructuredGrid();
if (aGrid != 0) {
for (int i = 0, n = aGrid->GetNumberOfCells(); i < n; i++) {
vtkCell* aCell = aGrid->GetCell(i);
if (aCell != 0) {
vtkTriangle* aTri = vtkTriangle::SafeDownCast(aCell);
vtkQuad* aQua = vtkQuad::SafeDownCast(aCell);
vtkPolygon* aPG = vtkPolygon::SafeDownCast(aCell);
vtkCell3D* a3d = vtkCell3D::SafeDownCast(aCell);
vtkConvexPointSet* aPH = vtkConvexPointSet::SafeDownCast(aCell);
if (aTri && myFilterType == SMESHGUI_TriaFilter ||
aQua && myFilterType == SMESHGUI_QuadFilter ||
(aTri || aQua || aPG) && myFilterType == SMESHGUI_FaceFilter ||
(a3d || aPH) && myFilterType == SMESHGUI_VolumeFilter) {
int anObjId = aVisualObj->GetElemObjId(i);
myIds.Add(anObjId);
}
}
}
}
*/
}
}
anIds->length(myIds.Extent());
TColStd_MapIteratorOfMapOfInteger anIter(myIds);
for (int i = 0; anIter.More(); anIter.Next() )
{
anIds[ i++ ] = anIter.Key();
}
return anIds._retn();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::onClose
// Purpose : SLOT called when "Close" button pressed. Close dialog
//=======================================================================
void SMESHGUI_MultiEditDlg::onClose()
{
if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
aViewWindow->SetSelectionMode(ActorSelection);
disconnect(mySelectionMgr, 0, this, 0);
disconnect(mySMESHGUI, 0, this, 0);
mySMESHGUI->ResetState();
SMESH::RemoveFilters();
SMESH::SetPickable();
//mySelectionMgr->clearSelected();
mySelectionMgr->clearFilters();
reject();
}
//=================================================================================
// function : onHelp()
// purpose :
//=================================================================================
void SMESHGUI_MultiEditDlg::onHelp()
{
LightApp_Application* app = (LightApp_Application*)(SUIT_Session::session()->activeApplication());
if (app)
app->onHelpContextModule(mySMESHGUI ? app->moduleName(mySMESHGUI->moduleName()) : QString(""), myHelpFileName);
else {
QString platform;
#ifdef WIN32
platform = "winapplication";
#else
platform = "application";
#endif
SUIT_MessageBox::warning(this, tr("WRN_WARNING"),
tr("EXTERNAL_BROWSER_CANNOT_SHOW_PAGE").
arg(app->resourceMgr()->stringValue("ExternalBrowser",
platform)).
arg(myHelpFileName));
}
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::onSelectionDone
// Purpose : SLOT called when selection changed
//=======================================================================
void SMESHGUI_MultiEditDlg::onSelectionDone()
{
if (myBusy || !isEnabled()) return;
myBusy = true;
const SALOME_ListIO& aList = mySelector->StoredIObjects();
int nbSel = aList.Extent();
myListBox->clearSelection();
if (mySubmeshChk->isChecked() || myGroupChk->isChecked()) {
QLineEdit* aNameEdit = mySubmeshChk->isChecked() ? mySubmesh : myGroup;
if (nbSel == 1) {
Handle(SALOME_InteractiveObject) anIO = aList.First();
QString aName = "";
SMESH::GetNameOfSelectedIObjects(mySelectionMgr, aName);
anIO.IsNull() ? aNameEdit->clear() : aNameEdit->setText(aName);
if (mySubmeshChk->isChecked()) {
SMESH::SMESH_subMesh_var aSubMesh =
SMESH::IObjectToInterface<SMESH::SMESH_subMesh>(anIO);
if (!aSubMesh->_is_nil())
myMesh = aSubMesh->GetFather();
} else {
SMESH::SMESH_GroupBase_var aGroup =
SMESH::IObjectToInterface<SMESH::SMESH_GroupBase>(anIO);
if (!aGroup->_is_nil())
myMesh = aGroup->GetMesh();
}
} else if (nbSel > 1) {
QString aStr = mySubmeshChk->isChecked() ?
tr("SMESH_SUBMESH_SELECTED") : tr("SMESH_GROUP_SELECTED");
aNameEdit->setText(aStr.arg(nbSel));
} else {
aNameEdit->clear();
}
} else if (nbSel > 0) {
QString aListStr = "";
Handle(SALOME_InteractiveObject) anIO = aList.First();
int aNbItems = SMESH::GetNameOfSelectedElements(mySelector,anIO,aListStr);
if (aNbItems > 0) {
QStringList anElements = aListStr.split(" ", QString::SkipEmptyParts);
for (QStringList::iterator it = anElements.begin(); it != anElements.end(); ++it) {
QList<QListWidgetItem*> items = myListBox->findItems(*it, Qt::MatchExactly);
QListWidgetItem* anItem;
foreach(anItem, items)
anItem->setSelected(true);
}
}
myMesh = SMESH::GetMeshByIO(anIO);
}
if (nbSel > 0) {
myActor = SMESH::FindActorByEntry(aList.First()->getEntry());
if (!myActor)
myActor = SMESH::FindActorByObject(myMesh);
SVTK_Selector* aSelector = SMESH::GetSelector();
Handle(VTKViewer_Filter) aFilter = aSelector->GetFilter(myFilterType);
if (!aFilter.IsNull())
aFilter->SetActor(myActor);
}
myBusy = false;
updateButtons();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::onDeactivate
// Purpose : SLOT called when dialog must be deativated
//=======================================================================
void SMESHGUI_MultiEditDlg::onDeactivate()
{
setEnabled(false);
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::enterEvent
// Purpose : Event filter
//=======================================================================
void SMESHGUI_MultiEditDlg::enterEvent (QEvent*)
{
if (!isEnabled()) {
mySMESHGUI->EmitSignalDeactivateDialog();
setEnabled(true);
setSelectionMode();
}
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::closeEvent
// Purpose :
//=======================================================================
void SMESHGUI_MultiEditDlg::closeEvent (QCloseEvent*)
{
onClose();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::hideEvent
// Purpose : caused by ESC key
//=======================================================================
void SMESHGUI_MultiEditDlg::hideEvent (QHideEvent*)
{
if (!isMinimized())
onClose();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::onFilterBtn
// Purpose : SLOT. Called when "Filter" button pressed.
// Start "Selection filters" dialog
//=======================================================================
void SMESHGUI_MultiEditDlg::onFilterBtn()
{
if (myFilterDlg == 0) {
myFilterDlg = new SMESHGUI_FilterDlg( mySMESHGUI, entityType() ? SMESH::VOLUME : SMESH::FACE);
connect(myFilterDlg, SIGNAL(Accepted()), SLOT(onFilterAccepted()));
} else {
myFilterDlg->Init(entityType() ? SMESH::VOLUME : SMESH::FACE);
}
myFilterDlg->SetSelection();
myFilterDlg->SetMesh(myMesh);
myFilterDlg->SetSourceWg(myListBox);
myFilterDlg->show();
}
//=======================================================================
// name : onFilterAccepted()
// Purpose : SLOT. Called when Filter dlg closed with OK button.
// Uncheck "Select submesh" and "Select group" checkboxes
//=======================================================================
void SMESHGUI_MultiEditDlg::onFilterAccepted()
{
myIds.Clear();
for (int i = 0, n = myListBox->count(); i < n; i++)
myIds.Add(myListBox->item(i)->text().toInt());
emit ListContensChanged();
if (mySubmeshChk->isChecked() || myGroupChk->isChecked()) {
mySubmeshChk->blockSignals(true);
myGroupChk->blockSignals(true);
mySubmeshChk->setChecked(false);
myGroupChk->setChecked(false);
mySubmeshChk->blockSignals(false);
myGroupChk->blockSignals(false);
}
updateButtons();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::isIdValid
// Purpose : Verify whether Id of element satisfies to filters from viewer
//=======================================================================
bool SMESHGUI_MultiEditDlg::isIdValid (const int theId) const
{
SVTK_Selector* aSelector = SMESH::GetSelector();
Handle(SMESHGUI_Filter) aFilter =
Handle(SMESHGUI_Filter)::DownCast(aSelector->GetFilter(myFilterType));
return (!aFilter.IsNull() && aFilter->IsObjValid(theId));
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::onAddBtn
// Purpose : SLOT. Called when "Add" button pressed.
// Add selected in viewer entities in list box
//=======================================================================
void SMESHGUI_MultiEditDlg::onAddBtn()
{
const SALOME_ListIO& aList = mySelector->StoredIObjects();
int nbSelected = aList.Extent();
if (nbSelected == 0)
return;
TColStd_IndexedMapOfInteger toBeAdded;
if (!mySubmeshChk->isChecked() && !myGroupChk->isChecked()) {
if (nbSelected > 0)
mySelector->GetIndex(aList.First(),toBeAdded);
} else if (mySubmeshChk->isChecked()) {
SALOME_ListIteratorOfListIO anIter(aList);
for ( ; anIter.More(); anIter.Next()) {
SMESH::SMESH_subMesh_var aSubMesh =
SMESH::IObjectToInterface<SMESH::SMESH_subMesh>(anIter.Value());
if (!aSubMesh->_is_nil()) {
if (aSubMesh->GetFather()->GetId() == myMesh->GetId()) {
SMESH::long_array_var anIds = aSubMesh->GetElementsId();
for (int i = 0, n = anIds->length(); i < n; i++) {
if (isIdValid(anIds[ i ]))
toBeAdded.Add(anIds[ i ]);
}
}
}
}
} else if (myGroupChk->isChecked()) {
SALOME_ListIteratorOfListIO anIter(aList);
for ( ; anIter.More(); anIter.Next()) {
SMESH::SMESH_GroupBase_var aGroup =
SMESH::IObjectToInterface<SMESH::SMESH_GroupBase>(anIter.Value());
if (!aGroup->_is_nil() && (aGroup->GetType() == SMESH::FACE &&
entityType() == 0 || aGroup->GetType() == SMESH::VOLUME &&
entityType() == 1)) {
if (aGroup->GetMesh()->GetId() == myMesh->GetId()) {
SMESH::long_array_var anIds = aGroup->GetListOfID();
for (int i = 0, n = anIds->length(); i < n; i++) {
if (isIdValid(anIds[ i ]))
toBeAdded.Add(anIds[ i ]);
}
}
}
}
} else {
}
myBusy = true;
bool isGroupOrSubmesh = (mySubmeshChk->isChecked() || myGroupChk->isChecked());
mySubmeshChk->setChecked(false);
myGroupChk->setChecked(false);
for(int i = 1; i <= toBeAdded.Extent(); i++)
if (myIds.Add(toBeAdded(i))) {
QListWidgetItem* item = new QListWidgetItem(QString("%1").arg(toBeAdded(i)));
myListBox->addItem(item);
item->setSelected(true);
}
myBusy = false;
emit ListContensChanged();
if (isGroupOrSubmesh)
onListSelectionChanged();
updateButtons();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::updateButtons
// Purpose : Enable/disable buttons of dialog in accordance with current state
//=======================================================================
void SMESHGUI_MultiEditDlg::updateButtons()
{
bool isOk = isValid(false);
myOkBtn->setEnabled(isOk);
myApplyBtn->setEnabled(isOk);
bool isListBoxNonEmpty = myListBox->count() > 0;
bool isToAll = myToAllChk->isChecked();
myFilterBtn->setEnabled(!isToAll);
myRemoveBtn->setEnabled(isListBoxNonEmpty && !isToAll);
mySortBtn->setEnabled(isListBoxNonEmpty &&!isToAll);
const SALOME_ListIO& aList = mySelector->StoredIObjects();
if (isToAll ||
myMesh->_is_nil() ||
aList.Extent() < 1 ||
(SMESH::IObjectToInterface<SMESH::SMESH_subMesh>(aList.First())->_is_nil() &&
SMESH::IObjectToInterface<SMESH::SMESH_GroupBase>(aList.First())->_is_nil() &&
SMESH::IObjectToInterface<SMESH::SMESH_Mesh>(aList.First())->_is_nil()))
myAddBtn->setEnabled(false);
else
myAddBtn->setEnabled(true);
mySubmeshChk->setEnabled(!isToAll);
mySubmeshBtn->setEnabled(mySubmeshChk->isChecked());
mySubmesh->setEnabled(mySubmeshChk->isChecked());
myGroupChk->setEnabled(!isToAll);
myGroupBtn->setEnabled(myGroupChk->isChecked());
myGroup->setEnabled(myGroupChk->isChecked());
if (!mySubmeshChk->isChecked())
mySubmesh->clear();
if (!myGroupChk->isChecked())
myGroup->clear();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::onRemoveBtn
// Purpose : SLOT. Called when "Remove" button pressed.
// Remove selected in list box entities
//=======================================================================
void SMESHGUI_MultiEditDlg::onRemoveBtn()
{
myBusy = true;
QList<QListWidgetItem*> selItems = myListBox->selectedItems();
QListWidgetItem* item;
foreach(item, selItems)
{
myIds.Remove(item->text().toInt());
delete item;
}
myBusy = false;
emit ListContensChanged();
updateButtons();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::onSortListBtn
// Purpose : SLOT. Called when "Sort list" button pressed.
// Sort entities of list box
//=======================================================================
void SMESHGUI_MultiEditDlg::onSortListBtn()
{
myBusy = true;
int i, k = myListBox->count();
if (k > 0)
{
QList<int> aSelected;
std::vector<int> anArray(k);
for (i = 0; i < k; i++)
{
int id = myListBox->item(i)->text().toInt();
anArray[ i ] = id;
if (myListBox->item(i)->isSelected())
aSelected.append(id);
}
std::sort(anArray.begin(), anArray.end());
myListBox->clear();
for (i = 0; i < k; i++) {
QListWidgetItem* item = new QListWidgetItem(QString::number(anArray[i]));
myListBox->addItem(item);
item->setSelected(aSelected.contains(anArray[i]));
}
}
myBusy = false;
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::onListSelectionChanged
// Purpose : SLOT. Called when selection in list box changed.
// Highlight in selected entities
//=======================================================================
void SMESHGUI_MultiEditDlg::onListSelectionChanged()
{
if (myActor == 0 || myBusy)
return;
if (mySubmeshChk->isChecked() || myGroupChk->isChecked())
return;
SMESH_Actor * anActor = SMESH::FindActorByObject(myMesh);
if (!anActor)
anActor = myActor;
TVisualObjPtr anObj = anActor->GetObject();
TColStd_MapOfInteger anIndexes;
int total = myListBox->count();
for (int i = 0; i < total; i++)
{
if (myListBox->item(i)->isSelected())
{
int anId = myListBox->item(i)->text().toInt();
if (anObj->GetElemVTKId(anId) >= 0) // avoid exception in hilight
anIndexes.Add(anId);
}
}
mySelector->AddOrRemoveIndex(anActor->getIO(),anIndexes,false);
if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
aViewWindow->highlight(anActor->getIO(),true,true);
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::onSubmeshChk
// Purpose : SLOT. Called when state of "SubMesh" check box changed.
// Activate/deactivate selection of submeshes
//=======================================================================
void SMESHGUI_MultiEditDlg::onSubmeshChk()
{
bool isChecked = mySubmeshChk->isChecked();
mySubmeshBtn->setEnabled(isChecked);
mySubmesh->setEnabled(isChecked);
if (!isChecked)
mySubmesh->clear();
if (isChecked && myGroupChk->isChecked())
myGroupChk->setChecked(false);
setSelectionMode();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::onGroupChk
// Purpose : SLOT. Called when state of "Group" check box changed.
// Activate/deactivate selection of groupes
//=======================================================================
void SMESHGUI_MultiEditDlg::onGroupChk()
{
bool isChecked = myGroupChk->isChecked();
myGroupBtn->setEnabled(isChecked);
myGroup->setEnabled(isChecked);
if (!isChecked)
myGroup->clear();
if (isChecked && mySubmeshChk->isChecked())
mySubmeshChk->setChecked(false);
setSelectionMode();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::onToAllChk
// Purpose : SLOT. Called when state of "Apply to all" check box changed.
// Activate/deactivate selection
//=======================================================================
void SMESHGUI_MultiEditDlg::onToAllChk()
{
bool isChecked = myToAllChk->isChecked();
if (isChecked)
myListBox->clear();
myIds.Clear();
emit ListContensChanged();
updateButtons();
setSelectionMode();
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::setSelectionMode
// Purpose : Set selection mode
//=======================================================================
void SMESHGUI_MultiEditDlg::setSelectionMode()
{
SMESH::RemoveFilters();
mySelectionMgr->clearSelected();
mySelectionMgr->clearFilters();
if (mySubmeshChk->isChecked()) {
if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
aViewWindow->SetSelectionMode(ActorSelection);
mySelectionMgr->installFilter(new SMESH_TypeFilter(SUBMESH));
}
else if (myGroupChk->isChecked()) {
if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
aViewWindow->SetSelectionMode(ActorSelection);
mySelectionMgr->installFilter(new SMESH_TypeFilter(GROUP));
}
if (entityType()) {
if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
aViewWindow->SetSelectionMode(VolumeSelection);
SMESH::SetFilter(new SMESHGUI_VolumesFilter());
} else {
if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
aViewWindow->SetSelectionMode(FaceSelection);
if (myFilterType == SMESH::TriaFilter)
SMESH::SetFilter(new SMESHGUI_TriangleFilter());
else if (myFilterType == SMESH::QuadFilter)
SMESH::SetFilter(new SMESHGUI_QuadrangleFilter());
else
SMESH::SetFilter(new SMESHGUI_FacesFilter());
}
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::onApply
// Purpose : SLOT. Called when "Apply" button clicked.
//=======================================================================
bool SMESHGUI_MultiEditDlg::onApply()
{
if (mySMESHGUI->isActiveStudyLocked())
return false;
if (!isValid(true))
return false;
SMESH::SMESH_MeshEditor_var aMeshEditor = myMesh->GetMeshEditor();
if (aMeshEditor->_is_nil())
return false;
myBusy = true;
SMESH::long_array_var anIds = getIds();
bool aResult = process(aMeshEditor, anIds.inout());
if (aResult) {
if (myActor) {
SALOME_ListIO sel;
mySelectionMgr->selectedObjects( sel );
mySelector->ClearIndex();
mySelectionMgr->setSelectedObjects( sel );
SMESH::UpdateView();
}
myListBox->clear();
myIds.Clear();
emit ListContensChanged();
updateButtons();
}
myBusy = false;
return aResult;
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::on3d2dChanged
// Purpose :
//=======================================================================
void SMESHGUI_MultiEditDlg::on3d2dChanged (int type)
{
if (myEntityType != type) {
myEntityType = type;
myListBox->clear();
myIds.Clear();
emit ListContensChanged();
if (type)
myFilterType = SMESH::VolumeFilter;
else
myFilterType = SMESH::FaceFilter;
updateButtons();
setSelectionMode();
}
}
//=======================================================================
// name : SMESHGUI_MultiEditDlg::entityType
// Purpose :
//=======================================================================
int SMESHGUI_MultiEditDlg::entityType()
{
return myEntityType;
}
//=================================================================================
// function : keyPressEvent()
// purpose :
//=================================================================================
void SMESHGUI_MultiEditDlg::keyPressEvent( QKeyEvent* e )
{
QDialog::keyPressEvent( e );
if ( e->isAccepted() )
return;
if ( e->key() == Qt::Key_F1 ) {
e->accept();
onHelp();
}
}
/*!
* Class : SMESHGUI_ChangeOrientationDlg
* Description : Modification of orientation of faces
*/
SMESHGUI_ChangeOrientationDlg
::SMESHGUI_ChangeOrientationDlg(SMESHGUI* theModule):
SMESHGUI_MultiEditDlg(theModule, SMESH::FaceFilter, true)
{
setWindowTitle(tr("CAPTION"));
myHelpFileName = "changing_orientation_of_elements_page.html";
}
SMESHGUI_ChangeOrientationDlg::~SMESHGUI_ChangeOrientationDlg()
{
}
bool SMESHGUI_ChangeOrientationDlg::process (SMESH::SMESH_MeshEditor_ptr theEditor,
const SMESH::long_array& theIds)
{
return theEditor->Reorient(theIds);
}
/*!
* Class : SMESHGUI_UnionOfTrianglesDlg
* Description : Construction of quadrangles by automatic association of triangles
*/
SMESHGUI_UnionOfTrianglesDlg
::SMESHGUI_UnionOfTrianglesDlg(SMESHGUI* theModule):
SMESHGUI_MultiEditDlg(theModule, SMESH::TriaFilter, false)
{
setWindowTitle(tr("CAPTION"));
myComboBoxFunctor->setEnabled(true);
myComboBoxFunctor->addItem(tr("WARP_ELEMENTS")); // for quadrangles only
myComboBoxFunctor->addItem(tr("TAPER_ELEMENTS")); // for quadrangles only
// Maximum angle
QWidget* aMaxAngleGrp = new QWidget(myCriterionGrp);
QHBoxLayout* aMaxAngleGrpLayout = new QHBoxLayout(aMaxAngleGrp);
aMaxAngleGrpLayout->setMargin(0);
aMaxAngleGrpLayout->setSpacing(SPACING);
QLabel* aLab = new QLabel (tr("MAXIMUM_ANGLE"), aMaxAngleGrp);
myMaxAngleSpin = new SMESHGUI_SpinBox (aMaxAngleGrp);
myMaxAngleSpin->RangeStepAndValidator(0, 180.0, 1.0, 3);
myMaxAngleSpin->SetValue(30.0);
aMaxAngleGrpLayout->addWidget(aLab);
aMaxAngleGrpLayout->addWidget(myMaxAngleSpin);
((QVBoxLayout*)(myCriterionGrp->layout()))->addWidget(aMaxAngleGrp);
myCriterionGrp->show();
myHelpFileName = "uniting_set_of_triangles_page.html";
}
SMESHGUI_UnionOfTrianglesDlg::~SMESHGUI_UnionOfTrianglesDlg()
{
}
bool SMESHGUI_UnionOfTrianglesDlg::isValid (const bool theMess)
{
bool ok = SMESHGUI_MultiEditDlg::isValid( theMess );
if( !ok )
return false;
QString msg;
ok = myMaxAngleSpin->isValid( msg, theMess );
if( !ok ) {
if( theMess ) {
QString str( tr( "SMESH_INCORRECT_INPUT" ) );
if ( !msg.isEmpty() )
str += "\n" + msg;
SUIT_MessageBox::critical( this, tr( "SMESH_ERROR" ), str );
}
return false;
}
return ok;
}
bool SMESHGUI_UnionOfTrianglesDlg::process (SMESH::SMESH_MeshEditor_ptr theEditor,
const SMESH::long_array& theIds)
{
SMESH::NumericalFunctor_var aCriterion = getNumericalFunctor();
double aMaxAngle = myMaxAngleSpin->GetValue() * PI / 180.0;
bool ok = theEditor->TriToQuad(theIds, aCriterion, aMaxAngle);
if( ok ) {
QStringList aParameters;
aParameters << myMaxAngleSpin->text();
myMesh->SetParameters( SMESHGUI::JoinObjectParameters(aParameters) );
}
return ok;
}
/*!
* Class : SMESHGUI_CuttingOfQuadsDlg
* Description : Automatic splitting of quadrangles into triangles
*/
SMESHGUI_CuttingOfQuadsDlg
::SMESHGUI_CuttingOfQuadsDlg(SMESHGUI* theModule):
SMESHGUI_MultiEditDlg(theModule, SMESH::QuadFilter, false)
{
setWindowTitle(tr("CAPTION"));
myPreviewActor = 0;
myPreviewChk = new QCheckBox (tr("PREVIEW"), mySelGrp);
QGridLayout* aLay = (QGridLayout*)(mySelGrp->layout());
aLay->addWidget(myPreviewChk, aLay->rowCount(), 0, 1, aLay->columnCount());
myCriterionGrp->show();
myChoiceWidget->show();
myComboBoxFunctor->setEnabled(false);
connect(myPreviewChk, SIGNAL(stateChanged(int)), this, SLOT(onPreviewChk()));
connect(myGroupChoice, SIGNAL(buttonClicked(int)), this, SLOT(onCriterionRB()));
connect(myComboBoxFunctor, SIGNAL(activated(int)), this, SLOT(onPreviewChk()));
connect(this, SIGNAL(ListContensChanged()), this, SLOT(onPreviewChk()));
myHelpFileName = "cutting_quadrangles_page.html";
}
SMESHGUI_CuttingOfQuadsDlg::~SMESHGUI_CuttingOfQuadsDlg()
{
}
void SMESHGUI_CuttingOfQuadsDlg::onClose()
{
erasePreview();
SMESHGUI_MultiEditDlg::onClose();
}
bool SMESHGUI_CuttingOfQuadsDlg::process (SMESH::SMESH_MeshEditor_ptr theEditor,
const SMESH::long_array& theIds)
{
switch (myGroupChoice->checkedId()) {
case 0: // use diagonal 1-3
return theEditor->SplitQuad(theIds, true);
case 1: // use diagonal 2-4
return theEditor->SplitQuad(theIds, false);
default: // use numeric functor
break;
}
SMESH::NumericalFunctor_var aCriterion = getNumericalFunctor();
return theEditor->QuadToTri(theIds, aCriterion);
}
void SMESHGUI_CuttingOfQuadsDlg::onCriterionRB()
{
if (myGroupChoice->checkedId() == 2) // Use numeric functor
myComboBoxFunctor->setEnabled(true);
else
myComboBoxFunctor->setEnabled(false);
onPreviewChk();
}
void SMESHGUI_CuttingOfQuadsDlg::onPreviewChk()
{
myPreviewChk->isChecked() ? displayPreview() : erasePreview();
}
void SMESHGUI_CuttingOfQuadsDlg::erasePreview()
{
if (myPreviewActor == 0)
return;
if (SVTK_ViewWindow* vf = SMESH::GetCurrentVtkView()) {
vf->RemoveActor(myPreviewActor);
vf->Repaint();
}
myPreviewActor->Delete();
myPreviewActor = 0;
}
void SMESHGUI_CuttingOfQuadsDlg::displayPreview()
{
if (myActor == 0)
return;
if (myPreviewActor != 0)
erasePreview();
// get Ids of elements
SMESH::long_array_var anElemIds = getIds();
if (getIds()->length() == 0)
return;
SMDS_Mesh* aMesh = myActor->GetObject()->GetMesh();
if (aMesh == 0)
return;
// 0 - use diagonal 1-3, 1 - use diagonal 2-4, 2 - use numerical functor
int aChoice = myGroupChoice->checkedId();
SMESH::NumericalFunctor_var aCriterion = SMESH::NumericalFunctor::_nil();
SMESH::SMESH_MeshEditor_var aMeshEditor = SMESH::SMESH_MeshEditor::_nil();
if (aChoice == 2) {
aCriterion = getNumericalFunctor();
aMeshEditor = myMesh->GetMeshEditor();
if (aMeshEditor->_is_nil())
return;
}
//Create grid
vtkUnstructuredGrid* aGrid = vtkUnstructuredGrid::New();
vtkIdType aNbCells = anElemIds->length() * 2;
vtkIdType aCellsSize = 4 * aNbCells;
vtkCellArray* aConnectivity = vtkCellArray::New();
aConnectivity->Allocate(aCellsSize, 0);
vtkPoints* aPoints = vtkPoints::New();
aPoints->SetNumberOfPoints(anElemIds->length() * 4);
vtkUnsignedCharArray* aCellTypesArray = vtkUnsignedCharArray::New();
aCellTypesArray->SetNumberOfComponents(1);
aCellTypesArray->Allocate(aNbCells * aCellTypesArray->GetNumberOfComponents());
vtkIdList *anIdList = vtkIdList::New();
anIdList->SetNumberOfIds(3);
TColStd_DataMapOfIntegerInteger anIdToVtk;
int aNodes[ 4 ];
int nbPoints = -1;
for (int i = 0, n = anElemIds->length(); i < n; i++)
{
const SMDS_MeshElement* anElem = aMesh->FindElement(anElemIds[ i ]);
if (anElem == 0 || anElem->NbNodes() != 4)
continue;
SMDS_ElemIteratorPtr anIter = anElem->nodesIterator();
int k = 0;
while (anIter->more()) {
const SMDS_MeshNode* aNode = static_cast<const SMDS_MeshNode*>(anIter->next());
if (aNode)
{
if (!anIdToVtk.IsBound(aNode->GetID()))
{
aPoints->SetPoint(++nbPoints, aNode->X(), aNode->Y(), aNode->Z());
anIdToVtk.Bind(aNode->GetID(), nbPoints);
}
aNodes[ k++ ] = aNode->GetID();
}
}
if (k != 4)
continue;
bool isDiag13 = true;
if (aChoice == 0) // use diagonal 1-3
{
isDiag13 = true;
}
else if (aChoice == 1) // use diagonal 2-4
{
isDiag13 = false;
}
else // use numerical functor
{
// compare two sets of possible triangles
int diag = aMeshEditor->BestSplit(anElemIds[i], aCriterion);
if (diag == 1) // 1-3
isDiag13 = true;
else if (diag == 2) // 2-4
isDiag13 = false;
else // error
continue;
}
if (isDiag13)
{
anIdList->SetId(0, anIdToVtk(aNodes[ 0 ]));
anIdList->SetId(1, anIdToVtk(aNodes[ 1 ]));
anIdList->SetId(2, anIdToVtk(aNodes[ 2 ]));
aConnectivity->InsertNextCell(anIdList);
aCellTypesArray->InsertNextValue(VTK_TRIANGLE);
anIdList->SetId(0, anIdToVtk(aNodes[ 2 ]));
anIdList->SetId(1, anIdToVtk(aNodes[ 3 ]));
anIdList->SetId(2, anIdToVtk(aNodes[ 0 ]));
aConnectivity->InsertNextCell(anIdList);
aCellTypesArray->InsertNextValue(VTK_TRIANGLE);
}
else
{
anIdList->SetId(0, anIdToVtk(aNodes[ 1 ]));
anIdList->SetId(1, anIdToVtk(aNodes[ 2 ]));
anIdList->SetId(2, anIdToVtk(aNodes[ 3 ]));
aConnectivity->InsertNextCell(anIdList);
aCellTypesArray->InsertNextValue(VTK_TRIANGLE);
anIdList->SetId(0, anIdToVtk(aNodes[ 3 ]));
anIdList->SetId(1, anIdToVtk(aNodes[ 0 ]));
anIdList->SetId(2, anIdToVtk(aNodes[ 1 ]));
aConnectivity->InsertNextCell(anIdList);
aCellTypesArray->InsertNextValue(VTK_TRIANGLE);
}
}
VTKViewer_CellLocationsArray* aCellLocationsArray = VTKViewer_CellLocationsArray::New();
aCellLocationsArray->SetNumberOfComponents(1);
aCellLocationsArray->SetNumberOfTuples(aNbCells);
aConnectivity->InitTraversal();
for(vtkIdType idType = 0, *pts, npts; aConnectivity->GetNextCell(npts, pts); idType++)
aCellLocationsArray->SetValue(idType, aConnectivity->GetTraversalLocation(npts));
aGrid->SetPoints(aPoints);
aGrid->SetCells(aCellTypesArray, aCellLocationsArray,aConnectivity);
// Create and display actor
vtkDataSetMapper* aMapper = vtkDataSetMapper::New();
aMapper->SetInput(aGrid);
myPreviewActor = SALOME_Actor::New();
myPreviewActor->PickableOff();
myPreviewActor->SetMapper(aMapper);
vtkProperty* aProp = vtkProperty::New();
aProp->SetRepresentationToWireframe();
aProp->SetColor(250, 0, 250);
aProp->SetLineWidth(myActor->GetLineWidth() + 1);
myPreviewActor->SetProperty(aProp);
SMESH::GetCurrentVtkView()->AddActor(myPreviewActor);
SMESH::GetCurrentVtkView()->Repaint();
aProp->Delete();
aPoints->Delete();
aConnectivity->Delete();
aGrid->Delete();
aMapper->Delete();
anIdList->Delete();
aCellTypesArray->Delete();
aCellLocationsArray->Delete();
}
Reference in New Issue View Git Blame Copy Permalink