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                                   const bool theDoFitAll)
 121 {
 122   GraphicsView_Viewer* aViewer = module()->getViewer( theViewerId );
 123   if( !aViewer )
 124     return;
 125
 126   GraphicsView_ViewPort* aViewPort = aViewer->getActiveViewPort();
 127   if( !aViewPort )
 128     return;
 129
 130   bool anIsDisplayed = false;
 131   GraphicsView_ObjectList anObjectList = aViewPort->getObjects();
 132   for( int i = 1, n = theObjs.Length(); i <= n; i++ )
 133   {
 134     Handle(HYDROData_Entity) anObj = theObjs.Value( i );
 135     if( anObj.IsNull() )
 136       continue;
 137
 138     HYDROGUI_Prs* aPrs = HYDROGUI_Tool::GetPresentation( anObj, anObjectList );
 139
 140     bool anIsInserted = ( aPrs != 0 );
 141     if( !aPrs || aPrs->getIsToUpdate() || theIsForced )
 142     {
 143       if( HYDROGUI_PrsDriver* aDriver = getDriver( anObj ) )
 144       {
 145         if( aDriver->Update( anObj, aPrs ) && aPrs && !anIsInserted )
 146           aViewPort->addItem( aPrs );
 147       }
 148     }
 149
 150     if( aPrs )
 151     {
 152       bool anIsVisible = module()->isObjectVisible( (size_t)aViewer, anObj );
 153       aPrs->setVisible( anIsVisible );
 154     }
 155   }
 156
 157   aViewPort->onBoundingRectChanged(); // specific of HYDRO module
 158   if ( theDoFitAll )
 159   {
 160     aViewPort->fitAll();
 161   }
 162 }
 163
 164 void HYDROGUI_Displayer::purgeObjects( const int theViewerId )
 165 {
 166   GraphicsView_Viewer* aViewer = module()->getViewer( theViewerId );
 167   if( !aViewer )
 168     return;
 169
 170   GraphicsView_ViewPort* aViewPort = aViewer->getActiveViewPort();
 171   if( !aViewPort )
 172     return;
 173
 174   GraphicsView_ObjectListIterator anIter( HYDROGUI_Tool::GetPrsList( aViewPort ) );
 175   while( anIter.hasNext() )
 176   {
 177     if( HYDROGUI_Prs* aPrs = dynamic_cast<HYDROGUI_Prs*>( anIter.next() ) )
 178     {
 179       Handle(HYDROData_Entity) anObject = aPrs->getObject();
 180       if( !anObject.IsNull() && anObject->IsRemoved() )
 181       {
 182         aViewPort->removeItem( aPrs );
 183         delete aPrs;
 184       }
 185     }
 186   }
 187 }
 188
 189 HYDROGUI_PrsDriver* HYDROGUI_Displayer::getDriver( const Handle(HYDROData_Entity)& theObj )
 190 {
 191   HYDROGUI_PrsDriver* aDriver = NULL;
 192   ObjectKind aKind = theObj->GetKind();
 193   PrsDriversMap::iterator anIter = myPrsDriversMap.find( aKind );
 194   if( anIter != myPrsDriversMap.end() )
 195     aDriver = anIter.value();
 196   else
 197   {
 198     switch( aKind )
 199     {
 200       case KIND_IMAGE:
 201         aDriver = new HYDROGUI_PrsImageDriver();
 202         break;
 203       case KIND_POLYLINE:
 204         aDriver = new HYDROGUI_PrsPolylineDriver();
 205         break;
 206       case KIND_ZONE:
 207         aDriver = new HYDROGUI_PrsZoneDriver();
 208         break;
 209       default:
 210         break;
 211     }
 212
 213     if ( aDriver )
 214       myPrsDriversMap[ aKind ] = aDriver;
 215   }
 216
 217   return aDriver;
 218 }