#!/usr/bin/python
###!/usr/bin/env python
###!/home/aymeric/Software/epd-7.0-2-rh5-x86/bin/python
####!/home/marshttp/EPD/epd-7.0-2-rh5-x86_64/bin/python
### here the version used to f2py the MCD Fortran routines

##################################################
### A Python CGI for the Mars Climate Database ###
### ------------------------------------------ ###
### Aymeric SPIGA 18-19/04/2012 ~ 11/08/2012   ###
### ------------------------------------------ ###
### (see mcdtest.py for examples of use)       ###
##################################################
### ajouts et corrections par Franck Guyon 09/2012
### ajouts suite a brainstorm equipe AS 10/2012

import cgi, cgitb 
import numpy as np
#from mcd import mcd
from modules import *
from modules import mcd

import cStringIO
import os as daos
import matplotlib.pyplot as mpl
from PIL import Image



### a function to read HTML arguments for coordinates
def gethtmlcoord(userinput,defmin,defmax):
   # accepted separators
   separators = [":",";",",","/"] 
   # initial values
   val = -9999. ; vals = None ; vale = None ; foundinterv = False
   if userinput == None:   userinput = "1"
   # the main work. either all -- or an interval -- or a single value.
   if userinput == "all":  isfree = 1 ; vals = defmin ; vale = defmax ; foundinterv = True
   else:
       for sep in separators:
           isfree = 1 ; ind = userinput.find(sep)
           if ind != -1: vals = float(userinput[:ind]) ; vale = float(userinput[ind+1:]) ; foundinterv = True
   if not foundinterv: isfree = 0 ; val = float(userinput)
   # return values
   return isfree, val, vals, vale


# for debugging in web browser
cgitb.enable()

# Create instance of FieldStorage 
form = cgi.FieldStorage() 

# create a MCD object
#query = mcd()
query=mcd.mcd() #FG: import from module mcd

# Get the kind of vertical coordinates and choose default behavior for "all"
try: query.zkey = int(form.getvalue("zkey"))
except: query.zkey = int(3)
if query.zkey == 2:    minxz = -5000.   ; maxxz = 100000.
elif query.zkey == 3:  minxz = 0.       ; maxxz = 120000.
elif query.zkey == 5:  minxz = -5000.   ; maxxz = 100000.
elif query.zkey == 4:  minxz = 1000.    ; maxxz = 0.001
elif query.zkey == 1:  minxz = 3396000. ; maxxz = 3596000.

# Get data from user-defined fields and define free dimensions
islatfree,  query.lat,  query.lats,  query.late  = gethtmlcoord( form.getvalue("latitude"),   -90.,  90. )
islonfree,  query.lon,  query.lons,  query.lone  = gethtmlcoord( form.getvalue("longitude"), -180., 180. )
isloctfree, query.loct, query.locts, query.locte = gethtmlcoord( form.getvalue("localtime"),    0.,  24. )
isaltfree,  query.xz,   query.xzs,   query.xze   = gethtmlcoord( form.getvalue("altitude"),  minxz, maxxz)

sumfree = islatfree + islonfree + isloctfree + isaltfree 
if sumfree > 2: exit() ## only 1D or 2D plots for the moment

try: query.datekey = int(form.getvalue("datekeyhtml"))
except: query.datekey = float(1)
if query.datekey == 1:
    try: query.xdate = float(form.getvalue("ls"))
    except: query.xdate = float(1)
else:
    try: query.xdate = float(form.getvalue("julian"))
    except: query.xdate = float(1)
    query.loct = 0.

try: query.hrkey = int(form.getvalue("hrkey"))
except: query.hrkey = int(1)
try: query.dust = int(form.getvalue("dust"))
except: query.dust  = int(1)
#        self.perturkey = 0  #integer perturkey ! perturbation type (0: none)
#        self.seedin    = 1  #random number generator seed (unused if perturkey=0)
#        self.gwlength  = 0. #gravity Wave wavelength (unused if perturkey=0)

# Get variables to plot
var1 = form.getvalue("var1")
var2 = form.getvalue("var2")
var3 = form.getvalue("var3")
var4 = form.getvalue("var4")

# fg: vartoplot is not None without form values
# vartoplot = [var1]
# fg: init var as with form values
if var1 == None: var1="t"
#if var2 == None: var2="p"

vartoplot = []
if var1 != "none": vartoplot = np.append(vartoplot,var1)
if var2 != "none" and var2 != None: vartoplot = np.append(vartoplot,var2)
if var3 != "none" and var3 != None: vartoplot = np.append(vartoplot,var3)
if var4 != "none" and var4 != None: vartoplot = np.append(vartoplot,var4)

iswind = form.getvalue("iswind")
if iswind == "on": iswindlog = True
else:              iswindlog = False
isfixedlt = form.getvalue("isfixedlt")
if isfixedlt == "on": input_fixedlt=True
else:                 input_fixedlt=False  

# reference name (to test which figures are already in the database)
reference = query.getnameset()+str(var1)+str(var2)+str(var3)+str(var4)+str(iswind)+str(isfixedlt)
figname = '../img/'+reference+'.png'
txtname = '../txt/'+reference
testexist = daos.path.isfile(figname)

# extract data from MCD if needed
if not testexist:

  ### 1D plots
  if sumfree == 1:

    ### getting data
    if isloctfree == 1:  	query.diurnal(nd=24) 
    elif islonfree == 1: 	query.zonal(nd=64)
    elif islatfree == 1: 	query.meridional(nd=48)
    elif isaltfree == 1: 	query.profile(nd=35)   
    else:			exit()  

    ### generic building of figure
    query.getascii(vartoplot,filename=txtname)
    query.htmlplot1d(vartoplot,figname=figname)
    #mpl.savefig("img/temp.png",dpi=85,bbox_inches='tight',pad_inches=0.25)
    #Image.open("../img/temp.png").save(figname,'JPEG')

  ### 2D plots
  elif sumfree == 2:

    ### getting data
    if islatfree == 1 and islonfree == 1:  	
        query.htmlmap2d(vartoplot,incwind=iswindlog,fixedlt=input_fixedlt,figname=figname) 
        #mpl.savefig("img/temp.png",dpi=110,bbox_inches='tight',pad_inches=0.4)
        #Image.open("img/temp.png").save(figname,'JPEG') ##lighter images   
        ### http://www.pythonware.com/library/pil/handbook/introduction.htm
    elif isaltfree == 1 and islonfree == 1:  	
        query.htmlplot2d(vartoplot,fixedlt=input_fixedlt,figname=figname)
    elif isaltfree == 1 and islatfree == 1:  	
        query.htmlplot2d(vartoplot,fixedlt=input_fixedlt,figname=figname)
    else:
        exit()  

## This is quite common
print "Content-type:text/html\n"
print "  "  #Apache needs a space after content-type

#entete="""<?xml version="1.0" encoding="UTF-8"?> 
#<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1 plus MathML 2.0//EN" "http://www.w3.org/Math/DTD/mathml2/xhtml-math11-f.dtd">
#<html xmlns="http://www.w3.org/1999/xhtml"> """

header="""<html><head><title>Mars Climate Database: The Web Interface</title></head><body>"""

print header
#print query.printset()
#print "<br />"


#print "<a href='../index.html'>Click here to start a new query</a><br />"
#print "<hr>"

## Now the part which differs
if sumfree == 0: 	query.update() ; query.htmlprinttabextvar(vartoplot)  #query.printmeanvar()
elif sumfree == 2: 	print "<img src='"+figname+"'><br />"
elif sumfree == 1:      
    print "<a href='"+txtname+"'>Click here to download an ASCII file containing data</a><br />"
    print "<hr>"
    print "<img src='"+figname+"'><br />"
else:			print "<h1>ERROR : sumfree is not or badly defined ...</h1></body></html>"


## This is quite common
bottom = "<hr><a href='../index.html'>Click here to start a new query</a>.<hr></body></html>"
#print bottom

##write to file object
#f = cStringIO.StringIO()
#mpl.savefig(f)
#f.seek(0)

##output to browser
#print "Content-type: image/png\n"
#print f.read()
#exit()

#print "Content-type:text/html\r\n\r\n"
#print "<html>"
#print "<head>"
#print "<title>MCD. Simple Python interface</title>"
#print "</head>"
#print "<body>"






## HTTP Header
#print "Content-Type:application/octet-stream; name=\"FileName\"\r\n";
#print "Content-Disposition: attachment; filename=\"FileName\"\r\n\n";
## Actual File Content will go hear.
#fo = open("foo.txt", "rb")
#str = fo.read();
#print str
## Close opend file
#fo.close()