From 5584086e40111f748f1d0b2e6d79fe05c0de0180 Mon Sep 17 00:00:00 2001 From: Achim Gsell Date: Fri, 13 Nov 2009 21:51:38 +0000 Subject: [PATCH] tag test added --- .gitattributes | 2 + test/H5Fed/tetmesh_read_tags.c | 208 ++++++++++++++++++++++++++++++++ test/H5Fed/tetmesh_write_tags.c | 185 ++++++++++++++++++++++++++++ 3 files changed, 395 insertions(+) create mode 100644 test/H5Fed/tetmesh_read_tags.c create mode 100644 test/H5Fed/tetmesh_write_tags.c diff --git a/.gitattributes b/.gitattributes index 202e2b7..f32d6c3 100644 --- a/.gitattributes +++ b/.gitattributes @@ -466,7 +466,9 @@ test/H5Fed/map_tet2globalid.c -text test/H5Fed/map_triangle2globalid.c -text test/H5Fed/tetmesh_adjacencies.c -text test/H5Fed/tetmesh_read.c -text +test/H5Fed/tetmesh_read_tags.c -text test/H5Fed/tetmesh_write.c -text +test/H5Fed/tetmesh_write_tags.c -text test/H5Fed/trimesh_read.c -text test/H5Fed/trimesh_write.c -text test/H5Fed/write_boundary.c -text diff --git a/test/H5Fed/tetmesh_read_tags.c b/test/H5Fed/tetmesh_read_tags.c new file mode 100644 index 0000000..97b1a74 --- /dev/null +++ b/test/H5Fed/tetmesh_read_tags.c @@ -0,0 +1,208 @@ +#include +#include +#include +#include +#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 tet { + h5_id_t global_id; + h5_id_t parent_id; + h5_id_t vids[4]; +}; +typedef struct tet tet_t; + +static h5_err_t +traverse_vertices ( + h5_file_t * f + ) { + h5_id_t local_id; + h5_float64_t P[3]; + h5_size_t real_num = 0; + + h5_size_t num = H5FedGetNumVerticesTotal ( f ); + + H5FedBeginTraverseVertices ( f ); + while ( (real_num < num) && + ((local_id = H5FedTraverseVertices ( f, P )) >= 0) ) { + size_t size; + h5_int64_t retval[3]; + H5FedGetMTag ( f, "testtag", local_id, &size, retval ); + if ( (retval[0] != local_id) || + (retval[1] != local_id+1) || + (retval[1] != local_id+1) ) { + fprintf ( stderr, "!!! Wrong tag values for vertix %lld\n", + local_id ); + exit ( 1 ); + } + real_num++; + } + H5FedEndTraverseVertices ( f ); + + if ( real_num != num ) { + fprintf ( stderr, "!!! Got %lld vertices, but expected %lld.\n", + real_num, num ); + exit ( 1 ); + } + return H5_SUCCESS; +} + +static h5_err_t +traverse_edges ( + h5_file_t * f + ) { + h5_id_t local_id, vids[4]; + + H5FedBeginTraverseEdges ( f ); + while ( (local_id = H5FedTraverseEdges ( f, vids )) >= 0 ) { + h5_id_t local_vids[4]; + H5FedLMapEdgeID2VertexIDs ( f, local_id, local_vids ); + size_t size; + h5_int64_t retval[3]; + H5FedGetMTag ( f, "testtag", local_id, &size, retval ); + if ( (retval[0] != local_id) || + (retval[1] != local_id+1) || + (retval[1] != local_id+1) ) { + fprintf ( stderr, "!!! Wrong tag values for edge %lld\n", + local_id ); + exit ( 1 ); + } + } + H5FedEndTraverseEdges ( f ); + return H5_SUCCESS; +} + +static h5_err_t +traverse_triangles ( + h5_file_t * f + ) { + h5_id_t local_id, vids[4]; + + H5FedBeginTraverseTriangles ( f ); + while ( (local_id = H5FedTraverseTriangles ( f, vids )) >= 0 ) { + size_t size; + h5_int64_t retval[3]; + H5FedGetMTag ( f, "testtag", local_id, &size, retval ); + if ( (retval[0] != local_id) || + (retval[1] != local_id+1) || + (retval[1] != local_id+1) ) { + fprintf ( stderr, "!!! Wrong tag values for edge %lld\n", + local_id ); + exit ( 1 ); + } + } + H5FedEndTraverseTriangles ( f ); + return H5_SUCCESS; +} + +static h5_err_t +traverse_tets ( + h5_file_t * f + ) { + h5_id_t local_id, vids[4]; + h5_size_t real_num = 0; + + h5_size_t num = H5FedGetNumElementsTotal ( f ); + + H5FedBeginTraverseElements ( f ); + while ( (real_num < num) && + ((local_id = H5FedTraverseElements ( f, vids )) >= 0) ) { + size_t size; + h5_int64_t retval[3]; + H5FedGetMTag ( f, "testtag", local_id, &size, retval ); + if ( (retval[0] != local_id) || + (retval[1] != local_id+1) || + (retval[1] != local_id+1) ) { + fprintf ( stderr, "!!! Wrong tag values for edge %lld\n", + local_id ); + exit ( 1 ); + } + real_num++; + } + H5FedEndTraverseElements ( f ); + if ( real_num != num ) { + fprintf ( stderr, "!!! Got %lld tets, but expected %lld.\n", + real_num, num ); + exit(1); + } + + return H5_SUCCESS; +} + +static h5_err_t +read_level ( + h5_file_t * f + ) { + traverse_vertices ( f ); + traverse_edges ( f ); + traverse_triangles ( f ); + traverse_tets ( f ); + return H5_SUCCESS; +} + +static h5_err_t +traverse_mesh ( + h5_file_t * f + ) { + + h5_id_t level_id; + h5_size_t num_levels = H5FedGetNumLevels ( f ); + printf ( " Number of levels in mesh: %lld\n", num_levels ); + for ( level_id = 2; level_id < num_levels; 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 ( 2 ); + H5SetErrorHandler ( H5AbortErrorhandler ); + + h5_file_t *f = H5OpenFile ( "simple_tet.h5", H5_O_RDONLY, 0 ); + h5_size_t num_meshes = H5FedGetNumMeshes ( f, H5_TETRAHEDRAL_MESH ); + printf ( " Number of meshes: %lld\n", num_meshes ); + + h5_id_t mesh_id; + for ( mesh_id = 0; mesh_id < num_meshes; mesh_id++ ) { + fprintf ( + stderr, + "Time used: %f\n", + (float)clock()/(float)CLOCKS_PER_SEC ); + H5FedOpenMesh ( f, mesh_id, H5_TETRAHEDRAL_MESH ); + fprintf ( + stderr, + "Time used: %f\n", + (float)clock()/(float)CLOCKS_PER_SEC ); + traverse_mesh ( f ); + } + + H5CloseFile ( f ); + return 0; +} diff --git a/test/H5Fed/tetmesh_write_tags.c b/test/H5Fed/tetmesh_write_tags.c new file mode 100644 index 0000000..9ae3f1e --- /dev/null +++ b/test/H5Fed/tetmesh_write_tags.c @@ -0,0 +1,185 @@ +#include +#include +#include +#include +#include +#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 tet { + h5_id_t global_id; + h5_id_t parent_id; + h5_id_t vids[4]; +}; +typedef struct tet tet_t; + +static h5_err_t +set_vertex_tags ( + h5_file_t * f + ) { + h5_id_t local_id; + h5_int64_t val[3]; + h5_float64_t P[3]; + + printf ( "\nSet tags to vertices\n" ); + H5FedBeginTraverseVertices ( f ); + while ( (local_id = H5FedTraverseVertices ( f, P )) >= 0 ) { + val[0] = local_id; + val[1] = local_id+1; + val[2] = local_id+2; + H5FedSetMTag ( + f, + "testtag", + local_id, + 3, + val + ); + h5_int64_t retval[3]; + size_t dims; + H5FedGetMTag ( + f, + "testtag", + local_id, + &dims, + retval ); + if ( memcmp ( val, retval, sizeof(val) ) ) { + fprintf ( stderr, "Oops!\n" ); + } + } + return H5FedEndTraverseVertices ( f ); +} + +static h5_err_t +set_edge_tags ( + h5_file_t * f + ) { + h5_id_t local_id, vids[4]; + h5_int64_t val[3]; + printf ( "\nSet tags to edges\n" ); + H5FedBeginTraverseEdges ( f ); + while ( (local_id = H5FedTraverseEdges ( f, vids )) >= 0 ) { + val[0] = local_id; + val[1] = local_id+1; + val[2] = local_id+2; + H5FedSetMTag ( + f, + "testtag", + local_id, + 3, + val + ); + h5_int64_t retval[3]; + size_t dims; + H5FedGetMTag ( + f, + "testtag", + local_id, + &dims, + retval ); + if ( memcmp ( val, retval, sizeof(val) ) ) { + fprintf ( stderr, "Oops!\n" ); + } + } + return H5FedEndTraverseEdges ( f ); +} + +static h5_err_t +set_tri_tags ( + h5_file_t * f + ) { + h5_id_t local_id, vids[4]; + h5_int64_t val[3]; + printf ( "\nSet tags to triangle\n" ); + H5FedBeginTraverseTriangles ( f ); + while ( (local_id = H5FedTraverseTriangles ( f, vids )) >= 0 ) { + val[0] = local_id; + val[1] = local_id+1; + val[2] = local_id+2; + H5FedSetMTag ( + f, + "testtag", + local_id, + 3, + val + ); + h5_int64_t retval[3]; + size_t dims; + H5FedGetMTag ( + f, + "testtag", + local_id, + &dims, + retval ); + if ( memcmp ( val, retval, sizeof(val) ) ) { + fprintf ( stderr, "Oops!\n" ); + } + } + return H5FedEndTraverseTriangles ( f ); +} + +static h5_err_t +set_tet_tags ( + h5_file_t * f + ) { + h5_id_t local_id, vids[4]; + h5_int64_t val[3]; + printf ( "\nSet tags to tetrahedra\n" ); + H5FedBeginTraverseElements ( f ); + while ( (local_id = H5FedTraverseElements (f, vids )) >= 0 ) { + val[0] = local_id; + val[1] = local_id+1; + val[2] = local_id+2; + H5FedSetMTag ( + f, + "testtag", + local_id, + 3, + val + ); + h5_int64_t retval[3]; + size_t dims; + H5FedGetMTag ( + f, + "testtag", + local_id, + &dims, + retval ); + if ( memcmp ( val, retval, sizeof(val) ) ) { + fprintf ( stderr, "Oops!\n" ); + } + } + return H5FedEndTraverseElements ( f ); +} + +int +main ( + int argc, + char *argv[] + ) { + H5SetVerbosityLevel ( 5 ); + H5SetErrorHandler ( H5AbortErrorhandler ); + + h5_file_t *f = H5OpenFile ( "simple_tet.h5", H5_O_RDONLY, 0 ); + H5FedOpenMesh ( f, 0, H5_TETRAHEDRAL_MESH ); + h5_size_t num_levels = H5FedGetNumLevels ( f ); + H5FedSetLevel ( f, num_levels-1 ); + H5FedAddMTagset ( f, "testtag", H5_INT64_T ); + set_vertex_tags ( f ); + set_edge_tags ( f ); + set_tri_tags ( f ); + set_tet_tags ( f ); + H5CloseFile ( f ); + return 0; +}