move more bitmap code to bitmap.loading
parent
3cabf3aeca
commit
41a2894083
|
@ -2,8 +2,7 @@
|
||||||
! See http://factorcode.org/license.txt for BSD license.
|
! See http://factorcode.org/license.txt for BSD license.
|
||||||
USING: accessors alien alien.c-types arrays byte-arrays columns
|
USING: accessors alien alien.c-types arrays byte-arrays columns
|
||||||
combinators compression.run-length endian fry grouping images
|
combinators compression.run-length endian fry grouping images
|
||||||
images.bitmap.loading images.bitmap.saving images.loader io
|
images.bitmap.loading images.loader io
|
||||||
io.binary io.encodings.8-bit io.encodings.binary
|
|
||||||
io.encodings.string io.files io.streams.limited kernel locals
|
io.encodings.string io.files io.streams.limited kernel locals
|
||||||
macros math math.bitwise math.functions namespaces sequences
|
macros math math.bitwise math.functions namespaces sequences
|
||||||
specialized-arrays.uint specialized-arrays.ushort strings
|
specialized-arrays.uint specialized-arrays.ushort strings
|
||||||
|
@ -11,153 +10,8 @@ summary ;
|
||||||
QUALIFIED-WITH: bitstreams b
|
QUALIFIED-WITH: bitstreams b
|
||||||
IN: images.bitmap
|
IN: images.bitmap
|
||||||
|
|
||||||
SINGLETON: bitmap-image
|
: write2 ( n -- ) 2 >le write ;
|
||||||
"bmp" bitmap-image register-image-class
|
: write4 ( n -- ) 4 >le write ;
|
||||||
|
|
||||||
! endpoints-triple is ciexyzX/Y/Z, 3x fixed-point-2.30 aka 3x uint
|
|
||||||
|
|
||||||
<PRIVATE
|
|
||||||
|
|
||||||
: color-lookup3 ( loading-bitmap -- seq )
|
|
||||||
[ color-index>> >array ]
|
|
||||||
[ color-palette>> 3 <sliced-groups> ] bi
|
|
||||||
'[ _ nth ] map concat ;
|
|
||||||
|
|
||||||
: color-lookup4 ( loading-bitmap -- seq )
|
|
||||||
[ color-index>> >array ]
|
|
||||||
[ color-palette>> 4 <sliced-groups> [ 3 head-slice ] map ] bi
|
|
||||||
'[ _ nth ] map concat ;
|
|
||||||
|
|
||||||
! os2v1 is 3bytes each, all others are 3 + 1 unused
|
|
||||||
: color-lookup ( loading-bitmap -- seq )
|
|
||||||
dup file-header>> header-length>> {
|
|
||||||
{ 12 [ color-lookup3 ] }
|
|
||||||
{ 64 [ color-lookup4 ] }
|
|
||||||
{ 40 [ color-lookup4 ] }
|
|
||||||
{ 108 [ color-lookup4 ] }
|
|
||||||
{ 124 [ color-lookup4 ] }
|
|
||||||
} case ;
|
|
||||||
|
|
||||||
ERROR: bmp-not-supported n ;
|
|
||||||
|
|
||||||
: uncompress-bitfield ( seq masks -- bytes' )
|
|
||||||
'[
|
|
||||||
_ [
|
|
||||||
[ bitand ] [ bit-count ] [ log2 ] tri - shift
|
|
||||||
] with map
|
|
||||||
] { } map-as B{ } concat-as ;
|
|
||||||
|
|
||||||
: bitmap>bytes ( loading-bitmap -- byte-array )
|
|
||||||
dup header>> bit-count>>
|
|
||||||
{
|
|
||||||
{ 32 [ color-index>> ] }
|
|
||||||
{ 24 [ color-index>> ] }
|
|
||||||
{ 16 [
|
|
||||||
[
|
|
||||||
! byte-array>ushort-array
|
|
||||||
2 group [ le> ] map
|
|
||||||
! 5 6 5
|
|
||||||
! { HEX: f800 HEX: 7e0 HEX: 1f } uncompress-bitfield
|
|
||||||
! 5 5 5
|
|
||||||
{ HEX: 7c00 HEX: 3e0 HEX: 1f } uncompress-bitfield
|
|
||||||
] change-color-index
|
|
||||||
color-index>>
|
|
||||||
] }
|
|
||||||
{ 8 [ color-lookup ] }
|
|
||||||
{ 4 [ [ 4 b:byte-array-n>seq ] change-color-index color-lookup ] }
|
|
||||||
{ 1 [ [ 1 b:byte-array-n>seq ] change-color-index color-lookup ] }
|
|
||||||
[ bmp-not-supported ]
|
|
||||||
} case >byte-array ;
|
|
||||||
|
|
||||||
: set-bitfield-widths ( loading-bitmap -- loading-bitmap' )
|
|
||||||
dup header>> bit-count>> {
|
|
||||||
{ 16 [ dup color-palette>> 4 group [ le> ] map ] }
|
|
||||||
{ 32 [ { HEX: ff0000 HEX: ff00 HEX: ff } ] }
|
|
||||||
} case reverse >>bitfields ;
|
|
||||||
|
|
||||||
ERROR: unsupported-bitfield-widths n ;
|
|
||||||
|
|
||||||
M: unsupported-bitfield-widths summary
|
|
||||||
drop "Bitmaps only support bitfield compression in 16/32bit images" ;
|
|
||||||
|
|
||||||
: uncompress-bitfield-widths ( loading-bitmap -- loading-bitmap' )
|
|
||||||
set-bitfield-widths
|
|
||||||
dup header>> bit-count>> {
|
|
||||||
{ 16 [
|
|
||||||
dup bitfields>> '[
|
|
||||||
byte-array>ushort-array _ uncompress-bitfield
|
|
||||||
] change-color-index
|
|
||||||
] }
|
|
||||||
{ 32 [ ] }
|
|
||||||
[ unsupported-bitfield-widths ]
|
|
||||||
} case ;
|
|
||||||
|
|
||||||
ERROR: unsupported-bitmap-compression compression ;
|
|
||||||
|
|
||||||
GENERIC: uncompress-bitmap* ( loading-bitmap header -- loading-bitmap )
|
|
||||||
|
|
||||||
: uncompress-bitmap ( loading-bitmap -- loading-bitmap )
|
|
||||||
dup header>> uncompress-bitmap* ;
|
|
||||||
|
|
||||||
M: os2-header uncompress-bitmap* ( loading-bitmap header -- loading-bitmap' )
|
|
||||||
drop ;
|
|
||||||
|
|
||||||
: do-run-length-uncompress ( loading-bitmap word -- loading-bitmap )
|
|
||||||
dupd '[
|
|
||||||
_ header>> [ width>> ] [ height>> ] bi
|
|
||||||
_ execute
|
|
||||||
] change-color-index ; inline
|
|
||||||
|
|
||||||
M: v-header uncompress-bitmap* ( loading-bitmap header -- loading-bitmap' )
|
|
||||||
compression>> {
|
|
||||||
{ f [ ] }
|
|
||||||
{ 0 [ ] }
|
|
||||||
{ 1 [ \ run-length-uncompress-bitmap8 do-run-length-uncompress ] }
|
|
||||||
{ 2 [ \ run-length-uncompress-bitmap4 do-run-length-uncompress ] }
|
|
||||||
{ 3 [ uncompress-bitfield-widths ] }
|
|
||||||
{ 4 [ "jpeg" unsupported-bitmap-compression ] }
|
|
||||||
{ 5 [ "png" unsupported-bitmap-compression ] }
|
|
||||||
} case ;
|
|
||||||
|
|
||||||
: bitmap-padding ( width -- n )
|
|
||||||
3 * 4 mod 4 swap - 4 mod ; inline
|
|
||||||
|
|
||||||
: loading-bitmap>bytes ( loading-bitmap -- byte-array )
|
|
||||||
uncompress-bitmap bitmap>bytes ;
|
|
||||||
|
|
||||||
: color-palette-length ( loading-bitmap -- n )
|
|
||||||
file-header>>
|
|
||||||
[ offset>> 14 - ] [ header-length>> ] bi - ;
|
|
||||||
|
|
||||||
: color-index-length ( header -- n )
|
|
||||||
{
|
|
||||||
[ width>> ]
|
|
||||||
[ planes>> * ]
|
|
||||||
[ bit-count>> * 31 + 32 /i 4 * ]
|
|
||||||
[ height>> abs * ]
|
|
||||||
} cleave ;
|
|
||||||
|
|
||||||
ERROR: unsupported-bitmap-file magic ;
|
|
||||||
|
|
||||||
PRIVATE>
|
|
||||||
|
|
||||||
: bitmap>color-index ( bitmap -- byte-array )
|
|
||||||
[
|
|
||||||
bitmap>>
|
|
||||||
4 <sliced-groups>
|
|
||||||
[ 3 head-slice <reversed> ] map
|
|
||||||
B{ } join
|
|
||||||
] [
|
|
||||||
dim>> first dup bitmap-padding dup 0 > [
|
|
||||||
[ 3 * group ] dip '[ _ <byte-array> append ] map
|
|
||||||
B{ } join
|
|
||||||
] [
|
|
||||||
2drop
|
|
||||||
] if
|
|
||||||
] bi ;
|
|
||||||
|
|
||||||
: reverse-lines ( byte-array width -- byte-array )
|
|
||||||
<sliced-groups> <reversed> concat ; inline
|
|
||||||
|
|
||||||
: save-bitmap ( image path -- )
|
: save-bitmap ( image path -- )
|
||||||
binary [
|
binary [
|
||||||
|
|
|
@ -1,12 +1,16 @@
|
||||||
! Copyright (C) 2009 Doug Coleman.
|
! Copyright (C) 2009 Doug Coleman.
|
||||||
! See http://factorcode.org/license.txt for BSD license.
|
! See http://factorcode.org/license.txt for BSD license.
|
||||||
USING: accessors arrays combinators images images.bitmap
|
USING: accessors arrays byte-arrays combinators
|
||||||
images.bitmap.private io io.binary io.encodings.8-bit
|
compression.run-length fry grouping images images.loader io
|
||||||
io.encodings.binary io.encodings.string io.streams.limited
|
io.binary io.encodings.8-bit io.encodings.binary
|
||||||
kernel math math.bitwise grouping sequences ;
|
io.encodings.string io.streams.limited kernel math math.bitwise
|
||||||
QUALIFIED-WITH: syntax S
|
sequences specialized-arrays.ushort summary ;
|
||||||
|
QUALIFIED-WITH: bitstreams b
|
||||||
IN: images.bitmap.loading
|
IN: images.bitmap.loading
|
||||||
|
|
||||||
|
SINGLETON: bitmap-image
|
||||||
|
"bmp" bitmap-image register-image-class
|
||||||
|
|
||||||
! http://www.fileformat.info/format/bmp/egff.htm
|
! http://www.fileformat.info/format/bmp/egff.htm
|
||||||
! http://www.digicamsoft.com/bmp/bmp.html
|
! http://www.digicamsoft.com/bmp/bmp.html
|
||||||
|
|
||||||
|
@ -176,6 +180,18 @@ UNION: os2-header os2v1-header os2v2-header ;
|
||||||
[ unknown-bitmap-header ]
|
[ unknown-bitmap-header ]
|
||||||
} case ;
|
} case ;
|
||||||
|
|
||||||
|
: color-index-length ( header -- n )
|
||||||
|
{
|
||||||
|
[ width>> ]
|
||||||
|
[ planes>> * ]
|
||||||
|
[ bit-count>> * 31 + 32 /i 4 * ]
|
||||||
|
[ height>> abs * ]
|
||||||
|
} cleave ;
|
||||||
|
|
||||||
|
: color-palette-length ( loading-bitmap -- n )
|
||||||
|
file-header>>
|
||||||
|
[ offset>> 14 - ] [ header-length>> ] bi - ;
|
||||||
|
|
||||||
: parse-color-palette ( loading-bitmap -- loading-bitmap )
|
: parse-color-palette ( loading-bitmap -- loading-bitmap )
|
||||||
dup color-palette-length read >>color-palette ;
|
dup color-palette-length read >>color-palette ;
|
||||||
|
|
||||||
|
@ -217,12 +233,113 @@ GENERIC: bitmap>component-order* ( loading-bitmap header -- object )
|
||||||
[ drop simple-bitmap>component-order ]
|
[ drop simple-bitmap>component-order ]
|
||||||
} case ;
|
} case ;
|
||||||
|
|
||||||
|
: color-lookup3 ( loading-bitmap -- seq )
|
||||||
|
[ color-index>> >array ]
|
||||||
|
[ color-palette>> 3 <sliced-groups> ] bi
|
||||||
|
'[ _ nth ] map concat ;
|
||||||
|
|
||||||
|
: color-lookup4 ( loading-bitmap -- seq )
|
||||||
|
[ color-index>> >array ]
|
||||||
|
[ color-palette>> 4 <sliced-groups> [ 3 head-slice ] map ] bi
|
||||||
|
'[ _ nth ] map concat ;
|
||||||
|
|
||||||
|
! os2v1 is 3bytes each, all others are 3 + 1 unused
|
||||||
|
: color-lookup ( loading-bitmap -- seq )
|
||||||
|
dup file-header>> header-length>> {
|
||||||
|
{ 12 [ color-lookup3 ] }
|
||||||
|
{ 64 [ color-lookup4 ] }
|
||||||
|
{ 40 [ color-lookup4 ] }
|
||||||
|
{ 108 [ color-lookup4 ] }
|
||||||
|
{ 124 [ color-lookup4 ] }
|
||||||
|
} case ;
|
||||||
|
|
||||||
M: os2v1-header bitmap>component-order* drop simple-bitmap>component-order ;
|
M: os2v1-header bitmap>component-order* drop simple-bitmap>component-order ;
|
||||||
M: os2v2-header bitmap>component-order* drop simple-bitmap>component-order ;
|
M: os2v2-header bitmap>component-order* drop simple-bitmap>component-order ;
|
||||||
M: v3-header bitmap>component-order* drop simple-bitmap>component-order ;
|
M: v3-header bitmap>component-order* drop simple-bitmap>component-order ;
|
||||||
M: v4-header bitmap>component-order* drop advanced-bitmap>component-order ;
|
M: v4-header bitmap>component-order* drop advanced-bitmap>component-order ;
|
||||||
M: v5-header bitmap>component-order* drop advanced-bitmap>component-order ;
|
M: v5-header bitmap>component-order* drop advanced-bitmap>component-order ;
|
||||||
|
|
||||||
|
: uncompress-bitfield ( seq masks -- bytes' )
|
||||||
|
'[
|
||||||
|
_ [
|
||||||
|
[ bitand ] [ bit-count ] [ log2 ] tri - shift
|
||||||
|
] with map
|
||||||
|
] { } map-as B{ } concat-as ;
|
||||||
|
|
||||||
|
ERROR: bmp-not-supported n ;
|
||||||
|
|
||||||
|
: bitmap>bytes ( loading-bitmap -- byte-array )
|
||||||
|
dup header>> bit-count>>
|
||||||
|
{
|
||||||
|
{ 32 [ color-index>> ] }
|
||||||
|
{ 24 [ color-index>> ] }
|
||||||
|
{ 16 [
|
||||||
|
[
|
||||||
|
! byte-array>ushort-array
|
||||||
|
2 group [ le> ] map
|
||||||
|
! 5 6 5
|
||||||
|
! { HEX: f800 HEX: 7e0 HEX: 1f } uncompress-bitfield
|
||||||
|
! 5 5 5
|
||||||
|
{ HEX: 7c00 HEX: 3e0 HEX: 1f } uncompress-bitfield
|
||||||
|
] change-color-index
|
||||||
|
color-index>>
|
||||||
|
] }
|
||||||
|
{ 8 [ color-lookup ] }
|
||||||
|
{ 4 [ [ 4 b:byte-array-n>seq ] change-color-index color-lookup ] }
|
||||||
|
{ 1 [ [ 1 b:byte-array-n>seq ] change-color-index color-lookup ] }
|
||||||
|
[ bmp-not-supported ]
|
||||||
|
} case >byte-array ;
|
||||||
|
|
||||||
|
: set-bitfield-widths ( loading-bitmap -- loading-bitmap' )
|
||||||
|
dup header>> bit-count>> {
|
||||||
|
{ 16 [ dup color-palette>> 4 group [ le> ] map ] }
|
||||||
|
{ 32 [ { HEX: ff0000 HEX: ff00 HEX: ff } ] }
|
||||||
|
} case reverse >>bitfields ;
|
||||||
|
|
||||||
|
ERROR: unsupported-bitfield-widths n ;
|
||||||
|
|
||||||
|
M: unsupported-bitfield-widths summary
|
||||||
|
drop "Bitmaps only support bitfield compression in 16/32bit images" ;
|
||||||
|
|
||||||
|
: uncompress-bitfield-widths ( loading-bitmap -- loading-bitmap' )
|
||||||
|
set-bitfield-widths
|
||||||
|
dup header>> bit-count>> {
|
||||||
|
{ 16 [
|
||||||
|
dup bitfields>> '[
|
||||||
|
byte-array>ushort-array _ uncompress-bitfield
|
||||||
|
] change-color-index
|
||||||
|
] }
|
||||||
|
{ 32 [ ] }
|
||||||
|
[ unsupported-bitfield-widths ]
|
||||||
|
} case ;
|
||||||
|
|
||||||
|
ERROR: unsupported-bitmap-compression compression ;
|
||||||
|
|
||||||
|
GENERIC: uncompress-bitmap* ( loading-bitmap header -- loading-bitmap )
|
||||||
|
|
||||||
|
: uncompress-bitmap ( loading-bitmap -- loading-bitmap )
|
||||||
|
dup header>> uncompress-bitmap* ;
|
||||||
|
|
||||||
|
M: os2-header uncompress-bitmap* ( loading-bitmap header -- loading-bitmap' )
|
||||||
|
drop ;
|
||||||
|
|
||||||
|
: do-run-length-uncompress ( loading-bitmap word -- loading-bitmap )
|
||||||
|
dupd '[
|
||||||
|
_ header>> [ width>> ] [ height>> ] bi
|
||||||
|
_ execute
|
||||||
|
] change-color-index ; inline
|
||||||
|
|
||||||
|
M: v-header uncompress-bitmap* ( loading-bitmap header -- loading-bitmap' )
|
||||||
|
compression>> {
|
||||||
|
{ f [ ] }
|
||||||
|
{ 0 [ ] }
|
||||||
|
{ 1 [ \ run-length-uncompress-bitmap8 do-run-length-uncompress ] }
|
||||||
|
{ 2 [ \ run-length-uncompress-bitmap4 do-run-length-uncompress ] }
|
||||||
|
{ 3 [ uncompress-bitfield-widths ] }
|
||||||
|
{ 4 [ "jpeg" unsupported-bitmap-compression ] }
|
||||||
|
{ 5 [ "png" unsupported-bitmap-compression ] }
|
||||||
|
} case ;
|
||||||
|
|
||||||
ERROR: unsupported-bitmap-file magic ;
|
ERROR: unsupported-bitmap-file magic ;
|
||||||
|
|
||||||
: load-bitmap ( path -- loading-bitmap )
|
: load-bitmap ( path -- loading-bitmap )
|
||||||
|
@ -243,6 +360,9 @@ ERROR: unsupported-bitmap-file magic ;
|
||||||
} case
|
} case
|
||||||
] with-input-stream ;
|
] with-input-stream ;
|
||||||
|
|
||||||
|
: loading-bitmap>bytes ( loading-bitmap -- byte-array )
|
||||||
|
uncompress-bitmap bitmap>bytes ;
|
||||||
|
|
||||||
M: bitmap-image load-image* ( path bitmap-image -- bitmap )
|
M: bitmap-image load-image* ( path bitmap-image -- bitmap )
|
||||||
drop load-bitmap
|
drop load-bitmap
|
||||||
[ image new ] dip
|
[ image new ] dip
|
||||||
|
|
Loading…
Reference in New Issue