Line data Source code
1 : // The libMesh Finite Element Library.
2 : // Copyright (C) 2002-2026 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 : #ifndef LIBMESH_EXODUSII_IO_HELPER_H
19 : #define LIBMESH_EXODUSII_IO_HELPER_H
20 :
21 : #include "libmesh/libmesh_config.h"
22 :
23 : #ifdef LIBMESH_HAVE_EXODUS_API
24 :
25 : // Local includes
26 : #include "libmesh/parallel_object.h"
27 : #include "libmesh/point.h"
28 : #include "libmesh/boundary_info.h" // BoundaryInfo::BCTuple
29 : #include "libmesh/enum_elem_type.h" // INVALID_ELEM
30 : #include "libmesh/exodus_header_info.h"
31 :
32 : // C++ includes
33 : #include <iostream>
34 : #include <string>
35 : #include <vector>
36 : #include <map>
37 :
38 : // Macros to simplify checking Exodus error codes
39 : #define EX_CHECK_ERR(code, msg) \
40 : do { \
41 : if ((code) < 0) { \
42 : libmesh_error_msg((msg)); \
43 : } } while (0)
44 :
45 : #define EX_EXCEPTIONLESS_CHECK_ERR(code, msg) \
46 : do { \
47 : if ((code) < 0) { \
48 : libMesh::err << (msg) << std::endl; \
49 : libmesh_exceptionless_error(); \
50 : } } while (0)
51 :
52 : // Before we include a header wrapped in a namespace, we'd better make
53 : // sure none of its dependencies end up in that namespace
54 : #include <errno.h>
55 : #include <stddef.h>
56 : #include <stdlib.h>
57 : #include <stdint.h>
58 :
59 : namespace libMesh
60 : {
61 :
62 : // Forward declarations
63 : class MeshBase;
64 : class DofObject;
65 :
66 : /**
67 : * This is the \p ExodusII_IO_Helper class. This class hides the
68 : * implementation details of interfacing with the Exodus binary
69 : * format.
70 : *
71 : * \author John W. Peterson
72 : * \date 2002
73 : */
74 8764 : class ExodusII_IO_Helper : public ParallelObject
75 : {
76 : public:
77 : /**
78 : * Constructor. Automatically initializes all the private members of
79 : * the class. Also allows you to set the verbosity level to v=true
80 : * (on) or v=false (off). The second argument, if true, tells the class to only
81 : * perform its actions if running on processor zero. If you initialize this
82 : * to false, the writing methods will run on all processors instead.
83 : */
84 : ExodusII_IO_Helper(const ParallelObject & parent,
85 : bool v=false,
86 : bool run_only_on_proc0=true,
87 : bool single_precision=false);
88 : /**
89 : * Special functions. This class does not manage any dynamically
90 : * allocated resources (file pointers, etc.) so it _should_ be
91 : * default copy/move constructable, but I don't know
92 : * if any existing code actually uses these operations.
93 : */
94 : ExodusII_IO_Helper (const ExodusII_IO_Helper &) = default;
95 : ExodusII_IO_Helper (ExodusII_IO_Helper &&) = default;
96 : virtual ~ExodusII_IO_Helper();
97 :
98 : /**
99 : * This class contains references so it can't be default
100 : * copy/move-assigned.
101 : */
102 : ExodusII_IO_Helper & operator= (const ExodusII_IO_Helper &) = delete;
103 : ExodusII_IO_Helper & operator= (ExodusII_IO_Helper &&) = delete;
104 :
105 : /**
106 : * \returns The ExodusII API version, in "nodot" format; e.g. 822
107 : * for 8.22
108 : */
109 : static int get_exodus_version();
110 :
111 : /**
112 : * \returns The current element type.
113 : *
114 : * \note The default behavior is for this value to be in all capital
115 : * letters, e.g. \p HEX27.
116 : */
117 : const char * get_elem_type() const;
118 :
119 : /**
120 : * Sets whether or not to write extra "side" elements. This is useful for
121 : * plotting SIDE_DISCONTINUOUS data.
122 : */
123 : void set_add_sides(bool add_sides);
124 :
125 : bool get_add_sides();
126 :
127 : /**
128 : * Opens an \p ExodusII mesh file named \p filename. If
129 : * read_only==true, the file will be opened with the EX_READ flag,
130 : * otherwise it will be opened with the EX_WRITE flag.
131 : */
132 : void open(const char * filename, bool read_only);
133 :
134 : /**
135 : * Reads an \p ExodusII mesh file header, leaving this object's
136 : * internal data structures unchanged.
137 : */
138 : ExodusHeaderInfo read_header() const;
139 :
140 : /**
141 : * Reads an \p ExodusII mesh file header, and stores required
142 : * information on this object.
143 : */
144 : void read_and_store_header_info();
145 :
146 : /**
147 : * Reads the QA records from an ExodusII file. We can use this to
148 : * detect when e.g. CUBIT 14+ was used to generate a Mesh file, and
149 : * work around certain known bugs in that version.
150 : */
151 : void read_qa_records();
152 :
153 : /**
154 : * Prints the \p ExodusII mesh file header, which includes the mesh
155 : * title, the number of nodes, number of elements, mesh dimension,
156 : * number of sidesets, and number of nodesets.
157 : */
158 : void print_header();
159 :
160 : /**
161 : * Reads the nodal data (x,y,z coordinates) from the \p ExodusII
162 : * mesh file.
163 : */
164 : void read_nodes();
165 :
166 : /**
167 : * Reads the optional \p node_num_map from the \p ExodusII mesh
168 : * file.
169 : */
170 : void read_node_num_map();
171 :
172 : /**
173 : * Reads the optional \p bex_cv_blocks from the \p ExodusII mesh
174 : * file.
175 : */
176 : void read_bex_cv_blocks();
177 :
178 : /**
179 : * Prints the nodal information, by default to \p libMesh::out.
180 : */
181 : void print_nodes(std::ostream & out_stream = libMesh::out);
182 :
183 : /**
184 : * Reads information for all of the blocks in the \p ExodusII mesh
185 : * file.
186 : */
187 : void read_block_info();
188 :
189 : /**
190 : * Get the block number for the given block index.
191 : */
192 : int get_block_id(int index);
193 :
194 : /**
195 : * Get the block name for the given block index if supplied in
196 : * the mesh file. Otherwise an empty string is returned.
197 : */
198 : std::string get_block_name(int index);
199 :
200 : /**
201 : * Get the side set id for the given side set index.
202 : */
203 : int get_side_set_id(int index);
204 :
205 : /**
206 : * Get the side set name for the given side set index if supplied in
207 : * the mesh file. Otherwise an empty string is returned.
208 : */
209 : std::string get_side_set_name(int index);
210 :
211 : /**
212 : * Get the node set id for the given node set index.
213 : */
214 : int get_node_set_id(int index);
215 :
216 : /**
217 : * Get the node set name for the given node set index if supplied in
218 : * the mesh file. Otherwise an empty string is returned.
219 : */
220 : std::string get_node_set_name(int index);
221 :
222 : /**
223 : * Reads all of the element connectivity for block \p block in the
224 : * \p ExodusII mesh file.
225 : */
226 : void read_elem_in_block(int block);
227 :
228 : /**
229 : * Read in edge blocks, storing information in the BoundaryInfo object.
230 : */
231 : void read_edge_blocks(MeshBase & mesh);
232 :
233 : /**
234 : * Reads the optional \p node_num_map from the \p ExodusII mesh
235 : * file.
236 : */
237 : void read_elem_num_map();
238 :
239 : /**
240 : * Reads information about all of the sidesets in the \p ExodusII
241 : * mesh file.
242 : */
243 : void read_sideset_info();
244 :
245 : /**
246 : * Reads information about all of the nodesets in the \p ExodusII
247 : * mesh file.
248 : */
249 : void read_nodeset_info();
250 :
251 : /**
252 : * Reads information about all of the elemsets in the \p ExodusII
253 : * mesh file.
254 : */
255 : void read_elemset_info();
256 :
257 : /**
258 : * Reads information about sideset \p id and inserts it into the
259 : * global sideset array at the position \p offset.
260 : */
261 : void read_sideset(int id, int offset);
262 :
263 : /**
264 : * Reads information about elemset \p id and inserts it into the
265 : * global elemset array at the position \p offset.
266 : */
267 : void read_elemset(int id, int offset);
268 :
269 : /**
270 : * New API that reads all nodesets simultaneously. This may be slightly
271 : * faster than reading them one at a time. Calls ex_get_concat_node_sets()
272 : * under the hood.
273 : */
274 : void read_all_nodesets();
275 :
276 : /**
277 : * Closes the \p ExodusII mesh file.
278 : *
279 : * This function is called from the ExodusII_IO destructor, so it should
280 : * not throw an exception.
281 : */
282 : void close() noexcept;
283 :
284 : /**
285 : * Reads and stores the timesteps in the 'time_steps' array.
286 : */
287 : void read_time_steps();
288 :
289 : /**
290 : * Reads the number of timesteps currently stored in the Exodus file
291 : * and stores it in the num_time_steps variable.
292 : */
293 : void read_num_time_steps();
294 :
295 : /**
296 : * Reads the nodal values for the variable 'nodal_var_name' at the
297 : * specified time into the 'nodal_var_values' array.
298 : */
299 : void read_nodal_var_values(std::string nodal_var_name, int time_step);
300 :
301 : /**
302 : * Reads elemental values for the variable 'elemental_var_name' at the
303 : * specified timestep into the 'elem_var_value_map' which is passed in.
304 : */
305 : void read_elemental_var_values(std::string elemental_var_name,
306 : int time_step,
307 : std::map<dof_id_type, Real> & elem_var_value_map);
308 :
309 : /**
310 : * Helper function that takes a (1-based) Exodus node/elem id and
311 : * determines the corresponding libMesh Node/Elem id. Takes into account
312 : * whether the user has chosen to set the Node/Elem unique ids based on
313 : * the {node,elem}_num_map or to let libMesh set them.
314 : */
315 : dof_id_type get_libmesh_node_id(int exodus_node_id);
316 : dof_id_type get_libmesh_elem_id(int exodus_elem_id);
317 :
318 : /**
319 : * Helper function that conditionally sets the unique_id of the
320 : * passed-in Node/Elem. Calling this function does nothing if
321 : * _set_unique_ids_from_maps == false, otherwise it sets the
322 : * unique_id based on the entries of the {node,elem_num_map}. The
323 : * input index is assumed to be a zero-based index into the
324 : * {node,elem}_num_map array.
325 : */
326 : void conditionally_set_node_unique_id(
327 : MeshBase & mesh, Node * node, int zero_based_node_num_map_index);
328 : void conditionally_set_elem_unique_id(
329 : MeshBase & mesh, Elem * elem, int zero_based_elem_num_map_index);
330 :
331 : private:
332 :
333 : /**
334 : * Internal implementation for the two sets of functions above.
335 : */
336 : dof_id_type get_libmesh_id(
337 : int exodus_id,
338 : const std::vector<int> & num_map);
339 :
340 : void set_dof_object_unique_id(
341 : MeshBase & mesh,
342 : DofObject * dof_object,
343 : int exodus_mapped_id);
344 :
345 : public:
346 :
347 : /**
348 : * Opens an \p ExodusII mesh file named \p filename for writing.
349 : */
350 : virtual void create(std::string filename);
351 :
352 : /**
353 : * Initializes the Exodus file.
354 : */
355 : virtual void initialize(std::string title, const MeshBase & mesh, bool use_discontinuous=false);
356 :
357 : /**
358 : * Writes the nodal coordinates contained in "mesh"
359 : */
360 : virtual void write_nodal_coordinates(const MeshBase & mesh, bool use_discontinuous=false);
361 :
362 : /**
363 : * Writes the elements contained in "mesh". FIXME: This only works
364 : * for Meshes having a single type of element in each subdomain!
365 : *
366 : * If \p use_discontinuous is true, we break apart elements, so that
367 : * shared nodes on faces/edges/vertices can take different values
368 : * from different elements. This is useful for plotting
369 : * discontinuous underlying variables
370 : *
371 : * If \p _add_sides is true, we also output side elements, so that
372 : * shared nodes on edges/vertices can take different values from
373 : * different elements. This is useful for plotting
374 : * SIDE_DISCONTINUOUS representing e.g. inter-element fluxes.
375 : */
376 : virtual void write_elements(const MeshBase & mesh,
377 : bool use_discontinuous=false);
378 :
379 : /**
380 : * Writes the sidesets contained in "mesh"
381 : */
382 : virtual void write_sidesets(const MeshBase & mesh);
383 :
384 : /**
385 : * Writes the nodesets contained in "mesh"
386 : */
387 : virtual void write_nodesets(const MeshBase & mesh);
388 :
389 : /**
390 : * Sets up the nodal variables
391 : */
392 : virtual void initialize_element_variables(std::vector<std::string> names,
393 : const std::vector<std::set<subdomain_id_type>> & vars_active_subdomains);
394 :
395 : /**
396 : * Sets up the nodal variables
397 : */
398 : void initialize_nodal_variables(std::vector<std::string> names);
399 :
400 : /**
401 : * Sets up the global variables
402 : */
403 : void initialize_global_variables(std::vector<std::string> names);
404 :
405 : /**
406 : * Writes the time for the timestep
407 : */
408 : void write_timestep(int timestep, Real time);
409 :
410 : /**
411 : * Write elemsets stored on the Mesh to the exo file.
412 : */
413 : void write_elemsets(const MeshBase & mesh);
414 :
415 : /**
416 : * Write sideset data for the requested timestep.
417 : */
418 : void
419 : write_sideset_data (const MeshBase & mesh,
420 : int timestep,
421 : const std::vector<std::string> & var_names,
422 : const std::vector<std::set<boundary_id_type>> & side_ids,
423 : const std::vector<std::map<BoundaryInfo::BCTuple, Real>> & bc_vals);
424 :
425 : /**
426 : * Read sideset variables, if any, into the provided data structures.
427 : */
428 : void
429 : read_sideset_data (const MeshBase & mesh,
430 : int timestep,
431 : std::vector<std::string> & var_names,
432 : std::vector<std::set<boundary_id_type>> & side_ids,
433 : std::vector<std::map<BoundaryInfo::BCTuple, Real>> & bc_vals);
434 :
435 : /**
436 : * Similar to read_sideset_data(), but instead of creating one
437 : * std::map per sideset per variable, creates a single map of (elem,
438 : * side, boundary_id) tuples, and stores the exo file array indexing
439 : * for any/all sideset variables on that sideset (they are all the
440 : * same). This function does not actually call exII::ex_get_sset_var()
441 : * to get values, and can be useful if only the original ordering of
442 : * (elem, side) pairs in the exo file is required in cases where a
443 : * separate approach is used to read the sideset data arrays.
444 : */
445 : void
446 : get_sideset_data_indices (const MeshBase & mesh,
447 : std::map<BoundaryInfo::BCTuple, unsigned int> & bc_array_indices);
448 :
449 : /**
450 : * Write nodeset data for the requested timestep.
451 : */
452 : void
453 : write_nodeset_data (int timestep,
454 : const std::vector<std::string> & var_names,
455 : const std::vector<std::set<boundary_id_type>> & node_boundary_ids,
456 : const std::vector<std::map<BoundaryInfo::NodeBCTuple, Real>> & bc_vals);
457 :
458 : /**
459 : * Read nodeset variables, if any, into the provided data structures.
460 : */
461 : void
462 : read_nodeset_data (int timestep,
463 : std::vector<std::string> & var_names,
464 : std::vector<std::set<boundary_id_type>> & node_boundary_ids,
465 : std::vector<std::map<BoundaryInfo::NodeBCTuple, Real>> & bc_vals);
466 :
467 : /**
468 : * Similar to read_nodeset_data(), but instead of creating one
469 : * std::map per nodeset per variable, creates a single map of
470 : * (node_id, boundary_id) tuples, and stores the exo file array
471 : * indexing for any/all nodeset variables on that nodeset (they are
472 : * all the same).
473 : */
474 : void
475 : get_nodeset_data_indices (std::map<BoundaryInfo::NodeBCTuple, unsigned int> & bc_array_indices);
476 :
477 : /**
478 : * Write elemset data for the requested timestep.
479 : */
480 : void
481 : write_elemset_data (int timestep,
482 : const std::vector<std::string> & var_names,
483 : const std::vector<std::set<elemset_id_type>> & elemset_ids_in,
484 : const std::vector<std::map<std::pair<dof_id_type, elemset_id_type>, Real>> & elemset_vals);
485 :
486 : /**
487 : * Read elemset variables, if any, into the provided data structures.
488 : */
489 : void
490 : read_elemset_data (int timestep,
491 : std::vector<std::string> & var_names,
492 : std::vector<std::set<elemset_id_type>> & elemset_ids_in,
493 : std::vector<std::map<std::pair<dof_id_type, elemset_id_type>, Real>> & elemset_vals);
494 :
495 : /**
496 : * Similar to read_elemset_data(), but instead of creating one
497 : * std::map per elemset per variable, creates a single map of
498 : * (elem_id, elemset_id) tuples, and stores the exo file array
499 : * indexing for any/all elemset variables on that elemset (they are
500 : * all the same).
501 : */
502 : void
503 : get_elemset_data_indices (std::map<std::pair<dof_id_type, elemset_id_type>, unsigned int> & elemset_array_indices);
504 :
505 : /**
506 : * Writes the vector of values to the element variables.
507 : *
508 : * The 'values' vector is assumed to be in the order:
509 : * {(u1, u2, u3, ..., uN), (v1, v2, v3, ..., vN), (w1, w2, w3, ..., wN)}
510 : * where N is the number of elements.
511 : *
512 : * This ordering is produced by calls to ES::build_elemental_solution_vector().
513 : * ES::build_discontinuous_solution_vector(), on the other hand, produces an
514 : * element-major ordering. See the function below for that case.
515 : */
516 : void write_element_values
517 : (const MeshBase & mesh,
518 : const std::vector<Real> & values,
519 : int timestep,
520 : const std::vector<std::set<subdomain_id_type>> & vars_active_subdomains);
521 :
522 : /**
523 : * Same as the function above, but assume the input 'values' vector is
524 : * in element-major order, i.e.
525 : * {(u1,v1,w1), (u2,v2,w2), ... (uN,vN,wN)}
526 : * This function is called by
527 : * ExodusII_IO::write_element_data_from_discontinuous_nodal_data()
528 : * because ES::build_discontinuous_solution_vector() builds the solution
529 : * vector in this order.
530 : *
531 : * \note If some variables are subdomain-restricted, then the tuples will
532 : * be of different lengths for each element, i.e.
533 : * {(u1,v1,w1), (u2,v2), ... (uN,vN,wN)}
534 : * if variable w is not active on element 2.
535 : */
536 : void write_element_values_element_major
537 : (const MeshBase & mesh,
538 : const std::vector<Real> & values,
539 : int timestep,
540 : const std::vector<std::set<subdomain_id_type>> & vars_active_subdomains,
541 : const std::vector<std::string> & derived_var_names,
542 : const std::map<subdomain_id_type, std::vector<std::string>> & subdomain_to_var_names);
543 :
544 : /**
545 : * Writes the vector of values to a nodal variable.
546 : */
547 : void write_nodal_values(int var_id, const std::vector<Real> & values, int timestep);
548 :
549 : /**
550 : * Writes the vector of information records.
551 : */
552 : void write_information_records(const std::vector<std::string> & records);
553 :
554 : /**
555 : * Writes the vector of global variables.
556 : */
557 : void write_global_values(const std::vector<Real> & values, int timestep);
558 :
559 : /**
560 : * Uses ex_update() to flush buffers to file.
561 : */
562 : void update();
563 :
564 : /**
565 : * Reads the vector of global variables.
566 : */
567 : void read_global_values(std::vector<Real> & values, int timestep);
568 :
569 : /**
570 : * Sets the underlying value of the boolean flag
571 : * _use_mesh_dimension_instead_of_spatial_dimension. By default,
572 : * the value of this flag is false.
573 : *
574 : * See the ExodusII_IO class documentation for a detailed
575 : * description of this flag.
576 : */
577 : void use_mesh_dimension_instead_of_spatial_dimension(bool val);
578 :
579 : /**
580 : * Set to true (the default) to write files in an HDF5-based file
581 : * format (when HDF5 is available), or to false to write files in
582 : * the old NetCDF3-based format. If HDF5 is unavailable, this
583 : * setting does nothing.
584 : */
585 : void set_hdf5_writing(bool write_hdf5);
586 :
587 : /**
588 : * Sets the value of _write_as_dimension.
589 : *
590 : * This directly controls the num_dim which is written to the Exodus
591 : * file. If non-zero, this value supersedes all other dimensions,
592 : * including:
593 : * 1.) MeshBase::spatial_dimension()
594 : * 2.) MeshBase::mesh_dimension()
595 : * 3.) Any value passed to use_mesh_dimension_instead_of_spatial_dimension()
596 : * This is useful/necessary for working around a bug in Paraview which
597 : * prevents the "Plot Over Line" filter from working on 1D meshes.
598 : */
599 : void write_as_dimension(unsigned dim);
600 :
601 : /**
602 : * Allows you to set a vector that is added to the coordinates of all
603 : * of the nodes. Effectively, this "moves" the mesh to a particular position
604 : */
605 : void set_coordinate_offset(Point p);
606 :
607 : /**
608 : * Set how many characters to use in names when opening a file for
609 : * writing.
610 : */
611 : void set_max_name_length(unsigned int max_length);
612 :
613 : /**
614 : * \returns A vector with three copies of each element in the provided name vector,
615 : * starting with r_, i_ and a_ respectively. If the "write_complex_abs" parameter
616 : * is true (default), the complex modulus is written, otherwise only the real and
617 : * imaginary parts are written.
618 : */
619 : std::vector<std::string>
620 : get_complex_names(const std::vector<std::string> & names,
621 : bool write_complex_abs) const;
622 :
623 : /**
624 : * returns a "tripled" copy of \p vars_active_subdomains, which is necessary in the
625 : * complex-valued case.
626 : */
627 : std::vector<std::set<subdomain_id_type>>
628 : get_complex_vars_active_subdomains
629 : (const std::vector<std::set<subdomain_id_type>> & vars_active_subdomains,
630 : bool write_complex_abs) const;
631 :
632 : /**
633 : * Takes a map from subdomain id -> vector of active variable names
634 : * as input and returns a corresponding map where the original
635 : * variable names have been replaced by their complex counterparts.
636 : * Used by the ExodusII_IO::write_element_data_from_discontinuous_nodal_data()
637 : * function.
638 : */
639 : std::map<subdomain_id_type, std::vector<std::string>>
640 : get_complex_subdomain_to_var_names
641 : (const std::map<subdomain_id_type, std::vector<std::string>> & subdomain_to_var_names,
642 : bool write_complex_abs) const;
643 :
644 : /**
645 : * This is the \p ExodusII_IO_Helper Conversion class. It provides
646 : * a data structure which contains \p ExodusII node/edge maps and
647 : * name conversions. It's defined below.
648 : */
649 : class Conversion;
650 :
651 : /**
652 : * This is the \p ExodusII_IO_Helper NamesData class.
653 : * It manages the C data structure necessary for writing out named
654 : * entities to ExodusII files.
655 : */
656 : class NamesData;
657 :
658 : /**
659 : * Prints the message defined in \p msg. Can be turned off if
660 : * verbosity is set to 0.
661 : */
662 : void message(std::string_view msg);
663 :
664 : /**
665 : * Prints the message defined in \p msg, and appends the number \p i
666 : * to the end of the message. Useful for printing messages in
667 : * loops. Can be turned off if verbosity is set to 0.
668 : */
669 : void message(std::string_view msg, int i);
670 :
671 : // File identification flag
672 : int ex_id;
673 :
674 : // General error flag
675 : int ex_err;
676 :
677 : // struct which contains data fields from the Exodus file header
678 : ExodusHeaderInfo header_info;
679 :
680 : // Problem title (Use vector<char> to emulate a char *)
681 : std::vector<char> & title;
682 :
683 : // Number of dimensions in the mesh
684 : int & num_dim;
685 :
686 : // Total number of nodes in the mesh
687 : int & num_nodes;
688 :
689 : // Total number of elements in the mesh
690 : int & num_elem;
691 :
692 : // Smallest element id which exceeds every element id in the mesh.
693 : // (this may exceed num_elem due to mapping)
694 : int end_elem_id() const;
695 :
696 : // Total number of element blocks
697 : int & num_elem_blk;
698 :
699 : // Total number of edges
700 : int & num_edge;
701 :
702 : // Total number of edge blocks. The sum of the number of edges in
703 : // each block must equal num_edge.
704 : int & num_edge_blk;
705 :
706 : // Total number of node sets
707 : int & num_node_sets;
708 :
709 : // Total number of side sets
710 : int & num_side_sets;
711 :
712 : // Total number of element sets
713 : int & num_elem_sets;
714 :
715 : // Number of global variables
716 : int num_global_vars;
717 :
718 : // Number of sideset variables
719 : int num_sideset_vars;
720 :
721 : // Number of nodeset variables
722 : int num_nodeset_vars;
723 :
724 : // Number of elemset variables
725 : int num_elemset_vars;
726 :
727 : // Number of elements in this block
728 : int num_elem_this_blk;
729 :
730 : // Number of nodes in each element
731 : int num_nodes_per_elem;
732 :
733 : // Number of attributes for a given block
734 : int num_attr;
735 :
736 : // Total number of elements in all side sets
737 : int num_elem_all_sidesets;
738 :
739 : // Total number of elements in all elem sets
740 : int num_elem_all_elemsets;
741 :
742 : // Vector of element block identification numbers
743 : std::vector<int> block_ids;
744 :
745 : // Vector of edge block identification numbers
746 : std::vector<int> edge_block_ids;
747 :
748 : // Vector of nodes in an element
749 : std::vector<int> connect;
750 :
751 : // Vector of the sideset IDs
752 : std::vector<int> ss_ids;
753 :
754 : // Vector of the nodeset IDs
755 : std::vector<int> nodeset_ids;
756 :
757 : // Vector of the elemset IDs
758 : std::vector<int> elemset_ids;
759 :
760 : // Number of sides in each sideset
761 : std::vector<int> num_sides_per_set;
762 :
763 : // Number of nodes in each nodeset
764 : std::vector<int> num_nodes_per_set;
765 :
766 : // Number of elems in each elemset
767 : std::vector<int> num_elems_per_set;
768 :
769 : // Number of distribution factors per sideset
770 : std::vector<int> num_df_per_set;
771 :
772 : // Number of distribution factors per nodeset
773 : std::vector<int> num_node_df_per_set;
774 :
775 : // Number of distribution factors per elemset
776 : std::vector<int> num_elem_df_per_set;
777 :
778 : // Starting indices for each nodeset in the node_sets_node_list vector.
779 : // Used in the calls to ex_{put,get}_concat_node_sets().
780 : std::vector<int> node_sets_node_index;
781 :
782 : // Starting indices for each nodeset in the node_sets_dist_fact vector.
783 : // Used in the calls to ex_{put,get}_concat_node_sets().
784 : std::vector<int> node_sets_dist_index;
785 :
786 : // Node ids for all nodes in nodesets, concatenated together.
787 : // Used in the calls to ex_{put,get}_concat_node_sets().
788 : std::vector<int> node_sets_node_list;
789 :
790 : // Distribution factors for all nodes in all nodesets, concatenated together.
791 : // Used in the calls to ex_{put,get}_concat_node_sets().
792 : std::vector<Real> node_sets_dist_fact;
793 :
794 : // List of element numbers in all sidesets
795 : std::vector<int> elem_list;
796 :
797 : // Side (face/edge) number actually on the boundary
798 : std::vector<int> side_list;
799 :
800 : // Side (face/edge) id number
801 : std::vector<int> id_list;
802 :
803 : // List of element numbers in all elemsets
804 : std::vector<int> elemset_list;
805 :
806 : // List of elemset ids for all elements in elemsets
807 : std::vector<int> elemset_id_list;
808 :
809 : // Optional mapping from internal [0,num_nodes) to arbitrary indices
810 : std::vector<int> node_num_map;
811 :
812 : // Optional mapping from internal [0,num_elem) to arbitrary indices
813 : std::vector<int> elem_num_map;
814 :
815 : // x locations of node points
816 : std::vector<Real> x;
817 :
818 : // y locations of node points
819 : std::vector<Real> y;
820 :
821 : // z locations of node points
822 : std::vector<Real> z;
823 :
824 : // Spline weights associated with node points, in IGA meshes
825 : std::vector<Real> w;
826 :
827 : // Number of Bezier Extraction coefficient vectors in a block
828 : unsigned int bex_num_elem_cvs;
829 :
830 : // Bezier Extraction connectivity indices, in IGA meshes
831 : std::vector<std::vector<long unsigned int>> bex_cv_conn;
832 :
833 : // Bezier Extraction coefficient vectors, in IGA meshes
834 : // bex_dense_constraint_vecs[block_num][vec_num][column_num] = coef
835 : std::vector<std::vector<std::vector<Real>>> bex_dense_constraint_vecs;
836 :
837 : // Type of element in a given block
838 : std::vector<char> elem_type;
839 :
840 : // Maps libMesh element numbers to Exodus element numbers
841 : // gets filled in when write_elements gets called
842 : std::map<dof_id_type, dof_id_type> libmesh_elem_num_to_exodus;
843 : std::vector<int> exodus_elem_num_to_libmesh;
844 :
845 : // Map of all node numbers connected to local node numbers to their exodus numbering.
846 : // The exodus numbers are stored in here starting with 1
847 : std::map<dof_id_type, dof_id_type> libmesh_node_num_to_exodus;
848 : std::vector<int> exodus_node_num_to_libmesh;
849 :
850 : // The number of timesteps in the file, as returned by ex_inquire
851 : int num_time_steps;
852 :
853 : // The timesteps stored in the solution file, filled by read_time_steps()
854 : std::vector<Real> time_steps;
855 :
856 : // The number of nodal variables in the Exodus file
857 : int num_nodal_vars;
858 :
859 : // The names of the nodal variables stored in the Exodus file
860 : std::vector<std::string> nodal_var_names;
861 :
862 : // Holds the nodal variable values for a given variable, one value
863 : // per node, indexed by libMesh node id.
864 : // This is a map so it can handle Nemesis files as well.
865 : std::map<dof_id_type, Real> nodal_var_values;
866 :
867 : // The number of elemental variables in the Exodus file
868 : int num_elem_vars;
869 :
870 : // The names of the elemental variables stored in the Exodus file
871 : std::vector<std::string> elem_var_names;
872 :
873 : // Holds the elemental variable values for a given variable, one value per element
874 : std::vector<Real> elem_var_values;
875 :
876 : // The names of the global variables stored in the Exodus file
877 : std::vector<std::string> global_var_names;
878 :
879 : // The names of the sideset variables stored in the Exodus file
880 : std::vector<std::string> sideset_var_names;
881 :
882 : // The names of the nodeset variables stored in the Exodus file
883 : std::vector<std::string> nodeset_var_names;
884 :
885 : // The names of the elemset variables stored in the Exodus file
886 : std::vector<std::string> elemset_var_names;
887 :
888 : // Maps of Ids to named entities
889 : std::map<int, std::string> id_to_block_names;
890 : std::map<int, std::string> id_to_edge_block_names;
891 : std::map<int, std::string> id_to_ss_names;
892 : std::map<int, std::string> id_to_ns_names;
893 : std::map<int, std::string> id_to_elemset_names;
894 :
895 : // On/Off message flag
896 : bool verbose;
897 :
898 : // Same as the ExodusII_IO flag by the same name. This flag is
899 : // also set whenever ExodusII_IO::set_unique_ids_from_maps() is called.
900 : bool set_unique_ids_from_maps;
901 :
902 : // This flag gets set after the Exodus file has been successfully opened for writing.
903 : // Both the create() and open() (if called with EX_WRITE) functions may set this flag.
904 : bool opened_for_writing;
905 :
906 : // This flag gets set after the open() function has been successfully called.
907 : // We call open() to open an ExodusII file for reading.
908 : bool opened_for_reading;
909 :
910 : // When either create() or open() is called, the Helper stores the
911 : // name of the opened file as current_filename. This way, the
912 : // ExodusII_IO object can check to see if, on subsequent writes, the
913 : // user is asking to write to a *different* filename from the one
914 : // that is currently open, and signal an error. The current
915 : // ExodusII_IO implementation is designed to work with a single file
916 : // only, so if you want to write to multiple Exodus files, use a
917 : // different ExodusII_IO object for each one.
918 : std::string current_filename;
919 :
920 : /**
921 : * Wraps calls to exII::ex_get_var_names() and exII::ex_get_var_param().
922 : * The enumeration controls whether nodal, elemental, global, etc.
923 : * variable names are read and which class members are filled in.
924 : * NODAL: num_nodal_vars nodal_var_names
925 : * ELEMENTAL: num_elem_vars elem_var_names
926 : * GLOBAL: num_global_vars global_var_names
927 : * SIDESET: num_sideset_vars sideset_var_names
928 : * NODESET: num_nodeset_vars nodeset_var_names
929 : */
930 : enum ExodusVarType {NODAL=0, ELEMENTAL=1, GLOBAL=2, SIDESET=3, NODESET=4, ELEMSET=5};
931 : void read_var_names(ExodusVarType type);
932 :
933 : const ExodusII_IO_Helper::Conversion &
934 : get_conversion(const ElemType type) const;
935 :
936 : const ExodusII_IO_Helper::Conversion &
937 : get_conversion(std::string type_str) const;
938 :
939 : /*
940 : * Returns the sum of node "offsets" that are to be expected from a
941 : * parallel nodal solution vector that has had "fake" nodes added on
942 : * each processor. This plus a node id gives a valid nodal solution
943 : * vector index.
944 : */
945 25251546 : dof_id_type node_id_to_vec_id(dof_id_type n) const
946 : {
947 25251546 : if (_added_side_node_offsets.empty())
948 23144286 : return n;
949 :
950 : // Find the processor id that has node_id in the parallel vec
951 12050 : const auto lb = std::upper_bound(_true_node_offsets.begin(),
952 2410 : _true_node_offsets.end(), n);
953 1205 : libmesh_assert(lb != _true_node_offsets.end());
954 14460 : const processor_id_type p = lb - _true_node_offsets.begin();
955 :
956 14460 : return n + (p ? _added_side_node_offsets[p-1] : 0);
957 : }
958 :
959 : /*
960 : * Returns the sum of both added node "offsets" on processors 0
961 : * through p-1 and real nodes added on processors 0 to p.
962 : * This is the starting index for added nodes' data.
963 : */
964 1725 : dof_id_type added_node_offset_on(processor_id_type p) const
965 : {
966 50 : libmesh_assert (p < _true_node_offsets.size());
967 : const dof_id_type added_node_offsets =
968 1800 : (_added_side_node_offsets.empty() || !p) ? 0 :
969 1475 : _added_side_node_offsets[p-1];
970 1825 : return _true_node_offsets[p] + added_node_offsets;
971 : }
972 :
973 :
974 : protected:
975 : /**
976 : * When appending: during initialization, check that variable names
977 : * in the file match those you attempt to initialize with.
978 : */
979 : void check_existing_vars(ExodusVarType type, std::vector<std::string> & names, std::vector<std::string> & names_from_file);
980 :
981 : /**
982 : * Wraps calls to exII::ex_put_var_names() and exII::ex_put_var_param().
983 : * The enumeration controls whether nodal, elemental, or global
984 : * variable names are read and which class members are filled in.
985 : */
986 : void write_var_names(ExodusVarType type, const std::vector<std::string> & names);
987 :
988 : // If true, whenever there is an I/O operation, only perform if if we are on processor 0.
989 : bool _run_only_on_proc0;
990 :
991 : // This flag gets set after the create() function has been successfully called.
992 : bool _opened_by_create;
993 :
994 : // True once the elem vars are initialized
995 : bool _elem_vars_initialized;
996 :
997 : // True once the global vars are initialized
998 : bool _global_vars_initialized;
999 :
1000 : // True once the nodal vars are initialized
1001 : bool _nodal_vars_initialized;
1002 :
1003 : // If true, use the Mesh's dimension (as determined by the dimension
1004 : // of the elements comprising the mesh) instead of the mesh's
1005 : // spatial dimension, when writing. By default this is false.
1006 : bool _use_mesh_dimension_instead_of_spatial_dimension;
1007 :
1008 : // If true, write an HDF5 file when available. If false, write the
1009 : // old format.
1010 : bool _write_hdf5;
1011 :
1012 : // The maximum name length to use when writing
1013 : unsigned int _max_name_length;
1014 :
1015 : // Set once the elem num map has been read
1016 : int _end_elem_id;
1017 :
1018 : // Use this for num_dim when writing the Exodus file. If non-zero, supersedes
1019 : // any value set in _use_mesh_dimension_instead_of_spatial_dimension.
1020 : unsigned _write_as_dimension;
1021 :
1022 : // On output, shift every point by _coordinate_offset
1023 : Point _coordinate_offset;
1024 :
1025 : // If true, forces single precision I/O
1026 : bool _single_precision;
1027 :
1028 : /**
1029 : * If we're adding "fake" sides to visualize SIDE_DISCONTINUOUS
1030 : * variables, _added_side_node_offsets[p] gives us the total
1031 : * solution vector offset to use on processor p+1 from the nodes on
1032 : * those previous ranks' sides.
1033 : */
1034 : std::vector<dof_id_type> _added_side_node_offsets;
1035 :
1036 : /**
1037 : * If we're adding "fake" sides to visualize SIDE_DISCONTINUOUS
1038 : * variables, we also need to know how many real nodes from previous
1039 : * ranks are taking up space in a solution vector.
1040 : */
1041 : std::vector<dof_id_type> _true_node_offsets;
1042 :
1043 : /**
1044 : * This class facilitates inline conversion of an input data vector
1045 : * to a different precision level, depending on the underlying type
1046 : * of Real and whether or not the single_precision flag is set. This
1047 : * should be used whenever floating point data is being written to
1048 : * the Exodus file. Note that if no precision conversion has to take
1049 : * place, there should be very little overhead involved in using
1050 : * this object.
1051 : */
1052 : struct MappedOutputVector
1053 : {
1054 : // If necessary, allocates space to store a version of vec_in in a
1055 : // different precision than it was input with.
1056 : MappedOutputVector(const std::vector<Real> & vec_in,
1057 : bool single_precision_in);
1058 :
1059 45894 : ~MappedOutputVector() = default;
1060 :
1061 : // Returns void * pointer to either the mapped data or the
1062 : // original data, as necessary.
1063 : void * data();
1064 :
1065 : private:
1066 : const std::vector<Real> & our_data;
1067 : bool single_precision;
1068 : std::vector<double> double_vec;
1069 : std::vector<float> float_vec;
1070 : };
1071 :
1072 : /**
1073 : * This class facilitates reading in vectors from Exodus file that
1074 : * may be of a different floating point type than Real. It employs
1075 : * basically the same approach as the MappedOutputVector, just going
1076 : * in the opposite direction. For more information, see the
1077 : * MappedOutputVector class docs.
1078 : */
1079 : struct MappedInputVector
1080 : {
1081 : MappedInputVector(std::vector<Real> & vec_in,
1082 : bool single_precision_in);
1083 : ~MappedInputVector();
1084 :
1085 : // Returns void * pointer to either the mapped data or the
1086 : // original data, as necessary.
1087 : void * data();
1088 :
1089 : private:
1090 : std::vector<Real> & our_data;
1091 : bool single_precision;
1092 : std::vector<double> double_vec;
1093 : std::vector<float> float_vec;
1094 : };
1095 :
1096 :
1097 : protected:
1098 :
1099 : /**
1100 : * read_var_names() dispatches to this function. We need to
1101 : * override it slightly for Nemesis.
1102 : */
1103 : virtual void read_var_names_impl(const char * var_type,
1104 : int & count,
1105 : std::vector<std::string> & result);
1106 :
1107 : private:
1108 :
1109 : /**
1110 : * Set to true iff we want to write separate "side" elements too.
1111 : */
1112 : bool _add_sides = false;
1113 :
1114 : /**
1115 : * write_var_names() dispatches to this function.
1116 : */
1117 : void write_var_names_impl(const char * var_type,
1118 : int & count,
1119 : const std::vector<std::string> & names);
1120 :
1121 : /**
1122 : * Defines equivalence classes of Exodus element types that map to
1123 : * libmesh ElemTypes.
1124 : */
1125 : std::map<std::string, ElemType> element_equivalence_map;
1126 : void init_element_equivalence_map();
1127 :
1128 : /**
1129 : * Associates libMesh ElemTypes with node/face/edge/etc. mappings
1130 : * of the corresponding Exodus element types.
1131 : *
1132 : * We have to map based on both ElemType and mesh dimension, because
1133 : * Exodus treats "TRI" side numbering in two different ways
1134 : * depending on whether a triangle is embedded in a 2D or a 3D mesh.
1135 : */
1136 : std::map<int, std::map<ElemType, ExodusII_IO_Helper::Conversion>> conversion_map;
1137 : void init_conversion_map();
1138 : };
1139 :
1140 :
1141 :
1142 1846300 : class ExodusII_IO_Helper::Conversion
1143 : {
1144 : public:
1145 :
1146 : /**
1147 : * Constructor. Zero initializes all variables.
1148 : */
1149 1904200 : Conversion()
1150 1904200 : : node_map(nullptr),
1151 1846300 : inverse_node_map(nullptr),
1152 1846300 : side_map(nullptr),
1153 1846300 : inverse_side_map(nullptr),
1154 1846300 : shellface_map(nullptr),
1155 1846300 : inverse_shellface_map(nullptr),
1156 1846300 : shellface_index_offset(0),
1157 1846300 : libmesh_type(INVALID_ELEM),
1158 1846300 : dim(0),
1159 1846300 : n_nodes(0),
1160 1904200 : exodus_type("")
1161 1904200 : {}
1162 :
1163 : /**
1164 : * \returns The ith component of the node map for this element.
1165 : *
1166 : * The node map maps the exodusII node numbering format to this
1167 : * library's format.
1168 : */
1169 : int get_node_map(int i) const;
1170 :
1171 : /**
1172 : * \returns The ith component of the inverse node map for this
1173 : * element.
1174 : *
1175 : * The inverse node map maps the libmesh node numbering to Exodus'
1176 : * node numbering.
1177 : *
1178 : * \note All elements except Hex27 currently have the same node
1179 : * numbering as libmesh elements.
1180 : */
1181 : int get_inverse_node_map(int i) const;
1182 :
1183 : /**
1184 : * \returns The ith component of the side map for this element.
1185 : *
1186 : * The side map maps the exodusII side numbering format to this
1187 : * library's format.
1188 : */
1189 : int get_side_map(int i) const;
1190 :
1191 : /**
1192 : * \returns The ith component of the side map for this element.
1193 : *
1194 : * The side map maps the libMesh side numbering format to this
1195 : * exodus's format.
1196 : */
1197 : int get_inverse_side_map(int i) const;
1198 :
1199 : /**
1200 : * \returns The ith component of the shellface map for this element.
1201 : * \note Nothing is currently using this.
1202 : */
1203 : int get_shellface_map(int i) const;
1204 :
1205 : /**
1206 : * \returns The ith component of the inverse shellface map for this element.
1207 : */
1208 : int get_inverse_shellface_map(int i) const;
1209 :
1210 : /**
1211 : * \returns The canonical element type for this element.
1212 : *
1213 : * The canonical element type is the standard element type
1214 : * understood by this library.
1215 : */
1216 : ElemType libmesh_elem_type() const;
1217 :
1218 : /**
1219 : * \returns The string corresponding to the Exodus type for this element.
1220 : */
1221 : std::string exodus_elem_type() const;
1222 :
1223 : /**
1224 : * \returns The shellface index offset.
1225 : */
1226 : std::size_t get_shellface_index_offset() const;
1227 :
1228 : /**
1229 : * An invalid_id that can be returned to signal failure in case
1230 : * something goes wrong.
1231 : */
1232 : static const int invalid_id;
1233 :
1234 : /**
1235 : * Pointer to the node map for this element.
1236 : */
1237 : const std::vector<int> * node_map;
1238 :
1239 : /**
1240 : * Pointer to the inverse node map for this element.
1241 : * For all elements except for the Hex27, this is the same
1242 : * as the node map.
1243 : */
1244 : const std::vector<int> * inverse_node_map;
1245 :
1246 : /**
1247 : * Pointer to the side map for this element.
1248 : */
1249 : const std::vector<int> * side_map;
1250 :
1251 : /**
1252 : * Pointer to the inverse side map for this element.
1253 : */
1254 : const std::vector<int> * inverse_side_map;
1255 :
1256 : /**
1257 : * Pointer to the shellface map for this element. Only the inverse
1258 : * is actually used currently, this one is provided for completeness
1259 : * and libmesh_ingore()d to avoid warnings.
1260 : */
1261 : const std::vector<int> * shellface_map;
1262 :
1263 : /**
1264 : * Pointer to the inverse shellface map for this element.
1265 : */
1266 : const std::vector<int> * inverse_shellface_map;
1267 :
1268 : /**
1269 : * The shellface index offset defines the offset due to a difference between libMesh
1270 : * and Exodus in indexing sidesets. This is relevant for shell elements, for
1271 : * example, since Exodus includes extra "shell face" sides in that case.
1272 : */
1273 : size_t shellface_index_offset;
1274 :
1275 : /**
1276 : * The canonical (i.e. standard for this library)
1277 : * element type.
1278 : */
1279 : ElemType libmesh_type;
1280 :
1281 : /**
1282 : * The element dimension; useful since we don't seem to have a cheap
1283 : * way to look this up from ElemType
1284 : */
1285 : int dim;
1286 :
1287 : /**
1288 : * The number of nodes per element; useful likewise
1289 : */
1290 : int n_nodes;
1291 :
1292 : /**
1293 : * The string corresponding to the Exodus type for this element
1294 : */
1295 : std::string exodus_type;
1296 : };
1297 :
1298 :
1299 :
1300 : /**
1301 : * This class is useful for managing anything that requires a char **
1302 : * input/output in ExodusII file. You must know the number of strings
1303 : * and the length of each one at the time you create it.
1304 : */
1305 42510 : class ExodusII_IO_Helper::NamesData
1306 : {
1307 : public:
1308 : /**
1309 : * Constructor. Allocates enough storage to hold n_strings of
1310 : * length string_length. (Actually allocates string_length+1 characters
1311 : * per string to account for the trailing '\0' character.)
1312 : */
1313 : explicit
1314 : NamesData(size_t n_strings, size_t string_length);
1315 :
1316 : /**
1317 : * Adds another name to the current data table.
1318 : */
1319 : void push_back_entry(const std::string & name);
1320 :
1321 : /**
1322 : * Provide access to the underlying C data table
1323 : */
1324 : char ** get_char_star_star();
1325 :
1326 : /**
1327 : * Provide access to the i'th underlying char *
1328 : */
1329 : char * get_char_star(int i);
1330 :
1331 : private:
1332 : // C++ data structures for managing string memory
1333 : std::vector<std::vector<char>> data_table;
1334 : std::vector<char *> data_table_pointers;
1335 :
1336 : size_t counter;
1337 : size_t table_size;
1338 : };
1339 :
1340 :
1341 68 : inline void ExodusII_IO_Helper::set_add_sides(bool add_sides)
1342 : {
1343 2414 : _add_sides = add_sides;
1344 68 : }
1345 :
1346 :
1347 2752 : inline bool ExodusII_IO_Helper::get_add_sides()
1348 : {
1349 14575 : return _add_sides;
1350 : }
1351 :
1352 :
1353 0 : inline int ExodusII_IO_Helper::end_elem_id() const
1354 : {
1355 0 : libmesh_assert_equal_to(std::size_t(num_elem), elem_num_map.size());
1356 357 : return _end_elem_id;
1357 : }
1358 :
1359 :
1360 : } // namespace libMesh
1361 :
1362 : #endif // LIBMESH_HAVE_EXODUS_API
1363 :
1364 : #endif // LIBMESH_EXODUSII_IO_HELPER_H
|
