src/HYDROGUI/HYDROGUI_Displayer.cxx

   1 // Copyright (C) 2007-2013  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.
  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 #include "HYDROGUI_Displayer.h"
  24
  25 #include "HYDROGUI_DataModel.h"
  26 #include "HYDROGUI_Module.h"
  27 #include "HYDROGUI_Prs.h"
  28 #include "HYDROGUI_PrsImageDriver.h"
  29 #include "HYDROGUI_PrsPolylineDriver.h"
  30 #include "HYDROGUI_PrsZoneDriver.h"
  31 #include "HYDROGUI_Tool.h"
  32
  33 #include <GraphicsView_Viewer.h>
  34 #include <GraphicsView_ViewPort.h>
  35
  36 HYDROGUI_Displayer::HYDROGUI_Displayer( HYDROGUI_Module* theModule )
  37 : HYDROGUI_AbstractDisplayer( theModule )
  38 {
  39 }
  40
  41 HYDROGUI_Displayer::~HYDROGUI_Displayer()
  42 {
  43 }
  44
  45 void HYDROGUI_Displayer::SetToUpdate( const HYDROData_SequenceOfObjects& theObjs,
  46                                       const int theViewerId )
  47 {
  48   GraphicsView_Viewer* aViewer = module()->getViewer( theViewerId );
  49   if( !aViewer )
  50     return;
  51
  52   GraphicsView_ViewPort* aViewPort = aViewer->getActiveViewPort();
  53   if( !aViewPort )
  54     return;
  55
  56   GraphicsView_ObjectList anObjectList = aViewPort->getObjects();
  57   for( int i = 1, n = theObjs.Length(); i <= n; i++ )
  58   {
  59     Handle(HYDROData_Entity) anObj = theObjs.Value( i );
  60     if( anObj.IsNull() )
  61       continue;
  62
  63     if( HYDROGUI_Prs* aPrs = HYDROGUI_Tool::GetPresentation( anObj, anObjectList ) )
  64       aPrs->setIsToUpdate( true );
  65   }
  66 }
  67
  68 void HYDROGUI_Displayer::EraseAll( const int theViewerId )
  69 {
  70   GraphicsView_Viewer* aViewer = module()->getViewer( theViewerId );
  71   if( !aViewer )
  72     return;
  73
  74   GraphicsView_ViewPort* aViewPort = aViewer->getActiveViewPort();
  75   if( !aViewPort )
  76     return;
  77
  78   GraphicsView_ObjectListIterator anIter( HYDROGUI_Tool::GetPrsList( aViewPort ) );
  79   while( anIter.hasNext() )
  80   {
  81     if( GraphicsView_Object* anObject = anIter.next() )
  82     {
  83       aViewPort->removeItem( anObject );
  84       delete anObject;
  85     }
  86   }
  87 }
  88
  89 void HYDROGUI_Displayer::Erase( const HYDROData_SequenceOfObjects& theObjs,
  90                                 const int theViewerId )
  91 {
  92   GraphicsView_Viewer* aViewer = module()->getViewer( theViewerId );
  93   if( !aViewer )
  94     return;
  95
  96   GraphicsView_ViewPort* aViewPort = aViewer->getActiveViewPort();
  97   if( !aViewPort )
  98     return;
  99
 100   HYDROGUI_DataModel* aModel = (HYDROGUI_DataModel*)module()->dataModel();
 101   if( aModel )
 102   {
 103     GraphicsView_ObjectList anObjectList = HYDROGUI_Tool::GetPrsList( aViewPort );
 104     for( int i = 1, n = theObjs.Length(); i <= n; i++ )
 105     {
 106       // the object may be null or dead
 107       const Handle(HYDROData_Entity)& anObj = theObjs.Value( i );
 108       if( HYDROGUI_Prs* aPrs = HYDROGUI_Tool::GetPresentation( anObj, anObjectList ) )
 109       {
 110         aViewPort->removeItem( aPrs );
 111         delete aPrs;
 112       }
 113     }
 114   }
 115 }
 116
 117 void HYDROGUI_Displayer::Display( const HYDROData_SequenceOfObjects& theObjs,
 118                                   const int theViewerId,
 119                                   const bool theIsForced )
 120 {
 121   GraphicsView_Viewer* aViewer = module()->getViewer( theViewerId );
 122   if( !aViewer )
 123     return;
 124
 125   GraphicsView_ViewPort* aViewPort = aViewer->getActiveViewPort();
 126   if( !aViewPort )
 127     return;
 128
 129   bool anIsDisplayed = false;
 130   GraphicsView_ObjectList anObjectList = aViewPort->getObjects();
 131   for( int i = 1, n = theObjs.Length(); i <= n; i++ )
 132   {
 133     Handle(HYDROData_Entity) anObj = theObjs.Value( i );
 134     if( anObj.IsNull() )
 135       continue;
 136
 137     HYDROGUI_Prs* aPrs = HYDROGUI_Tool::GetPresentation( anObj, anObjectList );
 138
 139     bool anIsInserted = ( aPrs != 0 );
 140     if( !aPrs || aPrs->getIsToUpdate() || theIsForced )
 141     {
 142       if( HYDROGUI_PrsDriver* aDriver = getDriver( anObj ) )
 143       {
 144         if( aDriver->Update( anObj, aPrs ) && aPrs && !anIsInserted )
 145           aViewPort->addItem( aPrs );
 146       }
 147     }
 148
 149     if( aPrs )
 150     {
 151       bool anIsVisible = module()->isObjectVisible( (size_t)aViewer, anObj );
 152       aPrs->setVisible( anIsVisible );
 153     }
 154   }
 155
 156   aViewPort->onBoundingRectChanged(); // specific of HYDRO module
 157   aViewPort->fitAll();
 158 }
 159
 160 void HYDROGUI_Displayer::purgeObjects( const int theViewerId )
 161 {
 162   GraphicsView_Viewer* aViewer = module()->getViewer( theViewerId );
 163   if( !aViewer )
 164     return;
 165
 166   GraphicsView_ViewPort* aViewPort = aViewer->getActiveViewPort();
 167   if( !aViewPort )
 168     return;
 169
 170   GraphicsView_ObjectListIterator anIter( HYDROGUI_Tool::GetPrsList( aViewPort ) );
 171   while( anIter.hasNext() )
 172   {
 173     if( HYDROGUI_Prs* aPrs = dynamic_cast<HYDROGUI_Prs*>( anIter.next() ) )
 174     {
 175       Handle(HYDROData_Entity) anObject = aPrs->getObject();
 176       if( !anObject.IsNull() && anObject->IsRemoved() )
 177       {
 178         aViewPort->removeItem( aPrs );
 179         delete aPrs;
 180       }
 181     }
 182   }
 183 }
 184
 185 HYDROGUI_PrsDriver* HYDROGUI_Displayer::getDriver( const Handle(HYDROData_Entity)& theObj )
 186 {
 187   HYDROGUI_PrsDriver* aDriver = NULL;
 188   ObjectKind aKind = theObj->GetKind();
 189   PrsDriversMap::iterator anIter = myPrsDriversMap.find( aKind );
 190   if( anIter != myPrsDriversMap.end() )
 191     aDriver = anIter.value();
 192   else
 193   {
 194     switch( aKind )
 195     {
 196       case KIND_IMAGE:
 197         aDriver = new HYDROGUI_PrsImageDriver();
 198         break;
 199       case KIND_POLYLINE:
 200         aDriver = new HYDROGUI_PrsPolylineDriver();
 201         break;
 202       case KIND_ZONE:
 203         aDriver = new HYDROGUI_PrsZoneDriver();
 204         break;
 205       default:
 206         break;
 207     }
 208
 209     if ( aDriver )
 210       myPrsDriversMap[ aKind ] = aDriver;
 211   }
 212
 213   return aDriver;
 214 }