narray_ffi 1.1.1 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 510a401c952751ea163855b174e72a2f871c25d4
-  data.tar.gz: 9e81c0380bc8e31ec91a82c5f72b4df4344b0516
+  metadata.gz: 7999260e1d1b305db5f0c534e2f5f1e6323fc3ba
+  data.tar.gz: 6e439735e3139f0df821b53201c9fe4158d2ecce
 SHA512:
-  metadata.gz: 5ef3e44b7ad2999b1cee2146a63d4f34828d27e588d2952bfdff3120e62d17588e7d11dad16801f6c2de80a2db8f96ce5dcef71d759200bba7b0e8083f512282
-  data.tar.gz: 85840bd05ac8574a205185d5a4f2582161b8135df5870d0e6a7a33dfbb0c1fb7c871a7dbf5c852da9a721951f334b6b27f6a02f4c63a927de192f11e4eaa66c2
+  metadata.gz: bf8a6c49ba47dda532a414e2b14d94dfe16c5c4a4fb311fb14839496a9b4fbe3f02b3e687420a32f8f4abc9c1e55f307b4dd3d5c5acdae195e14e51e11f6b25f
+  data.tar.gz: 7c0974b9fe5933755a955acb059b7da3e3f3a0cfcf21a9019d8146e5abf7fccef624084a70bc60659ff31e818ad4436b0c367f62d4ec9d500797824dc215cb43

data/lib/narray_ffi.rb

@@ -1,3 +1,63 @@
 require "narray"
 require "ffi"
 require "narray_ffi_c.so"
+
+class ANArray < NArray
+
+  FFITYPECODES = {
+    NArray::BYTE => :char,
+    NArray::SINT => :short,
+    NArray::INT => :int,
+    NArray::SFLOAT => :float,
+    NArray::FLOAT => :double,
+    NArray::SCOMPLEX => :float,
+    NArray::COMPLEX => :double
+  }
+
+  def self.new(typecode, alignment, *size)
+    raise "Wrong type code" if not FFITYPECODES[typecode]
+    raise "Invalid alignment" unless alignment > 0 and ( alignment & (alignment - 1) == 0 )
+    total = size[0]
+    size[1..-1].each { |sz| total = total * sz }
+    total = 2*total if typecode == NArray::COMPLEX or typecode == NArray::SCOMPLEX
+    mem = FFI::MemoryPointer::new( FFITYPECODES[typecode], total + alignment - 1 )
+    address = mem.address
+    offset = address & (alignment - 1)
+    offset = alignment - offset unless offset == 0
+    mem = mem.slice(offset, total*mem.type_size)
+    return NArray.to_na(mem, typecode, *size)
+  end
+
+  def self.byte(alignment, *size)
+    return self.new(NArray::BYTE, alignment, *size)
+  end
+
+  def self.sint(alignment, *size)
+    return self.new(NArray::SINT, alignment, *size)
+  end
+
+  def self.int(alignment, *size)
+    return self.new(NArray::INT, alignment, *size)
+  end
+
+  def self.sfloat(alignment, *size)
+    return self.new(NArray::SFLOAT, alignment, *size)
+  end
+
+  def self.float(alignment, *size)
+    return self.new(NArray::FLOAT, alignment, *size)
+  end
+
+  def self.scomplex(alignment, *size)
+    return self.new(NArray::SCOMPLEX, alignment, *size)
+  end
+
+  def self.complex(alignment, *size)
+    return self.new(NArray::COMPLEX, alignment, *size)
+  end
+
+  def self.object(alignment, *size)
+    return self.new(NArray::OBJECT, alignment, *size)
+  end
+
+end

data/narray_ffi.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name = 'narray_ffi'
-  s.version = "1.1.1"
+  s.version = "1.2.0"
   s.author = "Brice Videau"
   s.email = "brice.videau@imag.fr"
   s.homepage = "https://forge.imag.fr/projects/opencl-ruby/"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: narray_ffi
 version: !ruby/object:Gem::Version
-  version: 1.1.1
+  version: 1.2.0
 platform: ruby
 authors:
 - Brice Videau