1 // Copyright (C) 2007-2015 CEA/DEN, EDF R&D, OPEN CASCADE
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Lesser General Public License for more details.
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 // File : StdMeshersGUI_SubShapeSelectorWdg.cxx
21 // Author : Open CASCADE S.A.S. (dmv)
24 #include "StdMeshersGUI_SubShapeSelectorWdg.h"
27 #include "SMESHGUI_Utils.h"
28 #include "SMESHGUI_VTKUtils.h"
29 #include "SMESH_Actor.h"
30 #include "SMESH_Gen_i.hxx"
31 #include "SMESH_LogicalFilter.hxx"
32 #include "SMESH_PreviewActorsCollection.h"
33 #include "SMESH_Type.h"
35 // SALOME GUI includes
36 #include <LightApp_SelectionMgr.h>
37 #include <SALOME_ListIO.hxx>
38 #include <SUIT_OverrideCursor.h>
39 #include <SUIT_ResourceMgr.h>
40 #include <SVTK_Selector.h>
41 #include <SVTK_ViewModel.h>
42 #include <SVTK_ViewWindow.h>
46 #include <GEOM_TypeFilter.h>
47 #include <GEOM_CompoundFilter.h>
50 #include <QPushButton>
51 #include <QGridLayout>
52 #include <QListWidget>
57 #include <TColStd_MapOfInteger.hxx>
58 #include <TopoDS_Shape.hxx>
60 #include <TopExp_Explorer.hxx>
66 //================================================================================
70 //================================================================================
72 StdMeshersGUI_SubShapeSelectorWdg
73 ::StdMeshersGUI_SubShapeSelectorWdg( QWidget * parent, TopAbs_ShapeEnum aSubShType ):
78 QPixmap image0( SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap( "SMESH", tr( "ICON_SELECT" ) ) );
80 QGridLayout* edgesLayout = new QGridLayout( this );
81 edgesLayout->setMargin( MARGIN );
82 edgesLayout->setSpacing( SPACING );
84 myListWidget = new QListWidget( this );
85 myAddButton = new QPushButton( tr( "SMESH_BUT_ADD" ), this );
86 myRemoveButton = new QPushButton( tr( "SMESH_BUT_REMOVE" ), this );
87 myInfoLabel = new QLabel( this );
88 myPrevButton = new QPushButton( "<<", this );
89 myNextButton = new QPushButton( ">>", this );
90 myListWidget->setSelectionMode( QListWidget::ExtendedSelection );
92 edgesLayout->addWidget(myListWidget, 0, 0, 3, 3);
93 edgesLayout->addWidget(myAddButton, 0, 3);
94 edgesLayout->addWidget(myRemoveButton, 1, 3);
95 edgesLayout->addWidget(myInfoLabel, 3, 0, 1, 3);
96 edgesLayout->addWidget(myPrevButton, 4, 0);
97 edgesLayout->addWidget(myNextButton, 4, 2);
99 edgesLayout->setRowStretch(2, 5);
100 edgesLayout->setColumnStretch(1, 5);
102 myListWidget->setMinimumWidth(300);
103 myInfoLabel->setMinimumWidth(300);
104 myInfoLabel->setSizePolicy(QSizePolicy::MinimumExpanding, QSizePolicy::Fixed);
105 myInfoLabel->setAlignment(Qt::AlignCenter);
107 mySubShType = aSubShType;
112 //================================================================================
116 //================================================================================
118 StdMeshersGUI_SubShapeSelectorWdg::~StdMeshersGUI_SubShapeSelectorWdg()
120 if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI )) {
121 if ( myPreviewActor ) {
122 myPreviewActor->RemoveFromRender( myRenderer );
123 aViewWindow->Repaint();
125 delete myPreviewActor;
131 myMainShape.Nullify();
133 if ( mySelectionMgr && myFilter )
134 mySelectionMgr->removeFilter( myFilter );
135 delete myFilter; myFilter=0;
137 mySelectionMgr->clearSelected();
139 SUIT_SelectionFilter* filter;
140 foreach( filter, myGeomFilters )
144 //================================================================================
146 * Create a layout, initialize fields
148 //================================================================================
150 void StdMeshersGUI_SubShapeSelectorWdg::init()
153 myIsNotCorrected = true; // to dont call the GetCorrectedValue method twice
155 mySelectedIDs.clear();
157 myAddButton->setEnabled( false );
158 myRemoveButton->setEnabled( false );
160 mySMESHGUI = SMESHGUI::GetSMESHGUI();
161 mySelectionMgr = SMESH::GetSelectionMgr( mySMESHGUI );
162 mySelector = (SMESH::GetViewWindow( mySMESHGUI ))->GetSelector();
164 if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
165 aViewWindow->SetSelectionMode( ActorSelection );
170 connect( myAddButton, SIGNAL(clicked()), SLOT(onAdd()));
171 connect( myRemoveButton, SIGNAL(clicked()), SLOT(onRemove()));
172 connect( myPrevButton, SIGNAL(clicked()), SLOT(onPrevious()));
173 connect( myNextButton, SIGNAL(clicked()), SLOT(onNext()));
175 connect( mySelectionMgr, SIGNAL(currentSelectionChanged()), this, SLOT(selectionIntoArgument()));
176 connect( myListWidget, SIGNAL(itemSelectionChanged()), this, SLOT(onListSelectionChanged()));
181 //================================================================================
183 * \brief Install filters to select sub-shapes of mySubShType or their groups
185 //================================================================================
187 void StdMeshersGUI_SubShapeSelectorWdg::setFilter()
189 SalomeApp_Study* study = mySMESHGUI->activeStudy();
190 GEOM_TypeFilter* typeFilter = new GEOM_TypeFilter(study, mySubShType, /*isShapeType=*/true );
191 GEOM_CompoundFilter* gpoupFilter = new GEOM_CompoundFilter(study);
192 gpoupFilter->addSubType( mySubShType );
193 myGeomFilters.append( typeFilter );
194 myGeomFilters.append( gpoupFilter );
195 myFilter = new SMESH_LogicalFilter( myGeomFilters, SMESH_LogicalFilter::LO_OR );
196 mySelectionMgr->installFilter( myFilter );
199 //================================================================================
201 * Create a layout, initialize fields
203 //================================================================================
205 void StdMeshersGUI_SubShapeSelectorWdg::ShowPreview( bool visible)
207 if ( !myPreviewActor )
210 if ( myIsShown != visible ) {
211 myPreviewActor->SetShown( visible );
213 if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
214 aViewWindow->Repaint();
220 //================================================================================
222 * \brief Clears selected IDs. This is a workaround of a bug that
223 * SUIT_SelectionMgr::clearSelected() does not emit currentSelectionChanged
225 //================================================================================
227 void StdMeshersGUI_SubShapeSelectorWdg::ClearSelected()
229 mySelectedIDs.clear();
230 selectionIntoArgument();
233 //=================================================================================
234 // function : selectionIntoArgument()
235 // purpose : Called when selection as changed or other case
236 //=================================================================================
237 void StdMeshersGUI_SubShapeSelectorWdg::selectionIntoArgument()
239 if ( !myPreviewActor )
242 mySelectedIDs.clear();
246 mySelectionMgr->selectedObjects( aList );
247 int nbSel = aList.Extent();
250 SALOME_ListIteratorOfListIO anIt (aList);
252 for ( ; anIt.More(); anIt.Next()) // Loop on selected objects
254 Handle(SALOME_InteractiveObject) IO = anIt.Value();
256 GEOM::GEOM_Object_var aGeomObj = GetGeomObjectByEntry( IO->getEntry() );
257 if ( !CORBA::is_nil( aGeomObj ) ) { // Selected Object From Study
258 GEOM::GEOM_Object_var aGeomFatherObj = aGeomObj->GetMainShape();
259 QString aFatherEntry = "";
260 QString aMainFatherEntry = "";
262 if ( !CORBA::is_nil( aGeomFatherObj ) ) {
264 GEOM::GEOM_Object_var aGeomMain = GetGeomObjectByEntry( myEntry.c_str() );
265 if ( !CORBA::is_nil( aGeomMain ) && aGeomMain->GetType() == 37 ) { // Main Shape is a Group
266 GEOM::GEOM_Object_var aMainFatherObj = aGeomMain->GetMainShape();
267 if ( !CORBA::is_nil( aMainFatherObj ) )
268 aMainFatherEntry = aMainFatherObj->GetStudyEntry();
270 aFatherEntry = aGeomFatherObj->GetStudyEntry();
273 if (( ! aFatherEntry.isEmpty() ) &&
274 ( aFatherEntry == myEntry.c_str() || aFatherEntry == aMainFatherEntry ) )
276 if ( aGeomObj->GetType() == 37 /*GEOM_GROUP*/ ) { // Selected Group that belongs the main object
277 GEOMBase::GetShape(aGeomObj, shape);
278 if ( !shape.IsNull() ) {
279 TopExp_Explorer exp( shape, mySubShType );
280 for ( ; exp.More(); exp.Next() ) {
281 int index = myPreviewActor->GetIndexByShape( exp.Current() );
283 mySelectedIDs.append( index );
284 myPreviewActor->HighlightID( index );
288 } else if ( aGeomObj->GetType() == 28 /*GEOM_SUBSHAPE*/ ) {
289 GEOMBase::GetShape(aGeomObj, shape);
290 if ( !shape.IsNull() && shape.ShapeType() == mySubShType ) {
291 int index = myPreviewActor->GetIndexByShape( shape );
293 mySelectedIDs.append( index );
294 myPreviewActor->HighlightID( index );
299 } else { // Selected Actor from Actor Collection
300 QString anEntry = IO->getEntry();
302 int index = anEntry.lastIndexOf( str );
303 anEntry.remove(0, index+1);
304 int ind = anEntry.toInt();
306 mySelectedIDs.append( ind );
311 myAddButton->setEnabled( ( myListWidget->count() < myMaxSize || myMaxSize == -1 ) &&
312 mySelectedIDs.size() > 0 &&
313 ( mySelectedIDs.size() <= myMaxSize || myMaxSize == -1 ) );
315 //Connect Selected Ids in viewer and dialog's Ids list
316 bool signalsBlocked = myListWidget->blockSignals( true );
317 myListWidget->clearSelection();
318 if ( mySelectedIDs.size() > 0 ) {
319 for (int i = 0; i < mySelectedIDs.size(); i++) {
320 QString anID = QString(" %1").arg( mySelectedIDs.at(i) );
321 QList<QListWidgetItem*> anItems = myListWidget->findItems ( anID, Qt::MatchExactly );
322 QListWidgetItem* item;
323 foreach(item, anItems)
324 item->setSelected(true);
327 myListWidget->blockSignals( signalsBlocked );
330 //=================================================================================
331 // function : onAdd()
332 // purpose : Called when Add Button Clicked
333 //=================================================================================
334 void StdMeshersGUI_SubShapeSelectorWdg::onAdd()
336 if ( mySelectedIDs.size() < 1 )
339 myListWidget->blockSignals( true );
340 for (int i = 0; i < mySelectedIDs.size() && (myMaxSize == -1 || myListOfIDs.size() < myMaxSize); i++) {
341 if ( myListOfIDs.indexOf( mySelectedIDs.at(i) ) == -1 ) {
342 QString anID = QString(" %1").arg( mySelectedIDs.at(i) );
344 QListWidgetItem* anItem = new QListWidgetItem( anID, myListWidget );
345 anItem->setSelected(true);
347 myListOfIDs.append( mySelectedIDs.at(i) );
350 onListSelectionChanged();
351 myListWidget->blockSignals( false );
353 mySelectedIDs.clear();
354 myAddButton->setEnabled( false );
357 //=================================================================================
358 // function : onRemove()
359 // purpose : Called when Remove Button Clicked
360 //=================================================================================
361 void StdMeshersGUI_SubShapeSelectorWdg::onRemove()
363 if ( myListWidget->count() < 1 )
366 myListWidget->blockSignals( true );
367 QList<QListWidgetItem*> selItems = myListWidget->selectedItems();
368 QListWidgetItem* item;
369 foreach(item, selItems) {
370 QString idStr = item->text();
371 int id = idStr.toInt();
373 int index = myListOfIDs.indexOf( id );
374 myListOfIDs.removeAt( index );
378 onListSelectionChanged();
379 myListWidget->blockSignals( false );
381 myAddButton->setEnabled( !mySelectedIDs.isEmpty() );
384 void StdMeshersGUI_SubShapeSelectorWdg::onPrevious()
386 if ( myPreviewActor ) {
387 myPreviewActor->previous();
388 myListWidget->clearSelection();
390 if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
391 aViewWindow->Repaint();
395 void StdMeshersGUI_SubShapeSelectorWdg::onNext()
397 if ( myPreviewActor ) {
398 myPreviewActor->next();
399 myListWidget->clearSelection();
401 if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
402 aViewWindow->Repaint();
406 //=================================================================================
407 // function : onListSelectionChanged()
408 // purpose : Called when selection in element list is changed
409 //=================================================================================
410 void StdMeshersGUI_SubShapeSelectorWdg::onListSelectionChanged()
412 if ( !myPreviewActor )
415 mySelectionMgr->clearSelected();
416 TColStd_MapOfInteger aIndexes;
417 QList<QListWidgetItem*> selItems = myListWidget->selectedItems();
418 QListWidgetItem* anItem;
419 foreach(anItem, selItems)
420 myPreviewActor->HighlightID( anItem->text().toInt() );
422 // update remove button
423 myRemoveButton->setEnabled( selItems.size() > 0 );
426 //=================================================================================
427 // function : setGeomShape
428 // purpose : Called to set geometry whose sub-shapes are selected
429 //================================================================================
430 void StdMeshersGUI_SubShapeSelectorWdg::SetGeomShapeEntry( const QString& theEntry,
431 const QString& theMainShapeEntry )
433 if ( !theEntry.isEmpty() || theMainShapeEntry.isEmpty() )
435 myParamValue = theEntry;
436 myEntry = theEntry.toStdString();
437 myMainEntry = theMainShapeEntry.toStdString();
439 if ( myMainEntry.empty() ) myMainEntry = myEntry;
440 if ( myEntry.empty() ) myEntry = myMainEntry;
441 if ( myMainEntry.length() > myEntry.length() &&
442 theMainShapeEntry.startsWith( theEntry ))
443 std::swap( myMainEntry, myEntry );
445 myGeomShape = GetTopoDSByEntry( myEntry.c_str() );
446 if ( myEntry == myMainEntry )
447 myMainShape = myGeomShape;
449 myMainShape = GetTopoDSByEntry( myMainEntry.c_str() );
451 myIsNotCorrected = true;
455 //=================================================================================
456 // function : updateState
457 // purpose : update Widget state
458 //=================================================================================
459 void StdMeshersGUI_SubShapeSelectorWdg::updateState()
462 if ( !myGeomShape.IsNull() )
464 myInfoLabel->setVisible( false );
465 myPrevButton->setVisible( false );
466 myNextButton->setVisible( false );
468 myListWidget->setEnabled( state );
469 myAddButton->setEnabled( mySelectedIDs.size() > 0 );
472 SUIT_OverrideCursor wc;
473 myPreviewActor = new SMESH_PreviewActorsCollection();
474 myPreviewActor->SetSelector( mySelector );
475 myPreviewActor->Init( myGeomShape, myMainShape, mySubShType, myEntry.c_str() );
476 myPreviewActor->SetShown( false );
478 if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI )) {
479 myRenderer = aViewWindow->getRenderer();
480 myPreviewActor->AddToRender( myRenderer );
481 aViewWindow->Repaint();
487 //=================================================================================
488 // function : GetGeomObjectByEntry
489 // purpose : Called to get GeomObject
490 //=================================================================================
491 GEOM::GEOM_Object_var StdMeshersGUI_SubShapeSelectorWdg::GetGeomObjectByEntry( const QString& theEntry )
493 GEOM::GEOM_Object_var aGeomObj;
494 SALOMEDS::Study_var aStudy = SMESHGUI::GetSMESHGen()->GetCurrentStudy();
495 if ( !aStudy->_is_nil() )
497 SALOMEDS::SObject_var aSObj = aStudy->FindObjectID( theEntry.toLatin1().data() );
498 if (!aSObj->_is_nil() )
500 CORBA::Object_var obj = aSObj->GetObject();
501 aGeomObj = GEOM::GEOM_Object::_narrow(obj);
505 return aGeomObj._retn();
508 //=================================================================================
509 // function : setObjectByEntry
510 // purpose : Called to get GeomObject
511 //=================================================================================
512 TopoDS_Shape StdMeshersGUI_SubShapeSelectorWdg::GetTopoDSByEntry( const QString& theEntry )
515 GEOM::GEOM_Object_var aGeomObj = GetGeomObjectByEntry( theEntry );
516 GEOMBase::GetShape(aGeomObj, shape);
520 //=================================================================================
521 // function : GetListOfIds
522 // purpose : Called to get the list of SubShapes IDs
523 //=================================================================================
524 SMESH::long_array_var StdMeshersGUI_SubShapeSelectorWdg::GetListOfIDs()
526 SMESH::long_array_var anArray = new SMESH::long_array;
528 // if ( myMainEntry != "" && myIsNotCorrected )
529 // myListOfIDs = GetCorrectedListOfIDs( true );
531 int size = myListOfIDs.size();
532 anArray->length( size );
533 for (int i = 0; i < size; i++)
534 anArray[i] = myListOfIDs.at(i);
539 //=================================================================================
540 // function : SetListOfIds
541 // purpose : Called to set the list of SubShapes IDs. Returns false if any ID is invalid
542 //=================================================================================
543 bool StdMeshersGUI_SubShapeSelectorWdg::SetListOfIDs( SMESH::long_array_var theIds)
545 mySelectedIDs.clear();
547 int size = theIds->length();
548 for ( int i = 0; i < size; i++ )
549 mySelectedIDs.append( theIds[ i ] );
551 myListWidget->blockSignals( true );
552 myListWidget->clear();
553 myListWidget->blockSignals( false );
556 if ( myPreviewActor )
558 for ( int i = 0; i < size && isOk; i++ )
559 isOk = myPreviewActor->IsValidIndex( theIds[ i ] );
561 else if ( !myMainShape.IsNull() )
563 TopTools_IndexedMapOfShape aMainMap;
564 TopExp::MapShapes(myMainShape, aMainMap);
565 for ( int i = 0; i < size && isOk; i++ )
566 isOk = ( theIds[ i ] > 0 && theIds[ i ] <= aMainMap.Extent() );
568 // mySelectedIDs = GetCorrectedListOfIDs( false, &isOk );
573 //=================================================================================
574 // function : SetMainShapeEntry
575 // purpose : Called to set the Entry of main shape of the mesh
576 //=================================================================================
577 // void StdMeshersGUI_SubShapeSelectorWdg::SetMainShapeEntry( const QString& theEntry )
579 // myMainEntry = theEntry;
580 // myMainShape = GetTopoDSByEntry( theEntry );
581 // myIsNotCorrected = true;
584 //=================================================================================
585 // function : GetMainShapeEntry
586 // purpose : Called to get the Main Object Entry
587 //=================================================================================
588 const char* StdMeshersGUI_SubShapeSelectorWdg::GetMainShapeEntry()
590 if ( myMainEntry.empty() ) myMainEntry = "";
591 return myMainEntry.c_str();
594 //=================================================================================
595 // function : GetCorrectedListOfIds
596 // purpose : Called to convert the list of IDs from sub-shape IDs to main shape IDs
597 //=================================================================================
599 // StdMeshersGUI_SubShapeSelectorWdg::GetCorrectedListOfIDs( bool fromSubshapeToMainshape,
602 // if (( myMainShape.IsNull() || myGeomShape.IsNull() ) && fromSubshapeToMainshape )
603 // return myListOfIDs;
604 // else if (( myMainShape.IsNull() /*||*/&& myGeomShape.IsNull() ) && !fromSubshapeToMainshape )
605 // return mySelectedIDs;
607 // if ( !fromSubshapeToMainshape ) // called from SetListOfIDs
609 // if ( myMainShape.IsNull() )
610 // std::swap( myMainShape, myGeomShape );
614 // TopTools_IndexedMapOfShape aGeomMap, aMainMap;
615 // TopExp::MapShapes(myMainShape, aMainMap);
616 // if ( !myGeomShape.IsNull() )
617 // TopExp::MapShapes(myGeomShape, aGeomMap);
620 // if ( fromSubshapeToMainshape ) // convert indexes from sub-shape to mainshape
622 // int size = myListOfIDs.size();
623 // for (int i = 0; i < size; i++) {
624 // int index = myListOfIDs.at(i);
625 // if ( aGeomMap.Extent() < index )
631 // TopoDS_Shape aSubShape = aGeomMap.FindKey( index );
632 // if ( mySubShType != aSubShape.ShapeType() )
634 // if ( !aMainMap.Contains( aSubShape ))
637 // index = aMainMap.FindIndex( aSubShape );
639 // aList.append( index );
641 // myIsNotCorrected = false;
643 // else // convert indexes from main shape to sub-shape, or just check indices
645 // int size = mySelectedIDs.size();
646 // for (int i = 0; i < size; i++) {
647 // int index = mySelectedIDs.at(i);
648 // if ( aMainMap.Extent() < index )
654 // TopoDS_Shape aSubShape = aMainMap.FindKey( index );
655 // if ( mySubShType != aSubShape.ShapeType() )
657 // if ( !aGeomMap.Contains( aSubShape ) && !aGeomMap.IsEmpty() )
660 // index = aGeomMap.FindIndex( aSubShape );
662 // aList.append( index );
665 // if ( isOK ) *isOK = ok;
670 void StdMeshersGUI_SubShapeSelectorWdg::updateButtons()
672 if ( myPreviewActor ) {
673 int total = myPreviewActor->count();
674 int chunk = myPreviewActor->currentChunk();
675 int chunkSize = myPreviewActor->chunkSize();
676 int imin = chunk*chunkSize+1;
677 int imax = std::min((chunk+1)*chunkSize, total);
678 bool vis = imax > 0 && total > chunkSize;
679 myInfoLabel->setVisible( vis );
680 myPrevButton->setVisible( vis );
681 myNextButton->setVisible( vis );
682 myInfoLabel->setText( tr( "X_FROM_Y_ITEMS_SHOWN" ).arg(imin).arg(imax).arg(total) );
683 myPrevButton->setEnabled( myPreviewActor->hasPrevious() );
684 myNextButton->setEnabled( myPreviewActor->hasNext() );