Merge branch 'master' into pyup-update-dask-0.15.4-to-0.16.0

Author: jgerardsimcock

impactlab_tools/__init__.py

@@ -5,7 +5,7 @@
 from __future__ import absolute_import
 
 __author__ = """Climate Impact Lab"""
-__version__ = '0.1.0'
+__version__ = '0.2.0'
 
 _module_imports = (
 )

impactlab_tools/assets/hierid_regions.dbf

@@ -0,0 +1 @@
+GEOGCS["GCS_WGS_1984",DATUM["D_WGS_1984",SPHEROID["WGS_1984",6378137,298.257223563]],PRIMEM["Greenwich",0],UNIT["Degree",0.017453292519943295]]

impactlab_tools/assets/hierid_regions.prj

@@ -0,0 +1,63 @@
+import numpy as np
+
+import matplotlib
+import matplotlib.pyplot as plt
+from mpl_toolkits.basemap import Basemap
+from matplotlib.patches import Polygon
+from matplotlib.collections import PatchCollection
+
+import toolz
+
+
+def get_shape():
+    
+    return 'assets/hierid_regions'
+    
+
+@toolz.memoize
+def prep_polygons(
+        shapepath=None,
+        projection='cyl',
+        **kwargs):
+    
+    if shapepath is None:
+        shapepath = get_shape()
+
+    m = Basemap(projection=projection, llcrnrlat=-90, urcrnrlat=90,\
+            llcrnrlon=-180, urcrnrlon=180, resolution=None, **kwargs)
+    m.readshapefile(shapepath, 'shapes', drawbounds=False)
+    
+    poly = []
+
+    for ii in range(len(m.shapes)):
+        poly.append(Polygon(m.shapes[ii], closed=False))
+        
+    return m, poly
+
+def plot_by_hierid(da, ax=None, clim=None, cmap='jet', **kwargs):
+
+    if ax is None:
+        ax = plt.subplot(111)
+        
+    if clim is None:
+        clim = [da.min(), da.max()]
+        
+    m, poly = prep_polygons()
+
+    hierids = np.array([m.shapes_info[i]['hierid'] for i in range(len(m.shapes))])
+    hierids = hierids[np.in1d(hierids, da.hierid)]
+
+    color = da.sel(hierid=hierids).values
+
+    c = PatchCollection(poly, array=color, cmap=cmap, **kwargs)
+    c.set_clim(clim)     # set the range of colorbar here
+    ax.add_collection(c)
+    ax.set_xlim(-180, 180)
+    ax.set_ylim(-90,90)
+    ax.set_xticks(np.linspace(-180, 180, 7))
+    ax.set_yticks(np.linspace(-90, 90, 7))
+    ax.set_ylabel('Longitude')
+    ax.set_xlabel('Latitude')
+    plt.colorbar(c, ax=ax)
+
+    return ax

impactlab_tools/assets/hierid_regions.shp

@@ -5,6 +5,10 @@ xarray==0.9.6
 h5py==2.7.1
 numba==0.35.0
 blaze==0.10.1
-scipy==0.19.1
+scipy==1.0.0
 bottleneck==1.2.1
 dask==0.16.0
+basemap==1.0.7
+matplotlib==2.1.0
+
+

impactlab_tools/assets/hierid_regions.shx

@@ -2,7 +2,7 @@ Sphinx==1.6.5
 sphinx-rtd-theme==0.2.5b1
 pip==9.0.1
 wheel==0.30.0
-flake8==3.4.1
+flake8==3.5.0
 tox==2.9.1
 coverage==4.4.1
 pytest==3.2.3

impactlab_tools/utils/visualize.py

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.1.0
+current_version = 0.2.0
 commit = True
 tag = True
 

requirements_conda.txt

@@ -40,7 +40,7 @@
 
 setup(
     name='impactlab-tools',
-    version='0.1.0',
+    version='0.2.0',
     description="Python tools for Climate Impact Lab developers",
     long_description=readme,
     author="Climate Impact Lab",