%load_ext autoreload
%autoreload 2
from rush import *
import matplotlib.pyplot as plt
import numpy as np
import time
import glob

s = State()
s.load_puzzle("jams/1.txt")
neighbors = s.get_neighbors()
# Plot everything
N = len(neighbors)+1
K = int(np.ceil(np.sqrt(N))) # Figure out a square grid to put these on

plt.figure(figsize=(10, 10))
plt.subplot(K, K, 1)
plt.title("Original")
s.plot()
for i, n in enumerate(neighbors):
    plt.subplot(K, K, i+2)
    n.plot()
    plt.title("Neighbor {}".format(i))
plt.tight_layout()