120 lines
3.9 KiB
Python
120 lines
3.9 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import collections
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import webbrowser
|
||
|
|
import folium
|
||
|
|
from folium.plugins import HeatMap
|
||
|
|
|
||
|
|
|
||
|
|
TEXT_BASED_BROWSERS = [webbrowser.GenericBrowser, webbrowser.Elinks]
|
||
|
|
|
||
|
|
|
||
|
|
class Generator:
|
||
|
|
def __init__(self):
|
||
|
|
self.coordinates = collections.defaultdict(int)
|
||
|
|
self.max_coordinates = (0, 0)
|
||
|
|
self.max_magnitude = 0
|
||
|
|
|
||
|
|
def loadData(self, file_name):
|
||
|
|
"""Loads the google location data from the given json file.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
file_name {string} -- The name of the json file with the google
|
||
|
|
location data.
|
||
|
|
"""
|
||
|
|
if type(file_name) is str:
|
||
|
|
with open(file_name) as json_file:
|
||
|
|
data = json.load(json_file)
|
||
|
|
else:
|
||
|
|
assert type(file_name) is dict # call from library
|
||
|
|
data = file_name
|
||
|
|
|
||
|
|
for (i, loc) in enumerate(data["locations"]):
|
||
|
|
if "lat" not in loc or "lng" not in loc:
|
||
|
|
continue
|
||
|
|
#lat = round(loc["lat"] / 1e7, 6)
|
||
|
|
#lon = round(loc["lng"] / 1e7, 6)
|
||
|
|
lat = loc["lat"]
|
||
|
|
lon = loc["lng"]
|
||
|
|
self.coordinates[(lat, lon)] += 1
|
||
|
|
if self.coordinates[(lat, lon)] > self.max_magnitude:
|
||
|
|
self.max_coordinates = (lat, lon)
|
||
|
|
self.max_magnitude = self.coordinates[(lat, lon)]
|
||
|
|
|
||
|
|
def generateMap(self, map_zoom_start=6, heatmap_radius=7,
|
||
|
|
heatmap_blur=4, heatmap_min_opacity=0.2,
|
||
|
|
heatmap_max_zoom=4):
|
||
|
|
"""Generates a heatmap and saves it in the output_file.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
output_file {string} -- The name of the output file.
|
||
|
|
"""
|
||
|
|
map_data = [(coords[0], coords[1], magnitude)
|
||
|
|
for coords, magnitude in self.coordinates.items()]
|
||
|
|
|
||
|
|
# Generate map
|
||
|
|
m = folium.Map(location=self.max_coordinates,
|
||
|
|
zoom_start=map_zoom_start,
|
||
|
|
tiles="OpenStreetMap")
|
||
|
|
|
||
|
|
# Generate heat map
|
||
|
|
heatmap = folium.plugins.HeatMap(
|
||
|
|
map_data,
|
||
|
|
max_val=self.max_magnitude,
|
||
|
|
min_opacity=heatmap_min_opacity,
|
||
|
|
radius=heatmap_radius,
|
||
|
|
blur=heatmap_blur,
|
||
|
|
max_zoom=heatmap_max_zoom)
|
||
|
|
|
||
|
|
m.add_child(heatmap)
|
||
|
|
return m
|
||
|
|
|
||
|
|
def run(self, data_file, output_file):
|
||
|
|
"""Load the data, generate the heatmap and save it.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
data_file {string} -- The name of the json file with the google
|
||
|
|
location data.
|
||
|
|
output_file {[type]} -- The name of the output file.
|
||
|
|
"""
|
||
|
|
# print("Loading data from {}...".format(data_file))
|
||
|
|
self.loadData(data_file)
|
||
|
|
print("Generating heatmap...")
|
||
|
|
m = self.generateMap()
|
||
|
|
print("Saving map to {}...".format(output_file))
|
||
|
|
m.save(output_file)
|
||
|
|
|
||
|
|
|
||
|
|
def isTextBasedBrowser(browser):
|
||
|
|
"""Returns if browser is a text-based browser.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
browser {webbrowser.BaseBrowser} -- A browser.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
bool -- True if browser is text-based, False if browser is not
|
||
|
|
text-based.
|
||
|
|
"""
|
||
|
|
for tb_browser in TEXT_BASED_BROWSERS:
|
||
|
|
if type(browser) is tb_browser:
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
if len(sys.argv) > 1:
|
||
|
|
data_file = sys.argv[1]
|
||
|
|
output_file = "heatmap.html"
|
||
|
|
generator = Generator()
|
||
|
|
generator.run(data_file, output_file)
|
||
|
|
# Check if browser is text-based
|
||
|
|
# if not isTextBasedBrowser(webbrowser.get()):
|
||
|
|
# print("Opening {} in browser...".format(output_file))
|
||
|
|
# webbrowser.open('file://' + os.path.realpath(output_file))
|
||
|
|
else:
|
||
|
|
print("Usage: python geo_heatmap.py <file>")
|
||
|
|
sys.exit()
|