HexDecimal / libtcod-cffi

def console_fill_background(con,r,g,b):

    if len(r) != len(g) or len(r) != len(b):

        raise TypeError('R, G and B must all have the same size.')

    if (_numpy_available() and isinstance(r, _numpy.ndarray) and

        isinstance(g, _numpy.ndarray) and isinstance(b, _numpy.ndarray)):

        #numpy arrays, use numpy's ctypes functions

        r = _numpy.ascontiguousarray(r, dtype=_numpy.intc)

        g = _numpy.ascontiguousarray(g, dtype=_numpy.intc)

        b = _numpy.ascontiguousarray(b, dtype=_numpy.intc)

        cr = ffi.cast('int *', r.ctypes.data)

        cg = ffi.cast('int *', g.ctypes.data)

        cb = ffi.cast('int *', b.ctypes.data)

    else:

        # otherwise convert using ffi arrays

        cr = ffi.new('int[]', r)

        cg = ffi.new('int[]', g)

        cb = ffi.new('int[]', b)

    lib.TCOD_console_fill_background(con or ffi.NULL, cr, cg, cb)

def console_fill_foreground(con,r,g,b):

    if len(r) != len(g) or len(r) != len(b):

        raise TypeError('R, G and B must all have the same size.')

    if (_numpy_available() and isinstance(r, _numpy.ndarray) and

        isinstance(g, _numpy.ndarray) and isinstance(b, _numpy.ndarray)):

        #numpy arrays, use numpy's ctypes functions

        r = _numpy.ascontiguousarray(r, dtype=_numpy.intc)

        g = _numpy.ascontiguousarray(g, dtype=_numpy.intc)

        b = _numpy.ascontiguousarray(b, dtype=_numpy.intc)

        cr = ffi.cast('int *', r.ctypes.data)

        cg = ffi.cast('int *', g.ctypes.data)

        cb = ffi.cast('int *', b.ctypes.data)

    else:

        # otherwise convert using ffi arrays

        cr = ffi.new('int[]', r)

        cg = ffi.new('int[]', g)

        cb = ffi.new('int[]', b)

    lib.TCOD_console_fill_foreground(con or ffi.NULL, cr, cg, cb)