49 lines
1.3 KiB
Factor
49 lines
1.3 KiB
Factor
! Copyright (C) 2005, 2007 Slava Pestov.
|
|
! See http://factorcode.org/license.txt for BSD license.
|
|
IN: matrices
|
|
USING: arrays generic kernel sequences math ;
|
|
|
|
! Matrices
|
|
: zero-matrix ( m n -- matrix )
|
|
swap [ drop 0 <array> ] map-with ;
|
|
|
|
: identity-matrix ( n -- matrix )
|
|
#! Make a nxn identity matrix.
|
|
dup [ swap [ = 1 0 ? ] map-with ] map-with ;
|
|
|
|
! Matrix operations
|
|
: mneg ( m -- m ) [ vneg ] map ;
|
|
|
|
: n*m ( n m -- m ) [ n*v ] map-with ;
|
|
: m*n ( m n -- m ) swap n*m ;
|
|
: n/m ( n m -- m ) [ n/v ] map-with ;
|
|
: m/n ( m n -- m ) swap [ swap v/n ] map-with ;
|
|
|
|
: m+ ( m m -- m ) [ v+ ] 2map ;
|
|
: m- ( m m -- m ) [ v- ] 2map ;
|
|
: m* ( m m -- m ) [ v* ] 2map ;
|
|
: m/ ( m m -- m ) [ v/ ] 2map ;
|
|
|
|
: v.m ( v m -- v ) flip [ v. ] map-with ;
|
|
: m.v ( m v -- v ) swap [ v. ] map-with ;
|
|
: m. ( m m -- m ) flip swap [ m.v ] map-with ;
|
|
|
|
: mmin ( m -- n ) >r 1/0. r> [ [ min ] each ] each ;
|
|
: mmax ( m -- n ) >r -1/0. r> [ [ max ] each ] each ;
|
|
: mnorm ( m -- n ) dup mmax abs m/n ;
|
|
|
|
: cross-i ( vec1 vec2 -- i )
|
|
over third over second * >r
|
|
swap second swap third * r> - ;
|
|
|
|
: cross-j ( vec1 vec2 -- j )
|
|
over first over third * >r
|
|
swap third swap first * r> - ;
|
|
|
|
: cross-k ( vec1 vec2 -- k )
|
|
over first over second * >r
|
|
swap second swap first * r> - ;
|
|
|
|
: cross ( vec1 vec2 -- vec3 )
|
|
[ cross-i ] 2keep [ cross-j ] 2keep cross-k 3array ;
|