compute-regions.py

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

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

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

bus_stops = {}

with ZipFile('gtfs.zip') as archive:
	with archive.open('stops.txt') as file:
		for row in read_csv(map(bytes.decode, file)):
			location = Sijainti(float(row['stop_lat']), float(row['stop_lon']))
			reference = row['stop_id']
			bus_stops[reference] = location

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)