unmaintained/images/processing/rotation/rotation.factor

   1 ! Copyright (C) 2009 Kobi Lurie.
   2 ! See http://factorcode.org/license.txt for BSD license.
   3 USING: accessors arrays colors combinators
   4 combinators.short-circuit fry grouping images images.bitmap
   5 images.loader images.normalization kernel locals math sequences ;
   6 IN: images.processing.rotation
   7
   8 ERROR: unsupported-rotation degrees ;
   9
  10 <PRIVATE
  11
  12 : rotate-90 ( seq^3 -- seq^3 ) flip [ reverse ] map ;
  13 : rotate-180 ( seq^3 -- seq^3 ) reverse [ reverse ] map ;
  14 : rotate-270 ( seq^3 -- seq^3 ) flip reverse ;
  15
  16 : (rotate) ( seq n -- seq' )
  17     {
  18         { 0 [ ] }
  19         { 90 [ rotate-90 ] }
  20         { 180 [ rotate-180 ] }
  21         { 270 [ rotate-270 ] }
  22         [ unsupported-rotation ]
  23     } case ;
  24
  25 : rows-remove-pad ( byte-rows -- pixels' )
  26     [ dup length 4 mod head* ] map ;
  27
  28 : row-length ( image -- n )
  29     [ bitmap>> length ] [ dim>> second ] bi /i ;
  30
  31 : image>byte-rows ( image -- byte-rows )
  32     [ bitmap>> ] [ row-length ] bi group rows-remove-pad ;
  33
  34 : (seperate-to-pixels) ( byte-rows image -- pixel-rows )
  35     component-order>> bytes-per-pixel '[ _ group ] map ;
  36
  37 : image>pixel-rows ( image -- pixel-rows )
  38     [ image>byte-rows ] keep (seperate-to-pixels) ;
  39
  40 : flatten-table ( seq^3 -- seq )
  41     [ concat ] map concat ;
  42
  43 : ?reverse-dimensions ( image n -- )
  44     { 270 90 } member? [ [ reverse ] change-dim ] when drop ;
  45
  46 :  normalize-degree ( n -- n' ) 360 rem ;
  47
  48 : processing-effect ( image quot -- image' )
  49     '[ image>pixel-rows @ flatten-table ] [ bitmap<< ] [ ] tri ; inline
  50
  51 :: rotate' ( image n -- image )
  52     n normalize-degree :> n'
  53     image image>pixel-rows :> pixel-table
  54     image n' ?reverse-dimensions
  55     pixel-table n' (rotate) :> table-rotated
  56     image table-rotated flatten-table >>bitmap ;
  57
  58 PRIVATE>
  59
  60 : rotate ( image n -- image' )
  61     normalize-degree
  62     [ '[ _ (rotate) ] processing-effect ] [ ?reverse-dimensions ] 2bi ;
  63
  64 : reflect-y-axis ( image -- image )
  65     [ [ reverse ] map ] processing-effect ;
  66
  67 : reflect-x-axis ( image -- image )
  68     [ reverse ] processing-effect ;