You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
237 lines
9.2 KiB
237 lines
9.2 KiB
2 years ago
|
/*
|
||
|
|
* Copyright(C) 1999-2021 National Technology & Engineering Solutions
|
||
|
|
* of Sandia, LLC (NTESS). Under the terms of Contract DE-NA0003525 with
|
||
|
|
* NTESS, the U.S. Government retains certain rights in this software.
|
||
|
|
*
|
||
|
|
* See packages/seacas/LICENSE for details
|
||
|
|
*/
|
||
|
|
/*****************************************************************************
|
||
|
|
*
|
||
|
|
* testwt - test write an ExodusII database file
|
||
|
|
*
|
||
|
|
* author - Sandia National Laboratories
|
||
|
|
* Larry A. Schoof - Original
|
||
|
|
* Vic Yarberry - Added headers and error logging
|
||
|
|
* 7/7/93 Modified for use with Exodus 2.00
|
||
|
|
*
|
||
|
|
*
|
||
|
|
* environment - UNIX
|
||
|
|
*
|
||
|
|
* entry conditions -
|
||
|
|
*
|
||
|
|
* exit conditions -
|
||
|
|
*
|
||
|
|
* revision history -
|
||
|
|
*
|
||
|
|
* This is a test program for the C binding of the EXODUS II
|
||
|
|
* database write routines.
|
||
|
|
*
|
||
|
|
*
|
||
|
|
*****************************************************************************/
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
|
||
|
|
#include "exodusII.h"
|
||
|
|
|
||
|
|
#define STRINGIFY(x) #x
|
||
|
|
#define TOSTRING(x) STRINGIFY(x)
|
||
|
|
|
||
|
|
#define EXCHECK(funcall) \
|
||
|
|
do { \
|
||
|
|
int error = (funcall); \
|
||
|
|
printf("after %s, error = %d\n", TOSTRING(funcall), error); \
|
||
|
|
if (error != EX_NOERR) { \
|
||
|
|
fprintf(stderr, "Error calling %s\n", TOSTRING(funcall)); \
|
||
|
|
ex_close(exoid); \
|
||
|
|
exit(-1); \
|
||
|
|
} \
|
||
|
|
} while (0)
|
||
|
|
|
||
|
|
int main(int argc, char **argv)
|
||
|
|
{
|
||
|
|
ex_opts(EX_VERBOSE);
|
||
|
|
|
||
|
|
/* Specify compute and i/o word size */
|
||
|
|
int CPU_word_size = 8;
|
||
|
|
int IO_word_size = 8;
|
||
|
|
|
||
|
|
/* create EXODUS II file */
|
||
|
|
int exoid = ex_create("test-assembly.exo", /* filename path */
|
||
|
|
EX_CLOBBER, /* create mode */
|
||
|
|
&CPU_word_size, /* CPU double word size in bytes */
|
||
|
|
&IO_word_size); /* I/O double 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);
|
||
|
|
|
||
|
|
int num_elem_blk = 7;
|
||
|
|
int num_assembly = 4;
|
||
|
|
{
|
||
|
|
/* initialize file with parameters */
|
||
|
|
int num_dim = 3;
|
||
|
|
int num_nodes = 1;
|
||
|
|
int num_elem = 7;
|
||
|
|
int num_node_sets = 0;
|
||
|
|
int num_side_sets = 0;
|
||
|
|
ex_init_params par = {.num_dim = num_dim,
|
||
|
|
.num_nodes = num_nodes,
|
||
|
|
.num_elem = num_elem,
|
||
|
|
.num_elem_blk = num_elem_blk,
|
||
|
|
.num_node_sets = num_node_sets,
|
||
|
|
.num_side_sets = num_side_sets,
|
||
|
|
.num_assembly = num_assembly};
|
||
|
|
|
||
|
|
char *title = "This is a test";
|
||
|
|
ex_copy_string(par.title, title, MAX_LINE_LENGTH + 1);
|
||
|
|
EXCHECK(ex_put_init_ext(exoid, &par));
|
||
|
|
}
|
||
|
|
|
||
|
|
double coord[] = {0.0};
|
||
|
|
EXCHECK(ex_put_coord(exoid, coord, coord, coord));
|
||
|
|
|
||
|
|
/* ======================================================================== */
|
||
|
|
/* write element block parameters */
|
||
|
|
struct ex_block blocks[num_elem_blk];
|
||
|
|
for (int i = 0; i < num_elem_blk; i++) {
|
||
|
|
blocks[i] = (ex_block){.type = EX_ELEM_BLOCK, .num_entry = 1, .id = i + 10};
|
||
|
|
}
|
||
|
|
|
||
|
|
ex_copy_string(blocks[0].topology, "sphere", MAX_STR_LENGTH + 1);
|
||
|
|
ex_copy_string(blocks[1].topology, "sphere", MAX_STR_LENGTH + 1);
|
||
|
|
ex_copy_string(blocks[2].topology, "sphere", MAX_STR_LENGTH + 1);
|
||
|
|
ex_copy_string(blocks[3].topology, "sphere", MAX_STR_LENGTH + 1);
|
||
|
|
ex_copy_string(blocks[4].topology, "sphere", MAX_STR_LENGTH + 1);
|
||
|
|
ex_copy_string(blocks[5].topology, "sphere", MAX_STR_LENGTH + 1);
|
||
|
|
ex_copy_string(blocks[6].topology, "sphere", MAX_STR_LENGTH + 1);
|
||
|
|
|
||
|
|
blocks[0].num_nodes_per_entry = 1;
|
||
|
|
blocks[1].num_nodes_per_entry = 1;
|
||
|
|
blocks[2].num_nodes_per_entry = 1;
|
||
|
|
blocks[3].num_nodes_per_entry = 1;
|
||
|
|
blocks[4].num_nodes_per_entry = 1;
|
||
|
|
blocks[5].num_nodes_per_entry = 1;
|
||
|
|
blocks[6].num_nodes_per_entry = 1;
|
||
|
|
|
||
|
|
EXCHECK(ex_put_block_params(exoid, num_elem_blk, blocks));
|
||
|
|
|
||
|
|
int connect[] = {1};
|
||
|
|
for (int i = 0; i < num_elem_blk; i++) {
|
||
|
|
EXCHECK(ex_put_conn(exoid, EX_ELEM_BLOCK, blocks[i].id, connect, NULL, NULL));
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Write element block names */
|
||
|
|
char *block_names[] = {"block_A", "block_B", "block_C", "block_D",
|
||
|
|
"block_E", "block_F", "block_G"};
|
||
|
|
for (int i = 0; i < num_elem_blk; i++) {
|
||
|
|
EXCHECK(ex_put_name(exoid, EX_ELEM_BLOCK, blocks[i].id, block_names[i]));
|
||
|
|
}
|
||
|
|
|
||
|
|
/* ======================================================================== */
|
||
|
|
/* Define and Output Assemblies */
|
||
|
|
int64_t list_100[] = {200, 300, 400};
|
||
|
|
int64_t list_200[] = {10, 11, 12, 13};
|
||
|
|
int64_t list_300[] = {14, 15, 16};
|
||
|
|
int64_t list_400[] = {10, 16};
|
||
|
|
|
||
|
|
/* Define assemblies. Note that first and last do not have an entity list at time of definition.
|
||
|
|
*/
|
||
|
|
ex_assembly assembly[] = {{100, "Root", EX_ASSEMBLY, 3, NULL},
|
||
|
|
{200, "Child2", EX_ELEM_BLOCK, 4, list_200},
|
||
|
|
{300, "Child3", EX_ELEM_BLOCK, 3, list_300},
|
||
|
|
{400, "Child4", EX_ELEM_BLOCK, 2, NULL}};
|
||
|
|
|
||
|
|
for (int i = 0; i < num_assembly; i++) {
|
||
|
|
EXCHECK(ex_put_assembly(exoid, assembly[i]));
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Now verify that can put entity list at a later time... */
|
||
|
|
assembly[0].entity_list = list_100;
|
||
|
|
EXCHECK(ex_put_assembly(exoid, assembly[0]));
|
||
|
|
|
||
|
|
assembly[3].entity_list = list_400;
|
||
|
|
EXCHECK(ex_put_assembly(exoid, assembly[3]));
|
||
|
|
|
||
|
|
/* ======================================================================== */
|
||
|
|
/* Add some arbitrary attributes to the assemblies */
|
||
|
|
{
|
||
|
|
double scale = 1.5;
|
||
|
|
double offset[] = {1.1, 2.2, 3.3};
|
||
|
|
char *dimension = "length";
|
||
|
|
int units[] = {1, 0, 0, -1};
|
||
|
|
|
||
|
|
EXCHECK(ex_put_double_attribute(exoid, EX_ASSEMBLY, 100, "Scale", 1, &scale));
|
||
|
|
EXCHECK(ex_put_double_attribute(exoid, EX_ASSEMBLY, 200, "Offset", 3, offset));
|
||
|
|
EXCHECK(ex_put_text_attribute(exoid, EX_ASSEMBLY, 300, "Dimension", dimension));
|
||
|
|
EXCHECK(ex_put_integer_attribute(exoid, EX_ASSEMBLY, 400, "Units", 4, units));
|
||
|
|
|
||
|
|
ex_attribute attribute = {.entity_type = EX_ASSEMBLY,
|
||
|
|
.entity_id = 100,
|
||
|
|
.name = {"Units"},
|
||
|
|
.type = EX_INTEGER,
|
||
|
|
.value_count = 4,
|
||
|
|
.values = units};
|
||
|
|
EXCHECK(ex_put_attribute(exoid, attribute));
|
||
|
|
|
||
|
|
ex_attribute attr_offset = {EX_ASSEMBLY, 300, "Offset", EX_DOUBLE, 3, offset};
|
||
|
|
EXCHECK(ex_put_attribute(exoid, attr_offset));
|
||
|
|
|
||
|
|
/* Make sure this works for non-assemblies also... */
|
||
|
|
EXCHECK(ex_put_integer_attribute(exoid, EX_ELEM_BLOCK, 11, "Units", 4, units));
|
||
|
|
EXCHECK(ex_put_text_attribute(exoid, EX_GLOBAL, 0, "SOLID_MODEL", "STEP-X-43-1547836-Rev 0"));
|
||
|
|
}
|
||
|
|
|
||
|
|
/* ======================================================================== */
|
||
|
|
/* Transient Variables */
|
||
|
|
int num_assem_vars = 4;
|
||
|
|
EXCHECK(ex_put_reduction_variable_param(exoid, EX_ASSEMBLY, num_assem_vars));
|
||
|
|
|
||
|
|
{
|
||
|
|
char *var_names[] = {"Momentum_X", "Momentum_Y", "Momentum_Z", "Kinetic_Energy"};
|
||
|
|
EXCHECK(ex_put_reduction_variable_names(exoid, EX_ASSEMBLY, num_assem_vars, var_names));
|
||
|
|
}
|
||
|
|
|
||
|
|
{ /* Output time steps ... */
|
||
|
|
double *var_vals = (double *)calloc(num_assem_vars, CPU_word_size);
|
||
|
|
for (int ts = 0; ts < 10; ts++) {
|
||
|
|
double time_val = (double)(ts + 1) / 100.0;
|
||
|
|
|
||
|
|
EXCHECK(ex_put_time(exoid, ts + 1, &time_val));
|
||
|
|
|
||
|
|
/* write assembly variables */
|
||
|
|
for (int k = 0; k < num_assembly; k++) {
|
||
|
|
for (int var_idx = 0; var_idx < num_assem_vars; var_idx++) {
|
||
|
|
var_vals[var_idx] = (double)(var_idx + 2) * time_val + k;
|
||
|
|
}
|
||
|
|
EXCHECK(ex_put_reduction_vars(exoid, ts + 1, EX_ASSEMBLY, assembly[k].id, num_assem_vars,
|
||
|
|
var_vals));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
free(var_vals);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* close the EXODUS files
|
||
|
|
*/
|
||
|
|
EXCHECK(ex_close(exoid));
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*! Reduction Variables:
|
||
|
|
* - Transient values applied to the object instead of each member of the object.
|
||
|
|
* - for example, a value for the element block; not for each element in the block.
|
||
|
|
* - Each object type has the same variables
|
||
|
|
* - All values for a specific object are output at same time -- Vals for "assembly 100"
|
||
|
|
* - GLOBAL values can be considered reduction variables
|
||
|
|
* - for BW compat, can output either with:
|
||
|
|
* - ex_put_var(exoid, ts, EX_GLOBAL, ?, ?, #gvar, gvar), or
|
||
|
|
* - ex_put_reduction_var(exoid, ts, EX_GLOBAL, ?, #gvar, gvar)
|
||
|
|
* -
|
||
|
|
* - ex_put_reduc_var(exoid, ts, EX_ASSEMBLY, id, #var, var)
|
||
|
|
* - ex_get_reduc_var(exoid, ts, EX_ASSEMBLY, id, #var, &var)
|
||
|
|
*
|
||
|
|
* - ex_put_reduc_variable_names()
|
||
|
|
* - ex_put_reduc_variable_param()
|
||
|
|
|
||
|
|
*/
|
||
|
|
/* int ex_put_reduction_var(exoid, time_step, obj_type, obj_id, #values, values); */
|