smesh/src/SMESHGUI/SMESHGUI_VTKUtils.cxx

780 lines
23 KiB
C++
Raw Normal View History

2004-12-01 15:48:31 +05:00
// Copyright (C) 2003 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
//
2004-12-01 15:48:31 +05:00
// See http://www.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org
#include "SMESHGUI_VTKUtils.h"
#include "SMESHGUI_Utils.h"
2005-02-02 18:54:31 +05:00
#include "SMESHGUI_Filter.h"
2004-12-01 15:48:31 +05:00
#include <vtkRenderer.h>
#include <vtkActorCollection.h>
#include <TColStd_IndexedMapOfInteger.hxx>
#include <SUIT_Desktop.h>
#include <SUIT_Session.h>
#include <SUIT_Study.h>
2004-12-01 15:48:31 +05:00
#include "LightApp_SelectionMgr.h"
2004-12-01 15:48:31 +05:00
#include "SVTK_Selector.h"
#include "SVTK_ViewModel.h"
#include "SVTK_ViewWindow.h"
#include "SVTK_RenderWindow.h"
#include "SVTK_InteractorStyle.h"
#include "SVTK_RenderWindowInteractor.h"
2004-12-01 15:48:31 +05:00
#include "utilities.h"
#include "SALOMEconfig.h"
#include CORBA_CLIENT_HEADER(SMESH_Gen)
#include CORBA_CLIENT_HEADER(SMESH_Mesh)
#include CORBA_CLIENT_HEADER(SMESH_Group)
#include CORBA_CLIENT_HEADER(SMESH_Hypothesis)
#include "SMESHGUI.h"
2004-12-01 15:48:31 +05:00
#include "SMESH_Actor.h"
2005-01-20 11:25:54 +05:00
#include "SMESH_ObjectDef.h"
2004-12-01 15:48:31 +05:00
#include <SalomeApp_Application.h>
#include <LightApp_SelectionMgr.h>
#include <SalomeApp_Study.h>
#include <SALOMEDSClient_Study.hxx>
#include <SALOMEDSClient_SObject.hxx>
#include <SALOME_ListIO.hxx>
#include <SALOME_ListIteratorOfListIO.hxx>
#include <set>
2004-12-01 15:48:31 +05:00
using namespace std;
namespace SMESH{
typedef map<TKeyOfVisualObj,TVisualObjPtr> TVisualObjCont;
static TVisualObjCont VISUAL_OBJ_CONT;
TVisualObjPtr GetVisualObj(int theStudyId, const char* theEntry){
TVisualObjPtr aVisualObj;
try{
TVisualObjCont::key_type aKey(theStudyId,theEntry);
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 interafce
SMESH::SMESH_Mesh_var aMesh = SMESH::SMESH_Mesh::_narrow(anObj);
if(!aMesh->_is_nil()){
aVisualObj.reset(new SMESH_MeshObj(aMesh));
aVisualObj->Update();
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);
return aVisualObj;
}
//Try narrow to SMESH_Group interafce
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));
aVisualObj->Update();
TVisualObjCont::value_type aValue(aKey,aVisualObj);
2004-12-01 15:48:31 +05:00
VISUAL_OBJ_CONT.insert(aValue);
return aVisualObj;
}
}
//Try narrow to SMESH_subMesh interafce
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));
aVisualObj->Update();
TVisualObjCont::value_type aValue(aKey,aVisualObj);
2004-12-01 15:48:31 +05:00
VISUAL_OBJ_CONT.insert(aValue);
return aVisualObj;
}
}
}
}
}
}
}catch(...){
INFOS("GetMeshObj - There is no SMESH_Mesh object for the SALOMEDS::Strudy and Entry!!!");
}
return aVisualObj;
}
SVTK_ViewWindow*
GetViewWindow(const SalomeApp_Module* theModule)
2004-12-01 15:48:31 +05:00
{
2005-08-11 14:04:31 +06:00
if (SalomeApp_Application* anApp = theModule->getApp())
return dynamic_cast<SVTK_ViewWindow*>(anApp->desktop()->activeWindow());
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
QPtrVector<SUIT_ViewWindow> views = theMgr->getViews();
if( views.containsRef( 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 RepaintViewWindow(SVTK_ViewWindow* theWindow)
2005-01-20 11:25:54 +05:00
{
theWindow->Repaint();
2005-01-20 11:25:54 +05:00
}
void RenderViewWindow(SVTK_ViewWindow* theWindow)
{
theWindow->getRenderer()->Render();
theWindow->Repaint();
}
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);
}
}
}
return anActor;
}
void DisplayActor( SUIT_ViewWindow *theWnd, SMESH_Actor* theActor){
if(SVTK_ViewWindow* vtkWnd = GetVtkViewWindow(theWnd)){
vtkWnd->AddActor(theActor);
vtkWnd->Repaint();
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
}
}
2004-12-01 15:48:31 +05:00
void FitAll(){
if(SVTK_ViewWindow* wnd = GetCurrentVtkView() ){
wnd->onFitAll();
wnd->Repaint();
2004-12-01 15:48:31 +05:00
}
}
2004-12-01 15:48:31 +05:00
vtkRenderer* GetCurrentRenderer(){
if(SVTK_ViewWindow* wnd = GetCurrentVtkView() )
return wnd->getRenderer();
2004-12-01 15:48:31 +05:00
return NULL;
}
void RepaintCurrentView(){
if(SVTK_ViewWindow* wnd = GetCurrentVtkView() )
2005-06-27 17:41:01 +06:00
{
wnd->getRenderer()->Render();
wnd->Repaint(false);
}
2004-12-01 15:48:31 +05:00
}
void UpdateView(SUIT_ViewWindow *theWnd, EDisplaing theAction, const char* theEntry)
2004-12-01 15:48:31 +05:00
{
if(SVTK_ViewWindow* aViewWnd = GetVtkViewWindow(theWnd)){
vtkRenderer *aRenderer = aViewWnd->getRenderer();
2004-12-01 15:48:31 +05:00
vtkActorCollection *aCollection = aRenderer->GetActors();
aCollection->InitTraversal();
switch(theAction){
case eDisplayAll: {
while(vtkActor *anAct = aCollection->GetNextActor()){
if(SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)){
anActor->SetVisibility(true);
}
}
break;
}
case eDisplayOnly:
case eEraseAll: {
while(vtkActor *anAct = aCollection->GetNextActor()){
if(SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)){
anActor->SetVisibility(false);
}
}
}
default: {
if(SMESH_Actor *anActor = FindActorByEntry(theWnd,theEntry)){
2004-12-01 15:48:31 +05:00
switch(theAction) {
case eDisplay:
case eDisplayOnly:
anActor->SetVisibility(true);
break;
case eErase:
anActor->SetVisibility(false);
break;
}
} else {
switch(theAction){
case eDisplay:
case eDisplayOnly:{
SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( theWnd->getViewManager()->study() );
_PTR(Study) aDocument = aStudy->studyDS();
2004-12-01 15:48:31 +05:00
if((anActor = CreateActor(aDocument,theEntry,true))) {
DisplayActor(theWnd,anActor);
2004-12-01 15:48:31 +05:00
FitAll();
}
break;
}
}
}
}
}
}
}
void 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();
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())
Update(anActor->getIO(),anActor->GetVisibility());
}
}
}else{
SALOME_ListIteratorOfListIO anIter( selected );
2004-12-01 15:48:31 +05:00
for(; anIter.More(); anIter.Next()){
Handle(SALOME_InteractiveObject) anIO = anIter.Value();
Update(anIO,true);
}
}
RepaintCurrentView();
}
}
void Update(const Handle(SALOME_InteractiveObject)& theIO,
bool theDisplay)
{
_PTR(Study) aStudy = GetActiveStudyDocument();
2004-12-01 15:48:31 +05:00
CORBA::Long anId = aStudy->StudyId();
TVisualObjPtr aVisualObj = SMESH::GetVisualObj(anId,theIO->getEntry());
2005-06-10 09:37:51 +06:00
if( aVisualObj )
aVisualObj->Update();
2004-12-01 15:48:31 +05:00
if ( theDisplay )
UpdateView(SMESH::eDisplay,theIO->getEntry());
}
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;
}
QPtrVector<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 );
for ( int i=0, n=views.count(); i<n; i++ ) {
SVTK_ViewWindow* aVtkView = GetVtkViewWindow( views[i] );
if (!aVtkView) continue;
2004-12-01 15:48:31 +05:00
// update VTK viewer properties
SVTK_RenderWindowInteractor* anInteractor = aVtkView->getRWInteractor();
2004-12-01 15:48:31 +05:00
if (anInteractor) {
// mesh element selection
anInteractor->SetSelectionProp(aSelColor.red()/255., aSelColor.green()/255.,
aSelColor.blue()/255., SW );
2004-12-01 15:48:31 +05:00
// tolerances
anInteractor->SetSelectionTolerance(SP1, SP2);
2004-12-01 15:48:31 +05:00
// pre-selection
SVTK_InteractorStyle* aStyle =
dynamic_cast<SVTK_InteractorStyle*>( anInteractor->GetInteractorStyle() );
2004-12-01 15:48:31 +05:00
if (aStyle) {
aStyle->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
//----------------------------------------------------------------------------
SVTK_InteractorStyle* GetInteractorStyle(SUIT_ViewWindow *theWindow){
if(SVTK_ViewWindow* aWnd = GetVtkViewWindow(theWindow)){
if(SVTK_RenderWindowInteractor* anInteractor = aWnd->getRWInteractor()){
return dynamic_cast<SVTK_InteractorStyle*>( anInteractor->GetInteractorStyle() );
2004-12-01 15:48:31 +05:00
}
}
return NULL;
}
void SetFilter(const Handle(VTKViewer_Filter)& theFilter,
SVTK_InteractorStyle* theStyle)
2004-12-01 15:48:31 +05:00
{
if (theStyle)
theStyle->SetFilter(theFilter);
2004-12-01 15:48:31 +05:00
}
Handle(VTKViewer_Filter) GetFilter(int theId, SVTK_InteractorStyle* theStyle)
2004-12-01 15:48:31 +05:00
{
return theStyle->GetFilter(theId);
}
bool IsFilterPresent(int theId, SVTK_InteractorStyle* theStyle)
2004-12-01 15:48:31 +05:00
{
return theStyle->IsFilterPresent(theId);
}
void RemoveFilter(int theId, SVTK_InteractorStyle* theStyle)
{
2004-12-01 15:48:31 +05:00
theStyle->RemoveFilter(theId);
}
void RemoveFilters(SVTK_InteractorStyle* theStyle)
{
2005-02-02 18:51:36 +05:00
for ( int id = SMESHGUI_NodeFilter; theStyle && id < SMESHGUI_LastFilter; id++ )
theStyle->RemoveFilter( id );
}
2004-12-01 15:48:31 +05:00
bool IsValid(SALOME_Actor* theActor, int theCellId,
SVTK_InteractorStyle* theStyle)
2004-12-01 15:48:31 +05:00
{
return theStyle->IsValid(theActor,theCellId);
}
//----------------------------------------------------------------------------
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();
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);
typedef 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();
for(; anIter != anIdContainer.end(); anIter++){
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() );
}
}
}
}
}