source: trunk/UTIL/PYTHON/mcd/proto/cgi-bin/mcdcgi.py @ 796

###!/usr/bin/env python
#!/usr/bin/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

# 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 data from user-defined fields and define free dimensions
# FG: add tests if var==None to have values in local without forms ones
query.lat = -9999.
getlat = form.getvalue("latitude")
if getlat == None: getlat = "1"
if getlat == "all":  islatfree = 1 ; query.lats = -90. ; query.late = 90.
elif ";" in getlat:  islatfree = 1 ; ind = getlat.find(";") ; query.lats = float(getlat[:ind]) ; query.late = float(getlat[ind+1:])
else:                islatfree = 0 ; query.lat = float(getlat)

query.lon = -9999.
getlon = form.getvalue("longitude")
if getlon == None: getlon = "1"
if getlon == "all":  islonfree = 1 ; query.lons = -180. ; query.lone = 180.
elif ";" in getlon:  islonfree = 1 ; ind = getlon.find(";") ; query.lons = float(getlon[:ind]) ; query.lone = float(getlon[ind+1:])
else:                islonfree = 0 ; query.lon = float(getlon)

query.loct = -9999.
getloct = form.getvalue("localtime")
if getloct == None: getloct = "1"
if getloct == "all": isloctfree = 1 ; query.locts = 0. ; query.locte = 24.
elif ";" in getloct: isloctfree = 1 ; ind = getloct.find(";") ; query.locts = float(getloct[:ind]) ; query.locte = float(getloct[ind+1:])
else:                isloctfree = 0 ; query.loct = float(getloct)

query.xz = -9999.
getalt = form.getvalue("altitude")
if getalt == None: getalt = "1"
if getalt == "all":  isaltfree = 1 ; query.xzs = 0. ; query.xze = 120000.
elif ";" in getalt:  isaltfree = 1 ; ind = getalt.find(";") ; query.xzs = float(getalt[:ind]) ; query.xze = float(getalt[ind+1:])
else:                isaltfree = 0 ; query.xz = float(getalt)

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

try: query.xdate = float(form.getvalue("ls"))
except: query.xdate = float(1)
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.zkey      = 3  # specify that xz is the altitude above surface (m)
#        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'

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()
    elif islatfree == 1:        query.meridional()
    elif isaltfree == 1:        query.profile()   
    else:                       exit()  

    ### generic building of figure
    query.htmlplot1d(vartoplot,vertplot=isaltfree,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.latlon()   
    else:                                       exit()  

    ### figure    
    zetitle = "MCD v4.3 - Dust scenario "+str(query.dust)+" - Date is "+str(query.xdate)
    query.htmlmap2d(vartoplot,incwind=iswindlog,fixedlt=input_fixedlt,figname=figname,title=zetitle) 
    #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

## 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

## Now the part which differs
if sumfree == 0:        query.update() ; query.htmlprinttabextvar(vartoplot)  #query.printmeanvar()
elif sumfree >= 1:      print "<a href='../index.html'>Click here to start a new query</a><br /><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()