images.normalization: more unit tests and bug fix

db4
Keith Lazuka 2009-10-09 10:45:12 -04:00
parent 17c1713ebb
commit 5fb71a8fcd
3 changed files with 54 additions and 68 deletions

View File

@ -21,7 +21,7 @@ HELP: reorder-colors
ARTICLE: "images.normalization" "Image normalization" ARTICLE: "images.normalization" "Image normalization"
"The " { $vocab-link "images.normalization" } " vocab can be used to convert between " { $link image } " representations." "The " { $vocab-link "images.normalization" } " vocab can be used to convert between " { $link image } " representations."
$nl $nl
"You can normalize any image to RGBA with ubyte-components:" "You can normalize any image to a RGBA with ubyte-components representation:"
{ $subsections normalize-image } { $subsections normalize-image }
"Convert an image's pixel layout to match an arbitrary " { $link component-order } ":" "Convert an image's pixel layout to match an arbitrary " { $link component-order } ":"
{ $subsections reorder-colors } ; { $subsections reorder-colors } ;

View File

@ -4,93 +4,73 @@ USING: images images.normalization images.normalization.private
sequences tools.test ; sequences tools.test ;
IN: images.normalization.tests IN: images.normalization.tests
! R ! 1>x
[ B{ 255 255 } ]
[ B{ 0 1 } A L permute ] unit-test
[ B{ 255 255 255 255 } ]
[ B{ 0 1 } A RG permute ] unit-test
[ B{ 255 255 255 255 255 255 } ]
[ B{ 0 1 } A BGR permute ] unit-test
[ B{ 0 255 255 255 1 255 255 255 } ] [ B{ 0 255 255 255 1 255 255 255 } ]
[ B{ 0 1 } R RGBA permute ] unit-test [ B{ 0 1 } A ABGR permute ] unit-test
[ B{ 255 255 0 255 255 1 } ] ! 2>x
[ B{ 0 1 } R BGR permute ] unit-test
[ B{ 255 255 0 255 255 255 1 255 } ] [ B{ 0 2 } ]
[ B{ 0 1 } R BGRA permute ] unit-test [ B{ 0 1 2 3 } LA L permute ] unit-test
[ B{ 255 0 255 255 255 1 255 255 } ] [ B{ 255 255 255 255 } ]
[ B{ 0 1 } R ARGB permute ] unit-test [ B{ 0 1 2 3 } LA RG permute ] unit-test
! RGB [ B{ 255 255 255 255 255 255 } ]
[ B{ 0 1 2 3 } LA BGR permute ] unit-test
[ B{ 0 3 } ] [ B{ 1 255 255 255 3 255 255 255 } ]
[ B{ 0 1 2 3 4 5 } RGB R permute ] unit-test [ B{ 0 1 2 3 } LA ABGR permute ] unit-test
[ B{ 0 1 2 255 3 4 5 255 } ] ! 3>x
[ B{ 0 1 2 3 4 5 } RGB RGBA permute ] unit-test
[ B{ 255 255 } ]
[ B{ 0 1 2 3 4 5 } RGB L permute ] unit-test
[ B{ 0 1 3 4 } ]
[ B{ 0 1 2 3 4 5 } RGB RG permute ] unit-test
[ B{ 2 1 0 5 4 3 } ] [ B{ 2 1 0 5 4 3 } ]
[ B{ 0 1 2 3 4 5 } RGB BGR permute ] unit-test [ B{ 0 1 2 3 4 5 } RGB BGR permute ] unit-test
[ B{ 2 1 0 255 5 4 3 255 } ] [ B{ 255 2 1 0 255 5 4 3 } ]
[ B{ 0 1 2 3 4 5 } RGB BGRA permute ] unit-test [ B{ 0 1 2 3 4 5 } RGB ABGR permute ] unit-test
[ B{ 255 0 1 2 255 3 4 5 } ] ! 4>x
[ B{ 0 1 2 3 4 5 } RGB ARGB permute ] unit-test
! RGBA [ B{ 255 255 } ]
[ B{ 0 1 2 3 4 5 6 7 } RGBA L permute ] unit-test
[ B{ 0 4 } ] [ B{ 0 1 4 5 } ]
[ B{ 0 1 2 3 4 5 6 7 } RGBA R permute ] unit-test [ B{ 0 1 2 3 4 5 6 7 } RGBA RG permute ] unit-test
[ B{ 0 1 2 4 5 6 } ]
[ B{ 0 1 2 3 4 5 6 7 } RGBA RGB permute ] unit-test
[ B{ 2 1 0 6 5 4 } ] [ B{ 2 1 0 6 5 4 } ]
[ B{ 0 1 2 3 4 5 6 7 } RGBA BGR permute ] unit-test [ B{ 0 1 2 3 4 5 6 7 } RGBA BGR permute ] unit-test
[ B{ 2 1 0 3 6 5 4 7 } ]
[ B{ 0 1 2 3 4 5 6 7 } RGBA BGRA permute ] unit-test
[ B{ 3 0 1 2 7 4 5 6 } ]
[ B{ 0 1 2 3 4 5 6 7 } RGBA ARGB permute ] unit-test
! BGR
[ B{ 2 1 0 5 4 3 } ]
[ B{ 0 1 2 3 4 5 } BGR RGB permute ] unit-test
[ B{ 2 1 0 255 5 4 3 255 } ]
[ B{ 0 1 2 3 4 5 } BGR RGBA permute ] unit-test
[ B{ 0 1 2 255 3 4 5 255 } ]
[ B{ 0 1 2 3 4 5 } BGR BGRA permute ] unit-test
[ B{ 255 2 1 0 255 5 4 3 } ]
[ B{ 0 1 2 3 4 5 } BGR ARGB permute ] unit-test
! BGRA
[ B{ 2 1 0 6 5 4 } ]
[ B{ 0 1 2 3 4 5 6 7 } BGRA RGB permute ] unit-test
[ B{ 0 1 2 4 5 6 } ]
[ B{ 0 1 2 3 4 5 6 7 } BGRA BGR permute ] unit-test
[ B{ 2 1 0 3 6 5 4 7 } ]
[ B{ 0 1 2 3 4 5 6 7 } BGRA RGBA permute ] unit-test
[ B{ 3 2 1 0 7 6 5 4 } ] [ B{ 3 2 1 0 7 6 5 4 } ]
[ B{ 0 1 2 3 4 5 6 7 } BGRA ARGB permute ] unit-test [ B{ 0 1 2 3 4 5 6 7 } RGBA ABGR permute ] unit-test
! ARGB ! A little ad hoc testing
[ B{ 1 2 3 5 6 7 } ] [ B{ 0 4 } ]
[ B{ 0 1 2 3 4 5 6 7 } ARGB RGB permute ] unit-test [ B{ 0 1 2 3 4 5 6 7 } RGBA R permute ] unit-test
[ B{ 3 2 1 7 6 5 } ] [ B{ 255 0 1 2 255 4 5 6 } ]
[ B{ 0 1 2 3 4 5 6 7 } ARGB BGR permute ] unit-test [ B{ 0 1 2 3 4 5 6 7 } RGBA XRGB permute ] unit-test
[ B{ 3 2 1 0 7 6 5 4 } ] [ B{ 1 2 3 255 5 6 7 255 } ]
[ B{ 0 1 2 3 4 5 6 7 } ARGB BGRA permute ] unit-test [ B{ 0 1 2 3 4 5 6 7 } XRGB RGBA permute ] unit-test
[ B{ 1 2 3 0 5 6 7 4 } ] [ B{ 255 255 255 255 255 255 255 255 } ]
[ B{ 0 1 2 3 4 5 6 7 } ARGB RGBA permute ] unit-test [ B{ 0 1 } L RGBA permute ] unit-test

View File

@ -11,19 +11,25 @@ IN: images.normalization
<PRIVATE <PRIVATE
CONSTANT: don't-care 3 CONSTANT: don't-care 127
CONSTANT: fill-value 255
: permutation ( src dst -- seq ) : permutation ( src dst -- seq )
swap '[ _ index [ don't-care ] unless* ] { } map-as swap '[ _ index [ don't-care ] unless* ] { } map-as
4 don't-care pad-tail ; 4 don't-care pad-tail ;
: pad4 ( seq -- newseq ) 4 255 pad-tail ; : pad4 ( seq -- newseq ) 4 fill-value pad-tail ;
: shuffle ( seq permutation -- newseq )
swap '[
dup 4 >= [ drop fill-value ] [ _ nth ] if
] B{ } map-as ;
:: permute ( bytes src-order dst-order -- new-bytes ) :: permute ( bytes src-order dst-order -- new-bytes )
[let | src [ src-order name>> ] [let | src [ src-order name>> ]
dst [ dst-order name>> ] | dst [ dst-order name>> ] |
bytes src length group bytes src length group
[ pad4 src dst permutation vshuffle dst length head ] [ pad4 src dst permutation shuffle dst length head ]
map concat ] ; map concat ] ;
: (reorder-colors) ( image src-order dest-order -- image ) : (reorder-colors) ( image src-order dest-order -- image )