Return plots

Author: 1-Bart-1

ext/VortexStepMethodMakieExt.jl

@@ -9,14 +9,18 @@ The corner points are ordered as: LE1, TE1, TE2, LE2.
 This creates two triangles: (LE1, TE1, TE2) and (LE1, TE2, LE2).
 """
 function Makie.plot!(ax, panel::VortexStepMethod.Panel; color=(:red, 0.2), R_b_w=nothing, T_b_w=nothing, kwargs...)
+    plots = []
     points = [Point3f(panel.corner_points[:, i]) for i in 1:4]
     if !isnothing(R_b_w) && !isnothing(T_b_w)
         points = [Point3f(R_b_w * p + T_b_w) for p in points]
     end
     faces = [Makie.GLTriangleFace(1, 2, 3), Makie.GLTriangleFace(1, 3, 4)]
-    mesh!(ax, points, faces; color, kwargs...)
+    p = mesh!(ax, points, faces; color, kwargs...)
+    push!(plots, p)
     border_points = [points..., points[1]]
-    lines!(ax, border_points; color=:black)
+    p = lines!(ax, border_points; color=:black)
+    push!(plots, p)
+    return plots
 end
 
 """
@@ -25,9 +29,12 @@ end
 Plot a `BodyAerodynamics` object by plotting each of its panels.
 """
 function Makie.plot!(ax, body::VortexStepMethod.BodyAerodynamics; color=(:red, 0.2), R_b_w=nothing, T_b_w=nothing, kwargs...)
+    plots = []
     for panel in body.panels
-        Makie.plot!(ax, panel; color, R_b_w, T_b_w, kwargs...)
+        p = Makie.plot!(ax, panel; color, R_b_w, T_b_w, kwargs...)
+        push!(plots, p)
     end
+    return plots
 end
 
 function Makie.plot(panel::VortexStepMethod.Panel; size = (1200, 800), kwargs...)