Salome HOME
Merge branch 'hydro/imps_2015' into V7_dev
[modules/smesh.git] / src / StdMeshersGUI / StdMeshersGUI_SubShapeSelectorWdg.cxx
1 // Copyright (C) 2007-2015  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
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.
7 //
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.
12 //
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
16 //
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 //
19
20 // File   : StdMeshersGUI_SubShapeSelectorWdg.cxx
21 // Author : Open CASCADE S.A.S. (dmv)
22 // SMESH includes
23 //
24 #include "StdMeshersGUI_SubShapeSelectorWdg.h"
25
26 // SMESH Includes
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"
34
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>
43
44 // GEOM Includes
45 #include <GEOMBase.h>
46 #include <GEOM_TypeFilter.h>
47 #include <GEOM_CompoundFilter.h>
48
49 // Qt includes
50 #include <QPushButton>
51 #include <QGridLayout>
52 #include <QListWidget>
53 #include <QCheckBox>
54 #include <QLineEdit>
55
56 // OCCT includes
57 #include <TColStd_MapOfInteger.hxx>
58 #include <TopoDS_Shape.hxx>
59 #include <TopExp.hxx>
60 #include <TopExp_Explorer.hxx>
61
62
63 #define SPACING 6
64 #define MARGIN 0
65
66 //================================================================================
67 /*!
68  *  Constructor
69  */
70 //================================================================================
71
72 StdMeshersGUI_SubShapeSelectorWdg
73 ::StdMeshersGUI_SubShapeSelectorWdg( QWidget * parent, TopAbs_ShapeEnum aSubShType ): 
74   QWidget( parent ),
75   myMaxSize( -1 ),
76   myPreviewActor( 0 )
77 {
78   QPixmap image0( SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap( "SMESH", tr( "ICON_SELECT" ) ) );
79
80   QGridLayout* edgesLayout = new QGridLayout( this );
81   edgesLayout->setMargin( MARGIN );
82   edgesLayout->setSpacing( SPACING );
83   
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 );
91
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);
98
99   edgesLayout->setRowStretch(2, 5);
100   edgesLayout->setColumnStretch(1, 5);
101
102   myListWidget->setMinimumWidth(300);
103   myInfoLabel->setMinimumWidth(300);
104   myInfoLabel->setSizePolicy(QSizePolicy::MinimumExpanding, QSizePolicy::Fixed);
105   myInfoLabel->setAlignment(Qt::AlignCenter);
106
107   mySubShType = aSubShType;
108
109   init();
110 }
111
112 //================================================================================
113 /*!
114  *  Destructor
115  */
116 //================================================================================
117
118 StdMeshersGUI_SubShapeSelectorWdg::~StdMeshersGUI_SubShapeSelectorWdg()
119 {
120   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI )) {
121     if ( myPreviewActor ) {
122       myPreviewActor->RemoveFromRender( myRenderer );
123       aViewWindow->Repaint();
124
125       delete myPreviewActor;
126       myPreviewActor = 0;
127     }
128   }
129   myEntry = "";
130   myParamValue = "";
131   myMainShape.Nullify();
132
133   if ( mySelectionMgr && myFilter )
134     mySelectionMgr->removeFilter( myFilter );
135   delete myFilter; myFilter=0;
136
137   mySelectionMgr->clearSelected();
138
139   SUIT_SelectionFilter* filter;
140   foreach( filter, myGeomFilters )
141     delete filter;
142 }
143
144 //================================================================================
145 /*!
146  *  Create a layout, initialize fields
147  */
148 //================================================================================
149
150 void StdMeshersGUI_SubShapeSelectorWdg::init()
151 {
152   myParamValue = "";
153   myIsNotCorrected = true; // to dont call the GetCorrectedValue method twice
154   myListOfIDs.clear();
155   mySelectedIDs.clear();
156
157   myAddButton->setEnabled( false );
158   myRemoveButton->setEnabled( false );
159
160   mySMESHGUI     = SMESHGUI::GetSMESHGUI();
161   mySelectionMgr = SMESH::GetSelectionMgr( mySMESHGUI );
162   mySelector = (SMESH::GetViewWindow( mySMESHGUI ))->GetSelector();
163
164   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
165     aViewWindow->SetSelectionMode( ActorSelection );
166
167   myFilter=0;
168   //setFilter();
169
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()));
174   
175   connect( mySelectionMgr, SIGNAL(currentSelectionChanged()), this, SLOT(selectionIntoArgument()));
176   connect( myListWidget,   SIGNAL(itemSelectionChanged()),    this, SLOT(onListSelectionChanged()));
177
178   updateState();
179 }
180
181 //================================================================================
182 /*!
183  * \brief Install filters to select sub-shapes of mySubShType or their groups
184  */
185 //================================================================================
186
187 void StdMeshersGUI_SubShapeSelectorWdg::setFilter()
188 {
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 );
197 }
198
199 //================================================================================
200 /*!
201  *  Create a layout, initialize fields
202  */
203 //================================================================================
204
205 void StdMeshersGUI_SubShapeSelectorWdg::ShowPreview( bool visible)
206 {
207   if ( !myPreviewActor )
208     return;
209
210   if ( myIsShown != visible ) {
211     myPreviewActor->SetShown( visible );
212     
213     if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
214       aViewWindow->Repaint();
215
216     myIsShown = visible;
217   }
218 }
219
220 //================================================================================
221 /*!
222  * \brief Clears selected IDs. This is a workaround of a bug that
223  *        SUIT_SelectionMgr::clearSelected() does not emit currentSelectionChanged
224  */
225 //================================================================================
226
227 void StdMeshersGUI_SubShapeSelectorWdg::ClearSelected()
228 {
229   mySelectedIDs.clear();
230   selectionIntoArgument();
231 }
232
233 //=================================================================================
234 // function : selectionIntoArgument()
235 // purpose  : Called when selection as changed or other case
236 //=================================================================================
237 void StdMeshersGUI_SubShapeSelectorWdg::selectionIntoArgument()
238 {
239   if ( !myPreviewActor )
240     return;
241
242   mySelectedIDs.clear();
243
244   // get selected mesh
245   SALOME_ListIO aList;
246   mySelectionMgr->selectedObjects( aList );
247   int nbSel = aList.Extent();
248
249   if (nbSel > 0) {
250     SALOME_ListIteratorOfListIO anIt (aList);
251
252     for ( ; anIt.More(); anIt.Next()) // Loop on selected objects
253     {
254       Handle(SALOME_InteractiveObject) IO = anIt.Value();
255
256       GEOM::GEOM_Object_var aGeomObj = GetGeomObjectByEntry( IO->getEntry() );
257       if ( !CORBA::is_nil( aGeomObj ) ) { // Selected Object From Study
258         // commented for IPAL52836
259         //
260         // GEOM::GEOM_Object_var aGeomFatherObj = aGeomObj->GetMainShape();
261         // QString aFatherEntry = "";
262         // QString aMainFatherEntry = "";
263         // TopoDS_Shape shape;
264         // if ( !CORBA::is_nil( aGeomFatherObj ) ) {
265         //   // Get Main Shape
266         //   GEOM::GEOM_Object_var aGeomMain = GetGeomObjectByEntry( myEntry.c_str() );
267         //   if ( !CORBA::is_nil( aGeomMain ) && aGeomMain->GetType() == 37 ) {  // Main Shape is a Group
268         //     GEOM::GEOM_Object_var aMainFatherObj = aGeomMain->GetMainShape();
269         //     if ( !CORBA::is_nil( aMainFatherObj ) )
270         //       aMainFatherEntry = aMainFatherObj->GetStudyEntry();
271         //   }
272         //   aFatherEntry = aGeomFatherObj->GetStudyEntry();
273         // }
274
275         // if (( ! aFatherEntry.isEmpty() ) &&
276         //     ( aFatherEntry == myEntry.c_str() || aFatherEntry == aMainFatherEntry ) )
277         {
278           TopoDS_Shape shape;
279           if ( aGeomObj->GetType() == 37 /*GEOM_GROUP*/ ) { // Selected Group that belongs the main object
280             GEOMBase::GetShape(aGeomObj, shape);
281             if ( !shape.IsNull() ) {
282               TopExp_Explorer exp( shape, mySubShType );
283               for ( ; exp.More(); exp.Next() ) {
284                 int index = myPreviewActor->GetIndexByShape( exp.Current() );
285                 if ( index ) {
286                   mySelectedIDs.append( index );
287                   myPreviewActor->HighlightID( index );
288                 }
289               }
290             }
291           } else if ( aGeomObj->GetType() == 28 /*GEOM_SUBSHAPE*/ ||
292                       myEntry == IO->getEntry() )
293           {
294             GEOMBase::GetShape(aGeomObj, shape); 
295             if ( !shape.IsNull() && shape.ShapeType() == mySubShType ) {
296               int index = myPreviewActor->GetIndexByShape( shape );
297               if ( index ) {
298                 mySelectedIDs.append( index );
299                 myPreviewActor->HighlightID( index );
300               }
301             }
302           }
303         }
304       } else { // Selected Actor from Actor Collection
305         QString anEntry = IO->getEntry();
306         QString str = "_";
307         int index = anEntry.lastIndexOf( str );
308         anEntry.remove(0, index+1);
309         int ind = anEntry.toInt();
310         if ( ind )
311           mySelectedIDs.append( ind );
312       }
313     }
314   }
315   // update add button
316   myAddButton->setEnabled( ( myListWidget->count() < myMaxSize || myMaxSize == -1 ) &&
317                            mySelectedIDs.size() > 0 &&
318                            ( mySelectedIDs.size() <= myMaxSize || myMaxSize == -1 ) );
319
320   //Connect Selected Ids in viewer and dialog's Ids list
321   bool signalsBlocked = myListWidget->blockSignals( true );
322   myListWidget->clearSelection();
323   if ( mySelectedIDs.size() > 0 ) {
324     for (int i = 0; i < mySelectedIDs.size(); i++) {
325       QString anID = QString(" %1").arg( mySelectedIDs.at(i) );
326       QList<QListWidgetItem*> anItems = myListWidget->findItems ( anID, Qt::MatchExactly );
327       QListWidgetItem* item;
328       foreach(item, anItems)
329         item->setSelected(true);
330     }
331   }
332   myListWidget->blockSignals( signalsBlocked );
333 }
334
335 //=================================================================================
336 // function : onAdd()
337 // purpose  : Called when Add Button Clicked
338 //=================================================================================
339 void StdMeshersGUI_SubShapeSelectorWdg::onAdd()
340 {
341   if ( mySelectedIDs.size() < 1 )
342     return;
343
344   myListWidget->blockSignals( true );
345   for (int i = 0; i < mySelectedIDs.size() && (myMaxSize == -1 || myListOfIDs.size() < myMaxSize); i++) {
346     if ( myListOfIDs.indexOf( mySelectedIDs.at(i) ) == -1 ) {
347       QString anID = QString(" %1").arg( mySelectedIDs.at(i) );
348
349       QListWidgetItem* anItem = new QListWidgetItem( anID, myListWidget );
350       anItem->setSelected(true);
351       
352       myListOfIDs.append( mySelectedIDs.at(i) );
353     }
354   }
355   onListSelectionChanged();
356   myListWidget->blockSignals( false );
357
358   mySelectedIDs.clear();
359   myAddButton->setEnabled( false );
360 }
361          
362 //=================================================================================
363 // function : onRemove()
364 // purpose  : Called when Remove Button Clicked
365 //=================================================================================
366 void StdMeshersGUI_SubShapeSelectorWdg::onRemove()
367 {
368   if ( myListWidget->count() < 1 )
369     return;
370
371   myListWidget->blockSignals( true );
372   QList<QListWidgetItem*> selItems = myListWidget->selectedItems();
373   QListWidgetItem* item;
374   foreach(item, selItems) {
375     QString idStr = item->text();
376     int id = idStr.toInt();
377
378     int index = myListOfIDs.indexOf( id );
379     myListOfIDs.removeAt( index );
380     delete item;
381   }
382
383   onListSelectionChanged();
384   myListWidget->blockSignals( false );
385   
386   myAddButton->setEnabled( !mySelectedIDs.isEmpty() );
387 }
388
389 void StdMeshersGUI_SubShapeSelectorWdg::onPrevious()
390 {
391   if ( myPreviewActor ) {
392     myPreviewActor->previous();
393     myListWidget->clearSelection();
394     updateButtons();
395     if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
396       aViewWindow->Repaint();
397   }
398 }
399
400 void StdMeshersGUI_SubShapeSelectorWdg::onNext()
401 {
402   if ( myPreviewActor ) {
403     myPreviewActor->next();
404     myListWidget->clearSelection();
405     updateButtons();
406     if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
407       aViewWindow->Repaint();
408   }
409 }
410
411 //=================================================================================
412 // function : onListSelectionChanged()
413 // purpose  : Called when selection in element list is changed
414 //=================================================================================
415 void StdMeshersGUI_SubShapeSelectorWdg::onListSelectionChanged()
416 {
417   if ( !myPreviewActor )
418     return;
419
420   //mySelectionMgr->clearSelected();
421   TColStd_MapOfInteger aIndexes;
422   QList<QListWidgetItem*> selItems = myListWidget->selectedItems();
423   QListWidgetItem* anItem;
424   foreach(anItem, selItems)
425     myPreviewActor->HighlightID( anItem->text().toInt() );
426
427   // update remove button
428   myRemoveButton->setEnabled( selItems.size() > 0 );
429
430   emit selectionChanged();
431 }
432
433 //=================================================================================
434 // function : setGeomShape
435 // purpose  : Called to set geometry whose sub-shapes are selected
436 //================================================================================
437 void StdMeshersGUI_SubShapeSelectorWdg::SetGeomShapeEntry( const QString& theEntry,
438                                                            const QString& theMainShapeEntry )
439 {
440   if ( !theEntry.isEmpty() || theMainShapeEntry.isEmpty() )
441   {
442     myParamValue = theEntry;
443     myEntry      = theEntry.toStdString();
444     myMainEntry  = theMainShapeEntry.toStdString();
445
446     if ( myMainEntry.empty() ) myMainEntry = myEntry;
447     if ( myEntry.empty() )     myEntry     = myMainEntry;
448     if ( myMainEntry.length() > myEntry.length() &&
449          theMainShapeEntry.startsWith( theEntry ))
450       std::swap( myMainEntry, myEntry );
451
452     myGeomShape = GetTopoDSByEntry( myEntry.c_str() );
453     if ( myEntry == myMainEntry )
454       myMainShape = myGeomShape;
455     else
456       myMainShape = GetTopoDSByEntry( myMainEntry.c_str() );
457     updateState();
458     myIsNotCorrected = true;
459   }
460 }
461
462 //=================================================================================
463 // function : updateState
464 // purpose  : update Widget state
465 //=================================================================================
466 void StdMeshersGUI_SubShapeSelectorWdg::updateState()
467 {
468   bool state = false;
469   if ( !myGeomShape.IsNull() )
470     state = true;
471   myInfoLabel->setVisible( false );
472   myPrevButton->setVisible( false );
473   myNextButton->setVisible( false );
474   
475   myListWidget->setEnabled( state );
476   myAddButton->setEnabled( mySelectedIDs.size() > 0 );
477   
478   if (state) {
479     SUIT_OverrideCursor wc;
480     myPreviewActor = new SMESH_PreviewActorsCollection();
481     myPreviewActor->SetSelector( mySelector );
482     myPreviewActor->Init( myGeomShape, myMainShape, mySubShType, myEntry.c_str() );
483     myPreviewActor->SetShown( false );
484     myIsShown = false;
485     if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI )) {
486       myRenderer = aViewWindow->getRenderer();
487       myPreviewActor->AddToRender( myRenderer );
488       aViewWindow->Repaint();
489     }
490     updateButtons();
491   }
492 }
493
494 //=================================================================================
495 // function : GetGeomObjectByEntry
496 // purpose  : Called to get GeomObject
497 //=================================================================================
498 GEOM::GEOM_Object_var StdMeshersGUI_SubShapeSelectorWdg::GetGeomObjectByEntry( const QString& theEntry )
499 {
500   GEOM::GEOM_Object_var aGeomObj;
501   SALOMEDS::Study_var aStudy = SMESHGUI::GetSMESHGen()->GetCurrentStudy();
502   if ( !aStudy->_is_nil() )
503   {
504     SALOMEDS::SObject_var aSObj = aStudy->FindObjectID( theEntry.toLatin1().data() );
505     if (!aSObj->_is_nil() )
506     {
507       CORBA::Object_var obj = aSObj->GetObject();
508       aGeomObj = GEOM::GEOM_Object::_narrow(obj);
509       aSObj->UnRegister();
510     }
511   }
512   return aGeomObj._retn();
513 }
514
515 //=================================================================================
516 // function : setObjectByEntry
517 // purpose  : Called to get GeomObject
518 //=================================================================================
519 TopoDS_Shape StdMeshersGUI_SubShapeSelectorWdg::GetTopoDSByEntry( const QString& theEntry )
520 {
521   TopoDS_Shape shape;
522   GEOM::GEOM_Object_var aGeomObj = GetGeomObjectByEntry( theEntry );
523   GEOMBase::GetShape(aGeomObj, shape);
524   return shape;
525 }
526
527 //=================================================================================
528 // function : GetListOfIds
529 // purpose  : Called to get the list of SubShapes IDs
530 //=================================================================================
531 SMESH::long_array_var StdMeshersGUI_SubShapeSelectorWdg::GetListOfIDs()
532 {
533   SMESH::long_array_var anArray = new SMESH::long_array;
534
535   int size = myListOfIDs.size();
536   anArray->length( size );
537   for (int i = 0; i < size; i++)
538     anArray[i] = myListOfIDs.at(i);
539
540   return anArray;
541 }
542
543 //=================================================================================
544 // function : SetListOfIds
545 // purpose  : Called to set the list of SubShapes IDs. Returns false if any ID is invalid
546 //=================================================================================
547 bool StdMeshersGUI_SubShapeSelectorWdg::SetListOfIDs( SMESH::long_array_var theIds)
548 {
549   mySelectedIDs.clear();
550   myListOfIDs.clear();
551   int size = theIds->length();
552   for ( int i = 0; i < size; i++ )
553     mySelectedIDs.append( theIds[ i ] );
554
555   myListWidget->blockSignals( true );
556   myListWidget->clear();
557   myListWidget->blockSignals( false );
558
559   bool isOk = true;
560   if ( myPreviewActor )
561   {
562     for ( int i = 0; i < size && isOk; i++ )
563       isOk = myPreviewActor->IsValidIndex( theIds[ i ] );
564   }
565   else if ( !myMainShape.IsNull() )
566   {
567     TopTools_IndexedMapOfShape aMainMap;
568     TopExp::MapShapes(myMainShape, aMainMap);
569     for ( int i = 0; i < size && isOk; i++ )
570       isOk = ( theIds[ i ] > 0 && theIds[ i ] <= aMainMap.Extent() );
571   }
572   // mySelectedIDs = GetCorrectedListOfIDs( false, &isOk );
573   onAdd();
574   return isOk;
575 }
576
577 //=================================================================================
578 // function : GetMainShapeEntry
579 // purpose  : Called to get the Main Object Entry
580 //=================================================================================
581 const char* StdMeshersGUI_SubShapeSelectorWdg::GetMainShapeEntry()
582 {
583   if ( myMainEntry.empty() ) myMainEntry = "";
584   return myMainEntry.c_str();
585 }
586
587 void StdMeshersGUI_SubShapeSelectorWdg::updateButtons()
588 {
589   if ( myPreviewActor ) {
590     int total = myPreviewActor->count();
591     int chunk = myPreviewActor->currentChunk();
592     int chunkSize = myPreviewActor->chunkSize();
593     int imin = chunk*chunkSize+1;
594     int imax = std::min((chunk+1)*chunkSize, total);
595     bool vis = imax > 0 && total > chunkSize;
596     myInfoLabel->setVisible( vis );
597     myPrevButton->setVisible( vis );
598     myNextButton->setVisible( vis );
599     myInfoLabel->setText( tr( "X_FROM_Y_ITEMS_SHOWN" ).arg(imin).arg(imax).arg(total) );
600     myPrevButton->setEnabled( myPreviewActor->hasPrevious() );
601     myNextButton->setEnabled( myPreviewActor->hasNext() );
602   }
603 }