######################################################################################################
# LICENSE
# Copyright (C) 2020 - INPE - NATIONAL INSTITUTE FOR SPACE RESEARCH - BRAZIL
# This program is free software: you can redistribute it and/or modify it under the terms of the GNU 
# General Public License as published by the Free Software Foundation, either version 3 of the License, 
# or (at your option) any later version.
# This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without 
# even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General 
# Public License for more details.
# You should have received a copy of the GNU General Public License along with this program. 
# If not, see http://www.gnu.org/licenses/.
#######################################################################################################

__author__ = "Diego Souza"
__email__ = "[email protected]"
__status__ = "Prototype"

# Required modules

import folium
from folium import plugins
from folium.plugins import Draw
from folium.plugins import MeasureControl
from folium.plugins import MousePosition
from folium.plugins import FloatImage
import pandas as pd
import geopandas as gpd
from shapely.geometry import Point

#================================================================================

# Create the map

m = folium.Map(
    location=[-15, -58],
    zoom_start=5,
    tiles='cartodbdark_matter',
	crs='EPSG3857',
    control_scale=True,
	prefer_canvas=True
)

# Add the full screen button

plugins.Fullscreen(
    position='topright',
    title='Expand me',
    title_cancel='Exit me',
    force_separate_button=True
).add_to(m)


# Add the mouse position coordinates

formatter = "function(num) {return L.Util.formatNum(num, 3) + ' ยบ ';};"
MousePosition(
    position='bottomleft',
    separator=' | ',
    empty_string='NaN',
    lng_first=True,
    num_digits=20,
    prefix='Coordinates:',
    lat_formatter=formatter,
    lng_formatter=formatter,
).add_to(m)


# Add a mini map
minimap = plugins.MiniMap()
m.add_child(minimap)

# Add drawinf controls
draw = Draw()
draw.add_to(m)
m.add_child(MeasureControl())

# Add the lat lon pop up when clicking
m.add_child(folium.LatLngPopup())

#================================================================================

# Add custom logos and legends

image = ('showcast_fire.png')
FloatImage(image, bottom=7, left=1).add_to(m)

image = ('showcast_fire_legend.png')
FloatImage(image, bottom=18, left=1).add_to(m)

#================================================================================

# Add shapefiles

shape = ('ne_50m_admin_0_countries.shp')
countries = gpd.read_file(shape)

shape = ('ne_10m_admin_1_states_provinces.shp')
states = gpd.read_file(shape)

shape = ('ne_10m_coastline.shp')
coastlines = gpd.read_file(shape)

geoPath_countries = countries.geometry.to_json()
geoPath_states = states.geometry.to_json()
geoPath_coastlines = coastlines.geometry.to_json()

folium.GeoJson(
    geoPath_states,
	name='States',
	show=False,
    style_function=lambda feature: {
        'fillColor': '#00000000',
        'color': 'gold',
        'weight': 1.0,
    }
).add_to(m)

folium.GeoJson(
    geoPath_countries,
	name='Countries',
	show=False,
    style_function=lambda feature: {
        'fillColor': '#00000000',
        'color': 'black',
        'weight': 1.0,
    }
).add_to(m)

folium.GeoJson(
    geoPath_coastlines,
	name='Coastlnes',
	show=False,
    style_function=lambda feature: {
        'fillColor': '#00000000',
        'color': 'red',
        'weight': 1.0,
    }
).add_to(m)

#================================================================================

# Read the Fire / Hot Spot Shapefile and plot it as points over the map

inpe_mvf = gpd.read_file('INPE_MVF_202009050000.shp')
geometry = [Point(xy) for xy in zip(inpe_mvf['longitude'], inpe_mvf['latitude'])]
df = gpd.GeoDataFrame(inpe_mvf, geometry = geometry)
fires = folium.map.FeatureGroup()

latitudes = list(df.latitude)
longitudes = list(df.longitude)
labels = list(df.cod_sat)

dates = list(df.data_pas)
times = list(df.id_pas)

countries = list(df.name_0)
states = list(df.name_1)
cities = list(df.name_2)

for lat, lng, label, date, time, country, state, city in zip(latitudes, longitudes, labels, dates, times, countries, states, cities):
  if label == 'GOES-16':
    popup_text = "<b>Satellite:</b> " + label + '<br>' + '<b>Date:</b> ' + date + '<br>' + '<b>Time:</b> ' + time[-11:-9] + ':' + time[-9:-7] + ' UTC' + '<br>' + '<b>Country:</b> ' + country + '<br>' + '<b>State / Province:</b> ' + state + '<br>' + '<b>City:</b> ' + city
    folium.Circle(
       [lat, lng],
       radius=10,
	   fill=True,
	   #line_color='white',
       color='white',
       fill_color='yellow',
	   weight = 0.2,
       fill_opacity=1,
       ).add_child(folium.Popup(popup_text, max_width=200,min_width=200)).add_to(m)

for lat, lng, label, date, time, country, state, city in zip(latitudes, longitudes, labels, dates, times, countries, states, cities):
  if label == 'METOP-B':
    popup_text = "<b>Satellite:</b> " + label + '<br>' + '<b>Date:</b> ' + date + '<br>' + '<b>Country:</b> ' + country + '<br>' + '<b>State / Province:</b> ' + state + '<br>' + '<b>City:</b> ' + city#+ '<br>' + '<b>Time:</b> ' + time[-11:-9] + ':' + time[-9:-7] + ' UTC'
    folium.Circle(
       [lat, lng],
       radius=10,
	   fill=True,
	   #line_color='white',
       color='white',
       fill_color='blue',
	   weight = 0.2,
       fill_opacity=1,
       ).add_child(folium.Popup(popup_text, max_width=200,min_width=200)).add_to(m)

for lat, lng, label, date, time, country, state, city in zip(latitudes, longitudes, labels, dates, times, countries, states, cities):
  if label == 'NOAA-18':
    popup_text = "<b>Satellite:</b> " + label + '<br>' + '<b>Date:</b> ' + date + '<br>' + '<b>Country:</b> ' + country + '<br>' + '<b>State / Province:</b> ' + state + '<br>' + '<b>City:</b> ' + city#+ '<br>' + '<b>Time:</b> ' + time[-11:-9] + ':' + time[-9:-7] + ' UTC'
    folium.Circle(
       [lat, lng],
       radius=10,
	   fill=True,
	   #line_color='white',
       color='white',
       fill_color='red',
	   weight = 0.2,
       fill_opacity=1,
       ).add_child(folium.Popup(popup_text, max_width=200,min_width=200)).add_to(m)

for lat, lng, label, date, time, country, state, city in zip(latitudes, longitudes, labels, dates, times, countries, states, cities):
  if label == 'NOAA-19':
    popup_text = "<b>Satellite:</b> " + label + '<br>' + '<b>Date:</b> ' + date + '<br>' + '<b>Country:</b> ' + country + '<br>' + '<b>State / Province:</b> ' + state + '<br>' + '<b>City:</b> ' + city#+ '<br>' + '<b>Time:</b> ' + time[-11:-9] + ':' + time[-9:-7] + ' UTC'
    folium.Circle(
       [lat, lng],
       radius=10,
	   fill=True,
	   #line_color='white',
       color='white',
       fill_color='purple',
	   weight = 0.2,
       fill_opacity=1,
       ).add_child(folium.Popup(popup_text, max_width=200,min_width=200)).add_to(m)

for lat, lng, label, date, time, country, state, city in zip(latitudes, longitudes, labels, dates, times, countries, states, cities):
  if label == 'NPP':
    popup_text = "<b>Satellite:</b> " + label + '<br>' + '<b>Date:</b> ' + date + '<br>' + '<b>Country:</b> ' + country + '<br>' + '<b>State / Province:</b> ' + state + '<br>' + '<b>City:</b> ' + city#+ '<br>' + '<b>Time:</b> ' + time[-11:-9] + ':' + time[-9:-7] + ' UTC'
    folium.Circle(
       [lat, lng],
       radius=10,
	   fill=True,
	   #line_color='white',
       color='white',
       fill_color='green',
	   weight = 0.2,
       fill_opacity=1,
       ).add_child(folium.Popup(popup_text, max_width=200,min_width=200)).add_to(m)
	   
#================================================================================

# Add the background tile options

folium.TileLayer(
    name="Esri World Imagery",
    tiles='https://server.arcgisonline.com/ArcGIS/rest/services/World_Imagery/MapServer/tile/{z}/{y}/{x}',
    attr='Tiles &copy; Esri &mdash; Source: Esri, i-cubed, USDA, USGS, AEX, GeoEye, Getmapping, Aerogrid, IGN, IGP, UPR-EGP, and the GIS User Community',
).add_to(m)

folium.TileLayer('openstreetmap').add_to(m)
folium.TileLayer('Stamen Terrain').add_to(m)
folium.TileLayer('Stamen Toner').add_to(m)
folium.TileLayer('Stamen Water Color').add_to(m)
folium.TileLayer('cartodbpositron').add_to(m)
folium.TileLayer('cartodbdark_matter').add_to(m)
folium.LayerControl().add_to(m)

# Save the map as an HTML file
m.save('map.html')