在c ++中将hdf5文件读取到动态数组

时间:2013-06-14 14:07:28

标签: c++ hdf5

由于堆栈的大小限制,我试图将大型3D hdf5文件读入动态数组。我已经尝试了几种不同的方法,并通过分段错误失败了。下面是显示我的问题的示例代码。我非常感谢一些帮助!

//This example was based on several examples which came in the c++ examples directory of the hdf5 package.


#ifdef OLD_HEADER_FILENAME
#include <iostream.h>
#else
#include <iostream>
#endif
#include <string>
#include <new>
#include "hdf5.h"

#include "H5Cpp.h"

#ifndef H5_NO_NAMESPACE
    using namespace H5;
#endif

const H5std_string  FILE_NAME( "Test.h5" );
const H5std_string  DATASET_NAME( "Identity" );
const int   NX = 3;                    // dataset dimensions
const int   NY = 3;
const int   RANK = 2;

using namespace std;

int main (void)
{

    int buffer[3][3];


//  Make Identity Matrix for fill.
    for(int i=0; i<NX; i++){
        for(int j=0; j<NY; j++){
            if(i==j) buffer[i][j] = 1;
            else buffer[i][j]=0;
        }
    }
cout << "Lets check we filled it correctly: " << endl;
//  First let's check we do have I matrix

    for(int i=0; i<NX; i++){
        for(int j=0;j<NY;j++){
            cout << buffer[i][j] << "  ";
        } cout << endl;
    }




//  Now let's write into a file
    H5File file( FILE_NAME, H5F_ACC_TRUNC );
    hsize_t     dimsf[2];
    dimsf[0] = NX;
    dimsf[1] = NY;
    DataSpace dataspace( RANK, dimsf );
    IntType datatype( PredType::NATIVE_INT );
    datatype.setOrder( H5T_ORDER_LE );
    DataSet dataset = file.createDataSet( DATASET_NAME, datatype, dataspace );
    dataset.write( buffer, PredType::NATIVE_INT );



//  Ok great, now let's try opening this array using both static and dynamic(new) type arrays:

    H5File file1( FILE_NAME, H5F_ACC_RDONLY );
    DataSet dataset1 = file1.openDataSet( DATASET_NAME );

    /*
     * Get filespace for rank and dimension
     */
    DataSpace filespace = dataset1.getSpace();

    /*
     * Get number of dimensions in the file dataspace
     */
    int rank = filespace.getSimpleExtentNdims();

    /*
     * Get and print the dimension sizes of the file dataspace
     */
    hsize_t dims[2];    // dataset dimensions
    rank = filespace.getSimpleExtentDims( dims );
    /*
     * Define the memory space to read dataset.
     */
    DataSpace mspace1(RANK, dims);


    int newbuffer[NX][NY]; //static array.
    int **p2DArray;

    p2DArray = new int*[NX];
    for (uint i = 0; i < NX; ++i) {
        p2DArray[i] = new int[NY];
    }

    dataset1.read( newbuffer, PredType::NATIVE_INT, mspace1, filespace );
    dataset1.read( p2DArray, PredType::NATIVE_INT, mspace1, filespace );

//  Lastly lets print the arrays to make sure we get the same thing:
    cout << "Ok and now for the static array:" << endl;

    for(uint i=0; i<NX; i++){
        for(uint j=0;j<NY;j++){
            cout << newbuffer[i][j] << "  ";
        } cout << endl;
    }
    cout << "Ok and now for the dynamic array:" << endl;

    for(uint i=0; i<NX; i++){
        for(uint j=0;j<NY;j++){
            cout << p2DArray[i][j] << "  ";
        } cout << endl;
    }

return 0;
}

1 个答案:

答案 0 :(得分:5)

您的p2DArray必须是int的普通(1D)数组,而不是数组数组。这就是read方法所期望的。

只需分配足够的空间来存储NX * NY号码:

int *p2DArray = new int[NX*NY];

然后使用以下公式访问数组:p2DArray [i] [j] = p2DArray [i * NY + j]。