add strength_factor to Gravitational constructor

byuflowlab · Jan 31, 2025 · 8d9990d · 8d9990d
1 parent b153c7d
commit 8d9990d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/gravitational.jl b/test/gravitational.jl
@@ -31,11 +31,12 @@ function Gravitational(bodies::Matrix)
     return Gravitational(bodies2,potential)
 end
 
-function generate_gravitational(seed, n_bodies; radius_factor=0.1, TF=Float64)
+function generate_gravitational(seed, n_bodies; radius_factor=0.1, strength_factor=1/n_bodies, TF=Float64)
     Random.seed!(123)
     bodies = rand(TF,8,n_bodies)
     bodies[4,:] ./= (n_bodies^(1/3)*2)
     bodies[4,:] .*= radius_factor
+    bodies[5,:] .*= strength_factor
     system = Gravitational(bodies)
 end