从 C 中的 HDF5 文件确定固定长度字符串属性的长度

Determine length of fixed-length string attribute from HDF5 file in C

我有一个包含以下内容的 HDF5 文件:

$ h5dump foo.h5 
HDF5 "foo.h5" {
GROUP "/" {
   ATTRIBUTE "Version" {
      DATATYPE  H5T_STRING {
         STRSIZE 5;
         STRPAD H5T_STR_NULLTERM;
         CSET H5T_CSET_ASCII;
         CTYPE H5T_C_S1;
      }
      DATASPACE  SIMPLE { ( 1 ) / ( 1 ) }
      DATA {
      (0): "1.2.0"
      }
   }
...

如何从属性中提取 STRSIZE(在本例中为 5)?我目前可以通过分配比我认为需要的更多的内存来提取整个字符串:

char version_string[] = "hello_hello_hello";
hid_t attr = H5Aopen(file_id, "Version", H5P_DEFAULT);
hid_t atype = H5Tcopy(H5T_C_S1);
H5Tset_size(atype, 17);
herr_t err = H5Aread(attr, atype, version_string);
H5Aclose(attr);
H5Tclose(atype);

但是,我不知道如何为字符串分配 正确 的内存。

(免责声明:我不是 HDF5 用户)

通过查看文档,我认为您需要使用 H5Aget_storage_size 来确定要与 H5Aread 一起使用的缓冲区的大小。

    hid_t versionAttr = H5Aopen( file_id, "Version", H5P_DEFAULT );
    if( versionAttr < 0 ) goto cleanup1;

    size_t versionAttrSize = 0;
    {
        hsize_t size = H5Aget_storage_size( versionAttr );
        if( size == 0 ) {
            // get_storage_size failed for some reason but the documentation isn't helpful. Grrrr.
            // TODO: Show an error message here, I guess.
            goto cleanup2;
        }
        else if( size > SIZE_MAX ) {
            // The data on-disk is larger than can be represented in C on your platform.
            // TODO: Show an error message here, I guess.
            goto cleanup2;
        }
        else {
            versionAttrSize = (size_t)size;
        }
    }

    char* versionAttrBuffer = malloc( versionAttrSize );
    if( !versionAttrBuffer ) {
        // Computer broken.
        goto cleanup2;
    }

    herr_t versionAttrBufferReadError = H5Aread( versionAttr, H5T_C_S1, versionAttrBuffer );
    if( versionAttrBufferReadError < 0 ) {
        goto cleanup3;
    }

    // Do stuff with `versionAttrBuffer` here.
    printf( "String attribute length: %d\n", versionAttrSize );
    printf( "String attribute data: \"%s\"", versionAttrBuffer );

cleanup3:
    free( versionAttrBuffer );

cleanup2:
    herr_t cleanupErr = H5Aclose( versionAttr );
    if( cleanupErr < 0 ) { die( "This should never happen." ); };

cleanup1:
    // TODO: Close file?

另一种使在 C 中处理 HDF5 文件更容易的方法是使用 HDFql。要分配适量的内存来存储固定长度字符串的属性值,可以按如下方式完成(在 C 中使用 HDFql):

long long size;

char *data;

hdfql_variable_transient_register(&size);

hdfql_execute("SHOW SIZE foo.h5 \"Version\" INTO MEMORY 0");

printf("String size: %lld\n", size);

data = malloc(size);

hdfql_variable_transient_register(&data);

hdfql_execute("SELECT FROM foo.h5 \"Version\" INTO MEMORY 0");

printf("String data: %s\n", data);