Mercurial > bus-service / changeset

--- a/buses.py	Sat Jun 10 18:06:27 2017 +0300
+++ b/buses.py	Sat Jun 10 20:56:38 2017 +0300
@@ -62,6 +62,8 @@
 class Pysäkki:
 	def __init__(self, tunniste, nimi, sijainti):
 		self.tunniste, self.nimi, self.sijainti = tunniste, nimi, sijainti
+		self.cluster = None
+		self.pairs = set() # samannimiset lähellä olevat pysäkit
 	def __repr__(self):
 		return 'pysäkit[%r]' % self.tunniste
 	def aikataulu(self, määrä = 50):
@@ -131,6 +133,7 @@
 ajot_per_numero = {}
 palvelut = {}
 pysäkit = {}
+all_clusters = set()
 
 print('Ladataan linjat... ', file = stderr, end = '', flush = True)
 with open('gtfs/routes.txt') as tiedosto:
@@ -194,6 +197,131 @@
 		pysäkit[pysäkkitunniste].alue = alue
 print('%d pysäkkiä' % len(pysäkit), file = stderr)
 
+class BusStopCluster:
+	def __init__(self):
+		self.stops = set()
+		self._center = None
+		self.name = None
+	def add_stop(self, stop):
+		assert not stop.cluster
+		stop.cluster = self
+		self.stops.add(stop)
+		self._center = None
+	@property
+	def center(self):
+		if not self._center:
+			if self.stops:
+				from statistics import median
+				pointtype = type(next(iter(self.stops)).sijainti)
+				self._center = pointtype(
+					median(stop.sijainti.x for stop in self.stops),
+					median(stop.sijainti.y for stop in self.stops),
+				)
+			else:
+				raise ValueError('an empty cluster has no center point')
+		return self._center
+	def merge(self, other):
+		for bus_stop in other.stops:
+			bus_stop.cluster = self
+		self.stops |= other.stops
+		other.stops = set()
+		other._center = None
+
+from collections import defaultdict
+bus_stops_by_name = defaultdict(set)
+for bus_stop in pysäkit.values():
+	bus_stops_by_name[bus_stop.nimi].add(bus_stop)
+bus_stops_by_name = dict(bus_stops_by_name)
+
+# ryhmittele pysäkit nimen mukaan
+all_clusters = []
+def cluster_bus_stops():
+	sorted_bus_stops = sorted(pysäkit.values(), key = lambda bus_stop: bus_stop.nimi)
+	for bus_stop in sorted_bus_stops:
+		if not bus_stop.cluster:
+			stops_to_cluster = {bus_stop}
+			# etsi pysäkin samannimiset vastaparit
+			for pair_candidate in bus_stops_by_name[bus_stop.nimi]:
+				distance = pair_candidate.sijainti.etäisyys(bus_stop.sijainti)
+				if pair_candidate is not bus_stop and distance <= 0.3:
+					stops_to_cluster.add(pair_candidate)
+			for stop_to_cluster in stops_to_cluster:
+				if stop_to_cluster.cluster:
+					cluster = stop_to_cluster.cluster
+					break
+			else:
+				cluster = BusStopCluster()
+				all_clusters.append(cluster)
+			for stop_to_cluster in stops_to_cluster:
+				if not stop_to_cluster.cluster:
+					cluster.add_stop(stop_to_cluster)
+	# Merkitse muistiin pysäkkien vastaparit käyttäen hyväksi tämänhetkistä ryhmittelytietoa
+	for bus_stop in pysäkit.values():
+		if bus_stop.cluster:
+			bus_stop.pairs = bus_stop.cluster.stops - {bus_stop}
+	# Ryhmitä ne pysäkit, joilla ei ollut omaa vastaparia, muiden pysäkkien kanssa
+	for bus_stop in sorted_bus_stops:
+		if len(bus_stop.cluster.stops) == 1:
+			possibilities = set()
+			for cluster in all_clusters:
+				if cluster is not bus_stop.cluster:
+					distance = cluster.center.etäisyys(bus_stop.sijainti)
+					if distance <= 0.3:
+						possibilities.add((distance, cluster))
+			if possibilities:
+				best = min(possibilities)[1]
+				all_clusters.remove(bus_stop.cluster)
+				best.merge(bus_stop.cluster)
+
+def shared_elements_in_n_sets(sets):
+	from itertools import combinations
+	result = set()
+	for pair in combinations(sets, 2):
+		result |= pair[0] & pair[1]
+	return result
+
+def name_clusters():
+	from collections import defaultdict
+	from pprint import pprint
+	clusters_per_name = defaultdict(set)
+	for cluster in all_clusters:
+		name_representing_stop = min((len(pysäkki.tunniste), pysäkki.tunniste, pysäkki) for pysäkki in cluster.stops)[2]
+		name = name_representing_stop.nimi.lower().replace(' ', '-')
+		clusters_per_name[name].add(cluster)
+	for name, clusters in clusters_per_name.items():
+		if len(clusters) == 1:
+			# Ryhmä on ainoa jolla on varaus tälle nimelle. Sen kuin vaan.
+			next(iter(clusters)).name = name
+		else:
+			# Olisiko kaikki klusterit eri alueilla?
+			common_regions = shared_elements_in_n_sets({stop.alue for stop in cluster.stops} for cluster in clusters)
+			# Esitys: ryhmä -> ne alueet jotka ovat tälle ryhmälle ainutlaatuisia
+			proposal = {
+				cluster: {stop.alue for stop in cluster.stops} - common_regions - {None}
+				for cluster in clusters
+			}
+			# Jos enintään yksi klusteri tässä esityksessä on kokonaan ilman omaa aluetta, jolla se voisi eritellä,
+			# niin nimetään klusterit näiden alueiden mukaan.
+			# Se klusteri jolla ei ole omaa aluetta (jos on) jätetään ilman aluepäätettä.
+			if sum([1 for unique_areas in proposal.values() if not unique_areas]) <= 1:
+				for cluster, unique_areas in proposal.items():
+					individual_cluster_name = name
+					if unique_areas:
+						individual_cluster_name += '-' + min(unique_areas).lower().replace(' ', '-')
+					cluster.name = individual_cluster_name
+			else:
+				# Typerä reunatapaus. Indeksoidaan numeroin...
+				for n, (_, cluster) in enumerate(sorted(
+					min((stop.tunniste.lower(), cluster) for stop in cluster.stops)
+					for cluster in clusters
+				), 1):
+					individual_cluster_name = name + '-' + str(n)
+					cluster.name = individual_cluster_name
+
+print('Ryhmitellään pysäkit...')
+cluster_bus_stops()
+name_clusters()
+
 print('Ladataan aikataulut... ', end = '', flush = True, file = stderr)
 with open('gtfs/stop_times.txt') as file:
 	rivimäärä = sum(line.count('\n') for line in file)