mpl.style.use('default')
file_path = pathlib.Path(__file__).parent.absolute()
sys.path.append(str(file_path) + "/../LIB")
from plot_utils import save_fig
from fourier_solver import solve_pacman
from utils import read_performance_file
from numpy import pi
###############################################################################
# LATEX FONT
font = {'family': 'serif', 'size': 18}
rc('font', **font)
rc('text', usetex=True)
###############################################################################
# COLORMAP:
fc = farge_colormaps.farge_colormap_multi(taille=600,
limite_faible_fort=0.2,
etalement_du_zero=0.04,
type='pressure')
###############################################################################
# %% Specify:
###############################################################################
# %% get the data
quantity = "q_"
plt.close("all")
case = "CDF44_dealias_pacman"
#case = "CDF44_pacman"
ini = "pacman.ini"
eps_dir = "./adaptive_" + case + "_Bs17_Jmax7*eps1.0e-04_iter1/"
norm = 2
# %%
files = glob.glob(eps_dir + '/' + quantity + '*.h5')
#plt.close('all')
fig = plt.figure(2)
a = 2
plt.gcf().set_size_inches([5.0 * a, 5.0 * a])
plt.gcf().subplots_adjust(top=0.95, bottom=0.05, left=0.02, right=0.97)
overwrite = True
# for prefix, cmap in zip(['vort', 'p', 'div', 'Ux', 'Uy'],['vorticity','pressure','streamfunction','velocity','velocity']):
for prefix, cmap in zip(['div'], ['streamfunction']):
files = glob.glob('/home/engels/dev/WABBIT4-new-physics/WABBIT_div_test/' +
prefix + '_*.h5')
files.sort()
fc = farge_colormaps.farge_colormap_multi(taille=600,
limite_faible_fort=0.2,
etalement_du_zero=0.02,
type=cmap)
for file, i in zip(files[1:], range(len(files[1:]))):
file2 = file.replace('.h5', '.png')
if not os.path.isfile(file2) or overwrite:
wabbit_tools.plot_wabbit_file(file,
savepng=True,
cmap=fc,
ticks=False,
colorbar=True,
dpi=300,
caxis=[-0.06, 0.06],
title=True,
block_edge_color='w',
caxis_symmetric=True,
import flusi_tools as flusi
import numpy as np
import insect_tools
import os
import farge_colormaps
import glob
from matplotlib import rc
import matplotlib.gridspec as gs
from mpl_toolkits.axes_grid1.inset_locator import inset_axes
font = {'family' : 'sans-serif',
'sans-serif':['Helvetica'],
'size' : 22}
rc('font',**font)
rc('text', usetex=True)
fc_default = farge_colormaps.farge_colormap_multi(taille=600,etalement_du_zero=0.04, type='vorticity' )
def plot_modes(files,acoef_file,quantity,quantity_replace,pic_dir="./",fc=fc_default):
f = open(acoef_file)
data = [[float(num) for num in line.split()] for line in f]
f.close
data = np.asarray(data)
amin = np.amin(data)
amax = np.amax(data)
print("min(a_coef):", amin)
print("max(a_coef):", amax)
for mode,file in enumerate(files):
plt_file = file.split("/")[-1]
plt_file = plt_file.replace(quantity+"_",quantity_replace+"_")
plt_file = plt_file.replace('h5','png')
[float("%1.1e" % eps) for eps in np.logspace(-8, 0, 9)])
wdir = dirs["wabbit"]
fig_adapt = [plt.figure(41)]
ax_adapt = [fig.add_subplot() for fig in fig_adapt]
h5_fname = [
sorted(glob.glob(data["folder"] + '/Jmax6/' + qty + '*550*.h5'))[0]
for qty in data["qname"]
]
[l2error, linferror, Nblocks, Nblocksdense]=adapt2eps(h5_fname, \
wdir, eps_list, wabbit_setup['memory'], wabbit_setup['mpicommand'], wavelets="CDF44",\
normalization = "L2",create_log_file=True, show_plot=True, pic_dir=dir_list["images"])
# %% get the data
# COLORMAP:
fc = farge_colormaps.farge_colormap_multi(taille=600,
limite_faible_fort=0.2,
etalement_du_zero=0.04,
type='vorticity')
plt.close("all")
fig, axes = plt.subplots(nrows=3, ncols=2)
fig.subplots_adjust(bottom=0.001, top=0.999, wspace=0.01, hspace=-0.8)
for i, eps in enumerate(np.take(eps_list, [0, 2, 4, 6, 7, 8])):
files = sorted(glob.glob('./adapt_files/u[1,2]-eps%1.1e_' % eps + '*.h5'))
success = compute_vorticity(wdir, files[0], files[1],
wabbit_setup["memory"], "")
file = files[0].replace('u1-eps%1.1e_' % eps, 'vorx_')
file = os.path.split(file)[1]
ax,cb,hplot = wt.plot_wabbit_file(file,cmap=fc,caxis_symmetric=True,dpi=300, \
shading='flat', caxis=[-6,6], \
