compute-regions.py

#!/usr/bin/env python3
import sys, json
from misc import *
from geometry import *

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

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

bus_stops = {}

with open('gtfs/stops.txt') as file:
	for rivi in lue_csv(file):
		sijainti = Sijainti(float(rivi['stop_lat']), float(rivi['stop_lon']))
		tunniste = rivi['stop_id']
		bus_stops[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(Sijainti(float(x), float(y)))
	district_shapes.append(Monikulmio(*coordinates))

districts = {}
bus_stop_districts = {}

for name, stop_id in representatives.items():
	if stop_id is None:
		continue
	if stop_id not in bus_stops:
		print('Representative %r for region %r not found in schedule' % (stop_id, name), file = sys.stderr)
	else:
		for district_shape in district_shapes:
			if district_shape.sisältää_pisteen(bus_stops[stop_id]):
				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 bus_stops.items():
	for district, shape in districts.items():
		if shape.sisältää_pisteen(stop_position):
			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(bus_stops)
print('%.1f%% bus stops covered.' % (covered * 100 / total), file = sys.stderr)
json.dump(bus_stop_districts, sys.stdout, indent = 2)