import os
last_modified = None
if os.name == "posix":
    last_modified = !stat -f\
                    "# This notebook was last updated: %Sm"\
                     Spaghetti_Pointpatterns_Empirical.ipynb
elif os.name == "nt":
    last_modified = !for %a in (Spaghetti_Pointpatterns_Empirical.ipynb)\
                    do echo # This notebook was last updated: %~ta
    
if last_modified:
    get_ipython().set_next_input(last_modified[-1])

# This notebook was last updated: Dec  9 14:23:58 2018

from pysal.explore import spaghetti as spgh
from pysal.lib import examples
import geopandas as gpd
import matplotlib.pyplot as plt
import matplotlib.lines as mlines
from shapely.geometry import Point, LineString

%matplotlib inline

__author__ = "James Gaboardi <jgaboardi@gmail.com>"

ntw = spgh.Network(in_data=examples.get_path('streets.shp'))

# Crimes with attributes
ntw.snapobservations(examples.get_path('crimes.shp'),
                     'crimes',
                     attribute=True)

# Schools without attributes
ntw.snapobservations(examples.get_path('schools.shp'),
                     'schools',
                     attribute=False)

schools_df = spgh.element_as_gdf(ntw,
                                 pp_name='schools',
                                 snapped=False)

snapped_schools_df = spgh.element_as_gdf(ntw,
                                         pp_name='schools',
                                         snapped=True)

crimes_df = spgh.element_as_gdf(ntw,
                                pp_name='crimes',
                                snapped=False)

snapped_crimes_df = spgh.element_as_gdf(ntw,
                                        pp_name='crimes',
                                        snapped=True)

# network nodes and edges
vertices_df, arcs_df = spgh.element_as_gdf(ntw,
                                           vertices=True,
                                           arcs=True)

# legend patches
arcs = mlines.Line2D([], [], color='k', label='Network Arcs', alpha=.5)
vtxs = mlines.Line2D([], [], color='k', linewidth=0, markersize=2.5,
                     marker='o', label='Network Vertices', alpha=1)
schl = mlines.Line2D([], [], color='k', linewidth=0, markersize=25,
                     marker='X', label='School Locations', alpha=1)
snp_schl = mlines.Line2D([], [], color='k', linewidth=0, markersize=12,
                         marker='o', label='Snapped Schools', alpha=1)
crme = mlines.Line2D([], [], color='r', linewidth=0, markersize=7,
                     marker='x', label='Crime Locations', alpha=.75)
snp_crme = mlines.Line2D([], [], color='r', linewidth=0, markersize=3,
                         marker='o', label='Snapped Crimes', alpha=.75)

patches = [arcs, vtxs, schl, snp_schl, crme, snp_crme]

# plot figure
base = arcs_df.plot(color='k', alpha=.25, figsize=(12,12), zorder=0)
vertices_df.plot(ax=base, color='k', markersize=5, alpha=1)

crimes_df.plot(ax=base, color='r', marker='x',
               markersize=50, alpha=.5, zorder=1)
snapped_crimes_df.plot(ax=base, color='r',
                       markersize=20, alpha=.5, zorder=1)

schools_df.plot(ax=base, cmap='tab20', column='id', marker='X',
                markersize=500, alpha=.5, zorder=2)
snapped_schools_df.plot(ax=base,cmap='tab20', column='id',
                        markersize=200, alpha=.5, zorder=2)

# add legend
plt.legend(handles=patches, fancybox=True, framealpha=0.8,
           scatterpoints=1, fontsize="xx-large", bbox_to_anchor=(1.04, .6))