Cython version of the ordered dither function, immensely faster!

st_dither2.pyx

import numpy as np
cimport numpy as np
cimport cython

DTYPE = np.double
ctypedef np.double_t DTYPE_t
DTYPE2 = np.uint8
ctypedef np.uint8_t DTYPE2_t

@cython.boundscheck(False) # turn off bounds-checking for entire function
@cython.wraparound(False)  # turn off negative index wrapping for entire function
cpdef np.ndarray[DTYPE2_t,ndim=2] apply_ordered_dither(np.ndarray[DTYPE2_t,ndim=2] inputarray,np.ndarray[DTYPE_t,ndim=2] bayer_map, int bayer_size):
    cdef int width, height, x, y
    width = inputarray.shape[0]
    height = inputarray.shape[1]
    # width, height = inputarray.shape
    for y in range(height):
        for x in range(width):
            inputarray[x, y] = 0 if inputarray[x, y] < bayer_map[x % bayer_size, y % bayer_size] else 255
    return inputarray