smesh/src/SMESHGUI/SMESHGUI_VTKUtils.cxx

1144 lines
36 KiB
C++
Raw Normal View History

2009-02-17 10:27:49 +05:00
// 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
2009-02-17 10:27:49 +05:00
//
// SMESH SMESHGUI : GUI for SMESH component
// File : SMESHGUI_VTKUtils.cxx
// Author : Open CASCADE S.A.S.
// SMESH includes
//
2004-12-01 15:48:31 +05:00
#include "SMESHGUI_VTKUtils.h"
2009-02-17 10:27:49 +05:00
#include "SMESHGUI.h"
2004-12-01 15:48:31 +05:00
#include "SMESHGUI_Utils.h"
2005-02-02 18:54:31 +05:00
#include "SMESHGUI_Filter.h"
2004-12-01 15:48:31 +05:00
2009-02-17 10:27:49 +05:00
#include <SMESH_Actor.h>
#include <SMESH_ActorUtils.h>
#include <SMESH_ObjectDef.h>
#include <SMDS_Mesh.hxx>
2004-12-01 15:48:31 +05:00
2009-02-17 10:27:49 +05:00
// SALOME GUI includes
#include <SUIT_Desktop.h>
#include <SUIT_Session.h>
#include <SUIT_MessageBox.h>
2009-02-17 10:27:49 +05:00
#include <SUIT_ViewManager.h>
#include <SUIT_ResourceMgr.h>
2004-12-01 15:48:31 +05:00
#include <SALOME_ListIO.hxx>
#include <SALOME_ListIteratorOfListIO.hxx>
2004-12-01 15:48:31 +05:00
#include <SVTK_Selector.h>
#include <SVTK_ViewModel.h>
#include <SVTK_ViewWindow.h>
2004-12-01 15:48:31 +05:00
#include <LightApp_SelectionMgr.h>
#include <SalomeApp_Application.h>
#include <SalomeApp_Study.h>
2004-12-01 15:48:31 +05:00
2009-02-17 10:27:49 +05:00
// SALOME KERNEL includes
#include <utilities.h>
2009-02-17 10:27:49 +05:00
// IDL includes
#include <SALOMEconfig.h>
2004-12-01 15:48:31 +05:00
#include CORBA_CLIENT_HEADER(SMESH_Mesh)
#include CORBA_CLIENT_HEADER(SMESH_Group)
2009-02-17 10:27:49 +05:00
// VTK includes
#include <vtkRenderer.h>
#include <vtkActorCollection.h>
#include <vtkUnstructuredGrid.h>
2009-02-17 10:27:49 +05:00
// OCCT includes
#include <TColStd_IndexedMapOfInteger.hxx>
#include <Standard_ErrorHandler.hxx>
2009-02-17 10:27:49 +05:00
namespace SMESH
{
typedef std::map<TKeyOfVisualObj,TVisualObjPtr> TVisualObjCont;
2004-12-01 15:48:31 +05:00
static TVisualObjCont VISUAL_OBJ_CONT;
//=============================================================================
/*!
* \brief Allocate some memory at construction and release it at destruction.
* Is used to be able to continue working after mesh generation or visualization
* break due to lack of memory
*/
//=============================================================================
struct MemoryReserve
{
char* myBuf;
MemoryReserve(): myBuf( new char[1024*1024*1] ){} // 1M
void Free() { if (myBuf) { delete [] myBuf; myBuf = 0; }}
~MemoryReserve() { Free(); }
};
static MemoryReserve* theVISU_MemoryReserve = new MemoryReserve;
//================================================================================
/*!
* \brief Remove VisualObj and its actor from all views
*/
//================================================================================
void RemoveVisualObjectWithActors( const char* theEntry )
{
SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>
( SUIT_Session::session()->activeApplication() );
SUIT_ViewManager* aViewManager =
app ? app->getViewManager(SVTK_Viewer::Type(), true) : 0;
if ( aViewManager ) {
2009-02-17 10:27:49 +05:00
QVector<SUIT_ViewWindow*> views = aViewManager->getViews();
for ( int iV = 0; iV < views.count(); ++iV ) {
if ( SMESH_Actor* actor = FindActorByEntry( views[iV], theEntry)) {
if(SVTK_ViewWindow* vtkWnd = GetVtkViewWindow(views[iV]))
vtkWnd->RemoveActor(actor);
actor->Delete();
}
}
int aStudyId = aViewManager->study()->id();
TVisualObjCont::key_type aKey(aStudyId,theEntry);
TVisualObjCont::iterator anIter = VISUAL_OBJ_CONT.find(aKey);
if(anIter != VISUAL_OBJ_CONT.end()) {
// for unknown reason, object destructor is not called, so clear object manually
anIter->second->GetUnstructuredGrid()->SetCells(0,0,0);
anIter->second->GetUnstructuredGrid()->SetPoints(0);
}
VISUAL_OBJ_CONT.erase(aKey);
}
}
//================================================================================
/*!
* \brief Remove all VisualObjs and their actors from all views
*/
//================================================================================
void RemoveAllObjectsWithActors()
{
SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>
( SUIT_Session::session()->activeApplication() );
if (!app) return;
ViewManagerList viewMgrs = app->viewManagers();
for ( int iM = 0; iM < viewMgrs.count(); ++iM ) {
SUIT_ViewManager* aViewManager = viewMgrs.at( iM );
if ( aViewManager && aViewManager->getType() == SVTK_Viewer::Type()) {
2009-02-17 10:27:49 +05:00
QVector<SUIT_ViewWindow*> views = aViewManager->getViews();
for ( int iV = 0; iV < views.count(); ++iV ) {
if(SVTK_ViewWindow* vtkWnd = GetVtkViewWindow(views[iV])) {
vtkRenderer *aRenderer = vtkWnd->getRenderer();
vtkActorCollection *actors = aRenderer->GetActors();
for (int i = 0; i < actors->GetNumberOfItems(); ++i ) {
// size of actors changes inside the loop
while (SMESH_Actor *actor = dynamic_cast<SMESH_Actor*>(actors->GetItemAsObject(i)))
{
vtkWnd->RemoveActor(actor);
actor->Delete();
}
}
}
}
}
}
TVisualObjCont::iterator anIter = VISUAL_OBJ_CONT.begin();
for ( ; anIter != VISUAL_OBJ_CONT.end(); ++anIter ) {
// for unknown reason, object destructor is not called, so clear object manually
anIter->second->GetUnstructuredGrid()->SetCells(0,0,0);
anIter->second->GetUnstructuredGrid()->SetPoints(0);
}
VISUAL_OBJ_CONT.clear();
}
//================================================================================
/*!
* \brief Remove all VisualObjs of a study
*/
//================================================================================
void RemoveVisuData(int studyID)
{
SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>
( SUIT_Session::session()->activeApplication() );
if (!app) return;
ViewManagerList viewMgrs = app->viewManagers();
for ( int iM = 0; iM < viewMgrs.count(); ++iM ) {
SUIT_ViewManager* aViewManager = viewMgrs.at( iM );
if ( aViewManager && aViewManager->getType() == SVTK_Viewer::Type() &&
aViewManager->study()->id() == studyID ) {
2009-02-17 10:27:49 +05:00
QVector<SUIT_ViewWindow*> views = aViewManager->getViews();
for ( int iV = 0; iV < views.count(); ++iV ) {
if(SVTK_ViewWindow* vtkWnd = GetVtkViewWindow(views[iV])) {
vtkRenderer *aRenderer = vtkWnd->getRenderer();
vtkActorCollection *actors = aRenderer->GetActors();
for (int i = 0; i < actors->GetNumberOfItems(); ++i ) {
// size of actors changes inside the loop
while(SMESH_Actor *actor = dynamic_cast<SMESH_Actor*>(actors->GetItemAsObject(i)))
{
vtkWnd->RemoveActor(actor);
actor->Delete();
}
}
}
}
}
}
TVisualObjCont::iterator anIter = VISUAL_OBJ_CONT.begin();
for ( ; anIter != VISUAL_OBJ_CONT.end(); ++anIter ) {
int curId = anIter->first.first;
if ( curId == studyID ) {
// for unknown reason, object destructor is not called, so clear object manually
anIter->second->GetUnstructuredGrid()->SetCells(0,0,0);
anIter->second->GetUnstructuredGrid()->SetPoints(0);
VISUAL_OBJ_CONT.erase( anIter-- ); // dercement occures before erase()
}
}
}
//================================================================================
/*!
* \brief Notify the user on problems during visualization
*/
//================================================================================
void OnVisuException()
{
try {
#if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
OCC_CATCH_SIGNALS;
#endif
// PAL16774 (Crash after display of many groups). Salome sometimes crashes just
// after or at showing this message, so we do an additional check of available memory
// char* buf = new char[100*1024];
// delete [] buf;
2009-02-17 10:27:49 +05:00
SUIT_MessageBox::warning(SMESHGUI::desktop(), QObject::tr("SMESH_WRN_WARNING"),
QObject::tr("SMESH_VISU_PROBLEM"));
} catch (...) {
// no more memory at all: last resort
2009-02-17 10:27:49 +05:00
MESSAGE_BEGIN ( "SMESHGUI_VTKUtils::OnVisuException(), exception even at showing a message!!!" <<
std::endl << "Try to remove all visual data..." );
if (theVISU_MemoryReserve) {
delete theVISU_MemoryReserve;
theVISU_MemoryReserve = 0;
}
RemoveAllObjectsWithActors();
2009-02-17 10:27:49 +05:00
SUIT_MessageBox::warning(SMESHGUI::desktop(), QObject::tr("SMESH_WRN_WARNING"),
QObject::tr("SMESH_VISU_PROBLEM_CLEAR"));
MESSAGE_END ( "...done" );
}
}
//================================================================================
/*!
* \brief Returns an updated visual object
*/
//================================================================================
2004-12-01 15:48:31 +05:00
TVisualObjPtr GetVisualObj(int theStudyId, const char* theEntry){
TVisualObjPtr aVisualObj;
TVisualObjCont::key_type aKey(theStudyId,theEntry);
2004-12-01 15:48:31 +05:00
try{
#if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
OCC_CATCH_SIGNALS;
#endif
2004-12-01 15:48:31 +05:00
TVisualObjCont::iterator anIter = VISUAL_OBJ_CONT.find(aKey);
if(anIter != VISUAL_OBJ_CONT.end()){
aVisualObj = anIter->second;
}else{
SalomeApp_Application* app =
dynamic_cast<SalomeApp_Application*>( SMESHGUI::activeStudy()->application() );
_PTR(Study) aStudy = SMESHGUI::activeStudy()->studyDS();
_PTR(SObject) aSObj = aStudy->FindObjectID(theEntry);
if(aSObj){
_PTR(GenericAttribute) anAttr;
2004-12-01 15:48:31 +05:00
if(aSObj->FindAttribute(anAttr,"AttributeIOR")){
_PTR(AttributeIOR) anIOR = anAttr;
CORBA::String_var aVal = anIOR->Value().c_str();
CORBA::Object_var anObj = app->orb()->string_to_object( aVal.in() );
2004-12-01 15:48:31 +05:00
if(!CORBA::is_nil(anObj)){
//Try narrow to SMESH_Mesh interface
2004-12-01 15:48:31 +05:00
SMESH::SMESH_Mesh_var aMesh = SMESH::SMESH_Mesh::_narrow(anObj);
if(!aMesh->_is_nil()){
aVisualObj.reset(new SMESH_MeshObj(aMesh));
2005-08-11 14:04:31 +06:00
TVisualObjCont::value_type aValue(aKey,aVisualObj);
2004-12-01 15:48:31 +05:00
VISUAL_OBJ_CONT.insert(aValue);
}
//Try narrow to SMESH_Group interface
2004-12-01 15:48:31 +05:00
SMESH::SMESH_GroupBase_var aGroup = SMESH::SMESH_GroupBase::_narrow(anObj);
if(!aGroup->_is_nil()){
_PTR(SObject) aFatherSObj = aSObj->GetFather();
if(!aFatherSObj) return aVisualObj;
2004-12-01 15:48:31 +05:00
aFatherSObj = aFatherSObj->GetFather();
if(!aFatherSObj) return aVisualObj;
CORBA::String_var anEntry = aFatherSObj->GetID().c_str();
2004-12-01 15:48:31 +05:00
TVisualObjPtr aVisObj = GetVisualObj(theStudyId,anEntry.in());
if(SMESH_MeshObj* aMeshObj = dynamic_cast<SMESH_MeshObj*>(aVisObj.get())){
aVisualObj.reset(new SMESH_GroupObj(aGroup,aMeshObj));
TVisualObjCont::value_type aValue(aKey,aVisualObj);
2004-12-01 15:48:31 +05:00
VISUAL_OBJ_CONT.insert(aValue);
}
}
//Try narrow to SMESH_subMesh interface
2004-12-01 15:48:31 +05:00
SMESH::SMESH_subMesh_var aSubMesh = SMESH::SMESH_subMesh::_narrow(anObj);
if(!aSubMesh->_is_nil()){
_PTR(SObject) aFatherSObj = aSObj->GetFather();
if(!aFatherSObj) return aVisualObj;
2004-12-01 15:48:31 +05:00
aFatherSObj = aFatherSObj->GetFather();
if(!aFatherSObj) return aVisualObj;
CORBA::String_var anEntry = aFatherSObj->GetID().c_str();
2004-12-01 15:48:31 +05:00
TVisualObjPtr aVisObj = GetVisualObj(theStudyId,anEntry.in());
if(SMESH_MeshObj* aMeshObj = dynamic_cast<SMESH_MeshObj*>(aVisObj.get())){
aVisualObj.reset(new SMESH_subMeshObj(aSubMesh,aMeshObj));
TVisualObjCont::value_type aValue(aKey,aVisualObj);
2004-12-01 15:48:31 +05:00
VISUAL_OBJ_CONT.insert(aValue);
}
}
}
}
}
}
}catch(...){
INFOS("GetMeshObj - There is no SMESH_Mesh object for the SALOMEDS::Strudy and Entry!!!");
return TVisualObjPtr();
2004-12-01 15:48:31 +05:00
}
// Update object
bool objModified = false;
if ( aVisualObj ) {
try {
#if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
OCC_CATCH_SIGNALS;
#endif
objModified = aVisualObj->Update();
}
catch (...) {
#ifdef _DEBUG_
2009-02-17 10:27:49 +05:00
MESSAGE ( "Exception in SMESHGUI_VTKUtils::GetVisualObj()" );
#endif
RemoveVisualObjectWithActors( theEntry ); // remove this object
OnVisuException();
aVisualObj.reset();
}
}
if ( objModified ) {
// PAL16631. Mesurements showed that to show aVisualObj in SHADING(default) mode,
// ~10 times more memory is used than it occupies.
// Warn the user if there is less free memory than 30 sizes of a grid
// TODO: estimate memory usage in other modes and take current mode into account
int freeMB = SMDS_Mesh::CheckMemory(true);
int usedMB = aVisualObj->GetUnstructuredGrid()->GetActualMemorySize() / 1024;
if ( freeMB > 0 && usedMB * 30 > freeMB ) {
#ifdef _DEBUG_
2009-02-17 10:27:49 +05:00
MESSAGE ( "SMESHGUI_VTKUtils::GetVisualObj(), freeMB=" << freeMB
<< ", usedMB=" << usedMB );
#endif
2009-02-17 10:27:49 +05:00
bool continu = false;
if ( usedMB * 10 > freeMB )
// even dont try to show
2009-02-17 10:27:49 +05:00
SUIT_MessageBox::warning(SMESHGUI::desktop(), QObject::tr("SMESH_WRN_WARNING"),
QObject::tr("SMESH_NO_MESH_VISUALIZATION"));
else
// there is a chance to succeed
2009-02-17 10:27:49 +05:00
continu = SUIT_MessageBox::warning
(SMESHGUI::desktop(),
QObject::tr("SMESH_WRN_WARNING"),
QObject::tr("SMESH_CONTINUE_MESH_VISUALIZATION"),
2009-02-17 10:27:49 +05:00
SUIT_MessageBox::Yes | SUIT_MessageBox::No,
SUIT_MessageBox::Yes ) == SUIT_MessageBox::Yes;
if ( !continu ) {
// remove the corresponding actors from all views
RemoveVisualObjectWithActors( theEntry );
aVisualObj.reset();
}
}
}
2004-12-01 15:48:31 +05:00
return aVisualObj;
}
/*! Return active view window, if it instantiates SVTK_ViewWindow class,
* overwise find or create corresponding view window, make it active and return it.
* \note Active VVTK_ViewWindow can be returned, because it inherits SVTK_ViewWindow.
*/
SVTK_ViewWindow* GetViewWindow (const SalomeApp_Module* theModule,
bool createIfNotFound)
2004-12-01 15:48:31 +05:00
{
SalomeApp_Application* anApp;
if (theModule)
anApp = theModule->getApp();
else
anApp = dynamic_cast<SalomeApp_Application*>
(SUIT_Session::session()->activeApplication());
if (anApp) {
if (SVTK_ViewWindow* aView = dynamic_cast<SVTK_ViewWindow*>(anApp->desktop()->activeWindow()))
return aView;
SUIT_ViewManager* aViewManager =
anApp->getViewManager(SVTK_Viewer::Type(), createIfNotFound);
if (aViewManager) {
if (SUIT_ViewWindow* aViewWindow = aViewManager->getActiveView()) {
if (SVTK_ViewWindow* aView = dynamic_cast<SVTK_ViewWindow*>(aViewWindow)) {
aViewWindow->raise();
aViewWindow->setFocus();
return aView;
}
}
}
}
2005-08-11 14:04:31 +06:00
return NULL;
2004-12-01 15:48:31 +05:00
}
SVTK_ViewWindow* FindVtkViewWindow (SUIT_ViewManager* theMgr,
SUIT_ViewWindow * theWindow)
{
if( !theMgr )
return NULL;
2004-12-01 15:48:31 +05:00
2009-02-17 10:27:49 +05:00
QVector<SUIT_ViewWindow*> views = theMgr->getViews();
if( views.contains( theWindow ) )
return GetVtkViewWindow( theWindow );
else
return NULL;
2004-12-01 15:48:31 +05:00
}
SVTK_ViewWindow* GetVtkViewWindow(SUIT_ViewWindow* theWindow){
return dynamic_cast<SVTK_ViewWindow*>(theWindow);
2004-12-01 15:48:31 +05:00
}
/* SUIT_ViewWindow* GetActiveWindow()
2005-01-20 11:25:54 +05:00
{
SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( SUIT_Session::session()->activeApplication() );
if( !app )
return NULL;
SUIT_ViewManager* mgr = app->activeViewManager();
if( mgr )
return mgr->getActiveView();
else
return NULL;
}*/
2005-01-20 11:25:54 +05:00
SVTK_ViewWindow* GetCurrentVtkView(){
return GetVtkViewWindow( GetActiveWindow() );
}
2005-01-20 11:25:54 +05:00
void RepaintCurrentView()
{
if (SVTK_ViewWindow* wnd = GetCurrentVtkView())
{
try {
#if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
OCC_CATCH_SIGNALS;
#endif
wnd->getRenderer()->Render();
wnd->Repaint(false);
}
catch (...) {
#ifdef _DEBUG_
2009-02-17 10:27:49 +05:00
MESSAGE ( "Exception in SMESHGUI_VTKUtils::RepaintCurrentView()" );
#endif
OnVisuException();
}
}
}
void RepaintViewWindow(SVTK_ViewWindow* theWindow)
2005-01-20 11:25:54 +05:00
{
try {
#if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
OCC_CATCH_SIGNALS;
#endif
theWindow->getRenderer()->Render();
theWindow->Repaint();
}
catch (...) {
#ifdef _DEBUG_
2009-02-17 10:27:49 +05:00
MESSAGE ( "Exception in SMESHGUI_VTKUtils::RepaintViewWindow(SVTK_ViewWindow*)" );
#endif
OnVisuException();
}
2005-01-20 11:25:54 +05:00
}
void RenderViewWindow(SVTK_ViewWindow* theWindow)
{
try {
#if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
OCC_CATCH_SIGNALS;
#endif
theWindow->getRenderer()->Render();
theWindow->Repaint();
}
catch (...) {
#ifdef _DEBUG_
2009-02-17 10:27:49 +05:00
MESSAGE ( "Exception in SMESHGUI_VTKUtils::RenderViewWindow(SVTK_ViewWindow*)" );
#endif
OnVisuException();
}
}
void FitAll(){
if(SVTK_ViewWindow* wnd = GetCurrentVtkView() ){
try {
#if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
OCC_CATCH_SIGNALS;
#endif
wnd->onFitAll();
wnd->Repaint();
}
catch (...) {
#ifdef _DEBUG_
2009-02-17 10:27:49 +05:00
MESSAGE ( "Exception in SMESHGUI_VTKUtils::FitAll()" );
#endif
OnVisuException();
}
}
}
2005-01-20 11:25:54 +05:00
SMESH_Actor* FindActorByEntry(SUIT_ViewWindow *theWindow,
2004-12-01 15:48:31 +05:00
const char* theEntry)
{
if(SVTK_ViewWindow* aViewWindow = GetVtkViewWindow(theWindow)){
vtkRenderer *aRenderer = aViewWindow->getRenderer();
2004-12-01 15:48:31 +05:00
vtkActorCollection *aCollection = aRenderer->GetActors();
aCollection->InitTraversal();
while(vtkActor *anAct = aCollection->GetNextActor()){
if(SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)){
if(anActor->hasIO()){
Handle(SALOME_InteractiveObject) anIO = anActor->getIO();
if(anIO->hasEntry() && strcmp(anIO->getEntry(),theEntry) == 0){
return anActor;
}
}
}
}
}
return NULL;
}
2004-12-01 15:48:31 +05:00
SMESH_Actor* FindActorByEntry(const char* theEntry){
return FindActorByEntry(GetActiveWindow(),theEntry);
2004-12-01 15:48:31 +05:00
}
2004-12-01 15:48:31 +05:00
SMESH_Actor* FindActorByObject(CORBA::Object_ptr theObject){
SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( SUIT_Session::session()->activeApplication() );
if( !app )
return NULL;
2004-12-01 15:48:31 +05:00
if(!CORBA::is_nil(theObject)){
_PTR(Study) aStudy = GetActiveStudyDocument();
CORBA::String_var anIOR = app->orb()->object_to_string( theObject );
_PTR(SObject) aSObject = aStudy->FindObjectIOR(anIOR.in());
if(aSObject){
CORBA::String_var anEntry = aSObject->GetID().c_str();
2004-12-01 15:48:31 +05:00
return FindActorByEntry(anEntry.in());
}
}
return NULL;
}
SMESH_Actor* CreateActor(_PTR(Study) theStudy,
2004-12-01 15:48:31 +05:00
const char* theEntry,
int theIsClear)
{
SMESH_Actor *anActor = NULL;
CORBA::Long anId = theStudy->StudyId();
if(TVisualObjPtr aVisualObj = GetVisualObj(anId,theEntry)){
_PTR(SObject) aSObj = theStudy->FindObjectID(theEntry);
if(aSObj){
_PTR(GenericAttribute) anAttr;
2004-12-01 15:48:31 +05:00
if(aSObj->FindAttribute(anAttr,"AttributeName")){
_PTR(AttributeName) aName = anAttr;
std::string aNameVal = aName->Value();
2004-12-01 15:48:31 +05:00
anActor = SMESH_Actor::New(aVisualObj,theEntry,aNameVal.c_str(),theIsClear);
}
SMESH::SMESH_GroupBase_var aGroup = SMESH::SMESH_GroupBase::_narrow( SMESH::SObjectToObject( aSObj ));
if(!CORBA::is_nil(aGroup))
{
SALOMEDS::Color aColor = aGroup->GetColor();
if( !( aColor.R > 0 || aColor.G > 0 || aColor.B > 0 ) )
{
int r = 0, g = 0, b = 0;
SMESH::GetColor( "SMESH", "fill_color", r, g, b, QColor( 0, 170, 255 ) );
aColor.R = (float)r / 255.0;
aColor.G = (float)g / 255.0;
aColor.B = (float)b / 255.0;
aGroup->SetColor( aColor );
}
if( aGroup->GetType() == SMESH::NODE )
anActor->SetNodeColor( aColor.R, aColor.G, aColor.B );
else if( aGroup->GetType() == SMESH::EDGE )
anActor->SetEdgeColor( aColor.R, aColor.G, aColor.B );
else
anActor->SetSufaceColor( aColor.R, aColor.G, aColor.B );
}
2004-12-01 15:48:31 +05:00
}
}
return anActor;
}
void DisplayActor( SUIT_ViewWindow *theWnd, SMESH_Actor* theActor){
if(SVTK_ViewWindow* vtkWnd = GetVtkViewWindow(theWnd)){
try {
#if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
OCC_CATCH_SIGNALS;
#endif
vtkWnd->AddActor(theActor);
vtkWnd->Repaint();
}
catch (...) {
#ifdef _DEBUG_
2009-02-17 10:27:49 +05:00
MESSAGE ( "Exception in SMESHGUI_VTKUtils::DisplayActor()" );
#endif
OnVisuException();
}
2004-12-01 15:48:31 +05:00
}
}
void RemoveActor( SUIT_ViewWindow *theWnd, SMESH_Actor* theActor){
if(SVTK_ViewWindow* vtkWnd = GetVtkViewWindow(theWnd)){
vtkWnd->RemoveActor(theActor);
if(theActor->hasIO()){
Handle(SALOME_InteractiveObject) anIO = theActor->getIO();
if(anIO->hasEntry()){
std::string anEntry = anIO->getEntry();
SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( vtkWnd->getViewManager()->study() );
int aStudyId = aStudy->id();
TVisualObjCont::key_type aKey(aStudyId,anEntry);
VISUAL_OBJ_CONT.erase(aKey);
}
}
theActor->Delete();
vtkWnd->Repaint();
2004-12-01 15:48:31 +05:00
}
}
//================================================================================
/*!
* \brief Return true if there are no SMESH actors in a view
*/
//================================================================================
2004-12-01 15:48:31 +05:00
bool noSmeshActors(SUIT_ViewWindow *theWnd)
{
if(SVTK_ViewWindow* aViewWindow = GetVtkViewWindow(theWnd)) {
vtkRenderer *aRenderer = aViewWindow->getRenderer();
vtkActorCollection *aCollection = aRenderer->GetActors();
aCollection->InitTraversal();
while(vtkActor *anAct = aCollection->GetNextActor())
2009-02-17 10:27:49 +05:00
if(dynamic_cast<SMESH_Actor*>(anAct))
return false;
2004-12-01 15:48:31 +05:00
}
return true;
2004-12-01 15:48:31 +05:00
}
bool UpdateView(SUIT_ViewWindow *theWnd, EDisplaing theAction, const char* theEntry)
2004-12-01 15:48:31 +05:00
{
bool OK = false;
SVTK_ViewWindow* aViewWnd = GetVtkViewWindow(theWnd);
if (!aViewWnd)
return OK;
{
OK = true;
vtkRenderer *aRenderer = aViewWnd->getRenderer();
2004-12-01 15:48:31 +05:00
vtkActorCollection *aCollection = aRenderer->GetActors();
aCollection->InitTraversal();
switch (theAction) {
2004-12-01 15:48:31 +05:00
case eDisplayAll: {
while (vtkActor *anAct = aCollection->GetNextActor()) {
if (SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)) {
2004-12-01 15:48:31 +05:00
anActor->SetVisibility(true);
}
}
break;
}
case eDisplayOnly:
case eEraseAll: {
while (vtkActor *anAct = aCollection->GetNextActor()) {
if (SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)) {
2004-12-01 15:48:31 +05:00
anActor->SetVisibility(false);
}
}
}
default: {
if (SMESH_Actor *anActor = FindActorByEntry(theWnd,theEntry)) {
switch (theAction) {
2004-12-01 15:48:31 +05:00
case eDisplay:
case eDisplayOnly:
anActor->SetVisibility(true);
if (theAction == eDisplayOnly) aRenderer->ResetCameraClippingRange();
2004-12-01 15:48:31 +05:00
break;
case eErase:
anActor->SetVisibility(false);
break;
}
} else {
switch (theAction) {
2004-12-01 15:48:31 +05:00
case eDisplay:
case eDisplayOnly:
{
SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>(theWnd->getViewManager()->study());
_PTR(Study) aDocument = aStudy->studyDS();
// Pass non-visual objects (hypotheses, etc.), return true in this case
CORBA::Long anId = aDocument->StudyId();
if (TVisualObjPtr aVisualObj = GetVisualObj(anId,theEntry))
{
if ((anActor = CreateActor(aDocument,theEntry,true))) {
bool needFitAll = noSmeshActors(theWnd); // fit for the first object only
DisplayActor(theWnd,anActor);
// FitAll(); - PAL16770(Display of a group performs an automatic fit all)
if (needFitAll) FitAll();
} else {
OK = false;
}
}
break;
}
2004-12-01 15:48:31 +05:00
}
}
}
}
}
return OK;
2004-12-01 15:48:31 +05:00
}
bool UpdateView(EDisplaing theAction, const char* theEntry){
SalomeApp_Study* aStudy = dynamic_cast< SalomeApp_Study* >( GetActiveStudy() );
SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( aStudy->application() );
SUIT_ViewWindow *aWnd = app->activeViewManager()->getActiveView();
return UpdateView(aWnd,theAction,theEntry);
2004-12-01 15:48:31 +05:00
}
2004-12-01 15:48:31 +05:00
void UpdateView(){
if(SVTK_ViewWindow* aWnd = SMESH::GetCurrentVtkView()){
LightApp_SelectionMgr* mgr = SMESHGUI::selectionMgr();
SALOME_ListIO selected; mgr->selectedObjects( selected );
if( selected.Extent() == 0){
vtkRenderer* aRenderer = aWnd->getRenderer();
2004-12-01 15:48:31 +05:00
vtkActorCollection *aCollection = aRenderer->GetActors();
aCollection->InitTraversal();
while(vtkActor *anAct = aCollection->GetNextActor()){
if(SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)){
if(anActor->hasIO())
if (!Update(anActor->getIO(),anActor->GetVisibility()))
break; // avoid multiple warinings if visu failed
2004-12-01 15:48:31 +05:00
}
}
}else{
SALOME_ListIteratorOfListIO anIter( selected );
2009-02-17 10:27:49 +05:00
for( ; anIter.More(); anIter.Next()){
2004-12-01 15:48:31 +05:00
Handle(SALOME_InteractiveObject) anIO = anIter.Value();
if ( !Update(anIO,true) )
break; // avoid multiple warinings if visu failed
2004-12-01 15:48:31 +05:00
}
}
RepaintCurrentView();
}
}
bool Update(const Handle(SALOME_InteractiveObject)& theIO, bool theDisplay)
2004-12-01 15:48:31 +05:00
{
_PTR(Study) aStudy = GetActiveStudyDocument();
2004-12-01 15:48:31 +05:00
CORBA::Long anId = aStudy->StudyId();
if ( TVisualObjPtr aVisualObj = SMESH::GetVisualObj(anId,theIO->getEntry())) {
if ( theDisplay )
UpdateView(SMESH::eDisplay,theIO->getEntry());
return true;
}
return false;
2004-12-01 15:48:31 +05:00
}
2005-06-14 12:25:33 +06:00
void UpdateSelectionProp( SMESHGUI* theModule ) {
if( !theModule )
return;
SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( theModule->application() );
if( !app )
{
MESSAGE( "UpdateSelectionProp: Application is null" );
return;
}
SUIT_ViewManager* vm = app->activeViewManager();
2005-06-14 12:25:33 +06:00
if( !vm )
{
MESSAGE( "UpdateSelectionProp: View manager is null" );
return;
}
2009-02-17 10:27:49 +05:00
QVector<SUIT_ViewWindow*> views = vm->getViews();
2005-06-14 12:25:33 +06:00
SUIT_ResourceMgr* mgr = SMESH::GetResourceMgr( theModule );
if( !mgr )
{
MESSAGE( "UpdateSelectionProp: Resource manager is null" );
return;
}
2005-06-27 17:29:58 +06:00
QColor aHiColor = mgr->colorValue( "SMESH", "selection_object_color", Qt::white ),
aSelColor = mgr->colorValue( "SMESH", "selection_element_color", Qt::yellow ),
aPreColor = mgr->colorValue( "SMESH", "highlight_color", Qt::cyan );
2005-06-27 17:29:58 +06:00
int SW = mgr->integerValue( "SMESH", "selection_width", 5 ),
PW = mgr->integerValue( "SMESH", "highlight_width", 5 );
2005-06-27 17:29:58 +06:00
double SP1 = mgr->doubleValue( "SMESH", "selection_precision_node", 0.025 ),
SP2 = mgr->doubleValue( "SMESH", "selection_precision_element", 0.001 ),
SP3 = mgr->doubleValue( "SMESH", "selection_precision_object", 0.025 );
for ( int i=0, n=views.count(); i<n; i++ ){
// update VTK viewer properties
if(SVTK_ViewWindow* aVtkView = GetVtkViewWindow( views[i] )){
// mesh element selection
aVtkView->SetSelectionProp(aSelColor.red()/255.,
aSelColor.green()/255.,
aSelColor.blue()/255.,
SW );
// tolerances
aVtkView->SetSelectionTolerance(SP1, SP2, SP3);
// pre-selection
aVtkView->SetPreselectionProp(aPreColor.red()/255.,
aPreColor.green()/255.,
aPreColor.blue()/255.,
PW);
2004-12-01 15:48:31 +05:00
// update actors
vtkRenderer* aRenderer = aVtkView->getRenderer();
2004-12-01 15:48:31 +05:00
vtkActorCollection *aCollection = aRenderer->GetActors();
aCollection->InitTraversal();
while(vtkActor *anAct = aCollection->GetNextActor()){
if(SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)){
anActor->SetHighlightColor(aHiColor.red()/255.,
aHiColor.green()/255.,
2004-12-01 15:48:31 +05:00
aHiColor.blue()/255.);
anActor->SetPreHighlightColor(aPreColor.red()/255.,
aPreColor.green()/255.,
2004-12-01 15:48:31 +05:00
aPreColor.blue()/255.);
}
}
}
2004-12-01 15:48:31 +05:00
}
}
2004-12-01 15:48:31 +05:00
//----------------------------------------------------------------------------
SVTK_Selector*
GetSelector(SUIT_ViewWindow *theWindow)
{
if(SVTK_ViewWindow* aWnd = GetVtkViewWindow(theWindow))
return aWnd->GetSelector();
2004-12-01 15:48:31 +05:00
return NULL;
}
void SetFilter(const Handle(VTKViewer_Filter)& theFilter,
SVTK_Selector* theSelector)
2004-12-01 15:48:31 +05:00
{
if (theSelector)
theSelector->SetFilter(theFilter);
2004-12-01 15:48:31 +05:00
}
Handle(VTKViewer_Filter) GetFilter(int theId, SVTK_Selector* theSelector)
2004-12-01 15:48:31 +05:00
{
return theSelector->GetFilter(theId);
2004-12-01 15:48:31 +05:00
}
bool IsFilterPresent(int theId, SVTK_Selector* theSelector)
2004-12-01 15:48:31 +05:00
{
return theSelector->IsFilterPresent(theId);
2004-12-01 15:48:31 +05:00
}
void RemoveFilter(int theId, SVTK_Selector* theSelector)
{
theSelector->RemoveFilter(theId);
2004-12-01 15:48:31 +05:00
}
void RemoveFilters(SVTK_Selector* theSelector)
{
2009-02-17 10:27:49 +05:00
for ( int id = SMESH::NodeFilter; theSelector && id < SMESH::LastFilter; id++ )
theSelector->RemoveFilter( id );
2005-02-02 18:51:36 +05:00
}
2004-12-01 15:48:31 +05:00
bool IsValid(SALOME_Actor* theActor, int theCellId,
SVTK_Selector* theSelector)
2004-12-01 15:48:31 +05:00
{
return theSelector->IsValid(theActor,theCellId);
2004-12-01 15:48:31 +05:00
}
//----------------------------------------------------------------------------
void SetPointRepresentation(bool theIsVisible){
if(SVTK_ViewWindow* aViewWindow = GetCurrentVtkView()){
vtkRenderer *aRenderer = aViewWindow->getRenderer();
2004-12-01 15:48:31 +05:00
vtkActorCollection *aCollection = aRenderer->GetActors();
aCollection->InitTraversal();
while(vtkActor *anAct = aCollection->GetNextActor()){
if(SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)){
if(anActor->GetVisibility()){
anActor->SetPointRepresentation(theIsVisible);
}
}
}
RepaintCurrentView();
}
}
void SetPickable(SMESH_Actor* theActor){
if(SVTK_ViewWindow* aWnd = GetCurrentVtkView()){
2004-12-01 15:48:31 +05:00
int anIsAllPickable = (theActor == NULL);
vtkRenderer *aRenderer = aWnd->getRenderer();
2004-12-01 15:48:31 +05:00
vtkActorCollection *aCollection = aRenderer->GetActors();
aCollection->InitTraversal();
while(vtkActor *anAct = aCollection->GetNextActor()){
if(SALOME_Actor *anActor = dynamic_cast<SALOME_Actor*>(anAct)){
if(anActor->GetVisibility()){
anActor->SetPickable(anIsAllPickable);
}
}
}
if(theActor)
theActor->SetPickable(!anIsAllPickable);
RepaintCurrentView();
}
}
//----------------------------------------------------------------------------
2005-08-11 14:04:31 +06:00
int GetNameOfSelectedNodes(SVTK_Selector* theSelector,
const Handle(SALOME_InteractiveObject)& theIO,
2004-12-01 15:48:31 +05:00
QString& theName)
{
theName = "";
TColStd_IndexedMapOfInteger aMapIndex;
theSelector->GetIndex(theIO,aMapIndex);
for(int i = 1; i <= aMapIndex.Extent(); i++)
theName += QString(" %1").arg(aMapIndex(i));
return aMapIndex.Extent();
}
2005-08-11 14:04:31 +06:00
int GetNameOfSelectedElements(SVTK_Selector* theSelector,
const Handle(SALOME_InteractiveObject)& theIO,
QString& theName)
{
theName = "";
TColStd_IndexedMapOfInteger aMapIndex;
theSelector->GetIndex(theIO,aMapIndex);
typedef std::set<int> TIdContainer;
TIdContainer anIdContainer;
for( int i = 1; i <= aMapIndex.Extent(); i++)
anIdContainer.insert(aMapIndex(i));
TIdContainer::const_iterator anIter = anIdContainer.begin();
2009-02-17 10:27:49 +05:00
for( ; anIter != anIdContainer.end(); anIter++)
theName += QString(" %1").arg(*anIter);
return aMapIndex.Extent();
}
2005-08-11 14:04:31 +06:00
int GetEdgeNodes(SVTK_Selector* theSelector,
const TVisualObjPtr& theVisualObject,
2005-08-11 14:04:31 +06:00
int& theId1,
int& theId2)
{
2005-08-11 14:04:31 +06:00
const SALOME_ListIO& selected = theSelector->StoredIObjects();
if ( selected.Extent() != 1 )
return -1;
Handle(SALOME_InteractiveObject) anIO = selected.First();
if ( anIO.IsNull() || !anIO->hasEntry() )
return -1;
TColStd_IndexedMapOfInteger aMapIndex;
theSelector->GetIndex( anIO, aMapIndex );
if ( aMapIndex.Extent() != 2 )
return -1;
int anObjId = -1, anEdgeNum = -1;
for ( int i = 1; i <= aMapIndex.Extent(); i++ ) {
int aVal = aMapIndex( i );
if ( aVal > 0 )
anObjId = aVal;
else
anEdgeNum = abs( aVal ) - 1;
}
if ( anObjId == -1 || anEdgeNum == -1 )
return -1;
return theVisualObject->GetEdgeNodes( anObjId, anEdgeNum, theId1, theId2 ) ? 1 : -1;
}
//----------------------------------------------------------------------------
int GetNameOfSelectedNodes(LightApp_SelectionMgr *theMgr,
const Handle(SALOME_InteractiveObject)& theIO,
QString& theName)
2004-12-01 15:48:31 +05:00
{
theName = "";
if(theIO->hasEntry()){
2005-01-20 11:25:54 +05:00
if(FindActorByEntry(theIO->getEntry())){
2004-12-01 15:48:31 +05:00
TColStd_IndexedMapOfInteger aMapIndex;
theMgr->GetIndexes(theIO,aMapIndex);
2004-12-01 15:48:31 +05:00
for(int i = 1; i <= aMapIndex.Extent(); i++){
theName += QString(" %1").arg(aMapIndex(i));
}
return aMapIndex.Extent();
}
}
return -1;
}
int GetNameOfSelectedNodes(LightApp_SelectionMgr *theMgr, QString& theName){
2004-12-01 15:48:31 +05:00
theName = "";
SALOME_ListIO selected; theMgr->selectedObjects( selected );
if(selected.Extent() == 1){
Handle(SALOME_InteractiveObject) anIO = selected.First();
return GetNameOfSelectedNodes(theMgr,anIO,theName);
2004-12-01 15:48:31 +05:00
}
return -1;
}
2005-08-11 14:04:31 +06:00
2004-12-01 15:48:31 +05:00
int GetNameOfSelectedElements(LightApp_SelectionMgr *theMgr,
const Handle(SALOME_InteractiveObject)& theIO,
2004-12-01 15:48:31 +05:00
QString& theName)
{
theName = "";
if(theIO->hasEntry()){
2005-01-20 11:25:54 +05:00
if(FindActorByEntry(theIO->getEntry())){
2004-12-01 15:48:31 +05:00
TColStd_IndexedMapOfInteger aMapIndex;
theMgr->GetIndexes(theIO,aMapIndex);
2009-02-17 10:27:49 +05:00
typedef std::set<int> TIdContainer;
2004-12-01 15:48:31 +05:00
TIdContainer anIdContainer;
for( int i = 1; i <= aMapIndex.Extent(); i++)
anIdContainer.insert(aMapIndex(i));
TIdContainer::const_iterator anIter = anIdContainer.begin();
2009-02-17 10:27:49 +05:00
for( ; anIter != anIdContainer.end(); anIter++){
2004-12-01 15:48:31 +05:00
theName += QString(" %1").arg(*anIter);
}
return aMapIndex.Extent();
}
}
return -1;
}
int GetNameOfSelectedElements(LightApp_SelectionMgr *theMgr, QString& theName)
2004-12-01 15:48:31 +05:00
{
theName = "";
SALOME_ListIO selected; theMgr->selectedObjects( selected );
if( selected.Extent() == 1){
Handle(SALOME_InteractiveObject) anIO = selected.First();
return GetNameOfSelectedElements(theMgr,anIO,theName);
2004-12-01 15:48:31 +05:00
}
return -1;
}
int GetSelected(LightApp_SelectionMgr* theMgr,
TColStd_IndexedMapOfInteger& theMap,
2004-12-01 15:48:31 +05:00
const bool theIsElement)
{
theMap.Clear();
SALOME_ListIO selected; theMgr->selectedObjects( selected );
2004-12-01 15:48:31 +05:00
if ( selected.Extent() == 1 )
2004-12-01 15:48:31 +05:00
{
Handle(SALOME_InteractiveObject) anIO = selected.First();
2004-12-01 15:48:31 +05:00
if ( anIO->hasEntry() ) {
theMgr->GetIndexes( anIO, theMap );
2004-12-01 15:48:31 +05:00
}
}
return theMap.Extent();
}
int GetEdgeNodes( LightApp_SelectionMgr* theMgr, int& theId1, int& theId2 )
2004-12-01 15:48:31 +05:00
{
SALOME_ListIO selected; theMgr->selectedObjects( selected );
if ( selected.Extent() != 1 )
2004-12-01 15:48:31 +05:00
return -1;
Handle(SALOME_InteractiveObject) anIO = selected.First();
2004-12-01 15:48:31 +05:00
if ( anIO.IsNull() || !anIO->hasEntry() )
return -1;
SMESH_Actor *anActor = SMESH::FindActorByEntry( anIO->getEntry() );
if ( anActor == 0 )
return -1;
TColStd_IndexedMapOfInteger aMapIndex;
theMgr->GetIndexes( anIO, aMapIndex );
2004-12-01 15:48:31 +05:00
if ( aMapIndex.Extent() != 2 )
return -1;
2004-12-01 15:48:31 +05:00
int anObjId = -1, anEdgeNum = -1;
for ( int i = 1; i <= aMapIndex.Extent(); i++ ) {
int aVal = aMapIndex( i );
if ( aVal > 0 )
anObjId = aVal;
else
anEdgeNum = abs( aVal );
}
2004-12-01 15:48:31 +05:00
if ( anObjId == -1 || anEdgeNum == -1 )
return -1;
2004-12-01 15:48:31 +05:00
return anActor->GetObject()->GetEdgeNodes( anObjId, anEdgeNum, theId1, theId2 ) ? 1 : -1;
}
2004-12-01 15:48:31 +05:00
void SetControlsPrecision( const long theVal )
{
if( SVTK_ViewWindow* aWnd = SMESH::GetCurrentVtkView() )
2004-12-01 15:48:31 +05:00
{
vtkRenderer *aRenderer = aWnd->getRenderer();
2004-12-01 15:48:31 +05:00
vtkActorCollection *aCollection = aRenderer->GetActors();
aCollection->InitTraversal();
2005-08-11 14:04:31 +06:00
2004-12-01 15:48:31 +05:00
while ( vtkActor *anAct = aCollection->GetNextActor())
{
if ( SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>( anAct ) )
{
anActor->SetControlsPrecision( theVal );
anActor->SetControlMode( anActor->GetControlMode() );
}
}
}
}
2009-02-17 10:27:49 +05:00
} // end of namespace SMESH