-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdraftplot_FSBN2.py
119 lines (89 loc) · 2.8 KB
/
draftplot_FSBN2.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
from utils2 import *
import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mpl
import sys
#Define input parameters
dx = 10.
L = 5.e6
x = np.arange(0.,L,dx)
#step = .005
step = .0005
FSmean = np.logspace(np.log10(.0005),np.log10(.02),num=10)
BNmean = np.logspace(np.log10(.0005),np.log10(.01),num=10)
#FSmean = np.log(np.arange(.0005,.02+step,step))
#BNmean = np.log(np.arange(.0005,.01+step,step))
rmean = .004
width = 5.e5
width2 = 2.5e5
xmax = 3.*L/5.
Psi1 = np.zeros((len(BNmean),len(FSmean)))
Psi2 = np.zeros((len(BNmean),len(FSmean)))
Psi0 = np.zeros((len(BNmean),len(FSmean)))
for i,ii in enumerate(BNmean):
for j,jj in enumerate(FSmean):
print ii,jj
Rfs = np.exp(-((x-xmax)/width2)**2.)
Rfs = Rfs*jj/np.mean(Rfs)
Rbn = np.exp(-((x-0.)/width)**2.)
Rbn = Rbn*ii/np.mean(Rbn)
R = rmean+Rfs+Rbn
if R.min()>0:
#Solve equations
Hsg,S2,Msg = solvePW(x,R)
Hth,T1,FTint,T0 = solveAW(x,Msg)
#Get transports
psi1,psi2,psi0 = transport(Hth,Hsg,T1,S2,T0)
Psi1[i,j] = psi1[0]
Psi0[i,j] = psi0[-1]
Psi2[i,j] = psi2[-1]
else:
Psi1[i,j] = np.nan
Psi0[i,j] = np.nan
Psi2[i,j] = np.nan
#Get plotcolors
red,blu,pur = getcols()
#Make variable plot
prettyplot(10,3)
mpl.rcParams['figure.subplot.left'] = .07
mpl.rcParams['figure.subplot.right'] = .95
fig,ax = plt.subplots(1,3)
I,J = np.meshgrid(BNmean*L*1.e-3,FSmean*L*1.e-3)
cax = ax[0].contourf(I,J,Psi1.T*1.e-6,30,cmap=plt.get_cmap('Reds'))
cbar = plt.colorbar(cax,ax=ax[0])
cbar.ax.set_title('[Sv]')
cax = ax[1].contourf(I,J,Psi0.T*1.e-6,30,cmap=plt.get_cmap('Purples'))
cbar = plt.colorbar(cax,ax=ax[1])
cbar.ax.set_title('[Sv]')
cax = ax[2].contourf(I,J,Psi2.T*1.e-6,30,cmap=plt.get_cmap('Blues'))
cbar = plt.colorbar(cax,ax=ax[2])
cbar.ax.set_title('[Sv]')
for AX in ax:
AX.scatter(20.,65.,60,marker='X',color='.5')
AX.set_xscale('log')
AX.set_yscale('log')
Apd = 65.
Bpd = 20.
apd = np.argmin(FSmean*L*1.e-3-Apd)
bpd = np.argmin(BNmean*L*1.e-3-Bpd)
dp1da = (Psi1[bpd,apd+1]-Psi1[bpd,apd])*1.e-6
dp1db = (Psi1[bpd+1,apd]-Psi1[bpd,apd])*1.e-6
dp0da = (Psi0[bpd,apd+1]-Psi0[bpd,apd])*1.e-6
dp0db = (Psi0[bpd+1,apd]-Psi0[bpd,apd])*1.e-6
dp2da = (Psi2[bpd,apd+1]-Psi2[bpd,apd])*1.e-6
dp2db = (Psi2[bpd+1,apd]-Psi2[bpd,apd])*1.e-6
print 'psi1',dp1db/dp1da
print 'psi0',dp0db/dp0da
print 'psi2',dp2db/dp2da
for AX in ax[1:]:
AX.set_yticklabels([])
for AX in ax:
AX.set_xlabel('Baltic FW outflow [mSv]')
ax[0].set_ylabel(r'Arctic FW outflow [mSv]')
ax[0].set_title(r'AW inflow')
ax[1].set_title(r'DW outflow')
ax[2].set_title(r'PW outflow')
for d,dd in enumerate([r'$\textbf{a)}$',r'$\textbf{b)}$',r'$\textbf{c)}$']):
ax[d].text(0,1.05,dd,transform=ax[d].transAxes)
saveshow('rnfFSBN')
os.system('./copy_figs')