1 // Copyright (C) 2014-2015 EDF-R&D
2 // This library is free software; you can redistribute it and/or
3 // modify it under the terms of the GNU Lesser General Public
4 // License as published by the Free Software Foundation; either
5 // version 2.1 of the License, or (at your option) any later version.
7 // This library is distributed in the hope that it will be useful,
8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
10 // Lesser General Public License for more details.
12 // You should have received a copy of the GNU Lesser General Public
13 // License along with this library; if not, write to the Free Software
14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 #include "HYDROGUI_OCCDisplayer.h"
21 #include "HYDROGUI_DataModel.h"
22 #include "HYDROGUI_Module.h"
23 #include "HYDROGUI_Tool2.h"
24 #include <HYDROGUI_ShapeImage.h>
25 #include <HYDROGUI_ShapeBathymetry.h>
26 #include <HYDROGUI_ShapeLandCoverMap.h>
27 #include "HYDROGUI_Operation.h"
28 #include "HYDROGUI_DataObject.h"
29 #include "HYDROGUI_ZLayers.h"
31 #include <HYDROData_Bathymetry.h>
32 #include <HYDROData_Image.h>
33 #include <HYDROData_LandCoverMap.h>
34 #include <HYDROData_StricklerTable.h>
36 #include <AIS_InteractiveContext.hxx>
37 #include <AIS_ListIteratorOfListOfInteractive.hxx>
38 #include <AIS_ListOfInteractive.hxx>
40 #include <Aspect_ColorScale.hxx>
42 #include <TColStd_SequenceOfInteger.hxx>
44 #include <LightApp_Application.h>
45 #include <SUIT_Study.h>
47 #include <OCCViewer_ViewManager.h>
48 #include <OCCViewer_ViewModel.h>
49 #include <OCCViewer_ViewWindow.h>
50 #include <OCCViewer_ViewPort3d.h>
52 HYDROGUI_OCCDisplayer::HYDROGUI_OCCDisplayer( HYDROGUI_Module* theModule )
53 : HYDROGUI_AbstractDisplayer( theModule )
55 myToUpdateColorScale = false;
58 HYDROGUI_OCCDisplayer::~HYDROGUI_OCCDisplayer()
62 void HYDROGUI_OCCDisplayer::SetToUpdate( const HYDROData_SequenceOfObjects& theObjs,
63 const int theViewerId )
65 OCCViewer_Viewer* aViewer = module()->getOCCViewer( theViewerId );
69 for ( int i = 1, n = theObjs.Length(); i <= n; i++ )
71 Handle(HYDROData_Entity) anObj = theObjs.Value( i );
75 HYDROGUI_Shape* anObjShape = module()->getObjectShape( (size_t)aViewer, anObj );
79 anObjShape->setIsToUpdate( true );
83 int HYDROGUI_OCCDisplayer::AddPreviewZLayer( OCCViewer_ViewManager* theMgr )
86 OCCViewer_Viewer* aViewer = theMgr->getOCCViewer();
90 aLayer = CreateTopZLayer( aViewer->getViewer3d() );
92 // Hilight presentation should be on top
93 Handle(AIS_InteractiveContext) aCtx = aViewer->getAISContext();
94 if( !aCtx.IsNull() ) {
95 int aTopLayer = CreateTopZLayer( aViewer->getViewer3d() );
96 if ( aTopLayer > 0 ) {
97 UpdateZLayersOfHilightPresentationsOfDisplayedObjects( aCtx, aTopLayer );
104 void HYDROGUI_OCCDisplayer::RemoveZLayer( OCCViewer_ViewManager* theMgr,
110 OCCViewer_Viewer* aViewer = theMgr->getOCCViewer();
114 // Get existing Z layers
115 TColStd_SequenceOfInteger anExistingZLayers;
116 aViewer->getViewer3d()->GetAllZLayers( anExistingZLayers );
117 int aNbLayers = anExistingZLayers.Length();
119 if ( theLayer < aNbLayers )
120 aViewer->getViewer3d()->RemoveZLayer( theLayer );
123 void HYDROGUI_OCCDisplayer::EraseAll( const int theViewerId )
125 OCCViewer_Viewer* aViewer = module()->getOCCViewer( theViewerId );
129 module()->removeViewShapes( (size_t)aViewer );
130 UpdateColorScale( aViewer );
133 void HYDROGUI_OCCDisplayer::Erase( const HYDROData_SequenceOfObjects& theObjs,
134 const int theViewerId )
136 OCCViewer_Viewer* aViewer = module()->getOCCViewer( theViewerId );
140 for ( int i = 1, n = theObjs.Length(); i <= n; i++ )
142 Handle(HYDROData_Entity) anObj = theObjs.Value( i );
146 module()->removeObjectShape( (size_t)aViewer, anObj );
149 if ( !module()->isLandCoversScalarMapModeOn( (size_t)aViewer ) ) {
150 UpdateColorScale( aViewer );
154 HYDROGUI_Shape* HYDROGUI_OCCDisplayer::createShape( const int theViewerId,
155 const Handle(AIS_InteractiveContext)& theContext,
156 const Handle(HYDROData_Entity)& theObject )
158 HYDROGUI_Shape* aResShape = NULL;
159 if ( theContext.IsNull() || theObject.IsNull() )
162 if ( !HYDROGUI_Tool::IsObjectHasPresentation( theObject, OCCViewer_Viewer::Type() ) )
165 if( theObject->IsKind( STANDARD_TYPE( HYDROData_Image ) ) )
166 aResShape = new HYDROGUI_ShapeImage( theContext, Handle_HYDROData_Image::DownCast( theObject ) );
167 else if( theObject->IsKind( STANDARD_TYPE( HYDROData_Bathymetry ) ) )
168 aResShape = new HYDROGUI_ShapeBathymetry( this, theContext, Handle_HYDROData_Bathymetry::DownCast( theObject ) );
169 else if( theObject->IsKind( STANDARD_TYPE( HYDROData_LandCoverMap ) ) ) {
170 bool isScalarMode = module()->isLandCoversScalarMapModeOn( theViewerId );
171 aResShape = new HYDROGUI_ShapeLandCoverMap( this, theContext, Handle_HYDROData_LandCoverMap::DownCast( theObject ), -1, isScalarMode );
174 aResShape = new HYDROGUI_Shape( theContext, theObject );
176 module()->setObjectShape( theViewerId, theObject, aResShape );
181 void HYDROGUI_OCCDisplayer::Display( const HYDROData_SequenceOfObjects& theObjs,
182 const int theViewerId,
183 const bool theIsForced,
184 const bool theDoFitAll )
186 // Get OCC viewer by id
187 OCCViewer_Viewer* aViewer = module()->getOCCViewer( theViewerId );
191 // Get interactive context
192 Handle(AIS_InteractiveContext) aCtx = aViewer->getAISContext();
197 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( module()->getStudyId() );
201 // Assign Z layer indexes to the objects
202 aDoc->Show( theObjs );
204 // Sort objects by display order ( needed for Z layers assignment only )
205 HYDROData_SequenceOfObjects anUnorderedToDisplay = theObjs;
206 HYDROData_SequenceOfObjects anOrderedToDisplay;
207 HYDROData_SequenceOfObjects anAllOrderedObjects = aDoc->GetObjectsLayerOrder();
209 HYDROData_SequenceOfObjects::Iterator anAllOrderedIter( anAllOrderedObjects );
210 for ( ; anAllOrderedIter.More(); anAllOrderedIter.Next() ) {
211 QString anOrderedEntry =
212 HYDROGUI_DataObject::dataObjectEntry( anAllOrderedIter.Value() );
214 HYDROData_SequenceOfObjects::Iterator aToDisplayIter( anUnorderedToDisplay );
215 for ( ; aToDisplayIter.More(); aToDisplayIter.Next() ) {
216 Handle(HYDROData_Entity) anObjToDisplay = aToDisplayIter.Value();
217 QString anEntry = HYDROGUI_DataObject::dataObjectEntry( anObjToDisplay );
218 if ( anEntry == anOrderedEntry ) {
219 anOrderedToDisplay.Prepend( anObjToDisplay );
220 anUnorderedToDisplay.Remove( aToDisplayIter );
227 Handle(V3d_Viewer) aViewer3d = aViewer->getViewer3d();
230 HYDROGUI_ZLayersIterator aZLayersIt( aViewer->getViewer3d() );
231 if ( !aZLayersIt.More() ) {
235 // 1. Display the ordered objects:
236 HYDROData_SequenceOfObjects::Iterator anOrderedIter( anOrderedToDisplay );
237 for ( ; anOrderedIter.More(); anOrderedIter.Next() ) {
238 Handle(HYDROData_Entity) anObj = anOrderedIter.Value();
239 if ( Display( anObj, aViewer, theIsForced ) ) {
240 // set Z layer ( one Z layer for each ordered object )
241 int aZLayerId = aZLayersIt.LayerId();
242 SetZLayer( aViewer, anObj, aZLayerId );
243 SetZLayerSettings( aViewer3d, aZLayerId, true );
248 // 2. Display the unordered objects:
249 bool isDisplayed = false;
250 int anUnorderedZLayerId = aZLayersIt.LayerId();
251 HYDROData_SequenceOfObjects::Iterator anUnorderedIter( anUnorderedToDisplay );
252 for ( ; anUnorderedIter.More(); anUnorderedIter.Next() ) {
253 Handle(HYDROData_Entity) anObj = anUnorderedIter.Value();
254 if ( Display( anObj, aViewer, theIsForced) ) {
255 // set Z layer ( one Z layer for all unordered objects )
256 SetZLayer( aViewer, anObj, anUnorderedZLayerId );
257 if ( !isDisplayed ) {
258 SetZLayerSettings( aViewer3d, anUnorderedZLayerId, false );
264 // 3. Update the top Z layer index
269 // Update Z layer of the active operation
270 int aPreviewZLayerId = aZLayersIt.LayerId();
272 HYDROGUI_Module* aModule = module();
273 SUIT_Operation* anOp = aModule->activeOperation();
274 HYDROGUI_Operation* aHOp = anOp ? dynamic_cast<HYDROGUI_Operation*>( anOp ) : 0;
275 if ( aHOp && aHOp->getPreviewZLayer() >= 0 ) {
276 aHOp->updatePreviewZLayer( aPreviewZLayerId );
280 // Update Z layer of hilight presentations
281 int aHilightLayer = aZLayersIt.TopLayer();
282 UpdateZLayersOfHilightPresentationsOfDisplayedObjects( aCtx, aHilightLayer );
284 // Fit all / update selection
286 OCCViewer_ViewManager* aViewManager
287 = ::qobject_cast<OCCViewer_ViewManager*>( aViewer->getViewManager() );
288 if ( aViewManager ) {
289 OCCViewer_ViewWindow* aViewWindow =
290 ::qobject_cast<OCCViewer_ViewWindow*>( aViewManager->getActiveView() );
292 aViewWindow->onFitAll();
296 else if ( !aCtx.IsNull() ) { // TODO: determine if this code is necessary (added as a fix for issue# 359)
297 aCtx->UpdateSelected();
300 UpdateColorScale( aViewer );
303 void HYDROGUI_OCCDisplayer::purgeObjects( const int theViewerId )
305 OCCViewer_Viewer* aViewer = module()->getOCCViewer( theViewerId );
309 Handle(AIS_InteractiveContext) aCtx = aViewer->getAISContext();
313 AIS_ListOfInteractive aDisplayedObjects;
314 aCtx->DisplayedObjects( aDisplayedObjects );
316 AIS_ListIteratorOfListOfInteractive aListIter( aDisplayedObjects );
317 for ( ; aListIter.More(); aListIter.Next() )
319 Handle(AIS_InteractiveObject) aPrsObj = aListIter.Value();
320 if ( aPrsObj.IsNull() )
323 Handle(HYDROData_Entity) anOwnerObj =
324 Handle(HYDROData_Entity)::DownCast( aPrsObj->GetOwner() );
325 if ( !anOwnerObj.IsNull() && anOwnerObj->IsRemoved() )
326 module()->removeObjectShape( (size_t)aViewer, anOwnerObj );
330 QString HYDROGUI_OCCDisplayer::GetType() const
332 return OCCViewer_Viewer::Type();
335 bool HYDROGUI_OCCDisplayer::Display( const Handle(HYDROData_Entity)& theObject,
336 const OCCViewer_Viewer* theViewer,
337 const bool theIsForced )
341 if ( theObject.IsNull() || theObject->IsRemoved() || !theViewer ) {
345 // Get interactive context
346 Handle(AIS_InteractiveContext) aCtx = theViewer->getAISContext();
347 if( aCtx.IsNull() ) {
352 size_t aViewerId = (size_t)theViewer;
355 HYDROGUI_Shape* anObjShape = module()->getObjectShape( aViewerId, theObject );
358 anObjShape = createShape( aViewerId, aCtx, theObject );
360 anObjShape->setIsToUpdate( true );
367 if ( anObjShape->getIsToUpdate() || theIsForced ) {
368 anObjShape->update( false, false );
372 bool anIsVisible = module()->isObjectVisible( aViewerId, theObject );
373 anObjShape->setVisible( anIsVisible, false );
381 void HYDROGUI_OCCDisplayer::SetZLayer( const OCCViewer_Viewer* theViewer,
382 const Handle(HYDROData_Entity)& theObject,
383 const int theZLayerId )
385 if ( !theViewer /*|| ( theZLayerId < 0 )*/ ) {
389 // Get interactive context
390 Handle(AIS_InteractiveContext) aCtx = theViewer->getAISContext();
391 if( aCtx.IsNull() ) {
396 size_t aViewerId = (size_t)theViewer;
399 HYDROGUI_Shape* anObjShape = module()->getObjectShape( aViewerId, theObject );
403 aCtx->SetZLayer( anObjShape->getAISObject(), theZLayerId );
407 void HYDROGUI_OCCDisplayer::SetToUpdateColorScale()
409 myToUpdateColorScale = true;
412 void HYDROGUI_OCCDisplayer::UpdateColorScale( const OCCViewer_Viewer* theViewer )
414 if( !myToUpdateColorScale || !theViewer )
417 OCCViewer_ViewWindow* aWnd = dynamic_cast<OCCViewer_ViewWindow*>( theViewer->getViewManager()->getActiveView() );
418 Handle(V3d_View) aView = aWnd->getViewPort()->getView();
420 int aViewerId = (size_t)theViewer;//TODO: check if viewer id is correct
421 bool isLandCoverColoringOn = module()->isLandCoversScalarMapModeOn( aViewerId );
423 QList<HYDROGUI_Shape*> aLandCoverMapShapes = module()->getObjectShapes( aViewerId, KIND_LAND_COVER_MAP );
424 QList<HYDROGUI_Shape*> aBathShapes = module()->getObjectShapes( aViewerId, KIND_BATHYMETRY );
426 bool isDisplayColorScale = !aBathShapes.empty() || isLandCoverColoringOn;
427 Standard_Real anXPos = 0.05; //TODO
428 Standard_Real anYPos = 0.1; //TODO
429 Standard_Real aWidth = 0.2; //TODO
430 Standard_Real aHeight = 0.5; //TODO
431 Standard_Integer aTextHeight = 14; //TODO
432 Standard_Integer aNbIntervals = 20; //TODO
433 TCollection_ExtendedString aColorScaleTitle = "";//TODO
435 Standard_Real aColorScaleMin = 0, aColorScaleMax = 1;
438 Handle(HYDROData_StricklerTable) aTable;
439 QStringList aTableTypes;
440 if ( isLandCoverColoringOn ) {
441 aTable = module()->getLandCoverColoringTable( aViewerId );
442 if ( !aTable.IsNull() ) {
443 // TODO: non-empty title leads to buggy behaviour
444 // aColorScaleTitle = TCollection_ExtendedString( aTable->GetName().toLatin1().constData() );
445 aTable->GetCoefficientRange( aColorScaleMin, aColorScaleMax );
446 aTableTypes = aTable->GetTypes();
449 Standard_Real aMin, aMax;
451 foreach( HYDROGUI_Shape* aShape, aBathShapes )
453 HYDROGUI_ShapeBathymetry* aBathShape = dynamic_cast<HYDROGUI_ShapeBathymetry*>( aShape );
454 if( !aBathShape || !aBathShape->isVisible() )
457 aBathShape->GetRange( aMin, aMax );
459 if( isFirst || aMin < aColorScaleMin )
460 aColorScaleMin = aMin;
461 if( isFirst || aMax > aColorScaleMax )
462 aColorScaleMax = aMax;
468 Handle(Aspect_ColorScale) aColorScale;
469 if( isDisplayColorScale )
471 aColorScale = aView->ColorScale();
472 if( !aColorScale.IsNull() )
474 aColorScale->SetXPosition( anXPos );
475 aColorScale->SetYPosition( anYPos );
476 aColorScale->SetWidth( aWidth );
477 aColorScale->SetHeight( aHeight );
479 aColorScale->SetTextHeight( aTextHeight );
480 aColorScale->SetNumberOfIntervals( aNbIntervals );
482 aColorScale->SetTitle( aColorScaleTitle );
483 aColorScale->SetRange( aColorScaleMin, aColorScaleMax );
485 if ( !isLandCoverColoringOn ) {
486 foreach( HYDROGUI_Shape* aShape, aBathShapes ) {
487 HYDROGUI_ShapeBathymetry* aBathShape = dynamic_cast<HYDROGUI_ShapeBathymetry*>( aShape );
488 if( !aBathShape || !aBathShape->isVisible() )
491 aBathShape->UpdateWithColorScale( aColorScale );
495 if( !aView->ColorScaleIsDisplayed() )
496 aView->ColorScaleDisplay();
500 if( aView->ColorScaleIsDisplayed() )
501 aView->ColorScaleErase();
504 foreach( HYDROGUI_Shape* aShape, aLandCoverMapShapes ) {
505 HYDROGUI_ShapeLandCoverMap* aLandCoverMapShape =
506 dynamic_cast<HYDROGUI_ShapeLandCoverMap*>( aShape );
508 if ( !aLandCoverMapShape || !aLandCoverMapShape->isVisible() ) {
512 Handle(HYDROData_LandCoverMap) aLandCoverMap =
513 Handle(HYDROData_LandCoverMap)::DownCast( aLandCoverMapShape->getObject() );
515 if ( aLandCoverMap.IsNull() ) {
519 bool isScalarMode = aLandCoverMapShape->isScalarMapModeEnabled();
520 if( isScalarMode != isLandCoverColoringOn )
522 aLandCoverMapShape->setScalarMapModeEnabled( isLandCoverColoringOn );
523 theViewer->getAISContext()->Redisplay( aLandCoverMapShape->getAISObject() );
527 myToUpdateColorScale = false;