# ----------------------------------------------------------------------------
# This software is in the public domain, furnished "as is", without technical
# support, and with no warranty, express or implied, as to its usefulness for
# any purpose.
#
# Haines.py
#
# Author: dtomalak
# ----------------------------------------------------------------------------


ToolType = "numeric"
WeatherElementEdited = "Haines"
from numpy import *
HideTool = 0
#ScreenList = ["Haines"]
# You can screen the elements for which your tool will appear by using
# a ScreenList.  For example:
#
#ScreenList = ["T","Td"]
#ScreenList = ["SCALAR","VECTOR","WEATHER","DISCRETE"]

# If desired, Set up variables to be solicited from the user:
# VariableList = [
#         ("Variable name1" , defaultValue1, "numeric"),
#         ("Variable name2" , "default value2", "alphaNumeric"),
#         ("Variable name3" , ["default value1", "default value2"], "check",
#                       ["value1", "value2", "value3"]),
#         ("Variable name4" , "default value4", "radio",
#                       ["value1", "value2", "value3"]),
#         ("Variable name5" , defaultValue, "scale",
#                       [minValue, maxValue], resolution),
#         ("Variable name6" , "", "model"),
#         ("Variable name7" , "", "D2D_model"),
#         ("Label contents" , "", "label"),
#         ("", dialogHeight, "scrollbar"),
#        ]

# Set up Class
import SmartScript
import string, time
import AbsTime, TimeRange
# For available commands, see SmartScript


class Tool (SmartScript.SmartScript):
    def __init__(self, dbss):
        SmartScript.SmartScript.__init__(self, dbss)

    # Required Method: Execute
    #  %comment
    # Fill in the arguments you want to use -- WeatherElement1, WeatherElement2...

    
    def execute(self, GridTimeRange, varDict):
        "Sets Haines according to selected model"

         
        fwModel = self.getObject("FireModel", "ModelType")

        if fwModel == "NAM12":
            modelSource = "_D2D_NAM12"
        elif fwModel == "NAM40":
            modelSource = "_D2D_NAM40"
        else:
            modelSource = "_D2D_GFS40"
            

        ############################################
        #
        # Valid Modes are 'Low', 'Mid', or 'High'
        #
        ############################################

        Mode = 'Low'

##        month = time.strftime('%b%d', time.gmtime())
##        monthrun = month + run
        site =  self.getSiteID()
##        model = site + modelSource + monthrun
        model = site + modelSource
        print "Using " + model + " for Haines Index Calculation"
        print "Haines time range is: \n" + `GridTimeRange`
            
        T950 = self.getGrids(model,"t","MB950",GridTimeRange,noDataError=0)
        if T950 is None:
            self.noData()

        T850 = self.getGrids(model,"t","MB850",GridTimeRange,noDataError=0)
        if T850 is None:
            self.noData()

        T700 = self.getGrids(model,"t","MB700",GridTimeRange,noDataError=0)
        if T850 is None:
            self.noData()

        T500 = self.getGrids(model,"t","MB500",GridTimeRange,noDataError=0)
        if T850 is None:
            self.noData()

        RH850 = self.getGrids(model,"rh","MB850",GridTimeRange,noDataError=0)
        if RH850 is None:
            self.noData()

        RH700 = self.getGrids(model,"rh","MB700",GridTimeRange,noDataError=0)
        if RH850 is None:
            self.noData()

        #
        # Calculate 850 MB Depressions
        #

        Dwpt_Factor_850 = (log(RH850/100.0) + ((17.27 * (T850 - 273.3))/(237.3 + (T850 - 273.31))))/ 17.27
        DD850 = T850 - ((Dwpt_Factor_850 * 237.3) / ( 1.0 - Dwpt_Factor_850 ) + 273.3)

        #
        # Calculate 700 MB Depressions
        #
        
        Dwpt_Factor_700 = (log(RH700/100.0) + ((17.27 * (T700 - 273.3))/(237.3 + (T700 - 273.31))))/ 17.27
        DD700 = T700 - ((Dwpt_Factor_700 * 237.3) / ( 1.0 - Dwpt_Factor_700 ) + 273.3)

        if string.upper(Mode) == 'LOW':

            # find T difference between levels
            Tdiff = (T950 - T850)

            # compute A & B terms
            Aterm = 2
            Bterm = 2
            Aterm = where(less_equal(Tdiff,3),1,Aterm)
            Aterm[greater_equal(Tdiff,8)] = 3
            Bterm = where(less_equal(DD850,5),1,Bterm)
            Bterm[greater_equal(DD850,10)] = 3

            # compute Haines
            Haines = Aterm + Bterm        

            return Haines

        if string.upper(Mode) == 'MID':

            # find T difference between levels
            Tdiff = (T850 - T700)

            # compute A & B terms
            Aterm = 2
            Bterm = 2
            Aterm = where(less_equal(Tdiff,5),1,Aterm)
            Aterm[greater_equal(Tdiff,11)] = 3
            Bterm = where(less_equal(DD850,5),1,Bterm)
            Bterm[greater_equal(DD850,13)] = 3

            # compute Haines
            Haines = Aterm + Bterm        

            return Haines 

        if string.upper(Mode) == 'HIGH':

            # find T difference between levels
            Tdiff = (T700 - T500)

            # compute A & B terms
            Aterm = 2
            Bterm = 2
            Aterm = where(less_equal(Tdiff,17),1,Aterm)
            Aterm[greater_equal(Tdiff,22)] = 3
            Bterm = where(less_equal(DD700,14),3,Bterm)
            Bterm[greater_equal(DD700,21)] = 1

            # compute Haines
            Haines = Aterm + Bterm        

            return Haines