#--------------------------------------------
#Description:
#    Finds the joint effect of CO2
#    and CH4 on OLR, using the ccm radiation
#    model. Plots results as a contour plot
#
#    (Figure not used in the book)
#
#
#--------------------------------------------
#Use the function OLRDryAir to do the dry adiabat,
#or OLR to do the moist adiabat.


from ClimateUtilities import *
import ccmradFunctions as ccm

rh = .5 #Set relative humidity if you need it
Ts = 280. #Surface temperature
ps = 1.e5 #Surface air pressure, in Pa

ch4List = []
OLRList = []
OLRLists = []
ch4 = .01
while ch4 < 1.e4:
    OLRList = []
    co2List = []
    co2 = .1
    while co2 <= 1.e5:
        olr = ccm.OLR(ps,Ts,rh,co2,ch4)
        #olr = ccm.OLRDryAir(ps,Ts,co2,ch4)
        OLRList.append(olr)
        co2List.append(co2)
        print co2,ch4,olr
        co2 = 2.**.5*co2
    OLRLists.append(OLRList)
    ch4List.append(ch4)
    ch4 = 2.**.5*ch4
#
OLRLists = Numeric.array(OLRLists)
co2List = Numeric.array(co2List)
ch4List = Numeric.array(ch4List)
contour(OLRLists,x=Numeric.log10(co2List),y=Numeric.log10(ch4List))