stud/Karto/assignment4/draw.py
2019-11-29 14:11:51 +02:00

108 lines
3.2 KiB
Python
Executable File

#!/usr/bin/env python3
import numpy as np
import matplotlib.pyplot as plt
from collections import namedtuple
from shapely.geometry import LineString, asPolygon
from descartes import PolygonPatch
from measure import *
juosta = namedtuple('juosta', ['plotis', 'kryptis', 'dashes', 'spalva'])
kelias = namedtuple('kelias', ['virsunes', 'plotis', 'kat', 'dashes', 'spalva', 'juostos'])
CONTINUOUS = (1,0)
DASHDOTX2 = (10,3,2,3)
DASHED = (100,20)
a08 = kelias(
virsunes=[1,2,3],
plotis=A08_plotis,
kat=KAT1,
dashes=DASHDOTX2,
spalva='red',
juostos=(
juosta(L6+L5+L4, 'right', DASHED, 'lightgreen'),
juosta(L6+L5, 'right', DASHED, 'lightgreen'),
juosta(L6, 'right', CONTINUOUS, 'black'),
juosta(L7, 'left', CONTINUOUS, 'black'),
juosta(L7+L8, 'left', DASHED, 'lightgreen'),
juosta(L7+L8+L9, 'left', DASHED, 'lightgreen'),
),
)
a05 = kelias(
virsunes=[4,5,6,7,8,9,10],
plotis=A05_plotis,
kat=KAT2,
dashes=DASHDOTX2,
spalva='red',
juostos=(
juosta(L3, 'right', CONTINUOUS, 'brown'),
juosta(L2, 'left', CONTINUOUS, 'brown'),
),
)
a03 = kelias(
virsunes=[11,12,13,14,15,16,17,18],
plotis=A03_plotis,
kat=KAT3,
dashes=CONTINUOUS,
spalva='pink',
juostos=(
juosta(L1, 'right', DASHED, 'pink'),
),
)
g11 = kelias(
virsunes=[19,20,21,22,23],
plotis=G11_plotis,
kat=KAT4,
dashes=CONTINUOUS,
spalva='red',
juostos=(
juosta(L10+L11, 'right', CONTINUOUS, 'blue'),
juosta(L11, 'right', CONTINUOUS, 'lightblue'),
juosta(L12, 'left', CONTINUOUS, 'lightblue'),
juosta(L12+L13, 'left', CONTINUOUS, 'blue'),
),
)
N, E, S, W = (0,15), (15,0), (0,-15), (-15,0)
point_annotations = {
1: S, 2: E, 3: N, 4: E,
5: N, 6: S, 7: S, 8: N,
9: E, 10: N, 11: N, 12: N,
13: S, 14: W, 15: N, 16: N,
17: N, 18: E, 19: N, 20: S,
21: N, 22: N, 23: E, 24: N,
}
# implementacija
fig, ax = plt.subplots()
plt.grid(True)
# taškų anotacijos
for v in vertices:
ax.annotate(v.point, xy=v.xy, zorder=99, textcoords='offset points', xytext=point_annotations[v.point])
# kelių piešimas
for kelias in [a08, a05, a03, g11]:
# ašis
kelias_line = LineString([Points[i].xy for i in kelias.virsunes ])
ax.plot(*kelias_line.xy, linewidth=2, dashes=kelias.dashes, color=kelias.spalva, zorder=kelias.kat)
# offset'ai
offset_multiplier = kelias.plotis/(kelias.juostos[0].plotis + kelias.juostos[-1].plotis)
offset_lines = []
for offset in kelias.juostos:
l = kelias_line.parallel_offset(offset.plotis * offset_multiplier, offset.kryptis, join_style=2)
offset_lines.append(l)
ax.plot(*l.xy, linewidth=.5, dashes=offset.dashes, color=offset.spalva, zorder=kelias.kat)
# kelio poligonas su plotu
kelias_poly = np.vstack((offset_lines[0].coords, offset_lines[-1].coords))
ax.add_patch(PolygonPatch(asPolygon(kelias_poly), fc='white', zorder=kelias.kat, linewidth=0))
plt.show()