1 // Copyright (C) 2007-2013 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 #include <HYDROGUI_Shape.h>
24 #include <HYDROGUI_Tool.h>
25 #include <HYDROData_Channel.h>
26 #include <HYDROData_Document.h>
27 #include <HYDROData_DummyObject3D.h>
28 #include <HYDROData_ImmersibleZone.h>
29 #include <HYDROData_Obstacle.h>
30 #include <HYDROData_PolylineXY.h>
31 #include <HYDROData_Polyline3D.h>
32 #include <HYDROData_Profile.h>
33 #include <HYDROData_ShapesGroup.h>
34 #include <HYDROData_Stream.h>
35 #include <HYDROData_Zone.h>
37 #include <AIS_Shape.hxx>
38 #include <BRep_Builder.hxx>
39 #include <BRepBuilderAPI_MakeFace.hxx>
40 #include <Graphic3d_AspectFillArea3d.hxx>
41 #include <Prs3d_IsoAspect.hxx>
42 #include <Prs3d_ShadingAspect.hxx>
44 #include <TopoDS_Face.hxx>
45 #include <TopoDS_Wire.hxx>
46 #include <TopExp_Explorer.hxx>
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_WireFrame )
64 HYDROGUI_Shape::~HYDROGUI_Shape()
68 if ( !myShape.IsNull() )
72 Handle(AIS_InteractiveContext) HYDROGUI_Shape::getContext() const
77 Handle(HYDROData_Entity) HYDROGUI_Shape::getObject() const
82 TopoDS_Shape HYDROGUI_Shape::getTopoShape() const
87 bool HYDROGUI_Shape::getIsToUpdate() const
92 void HYDROGUI_Shape::setIsToUpdate( bool theState )
94 myIsToUpdate = theState;
97 bool HYDROGUI_Shape::isVisible() const
102 Handle(AIS_InteractiveObject) HYDROGUI_Shape::getAISObject() const
107 void HYDROGUI_Shape::display( const bool theIsUpdateViewer )
109 if ( myContext.IsNull() || myShape.IsNull() || !isVisible() )
112 displayShape( theIsUpdateViewer );
115 void HYDROGUI_Shape::erase( const bool theIsUpdateViewer )
117 if ( myContext.IsNull() || myShape.IsNull() )
120 myContext->Erase( myShape, theIsUpdateViewer );
123 void HYDROGUI_Shape::update( bool isUpdateViewer,
124 bool isDeactivateSelection )
127 setIsToUpdate( false );
129 if ( myContext.IsNull() )
132 // Try to retrieve information from object
133 if ( !myObject.IsNull() )
135 Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myObject->Label() );
137 if ( myObject->IsKind( STANDARD_TYPE(HYDROData_ImmersibleZone) ) )
139 Handle(HYDROData_ImmersibleZone) aZoneObj =
140 Handle(HYDROData_ImmersibleZone)::DownCast( myObject );
142 TopoDS_Shape aZoneShape = aZoneObj->GetTopShape();
143 if ( !aZoneShape.IsNull() ) {
144 if ( aZoneShape.ShapeType() == TopAbs_FACE ) {
145 TopoDS_Face aZoneFace = TopoDS::Face( aZoneShape );
146 setFace( aZoneFace, false, false, "" );
148 myTopoShape = aZoneShape;
149 //TODO: myDisplayMode = myTextureFileName.isEmpty() ? AIS_Shaded : AIS_Shaded+2;
150 myDisplayMode = AIS_Shaded;
153 updateShape( false, false );
157 QColor aFillingColor = aZoneObj->GetFillingColor();
158 QColor aBorderColor = aZoneObj->GetBorderColor();
160 setFillingColor( aFillingColor, false, false );
161 setBorderColor( aBorderColor, false, false );
163 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_PolylineXY) ) )
165 Handle(HYDROData_PolylineXY) aPolyline =
166 Handle(HYDROData_PolylineXY)::DownCast( myObject );
168 TopoDS_Shape aPolylineShape = aPolyline->GetShape();
170 if ( !aPolylineShape.IsNull() ) {
171 if ( aPolylineShape.ShapeType() == TopAbs_WIRE ) {
172 TopoDS_Wire aPolylineWire = TopoDS::Wire( aPolylineShape );
173 setWire( aPolylineWire, false, false );
175 myTopoShape = aPolylineShape;
176 // Set shading mode to avoid that hilight presentation is equal to "normal" object presentation.
177 // Note that hilight presentation is always to be on top ( i.e. in the top Z layer ).
178 myDisplayMode = AIS_Shaded;
181 updateShape( false, false );
185 QColor aWireColor = aPolyline->GetWireColor();
186 setBorderColor( aWireColor, false, false );
188 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_Polyline3D) ) )
190 Handle(HYDROData_Polyline3D) aPolyline =
191 Handle(HYDROData_Polyline3D)::DownCast( myObject );
193 TopoDS_Shape aPolylineShape = aPolyline->GetShape3D();
195 if ( !aPolylineShape.IsNull() ) {
196 if ( aPolylineShape.ShapeType() == TopAbs_WIRE ) {
197 TopoDS_Wire aPolylineWire = TopoDS::Wire( aPolylineShape );
198 setWire( aPolylineWire, false, false );
200 myTopoShape = aPolylineShape;
201 // Set shading mode to avoid that hilight presentation is equal to "normal" object presentation.
202 // Note that hilight presentation is always to be on top ( i.e. in the top Z layer ).
203 myDisplayMode = AIS_Shaded;
206 updateShape( false, false );
210 QColor aWireColor = aPolyline->GetBorderColor();
211 setBorderColor( aWireColor, false, false );
213 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_Zone) ) )
215 Handle(HYDROData_Zone) aZone =
216 Handle(HYDROData_Zone)::DownCast( myObject );
218 TopoDS_Face aZoneFace = TopoDS::Face( aZone->GetShape() );
220 setFace( aZoneFace, false, false, "" );
221 if (aZone->IsMergingNeed() && aZone->GetMergeType() == HYDROData_Zone::Merge_UNKNOWN )
223 // Red color for a zone with bathymetry conflict
224 setFillingColor( Qt::red );
228 // Generate the filling color for zone
229 QStringList aGeomObjectsNames;
231 HYDROData_SequenceOfObjects aRefObjects = aZone->GetGeometryObjects();
232 HYDROData_SequenceOfObjects::Iterator anIter( aRefObjects );
233 for ( ; anIter.More(); anIter.Next() )
235 Handle(HYDROData_Object) aRefbject =
236 Handle(HYDROData_Object)::DownCast( anIter.Value() );
237 if ( aRefbject.IsNull() )
240 QString aRefObjectName = aRefbject->GetName();
241 if ( aRefObjectName.isEmpty() )
244 aGeomObjectsNames.append( aRefObjectName );
247 setFillingColor( HYDROGUI_Tool::GenerateFillingColor( aDocument, aGeomObjectsNames ) );
250 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_Profile) ) )
252 Handle(HYDROData_Profile) aProfile =
253 Handle(HYDROData_Profile)::DownCast( myObject );
255 TopoDS_Wire aProfileWire;
257 if ( aProfile->IsValid() ) {
258 TopoDS_Shape aProfileShape = aProfile->GetShape3D();
260 if ( !aProfileShape.IsNull() &&
261 aProfileShape.ShapeType() == TopAbs_WIRE ) {
262 aProfileWire = TopoDS::Wire( aProfileShape );
266 setWire( aProfileWire, false, false );
268 QColor aWireColor = aProfile->GetBorderColor();
269 setBorderColor( aWireColor, false, false );
271 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_Stream) ) ||
272 myObject->IsKind( STANDARD_TYPE(HYDROData_Channel) ) ||
273 myObject->IsKind( STANDARD_TYPE(HYDROData_Obstacle) ) )
275 Handle(HYDROData_Object) aGeomObject =
276 Handle(HYDROData_Object)::DownCast( myObject );
278 TopoDS_Shape anObjShape = aGeomObject->GetTopShape();
280 setShape( anObjShape, false, false );
282 QColor aFillingColor = aGeomObject->GetFillingColor();
283 QColor aBorderColor = aGeomObject->GetBorderColor();
285 setFillingColor( aFillingColor, false, false );
286 setBorderColor( aBorderColor, false, false );
288 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_DummyObject3D) ) )
290 Handle(HYDROData_DummyObject3D) anObject3D =
291 Handle(HYDROData_DummyObject3D)::DownCast( myObject );
292 TopoDS_Shape aShape3D = anObject3D->GetShape();
294 setShape( aShape3D, false, false );
296 QColor aFillingColor = anObject3D->GetFillingColor();
297 QColor aBorderColor = anObject3D->GetBorderColor();
299 setFillingColor( aFillingColor, false, false );
300 setBorderColor( aBorderColor, false, false );
302 else if ( myObject->IsKind( STANDARD_TYPE(HYDROData_ShapesGroup) ) )
304 Handle(HYDROData_ShapesGroup) aShapesGroup =
305 Handle(HYDROData_ShapesGroup)::DownCast( myObject );
307 TopTools_SequenceOfShape aShapes;
308 aShapesGroup->GetShapes( aShapes );
310 TopoDS_Compound aCompound;
311 BRep_Builder aCompoundBuilder;
312 aCompoundBuilder.MakeCompound( aCompound );
314 for ( int i = 1, n = aShapes.Length(); i <= n; ++i )
316 const TopoDS_Shape& aShape = aShapes.Value( i );
317 aCompoundBuilder.Add( aCompound, aShape );
320 setShape( aCompound, false, false );
324 if ( myShape.IsNull() || !isVisible() )
327 displayShape( isUpdateViewer );
329 if (isDeactivateSelection)
330 myContext->Deactivate(myShape);
333 void HYDROGUI_Shape::setVisible( const bool theState,
334 const bool theIsUpdateViewer )
336 myIsVisible = theState;
338 if ( myShape.IsNull() )
341 if ( ( myIsVisible && myContext->IsDisplayed( myShape ) ) ||
342 ( !myIsVisible && !myContext->IsDisplayed( myShape ) ) )
346 displayShape( theIsUpdateViewer );
349 myContext->Erase( myShape, theIsUpdateViewer );
352 void HYDROGUI_Shape::highlight( bool theIsHighlight, bool isUpdateViewer )
354 if ( myIsHighlight == theIsHighlight )
357 myIsHighlight = theIsHighlight;
359 if ( myContext.IsNull() || myShape.IsNull() )
362 colorShapeBorder( getActiveColor() );
363 displayShape( isUpdateViewer );
366 bool HYDROGUI_Shape::isHighlighted() const
368 return myIsHighlight;
371 void HYDROGUI_Shape::setWire( const TopoDS_Wire& theWire,
372 const bool theToDisplay,
373 const bool theIsUpdateViewer )
375 myTopoShape = theWire;
376 // To avoid that hilight presentation is equal to "normal" object presentation.
377 // Note that hilight presentation is always to be on top ( i.e. in the top Z layer ).
378 myDisplayMode = AIS_Shaded;
381 updateShape( theToDisplay, theIsUpdateViewer );
384 void HYDROGUI_Shape::setFaces( const TopoDS_Compound& theWires,
385 const bool theToDisplay,
386 const bool theIsUpdateViewer )
388 TopExp_Explorer anExp( theWires, TopAbs_WIRE );
389 TopoDS_Compound aCompound;
390 BRep_Builder aBuilder;
391 aBuilder.MakeCompound( aCompound );
393 for ( ; anExp.More(); anExp.Next() ) {
394 TopoDS_Wire aWire = TopoDS::Wire( anExp.Current() );
395 if ( aWire.IsNull() ) {
399 BRepBuilderAPI_MakeFace aMakeFace( aWire, Standard_True );
401 if( aMakeFace.IsDone() ) {
402 aBuilder.Add( aCompound, aMakeFace.Face() );
406 myTopoShape = aCompound;
407 myDisplayMode = AIS_Shaded;
410 updateShape( theToDisplay, theIsUpdateViewer );
413 void HYDROGUI_Shape::setFace( const TopoDS_Wire& theWire,
414 const bool theToDisplay,
415 const bool theIsUpdateViewer,
416 const QString& theTextureFileName )
418 BRepBuilderAPI_MakeFace aFaceBuilder( theWire, Standard_True );
419 aFaceBuilder.Build();
420 if( aFaceBuilder.IsDone() )
422 TopoDS_Face aFace = aFaceBuilder.Face();
423 setFace( aFace, theToDisplay, theIsUpdateViewer, theTextureFileName );
427 void HYDROGUI_Shape::setFace( const TopoDS_Face& theFace,
428 const bool theToDisplay,
429 const bool theIsUpdateViewer,
430 const QString& theTextureFileName )
432 myTopoShape = theFace;
433 myDisplayMode = theTextureFileName.isEmpty() ? AIS_Shaded : AIS_Shaded+2;
434 //Note: AIS_Shaded+2 is the same as AIS_ExactHLR
435 //TODO: it would be more suitable to use TexturedShape mode from GEOM_AISShape
438 updateShape( theToDisplay, theIsUpdateViewer );
441 void HYDROGUI_Shape::setShape( const TopoDS_Shape& theShape,
442 const bool theToDisplay,
443 const bool theIsUpdateViewer )
445 myTopoShape = theShape;
446 myDisplayMode = AIS_Shaded;
449 updateShape( theToDisplay, theIsUpdateViewer );
452 void HYDROGUI_Shape::setFillingColor( const QColor& theColor,
453 const bool theToDisplay,
454 const bool theIsUpdateViewer )
456 myFillingColor = theColor;
457 updateShape( theToDisplay, theIsUpdateViewer );
460 QColor HYDROGUI_Shape::getFillingColor() const
462 return myFillingColor;
465 void HYDROGUI_Shape::setBorderColor( const QColor& theColor,
466 const bool theToDisplay,
467 const bool theIsUpdateViewer )
469 myBorderColor = theColor;
470 updateShape( theToDisplay, theIsUpdateViewer );
473 QColor HYDROGUI_Shape::getBorderColor() const
475 return myBorderColor;
478 void HYDROGUI_Shape::setHighlightColor( const QColor& theColor )
480 myHighlightColor = theColor;
483 QColor HYDROGUI_Shape::getHighlightColor() const
485 return myHighlightColor;
488 void HYDROGUI_Shape::setZLayer( const int theZLayer )
490 if ( myZLayer == theZLayer )
493 myZLayer = theZLayer;
494 if ( !myShape.IsNull() && isVisible() && !myContext.IsNull() && myZLayer >= 0 )
495 myContext->SetZLayer( myShape, myZLayer );
498 Handle_AIS_InteractiveObject HYDROGUI_Shape::createShape() const
500 if( myTopoShape.IsNull() )
501 return Handle_AIS_InteractiveObject();
503 return new AIS_Shape( myTopoShape );
506 void HYDROGUI_Shape::buildShape()
508 // Erase previously created shape
511 myShape = createShape();
512 if( myShape.IsNull() )
515 Handle_AIS_Shape anAISShape = Handle_AIS_Shape::DownCast( myShape );
516 if( !anAISShape.IsNull() )
517 anAISShape ->SetHLRAngleAndDeviation( 0.001 );
519 if ( !myObject.IsNull() )
520 myShape->SetOwner( myObject );
522 myShape->SetTransparency( 0 );
523 myShape->SetDisplayMode( (AIS_DisplayMode)myDisplayMode );
525 // Init default params for shape
526 const Handle(AIS_Drawer)& anAttributes = myShape->Attributes();
527 if ( !anAttributes.IsNull() )
529 Handle(Prs3d_IsoAspect) anIsoAspect = anAttributes->UIsoAspect();
530 if ( !anIsoAspect.IsNull() ) {
531 anIsoAspect->SetNumber( 0 );
532 anAttributes->SetUIsoAspect( anIsoAspect );
535 anIsoAspect = anAttributes->VIsoAspect();
536 if ( !anIsoAspect.IsNull() ) {
537 anIsoAspect->SetNumber( 0 );
538 anAttributes->SetVIsoAspect( anIsoAspect );
541 if ( myDisplayMode == AIS_Shaded )
543 Handle(Prs3d_ShadingAspect) aShadingAspect = anAttributes->ShadingAspect();
544 if ( !aShadingAspect.IsNull() )
546 Graphic3d_MaterialAspect aMatAspect( Graphic3d_NOM_PLASTIC );
547 //aMatAspect.SetAmbient( 1 );
548 //aMatAspect.SetDiffuse( 0 );
550 aShadingAspect->Aspect()->SetFrontMaterial( aMatAspect );
551 aShadingAspect->Aspect()->SetBackMaterial( aMatAspect );
554 else if ( myDisplayMode == AIS_WireFrame )
556 anAttributes->SetWireDraw( true );
561 void HYDROGUI_Shape::updateShape( const bool theToDisplay,
562 const bool theIsUpdateViewer )
564 if ( myShape.IsNull() )
567 const Handle(AIS_Drawer)& anAttributes = myShape->Attributes();
568 if ( !anAttributes.IsNull() )
570 if ( myDisplayMode == AIS_Shaded )
572 // Coloring face filling
573 Handle(Prs3d_ShadingAspect) aShadingAspect = anAttributes->ShadingAspect();
574 if ( !aShadingAspect.IsNull() )
576 Quantity_Color aFillingColor( getQuantityColorVal( myFillingColor.red() ),
577 getQuantityColorVal( myFillingColor.green() ),
578 getQuantityColorVal( myFillingColor.blue() ),
581 aShadingAspect->SetColor( aFillingColor );
582 aShadingAspect->SetTransparency( 1 - getQuantityColorVal( myFillingColor.alpha() ) );
585 else if ( myDisplayMode == AIS_WireFrame )
590 colorShapeBorder( getActiveColor() );
593 if ( !theToDisplay || !isVisible() || myContext.IsNull() )
596 displayShape( theIsUpdateViewer );
599 void HYDROGUI_Shape::displayShape( const bool theIsUpdateViewer )
601 myContext->Display( myShape, Standard_False );
604 myContext->SetZLayer( myShape, myZLayer );
606 myContext->UpdateCurrentViewer();
609 QColor HYDROGUI_Shape::getActiveColor() const
611 return isHighlighted() ? myHighlightColor : myBorderColor;
614 double HYDROGUI_Shape::getQuantityColorVal( const int theColorVal )
616 return theColorVal == 0 ? 0 : ( (double)theColorVal / 255 );
619 void HYDROGUI_Shape::colorShapeBorder( const QColor& theColor )
621 if ( myShape.IsNull() )
624 const Handle(AIS_Drawer)& anAttributes = myShape->Attributes();
625 if ( anAttributes.IsNull() )
628 Quantity_Color aBorderColor( getQuantityColorVal( theColor.red() ),
629 getQuantityColorVal( theColor.green() ),
630 getQuantityColorVal( theColor.blue() ),
633 if( !myTopoShape.IsNull() )
635 if ( myTopoShape.ShapeType() == TopAbs_WIRE ) // Note that we display polylines in shaded mode
637 myShape->SetColor( aBorderColor );
639 else if ( myDisplayMode == AIS_Shaded )
641 if ( theColor.alpha() == 0 )
643 anAttributes->SetFaceBoundaryDraw( false );
647 anAttributes->SetFaceBoundaryDraw( true );
649 Handle(Prs3d_LineAspect) aBoundaryAspect = anAttributes->FaceBoundaryAspect();
650 if ( !aBoundaryAspect.IsNull() )
652 aBoundaryAspect->SetColor( aBorderColor );
653 anAttributes->SetFaceBoundaryAspect( aBoundaryAspect );
655 Handle(Prs3d_LineAspect) aWireAspect = anAttributes->WireAspect();
656 if ( !aWireAspect.IsNull() )
658 aWireAspect->SetColor( aBorderColor );
659 anAttributes->SetWireAspect( aWireAspect );
663 else if ( myDisplayMode == AIS_WireFrame )
665 myShape->SetColor( aBorderColor );