mo8it/ReCo.jl
1
0
Fork 0
mirror of https://gitlab.rlp.net/mobitar/ReCo.jl.git synced 2024-11-08 22:21:08 +00:00
Code Issues Releases Activity

Fixed verlet_and_cell_lists

Browse source
This commit is contained in:
Mo8it 2022-02-01 22:57:56 +01:00
parent f61d06ce1e
commit 341b1e8a85
10 changed files with 30 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
.gitignore vendored
View file

@ -1,4 +1,4 @@
#Manifest.toml Manifest.toml
exports exports
# Jupyter notebooks # Jupyter notebooks

18
Manifest.toml
View file

@ -131,10 +131,10 @@ uuid = "69e1c6dd-3888-40e6-b3c8-31ac5f578864"
version = "0.7.2" version = "0.7.2"
[[deps.ChainRules]] [[deps.ChainRules]]
deps = ["ChainRulesCore", "Compat", "IrrationalConstants", "LinearAlgebra", "Random", "RealDot", "Statistics"] deps = ["ChainRulesCore", "Compat", "IrrationalConstants", "LinearAlgebra", "Random", "RealDot", "SparseArrays", "Statistics"]
git-tree-sha1 = "3ce62c97bd166b8c810889fba131531c61e41dc8" git-tree-sha1 = "849d4cb467ea3ecbbd3efe68dacd36f9429b543c"
uuid = "082447d4-558c-5d27-93f4-14fc19e9eca2" uuid = "082447d4-558c-5d27-93f4-14fc19e9eca2"
version = "1.25.0" version = "1.26.0"
[[deps.ChainRulesCore]] [[deps.ChainRulesCore]]
deps = ["Compat", "LinearAlgebra", "SparseArrays"] deps = ["Compat", "LinearAlgebra", "SparseArrays"]
@ -294,9 +294,9 @@ uuid = "8ba89e20-285c-5b6f-9357-94700520ee1b"
[[deps.Distributions]] [[deps.Distributions]]
deps = ["ChainRulesCore", "DensityInterface", "FillArrays", "LinearAlgebra", "PDMats", "Printf", "QuadGK", "Random", "SparseArrays", "SpecialFunctions", "Statistics", "StatsBase", "StatsFuns", "Test"] deps = ["ChainRulesCore", "DensityInterface", "FillArrays", "LinearAlgebra", "PDMats", "Printf", "QuadGK", "Random", "SparseArrays", "SpecialFunctions", "Statistics", "StatsBase", "StatsFuns", "Test"]
git-tree-sha1 = "c6dd4a56078a7760c04b882d9d94a08a4669598d" git-tree-sha1 = "24d26ca2197c158304ab2329af074fbe14c988e4"
uuid = "31c24e10-a181-5473-b8eb-7969acd0382f" uuid = "31c24e10-a181-5473-b8eb-7969acd0382f"
version = "0.25.44" version = "0.25.45"
[[deps.DocStringExtensions]] [[deps.DocStringExtensions]]
deps = ["LibGit2"] deps = ["LibGit2"]
@ -456,9 +456,9 @@ version = "3.4.1"
[[deps.GLFW_jll]] [[deps.GLFW_jll]]
deps = ["Artifacts", "JLLWrappers", "Libdl", "Libglvnd_jll", "Pkg", "Xorg_libXcursor_jll", "Xorg_libXi_jll", "Xorg_libXinerama_jll", "Xorg_libXrandr_jll"] deps = ["Artifacts", "JLLWrappers", "Libdl", "Libglvnd_jll", "Pkg", "Xorg_libXcursor_jll", "Xorg_libXi_jll", "Xorg_libXinerama_jll", "Xorg_libXrandr_jll"]
git-tree-sha1 = "0c603255764a1fa0b61752d2bec14cfbd18f7fe8" git-tree-sha1 = "51d2dfe8e590fbd74e7a842cf6d13d8a2f45dc01"
uuid = "0656b61e-2033-5cc2-a64a-77c0f6c09b89" uuid = "0656b61e-2033-5cc2-a64a-77c0f6c09b89"
version = "3.3.5+1" version = "3.3.6+0"
[[deps.GLMakie]] [[deps.GLMakie]]
deps = ["ColorTypes", "Colors", "FileIO", "FixedPointNumbers", "FreeTypeAbstraction", "GLFW", "GeometryBasics", "LinearAlgebra", "Makie", "Markdown", "MeshIO", "ModernGL", "Observables", "Printf", "Serialization", "ShaderAbstractions", "StaticArrays"] deps = ["ColorTypes", "Colors", "FileIO", "FixedPointNumbers", "FreeTypeAbstraction", "GLFW", "GeometryBasics", "LinearAlgebra", "Makie", "Markdown", "MeshIO", "ModernGL", "Observables", "Printf", "Serialization", "ShaderAbstractions", "StaticArrays"]
@ -674,9 +674,9 @@ version = "0.4.19"
[[deps.JLLWrappers]] [[deps.JLLWrappers]]
deps = ["Preferences"] deps = ["Preferences"]
git-tree-sha1 = "22df5b96feef82434b07327e2d3c770a9b21e023" git-tree-sha1 = "abc9885a7ca2052a736a600f7fa66209f96506e1"
uuid = "692b3bcd-3c85-4b1f-b108-f13ce0eb3210" uuid = "692b3bcd-3c85-4b1f-b108-f13ce0eb3210"
version = "1.4.0" version = "1.4.1"
[[deps.JSON]] [[deps.JSON]]
deps = ["Dates", "Mmap", "Parsers", "Unicode"] deps = ["Dates", "Mmap", "Parsers", "Unicode"]

6
graphics/elliptical_distance.jl
View file

@ -10,14 +10,14 @@ function gen_elliptical_distance_graphics()
fig = gen_figure() fig = gen_figure()
ax = Axis(fig[1, 1]; xlabel=L"x", ylabel=L"y", aspect=AxisAspect(1)) ax = Axis(fig[1, 1]; xlabel=L"x", ylabel=L"y")
elliptical_a_b_ratio = 0.4 elliptical_b_a_ratio = 0.4
as = 1:1:3 as = 1:1:3
for a in as for a in as
x = collect(LinRange(-a, a, 1000)) x = collect(LinRange(-a, a, 1000))
y = @. sqrt(a^2 - x^2) * elliptical_a_b_ratio y = @. sqrt(a^2 - x^2) * elliptical_b_a_ratio
append!(x, reverse(x)) append!(x, reverse(x))
append!(y, reverse(-y)) append!(y, reverse(-y))
lines!(x, y; label=L"a = %$a") lines!(x, y; label=L"a = %$a")

6
graphics/verlet_and_cell_lists.jl
View file

@ -5,7 +5,7 @@ using StaticArrays: SVector
using ReCo: ReCo using ReCo: ReCo
function gen_verlet_and_cell_lists_graphics() function gen_verlet_and_cell_lists_graphics()
Random.seed!(23) Random.seed!(3)
box_length = 100 box_length = 100
@ -26,11 +26,11 @@ function gen_verlet_and_cell_lists_graphics()
circle(Point(reference_particle_x, reference_particle_y), R_particle, :fill) circle(Point(reference_particle_x, reference_particle_y), R_particle, :fill)
reference_particle_c = SVector(reference_particle_x, reference_particle_y) reference_particle_c = SVector(reference_particle_x, reference_particle_y)
N = 91 N = 107
particle_cs = Vector{SVector{2,Float64}}(undef, N) particle_cs = Vector{SVector{2,Float64}}(undef, N)
distance_limit = 0.99 * 2 * particle_radius distance_limit = 0.98 * 2 * R_particle
x = y = distance = 0.0 x = y = distance = 0.0

6
src/RL/EnvHelper.jl
View file

@ -9,7 +9,7 @@ struct EnvHelperSharedProps{H<:AbstractHook}
n_actions_updates_per_episode::Int64 n_actions_updates_per_episode::Int64
reward_normalization::Float64 reward_normalization::Float64
elliptical_a_b_ratio::Float64 elliptical_b_a_ratio::Float64
n_particles::Int64 n_particles::Int64
@ -25,7 +25,7 @@ struct EnvHelperSharedProps{H<:AbstractHook}
hook::H, hook::H,
n_steps_before_actions_update::Int64, n_steps_before_actions_update::Int64,
n_actions_updates_per_episode::Int64, n_actions_updates_per_episode::Int64,
elliptical_a_b_ratio::Float64, elliptical_b_a_ratio::Float64,
n_particles::Int64, n_particles::Int64,
) where {H<:AbstractHook} ) where {H<:AbstractHook}
reward_normalization = n_particles * n_actions_updates_per_episode reward_normalization = n_particles * n_actions_updates_per_episode
@ -37,7 +37,7 @@ struct EnvHelperSharedProps{H<:AbstractHook}
n_steps_before_actions_update, n_steps_before_actions_update,
n_actions_updates_per_episode, n_actions_updates_per_episode,
reward_normalization, reward_normalization,
elliptical_a_b_ratio, elliptical_b_a_ratio,
n_particles, n_particles,
fill(0, n_particles), fill(0, n_particles),
fill(0, n_particles), fill(0, n_particles),

4
src/RL/Envs/COMCompass.jl
View file

@ -61,7 +61,7 @@ mutable struct COMCompassEnvHelper <: EnvHelper
function COMCompassEnvHelper(shared::EnvHelperSharedProps, half_box_len::Float64) function COMCompassEnvHelper(shared::EnvHelperSharedProps, half_box_len::Float64)
max_elliptical_distance = sqrt( max_elliptical_distance = sqrt(
half_box_len^2 + (half_box_len / shared.elliptical_a_b_ratio)^2 half_box_len^2 + (half_box_len / shared.elliptical_b_a_ratio)^2
) )
return new( return new(
@ -145,7 +145,7 @@ function update_reward!(
env_helper.center_of_mass, env_helper.center_of_mass,
env_helper.gyration_tensor_eigvec_to_smaller_eigval, env_helper.gyration_tensor_eigvec_to_smaller_eigval,
env_helper.gyration_tensor_eigvec_to_bigger_eigval, env_helper.gyration_tensor_eigvec_to_bigger_eigval,
env_helper.shared.elliptical_a_b_ratio, env_helper.shared.elliptical_b_a_ratio,
env_helper.half_box_len, env_helper.half_box_len,
) )

4
src/RL/Envs/LocalCOMWithAdditionalShapeRewardEnv.jl
View file

@ -58,7 +58,7 @@ mutable struct LocalCOMWithAdditionalShapeRewardEnvHelper <: EnvHelper
shared::EnvHelperSharedProps, half_box_len::Float64, skin_radius::Float64 shared::EnvHelperSharedProps, half_box_len::Float64, skin_radius::Float64
) )
max_elliptical_distance = sqrt( max_elliptical_distance = sqrt(
half_box_len^2 + (half_box_len / shared.elliptical_a_b_ratio)^2 half_box_len^2 + (half_box_len / shared.elliptical_b_a_ratio)^2
) )
max_distance_to_local_center_of_mass = skin_radius max_distance_to_local_center_of_mass = skin_radius
@ -188,7 +188,7 @@ function update_reward!(
env_helper.center_of_mass, env_helper.center_of_mass,
env_helper.gyration_tensor_eigvec_to_smaller_eigval, env_helper.gyration_tensor_eigvec_to_smaller_eigval,
env_helper.gyration_tensor_eigvec_to_bigger_eigval, env_helper.gyration_tensor_eigvec_to_bigger_eigval,
env_helper.shared.elliptical_a_b_ratio, env_helper.shared.elliptical_b_a_ratio,
env_helper.half_box_len, env_helper.half_box_len,
) )

4
src/RL/Envs/OriginCompass.jl
View file

@ -61,7 +61,7 @@ mutable struct OriginCompassEnvHelper <: EnvHelper
function OriginCompassEnvHelper(shared::EnvHelperSharedProps, half_box_len::Float64) function OriginCompassEnvHelper(shared::EnvHelperSharedProps, half_box_len::Float64)
max_elliptical_distance = sqrt( max_elliptical_distance = sqrt(
half_box_len^2 + (half_box_len / shared.elliptical_a_b_ratio)^2 half_box_len^2 + (half_box_len / shared.elliptical_b_a_ratio)^2
) )
return new( return new(
@ -143,7 +143,7 @@ function update_reward!(
env_helper.center_of_mass, env_helper.center_of_mass,
env_helper.gyration_tensor_eigvec_to_smaller_eigval, env_helper.gyration_tensor_eigvec_to_smaller_eigval,
env_helper.gyration_tensor_eigvec_to_bigger_eigval, env_helper.gyration_tensor_eigvec_to_bigger_eigval,
env_helper.shared.elliptical_a_b_ratio, env_helper.shared.elliptical_b_a_ratio,
env_helper.half_box_len, env_helper.half_box_len,
) )

6
src/RL/RL.jl
View file

@ -66,7 +66,7 @@ end
function run_rl(; function run_rl(;
EnvType::Type{E}, EnvType::Type{E},
process_dir::String, process_dir::String,
elliptical_a_b_ratio::Float64=1.0, elliptical_b_a_ratio::Float64=1.0,
n_episodes::Int64=200, n_episodes::Int64=200,
episode_duration::Float64=50.0, episode_duration::Float64=50.0,
update_actions_at::Float64=0.1, update_actions_at::Float64=0.1,
@ -80,7 +80,7 @@ function run_rl(;
show_simulation_progress::Bool=true, show_simulation_progress::Bool=true,
n_episodes_before_env_helper_saving::Int64=10, n_episodes_before_env_helper_saving::Int64=10,
) where {E<:Env} ) where {E<:Env}
@assert 0.0 <= elliptical_a_b_ratio <= 1.0 @assert 0.0 <= elliptical_b_a_ratio <= 1.0
@assert n_episodes > 0 @assert n_episodes > 0
@assert episode_duration > 0 @assert episode_duration > 0
@assert update_actions_at in 0.001:0.001:episode_duration @assert update_actions_at in 0.001:0.001:episode_duration
@ -116,7 +116,7 @@ function run_rl(;
hook, hook,
n_steps_before_actions_update, n_steps_before_actions_update,
n_actions_updates_per_episode, n_actions_updates_per_episode,
elliptical_a_b_ratio, elliptical_b_a_ratio,
n_particles, n_particles,
) )

4
src/Shape.jl
View file

@ -161,7 +161,7 @@ function elliptical_distance(
COM::SVector{2,<:Real}, COM::SVector{2,<:Real},
gyration_tensor_eigvec_to_smaller_eigval::SVector{2,<:Real}, gyration_tensor_eigvec_to_smaller_eigval::SVector{2,<:Real},
gyration_tensor_eigvec_to_bigger_eigval::SVector{2,<:Real}, gyration_tensor_eigvec_to_bigger_eigval::SVector{2,<:Real},
elliptical_a_b_ratio::Real, elliptical_b_a_ratio::Real,
half_box_len::Real, half_box_len::Real,
) )
v′ = ReCo.restrict_coordinates(v - COM, half_box_len) v′ = ReCo.restrict_coordinates(v - COM, half_box_len)
@ -169,7 +169,7 @@ function elliptical_distance(
x = LA.dot(v′, gyration_tensor_eigvec_to_bigger_eigval) x = LA.dot(v′, gyration_tensor_eigvec_to_bigger_eigval)
y = LA.dot(v′, gyration_tensor_eigvec_to_smaller_eigval) y = LA.dot(v′, gyration_tensor_eigvec_to_smaller_eigval)
return sqrt(x^2 + (y / elliptical_a_b_ratio)^2) return sqrt(x^2 + (y / elliptical_b_a_ratio)^2)
end end
end # module end # module