medit_io.C
Go to the documentation of this file.
1 // The libMesh Finite Element Library.
2 // Copyright (C) 2002-2014 Benjamin S. Kirk, John W. Peterson, Roy H. Stogner
3 
4 // This library is free software; you can redistribute it and/or
5 // modify it under the terms of the GNU Lesser General Public
6 // License as published by the Free Software Foundation; either
7 // version 2.1 of the License, or (at your option) any later version.
8 
9 // This library is distributed in the hope that it will be useful,
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 // Lesser General Public License for more details.
13 
14 // You should have received a copy of the GNU Lesser General Public
15 // License along with this library; if not, write to the Free Software
16 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 
18 
19 
20 // C++ includes
21 #include <iomanip>
22 #include <fstream>
23 
24 // Local includes
25 #include "libmesh/libmesh_config.h"
27 #include "libmesh/mesh_base.h"
28 #include "libmesh/medit_io.h"
29 #include "libmesh/elem.h"
30 
31 namespace libMesh
32 {
33 
34 
35 // ------------------------------------------------------------
36 // MEDITIO members
37 void MEDITIO::write (const std::string& fname)
38 {
39  if (this->mesh().processor_id() == 0)
40  if (!this->binary())
41  this->write_ascii (fname);
42 }
43 
44 
45 
46 void MEDITIO::write_nodal_data (const std::string& fname,
47  const std::vector<Number>& soln,
48  const std::vector<std::string>& names)
49 {
50  START_LOG("write_nodal_data()", "MEDITIO");
51 
52  if (this->mesh().processor_id() == 0)
53  if (!this->binary())
54  this->write_ascii (fname, &soln, &names);
55 
56  STOP_LOG("write_nodal_data()", "MEDITIO");
57 }
58 
59 
60 
61 void MEDITIO::write_ascii (const std::string& fname,
62  const std::vector<Number>* vec,
63  const std::vector<std::string>* solution_names)
64 {
65  // Current lacks in implementation:
66  // (i) only 3D meshes.
67  // (ii) only QUAD4, TRI3, TET4 elements, others are omitted !
68  // (iii) no distinction between materials.
69  // (iv) no vector output, just first scalar as output
70 
71  // libmesh_assert three dimensions (should be extended later)
72  libmesh_assert_equal_to (MeshOutput<MeshBase>::mesh().mesh_dimension(), 3);
73 
74  // Open the output file stream
75  std::ofstream out_stream (fname.c_str());
76 
77  // Make sure it opened correctly
78  if (!out_stream.good())
79  libmesh_file_error(fname.c_str());
80 
81  // Get a reference to the mesh
82  const MeshBase& the_mesh = MeshOutput<MeshBase>::mesh();
83 
84  // Begin interfacing with the MEdit data file
85  {
86  // header:
87  out_stream << "MeshVersionFormatted 1\n";
88  out_stream << "Dimension 3\n";
89  out_stream << "# Mesh generated by libmesh\n\n";
90 
91  // write the nodes:
92  out_stream << "# Set of mesh vertices\n";
93  out_stream << "Vertices\n";
94  out_stream << the_mesh.n_nodes() << "\n";
95 
96  for (unsigned int v=0; v<the_mesh.n_nodes(); v++)
97  out_stream << the_mesh.point(v)(0) << " " << the_mesh.point(v)(1) << " " << the_mesh.point(v)(2) << " 0\n";
98  }
99 
100  {
101  // write the connectivity:
102  out_stream << "\n# Set of Polys\n\n";
103 
104  // count occurrences of output elements:
105  int n_tri3 = 0;
106  int n_quad4 = 0;
107  int n_tet4 = 0;
108 
109  {
112 
113  for ( ; it != end; ++it)
114  {
115  if ((*it)->type() == TRI3) n_tri3++;
116  if ((*it)->type() == QUAD4) n_quad4++;
117  if ((*it)->type() == QUAD9) n_quad4+=4; // (QUAD9 is written as 4 QUAD4.)
118  if ((*it)->type() == TET4) n_tet4++;
119  } // for
120  }
121 
122  // First: write out TRI3 elements:
123  out_stream << "Triangles\n";
124  out_stream << n_tri3 << "\n";
125 
126  {
129 
130  for ( ; it != end; ++it)
131  if ((*it)->type() == TRI3)
132  out_stream << (*it)->node(0)+1 << " " << (*it)->node(1)+1 << " " << (*it)->node(2)+1 << " 0\n";
133  }
134 
135  // Second: write out QUAD4 elements:
136  out_stream << "Quadrilaterals\n";
137  out_stream << n_quad4 << "\n";
138 
139  {
142 
143  for ( ; it != end; ++it)
144  if ((*it)->type() == QUAD4)
145  {
146  out_stream << (*it)->node(0)+1 << " "
147  << (*it)->node(1)+1 << " "
148  << (*it)->node(2)+1 << " "
149  << (*it)->node(3)+1 <<" 0\n";
150  } // if
151  else if ((*it)->type() == QUAD9)
152  {
153  out_stream << (*it)->node(0)+1 << " "
154  << (*it)->node(4)+1 << " "
155  << (*it)->node(8)+1 << " "
156  << (*it)->node(7)+1 <<" 0\n";
157  out_stream << (*it)->node(7)+1 << " "
158  << (*it)->node(8)+1 << " "
159  << (*it)->node(6)+1 << " "
160  << (*it)->node(3)+1 <<" 0\n";
161  out_stream << (*it)->node(4)+1 << " "
162  << (*it)->node(1)+1 << " "
163  << (*it)->node(5)+1 << " "
164  << (*it)->node(8)+1 <<" 0\n";
165  out_stream << (*it)->node(8)+1 << " "
166  << (*it)->node(5)+1 << " "
167  << (*it)->node(2)+1 << " "
168  << (*it)->node(6)+1 <<" 0\n";
169  } // if
170  }
171 
172 
173  // Third: write out TET4 elements:
174  out_stream << "Tetrahedra\n";
175  out_stream << n_tet4 << "\n";
176 
177  {
180 
181  for ( ; it != end; ++it)
182  if ((*it)->type() == TET4)
183  {
184  out_stream << (*it)->node(0)+1 << " "
185  << (*it)->node(1)+1 << " "
186  << (*it)->node(2)+1 << " "
187  << (*it)->node(3)+1 <<" 0\n";
188  } // if
189  }
190 
191  }
192  // end of the out file
193  out_stream << '\n' << "# end of file\n";
194 
195  // optionally write the data
196  if ((solution_names != NULL) &&
197  (vec != NULL))
198  {
199  // Open the ".bb" file stream
200  std::size_t idx = fname.find_last_of(".");
201  std::string bbname = fname.substr(0,idx) + ".bb";
202 
203  std::ofstream bbout (bbname.c_str());
204 
205  // Make sure it opened correctly
206  if (!bbout.good())
207  libmesh_file_error(bbname.c_str());
208 
209  // Header: 3: 3D mesh, 1: scalar output, 2: node-indexed
210  const std::size_t n_vars = solution_names->size();
211  bbout << "3 1 " << the_mesh.n_nodes() << " 2\n";
212  for (dof_id_type n=0; n<the_mesh.n_nodes(); n++)
213  bbout << std::setprecision(10) << (*vec)[n*n_vars + scalar_idx] << " ";
214  bbout << "\n";
215  } // endif
216 }
217 
218 } // namespace libMesh

Site Created By: libMesh Developers
Last modified: February 07 2014 16:57:06 UTC

Hosted By:
SourceForge.net Logo