Skip to content

Heatmap #12

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
kenseii wants to merge 13 commits into
base: master
Choose a base branch
from
Open

Heatmap #12

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
6ae3318
added a basic scaffolding of the heatmap placeholder
kenseii Oct 4, 2019
a045249
updated the heatmap writer
kenseii Oct 4, 2019
f2f4ff8
using the correct heatmap instance
kenseii Oct 4, 2019
dfe3ed9
had forgotten to uncomment the heatmap writer
kenseii Oct 4, 2019
437a2d2
trying to view the format of heatmap
kenseii Oct 4, 2019
88660c2
changed the html writer
kenseii Oct 4, 2019
1c71543
renamed the heat to heatmap
kenseii Oct 4, 2019
0dce3e8
changed the location of the heatmap library as it was getting added i…
kenseii Oct 4, 2019
558bdf7
added all the plugin variables
kenseii Oct 5, 2019
dd5d5f0
attempt to add all the options at the same time
kenseii Oct 5, 2019
658d08e
set default gradient
kenseii Oct 5, 2019
e4d4fa8
set default gradient
kenseii Oct 5, 2019
775c5fa
cleanup commented parts
kenseii Oct 5, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions llplot/google_maps_templates.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,13 @@
}}).addTo(llMap);
"""

HEATMAP = """
var heatmap = L.heatLayer({latlng}, {{
radius: {radius},

}}).addTo(llMap);
"""

# FIXME: This generate a Xmark in cartesian frame rather than in lat/long.
XMARK = """

Expand Down
113 changes: 35 additions & 78 deletions llplot/llplot.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
from collections import namedtuple

from llplot.color_dicts import mpl_color_map, html_color_codes
from llplot.google_maps_templates import SYMBOLS, CIRCLE
from llplot.google_maps_templates import SYMBOLS, CIRCLE, HEATMAP


Symbol = namedtuple('Symbol', ['symbol', 'lat', 'long', 'size'])
Expand Down Expand Up @@ -161,55 +161,35 @@ def plot(self, lats, lngs, color=None, c=None, **kwargs):
path = zip(lats, lngs)
self.paths.append((path, settings))

def heatmap(self, lats, lngs, threshold=10, radius=10, gradient=None, opacity=0.6, maxIntensity=1, dissipating=True):
def heatmap(self, coordinates_intensity, blur=15, radius=10, gradient={0.4: 'blue', 0.65: 'lime', 1: 'red'}, minOpacity=0.6, max=1.0, maxZoom=10):
"""
:param lats: list of latitudes
:param lngs: list of longitudes
:param coordinates_intensity: (list of lists) the inner list contains [lat, long, intensity]
:param maxIntensity:(int) max frequency to use when plotting. Default (None) uses max value on map domain.
:param threshold:
:param radius: The hardest param. Example (string):
:return:
"""
settings = {}
# Try to give anyone using threshold a heads up.
if threshold != 10:
warnings.warn("The 'threshold' kwarg is deprecated, replaced in favor of maxIntensity.")
settings['threshold'] = threshold
settings['radius'] = radius
settings['blur'] = blur
settings['radius'] = radius or 25
settings['gradient'] = gradient
settings['opacity'] = opacity
settings['maxIntensity'] = maxIntensity
settings['dissipating'] = dissipating
settings['minOpacity'] = minOpacity
settings['max'] = max
settings['maxZoom'] = maxZoom
settings = self._process_heatmap_kwargs(settings)

heatmap_points = []
for lat, lng in zip(lats, lngs):
heatmap_points.append((lat, lng))
self.heatmap_points.append((heatmap_points, settings))

self.heatmap_points.append((coordinates_intensity, settings))


def fit_bounds(self, nelat, nelng, swlat, swlng):
self.bounding_box = [nelat, nelng, swlat, swlng]

def _process_heatmap_kwargs(self, settings_dict):
settings_string = ''
settings_string += "heatmap.set('threshold', %d);\n" % settings_dict['threshold']
settings_string += "heatmap.set('radius', %d);\n" % settings_dict['radius']
settings_string += "heatmap.set('maxIntensity', %d);\n" % settings_dict['maxIntensity']
settings_string += "heatmap.set('opacity', %f);\n" % settings_dict['opacity']

dissipation_string = 'true' if settings_dict['dissipating'] else 'false'
settings_string += "heatmap.set('dissipating', %s);\n" % (dissipation_string)
settings_string += "heatmap.setOptions(%s);\n" % settings_dict

gradient = settings_dict['gradient']
if gradient:
gradient_string = "var gradient = [\n"
for r, g, b, a in gradient:
gradient_string += "\t" + "'rgba(%d, %d, %d, %d)',\n" % (r, g, b, a)
gradient_string += '];' + '\n'
gradient_string += "heatmap.set('gradient', gradient);\n"

settings_string += gradient_string

return settings_string

Expand Down Expand Up @@ -264,6 +244,11 @@ def draw(self, htmlfile, img_path=None, header=None, footer=None):
'integrity="sha512-nMMmRyTVoLYqjP9hrbed9S+FzjZHW5gY1TWCHA5ckwXZBadntCNs8kEqAWdrb9O7rxbCaA4lKTIWjDXZxflOcA=="'
'crossorigin=""></script>'
)
f.write(
'<script src="https://cdnjs.cloudflare.com/ajax/libs/leaflet.heat/0.2.0/leaflet-heat.js" '
'integrity="sha256-65UqrlgGoRAnKfKRuriH3eeDrOhZgZo1SCenduc+SGo=" '
'crossorigin="anonymous"></script>'
)
f.write(
'<meta http-equiv="content-type" content="text/html; charset=UTF-8"/>\n')
f.write('<title>Leaflet - llplot </title>\n')
Expand All @@ -281,7 +266,7 @@ def draw(self, htmlfile, img_path=None, header=None, footer=None):
self.write_circles(f)
# self.write_symbols(f)
# self.write_shapes(f)
# self.write_heatmap(f)
self.write_heatmap(f)
# self.write_ground_overlay(f)
self.write_fitbounds(f)
f.write('\t}\n')
Expand Down Expand Up @@ -491,21 +476,24 @@ def write_polygon(self, f, path, settings):
f.write('\n\n')

def write_heatmap(self, f):
for heatmap_points, settings_string in self.heatmap_points:
f.write('var heatmap_points = [\n')
for heatmap_lat, heatmap_lng in heatmap_points:
f.write('new google.maps.LatLng(%f, %f),\n' %
(heatmap_lat, heatmap_lng))
f.write('];\n')
f.write('\n')
f.write('var pointArray = new google.maps.MVCArray(heatmap_points);' + '\n')
f.write('var heatmap;' + '\n')
f.write('heatmap = new google.maps.visualization.HeatmapLayer({' + '\n')
f.write('\n')
f.write('data: pointArray' + '\n')
f.write('});' + '\n')
f.write('heatmap.setMap(map);' + '\n')
f.write(settings_string)
# for heatmap_points, settings_string in self.heatmap_points:

f.write(HEATMAP.format(latlng=self.heatmap_points[0][0],radius=25))
f.write(self.heatmap_points[0][1])
# f.write('var heatmap_points = [\n')
# for heatmap_lat, heatmap_lng in heatmap_points:
# f.write('new google.maps.LatLng(%f, %f),\n' %
# (heatmap_lat, heatmap_lng))
# f.write('];\n')
# f.write('\n')
# f.write('var pointArray = new google.maps.MVCArray(heatmap_points);' + '\n')
# f.write('var heatmap;' + '\n')
# f.write('heatmap = new google.maps.visualization.HeatmapLayer({' + '\n')
# f.write('\n')
# f.write('data: pointArray' + '\n')
# f.write('});' + '\n')
# f.write('heatmap.setMap(map);' + '\n')
# f.write(settings_string)

def write_ground_overlay(self, f):

Expand All @@ -531,34 +519,3 @@ def write_fitbounds(self, f):
self.bounding_box[2], self.bounding_box[3]))
f.write('llMap.fitBounds(bounds);\n')

if __name__ == "__main__":

mymap = LeafletPlotter(37.428, -122.145, 16)
# mymap = GoogleMapPlotter.from_geocode("Stanford University")

mymap.grid(37.42, 37.43, 0.001, -122.15, -122.14, 0.001)
mymap.marker(37.427, -122.145, "yellow")
mymap.marker(37.428, -122.146, "cornflowerblue")
mymap.marker(37.429, -122.144, "k")
lat, lng = mymap.geocode("Stanford University")
mymap.marker(lat, lng, "red")
mymap.circle(37.429, -122.145, 100, "#FF0000", ew=2)
path = [(37.429, 37.428, 37.427, 37.427, 37.427),
(-122.145, -122.145, -122.145, -122.146, -122.146)]
path2 = [[i+.01 for i in path[0]], [i+.02 for i in path[1]]]
path3 = [(37.433302 , 37.431257 , 37.427644 , 37.430303), (-122.14488, -122.133121, -122.137799, -122.148743)]
path4 = [(37.423074, 37.422700, 37.422410, 37.422188, 37.422274, 37.422495, 37.422962, 37.423552, 37.424387, 37.425920, 37.425937),
(-122.150288, -122.149794, -122.148936, -122.148142, -122.146747, -122.14561, -122.144773, -122.143936, -122.142992, -122.147863, -122.145953)]
mymap.plot(path[0], path[1], "plum", edge_width=10)
mymap.plot(path2[0], path2[1], "red")
mymap.polygon(path3[0], path3[1], edge_color="cyan", edge_width=5, face_color="blue", face_alpha=0.1)
mymap.heatmap(path4[0], path4[1], threshold=10, radius=40)
mymap.heatmap(path3[0], path3[1], threshold=10, radius=40, dissipating=False, gradient=[(30,30,30,0), (30,30,30,1), (50, 50, 50, 1)])
mymap.scatter(path4[0], path4[1], c='r', marker=True)
mymap.scatter(path4[0], path4[1], s=90, marker=False, alpha=0.9, symbol='x', c='red', edge_width=4)
# Get more points with:
# http://www.findlatitudeandlongitude.com/click-lat-lng-list/
scatter_path = ([37.424435, 37.424417, 37.424417, 37.424554, 37.424775, 37.425099, 37.425235, 37.425082, 37.424656, 37.423957, 37.422952, 37.421759, 37.420447, 37.419135, 37.417822, 37.417209],
[-122.142048, -122.141275, -122.140503, -122.139688, -122.138872, -122.138078, -122.137241, -122.136405, -122.135568, -122.134731, -122.133894, -122.133057, -122.13222, -122.131383, -122.130557, -122.129999])
mymap.scatter(scatter_path[0], scatter_path[1], c='r', marker=True)
mymap.draw('./mymap.html')