aboutsummaryrefslogtreecommitdiff
path: root/src/util/hdf5_convert_from_sdf.c
blob: ef0ca79906c50d8cb6b8c987c3c1fe04f93535e0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
 /*@@
   @file      hdf5_convert_from_sdf.c
   @date      Tue 24 August 2004
   @author    Thomas Radke
   @desc
              Utility program to convert SDF datafiles into IOHDF5
              datafiles.
   @enddesc
   @version   $Id$
 @@*/

#define  MAXDIM       3
#define  MAXNAMESIZE  100

#include <stdio.h>
#include <stdlib.h>


/* Cactus includes (defines CCTK_FILEVERSION) */
#include "cctk.h"

/* SDF includes */
#include <bbhutil.h>
#include <sdf_priv.h>

/* HDF5 include */
#include <hdf5.h>

#define GLOBAL_ATTRIBUTES_GROUP "Global Attributes"

/* the rcs ID and its dummy function to use it */
static const char *rcsid = "$Header$";
CCTK_FILEVERSION(CactusPUGHIO_IOHDF5_util_hdf5_convert_from_sdf_c)


/*****************************************************************************/
/*                           macro definitions                               */
/*****************************************************************************/
/* uncomment the following to get some debugging output */
/* #define IOHDF5_DEBUG  1 */

/* macro to do an HDF5 call, check its return code, and print a warning
   in case of an error */
#define CHECK_ERROR(hdf5_call)                                                \
          do                                                                  \
          {                                                                   \
            int _error_code = hdf5_call;                                      \
                                                                              \
                                                                              \
            if (_error_code < 0)                                              \
            {                                                                 \
              fprintf (stderr, "WARNING: line %d: HDF5 call '%s' returned "   \
                               "error code %d\n",                             \
                                __LINE__, #hdf5_call, _error_code);           \
              exit (-1);                                                      \
            }                                                                 \
          } while (0)


 /*@@
   @routine    main
   @date       Tue 24 August 2004
   @author     Thomas Radke
   @desc
               Main routine of the SDF-to-HDF5 converter
   @enddesc

   @var        argc
   @vdesc      number of command line arguments
   @vtype      int
   @vio        in
   @endvar
   @var        argv
   @vdesc      command line arguments
   @vtype      char *[]
   @vio        in
   @endvar

   @returntype int
   @returndesc
               0 for success, negative return values indicate an error
   @endreturndesc
@@*/
int main (int argc, char **argv)
{
  int i, j;
  int version, rank, dsize, csize, grouptype;
  int *dims;
  double timestep;
  char *varname, *groupname, *coordname, *tag;
  gft_sdf_file_data *infile;
  double *coords, *data, *bbox;
  hid_t outfile, dataset, dataspace, attr, attrDataspace;
  hid_t group, hdf5String;
  hsize_t hdf5Dims[MAXDIM];
  int iteration, timelevel, ntimelevels;
  char hdf5DatasetName[2 * MAXNAMESIZE];


  if (argc <= 2)
  {
    fprintf (stderr, "Usage: %s <inputfile> <outputfile>\n", argv[0]);
    fprintf (stderr, "   eg. %s foo.sdf bar.h5\n", argv[0]);
    return (0);
  }

  infile = gft_open_sdf_file (argv[1]);
  if (! infile)
  {
    printf ("Could not open SDF input file '%s'\n", argv[1]);
    return (-1);
  }

  outfile = H5Fcreate (argv[2], H5F_ACC_TRUNC, H5P_DEFAULT, H5P_DEFAULT);
  if (outfile < 0)
  {
    fprintf (stderr, "Could not create HDF5 output file '%s'\n", argv[2]);
    return (-1);
  }

  printf ("\n  ----------------------------\n"
            "  Cactus SDF-to-HDF5 Converter\n"
            "  ----------------------------\n");

  /* add a dummy GLOBAL_ATTRIBUTES_GROUP so that the HDF5 file is recognized as
     unchunked Cactus data */
  CHECK_ERROR (group = H5Gcreate (outfile, GLOBAL_ATTRIBUTES_GROUP, 0));
  CHECK_ERROR (attrDataspace = H5Screate (H5S_SCALAR));

  CHECK_ERROR (attr = H5Acreate (group, "nprocs", H5T_NATIVE_INT,
                                      attrDataspace, H5P_DEFAULT));
  i = 1;
  CHECK_ERROR (H5Awrite (attr, H5T_NATIVE_INT, &i));
  CHECK_ERROR (H5Aclose (attr));

  CHECK_ERROR (attr = H5Acreate (group, "ioproc_every", H5T_NATIVE_INT,
                                      attrDataspace, H5P_DEFAULT));
  i = 1;
  CHECK_ERROR (H5Awrite (attr, H5T_NATIVE_INT, &i));
  CHECK_ERROR (H5Aclose (attr));

  CHECK_ERROR (attr = H5Acreate (group, "unchunked", H5T_NATIVE_INT,
                                      attrDataspace, H5P_DEFAULT));
  i = 1;
  CHECK_ERROR (H5Awrite (attr, H5T_NATIVE_INT, &i));
  CHECK_ERROR (H5Aclose (attr));

  CHECK_ERROR (H5Gclose (group));

  CHECK_ERROR (hdf5String = H5Tcopy (H5T_C_S1));

  /*** FIXME: get iteration number and timelevel information from user ***/
  iteration = timelevel = 0;
  ntimelevels = 1;
  grouptype = CCTK_GF;
  groupname = "grid::coordinates";

  while (low_read_sdf_stream (1, infile->fp, &timestep, &version, &rank, &dsize,
                              &csize, &varname, &coordname, &tag, &dims, &bbox,
                              &coords, &data))
  {
    /*** FIXME: give the dataset a valid variable name ***/
    varname = strdup ("grid::x");

    printf ("Processing dataset '%s' (timestep %f)\n", varname, timestep);

    /* convert from int to hsize_t */
    for (j = 0; j < rank; j++)
    {
      hdf5Dims[j] = dims[j];
    }
    CHECK_ERROR (dataspace = H5Screate_simple (rank, hdf5Dims, NULL));

    sprintf (hdf5DatasetName, "/%s timelevel %d at iteration %d",
             varname, timelevel, iteration);
    CHECK_ERROR (dataset = H5Dcreate (outfile, hdf5DatasetName,
                                      H5T_NATIVE_DOUBLE,dataspace,H5P_DEFAULT));
    CHECK_ERROR (H5Dwrite (dataset, H5T_NATIVE_DOUBLE, H5S_ALL, H5S_ALL,
                           H5P_DEFAULT, data));

    /* attach necessary attributes */
    CHECK_ERROR (H5Tset_size (hdf5String, strlen (groupname)));
    CHECK_ERROR (attr = H5Acreate (dataset, "groupname", hdf5String,
                                   attrDataspace, H5P_DEFAULT));
    CHECK_ERROR (H5Awrite (attr, hdf5String, groupname));
    CHECK_ERROR (H5Aclose (attr));

    CHECK_ERROR (attr = H5Acreate (dataset, "grouptype", H5T_NATIVE_INT,
                                   attrDataspace, H5P_DEFAULT));
    CHECK_ERROR (H5Awrite (attr, H5T_NATIVE_INT, &grouptype));
    CHECK_ERROR (H5Aclose (attr));

    CHECK_ERROR (attr = H5Acreate (dataset, "ntimelevels", H5T_NATIVE_INT,
                                   attrDataspace, H5P_DEFAULT));
    CHECK_ERROR (H5Awrite (attr, H5T_NATIVE_INT, &ntimelevels));
    CHECK_ERROR (H5Aclose (attr));

    CHECK_ERROR (H5Dclose (dataset));

    CHECK_ERROR (H5Sclose (dataspace));

    free (varname);
    free (coordname);
    free (tag);
    free (dims);
    free (bbox);
    free (coords);
    free (data);

    /*** FIXME: increment iteration number to distinguish datasets ***/
    iteration++;
  }

  CHECK_ERROR (H5Sclose (attrDataspace));
  CHECK_ERROR (H5Tclose (hdf5String));
  CHECK_ERROR (H5Fclose (outfile));
  gsfd_close (infile);

  return (0);
}