Function hilbert_path_square

utils/utils_zigzag.py:242–282

(num_dims=2, N=4)

def hilbert_path_square(num_dims=2, N=4):
    import numpy as np
    from hilbert import decode

    img_size_power = int(math.sqrt(N))
    assert img_size_power**2 == N, f"{N} should be a square number"
    print("img_size_power", img_size_power)

    def draw_curve(num_bits):
        # The maximum Hilbert integer.
        max_h = 2 ** (num_bits * num_dims)

        # Generate a sequence of Hilbert integers.
        hilberts = np.arange(max_h)
        print("image size:", N)
        order_index = np.zeros((N, N), dtype=int)

        # Compute the 2-dimensional locations.
        locs = decode(hilberts, num_dims, num_bits)
        for i, loc in enumerate(locs):
            order_index[loc[0], loc[1]] = i
        # print(locs.shape, locs)
        print(order_index)
        return order_index

    res = draw_curve(img_size_power)
    res_mirror = np.transpose(res)
    ro90 = np.rot90(res, 1)
    ro90_mirror = np.transpose(ro90)
    ro180 = np.rot90(res, 2)
    ro180_mirror = np.transpose(ro180)
    ro270 = np.rot90(res, 3)
    ro270_mirror = np.transpose(ro270)
    res = [res, res_mirror, ro90, ro90_mirror, ro180, ro180_mirror, ro270, ro270_mirror]
    print("***")
    for _ in res:
        print(_)
    res = [_.flatten() for _ in res]
    for _ in res:
        print(_.shape)
    return res


def hilbert_path(N=16):