source code
# The Computer Language Benchmarks Game
# https://salsa.debian.org/benchmarksgame-team/benchmarksgame/
#
# Contributed by Andrei Fomiga, Stefan Karpinski, Viral B. Shah, Jeff
# Bezanson, smallnamespaces, and Adam Beckmeyer.
#
# This implementation was specifically written so that for-loops have
# integer-literal ranges as their iterable. This makes the number of
# loop iterations available as a compile-time constant, so llvm can
# unroll them. To convince llvm to do so, this script should be run
# with the evironment variable:
# JULIA_LLVM_ARGS='-unroll-threshold=500'.
using Printf, Base.Cartesian
const SOLAR_MASS = 4 * pi * pi
const DAYS_PER_YEAR = 365.24
# Precalculate the pairs of bodies that must be compared so that it
# doesn't have to be done each loop.
const PAIRS = Tuple((i, j) for i=1:4 for j=i+1:5)
# Use a struct instead of mutable struct since a struct can be stored
# inline in an array avoiding the overhead of following a pointer
struct Body
x::NTuple{3,Float64}
v::NTuple{3,Float64}
m::Float64
end
function init_sun(bodies)
p = (0.0, 0.0, 0.0)
for b in bodies
p = p .- b.v .* b.m
end
Body((0.0, 0.0, 0.0), p ./ SOLAR_MASS, SOLAR_MASS)
end
# Advance all bodies in the system by one timestep of 0.01. This
# function always uses a timestep of 0.01 and assumes that there are
# exactly 5 bodies in the system.
@inline function advance!(bodies)
# In a system with 5 bodies, there are 10 unique pairs of
# bodies. Δx holds the difference in position between these pairs
# of bodies.
Δx = @ntuple 10 k-> @inbounds bodies[PAIRS[k][1]].x .- bodies[PAIRS[k][2]].x
dsq = @ntuple 10 k-> sum(Δx[k] .* Δx[k])
# When called with @fastmath, 1 / sqrt(x::Float32) will use
# SSE single-precision rsqrt approximation followed by a single
# iteration of the Newton-Raphson method. This, followed by an
# additional double-precision Newton-Raphson iteration gives
# sufficient precision for this problem and is significantly
# faster than double-precision division and sqrt.
rd = @ntuple 10 k-> @fastmath Float64(1 / sqrt(Float32(dsq[k])))
# This is a Newton-Raphson iteration.
rd = @ntuple 10 k-> 1.5rd[k] - 0.5dsq[k] * rd[k] * (rd[k] * rd[k])
# Alternatively 0.01rd[k] / dsq[k] may be faster. This is what
# other fast implementations do, but on my machine the 2
# multiplications are faster than a single division.
mag = @ntuple 10 k-> 0.01rd[k] * (rd[k] * rd[k])
# Update the velocities of each body using the precalculated mag.
k = 1
@inbounds for i=1:4
bi = bodies[i]
# For body i, since velocity is the only part of the object
# changing on each iteration, it's more efficient to update
# outside the vector.
iv = bi.v
for j=i+1:5
iv = iv .- Δx[k] .* (mag[k] * bodies[j].m)
bodies[j] = Body(bodies[j].x,
bodies[j].v .+ Δx[k] .* (mag[k] * bi.m),
bodies[j].m)
k += 1
end
bodies[i] = Body(bi.x, iv, bi.m)
end
# Advance body positions using the updated velocities.
@inbounds for i=1:5
bi = bodies[i]
bodies[i] = Body(bi.x .+ bi.v .* 0.01, bi.v, bi.m)
end
end
# Total energy of the system
function energy(bodies)
e = 0.0
# Kinetic energy of bodies
@inbounds for b in bodies
e += 0.5b.m * sum(b.v .* b.v)
end
# Potential energy between body i and body j
@inbounds for (i, j) in PAIRS
Δx = bodies[i].x .- bodies[j].x
e -= bodies[i].m * bodies[j].m / √sum(Δx .* Δx)
end
e
end
# Mutate bodies array according to symplectic integrator in advance!
# for n iterations.
nbody!(bodies, n) = for i=1:n
advance!(bodies)
end
# Doing the allocation of the Vector{Body} as a global constant
# instead of within the nbody! function speeds up inference
# considerably. Inference takes less than 60% of the time it would
# otherwise for an overall speedup of 2%-3%.
const bodies = [
# Jupiter
Body(( 4.84143144246472090e+0, # x
-1.16032004402742839e+0, # y
-1.03622044471123109e-1), # z
( 1.66007664274403694e-3DAYS_PER_YEAR, # vx
7.69901118419740425e-3DAYS_PER_YEAR, # vy
-6.90460016972063023e-5DAYS_PER_YEAR), # vz
9.54791938424326609e-4SOLAR_MASS) # mass
# Saturn
Body(( 8.34336671824457987e+0,
4.12479856412430479e+0,
-4.03523417114321381e-1),
(-2.76742510726862411e-3DAYS_PER_YEAR,
4.99852801234917238e-3DAYS_PER_YEAR,
2.30417297573763929e-5DAYS_PER_YEAR),
2.85885980666130812e-4SOLAR_MASS)
# Uranus
Body(( 1.28943695621391310e+1,
-1.51111514016986312e+1,
-2.23307578892655734e-1),
( 2.96460137564761618e-3DAYS_PER_YEAR,
2.37847173959480950e-3DAYS_PER_YEAR,
-2.96589568540237556e-5DAYS_PER_YEAR),
4.36624404335156298e-5SOLAR_MASS)
# Neptune
Body(( 1.53796971148509165e+1,
-2.59193146099879641e+1,
1.79258772950371181e-1),
( 2.68067772490389322e-3DAYS_PER_YEAR,
1.62824170038242295e-3DAYS_PER_YEAR,
-9.51592254519715870e-5DAYS_PER_YEAR),
5.15138902046611451e-5SOLAR_MASS)
]
push!(bodies, init_sun(bodies))
if !isinteractive()
@printf("%.9f\n", energy(bodies))
nbody!(bodies, parse(Int, ARGS[1]))
@printf("%.9f\n", energy(bodies))
end