#!/usr/bin/env python
# coding: utf-8

# # SlideRule ATL03 Subsetting: Interactive Tutorial
# 
# SlideRule is an on-demand science data processing service that runs in on Amazon Web Services and responds to REST API calls to process and return science results.  SlideRule was designed to enable researchers and other data systems to have low-latency access to custom-generated, high-level, analysis-ready data products using processing parameters supplied at the time of the request. 
# 
# [Documentation for using SlideRule](https://slideruleearth.io/rtd) is available from the [project website](https://slideruleearth.io) 
# 
# ### Background
# SlideRule can subset ATL03 geolocated photon height data _on-demand_ and calculate photon classifications to suit different needs.
# 
# ### Jupyter and SlideRule
# [Jupyter widgets](https://ipywidgets.readthedocs.io) are used to set parameters for the SlideRule API.  
# 
# Regions of interest for submitting to SlideRule are drawn on a [ipyleaflet](https://ipyleaflet.readthedocs.io) map.  
# The results from SlideRule can be displayed on the interactive [ipyleaflet](https://ipyleaflet.readthedocs.io) map along with additional contextual layers.

# #### Load necessary packages

# In[ ]:


import warnings
# turn off warnings for demo
warnings.filterwarnings('ignore')# autoreload

from sliderule import icesat2, ipysliderule, sliderule, io, earthdata
import geopandas
import logging

get_ipython().run_line_magic('load_ext', 'autoreload')
get_ipython().run_line_magic('autoreload', '2')


# ### Initiate SlideRule API
# - Sets the URL for accessing the SlideRule service
# - Builds a table of servers available for processing data

# In[ ]:


# set the url for the sliderule service
# set the logging level
icesat2.init("slideruleearth.io", loglevel=logging.WARNING)


# ### Set options for making science data processing requests to SlideRule
# 
# SlideRule can provide different sources for photon classification. This is useful for example in cases using the photon returns to estimate ocean or lake bathymetry, vegetated canopy heights, or ground position in the presence of a vegetated canopy. 
# - ATL03 photon confidence values, based on algorithm-specific classification types for land, ocean, sea-ice, land-ice, or inland water
# - [ATL08 Land and Vegetation Height product](https://nsidc.org/data/atl08) photon classification
# - Experimental YAPC (Yet Another Photon Classification) photon-density-based classification

# In[ ]:


# display widgets for setting SlideRule parameters
SRwidgets = ipysliderule.widgets()
SRwidgets.set_atl03_defaults()
SRwidgets.VBox(SRwidgets.atl03())


# ### Interactive Mapping with Leaflet
# 
# Interactive maps within the SlideRule python API are built upon [ipyleaflet](https://ipyleaflet.readthedocs.io).
# 
# #### Leaflet Basemaps and Layers
# 
# There are 3 projections available within SlideRule for mapping ([Global](https://epsg.io/3857), [North](https://epsg.io/5936) and [South](https://epsg.io/3031)).  There are also contextual layers available for each projection.
# 
# <table>
#   <tbody>
#     <tr>
#       <th align='center' max-width="30%"><a href="https://epsg.io/3857">Global (Web Mercator, EPSG:3857)</a></th>
#       <th align='center' max-width="30%"><a href="https://epsg.io/5936">North (Alaska Polar Stereographic, EPSG:5936)</a></th>
#       <th align='center' max-width="30%"><a href="https://epsg.io/3031">South (Antarctic Polar Stereographic, EPSG:3031)</a></th>
#     </tr>
#     <tr>
#       <td align='left' valign='top' width="30%">
#         <ul style="line-height: 1.5em">
#             <li><a href="https://apps.nationalmap.gov/3depdem/">USGS 3DEP Hillshade</a></li>
#             <li><a href="https://asterweb.jpl.nasa.gov/gdem.asp">ASTER GDEM Hillshade</a></li>
#             <li><a href="https://www.arcgis.com/home/item.html?id=10df2279f9684e4a9f6a7f08febac2a9">ESRI Imagery</a></li>
#             <li><a href="http://glims.colorado.edu/glacierdata/">Global Land Ice Measurements from Space (GLIMS)</a></li>
#             <li><a href="https://www.glims.org/RGI/">Randolph Glacier Inventory (RGI)</a></li>
#         </ul>
#        </td>
#        <td align='left' valign='top' width="30%">
#         <ul style="line-height: 1.5em">
#             <li><a href="http://goto.arcgisonline.com/maps/Arctic_Imagery">ESRI Imagery</a></li>
#             <li><a href="https://www.pgc.umn.edu/data/arcticdem">ArcticDEM</a></li>
#         </ul>
#        </td>
#        <td align='left' valign='top' width="30%">
#         <ul style="line-height: 1.5em">
#             <li><a href="https://lima.usgs.gov/">Landsat Image Mosaic of Antarctica (LIMA)</a></li>
#             <li><a href="https://nsidc.org/data/nsidc-0280">MODIS Mosaic of Antarctica (MOA)</a></li>
#             <li><a href="https://nsidc.org/data/NSIDC-0103">Radarsat Antarctic Mapping Project (RAMP)</a></li>
#             <li><a href="https://www.pgc.umn.edu/data/rema">Reference Elevation Model of Antarctica (REMA)</a></li>
#         </ul>
#       </td>
#     </tr>
#   </tbody>
# </table>
# 
# In addition, most [xyzservice providers](https://xyzservices.readthedocs.io/en/stable/introduction.html) can be added as contextual layers to the global Web Mercator maps

# In[ ]:


SRwidgets.VBox([
    SRwidgets.projection,
    SRwidgets.layers,
    SRwidgets.raster_functions
])


# ### Select regions of interest for submitting to SlideRule
# 
# Here, we create polygons or bounding boxes for our regions of interest.  
# This map is also our viewer for inspecting our SlideRule ICESat-2 data returns.

# In[ ]:


# create ipyleaflet map in specified projection
m = ipysliderule.leaflet(SRwidgets.projection.value)
m.map


# In[ ]:


m.add_layer(
    layers=SRwidgets.layers.value,
    rendering_rule=SRwidgets.rendering_rule
)


# ### Inspect list of available granules
# - SlideRule will query the [NASA Common Metadata Repository (CMR)](https://cmr.earthdata.nasa.gov/) for ATL03 data within our region of interest

# In[ ]:


get_ipython().run_cell_magic('time', '', '# sliderule asset and data release\nasset = SRwidgets.asset.value\nrelease = SRwidgets.release.value\n# find granule for each region of interest\ngranules_list = []\n# for each region of interest\nsliderule.logger.warning(\'No valid regions to run\') if not m.regions else None\nfor poly in m.regions:\n    granules = earthdata.cmr(short_name="ATL03",\n        polygon=poly,\n        time_start=SRwidgets.time_start,\n        time_end=SRwidgets.time_end,\n        version=release)\n    granules_list.extend(granules)\n# inspect granules list\nprint(f\'Available granules: {len(granules_list)}\')\n')


# ### Transmit requests to SlideRule
# - When using the `icesat2` asset, the ICESat-2 ATL03 data are then accessed from the NSIDC AWS s3 bucket in `us-west-2`
# - The ATL03 granules is spatially subset within SlideRule to our exact region of interest
# - Photon classification parameters can then be extracted or calculated for our ATL03 data
# - The completed data is streamed concurrently back and combined into a geopandas GeoDataFrame within the Python client

# In[ ]:


# build sliderule parameters using latest values from widget
parms = SRwidgets.build_atl03()

# clear existing geodataframe results
elevations = [sliderule.emptyframe()]

# for each region of interest
for poly in m.regions:
    # add polygon from map to sliderule parameters
    parms["poly"] = poly
    # make the request to the SlideRule (ATL03-SR) endpoint
    # and pass it the request parameters to request ATL03 Data
    elevations.append(icesat2.atl03sp(parms, resources=granules_list))

gdf = geopandas.pd.concat(elevations)


# ### Review GeoDataFrame output
# Can inspect the columns, number of returns and returns at the top of the GeoDataFrame.
# 
# See the [ICESat-2 documentation](https://slideruleearth.io/rtd/user_guide/ICESat-2.html#elevations) for descriptions of each column

# In[ ]:


print(f'Returned {gdf.shape[0]} records')
gdf.head()


# ### Add GeoDataFrame to map
# 
# For stability of the leaflet map, SlideRule will as a default limit the plot to have up to 10000 points from the GeoDataFrame
# 
# GeoDataFrames can be plotted in any available [matplotlib colormap](https://matplotlib.org/stable/tutorials/colors/colormaps.html)

# In[ ]:


SRwidgets.VBox([
    SRwidgets.variable,
    SRwidgets.cmap,
    SRwidgets.reverse,
])


# In[ ]:


get_ipython().run_line_magic('matplotlib', 'inline')
# ATL03 fields for hover tooltip
fields = gdf.leaflet.default_atl03_fields()
gdf.leaflet.GeoData(m.map, column_name=SRwidgets.variable.value, cmap=SRwidgets.colormap,
    max_plot_points=10000, tooltip=True, colorbar=True, fields=fields)
# install handlers and callbacks
gdf.leaflet.set_observables(SRwidgets)
gdf.leaflet.add_selected_callback(SRwidgets.atl03_click_handler)
m.add_region_callback(gdf.leaflet.handle_region)


# ### Create plots for a single track
# - scatter: Will plot data returned by SlideRule for a single RGT, ground track and cycle

# In[ ]:


SRwidgets.VBox([
    SRwidgets.plot_classification,
    SRwidgets.rgt,
    SRwidgets.ground_track,
    SRwidgets.cycle,
])


# In[ ]:


get_ipython().run_line_magic('matplotlib', 'widget')
gdf.icesat2.plot(data_type='atl03', kind='scatter', title='Photon Cloud',
    cmap=SRwidgets.colormap, legend=True, legend_frameon=True,
    classification=SRwidgets.plot_classification.value,
    segments=False, **SRwidgets.plot_kwargs)


# ### Save GeoDataFrame to output file

# In[ ]:


display(SRwidgets.filesaver)


# In[ ]:


# append sliderule api version to attributes
version = sliderule.get_version()
parms['version'] = version['icesat2']['version']
parms['commit'] = version['icesat2']['commit']
# save to file in format (HDF5 or netCDF)
io.to_file(gdf, SRwidgets.file,
    format=SRwidgets.format,
    parameters=parms,
    regions=m.regions,
    verbose=True)


# ### Read GeoDataFrame from input file

# In[ ]:


display(SRwidgets.fileloader)


# In[ ]:


# read from file in format (HDF5 or netCDF)
gdf,parms,regions = io.from_file(SRwidgets.file,
    format=SRwidgets.format,
    return_parameters=True,
    return_regions=True)


# ### Review GeoDataFrame input from file

# In[ ]:


gdf.head()


# ### Set parameters and add saved regions to map

# In[ ]:


SRwidgets.set_values(parms)
m.add_region(regions)


# In[ ]: