void write_to_exodus(int rank, int num_procs, char * out_file_name)
/*****************************************************************************/
{
int exo_access = EX_CLOBBER;
int cpu_word_size = sizeof(double);
int io_word_size = sizeof(float);
int out_id;
int i;
int b;
ex_init_params exinit;
int error = 0;
out_id = ex_create(out_file_name, exo_access, &cpu_word_size,
&io_word_size);
if (out_id < 0){
printf("error opening file");
}
strncpy( exinit.title, mss.title, MAX_LINE_LENGTH-1 );
exinit.title[MAX_LINE_LENGTH-1] = 0;
exinit.num_dim = mss.num_dim;
exinit.num_nodes = mss.num_nodes;
exinit.num_edge = 0;
exinit.num_edge_blk = 0;
exinit.num_face = 0;
exinit.num_face_blk = 0;
exinit.num_elem = mss.num_elem;
exinit.num_elem_blk = mss.num_elem_blk;
exinit.num_node_sets = mss.num_node_sets;
exinit.num_edge_sets = 0;
exinit.num_face_sets = 0;
exinit.num_side_sets = mss.num_side_sets;
exinit.num_elem_sets = 0;
exinit.num_node_maps = 0;
exinit.num_edge_maps = 0;
exinit.num_face_maps = 0;
exinit.num_elem_maps = 0;
PERROR;
if ( ex_put_init_ext(out_id, &exinit) < 0 )
++error;
PERROR;
/*now write parallel global information*/
if ( ne_put_init_global( out_id,
mss.num_nodes_global,
mss.num_elems_global,
mss.num_elm_blks_global,
mss.num_node_sets_global,
mss.num_side_sets_global ) < 0 )
++error;
PERROR;
if ( ne_put_init_info( out_id, mss.num_total_proc, mss.num_proc_in_file,
mss.type ) < 0 )
++error;
PERROR;
if ( ne_put_eb_info_global(out_id,mss.elem_blk_ids_global,mss.elem_blk_cnts_global) < 0 )
++error;
PERROR;
if ( mss.num_node_sets_global > 0 ) {
if ( ne_put_ns_param_global( out_id,
mss.ns_ids_global,
mss.ns_cnts_global,
mss.ns_df_cnts_global ) < 0 )
++error;
}
PERROR;
if ( mss.num_side_sets_global > 0 ) {
if ( ne_put_ss_param_global( out_id,
mss.ss_ids_global,
mss.ss_cnts_global,
mss.ss_df_cnts_global ) < 0 )
++error;
}
PERROR;
/*writingparallel info*/
if ( ne_put_loadbal_param( out_id,
mss.num_internal_nodes,
mss.num_border_nodes,
mss.num_external_nodes,
mss.num_internal_elems,
mss.num_border_elems,
mss.num_node_comm_maps,
mss.num_elem_comm_maps,
rank ) < 0 )
++error;
PERROR;
if ( ne_put_cmap_params( out_id,
mss.node_cmap_ids,
(int*)mss.node_cmap_node_cnts,
mss.elem_cmap_ids,
(int*)mss.elem_cmap_elem_cnts,
rank ) < 0 )
++error;
PERROR;
if ( ne_put_elem_map( out_id,
mss.internal_elements,
mss.border_elements,
rank ) < 0 )
++error;
PERROR;
if ( ne_put_node_map( out_id,
mss.internal_nodes,
mss.border_nodes,
mss.external_nodes,
rank ) < 0 )
++error;
PERROR;
for (i = 0; i < mss.num_node_comm_maps; i++) {
if ( ne_put_node_cmap( out_id,
mss.node_cmap_ids[i],
mss.comm_node_ids[i],
mss.comm_node_proc_ids[i],
rank ) < 0 )
++error;
}
PERROR;
for (i = 0; i < mss.num_elem_comm_maps; i++) {
if ( ne_put_elem_cmap( out_id,
mss.elem_cmap_ids[i],
mss.comm_elem_ids[i],
mss.comm_side_ids[i],
mss.comm_elem_proc_ids[i],
rank ) < 0 )
++error;
}
PERROR;
/*coords*/
error += ex_put_coord(out_id, mss.coord, (mss.coord)+mss.num_nodes, (mss.coord)+2*mss.num_nodes);
PERROR;
error += ex_put_coord_names(out_id, mss.bptr);
PERROR;
/*map*/
error += ex_put_map(out_id, mss.element_order_map);
PERROR;
error += ex_put_elem_num_map(out_id, mss.global_element_numbers);
PERROR;
error += ex_put_node_num_map(out_id, mss.global_node_numbers);
PERROR;
/*block info*/
for(b = 0; b < mss.num_elem_blk; b++)
{
int gpe = 0;
int fpe = 0;
error += ex_put_block( out_id,
EX_ELEM_BLOCK,
mss.block_id[b],
mss.element_types[b],
mss.elements[b],
mss.nodes_per_element[b],
gpe, fpe,
mss.element_attributes[b] ); /* num attr */
PERROR;
}
/* write element connectivity information */
for (b = 0; b < mss.num_elem_blk; b++) {
if ( mss.elements[b] > 0 ){
error += ex_put_elem_conn(out_id,mss.block_id[b],mss.elmt_node_linkage[b]);
PERROR;
}
}
/* write in nodal boundary sets for the body. */
for(i = 0; i < mss.num_node_sets; i++) {
error += ex_put_node_set_param(out_id, mss.node_set_id[i],
mss.num_nodes_in_node_set[i],
mss.num_df_in_node_set[i]);
PERROR;
if(mss.num_nodes_in_node_set[i])
error += ex_put_node_set(out_id, mss.node_set_id[i], mss.node_set_nodes[i]);
PERROR;
}
for(i = 0; i < mss.num_side_sets; i++) {
error += ex_put_side_set_param(out_id, mss.side_set_id[i],
mss.num_elements_in_side_set[i],
mss.num_df_in_side_set[i]);
PERROR;
if(mss.num_elements_in_side_set[i])
error += ex_put_side_set(out_id, mss.side_set_id[i],
mss.side_set_elements[i],
mss.side_set_faces[i]);
PERROR;
}
error += ex_put_qa(out_id, mss.num_qa_records, mss.qaRecord);
PERROR;
ex_close(out_id);
}
int main (int argc, char **argv)
{
int exoid, num_dim, num_nodes, num_elem, num_elem_blk;
int num_elem_in_block[10], num_nodes_per_elem[10], num_attr[10];
int num_node_sets, num_side_sets, error;
int i, j, k, m, *elem_map, *connect, *node_map;
int node_list[100],elem_list[100],side_list[100];
int ebids[10], ids[10];
int num_nodes_per_set[10], num_elem_per_set[10];
int num_df_per_set[10];
int df_ind[10], node_ind[10], elem_ind[10];
int num_qa_rec, num_info;
int num_glo_vars, num_nod_vars, num_ele_vars;
int *truth_tab;
int whole_time_step, num_time_steps;
int CPU_word_size,IO_word_size;
int prop_array[2];
float *glob_var_vals, *nodal_var_vals, *elem_var_vals;
float time_value;
float x[100], y[100], z[100], *dummy;
float attrib[100], dist_fact[100];
char *coord_names[3], *qa_record[2][4], *info[3], *var_names[3];
char *prop_names[2];
ex_opts (EX_VERBOSE | EX_ABORT);
dummy = 0; /* assign this so the Cray compiler doesn't complain */
/* Specify compute and i/o word size */
CPU_word_size = 0; /* float or double */
IO_word_size = 0; /* use system default (4 bytes) */
/* create EXODUS II file */
exoid = ex_create ("test.exo", /* filename path */
EX_CLOBBER, /* create mode */
&CPU_word_size, /* CPU float word size in bytes */
&IO_word_size); /* I/O float word size in bytes */
printf ("after ex_create for test.exo, exoid = %d\n", exoid);
printf (" cpu word size: %d io word size: %d\n",CPU_word_size,IO_word_size);
/* ncopts = NC_VERBOSE; */
/* initialize file with parameters */
num_dim = 3;
num_nodes = 28;
num_elem = 8;
num_elem_blk = 7;
num_node_sets = 2;
num_side_sets = 5;
/* num_side_sets = 6; Uncomment to test NULL side sets */
error = ex_put_init (exoid, "This is testwt1", num_dim, num_nodes, num_elem,
num_elem_blk, num_node_sets, num_side_sets);
printf ("after ex_put_init, error = %d\n", error);
/* write nodal coordinates values and names to database */
/* Quad #1 */
x[0] = 0.0; y[0] = 0.0; z[0] = 0.0;
x[1] = 1.0; y[1] = 0.0; z[1] = 0.0;
x[2] = 1.0; y[2] = 1.0; z[2] = 0.0;
x[3] = 0.0; y[3] = 1.0; z[3] = 0.0;
/* Quad #2 */
x[4] = 1.0; y[4] = 0.0; z[4] = 0.0;
x[5] = 2.0; y[5] = 0.0; z[5] = 0.0;
x[6] = 2.0; y[6] = 1.0; z[6] = 0.0;
x[7] = 1.0; y[7] = 1.0; z[7] = 0.0;
/* Hex #1 */
x[8] = 0.0; y[8] = 0.0; z[8] = 0.0;
x[9] = 10.0; y[9] = 0.0; z[9] = 0.0;
x[10] = 10.0; y[10] = 0.0; z[10] =-10.0;
x[11] = 1.0; y[11] = 0.0; z[11] =-10.0;
x[12] = 1.0; y[12] = 10.0; z[12] = 0.0;
x[13] = 10.0; y[13] = 10.0; z[13] = 0.0;
x[14] = 10.0; y[14] = 10.0; z[14] =-10.0;
x[15] = 1.0; y[15] = 10.0; z[15] =-10.0;
/* Tetra #1 */
x[16] = 0.0; y[16] = 0.0; z[16] = 0.0;
x[17] = 1.0; y[17] = 0.0; z[17] = 5.0;
x[18] = 10.0; y[18] = 0.0; z[18] = 2.0;
x[19] = 7.0; y[19] = 5.0; z[19] = 3.0;
/* Circle #1 */
x[20] = 100.0; y[20] = 100.0; z[20] = 0.0;
/* Sphere #1 */
x[21] = 50.0; y[21] = 50.0; z[21] = 20.0;
/* Wedge #1 */
x[22] = 3.0; y[22] = 0.0; z[22] = 6.0;
x[23] = 6.0; y[23] = 0.0; z[23] = 0.0;
x[24] = 0.0; y[24] = 0.0; z[24] = 0.0;
x[25] = 3.0; y[25] = 2.0; z[25] = 6.0;
x[26] = 6.0; y[26] = 2.0; z[26] = 2.0;
x[27] = 0.0; y[27] = 2.0; z[27] = 0.0;
error = ex_put_coord (exoid, x, y, z);
printf ("after ex_put_coord, error = %d\n", error);
coord_names[0] = "xcoor";
coord_names[1] = "ycoor";
coord_names[2] = "zcoor";
error = ex_put_coord_names (exoid, coord_names);
printf ("after ex_put_coord_names, error = %d\n", error);
/* write element order map */
elem_map = (int *) calloc(num_elem, sizeof(int));
for (i=1; i<=num_elem; i++)
{
elem_map[i-1] = i;
}
error = ex_put_map (exoid, elem_map);
printf ("after ex_put_map, error = %d\n", error);
free (elem_map);
/* write element numbering map */
elem_map = (int *) calloc(num_elem, sizeof(int));
for (i=1; i<=num_elem; i++)
{
elem_map[i-1] = i*2;
}
error = ex_put_elem_num_map (exoid, elem_map);
printf ("after ex_put_elem_num_map, error = %d\n", error);
free (elem_map);
/* write node numbering map */
node_map = (int *) calloc(num_nodes, sizeof(int));
for (i=1; i<=num_nodes; i++)
{
node_map[i-1] = i*3;
}
error = ex_put_node_num_map (exoid, node_map);
printf ("after ex_put_node_num_map, error = %d\n", error);
free (node_map);
/* write element block parameters */
num_elem_in_block[0] = 1; /* element 1: Quad 1 */
num_elem_in_block[1] = 2; /* elements 2, 3: Quad 1 & 2 */
num_elem_in_block[2] = 1; /* element 4: Hex */
num_elem_in_block[3] = 1; /* element 5: Tetra */
num_elem_in_block[4] = 1; /* element 6: Circle */
num_elem_in_block[5] = 1; /* element 7: Sphere */
num_elem_in_block[6] = 1; /* element 8: Wedge */
num_nodes_per_elem[0] = 4; /* elements in block #1 are 4-node quads */
num_nodes_per_elem[1] = 4; /* elements in block #2 are 4-node quads */
num_nodes_per_elem[2] = 8; /* elements in block #3 are 8-node hexes */
num_nodes_per_elem[3] = 4; /* elements in block #3 are 4-node tetras */
num_nodes_per_elem[4] = 1; /* elements in block #4 are 1-node circles */
num_nodes_per_elem[5] = 1; /* elements in block #5 are 1-node spheres */
num_nodes_per_elem[6] = 6; /* elements in block #6 are 6-node wedges */
ebids[0] = 10;
ebids[1] = 11;
ebids[2] = 12;
ebids[3] = 13;
ebids[4] = 14;
ebids[5] = 15;
ebids[6] = 16;
num_attr[0] = 3;
num_attr[1] = 3;
num_attr[2] = 3;
num_attr[3] = 3;
num_attr[4] = 3;
num_attr[5] = 3;
num_attr[6] = 3;
error = ex_put_elem_block (exoid, ebids[0], "quad", num_elem_in_block[0],
num_nodes_per_elem[0], num_attr[0]);
printf ("after ex_put_elem_block, error = %d\n", error);
error = ex_put_elem_block (exoid, ebids[1], "quad", num_elem_in_block[1],
num_nodes_per_elem[1], num_attr[1]);
printf ("after ex_put_elem_block, error = %d\n", error);
error = ex_put_elem_block (exoid, ebids[2], "hex", num_elem_in_block[2],
num_nodes_per_elem[2], num_attr[2]);
printf ("after ex_put_elem_block, error = %d\n", error);
error = ex_put_elem_block (exoid, ebids[3], "tetra", num_elem_in_block[3],
num_nodes_per_elem[3], num_attr[3]);
printf ("after ex_put_elem_block, error = %d\n", error);
error = ex_put_elem_block (exoid, ebids[4], "circle", num_elem_in_block[4],
num_nodes_per_elem[4], num_attr[4]);
printf ("after ex_put_elem_block, error = %d\n", error);
error = ex_put_elem_block (exoid, ebids[5], "sphere", num_elem_in_block[5],
num_nodes_per_elem[5], num_attr[5]);
printf ("after ex_put_elem_block, error = %d\n", error);
error = ex_put_elem_block (exoid, ebids[6], "wedge", num_elem_in_block[6],
num_nodes_per_elem[6], num_attr[6]);
printf ("after ex_put_elem_block, error = %d\n", error);
/* write element block properties */
prop_names[0] = "MATL";
prop_names[1] = "DENSITY";
error = ex_put_prop_names(exoid,EX_ELEM_BLOCK,2,prop_names);
printf ("after ex_put_prop_names, error = %d\n", error);
error = ex_put_prop(exoid, EX_ELEM_BLOCK, ebids[0], "MATL", 10);
printf ("after ex_put_prop, error = %d\n", error);
error = ex_put_prop(exoid, EX_ELEM_BLOCK, ebids[1], "MATL", 20);
printf ("after ex_put_prop, error = %d\n", error);
error = ex_put_prop(exoid, EX_ELEM_BLOCK, ebids[2], "MATL", 30);
printf ("after ex_put_prop, error = %d\n", error);
error = ex_put_prop(exoid, EX_ELEM_BLOCK, ebids[3], "MATL", 40);
printf ("after ex_put_prop, error = %d\n", error);
error = ex_put_prop(exoid, EX_ELEM_BLOCK, ebids[4], "MATL", 50);
printf ("after ex_put_prop, error = %d\n", error);
error = ex_put_prop(exoid, EX_ELEM_BLOCK, ebids[5], "MATL", 60);
printf ("after ex_put_prop, error = %d\n", error);
error = ex_put_prop(exoid, EX_ELEM_BLOCK, ebids[6], "MATL", 70);
printf ("after ex_put_prop, error = %d\n", error);
/* write element connectivity */
connect = (int *) calloc(8, sizeof(int));
connect[0] = 1; connect[1] = 2; connect[2] = 3; connect[3] = 4;
error = ex_put_elem_conn (exoid, ebids[0], connect);
printf ("after ex_put_elem_conn, error = %d\n", error);
connect[0] = 1; connect[1] = 2; connect[2] = 3; connect[3] = 4;
connect[4] = 5; connect[5] = 6; connect[6] = 7; connect[7] = 8;
error = ex_put_elem_conn (exoid, ebids[1], connect);
printf ("after ex_put_elem_conn, error = %d\n", error);
connect[0] = 9; connect[1] = 10; connect[2] = 11; connect[3] = 12;
connect[4] = 13; connect[5] = 14; connect[6] = 15; connect[7] = 16;
error = ex_put_elem_conn (exoid, ebids[2], connect);
printf ("after ex_put_elem_conn, error = %d\n", error);
connect[0] = 17; connect[1] = 18; connect[2] = 19; connect[3] = 20;
error = ex_put_elem_conn (exoid, ebids[3], connect);
printf ("after ex_put_elem_conn, error = %d\n", error);
connect[0] = 21;
error = ex_put_elem_conn (exoid, ebids[4], connect);
printf ("after ex_put_elem_conn, error = %d\n", error);
connect[0] = 22;
error = ex_put_elem_conn (exoid, ebids[5], connect);
printf ("after ex_put_elem_conn, error = %d\n", error);
connect[0] = 23; connect[1] = 24; connect[2] = 25;
connect[3] = 26; connect[4] = 27; connect[5] = 28;
error = ex_put_elem_conn (exoid, ebids[6], connect);
printf ("after ex_put_elem_conn, error = %d\n", error);
free (connect);
/* write element block attributes (3 per block) */
attrib[0] = 1.0; attrib[1] = 2.0; attrib[2] = 3.0;
attrib[3] = 1.11; attrib[4] = 2.11; attrib[5] = 3.11;
attrib[6] = 1.12; attrib[7] = 2.12; attrib[8] = 3.12;
attrib[9] = 1.2; attrib[10] = 2.2; attrib[11] = 3.2;
attrib[12] = 1.3; attrib[13] = 2.3; attrib[14] = 3.3;
attrib[15] = 1.4; attrib[16] = 2.4; attrib[17] = 3.4;
attrib[18] = 1.5; attrib[19] = 2.5; attrib[20] = 3.5;
attrib[21] = 1.6; attrib[22] = 2.6; attrib[23] = 3.6;
error = ex_put_elem_attr (exoid, ebids[0], &attrib[0]);
printf ("after ex_put_elem_attr, error = %d\n", error);
error = ex_put_elem_attr (exoid, ebids[1], &attrib[3]);
printf ("after ex_put_elem_attr, error = %d\n", error);
error = ex_put_elem_attr (exoid, ebids[2], &attrib[9]);
printf ("after ex_put_elem_attr, error = %d\n", error);
error = ex_put_elem_attr (exoid, ebids[3], &attrib[12]);
printf ("after ex_put_elem_attr, error = %d\n", error);
error = ex_put_elem_attr (exoid, ebids[4], &attrib[15]);
printf ("after ex_put_elem_attr, error = %d\n", error);
error = ex_put_elem_attr (exoid, ebids[5], &attrib[18]);
printf ("after ex_put_elem_attr, error = %d\n", error);
error = ex_put_elem_attr (exoid, ebids[6], &attrib[21]);
printf ("after ex_put_elem_attr, error = %d\n", error);
/* write individual node sets */
/* COMMENTED OUT ...
error = ex_put_node_set_param (exoid, 20, 5, 5);
printf ("after ex_put_node_set_param, error = %d\n", error);
node_list[0] = 100; node_list[1] = 101; node_list[2] = 102;
node_list[3] = 103; node_list[4] = 104;
dist_fact[0] = 1.0; dist_fact[1] = 2.0; dist_fact[2] = 3.0;
dist_fact[3] = 4.0; dist_fact[4] = 5.0;
error = ex_put_node_set (exoid, 20, node_list);
printf ("after ex_put_node_set, error = %d\n", error);
error = ex_put_node_set_dist_fact (exoid, 20, dist_fact);
printf ("after ex_put_node_set_dist_fact, error = %d\n", error);
error = ex_put_node_set_param (exoid, 21, 3, 3);
printf ("after ex_put_node_set_param, error = %d\n", error);
node_list[0] = 200; node_list[1] = 201; node_list[2] = 202;
dist_fact[0] = 1.1; dist_fact[1] = 2.1; dist_fact[2] = 3.1;
error = ex_put_node_set (exoid, 21, node_list);
printf ("after ex_put_node_set, error = %d\n", error);
error = ex_put_node_set_dist_fact (exoid, 21, dist_fact);
printf ("after ex_put_node_set_dist_fact, error = %d\n", error);
error = ex_put_prop(exoid, EX_NODE_SET, 20, "FACE", 4);
printf ("after ex_put_prop, error = %d\n", error);
error = ex_put_prop(exoid, EX_NODE_SET, 21, "FACE", 5);
printf ("after ex_put_prop, error = %d\n", error);
prop_array[0] = 1000;
prop_array[1] = 2000;
error = ex_put_prop_array(exoid, EX_NODE_SET, "VELOCITY", prop_array);
printf ("after ex_put_prop_array, error = %d\n", error);
END COMMENTED OUT SECTION */
/* write concatenated node sets; this produces the same information as
* the above code which writes individual node sets
*/
/* THIS SECTION IS NOT COMMENTED OUT */
ids[0] = 20; ids[1] = 21;
num_nodes_per_set[0] = 5;
num_nodes_per_set[1] = 3;
/* num_nodes_per_set[1] = 0; Uncomment to test NULL node sets */
node_ind[0] = 0; node_ind[1] = 5;
node_list[0] = 100; node_list[1] = 101; node_list[2] = 102;
node_list[3] = 103; node_list[4] = 104;
node_list[5] = 200; node_list[6] = 201; node_list[7] = 202;
num_df_per_set[0] = 5; num_df_per_set[1] = 3;
df_ind[0] = 0; df_ind[1] = 5;
dist_fact[0] = 1.0; dist_fact[1] = 2.0; dist_fact[2] = 3.0;
dist_fact[3] = 4.0; dist_fact[4] = 5.0;
dist_fact[5] = 1.1; dist_fact[6] = 2.1; dist_fact[7] = 3.1;
error = ex_put_concat_node_sets (exoid, ids, num_nodes_per_set,
num_df_per_set, node_ind,
df_ind, node_list, dist_fact);
printf ("after ex_put_concat_node_sets, error = %d\n", error);
error = ex_put_prop(exoid, EX_NODE_SET, 20, "FACE", 4);
printf ("after ex_put_prop, error = %d\n", error);
error = ex_put_prop(exoid, EX_NODE_SET, 21, "FACE", 5);
printf ("after ex_put_prop, error = %d\n", error);
prop_array[0] = 1000;
prop_array[1] = 2000;
error = ex_put_prop_array(exoid, EX_NODE_SET, "VELOCITY", prop_array);
printf ("after ex_put_prop_array, error = %d\n", error);
/* END NOT COMMENTED OUT SECTION */
/* write individual side sets */
/* COMMENTED OUT SECTION ...
error = ex_put_side_set_param (exoid, 30, 2, 4);
printf ("after ex_put_side_set_param, error = %d\n", error);
elem_list[0] = 1; elem_list[1] = 1;
node_list[0] = 1; node_list[1] = 2;
node_list[2] = 5; node_list[3] = 6;
side_list[0] = 1; side_list[1] = 1;
dist_fact[0] = 30.0; dist_fact[1] = 30.1; dist_fact[2] = 30.2;
dist_fact[3] = 30.3;
error = ex_put_side_set (exoid, 30, elem_list, side_list);
printf ("after ex_put_side_set, error = %d\n", error);
error = ex_put_side_set_dist_fact (exoid, 30, dist_fact);
printf ("after ex_put_side_set_dist_fact, error = %d\n", error);
error = ex_put_side_set_param (exoid, 31, 2, 4);
printf ("after ex_put_side_set_param, error = %d\n", error);
elem_list[0] = 2; elem_list[1] = 2;
node_list[0] = 6; node_list[1] = 7;
node_list[2] = 7; node_list[3] = 8;
side_list[0] = 3; side_list[1] = 3;
dist_fact[0] = 31.0; dist_fact[1] = 31.1; dist_fact[2] = 31.2;
dist_fact[3] = 31.3;
error = ex_put_side_set (exoid, 31, elem_list, side_list);
printf ("after ex_put_side_set, error = %d\n", error);
error = ex_put_side_set_dist_fact (exoid, 31, dist_fact);
printf ("after ex_put_side_set_dist_fact, error = %d\n", error);
error = ex_put_prop(exoid, EX_SIDE_SET, 30, "COLOR", 100);
printf ("after ex_put_prop, error = %d\n", error);
error = ex_put_prop(exoid, EX_SIDE_SET, 31, "COLOR", 101);
printf ("after ex_put_prop, error = %d\n", error);
END COMMENTED OUT SECTION */
/* write concatenated side sets; this produces the same information as
* the above code which writes individual side sets
*/
/* THIS SECTION IS NOT COMMENTED OUT */
ids[0] = 30;
ids[1] = 31;
ids[2] = 32;
ids[3] = 33;
ids[4] = 34;
ids[5] = 35;
/* side set #1 - quad */
node_list[0] = 8; node_list[1] = 5;
node_list[2] = 6; node_list[3] = 7;
/* side set #2 - quad/hex, spanning 2 element types */
node_list[4] = 2; node_list[5] = 3;
node_list[6] = 7; node_list[7] = 8;
/* side set #3 - hex */
node_list[8] = 9; node_list[9] = 12;
node_list[10] = 11; node_list[11] = 10;
node_list[12] = 11; node_list[13] = 12;
node_list[14] = 16; node_list[15] = 15;
node_list[16] = 16; node_list[17] = 15;
node_list[18] = 11; node_list[19] = 12;
node_list[20] = 10; node_list[21] = 11;
node_list[22] = 15; node_list[23] = 14;
node_list[24] = 13; node_list[25] = 16;
node_list[26] = 12; node_list[27] = 9;
node_list[28] = 14; node_list[29] = 13;
node_list[30] = 9; node_list[31] = 10;
node_list[32] = 16; node_list[33] = 13;
node_list[34] = 14; node_list[35] = 15;
/* side set #4 - tetras */
node_list[36] = 17; node_list[37] = 18;
node_list[38] = 20;
node_list[39] = 18; node_list[40] = 19;
node_list[41] = 20;
node_list[42] = 20; node_list[43] = 19;
node_list[44] = 17;
node_list[45] = 19; node_list[46] = 18;
node_list[47] = 17;
/* side set #5 - circle and sphere */
node_list[48] = 21;
node_list[49] = 22;
/* side set #6 - wedges */
node_list[50] = 27; node_list[51] = 26;
node_list[52] = 23; node_list[53] = 24;
node_list[54] = 28; node_list[55] = 27;
node_list[56] = 24; node_list[57] = 25;
node_list[58] = 28; node_list[59] = 25;
node_list[60] = 23; node_list[61] = 26;
node_list[62] = 25; node_list[63] = 24;
node_list[64] = 23;
node_list[65] = 26; node_list[66] = 27;
node_list[67] = 28;
node_ind[0] = 0;
node_ind[1] = 4;
node_ind[2] = 8;
node_ind[3] = 36;
node_ind[4] = 47;
node_ind[5] = 49;
num_elem_per_set[0] = 2; /* two sides uses 2 elements */
num_elem_per_set[1] = 2;
num_elem_per_set[2] = 7;
num_elem_per_set[3] = 4;
num_elem_per_set[4] = 2;
num_elem_per_set[5] = 5;
/* num_elem_per_set[5] = 0; Uncomment to test NULL side sets */
num_nodes_per_set[0] = 4;
num_nodes_per_set[1] = 4;
num_nodes_per_set[2] = 28;
num_nodes_per_set[3] = 12;
num_nodes_per_set[4] = 2;
num_nodes_per_set[5] = 18;
elem_ind[0] = 0;
elem_ind[1] = 2;
elem_ind[2] = 4;
elem_ind[3] = 11;
elem_ind[4] = 15;
elem_ind[5] = 17;
elem_list[0] = 3; elem_list[1] = 3; /* side set 1: Quad #2 */
elem_list[2] = 1; elem_list[3] = 3; /* side set 2: Quad #1 & #2 */
elem_list[4] = 4; elem_list[5] = 4; /* side set 3: Hex */
elem_list[6] = 4; elem_list[7] = 4;
elem_list[8] = 4; elem_list[9] = 4;
elem_list[10] = 4;
elem_list[11] = 5; elem_list[12] = 5; /* side set 4: Tetra */
elem_list[13] = 5; elem_list[14] = 5;
elem_list[15] = 6; elem_list[16] = 7; /* side set 5: Circle & Sphere */
elem_list[17] = 8; elem_list[18] = 8; /* side set 6: Wedge */
elem_list[19] = 8; elem_list[20] = 8;
elem_list[21] = 8;
#if 0
/* side set 0 */
side_list[0]= 4; side_list[1]= 2;
/* side set 1 */
side_list[2]= 2; side_list[3]= 3;
/* side set 2 */
side_list[4]= 5; side_list[5]= 3;
side_list[6]= 3; side_list[7]= 2;
side_list[8]= 4; side_list[9]= 1;
side_list[10]= 6;
/* side set 3 */
side_list[11]= 1; side_list[12]= 2;
side_list[13]= 3; side_list[14]= 4;
/* side set 4 */
side_list[15] = 1; side_list[16] = 1;
/* side set 5 */
side_list[17]= 1; side_list[18]= 2;
side_list[19]= 3; side_list[20]= 4;
side_list[21]= 5;
#endif
error = ex_cvt_nodes_to_sides(exoid,
num_elem_per_set,
num_nodes_per_set,
elem_ind,
node_ind,
elem_list,
node_list,
side_list);
printf ("after ex_cvt_nodes_to_sides, error = %d\n", error);
#if 0
for (i=0;i<num_side_sets;i++)
{
printf("side set %d\n",i);
for (j=0;j<num_elem_per_set[i];j++)
printf(" side_list[%d]: %d\n",j,side_list[j+elem_ind[i]]);
}
#endif
num_df_per_set[0] = 4;
num_df_per_set[1] = 4;
num_df_per_set[2] = 0;
num_df_per_set[3] = 0;
num_df_per_set[4] = 0;
num_df_per_set[5] = 0;
df_ind[0] = 0;
df_ind[1] = 4;
/* side set #1 df */
dist_fact[0] = 30.0; dist_fact[1] = 30.1;
dist_fact[2] = 30.2; dist_fact[3] = 30.3;
/* side set #2 df */
dist_fact[4] = 31.0; dist_fact[5] = 31.1;
dist_fact[6] = 31.2; dist_fact[7] = 31.3;
error = ex_put_concat_side_sets (exoid, ids, num_elem_per_set,
num_df_per_set, elem_ind, df_ind,
elem_list, side_list, dist_fact);
printf ("after ex_put_concat_side_sets, error = %d\n", error);
error = ex_put_prop(exoid, EX_SIDE_SET, 30, "COLOR", 100);
printf ("after ex_put_prop, error = %d\n", error);
error = ex_put_prop(exoid, EX_SIDE_SET, 31, "COLOR", 101);
printf ("after ex_put_prop, error = %d\n", error);
/* END COMMENTED OUT SECTION */
/* write QA records */
num_qa_rec = 2;
qa_record[0][0] = "TESTWT1";
qa_record[0][1] = "testwt1";
qa_record[0][2] = "03/16/94";
qa_record[0][3] = "15:41:33";
qa_record[1][0] = "FASTQ";
qa_record[1][1] = "fastq";
qa_record[1][2] = "07/07/93";
qa_record[1][3] = "16:41:33";
error = ex_put_qa (exoid, num_qa_rec, qa_record);
printf ("after ex_put_qa, error = %d\n", error);
/* write information records */
num_info = 3;
info[0] = "This is the first information record.";
info[1] = "This is the second information record.";
info[2] = "This is the third information record.";
error = ex_put_info (exoid, num_info, info);
printf ("after ex_put_info, error = %d\n", error);
/* write results variables parameters and names */
num_glo_vars = 1;
var_names[0] = "glo vars";
error = ex_put_var_param (exoid, "g", num_glo_vars);
printf ("after ex_put_var_param, error = %d\n", error);
error = ex_put_var_name(exoid, "g", 1, var_names[0]);
printf ("after ex_put_var_name, error = %d\n", error);
num_nod_vars = 2;
var_names[0] = "nod_var0";
var_names[1] = "nod_var1";
error = ex_put_var_param (exoid, "n", num_nod_vars);
printf ("after ex_put_var_param, error = %d\n", error);
error = ex_put_var_names (exoid, "n", num_nod_vars, var_names);
printf ("after ex_put_var_names, error = %d\n", error);
num_ele_vars = 3;
var_names[0] = "ele_var0";
var_names[1] = "ele_var1";
var_names[2] = "ele_var2";
error = ex_put_var_param (exoid, "e", num_ele_vars);
printf ("after ex_put_var_param, error = %d\n", error);
error = ex_put_var_names (exoid, "e", num_ele_vars, var_names);
printf ("after ex_put_var_names, error = %d\n", error);
/* write element variable truth table */
truth_tab = (int *) calloc ((num_elem_blk*num_ele_vars), sizeof(int));
k = 0;
for (i=0; i<num_elem_blk; i++)
{
for (j=0; j<num_ele_vars; j++)
{
truth_tab[k] = k+1;
k++;
}
}
truth_tab[6] = 0;
/* commented out to test ex_get_elem_var_tab in testrd1
error = ex_put_elem_var_tab (exoid, num_elem_blk, num_ele_vars, truth_tab);
printf ("after ex_put_elem_var_tab, error = %d\n", error); */
free (truth_tab);
/* for each time step, write the analysis results;
* the code below fills the arrays glob_var_vals,
* nodal_var_vals, and elem_var_vals with values for debugging purposes;
* obviously the analysis code will populate these arrays
*/
whole_time_step = 1;
num_time_steps = 10;
glob_var_vals = (float *) calloc (num_glo_vars, CPU_word_size);
nodal_var_vals = (float *) calloc (num_nodes, CPU_word_size);
elem_var_vals = (float *) calloc (4, CPU_word_size);
for (i=0; i<num_time_steps; i++)
{
time_value = (float)(i+1)/100.;
/* write time value */
error = ex_put_time (exoid, whole_time_step, &time_value);
printf ("after ex_put_time, error = %d\n", error);
/* write global variables */
for (j=0; j<num_glo_vars; j++)
{
glob_var_vals[j] = (float)(j+2) * time_value;
}
error = ex_put_glob_vars (exoid, whole_time_step, num_glo_vars,
glob_var_vals);
printf ("after ex_put_glob_vars, error = %d\n", error);
/* write nodal variables */
for (k=1; k<=num_nod_vars; k++)
{
for (j=0; j<num_nodes; j++)
{
nodal_var_vals[j] = (float)k + ((float)(j+1) * time_value);
}
error = ex_put_nodal_var (exoid, whole_time_step, k, num_nodes,
nodal_var_vals);
printf ("after ex_put_nodal_var, error = %d\n", error);
}
/* write element variables */
for (k=1; k<=num_ele_vars; k++)
{
for (j=0; j<num_elem_blk; j++)
{
for (m=0; m<num_elem_in_block[j]; m++)
{
elem_var_vals[m] = (float)(k+1) + (float)(j+2) +
((float)(m+1)*time_value);
/* printf("elem_var_vals[%d]: %f\n",m,elem_var_vals[m]); */
}
if (k == 1 && j == 2)
continue; /* skip element block 3, variable 1 */
else
{
error = ex_put_elem_var (exoid, whole_time_step, k, ebids[j],
num_elem_in_block[j], elem_var_vals);
printf ("after ex_put_elem_var, error = %d\n", error);
}
}
}
whole_time_step++;
/* update the data file; this should be done at the end of every time step
* to ensure that no data is lost if the analysis dies
*/
error = ex_update (exoid);
printf ("after ex_update, error = %d\n", error);
}
free(glob_var_vals);
free(nodal_var_vals);
free(elem_var_vals);
/* close the EXODUS files
*/
error = ex_close (exoid);
printf ("after ex_close, error = %d\n", error);
return 0;
}
void
PeridigmNS::InterfaceData::InitializeExodusOutput(Teuchos::RCP<Epetra_Vector> exodusMeshElementConnectivity, Teuchos::RCP<Epetra_Vector> exodusMeshNodePositions){
if(comm->NumProc()>1){
filename << "Interfaces.e." << comm->NumProc() << "." << comm->MyPID();
}
else{
filename << "Interfaces.e";
}
std::string outputFileNameStr = filename.str();
std::vector<char> writable(outputFileNameStr.size() + 1);
std::copy(outputFileNameStr.begin(), outputFileNameStr.end(), writable.begin());
int spaDim = 3; // force three dimensional output
const int numShells = numOwnedPoints;
const int numNodes = exodusMeshNodePositions->Map().NumMyElements();
int error_int;
int CPU_word_size = 0;
int IO_word_size = 0;
/* create EXODUS II file */
const int output_exoid = ex_create (&writable[0],EX_CLOBBER,&CPU_word_size, &IO_word_size);
exoid = output_exoid;
// scan the connectivity to see if there are quads and tets:
numQuads = 0;
numTris = 0;
for(int i=0;i<numOwnedPoints;++i){
if(interfaceNodesMap->ElementSize(i)==4) numQuads++;
if(interfaceNodesMap->ElementSize(i)==3) numTris++;
}
TEUCHOS_TEST_FOR_EXCEPTION(numQuads+numTris!=numShells,std::logic_error,"numQuads " << numQuads << " + numTris " << numTris << " should sum up to numShells " << numShells);
const int numBlocks = 2;
error_int = ex_put_init(exoid, &writable[0], spaDim, numNodes, numShells, numBlocks, 0, 0);
TEUCHOS_TEST_FOR_EXCEPTION(error_int,std::logic_error,"ex_put_init(): Failure");
// write initial coordinates and node/element maps
float * x = new float[numNodes];
float * y = new float[numNodes];
float * z = new float[numNodes];
int * shellMap = new int[numShells];
int * nodeMap = new int[numNodes];
for(int i=0;i<numNodes;++i){
int nodeIndex = exodusMeshNodePositions->Map().FirstPointInElement(i);
x[i] = (*exodusMeshNodePositions)[nodeIndex+0];
y[i] = (*exodusMeshNodePositions)[nodeIndex+1];
z[i] = (*exodusMeshNodePositions)[nodeIndex+2];
nodeMap[i] = exodusMeshNodePositions->Map().GID(i) + 1; // numbering is one based in exodus
}
for(int i=0;i<numShells;++i){
shellMap[i] = interfaceNodesMap->GID(i) +1; // numbering is one based in exodus
}
error_int = ex_put_coord(exoid, x, y, z);
char * coord_names[3];
coord_names[0] = (char*) "coordinates_x";
coord_names[1] = (char*) "coordinates_y";
coord_names[2] = (char*) "coordinates_x";
error_int = ex_put_coord_names(exoid, coord_names);
TEUCHOS_TEST_FOR_EXCEPTION(error_int,std::logic_error,"ex_put_coord_names(): Failure");
error_int = ex_put_elem_num_map(exoid, shellMap);
TEUCHOS_TEST_FOR_EXCEPTION(error_int,std::logic_error,"ex_put_elem_num_map(): Failure");
error_int = ex_put_node_num_map(exoid, nodeMap);
TEUCHOS_TEST_FOR_EXCEPTION(error_int,std::logic_error,"ex_put_node_num_map(): Failure");
delete[] shellMap;
delete[] nodeMap;
delete[] x; delete[] y; delete[] z;
// write quad blocks:
int blockIndex = 0;
blockIndex ++;
int num_nodes_per_elem_q4 = 4;
const std::string elem_type_str_q4 = numQuads > 0 ? "QUAD4" : "NULL";
char * elem_type_q4 = const_cast<char *>(elem_type_str_q4.c_str());
const int numElemInBlock_q4 = numQuads;
error_int = ex_put_elem_block(exoid, blockIndex, elem_type_q4, numElemInBlock_q4, num_nodes_per_elem_q4, 0); // no attributes put in output file
TEUCHOS_TEST_FOR_EXCEPTION(error_int,std::logic_error,"ex_put_elem_block(): Failure");
blockIndex ++;
int num_nodes_per_elem_t3 = 3;
const std::string elem_type_str_t3 = numTris > 0 ? "TRI3": "NULL";
char * elem_type_t3 = const_cast<char *>(elem_type_str_t3.c_str());
const int numElemInBlock_t3 = numTris;
error_int = ex_put_elem_block(exoid, blockIndex, elem_type_t3, numElemInBlock_t3, num_nodes_per_elem_t3, 0); // no attributes put in output file
TEUCHOS_TEST_FOR_EXCEPTION(error_int,std::logic_error,"ex_put_elem_block(): Failure");
// write elem connectivities
// HEADS UP: the connectivities will not write to file until ex_close is called
// also note that the nodes must be the local node ids
const int connLength = interfaceNodes->Map().NumMyElements();
blockIndex = 0;
blockIndex ++;
int conn_index = 0;
int * block_connect_q4 = new int[numQuads*num_nodes_per_elem_q4];
for(int it=0;it<connLength;++it)
{
if(interfaceNodes->Map().ElementSize(it)!=4) continue;
int elemIndex = interfaceNodes->Map().FirstPointInElement(it);
for(int nn=0;nn<4;++nn){
int node = static_cast<int>( (*interfaceNodes)[elemIndex+nn] );
block_connect_q4[conn_index*num_nodes_per_elem_q4+nn] = exodusMeshNodePositions->Map().LID(node) + 1;// nodes are 1 based in exodus
}
conn_index++;
}
error_int = ex_put_elem_conn(exoid, blockIndex, block_connect_q4);
delete[] block_connect_q4;
blockIndex ++;
conn_index = 0;
int * block_connect_t3 = new int[numTris*num_nodes_per_elem_t3];
for(int it=0;it<connLength;++it)
{
if(interfaceNodes->Map().ElementSize(it)!=3) continue;
int elemIndex = interfaceNodes->Map().FirstPointInElement(it);
for(int nn=0;nn<3;++nn){
int node = static_cast<int>( (*interfaceNodes)[elemIndex+nn] );
block_connect_t3[conn_index*num_nodes_per_elem_t3+nn] = exodusMeshNodePositions->Map().LID(node) + 1;// nodes are 1 based in exodus
}
conn_index++;
}
error_int = ex_put_elem_conn(exoid, blockIndex, block_connect_t3);
delete[] block_connect_t3;
int numVariables = 1; // TODO careful with this, if more fields are added to the interface data this must be updated
char** eleVarNames = new char*[numVariables];
std::vector<std::string> strNames;
strNames.push_back("interface_aperture");
for (int i=0;i<numVariables;++i)
eleVarNames[i] = (char*) (strNames[i].c_str());
error_int = ex_put_var_param(exoid, (char*) "e", numVariables);
error_int = ex_put_var_names(exoid, (char*) "e", numVariables, &eleVarNames[0]);
delete [] eleVarNames;
// write the truth table
int * truth_tab = new int[numVariables*numBlocks];
for(int i=0;i<numVariables*numBlocks;++i)
truth_tab[i] = 1;
error_int = ex_put_elem_var_tab (exoid, numBlocks, numVariables, truth_tab);
delete [] truth_tab;
error_int = ex_update(exoid);
TEUCHOS_TEST_FOR_EXCEPTION(error_int,std::logic_error,"Exodus file close failed.");
error_int = ex_close(exoid);
TEUCHOS_TEST_FOR_EXCEPTION(error_int,std::logic_error,"Exodus file close failed.");
}
int main (int argc, char **argv)
{
int num_glo_vars = 10;
int num_nod_vars = 2;
int CPU_word_size = 8;
int IO_word_size = 8;
const char* title = "This is a 2D mesh example with tri, quad, beam, truss, circle";
int ebids[] = {100, 200, 300, 400, 500};
int num_dim = 2;
int num_nodes = 13;
int num_elem = 20;
int num_elem_blk = 5;
int num_node_sets = 2;
int num_side_sets = 2;
/* create EXODUS II file */
int exoid = ex_create ("twod.e", /* filename path */
EX_CLOBBER, /* create mode */
&CPU_word_size, /* CPU float word size in bytes */
&IO_word_size); /* I/O float word size in bytes */
ex_opts(EX_VERBOSE);
/* initialize file with parameters */
ex_put_init (exoid, title, num_dim, num_nodes, num_elem,
num_elem_blk, num_node_sets, num_side_sets);
/* write nodal coordinates values and names to database */
{
double x[13], y[13];
x[0] = 0.0; y[0] = 0.0;
x[1] = -0.5; y[1] = -0.5;
x[2] = 0.5; y[2] = -0.5;
x[3] = 0.5; y[3] = 0.5;
x[4] = -0.5; y[4] = 0.5;
x[5] = -1.0; y[5] = -1.0;
x[6] = 1.0; y[6] = -1.0;
x[7] = 1.0; y[7] = 1.0;
x[8] = -1.0; y[8] = 1.0;
x[9] = -2.0; y[9] = 0.0;
x[10] = 0.0; y[10] = -2.0;
x[11] = 2.0; y[11] = 0.0;
x[12] = 0.0; y[12] = 2.0;
ex_put_coord (exoid, x, y, 0);
}
{
const char* coord_names[] = {"xcoor", "ycoor"};
ex_put_coord_names (exoid, (char**)coord_names);
}
{
int node_map[] = {10,20,30,40,50,60,70,80,90,100,110,120,130};
ex_put_node_num_map(exoid, node_map);
}
/* write element order map */
{
int elem_map[] = {11,21,31,41, 52, 62, 72, 82, 93,103,113,123,133,143,153,163, 174,184,194,204};
ex_put_elem_num_map (exoid, elem_map);
}
/* write element block parameters */
{
const char* block_names[] = {"Triangles", "Quadrilaterals", "", "Trusses", "Circles"};
int num_elem_in_block[] = {4, 4, 4, 4, 4};
int num_nodes_per_elem[] = {3, 4, 2, 2, 1};
ex_put_elem_block (exoid, ebids[0], "triangle", num_elem_in_block[0], num_nodes_per_elem[0], 0);
ex_put_elem_block (exoid, ebids[1], "quad", num_elem_in_block[1], num_nodes_per_elem[1], 0);
ex_put_elem_block (exoid, ebids[2], "beam", num_elem_in_block[2], num_nodes_per_elem[2], 3);
ex_put_elem_block (exoid, ebids[3], "truss", num_elem_in_block[3], num_nodes_per_elem[3], 1);
ex_put_elem_block (exoid, ebids[4], "circle", num_elem_in_block[4], num_nodes_per_elem[4], 2);
/* Write element block names */
ex_put_names(exoid, EX_ELEM_BLOCK, (char**)block_names);
}
/* write element connectivity */
{
int conn_t[] = {2,3,1, 3,4,1, 4,5,1, 5,2,1};
int conn_q[] = {6,7,3,2, 7,8,4,3, 8,9,5,4, 9,6,2,5};
int conn_B[] = {11,7, 8,13, 13,9, 6,11};
int conn_T[] = {10,6, 9,10, 7,12, 12,8};
int conn_c[] = {6,7,8,9};
ex_put_elem_conn (exoid, ebids[0], conn_t);
ex_put_elem_conn (exoid, ebids[1], conn_q);
ex_put_elem_conn (exoid, ebids[2], conn_B);
ex_put_elem_conn (exoid, ebids[3], conn_T);
ex_put_elem_conn (exoid, ebids[4], conn_c);
}
/* write element block attributes */
{
const char* attn_T[] = {"Area"};
double attr_T[] = {1.0, 1.1, 1.2, 1.3};
const char* attn_B[] = {"A", "I", "J"};
double attr_B[] = {1.0, 100.0, 200.0, 1.1, 100.1, 200.1, 1.2, 100.2, 200.2, 1.3, 100.3, 200.3};
const char* attn_c[] = {"Radius", "A"};
double attr_c[] = {1.0, 3.14, 1.1, 4.14, 1.2, 5.14, 1.3, 6.14};
ex_put_elem_attr (exoid, ebids[2], attr_B);
ex_put_elem_attr (exoid, ebids[3], attr_T);
ex_put_elem_attr (exoid, ebids[4], attr_c);
ex_put_elem_attr_names (exoid, ebids[2], (char**)attn_B);
ex_put_elem_attr_names (exoid, ebids[3], (char**)attn_T);
ex_put_elem_attr_names (exoid, ebids[4], (char**)attn_c);
}
/* write individual node sets */
{
int num_nodes_in_nset[] = {5, 8};
int nsids[] = {20, 22};
int nod1[] = {5,4,3,2,1};
int nod2[] = {6,7,8,9,2,3,4,5};
const char* nset_names[] = {"Triangle_Nodes", "Quadrilateral_Nodes"};
ex_put_set_param (exoid, EX_NODE_SET, nsids[0], num_nodes_in_nset[0], 0);
ex_put_set_param (exoid, EX_NODE_SET, nsids[1], num_nodes_in_nset[1], 0);
ex_put_set (exoid, EX_NODE_SET, nsids[0], nod1, 0);
ex_put_set (exoid, EX_NODE_SET, nsids[1], nod2, 0);
ex_put_names(exoid, EX_NODE_SET, (char**)nset_names);
}
{
/* write individual side sets */
int num_face_in_sset[] = {4,4};
int ssids[] = {100,200};
int ss1el[] = {1,2,3,4};
int ss1si[] = {1,1,1,1};
int ss2el[] = {5,7,6,8};
int ss2si[] = {1,1,1,1};
const char* sset_names[] = {"A", "B"};
ex_put_set_param (exoid, EX_SIDE_SET, ssids[0], num_face_in_sset[0], 0);
ex_put_set_param (exoid, EX_SIDE_SET, ssids[1], num_face_in_sset[1], 0);
ex_put_set (exoid, EX_SIDE_SET, ssids[0], ss1el, ss1si);
ex_put_set (exoid, EX_SIDE_SET, ssids[1], ss2el, ss2si);
ex_put_names(exoid, EX_SIDE_SET, (char**)sset_names);
}
/* write results variables parameters and names */
{
const char* gvarn[] = {"g_01", "g_02", "g_03", "g_04", "g_05", "g_06", "g_07", "g_08", "g_09", "g_10"};
ex_put_variable_param (exoid, EX_GLOBAL, num_glo_vars);
ex_put_variable_names (exoid, EX_GLOBAL, num_glo_vars, (char**)gvarn);
}
{
const char* nvarn[] = {"disp_x", "disp_y"};
ex_put_variable_param (exoid, EX_NODAL, num_nod_vars);
ex_put_variable_names (exoid, EX_NODAL, num_nod_vars, (char**)nvarn);
}
#if 0
num_ele_vars = 3;
/* 0 1 2 3 */
/* 12345678901234567890123456789012 */
var_names[0] = "this_variable_name_is_short";
var_names[1] = "this_variable_name_is_just_right";
var_names[2] = "this_variable_name_is_tooooo_long";
ex_put_var_param (exoid, "e", num_ele_vars);
printf ("after ex_put_var_param, %d\n", error);
if (error) {
ex_close (exoid);
exit(-1);
}
ex_put_var_names (exoid, "e", num_ele_vars, var_names);
printf ("after ex_put_var_names, %d\n", error);
if (error) {
ex_close (exoid);
exit(-1);
}
{
num_nset_vars = 3;
var_names[0] = "ns_var0";
var_names[1] = "ns_var1";
var_names[2] = "ns_var2";
ex_put_var_param (exoid, "m", num_nset_vars);
printf ("after ex_put_var_param, %d\n", error);
if (error) {
ex_close (exoid);
exit(-1);
}
ex_put_var_names (exoid, "m", num_nset_vars, var_names);
printf ("after ex_put_var_names, %d\n", error);
if (error) {
ex_close (exoid);
exit(-1);
}
}
{
num_sset_vars = 3;
var_names[0] = "ss_var0";
var_names[1] = "ss_var1";
var_names[2] = "ss_var2";
ex_put_var_param (exoid, "s", num_sset_vars);
printf ("after ex_put_var_param, %d\n", error);
if (error) {
ex_close (exoid);
exit(-1);
}
ex_put_var_names (exoid, "s", num_sset_vars, var_names);
printf ("after ex_put_var_names, %d\n", error);
if (error) {
ex_close (exoid);
exit(-1);
}
}
#endif
/* for each time step, write the analysis results;
* the code below fills the arrays glob_var_vals,
* nodal_var_vals, and elem_var_vals with values for debugging purposes;
* obviously the analysis code will populate these arrays
*/
{
int i, j, k;
int whole_time_step = 1;
int num_time_steps = 10;
double gvar[10];
double nvar[20];
for (i=0; i<num_time_steps; i++) {
double time_value = (double)(i)/100.;
ex_put_time (exoid, whole_time_step, &time_value);
for (j=0; j<num_glo_vars; j++) {
gvar[j] = (double)(j+2) * time_value;
}
ex_put_glob_vars (exoid, whole_time_step, num_glo_vars, gvar);
/* write nodal variables */
for (k=0; k < num_nod_vars; k++) {
for (j=0; j<num_nodes; j++) {
nvar[j] = (double)k + ((double)(j+1) * time_value);
}
ex_put_nodal_var (exoid, whole_time_step, k+1, num_nodes, nvar);
}
#if 0
/* write element variables */
for (k=1; k<=num_ele_vars; k++)
{
for (j=0; j<num_elem_blk; j++)
{
for (m=0; m<num_elem_in_block[j]; m++)
{
elem_var_vals[m] = (float)(k+1) + (float)(j+2) +
((float)(m+1)*time_value);
/* printf("elem_var_vals[%d]: %f\n",m,elem_var_vals[m]); */
}
ex_put_elem_var (exoid, whole_time_step, k, ebids[j],
num_elem_in_block[j], elem_var_vals);
printf ("after ex_put_elem_var, %d\n", error);
if (error) {
ex_close (exoid);
exit(-1);
}
}
}
/* write sideset variables */
for (k=1; k<=num_sset_vars; k++)
{
for (j=0; j<num_side_sets; j++)
{
for (m=0; m<num_face_in_sset[j]; m++)
{
sset_var_vals[m] = (float)(k+2) + (float)(j+3) +
((float)(m+1)*time_value);
/* printf("sset_var_vals[%d]: %f\n",m,sset_var_vals[m]); */
}
ex_put_sset_var (exoid, whole_time_step, k, ssids[j],
num_face_in_sset[j], sset_var_vals);
printf ("after ex_put_sset_var, %d\n", error);
if (error) {
ex_close (exoid);
exit(-1);
}
}
}
/* write nodeset variables */
for (k=1; k<=num_nset_vars; k++)
{
for (j=0; j<num_node_sets; j++)
{
for (m=0; m<num_nodes_in_nset[j]; m++)
{
nset_var_vals[m] = (float)(k+3) + (float)(j+4) +
((float)(m+1)*time_value);
/* printf("nset_var_vals[%d]: %f\n",m,nset_var_vals[m]); */
}
ex_put_nset_var (exoid, whole_time_step, k, nsids[j],
num_nodes_in_nset[j], nset_var_vals);
printf ("after ex_put_nset_var, %d\n", error);
if (error) {
ex_close (exoid);
exit(-1);
}
}
}
#endif
whole_time_step++;
}
}
ex_close (exoid);
return 0;
}
int main (int argc, char *argv[]){
char **str2,*line,*curr;
const char* ext=".exo";
int
i,j,k,n,n1,cpu_word_size,io_word_size,exo_file,
num_axes,num_nodes,num_elements,num_blocks,
num_side_sets,num_node_sets,num_time_steps,
num_global_vars,
num_nodal_vars,num_element_vars,*ids,*iscr,
*nsssides,*nssdfac,*elem_list,*side_list,
*nnsnodes,*nnsdfac,*node_list;
double
*scr,*x,*y,*z,
*escr;
char * blknames = NULL;
int *num_elem_in_block = NULL;
/* QA Info */
printf("%s: %s, %s\n", qainfo[0], qainfo[2], qainfo[1]);
/* usage message*/
if(argc != 2){
printf("%s matlab_file_name.\n",argv[0]);
printf(" the matlab_file_name is required\n");
printf("%d", argc);
exit(1);
}
/*open input file*/
mat_file = Mat_Open(argv[1], MAT_ACC_RDONLY);
if (mat_file == NULL) {
printf("Error opening matlab file %s\n", argv[1]);
return(1);
}
/*open output file*/
cpu_word_size=sizeof(double);
io_word_size=sizeof(double);
/* QA records */
ext=".exo";
line = (char *) calloc (2049,sizeof(char));
strcpy(line,argv[1]);
strtok(line,".");
strcat(line,ext);
exo_file = ex_create(line,EX_CLOBBER,&cpu_word_size,&io_word_size);
if (exo_file < 0){
printf("error creating %s\n",line);
exit(1);
}
/* print */
fprintf(stderr,"translating %s to %s ... ",argv[1],line);
/* read database parameters */
matGetInt("naxes", 1, 1,&num_axes);
matGetInt("nnodes", 1, 1,&num_nodes);
matGetInt("nelems", 1, 1,&num_elements);
matGetInt("nblks", 1, 1,&num_blocks);
matGetInt("nnsets", 1, 1,&num_node_sets);
matGetInt("nssets", 1, 1,&num_side_sets);
matGetInt("nsteps", 1, 1,&num_time_steps);
matGetInt("ngvars", 1, 1,&num_global_vars);
matGetInt("nnvars", 1, 1,&num_nodal_vars);
matGetInt("nevars", 1, 1,&num_element_vars);
/*export parameters */
ex_put_init(exo_file,line,
num_axes,num_nodes,num_elements,num_blocks,
num_node_sets,num_side_sets);
free(line);
if ( num_global_vars > 0 ){
ex_put_variable_param(exo_file,EX_GLOBAL,num_global_vars);
}
if ( num_nodal_vars > 0 ){
ex_put_variable_param(exo_file,EX_NODAL,num_nodal_vars);
}
if ( num_element_vars > 0 ){
ex_put_variable_param(exo_file,EX_ELEM_BLOCK,num_element_vars);
}
/* nodal coordinates */
x = (double *) calloc(num_nodes,sizeof(double));
y = (double *) calloc(num_nodes,sizeof(double));
if (num_axes == 3)
z = (double *) calloc(num_nodes,sizeof(double));
else
z = NULL;
matGetDbl("x0", num_nodes, 1, x);
matGetDbl("y0", num_nodes, 1, y);
if (num_axes == 3)
matGetDbl("z0", num_nodes,1,z);
ex_put_coord(exo_file,x,y,z);
free(x);
free(y);
if (num_axes == 3){
free(z);
}
/* side sets (section by dgriffi) */
if(num_side_sets > 0){
/* ssids */
ids = (int *) calloc(num_side_sets,sizeof(int));
matGetInt("ssids",num_side_sets, 1,ids);
/* nsssides */
nsssides = (int *) calloc(num_side_sets,sizeof(int));
matGetInt("nsssides",num_side_sets,1,nsssides);
/* nssdfac */
nssdfac = (int *) calloc(num_side_sets,sizeof(int));
matGetInt("nssdfac",num_side_sets,1,nssdfac);
for(i=0;i<num_side_sets;i++){
char name[32];
ex_put_set_param(exo_file,EX_SIDE_SET,ids[i],nsssides[i],nssdfac[i]);
elem_list = (int *) calloc(nsssides[i],sizeof(int));
side_list = (int *) calloc(nsssides[i],sizeof(int));
escr = (double *) calloc(nssdfac[i],sizeof(double));
sprintf(name,"sselem%02d",i+1);
matGetInt(name,nsssides[i],1,elem_list);
sprintf(name,"ssside%02d",i+1);
matGetInt(name,nsssides[i],1,side_list);
ex_put_set(exo_file,EX_SIDE_SET,ids[i],elem_list,side_list);
free(elem_list);
free(side_list);
sprintf(name,"ssfac%02d",i+1);
matGetDbl(name,nssdfac[i],1,escr);
ex_put_set_dist_fact(exo_file,EX_SIDE_SET,ids[i],escr);
free(escr);
}
free(nsssides);
free(nssdfac);
free(ids);
}
/* node sets (section by dgriffi) */
if(num_node_sets > 0){
/* nsids */
ids = (int *) calloc(num_node_sets,sizeof(int));
matGetInt("nsids",num_node_sets, 1,ids);
/* nnsnodes */
nnsnodes = (int *) calloc(num_node_sets,sizeof(int));
matGetInt("nnsnodes",num_node_sets,1,nnsnodes);
/* nnsdfac */
nnsdfac = (int *) calloc(num_node_sets,sizeof(int));
matGetInt("nnsdfac",num_node_sets,1,nnsdfac);
for(i=0;i<num_node_sets;i++){
char name[32];
ex_put_set_param(exo_file,EX_NODE_SET,ids[i],nnsnodes[i],nnsdfac[i]);
node_list = (int *) calloc(nnsnodes[i],sizeof(int));
escr = (double *) calloc(nnsdfac[i],sizeof(double));
sprintf(name,"nsnod%02d",i+1);
matGetInt(name,nnsnodes[i],1,node_list);
ex_put_set(exo_file,EX_NODE_SET,ids[i],node_list,NULL);
free(node_list);
sprintf(name,"nsfac%02d",i+1);
matGetDbl(name,nnsdfac[i],1,escr);
ex_put_set_dist_fact(exo_file,EX_NODE_SET,ids[i],escr);
free(escr);
}
free(nnsdfac);
free(nnsnodes);
free(ids);
}
/* element blocks */
/* get elem block ids */
ids = (int *) calloc(num_blocks,sizeof(int));
matGetInt("blkids",num_blocks,1,ids);
/* get elem block types */
blknames = (char *) calloc(num_blocks*(MAX_STR_LENGTH+1),sizeof(char));
matGetStr("blknames",blknames);
num_elem_in_block = (int *) calloc(num_blocks,sizeof(int));
curr = blknames;
curr = strtok(curr,"\n");
for(i=0;i<num_blocks;i++){
char name[32];
sprintf(name,"blk%02d",i+1);
n1 = matArrNRow(name);
n = matArrNCol(name);
iscr = (int *) calloc(n*n1,sizeof(int));
matGetInt(name,n1,n,iscr);
num_elem_in_block[i]=n;
ex_put_elem_block(exo_file,ids[i],curr,n,n1,0);
ex_put_conn(exo_file,EX_ELEM_BLOCK,ids[i],iscr,NULL,NULL);
free(iscr);
curr = strtok(NULL, "\n");
}
free(blknames);
/* time values */
if (num_time_steps > 0 ) {
scr = (double *) calloc(num_time_steps,sizeof(double));
matGetDbl( "time", num_time_steps, 1,scr);
for (i=0;i<num_time_steps;i++){
ex_put_time(exo_file,i+1,&scr[i]);
}
free(scr);
}
/* global variables */
if (num_global_vars > 0 ){
int max_name_length = ex_inquire_int(exo_file, EX_INQ_DB_MAX_USED_NAME_LENGTH);
char *str = (char *) calloc(num_global_vars * (max_name_length+1), sizeof(char));
matGetStr("gnames",str);
str2 = (char **) calloc(num_global_vars,sizeof(char*));
curr = strtok(str,"\n");
for(i=0;i<num_global_vars;i++){
str2[i]=curr;
curr = strtok(NULL,"\n");
}
ex_put_variable_names(exo_file, EX_GLOBAL, num_global_vars, str2);
free(str);
free(str2);
{
double * global_var_vals;
double * temp;
global_var_vals = (double *) calloc(num_global_vars*num_time_steps,sizeof(double));
temp = (double *) calloc(num_time_steps,sizeof(double));
for (j=0;j<num_global_vars;j++) {
char name[32];
sprintf(name,"gvar%02d",j+1);
matGetDbl(name,num_time_steps,1,temp);
for (i=0; i < num_time_steps; i++) {
global_var_vals[num_global_vars*i+j]=temp[i];
}
}
for (i=0; i<num_time_steps; i++) {
size_t offset = num_global_vars * i;
ex_put_var(exo_file,i+1,EX_GLOBAL,1,0,num_global_vars,&global_var_vals[offset]);
}
free(temp);
free(global_var_vals);
}
}
/* nodal variables */ /* section by dtg */
if (num_nodal_vars > 0){
int max_name_length = ex_inquire_int(exo_file, EX_INQ_DB_MAX_USED_NAME_LENGTH);
char *str = (char *) calloc(num_nodal_vars * (max_name_length+1), sizeof(char));
matGetStr("nnames",str);
str2 = (char **) calloc(num_nodal_vars,sizeof(char*));
curr = strtok(str,"\n");
for(i=0;i<num_nodal_vars;i++){
str2[i]=curr;
curr = strtok(NULL,"\n");
}
ex_put_variable_names(exo_file, EX_NODAL, num_nodal_vars, str2);
free(str);
free(str2);
{
double * nodal_var_vals;
for (i=0;i<num_nodal_vars;i++) {
char name[32];
nodal_var_vals = (double *) calloc(num_nodes*num_time_steps,sizeof(double));
sprintf(name,"nvar%02d",i+1);
matGetDbl(name,num_nodes,num_time_steps,nodal_var_vals);
for (j=0;j<num_time_steps;j++) {
ex_put_var(exo_file,j+1,EX_NODAL,i+1,num_nodes,1,nodal_var_vals+num_nodes*j);
}
free(nodal_var_vals);
}
}
}
/* elemental variables */ /* section by dtg */
if (num_element_vars > 0){
int max_name_length = ex_inquire_int(exo_file, EX_INQ_DB_MAX_USED_NAME_LENGTH);
char *str = (char *) calloc(num_element_vars * (max_name_length+1), sizeof(char));
matGetStr("enames",str);
str2 = (char **) calloc(num_element_vars,sizeof(char*));
curr = strtok(str,"\n");
for(i=0;i<num_element_vars;i++){
str2[i]=curr;
curr = strtok(NULL,"\n");
}
ex_put_variable_names(exo_file, EX_ELEM_BLOCK, num_element_vars, str2);
free(str);
free(str2);
{
double * element_var_vals;
for (i=0;i<num_element_vars;i++) {
char name[32];
element_var_vals = (double *) calloc(num_elements*num_time_steps,sizeof(double));
sprintf(name,"evar%02d",i+1);
matGetDbl(name,num_elements,num_time_steps,element_var_vals);
n=0;
for (j=0;j<num_time_steps;j++) {
for (k=0;k<num_blocks;k++) {
ex_put_var(exo_file,j+1,EX_ELEM_BLOCK, i+1,ids[k],num_elem_in_block[k],element_var_vals+n);
n=n+num_elem_in_block[k];
}
}
free(element_var_vals);
}
}
}
free(ids);
/* node and element number maps */
ids = (int *) calloc (num_nodes,sizeof(int));
if ( !matGetInt("node_num_map",num_nodes,1,ids)){
ex_put_node_num_map(exo_file,ids);
}
free(ids);
ids = (int *) calloc (num_elements,sizeof(int));
if ( !matGetInt("elem_num_map",num_elements,1,ids)){
ex_put_elem_num_map(exo_file,ids);
}
free(ids);
free(num_elem_in_block);
/* close exo file */
ex_close(exo_file);
/* close mat file */
Mat_Close(mat_file);
/* */
fprintf(stderr,"done.\n");
/* exit status */
add_to_log("mat2exo", 0);
return(0);
}
