import jax.numpy as jnp
import matplotlib.pyplot as plt

from differt2d.geometry import Wall
from differt2d.scene import Scene

ax = plt.gca()
scene = Scene.square_scene()
wall = Wall(xys=jnp.array([[.8, .2], [.8, .8]]))
scene = scene.add_objects(wall)
scene.plot(ax)

for _, _, path, _ in scene.all_valid_paths():
    path.plot(ax, zorder=-1)  # -1 to draw below the scene objects

plt.show()