Set get_bounds method

Author: Martin Journois

folium/element.py

@@ -13,7 +13,7 @@
 import base64
 
 from .six import urlopen, text_type, binary_type
-from .utilities import _camelify, _parse_size
+from .utilities import _camelify, _parse_size, none_min, none_max
 
 
 ENV = Environment(loader=PackageLoader('folium', 'templates'))
@@ -40,6 +40,32 @@ def __init__(self, template=None, template_name=None):
     def get_name(self):
         return _camelify(self._name) + '_' + self._id
 
+    def _get_self_bounds(self):
+        """Computes the bounds of the object itself (not including it's children)
+        in the form [[lat_min, lon_min], [lat_max, lon_max]]
+        """
+        return [[None,None],[None,None]]
+
+    def get_bounds(self):
+        """Computes the bounds of the object and all it's children
+        in the form [[lat_min, lon_min], [lat_max, lon_max]].
+        """
+        bounds = self._get_self_bounds()
+
+        for child in self._children.values():
+            child_bounds = child.get_bounds()
+            bounds = [
+                [
+                    none_min(bounds[0][0], child_bounds[0][0]),
+                    none_min(bounds[0][1], child_bounds[0][1]),
+                    ],
+                [
+                    none_max(bounds[1][0], child_bounds[1][0]),
+                    none_max(bounds[1][1], child_bounds[1][1]),
+                    ],
+                ]
+        return bounds
+
     def add_children(self, child, name=None, index=None):
         """Add a children."""
         if name is None:

folium/features.py

@@ -126,7 +126,6 @@ def render(self, **kwargs):
             JavascriptLink("https://cdnjs.cloudflare.com/ajax/libs/leaflet-dvf/0.2/leaflet-dvf.markers.min.js"),  # noqa
             name='dvf_js')
 
-
 class Vega(Element):
     def __init__(self, data, width=None, height=None,
                  left="0%", top="0%", position='relative'):
@@ -277,6 +276,12 @@ def style_data(self):
                 self.style_function(feature))
         return json.dumps(self.data)
 
+    def _get_self_bounds(self):
+        """Computes the bounds of the object itself (not including it's children)
+        in the form [[lat_min, lon_min], [lat_max, lon_max]]
+        """
+        raise NotImplementedError
+
 class TopoJson(MacroElement):
     def __init__(self, data, object_path):
         """
@@ -315,6 +320,13 @@ def render(self, **kwargs):
             JavascriptLink("https://cdnjs.cloudflare.com/ajax/libs/topojson/1.6.9/topojson.min.js"),  # noqa
             name='topojson')
 
+    def _get_self_bounds(self):
+        """Computes the bounds of the object itself (not including it's children)
+        in the form [[lat_min, lon_min], [lat_max, lon_max]]
+        """
+        raise NotImplementedError
+        return [[self.location[0],self.location[1]],[self.location[0],self.location[1]]]
+
 class ColorScale(MacroElement):
     def __init__(self, color_domain, color_code, caption=""):
         """
@@ -344,7 +356,6 @@ def render(self, **kwargs):
             JavascriptLink("https://cdnjs.cloudflare.com/ajax/libs/d3/3.5.5/d3.min.js"),  # noqa
             name='d3')
 
-
 class MarkerCluster(Layer):
     """Adds a MarkerCluster layer on the map."""
     def __init__(self, overlay=True, control=True):
@@ -569,6 +580,11 @@ def __init__(self, locations, color=None, weight=None,
             {% endmacro %}
             """)  # noqa
 
+    def _get_self_bounds(self):
+        """Computes the bounds of the object itself (not including it's children)
+        in the form [[lat_min, lon_min], [lat_max, lon_max]]
+        """
+        raise NotImplementedError
 
 class MultiPolyLine(MacroElement):
     def __init__(self, locations, color=None, weight=None,
@@ -616,6 +632,11 @@ def __init__(self, locations, color=None, weight=None,
                 {{this._parent.get_name()}}.addLayer({{this.get_name()}});
             {% endmacro %}
             """)  # noqa
+    def _get_self_bounds(self):
+        """Computes the bounds of the object itself (not including it's children)
+        in the form [[lat_min, lon_min], [lat_max, lon_max]]
+        """
+        raise NotImplementedError
 
 class CustomIcon(Icon):
     def __init__(self, icon_image, icon_size=None, icon_anchor=None,

folium/map.py

@@ -16,7 +16,6 @@
 
 from .element import Element, Figure, MacroElement, Html
 
-
 class Map(MacroElement):
     def __init__(self, location=None, width='100%', height='100%',
                  left="0%", top="0%", position='relative',
@@ -168,6 +167,12 @@ def add_tile_layer(self, tiles='OpenStreetMap', name=None,
                                detect_retina=detect_retina)
         self.add_children(tile_layer, name=tile_layer.tile_name)
 
+    def _get_self_bounds(self):
+        """Computes the bounds of the object itself (not including it's children)
+        in the form [[lat_min, lon_min], [lat_max, lon_max]]
+        """
+        return [[self.location[0],self.location[1]],[self.location[0],self.location[1]]]
+
 class Layer(MacroElement):
     """An abstract class for everything that is a Layer on the map.
     It will be used to define whether an object will be included in LayerControls.
@@ -243,7 +248,6 @@ def __init__(self, tiles='OpenStreetMap', name=None,
         {% endmacro %}
         """)
 
-
 class FeatureGroup(Layer):
     def __init__(self, name=None, overlay=True, control=True):
         """
@@ -273,7 +277,6 @@ def __init__(self, name=None, overlay=True, control=True):
         {% endmacro %}
         """)
 
-
 class LayerControl(MacroElement):
     """Adds a layer control to the map."""
     def __init__(self):
@@ -374,7 +377,6 @@ def __init__(self, color='blue', icon_color='white', icon='info-sign',
             {% endmacro %}
             """)
 
-
 class Marker(MacroElement):
     def __init__(self, location, popup=None, icon=None):
         """Create a simple stock Leaflet marker on the map, with optional
@@ -421,6 +423,11 @@ def __init__(self, location, popup=None, icon=None):
             {% endmacro %}
             """)
 
+    def _get_self_bounds(self):
+        """Computes the bounds of the object itself (not including it's children)
+        in the form [[lat_min, lon_min], [lat_max, lon_max]]
+        """
+        return [[self.location[0],self.location[1]],[self.location[0],self.location[1]]]
 
 class Popup(Element):
     def __init__(self, html=None, max_width=300):
@@ -469,7 +476,6 @@ def render(self, **kwargs):
             self._template.render(this=self, kwargs=kwargs)),
             name=self.get_name())
 
-
 class FitBounds(MacroElement):
     def __init__(self, bounds, padding_top_left=None,
                  padding_bottom_right=None, padding=None, max_zoom=None):
@@ -491,7 +497,6 @@ def __init__(self, bounds, padding_top_left=None,
             the same value.
         max_zoom: int, default None
             Maximum zoom to be used.
-
         """
         super(FitBounds, self).__init__()
         self._name = 'FitBounds'

folium/utilities.py

@@ -428,3 +428,19 @@ def _locations_tolist(x):
         return list(map(_locations_tolist, x))
     else:
         return x
+
+def none_min(x,y):
+    if x is None:
+        return y
+    elif y is None:
+        return x
+    else:
+        return min(x,y)
+
+def none_max(x,y):
+    if x is None:
+        return y
+    elif y is None:
+        return x
+    else:
+        return max(x,y)