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_Shape.h>
20 #include <HYDROGUI_Tool.h>
21 #include <HYDROGUI_Polyline.h>
22 #include <HYDROGUI_AISShape.h>
24 #include <HYDROData_Channel.h>
25 #include <HYDROData_Document.h>
26 #include <HYDROData_DummyObject3D.h>
27 #include <HYDROData_ImmersibleZone.h>
28 #include <HYDROData_Obstacle.h>
29 #include <HYDROData_PolylineXY.h>
30 #include <HYDROData_Polyline3D.h>
31 #include <HYDROData_Profile.h>
32 #include <HYDROData_ShapesGroup.h>
33 #include <HYDROData_Stream.h>
34 #include <HYDROData_Zone.h>
35 #include <HYDROData_LandCoverMap.h>
37 #include <BRep_Builder.hxx>
38 #include <BRepBuilderAPI_MakeFace.hxx>
39 #include <Prs3d_IsoAspect.hxx>
41 #include <TopoDS_Face.hxx>
42 #include <TopoDS_Wire.hxx>
43 #include <TopExp_Explorer.hxx>
45 #include <SUIT_ResourceMgr.h>
46 #include <SUIT_Session.h>
48 HYDROGUI_Shape::HYDROGUI_Shape( const Handle(AIS_InteractiveContext)& theContext,
49 const Handle(HYDROData_Entity)& theObject,
51 : myContext( theContext ),
52 myObject( theObject ),
53 myZLayer( theZLayer ),
54 myIsHighlight( false ),
55 myFillingColor( Qt::transparent ),
56 myBorderColor( Qt::black ),
57 myHighlightColor( Qt::white ),
58 myIsToUpdate( false ),
60 myDisplayMode( AIS_Shaded ),
61 mySelectionMode( AIS_Shape::SelectionMode( TopAbs_SHAPE ) )
65 HYDROGUI_Shape::~HYDROGUI_Shape()
70 Handle(AIS_InteractiveContext) HYDROGUI_Shape::getContext() const
75 Handle(HYDROData_Entity) HYDROGUI_Shape::getObject() const
80 TopoDS_Shape HYDROGUI_Shape::getTopoShape() const
85 bool HYDROGUI_Shape::getIsToUpdate() const
90 void HYDROGUI_Shape::setIsToUpdate( bool theState )
92 myIsToUpdate = theState;
95 bool HYDROGUI_Shape::isVisible() const
100 QList<Handle(AIS_InteractiveObject)> HYDROGUI_Shape::getAISObjects() const
105 void HYDROGUI_Shape::display( const bool theIsUpdateViewer )
107 if ( myContext.IsNull() || myShapes.empty() || !isVisible() )
110 displayShape( theIsUpdateViewer );
113 void HYDROGUI_Shape::erase( const bool theIsUpdateViewer )
115 if ( myContext.IsNull() || myShapes.empty() )
118 eraseShape( theIsUpdateViewer );
121 void HYDROGUI_Shape::update( bool isUpdateViewer,
122 bool isDeactivateSelection )
125 setIsToUpdate( false );
127 if ( myContext.IsNull() )
130 // Try to retrieve information from object
131 if ( !myObject.IsNull() )
133 Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myObject->Label() );
135 if ( myObject->IsKind( STANDARD_TYPE(HYDROData_ImmersibleZone) ) )
137 Handle(HYDROData_ImmersibleZone) aZoneObj =
138 Handle(HYDROData_ImmersibleZone)::DownCast( myObject );
140 TopoDS_Shape aZoneShape = aZoneObj->GetTopShape();
141 if ( !aZoneShape.IsNull() ) {
142 if ( aZoneShape.ShapeType() == TopAbs_FACE ) {
143 TopoDS_Face aZoneFace = TopoDS::Face( aZoneShape );
144 setFace( aZoneFace, false, false, "" );
146 myTopoShape = aZoneShape;
147 //TODO: myDisplayMode = myTextureFileName.isEmpty() ? AIS_Shaded : AIS_Shaded+2;
148 myDisplayMode = AIS_Shaded;
151 updateShape( false, false );
155 QColor aFillingColor = aZoneObj->GetFillingColor();
156 QColor aBorderColor = aZoneObj->GetBorderColor();
158 setFillingColor( aFillingColor, false, false );
159 setBorderColor( aBorderColor, false, false );
161 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_PolylineXY) ) )
163 Handle(HYDROData_PolylineXY) aPolyline =
164 Handle(HYDROData_PolylineXY)::DownCast( myObject );
166 TopoDS_Shape aPolylineShape = aPolyline->GetShape();
168 if ( !aPolylineShape.IsNull() )
170 if ( aPolylineShape.ShapeType() == TopAbs_WIRE )
172 TopoDS_Wire aPolylineWire = TopoDS::Wire( aPolylineShape );
173 setWire( aPolylineWire, false, false );
177 myTopoShape = aPolylineShape;
178 // Set shading mode to avoid that hilight presentation is equal to "normal" object presentation.
179 // Note that hilight presentation is always to be on top ( i.e. in the top Z layer ).
180 myDisplayMode = AIS_Shaded;
183 updateShape( false, false );
187 QColor aWireColor = aPolyline->GetWireColor();
188 setBorderColor( aWireColor, false, false );
190 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_Polyline3D) ) )
192 Handle(HYDROData_Polyline3D) aPolyline =
193 Handle(HYDROData_Polyline3D)::DownCast( myObject );
195 TopoDS_Shape aPolylineShape = aPolyline->GetShape3D();
197 if ( !aPolylineShape.IsNull() ) {
198 if ( aPolylineShape.ShapeType() == TopAbs_WIRE ) {
199 TopoDS_Wire aPolylineWire = TopoDS::Wire( aPolylineShape );
200 setWire( aPolylineWire, false, false );
202 myTopoShape = aPolylineShape;
203 // Set shading mode to avoid that hilight presentation is equal to "normal" object presentation.
204 // Note that hilight presentation is always to be on top ( i.e. in the top Z layer ).
205 myDisplayMode = AIS_Shaded;
208 updateShape( false, false );
212 QColor aWireColor = aPolyline->GetBorderColor();
213 setBorderColor( aWireColor, false, false );
215 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_Zone) ) )
217 Handle(HYDROData_Zone) aZone =
218 Handle(HYDROData_Zone)::DownCast( myObject );
220 TopoDS_Face aZoneFace = TopoDS::Face( aZone->GetShape() );
222 setFace( aZoneFace, false, false, "" );
223 if( aZone->IsMergingNeed() && aZone->GetMergeType() == HYDROData_Zone::Merge_UNKNOWN )
225 // Red color for a zone with bathymetry conflict
226 setFillingColor( Qt::red );
230 // Set the filling color for zone
231 setFillingColor( aZone->GetColor( Qt::darkBlue ) );
234 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_Profile) ) )
236 Handle(HYDROData_Profile) aProfile =
237 Handle(HYDROData_Profile)::DownCast( myObject );
239 TopoDS_Wire aProfileWire;
241 if ( aProfile->IsValid() ) {
242 TopoDS_Shape aProfileShape = aProfile->GetShape3D();
244 if ( !aProfileShape.IsNull() &&
245 aProfileShape.ShapeType() == TopAbs_WIRE ) {
246 aProfileWire = TopoDS::Wire( aProfileShape );
250 setWire( aProfileWire, false, false );
252 QColor aWireColor = aProfile->GetBorderColor();
253 setBorderColor( aWireColor, false, false );
255 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_Stream) ) ||
256 myObject->IsKind( STANDARD_TYPE(HYDROData_Channel) ) ||
257 myObject->IsKind( STANDARD_TYPE(HYDROData_Obstacle) ) )
259 Handle(HYDROData_Object) aGeomObject =
260 Handle(HYDROData_Object)::DownCast( myObject );
262 TopoDS_Shape anObjShape = aGeomObject->GetTopShape();
264 setShape( anObjShape, false, false );
266 QColor aFillingColor = aGeomObject->GetFillingColor();
267 QColor aBorderColor = aGeomObject->GetBorderColor();
269 setFillingColor( aFillingColor, false, false );
270 setBorderColor( aBorderColor, false, false );
272 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_DummyObject3D) ) )
274 Handle(HYDROData_DummyObject3D) anObject3D =
275 Handle(HYDROData_DummyObject3D)::DownCast( myObject );
276 TopoDS_Shape aShape3D = anObject3D->GetShape();
278 setShape( aShape3D, false, false );
280 QColor aFillingColor = anObject3D->GetFillingColor();
281 QColor aBorderColor = anObject3D->GetBorderColor();
283 setFillingColor( aFillingColor, false, false );
284 setBorderColor( aBorderColor, false, false );
286 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_ShapesGroup) ) )
288 Handle(HYDROData_ShapesGroup) aShapesGroup =
289 Handle(HYDROData_ShapesGroup)::DownCast( myObject );
291 TopTools_SequenceOfShape aShapes;
292 aShapesGroup->GetShapes( aShapes );
294 TopoDS_Compound aCompound;
295 BRep_Builder aCompoundBuilder;
296 aCompoundBuilder.MakeCompound( aCompound );
298 for ( int i = 1, n = aShapes.Length(); i <= n; ++i )
300 const TopoDS_Shape& aShape = aShapes.Value( i );
301 aCompoundBuilder.Add( aCompound, aShape );
304 setShape( aCompound, false, false );
308 if ( myShapes.empty() || !isVisible() )
311 displayShape( isUpdateViewer );
313 if (isDeactivateSelection)
314 foreach( Handle(AIS_InteractiveObject) aShape, myShapes )
315 myContext->Deactivate( aShape );
318 void HYDROGUI_Shape::setVisible( const bool theState,
319 const bool theIsUpdateViewer )
321 myIsVisible = theState;
323 if ( myShapes.empty() )
326 if ( ( myIsVisible && myContext->IsDisplayed( myShapes[0] ) ) ||
327 ( !myIsVisible && !myContext->IsDisplayed( myShapes[0] ) ) )
331 displayShape( theIsUpdateViewer );
334 eraseShape( theIsUpdateViewer );
337 void HYDROGUI_Shape::highlight( bool theIsHighlight, bool isUpdateViewer )
339 if ( myIsHighlight == theIsHighlight )
342 myIsHighlight = theIsHighlight;
344 if ( myContext.IsNull() || myShapes.empty() )
347 colorShapeBorder( getActiveColor() );
348 displayShape( isUpdateViewer );
351 bool HYDROGUI_Shape::isHighlighted() const
353 return myIsHighlight;
356 void HYDROGUI_Shape::setWire( const TopoDS_Wire& theWire,
357 const bool theToDisplay,
358 const bool theIsUpdateViewer )
360 myTopoShape = theWire;
361 // To avoid that hilight presentation is equal to "normal" object presentation.
362 // Note that hilight presentation is always to be on top ( i.e. in the top Z layer ).
363 myDisplayMode = AIS_Shaded;
366 updateShape( theToDisplay, theIsUpdateViewer );
369 void HYDROGUI_Shape::setFaces( const TopoDS_Compound& theWires,
370 const bool theToDisplay,
371 const bool theIsUpdateViewer )
373 TopExp_Explorer anExp( theWires, TopAbs_WIRE );
374 TopoDS_Compound aCompound;
375 BRep_Builder aBuilder;
376 aBuilder.MakeCompound( aCompound );
378 for ( ; anExp.More(); anExp.Next() ) {
379 TopoDS_Wire aWire = TopoDS::Wire( anExp.Current() );
380 if ( aWire.IsNull() ) {
384 BRepBuilderAPI_MakeFace aMakeFace( aWire, Standard_True );
386 if( aMakeFace.IsDone() ) {
387 aBuilder.Add( aCompound, aMakeFace.Face() );
391 myTopoShape = aCompound;
392 myDisplayMode = AIS_Shaded;
395 updateShape( theToDisplay, theIsUpdateViewer );
398 void HYDROGUI_Shape::setFace( const TopoDS_Wire& theWire,
399 const bool theToDisplay,
400 const bool theIsUpdateViewer,
401 const QString& theTextureFileName )
403 BRepBuilderAPI_MakeFace aFaceBuilder( theWire, Standard_True );
404 aFaceBuilder.Build();
405 if( aFaceBuilder.IsDone() )
407 TopoDS_Face aFace = aFaceBuilder.Face();
408 setFace( aFace, theToDisplay, theIsUpdateViewer, theTextureFileName );
412 void HYDROGUI_Shape::setFace( const TopoDS_Face& theFace,
413 const bool theToDisplay,
414 const bool theIsUpdateViewer,
415 const QString& theTextureFileName )
417 myTopoShape = theFace;
418 myDisplayMode = theTextureFileName.isEmpty() ? AIS_Shaded : AIS_Shaded+2;
419 //Note: AIS_Shaded+2 is the same as AIS_ExactHLR
420 //TODO: it would be more suitable to use TexturedShape mode from GEOM_AISShape
423 updateShape( theToDisplay, theIsUpdateViewer );
426 void HYDROGUI_Shape::setShape( const TopoDS_Shape& theShape,
427 const bool theToDisplay,
428 const bool theIsUpdateViewer,
429 const int theDisplayMode )
431 myTopoShape = theShape;
432 myDisplayMode = theDisplayMode;
435 updateShape( theToDisplay, theIsUpdateViewer );
438 void HYDROGUI_Shape::setFillingColor( const QColor& theColor,
439 const bool theToDisplay,
440 const bool theIsUpdateViewer )
442 myFillingColor = theColor;
443 updateShape( theToDisplay, theIsUpdateViewer );
446 QColor HYDROGUI_Shape::getFillingColor() const
448 return myFillingColor;
451 void HYDROGUI_Shape::setBorderColor( const QColor& theColor,
452 const bool theToDisplay,
453 const bool theIsUpdateViewer )
455 myBorderColor = theColor;
456 updateShape( theToDisplay, theIsUpdateViewer );
459 QColor HYDROGUI_Shape::getBorderColor() const
461 return myBorderColor;
464 void HYDROGUI_Shape::setHighlightColor( const QColor& theColor )
466 myHighlightColor = theColor;
469 QColor HYDROGUI_Shape::getHighlightColor() const
471 return myHighlightColor;
474 void HYDROGUI_Shape::setZLayer( const int theZLayer )
476 if ( myZLayer == theZLayer )
479 myZLayer = theZLayer;
480 if( isVisible() && !myContext.IsNull() && myZLayer >= 0 )
481 foreach( Handle(AIS_InteractiveObject) aShape, myShapes )
482 myContext->SetZLayer( aShape, myZLayer );
485 QList<Handle(AIS_InteractiveObject)> HYDROGUI_Shape::createShape() const
487 QList<Handle(AIS_InteractiveObject)> shapes;
488 if( myTopoShape.IsNull() ) {
492 TopAbs_ShapeEnum aShapeType = myTopoShape.ShapeType();
493 bool IsWireEdgeCompound = aShapeType==TopAbs_COMPOUND;
494 if (IsWireEdgeCompound) {
495 TopoDS_Iterator itr(myTopoShape);
496 while (itr.More() && IsWireEdgeCompound) {
497 if (itr.Value().ShapeType() != TopAbs_WIRE && itr.Value().ShapeType() != TopAbs_EDGE)
498 IsWireEdgeCompound = false;
503 if ( aShapeType==TopAbs_EDGE || aShapeType==TopAbs_WIRE || IsWireEdgeCompound)
505 SUIT_ResourceMgr* aResMgr = 0;
506 SUIT_Session* aSession = SUIT_Session::session();
508 aResMgr = SUIT_Session::session()->resourceMgr();
514 aResMgr->value( "polyline", "arrow_type", aType );
515 aResMgr->value( "polyline", "arrow_size", aSize );
518 shapes.append( HYDROGUI_Polyline::createPresentations( myTopoShape, aType, aSize ) );
522 shapes.append( new HYDROGUI_AISShape( myTopoShape ) );
527 void HYDROGUI_Shape::buildShape()
529 // Erase previously created shape
532 myShapes = createShape();
533 if( myShapes.empty() )
536 foreach( Handle(AIS_InteractiveObject) aShape, myShapes )
538 Handle(AIS_Shape) anAISShape = Handle(AIS_Shape)::DownCast( aShape );
539 if( !anAISShape.IsNull() )
540 anAISShape ->SetHLRAngleAndDeviation( 0.001 );
542 if ( !myObject.IsNull() )
543 aShape->SetOwner( myObject );
545 if ( !myObject.IsNull() && !myObject->IsKind( STANDARD_TYPE(HYDROData_LandCoverMap) ) )
546 aShape->SetTransparency( 0 );
547 aShape->SetDisplayMode( (AIS_DisplayMode)myDisplayMode );
549 // Init default params for shape
550 const Handle(Prs3d_Drawer)& anAttributes = aShape->Attributes();
551 if ( !anAttributes.IsNull() )
553 Handle(Prs3d_IsoAspect) anIsoAspect = anAttributes->UIsoAspect();
554 if ( !anIsoAspect.IsNull() ) {
555 anIsoAspect->SetNumber( 0 );
556 anAttributes->SetUIsoAspect( anIsoAspect );
559 anIsoAspect = anAttributes->VIsoAspect();
560 if ( !anIsoAspect.IsNull() )
562 anIsoAspect->SetNumber( 0 );
563 anAttributes->SetVIsoAspect( anIsoAspect );
566 if ( myDisplayMode == AIS_Shaded )
568 aShape->SetMaterial( Graphic3d_NOM_PLASTIC );
570 else if ( myDisplayMode == AIS_WireFrame )
572 anAttributes->SetWireDraw( true );
578 void HYDROGUI_Shape::updateShape( const bool theToDisplay,
579 const bool theIsUpdateViewer )
581 foreach( Handle(AIS_InteractiveObject) aShape, myShapes )
583 const Handle(Prs3d_Drawer)& anAttributes = aShape->Attributes();
584 if ( !anAttributes.IsNull() )
586 if ( myDisplayMode == AIS_Shaded && myFillingColor.isValid() )
588 Quantity_Color aFillingColor( getQuantityColorVal( myFillingColor.red() ),
589 getQuantityColorVal( myFillingColor.green() ),
590 getQuantityColorVal( myFillingColor.blue() ),
593 aShape->SetColor( aFillingColor );
594 aShape->SetTransparency( 1 - getQuantityColorVal( myFillingColor.alpha() ) );
596 else if ( myDisplayMode == AIS_WireFrame )
604 colorShapeBorder( myBorderColor );
605 if ( !theToDisplay || !isVisible() || myContext.IsNull() )
608 displayShape( theIsUpdateViewer );
611 void HYDROGUI_Shape::displayShape( const bool theIsUpdateViewer )
613 if ( myContext->HasOpenedContext() && mySelectionMode > 0 )
614 myContext->CloseLocalContext();
616 foreach( Handle(AIS_InteractiveObject) aShape, myShapes )
618 if ( mySelectionMode > 0 )
619 // Display object in local context with selection
620 myContext->Display( aShape, myDisplayMode, mySelectionMode, Standard_False, Standard_False );
623 if ( !myContext->HasOpenedContext() )
624 // Ordinary display of object published in the Object Browser
625 myContext->Display( aShape, Standard_False );
627 // Display object in local context without selection
628 myContext->Display( aShape, myDisplayMode, -1, Standard_False, Standard_False );
632 if ( mySelectionMode > 0 )
634 myContext->OpenLocalContext();
635 foreach( Handle(AIS_InteractiveObject) aShape, myShapes )
636 myContext->Activate( aShape, mySelectionMode, Standard_True );
641 foreach( Handle(AIS_InteractiveObject) aShape, myShapes )
642 myContext->SetZLayer( aShape, myZLayer );
645 if ( theIsUpdateViewer ) {
646 myContext->UpdateCurrentViewer();
650 void HYDROGUI_Shape::eraseShape( const bool theIsUpdateViewer )
652 if ( myContext->HasOpenedContext() && mySelectionMode > 0 )
653 myContext->CloseLocalContext();
655 foreach( Handle(AIS_InteractiveObject) aShape, myShapes )
656 myContext->Erase( aShape, Standard_False );
658 if( theIsUpdateViewer )
659 myContext->UpdateCurrentViewer();
662 QColor HYDROGUI_Shape::getActiveColor() const
664 return isHighlighted() ? myHighlightColor : myBorderColor;
667 double HYDROGUI_Shape::getQuantityColorVal( const int theColorVal )
669 return theColorVal == 0 ? 0 : ( (double)theColorVal / 255 );
672 void HYDROGUI_Shape::colorShapeBorder( const QColor& theColor )
674 foreach( Handle(AIS_InteractiveObject) aShape1, myShapes )
676 const Handle(Prs3d_Drawer)& anAttributes = aShape1->Attributes();
677 if ( anAttributes.IsNull() )
680 Quantity_Color aBorderColor( getQuantityColorVal( theColor.red() ),
681 getQuantityColorVal( theColor.green() ),
682 getQuantityColorVal( theColor.blue() ),
685 if( !myTopoShape.IsNull() )
687 if ( myTopoShape.ShapeType() == TopAbs_WIRE ) // Note that we display polylines in shaded mode
689 aShape1->SetColor( aBorderColor );
691 else if ( myDisplayMode == AIS_Shaded )
693 Handle(HYDROGUI_AISShape) aShape = Handle(HYDROGUI_AISShape)::DownCast( aShape1 );
694 if( !aShape.IsNull() )
695 aShape->SetBorderColor( aBorderColor );
697 else if ( myDisplayMode == AIS_WireFrame )
699 aShape1->SetColor( aBorderColor );
705 void HYDROGUI_Shape::setDisplayMode( int theDisplayMode )
707 myDisplayMode = theDisplayMode;
710 void HYDROGUI_Shape::setSelectionMode( int theSelectionMode )
712 mySelectionMode = theSelectionMode;