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
|
#ifndef __H5VOL_DATASET_FUNC
#define __H5VOL_DATASET_FUNC
#include "H5Vol_def.h"
void *H5VL_adios2_dataset_create(void *obj, const H5VL_loc_params_t *loc_params, const char *name,
hid_t lcpl_id, hid_t type_id, hid_t space_id, hid_t dcpl_id,
hid_t dapl_id, hid_t dxpl_id, void **req)
{
REQUIRE_NOT_NULL_ERR(loc_params, NULL);
REQUIRE_NOT_NULL_ERR(obj, NULL);
H5VL_ObjDef_t *vol = (H5VL_ObjDef_t *)obj;
if (H5I_FILE == loc_params->obj_type)
{
REQUIRE_SUCC((ROOT == vol->m_ObjType), NULL);
H5VL_FileDef_t *handle = (H5VL_FileDef_t *)(vol->m_ObjPtr);
H5VL_VarDef_t *varDef = gCreateVarDef(name, handle->m_Engine, NULL, space_id, type_id);
gADIOS2DefineVar(vol->m_FileIO, varDef);
return gVarToVolObj(varDef, vol);
}
else if (H5I_GROUP == loc_params->obj_type)
{
REQUIRE_SUCC((GROUP == vol->m_ObjType), NULL);
size_t pathSize = strlen(vol->m_Path) + 1 + strlen(name);
char fullPath[pathSize + 1];
if (vol->m_Path[strlen(vol->m_Path) - 1] == '/')
{
pathSize = pathSize - 1;
snprintf(fullPath, sizeof(fullPath), "%s%s", vol->m_Path, name);
}
else
snprintf(fullPath, sizeof(fullPath), "%s/%s", vol->m_Path, name);
fullPath[pathSize] = '\0';
H5VL_ObjDef_t *curr = vol;
while (curr)
{
if (curr->m_Parent == NULL)
{
H5VL_VarDef_t *varDef =
gCreateVarDef(fullPath, ((H5VL_FileDef_t *)(curr->m_ObjPtr))->m_Engine, NULL,
space_id, type_id);
gADIOS2DefineVar(vol->m_FileIO, varDef);
return gVarToVolObj(varDef, vol);
}
else
{
curr = (H5VL_ObjDef_t *)(curr->m_Parent);
}
}
}
return NULL;
}
void *H5VL_adios2_dataset_open(void *obj, const H5VL_loc_params_t *loc_params, const char *name,
hid_t dapl_id, hid_t dxpl_id, void **req)
{
REQUIRE_NOT_NULL_ERR(loc_params, NULL);
REQUIRE_NOT_NULL_ERR(obj, NULL);
H5VL_ObjDef_t *vol = (H5VL_ObjDef_t *)obj;
if (H5I_FILE == loc_params->obj_type)
{
REQUIRE_SUCC((ROOT == vol->m_ObjType), NULL);
// return gGetVarObjDef(name, vol);
}
if (H5I_GROUP == loc_params->obj_type)
{
REQUIRE_SUCC((GROUP == vol->m_ObjType), NULL);
// return gGetVarObjDef(name, vol);
}
H5VL_ObjDef_t *result = gGetVarObjDef(name, vol);
if (NULL == result)
SHOW_ERROR_MSG("H5VL_ADIOS2: Error: No such variable: %s in file.\n ", name);
return result;
}
herr_t H5VL_adios2_dataset_read(size_t count, void *dset_array[], hid_t mem_type_id_array[],
hid_t mem_space_id_array[], hid_t file_space_id_array[],
hid_t dxpl_id, void *buf_array[],
void **req) // last parameter is unused as in h5
{
herr_t returnValue = 0;
for (size_t i = 0; i < count; i++)
{
REQUIRE_NOT_NULL_ERR(dset_array[i], -1);
H5VL_ObjDef_t *vol = (H5VL_ObjDef_t *)(dset_array[i]);
H5VL_VarDef_t *var = (H5VL_VarDef_t *)(vol->m_ObjPtr);
var->m_HyperSlabID = file_space_id_array[i];
var->m_MemSpaceID = mem_space_id_array[i];
var->m_Data = buf_array[i];
if (gADIOS2ReadVar(var) < 0)
returnValue = -1;
}
return returnValue;
}
herr_t H5VL_adios2_dataset_get(void *dset, H5VL_dataset_get_args_t *args, hid_t dxpl_id, void **req)
{
REQUIRE_NOT_NULL_ERR(dset, -1);
H5VL_ObjDef_t *vol = (H5VL_ObjDef_t *)dset;
H5VL_VarDef_t *varDef = (H5VL_VarDef_t *)(vol->m_ObjPtr);
switch (args->op_type)
{
case H5VL_DATASET_GET_SPACE: {
REQUIRE_SUCC_MSG((varDef->m_ShapeID >= 0), -1, "H5VOL-ADIOS2: Unable to get space id.");
args->args.get_space.space_id = H5Scopy(varDef->m_ShapeID);
break;
}
case H5VL_DATASET_GET_TYPE: {
args->args.get_type.type_id = H5Tcopy(varDef->m_TypeID);
break;
}
default: {
return -1;
}
}
return 0;
}
herr_t H5VL_adios2_dataset_write(size_t count, void *dset_array[], hid_t mem_type_id_array[],
hid_t mem_space_id_array[], hid_t file_space_id_array[],
hid_t dxpl_id, const void *buf_array[], void **req)
{
for (size_t i = 0; i < count; i++)
{
REQUIRE_NOT_NULL_ERR(dset_array[0], -1);
H5VL_ObjDef_t *vol = (H5VL_ObjDef_t *)(dset_array[0]);
H5VL_VarDef_t *varDef = (H5VL_VarDef_t *)(vol->m_ObjPtr);
varDef->m_Data = (void *)(buf_array[i]);
if (file_space_id_array[i] > 0)
varDef->m_HyperSlabID = file_space_id_array[i];
else
varDef->m_HyperSlabID = varDef->m_ShapeID;
if (mem_space_id_array[i] > 0)
varDef->m_MemSpaceID = mem_space_id_array[i];
else
varDef->m_MemSpaceID = varDef->m_ShapeID;
varDef->m_PropertyID = dxpl_id; // plist_id;
gADIOS2CreateVar(vol->m_FileIO, varDef);
}
return 0;
}
herr_t H5VL_adios2_dataset_close(void *dset, hid_t dxpl_id, void **req)
{
if (NULL == dset)
return 0;
H5VL_ObjDef_t *vol = (H5VL_ObjDef_t *)dset;
REQUIRE_SUCC((VAR == vol->m_ObjType), -1);
H5VL_VarDef_t *varDef = (H5VL_VarDef_t *)(vol->m_ObjPtr);
// H5VL_VarDef_t *varDef = (H5VL_VarDef_t *)dset;
SAFE_FREE(varDef->m_Name);
H5Sclose(varDef->m_ShapeID);
SAFE_FREE(varDef);
gFreeVol(vol);
return 0;
}
#endif
|
