annotate matrix.yeti @ 16:e8d404787c35

Start on tests

module matrix;

zeros n = array(map \0 [1..n]);
ones  n = array(map \1 [1..n]);

generateMatrix f rows cols = array
   (map do row: array
       (map (f row) [0..cols-1])
    done [0..rows-1]);

constMatrix n = generateMatrix do row col: n done;

randomMatrix = generateMatrix do row col: Math#random() done;

zeroMatrix = constMatrix 0;
identityMatrix = constMatrix 1;

width m = if length m > 0 then length m[0] else 0 fi;

height m = length m;

dimensions m = { cols = width m, rows = height m };

transposed m = array
   (map do n: array
       (map do a: a[n-1] done m)
    done [1..width m]);

interleaved m = array(concat(transposed m));

deinterleaved rows v =
    generateMatrix do row col:
        v[rows * col + row]
    done rows (length v / rows);

{
zeros, ones,
generateMatrix, constMatrix, randomMatrix, zeroMatrix, identityMatrix,
width, height, dimensions,
transposed,
interleaved, deinterleaved,
}