HDF5简易使用

最新推荐文章于 2024-07-21 20:37:57 发布

MagiChos

最新推荐文章于 2024-07-21 20:37:57 发布

阅读量5.3k

点赞数

分类专栏： HDF5 文章标签：数据

本文链接：https://blog.csdn.net/hanshuobest/article/details/52269850

版权

HDF5 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

HDF5是用于存储科学数据的文件格式，其最大的优点是能够容纳下不同类型的数据。
我么首先创建一个HDF5格式的数据，用来保存点云的三维坐标，其代码如下：
int writeHDF5data(std::vector &cloud , std::string &filename)
{
if (cloud.empty())
{
std::cout<<”欲保存点云数据为空！\n” ;
return 0 ;
}
int num_of_cloud = cloud.size() ;
float data = new float[num_of_cloud 3] ;

for (int i = 0 ; i < num_of_cloud ; ++i)
{
    data[3 * i + 0] = cloud[i].get_x() ;
    data[3 * i + 1] = cloud[i].get_y() ;
    data[3 * i + 2] = cloud[i].get_z() ;
}

H5File file(filename.c_str() , H5F_ACC_TRUNC) ;
hsize_t dimsf[2] ;//定义维度
dimsf[0] = num_of_cloud ;
dimsf[1] = 3 ;
DataSpace dataspace(2 , dimsf) ;//分配数据空间

IntType datatype(PredType::NATIVE_FLOAT) ;//设置的数据类型，可以根据需要设定成整型，浮点型，以及double型等
datatype.setOrder(H5T_ORDER_LE) ;

DataSet dataset = file.createDataSet(DATASET_NAME , datatype , dataspace) ;//创建数据集
dataset.write(data , PredType::NATIVE_FLOAT) ;
return 1 ;

PointXYZ为我自定义的点结构体，用于保存点的X，Y，Z坐标值。创建完hdf5格式的数据后，因为是二进制无法查看，我么可以采用matlab的h5read()函数进行读取查看。
创建完hdf5格式的数据之后，如果我们需要读取它，其代码如下：

int readHDF5data(std::string &filename , std::vector<PointXYZ> &cloud)
{
    H5File file(filename.c_str() , H5F_ACC_RDONLY) ;
    DataSet dataset = file.openDataSet(DATASET_NAME) ;//数据集名字一定要与创建数据集的名字相同

    H5T_class_t type_class = dataset.getTypeClass() ;
    DataSpace dataspace = dataset.getSpace() ; //获取数据集的数据空间
    int rank = dataspace.getSimpleExtentNdims() ;//获取数据空间的维度

    hsize_t dims_out[2] ;
    int ndims = dataspace.getSimpleExtentDims(dims_out , NULL) ;//检索数据空间的维度和最大尺寸
#ifdef _DEBUG
    std::cout<<"rank"<<rank<<", dimensions "<<dims_out[0]<<" x "<<dims_out[1]<<std::endl ;
#endif // _DEBUG
    if (type_class == H5T_INTEGER)
    {
        int *data = new int[dims_out[0] * dims_out[1]] ;
        dataset.read(data , PredType::NATIVE_INT) ;
    }
    else if (type_class = H5T_FLOAT)
    {
        float *data = new float[dims_out[0] * dims_out[1]] ;
        dataset.read(data , PredType::NATIVE_FLOAT) ;
    }

    return 1 ;
}