Rosetta Code

N-body problem: Difference between revisions

Page Discussion

N-body problem (view source)

Revision as of 14:37, 24 February 2022

15,184 bytes added ,  2 years ago
m
→‎{{header|Phix}}: added syntax colouring, marked p2js compatible
(Added 11l)
m (→‎{{header|Phix}}: added syntax colouring, marked p2js compatible)
Line 4,966:
=={{header|Phix}}==
{{trans|Kotlin}}
<!--<lang Phix>sequence origin = {0,0,0}(phixonline)-->
<span style="color: #008080;">with</span> <span style="color: #008080;">javascript_semantics</span>
<span style="color: #008080;">constant</span> <span style="color: #000000;">origin</span> <span style="color: #0000FF;">=</span> <span style="color: #0000FF;">{</span><span style="color: #000000;">0</span><span style="color: #0000FF;">,</span><span style="color: #000000;">0</span><span style="color: #0000FF;">,</span><span style="color: #000000;">0</span><span style="color: #0000FF;">},</span>
constant nbody = """
<span style="color: #000000;">nbody</span> <span style="color: #0000FF;">=</span> <span style="color: #008000;">"""
0.01 3 20
0.01 3 20
1
1
0 0 0
0.01 0 0
0.101 0 0
0.1
1 1 0
0 1 01 0.02
0 0 0.02
0.001
0.001
0 1 1
0 1 1
0.01 -0.01 -0.01
0.01 -0.01 -0.01
"""
"""</span><span style="color: #0000FF;">,</span>
 
<span style="color: #000000;">lines</span> <span style="color: #0000FF;">=</span> <span style="color: #7060A8;">split</span><span style="color: #0000FF;">(</span><span style="color: #000000;">nbody</span><span style="color: #0000FF;">,</span><span style="color: #008000;">"\n"</span><span style="color: #0000FF;">),</span>
sequence lines = split(nbody,"\n")
<span style="color: #0000FF;">{{</span><span style="color: #000000;">grav_constant</span><span style="color: #0000FF;">,</span><span style="color: #000000;">bodies</span><span style="color: #0000FF;">,</span><span style="color: #000000;">timeSteps</span><span style="color: #0000FF;">}}</span> <span style="color: #0000FF;">=</span> <span style="color: #7060A8;">scanf</span><span style="color: #0000FF;">(</span><span style="color: #000000;">lines</span><span style="color: #0000FF;">[</span><span style="color: #000000;">1</span><span style="color: #0000FF;">],</span><span style="color: #008000;">"%f %f %f"</span><span style="color: #0000FF;">)</span>
atom {{grav_constant,bodies,timeSteps}} = scanf(lines[1],"%f %f %f")
<span style="color: #004080;">sequence</span> <span style="color: #000000;">masses</span> <span style="color: #0000FF;">=</span> <span style="color: #7060A8;">repeat</span><span style="color: #0000FF;">(</span><span style="color: #000000;">0</span><span style="color: #0000FF;">,</span><span style="color: #000000;">bodies</span><span style="color: #0000FF;">),</span>
sequence {masses, positions, velocities, accelerations} @= repeat(0,bodies)
<span style="color: #000000;">positions</span> <span style="color: #0000FF;">=</span> <span style="color: #7060A8;">repeat</span><span style="color: #0000FF;">(</span><span style="color: #000000;">0</span><span style="color: #0000FF;">,</span><span style="color: #000000;">bodies</span><span style="color: #0000FF;">),</span>
for i=1 to bodies do
<span style="color: #000000;">velocities</span> <span style="color: #0000FF;">=</span> <span style="color: #7060A8;">repeat</span><span style="color: #0000FF;">(</span><span style="color: #000000;">0</span><span style="color: #0000FF;">,</span><span style="color: #000000;">bodies</span><span style="color: #0000FF;">),</span>
{{masses[i]}} = scanf(lines[i*3-1],"%f")
<span style="color: #000000;">accelerations</span> <span style="color: #0000FF;">=</span> <span style="color: #7060A8;">repeat</span><span style="color: #0000FF;">(</span><span style="color: #000000;">0</span><span style="color: #0000FF;">,</span><span style="color: #000000;">bodies</span><span style="color: #0000FF;">)</span>
{positions[i]} = scanf(lines[i*3],"%f %f %f")
<span style="color: #008080;">for</span> <span style="color: #000000;">i</span><span style="color: #0000FF;">=</span><span style="color: #000000;">1</span> <span style="color: #008080;">to</span> <span style="color: #000000;">bodies</span> <span style="color: #008080;">do</span>
{velocities[i]} = scanf(lines[i*3+1],"%f %f %f")
<span style="color: #0000FF;">{{</span><span style="color: #000000;">masses</span><span style="color: #0000FF;">[</span><span style="color: #000000;">i</span><span style="color: #0000FF;">]}}</span> <span style="color: #0000FF;">=</span> <span style="color: #7060A8;">scanf</span><span style="color: #0000FF;">(</span><span style="color: #000000;">lines</span><span style="color: #0000FF;">[</span><span style="color: #000000;">i</span><span style="color: #0000FF;">*</span><span style="color: #000000;">3</span><span style="color: #0000FF;">-</span><span style="color: #000000;">1</span><span style="color: #0000FF;">],</span><span style="color: #008000;">"%f"</span><span style="color: #0000FF;">)</span>
end for
<span style="color: #0000FF;">{</span><span style="color: #000000;">positions</span><span style="color: #0000FF;">[</span><span style="color: #000000;">i</span><span style="color: #0000FF;">]}</span> <span style="color: #0000FF;">=</span> <span style="color: #7060A8;">scanf</span><span style="color: #0000FF;">(</span><span style="color: #000000;">lines</span><span style="color: #0000FF;">[</span><span style="color: #000000;">i</span><span style="color: #0000FF;">*</span><span style="color: #000000;">3</span><span style="color: #0000FF;">],</span><span style="color: #008000;">"%f %f %f"</span><span style="color: #0000FF;">)</span>
printf(1,"Body : x y z |")
<span style="color: #0000FF;">{</span><span style="color: #000000;">velocities</span><span style="color: #0000FF;">[</span><span style="color: #000000;">i</span><span style="color: #0000FF;">]}</span> <span style="color: #0000FF;">=</span> <span style="color: #7060A8;">scanf</span><span style="color: #0000FF;">(</span><span style="color: #000000;">lines</span><span style="color: #0000FF;">[</span><span style="color: #000000;">i</span><span style="color: #0000FF;">*</span><span style="color: #000000;">3</span><span style="color: #0000FF;">+</span><span style="color: #000000;">1</span><span style="color: #0000FF;">],</span><span style="color: #008000;">"%f %f %f"</span><span style="color: #0000FF;">)</span>
printf(1," vx vy vz\n")
<span style="color: #008080;">end</span> <span style="color: #008080;">for</span>
 
<span style="color: #7060A8;">printf</span><span style="color: #0000FF;">(</span><span style="color: #000000;">1</span><span style="color: #0000FF;">,</span><span style="color: #008000;">"Body : x y z |"</span><span style="color: #0000FF;">)</span>
function vmod(sequence a)
<span style="color: #7060A8;">printf</span><span style="color: #0000FF;">(</span><span style="color: #000000;">1</span><span style="color: #0000FF;">,</span><span style="color: #008000;">" vx vy vz\n"</span><span style="color: #0000FF;">)</span>
return sqrt(sum(sq_mul(a,a)))
end function
<span style="color: #008080;">function</span> <span style="color: #000000;">vmod</span><span style="color: #0000FF;">(</span><span style="color: #004080;">sequence</span> <span style="color: #000000;">a</span><span style="color: #0000FF;">)</span>
 
<span style="color: #008080;">return</span> <span style="color: #7060A8;">sqrt</span><span style="color: #0000FF;">(</span><span style="color: #7060A8;">sum</span><span style="color: #0000FF;">(</span><span style="color: #7060A8;">sq_mul</span><span style="color: #0000FF;">(</span><span style="color: #000000;">a</span><span style="color: #0000FF;">,</span><span style="color: #000000;">a</span><span style="color: #0000FF;">)))</span>
procedure computeAccelerations()
<span style="color: #008080;">end</span> <span style="color: #008080;">function</span>
for i=1 to bodies do
sequence ai = origin
<span style="color: #008080;">procedure</span> <span style="color: #000000;">computeAccelerations</span><span style="color: #0000FF;">()</span>
for j=1 to bodies do
<span style="color: #008080;">for</span> <span style="color: #000000;">i</span><span style="color: #0000FF;">=</span><span style="color: #000000;">1</span> <span style="color: #008080;">to</span> <span style="color: #000000;">bodies</span> <span style="color: #008080;">do</span>
if i!=j then
<span style="color: #004080;">sequence</span> <span style="color: #000000;">ai</span> <span style="color: #0000FF;">=</span> <span style="color: #000000;">origin</span>
sequence t = sq_sub(positions[j],positions[i])
<span style="color: #008080;">for</span> <span style="color: #000000;">j</span><span style="color: #0000FF;">=</span><span style="color: #000000;">1</span> <span style="color: #008080;">to</span> <span style="color: #000000;">bodies</span> <span style="color: #008080;">do</span>
atom temp = grav_constant*masses[j]/power(vmod(t),3)
<span style="color: #008080;">if</span> <span style="color: #000000;">i</span><span style="color: #0000FF;">!=</span><span style="color: #000000;">j</span> <span style="color: #008080;">then</span>
ai = sq_add(ai,sq_mul(t,temp))
<span style="color: #004080;">sequence</span> <span style="color: #000000;">t</span> <span style="color: #0000FF;">=</span> <span style="color: #7060A8;">sq_sub</span><span style="color: #0000FF;">(</span><span style="color: #000000;">positions</span><span style="color: #0000FF;">[</span><span style="color: #000000;">j</span><span style="color: #0000FF;">],</span><span style="color: #000000;">positions</span><span style="color: #0000FF;">[</span><span style="color: #000000;">i</span><span style="color: #0000FF;">])</span>
end if
<span style="color: #004080;">atom</span> <span style="color: #000000;">temp</span> <span style="color: #0000FF;">=</span> <span style="color: #000000;">grav_constant</span><span style="color: #0000FF;">*</span><span style="color: #000000;">masses</span><span style="color: #0000FF;">[</span><span style="color: #000000;">j</span><span style="color: #0000FF;">]/</span><span style="color: #7060A8;">power</span><span style="color: #0000FF;">(</span><span style="color: #000000;">vmod</span><span style="color: #0000FF;">(</span><span style="color: #000000;">t</span><span style="color: #0000FF;">),</span><span style="color: #000000;">3</span><span style="color: #0000FF;">)</span>
end for
<span style="color: #000000;">ai</span> <span style="color: #0000FF;">=</span> <span style="color: #7060A8;">sq_add</span><span style="color: #0000FF;">(</span><span style="color: #000000;">ai</span><span style="color: #0000FF;">,</span><span style="color: #7060A8;">sq_mul</span><span style="color: #0000FF;">(</span><span style="color: #000000;">t</span><span style="color: #0000FF;">,</span><span style="color: #000000;">temp</span><span style="color: #0000FF;">))</span>
accelerations[i] = ai
<span style="color: #008080;">end</span> <span style="color: #008080;">if</span>
end for
<span style="color: #008080;">end</span> <span style="color: #008080;">for</span>
end procedure
<span style="color: #000000;">accelerations</span><span style="color: #0000FF;">[</span><span style="color: #000000;">i</span><span style="color: #0000FF;">]</span> <span style="color: #0000FF;">=</span> <span style="color: #000000;">ai</span>
<span style="color: #008080;">end</span> <span style="color: #008080;">for</span>
procedure computePositions()
<span style="color: #008080;">end</span> <span style="color: #008080;">procedure</span>
positions = sq_add(positions,sq_add(velocities,sq_mul(accelerations,0.5)))
end procedure
<span style="color: #008080;">procedure</span> <span style="color: #000000;">computePositions</span><span style="color: #0000FF;">()</span>
 
<span style="color: #000000;">positions</span> <span style="color: #0000FF;">=</span> <span style="color: #7060A8;">sq_add</span><span style="color: #0000FF;">(</span><span style="color: #000000;">positions</span><span style="color: #0000FF;">,</span><span style="color: #7060A8;">sq_add</span><span style="color: #0000FF;">(</span><span style="color: #000000;">velocities</span><span style="color: #0000FF;">,</span><span style="color: #7060A8;">sq_mul</span><span style="color: #0000FF;">(</span><span style="color: #000000;">accelerations</span><span style="color: #0000FF;">,</span><span style="color: #000000;">0.5</span><span style="color: #0000FF;">)))</span>
procedure computeVelocities()
<span style="color: #008080;">end</span> <span style="color: #008080;">procedure</span>
velocities = sq_add(velocities,accelerations)
end procedure
<span style="color: #008080;">procedure</span> <span style="color: #000000;">computeVelocities</span><span style="color: #0000FF;">()</span>
 
<span style="color: #000000;">velocities</span> <span style="color: #0000FF;">=</span> <span style="color: #7060A8;">sq_add</span><span style="color: #0000FF;">(</span><span style="color: #000000;">velocities</span><span style="color: #0000FF;">,</span><span style="color: #000000;">accelerations</span><span style="color: #0000FF;">)</span>
procedure resolveCollisions()
<span style="color: #008080;">end</span> <span style="color: #008080;">procedure</span>
for i=1 to bodies do
for j=i+1 to bodies do
<span style="color: #008080;">procedure</span> <span style="color: #000000;">resolveCollisions</span><span style="color: #0000FF;">()</span>
if positions[i]=positions[j] then
<span style="color: #008080;">for</span> <span style="color: #000000;">i</span><span style="color: #0000FF;">=</span><span style="color: #000000;">1</span> <span style="color: #008080;">to</span> <span style="color: #000000;">bodies</span> <span style="color: #008080;">do</span>
{velocities[i],velocities[j]} = {velocities[j],velocities[i]}
<span style="color: #008080;">for</span> <span style="color: #000000;">j</span><span style="color: #0000FF;">=</span><span style="color: #000000;">i</span><span style="color: #0000FF;">+</span><span style="color: #000000;">1</span> <span style="color: #008080;">to</span> <span style="color: #000000;">bodies</span> <span style="color: #008080;">do</span>
end if
<span style="color: #008080;">if</span> <span style="color: #000000;">positions</span><span style="color: #0000FF;">[</span><span style="color: #000000;">i</span><span style="color: #0000FF;">]=</span><span style="color: #000000;">positions</span><span style="color: #0000FF;">[</span><span style="color: #000000;">j</span><span style="color: #0000FF;">]</span> <span style="color: #008080;">then</span>
end for
<span style="color: #0000FF;">{</span><span style="color: #000000;">velocities</span><span style="color: #0000FF;">[</span><span style="color: #000000;">i</span><span style="color: #0000FF;">],</span><span style="color: #000000;">velocities</span><span style="color: #0000FF;">[</span><span style="color: #000000;">j</span><span style="color: #0000FF;">]}</span> <span style="color: #0000FF;">=</span> <span style="color: #0000FF;">{</span><span style="color: #000000;">velocities</span><span style="color: #0000FF;">[</span><span style="color: #000000;">j</span><span style="color: #0000FF;">],</span><span style="color: #000000;">velocities</span><span style="color: #0000FF;">[</span><span style="color: #000000;">i</span><span style="color: #0000FF;">]}</span>
end for
<span style="color: #008080;">end</span> <span style="color: #008080;">if</span>
end procedure
<span style="color: #008080;">end</span> <span style="color: #008080;">for</span>
 
<span style="color: #008080;">end</span> <span style="color: #008080;">for</span>
procedure simulate()
<span style="color: #008080;">end</span> <span style="color: #008080;">procedure</span>
computeAccelerations()
computePositions()
<span style="color: #008080;">procedure</span> <span style="color: #000000;">simulate</span><span style="color: #0000FF;">()</span>
computeVelocities()
<span style="color: #000000;">computeAccelerations</span><span style="color: #0000FF;">()</span>
resolveCollisions()
<span style="color: #000000;">computePositions</span><span style="color: #0000FF;">()</span>
end procedure
<span style="color: #000000;">computeVelocities</span><span style="color: #0000FF;">()</span>
<span style="color: #000000;">resolveCollisions</span><span style="color: #0000FF;">()</span>
procedure printResults()
<span style="color: #008080;">end</span> <span style="color: #008080;">procedure</span>
string fmt = "Body %d : %9.6f %9.6f %9.6f | %9.6f %9.6f %9.6f\n"
for i=1 to bodies do
<span style="color: #008080;">procedure</span> <span style="color: #000000;">printResults</span><span style="color: #0000FF;">()</span>
printf(1,fmt,{i}&positions[i]&velocities[i])
<span style="color: #004080;">string</span> <span style="color: #000000;">fmt</span> <span style="color: #0000FF;">=</span> <span style="color: #008000;">"Body %d : %9.6f %9.6f %9.6f | %9.6f %9.6f %9.6f\n"</span>
end for
<span style="color: #008080;">for</span> <span style="color: #000000;">i</span><span style="color: #0000FF;">=</span><span style="color: #000000;">1</span> <span style="color: #008080;">to</span> <span style="color: #000000;">bodies</span> <span style="color: #008080;">do</span>
end procedure
<span style="color: #7060A8;">printf</span><span style="color: #0000FF;">(</span><span style="color: #000000;">1</span><span style="color: #0000FF;">,</span><span style="color: #000000;">fmt</span><span style="color: #0000FF;">,{</span><span style="color: #000000;">i</span><span style="color: #0000FF;">}&</span><span style="color: #000000;">positions</span><span style="color: #0000FF;">[</span><span style="color: #000000;">i</span><span style="color: #0000FF;">]&</span><span style="color: #000000;">velocities</span><span style="color: #0000FF;">[</span><span style="color: #000000;">i</span><span style="color: #0000FF;">])</span>
<span style="color: #008080;">end</span> <span style="color: #008080;">for</span>
for i=1 to timeSteps do
<span style="color: #008080;">end</span> <span style="color: #008080;">procedure</span>
printf(1,"\nCycle %d\n",i)
simulate()
<span style="color: #008080;">for</span> <span style="color: #000000;">i</span><span style="color: #0000FF;">=</span><span style="color: #000000;">1</span> <span style="color: #008080;">to</span> <span style="color: #000000;">timeSteps</span> <span style="color: #008080;">do</span>
printResults()
<span style="color: #7060A8;">printf</span><span style="color: #0000FF;">(</span><span style="color: #000000;">1</span><span style="color: #0000FF;">,</span><span style="color: #008000;">"\nCycle %d\n"</span><span style="color: #0000FF;">,</span><span style="color: #000000;">i</span><span style="color: #0000FF;">)</span>
end for</lang>
<span style="color: #000000;">simulate</span><span style="color: #0000FF;">()</span>
<span style="color: #000000;">printResults</span><span style="color: #0000FF;">()</span>
<span style="color: #008080;">end</span> <span style="color: #008080;">for</span>
<!--</lang>-->
Output same as Python (and C and C# and D and Kotlin and Go)
 
Petelomax
7,820

edits

Retrieved from "https://rosettacode.org/wiki/Special:MobileDiff/246244"