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
|
#include <cassert>
#include <cstdlib>
// force HDF5 1.8.x installations to use the new API
#define H5Dcreate_vers 2
#include <hdf5.h>
#include "cctk.h"
#include "cctk_Parameters.h"
#include "vect.hh"
#include "tensor_component.hh"
namespace CarpetIOF5 {
namespace F5 {
tensor_component_t::
tensor_component_t (data_region_t & data_region,
int const variable)
: m_data_region (data_region),
m_variable (variable)
{
DECLARE_CCTK_PARAMETERS;
assert (variable >= 0 and variable < CCTK_NumVars());
char const * const name = CCTK_VarName (variable);
assert (name != 0);
m_name = string (name);
int const vartype = CCTK_VarTypeI (variable);
assert (vartype >= 0);
hid_t const hdf5_datatype = hdf5_datatype_from_cactus_datatype (vartype);
assert (hdf5_datatype >= 0);
bbox<int, dim> const & region = m_data_region.get_region();
vect<hsize_t, dim> const dims
= (region.shape() / region.stride()).reverse();
m_dataspace = H5Screate_simple (dim, & dims [0], 0);
assert (m_dataspace >= 0);
m_properties = H5Pcreate (H5P_DATASET_CREATE);
assert (m_properties >= 0);
vect<int, dim> const user_chunk_size
(chunk_size_x, chunk_size_y, chunk_size_z);
bool const need_chunks
= any (user_chunk_size > 0) or compression_level > 0 or write_checksum;
if (need_chunks)
{
vect<hsize_t, dim> const chunk_size
= either (user_chunk_size > 0,
vect<hsize_t, dim> (user_chunk_size),
dims);
herr_t const herr = H5Pset_chunk (m_properties, dim, & chunk_size [0]);
assert (not herr);
}
if (compression_level > 0)
{
herr_t const herr = H5Pset_deflate (m_properties, compression_level);
assert (not herr);
}
if (write_checksum)
{
herr_t const herr = H5Pset_fletcher32 (m_properties);
assert (not herr);
}
m_dataset
= H5Dcreate (m_data_region.get_hdf5_data_region(), m_name.c_str(),
hdf5_datatype, m_dataspace,
H5P_DEFAULT, m_properties, H5P_DEFAULT);
assert (m_dataset >= 0);
write_or_check_attribute
(m_dataset, "iorigin", region.lower() / region.stride());
assert (invariant());
}
tensor_component_t::
~ tensor_component_t ()
{
herr_t herr;
herr = H5Dclose (m_dataset);
assert (not herr);
herr = H5Sclose (m_dataspace);
assert (not herr);
herr = H5Pclose (m_properties);
assert (not herr);
}
data_region_t & tensor_component_t::
get_data_region ()
const
{
return m_data_region;
}
hid_t tensor_component_t::
get_variable ()
const
{
return m_variable;
}
// TODO: This assumes that the shape of data is the same as the
// shape of the region; this may not be so if not all of the data
// are written out
void tensor_component_t::
write (void const * const data,
int const cactus_datatype)
const
{
hid_t const memory_hdf5_datatype
= hdf5_datatype_from_cactus_datatype (cactus_datatype);
assert (memory_hdf5_datatype >= 0);
bbox<int, dim> const & region = m_data_region.get_region();
vect<hsize_t, dim> const dims
= (region.shape() / region.stride()).reverse();
hid_t const memory_dataspace
= H5Screate_simple (dim, & dims [0], & dims [0]);
assert (memory_dataspace >= 0);
hid_t const transfer_properties = H5Pcreate (H5P_DATASET_XFER);
assert (transfer_properties >= 0);
herr_t herr;
herr
= H5Dwrite (m_dataset, memory_hdf5_datatype, memory_dataspace,
m_dataspace, transfer_properties, data);
assert (not herr);
herr = H5Pclose (transfer_properties);
assert (not herr);
herr = H5Sclose (memory_dataspace);
assert (not herr);
}
bool tensor_component_t::
invariant ()
const
{
return (m_variable >= 0 and m_variable < CCTK_NumVars()
and m_properties >= 0
and m_dataset >= 0
and m_dataspace >= 0);
}
} // namespace F5
} // namespace CarpetIOF5
|
