0030389: Data Exchange - StlAPI_Writer does not check if the face has triangulation
[occt.git] / src / StlAPI / StlAPI_Writer.cxx
1 // Copyright (c) 1999-2014 OPEN CASCADE SAS
2 //
3 // This file is part of Open CASCADE Technology software library.
4 //
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.
10 //
11 // Alternatively, this file may be used under the terms of Open CASCADE
12 // commercial license or contractual agreement.
13
14 #include <StlAPI_Writer.hxx>
15
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>
22 #include <RWStl.hxx>
23 #include <BRep_Tool.hxx>
24 #include <TopoDS.hxx>
25 #include <TopoDS_Face.hxx>
26 #include <TopExp_Explorer.hxx>
27 #include <Poly_Triangulation.hxx>
28
29 //=============================================================================
30 //function : StlAPI_Writer
31 //purpose  :
32 //=============================================================================
33 StlAPI_Writer::StlAPI_Writer()
34 : myASCIIMode (Standard_True)
35 {
36   //
37 }
38
39 //=============================================================================
40 //function : Write
41 //purpose  :
42 //=============================================================================
43 Standard_Boolean StlAPI_Writer::Write (const TopoDS_Shape&    theShape,
44                                        const Standard_CString theFileName)
45 {
46   Standard_Integer aNbNodes = 0;
47   Standard_Integer aNbTriangles = 0;
48
49   // calculate total number of the nodes and triangles
50   for (TopExp_Explorer anExpSF (theShape, TopAbs_FACE); anExpSF.More(); anExpSF.Next())
51   {
52     TopLoc_Location aLoc;
53     Handle(Poly_Triangulation) aTriangulation = BRep_Tool::Triangulation (TopoDS::Face (anExpSF.Current()), aLoc);
54     if (! aTriangulation.IsNull())
55     {
56       aNbNodes += aTriangulation->NbNodes ();
57       aNbTriangles += aTriangulation->NbTriangles ();
58     }
59   }
60
61   if (aNbTriangles == 0)
62   {
63     // No triangulation on the shape
64     return Standard_False;
65   }
66
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())
75   {
76     const TopoDS_Shape& aFace = anExpSF.Current();
77     TopLoc_Location aLoc;
78     Handle(Poly_Triangulation) aTriangulation = BRep_Tool::Triangulation (TopoDS::Face (aFace), aLoc);
79     if (aTriangulation.IsNull())
80     {
81       ++aNbFacesNoTri;
82       continue;
83     }
84
85     const TColgp_Array1OfPnt& aNodes = aTriangulation->Nodes();
86     const Poly_Array1OfTriangle& aTriangles = aTriangulation->Triangles();
87
88     // copy nodes
89     gp_Trsf aTrsf = aLoc.Transformation();
90     for (Standard_Integer aNodeIter = aNodes.Lower(); aNodeIter <= aNodes.Upper(); ++aNodeIter)
91     {
92       gp_Pnt aPnt = aNodes (aNodeIter);
93       aPnt.Transform (aTrsf);
94       aMesh->ChangeNode (aNodeIter + aNodeOffset) = aPnt;
95     }
96
97     // copy triangles
98     const TopAbs_Orientation anOrientation = anExpSF.Current().Orientation();
99     for (Standard_Integer aTriIter = aTriangles.Lower(); aTriIter <= aTriangles.Upper(); ++aTriIter)
100     {
101       Poly_Triangle aTri = aTriangles (aTriIter);
102
103       Standard_Integer anId[3];
104       aTri.Get (anId[0], anId[1], anId[2]);
105       if (anOrientation == TopAbs_REVERSED)
106       {
107         // Swap 1, 2.
108         Standard_Integer aTmpIdx = anId[1];
109         anId[1] = anId[2];
110         anId[2] = aTmpIdx;
111       }
112
113       // Update nodes according to the offset.
114       anId[0] += aNodeOffset;
115       anId[1] += aNodeOffset;
116       anId[2] += aNodeOffset;
117
118       aTri.Set (anId[0], anId[1], anId[2]);
119       aMesh->ChangeTriangle (aTriIter + aTriangleOffet) =  aTri;
120     }
121
122     aNodeOffset += aNodes.Size();
123     aTriangleOffet += aTriangles.Size();
124   }
125
126   OSD_Path aPath (theFileName);
127   Standard_Boolean isDone = (myASCIIMode
128        ? RWStl::WriteAscii  (aMesh, aPath)
129        : RWStl::WriteBinary (aMesh, aPath));
130
131   if (isDone && (aNbFacesNoTri > 0))
132   {
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);
140   }
141
142   return isDone;
143 }