math.extras: adding jacobi and legendere symbols.

diff --git a/extra/math/extras/extras-tests.factor b/extra/math/extras/extras-tests.factor
new file mode 100644 (file)
index 0000000..aa93071
--- /dev/null
+++ b/extra/math/extras/extras-tests.factor
@@ -0,0 +1,11 @@
+! Copyright (C) 2012 John Benediktsson
+! See http://factorcode.org/license.txt for BSD license
+
+USING: math.extras tools.test ;
+
+IN: math.extras.test
+
+{ -1 } [ -1 7 jacobi ] unit-test
+{ 0 } [ 3 3 jacobi ] unit-test
+{ -1 } [ 127 703 jacobi ] unit-test
+{ 1 } [ -4 197 jacobi ] unit-test

diff --git a/extra/math/extras/extras.factor b/extra/math/extras/extras.factor
index d2cf314ac90bf9681ead4c890cfb0582d9aa22c8..2b9c556acbfc37d944f38763dcd75c940b0561af 100644 (file)
--- a/extra/math/extras/extras.factor
+++ b/extra/math/extras/extras.factor
@@ -2,8 +2,8 @@
 ! See http://factorcode.org/license.txt for BSD license
 
 USING: combinators.short-circuit kernel math math.combinatorics
-math.functions math.order math.ranges math.statistics
-math.vectors memoize sequences ;
+math.functions math.order math.primes math.ranges
+math.statistics math.vectors memoize sequences ;
 
 IN: math.extras
 
@@ -51,3 +51,37 @@ PRIVATE>
 
 : chi2P ( chi df -- p )
     dup df-check [ 2.0 / ] [ 2 /i ] bi* (chi2P) 1.0 min ;
+
+<PRIVATE
+
+: check-jacobi ( m -- m )
+    dup { [ integer? ] [ 0 > ] [ odd? ] } 1&&
+    [ "modulus must be odd positive integer" throw ] unless ;
+
+: mod' ( x y -- n )
+    [ mod ] keep over zero? [ drop ] [
+        2dup [ sgn ] bi@ = [ drop ] [ + ] if
+    ] if ;
+
+PRIVATE>
+
+: jacobi ( a m -- n )
+    check-jacobi [ mod' ] keep 1
+    [ pick zero? ] [
+        [ pick even? ] [
+            [ 2 / ] 2dip
+            over 8 mod' { 3 5 } member? [ neg ] when
+        ] while swapd
+        2over [ 4 mod' 3 = ] both? [ neg ] when
+        [ [ mod' ] keep ] dip
+    ] until [ nip 1 = ] dip 0 ? ;
+
+<PRIVATE
+
+: check-legendere ( m -- m )
+    dup prime? [ "modulus must be prime positive integer" throw ] unless ;
+
+PRIVATE>
+
+: legendere ( a m -- n )
+    check-legendere jacobi ;