Files
SRS_evac_sim/archive/geometry.py
Varyngoth 676659e5b9 First
2026-01-28 13:31:49 -04:00

96 lines
2.2 KiB
Python

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())