1 // Copyright (C) 2007-2015 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, or (at your option) any later version.
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
24 \class GEOM_AISShape GEOM_AISShape.hxx
28 #include "GEOM_AISShape.hxx"
29 #include "GEOM_AISVector.hxx"
31 #include <GEOMUtils.hxx>
33 #include <Basics_OCCTVersion.hxx>
35 // Open CASCADE Includes
36 #include <AIS_InteractiveContext.hxx>
37 #include <BRep_Tool.hxx>
38 #include <GCPnts_AbscissaPoint.hxx>
39 #include <GeomAdaptor_Curve.hxx>
43 #include <Graphic3d_AspectFillArea3d.hxx>
44 #include <Graphic3d_AspectLine3d.hxx>
45 #include <Graphic3d_AspectMarker3d.hxx>
46 #include <Graphic3d_AspectText3d.hxx>
48 #include <Prs3d_ShadingAspect.hxx>
49 #include <Prs3d_Arrow.hxx>
50 #include <Prs3d_IsoAspect.hxx>
51 #if OCC_VERSION_LARGE > 0x06070200
52 #include <Prs3d_VertexDrawMode.hxx>
55 #include <SelectBasics_SensitiveEntity.hxx>
56 #include <SelectMgr_EntityOwner.hxx>
57 #include <StdSelect_BRepOwner.hxx>
58 #include <SelectMgr_IndexedMapOfOwner.hxx>
59 #include <SelectMgr_Selection.hxx>
60 #include <StdSelect_DisplayMode.hxx>
61 #include <StdPrs_ShadedShape.hxx>
62 #include <StdPrs_WFDeflectionShape.hxx>
64 #include <TColStd_IndexedMapOfInteger.hxx>
65 #include <TColStd_ListIteratorOfListOfInteger.hxx>
66 #include <TColStd_ListOfInteger.hxx>
68 #include <TopoDS_Shape.hxx>
69 #include <TopTools_IndexedMapOfShape.hxx>
71 #include <TopExp_Explorer.hxx>
73 #include <TopoDS_Edge.hxx>
74 #include <TopoDS_Shape.hxx>
75 #include <TopoDS_Vertex.hxx>
77 #include <TColStd_SequenceOfInteger.hxx>
79 #include <V3d_View.hxx>
81 #include <SalomeApp_Tools.h>
82 #include <SUIT_Session.h>
83 #include <SUIT_ResourceMgr.h>
85 IMPLEMENT_STANDARD_HANDLE (GEOM_AISShape, SALOME_AISShape)
86 IMPLEMENT_STANDARD_RTTIEXT(GEOM_AISShape, SALOME_AISShape)
88 GEOM_AISShape::TopLevelDispMode GEOM_AISShape::myTopLevelDm = GEOM_AISShape::TopKeepCurrent;
89 Quantity_Color GEOM_AISShape::myTopLevelColor;
92 static void getEntityOwners( const Handle(AIS_InteractiveObject)& theObj,
93 const Handle(AIS_InteractiveContext)& theIC,
94 SelectMgr_IndexedMapOfOwner& theMap )
96 if ( theObj.IsNull() || theIC.IsNull() )
99 TColStd_ListOfInteger modes;
100 theIC->ActivatedModes( theObj, modes );
102 TColStd_ListIteratorOfListOfInteger itr( modes );
103 for (; itr.More(); itr.Next() ) {
105 if ( !theObj->HasSelection( m ) )
108 Handle(SelectMgr_Selection) sel = theObj->Selection( m );
110 for ( sel->Init(); sel->More(); sel->Next() ) {
111 #if OCC_VERSION_LARGE > 0x06080100
112 const SelectMgr_HSensitiveEntity aHSenEntity = sel->Sensitive();
113 if( aHSenEntity.IsNull() )
116 Handle(SelectBasics_SensitiveEntity) entity = aHSenEntity->BaseSensitive();
118 Handle(SelectBasics_SensitiveEntity) entity = sel->Sensitive();
120 if ( entity.IsNull() )
122 Handle(SelectMgr_EntityOwner) owner =
123 Handle(SelectMgr_EntityOwner)::DownCast(entity->OwnerId());
124 if ( !owner.IsNull() )
130 static void indicesToOwners( const TColStd_IndexedMapOfInteger& aIndexMap,
131 const TopoDS_Shape& aMainShape,
132 const SelectMgr_IndexedMapOfOwner& anAllMap,
133 SelectMgr_IndexedMapOfOwner& aToHiliteMap )
135 TopTools_IndexedMapOfShape aMapOfShapes;
136 TopExp::MapShapes(aMainShape, aMapOfShapes);
138 for ( Standard_Integer i = 1, n = anAllMap.Extent(); i <= n; i++ ) {
139 Handle(StdSelect_BRepOwner) anOwner = Handle(StdSelect_BRepOwner)::DownCast(anAllMap( i ));
140 if ( anOwner.IsNull() || !anOwner->HasShape() )
143 const TopoDS_Shape& aSubShape = anOwner->Shape();
144 Standard_Integer aSubShapeId = aMapOfShapes.FindIndex( aSubShape );
145 if ( !aSubShapeId || !aIndexMap.Contains( aSubShapeId ) )
148 if ( !aToHiliteMap.Contains( anOwner ) )
149 aToHiliteMap.Add( anOwner );
153 GEOM_AISShape::GEOM_AISShape(const TopoDS_Shape& shape,
154 const Standard_CString aName)
155 : SALOME_AISShape(shape),
157 myDisplayVectors(false),
158 myDisplayVertices(false),
159 myFieldDataType(GEOM::FDT_Double),
161 myFieldStepRangeMin(0),
162 myFieldStepRangeMax(0)
164 //rnv: Commented to avoid bug with local selection
165 //SetHilightMode( CustomHighlight ); // override setting the mode to 0 inside AIS_Shape constructor
167 myShadingColor = Quantity_Color( Quantity_NOC_GOLDENROD );
168 myPrevDisplayMode = 0;
170 myEdgesInShadingColor = Quantity_Color( Quantity_NOC_GOLDENROD );
172 myTopLevel = Standard_False;
173 Graphic3d_MaterialAspect aMatAspect;
174 if ( !HasMaterial() ) {
175 aMatAspect.SetAmbient( 0.5 );
176 aMatAspect.SetDiffuse( 0.5 );
177 aMatAspect.SetEmissive( 0.5 );
178 aMatAspect.SetShininess(0.5 );
179 aMatAspect.SetSpecular( 0.5 );
181 myDrawer->ShadingAspect()->Aspect()->SetFrontMaterial(aMatAspect);
182 myDrawer->ShadingAspect()->Aspect()->SetBackMaterial(aMatAspect);
186 GEOM_AISShape::~GEOM_AISShape()
190 void GEOM_AISShape::setIO(const Handle(SALOME_InteractiveObject)& io){
194 Handle(SALOME_InteractiveObject) GEOM_AISShape::getIO(){
195 Handle(SALOME_InteractiveObject) IO;
196 if ( !GetOwner().IsNull() )
197 IO = Handle(SALOME_InteractiveObject)::DownCast( GetOwner() );
201 Standard_Boolean GEOM_AISShape::hasIO(){
202 return !getIO().IsNull();
205 void GEOM_AISShape::setName(const Standard_CString aName)
209 Handle(SALOME_InteractiveObject) IO = getIO();
214 Standard_CString GEOM_AISShape::getName(){
215 return myName.ToCString();
218 void GEOM_AISShape::Compute(const Handle(PrsMgr_PresentationManager3d)& aPresentationManager,
219 const Handle(Prs3d_Presentation)& aPrs,
220 const Standard_Integer aMode)
222 if (IsInfinite()) aPrs->SetInfiniteState(Standard_True); //pas de prise en compte lors du FITALL
224 Handle(AIS_InteractiveContext) anIC = GetContext();
225 // AKL: use old behavior to avoid keeping object's wireframe
226 // if to change shape properties (for example: 'Clear Top Level State','Color', 'Isos')
227 // calling popup menu over(!) the shape in OCC viewer.
228 anIC->SetToHilightSelected( false );
230 bool anIsField = !myFieldStepData.isEmpty();
231 bool anIsColorField = anIsField && myFieldDataType != GEOM::FDT_String;
232 bool anIsTextField = anIsField && myFieldDataType == GEOM::FDT_String;
234 #if OCC_VERSION_LARGE > 0x06070200
235 if (isShowVertices())
236 myDrawer->SetVertexDrawMode(Prs3d_VDM_All);
239 // StdSelect_DisplayMode d = (StdSelect_DisplayMode) aMode;
240 bool isTopLev = isTopLevel() && switchTopLevel();
243 case CustomHighlight:
246 SetColor(topLevelColor());
247 Handle(Prs3d_LineAspect) anAspect = Attributes()->WireAspect();
248 anAspect->SetColor( topLevelColor() );
249 Attributes()->SetWireAspect( anAspect );
251 if( !isTopLev && anIsColorField && myFieldDimension == 1 )
252 drawField( aPrs, false, aMode == CustomHighlight );
254 StdPrs_WFDeflectionShape::Add(aPrs,myshape,myDrawer);
259 shadingMode(aPresentationManager, aPrs, aMode);
262 case ShadingWithEdges:
264 myDrawer->SetFaceBoundaryDraw( Standard_True );
265 shadingMode(aPresentationManager, aPrs, Shading);
266 if( anIsColorField && myFieldDimension == 1 ) {
267 myDrawer->SetFaceBoundaryDraw( Standard_False );
274 #ifdef USE_TEXTURED_SHAPE
275 AIS_TexturedShape::Compute(aPresentationManager, aPrs, aMode);
277 AIS_Shape::Compute(aPresentationManager, aPrs, aMode);
283 const bool isVector = IsKind(STANDARD_TYPE(GEOM_AISVector));
284 TopExp_Explorer Exp ( myshape, TopAbs_EDGE );
285 for ( ; Exp.More(); Exp.Next() ) {
286 TopoDS_Vertex aV1, aV2;
287 TopoDS_Edge anEdgeE = TopoDS::Edge(Exp.Current());
289 // draw curve direction (issue 0021087)
290 anEdgeE.Orientation( TopAbs_FORWARD );
292 if ( anEdgeE.IsNull() ) continue;
294 TopExp::Vertices(anEdgeE, aV1, aV2);
295 gp_Pnt aP1 = BRep_Tool::Pnt(aV1);
296 gp_Pnt aP2 = BRep_Tool::Pnt(aV2);
300 Handle(Geom_Curve) C = BRep_Tool::Curve(anEdgeE,fp,lp);
302 if ( C.IsNull() ) continue;
304 if ( anEdgeE.Orientation() == TopAbs_FORWARD )
305 C->D1(lp, aP2, aDirVec);
307 C->D1(fp, aP1, aDirVec);
311 GeomAdaptor_Curve aAdC;
312 aAdC.Load(C, fp, lp);
313 Standard_Real aDist = GCPnts_AbscissaPoint::Length(aAdC, fp, lp);
315 if (aDist > gp::Resolution()) {
317 if ( anEdgeE.Orientation() == TopAbs_FORWARD )
322 Prs3d_Arrow::Draw(aPrs, aP2, aDir, M_PI/180.*5., aDist/10.);
327 // draw color field on vertices
328 if( anIsColorField && myFieldDimension == 0 && aMode != CustomHighlight )
333 drawField( aPrs, true );
338 // aPrs->ReCompute(); // for hidden line recomputation if necessary...
341 void GEOM_AISShape::SetShadingColor(const Quantity_Color &aCol)
343 myShadingColor = aCol;
346 void GEOM_AISShape::SetEdgesInShadingColor(const Quantity_Color &aCol)
348 myEdgesInShadingColor = aCol;
351 void GEOM_AISShape::SetLabelColor(const Quantity_Color &aCol) {
355 void GEOM_AISShape::highlightSubShapes(const TColStd_IndexedMapOfInteger& aIndexMap,
356 const Standard_Boolean aHighlight )
358 Handle(AIS_InteractiveObject) anObj = this;
359 Handle(AIS_InteractiveContext) anIC = GetContext();
360 if ( anIC.IsNull() || !anIC->HasOpenedContext() )
363 Standard_Boolean isAutoHilight = anIC->AutomaticHilight();
364 anIC->SetAutomaticHilight( false );
366 anIC->ClearSelected( false );
369 SelectMgr_IndexedMapOfOwner anAllMap, aToHiliteMap;
371 // Get entity owners for all activated selection modes
372 getEntityOwners( anObj, anIC, anAllMap );
374 // Convert <aIndexMap> into the map of owners to highlight/unhighlight
375 indicesToOwners( aIndexMap, Shape(), anAllMap, aToHiliteMap );
378 for ( Standard_Integer i = 1, n = aToHiliteMap.Extent(); i <= n; i++ )
379 anIC->AddOrRemoveSelected( aToHiliteMap( i ), false );
382 anIC->SetAutomaticHilight( isAutoHilight );
383 anIC->HilightSelected( false );
386 void GEOM_AISShape::SetDisplayVectors(bool isDisplayed)
388 myDisplayVectors = isDisplayed;
391 void GEOM_AISShape::SetDisplayVertices(bool isDisplayed)
393 myDisplayVertices = isDisplayed;
396 void GEOM_AISShape::SetDisplayName(bool isDisplayed)
398 myDisplayName = isDisplayed;
401 void GEOM_AISShape::shadingMode(const Handle(PrsMgr_PresentationManager3d)& aPresentationManager,
402 const Handle(Prs3d_Presentation)& aPrs,
403 const Standard_Integer aMode)
405 myDrawer->ShadingAspect()->Aspect()->SetDistinguishOn();
407 Graphic3d_MaterialAspect aMatAspect(Graphic3d_NOM_PLASTIC);
408 aMatAspect.SetTransparency(Transparency());
409 Graphic3d_MaterialAspect currentFrontMaterial = myDrawer->ShadingAspect()->Aspect()->FrontMaterial();
410 Graphic3d_MaterialAspect currentBackMaterial = myDrawer->ShadingAspect()->Aspect()->BackMaterial();
411 myDrawer->ShadingAspect()->Aspect()->SetFrontMaterial( isTopLevel() ? aMatAspect : currentFrontMaterial );
412 myDrawer->ShadingAspect()->Aspect()->SetBackMaterial ( isTopLevel() ? aMatAspect : currentBackMaterial );
414 if( isTopLevel() && switchTopLevel() )
415 myDrawer->ShadingAspect()->SetColor( topLevelColor() );
417 if(myDrawer->ShadingAspect()->Aspect()->FrontMaterial().MaterialType( Graphic3d_MATERIAL_ASPECT ))
418 myDrawer->ShadingAspect()->SetColor(myShadingColor);
420 myDrawer->ShadingAspect()->SetColor(myDrawer->ShadingAspect()->Aspect()->FrontMaterial().AmbientColor());
423 bool anIsColorField = !myFieldStepData.isEmpty() && myFieldDataType != GEOM::FDT_String;
424 if( anIsColorField && ( myFieldDimension == 2 || myFieldDimension == 3 || myFieldDimension == -1 ) )
430 // PAL12113: AIS_Shape::Compute() works correctly with shapes containing no faces
431 //StdPrs_ShadedShape::Add(aPrs,myshape,myDrawer);
432 #ifdef USE_TEXTURED_SHAPE
433 AIS_TexturedShape::Compute(aPresentationManager, aPrs, aMode);
435 AIS_Shape::Compute(aPresentationManager, aPrs, aMode);
440 Standard_Boolean GEOM_AISShape::isTopLevel() {
444 void GEOM_AISShape::setTopLevel(Standard_Boolean f) {
447 myPrevDisplayMode = DisplayMode();
449 switch(topLevelDisplayMode()) {
450 case TopWireFrame : dm = Wireframe; break;
451 case TopShading : dm = Shading; break;
452 case TopShadingWithEdges : dm = ShadingWithEdges; break;
453 default : dm = myPrevDisplayMode; break;
458 SetDisplayMode(myPrevDisplayMode);
463 void GEOM_AISShape::setPrevDisplayMode(const Standard_Integer mode) {
464 myPrevDisplayMode = mode;
467 Quantity_Color GEOM_AISShape::topLevelColor() {
468 return myTopLevelColor;
471 void GEOM_AISShape::setTopLevelColor(const Quantity_Color c) {
475 GEOM_AISShape::TopLevelDispMode GEOM_AISShape::topLevelDisplayMode() {
478 void GEOM_AISShape::setTopLevelDisplayMode(const GEOM_AISShape::TopLevelDispMode dm) {
482 Standard_Boolean GEOM_AISShape::switchTopLevel() {
483 return myTopLevelDm != TopShowAdditionalWActor;
486 Standard_Boolean GEOM_AISShape::toActivate() {
487 return ( myTopLevel && myTopLevelDm == TopShowAdditionalWActor ) ? false : true;
490 void GEOM_AISShape::setFieldStepInfo( const GEOM::field_data_type theFieldDataType,
491 const int theFieldDimension,
492 const QList<QVariant>& theFieldStepData,
493 const TCollection_AsciiString& theFieldStepName,
494 const double theFieldStepRangeMin,
495 const double theFieldStepRangeMax )
497 myFieldDataType = theFieldDataType;
498 myFieldDimension = theFieldDimension;
499 myFieldStepData = theFieldStepData;
500 myFieldStepName = theFieldStepName;
501 myFieldStepRangeMin = theFieldStepRangeMin;
502 myFieldStepRangeMax = theFieldStepRangeMax;
505 void GEOM_AISShape::getFieldStepInfo( GEOM::field_data_type& theFieldDataType,
506 int& theFieldDimension,
507 QList<QVariant>& theFieldStepData,
508 TCollection_AsciiString& theFieldStepName,
509 double& theFieldStepRangeMin,
510 double& theFieldStepRangeMax ) const
512 theFieldDataType = myFieldDataType;
513 theFieldDimension = myFieldDimension;
514 theFieldStepData = myFieldStepData;
515 theFieldStepName = myFieldStepName;
516 theFieldStepRangeMin = myFieldStepRangeMin;
517 theFieldStepRangeMax = myFieldStepRangeMax;
520 void GEOM_AISShape::drawField( const Handle(Prs3d_Presentation)& thePrs,
521 const bool theIsString,
522 const bool theIsHighlight )
524 if( myFieldStepData.isEmpty() )
527 QListIterator<QVariant> aFieldStepDataIter( myFieldStepData );
529 TopAbs_ShapeEnum aShapeType = TopAbs_SHAPE;
530 switch( myFieldDimension )
532 case 0: aShapeType = TopAbs_VERTEX; break;
533 case 1: aShapeType = TopAbs_EDGE; break;
534 case 2: aShapeType = TopAbs_FACE; break;
535 case 3: aShapeType = TopAbs_SOLID; break;
536 case -1: aShapeType = TopAbs_VERTEX; break;
539 TopTools_IndexedMapOfShape aShapeMap;
540 TopExp::MapShapes( myshape, aShapeMap );
542 TColStd_IndexedMapOfInteger anIndexMap;
544 TopExp_Explorer anExp;
545 for( anExp.Init( myshape, aShapeType ); anExp.More(); anExp.Next() )
547 TopoDS_Shape aSubShape = anExp.Current();
548 if( !aSubShape.IsNull() )
550 Standard_Integer aSubShapeIndex = aShapeMap.FindIndex( aSubShape );
551 if( anIndexMap.Contains( aSubShapeIndex ) )
554 anIndexMap.Add( aSubShapeIndex );
556 Handle(Graphic3d_Group) aGroup = Prs3d_Root::NewGroup( thePrs );
560 if( aFieldStepDataIter.hasNext() )
562 const QVariant& aVariant = aFieldStepDataIter.next();
564 aString = aVariant.toString();
566 aQColor = aVariant.value<QColor>();
574 if( computeMassCenter( aSubShape, aCenter ) )
576 Graphic3d_Vertex aVertex( aCenter.X(), aCenter.Y(), aCenter.Z() );
578 Handle(Graphic3d_AspectText3d) anAspectText3d = new Graphic3d_AspectText3d();
579 anAspectText3d->SetStyle( Aspect_TOST_ANNOTATION );
580 anAspectText3d->SetColor( myLabelColor );
581 aGroup->SetPrimitivesAspect( anAspectText3d );
583 aGroup->Text( aString.toLatin1().constData(), aVertex, 14 );
588 Quantity_Color aColor( aQColor.redF(), aQColor.greenF(), aQColor.blueF(), Quantity_TOC_RGB );
589 if( myFieldDimension == 0 )
591 TopoDS_Vertex aVertexShape = TopoDS::Vertex( aSubShape );
592 if( !aVertexShape.IsNull() )
594 gp_Pnt aPnt = BRep_Tool::Pnt( aVertexShape );
596 Handle(Graphic3d_AspectMarker3d) anAspectMarker3d = new Graphic3d_AspectMarker3d();
597 anAspectMarker3d->SetColor( aColor );
598 anAspectMarker3d->SetType( Aspect_TOM_POINT );
599 anAspectMarker3d->SetScale( 10.0 );
600 aGroup->SetPrimitivesAspect( anAspectMarker3d );
602 Handle(Graphic3d_ArrayOfPoints) anArray = new Graphic3d_ArrayOfPoints( 1 );
603 anArray->AddVertex( aPnt.X(), aPnt.Y(), aPnt.Z() );
605 aGroup->AddPrimitiveArray( anArray );
608 else if( myFieldDimension == 1 )
610 myDrawer->WireAspect()->SetColor( aColor );
612 myDrawer->WireAspect()->SetWidth( myOwnWidth );
614 myDrawer->WireAspect()->SetWidth( myOwnWidth + 4 );
615 StdPrs_WFDeflectionShape::Add( thePrs, aSubShape, myDrawer );
617 else if( myFieldDimension == 2 ||
618 myFieldDimension == 3 ||
619 myFieldDimension == -1 )
621 myDrawer->ShadingAspect()->SetColor( aColor );
622 StdPrs_ShadedShape::Add( thePrs, aSubShape, myDrawer );
629 void GEOM_AISShape::drawName( const Handle(Prs3d_Presentation)& thePrs )
631 Handle(Graphic3d_Group) aGroup = Prs3d_Root::NewGroup( thePrs );
633 gp_Ax3 anAx3 = GEOMUtils::GetPosition(myshape);
634 gp_Pnt aCenter = anAx3.Location();
636 Graphic3d_Vertex aVertex( aCenter.X(), aCenter.Y(), aCenter.Z() );
638 Handle(Graphic3d_AspectText3d) anAspectText3d = new Graphic3d_AspectText3d();
639 anAspectText3d->SetStyle( Aspect_TOST_ANNOTATION );
640 anAspectText3d->SetColor( myLabelColor );
641 aGroup->SetPrimitivesAspect( anAspectText3d );
643 const char* aName = getIO()->getName();
644 aGroup->Text( TCollection_ExtendedString( aName ), aVertex, 16 );
647 Standard_Boolean GEOM_AISShape::computeMassCenter( const TopoDS_Shape& theShape,
650 Standard_Real aX = 0, aY = 0, aZ = 0;
651 Standard_Integer aNbPoints = 0;
653 TopExp_Explorer anExp;
654 for( anExp.Init( theShape, TopAbs_VERTEX ); anExp.More(); anExp.Next() )
656 TopoDS_Vertex aVertex = TopoDS::Vertex( anExp.Current() );
657 if( !aVertex.IsNull() )
659 gp_Pnt aPnt = BRep_Tool::Pnt( aVertex );
668 return Standard_False;
670 theCenter.SetCoord( aX / (Standard_Real)aNbPoints,
671 aY / (Standard_Real)aNbPoints,
672 aZ / (Standard_Real)aNbPoints );
673 return Standard_True;