Files
src_old/test/H5Fed/read_trianglemesh.c
T

169 lines
3.6 KiB
C

#include <stdio.h>
#include <stdlib.h>
#include <hdf5.h>
#include "H5Part.h"
#include "H5Fed.h"
#ifndef PARALLEL_IO
#ifndef MPI_COMM_WORLD
#define MPI_COMM_WORLD 0
#endif
#endif
struct vertex {
h5_float64_t P[3];
};
typedef struct vertex vertex_t;
struct entity {
h5_id_t global_id;
h5_id_t parent_id;
h5_id_t vids[3];
};
typedef struct entity entity_t;
h5_err_t
read_vertices (
h5_file_t * f
) {
h5_id_t id, local_id;
h5_float64_t P[3];
h5_size_t real_num = 0;
h5_id_t level_id = H5FedGetLevel ( f );
h5_size_t num = H5FedGetNumVertices ( f );
printf ( " Number of vertices on level %lld: %lld\n", level_id, num );
h5_err_t h5err = H5FedBeginTraverseVertices ( f );
if ( h5err < 0 ) return h5err;
while ( (real_num < num) &&
((local_id = H5FedTraverseVertices ( f, &id, P )) >= 0) ) {
printf ( " Vertex[%lld]: local id: %lld, coords: %f %f %f \n",
id, local_id, P[0], P[1], P[2] );
real_num++;
}
if ( real_num != num ) {
fprintf ( stderr, "!!! Got %lld vertices, but expected %lld.\n",
real_num, num );
return -1;
}
return H5_SUCCESS;
}
h5_err_t
read_entities (
h5_file_t * f
) {
h5_id_t id, local_id, parent_id, vids[4];
h5_size_t real_num = 0;
h5_id_t level_id = H5FedGetLevel ( f );
h5_size_t num = H5FedGetNumElements ( f );
printf ( " Number of triangles on level %lld: %lld\n", level_id, num );
h5_err_t h5err = H5FedBeginTraverseElements ( f );
if ( h5err < 0 ) return h5err;
while ( (real_num < num) &&
((local_id = H5FedTraverseElements (
f, &id, &parent_id, vids )) >= 0) ) {
printf (
" entitiy[%lld]: local id: %lld, parent id: %lld, "
"vids: %lld %lld %lld\n",
id, local_id, parent_id, vids[0], vids[1], vids[2] );
real_num++;
}
if ( real_num != num ) {
fprintf ( stderr, "!!! Got %lld tets, but expected %lld.\n",
real_num, num );
return -1;
}
return H5_SUCCESS;
}
h5_err_t
read_level (
h5_file_t * f
) {
h5_err_t h5err = read_vertices ( f );
if ( h5err < 0 ) {
fprintf ( stderr, "!!! Oops ...\n" );
return -1;
}
h5err = read_entities ( f );
if ( h5err < 0 ) {
fprintf ( stderr, "!!! Oops ...\n" );
return -1;
}
return H5_SUCCESS;
}
h5_err_t
read_mesh (
h5_file_t * f
) {
h5_id_t level_id;
h5_size_t num_levels = H5FedGetNumLevels ( f );
if ( num_levels < 0 ) {
fprintf ( stderr, "!!! Oops ...\n" );
return -1;
}
printf ( " Number of levels in mesh: %lld\n", num_levels );
for ( level_id = 0; level_id < num_levels; level_id++ ) {
printf ( " Going to level %lld\n", level_id );
h5_err_t h5err = H5FedSetLevel ( f, level_id );
if ( h5err < 0 ) {
fprintf ( stderr, "!!! Can't set level %lld.\n", level_id );
return -1;
}
h5err = read_level ( f );
if ( h5err < 0 ) {
fprintf ( stderr, "!!! Oops ...\n" );
return -1;
}
}
return H5_SUCCESS;
}
int
main (
int argc,
char *argv[]
) {
H5SetVerbosityLevel ( 4 );
h5_file_t *f = H5OpenFile ( "simple_triangle.h5", H5_O_RDONLY, 0 );
if ( f == NULL ) {
fprintf ( stderr, "!!! Can't open file.\n" );
return -1;
}
h5_size_t num_meshes = H5FedGetNumMeshes ( f, H5_TRIANGLE_MESH );
printf ( " Number of meshes: %lld\n", num_meshes );
h5_id_t mesh_id;
for ( mesh_id = 0; mesh_id < num_meshes; mesh_id++ ) {
h5_err_t h5err = H5FedOpenMesh ( f, mesh_id, H5_TRIANGLE_MESH );
if ( h5err < 0 ) {
fprintf ( stderr, "!!! Can't open mesh %lld\n", mesh_id );
return -1;
}
h5err = read_mesh ( f );
if ( h5err < 0 ) {
fprintf ( stderr, "!!! Oops ...\n" );
return 1;
}
}
h5_err_t h5err = H5CloseFile ( f );
if ( h5err < 0 ) {
fprintf ( stderr, "!!! Can't close file.\n" );
return -1;
}
return 0;
}