factor/extra/math/haar/haar.factor

16 lines
435 B
Factor

! Haar wavelet transform -- http://dmr.ath.cx/gfx/haar/
USING: sequences math kernel splitting grouping columns ;
IN: math.haar
: averages ( seq -- seq )
[ first2 + 2 / ] map ;
: differences ( seq averages -- differences )
>r 0 <column> r> [ - ] 2map ;
: haar-step ( seq -- differences averages )
2 group dup averages [ differences ] keep ;
: haar ( seq -- seq )
dup length 1 <= [ haar-step haar prepend ] unless ;