side by side

Author: BatyLeo

Project.toml

@@ -10,6 +10,7 @@ ConstrainedShortestPaths = "b3798467-87dc-4d99-943d-35a1bd39e395"
 DataDeps = "124859b0-ceae-595e-8997-d05f6a7a8dfe"
 Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
+FileIO = "5789e2e9-d7fb-5bc7-8068-2c6fae9b9549"
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
 HiGHS = "87dc4568-4c63-4d18-b0c0-bb2238e4078b"
@@ -40,6 +41,7 @@ ConstrainedShortestPaths = "0.6.0"
 DataDeps = "0.7"
 Distributions = "0.25"
 DocStringExtensions = "0.9"
+FileIO = "1.17.0"
 Flux = "0.14, 0.15, 0.16"
 Graphs = "1.11"
 HiGHS = "1.9"

anticipative_animation.gif

@@ -109,9 +109,9 @@ indices of locations visited by that route (excluding the depot).
 function plot_routes(
     state::DVSPState,
     routes::Vector{Vector{Int}};
-    route_colors=nothing,
-    route_linewidth=3,  # Increased from 2 to 3
-    route_alpha=0.7,
+    # route_colors=nothing,
+    route_linewidth=2,  # Increased from 2 to 3
+    # route_alpha=0.7,
     show_route_labels=true,
     kwargs...,
 )
@@ -128,9 +128,9 @@ function plot_routes(
     y_depot = y[1]
 
     # Default route colors if not provided
-    if isnothing(route_colors)
-        route_colors = [:blue, :purple, :orange, :brown, :pink, :gray, :olive, :cyan]
-    end
+    # if isnothing(route_colors)
+    #     route_colors = [:blue, :purple, :orange, :brown, :pink, :gray, :olive, :cyan]
+    # end
 
     # Plot each route
     for (route_idx, route) in enumerate(routes)
@@ -140,15 +140,15 @@ function plot_routes(
             route_y = vcat(y_depot, y[route], y_depot)
 
             # Select color for this route
-            color = route_colors[(route_idx - 1) % length(route_colors) + 1]
+            # color = route_colors[(route_idx - 1) % length(route_colors) + 1]
 
             # Plot the route with more visible styling
             label = show_route_labels ? "Route $route_idx" : nothing
             plot!(
                 fig,
                 route_x,
                 route_y;
-                # color=color,
+                # color=:black,
                 linewidth=route_linewidth,
                 alpha=1.0,  # Make routes fully opaque
                 label=label,
@@ -674,3 +674,267 @@ function animate_epochs(
 
     return anim
 end
+
+"""
+Animate multiple solutions where each solution is provided as its own vector of
+`DataSample` objects (one per epoch). This treats each solution's `DataSample`
+as the canonical source for that column in the side-by-side animation.
+"""
+function animate_solutions_side_by_side(
+    solutions_data_samples::AbstractVector;
+    solution_names=nothing,
+    filename="dvsp_solutions_side_by_side.gif",
+    fps=1,
+    figsize=(1200, 600),
+    margin=0.1,
+    legend_margin_factor=0.15,
+    titlefontsize=14,
+    guidefontsize=12,
+    legendfontsize=11,
+    tickfontsize=10,
+    show_axis_labels=true,
+    show_cost_bar=true,
+    cost_bar_width=0.05,
+    cost_bar_margin=0.02,
+    cost_bar_color_palette=:turbo,
+    kwargs...,
+)
+    n_solutions = length(solutions_data_samples)
+    if n_solutions == 0
+        error("No solutions provided")
+    end
+
+    # Ensure all solution sequences have the same number of epochs
+    n_epochs = length(solutions_data_samples[1])
+    for (i, s) in enumerate(solutions_data_samples)
+        if length(s) != n_epochs
+            error(
+                "All solution DataSample vectors must have the same length. Solution $i has length $(length(s)) but expected $n_epochs",
+            )
+        end
+    end
+
+    if isnothing(solution_names)
+        solution_names = ["Solution $(i)" for i in 1:n_solutions]
+    end
+
+    # Collect global coordinates and start times across all solutions/epochs
+    all_coordinates = []
+    all_start_times = []
+    epoch_costs_per_solution = [Float64[] for _ in 1:n_solutions]
+
+    for j in 1:n_solutions
+        samples = solutions_data_samples[j]
+        for (t, sample) in enumerate(samples)
+            if !isnothing(sample.instance)
+                append!(all_coordinates, coordinate(sample.instance))
+                append!(all_start_times, start_time(sample.instance))
+
+                if sample.y_true isa BitMatrix
+                    routes = decode_bitmatrix_to_routes(sample.y_true)
+                else
+                    routes = sample.y_true isa Vector{Int} ? [sample.y_true] : sample.y_true
+                end
+                c = isnothing(routes) ? 0.0 : cost(sample.instance, routes)
+                push!(epoch_costs_per_solution[j], c)
+            else
+                push!(epoch_costs_per_solution[j], NaN)
+            end
+        end
+    end
+
+    if isempty(all_coordinates)
+        error("No valid coordinates found in solution data samples")
+    end
+
+    # Global limits
+    xlims = (
+        minimum(p.x for p in all_coordinates) - margin,
+        maximum(p.x for p in all_coordinates) + margin,
+    )
+
+    y_min = minimum(p.y for p in all_coordinates) - margin
+    y_max = maximum(p.y for p in all_coordinates) + margin
+    y_range = y_max - y_min
+    legend_margin = y_range * legend_margin_factor
+    ylims = (y_min, y_max + legend_margin)
+
+    clims = if !isempty(all_start_times)
+        (minimum(all_start_times), maximum(all_start_times))
+    else
+        (0.0, 1.0)
+    end
+
+    # Robust cumulative costs per solution
+    robust_cumulative = Vector{Vector{Float64}}(undef, n_solutions)
+    for j in 1:n_solutions
+        robust = Float64[]
+        s = 0.0
+        for c in epoch_costs_per_solution[j]
+            if !isnan(c)
+                s += c
+            end
+            push!(robust, s)
+        end
+        robust_cumulative[j] = robust
+    end
+
+    function has_routes_local(routes)
+        if isnothing(routes)
+            return false
+        elseif routes isa Vector{Vector{Int}}
+            return any(!isempty(route) for route in routes)
+        elseif routes isa Vector{Int}
+            return !isempty(routes)
+        elseif routes isa BitMatrix
+            return any(routes)
+        else
+            return false
+        end
+    end
+
+    anim = @animate for t in 1:n_epochs
+        col_plots = []
+        for j in 1:n_solutions
+            sample = solutions_data_samples[j][t]
+            state = sample.instance
+            routes = sample.y_true
+
+            if isnothing(state)
+                fig = plot(;
+                    xlims=xlims,
+                    ylims=ylims,
+                    title="$(solution_names[j]) - Epoch $t (No Data)",
+                    titlefontsize=titlefontsize,
+                    guidefontsize=guidefontsize,
+                    tickfontsize=tickfontsize,
+                    legend=false,
+                    kwargs...,
+                )
+            else
+                if has_routes_local(routes)
+                    fig = plot_routes(
+                        state,
+                        routes;
+                        xlims=xlims,
+                        ylims=ylims,
+                        clims=clims,
+                        title="$(solution_names[j]) - Epoch $(state.current_epoch)",
+                        titlefontsize=titlefontsize,
+                        guidefontsize=guidefontsize,
+                        legendfontsize=legendfontsize,
+                        tickfontsize=tickfontsize,
+                        show_axis_labels=show_axis_labels,
+                        markerstrokewidth=0.5,
+                        show_route_labels=false,
+                        show_colorbar=false,
+                        size=(floor(Int, figsize[1] / n_solutions), figsize[2]),
+                        kwargs...,
+                    )
+                else
+                    fig = plot_state(
+                        state;
+                        xlims=xlims,
+                        ylims=ylims,
+                        clims=clims,
+                        title="$(solution_names[j]) - Epoch $(state.current_epoch)",
+                        titlefontsize=titlefontsize,
+                        guidefontsize=guidefontsize,
+                        legendfontsize=legendfontsize,
+                        tickfontsize=tickfontsize,
+                        show_axis_labels=show_axis_labels,
+                        markerstrokewidth=0.5,
+                        size=(floor(Int, figsize[1] / n_solutions), figsize[2]),
+                        kwargs...,
+                    )
+                end
+            end
+
+            # cost bar
+            if show_cost_bar
+                current_cost = robust_cumulative[j][t]
+                max_cost = maximum([robust_cumulative[k][end] for k in 1:n_solutions])
+
+                x_min, x_max = xlims
+                x_range = x_max - x_min
+                bar_x_start = x_max - cost_bar_width * x_range
+                bar_x_end = x_max - cost_bar_margin * x_range
+
+                y_min, y_max = ylims
+                y_range = y_max - y_min
+                bar_y_start = y_min + 0.1 * y_range
+                bar_y_end = y_max - 0.1 * y_range
+                bar_height = bar_y_end - bar_y_start
+
+                if max_cost > 0
+                    filled_height = (current_cost / max_cost) * bar_height
+                else
+                    filled_height = 0.0
+                end
+
+                plot!(
+                    fig,
+                    [bar_x_start, bar_x_end, bar_x_end, bar_x_start, bar_x_start],
+                    [bar_y_start, bar_y_start, bar_y_end, bar_y_end, bar_y_start];
+                    seriestype=:shape,
+                    color=:white,
+                    alpha=0.8,
+                    linecolor=:black,
+                    linewidth=2,
+                    label="",
+                )
+
+                if filled_height > 0
+                    cmap = Plots.cgrad(cost_bar_color_palette)
+                    ratio = max_cost > 0 ? current_cost / max_cost : 0.0
+                    color_at_val = Plots.get(cmap, ratio)
+                    plot!(
+                        fig,
+                        [bar_x_start, bar_x_end, bar_x_end, bar_x_start, bar_x_start],
+                        [
+                            bar_y_start,
+                            bar_y_start,
+                            bar_y_start + filled_height,
+                            bar_y_start + filled_height,
+                            bar_y_start,
+                        ];
+                        seriestype=:shape,
+                        color=color_at_val,
+                        alpha=0.7,
+                        linecolor=:darkred,
+                        linewidth=1,
+                        label="",
+                    )
+                end
+
+                cost_text_y = bar_y_start + filled_height + 0.02 * y_range
+                if cost_text_y > bar_y_end
+                    cost_text_y = bar_y_end
+                end
+                plot!(
+                    fig,
+                    [bar_x_start + (bar_x_end - bar_x_start) / 2],
+                    [cost_text_y];
+                    seriestype=:scatter,
+                    markersize=0,
+                    label="",
+                    annotations=(
+                        bar_x_start - 0.04 * x_range,
+                        cost_text_y,
+                        (@sprintf("%.1f", current_cost), :center, guidefontsize),
+                    ),
+                )
+            end
+
+            push!(col_plots, fig)
+        end
+
+        combined = plot(
+            col_plots...; layout=(1, n_solutions), size=figsize, link=:both, clims=clims
+        )
+        combined
+    end
+
+    gif(anim, filename; fps=fps)
+    return anim
+end