Add High level classes

Author: kojix2

lib/hdf5.rb

@@ -33,3 +33,7 @@ def search_hdf5lib(name = nil)
 end
 
 require_relative 'hdf5/ffi3'
+
+require_relative 'hdf5/file'
+require_relative 'hdf5/group'
+require_relative 'hdf5/dataset'

lib/hdf5/dataset.rb

@@ -0,0 +1,52 @@
+module HDF5
+  class Dataset
+    def initialize(parent_id, name)
+      @dataset_id = HDF5::FFI.H5Dopen1(parent_id, name)
+    end
+
+    def write(data)
+      HDF5::FFI.H5Dwrite(@dataset_id, data)
+    end
+
+    def close
+      HDF5::FFI.H5Dclose(@dataset_id)
+    end
+
+    def dtype
+      datatype_id = HDF5::FFI.H5Dget_type(@dataset_id)
+      HDF5::FFI.H5Tget_class(datatype_id)
+    end
+
+    def shape
+      dataspace_id = HDF5::FFI.H5Dget_space(@dataset_id)
+      raise 'Failed to get dataspace' if dataspace_id < 0
+
+      ndims = HDF5::FFI.H5Sget_simple_extent_ndims(dataspace_id)
+      raise 'Failed to get number of dimensions' if ndims < 0
+
+      dims = ::FFI::MemoryPointer.new(:ulong_long, ndims)
+      HDF5::FFI.H5Sget_simple_extent_dims(dataspace_id, dims, nil)
+
+      dims.read_array_of_uint64(ndims)
+    ensure
+      HDF5::FFI.H5Sclose(dataspace_id) if dataspace_id && dataspace_id >= 0
+    end
+
+    # def read
+    #   dtype = dtype()
+    #   shape = shape()
+
+    #   total_elements = shape.inject(:*)
+    #   case dtype
+    #   when :H5T_INTEGER
+    #     read_integer_data(total_elements)
+    #   when :H5T_FLOAT
+    #     read_float_data(total_elements)
+    #   when :H5T_STRING
+    #     read_string_data(total_elements)
+    #   else
+    #     raise 'Unsupported datatype'
+    #   end
+    # end
+  end
+end

lib/hdf5/file.rb

@@ -0,0 +1,68 @@
+module HDF5
+  class File
+    def initialize(filename, mode = 0)
+      @filename = filename
+      @mode = mode
+      @file_id = HDF5::FFI.H5Fopen(filename, mode, 0)
+    end
+
+    def close
+      HDF5::FFI.H5Fclose(@file_id)
+    end
+
+    def create_group(name)
+      Group.new(@file_id, name)
+    end
+
+    def create_dataset(name, data)
+      Dataset.new(@file_id, name, data)
+    end
+
+    def [](name)
+      if HDF5::FFI.group_exists?(@file_id, name)
+        Group.new(@file_id, name)
+      elsif HDF5::FFI.dataset_exists?(@file_id, name)
+        Dataset.new(@file_id, name, nil)
+      else
+        raise 'Group or Dataset not found'
+      end
+    end
+
+    def list_entries
+      list = []
+      callback = ::FFI::Function.new(:int, %i[int64_t string pointer pointer]) do |_, name, _, _|
+        list << name
+        0 # continue
+      end
+
+      status = HDF5::FFI.H5Literate(@file_id, :H5_INDEX_NAME, :H5_ITER_NATIVE, nil, callback, nil)
+      raise 'Failed to iterate over file entries' if status < 0
+
+      list
+    end
+
+    def [](name)
+      if group?(name)
+        Group.new(@file_id, name)
+      elsif dataset?(name)
+        Dataset.new(@file_id, name)
+      else
+        raise 'Unknown object type'
+      end
+    end
+
+    private
+
+    def group?(name)
+      info = HDF5::FFI::H5OInfoT.new
+      HDF5::FFI.H5Oget_info_by_name(@file_id, name, info, 0)
+      info[:type] == :H5O_TYPE_GROUP
+    end
+
+    def dataset?(name)
+      info = HDF5::FFI::H5OInfoT.new
+      HDF5::FFI.H5Oget_info_by_name(@file_id, name, info, 0)
+      info[:type] == :H5O_TYPE_DATASET
+    end
+  end
+end

lib/hdf5/group.rb

@@ -0,0 +1,48 @@
+module HDF5
+  class Group
+    def initialize(file_id, name)
+      @group_id = HDF5::FFI.H5Gopen1(file_id, name)
+      raise "Failed to open group: #{name}" if @group_id < 0
+    end
+
+    def close
+      HDF5::FFI.H5Gclose(@group_id)
+    end
+
+    def list_datasets
+      datasets = []
+      callback = ::FFI::Function.new(:int, %i[int64_t string pointer pointer]) do |_, name, _, _|
+        datasets << name
+        0 # continue
+      end
+      status = HDF5::FFI.H5Literate(@group_id, :H5_INDEX_NAME, :H5_ITER_NATIVE, nil, callback, nil)
+      raise 'Failed to list datasets' if status < 0
+
+      datasets
+    end
+
+    def [](name)
+      if group?(name)
+        Group.new(@group_id, name)
+      elsif dataset?(name)
+        Dataset.new(@group_id, name)
+      else
+        raise 'Group or Dataset not found'
+      end
+    end
+
+    private
+
+    def group?(name)
+      info = HDF5::FFI::H5OInfoT.new
+      HDF5::FFI.H5Oget_info_by_name(@group_id, name, info, 0)
+      info[:type] == :H5O_TYPE_GROUP
+    end
+
+    def dataset?(name)
+      info = HDF5::FFI::H5OInfoT.new
+      HDF5::FFI.H5Oget_info_by_name(@group_id, name, info, 0)
+      info[:type] == :H5O_TYPE_DATASET
+    end
+  end
+end