ndav-numo-narray 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: bdab89f32a1dc5913ae3c0df46237d9dac3e1c5eb49da215409a34e0ca504e7b
+  data.tar.gz: 1a23f2a704bbd19a6c27648ae663b8b32c9b3977f893f48888baf92f7fadb50e
+SHA512:
+  metadata.gz: 49afc07a7c360770f17109277b99a77daa68ba3bfa944e981525214668a777171ae2a9661fcdf1d2412008c4bb5cfb6aaa4694e23482b2f7f22686e41bb82f0f
+  data.tar.gz: eb25644d06ec6f69f639f847897ccfefad78563faf3f1e1106c3db6e094bcea286fa6214004a240de4566f18e83acbb3b379d959bec0cff5f9e1158f9d51c875

data/.gitignore

@@ -0,0 +1,8 @@
+Gemfile.lock
+pkg/
+*.so
+*.bundle
+*.dll
+*.o
+ext/Makefile
+mkmf.log

data/Gemfile

@@ -0,0 +1,3 @@
+source "https://rubygems.org"
+
+gemspec

data/LICENSE.txt

@@ -0,0 +1,32 @@
+BSD 3-Clause License
+
+Copyright (c) 2026 Kitaiti Makoto
+
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright
+   notice, this list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright
+   notice, this list of conditions and the following disclaimer in the
+   documentation and/or other materials provided with the distribution.
+
+3. Neither the names of Facebook, Deepmind Technologies, NYU, NEC Laboratories America
+   and IDIAP Research Institute nor the names of its contributors may be
+   used to endorse or promote products derived from this software without
+   specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+POSSIBILITY OF SUCH DAMAGE.

data/README.md

@@ -0,0 +1,12 @@
+NDAV::Numo::NArray
+==================
+
+MemoryView and pointer support for [Numo::NArray][] ([numo-narray-alt][]).
+
+LICENSE
+-------
+
+3-Clause BSD License. See LICENSE.txt file.
+
+[Numo::NArray]: https://ruby-numo.github.io/narray/
+[numo-narray-alt]: https://github.com/yoshoku/numo-narray-alt

data/Rakefile

@@ -0,0 +1,47 @@
+require "rake/clean"
+require "rake/testtask"
+require "rubygems/tasks"
+
+task default: :test
+
+Gem::Tasks.new
+
+SRC = FileList["ext/*.{h,c}"]
+
+LIB_NAME = "narray.#{RbConfig::CONFIG["DLEXT"]}"
+LIB_BUILD = File.join("ext", LIB_NAME)
+LIB_PATH = File.join("lib", "ndav", "numo", LIB_NAME)
+MAKEFILE = "ext/Makefile"
+
+file LIB_PATH => MAKEFILE do |t|
+  chdir "ext" do
+    sh "make"
+  end
+  copy LIB_BUILD, LIB_PATH
+end
+
+file MAKEFILE => SRC + ["ext/extconf.rb"] do |t|
+  chdir "ext" do
+    ruby "extconf.rb"
+  end
+end
+
+if File.exist? MAKEFILE
+  task :make_clean do
+    chdir "ext" do
+      sh "make", "clean"
+    end
+  end
+  task clean: :make_clean
+
+  task :make_distclean do
+    chdir "ext" do
+      sh "make", "distclean"
+    end
+  end
+  task clobber: :make_distclean
+end
+
+CLEAN.include LIB_PATH
+
+Rake::TestTask.new test: "lib/ndav/numo/narray.#{RbConfig::CONFIG["DLEXT"]}"

data/ext/extconf.rb

@@ -0,0 +1,9 @@
+require "mkmf"
+
+dependency = "numo-narray-alt"
+header = "numo/narray.h"
+header_path = File.join(Gem::Specification.find_by_name(dependency).full_gem_path, "lib/numo", header)
+$INCFLAGS = "-I#{File.dirname(File.dirname(header_path))} #{$INCFLAGS}"
+have_header(header) || abort "#{header} not found in #{dependency}"
+
+create_makefile "ndav/numo/narray"

data/ext/narray.c

@@ -0,0 +1,28 @@
+#ifndef NDAV_NUMO_NARRAY_H
+#define NDAV_NUMO_NARRAY_H 1
+
+#include <ruby.h>
+#include <ruby/memory_view.h>
+#include <numo/narray.h>
+
+static VALUE
+ndav_numo_narray_s_address(VALUE self, VALUE obj)
+{
+  void* ptr = na_get_pointer_for_read(obj);
+
+  return PTR2NUM(ptr);
+}
+
+void
+Init_narray(void)
+{
+  rb_require("numo/narray/alt");
+
+  VALUE cNDAV = rb_define_class("NDAV", rb_cObject);
+  VALUE mNumo = rb_define_module_under(cNDAV, "Numo");
+  VALUE mNArray = rb_define_module_under(mNumo, "NArray");
+
+  rb_define_singleton_method(mNArray, "address", ndav_numo_narray_s_address, 1);
+}
+
+#endif /* NDAV_NUMO_NARRAY_H */

data/lib/ndav/numo/narray.rb

@@ -0,0 +1,67 @@
+require "ndav"
+require "numo/narray/alt"
+require "ndav/numo/narray.so"
+
+class NDAV
+  module Numo
+    module NArray
+      CLASS_TO_FORMAT = {
+        ::Numo::Int8 => "c",
+        ::Numo::UInt8 => "C",
+        ::Numo::Int16 => "s",
+        ::Numo::UInt16 => "S",
+        ::Numo::Int32 => "l",
+        ::Numo::UInt32 => "L",
+        ::Numo::Int64 => "q",
+        ::Numo::UInt64 => "Q",
+        ::Numo::SFloat => "f",
+        ::Numo::DFloat => "d"
+      }
+
+      FORMAT_TO_CLASS = CLASS_TO_FORMAT.invert
+
+      module FromNDAV
+        def from_ndav(ndav)
+          cls = FORMAT_TO_CLASS[ndav.format]
+          raise ArgumentError, "unsupported format: #{ndav.format}, currently supported: #{FORMAT_TO_CLASS.keys}" unless cls
+
+          array = cls.zeros(ndav.shape)
+          address = ::NDAV::Numo::NArray.address(array)
+          ::Fiddle::Pointer.write address, ndav.to_s
+          array.instance_variable_set(:@ndav, ndav) # keep reference to NDAV to prevent GC
+          array
+        end
+      end
+
+      module ToNDAV
+        def to_ndav
+          format = CLASS_TO_FORMAT[self.class]
+          raise ArgumentError, "unsupported class: #{self.class}, currently supported: #{CLASS_TO_FORMAT.keys}" unless format
+
+          ptr = ::Fiddle::Pointer.new(::NDAV::Numo::NArray.address(self), byte_size)
+          NDAV.new(ptr, shape:, format:, lifetime: self)
+        end
+      end
+
+      module Converter
+        def NumoNArray(array, *, **)
+          if array.kind_of? ::Numo::NArray
+            array
+          elsif array.respond_to? :to_numo_narray
+            array.to_numo_narray
+          elsif array.respond_to? :to_ndav
+            array.to_ndav(*, **).to_numo_narray
+          elsif array.respond_to? :to_narray # Red Arrow
+            array.to_narray
+          elsif array.respond_to? :numo # OnnxRuntime::OrtValue, Torch::Tensor
+            array.numo
+          else
+            ::NDAV.new(array, *, **).to_numo_narray
+          end
+        end
+      end
+    end
+  end
+
+  register ::Numo::NArray, Numo::NArray, name: :numo_narray
+end

data/ndav-numo-narray.gemspec

@@ -0,0 +1,23 @@
+Gem::Specification.new do |s|
+  s.name = "ndav-numo-narray"
+  s.version = "0.0.1"
+  s.authors = ["Kitaiti Makoto"]
+  s.summary = "N-Dimensional Array View for Numo::NArray"
+  s.licenses = ["BSD-3-Clause"]
+  s.homepage = "https://gitlab.com/KitaitiMakoto/ndav-numo-narray"
+
+  s.files = Dir.chdir(__dir__) {`git ls-files -z`.split("\x0")}
+  s.extensions << "ext/extconf.rb"
+
+  s.add_runtime_dependency "ndav", ">= 0.0.3"
+  s.add_runtime_dependency "numo-narray-alt"
+
+  s.add_development_dependency "rake"
+  s.add_development_dependency "test-unit"
+  s.add_development_dependency "test-unit-notify"
+  s.add_development_dependency "test-unit-rr"
+  s.add_development_dependency "terminal-notifier" if RUBY_PLATFORM.match?(/darwin/)
+  s.add_development_dependency "rubygems-tasks"
+  s.add_development_dependency "fiddle"
+  s.add_development_dependency "red-arrow"
+end

data/test/helper.rb

@@ -0,0 +1,5 @@
+require "ndav/numo/narray"
+require "test/unit"
+require "test/unit/notify"
+require "arrow"
+require "fiddle"

data/test/test_address.rb

@@ -0,0 +1,28 @@
+require_relative "helper"
+
+class TestAddress < Test::Unit::TestCase
+  def test_address
+    numo = ::Numo::Int16[1, 2, 3]
+
+    assert_raise TypeError do
+      ::NDAV::Numo::NArray.address(Object.new)
+    end
+
+    assert_instance_of Integer, ::NDAV::Numo::NArray.address(numo)
+  end
+
+  def test_pointer
+    numo = ::Numo::Int16[1, 2, 3]
+    address = ::NDAV::Numo::NArray.address(numo)
+
+    ptr = ::Fiddle::Pointer[address]
+    ptr.size = numo.byte_size
+    assert_equal [1, 2, 3].pack("s*"), ptr.to_s(numo.byte_size)
+
+    numo[0] = 4
+    assert_equal [4, 2, 3].pack("s*"), ptr.to_s(numo.byte_size)
+
+    ::Fiddle::Pointer.write(address, [7, 8, 9].pack("s*"))
+    assert_equal ::Numo::Int16[7, 8, 9], numo
+  end
+end

data/test/test_converter.rb

@@ -0,0 +1,49 @@
+require_relative "helper"
+
+class TestConverter < Test::Unit::TestCase
+  def setup
+    @numo_narray = Numo::Int16[1, 2, 3]
+    @ndav = NDAV.new(Arrow::Int16Array.new([1, 2, 3]))
+  end
+
+  def test_ndav_from_numo_narray
+    assert_kind_of ::NDAV, ::NDAV.from_numo_narray(@numo_narray)
+  end
+
+  def test_numo_narray_from_ndav
+    assert_kind_of ::Numo::Int16, ::Numo::Int16.from_ndav(@ndav)
+  end
+
+  def test_ndav_to_numo_narray
+    assert_kind_of ::Numo::Int16, @ndav.to_numo_narray
+  end
+
+  def test_numo_narray_to_ndav
+    assert_kind_of ::NDAV, @numo_narray.to_ndav
+  end
+
+  def test_NDAV
+    assert_kind_of ::NDAV, ::NDAV::Converter::NDAV(@numo_narray)
+  end
+
+  def test_NumoNArray
+    assert_kind_of ::Numo::Int16, ::NDAV::Converter::NumoNArray(@ndav)
+  end
+
+  def test_include_converter
+    c = Class.new {
+      include ::NDAV::Converter
+
+      def to_ndav(array)
+        NDAV(array)
+      end
+
+      def to_numo_narray(ndav)
+        NumoNArray(ndav)
+      end
+    }
+
+    assert_kind_of ::NDAV, c.new.to_ndav(@numo_narray)
+    assert_kind_of ::Numo::Int16, c.new.to_numo_narray(@ndav)
+  end
+end

data/test/test_package.rb

@@ -0,0 +1,26 @@
+require_relative "helper"
+require "tempfile"
+require "tmpdir"
+
+class TestPackage < Test::Unit::TestCase
+  def test_build
+    Tempfile.create do |file|
+      assert system("gem", "build", "ndav-numo-narray.gemspec", "--output", file.to_path, exception: true)
+      assert file.size > 0
+      assert_path_exist file
+    end
+  end
+
+  def test_install
+    gemspec = Gem::Specification.load("ndav-numo-narray.gemspec")
+    Dir.mktmpdir do |dir|
+      dir = File.realpath(dir)
+      gemfile = File.join(dir, gemspec.file_name)
+      system "gem", "build", "ndav-numo-narray.gemspec", "--output", gemfile, exception: true
+      install_dir = File.join(dir, "install")
+      FileUtils.mkdir install_dir
+      system "gem", "install", "--install-dir", install_dir, "--no-document", gemfile, exception: true
+      assert_path_exist File.join(install_dir, "extensions", Gem::Platform.local.to_s, RbConfig::CONFIG['ruby_version'], gemspec.full_name, "ndav/numo/narray.#{RbConfig::CONFIG["DLEXT"]}")
+    end
+  end
+end

metadata

@@ -0,0 +1,191 @@
+--- !ruby/object:Gem::Specification
+name: ndav-numo-narray
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Kitaiti Makoto
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ndav
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.0.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.0.3
+- !ruby/object:Gem::Dependency
+  name: numo-narray-alt
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: test-unit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: test-unit-notify
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: test-unit-rr
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: terminal-notifier
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubygems-tasks
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: fiddle
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: red-arrow
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+executables: []
+extensions:
+- ext/extconf.rb
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- ext/extconf.rb
+- ext/narray.c
+- lib/ndav/numo/narray.rb
+- ndav-numo-narray.gemspec
+- test/helper.rb
+- test/test_address.rb
+- test/test_converter.rb
+- test/test_package.rb
+homepage: https://gitlab.com/KitaitiMakoto/ndav-numo-narray
+licenses:
+- BSD-3-Clause
+metadata: {}
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 4.0.6
+specification_version: 4
+summary: N-Dimensional Array View for Numo::NArray
+test_files: []