geom/src/OBJECT/GEOM_AISShape.cxx

701 lines
22 KiB
C++
Raw Normal View History

2013-04-01 18:25:01 +06:00
// Copyright (C) 2007-2013 CEA/DEN, EDF R&D, OPEN CASCADE
2009-02-13 17:16:39 +05:00
//
2012-08-09 13:58:02 +06:00
// Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
2003-07-09 20:44:57 +06:00
//
2012-08-09 13:58:02 +06:00
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License.
2003-07-09 20:44:57 +06:00
//
2012-08-09 13:58:02 +06:00
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// Lesser General Public License for more details.
2003-07-09 20:44:57 +06:00
//
2012-08-09 13:58:02 +06:00
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
2003-07-09 20:44:57 +06:00
//
2012-08-09 13:58:02 +06:00
// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
2009-02-13 17:16:39 +05:00
//
2012-08-09 13:58:02 +06:00
2009-02-13 17:16:39 +05:00
// GEOM OBJECT : interactive object for Geometry entities visualization
2003-07-09 20:44:57 +06:00
// File : GEOM_AISShape.cxx
// Author : Nicolas REJNERI
// Module : GEOM
2012-08-09 13:58:02 +06:00
2003-05-12 21:09:12 +06:00
/*!
\class GEOM_AISShape GEOM_AISShape.hxx
\brief ....
*/
#include "GEOM_AISShape.ixx"
#include "SALOME_InteractiveObject.hxx"
2012-08-09 13:58:02 +06:00
#include "GEOM_AISVector.hxx"
2003-05-12 21:09:12 +06:00
// Open CASCADE Includes
#include <AIS_Drawer.hxx>
2004-12-01 15:39:14 +05:00
#include <AIS_InteractiveContext.hxx>
2012-08-09 13:58:02 +06:00
#include <BRep_Tool.hxx>
#include <GCPnts_AbscissaPoint.hxx>
#include <GeomAdaptor_Curve.hxx>
#include <gp_Pnt.hxx>
#include <gp_Dir.hxx>
#include <gp_Vec.hxx>
2004-12-01 15:39:14 +05:00
#include <Graphic3d_AspectFillArea3d.hxx>
2012-08-09 13:58:02 +06:00
#include <Graphic3d_AspectLine3d.hxx>
#include <Graphic3d_AspectMarker3d.hxx>
#include <Graphic3d_AspectText3d.hxx>
2012-08-09 13:58:02 +06:00
2003-05-12 21:09:12 +06:00
#include <Prs3d_ShadingAspect.hxx>
2012-08-09 13:58:02 +06:00
#include <Prs3d_Arrow.hxx>
#include <Prs3d_IsoAspect.hxx>
2004-12-01 15:39:14 +05:00
#include <SelectBasics_SensitiveEntity.hxx>
#include <SelectMgr_EntityOwner.hxx>
#include <StdSelect_BRepOwner.hxx>
2004-12-01 15:39:14 +05:00
#include <SelectMgr_IndexedMapOfOwner.hxx>
#include <SelectMgr_Selection.hxx>
2003-05-12 21:09:12 +06:00
#include <StdSelect_DisplayMode.hxx>
#include <StdPrs_ShadedShape.hxx>
2012-08-09 13:58:02 +06:00
#include <StdPrs_WFDeflectionShape.hxx>
2004-12-01 15:39:14 +05:00
#include <TColStd_IndexedMapOfInteger.hxx>
#include <TColStd_ListIteratorOfListOfInteger.hxx>
#include <TColStd_ListOfInteger.hxx>
#include <TopExp.hxx>
#include <TopoDS_Shape.hxx>
#include <TopTools_IndexedMapOfShape.hxx>
2012-08-09 13:58:02 +06:00
#include <TopExp.hxx>
#include <TopExp_Explorer.hxx>
#include <TopoDS.hxx>
#include <TopoDS_Edge.hxx>
#include <TopoDS_Shape.hxx>
#include <TopoDS_Vertex.hxx>
#include <TColStd_SequenceOfInteger.hxx>
#include <V3d_View.hxx>
#include <SalomeApp_Tools.h>
#include <SUIT_Session.h>
#include <SUIT_ResourceMgr.h>
GEOM_AISShape::TopLevelDispMode GEOM_AISShape::myTopLevelDm = GEOM_AISShape::TopKeepCurrent;
Quantity_Color GEOM_AISShape::myTopLevelColor;
2004-12-01 15:39:14 +05:00
2005-08-11 10:43:57 +06:00
2004-12-01 15:39:14 +05:00
static void getEntityOwners( const Handle(AIS_InteractiveObject)& theObj,
2012-08-09 13:58:02 +06:00
const Handle(AIS_InteractiveContext)& theIC,
SelectMgr_IndexedMapOfOwner& theMap )
2004-12-01 15:39:14 +05:00
{
if ( theObj.IsNull() || theIC.IsNull() )
return;
2003-05-12 21:09:12 +06:00
2004-12-01 15:39:14 +05:00
TColStd_ListOfInteger modes;
theIC->ActivatedModes( theObj, modes );
TColStd_ListIteratorOfListOfInteger itr( modes );
for (; itr.More(); itr.Next() ) {
int m = itr.Value();
if ( !theObj->HasSelection( m ) )
continue;
Handle(SelectMgr_Selection) sel = theObj->Selection( m );
for ( sel->Init(); sel->More(); sel->Next() ) {
Handle(SelectBasics_SensitiveEntity) entity = sel->Sensitive();
if ( entity.IsNull() )
2012-08-09 13:58:02 +06:00
continue;
2004-12-01 15:39:14 +05:00
Handle(SelectMgr_EntityOwner) owner =
2012-08-09 13:58:02 +06:00
Handle(SelectMgr_EntityOwner)::DownCast(entity->OwnerId());
2004-12-01 15:39:14 +05:00
if ( !owner.IsNull() )
2012-08-09 13:58:02 +06:00
theMap.Add( owner );
2004-12-01 15:39:14 +05:00
}
}
}
static void indicesToOwners( const TColStd_IndexedMapOfInteger& aIndexMap,
2012-08-09 13:58:02 +06:00
const TopoDS_Shape& aMainShape,
const SelectMgr_IndexedMapOfOwner& anAllMap,
SelectMgr_IndexedMapOfOwner& aToHiliteMap )
2004-12-01 15:39:14 +05:00
{
TopTools_IndexedMapOfShape aMapOfShapes;
TopExp::MapShapes(aMainShape, aMapOfShapes);
for ( Standard_Integer i = 1, n = anAllMap.Extent(); i <= n; i++ ) {
Handle(StdSelect_BRepOwner) anOwner = Handle(StdSelect_BRepOwner)::DownCast(anAllMap( i ));
2004-12-01 15:39:14 +05:00
if ( anOwner.IsNull() || !anOwner->HasShape() )
continue;
const TopoDS_Shape& aSubShape = anOwner->Shape();
Standard_Integer aSubShapeId = aMapOfShapes.FindIndex( aSubShape );
if ( !aSubShapeId || !aIndexMap.Contains( aSubShapeId ) )
continue;
if ( !aToHiliteMap.Contains( anOwner ) )
aToHiliteMap.Add( anOwner );
}
}
2003-05-12 21:09:12 +06:00
GEOM_AISShape::GEOM_AISShape(const TopoDS_Shape& shape,
2012-08-09 13:58:02 +06:00
const Standard_CString aName)
: SALOME_AISShape(shape),
myName(aName),
myDisplayVectors(false),
myFieldDataType(GEOM::FDT_Double),
myFieldDimension(0),
myFieldStepRangeMin(0),
myFieldStepRangeMax(0)
2003-05-12 21:09:12 +06:00
{
SetHilightMode( CustomHighlight ); // override setting the mode to 0 inside AIS_Shape constructor
2003-05-12 21:09:12 +06:00
myShadingColor = Quantity_Color( Quantity_NOC_GOLDENROD );
2012-08-09 13:58:02 +06:00
myPrevDisplayMode = 0;
storeBoundaryColors();
myEdgesInShadingColor = Quantity_Color( Quantity_NOC_GOLDENROD );
myUIsoNumber = -1;
myVIsoNumber = -1;
myTopLevel = Standard_False;
Graphic3d_MaterialAspect aMatAspect;
if ( !HasMaterial() ) {
2013-06-17 18:15:50 +06:00
aMatAspect.SetAmbient( 0.5 );
aMatAspect.SetDiffuse( 0.5 );
aMatAspect.SetEmissive( 0.5 );
aMatAspect.SetShininess(0.5 );
aMatAspect.SetSpecular( 0.5 );
myDrawer->ShadingAspect()->Aspect()->SetFrontMaterial(aMatAspect);
myDrawer->ShadingAspect()->Aspect()->SetBackMaterial(aMatAspect);
2012-08-09 13:58:02 +06:00
}
2003-05-12 21:09:12 +06:00
}
void GEOM_AISShape::setIO(const Handle(SALOME_InteractiveObject)& io){
2005-03-17 19:02:35 +05:00
SetOwner( io );
2003-05-12 21:09:12 +06:00
}
Handle(SALOME_InteractiveObject) GEOM_AISShape::getIO(){
2005-03-17 19:02:35 +05:00
Handle(SALOME_InteractiveObject) IO;
if ( !GetOwner().IsNull() )
IO = Handle(SALOME_InteractiveObject)::DownCast( GetOwner() );
return IO;
2003-05-12 21:09:12 +06:00
}
Standard_Boolean GEOM_AISShape::hasIO(){
2005-03-17 19:02:35 +05:00
return !getIO().IsNull();
2003-05-12 21:09:12 +06:00
}
void GEOM_AISShape::setName(const Standard_CString aName)
{
myName = aName;
2003-05-12 21:09:12 +06:00
2005-03-17 19:02:35 +05:00
Handle(SALOME_InteractiveObject) IO = getIO();
if ( !IO.IsNull() )
IO->setName(aName);
2003-05-12 21:09:12 +06:00
}
Standard_CString GEOM_AISShape::getName(){
return myName.ToCString();
2003-05-12 21:09:12 +06:00
}
void GEOM_AISShape::Compute(const Handle(PrsMgr_PresentationManager3d)& aPresentationManager,
2012-08-09 13:58:02 +06:00
const Handle(Prs3d_Presentation)& aPrs,
const Standard_Integer aMode)
2003-05-12 21:09:12 +06:00
{
if (IsInfinite()) aPrs->SetInfiniteState(Standard_True); //pas de prise en compte lors du FITALL
2012-08-09 13:58:02 +06:00
Handle(AIS_InteractiveContext) anIC = GetContext();
bool anIsField = !myFieldStepData.isEmpty();
bool anIsColorField = anIsField && myFieldDataType != GEOM::FDT_String;
bool anIsTextField = anIsField && myFieldDataType == GEOM::FDT_String;
2012-08-09 13:58:02 +06:00
// StdSelect_DisplayMode d = (StdSelect_DisplayMode) aMode;
2013-02-12 17:35:16 +06:00
bool isTopLev = isTopLevel() && switchTopLevel();
2012-08-09 13:58:02 +06:00
switch (aMode) {
case 0://StdSelect_DM_Wireframe:
case CustomHighlight:
2003-05-12 21:09:12 +06:00
{
2012-08-09 13:58:02 +06:00
restoreIsoNumbers();
// Restore wireframe edges colors
restoreBoundaryColors();
if(isTopLev) {
2013-06-17 18:15:50 +06:00
SetColor(topLevelColor());
Handle(Prs3d_LineAspect) anAspect = Attributes()->WireAspect();
anAspect->SetColor( topLevelColor() );
Attributes()->SetWireAspect( anAspect );
2012-08-09 13:58:02 +06:00
}
if( !isTopLev && anIsColorField && myFieldDimension == 1 )
drawField( aPrs, false, aMode == CustomHighlight );
else
StdPrs_WFDeflectionShape::Add(aPrs,myshape,myDrawer);
2003-05-12 21:09:12 +06:00
break;
}
2012-08-09 13:58:02 +06:00
case 1://StdSelect_DM_Shading:
2003-05-12 21:09:12 +06:00
{
2012-08-09 13:58:02 +06:00
restoreIsoNumbers();
shadingMode(aPresentationManager, aPrs, aMode);
// Store wireframe edges colors
storeBoundaryColors();
break;
}
case 3: //StdSelect_DM_HLR:
{
if(!isTopLev)
2013-06-17 18:15:50 +06:00
AIS_TexturedShape::Compute(aPresentationManager, aPrs, aMode);
2012-08-09 13:58:02 +06:00
else
2013-06-17 18:15:50 +06:00
shadingMode(aPresentationManager, aPrs, AIS_Shaded);
2012-08-09 13:58:02 +06:00
break;
}
}
2003-05-12 21:09:12 +06:00
2012-08-09 13:58:02 +06:00
if ( aMode == ShadingWithEdges ) {
// Temporary store number of iso lines in order to recover its later
// when display mode is achnged to 'Wirefame' or 'Shading'.
// Iso lines are not displayed in 'Shading with edges' mode.
storeIsoNumbers();
2003-05-12 21:09:12 +06:00
2012-08-09 13:58:02 +06:00
// Reset number of iso lines to 0
resetIsoNumbers();
//Shaded faces
shadingMode(aPresentationManager, aPrs, AIS_Shaded);
// Store wireframe edges colors
storeBoundaryColors();
// Coloring edges
Handle(Prs3d_LineAspect) anAspect = myDrawer->UnFreeBoundaryAspect();
anAspect->SetColor( myEdgesInShadingColor );
myDrawer->SetUnFreeBoundaryAspect( anAspect );
anAspect = myDrawer->FreeBoundaryAspect();
anAspect->SetColor( myEdgesInShadingColor );
myDrawer->SetFreeBoundaryAspect( anAspect );
// Add edges to presentation
if( anIsColorField && myFieldDimension == 1 )
drawField( aPrs );
else
StdPrs_WFDeflectionShape::Add(aPrs,myshape,myDrawer);
2012-08-09 13:58:02 +06:00
}
if (isShowVectors())
{
const bool isVector = IsKind(STANDARD_TYPE(GEOM_AISVector));
TopExp_Explorer Exp ( myshape, TopAbs_EDGE );
for ( ; Exp.More(); Exp.Next() ) {
TopoDS_Vertex aV1, aV2;
TopoDS_Edge anEdgeE = TopoDS::Edge(Exp.Current());
if ( !isVector )
// draw curve direction (issue 0021087)
anEdgeE.Orientation( TopAbs_FORWARD );
if ( anEdgeE.IsNull() ) continue;
TopExp::Vertices(anEdgeE, aV1, aV2);
gp_Pnt aP1 = BRep_Tool::Pnt(aV1);
gp_Pnt aP2 = BRep_Tool::Pnt(aV2);
double fp,lp;
gp_Vec aDirVec;
Handle(Geom_Curve) C = BRep_Tool::Curve(anEdgeE,fp,lp);
if ( C.IsNull() ) continue;
if ( anEdgeE.Orientation() == TopAbs_FORWARD )
C->D1(lp, aP2, aDirVec);
else {
C->D1(fp, aP1, aDirVec);
aP2 = aP1;
}
GeomAdaptor_Curve aAdC;
aAdC.Load(C, fp, lp);
Standard_Real aDist = GCPnts_AbscissaPoint::Length(aAdC, fp, lp);
if (aDist > gp::Resolution()) {
gp_Dir aDir;
if ( anEdgeE.Orientation() == TopAbs_FORWARD )
aDir = aDirVec;
else
aDir = -aDirVec;
Prs3d_Arrow::Draw(aPrs, aP2, aDir, M_PI/180.*5., aDist/10.);
}
2003-05-12 21:09:12 +06:00
}
}
// draw color field on vertices
if( anIsColorField && myFieldDimension == 0 && aMode != CustomHighlight )
drawField( aPrs );
// draw text field
if( anIsTextField )
drawField( aPrs, true );
2003-05-12 21:09:12 +06:00
// aPrs->ReCompute(); // for hidden line recomputation if necessary...
}
void GEOM_AISShape::SetShadingColor(const Quantity_Color &aCol)
{
myShadingColor = aCol;
}
2004-12-01 15:39:14 +05:00
2012-08-09 13:58:02 +06:00
void GEOM_AISShape::SetEdgesInShadingColor(const Quantity_Color &aCol)
{
myEdgesInShadingColor = aCol;
}
2004-12-01 15:39:14 +05:00
void GEOM_AISShape::highlightSubShapes(const TColStd_IndexedMapOfInteger& aIndexMap,
2012-08-09 13:58:02 +06:00
const Standard_Boolean aHighlight )
2004-12-01 15:39:14 +05:00
{
Handle(AIS_InteractiveObject) anObj = this;
Handle(AIS_InteractiveContext) anIC = GetContext();
if ( anIC.IsNull() || !anIC->HasOpenedContext() )
return;
Standard_Boolean isAutoHilight = anIC->AutomaticHilight();
anIC->SetAutomaticHilight( false );
anIC->ClearSelected( false );
if ( aHighlight ) {
SelectMgr_IndexedMapOfOwner anAllMap, aToHiliteMap;
// Get entity owners for all activated selection modes
getEntityOwners( anObj, anIC, anAllMap );
// Convert <aIndexMap> into the map of owners to highlight/unhighlight
indicesToOwners( aIndexMap, Shape(), anAllMap, aToHiliteMap );
for ( Standard_Integer i = 1, n = aToHiliteMap.Extent(); i <= n; i++ )
anIC->AddOrRemoveSelected( aToHiliteMap( i ), false );
}
anIC->SetAutomaticHilight( isAutoHilight );
anIC->HilightSelected( false );
}
2012-08-09 13:58:02 +06:00
void GEOM_AISShape::SetDisplayVectors(bool isDisplayed)
{
myDisplayVectors = isDisplayed;
}
void GEOM_AISShape::shadingMode(const Handle(PrsMgr_PresentationManager3d)& aPresentationManager,
2013-06-17 18:15:50 +06:00
const Handle(Prs3d_Presentation)& aPrs,
const Standard_Integer aMode)
2012-08-09 13:58:02 +06:00
{
myDrawer->ShadingAspect()->Aspect()->SetDistinguishOn();
Graphic3d_MaterialAspect aMatAspect(Graphic3d_NOM_PLASTIC);
2013-02-12 17:35:16 +06:00
aMatAspect.SetTransparency(Transparency());
Graphic3d_MaterialAspect currentFrontMaterial = myDrawer->ShadingAspect()->Aspect()->FrontMaterial();
Graphic3d_MaterialAspect currentBackMaterial = myDrawer->ShadingAspect()->Aspect()->BackMaterial();
myDrawer->ShadingAspect()->Aspect()->SetFrontMaterial( isTopLevel() ? aMatAspect : currentFrontMaterial );
myDrawer->ShadingAspect()->Aspect()->SetBackMaterial ( isTopLevel() ? aMatAspect : currentBackMaterial );
if( isTopLevel() && switchTopLevel() )
2012-08-09 13:58:02 +06:00
myDrawer->ShadingAspect()->SetColor( topLevelColor() );
else {
if(myDrawer->ShadingAspect()->Aspect()->FrontMaterial().MaterialType( Graphic3d_MATERIAL_ASPECT ))
myDrawer->ShadingAspect()->SetColor(myShadingColor);
else
myDrawer->ShadingAspect()->SetColor(myDrawer->ShadingAspect()->Aspect()->FrontMaterial().AmbientColor());
}
bool anIsColorField = !myFieldStepData.isEmpty() && myFieldDataType != GEOM::FDT_String;
if( anIsColorField && ( myFieldDimension == 2 || myFieldDimension == 3 || myFieldDimension == -1 ) )
{
drawField( aPrs );
}
else
{
// PAL12113: AIS_Shape::Compute() works correctly with shapes containing no faces
//StdPrs_ShadedShape::Add(aPrs,myshape,myDrawer);
AIS_Shape::Compute(aPresentationManager, aPrs, aMode);
}
2012-08-09 13:58:02 +06:00
}
void GEOM_AISShape::storeIsoNumbers()
{
myUIsoNumber = myDrawer->UIsoAspect()->Number();
myVIsoNumber = myDrawer->VIsoAspect()->Number();
}
void GEOM_AISShape::restoreIsoNumbers()
{
if ( myUIsoNumber > 0 ) {
// Restore number of U iso lines
Handle(Prs3d_IsoAspect) anAspect = myDrawer->UIsoAspect();
anAspect->SetNumber( myUIsoNumber );
myDrawer->SetUIsoAspect( anAspect );
}
if ( myVIsoNumber > 0 ) {
// Restore number of V iso lines
Handle(Prs3d_IsoAspect) anAspect = myDrawer->VIsoAspect();
anAspect->SetNumber( myVIsoNumber );
myDrawer->SetVIsoAspect( anAspect );
}
}
void GEOM_AISShape::resetIsoNumbers()
{
Handle(Prs3d_IsoAspect) anAspect = myDrawer->UIsoAspect();
anAspect->SetNumber( 0 );
myDrawer->SetUIsoAspect( anAspect );
anAspect = myDrawer->VIsoAspect();
anAspect->SetNumber( 0 );
myDrawer->SetVIsoAspect( anAspect );
}
void GEOM_AISShape::storeBoundaryColors()
{
Aspect_TypeOfLine aLT;
Standard_Real aW;
myDrawer->FreeBoundaryAspect()->Aspect()->Values( myFreeBoundaryColor, aLT, aW);
myDrawer->UnFreeBoundaryAspect()->Aspect()->Values( myUnFreeBoundaryColor, aLT, aW);
}
void GEOM_AISShape::restoreBoundaryColors()
{
Handle(Prs3d_LineAspect) anAspect = myDrawer->FreeBoundaryAspect();
anAspect->SetColor( myFreeBoundaryColor );
myDrawer->SetFreeBoundaryAspect( anAspect );
anAspect = myDrawer->UnFreeBoundaryAspect();
anAspect->SetColor( myUnFreeBoundaryColor );
myDrawer->SetUnFreeBoundaryAspect( anAspect );
}
Standard_Boolean GEOM_AISShape::isTopLevel() {
return myTopLevel;
}
void GEOM_AISShape::setTopLevel(Standard_Boolean f) {
if(f) {
if(f != myTopLevel)
myPrevDisplayMode = DisplayMode();
Standard_Integer dm;
switch(topLevelDisplayMode()) {
case TopKeepCurrent : dm = myPrevDisplayMode; break;
case TopWireFrame : dm = AIS_WireFrame; break;
case TopShadingWithEdges : dm = ShadingWithEdges; break;
default : dm = AIS_Shaded; break;
}
SetDisplayMode(dm);
} else {
if(f != myTopLevel)
SetDisplayMode(myPrevDisplayMode);
}
myTopLevel = f;
}
void GEOM_AISShape::setPrevDisplayMode(const Standard_Integer mode) {
myPrevDisplayMode = mode;
}
Quantity_Color GEOM_AISShape::topLevelColor() {
return myTopLevelColor;
}
void GEOM_AISShape::setTopLevelColor(const Quantity_Color c) {
myTopLevelColor = c;
}
GEOM_AISShape::TopLevelDispMode GEOM_AISShape::topLevelDisplayMode() {
return myTopLevelDm;
}
void GEOM_AISShape::setTopLevelDisplayMode(const GEOM_AISShape::TopLevelDispMode dm) {
myTopLevelDm = dm;
}
Standard_Boolean GEOM_AISShape::switchTopLevel() {
2013-06-17 18:15:50 +06:00
return myTopLevelDm != TopShowAdditionalWActor;
2012-08-09 13:58:02 +06:00
}
Standard_Boolean GEOM_AISShape::toActivate() {
2013-06-17 18:15:50 +06:00
return Standard_True;
2012-10-08 17:16:36 +06:00
}
void GEOM_AISShape::setFieldStepInfo( const GEOM::field_data_type theFieldDataType,
const int theFieldDimension,
const QList<QVariant>& theFieldStepData,
const TCollection_AsciiString& theFieldStepName,
const double theFieldStepRangeMin,
const double theFieldStepRangeMax )
{
myFieldDataType = theFieldDataType;
myFieldDimension = theFieldDimension;
myFieldStepData = theFieldStepData;
myFieldStepName = theFieldStepName;
myFieldStepRangeMin = theFieldStepRangeMin;
myFieldStepRangeMax = theFieldStepRangeMax;
}
void GEOM_AISShape::getFieldStepInfo( GEOM::field_data_type& theFieldDataType,
int& theFieldDimension,
QList<QVariant>& theFieldStepData,
TCollection_AsciiString& theFieldStepName,
double& theFieldStepRangeMin,
double& theFieldStepRangeMax ) const
{
theFieldDataType = myFieldDataType;
theFieldDimension = myFieldDimension;
theFieldStepData = myFieldStepData;
theFieldStepName = myFieldStepName;
theFieldStepRangeMin = myFieldStepRangeMin;
theFieldStepRangeMax = myFieldStepRangeMax;
}
void GEOM_AISShape::drawField( const Handle(Prs3d_Presentation)& thePrs,
const bool theIsString,
const bool theIsHighlight )
{
if( myFieldStepData.isEmpty() )
return;
QListIterator<QVariant> aFieldStepDataIter( myFieldStepData );
TopAbs_ShapeEnum aShapeType = TopAbs_SHAPE;
switch( myFieldDimension )
{
case 0: aShapeType = TopAbs_VERTEX; break;
case 1: aShapeType = TopAbs_EDGE; break;
case 2: aShapeType = TopAbs_FACE; break;
case 3: aShapeType = TopAbs_SOLID; break;
case -1: aShapeType = TopAbs_VERTEX; break;
}
TopTools_IndexedMapOfShape aShapeMap;
TopExp::MapShapes( myshape, aShapeMap );
TColStd_IndexedMapOfInteger anIndexMap;
TopExp_Explorer anExp;
for( anExp.Init( myshape, aShapeType ); anExp.More(); anExp.Next() )
{
TopoDS_Shape aSubShape = anExp.Current();
if( !aSubShape.IsNull() )
{
Standard_Integer aSubShapeIndex = aShapeMap.FindIndex( aSubShape );
if( anIndexMap.Contains( aSubShapeIndex ) )
continue;
anIndexMap.Add( aSubShapeIndex );
Handle(Graphic3d_Group) aGroup = Prs3d_Root::NewGroup( thePrs );
QColor aQColor;
QString aString;
if( aFieldStepDataIter.hasNext() )
{
const QVariant& aVariant = aFieldStepDataIter.next();
if( theIsString )
aString = aVariant.toString();
else
aQColor = aVariant.value<QColor>();
}
else
break;
if( theIsString )
{
gp_Pnt aCenter;
if( computeMassCenter( aSubShape, aCenter ) )
{
Graphic3d_Vertex aVertex( aCenter.X(), aCenter.Y(), aCenter.Z() );
Handle(Graphic3d_AspectText3d) anAspectText3d = new Graphic3d_AspectText3d();
anAspectText3d->SetStyle( Aspect_TOST_ANNOTATION );
aGroup->SetPrimitivesAspect( anAspectText3d );
aGroup->Text( aString.toLatin1().constData(), aVertex, 14 );
}
}
else
{
Quantity_Color aColor( aQColor.redF(), aQColor.greenF(), aQColor.blueF(), Quantity_TOC_RGB );
if( myFieldDimension == 0 )
{
TopoDS_Vertex aVertexShape = TopoDS::Vertex( aSubShape );
if( !aVertexShape.IsNull() )
{
gp_Pnt aPnt = BRep_Tool::Pnt( aVertexShape );
Handle(Graphic3d_AspectMarker3d) anAspectMarker3d = new Graphic3d_AspectMarker3d();
anAspectMarker3d->SetColor( aColor );
anAspectMarker3d->SetType( Aspect_TOM_POINT );
anAspectMarker3d->SetScale( 10.0 );
aGroup->SetPrimitivesAspect( anAspectMarker3d );
Handle(Graphic3d_ArrayOfPoints) anArray = new Graphic3d_ArrayOfPoints( 1 );
anArray->AddVertex( aPnt.X(), aPnt.Y(), aPnt.Z() );
aGroup->AddPrimitiveArray( anArray );
}
}
else if( myFieldDimension == 1 )
{
myDrawer->WireAspect()->SetColor( aColor );
if( theIsHighlight )
myDrawer->WireAspect()->SetWidth( myOwnWidth );
else
myDrawer->WireAspect()->SetWidth( myOwnWidth + 4 );
StdPrs_WFDeflectionShape::Add( thePrs, aSubShape, myDrawer );
}
else if( myFieldDimension == 2 ||
myFieldDimension == 3 ||
myFieldDimension == -1 )
{
myDrawer->ShadingAspect()->SetColor( aColor );
StdPrs_ShadedShape::Add( thePrs, aSubShape, myDrawer );
}
}
}
}
}
Standard_Boolean GEOM_AISShape::computeMassCenter( const TopoDS_Shape& theShape,
gp_Pnt& theCenter )
{
Standard_Real aX = 0, aY = 0, aZ = 0;
Standard_Integer aNbPoints = 0;
TopExp_Explorer anExp;
for( anExp.Init( theShape, TopAbs_VERTEX ); anExp.More(); anExp.Next() )
{
TopoDS_Vertex aVertex = TopoDS::Vertex( anExp.Current() );
if( !aVertex.IsNull() )
{
gp_Pnt aPnt = BRep_Tool::Pnt( aVertex );
aX += aPnt.X();
aY += aPnt.Y();
aZ += aPnt.Z();
aNbPoints++;
}
}
if( aNbPoints == 0 )
return Standard_False;
theCenter.SetCoord( aX / (Standard_Real)aNbPoints,
aY / (Standard_Real)aNbPoints,
aZ / (Standard_Real)aNbPoints );
return Standard_True;
}