factor: trim using lists

[factor.git] / extra / benchmark / nbody-simd / nbody-simd.factor
diff --git a/extra/benchmark/nbody-simd/nbody-simd.factor b/extra/benchmark/nbody-simd/nbody-simd.factor

index e8bef58923beae7076aa7f7d4c680b96a96a718a..c50481de5d08ecbe773cc45df3ca8d0f90bb62c5 100644 (file)
--- a/extra/benchmark/nbody-simd/nbody-simd.factor
+++ b/extra/benchmark/nbody-simd/nbody-simd.factor
@@ -1,9 +1,8 @@
-! Copyright (C) 2008, 2009 Slava Pestov.
+! Copyright (C) 2008, 2010 Slava Pestov.
  ! See http://factorcode.org/license.txt for BSD license.
-USING: accessors fry kernel locals math math.constants
-math.functions math.vectors math.vectors.simd prettyprint
-combinators.smart sequences hints classes.struct
-specialized-arrays ;
+USING: accessors alien.c-types kernel math math.constants
+math.functions math.vectors math.vectors.simd.cords math.parser
+combinators.smart sequences classes.struct specialized-arrays io ;
  IN: benchmark.nbody-simd
  
  : solar-mass ( -- x ) 4 pi sq * ; inline
@@ -17,35 +16,35 @@ STRUCT: body
  SPECIALIZED-ARRAY: body
  
  : <body> ( location velocity mass -- body )
-    [ days-per-year v*n ] [ solar-mass * ] bi* body <struct-boa> ; inline
+    [ days-per-year v*n ] [ solar-mass * ] bi* body boa ; inline
  
  : <jupiter> ( -- body )
-    double-4{ 4.84143144246472090e+00 -1.16032004402742839e+00 -1.03622044471123109e-01 0.0 }
+    double-4{ 4.84143144246472090e00 -1.16032004402742839e00 -1.03622044471123109e-01 0.0 }
      double-4{ 1.66007664274403694e-03 7.69901118419740425e-03 -6.90460016972063023e-05 0.0 }
      9.54791938424326609e-04
      <body> ;
  
  : <saturn> ( -- body )
-    double-4{ 8.34336671824457987e+00 4.12479856412430479e+00 -4.03523417114321381e-01 0.0 }
+    double-4{ 8.34336671824457987e00 4.12479856412430479e00 -4.03523417114321381e-01 0.0 }
      double-4{ -2.76742510726862411e-03 4.99852801234917238e-03 2.30417297573763929e-05 0.0 }
      2.85885980666130812e-04
      <body> ;
  
  : <uranus> ( -- body )
-    double-4{ 1.28943695621391310e+01 -1.51111514016986312e+01 -2.23307578892655734e-01 0.0 }
+    double-4{ 1.28943695621391310e01 -1.51111514016986312e01 -2.23307578892655734e-01 0.0 }
      double-4{ 2.96460137564761618e-03 2.37847173959480950e-03 -2.96589568540237556e-05 0.0 }
      4.36624404335156298e-05
      <body> ;
  
  : <neptune> ( -- body )
-    double-4{ 1.53796971148509165e+01 -2.59193146099879641e+01 1.79258772950371181e-01 0.0 }
+    double-4{ 1.53796971148509165e01 -2.59193146099879641e01 1.79258772950371181e-01 0.0 }
      double-4{ 2.68067772490389322e-03 1.62824170038242295e-03 -9.51592254519715870e-05 0.0 }
      5.15138902046611451e-05
      <body> ;
  
  : <sun> ( -- body )
      double-4{ 0 0 0 0 } double-4{ 0 0 0 0 } 1 <body> ;
-    
+
  : offset-momentum ( body offset -- body )
      vneg solar-mass v/n >>velocity ; inline
  
@@ -58,7 +57,7 @@ SPECIALIZED-ARRAY: body
      body-array{ } output>sequence
      dup init-bodies ; inline
  
-:: each-pair ( bodies pair-quot: ( other-body body -- ) each-quot: ( body -- ) -- )
+:: each-pair ( ... bodies pair-quot: ( ... other-body body -- ... ) each-quot: ( ... body -- ... ) -- )
      bodies [| body i |
          body each-quot call
          bodies i 1 + tail-slice [
@@ -94,8 +93,10 @@ SPECIALIZED-ARRAY: body
  : nbody ( n -- )
      >fixnum
      <nbody-system>
-    [ energy . ] [ '[ _ 0.01 advance ] times ] [ energy . ] tri ;
+    [ energy number>string print ]
+    [ '[ _ 0.01 advance ] times ]
+    [ energy number>string print ] tri ;
  
-: nbody-main ( -- ) 1000000 nbody ;
+: nbody-simd-benchmark ( -- ) 1000000 nbody ;
  
-MAIN: nbody-main
+MAIN: nbody-simd-benchmark