Mercurial > bus-service / file revision

#!/usr/bin/env python3
from pprint import pprint
from PyQt5.QtCore import QPointF, Qt
from PyQt5.QtGui import QPolygonF
from PyQt5.QtWidgets import QApplication
import sys, json
from misc import *

app = QApplication(sys.argv)

with open('alueet.gmp') as file:
	data = file.read().split('@')

with open('alue-edustajat.json') as file:
	representatives = json.load(file)

pysäkit = {}

with open('gtfs/stops.txt') as file:
	for rivi in lue_csv(file):
		sijainti = QPointF(float(rivi['stop_lat']), float(rivi['stop_lon']))
		tunniste = rivi['stop_id']
		pysäkit[tunniste] = sijainti

district_shapes = []
for polygon in data[1].splitlines():
	polygon = polygon.split('^')
	coordinates = []
	for point in polygon[3].split('~'):
		x, y = point.split(',')
		coordinates.append(QPointF(float(x), float(y)))
	district_shapes.append(QPolygonF(coordinates))

districts = {}
bus_stop_districts = {}

for name, stop_id in representatives.items():
	if stop_id is None:
		continue
	for district_shape in district_shapes:
		if district_shape.containsPoint(pysäkit[stop_id], Qt.OddEvenFill):
			assert name not in districts
			districts[name] = district_shape
			district_shapes.remove(district_shape)
			bus_stop_districts[stop_id] = name
			break
	else:
		print('cannot find a shape for %r' % name, file = sys.stderr)

for stop_id, stop_position in pysäkit.items():
	for district, shape in districts.items():
		if shape.containsPoint(stop_position, Qt.OddEvenFill):
			bus_stop_districts[stop_id] = district
			break
	else:
		bus_stop_districts[stop_id] = None

covered = sum(int(bool(k)) for k in bus_stop_districts.values())
total = len(pysäkit)
print('%.1f%% bus stops covered.' % (covered * 100 / total), file = sys.stderr)

json.dump(bus_stop_districts, sys.stdout, indent = 2)