def b_model(params, data):
phi_0, X, Y, Z, phase_0 = params
time, x, y, z, U, V, N, f = data
k = 2*np.pi/X
l = 2*np.pi/Y
m = 2*np.pi/Z
om = gw.omega(N, k, m, l, f)
b = gw.b(x, y, z, time, phi_0, k, l, m, om, N, U=U, V=V, phase_0=phase_0)
return b
def full_model(params, data):
phi_0, X, Y, Z, phase_0 = params
time, x, y, z, U, V, N, f, w_data, u_data, v_data, b_data = data
k = 2*np.pi/X
l = 2*np.pi/Y
m = 2*np.pi/Z
om = gw.omega(N, k, m, l, f)
u = gw.u(x, y, z, time, phi_0, k, l, m, om, f=f, U=U, V=V, phase_0=phase_0)
v = gw.v(x, y, z, time, phi_0, k, l, m, om, f=f, U=U, V=V, phase_0=phase_0)
w = gw.w(x, y, z, time, phi_0, k, l, m, om, N, U=U, V=V, phase_0=phase_0)
b = gw.b(x, y, z, time, phi_0, k, l, m, om, N, U=U, V=V, phase_0=phase_0)
return np.hstack((w - w_data, u - u_data, v - v_data, b - b_data))
def b_model(params, data):
X, Y, Z, phase_0 = params
time, dist, depth, U, V, W, B, N, f = data
k = 2*np.pi/X
l = 2*np.pi/Y
m = 2*np.pi/Z
om = gw.omega(N, k, m, l, f)
phi_0 = np.max(W)*(N**2 - f**2)*m/(om*(k**2 + l**2 + m**2))
b = gw.b(dist, 0., depth, time, phi_0, k, l, m, om, N, phase_0=phase_0)
resid = b - B
return 250.*resid
def b_model(params, pfl, zlim, deg):
phi_0, X, Z, phase_0 = params
zmin, zmax = zlim
nope = np.isnan(pfl.z) | (pfl.z < zmin) | (pfl.z > zmax)
t = 60*60*24*(pfl.UTC - np.nanmin(pfl.UTC))
x = 1000.*(pfl.dist_ctd - np.nanmin(pfl.dist_ctd))
k = 2*np.pi/X
l = 0.
m = 2*np.pi/Z
f = gsw.f(pfl.lat_start)
N = np.mean(np.sqrt(pfl.N2_ref[~nope]))
om = gw.omega(N, k, m, l, f)
b = gw.b(x, 0., pfl.z, t, phi_0, k, l, m, om, N, phase_0=phase_0)
resid = b[~nope] - pfl.b[~nope]
return 250.*resid