import sys
import matplotlib
matplotlib.use('QtAgg')
import matplotlib.pyplot as plt
from shapely import Polygon
from shapely.plotting import plot_polygon

def current():
    queue = Polygon([
        (18.767,9.395),
        (15.214,9.395),
        (15.214,4.569),
        (15.214,0),
        (16.924,0),
        (16.924,4.569),
        (25.0,4.569),
        (25.0,16.823),
        #(26.163,4.569),
        #(28.214,5.07),
        #(28.214,17.761),
        #(25.787,17.761),
        ])
    crosswalk = Polygon([
        (4,6.068),
        (4,7.896),
        (4,7.896),
        (15.214,7.896),
        (15.214,6.068),
        (4,6.068)
        ])
    grass = Polygon([
        (4,0),
        (0,0),
        (0,17.761),
        (4,17.761)
        ])
    plot_polygon(queue,color="blue",add_points=False)
    plot_polygon(crosswalk,color="red",add_points=False)
    plot_polygon(grass,color="blue",add_points=False)

def new():
    new_queue = Polygon([
        (19.531,10.306),
        (13.88,10.306),
        (12.98,9.896),
        (12.98,4.569),
        (12.98,0),
        (16.924,0),
        (16.924,4.569),
        (25.0,4.569),
        (25.0,16.823),
        #(26.163,4.569),
        #(28.214,5.07),
        #(28.214,17.761),
        #(25.787,17.761),
        ])
    new_crosswalk = Polygon([
        (6.23,4.982),
        (6.23,8.982),
        (12.98,8.982),
        (12.98,4.982),
        ])
    new_grass = Polygon([
        (6.23,0),
        (0,0),
        (0,17.761),
        (4,17.761),
        (4,10.306),
        (5.33,10.306),
        (6.23,9.896)
        ])
    plot_polygon(new_queue,color="blue",add_points=False)
    plot_polygon(new_crosswalk,color="red",add_points=False)
    plot_polygon(new_grass,color="blue",add_points=False)
    return 

def spawn():
    spawn_area = Polygon([
        (25.0,16.823),
        (25.0,4.569),
        (26.163,4.569),
        (28.214,5.07),
        (28.214,17.761),
        (25.787,17.761),
        ])
    plot_polygon(spawn_area,color="green",add_points=False)
    return spawn_area

if __name__ == "__main__":
    from PyQt6 import QtWidgets
    app = QtWidgets.QApplication(sys.argv)
    new()
    current()
    spawn()
    plt.show(block=False)
    sys.exit(app.exec())