steve
/
factor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

optimization

cvs
Slava Pestov 2006-01-11 05:40:11 +00:00
parent 2e919a82d1
commit b843de7707
1 changed files with 8 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
library/math/vectors.factor
View File

@ -1,9 +1,8 @@
! Copyright (C) 2005 Slava Pestov.
! See http://factor.sf.net/license.txt for BSD license.
! Copyright (C) 2005, 2006 Slava Pestov.
! See http://factorcode.org/license.txt for BSD license.
IN: math
USING: arrays generic kernel sequences ;
! Vectors
: vneg ( v -- v ) [ neg ] map ;
: n*v ( n v -- v ) [ * ] map-with ;
@ -18,22 +17,13 @@ USING: arrays generic kernel sequences ;
: vmax ( v v -- v ) [ max ] 2map ;
: vmin ( v v -- v ) [ min ] 2map ;
: v. ( v v -- x ) 0 [ * + ] 2reduce ;
: norm-sq ( v -- n ) 0 [ absq + ] reduce ;
: norm ( vec -- n ) norm-sq sqrt ;
: normalize ( vec -- uvec ) dup norm v/n ;
: sum ( v -- n ) 0 [ + ] reduce ;
: product ( v -- n ) 1 [ * ] reduce ;
: set-axis ( x y axis -- v )
2dup v* >r >r drop dup r> v* v- r> v+ ;
: v. ( v v -- x )
#! Dot product.
0 [ * + ] 2reduce ;
: norm-sq ( v -- n ) 0 [ absq + ] reduce ;
: norm ( vec -- n )
#! Length of a vector.
norm-sq sqrt ;
: normalize ( vec -- uvec )
#! Unit vector with same direction as vec.
dup norm v/n ;
dup length [ >r 0 = pick pick ? r> swap nth ] 2map 2nip ;