Skip to content

update watson example to create a video and beautify the plot #58

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
leonhaufe merged 2 commits into from
Jan 28, 2025
Merged

update watson example to create a video and beautify the plot #58

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
3e35714
update watson example to create a video and beautify the plot
moritz155 Jan 28, 2025
83f4a6e
Merge branch 'main' into feature/improve-watson-example
leonhaufe Jan 28, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
44 changes: 40 additions & 4 deletions test/watson_example.jl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,17 @@ function projectile_simulation()
total_distance=Float64[],
time_of_flight=Float64[]
)
for d in dicts
dicts_distint = find_minimal_distinct_params(dicts)
for d in dicts_distint
results = exec_simulation(d, results)
end
results_obs = Observable(results)

fig = parallelplot(results_obs, curve=true, figure = (size = (1000, 600),))
save("projectile_simulation_initial.png", fig)
record(fig, "projectile_simulation.mp4", 1:length(dicts), framerate = 1) do t
results_obs[] = exec_simulation(dicts[t], results)
end
return results
end

Expand All @@ -25,7 +33,6 @@ function prepare_simulation()
launch_angles = 0:45:90 |> collect
air_resistances = 0.0:0.5:1.0 |> collect
gravities = 9.0:0.5:10.0 |> collect
display(initial_velocities)
allparams = Dict(
"initial_velocities" => initial_velocities,
"launch_angles" => launch_angles,
Expand Down Expand Up @@ -69,14 +76,43 @@ function exec_simulation(d::Dict, results)
return results
end

function find_minimal_distinct_params(arr_of_dicts)
# Get all unique values for each parameter
params = keys(first(arr_of_dicts))
param_values = Dict(
param => unique([Float64(d[param]) for d in arr_of_dicts]) # Ensure all values are Float64
for param in params
)

# Only two different values for each parameter
minimal_values = Dict(
param => param_values[param][1:min(2, length(param_values[param]))]
for param in params
)

# Create minimal set with proper types
result = Vector{Dict{String, Float64}}() # Specify the exact type of the result
push!(result, Dict(param => minimal_values[param][1] for param in params))

for param in params
if length(minimal_values[param]) > 1
new_dict = copy(result[1]) # Copy the base dict
new_dict[param] = minimal_values[param][2] # Set the second value for the current parameter
push!(result, new_dict)
end
end

return result
end

# Main execution
function main()
results = projectile_simulation()
println("Total parameter combinations: ", nrow(results))
println("\nSample results:")
display(first(results, 5))
fig = parallelplot(results, color_feature=5)
save("projectile_simulation.png", fig)
fig = parallelplot(results)
save("projectile_simulation_final.png", fig)
end

# Run the simulation
Expand Down
Loading