1 // Copyright (c) 1999-2014 OPEN CASCADE SAS
3 // This file is part of Open CASCADE Technology software library.
5 // This library is free software; you can redistribute it and/or modify it under
6 // the terms of the GNU Lesser General Public License version 2.1 as published
7 // by the Free Software Foundation, with special exception defined in the file
8 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
9 // distribution for complete text of the license and disclaimer of any warranty.
11 // Alternatively, this file may be used under the terms of Open CASCADE
12 // commercial license or contractual agreement.
14 #include <StlAPI_Writer.hxx>
16 #include <Bnd_Box.hxx>
17 #include <BRepBndLib.hxx>
18 #include <Message.hxx>
19 #include <Message_Messenger.hxx>
20 #include <OSD_Path.hxx>
21 #include <OSD_OpenFile.hxx>
23 #include <BRep_Tool.hxx>
25 #include <TopoDS_Face.hxx>
26 #include <TopExp_Explorer.hxx>
27 #include <Poly_Triangulation.hxx>
29 //=============================================================================
30 //function : StlAPI_Writer
32 //=============================================================================
33 StlAPI_Writer::StlAPI_Writer()
34 : myASCIIMode (Standard_True)
39 //=============================================================================
42 //=============================================================================
43 Standard_Boolean StlAPI_Writer::Write (const TopoDS_Shape& theShape,
44 const Standard_CString theFileName)
46 Standard_Integer aNbNodes = 0;
47 Standard_Integer aNbTriangles = 0;
49 // calculate total number of the nodes and triangles
50 for (TopExp_Explorer anExpSF (theShape, TopAbs_FACE); anExpSF.More(); anExpSF.Next())
53 Handle(Poly_Triangulation) aTriangulation = BRep_Tool::Triangulation (TopoDS::Face (anExpSF.Current()), aLoc);
54 if (! aTriangulation.IsNull())
56 aNbNodes += aTriangulation->NbNodes ();
57 aNbTriangles += aTriangulation->NbTriangles ();
61 if (aNbTriangles == 0)
63 // No triangulation on the shape
64 return Standard_False;
67 // create temporary triangulation
68 Handle(Poly_Triangulation) aMesh = new Poly_Triangulation (aNbNodes, aNbTriangles, Standard_False);
69 // count faces missing triangulation
70 Standard_Integer aNbFacesNoTri = 0;
71 // fill temporary triangulation
72 Standard_Integer aNodeOffset = 0;
73 Standard_Integer aTriangleOffet = 0;
74 for (TopExp_Explorer anExpSF (theShape, TopAbs_FACE); anExpSF.More(); anExpSF.Next())
76 const TopoDS_Shape& aFace = anExpSF.Current();
78 Handle(Poly_Triangulation) aTriangulation = BRep_Tool::Triangulation (TopoDS::Face (aFace), aLoc);
79 if (aTriangulation.IsNull())
85 const TColgp_Array1OfPnt& aNodes = aTriangulation->Nodes();
86 const Poly_Array1OfTriangle& aTriangles = aTriangulation->Triangles();
89 gp_Trsf aTrsf = aLoc.Transformation();
90 for (Standard_Integer aNodeIter = aNodes.Lower(); aNodeIter <= aNodes.Upper(); ++aNodeIter)
92 gp_Pnt aPnt = aNodes (aNodeIter);
93 aPnt.Transform (aTrsf);
94 aMesh->ChangeNode (aNodeIter + aNodeOffset) = aPnt;
98 const TopAbs_Orientation anOrientation = anExpSF.Current().Orientation();
99 for (Standard_Integer aTriIter = aTriangles.Lower(); aTriIter <= aTriangles.Upper(); ++aTriIter)
101 Poly_Triangle aTri = aTriangles (aTriIter);
103 Standard_Integer anId[3];
104 aTri.Get (anId[0], anId[1], anId[2]);
105 if (anOrientation == TopAbs_REVERSED)
108 Standard_Integer aTmpIdx = anId[1];
113 // Update nodes according to the offset.
114 anId[0] += aNodeOffset;
115 anId[1] += aNodeOffset;
116 anId[2] += aNodeOffset;
118 aTri.Set (anId[0], anId[1], anId[2]);
119 aMesh->ChangeTriangle (aTriIter + aTriangleOffet) = aTri;
122 aNodeOffset += aNodes.Size();
123 aTriangleOffet += aTriangles.Size();
126 OSD_Path aPath (theFileName);
127 Standard_Boolean isDone = (myASCIIMode
128 ? RWStl::WriteAscii (aMesh, aPath)
129 : RWStl::WriteBinary (aMesh, aPath));
131 if (isDone && (aNbFacesNoTri > 0))
133 // Print warning with number of faces missing triangulation
134 TCollection_AsciiString aWarningMsg =
135 TCollection_AsciiString ("Warning: ") +
136 TCollection_AsciiString (aNbFacesNoTri) +
137 TCollection_AsciiString ((aNbFacesNoTri == 1) ? " face has" : " faces have") +
138 TCollection_AsciiString (" been skipped due to null triangulation");
139 Message::DefaultMessenger()->Send (aWarningMsg, Message_Warning);