Initial commit

Author: jmhummel

cache.p

@@ -0,0 +1,9 @@
+api_keys:
+    # https://developers.google.com/maps/documentation/static-maps/
+    google_static_maps: YOURKEYHERE
+    # https://developers.google.com/maps/documentation/distance-matrix/
+    google_distance_matrix: YOURKEYHERE
+    # https://developers.google.com/maps/documentation/directions/
+    google_directions: YOURKEYHERE
+    # https://developers.google.com/maps/documentation/geocoding/
+    google_geocoding: YOURKEYHERE

config.yml

@@ -0,0 +1,41 @@
+#!/usr/bin/env python
+
+import requests
+import pickle
+
+import maputils
+
+API_KEY = maputils.config['api_keys']['google_distance_matrix']
+BASE_URL = 'https://maps.googleapis.com/maps/api/distancematrix'
+CAPITALS = maputils.CAPITALS
+
+# cachedDistDur = pickle.load(open('cache.p', 'rb'))
+cachedDistDur = {}
+
+def getDistDur(origin, destination):
+	# print origin, destination
+	key = tuple(sorted([origin, destination]))
+	if key in cachedDistDur:
+		# print origin, destination, 'from cache'
+		return cachedDistDur[key]
+	origin = origin.replace(' ','+')
+	destination = destination.replace(' ','+')
+	r = requests.get(BASE_URL + '/json?units=imperial&origins=' + origin + '&destinations=' + destination + '&key=' + API_KEY)
+	if r.json()['status'] == 'OK':
+		dist = r.json()['rows'][0]['elements'][0]['distance']['value']
+		time = r.json()['rows'][0]['elements'][0]['duration']['value']
+		result = (dist, time)
+		cachedDistDur[key] = result
+		print key, result
+		# print origin, destination, result
+		return result
+	else:
+		raise Exception('Over api limit!')
+
+for p1 in CAPITALS[:50]:
+	print p1
+	for p2 in CAPITALS[:50]:
+		if p1 != p2:
+			getDistDur(p1, p2)
+
+pickle.dump(cachedDistDur, open('cache.p', 'wb'), -1)

createCache.py

@@ -0,0 +1,41 @@
+#!/usr/bin/env python
+
+import requests
+import pickle
+import polyline
+
+import maputils
+
+API_KEY = maputils.config['api_keys']['google_directions']
+BASE_URL = 'https://maps.googleapis.com/maps/api/directions/json?'
+CAPITALS = maputils.CAPITALS
+
+# directionsCache = pickle.load(open('directions.p','rb'))
+directionsCache = {}
+
+def getDirections(origin, destination):
+	if (origin, destination) in directionsCache:
+		directions = directionsCache[(origin, destination)]
+		# print origin, destination
+	else:
+		url = BASE_URL + 'origin=' + origin + '&destination=' + destination + '&key=' + API_KEY
+		print origin, destination, '\t' + url
+		r = requests.get(url)
+		# print url
+		if r.json()['status'] == 'OK':
+			poly = r.json()['routes'][0]['overview_polyline']['points']
+			directions = polyline.decode(poly)
+		else:
+			print r.json()
+			raise Exception('Something went wrong')
+		directionsCache[(origin, destination)] = directions
+		directionsCache[(destination, origin)] = list(reversed(directions))
+	return directions
+
+for p1 in CAPITALS[:50]:
+	# print p1
+	for p2 in CAPITALS[:50]:
+		if p1 != p2:
+			getDirections(p1, p2)
+
+pickle.dump(directionsCache, open('directions.p','wb'), -1)

createDirectionsCache.py

@@ -0,0 +1,23 @@
+#!/usr/bin/env python
+
+import requests
+import pickle
+import yaml
+import maputils
+
+CAPITALS = maputils.CAPITALS
+API_KEY = maputils.config['api_keys']['google_geocoding']
+BASE_URL = 'https://maps.googleapis.com/maps/api/geocode'
+
+def getCoords(place):
+	place = place.replace(' ','+')
+	url = BASE_URL + '/json?address=' + place + '&key=' + API_KEY
+	r = requests.get(url)
+	lat = r.json()['results'][0]['geometry']['location']['lat']
+	lon = r.json()['results'][0]['geometry']['location']['lng']
+	return (lat, lon)
+
+geoCache = {}
+for place in CAPITALS:
+	geoCache[place] = getCoords(place)
+pickle.dump(geoCache, open('geoCache.p','wb'), -1)

createGeoCache.py

@@ -0,0 +1,20 @@
+import matplotlib.pyplot as plt
+from mpl_toolkits.basemap import Basemap
+import pickle
+
+# create the figure and axes instances.
+fig = plt.figure()
+ax = fig.add_axes([0.1,0.1,0.8,0.8])
+# setup of basemap ('lcc' = lambert conformal conic).
+# use major and minor sphere radii from WGS84 ellipsoid.
+m = Basemap(llcrnrlon=-145.5,llcrnrlat=1.,urcrnrlon=-2.566,urcrnrlat=46.352,\
+            rsphere=(6378137.00,6356752.3142),\
+            resolution='l',area_thresh=1000.,projection='lcc',\
+            lat_1=50.,lon_0=-107.,ax=ax)
+
+# draw coastlines and political boundaries.
+m.drawcoastlines()
+m.drawcountries()
+m.drawstates()
+
+pickle.dump(m,open('map.p','wb'),-1)

createMap.py

@@ -0,0 +1,25 @@
+import polyline
+import requests
+import csv
+import pickle
+import shutil
+from subprocess import call
+
+import maputils
+
+csvfile = open('stats.tsv', 'rb')
+stats = csv.reader(csvfile, delimiter='\t')
+route = list(eval(list(stats)[-1][1]))
+
+poly = maputils.getPoly(route, 0.2)
+url = maputils.mapUrl(poly)
+
+filename = 'goog.png'
+r = requests.get(url, stream=True)
+if r.status_code == 200:
+    with open(filename, 'wb') as f:
+        r.raw.decode_content = True
+        shutil.copyfileobj(r.raw, f) 
+        call(["open", filename])
+else:
+	print r.status_code, 'Error'

directions.p

@@ -0,0 +1,33 @@
+import polyline
+import requests
+import pickle
+import shutil
+import csv
+import math
+
+import maputils
+
+csvfile = open('stats.tsv', 'rb')
+stats = csv.reader(csvfile, delimiter='\t')
+routes = [ list(eval(row[1])) for row in stats ]
+
+for i in xrange(len(routes)):
+# for i in xrange(len(routes)-1,len(routes)):
+
+	filename = 'pics/' + '{:03}'.format(i) + '.png'
+	# route = routes[i]
+	route = routes[i]
+
+	poly = maputils.getPoly(route, 0.15)
+	url = maputils.mapUrl(poly)
+
+	r = requests.get(url, stream=True)
+	if r.status_code == 200:
+	    with open(filename, 'wb') as f:
+	        r.raw.decode_content = True
+	        shutil.copyfileobj(r.raw, f) 
+	        print len(url), len(url) - len(poly), len(poly), len(poly)/float(len(polyline.decode(poly))), filename #, routes[i]
+	else:
+		print len(url), len(url) - len(poly), len(poly), len(poly)/float(len(polyline.decode(poly))), r.status_code, 'ERROR'
+		break
+