"""
Implementation of some CFFI functions
"""

from __future__ import print_function, absolute_import, division

from numba.targets.imputils import implement, Registry
from numba import types
from . import arrayobj

registry = Registry()

@registry.register
@implement('ffi.from_buffer', types.Kind(types.Array))
def from_buffer(context, builder, sig, args):
    assert len(sig.args) == 1
    assert len(args) == 1
    [fromty] = sig.args
    [val] = args
    # Type inference should have prevented passing a buffer from an
    # array to a pointer of the wrong type
    assert fromty.dtype == sig.return_type.dtype
    ary = arrayobj.make_array(fromty)(context, builder, val)
    return ary.data
