def compare_flow(var_key, *cases):
"""
Compare distinct flow cases for a given variable on a
grid of plots.
:param var_key: Key of variable.
:param cases: flowCase instances.
:return: 1:success.
"""
# Check valid key
assert var_key in cases[
0].flow_dict, "Invalid variable key given: %r" % var_key
# Check required amount of subplots
num_of_plots = len(cases)
grid_length = int(num_of_plots**0.5) + 1
# Create figure
fig = plot.empty_plot(no_axis=True)
# Add Title
ax = fig.add_subplot(111)
ax.set_title('Data basis with %r cases' % num_of_plots)
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
# Plot individual cases
for i, case in enumerate(cases):
ax = fig.add_subplot(grid_length, grid_length,
i + 1) # Add plot to grid
ax.get_xaxis().set_visible(False) # No axes
ax.get_yaxis().set_visible(False)
x = case.flow_dict['x']
y = case.flow_dict['y']
nx = case.flow_dict['nx']
ny = case.flow_dict['ny']
var = case.flow_dict[var_key]
plot.contouring(x.reshape(nx, ny),
y.reshape(nx, ny),
var.reshape(nx, ny),
xlabel='',
ylabel='',
colorbar=False,
append_to_fig_ax=(fig, ax))
plot.show()
return 1
# compare_profiles('pm', 'x', 0, *hifi_data)
#
# # Fig 27 (a), (c)
# compare_profiles('um', 'y', 4, *hifi_data, xlim=[-0.2, 1.2], ylim=[-0.1, 3.1])
# compare_profiles('uu', 'y', 4, *hifi_data, xlim=[0, 0.1], ylim=[-0.1, 3.1])
#
# # Fig 23, tiny separation bubble
# hifi_data[3].show_flow('um', contour_level=[-0.1, 0, 0.1, 0.2, 0.3], xlim=[0.55, 0.93], ylim=[0.58, 0.82])
#
# # Fig 24, tiny recirculation bubble
# hifi_data[2].show_flow('um', contour_level=[-0.1, 0, 0.1, 0.2, 0.3], xlim=[6.9, 7.5], ylim=[-0.01, 0.31])
#
# # Fig 29, tke comparison
# hifi_data[0].show_flow('k', contour=False)
# hifi_data[-1].show_flow('k', contour=False)
# Fig 30, lumley triangles with invariants
loc = 4
p_IIb = get_profile_data(hifi_data[0], 'IIb', 'y', loc)
p_IIIb = get_profile_data(hifi_data[0], 'IIIb', 'y', loc)
p2_IIb = get_profile_data(hifi_data[-1], 'IIb', 'y', loc)
p2_IIIb = get_profile_data(hifi_data[-1], 'IIIb', 'y', loc)
fig, ax = empty_plot()
lining(p_IIIb[0][221:-1], p_IIb[0][221:-1], xlim=[-0.03, 0.09], ylim=[0, 0.41], append_to_fig_ax=(fig, ax), linestyle='-s')
lining(p2_IIIb[0][221:-1], p2_IIb[0][221:-1], xlim=[-0.03, 0.09], ylim=[0, 0.41], append_to_fig_ax=(fig, ax))
show()
# fig, ax = compare_profiles('uu', 'y+', 0.7, hifi_data[2], xlim=[1, 1e3], ylim=[-0.005, 0.017], xlog=True)
# compare_profiles('vv', 'y+', 0.7, hifi_data[2], xlim=[1, 1e3], ylim=None, xlog=True, append_to_fig_ax=(fig, ax))
# compare_profiles('ww', 'y+', 0.7, hifi_data[2], xlim=[1, 1e3], ylim=None, xlog=True, append_to_fig_ax=(fig, ax))
# compare_profiles('uv', 'y+', 0.7, hifi_data[2], xlim=[1, 1e3], ylim=None, xlog=True, append_to_fig_ax=(fig, ax))
# Fig 2, Evolution of Clauser beta
compare_integral_quantity('beta', 'x', *hifi_data, xlim=[0, 1], ylim=[0, 20])
# compare_integral_quantity('beta', 'Ret', *hifi_data, xlim=[0, 400], ylim=[0, 20])
# Fig 4, Evolution of Reynolds number wrt momentum thickness and friction (tau)
compare_integral_quantity('Reth', 'x', *hifi_data, xlim=[0, 1], ylim=[0, 3500])
# compare_integral_quantity('Ret', 'x', *hifi_data, xlim=[0, 1], ylim=[0, 500])
# Fig 6, Inner-scaled mean velocity profiles
compare_profiles('u+',
'y+',
0.4,
*hifi_data,
xlim=[1, 1e3],
ylim=[0, 30],
xlog=True)
compare_profiles('u+',
'y+',
0.75,
*hifi_data,
xlim=[1, 1e3],
ylim=[0, 30],
xlog=True)
plot.show()