proto-HDF5/subsets.cpp at master · Stride-Apollo/proto-HDF5 · GitHub

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
#include <iostream>
#include <string>

#include "H5Cpp.h"

#ifndef H5_NO_NAMESPACE
	using namespace H5;
	#ifndef H5_NO_STD
		using std::cout;
		using std::endl;
	#endif
#endif

const H5std_string FILE_NAME("h5tutr_subset.h5");
const H5std_string DATASET_NAME("IntArray");
const int RANK = 2;
const int DIM0_SUB = 3;
const int DIM1_SUB = 4;
const int DIM0 = 8;
const int DIM1 = 10;

int justaFunction() {
	int data[DIM0][DIM1], sdata[DIM0_SUB][DIM1_SUB], rdata[DIM0][DIM1];
	int i,j;

	try {
		Exception::dontPrint();
		H5File file(FILE_NAME, H5F_ACC_TRUNC);

		hsize_t dims[2];
		dims[0] = DIM0;
		dims[1] = DIM1;
		DataSpace dataspace = DataSpace(RANK, dims);
		DataSet dataset(file.createDataSet(DATASET_NAME, PredType::STD_I32BE, dataspace));

		for(j=0; j<DIM0; j++) {
			for(i=0; i<DIM1; i++) {
				if(i < (DIM1/2)) {
					data[j][i] = 1;
				} else {
					data[j][i] = 2;
				}
			}
		}

		dataset.write(data, PredType::NATIVE_INT);

		cout << endl << "Data Written to file:" << endl;
		for (j=0; j<DIM0; j++) {
			for (i=0; i<DIM1; i++) {
				cout << " " << data[j][i];
			}
			cout << endl;
		}
		dataspace.close();
		dataset.close();
		file.close();

		hsize_t offset[2], count[2], stride[2], block[2];
		hsize_t dimsm[2];

		file.openFile(FILE_NAME, H5F_ACC_RDWR);
		dataset = file.openDataSet(DATASET_NAME);

		offset[0] = 1;
		offset[1] = 2;

		count[0] = DIM0_SUB;
		count[1] = DIM1_SUB;

		stride[0] = 1;
		stride[1] = 1;

		block[0] = 1;
		block[1] = 1;

		dimsm[0] = DIM0_SUB;
		dimsm[1] = DIM1_SUB;

		DataSpace memspace(RANK, dimsm, NULL);
		dataspace = dataset.getSpace();
		dataspace.selectHyperslab(H5S_SELECT_SET, count, offset, stride, block);

		cout << endl << "Write subset to file specifying: " << endl;
		cout << " offset=1x2 stride=1x1 count=3x4 block=1x1" << endl;
		for (j=0; j < DIM0_SUB; j++) {
			for (i=0; i < DIM1_SUB; i++) {
				sdata[j][i] = 5;
			}
		}

		dataset.write(sdata, PredType::NATIVE_INT, memspace, dataspace);
		dataset.read(rdata, PredType::NATIVE_INT);

		cout << endl << "Data in File after Subset is Written:" << endl;
		for (i = 0; i < DIM0; i++) {
			for (j = 0; j < DIM1; j++) {
				cout << " " << rdata[i][j];
			}
			cout << endl;
		}
		cout << endl;

		dataspace.close();
		memspace.close();
		dataset.close();
		file.close();

	} catch(GroupIException error) {
		error.printError();
		return -1;
	} catch(AttributeIException error) {
		error.printError();
		return -1;
	} catch(FileIException error) {
		error.printError();
		return -1;
	} catch(DataSetIException error) {
		error.printError();
		return -1;
	} catch(DataSpaceIException error) {
		error.printError();
		return -1;
	}
	return 0;
}