135 lines
2.4 KiB
C
135 lines
2.4 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_id_t global_id;
|
|
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;
|
|
|
|
|
|
vertex_t V0[4] = {
|
|
{ 0, {-1.0, 0.0, 0.0} },
|
|
{ 1, { 1.0, 0.0, 0.0} },
|
|
{ 2, { 0.0, 1.0, 0.0} },
|
|
{ 3, { 0.0, -1.0, 0.0} }
|
|
};
|
|
|
|
vertex_t V1[1] = {
|
|
{ 4, {0.0, 0.0, 0.0 } }
|
|
};
|
|
|
|
entity_t T0[2] = {
|
|
{ 1, -1, { 0, 1, 2 } },
|
|
{ 0, -1, { 0, 1, 3 } }
|
|
};
|
|
|
|
entity_t T1[4] = {
|
|
{ 2, 1, { 1, 2, 4 } },
|
|
{ 3, 1, { 0, 2, 4 } },
|
|
{ 4, 0, { 0, 3, 4 } },
|
|
{ 5, 0, { 1, 3, 4 } }
|
|
};
|
|
|
|
h5_err_t
|
|
add_level (
|
|
h5_file *f,
|
|
vertex_t V[],
|
|
int num_verts,
|
|
entity_t T[],
|
|
int num_entities
|
|
) {
|
|
|
|
h5_err_t h5err = H5FedAddLevel ( f );
|
|
if ( h5err < 0 ) {
|
|
fprintf ( stderr, "!!! Can't add level.\n" );
|
|
return -1;
|
|
}
|
|
h5err = H5FedAddNumVertices ( f, num_verts );
|
|
if ( h5err < 0 ) {
|
|
fprintf ( stderr, "!!! Can't set number of vertices.\n" );
|
|
return -1;
|
|
}
|
|
|
|
int i;
|
|
for ( i = 0; i<num_verts; i++ ) {
|
|
h5err = H5FedStoreVertex (
|
|
f,
|
|
V[i].global_id,
|
|
V[i].P );
|
|
if ( h5err < 0 ) {
|
|
fprintf ( stderr, "!!! Can't store vertex.\n" );
|
|
return -1;
|
|
}
|
|
}
|
|
h5err = H5FedAddNumEntities ( f, num_entities );
|
|
if ( h5err < 0 ) {
|
|
fprintf ( stderr, "!!! Can't set number of tets.\n" );
|
|
return -1;
|
|
}
|
|
|
|
for ( i = 0; i<num_entities; i++ ) {
|
|
h5err = H5FedStoreTriangle (
|
|
f,
|
|
T[i].global_id,
|
|
T[i].parent_id,
|
|
T[i].vids );
|
|
|
|
if ( h5err < 0 ) {
|
|
fprintf ( stderr, "!!! Can't store tet.\n" );
|
|
return -1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
main (
|
|
int argc,
|
|
char *argv[]
|
|
) {
|
|
|
|
H5PartSetVerbosityLevel ( 4 );
|
|
|
|
h5_file *f = H5OpenFile ( "simple_triangle.h5", 0 );
|
|
if ( f == NULL ) {
|
|
fprintf ( stderr, "!!! Can't open file.\n" );
|
|
return -1;
|
|
}
|
|
|
|
h5_err_t h5err = H5FedAddMesh ( f, TRIANGLE_MESH );
|
|
if ( h5err < 0 ) {
|
|
fprintf ( stderr, "!!! Can't set step.\n" );
|
|
return -1;
|
|
}
|
|
|
|
h5err = add_level ( f, V0, sizeof(V0)/sizeof(V0[0]), T0, sizeof(T0)/sizeof(T0[0]) );
|
|
if ( h5err < 0 ) return h5err;
|
|
h5err = add_level ( f, V1, sizeof(V1)/sizeof(V1[0]), T1, sizeof(T1)/sizeof(T1[0]) );
|
|
if ( h5err < 0 ) return h5err;
|
|
|
|
h5err = H5CloseFile ( f );
|
|
if ( h5err < 0 ) {
|
|
fprintf ( stderr, "!!! Can't close file.\n" );
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|