|
1 /* The Great Computer Language Shootout |
|
2 http://shootout.alioth.debian.org/ |
|
3 contributed by Isaac Gouy */ |
|
4 |
|
5 var PI = 3.141592653589793; |
|
6 var SOLAR_MASS = 4 * PI * PI; |
|
7 var DAYS_PER_YEAR = 365.24; |
|
8 |
|
9 function Body(x,y,z,vx,vy,vz,mass){ |
|
10 this.x = x; |
|
11 this.y = y; |
|
12 this.z = z; |
|
13 this.vx = vx; |
|
14 this.vy = vy; |
|
15 this.vz = vz; |
|
16 this.mass = mass; |
|
17 } |
|
18 |
|
19 Body.prototype.offsetMomentum = function(px,py,pz) { |
|
20 this.vx = -px / SOLAR_MASS; |
|
21 this.vy = -py / SOLAR_MASS; |
|
22 this.vz = -pz / SOLAR_MASS; |
|
23 return this; |
|
24 } |
|
25 |
|
26 function Jupiter(){ |
|
27 return new Body( |
|
28 4.84143144246472090e+00, |
|
29 -1.16032004402742839e+00, |
|
30 -1.03622044471123109e-01, |
|
31 1.66007664274403694e-03 * DAYS_PER_YEAR, |
|
32 7.69901118419740425e-03 * DAYS_PER_YEAR, |
|
33 -6.90460016972063023e-05 * DAYS_PER_YEAR, |
|
34 9.54791938424326609e-04 * SOLAR_MASS |
|
35 ); |
|
36 } |
|
37 |
|
38 function Saturn(){ |
|
39 return new Body( |
|
40 8.34336671824457987e+00, |
|
41 4.12479856412430479e+00, |
|
42 -4.03523417114321381e-01, |
|
43 -2.76742510726862411e-03 * DAYS_PER_YEAR, |
|
44 4.99852801234917238e-03 * DAYS_PER_YEAR, |
|
45 2.30417297573763929e-05 * DAYS_PER_YEAR, |
|
46 2.85885980666130812e-04 * SOLAR_MASS |
|
47 ); |
|
48 } |
|
49 |
|
50 function Uranus(){ |
|
51 return new Body( |
|
52 1.28943695621391310e+01, |
|
53 -1.51111514016986312e+01, |
|
54 -2.23307578892655734e-01, |
|
55 2.96460137564761618e-03 * DAYS_PER_YEAR, |
|
56 2.37847173959480950e-03 * DAYS_PER_YEAR, |
|
57 -2.96589568540237556e-05 * DAYS_PER_YEAR, |
|
58 4.36624404335156298e-05 * SOLAR_MASS |
|
59 ); |
|
60 } |
|
61 |
|
62 function Neptune(){ |
|
63 return new Body( |
|
64 1.53796971148509165e+01, |
|
65 -2.59193146099879641e+01, |
|
66 1.79258772950371181e-01, |
|
67 2.68067772490389322e-03 * DAYS_PER_YEAR, |
|
68 1.62824170038242295e-03 * DAYS_PER_YEAR, |
|
69 -9.51592254519715870e-05 * DAYS_PER_YEAR, |
|
70 5.15138902046611451e-05 * SOLAR_MASS |
|
71 ); |
|
72 } |
|
73 |
|
74 function Sun(){ |
|
75 return new Body(0.0, 0.0, 0.0, 0.0, 0.0, 0.0, SOLAR_MASS); |
|
76 } |
|
77 |
|
78 |
|
79 function NBodySystem(bodies){ |
|
80 this.bodies = bodies; |
|
81 var px = 0.0; |
|
82 var py = 0.0; |
|
83 var pz = 0.0; |
|
84 var size = this.bodies.length; |
|
85 /* BEGIN LOOP */ |
|
86 for (var i=0; i<size; i++){ |
|
87 var b = this.bodies[i]; |
|
88 var m = b.mass; |
|
89 px += b.vx * m; |
|
90 py += b.vy * m; |
|
91 pz += b.vz * m; |
|
92 } |
|
93 /* END LOOP */ |
|
94 this.bodies[0].offsetMomentum(px,py,pz); |
|
95 } |
|
96 |
|
97 NBodySystem.prototype.advance = function(dt){ |
|
98 var dx, dy, dz, distance, mag; |
|
99 var size = this.bodies.length; |
|
100 |
|
101 /* BEGIN LOOP */ |
|
102 for (var i=0; i<size; i++) { |
|
103 var bodyi = this.bodies[i]; |
|
104 /* BEGIN LOOP */ |
|
105 for (var j=i+1; j<size; j++) { |
|
106 var bodyj = this.bodies[j]; |
|
107 dx = bodyi.x - bodyj.x; |
|
108 dy = bodyi.y - bodyj.y; |
|
109 dz = bodyi.z - bodyj.z; |
|
110 |
|
111 distance = Math.sqrt(dx*dx + dy*dy + dz*dz); |
|
112 mag = dt / (distance * distance * distance); |
|
113 |
|
114 bodyi.vx -= dx * bodyj.mass * mag; |
|
115 bodyi.vy -= dy * bodyj.mass * mag; |
|
116 bodyi.vz -= dz * bodyj.mass * mag; |
|
117 |
|
118 bodyj.vx += dx * bodyi.mass * mag; |
|
119 bodyj.vy += dy * bodyi.mass * mag; |
|
120 bodyj.vz += dz * bodyi.mass * mag; |
|
121 } |
|
122 /* END LOOP */ |
|
123 } |
|
124 /* END LOOP */ |
|
125 |
|
126 /* BEGIN LOOP */ |
|
127 for (var i=0; i<size; i++) { |
|
128 var body = this.bodies[i]; |
|
129 body.x += dt * body.vx; |
|
130 body.y += dt * body.vy; |
|
131 body.z += dt * body.vz; |
|
132 } |
|
133 /* END LOOP */ |
|
134 } |
|
135 |
|
136 NBodySystem.prototype.energy = function(){ |
|
137 var dx, dy, dz, distance; |
|
138 var e = 0.0; |
|
139 var size = this.bodies.length; |
|
140 |
|
141 /* BEGIN LOOP */ |
|
142 for (var i=0; i<size; i++) { |
|
143 var bodyi = this.bodies[i]; |
|
144 |
|
145 e += 0.5 * bodyi.mass * |
|
146 ( bodyi.vx * bodyi.vx |
|
147 + bodyi.vy * bodyi.vy |
|
148 + bodyi.vz * bodyi.vz ); |
|
149 |
|
150 /* BEGIN LOOP */ |
|
151 for (var j=i+1; j<size; j++) { |
|
152 var bodyj = this.bodies[j]; |
|
153 dx = bodyi.x - bodyj.x; |
|
154 dy = bodyi.y - bodyj.y; |
|
155 dz = bodyi.z - bodyj.z; |
|
156 |
|
157 distance = Math.sqrt(dx*dx + dy*dy + dz*dz); |
|
158 e -= (bodyi.mass * bodyj.mass) / distance; |
|
159 } |
|
160 /* END LOOP */ |
|
161 } |
|
162 /* END LOOP */ |
|
163 return e; |
|
164 } |
|
165 |
|
166 var ret; |
|
167 |
|
168 /* BEGIN LOOP */ |
|
169 for ( var n = 3; n <= 24; n *= 2 ) { |
|
170 (function(){ |
|
171 var bodies = new NBodySystem( Array( |
|
172 Sun(),Jupiter(),Saturn(),Uranus(),Neptune() |
|
173 )); |
|
174 var max = n * 100; |
|
175 |
|
176 ret = bodies.energy(); |
|
177 /* BEGIN LOOP */ |
|
178 for (var i=0; i<max; i++){ |
|
179 bodies.advance(0.01); |
|
180 } |
|
181 /* END LOOP */ |
|
182 ret = bodies.energy(); |
|
183 })(); |
|
184 } |
|
185 /* END LOOP */ |