1 // Copyright (C) 2007-2010 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License.
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 // Lesser General Public License for more details.
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 // SMESH SMESHGUI : GUI for SMESH component
24 // File : SMESHGUI_VTKUtils.cxx
25 // Author : Open CASCADE S.A.S.
28 #include "SMESHGUI_VTKUtils.h"
31 #include "SMESHGUI_Utils.h"
32 #include "SMESHGUI_Filter.h"
34 #include <SMESH_Actor.h>
35 #include <SMESH_ActorUtils.h>
36 #include <SMESH_ObjectDef.h>
37 #include <SMDS_Mesh.hxx>
39 // SALOME GUI includes
40 #include <SUIT_Desktop.h>
41 #include <SUIT_Session.h>
42 #include <SUIT_MessageBox.h>
43 #include <SUIT_ViewManager.h>
44 #include <SUIT_ResourceMgr.h>
46 #include <SALOME_ListIO.hxx>
47 #include <SALOME_ListIteratorOfListIO.hxx>
49 #include <SVTK_Selector.h>
50 #include <SVTK_ViewModel.h>
51 #include <SVTK_ViewWindow.h>
53 #include <VTKViewer_Algorithm.h>
55 #include <LightApp_SelectionMgr.h>
56 #include <SalomeApp_Application.h>
57 #include <SalomeApp_Study.h>
59 // SALOME KERNEL includes
60 #include <utilities.h>
63 #include <SALOMEconfig.h>
64 #include CORBA_CLIENT_HEADER(SMESH_Mesh)
65 #include CORBA_CLIENT_HEADER(SMESH_Group)
68 #include <vtkRenderer.h>
69 #include <vtkActorCollection.h>
70 #include <vtkUnstructuredGrid.h>
73 #include <TColStd_IndexedMapOfInteger.hxx>
74 #include <Standard_ErrorHandler.hxx>
78 typedef std::map<TKeyOfVisualObj,TVisualObjPtr> TVisualObjCont;
79 static TVisualObjCont VISUAL_OBJ_CONT;
81 //=============================================================================
83 * \brief Allocate some memory at construction and release it at destruction.
84 * Is used to be able to continue working after mesh generation or visualization
85 * break due to lack of memory
87 //=============================================================================
92 MemoryReserve(): myBuf( new char[1024*1024*1] ){} // 1M
93 void Free() { if (myBuf) { delete [] myBuf; myBuf = 0; }}
94 ~MemoryReserve() { Free(); }
96 static MemoryReserve* theVISU_MemoryReserve = new MemoryReserve;
98 //================================================================================
100 * \brief Remove VisualObj and its actor from all views
102 //================================================================================
104 void RemoveVisualObjectWithActors( const char* theEntry )
106 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>
107 ( SUIT_Session::session()->activeApplication() );
108 SUIT_ViewManager* aViewManager =
109 app ? app->getViewManager(SVTK_Viewer::Type(), true) : 0;
110 if ( aViewManager ) {
111 QVector<SUIT_ViewWindow*> views = aViewManager->getViews();
112 for ( int iV = 0; iV < views.count(); ++iV ) {
113 if ( SMESH_Actor* actor = FindActorByEntry( views[iV], theEntry)) {
114 if(SVTK_ViewWindow* vtkWnd = GetVtkViewWindow(views[iV]))
115 vtkWnd->RemoveActor(actor);
119 int aStudyId = aViewManager->study()->id();
120 TVisualObjCont::key_type aKey(aStudyId,theEntry);
121 TVisualObjCont::iterator anIter = VISUAL_OBJ_CONT.find(aKey);
122 if(anIter != VISUAL_OBJ_CONT.end()) {
123 // for unknown reason, object destructor is not called, so clear object manually
124 anIter->second->GetUnstructuredGrid()->SetCells(0,0,0);
125 anIter->second->GetUnstructuredGrid()->SetPoints(0);
127 VISUAL_OBJ_CONT.erase(aKey);
130 //================================================================================
132 * \brief Remove all VisualObjs and their actors from all views
134 //================================================================================
136 void RemoveAllObjectsWithActors()
138 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>
139 ( SUIT_Session::session()->activeApplication() );
141 ViewManagerList viewMgrs = app->viewManagers();
142 for ( int iM = 0; iM < viewMgrs.count(); ++iM ) {
143 SUIT_ViewManager* aViewManager = viewMgrs.at( iM );
144 if ( aViewManager && aViewManager->getType() == SVTK_Viewer::Type()) {
145 QVector<SUIT_ViewWindow*> views = aViewManager->getViews();
146 for ( int iV = 0; iV < views.count(); ++iV ) {
147 if(SVTK_ViewWindow* vtkWnd = GetVtkViewWindow(views[iV])) {
148 vtkRenderer *aRenderer = vtkWnd->getRenderer();
149 VTK::ActorCollectionCopy aCopy(aRenderer->GetActors());
150 vtkActorCollection *actors = aCopy.GetActors();
151 for (int i = 0; i < actors->GetNumberOfItems(); ++i ) {
152 // size of actors changes inside the loop
153 if (SMESH_Actor *actor = dynamic_cast<SMESH_Actor*>(actors->GetItemAsObject(i)))
155 vtkWnd->RemoveActor(actor);
163 TVisualObjCont::iterator anIter = VISUAL_OBJ_CONT.begin();
164 for ( ; anIter != VISUAL_OBJ_CONT.end(); ++anIter ) {
165 // for unknown reason, object destructor is not called, so clear object manually
166 anIter->second->GetUnstructuredGrid()->SetCells(0,0,0);
167 anIter->second->GetUnstructuredGrid()->SetPoints(0);
169 VISUAL_OBJ_CONT.clear();
172 //================================================================================
174 * \brief Remove all VisualObjs of a study
176 //================================================================================
178 void RemoveVisuData(int studyID)
180 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>
181 ( SUIT_Session::session()->activeApplication() );
183 ViewManagerList viewMgrs = app->viewManagers();
184 for ( int iM = 0; iM < viewMgrs.count(); ++iM ) {
185 SUIT_ViewManager* aViewManager = viewMgrs.at( iM );
186 if ( aViewManager && aViewManager->getType() == SVTK_Viewer::Type() &&
187 aViewManager->study()->id() == studyID ) {
188 QVector<SUIT_ViewWindow*> views = aViewManager->getViews();
189 for ( int iV = 0; iV < views.count(); ++iV ) {
190 if(SVTK_ViewWindow* vtkWnd = GetVtkViewWindow(views[iV])) {
191 vtkRenderer *aRenderer = vtkWnd->getRenderer();
192 VTK::ActorCollectionCopy aCopy(aRenderer->GetActors());
193 vtkActorCollection *actors = aCopy.GetActors();
194 for (int i = 0; i < actors->GetNumberOfItems(); ++i ) {
195 // size of actors changes inside the loop
196 if(SMESH_Actor *actor = dynamic_cast<SMESH_Actor*>(actors->GetItemAsObject(i)))
198 vtkWnd->RemoveActor(actor);
206 TVisualObjCont::iterator anIter = VISUAL_OBJ_CONT.begin();
207 for ( ; anIter != VISUAL_OBJ_CONT.end(); ) {
208 int curId = anIter->first.first;
209 if ( curId == studyID ) {
210 // for unknown reason, object destructor is not called, so clear object manually
211 anIter->second->GetUnstructuredGrid()->SetCells(0,0,0);
212 anIter->second->GetUnstructuredGrid()->SetPoints(0);
213 VISUAL_OBJ_CONT.erase( anIter++ ); // anIter++ returns a copy of self before incrementing
221 //================================================================================
223 * \brief Notify the user on problems during visualization
225 //================================================================================
227 void OnVisuException()
230 #if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
233 // PAL16774 (Crash after display of many groups). Salome sometimes crashes just
234 // after or at showing this message, so we do an additional check of available memory
235 // char* buf = new char[100*1024];
237 SUIT_MessageBox::warning(SMESHGUI::desktop(), QObject::tr("SMESH_WRN_WARNING"),
238 QObject::tr("SMESH_VISU_PROBLEM"));
240 // no more memory at all: last resort
241 MESSAGE_BEGIN ( "SMESHGUI_VTKUtils::OnVisuException(), exception even at showing a message!!!" <<
242 std::endl << "Try to remove all visual data..." );
243 if (theVISU_MemoryReserve) {
244 delete theVISU_MemoryReserve;
245 theVISU_MemoryReserve = 0;
247 RemoveAllObjectsWithActors();
248 SUIT_MessageBox::warning(SMESHGUI::desktop(), QObject::tr("SMESH_WRN_WARNING"),
249 QObject::tr("SMESH_VISU_PROBLEM_CLEAR"));
250 MESSAGE_END ( "...done" );
253 //================================================================================
255 * \brief Returns an updated visual object
257 //================================================================================
259 TVisualObjPtr GetVisualObj(int theStudyId, const char* theEntry){
260 TVisualObjPtr aVisualObj;
261 TVisualObjCont::key_type aKey(theStudyId,theEntry);
263 #if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
266 TVisualObjCont::iterator anIter = VISUAL_OBJ_CONT.find(aKey);
267 if(anIter != VISUAL_OBJ_CONT.end()){
268 aVisualObj = anIter->second;
270 SalomeApp_Application* app =
271 dynamic_cast<SalomeApp_Application*>( SMESHGUI::activeStudy()->application() );
272 _PTR(Study) aStudy = SMESHGUI::activeStudy()->studyDS();
273 _PTR(SObject) aSObj = aStudy->FindObjectID(theEntry);
275 _PTR(GenericAttribute) anAttr;
276 if(aSObj->FindAttribute(anAttr,"AttributeIOR")){
277 _PTR(AttributeIOR) anIOR = anAttr;
278 CORBA::String_var aVal = anIOR->Value().c_str();
279 CORBA::Object_var anObj = app->orb()->string_to_object( aVal.in() );
280 if(!CORBA::is_nil(anObj)){
281 //Try narrow to SMESH_Mesh interface
282 SMESH::SMESH_Mesh_var aMesh = SMESH::SMESH_Mesh::_narrow(anObj);
283 if(!aMesh->_is_nil()){
284 aVisualObj.reset(new SMESH_MeshObj(aMesh));
285 TVisualObjCont::value_type aValue(aKey,aVisualObj);
286 VISUAL_OBJ_CONT.insert(aValue);
288 //Try narrow to SMESH_Group interface
289 SMESH::SMESH_GroupBase_var aGroup = SMESH::SMESH_GroupBase::_narrow(anObj);
290 if(!aGroup->_is_nil()){
291 _PTR(SObject) aFatherSObj = aSObj->GetFather();
292 if(!aFatherSObj) return aVisualObj;
293 aFatherSObj = aFatherSObj->GetFather();
294 if(!aFatherSObj) return aVisualObj;
295 CORBA::String_var anEntry = aFatherSObj->GetID().c_str();
296 TVisualObjPtr aVisObj = GetVisualObj(theStudyId,anEntry.in());
297 if(SMESH_MeshObj* aMeshObj = dynamic_cast<SMESH_MeshObj*>(aVisObj.get())){
298 aVisualObj.reset(new SMESH_GroupObj(aGroup,aMeshObj));
299 TVisualObjCont::value_type aValue(aKey,aVisualObj);
300 VISUAL_OBJ_CONT.insert(aValue);
303 //Try narrow to SMESH_subMesh interface
304 SMESH::SMESH_subMesh_var aSubMesh = SMESH::SMESH_subMesh::_narrow(anObj);
305 if(!aSubMesh->_is_nil()){
306 _PTR(SObject) aFatherSObj = aSObj->GetFather();
307 if(!aFatherSObj) return aVisualObj;
308 aFatherSObj = aFatherSObj->GetFather();
309 if(!aFatherSObj) return aVisualObj;
310 CORBA::String_var anEntry = aFatherSObj->GetID().c_str();
311 TVisualObjPtr aVisObj = GetVisualObj(theStudyId,anEntry.in());
312 if(SMESH_MeshObj* aMeshObj = dynamic_cast<SMESH_MeshObj*>(aVisObj.get())){
313 aVisualObj.reset(new SMESH_subMeshObj(aSubMesh,aMeshObj));
314 TVisualObjCont::value_type aValue(aKey,aVisualObj);
315 VISUAL_OBJ_CONT.insert(aValue);
323 INFOS("GetMeshObj - There is no SMESH_Mesh object for the SALOMEDS::Strudy and Entry!!!");
324 return TVisualObjPtr();
327 bool objModified = false;
330 #if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
333 objModified = aVisualObj->Update();
337 MESSAGE ( "Exception in SMESHGUI_VTKUtils::GetVisualObj()" );
339 RemoveVisualObjectWithActors( theEntry ); // remove this object
346 // PAL16631. Mesurements showed that to show aVisualObj in SHADING(default) mode,
347 // ~10 times more memory is used than it occupies.
348 // Warn the user if there is less free memory than 30 sizes of a grid
349 // TODO: estimate memory usage in other modes and take current mode into account
350 int freeMB = SMDS_Mesh::CheckMemory(true);
351 int usedMB = aVisualObj->GetUnstructuredGrid()->GetActualMemorySize() / 1024;
352 if ( freeMB > 0 && usedMB * 30 > freeMB ) {
354 MESSAGE ( "SMESHGUI_VTKUtils::GetVisualObj(), freeMB=" << freeMB
355 << ", usedMB=" << usedMB );
357 bool continu = false;
358 if ( usedMB * 10 > freeMB )
359 // even dont try to show
360 SUIT_MessageBox::warning(SMESHGUI::desktop(), QObject::tr("SMESH_WRN_WARNING"),
361 QObject::tr("SMESH_NO_MESH_VISUALIZATION"));
363 // there is a chance to succeed
364 continu = SUIT_MessageBox::warning
365 (SMESHGUI::desktop(),
366 QObject::tr("SMESH_WRN_WARNING"),
367 QObject::tr("SMESH_CONTINUE_MESH_VISUALIZATION"),
368 SUIT_MessageBox::Yes | SUIT_MessageBox::No,
369 SUIT_MessageBox::Yes ) == SUIT_MessageBox::Yes;
371 // remove the corresponding actors from all views
372 RemoveVisualObjectWithActors( theEntry );
382 /*! Return active view window, if it instantiates SVTK_ViewWindow class,
383 * overwise find or create corresponding view window, make it active and return it.
384 * \note Active VVTK_ViewWindow can be returned, because it inherits SVTK_ViewWindow.
386 SVTK_ViewWindow* GetViewWindow (const SalomeApp_Module* theModule,
387 bool createIfNotFound)
389 SalomeApp_Application* anApp;
391 anApp = theModule->getApp();
393 anApp = dynamic_cast<SalomeApp_Application*>
394 (SUIT_Session::session()->activeApplication());
397 if (SVTK_ViewWindow* aView = dynamic_cast<SVTK_ViewWindow*>(anApp->desktop()->activeWindow()))
400 SUIT_ViewManager* aViewManager =
401 anApp->getViewManager(SVTK_Viewer::Type(), createIfNotFound);
403 if (SUIT_ViewWindow* aViewWindow = aViewManager->getActiveView()) {
404 if (SVTK_ViewWindow* aView = dynamic_cast<SVTK_ViewWindow*>(aViewWindow)) {
405 aViewWindow->raise();
406 aViewWindow->setFocus();
415 SVTK_ViewWindow* FindVtkViewWindow (SUIT_ViewManager* theMgr,
416 SUIT_ViewWindow * theWindow)
421 QVector<SUIT_ViewWindow*> views = theMgr->getViews();
422 if( views.contains( theWindow ) )
423 return GetVtkViewWindow( theWindow );
428 SVTK_ViewWindow* GetVtkViewWindow(SUIT_ViewWindow* theWindow){
429 return dynamic_cast<SVTK_ViewWindow*>(theWindow);
432 /* SUIT_ViewWindow* GetActiveWindow()
434 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( SUIT_Session::session()->activeApplication() );
437 SUIT_ViewManager* mgr = app->activeViewManager();
439 return mgr->getActiveView();
444 SVTK_ViewWindow* GetCurrentVtkView(){
445 return GetVtkViewWindow( GetActiveWindow() );
449 void RepaintCurrentView()
451 if (SVTK_ViewWindow* wnd = GetCurrentVtkView())
454 #if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
457 wnd->getRenderer()->Render();
462 MESSAGE ( "Exception in SMESHGUI_VTKUtils::RepaintCurrentView()" );
469 void RepaintViewWindow(SVTK_ViewWindow* theWindow)
472 #if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
475 theWindow->getRenderer()->Render();
476 theWindow->Repaint();
480 MESSAGE ( "Exception in SMESHGUI_VTKUtils::RepaintViewWindow(SVTK_ViewWindow*)" );
486 void RenderViewWindow(SVTK_ViewWindow* theWindow)
489 #if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
492 theWindow->getRenderer()->Render();
493 theWindow->Repaint();
497 MESSAGE ( "Exception in SMESHGUI_VTKUtils::RenderViewWindow(SVTK_ViewWindow*)" );
504 if(SVTK_ViewWindow* wnd = GetCurrentVtkView() ){
506 #if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
514 MESSAGE ( "Exception in SMESHGUI_VTKUtils::FitAll()" );
522 SMESH_Actor* FindActorByEntry(SUIT_ViewWindow *theWindow,
523 const char* theEntry)
525 if(SVTK_ViewWindow* aViewWindow = GetVtkViewWindow(theWindow)){
526 vtkRenderer *aRenderer = aViewWindow->getRenderer();
527 VTK::ActorCollectionCopy aCopy(aRenderer->GetActors());
528 vtkActorCollection *aCollection = aCopy.GetActors();
529 aCollection->InitTraversal();
530 while(vtkActor *anAct = aCollection->GetNextActor()){
531 if(SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)){
532 if(anActor->hasIO()){
533 Handle(SALOME_InteractiveObject) anIO = anActor->getIO();
534 if(anIO->hasEntry() && strcmp(anIO->getEntry(),theEntry) == 0){
545 SMESH_Actor* FindActorByEntry(const char* theEntry){
546 return FindActorByEntry(GetActiveWindow(),theEntry);
550 SMESH_Actor* FindActorByObject(CORBA::Object_ptr theObject){
551 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( SUIT_Session::session()->activeApplication() );
555 if(!CORBA::is_nil(theObject)){
556 _PTR(Study) aStudy = GetActiveStudyDocument();
557 CORBA::String_var anIOR = app->orb()->object_to_string( theObject );
558 _PTR(SObject) aSObject = aStudy->FindObjectIOR(anIOR.in());
560 CORBA::String_var anEntry = aSObject->GetID().c_str();
561 return FindActorByEntry(anEntry.in());
568 SMESH_Actor* CreateActor(_PTR(Study) theStudy,
569 const char* theEntry,
572 SMESH_Actor *anActor = NULL;
573 CORBA::Long anId = theStudy->StudyId();
574 if(TVisualObjPtr aVisualObj = GetVisualObj(anId,theEntry)){
575 _PTR(SObject) aSObj = theStudy->FindObjectID(theEntry);
577 _PTR(GenericAttribute) anAttr;
578 if(aSObj->FindAttribute(anAttr,"AttributeName")){
579 _PTR(AttributeName) aName = anAttr;
580 std::string aNameVal = aName->Value();
581 anActor = SMESH_Actor::New(aVisualObj,theEntry,aNameVal.c_str(),theIsClear);
584 SMESH::SMESH_GroupBase_var aGroup = SMESH::SMESH_GroupBase::_narrow( SMESH::SObjectToObject( aSObj ));
585 if(!CORBA::is_nil(aGroup))
587 SALOMEDS::Color aColor = aGroup->GetColor();
588 if( !( aColor.R > 0 || aColor.G > 0 || aColor.B > 0 ) )
590 int r = 0, g = 0, b = 0;
591 SMESH::GetColor( "SMESH", "fill_color", r, g, b, QColor( 0, 170, 255 ) );
592 aColor.R = (float)r / 255.0;
593 aColor.G = (float)g / 255.0;
594 aColor.B = (float)b / 255.0;
595 aGroup->SetColor( aColor );
597 if( aGroup->GetType() == SMESH::NODE )
598 anActor->SetNodeColor( aColor.R, aColor.G, aColor.B );
599 else if( aGroup->GetType() == SMESH::EDGE )
600 anActor->SetEdgeColor( aColor.R, aColor.G, aColor.B );
601 else if( aGroup->GetType() == SMESH::ELEM0D )
602 anActor->Set0DColor( aColor.R, aColor.G, aColor.B );
604 anActor->SetSufaceColor( aColor.R, aColor.G, aColor.B );
612 void DisplayActor( SUIT_ViewWindow *theWnd, SMESH_Actor* theActor){
613 if(SVTK_ViewWindow* vtkWnd = GetVtkViewWindow(theWnd)){
615 #if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
618 vtkWnd->AddActor(theActor);
623 MESSAGE ( "Exception in SMESHGUI_VTKUtils::DisplayActor()" );
631 void RemoveActor( SUIT_ViewWindow *theWnd, SMESH_Actor* theActor){
632 if(SVTK_ViewWindow* vtkWnd = GetVtkViewWindow(theWnd)){
633 vtkWnd->RemoveActor(theActor);
634 if(theActor->hasIO()){
635 Handle(SALOME_InteractiveObject) anIO = theActor->getIO();
636 if(anIO->hasEntry()){
637 std::string anEntry = anIO->getEntry();
638 SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( vtkWnd->getViewManager()->study() );
639 int aStudyId = aStudy->id();
640 TVisualObjCont::key_type aKey(aStudyId,anEntry);
641 VISUAL_OBJ_CONT.erase(aKey);
649 //================================================================================
651 * \brief Return true if there are no SMESH actors in a view
653 //================================================================================
655 bool noSmeshActors(SUIT_ViewWindow *theWnd)
657 if(SVTK_ViewWindow* aViewWindow = GetVtkViewWindow(theWnd)) {
658 vtkRenderer *aRenderer = aViewWindow->getRenderer();
659 VTK::ActorCollectionCopy aCopy(aRenderer->GetActors());
660 vtkActorCollection *aCollection = aCopy.GetActors();
661 aCollection->InitTraversal();
662 while(vtkActor *anAct = aCollection->GetNextActor())
663 if(dynamic_cast<SMESH_Actor*>(anAct))
669 bool UpdateView(SUIT_ViewWindow *theWnd, EDisplaing theAction, const char* theEntry)
672 SVTK_ViewWindow* aViewWnd = GetVtkViewWindow(theWnd);
678 vtkRenderer *aRenderer = aViewWnd->getRenderer();
679 VTK::ActorCollectionCopy aCopy(aRenderer->GetActors());
680 vtkActorCollection *aCollection = aCopy.GetActors();
681 aCollection->InitTraversal();
685 while (vtkActor *anAct = aCollection->GetNextActor()) {
686 if (SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)) {
687 anActor->SetVisibility(true);
694 while (vtkActor *anAct = aCollection->GetNextActor()) {
695 if (SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)) {
696 anActor->SetVisibility(false);
701 if (SMESH_Actor *anActor = FindActorByEntry(theWnd,theEntry)) {
705 anActor->SetVisibility(true);
706 if (theAction == eDisplayOnly) aRenderer->ResetCameraClippingRange();
709 anActor->SetVisibility(false);
717 SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>(theWnd->getViewManager()->study());
718 _PTR(Study) aDocument = aStudy->studyDS();
719 // Pass non-visual objects (hypotheses, etc.), return true in this case
720 CORBA::Long anId = aDocument->StudyId();
721 TVisualObjPtr aVisualObj;
722 if ( (aVisualObj = GetVisualObj(anId,theEntry)) && aVisualObj->IsValid())
724 if ((anActor = CreateActor(aDocument,theEntry,true))) {
725 bool needFitAll = noSmeshActors(theWnd); // fit for the first object only
726 DisplayActor(theWnd,anActor);
727 // FitAll(); - PAL16770(Display of a group performs an automatic fit all)
728 if (needFitAll) FitAll();
744 bool UpdateView(EDisplaing theAction, const char* theEntry){
745 SalomeApp_Study* aStudy = dynamic_cast< SalomeApp_Study* >( GetActiveStudy() );
746 SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( aStudy->application() );
747 SUIT_ViewWindow *aWnd = app->activeViewManager()->getActiveView();
748 return UpdateView(aWnd,theAction,theEntry);
752 if(SVTK_ViewWindow* aWnd = SMESH::GetCurrentVtkView()){
753 LightApp_SelectionMgr* mgr = SMESHGUI::selectionMgr();
754 SALOME_ListIO selected; mgr->selectedObjects( selected );
756 if( selected.Extent() == 0){
757 vtkRenderer* aRenderer = aWnd->getRenderer();
758 VTK::ActorCollectionCopy aCopy(aRenderer->GetActors());
759 vtkActorCollection *aCollection = aCopy.GetActors();
760 aCollection->InitTraversal();
761 while(vtkActor *anAct = aCollection->GetNextActor()){
762 if(SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)){
764 if (!Update(anActor->getIO(),anActor->GetVisibility()))
765 break; // avoid multiple warinings if visu failed
769 SALOME_ListIteratorOfListIO anIter( selected );
770 for( ; anIter.More(); anIter.Next()){
771 Handle(SALOME_InteractiveObject) anIO = anIter.Value();
772 if ( !Update(anIO,true) )
773 break; // avoid multiple warinings if visu failed
776 RepaintCurrentView();
781 bool Update(const Handle(SALOME_InteractiveObject)& theIO, bool theDisplay)
783 _PTR(Study) aStudy = GetActiveStudyDocument();
784 CORBA::Long anId = aStudy->StudyId();
785 if ( TVisualObjPtr aVisualObj = SMESH::GetVisualObj(anId,theIO->getEntry())) {
787 UpdateView(SMESH::eDisplay,theIO->getEntry());
794 void UpdateSelectionProp( SMESHGUI* theModule ) {
798 SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( theModule->application() );
801 MESSAGE( "UpdateSelectionProp: Application is null" );
805 SUIT_ViewManager* vm = app->activeViewManager();
808 MESSAGE( "UpdateSelectionProp: View manager is null" );
812 QVector<SUIT_ViewWindow*> views = vm->getViews();
814 SUIT_ResourceMgr* mgr = SMESH::GetResourceMgr( theModule );
817 MESSAGE( "UpdateSelectionProp: Resource manager is null" );
821 QColor aHiColor = mgr->colorValue( "SMESH", "selection_object_color", Qt::white ),
822 aSelColor = mgr->colorValue( "SMESH", "selection_element_color", Qt::yellow ),
823 aPreColor = mgr->colorValue( "SMESH", "highlight_color", Qt::cyan );
825 int SW = mgr->integerValue( "SMESH", "selection_width", 5 ),
826 PW = mgr->integerValue( "SMESH", "highlight_width", 5 );
828 // adjust highlight_width to the width of mesh entities
829 int aElem0DSize = mgr->integerValue("SMESH", "elem0d_size", 5);
830 int aLineWidth = mgr->integerValue("SMESH", "element_width", 1);
831 int maxSize = aElem0DSize;
832 if (aElem0DSize > maxSize) maxSize = aElem0DSize;
833 if (aLineWidth > maxSize) maxSize = aLineWidth;
834 if (PW < maxSize + 2) PW = maxSize + 2;
836 double SP1 = mgr->doubleValue( "SMESH", "selection_precision_node", 0.025 ),
837 SP2 = mgr->doubleValue( "SMESH", "selection_precision_element", 0.001 ),
838 SP3 = mgr->doubleValue( "SMESH", "selection_precision_object", 0.025 );
840 for ( int i=0, n=views.count(); i<n; i++ ){
841 // update VTK viewer properties
842 if(SVTK_ViewWindow* aVtkView = GetVtkViewWindow( views[i] )){
843 // mesh element selection
844 aVtkView->SetSelectionProp(aSelColor.red()/255.,
845 aSelColor.green()/255.,
846 aSelColor.blue()/255.,
849 aVtkView->SetSelectionTolerance(SP1, SP2, SP3);
852 aVtkView->SetPreselectionProp(aPreColor.red()/255.,
853 aPreColor.green()/255.,
854 aPreColor.blue()/255.,
857 vtkRenderer* aRenderer = aVtkView->getRenderer();
858 VTK::ActorCollectionCopy aCopy(aRenderer->GetActors());
859 vtkActorCollection *aCollection = aCopy.GetActors();
860 aCollection->InitTraversal();
861 while(vtkActor *anAct = aCollection->GetNextActor()){
862 if(SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)){
863 anActor->SetHighlightColor(aHiColor.red()/255.,
864 aHiColor.green()/255.,
865 aHiColor.blue()/255.);
866 anActor->SetPreHighlightColor(aPreColor.red()/255.,
867 aPreColor.green()/255.,
868 aPreColor.blue()/255.);
876 //----------------------------------------------------------------------------
878 GetSelector(SUIT_ViewWindow *theWindow)
880 if(SVTK_ViewWindow* aWnd = GetVtkViewWindow(theWindow))
881 return aWnd->GetSelector();
886 void SetFilter(const Handle(VTKViewer_Filter)& theFilter,
887 SVTK_Selector* theSelector)
890 theSelector->SetFilter(theFilter);
893 Handle(VTKViewer_Filter) GetFilter(int theId, SVTK_Selector* theSelector)
895 return theSelector->GetFilter(theId);
898 bool IsFilterPresent(int theId, SVTK_Selector* theSelector)
900 return theSelector->IsFilterPresent(theId);
903 void RemoveFilter(int theId, SVTK_Selector* theSelector)
905 theSelector->RemoveFilter(theId);
908 void RemoveFilters(SVTK_Selector* theSelector)
910 for ( int id = SMESH::NodeFilter; theSelector && id < SMESH::LastFilter; id++ )
911 theSelector->RemoveFilter( id );
914 bool IsValid(SALOME_Actor* theActor, int theCellId,
915 SVTK_Selector* theSelector)
917 return theSelector->IsValid(theActor,theCellId);
921 //----------------------------------------------------------------------------
922 void SetPointRepresentation(bool theIsVisible){
923 if(SVTK_ViewWindow* aViewWindow = GetCurrentVtkView()){
924 vtkRenderer *aRenderer = aViewWindow->getRenderer();
925 VTK::ActorCollectionCopy aCopy(aRenderer->GetActors());
926 vtkActorCollection *aCollection = aCopy.GetActors();
927 aCollection->InitTraversal();
928 while(vtkActor *anAct = aCollection->GetNextActor()){
929 if(SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)){
930 if(anActor->GetVisibility()){
931 anActor->SetPointRepresentation(theIsVisible);
935 RepaintCurrentView();
940 void SetPickable(SMESH_Actor* theActor){
941 if(SVTK_ViewWindow* aWnd = GetCurrentVtkView()){
942 int anIsAllPickable = (theActor == NULL);
943 vtkRenderer *aRenderer = aWnd->getRenderer();
944 VTK::ActorCollectionCopy aCopy(aRenderer->GetActors());
945 vtkActorCollection *aCollection = aCopy.GetActors();
946 aCollection->InitTraversal();
947 while(vtkActor *anAct = aCollection->GetNextActor()){
948 if(SALOME_Actor *anActor = dynamic_cast<SALOME_Actor*>(anAct)){
949 if(anActor->GetVisibility()){
950 anActor->SetPickable(anIsAllPickable);
955 theActor->SetPickable(!anIsAllPickable);
956 RepaintCurrentView();
961 //----------------------------------------------------------------------------
962 int GetNameOfSelectedNodes(SVTK_Selector* theSelector,
963 const Handle(SALOME_InteractiveObject)& theIO,
967 TColStd_IndexedMapOfInteger aMapIndex;
968 theSelector->GetIndex(theIO,aMapIndex);
970 for(int i = 1; i <= aMapIndex.Extent(); i++)
971 theName += QString(" %1").arg(aMapIndex(i));
973 return aMapIndex.Extent();
976 int GetNameOfSelectedElements(SVTK_Selector* theSelector,
977 const Handle(SALOME_InteractiveObject)& theIO,
981 TColStd_IndexedMapOfInteger aMapIndex;
982 theSelector->GetIndex(theIO,aMapIndex);
984 typedef std::set<int> TIdContainer;
985 TIdContainer anIdContainer;
986 for( int i = 1; i <= aMapIndex.Extent(); i++)
987 anIdContainer.insert(aMapIndex(i));
989 TIdContainer::const_iterator anIter = anIdContainer.begin();
990 for( ; anIter != anIdContainer.end(); anIter++)
991 theName += QString(" %1").arg(*anIter);
993 return aMapIndex.Extent();
997 int GetEdgeNodes(SVTK_Selector* theSelector,
998 const TVisualObjPtr& theVisualObject,
1002 const SALOME_ListIO& selected = theSelector->StoredIObjects();
1004 if ( selected.Extent() != 1 )
1007 Handle(SALOME_InteractiveObject) anIO = selected.First();
1008 if ( anIO.IsNull() || !anIO->hasEntry() )
1011 TColStd_IndexedMapOfInteger aMapIndex;
1012 theSelector->GetIndex( anIO, aMapIndex );
1013 if ( aMapIndex.Extent() != 2 )
1016 int anObjId = -1, anEdgeNum = -1;
1017 for ( int i = 1; i <= aMapIndex.Extent(); i++ ) {
1018 int aVal = aMapIndex( i );
1022 anEdgeNum = abs( aVal ) - 1;
1025 if ( anObjId == -1 || anEdgeNum == -1 )
1028 return theVisualObject->GetEdgeNodes( anObjId, anEdgeNum, theId1, theId2 ) ? 1 : -1;
1031 //----------------------------------------------------------------------------
1032 int GetNameOfSelectedNodes(LightApp_SelectionMgr *theMgr,
1033 const Handle(SALOME_InteractiveObject)& theIO,
1037 if(theIO->hasEntry()){
1038 if(FindActorByEntry(theIO->getEntry())){
1039 TColStd_IndexedMapOfInteger aMapIndex;
1040 theMgr->GetIndexes(theIO,aMapIndex);
1041 for(int i = 1; i <= aMapIndex.Extent(); i++){
1042 theName += QString(" %1").arg(aMapIndex(i));
1044 return aMapIndex.Extent();
1050 int GetNameOfSelectedNodes(LightApp_SelectionMgr *theMgr, QString& theName){
1052 SALOME_ListIO selected; theMgr->selectedObjects( selected );
1053 if(selected.Extent() == 1){
1054 Handle(SALOME_InteractiveObject) anIO = selected.First();
1055 return GetNameOfSelectedNodes(theMgr,anIO,theName);
1061 int GetNameOfSelectedElements(LightApp_SelectionMgr *theMgr,
1062 const Handle(SALOME_InteractiveObject)& theIO,
1066 if(theIO->hasEntry()){
1067 if(FindActorByEntry(theIO->getEntry())){
1068 TColStd_IndexedMapOfInteger aMapIndex;
1069 theMgr->GetIndexes(theIO,aMapIndex);
1070 typedef std::set<int> TIdContainer;
1071 TIdContainer anIdContainer;
1072 for( int i = 1; i <= aMapIndex.Extent(); i++)
1073 anIdContainer.insert(aMapIndex(i));
1074 TIdContainer::const_iterator anIter = anIdContainer.begin();
1075 for( ; anIter != anIdContainer.end(); anIter++){
1076 theName += QString(" %1").arg(*anIter);
1078 return aMapIndex.Extent();
1085 int GetNameOfSelectedElements(LightApp_SelectionMgr *theMgr, QString& theName)
1088 SALOME_ListIO selected; theMgr->selectedObjects( selected );
1090 if( selected.Extent() == 1){
1091 Handle(SALOME_InteractiveObject) anIO = selected.First();
1092 return GetNameOfSelectedElements(theMgr,anIO,theName);
1097 int GetSelected(LightApp_SelectionMgr* theMgr,
1098 TColStd_IndexedMapOfInteger& theMap,
1099 const bool theIsElement)
1102 SALOME_ListIO selected; theMgr->selectedObjects( selected );
1104 if ( selected.Extent() == 1 )
1106 Handle(SALOME_InteractiveObject) anIO = selected.First();
1107 if ( anIO->hasEntry() ) {
1108 theMgr->GetIndexes( anIO, theMap );
1111 return theMap.Extent();
1115 int GetEdgeNodes( LightApp_SelectionMgr* theMgr, int& theId1, int& theId2 )
1117 SALOME_ListIO selected; theMgr->selectedObjects( selected );
1119 if ( selected.Extent() != 1 )
1122 Handle(SALOME_InteractiveObject) anIO = selected.First();
1123 if ( anIO.IsNull() || !anIO->hasEntry() )
1126 SMESH_Actor *anActor = SMESH::FindActorByEntry( anIO->getEntry() );
1130 TColStd_IndexedMapOfInteger aMapIndex;
1131 theMgr->GetIndexes( anIO, aMapIndex );
1132 if ( aMapIndex.Extent() != 2 )
1135 int anObjId = -1, anEdgeNum = -1;
1136 for ( int i = 1; i <= aMapIndex.Extent(); i++ ) {
1137 int aVal = aMapIndex( i );
1141 anEdgeNum = abs( aVal );
1144 if ( anObjId == -1 || anEdgeNum == -1 )
1147 return anActor->GetObject()->GetEdgeNodes( anObjId, anEdgeNum, theId1, theId2 ) ? 1 : -1;
1150 void SetControlsPrecision( const long theVal )
1152 if( SVTK_ViewWindow* aWnd = SMESH::GetCurrentVtkView() )
1154 vtkRenderer *aRenderer = aWnd->getRenderer();
1155 VTK::ActorCollectionCopy aCopy(aRenderer->GetActors());
1156 vtkActorCollection *aCollection = aCopy.GetActors();
1157 aCollection->InitTraversal();
1159 while ( vtkActor *anAct = aCollection->GetNextActor())
1161 if ( SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>( anAct ) )
1163 anActor->SetControlsPrecision( theVal );
1164 anActor->SetControlMode( anActor->GetControlMode() );
1170 } // end of namespace SMESH