*
* this exec decomposes the monthly river discharge data set into individual basins,
* applies the inflation factors listed in Dai and Trenberth (2002) Table 3, recombines them
* and then writes out the data set: gridded_discharge_mo_360x720x480_corrected.nc
*
reinit
set display color white
c
sdfopen /aosc/horse/carton/rivers/data/gridded_discharge_mo_360x720x480.nc
sdfopen /aosc/horse/carton/rivers/ctls/bas_OCL_expanded_15.nc
set dfile 2
set x 1 1440
set y 1 720
set t 1
set z 1
msk2 = msk.2
set dfile 1
set lon 0 360
set y 1 360
set z 1
set t 1
msk3 = lterp(msk2,discharge.1)
atlantic = maskout(1.30,0.1-abs(msk3-1))
pacific = maskout(1.535,0.1-abs(msk3-2))
indian = maskout(2.323,0.1-abs(msk3-3))
arctic = maskout(1.,0.1-abs(msk3-11))
bbg = maskout(2.323,0.1-abs(msk3-56))

atlantic = maskout(1.38,0.1-abs(msk3-1))
pacific = maskout(1.627,0.1-abs(msk3-2))
indian = maskout(2.462,0.1-abs(msk3-3))
arctic = maskout(1.06,0.1-abs(msk3-11))
bbg = maskout(2.462,0.1-abs(msk3-56))

set lon 0 35
set lat 30 45
set z 1
set t 1
med = maskout(1.38,0.1-abs(msk3-4))
set lon 125 145
set lat 35 55
set z 1
set t 1
japan = maskout(1.627,0.1-abs(msk3-12))
set lon 0 360
set y 1 360
set z 1


set t 1 480
discharge2 = const(discharge*bbg,0,-u)+const(discharge*indian,0,-u)+const(discharge*pacific,0,-u)+const(discharge*atlantic,0,-u)+const(discharge*arctic,0,-u)+const(discharge*japan,0,-u)+const(discharge*med,0,-u)
set sdfwrite -3dt -rt -nc4 gridded_discharge_mo_360x720x480_corrected2.nc
sdfwrite discharge2
set sdfattr discharge2 String units m3/s
set sdfattr discharge2 String long_name Corrected river discharge
clear sdfwrite

set display color white
c
set x 1
set y 1
set t 1 480
ts = asum(discharge2,g)
set display color white
c
set vpage 0 8 5 9
set yaxis 0.5 2 0.5
set axlim 0.5 2.
set grads off
set grid off
set cmark 0
set ccolor 1
d ts/1.e6
set cmark 0
set ccolor 3
d tloop(ave(ts/1.e6,t-5,t+6))
draw ylab Sverdrups
draw title GLOBAL MONTHLY DISCHARGE
set vpage off
set strsiz .09
set string 1 l 4 0
draw string 2.07 8.06 AVERAGE = 1.14Sv