#!/usr/bin/python

import numpy as np
import cgi, cgitb
cgitb.enable()
import gzip, os, sys, datetime, re, time, urllib, collections
import MySQLdb as mysqldb

## Set timezone first

os.environ['TZ'] = 'Etc/Greenwich'
time.tzset()

## Get arguments from storage or use default if any errors are thrown

form = cgi.FieldStorage()
try:
	net = int(form["net"].value)
except:
	net = 1
try:
	zoom = int(form["zoom"].value)
except:
	zoom = 9
try:
	latcen = float(form["latcen"].value)
except:
	latcen = 40.76623
try:
	loncen = float(form["loncen"].value)
except:
	loncen = -111.84755
try:
	limitstring = form["limit"].value
except:
	limitstring = 'None'

if(len(limitstring) == 0):
	limitstring = 'None'
	
if(net > 0):
	latn = latcen + 3
	lats = latcen - 3
	lone = loncen + 5
	lonw = loncen - 5
else:
	latn = latcen + 1.25
	lats = latcen - 1.25
	lone = loncen + 1.5
	lonw = loncen - 1.5

templimit_active = 0
templimit_low = -1000.0
templimit_high = 1000.0
relhlimit_active = 0
relhlimit_low = -1000.0
relhlimit_high = 1000.0
skntlimit_active = 0
skntlimit_low = -1000.0
skntlimit_high = 1000.0
gustlimit_active = 0
gustlimit_low = -1000.0
gustlimit_high = 1000.0
	
if(limitstring != 'None'):
	if(',' in limitstring):
		limitlist = limitstring.split(',')
	else:
		limitlist = [limitstring]
	for limittype in limitlist:
		if(limittype[0] == 'T'):
			templimit_active = 1
			if(limittype[1] == 'G'):
				try:
					templimit_low = float(limittype[2:])
				except:
					skip = 1
			if(limittype[1] == 'L'):
				try:
					templimit_high = float(limittype[2:])
				except:
					skip = 1
		if(limittype[0] == 'R'):
			relhlimit_active = 1
			if(limittype[1] == 'G'):
				try:
					relhlimit_low = float(limittype[2:])
				except:
					skip = 1
			if(limittype[1] == 'L'):
				try:
					relhlimit_high = float(limittype[2:])
				except:
					skip = 1
		if(limittype[0] == 'S'):
			skntlimit_active = 1
			if(limittype[1] == 'G'):
				try:
					skntlimit_low = float(limittype[2:])*0.868976
				except:
					skip = 1
			if(limittype[1] == 'L'):
				try:
					skntlimit_high = float(limittype[2:])*0.868976
				except:
					skip = 1
		if(limittype[0] == 'G'):
			gustlimit_active = 1
			if(limittype[1] == 'G'):
				try:
					gustlimit_low = float(limittype[2:])*0.868976
				except:
					skip = 1
			if(limittype[1] == 'L'):
				try:
					gustlimit_high = float(limittype[2:])*0.868976
				except:
					skip = 1


## Dictionary of variable options and what what color lists to use...

allvarlist = '3,5,6,7,8,14,15,35,45,53,60,71,73';
dbvars = {
3: 'TMPF',
5: 'RELH',
6: 'SKNT',
7: 'DRCT',
8: 'GUST',
14: 'SOLR',
35: 'QFLG',
15: 'TSOI',
45: 'TRD',
53: 'VSBY',
60: 'GUST',
71: 'VOLT',
4000: 'CDRT',
4001: 'DATTIM',
4002: 'SKNTINT',
4003: 'DWPF'
}

networkoptions = {
0: 'All Networks'
}

## Determine time variables!

minback = 90
curdattim = datetime.datetime.strftime(datetime.datetime.utcnow(),'%Y%m%d%H%M')
nowyr = datetime.datetime.strftime(datetime.datetime.utcnow(),'%Y')
olddattim = datetime.datetime.strftime(datetime.datetime.utcnow()-datetime.timedelta(minutes=minback),'%Y%m%d%H%M')

curtimstr = datetime.datetime.strftime(datetime.datetime.utcnow(),'%Y-%m-%d %H:%MZ')

carddrctlist = ['N','NNE','NE','ENE','E','ESE','SE','SSE','S','SSW','SW','WSW','W','WNW','NW','NNW','N']
carddrctmax = [11.25,33.75,56.25,78.75,101.25,123.75,146.25,168.75,191.25,213.75,236.25,258.75,281.25,303.75,326.25,348.75,360.01]
carddrctmin = [ 0.00,11.25,33.75,56.25, 78.75,101.25,123.75,146.25,168.75,191.25,213.75,236.25,258.75,281.25,303.75,326.25,348.75]
	
## Connect to database and get data!
	
dbconnection = None
try:
	stninfoids = []
	dbconnection = mysqldb.connect('meso2.chpc.utah.edu','metwww','wind','mesobest')
	dbcur = dbconnection.cursor()

	dbquerynets = 'SELECT ID,MNAME FROM MNET WHERE ID<1000'
	dbcur.execute(dbquerynets)
	mesonets = dbcur.fetchall()
	for netrow in mesonets:
		networkoptions[netrow[0]] = netrow[1]

	if(net > 0):
		dbquerymeta = 'SELECT ID,STID,LATITUDE,LONGITUDE,MNET_ID FROM STNINFO WHERE LATITUDE > '+str(lats)+' AND LATITUDE < '+str(latn)+' AND LONGITUDE > '+str(lonw)+' AND LONGITUDE < '+str(lone)+' AND MNET_ID ='+str(net)
	else:
		dbquerymeta = 'SELECT ID,STID,LATITUDE,LONGITUDE,MNET_ID FROM STNINFO WHERE LATITUDE > '+str(lats)+' AND LATITUDE < '+str(latn)+' AND LONGITUDE > '+str(lonw)+' AND LONGITUDE < '+str(lone)+' AND MNET_ID < 1000'		
	dbcur.execute(dbquerymeta)
	metadata = dbcur.fetchall()

	dbquerydata = 'SELECT STNINFO_ID,DATTIM,VALUE,VARIABLE_ID FROM DATA_LATEST WHERE DATTIM >= '+str(olddattim)+' AND DATTIM <= '+str(curdattim)+' AND VARIABLE_ID in ('+allvarlist+') ORDER BY VARIABLE_ID'
	dbcur.execute(dbquerydata)
	obsdata = dbcur.fetchall()
	
	dbconnection.close()
except mysqldb.Error, e:
	skip = 1

## Now organize lists!

# print "Content-Type: text/html\n"

stninfoidlist = []
latdict = {}
londict = {}
mnetdict = {}
stiddict = {}
showstndict = {}

allstninfoidlist = []
allstidlist = []
alllatlist = []
alllonlist = []
allmnetlist = []
allshowlist = []
obsdatalist = {}
stndattims = {}
obsfloattimelist = {}

obsdatalist = {}
stndattims = {}

for metarow in metadata:
	stninfoidlist.append(metarow[0])
	stiddict[metarow[0]] = metarow[1]
	latdict[metarow[0]] = metarow[2]
	londict[metarow[0]] = metarow[3]
	mnetdict[metarow[0]] = metarow[4]
	if(limitstring != 'None'):
		showstndict[metarow[0]] = 0
	else:
		showstndict[metarow[0]] = 1

if(len(obsdata) > 0):
	for ob in obsdata:
		obval = ob[2]
		if(ob[3] == 3 and templimit_active == 1):
			if(obval >= templimit_low and obval <= templimit_high):
				showstndict[ob[0]] = 1
		if(ob[3] == 5 and relhlimit_active == 1):
			if(obval >= relhlimit_low and obval <= relhlimit_high):
				showstndict[ob[0]] = 1
		if(ob[3] == 6 and skntlimit_active == 1):
			if(obval >= skntlimit_low and obval <= skntlimit_high):
				showstndict[ob[0]] = 1
		if(ob[3] == 8 and gustlimit_active == 1):
			if(obval >= gustlimit_low and obval <= gustlimit_high):
				showstndict[ob[0]] = 1
		if(ob[3] == 73):
			stndattims[ob[0]] = ob[1]
			if(olddattim >= ob[1] and ob[0] in stninfoidlist):
				allstninfoidlist.append(ob[0])
				allstidlist.append(stiddict[ob[0]])
				alllatlist.append(latdict[ob[0]])
				alllonlist.append(londict[ob[0]])
				allmnetlist.append(mnetdict[ob[0]])
				allshowlist.append(showstndict[ob[0]])					
	for key in dbvars:
		obsdatalist[dbvars[key]] = ['NA'] * len(allstninfoidlist)
	for ob in obsdata:
		if(ob[0] in allstninfoidlist and ob[0] in stndattims and ob[3] in dbvars and ob[1] == stndattims[ob[0]]):
			mylistindex = allstninfoidlist.index(ob[0])
			obval = ob[2]			
			obsdatalist['DATTIM'][mylistindex] = str(ob[1])[0:8]+'/'+str(ob[1])[8:12]+'Z'
			if(ob[3] == 8):
				obsfloattimelist[mylistindex] = float(ob[1])
			if(ob[3] == 6):
				if(obval > 0):
					if(obval >= 5):
						skntint = ("%d" % ((int(obval)/5)*5))
					else:
						skntint = '1'
				else:
					if(obval == 0):
						skntint = '0'
					else:
						skntint = 'v'
				obsdatalist['SKNTINT'][mylistindex] = skntint
			if(ob[3] == 6 or ob[3] == 8 or ob[3] == 60):
				obval = obval * 1.15077945
			if(ob[3] == 7):
				for k in range(0,len(carddrctlist)):
					if(obval >= carddrctmin[k] and obval < carddrctmax[k]):
						obdrct = carddrctlist[k]
				obsdatalist['CDRT'][mylistindex] = obdrct
			if(ob[3] == 35):
				if(int(obval) == -1):
					obval = '#FF0000'
				elif(int(obval) == 1):
					obval = '#999900'
				else:
					obval = '#000000'
			if(ob[3] == 60):
				try:
					if(float(ob[1]) >= obsfloattimelist[mylistindex]):
						if(obval > float(obsdatalist[dbvars[ob[3]]][mylistindex])):
							obsdatalist[dbvars[ob[3]]][mylistindex] = ("%d" % np.around(obval,0))
				except:
					skip = 1
			else:			
				try:
					if(ob[3] == 71):
						obsdatalist[dbvars[ob[3]]][mylistindex] = ("%.2f" % obval)
					else:
						obsdatalist[dbvars[ob[3]]][mylistindex] = ("%d" % np.around(obval,0))
				except:
					obsdatalist[dbvars[ob[3]]][mylistindex] = obval

for i in range(0,len(obsdatalist['TMPF']),1):
	if(obsdatalist['TMPF'][i] != 'NA' and obsdatalist['RELH'][i] != 'NA'):
		tmpc = (5./9.)*(float(obsdatalist['TMPF'][i]) - 32)
		es = 6.11*(10**((7.5*tmpc)/(237.7+tmpc)))
		e = es*(float(obsdatalist['RELH'][i])/100.)
		if(e != 0):
			alogs = 0.43429*np.log(e) - 0.43429*np.log(6.11)
			dwpf = (237.7*alogs)/(7.5-alogs)
			dwpf = (dwpf*(9./5.))+32
			obsdatalist['DWPF'][i] = ("%d" % np.around(dwpf,0))
		else:
			obsdatalist['DWPF'][i] = obsdatalist['TMPF'][i]

	# height: 900px;

			
print "Content-Type: text/html\n"			
print'''<!DOCTYPE html>
<html>
<head>
<title>Current Observations - Interactive Map</title>
<script src="/~u0790486/js/wxinfo/wxopen.js"></script>
<style>
#map {
	width: 100%;
}
</style>
</head>
<body>
<script src="/~u0790486/js/wxinfo/wxmenu.js"></script>
<script src="http://leaflet.github.io/Leaflet.label/leaflet.label.js"></script>
<script src="/~u0790486/js/leaflet/dist/Marker.Rotate.js"></script>'''

print'''
<div style="width:900px">
<font color="white">'''
print'''<br>Select Network for Current Observations:
<select name="net" id="mnetlistselect" onchange="onFormClick()">'''
for option in sorted(networkoptions, key=networkoptions.get):
	if str(net) == str(option):
		print '''	<option selected="selected" value="'''+str(option)+'''">'''+networkoptions[option]+'''</option>'''
	else:
		print '''	<option value="'''+str(option)+'''">'''+networkoptions[option]+'''</option>'''
print'''</select>
<span style="padding-left:10px">
<input type="hidden" name="latcen" value="'''+str(latcen)+'''">
<input type="hidden" name="loncen" value="'''+str(loncen)+'''">
</font>
</div><br>
<div id="map" style="text-align:left"></div>
<script>
var allstids = '''+str(allstidlist)+'''
var alllats = '''+str(alllatlist)+'''
var allactive = '''+str(allshowlist)+'''
var alllons = '''+str(alllonlist)
for key in dbvars:
	print '''var all'''+dbvars[key]+''' = '''+str(obsdatalist[dbvars[key]])
print '''var cloudmadeUrl = 'http://stamen-tiles-{s}.a.ssl.fastly.net/terrain/{z}/{x}/{y}.{ext}',
mbAttr = 'Map tiles by <a href="http://stamen.com">Stamen Design</a>, <a href="http://creativecommons.org/licenses/by/3.0">CC BY 3.0</a> &mdash; Map data &copy; <a href="http://www.openstreetmap.org/copyright">OpenStreetMap</a>',
cloudmade = new L.TileLayer(cloudmadeUrl, {
	minZoom: 4, 
	maxZoom: 18, 
	attribution: mbAttr,
	subdomains: 'abcd',
	ext: 'png',
	bounds: [[22, -132], [70, -56]]
});
map = new L.Map('map', {
	center: new L.LatLng('''+str(latcen)+''','''+str(loncen)+''','''+str(zoom)+'''), 
	zoom: '''+str(zoom)+''', 
	layers: [cloudmade]
});

var topinfo = L.control();
topinfo.onAdd = function (map) {
	this._div = L.DomUtil.create('div','legendinfo');
	this.update();
	return this._div;
};
topinfo.update = function (props) {
	this._div.innerHTML = '<font color="black">Most recent data within '''+str(minback)+''' minutes of '''+curtimstr+'''</font>';		
};
topinfo.addTo(map);

var datamarkerlist = [];
var tempmarkerlist = [];
var dwptmarkerlist = [];
var gustmarkerlist = [];
for (var j=0;j<allstids.length;j++) {
	if(allDATTIM[j] != 'NA' && allactive[j] == 1) {
		iconHTML = '<a href="http://mesowest.utah.edu/cgi-bin/droman/meso_base_dyn.cgi?stn='+allstids[j]+'" target="_blank"><b><font color="'+allQFLG[j]+'">'+allstids[j]+'</font></a><br>'+allDATTIM[j]+'</b><br>'
		if(allTMPF[j] != 'NA') {	
			var apitempmarker = L.marker([alllats[j],alllons[j]], {
				icon: L.divIcon({html: '<div class="tmpflabel">'+allTMPF[j]+'</div>',iconAnchor:[20,20]})
			});
			tempmarkerlist.push(apitempmarker);
			iconHTML += 'Temp: '+allTMPF[j]+' F<br>'
		}
		if(allDWPF[j] != 'NA') {
			var apidwpfmarker = L.marker([alllats[j],alllons[j]], {
				icon: L.divIcon({html: '<div class="dwpflabel">'+allDWPF[j]+'</div>',iconAnchor:[20,-3]})
			});
			dwptmarkerlist.push(apidwpfmarker);
			iconHTML += 'RH: '+allRELH[j]+'%<br>'
		}
		if(allDRCT[j] != 'NA' && allSKNTINT[j] != 'NA') {
			if(allQFLG[j] != '#FF0000') {
				var mystnplot = new L.Marker([alllats[j],alllons[j]], {
					icon: L.icon({iconUrl: 'http://home.chpc.utah.edu/~u0790486/css/barbs/barb_'+allSKNTINT[j]+'.png',iconSize: [80, 80],iconAnchor: [40, 40],}),
					focus: false,
					iconAngle: allDRCT[j],
				});
				iconHTML += 'Wind: '+allCDRT[j]+' at '+allSKNT[j]+' mph<br>';
			}
			else {
				var mystnplot = new L.Marker([alllats[j],alllons[j]], {
					icon: L.icon({iconUrl: 'http://home.chpc.utah.edu/~u0790486/css/barbs/barb_v.png',iconSize: [80, 80],iconAnchor: [40, 40],}),
				});
				iconHTML += 'Wind: '+allCDRT[j]+' at '+allSKNT[j]+' mph<br>';
			}
		}
		else {
			var mystnplot = new L.Marker([alllats[j],alllons[j]], {
				icon: L.icon({iconUrl: 'http://home.chpc.utah.edu/~u0790486/css/barbs/barb_v.png',iconSize: [80, 80],iconAnchor: [40, 40],}),
			});
		}
		if(allGUST[j] >= 20) {
			var apigustmarker = L.marker([alllats[j],alllons[j]], {
				icon: L.divIcon({html: '<div class="gustlabel">'+allGUST[j]+'</div>',iconAnchor:[-5,20]})
			});
			gustmarkerlist.push(apigustmarker);
			iconHTML += 'Gusting to '+allGUST[j]+' mph<br>';
		}
		if(allVOLT[j] != 'NA') {
			iconHTML += 'Voltage: '+allVOLT[j]+' V';
		}
		
		mystnplot.bindPopup(iconHTML);
		datamarkerlist.push(mystnplot);
	}
}
obslayer = L.layerGroup(datamarkerlist).addTo(map);
obslayer = L.layerGroup(tempmarkerlist).addTo(map);
obslayer = L.layerGroup(dwptmarkerlist).addTo(map);
obslayer = L.layerGroup(gustmarkerlist).addTo(map);
function onFormClick() {
	var centerlat = map.getCenter().lat;
	var centerlon = map.getCenter().lng;
	var centerzoom = map.getZoom()
	var mnetlisting = document.getElementById("mnetlistselect");
	var mnetselectoption = mnetlisting.options[mnetlisting.selectedIndex].value;
	window.open('/~u0790486/wxinfo/cgi-bin/wxmap.cgi?latcen='+centerlat+'&loncen='+centerlon+'&net='+mnetselectoption+'&zoom='+centerzoom+'&limit='''+limitstring+'''',"_self");
}
var newlocpopup = L.popup();
function onMapClick(e) {
	var mnetlisting = document.getElementById("mnetlistselect");
	var mnetselectoption = mnetlisting.options[mnetlisting.selectedIndex].value;
	var centerzoom = map.getZoom()
	newlocpopup
		.setLatLng(e.latlng)
		.setContent('<a href="/~u0790486/wxinfo/cgi-bin/wxmap.cgi?latcen='+e.latlng.lat+'&loncen='+e.latlng.lng+'&net='+mnetselectoption+'&zoom='+centerzoom+'&limit='''+limitstring+'''">Click to Reload Data</a>')
		.openOn(map);
}
map.on('click', onMapClick)
var nexradurl = 'http://mesonet.agron.iastate.edu/cache/tile.py/1.0.0/nexrad-n0r/{z}/{x}/{y}.png?'+(new Date()).getTime(),
nexradAttribution = 'Overlay data &copy; '''+nowyr+''' Iowa Environmental Mesonet'
nexrad = new L.TileLayer(nexradurl, {
	opacity: 0.7,
	attribution: nexradAttribution
}).addTo(map);		
wwlayer = L.tileLayer.wms("http://mesonet.agron.iastate.edu/cgi-bin/wms/us/wwa.cgi?"+(new Date()).getTime(), {
	layers: 'wwa',
	format: 'image/png',
	transparent: true,
	attribution: nexradAttribution
});
vislayer = new L.TileLayer('http://mesonet.agron.iastate.edu/cache/tile.py/1.0.0/goes-vis-1km-900913/{z}/{x}/{y}.png?'+(new Date()).getTime(), {
	opacity: 0.7,
	attribution: nexradAttribution
});
irlayer = new L.TileLayer('http://mesonet.agron.iastate.edu/cache/tile.py/1.0.0/goes-ir-4km-900913/{z}/{x}/{y}.png?'+(new Date()).getTime(), {
	opacity: 0.7,
	attribution: nexradAttribution
});
irlayeralaska = new L.TileLayer('http://mesonet.agron.iastate.edu/cache/tile.py/1.0.0/alaska-ir-900913/{z}/{x}/{y}.png?'+(new Date()).getTime(), {
	opacity: 0.7,
	attribution: nexradAttribution
});
var myoverlays = {
	"Radar": nexrad,
	"Visible Satellite": vislayer,
	"IR Satellite": irlayer,
	"Watches/Warnings": wwlayer
}
L.control.layers(myoverlays,null,{position: 'topleft',collapsed:false}).addTo(map);
$(window).on("resize", function() {
    $("#map").height($(window).height()).width($(window).width());
    map.invalidateSize();
}).trigger("resize");
</script>
<script src="/~u0790486/js/wxinfo/wxclose.js"></script>
</body>
</html>'''

sys.exit()