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 }