src_old/test/H5Fed/tetmesh_read.c

#include <stdio.h>
#include <stdlib.h>
#include <time.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 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 );
	printf ( "    Number of vertices on level: %lld\n", num );

	H5FedBeginTraverseVertices ( f );
	while ( (real_num < num) &&
		((local_id = H5FedTraverseVertices ( f, P )) >= 0) ) {
		char v[256];
		snprintf ( v, sizeof(v), "=%llx=", local_id );
		printf ( "| %-18s | (%f, %f, %f) |\n",
			 v, P[0], P[1], P[2] );
		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 );
	printf ( "Edges on level %lld:\n", H5FedGetLevel(f) );
	while ( (local_id = H5FedTraverseEdges ( f, vids )) >= 0 ) {
		char v[256];
		char k[256];
		h5_id_t local_vids[4];
		snprintf ( k, sizeof(k), "=%llx=", local_id );
		H5FedLMapEdgeID2VertexIDs ( f, local_id, local_vids );
		snprintf ( v, sizeof(v), "=[%lld,%lld]=",
				   local_vids[0], local_vids[1] );
		printf ( "| %-18s | %-18s |\n", k, v );
	}
	H5FedEndTraverseEdges ( f );
	return H5_SUCCESS;
}

static h5_err_t
traverse_triangles (
	h5_file_t * f
	) {
	h5_id_t local_id, vids[4];

	H5FedBeginTraverseTriangles ( f );
	printf ( "Triangles on level %lld:\n", H5FedGetLevel(f) );
	while ( (local_id = H5FedTraverseTriangles ( f, vids )) >= 0 ) {
		char v[256];
		char d[256];
		h5_id_t local_vids[4];
		snprintf ( d, sizeof(d), "=%llx=", local_id );
		H5FedLMapTriangleID2VertexIDs ( f, local_id, local_vids );
		snprintf ( v, sizeof(v), "=[%lld,%lld,%lld]=",
			   local_vids[0], local_vids[1], local_vids[2] );
		printf ( "| %-18s | %-18s |\n", d, v );
	}
	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 );
	printf ( "    Number of tetrahedra on level: %lld\n", num );

	H5FedBeginTraverseElements ( f );
	while ( (real_num < num) &&
		((local_id = H5FedTraverseElements ( f, vids )) >= 0) ) {
		char v[256];
		char t[256];
		h5_id_t local_vids[4];
		snprintf ( t, sizeof(t), "=%llx=", local_id );
		H5FedLMapTetID2VertexIDs ( f, local_id, local_vids );
		snprintf ( v, sizeof(v), "=[%lld,%lld,%lld,%lld]=",
			   local_vids[0], local_vids[1], local_vids[2],local_vids[3] );
		printf ( "| %-18s | %-18s |\n", t, v );
		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;
}