Adding arrayfire.Array class and necessary helper functions

Author: pavanky

arrayfire/array.lua

@@ -1,5 +1,5 @@
 require('arrayfire.lib')
-require('class')
+require('arrayfire.defines')
 local ffi  = require( "ffi" )
 
 local funcs = {}
@@ -61,3 +61,66 @@ funcs[34] = [[
 ]]
 
 af.lib.cdef(funcs)
+
+local Array = {}
+Array.__index = Array
+
+local c_dim4_t = af.ffi.c_dim4_t
+local c_uint_t = af.ffi.c_uint_t
+local c_array_p = af.ffi.c_array_p
+
+local add_finalizer = function(arr_ptr)
+   return ffi.gc(arr_ptr[0], af.clib.af_release_array)
+end
+
+Array.__init = function(data, dims, dtype, source)
+   local self = setmetatable({}, Array)
+
+   if data then
+      assert(af.istable(data))
+   end
+
+   if dims then
+      assert(af.istable(dims))
+   end
+
+   c_dims = c_dim4_t(dims or (data and {#data} or {}))
+   c_ndims = c_uint_t(dims and #dims or (data and 1 or 0))
+
+   nelement = 1
+   for i = 1,tonumber(c_ndims) do
+      nelement = nelement * c_dims[i - 1]
+   end
+   nelement = tonumber(nelement)
+
+   local atype = dtype or af.dtype.f32
+   local res = c_array_p()
+   if not data then
+      af.clib.af_create_handle(res, c_ndims, c_dims, atype)
+   else
+      c_data = ffi.new(af.dtype_names[atype + 1] .. '[?]', nelement, data)
+      af.clib.af_create_array(res, c_data, c_ndims, c_dims, atype)
+   end
+   self.__arr = add_finalizer(res)
+   return self
+end
+
+Array.__tostring = function(self)
+   return 'arrayfire.Array\n'
+end
+
+Array.get = function(self)
+   return self.__arr
+end
+
+setmetatable(
+   Array,
+   {
+      __call = function(cls, ...)
+         return cls.__init(...)
+      end
+   }
+)
+
+af.Array = Array
+af.ffi.add_finalizer = add_finalizer

arrayfire/lib.lua

@@ -24,12 +24,24 @@ if ffi.abi('32bit') then
 else
    ffi.cdef[[typedef long long dim_t;]]
 end
+
 ffi.cdef[[
-    typedef void * af_array;
-    typedef int af_dtype;
-    typedef int af_err;
-    typedef int af_source;
-    af_err af_get_version (int *, int *, int *);
+typedef void * af_array;
+typedef int af_dtype;
+typedef int af_err;
+typedef int af_source;
+
+typedef struct af_cfloat {
+    float real;
+    float imag;
+} af_cfloat;
+
+typedef struct af_cdouble {
+    double real;
+    double imag;
+} af_cdouble;
+
+af_err af_get_version (int *, int *, int *);
 ]]
 
 local major, minor, patch = ffi.new("int[1]"), ffi.new("int[1]"), ffi.new("int[1]")
@@ -46,3 +58,34 @@ af.lib.cdef = function(funcs)
       end
    end
 end
+
+
+af.isnumber = function(val)
+   return type(val) == "number"
+end
+
+af.istable = function(val)
+   return type(val) == "table"
+end
+
+af.ffi = {}
+
+af.ffi.c_void_p = function()
+   return ffi.new('void *')
+end
+
+af.ffi.c_array_p = function(ptr)
+   return ffi.new('void *[1]', ptr)
+end
+
+af.ffi.c_dim_t = function(number)
+   return ffi.new('dim_t', number)
+end
+
+af.ffi.c_uint_t = function(number)
+   return ffi.new('unsigned int', number)
+end
+
+af.ffi.c_dim4_t = function(dims)
+   return ffi.new('dim_t[4]', dims or {1, 1, 1, 1})
+end

arrayfire/util.lua

@@ -27,3 +27,7 @@ funcs[34] = [[
 ]]
 
 af.lib.cdef(funcs)
+
+af.print = function(arr)
+   af.clib.af_print_array(arr:get())
+end