Branch data Line data Source code
1 : : // *****************************************************************************
2 : : /*!
3 : : \file src/IO/GmshMeshWriter.cpp
4 : : \copyright 2012-2015 J. Bakosi,
5 : : 2016-2018 Los Alamos National Security, LLC.,
6 : : 2019-2021 Triad National Security, LLC.
7 : : All rights reserved. See the LICENSE file for details.
8 : : \brief Gmsh mesh writer class definition
9 : : \details Gmsh mesh writer class definition. Currently, this class supports
10 : : line, triangle, tetrahedron, and point Gmsh element types.
11 : : */
12 : : // *****************************************************************************
13 : :
14 : : #include <iterator>
15 : : #include <iomanip>
16 : : #include <algorithm>
17 : : #include <cstddef>
18 : : #include <ostream>
19 : : #include <string>
20 : : #include <utility>
21 : :
22 : : #include "Exception.hpp"
23 : : #include "UnsMesh.hpp"
24 : : #include "PrintUtil.hpp"
25 : : #include "GmshMeshWriter.hpp"
26 : : #include "GmshMeshIO.hpp"
27 : :
28 : : using tk::GmshMeshWriter;
29 : :
30 : 4 : GmshMeshWriter::GmshMeshWriter( const std::string& filename,
31 : : GmshFileType type,
32 : : tk::real version,
33 : 4 : int datasize ) :
34 : 4 : Writer( filename ), m_type( type )
35 : : // *****************************************************************************
36 : : // Constructor: write mandatory "$MeshFormat" section
37 : : //! \param[in] filename File to open as a Gmsh file
38 : : //! \param[in] type Gmsh file type: ASCII or binary
39 : : //! \param[in] version Gmsh file version
40 : : //! \param[in] datasize Size of double precision number on machine
41 : : // *****************************************************************************
42 : : {
43 : : using tk::operator<<;
44 : :
45 : : // Write beginning of header: $MeshFormat
46 [ + - ]: 4 : m_outFile << "$MeshFormat\n";
47 [ - + ][ - - ]: 4 : ErrChk( !m_outFile.bad(), "Failed to write to file: " + m_filename );
[ - - ][ - - ]
[ - - ][ - - ]
48 : :
49 : : // Write "version-number file-type data-size"
50 [ + - ][ + - ]: 4 : m_outFile << version << " " << type << " " << datasize << "\n";
51 [ - + ][ - - ]: 4 : ErrChk( !m_outFile.bad(), "Failed to write to file: " + m_filename );
[ - - ][ - - ]
[ - - ][ - - ]
52 : :
53 [ + - ]: 4 : if (isBinary()) {
54 : 4 : int one = 1;
55 [ + - ]: 4 : m_outFile.write( reinterpret_cast<char*>(&one), sizeof(int) );
56 : : m_outFile << std::endl;
57 : : }
58 : :
59 : : // Write end of header: $EndMeshFormat
60 : : m_outFile << "$EndMeshFormat" << std::endl;
61 [ - + ][ - - ]: 4 : ErrChk( !m_outFile.bad(), "Failed to write to file: " + m_filename );
[ - - ][ - - ]
[ - - ][ - - ]
62 : 4 : }
63 : :
64 : : void
65 : 4 : GmshMeshWriter::writeMesh( const UnsMesh& mesh )
66 : : // *****************************************************************************
67 : : // Write Gmsh mesh file
68 : : //! \param[in] mesh Unstructured mesh object
69 : : // *****************************************************************************
70 : : {
71 : : // Write sections
72 : 4 : writeNodes( mesh );
73 : 4 : writeElements( mesh );
74 : 4 : }
75 : :
76 : : void
77 : 4 : GmshMeshWriter::writeNodes( const UnsMesh& mesh )
78 : : // *****************************************************************************
79 : : // Write "$Nodes--$EndNodes" section
80 : : //! \param[in] mesh Unstructured mesh object
81 : : // *****************************************************************************
82 : : {
83 : 4 : m_outFile << "$Nodes" << std::endl;
84 : :
85 : : // Write out number of nodes
86 : : m_outFile << mesh.nnode() << std::endl;
87 : :
88 : : // Write node ids and coordinates: node-number x-coord y-coord z-coord
89 [ + - ]: 4 : if (isASCII()) {
90 [ - - ]: 0 : for (std::size_t i=0; i<mesh.nnode(); ++i) {
91 : 0 : m_outFile << i+1 << " " << std::setprecision(16)
92 : 0 : << mesh.x()[i] << " "
93 : 0 : << mesh.y()[i] << " "
94 : 0 : << mesh.z()[i] << std::endl;
95 : : }
96 : : } else {
97 [ + + ]: 19247 : for (std::size_t i=0; i<mesh.nnode(); ++i) {
98 : : // gmsh likes one-based node ids
99 : 19243 : int I = static_cast< int >( i+1 );
100 : : m_outFile.write(
101 : 19243 : reinterpret_cast<const char*>(&I), sizeof(int) );
102 : : m_outFile.write(
103 : 19243 : reinterpret_cast<const char*>(&mesh.x()[i]), sizeof(double) );
104 : : m_outFile.write(
105 : 19243 : reinterpret_cast<const char*>(&mesh.y()[i]), sizeof(double) );
106 : : m_outFile.write(
107 : 19243 : reinterpret_cast<const char*>(&mesh.z()[i]), sizeof(double) );
108 : : }
109 : : m_outFile << std::endl;
110 : : }
111 : :
112 : : m_outFile << "$EndNodes" << std::endl;
113 : 4 : }
114 : :
115 : : void
116 : 4 : GmshMeshWriter::writeElements( const UnsMesh& mesh )
117 : : // *****************************************************************************
118 : : // Write "$Elements--$EndElements" section
119 : : //! \param[in] mesh Unstructured mesh object
120 : : // *****************************************************************************
121 : : {
122 : 4 : m_outFile << "$Elements" << std::endl;
123 : :
124 : : // Write out number of elements
125 : 4 : m_outFile << mesh.lininpoel().size()/2 +
126 : 4 : mesh.triinpoel().size()/3 +
127 : 4 : mesh.tetinpoel().size()/4
128 : : << std::endl;
129 : :
130 : : // Write out line element ids and connectivity (node list)
131 : 4 : writeElemBlock( 2, GmshElemType::LIN, mesh.lininpoel() );
132 : :
133 : : // Write out triangle element ids and connectivity (node list)
134 : 4 : writeElemBlock( 3, GmshElemType::TRI, mesh.triinpoel() );
135 : :
136 : : // Write out terahedron element ids and connectivity (node list)
137 : 4 : writeElemBlock( 4, GmshElemType::TET, mesh.tetinpoel() );
138 : :
139 [ + - ]: 4 : if (isBinary()) m_outFile << std::endl;
140 : : m_outFile << "$EndElements" << std::endl;
141 : 4 : }
142 : :
143 : : void
144 [ + + ]: 12 : GmshMeshWriter::writeElemBlock( std::size_t nnpe,
145 : : GmshElemType type,
146 : : const std::vector< std::size_t >& inpoel )
147 : : // *****************************************************************************
148 : : // Write element block: element ids and connectivity (node list)
149 : : //! \param[in] nnpe Number of nodes per element
150 : : //! \param[in] type Element type
151 : : //! \param[in] inpoel Element connectivity (must be zero-based)
152 : : // *****************************************************************************
153 : : {
154 : : // Return if connectivity is empty, there is no such element block in mesh
155 [ + + ]: 12 : if (inpoel.empty()) return;
156 : :
157 : : // Make sure element connectivity starts with zero
158 : : Assert( *std::minmax_element( begin(inpoel), end(inpoel) ).first == 0,
159 : : "node ids should start from zero" );
160 : :
161 : : // Get number of elements in mesh
162 [ + - ]: 8 : auto n = inpoel.size()/nnpe;
163 : :
164 : : // Ignore element tags
165 : 8 : std::vector< std::vector< int > > tg;
166 [ + - ]: 8 : tg.resize( n );
167 [ + + ][ + - ]: 109340 : for (auto& t : tg) t.push_back( 0 );
168 : :
169 [ - + ]: 8 : if (isASCII()) {
170 : :
171 [ - - ]: 0 : for (std::size_t i=0; i<n; i++) {
172 : : // elm-number elm-type number-of-tags < tag > ... node-number-list
173 [ - - ][ - - ]: 0 : m_outFile << i+1 << " " << type << " " << tg[i].size() << " ";
[ - - ][ - - ]
174 [ - - ]: 0 : copy( tg[i].begin(), tg[i].end()-1,
175 : : std::ostream_iterator< int >( m_outFile, " " ) );
176 [ - - ][ - - ]: 0 : m_outFile << tg[i].back() << " ";
177 : :
178 : : // gmsh likes one-based node ids
179 [ - - ][ - - ]: 0 : for (std::size_t k=0; k<nnpe; k++) m_outFile << inpoel[i*nnpe+k]+1 << " ";
[ - - ]
180 : : m_outFile << std::endl;
181 : : }
182 : :
183 : : } else {
184 : :
185 [ + - ]: 8 : int ntags = static_cast< int >( tg[0].size() );
186 : 8 : int nel = static_cast< int >( n );
187 : : // elm-type num-of-elm-follow number-of-tags
188 [ + - ]: 8 : m_outFile.write( reinterpret_cast<char*>(&type), sizeof(int) );
189 [ + - ]: 8 : m_outFile.write( reinterpret_cast<char*>(&nel), sizeof(int) );
190 [ + - ]: 8 : m_outFile.write( reinterpret_cast<char*>(&ntags), sizeof(int) );
191 [ + + ]: 109340 : for (std::size_t i=0; i<n; i++) {
192 : 109332 : int I = static_cast< int >( i );
193 : : // gmsh likes one-based node ids
194 : : std::vector< int > Inpoel;
195 [ + + ]: 536532 : for (std::size_t k=0; k<nnpe; ++k)
196 [ + - ][ - - ]: 427200 : Inpoel.push_back( static_cast< int >( inpoel[i*nnpe+k]+1 ) );
197 : : // element id
198 [ + - ]: 109332 : m_outFile.write( reinterpret_cast<const char*>(&I), sizeof(int) );
199 : : // element tags
200 : : m_outFile.write( reinterpret_cast<const char*>(tg[i].data()),
201 [ + - ]: 109332 : static_cast<std::streamsize>(tg[i].size()*sizeof(int)) );
202 : : // element node list (i.e. connectivity)
203 : 109332 : m_outFile.write( reinterpret_cast<const char*>(Inpoel.data()),
204 [ + - ]: 109332 : static_cast<std::streamsize>(nnpe*sizeof(int)) );
205 : : }
206 : :
207 : : }
208 : : }
|
