h5_file_path = '/tiara/home/glwu/Dedalus_test/test_HDF5/simulation_data_isoT_add_eq12.h5'
with h5py.File(h5_file_path, 'r') as file:
# 假設 HDF5 檔案中包含三維數據,並且每個變數與變數名稱相符
x = file['x'][:] # x 座標
y = file['y'][:] # y 座標
z = file['z'][:] # z 座標
ug = file['ug'][:] # u 的數值解
vg = file['vg'][:] # v 的數值解
wg = file['wg'][:] # w 的數值解
Pg = file['Pg'][:] # 數值解
Tg = file['Tg'][:] # 數值解
gg = file['gg'][:] # 解析解
# reload data
u['g'] = ug
v['g'] = vg
w['g'] = wg
P['g'] = Pg
T['g'] = Tg
print("==== check eq.10 ====")
# -grav_norm/cs_sq_norm*dlnrhodlnz*w -grav_norm/cs_sq_norm*f*dz(w) + dx(u) + dy(v) = 0
term1_1 = -grav_norm / cs_sq_norm * dlnrhodlnz * w
term1_2 = -grav_norm / cs_sq_norm * f * dz(w)
term1_3 = dx(u)
term1_4 = dy(v)
term1_1_val = term1_1.evaluate()
term1_2_val = term1_2.evaluate()
term1_3_val = term1_3.evaluate()
term1_4_val = term1_4.evaluate()