images.normalization: more unit tests and bug fix
parent
17c1713ebb
commit
5fb71a8fcd
|
@ -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 } ;
|
||||||
|
|
|
@ -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
|
||||||
|
|
||||||
|
|
|
@ -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 )
|
||||||
|
|
Loading…
Reference in New Issue