fn idx3d(i: usize, j: usize, k: usize, nx: usize, ny: usize) -> usize
Index into 3D array (Fortran/column-major order)