mirror of
https://gitlab.rlp.net/mobitar/ReCo.jl.git
synced 2024-11-12 22:40:44 +00:00
116 lines
3.8 KiB
Julia
116 lines
3.8 KiB
Julia
function animate(directory::String, sol::Solution, variables; framerate=0)
|
|
println("Generating animation...")
|
|
|
|
fig = Figure(; resolution=(1080, 1080))
|
|
ax = Axis(
|
|
fig[1, 1];
|
|
limits=(-variables.l, variables.l, -variables.l, variables.l),
|
|
aspect=AxisAspect(1),
|
|
xlabel=L"x",
|
|
ylabel=L"y",
|
|
)
|
|
|
|
color_scheme = ColorSchemes.cyclic_mrybm_35_75_c68_n256_s25
|
|
|
|
Colorbar(fig[1, 2]; limits=(0, 2), colormap=color_scheme, label=L"\frac{\varphi}{\pi}")
|
|
|
|
animation_path = "$directory/animation.mkv"
|
|
|
|
record_framerate = 0
|
|
if framerate > 0
|
|
record_framerate = framerate
|
|
else
|
|
record_framerate = variables.framerate
|
|
end
|
|
|
|
record(fig, animation_path; framerate=record_framerate) do io
|
|
circles = Observable(Vector{Circle}(undef, variables.N))
|
|
colors = Observable(Vector{RGBAf}(undef, variables.N))
|
|
|
|
if variables.debug
|
|
segments_x = Observable(zeros(2 * variables.N))
|
|
segments_y = Observable(zeros(2 * variables.N))
|
|
|
|
interaction_circles = Observable(Vector{Circle}(undef, variables.N))
|
|
skin_circles = Observable(Vector{Circle}(undef, variables.N))
|
|
|
|
interaction_colors = Observable(Vector{RGBAf}(undef, variables.N))
|
|
skin_colors = Observable(Vector{RGBAf}(undef, variables.N))
|
|
end
|
|
|
|
π2 = 2 * π
|
|
particle_radius = variables.particle_diameter / 2
|
|
|
|
@showprogress 0.6 for frame in 1:(variables.n_snapshots)
|
|
@simd for i in 1:(variables.N)
|
|
circles[][i] = Circle(Point2(sol.center[i, frame]), particle_radius)
|
|
|
|
color = get(color_scheme, rem2pi(sol.φ[i, frame], RoundDown) / π2)
|
|
colors[][i] = RGBAf(color)
|
|
|
|
if variables.debug
|
|
interaction_circles[][i] = Circle(
|
|
Point2(sol.center[i, frame]), variables.interaction_r
|
|
)
|
|
skin_circles[][i] = Circle(
|
|
Point2(sol.center[i, frame]), variables.skin_r
|
|
)
|
|
|
|
interaction_colors[][i] = RGBAf(color, 0.08)
|
|
skin_colors[][i] = RGBAf(color, 0.04)
|
|
end
|
|
end
|
|
|
|
if frame > 1
|
|
if variables.debug
|
|
@simd for i in 1:(variables.N)
|
|
first_ind = 2 * i - 1
|
|
second_ind = 2 * i
|
|
frame_min_1 = frame - 1
|
|
|
|
segments_x[][first_ind] = sol.center[i, frame_min_1][1]
|
|
segments_x[][second_ind] = sol.center[i, frame][1]
|
|
|
|
segments_y[][first_ind] = sol.center[i, frame_min_1][2]
|
|
segments_y[][second_ind] = sol.center[i, frame][2]
|
|
end
|
|
|
|
if frame == 2
|
|
linesegments!(ax, segments_x, segments_y; color=colors)
|
|
end
|
|
end
|
|
else # First frame
|
|
poly!(ax, circles; color=colors)
|
|
|
|
if variables.debug
|
|
poly!(ax, interaction_circles; color=interaction_colors)
|
|
poly!(ax, skin_circles; color=skin_colors)
|
|
end
|
|
|
|
println("Recording started!")
|
|
end
|
|
|
|
notify(circles)
|
|
notify(colors)
|
|
|
|
if variables.debug
|
|
notify(interaction_circles)
|
|
notify(interaction_colors)
|
|
|
|
notify(skin_circles)
|
|
notify(skin_colors)
|
|
|
|
notify(segments_x)
|
|
notify(segments_y)
|
|
end
|
|
|
|
ax.title = "t = $(round(sol.t[frame], digits=3))"
|
|
|
|
recordframe!(io)
|
|
end
|
|
end
|
|
|
|
println("Animation done and saved to $animation_path.")
|
|
|
|
return nothing
|
|
end
|