OpenCascade/occt
1
0
Fork 0
mirror of https://git.dev.opencascade.org/repos/occt.git synced 2025-04-03 17:56:21 +03:00
Code Packages Releases Activity

GLTF Export - Edge and Vertex support #243

Browse Source 
Added functionality to export TopoDS_Vertex and TopoDS_Edge to GLTF format
This commit is contained in:
mzernova 2025-01-05 22:43:23 +00:00
parent 053e01ec68
commit 174b985fd5
12 changed files with 1367 additions and 436 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

760
src/RWGltf/RWGltf_CafWriter.cxx
View File

File diff suppressed because it is too large Load Diff

103
src/RWGltf/RWGltf_CafWriter.hxx
View File

@ -14,23 +14,30 @@
#ifndef _RWGltf_CafWriter_HeaderFiler
#define _RWGltf_CafWriter_HeaderFiler
#include <Message_ProgressScope.hxx>
#include <NCollection_DataMap.hxx>
#include <TColStd_IndexedDataMapOfStringString.hxx>
#include <TColStd_MapOfAsciiString.hxx>
#include <TDF_LabelSequence.hxx>
#include <TopTools_ShapeMapHasher.hxx>
#include <RWGltf_DracoParameters.hxx>
#include <RWGltf_GltfArrayType.hxx>
#include <RWGltf_GltfBufferView.hxx>
#include <RWGltf_GltfFace.hxx>
#include <RWGltf_WriterTrsfFormat.hxx>
#include <RWMesh_CoordinateSystemConverter.hxx>
#include <RWMesh_NameFormat.hxx>
#include <XCAFPrs_DocumentNode.hxx>
#include <XCAFPrs_Style.hxx>
#include <Poly_Triangle.hxx>
#include <memory>
class Message_ProgressRange;
class RWMesh_ShapeIterator;
class RWMesh_FaceIterator;
class RWMesh_EdgeIterator;
class RWMesh_VertexIterator;
class RWGltf_GltfOStreamWriter;
class RWGltf_GltfMaterialMap;
class RWGltf_GltfSceneNodeMap;
@ -202,8 +209,12 @@ protected:
const Message_ProgressRange& theProgress);
protected:
//! Return TRUE if face mesh should be skipped (e.g. because it is invalid or empty).
Standard_EXPORT virtual Standard_Boolean toSkipFaceMesh(const RWMesh_FaceIterator& theFaceIter);
//! Return TRUE if face shape should be skipped (e.g. because it is invalid or empty).
Standard_EXPORT virtual Standard_Boolean toSkipShape(
const RWMesh_ShapeIterator& theShapeIter) const;
//! Return TRUE if shape has triangulation (not vertex or edge).
Standard_EXPORT virtual Standard_Boolean hasTriangulation(const RWGltf_GltfFace& theShape) const;
//! Generate name for specified labels.
//! @param[in] theFormat name format to apply
@ -216,13 +227,13 @@ protected:
//! Write mesh nodes into binary file.
//! @param[out] theGltfFace glTF face definition
//! @param[out] theBinFile output file to write into
//! @param[in] theFaceIter current face to write
//! @param[in] theShapeIter current shape to write
//! @param[in][out] theAccessorNb last accessor index
//! @param[in][out] theMesh mesh
Standard_EXPORT virtual void saveNodes(
RWGltf_GltfFace& theGltfFace,
std::ostream& theBinFile,
const RWMesh_FaceIterator& theFaceIter,
const RWMesh_ShapeIterator& theShapeIter,
Standard_Integer& theAccessorNb,
const std::shared_ptr<RWGltf_CafWriter::Mesh>& theMesh) const;
@ -235,7 +246,7 @@ protected:
Standard_EXPORT virtual void saveNormals(
RWGltf_GltfFace& theGltfFace,
std::ostream& theBinFile,
RWMesh_FaceIterator& theFaceIter,
const RWMesh_FaceIterator& theFaceIter,
Standard_Integer& theAccessorNb,
const std::shared_ptr<RWGltf_CafWriter::Mesh>& theMesh) const;
@ -255,15 +266,42 @@ protected:
//! Write mesh indexes into binary file.
//! @param[out] theGltfFace glTF face definition
//! @param[out] theBinFile output file to write into
//! @param[in] theFaceIter current face to write
//! @param[in] theShapeIter current shape to write
//! @param[in][out] theAccessorNb last accessor index
//! @param[in][out] theMesh mesh
Standard_EXPORT virtual void saveIndices(RWGltf_GltfFace& theGltfFace,
std::ostream& theBinFile,
const RWMesh_FaceIterator& theFaceIter,
Standard_Integer& theAccessorNb,
Standard_EXPORT virtual void saveIndices(RWGltf_GltfFace& theGltfFace,
std::ostream& theBinFile,
const RWMesh_ShapeIterator& theShapeIter,
Standard_Integer& theAccessorNb,
const std::shared_ptr<RWGltf_CafWriter::Mesh>& theMesh);
//! Write triangle indexes into binary file.
//! @param[out] theGltfFace glTF face definition
//! @param[out] theBinFile output file to write into
//! @param[in] theFaceIter current face to write
//! @param[in][out] theMesh mesh
Standard_EXPORT virtual void saveTriangleIndices(
RWGltf_GltfFace& theGltfFace,
std::ostream& theBinFile,
const RWMesh_FaceIterator& theFaceIter,
const std::shared_ptr<RWGltf_CafWriter::Mesh>& theMesh);
//! Write edge indexes into binary file.
//! @param[out] theGltfFace glTF face definition
//! @param[out] theBinFile output file to write into
//! @param[in] theEdgeIter current edge to write
Standard_EXPORT virtual void saveEdgeIndices(RWGltf_GltfFace& theGltfFace,
std::ostream& theBinFile,
const RWMesh_EdgeIterator& theEdgeIter);
//! Write vertex indexes into binary file.
//! @param[out] theGltfFace glTF face definition
//! @param[out] theBinFile output file to write into
//! @param[in] theVertexIter current vertex to write
Standard_EXPORT virtual void saveVertexIndices(RWGltf_GltfFace& theGltfFace,
std::ostream& theBinFile,
const RWMesh_VertexIterator& theVertexIter);
protected:
//! Write bufferView for vertex positions within RWGltf_GltfRootElement_Accessors section
//! @param[in] theGltfFace face definition to write
@ -369,6 +407,51 @@ protected:
Standard_EXPORT virtual void writeExtrasAttributes(
const Handle(TDataStd_NamedData)& theNamedData);
//! Dispatch shapes
//! @param[in] theDocNode Document node containing shape data
//! @param[in] thePSentryBin Progress scope for the operation
//! @param[in,out] theMergedFaces Data map to store merged faces
//! @param[in,out] theShapeIter Shape iterator to traverse shapes
Standard_EXPORT virtual void dispatchShapes(
const XCAFPrs_DocumentNode& theDocNode,
const Message_ProgressScope& thePSentryBin,
NCollection_DataMap<XCAFPrs_Style, Handle(RWGltf_GltfFace)>& theMergedFaces,
RWMesh_ShapeIterator& theShapeIter);
//! Write shape into binary file
//! @param[out] theGltfFace glTF face definition
//! @param[out] theBinFile Output file to write into
//! @param[in] theShapeIter Current shape iterator
//! @param[in,out] theAccessorNb Last accessor index
//! @param[in,out] theMesh Mesh data
//! @param[in] theArrType Array type for glTF
//! @param[in] thePSentryBin Progress scope for the operation
//! @return True if shapes were successfully written to the binary file, false otherwise
Standard_EXPORT bool writeShapesToBin(RWGltf_GltfFace& theGltfFace,
std::ostream& theBinFile,
RWMesh_ShapeIterator& theShapeIter,
Standard_Integer& theAccessorNb,
const std::shared_ptr<RWGltf_CafWriter::Mesh>& theMesh,
const RWGltf_GltfArrayType theArrType,
const Message_ProgressScope& thePSentryBin);
//! Write shapes to RWGltf_GltfRootElement_Meshes section
//! @param[in] theShapeIter Shape iterator to traverse shapes
//! @param[in,out] theNbFacesInNode Number of faces in the current node
//! @param[in,out] theDracoBufInd Draco buffer index
//! @param[in,out] theToStartPrims Flag to indicate if primitives should be started
//! @param[in] theNodeName Name of the current node
//! @param[in,out] theWrittenShapes Map to store written shapes
//! @param[in,out] theDracoBufIndMap Map to store Draco buffer indices
Standard_EXPORT virtual void writeShapes(
RWMesh_ShapeIterator& theShapeIter,
Standard_Integer& theNbFacesInNode,
Standard_Integer& theDracoBufInd,
Standard_Boolean& theToStartPrims,
const TCollection_AsciiString& theNodeName,
NCollection_Map<Handle(RWGltf_GltfFaceList)>& theWrittenShapes,
NCollection_IndexedDataMap<int, int>& theDracoBufIndMap);
protected:
//! Shape + Style pair.
struct RWGltf_StyledShape

6
src/RWMesh/FILES
View File

@ -5,13 +5,19 @@ RWMesh_CafReader.hxx
RWMesh_CoordinateSystem.hxx
RWMesh_CoordinateSystemConverter.cxx
RWMesh_CoordinateSystemConverter.hxx
RWMesh_EdgeIterator.cxx
RWMesh_EdgeIterator.hxx
RWMesh_FaceIterator.cxx
RWMesh_FaceIterator.hxx
RWMesh_MaterialMap.cxx
RWMesh_MaterialMap.hxx
RWMesh_NameFormat.hxx
RWMesh_NodeAttributes.hxx
RWMesh_ShapeIterator.cxx
RWMesh_ShapeIterator.hxx
RWMesh_TriangulationReader.cxx
RWMesh_TriangulationReader.hxx
RWMesh_TriangulationSource.cxx
RWMesh_TriangulationSource.hxx
RWMesh_VertexIterator.cxx
RWMesh_VertexIterator.hxx

78
src/RWMesh/RWMesh_EdgeIterator.cxx Normal file
View File

@ -0,0 +1,78 @@
// Copyright (c) 2025 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.
#include <RWMesh_EdgeIterator.hxx>
#include <BRep_Tool.hxx>
#include <TopExp.hxx>
#include <TopoDS.hxx>
#include <XCAFDoc_ShapeTool.hxx>
#include <XCAFPrs.hxx>
// =======================================================================
// function : RWMesh_EdgeIterator
// purpose :
// =======================================================================
RWMesh_EdgeIterator::RWMesh_EdgeIterator(const TDF_Label& theLabel,
const TopLoc_Location& theLocation,
const Standard_Boolean theToMapColors,
const XCAFPrs_Style& theStyle)
: RWMesh_ShapeIterator(theLabel, theLocation, TopAbs_EDGE, theToMapColors, theStyle)
{
Next();
}
// =======================================================================
// function : RWMesh_EdgeIterator
// purpose :
// =======================================================================
RWMesh_EdgeIterator::RWMesh_EdgeIterator(const TopoDS_Shape& theShape,
const XCAFPrs_Style& theStyle)
: RWMesh_ShapeIterator(theShape, TopAbs_EDGE, theStyle)
{
Next();
}
// =======================================================================
// function : Next
// purpose :
// =======================================================================
void RWMesh_EdgeIterator::Next()
{
for (; myIter.More(); myIter.Next())
{
myEdge = TopoDS::Edge(myIter.Current());
myPolygon3D = BRep_Tool::Polygon3D(myEdge, myLocation);
myTrsf = myLocation.Transformation();
if (myPolygon3D.IsNull() || myPolygon3D->NbNodes() == 0)
{
resetEdge();
continue;
}
initEdge();
myIter.Next();
return;
}
resetEdge();
}
// =======================================================================
// function : initEdge
// purpose :
// =======================================================================
void RWMesh_EdgeIterator::initEdge()
{
initShape();
}

118
src/RWMesh/RWMesh_EdgeIterator.hxx Normal file
View File

@ -0,0 +1,118 @@
// Copyright (c) 2025 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.
#ifndef _RWMesh_EdgeIterator_HeaderFile
#define _RWMesh_EdgeIterator_HeaderFile
#include <NCollection_DataMap.hxx>
#include <Poly_Polygon3D.hxx>
#include <RWMesh_ShapeIterator.hxx>
#include <TopExp_Explorer.hxx>
#include <TopTools_IndexedDataMapOfShapeListOfShape.hxx>
#include <TopoDS_Edge.hxx>
#include <XCAFPrs_Style.hxx>
#include <gp_Trsf.hxx>
#include <algorithm>
class TDF_Label;
//! Auxiliary class to iterate through edges.
//! Provides functionality to iterate through the edges of a shape.
//! It inherits from `RWMesh_ShapeIterator` and implements
//! methods to access and manipulate edge data.
class RWMesh_EdgeIterator : public RWMesh_ShapeIterator
{
public:
//! Main constructor.
//! @param[in] theLabel The label of the shape.
//! @param[in] theLocation The location of the shape.
//! @param[in] theToMapColors Flag to indicate if colors should be mapped.
//! @param[in] theStyle The style of the shape.
Standard_EXPORT RWMesh_EdgeIterator(const TDF_Label& theLabel,
const TopLoc_Location& theLocation,
const Standard_Boolean theToMapColors = false,
const XCAFPrs_Style& theStyle = XCAFPrs_Style());
//! Auxiliary constructor.
//! @param[in] theShape The shape to iterate.
//! @param[in] theStyle The style of the shape.
Standard_EXPORT RWMesh_EdgeIterator(const TopoDS_Shape& theShape,
const XCAFPrs_Style& theStyle = XCAFPrs_Style());
//! Return true if iterator points to the valid triangulation.
bool More() const Standard_OVERRIDE { return !myPolygon3D.IsNull(); }
//! Find next value.
Standard_EXPORT void Next() Standard_OVERRIDE;
//! Return current edge.
const TopoDS_Edge& Edge() const { return myEdge; }
//! Return current edge.
const TopoDS_Shape& Shape() const Standard_OVERRIDE { return myEdge; }
//! Return current edge data.
const Handle(Poly_Polygon3D)& Polygon3D() const { return myPolygon3D; }
//! Return true if geometry data is defined.
bool IsEmpty() const Standard_OVERRIDE
{
return myPolygon3D.IsNull() || myPolygon3D->NbNodes() < 1;
}
public:
//! Lower element index in current triangulation.
Standard_Integer ElemLower() const Standard_OVERRIDE { return 1; }
//! Upper element index in current triangulation.
Standard_Integer ElemUpper() const Standard_OVERRIDE { return myPolygon3D->NbNodes(); }
public:
//! Return number of nodes for the current edge.
Standard_Integer NbNodes() const Standard_OVERRIDE
{
return !myPolygon3D.IsNull() ? myPolygon3D->NbNodes() : 0;
}
//! Lower node index in current triangulation.
Standard_Integer NodeLower() const Standard_OVERRIDE { return 1; }
//! Upper node index in current triangulation.
Standard_Integer NodeUpper() const Standard_OVERRIDE { return myPolygon3D->NbNodes(); }
public:
//! Return the node with specified index with applied transformation.
gp_Pnt node(const Standard_Integer theNode) const Standard_OVERRIDE
{
return myPolygon3D->Nodes().Value(theNode);
}
private:
//! Reset information for current edge.
void resetEdge()
{
myPolygon3D.Nullify();
myEdge.Nullify();
resetShape();
}
//! Initialize edge properties.
void initEdge();
private:
TopoDS_Edge myEdge; //!< current edge
Handle(Poly_Polygon3D) myPolygon3D; //!< geometry of current edge
};
#endif // _RWMesh_EdgeIterator_HeaderFile

136
src/RWMesh/RWMesh_FaceIterator.cxx
View File

@ -28,28 +28,11 @@ RWMesh_FaceIterator::RWMesh_FaceIterator(const TDF_Label& theLabel,
const TopLoc_Location& theLocation,
const Standard_Boolean theToMapColors,
const XCAFPrs_Style& theStyle)
: myDefStyle(theStyle),
myToMapColors(theToMapColors),
: RWMesh_ShapeIterator(theLabel, theLocation, TopAbs_FACE, theToMapColors, theStyle),
mySLTool(1, 1e-12),
myHasNormals(false),
myIsMirrored(false),
myHasFaceColor(false)
myIsMirrored(false)
{
TopoDS_Shape aShape;
if (!XCAFDoc_ShapeTool::GetShape(theLabel, aShape) || aShape.IsNull())
{
return;
}
aShape.Location(theLocation, false);
myFaceIter.Init(aShape, TopAbs_FACE);
if (theToMapColors)
{
dispatchStyles(theLabel, theLocation, theStyle);
myStyles.Bind(aShape, theStyle);
}
Next();
}
@ -59,90 +42,14 @@ RWMesh_FaceIterator::RWMesh_FaceIterator(const TDF_Label& theLabel,
// =======================================================================
RWMesh_FaceIterator::RWMesh_FaceIterator(const TopoDS_Shape& theShape,
const XCAFPrs_Style& theStyle)
: myDefStyle(theStyle),
myToMapColors(true),
: RWMesh_ShapeIterator(theShape, TopAbs_FACE, theStyle),
mySLTool(1, 1e-12),
myHasNormals(false),
myIsMirrored(false),
myHasFaceColor(false)
myIsMirrored(false)
{
if (theShape.IsNull())
{
return;
}
myFaceIter.Init(theShape, TopAbs_FACE);
Next();
}
// =======================================================================
// function : dispatchStyles
// purpose :
// =======================================================================
void RWMesh_FaceIterator::dispatchStyles(const TDF_Label& theLabel,
const TopLoc_Location& theLocation,
const XCAFPrs_Style& theStyle)
{
TopLoc_Location aDummyLoc;
XCAFPrs_IndexedDataMapOfShapeStyle aStyles;
XCAFPrs::CollectStyleSettings(theLabel, aDummyLoc, aStyles);
Standard_Integer aNbTypes[TopAbs_SHAPE] = {};
for (Standard_Integer aTypeIter = TopAbs_FACE; aTypeIter >= TopAbs_COMPOUND; --aTypeIter)
{
if (aTypeIter != TopAbs_FACE && aNbTypes[aTypeIter] == 0)
{
continue;
}
for (XCAFPrs_IndexedDataMapOfShapeStyle::Iterator aStyleIter(aStyles); aStyleIter.More();
aStyleIter.Next())
{
const TopoDS_Shape& aKeyShape = aStyleIter.Key();
const TopAbs_ShapeEnum aKeyShapeType = aKeyShape.ShapeType();
if (aTypeIter == TopAbs_FACE)
{
++aNbTypes[aKeyShapeType];
}
if (aTypeIter != aKeyShapeType)
{
continue;
}
XCAFPrs_Style aCafStyle = aStyleIter.Value();
if (!aCafStyle.IsSetColorCurv() && theStyle.IsSetColorCurv())
{
aCafStyle.SetColorCurv(theStyle.GetColorCurv());
}
if (!aCafStyle.IsSetColorSurf() && theStyle.IsSetColorSurf())
{
aCafStyle.SetColorSurf(theStyle.GetColorSurfRGBA());
}
if (aCafStyle.Material().IsNull() && !theStyle.Material().IsNull())
{
aCafStyle.SetMaterial(theStyle.Material());
}
TopoDS_Shape aKeyShapeLocated = aKeyShape.Located(theLocation);
if (aKeyShapeType == TopAbs_FACE)
{
myStyles.Bind(aKeyShapeLocated, aCafStyle);
}
else
{
for (TopExp_Explorer aFaceIter(aKeyShapeLocated, TopAbs_FACE); aFaceIter.More();
aFaceIter.Next())
{
if (!myStyles.IsBound(aFaceIter.Current()))
{
myStyles.Bind(aFaceIter.Current(), aCafStyle);
}
}
}
}
}
}
// =======================================================================
// function : normal
// purpose :
@ -177,11 +84,11 @@ gp_Dir RWMesh_FaceIterator::normal(Standard_Integer theNode) const
// =======================================================================
void RWMesh_FaceIterator::Next()
{
for (; myFaceIter.More(); myFaceIter.Next())
for (; myIter.More(); myIter.Next())
{
myFace = TopoDS::Face(myFaceIter.Current());
myPolyTriang = BRep_Tool::Triangulation(myFace, myFaceLocation);
myTrsf = myFaceLocation.Transformation();
myFace = TopoDS::Face(myIter.Current());
myPolyTriang = BRep_Tool::Triangulation(myFace, myLocation);
myTrsf = myLocation.Transformation();
if (myPolyTriang.IsNull() || myPolyTriang->NbTriangles() == 0)
{
resetFace();
@ -189,7 +96,7 @@ void RWMesh_FaceIterator::Next()
}
initFace();
myFaceIter.Next();
myIter.Next();
return;
}
@ -202,9 +109,8 @@ void RWMesh_FaceIterator::Next()
// =======================================================================
void RWMesh_FaceIterator::initFace()
{
myHasNormals = false;
myHasFaceColor = false;
myIsMirrored = myTrsf.VectorialPart().Determinant() < 0.0;
myHasNormals = false;
myIsMirrored = myTrsf.VectorialPart().Determinant() < 0.0;
if (myPolyTriang->HasNormals())
{
myHasNormals = true;
@ -221,24 +127,6 @@ void RWMesh_FaceIterator::initFace()
myHasNormals = true;
}
}
if (!myToMapColors)
{
return;
}
if (!myStyles.Find(myFace, myFaceStyle))
{
myFaceStyle = myDefStyle;
}
if (!myFaceStyle.Material().IsNull())
{
myHasFaceColor = true;
myFaceColor = myFaceStyle.Material()->BaseColor();
}
else if (myFaceStyle.IsSetColorSurf())
{
myHasFaceColor = true;
myFaceColor = myFaceStyle.GetColorSurfRGBA();
}
initShape();
}

100
src/RWMesh/RWMesh_FaceIterator.hxx
View File

@ -14,50 +14,60 @@
#ifndef _RWMesh_FaceIterator_HeaderFile
#define _RWMesh_FaceIterator_HeaderFile
#include <RWMesh_ShapeIterator.hxx>
#include <BRepLProp_SLProps.hxx>
#include <gp_Trsf.hxx>
#include <NCollection_DataMap.hxx>
#include <Poly_Triangulation.hxx>
#include <TopExp_Explorer.hxx>
#include <TopoDS_Face.hxx>
#include <TopTools_IndexedDataMapOfShapeListOfShape.hxx>
#include <XCAFPrs_Style.hxx>
#include <algorithm>
class TDF_Label;
//! Auxiliary class to iterate through triangulated faces.
class RWMesh_FaceIterator
//! Class is designed to provide an interface for iterating over the faces
//! of a shape, specifically focusing on triangulated faces.
//! It inherits from the `RWMesh_ShapeIterator` base class and
//! extends its functionality to handle faces with triangulation data.
class RWMesh_FaceIterator : public RWMesh_ShapeIterator
{
public:
//! Main constructor.
//! @param[in] theLabel Label containing the face data
//! @param[in] theLocation Location of the face
//! @param[in] theToMapColors Flag to indicate if colors should be mapped
//! @param[in] theStyle Style information for the face
Standard_EXPORT RWMesh_FaceIterator(const TDF_Label& theLabel,
const TopLoc_Location& theLocation,
const Standard_Boolean theToMapColors = false,
const XCAFPrs_Style& theStyle = XCAFPrs_Style());
//! Auxiliary constructor.
//! @param[in] theShape Shape containing the face data
//! @param[in] theStyle Style information for the face
Standard_EXPORT RWMesh_FaceIterator(const TopoDS_Shape& theShape,
const XCAFPrs_Style& theStyle = XCAFPrs_Style());
//! Return explored shape.
const TopoDS_Shape& ExploredShape() const { return myFaceIter.ExploredShape(); }
//! Return true if iterator points to the valid triangulation.
bool More() const { return !myPolyTriang.IsNull(); }
bool More() const Standard_OVERRIDE { return !myPolyTriang.IsNull(); }
//! Find next value.
Standard_EXPORT void Next();
Standard_EXPORT void Next() Standard_OVERRIDE;
//! Return current face.
const TopoDS_Face& Face() const { return myFace; }
//! Return current face.
const TopoDS_Shape& Shape() const Standard_OVERRIDE { return myFace; }
//! Return current face triangulation.
const Handle(Poly_Triangulation)& Triangulation() const { return myPolyTriang; }
//! Return true if mesh data is defined.
bool IsEmptyMesh() const
bool IsEmptyMesh() const { return IsEmpty(); }
//! Return true if mesh data is defined.
bool IsEmpty() const Standard_OVERRIDE
{
return myPolyTriang.IsNull()
|| (myPolyTriang->NbNodes() < 1 && myPolyTriang->NbTriangles() < 1);
@ -65,23 +75,23 @@ public:
public:
//! Return face material.
const XCAFPrs_Style& FaceStyle() const { return myFaceStyle; }
const XCAFPrs_Style& FaceStyle() const { return myStyle; }
//! Return TRUE if face color is set.
bool HasFaceColor() const { return myHasFaceColor; }
bool HasFaceColor() const { return myHasColor; }
//! Return face color.
const Quantity_ColorRGBA& FaceColor() const { return myFaceColor; }
const Quantity_ColorRGBA& FaceColor() const { return myColor; }
public:
//! Return number of elements of specific type for the current face.
Standard_Integer NbTriangles() const { return myPolyTriang->NbTriangles(); }
//! Lower element index in current triangulation.
Standard_Integer ElemLower() const { return 1; }
Standard_Integer ElemLower() const Standard_OVERRIDE { return 1; }
//! Upper element index in current triangulation.
Standard_Integer ElemUpper() const { return myPolyTriang->NbTriangles(); }
Standard_Integer ElemUpper() const Standard_OVERRIDE { return myPolyTriang->NbTriangles(); }
//! Return triangle with specified index with applied Face orientation.
Poly_Triangle TriangleOriented(Standard_Integer theElemIndex) const
@ -118,21 +128,16 @@ public:
}
//! Return number of nodes for the current face.
Standard_Integer NbNodes() const { return !myPolyTriang.IsNull() ? myPolyTriang->NbNodes() : 0; }
Standard_Integer NbNodes() const Standard_OVERRIDE
{
return !myPolyTriang.IsNull() ? myPolyTriang->NbNodes() : 0;
}
//! Lower node index in current triangulation.
Standard_Integer NodeLower() const { return 1; }
Standard_Integer NodeLower() const Standard_OVERRIDE { return 1; }
//! Upper node index in current triangulation.
Standard_Integer NodeUpper() const { return myPolyTriang->NbNodes(); }
//! Return the node with specified index with applied transformation.
gp_Pnt NodeTransformed(const Standard_Integer theNode) const
{
gp_Pnt aNode = node(theNode);
aNode.Transform(myTrsf);
return aNode;
}
Standard_Integer NodeUpper() const Standard_OVERRIDE { return myPolyTriang->NbNodes(); }
//! Return texture coordinates for the node.
gp_Pnt2d NodeTexCoord(const Standard_Integer theNode) const
@ -142,7 +147,10 @@ public:
public:
//! Return the node with specified index with applied transformation.
gp_Pnt node(const Standard_Integer theNode) const { return myPolyTriang->Node(theNode); }
gp_Pnt node(const Standard_Integer theNode) const Standard_OVERRIDE
{
return myPolyTriang->Node(theNode);
}
//! Return normal at specified node index without face transformation applied.
Standard_EXPORT gp_Dir normal(Standard_Integer theNode) const;
@ -154,44 +162,26 @@ public:
}
private:
//! Dispatch face styles.
void dispatchStyles(const TDF_Label& theLabel,
const TopLoc_Location& theLocation,
const XCAFPrs_Style& theStyle);
//! Reset information for current face.
void resetFace()
{
myPolyTriang.Nullify();
myFace.Nullify();
myHasNormals = false;
myHasFaceColor = false;
myFaceColor = Quantity_ColorRGBA();
myFaceStyle = XCAFPrs_Style();
myHasNormals = false;
resetShape();
}
//! Initialize face properties.
void initFace();
private:
NCollection_DataMap<TopoDS_Shape, XCAFPrs_Style, TopTools_ShapeMapHasher>
myStyles; //!< Face -> Style map
// clang-format off
XCAFPrs_Style myDefStyle; //!< default style for faces without dedicated style
Standard_Boolean myToMapColors; //!< flag to dispatch styles
TopExp_Explorer myFaceIter; //!< face explorer
TopoDS_Face myFace; //!< current face
Handle(Poly_Triangulation) myPolyTriang; //!< triangulation of current face
TopLoc_Location myFaceLocation; //!< current face location
mutable BRepLProp_SLProps mySLTool; //!< auxiliary tool for fetching normals from surface
BRepAdaptor_Surface myFaceAdaptor; //!< surface adaptor for fetching normals from surface
Standard_Boolean myHasNormals; //!< flag indicating that current face has normals
gp_Trsf myTrsf; //!< current face transformation
Standard_Boolean myIsMirrored; //!< flag indicating that face triangles should be mirrored
XCAFPrs_Style myFaceStyle; //!< current face style
Quantity_ColorRGBA myFaceColor; //!< current face color
Standard_Boolean myHasFaceColor; //!< flag indicating that current face has assigned color
TopoDS_Face myFace; //!< current face
Handle(Poly_Triangulation) myPolyTriang; //!< triangulation of current face
mutable BRepLProp_SLProps mySLTool; //!< auxiliary tool for fetching normals from surface
BRepAdaptor_Surface myFaceAdaptor; //!< surface adaptor for fetching normals from surface
Standard_Boolean myHasNormals; //!< flag indicating that current face has normals
Standard_Boolean myIsMirrored; //!< flag indicating that face triangles should be mirrored
// clang-format on
};

163
src/RWMesh/RWMesh_ShapeIterator.cxx Normal file
View File

@ -0,0 +1,163 @@
// Copyright (c) 2025 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.
#include <RWMesh_ShapeIterator.hxx>
#include <TopExp.hxx>
#include <TopoDS.hxx>
#include <XCAFDoc_ShapeTool.hxx>
#include <XCAFPrs.hxx>
// =======================================================================
// function : RWMesh_ShapeIterator
// purpose :
// =======================================================================
RWMesh_ShapeIterator::RWMesh_ShapeIterator(const TDF_Label& theLabel,
const TopLoc_Location& theLocation,
const TopAbs_ShapeEnum theShapeType,
const Standard_Boolean theToMapColors,
const XCAFPrs_Style& theStyle)
: myDefStyle(theStyle),
myToMapColors(theToMapColors),
myShapeType(theShapeType),
myHasColor(false)
{
TopoDS_Shape aShape;
if (!XCAFDoc_ShapeTool::GetShape(theLabel, aShape) || aShape.IsNull())
{
return;
}
aShape.Location(theLocation, false);
myIter.Init(aShape, myShapeType);
if (theToMapColors)
{
dispatchStyles(theLabel, theLocation, theStyle);
myStyles.Bind(aShape, theStyle);
}
}
// =======================================================================
// function : RWMesh_ShapeIterator
// purpose :
// =======================================================================
RWMesh_ShapeIterator::RWMesh_ShapeIterator(const TopoDS_Shape& theShape,
const TopAbs_ShapeEnum theShapeType,
const XCAFPrs_Style& theStyle)
: myDefStyle(theStyle),
myToMapColors(true),
myShapeType(theShapeType),
myHasColor(false)
{
if (theShape.IsNull())
{
return;
}
myIter.Init(theShape, myShapeType);
}
// =======================================================================
// function : dispatchStyles
// purpose :
// =======================================================================
void RWMesh_ShapeIterator::dispatchStyles(const TDF_Label& theLabel,
const TopLoc_Location& theLocation,
const XCAFPrs_Style& theStyle)
{
TopLoc_Location aDummyLoc;
XCAFPrs_IndexedDataMapOfShapeStyle aStyles;
XCAFPrs::CollectStyleSettings(theLabel, aDummyLoc, aStyles);
Standard_Integer aNbTypes[TopAbs_SHAPE] = {};
for (Standard_Integer aTypeIter = myShapeType; aTypeIter >= TopAbs_COMPOUND; --aTypeIter)
{
if (aTypeIter != myShapeType && aNbTypes[aTypeIter] == 0)
{
continue;
}
for (XCAFPrs_IndexedDataMapOfShapeStyle::Iterator aStyleIter(aStyles); aStyleIter.More();
aStyleIter.Next())
{
const TopoDS_Shape& aKeyShape = aStyleIter.Key();
const TopAbs_ShapeEnum aKeyShapeType = aKeyShape.ShapeType();
if (aTypeIter == myShapeType)
{
++aNbTypes[aKeyShapeType];
}
if (aTypeIter != aKeyShapeType)
{
continue;
}
XCAFPrs_Style aCafStyle = aStyleIter.Value();
if (!aCafStyle.IsSetColorCurv() && theStyle.IsSetColorCurv())
{
aCafStyle.SetColorCurv(theStyle.GetColorCurv());
}
if (!aCafStyle.IsSetColorSurf() && theStyle.IsSetColorSurf())
{
aCafStyle.SetColorSurf(theStyle.GetColorSurfRGBA());
}
if (aCafStyle.Material().IsNull() && !theStyle.Material().IsNull())
{
aCafStyle.SetMaterial(theStyle.Material());
}
TopoDS_Shape aKeyShapeLocated = aKeyShape.Located(theLocation);
if (aKeyShapeType == myShapeType)
{
myStyles.Bind(aKeyShapeLocated, aCafStyle);
}
else
{
for (TopExp_Explorer aShapeIter(aKeyShapeLocated, myShapeType); aShapeIter.More();
aShapeIter.Next())
{
if (!myStyles.IsBound(aShapeIter.Current()))
{
myStyles.Bind(aShapeIter.Current(), aCafStyle);
}
}
}
}
}
}
// =======================================================================
// function : initEdge
// purpose :
// =======================================================================
void RWMesh_ShapeIterator::initShape()
{
myHasColor = false;
if (!myToMapColors)
{
return;
}
if (!myStyles.Find(Shape(), myStyle))
{
myStyle = myDefStyle;
}
if (!myStyle.Material().IsNull())
{
myHasColor = true;
myColor = myStyle.Material()->BaseColor();
}
else if (myStyle.IsSetColorSurf())
{
myHasColor = true;
myColor = myStyle.GetColorSurfRGBA();
}
}

131
src/RWMesh/RWMesh_ShapeIterator.hxx Normal file
View File

@ -0,0 +1,131 @@
// Copyright (c) 2025 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.
#ifndef _RWMesh_ShapeIterator_HeaderFile
#define _RWMesh_ShapeIterator_HeaderFile
#include <BRepLProp_SLProps.hxx>
#include <NCollection_DataMap.hxx>
#include <Poly_Triangulation.hxx>
#include <TopExp_Explorer.hxx>
#include <TopTools_IndexedDataMapOfShapeListOfShape.hxx>
#include <TopoDS_Edge.hxx>
#include <XCAFPrs_Style.hxx>
#include <gp_Trsf.hxx>
#include <algorithm>
class TDF_Label;
//! This is a virtual base class for other shape iterators.
//! Provides an abstract interface for iterating over the elements of a shape.
//! It defines a set of pure virtual methods that must be implemented by
//! derived classes to handle specific types of shapes and their elements.
class RWMesh_ShapeIterator
{
public:
//! Return explored shape.
const TopoDS_Shape& ExploredShape() const { return myIter.ExploredShape(); }
//! Return shape.
Standard_EXPORT virtual const TopoDS_Shape& Shape() const = 0;
//! Return true if iterator points to the valid triangulation.
Standard_EXPORT virtual bool More() const = 0;
//! Find next value.
Standard_EXPORT virtual void Next() = 0;
//! Return true if mesh data is defined.
Standard_EXPORT virtual bool IsEmpty() const = 0;
//! Return shape material.
const XCAFPrs_Style& Style() const { return myStyle; }
//! Return TRUE if shape color is set.
bool HasColor() const { return myHasColor; }
//! Return shape color.
const Quantity_ColorRGBA& Color() const { return myColor; }
//! Lower element index in current triangulation.
Standard_EXPORT virtual Standard_Integer ElemLower() const = 0;
//! Upper element index in current triangulation.
Standard_EXPORT virtual Standard_Integer ElemUpper() const = 0;
//! Return number of nodes for the current shape.
Standard_EXPORT virtual Standard_Integer NbNodes() const = 0;
//! Lower node index in current shape.
Standard_EXPORT virtual Standard_Integer NodeLower() const = 0;
//! Upper node index in current shape.
Standard_EXPORT virtual Standard_Integer NodeUpper() const = 0;
//! Return the node with specified index with applied transformation.
gp_Pnt NodeTransformed(const Standard_Integer theNode) const
{
gp_Pnt aNode = node(theNode);
aNode.Transform(myTrsf);
return aNode;
}
protected:
//! Return the node with specified index with applied transformation.
virtual gp_Pnt node(const Standard_Integer theNode) const = 0;
//! Main constructor.
RWMesh_ShapeIterator(const TDF_Label& theLabel,
const TopLoc_Location& theLocation,
const TopAbs_ShapeEnum theShapeType,
const Standard_Boolean theToMapColors = false,
const XCAFPrs_Style& theStyle = XCAFPrs_Style());
//! Auxiliary constructor.
RWMesh_ShapeIterator(const TopoDS_Shape& theShape,
const TopAbs_ShapeEnum theShapeType,
const XCAFPrs_Style& theStyle = XCAFPrs_Style());
//! Dispatch shape styles.
void dispatchStyles(const TDF_Label& theLabel,
const TopLoc_Location& theLocation,
const XCAFPrs_Style& theStyle);
//! Reset information for current shape.
void resetShape()
{
myHasColor = false;
myColor = Quantity_ColorRGBA();
myStyle = XCAFPrs_Style();
}
//! Initialize shape properties.
void initShape();
protected:
NCollection_DataMap<TopoDS_Shape, XCAFPrs_Style, TopTools_ShapeMapHasher>
myStyles; //!< Shape -> Style map
XCAFPrs_Style myDefStyle; //!< default style for shapes without dedicated style
Standard_Boolean myToMapColors; //!< flag to dispatch styles
TopExp_Explorer myIter; //!< shape explorer
TopLoc_Location myLocation; //!< current shape location
gp_Trsf myTrsf; //!< current shape transformation
XCAFPrs_Style myStyle; //!< current shape style
Quantity_ColorRGBA myColor; //!< current shape color
TopAbs_ShapeEnum myShapeType; //!< type of shape
Standard_Boolean myHasColor; //!< flag indicating that current shape has assigned color
};
#endif // _RWMesh_ShapeIterator_HeaderFile

78
src/RWMesh/RWMesh_VertexIterator.cxx Normal file
View File

@ -0,0 +1,78 @@
// Copyright (c) 2025 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.
#include <RWMesh_VertexIterator.hxx>
#include <BRep_Tool.hxx>
#include <TopExp.hxx>
#include <TopoDS.hxx>
#include <XCAFDoc_ShapeTool.hxx>
#include <XCAFPrs.hxx>
// =======================================================================
// function : RWMesh_VertexIterator
// purpose :
// =======================================================================
RWMesh_VertexIterator::RWMesh_VertexIterator(const TDF_Label& theLabel,
const TopLoc_Location& theLocation,
const Standard_Boolean theToMapColors,
const XCAFPrs_Style& theStyle)
: RWMesh_ShapeIterator(theLabel, theLocation, TopAbs_VERTEX, theToMapColors, theStyle)
{
Next();
}
// =======================================================================
// function : RWMesh_VertexIterator
// purpose :
// =======================================================================
RWMesh_VertexIterator::RWMesh_VertexIterator(const TopoDS_Shape& theShape,
const XCAFPrs_Style& theStyle)
: RWMesh_ShapeIterator(theShape, TopAbs_VERTEX, theStyle)
{
Next();
}
// =======================================================================
// function : Next
// purpose :
// =======================================================================
void RWMesh_VertexIterator::Next()
{
for (; myIter.More(); myIter.Next())
{
myVertex = TopoDS::Vertex(myIter.Current());
myPoint = BRep_Tool::Pnt(myVertex);
myTrsf = myLocation.Transformation();
if (myVertex.IsNull())
{
resetVertex();
continue;
}
initVertex();
myIter.Next();
return;
}
resetVertex();
}
// =======================================================================
// function : initEdge
// purpose :
// =======================================================================
void RWMesh_VertexIterator::initVertex()
{
initShape();
}

108
src/RWMesh/RWMesh_VertexIterator.hxx Normal file
View File

@ -0,0 +1,108 @@
// Copyright (c) 2025 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.
#ifndef _RWMesh_VertexIterator_HeaderFile
#define _RWMesh_VertexIterator_HeaderFile
#include <NCollection_DataMap.hxx>
#include <Poly_Polygon3D.hxx>
#include <RWMesh_ShapeIterator.hxx>
#include <TopExp_Explorer.hxx>
#include <TopTools_IndexedDataMapOfShapeListOfShape.hxx>
#include <TopoDS_Vertex.hxx>
#include <XCAFPrs_Style.hxx>
#include <gp_Trsf.hxx>
#include <algorithm>
class TDF_Label;
//! Auxiliary class to iterate through vertices.
//! Provides functionality to iterate through the vertices of a shape.
//! It inherits from `RWMesh_ShapeIterator` and implements
//! methods to access and manipulate vertex data.
class RWMesh_VertexIterator : public RWMesh_ShapeIterator
{
public:
//! Main constructor.
//! @param[in] theLabel The label of the shape.
//! @param[in] theLocation The location of the shape.
//! @param[in] theToMapColors Flag to indicate if colors should be mapped.
//! @param[in] theStyle The style of the shape.
Standard_EXPORT RWMesh_VertexIterator(const TDF_Label& theLabel,
const TopLoc_Location& theLocation,
const Standard_Boolean theToMapColors = false,
const XCAFPrs_Style& theStyle = XCAFPrs_Style());
//! Auxiliary constructor.
//! @param[in] theShape The shape to iterate.
//! @param[in] theStyle The style of the shape.
Standard_EXPORT RWMesh_VertexIterator(const TopoDS_Shape& theShape,
const XCAFPrs_Style& theStyle = XCAFPrs_Style());
//! Return true if iterator points to the valid triangulation.
bool More() const Standard_OVERRIDE { return !myVertex.IsNull(); }
//! Find next value.
Standard_EXPORT void Next() Standard_OVERRIDE;
//! Return current edge.
const TopoDS_Vertex& Vertex() const { return myVertex; }
//! Return current vertex.
const TopoDS_Shape& Shape() const Standard_OVERRIDE { return myVertex; }
//! Return current vertex data.
const gp_Pnt& Point() const { return myPoint; }
//! Return true if geometry data is defined.
bool IsEmpty() const Standard_OVERRIDE { return myVertex.IsNull(); }
public:
//! Lower element index in current triangulation.
Standard_Integer ElemLower() const Standard_OVERRIDE { return 1; }
//! Upper element index in current triangulation.
Standard_Integer ElemUpper() const Standard_OVERRIDE { return 1; }
public:
//! Return number of nodes for the current edge.
Standard_Integer NbNodes() const Standard_OVERRIDE { return 1; }
//! Lower node index in current triangulation.
Standard_Integer NodeLower() const Standard_OVERRIDE { return 1; }
//! Upper node index in current triangulation.
Standard_Integer NodeUpper() const Standard_OVERRIDE { return 1; }
public:
//! Return the node with specified index with applied transformation.
gp_Pnt node(const Standard_Integer /*theNode*/) const Standard_OVERRIDE { return myPoint; }
private:
//! Reset information for current vertex.
void resetVertex()
{
myVertex.Nullify();
resetShape();
}
//! Initialize vertex properties.
void initVertex();
private:
TopoDS_Vertex myVertex; //!< current vertex
gp_Pnt myPoint; //!< geometry of current vertex
};
#endif // _RWMesh_VertexIterator_HeaderFile

22
tests/de_mesh/gltf_write/draco_mesh Normal file
View File

@ -0,0 +1,22 @@
puts "========"
puts "278: Data Exchange, Fail to export to GLTF with draco buffer index out of bounds"
puts "========"
vclear
vclose ALL
Close *
set aTmpGltf "${imagedir}/${casename}_tmp.glb"
ReadStep D [locate_data_file "bug31670_russian.stp"]
XGetOneShape s D
incmesh s 0.1
XNewDoc D
XAddShape D s
WriteGltf D "$aTmpGltf" -draco -mergefaces
ReadGltf D1 "$aTmpGltf"
XGetOneShape s1 D1
checknbshapes s1 -face 11 -compound 10 -shape 21