src/DependencyTree/DependencyTree_ViewModel.cxx

   1 // Copyright (C) 2014  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 // internal includes
  21 #include "DependencyTree_ViewModel.h"
  22 #include "DependencyTree_View.h"
  23
  24 // GUI includes
  25 #include <SUIT_Session.h>
  26 #include <SalomeApp_Study.h>
  27 #include <LightApp_SelectionMgr.h>
  28 #include <SALOME_ListIteratorOfListIO.hxx>
  29 #include <SALOME_ListIO.hxx>
  30 #include <OCCViewer_ViewManager.h>
  31
  32 // GEOM includes
  33 #include <GEOM_Displayer.h>
  34
  35 // QT includes
  36 #include <QMenu>
  37
  38 DependencyTree_ViewModel::DependencyTree_ViewModel( const QString& title )
  39 : GraphicsView_Viewer( title )
  40 {
  41 }
  42
  43 DependencyTree_ViewModel::DependencyTree_ViewModel( const QString& title, QWidget* w )
  44 : GraphicsView_Viewer( title,  w )
  45 {
  46 }
  47
  48 DependencyTree_ViewModel::~DependencyTree_ViewModel()
  49 {
  50 }
  51
  52 //=================================================================================
  53 // function : onShowSelected()
  54 // purpose  : slot for showing selected objects in OCC Viewer
  55 //=================================================================================
  56 void DependencyTree_ViewModel::onShowSelected()
  57 {
  58   SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
  59   if ( !app ) return;
  60
  61   app->getViewManager( OCCViewer_Viewer::Type(), /*create=*/ true );
  62
  63   LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
  64   if ( !aSelMgr ) return;
  65
  66   SALOME_ListIO aSelList;
  67   aSelMgr->selectedObjects(aSelList);
  68
  69   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
  70   GEOM_Displayer* disp = new GEOM_Displayer( appStudy );
  71
  72   OCCViewer_ViewManager* anOCCVM = ( OCCViewer_ViewManager* ) app->getViewManager( OCCViewer_Viewer::Type(), /*create=*/ true );
  73
  74   if ( SUIT_ViewModel* viewModel = anOCCVM->getViewModel() ) {
  75     if ( SALOME_View* viewFrame = dynamic_cast<SALOME_View*>( viewModel ) ) {
  76       SALOME_ListIteratorOfListIO Iter( aSelList );
  77       for ( ; Iter.More(); Iter.Next() )
  78         disp->Display( Iter.Value(), false, viewFrame );
  79       viewFrame->Repaint();
  80     }
  81   }
  82 }
  83
  84 //=================================================================================
  85 // function : onShowOnlySelected()
  86 // purpose  : slot for showing only selected objects in OCC Viewer
  87 //=================================================================================
  88 void DependencyTree_ViewModel::onShowOnlySelected()
  89 {
  90   SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
  91   if ( !app ) return;
  92
  93   LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
  94   if ( !aSelMgr ) return;
  95
  96   SALOME_ListIO aSelList;
  97   aSelMgr->selectedObjects( aSelList );
  98
  99   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
 100   GEOM_Displayer* disp = new GEOM_Displayer( appStudy );
 101
 102   OCCViewer_ViewManager* anOCCVM = (OCCViewer_ViewManager*) app->getViewManager( OCCViewer_Viewer::Type(), /*create=*/ true );
 103
 104   if ( SUIT_ViewModel* viewModel = anOCCVM->getViewModel() ) {
 105     if ( SALOME_View* viewFrame = dynamic_cast<SALOME_View*>( viewModel ) ) {
 106       disp->EraseAll( true, false, viewFrame );
 107       SALOME_ListIteratorOfListIO Iter( aSelList );
 108       for ( ; Iter.More(); Iter.Next() )
 109         disp->Display( Iter.Value(), false, viewFrame );
 110       viewFrame->Repaint();
 111     }
 112   }
 113 }
 114
 115 //=================================================================================
 116 // function : contextMenuPopup()
 117 // purpose  : process calling of context menu popup
 118 //=================================================================================
 119 void DependencyTree_ViewModel::contextMenuPopup( QMenu* theMenu )
 120 {
 121   GraphicsView_Viewer::contextMenuPopup( theMenu );
 122
 123   if( DependencyTree_View* viewPort = dynamic_cast<DependencyTree_View*>( getActiveViewPort() ) )
 124   {
 125     int aNbSelected = viewPort->nbSelected();
 126     if( aNbSelected > 0 ) {
 127       theMenu->clear();
 128       theMenu->addAction( tr( "MEN_DISPLAY" ), this, SLOT( onShowSelected() ) );
 129       theMenu->addAction( tr( "MEN_DISPLAY_ONLY" ), this, SLOT( onShowOnlySelected() ) );
 130       theMenu->addAction( tr( "REBUILD_THE_TREE"), viewPort, SLOT( onRebuildModel() ) );
 131     }
 132   }
 133 }