Salome HOME
Merge multi-study removal branch.
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_Displayer.cxx
1 // Copyright (C) 2007-2016  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
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, or (at your option) any later version.
10 //
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.
15 //
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
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22
23 // SMESH SMESHGUI : Displayer for SMESH module
24 // File   : SMESHGUI_Displayer.cxx
25 // Author : Alexander SOLOVYOV, Open CASCADE S.A.S.
26 // SMESH includes
27 //
28
29 #include "SMESHGUI_Displayer.h"
30 #include "SMESHGUI_VTKUtils.h"
31 #include "SMESHGUI_Utils.h"
32
33 // SALOME GUI includes
34 #include <SalomeApp_Study.h>
35 #include <SalomeApp_Application.h>
36 #include <SUIT_ViewManager.h>
37 #include <SVTK_ViewModel.h>
38 #include <SVTK_ViewWindow.h>
39
40
41 // IDL includes
42 #include <SALOMEconfig.h>
43 #include CORBA_SERVER_HEADER(SMESH_Group)
44 #include CORBA_SERVER_HEADER(SMESH_Mesh)
45
46
47 SMESHGUI_Displayer::SMESHGUI_Displayer( SalomeApp_Application* app )
48 : LightApp_Displayer(),
49   myApp( app )
50 {
51 }
52
53 SMESHGUI_Displayer::~SMESHGUI_Displayer()
54 {
55 }
56
57 SALOME_Prs* SMESHGUI_Displayer::buildPresentation( const QString& entry, SALOME_View* theViewFrame )
58 {
59   SALOME_Prs* prs = 0;
60
61   SALOME_View* aViewFrame = theViewFrame ? theViewFrame : GetActiveView();
62
63   if ( aViewFrame )
64   {
65     SVTK_Viewer* vtk_viewer = dynamic_cast<SVTK_Viewer*>( aViewFrame );
66     if( vtk_viewer )
67     {
68       SUIT_ViewWindow* wnd = vtk_viewer->getViewManager()->getActiveView();
69       SMESH_Actor* anActor = SMESH::FindActorByEntry( wnd, entry.toLatin1().data() );
70       if( !anActor )
71         anActor = SMESH::CreateActor( entry.toLatin1().data(), true );
72       if( anActor )
73       {
74         SMESH::DisplayActor( wnd, anActor );
75         prs = LightApp_Displayer::buildPresentation( entry.toLatin1().data(), aViewFrame );
76       }
77       if( prs )
78         UpdatePrs( prs );
79       else if( anActor )
80         SMESH::RemoveActor( vtk_viewer->getViewManager()->getActiveView(), anActor );
81     }
82   }
83
84   return prs;
85 }
86
87 SalomeApp_Study* SMESHGUI_Displayer::study() const
88 {
89   return dynamic_cast<SalomeApp_Study*>( myApp->activeStudy() );
90 }
91
92 bool SMESHGUI_Displayer::canBeDisplayed( const QString& entry, const QString& viewer_type ) const {
93   bool res = false;
94   if(viewer_type != SVTK_Viewer::Type())
95     return res;
96   
97   _PTR(SObject) obj = SMESH::getStudy()->FindObjectID( (const char*)entry.toLatin1() );
98   CORBA::Object_var anObj = SMESH::SObjectToObject( obj );
99   
100     /*
101     if( !CORBA::is_nil( anObj ) ) {
102     SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( anObj );
103     if ( ! mesh->_is_nil() )
104     res = (mesh->NbNodes() > 0);
105     
106     SMESH::SMESH_subMesh_var aSubMeshObj = SMESH::SMESH_subMesh::_narrow( anObj );
107     if ( !aSubMeshObj->_is_nil() )
108     res = (aSubMeshObj->GetNumberOfNodes(true) > 0);
109     
110     SMESH::SMESH_GroupBase_var aGroupObj = SMESH::SMESH_GroupBase::_narrow( anObj );
111     if ( !aGroupObj->_is_nil() )
112     res = !aGroupObj->IsEmpty();
113     }*/
114   if( !CORBA::is_nil( anObj ) ) {
115     if(!SMESH::SMESH_Mesh::_narrow( anObj )->_is_nil() ||
116        !SMESH::SMESH_subMesh::_narrow( anObj )->_is_nil() ||
117        !SMESH::SMESH_GroupBase::_narrow( anObj )->_is_nil())
118     res = true;
119   }
120   return res;
121 }