Function test_hibert_fig

utils/utils_zigzag.py:207–239

(num_dims=2, img_size_power=3)

def test_hibert_fig(num_dims=2, img_size_power=3):
    import numpy as np
    import matplotlib.pyplot as plt
    from hilbert import decode

    def draw_curve(ax, 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:", 2**img_size_power)
        order_index = np.zeros((2**img_size_power, 2**img_size_power), 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)
        # Draw
        ax.plot(locs[:, 0], locs[:, 1], ".-")
        ax.set_aspect("equal")
        # ax.set_title("%d bits per dimension" % (num_bits))
        # ax.set_xlabel("dim 1")
        # ax.set_ylabel("dim 2")

    fig = plt.figure(figsize=(16, 4))
    for ii, num_bits in enumerate([img_size_power]):
        ax = fig.add_subplot(1, 4, ii + 1)
        draw_curve(ax, num_bits)
    plt.savefig("example_2d.png", bbox_inches="tight")


def hilbert_path_square(num_dims=2, N=4):