Function index_3d_from_linear 

pub fn index_3d_from_linear(
    index: usize,
    dim_x: usize,
    dim_y: usize,
) -> (usize, usize, usize)

Convert a linear index back to 3D coordinates.