libs/math/matrices.factor

   1 ! Copyright (C) 2005 Slava Pestov.
   2 ! See http://factor.sf.net/license.txt for BSD license.
   3 IN: math-contrib
   4 USING: arrays generic kernel sequences math ;
   5
   6 ! Matrices
   7 : zero-matrix ( m n -- matrix )
   8     swap [ drop 0 <array> ] map-with ;
   9
  10 : identity-matrix ( n -- matrix )
  11     #! Make a nxn identity matrix.
  12     dup [ swap [ = 1 0 ? ] map-with ] map-with ;
  13
  14 ! Matrix operations
  15 : mneg ( m -- m ) [ vneg ] map ;
  16
  17 : n*m ( n m -- m ) [ n*v ] map-with ;
  18 : m*n ( m n -- m ) swap n*m ;
  19 : n/m ( n m -- m ) [ n/v ] map-with ;
  20 : m/n ( m n -- m ) swap [ swap v/n ] map-with ;
  21
  22 : m+   ( m m -- m ) [ v+ ] 2map ;
  23 : m-   ( m m -- m ) [ v- ] 2map ;
  24 : m*   ( m m -- m ) [ v* ] 2map ;
  25 : m/   ( m m -- m ) [ v/ ] 2map ;
  26
  27 : v.m ( v m -- v ) flip [ v. ] map-with ;
  28 : m.v ( m v -- v ) swap [ v. ] map-with ;
  29 : m.  ( m m -- m ) flip swap [ m.v ] map-with ;
  30
  31 : mmin ( m -- n ) >r 1/0. r> [ [ min ] each ] each ;
  32 : mmax ( m -- n ) >r -1/0. r> [ [ max ] each ] each ;
  33 : mnorm ( m -- n ) dup mmax abs m/n ;
  34 : m-almost= ( m n -- ? )
  35     t -rot [ [ almost= and ] 2each ] 2each ;
  36