def f(q, t):
z, x, Nz, Nx = q.reshape(4, Num)
# all the variables are in global CS
r = np.array([z, x])
X = p.X(r)
Y = mf.Yabs(r)
gamma = p.gamma(Te)
N = np.array([Nz, Nx])
dnz_dN = ri.dnz_dN(r, N)
dnx_dN = ri.dnx_dN(r, N)
dnz_dr = ri.dnz_dr(r, N)
dnx_dr = ri.dnx_dr(r, N)
dX_dr = p.dX_dr(r)
dY_dr = mf.dY_dr(r)
dgamma_dr = p.dgamma_dr(r)
nz, nx = ri.nz(r, N), ri.nx(r, N)
n = np.array([nz, nx])
if nx < 1:
vg = -dD_dN(X, Y, gamma, nz, nx, dnz_dN, dnx_dN) / dD_dw(
X, Y, gamma, nz, nx)
a = dD_dr(X, Y, gamma, nz, nx, dnz_dr, dnx_dr, dX_dr, dY_dr,
dgamma_dr) / dD_dw(X, Y, gamma, nz, nx)
if conf.verbose == True:
print('disp rel cold:{}, disp rel:{},x:{} '.format(
cold_disp_rel(X, Y, nz, nx, "Xm"),
disp_rel(X, Y, gamma, nz, nx), x))
#if disp_rel(X,Y,gamma,nz,nx)>0:
#break
else:
vg = -np.real(
ebw.dD_dN(X, Y, gamma, nz, nx, dnz_dN, dnx_dN) /
ebw.dD_dw(X, Y, gamma, nz, nx))
if vg[0] > 1 or vg[1] > 1:
print("vg", vg[0], vg[1])
if conf.verbose == True:
print(
"EBW", 'x:{},X:{},dispRel:{},nx:{},nz:{},'.format(
x, X, disp_rel_ebw(X, Y, gamma, nz, nx), nx, nz))
#if np.abs(disp_rel_ebw(X,Y,gamma,nz,nx))>1e-4:
#break
a = ebw.dD_dr(X, Y, gamma, nz, nx, dnz_dr, dnx_dr, dX_dr, dY_dr,
dgamma_dr) / ebw.dD_dw(X, Y, gamma, nz, nx)
arr = np.array([vg, a])
arr = np.real(np.real(arr.astype(np.complex64)))
return arr.reshape(4 * Num)
def f(q, t):
z, x, Nz, Nx = q.reshape(4, Num)
# all the variables are in global CS
r = np.array([z, x])
N = np.array([Nz, Nx])
X = p.X(r)
Y = mf.Yabs(r)
gamma = p.gamma(10)
"""
if localNx<1:
vg = -np.real( dD_dN( X,Y,gamma,localNz,localNx,dNz_dN,dNx_dN ) / dD_dw( X,Y,gamma,localNz,localNx ) )
a = np.real( dD_dr(X,Y,gamma,localNz,localNx,dNz_dr,dNx_dr,dX_dr,dY_dr,dgamma_dr) / dD_dw( X,Y,gamma,localNz,localNx ) )
A,B,C=get_ABC(X,Y,gamma,localNz,localNx)
#print A*localNx**4
arr = np.array([vg,a])
"""
dnz_dN = ri.dnz_dN(r, N)
dnx_dN = ri.dnx_dN(r, N)
dnz_dr = ri.dnz_dr(r, N)
dnx_dr = ri.dnx_dr(r, N)
dX_dr = p.dX_dr(r)
dY_dr = mf.dY_dr(r)
dgamma_dr = p.dgamma_dr(r)
nz, nx = ri.nz(r, N), ri.nx(r, N)
n = np.array([nz, nx])
if nx < 1:
vg = -np.real(dD_dN(X, Y, gamma, nz, nx, dnz_dN, dnx_dN))
a = np.real(
dD_dr(X, Y, gamma, nz, nx, dnz_dr, dnx_dr, dX_dr, dY_dr,
dgamma_dr))
arr = np.array([vg, a])
if conf.verbose == True:
print "dr", disp_rel(X, Y, gamma, nz, nx), x
else:
vg = -np.real(ebw.dD_dN(X, Y, gamma, nz, nx, dnz_dN, dnx_dN))
if vg[0] > 1 or vg[1] > 1:
print "vg", vg[0], vg[1]
if conf.verbose == True:
print "EBW", x, X, disp_rel_ebw(X, Y, gamma, nz, nx), nx, nz
a = np.real(
ebw.dD_dr(X, Y, gamma, nz, nx, dnz_dr, dnx_dr, dX_dr, dY_dr,
dgamma_dr))
arr = np.array([vg, a])
return arr.reshape(4)
def hot_disp_rel_wrapp(nx,*data): X,Y,gamma,nz = data return np.real(disp_rel(X,Y,gamma,nz,nx))
path = os.path.abspath("results/slab/harm1")
files = os.listdir(os.path.abspath(path))
soln = np.load(path + "/" + files[0]).item()["soln"]
soln_init_cond = np.load(path + "/" + files[0]).item()["init_cond"]
params.Y = float(files[0][files[0].index("_") + 1:-4])
NN = []
acc = soln_init_cond[2:4]
dt = 10. / 2e4
for i in range(0, soln[:, 0].size)[0:2]:
r = soln[i, 0:2]
N = soln[i, 2:4]
X = p.X(r)
Y = mf.Yabs(r)
Nz, Nx = N
acc = acc + e.f(soln[i, 0:2].reshape(2, 1), soln[i, 2:4].reshape(
2, 1)) * dt * 200
#print acc[1]
NN.append(acc)
nz, nx = ri.nz(r.reshape(2, 1),
N.reshape(2, 1)), ri.nx(r.reshape(2, 1), N.reshape(2, 1))
if np.abs(nx) < 1:
print soln[i, 1], disp_rel(X, Y, p.gamma(10), nz,
nx), cold_disp_rel(X, Y, nz, nx, "Xm")
else:
print soln[i, 1], ebw.disp_rel(X, Y, p.gamma(10), nz,
nx), cold_disp_rel(X, Y, nz, nx, "Xm")
NN = np.array(NN)