/*
 * Copyright (c) 2005 Sandia Corporation. Under the terms of Contract
 * DE-AC04-94AL85000 with Sandia Corporation, the U.S. Governement
 * retains certain rights in this software.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 * 
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 * 
 *     * Redistributions in binary form must reproduce the above
 *       copyright notice, this list of conditions and the following
 *       disclaimer in the documentation and/or other materials provided
 *       with the distribution.  
 * 
 *     * Neither the name of Sandia Corporation nor the names of its
 *       contributors may be used to endorse or promote products derived
 *       from this software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 * 
 */
/*****************************************************************************
*
* testrdwt - test reading from one ExodusII file and writing to another
*            ExodusII file open concurrently
*
* author - Sandia National Laboratories
*          Larry A. Schoof - Original
*          
* environment - UNIX
*
* entry conditions - 
*
* exit conditions - 
*
* revision history - 
*
*  This is a test program for the C binding of the EXODUS II 
*  database read and write routines. It tests reading from an open EXODUSII
*  file and writing to another concurrently opened EXODUSII file.
*
*
*****************************************************************************/


#include <stdlib.h>
#include <stdio.h>
#include "exodusII.h"
#include "netcdf.h"
int main (int argc, char **argv)
{
   int exoid, exoid2, num_dim, num_nodes, num_elem, num_elem_blk;
   int num_elem_in_block, num_node_sets, num_nodes_per_elem, num_attr;
   int num_side_sets, error;
   int i, j;
   int *elem_map, *connect, *node_list, *node_ctr_list, *elem_list, *side_list;
   int *ids;
   int num_nodes_in_set, num_elem_in_set;
   int num_sides_in_set, num_df_in_set;
   int num_qa_rec, num_info;
   int CPU_word_size,IO_word_size;
   int num_props, prop_value, *prop_values;

   float *x, *y, *z;
   float *dist_fact;
   float version, fdum;

   char *coord_names[3], *qa_record[2][4], *info[3];
   char title[MAX_LINE_LENGTH+1], elem_type[MAX_STR_LENGTH+1];
   char *prop_names[3];
   char *cdum = 0;

/* Specify compute and i/o word size */

   CPU_word_size = 0;                   /* sizeof(float) */
   IO_word_size = 4;                    /* float */

/* open EXODUS II file for reading */

   ex_opts (EX_VERBOSE | EX_ABORT);

   exoid = ex_open ("test.exo",         /* filename path */
                    EX_READ,            /* access mode */
                    &CPU_word_size,     /* CPU float word size in bytes */
                    &IO_word_size,      /* I/O float word size in bytes */
                    &version);          /* returned version number */
   printf ("after ex_open for test.exo\n");
   printf (" cpu word size: %d io word size: %d\n",CPU_word_size,IO_word_size);

/* create EXODUS II file for writing */

   exoid2= ex_create ("test2.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 test2.exo, exoid = %d\n", exoid2);

   /* ncopts = NC_VERBOSE; */

/* read initialization parameters */

   error = ex_get_init (exoid, title, &num_dim, &num_nodes, &num_elem,
                        &num_elem_blk, &num_node_sets, &num_side_sets);

   printf ("after ex_get_init, error = %d\n", error);

/* write initialization parameters */

   error = ex_put_init (exoid2, title, num_dim, num_nodes, num_elem,
                        num_elem_blk, num_node_sets, num_side_sets);

   printf ("after ex_put_init, error = %d\n", error);

/* read nodal coordinate values */

   x = (float *) calloc(num_nodes, sizeof(float));
   y = (float *) calloc(num_nodes, sizeof(float));
   if (num_dim >= 3)
     z = (float *) calloc(num_nodes, sizeof(float));
   else
     z = 0;
 
   error = ex_get_coord (exoid, x, y, z);
   printf ("\nafter ex_get_coord, error = %3d\n", error);
 
/* write nodal coordinate values */

   error = ex_put_coord (exoid2, x, y, z);
   printf ("after ex_put_coord, error = %d\n", error);

   free (x);
   free (y);
   if (num_dim >= 3)
     free (z);
 
/* read nodal coordinate names */

   for (i=0; i<num_dim; i++)
   {
     coord_names[i] = (char *) calloc ((MAX_STR_LENGTH+1), sizeof(char));
   }
 
   error = ex_get_coord_names (exoid, coord_names);
   printf ("\nafter ex_get_coord_names, error = %3d\n", error);
 
/* write nodal coordinate names */

   error = ex_put_coord_names (exoid2, coord_names);
   printf ("after ex_put_coord_names, error = %d\n", error);

   for (i=0; i<num_dim; i++) {
     free(coord_names[i]);
   }

/* read element order map */

   elem_map = (int *) calloc(num_elem, sizeof(int));
 
   error = ex_get_map (exoid, elem_map);
   printf ("\nafter ex_get_map, error = %3d\n", error);
 
/* write element order map */

   error = ex_put_map (exoid2, elem_map);
   printf ("after ex_put_map, error = %d\n", error);

   free (elem_map);

/* read and write element block parameters and element connectivity */

   ids = (int *) calloc(num_elem_blk, sizeof(int));
   error = ex_get_elem_blk_ids (exoid, ids);
   printf ("\nafter ex_get_elem_blk_ids, error = %3d\n", error);
   
   for (i=0; i<num_elem_blk; i++)
   {
     error = ex_get_elem_block (exoid, ids[i], elem_type,
                                &num_elem_in_block,
                                &num_nodes_per_elem, &num_attr);
     printf ("\nafter ex_get_elem_block, error = %d\n", error);

     error = ex_put_elem_block (exoid2, ids[i], elem_type, num_elem_in_block,
                                num_nodes_per_elem, num_attr);
     printf ("after ex_put_elem_block, error = %d\n", error);
 
     connect = (int *) calloc((num_nodes_per_elem * num_elem_in_block),
                               sizeof(int));
 
     error = ex_get_elem_conn (exoid, ids[i], connect);
     printf ("\nafter ex_get_elem_conn, error = %d\n", error);
 
     error = ex_put_elem_conn (exoid2, ids[i], connect);
     printf ("after ex_put_elem_conn, error = %d\n", error);

     free (connect);
   }

/* read and write element block properties */

   error = ex_inquire (exoid, EX_INQ_EB_PROP, &num_props, &fdum, cdum);
   printf ("\nafter ex_inquire, error = %d\n", error);
 
   for (i=0; i<num_props; i++)
   {
      prop_names[i] = (char *) calloc ((MAX_STR_LENGTH+1), sizeof(char));
   }
 
   error = ex_get_prop_names(exoid,EX_ELEM_BLOCK,prop_names);
   printf ("after ex_get_prop_names, error = %d\n", error);
 
   error = ex_put_prop_names(exoid2,EX_ELEM_BLOCK,num_props,prop_names);
   printf ("after ex_put_prop_names, error = %d\n", error);
 
   for (i=0; i<num_props; i++)
   {
     for (j=0; j<num_elem_blk; j++)
     {
       error = ex_get_prop(exoid, EX_ELEM_BLOCK, ids[j], prop_names[i],
                           &prop_value);
       printf ("after ex_get_prop, error = %d\n", error);

       if (i>0) {   /* first property is the ID which is already stored */
          error = ex_put_prop(exoid2, EX_ELEM_BLOCK, ids[j], prop_names[i], 
                              prop_value);
          printf ("after ex_put_prop, error = %d\n", error);
       }
     }
   }

   for (i=0; i<num_props; i++)
     free(prop_names[i]);
 

   free (ids);

/* read and write individual node sets */

   ids = (int *) calloc(num_node_sets, sizeof(int));
 
   error = ex_get_node_set_ids (exoid, ids);
   printf ("\nafter ex_get_node_set_ids, error = %3d\n", error);
 
   for (i=0; i<num_node_sets; i++)
   {
      error = ex_get_node_set_param (exoid, ids[i],
				     &num_nodes_in_set, &num_df_in_set);
      printf ("\nafter ex_get_node_set_param, error = %3d\n", error);
 
      error = ex_put_node_set_param (exoid2, ids[i], num_nodes_in_set, 
                                     num_df_in_set);
      printf ("after ex_put_node_set_param, error = %d\n", error);

      node_list = (int *) calloc(num_nodes_in_set, sizeof(int));
      dist_fact = (float *) calloc(num_nodes_in_set, sizeof(float));
 
      error = ex_get_node_set (exoid, ids[i], node_list);
      printf ("\nafter ex_get_node_set, error = %3d\n", error);
 
      error = ex_put_node_set (exoid2, ids[i], node_list);
      printf ("after ex_put_node_set, error = %d\n", error);

      if (num_df_in_set > 0)
      {
        error = ex_get_node_set_dist_fact (exoid, ids[i], dist_fact);
        printf ("\nafter ex_get_node_set_dist_fact, error = %3d\n", error);

        error = ex_put_node_set_dist_fact (exoid2, ids[i], dist_fact);
        printf ("after ex_put_node_set, error = %d\n", error);

      }
 
      free (node_list);
      free (dist_fact);
   }
   free(ids);

   /* read node set properties */
   error = ex_inquire (exoid, EX_INQ_NS_PROP, &num_props, &fdum, cdum);
   printf ("\nafter ex_inquire, error = %d\n", error);
 
   for (i=0; i<num_props; i++)
   {
      prop_names[i] = (char *) calloc ((MAX_STR_LENGTH+1), sizeof(char));
   }
   prop_values = (int *) calloc (num_node_sets, sizeof(int));
 
   error = ex_get_prop_names(exoid,EX_NODE_SET,prop_names);
   printf ("after ex_get_prop_names, error = %d\n", error);

   error = ex_put_prop_names(exoid2,EX_NODE_SET,num_props,prop_names);
   printf ("after ex_put_prop_names, error = %d\n", error);
 
   for (i=0; i<num_props; i++)
   {
     error = ex_get_prop_array(exoid, EX_NODE_SET, prop_names[i],
                         prop_values);
     printf ("after ex_get_prop_array, error = %d\n", error);

     error = ex_put_prop_array(exoid2, EX_NODE_SET, prop_names[i], prop_values);
     printf ("after ex_put_prop_array, error = %d\n", error);

   }
   for (i=0; i<num_props; i++)
     free(prop_names[i]);
   free(prop_values);

/* read and write individual side sets */

   ids = (int *) calloc(num_side_sets, sizeof(int));
 
   error = ex_get_side_set_ids (exoid, ids);
   printf ("\nafter ex_get_side_set_ids, error = %3d\n", error);
 
   for (i=0; i<num_side_sets; i++)
   {
      error = ex_get_side_set_param (exoid, ids[i], &num_sides_in_set,
                                     &num_df_in_set);
      printf ("\nafter ex_get_side_set_param, error = %3d\n", error);
 
      error = ex_put_side_set_param (exoid2, ids[i], num_sides_in_set, 
                                     num_df_in_set);
      printf ("after ex_put_side_set_param, error = %d\n", error);

      /* Note: The # of elements is same as # of sides!  */
      num_elem_in_set = num_sides_in_set;
      elem_list = (int *) calloc(num_elem_in_set, sizeof(int));
      side_list = (int *) calloc(num_sides_in_set, sizeof(int));
      node_ctr_list = (int *) calloc(num_elem_in_set, sizeof(int));
      node_list = (int *) calloc(num_elem_in_set*21, sizeof(int));
      dist_fact = (float *) calloc(num_df_in_set, sizeof(float));
 
      error = ex_get_side_set (exoid, ids[i], elem_list, side_list);
      printf ("\nafter ex_get_side_set, error = %3d\n", error);
 
      error = ex_put_side_set (exoid2, ids[i], elem_list, side_list);
      printf ("after ex_put_side_set, error = %d\n", error);

      error = ex_get_side_set_node_list (exoid, ids[i], node_ctr_list,
                                         node_list);
      printf ("\nafter ex_get_side_set_node_list, error = %3d\n", error);
 
      if (num_df_in_set > 0)
      {
        error = ex_get_side_set_dist_fact (exoid, ids[i], dist_fact);
        printf ("\nafter ex_get_side_set_dist_fact, error = %3d\n", error);

        error = ex_put_side_set_dist_fact (exoid2, ids[i], dist_fact);
        printf ("after ex_put_side_set_dist_fact, error = %d\n", error);

      }
 
      free (elem_list);
      free (side_list);
      free (node_ctr_list);
      free (node_list);
      free (dist_fact);
 
   }


   /* read side set properties */
   error = ex_inquire (exoid, EX_INQ_SS_PROP, &num_props, &fdum, cdum);
   printf ("\nafter ex_inquire, error = %d\n", error);
 
   for (i=0; i<num_props; i++)
   {
      prop_names[i] = (char *) calloc ((MAX_STR_LENGTH+1), sizeof(char));
   }
 
   error = ex_get_prop_names(exoid,EX_SIDE_SET,prop_names);
   printf ("after ex_get_prop_names, error = %d\n", error);
 
 
   for (i=0; i<num_props; i++)
   {
     for (j=0; j<num_side_sets; j++)
     {
       error = ex_get_prop(exoid, EX_SIDE_SET, ids[j], prop_names[i],
                           &prop_value);
       printf ("after ex_get_prop, error = %d\n", error);

       if (i>0) {  /* first property is ID so it is already stored */
         error = ex_put_prop(exoid2, EX_SIDE_SET, ids[j], prop_names[i], 
                             prop_value);
         printf ("after ex_put_prop, error = %d\n", error);
       }
     }
   }
   for (i=0; i<num_props; i++)
     free(prop_names[i]);
   free (ids);

/* read and write QA records */

   ex_inquire (exoid, EX_INQ_QA, &num_qa_rec, &fdum, cdum);

   for (i=0; i<num_qa_rec; i++) {
      for (j=0; j<4; j++) {
	qa_record[i][j] = (char *) calloc ((MAX_STR_LENGTH+1), sizeof(char));
      }
   }

   error = ex_get_qa (exoid, qa_record);
   printf ("\nafter ex_get_qa, error = %3d\n", error);

   error = ex_put_qa (exoid2, num_qa_rec, qa_record);
   printf ("after ex_put_qa, error = %d\n", error);

   for (i=0; i<num_qa_rec; i++) {
      for (j=0; j<4; j++) {
	free(qa_record[i][j]);
      }
   }
/* read and write information records */

   error = ex_inquire (exoid, EX_INQ_INFO, &num_info, &fdum, cdum);
   printf ("\nafter ex_inquire, error = %3d\n", error);

   for (i=0; i<num_info; i++)
   {
      info[i] = (char *) calloc ((MAX_LINE_LENGTH+1), sizeof(char));
   }

   error = ex_get_info (exoid, info);
   printf ("\nafter ex_get_info, error = %3d\n", error);

   error = ex_put_info (exoid2, num_info, info);
   printf ("after ex_put_info, error = %d\n", error);

   for (i=0; i<num_info; i++)
   {
     free(info[i]);
   }

/* close the EXODUS files */

   error = ex_close (exoid);
   printf ("after ex_close, error = %d\n", error);
   error = ex_close (exoid2);
   printf ("after ex_close (2), error = %d\n", error);
   return 0;
}