Skip to content

Commit

Permalink
Enable neutral and electron constraints plots
Browse files Browse the repository at this point in the history 
The neutral plots were deactivated by a bug. The electron plots were
commented out, but can be made now.
  • Loading branch information
@johnomotani
johnomotani committed Jan 11, 2025
1 parent 79cbc89 commit 4d9eb97
Showing 1 changed file with 90 additions and 90 deletions.
180 changes: 90 additions & 90 deletions makie_post_processing/makie_post_processing/src/makie_post_processing.jl
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5340,7 +5340,7 @@ function constraints_plots(run_info; plot_prefix=plot_prefix)
end

# Neutrals
if any(ri.n_neutral_species > 1
if any(ri.n_neutral_species > 0
&& (ri.evolve_density || ri.evolve_upar || ri.evolve_ppar)
for ri ∈ run_info)

Expand Down Expand Up @@ -5388,41 +5388,41 @@ function constraints_plots(run_info; plot_prefix=plot_prefix)
end

# Electrons
#if any(ri.composition.electron_physics ∈ (kinetic_electrons,
# kinetic_electrons_with_temperature_equation)
# for ri ∈ run_info)

# fig, ax = get_1d_ax(; xlabel="z", ylabel="constraint coefficient")
# for ri ∈ run_info
# if length(run_info) > 1
# prefix = ri.run_name * ", "
# else
# prefix = ""
# end

# varname = "electron_constraints_A_coefficient"
# label = prefix * "(A-1)"
# data = get_variable(ri, varname; it=it0, ir=ir0)
# data .-= 1.0
# plot_vs_z(ri, varname; label=label, data=data, ax=ax, input=input)

# varname = "electron_constraints_B_coefficient"
# label = prefix * "B"
# plot_vs_z(ri, varname; label=label, ax=ax, it=it0, ir=ir0,
# input=input)

# varname = "electron_constraints_C_coefficient"
# label = prefix * "C"
# plot_vs_z(ri, varname; label=label, ax=ax, it=it0, ir=ir0,
# input=input)
# end
# put_legend_below(fig, ax)
# # Ensure the first row width is 3/4 of the column width so that
# # the plot does not get squashed by the legend
# rowsize!(fig.layout, 1, Aspect(1, 3/4))
# resize_to_layout!(fig)
# save(plot_prefix * "electron_constraints.pdf", fig)
#end
if any(ri.composition.electron_physics ∈ (kinetic_electrons,
kinetic_electrons_with_temperature_equation)
for ri ∈ run_info)

fig, ax = get_1d_ax(; xlabel="z", ylabel="constraint coefficient")
for ri ∈ run_info
if length(run_info) > 1
prefix = ri.run_name * ", "
else
prefix = ""
end

varname = "electron_constraints_A_coefficient"
label = prefix * "(A-1)"
data = get_variable(ri, varname; it=it0, ir=ir0)
data .-= 1.0
plot_vs_z(ri, varname; label=label, data=data, ax=ax, input=input)

varname = "electron_constraints_B_coefficient"
label = prefix * "B"
plot_vs_z(ri, varname; label=label, ax=ax, it=it0, ir=ir0,
input=input)

varname = "electron_constraints_C_coefficient"
label = prefix * "C"
plot_vs_z(ri, varname; label=label, ax=ax, it=it0, ir=ir0,
input=input)
end
put_legend_below(fig, ax)
# Ensure the first row width is 3/4 of the column width so that
# the plot does not get squashed by the legend
rowsize!(fig.layout, 1, Aspect(1, 3/4))
resize_to_layout!(fig)
save(plot_prefix * "electron_constraints.pdf", fig)
end
end

if input.animate
Expand Down Expand Up @@ -5496,7 +5496,7 @@ function constraints_plots(run_info; plot_prefix=plot_prefix)
end

# Neutrals
if any(ri.n_neutral_species > 1
if any(ri.n_neutral_species > 0
&& (ri.evolve_density || ri.evolve_upar || ri.evolve_ppar)
for ri ∈ run_info)

Expand Down Expand Up @@ -5564,59 +5564,59 @@ function constraints_plots(run_info; plot_prefix=plot_prefix)
end

# Electrons
#if any(ri.composition.electron_physics ∈ (kinetic_electrons,
# kinetic_electrons_with_temperature_equation)
# for ri ∈ run_info)

# frame_index = Observable(1)
# fig, ax = get_1d_ax(; xlabel="z", ylabel="constraint coefficient")

# # Calculate plot limits manually so we can exclude the first time point, which
# # often has a large value for (A-1) due to the way initialisation is done,
# # which can make the subsequent values hard to see.
# ymin = Inf
# ymax = -Inf
# for ri ∈ run_info
# if length(run_info) > 1
# prefix = ri.run_name * ", "
# else
# prefix = ""
# end

# varname = "electron_constraints_A_coefficient"
# label = prefix * "(A-1)"
# data = get_variable(ri, varname; ir=ir0)
# data .-= 1.0
# ymin = min(ymin, minimum(data[:,2:end]))
# ymax = max(ymax, maximum(data[:,2:end]))
# animate_vs_z(ri, varname; label=label, data=data,
# frame_index=frame_index, ax=ax, input=input)

# varname = "electron_constraints_B_coefficient"
# label = prefix * "B"
# data = get_variable(ri, varname; ir=ir0)
# ymin = min(ymin, minimum(data[:,2:end]))
# ymax = max(ymax, maximum(data[:,2:end]))
# animate_vs_z(ri, varname; label=label, data=data,
# frame_index=frame_index, ax=ax, ir=ir0, input=input)

# varname = "electron_constraints_C_coefficient"
# label = prefix * "C"
# data = get_variable(ri, varname; ir=ir0)
# ymin = min(ymin, minimum(data[:,2:end]))
# ymax = max(ymax, maximum(data[:,2:end]))
# animate_vs_z(ri, varname; label=label, data=data,
# frame_index=frame_index, ax=ax, ir=ir0, input=input)
# end
# put_legend_below(fig, ax)
# # Ensure the first row width is 3/4 of the column width so that
# # the plot does not get squashed by the legend
# rowsize!(fig.layout, 1, Aspect(1, 3/4))
# resize_to_layout!(fig)
# ylims!(ax, ymin, ymax)
# save_animation(fig, frame_index, nt,
# plot_prefix * "electron_constraints." * input.animation_ext)
#end
if any(ri.composition.electron_physics ∈ (kinetic_electrons,
kinetic_electrons_with_temperature_equation)
for ri ∈ run_info)

frame_index = Observable(1)
fig, ax = get_1d_ax(; xlabel="z", ylabel="constraint coefficient")

# Calculate plot limits manually so we can exclude the first time point, which
# often has a large value for (A-1) due to the way initialisation is done,
# which can make the subsequent values hard to see.
ymin = Inf
ymax = -Inf
for ri ∈ run_info
if length(run_info) > 1
prefix = ri.run_name * ", "
else
prefix = ""
end

varname = "electron_constraints_A_coefficient"
label = prefix * "(A-1)"
data = get_variable(ri, varname; ir=ir0)
data .-= 1.0
ymin = min(ymin, minimum(data[:,2:end]))
ymax = max(ymax, maximum(data[:,2:end]))
animate_vs_z(ri, varname; label=label, data=data,
frame_index=frame_index, ax=ax, input=input)

varname = "electron_constraints_B_coefficient"
label = prefix * "B"
data = get_variable(ri, varname; ir=ir0)
ymin = min(ymin, minimum(data[:,2:end]))
ymax = max(ymax, maximum(data[:,2:end]))
animate_vs_z(ri, varname; label=label, data=data,
frame_index=frame_index, ax=ax, ir=ir0, input=input)

varname = "electron_constraints_C_coefficient"
label = prefix * "C"
data = get_variable(ri, varname; ir=ir0)
ymin = min(ymin, minimum(data[:,2:end]))
ymax = max(ymax, maximum(data[:,2:end]))
animate_vs_z(ri, varname; label=label, data=data,
frame_index=frame_index, ax=ax, ir=ir0, input=input)
end
put_legend_below(fig, ax)
# Ensure the first row width is 3/4 of the column width so that
# the plot does not get squashed by the legend
rowsize!(fig.layout, 1, Aspect(1, 3/4))
resize_to_layout!(fig)
ylims!(ax, ymin, ymax)
save_animation(fig, frame_index, nt,
plot_prefix * "electron_constraints." * input.animation_ext)
end
end
catch e
return makie_post_processing_error_handler(
Expand Down

0 comments on commit 4d9eb97

Please sign in to comment.