import matplotlib.pyplot as plt
from IPython.display import SVG, display  # noqa: A004

from maze_dataset import LatticeMazeGenerators, MazeDataset, MazeDatasetConfig
from maze_dataset.plotting.plot_svg_fancy import plot_svg_fancy

cfg = MazeDatasetConfig(
	name="test",
	grid_n=5,
	n_mazes=5,
	maze_ctor=LatticeMazeGenerators.gen_dfs,
)
ds = MazeDataset.from_config(cfg)
maze = ds[0]
pixels = maze.as_pixels()

plt.imshow(pixels)
plt.axis("off")

(np.float64(-0.5), np.float64(10.5), np.float64(10.5), np.float64(-0.5))

svg_string: str = plot_svg_fancy(pixels)

display(SVG(svg_string))

from maze_dataset.plotting import MazePlot

MazePlot(maze).plot(plain=True)

<maze_dataset.plotting.plot_maze.MazePlot at 0x740c982ccaa0>

print(maze.as_ascii())

###########
#    XXX# #
# ###X#X# #
#   #X#S  #
#####X#####
#XXXXX#EXX#
#X### ###X#
#X#     #X#
#X#######X#
#XXXXXXXXX#
###########

from maze_dataset.plotting.print_tokens import (
	color_maze_tokens_AOTP,
)
from maze_dataset.tokenization import MazeTokenizer

print(color_maze_tokens_AOTP(maze.as_tokens(MazeTokenizer()), "html"))

<span style="color: black; background-color: rgb(176,152,232)">&nbsp&lt;ADJLIST_START&gt; (0,0) &lt;--&gt; (1,0) ; (2,0) &lt;--&gt; (3,0) ; (4,1) &lt;--&gt; (4,0) ; (2,0) &lt;--&gt; (2,1) ; (1,0) &lt;--&gt; (1,1) ; (3,4) &lt;--&gt; (2,4) ; (4,2) &lt;--&gt; (4,3) ; (0,0) &lt;--&gt; (0,1) ; (0,3) &lt;--&gt; (0,2) ; (4,4) &lt;--&gt; (3,4) ; (4,3) &lt;--&gt; (4,4) ; (4,1) &lt;--&gt; (4,2) ; (2,1) &lt;--&gt; (2,2) ; (1,4) &lt;--&gt; (0,4) ; (1,2) &lt;--&gt; (0,2) ; (2,4) &lt;--&gt; (2,3) ; (4,0) &lt;--&gt; (3,0) ; (2,2) &lt;--&gt; (3,2) ; (1,2) &lt;--&gt; (2,2) ; (1,3) &lt;--&gt; (0,3) ; (3,2) &lt;--&gt; (3,3) ; (0,2) &lt;--&gt; (0,1) ; (3,1) &lt;--&gt; (3,2) ; (1,3) &lt;--&gt; (1,4) ; &lt;ADJLIST_END&gt;&nbsp</span> <span style="color: black; background-color: rgb(154,239,123)">&nbsp&lt;ORIGIN_START&gt; (1,3) &lt;ORIGIN_END&gt;&nbsp</span> <span style="color: black; background-color: rgb(246,136,136)">&nbsp&lt;TARGET_START&gt; (2,3) &lt;TARGET_END&gt;&nbsp</span> <span style="color: black; background-color: rgb(111,187,254)">&nbsp&lt;PATH_START&gt; (1,3) (0,3) (0,2) (1,2) (2,2) (2,1) (2,0) (3,0) (4,0) (4,1) (4,2) (4,3) (4,4) (3,4) (2,4) (2,3) &lt;PATH_END&gt;&nbsp</span>