native.colourize_heightmap.argtypes = [ctypes.POINTER(ctypes.c_double), ctypes.POINTER(ctypes.c_double), ctypes.c_int, ctypes.c_int]
colourize_heightmap = native.colourize_heightmap

    y_true = tf.math.asinh(y_true - 4.94e-4)
    y_pred = tf.math.asinh(y_pred - 4.94e-4)

    y_true = tf.math.asinh(y_true - 0.1)
    y_pred = tf.math.asinh(y_pred - 0.1)

        }
    }
}
#[no_mangle]
pub extern "C" fn colourize_heightmap(target: *mut f64, source: *const f64, width: u32, height: u32) {
    const SEA_LEVEL: f64 = 0.1;
    let target = unsafe {
        std::slice::from_raw_parts_mut(target, (width as usize) * (height as usize) * 3)
    };
    let source = unsafe {
        std::slice::from_raw_parts(source, (width as usize) * (height as usize))
    };
    for y in 0 .. (height as usize) {
        let rowstart_s = y * (width as usize);
        let snow_line = {
            let y = (y as f64) / (height as f64);
            y * (1. - y) * 2.
        };
        for x in 0 .. (width as usize) {
            let ix = rowstart_s + x;
            let s = source[ix];
            let (r,g,b) = if s > snow_line {
                (1.,1.,1.)
            } else if s > SEA_LEVEL {
                let o = 0.9 * s;
                (o,0.9,o)
            } else {
                (0.,0.,0.9)
            };
            let ixt = ix * 3;
            target[ixt] = r;
            target[ixt + 1] = g;
            target[ixt + 2] = b;

  last_rendered = 0

  last_rendered = -15

        outputs = tf.math.multiply(255, tf.reshape(outputs ** 0.35, (512, 1024, outputs.shape[1])))

        outputs = tf.math.multiply(255, colourize_heightmap(tf.reshape(outputs, (512, 1024, outputs.shape[1]))))

def colourize_heightmap(source):
    shape = source.shape
    n = numpy.zeros((shape[0],shape[1],3), numpy.float64)
    source = tf.cast(source, tf.float64).numpy()
    helpers.colourize_heightmap(ctypes.cast(ctypes.c_voidp(n.ctypes.data), ctypes.POINTER(ctypes.c_double)), ctypes.cast(ctypes.c_voidp(source.ctypes.data), ctypes.POINTER(ctypes.c_double)), ctypes.c_int(shape[1]), ctypes.c_int(shape[0]))
    return tf.constant(n)

    outputs = tf.cast(outputs ** 0.35 * 255, tf.uint8)

    outputs = tf.cast(drawmap.colourize_heightmap(outputs) * 255, tf.uint8)