local ffi = require 'ffi'
local util = require('multiverso.util')

local tbh = torch.class('ArrayTableHanlder')

ffi.cdef[[
    void MV_NewArrayTable(int size, TableHandler* out);
    void MV_GetArrayTable(TableHandler handler, float* data, int size);
    void MV_AddArrayTable(TableHandler handler, float* data, int size);
]]

function tbh:new(size)
    tbh = {}
    size = size or 0
    setmetatable(tbh, self)
    self.__index = self
    tbh._handler = ffi.new("TableHandler[1]")
    tbh._size = ffi.new("int", size)
    libmv.MV_NewArrayTable(
        tbh._size,
        tbh._handler
    )
    return tbh
end

function tbh:get()
    cdata = ffi.new("float[?]", self._size)
    libmv.MV_GetArrayTable(self._handler[0], cdata, self._size)
    return util.cdata2tensor(cdata, tonumber(self._size))
end

function tbh:add(data)
    cdata = util.tensor2cdata(data)
    libmv.MV_AddArrayTable(self._handler[0], cdata, self._size)
end

return tbh