"

 COPYRIGHT (c) 1993 by Claus Gittinger

	      All Rights Reserved

 This software is furnished under a license and may be used

 only in accordance with the terms of that license and with the

 inclusion of the above copyright notice.   This software may not

 be provided or otherwise made available to, or used by, any

 other person.  No title to or ownership of the software is

 hereby transferred.

"

ImageReader subclass:#WindowsIconReader

	instanceVariableNames:''

	classVariableNames:''

	poolDictionaries:''

	category:'Graphics-Images-Support'

!

!WindowsIconReader class methodsFor:'documentation'!

copyright

"

 COPYRIGHT (c) 1993 by Claus Gittinger

	      All Rights Reserved

 This software is furnished under a license and may be used

 only in accordance with the terms of that license and with the

 inclusion of the above copyright notice.   This software may not

 be provided or otherwise made available to, or used by, any

 other person.  No title to or ownership of the software is

 hereby transferred.

"

!

documentation

"

    this class provides methods for loading Windows and OS2 icon files.

    Image writing is not supported.

    The reader tries to figure out which version of BMP/ICO is used.

    It seems to be able to load most formats, but who knows ...

    [See also:]

        Image Form Icon

        BlitImageReader FaceReader GIFReader JPEGReader PBMReader PCXReader 

        ST80FormReader SunRasterReader TargaReader TIFFReader  

        XBMReader XPMReader XWDReader 

"

! !

!WindowsIconReader class methodsFor:'initialization'!

initialize

    "tell Image-class, that a new fileReader is present

     for the '.bmp' and '.ico' extensions."

    Image addReader:self suffix:'bmp'.

    Image addReader:self suffix:'ico'.

    "Modified: 1.2.1997 / 15:03:59 / cg"

! !

!WindowsIconReader class methodsFor:'testing'!

isValidImageFile:aFileName

    "return true, if aFileName contains a valid windows bitmap-file image"

    |inStream header ok|

    inStream := self streamReadingFile:aFileName.

    inStream isNil ifTrue:[^ false].

    inStream binary.

    ok := false.

    inStream size > 16 ifTrue:[

        header := ByteArray uninitializedNew:4.

        inStream nextBytes:4 into:header.

        (header startsWith:#(66 77)) ifTrue:[     "BM"

            ok := true.

"/            'WINREADER: Win3.x or OS/2 vsn 2 BM format' infoPrintNL.

        ].

        (header startsWith:#(66 65)) ifTrue:[     "BA"

            ok := true.

"/            'WINREADER: OS/2 vsn 2 BA format' infoPrintNL.

        ].

        (header startsWith:#(73 67)) ifTrue:[     "IC"

            ok := true.

"/            'WINREADER: OS/2 IC format' infoPrintNL.

        ].

        (header startsWith:#(80 84)) ifTrue:[     "PT"

            ok := true.

"/            'WINREADER: OS/2 PT format' infoPrintNL.

        ].

        (header startsWith:#(0 0 1 0)) ifTrue:[

            ok := true.

"/            'WINREADER: Win3.x ICO format' infoPrintNL.

        ].

    ].

    inStream close.

    ^ ok

    "

     WindowsIconReader isValidImageFile:'/phys/clam2/LocalLibrary/Images/OS2_icons/dos.ico'

    "

    "Created: 17.9.1995 / 17:14:20 / claus"

! !

!WindowsIconReader methodsFor:'private'!

loadBMPWidth:w height:h depth:d compression:c from:aStream into:data

    "helper: load a BMP image"

    |buff idx1 idx2 bytesPerRow|

    d == 8 ifTrue:[

        (self class loadBMP8Width:w height:h compression:c from:aStream into:data) ifFalse:[

            'WinIconReader [warning]: read/decompression failed' infoPrintNL.

            ^ false

        ].

        ^ true

    ].

    d == 4 ifTrue:[

        (self class loadBMP4to8Width:w height:h compression:c from:aStream into:data) ifFalse:[

            'WinIconReader [warning]: read/decompression failed' infoPrintNL.

            ^ false

        ].

        ^ true

    ].

    d == 2 ifTrue:[

        (self class loadBMP2to8Width:w height:h from:aStream into:data) ifFalse:[

            'WinIconReader [warning]: read failed' infoPrintNL.

            ^ false

        ].

        ^ true

    ].

    d == 1 ifTrue:[

        (self class loadBMP1to8Width:w height:h from:aStream into:data) ifFalse:[

            'WinIconReader [warning]: read failed' infoPrintNL.

            ^ false

        ].

        ^ true

    ].

    d == 24 ifTrue:[

        bytesPerRow := w * 3.

        ((aStream nextBytes:(h * bytesPerRow) into:data) ~~ (h * bytesPerRow)) ifTrue:[

            'WinIconReader [warning]: read failed' infoPrintNL.

            ^ false

        ].

        "/ stupid - last row comes first

        buff := ByteArray uninitializedNew:bytesPerRow.

        idx1 := 1.

        idx2 := 1 + (h-1 * bytesPerRow).

        [idx1 < idx2] whileTrue:[

            buff replaceFrom:1 to:bytesPerRow with:data startingAt:idx1.

            data replaceFrom:idx1 to:(idx1 + bytesPerRow - 1) with:data startingAt:idx2.

            data replaceFrom:idx2 to:(idx2 + bytesPerRow - 1) with:buff startingAt:1.

            idx1 := idx1 + bytesPerRow.

            idx2 := idx2 - bytesPerRow.

        ].

        ^ true

    ].

    'WinIconReader [warning]: unsupported depth:' infoPrint. d infoPrintNL.

    "Created: 17.9.1995 / 18:48:11 / claus"

    "Modified: 28.1.1997 / 01:46:07 / cg"

! !

!WindowsIconReader methodsFor:'reading from file'!

fromOS2File:aFilename

    "read an image from an OS/2 BMP file"

    |reader stream|

    stream := self streamReadingFile:aFilename.

    stream isNil ifTrue:[^ nil].

    reader := (self new) fromOS2Stream:stream.

    stream close.

    reader notNil ifTrue:[^ reader image].

    ^ nil

    "Modified: 23.4.1996 / 13:09:28 / cg"

!

fromOS2Stream:aStream

    "read an image from an OS/2 BMP stream"

    | header inDepth

      rawMap rMap gMap bMap srcIndex dstIndex inBytesPerRow

      data4 mask tmp bytesPerRow nColors nByte|

    inStream := aStream.

    aStream binary.

    "read the header"

    header := ByteArray uninitializedNew:8r110.

    aStream nextBytes:16 into:header.

    (header startsWith:#(73 67)) ifTrue:[

        "IC format"

        aStream nextBytes:10 into:header startingAt:17.

        width := header at:7.

        height := header at:9.

        inDepth := 2 "header at:11". "where is it"

    ] ifFalse:[

        aStream nextBytes:(8r110-16) into:header startingAt:17.

        width := header at:8r101.

        height := header at:8r103.

        inDepth := header at:8r107.

    ].

    "read the colormap"

    nColors := 1 bitShift:inDepth.

    rawMap := ByteArray uninitializedNew:(nColors*3).

    aStream nextBytes:(nColors*3) into:rawMap.

    rMap := Array new:nColors.

    gMap := Array new:nColors.

    bMap := Array new:nColors.

    srcIndex := 1.

    1 to:nColors do:[:i |

        bMap at:i put:(rawMap at:srcIndex).

        srcIndex := srcIndex + 1.

        gMap at:i put:(rawMap at:srcIndex).

        srcIndex := srcIndex + 1.

        rMap at:i put:(rawMap at:srcIndex).

        srcIndex := srcIndex + 1.

    ].

    "read mask"

    nByte := width * height + 7 // 8.

    mask := ByteArray uninitializedNew:nByte.

    aStream nextBytes:nByte into:mask.

    "what is this"

    aStream nextBytes:nByte into:mask.

"/    "read the data bits"

"/

"/    bytesPerRow := width * inDepth + 7 // 8.

"/    data4 := ByteArray uninitializedNew:(height * bytesPerRow).

"/    inDepth == 8 ifTrue:[

"/    ].

"/    aStream nextBytes:(height * bytesPerRow) into:data4.

"/

"/    "stupid: last row first"

"/

"/    tmp := ByteArray new:(height * bytesPerRow).

"/    srcIndex := 1.

"/    dstIndex := (height - 1) * bytesPerRow + 1.

"/    1 to:height do:[:row |

"/        tmp replaceFrom:dstIndex to:(dstIndex + bytesPerRow - 1)

"/                   with:data4 startingAt:srcIndex.

"/        srcIndex := srcIndex + bytesPerRow.

"/        dstIndex := dstIndex - bytesPerRow.

"/    ].

"/    data4 := tmp.

"/

"/    "expand into bytes"

"/

"/    data := ByteArray new:(width * height).

"/    data4 expandPixels:inDepth width:width height:height

"/                  into:data mapping:nil.

"/

    bytesPerRow := width * inDepth + 7 // 8.

    "/ bmp data is always 32bit aligned; if required,

    inBytesPerRow := ((bytesPerRow + 3) // 4) * 4.

    data := ByteArray uninitializedNew:(height * width "bytesPerRow").

    (self loadBMPWidth:width height:height depth:inDepth compression:0 from:aStream into:data) ifFalse:[

        ^ nil

    ].

    photometric := #palette.

    samplesPerPixel := 1.

    bitsPerSample := #(8).

    colorMap := Colormap redVector:rMap greenVector:gMap blueVector:bMap.

    "

     |i f|

     i := Image fromFile:'/LocalLibrary/Images/OS2/dos3.ico'.

     f := i asFormOn:Display.

     v displayOpaqueForm:(f magnifyBy:2@2) x:5 y:5

    "

    "Modified: 17.9.1995 / 18:49:24 / claus"

    "Modified: 23.4.1996 / 13:09:40 / cg"

!

fromStream:aStream 

    "figure out which format the stream contains

     (there are various different bmp/ico formats around)

     and read the image."

    |fileSize header|

    inStream := aStream.

    aStream binary.

    fileSize := aStream size.

    fileSize < 16 ifTrue:[

        'WinIconReader [warning]: short file' infoPrintNL.

        ^ nil

    ].

    header := ByteArray uninitializedNew:4.

    aStream nextBytes:4 into:header.

    (header startsWith:#(66 77)) ifTrue:[     "BM"

        aStream position:1.

"/        'WinIconReader [info]: Win3.x or OS/2 vsn 2 BM format' infoPrintNL.

        ^ self fromWindowsBMPStream:aStream

    ].

    (header startsWith:#(66 65)) ifTrue:[     "BA"

        aStream position:1.

"/        'WinIconReader [info]: OS/2 vsn 2 BA format' infoPrintNL.

        ^ self fromOS2Stream:aStream

    ].

    (header startsWith:#(73 67)) ifTrue:[     "IC"

        aStream position:1.

"/        'WinIconReader [info]: OS/2 IC format' infoPrintNL.

        ^ self fromOS2Stream:aStream

    ].

    (header startsWith:#(80 84)) ifTrue:[     "PT"

        aStream position:1.

"/        'WinIconReader [info]: OS/2 PT format' infoPrintNL.

        ^ self fromOS2Stream:aStream

    ].

    (header startsWith:#(16r53 16r5A)) ifTrue:[     "SZ"

        'WinIconReader [warning]: SZ format not supported:' infoPrintNL.

        ^ nil.

        aStream position:1.

        'WinIconReader [info]: OS/2 PT format' infoPrintNL.

        ^ self fromOS2Stream:aStream

    ].

    (header startsWith:#(0 0 1 0)) ifTrue:[

        aStream position:1.

"/        'WinIconReader [info]: Win3.x ICO format' infoPrintNL.

        ^ self fromWindowsICOStream:aStream

    ].

    'WinIconReader [warning]: format not supported:' infoPrint.

    ((header at:1) printStringRadix:16) infoPrint.

    ' ' infoPrint.

    ((header at:2) printStringRadix:16) infoPrintNL.

    ^ nil

    "

     Image fromFile:'/phys/clam//LocalLibrary/Images/OS2_icons/dos.ico'

    "

    "Modified: 17.9.1995 / 18:59:07 / claus"

    "Modified: 28.1.1997 / 01:48:31 / cg"

!

fromWindowsBMPFile: aFilename 

    "read an image from a windows BMP file"

    |reader stream|

    stream := self class streamReadingFile:aFilename.

    stream isNil ifTrue:[^ nil].

    reader := (self new) fromWindowsBMPStream:stream.

    stream close.

    reader notNil ifTrue:[^ reader image].

    ^ nil

    "Modified: 23.4.1996 / 13:09:53 / cg"

!

fromWindowsBMPStream:aStream 

    "read an image from a windows BMP stream"

    | fileSize header inDepth inPlanes compression

      imgSize resH resV numColor numImportantColor

      dataStart

      rawMap rMap gMap bMap srcIndex dstIndex inBytesPerRow

      data4 mask tmp bytesPerRow fourBytesPerColorInfo|

    inStream := aStream.

    aStream binary.

    fileSize := aStream size.

    "read the header"

    header := ByteArray uninitializedNew:16r54.

    aStream nextBytes:18 into:header.

    ((header at:(16r0E + 1)) == 40) ifTrue:[    "header-size"

        "

         its an Windows3.x BMP file

         or OS/2 vsn 2 BMP file

        "

"/        'WinIconReader [info]: Win3.x or OS/2 vsn 2 format' infoPrintNL.

        aStream nextBytes:(40-4) into:header startingAt:19.

        width := header wordAt:(16r12 + 1) MSB:false.  "(header at:19) + ((header at:20) * 256).   "

        height := header wordAt:(16r16 + 1) MSB:false. "(header at:23) + ((header at:24) * 256).   "

        inPlanes := header wordAt:(16r1A + 1) MSB:false.

        inDepth := header wordAt:(16r1C + 1) MSB:false.

        compression := header wordAt:(16r1E + 1) MSB:false.

        imgSize := header doubleWordAt:(16r22 + 1) MSB:false.

        resH := header doubleWordAt:(16r26 + 1) MSB:false.

        resV := header doubleWordAt:(16r2A + 1) MSB:false.

        numColor := header doubleWordAt:(16r2E + 1) MSB:false.

        numImportantColor := header doubleWordAt:(16r32 + 1) MSB:false.

        numColor == 0 ifTrue:[

            "

             some bmp-writers seem to leave this as zero (which is wrong)

            "

            inDepth <= 8 ifTrue:[

                numColor := 1 bitShift:inDepth.

                'WinIconReader [warning]: missing nColor in header - assume ' infoPrint. numColor infoPrintNL

            ]

        ].

        numColor ~~ 0 ifTrue:[

            rawMap := ByteArray uninitializedNew:(numColor * 4).

            aStream nextBytes:(numColor * 4) into:rawMap.

            fourBytesPerColorInfo := true.

        ].

        dataStart := header wordAt:(16r0A + 1) MSB:false

    ] ifFalse:[

        ((header at:(16r0E + 1)) == 12) ifTrue:[     "core-info header size"

            "

             its an OS/2 (vsn1.2) BMP file

            "

"/           'WinIconReader [info]: OS/2 vsn 1.2 format' infoPrintNL.

            aStream nextBytes:(12-4) into:header startingAt:19.

            width := header wordAt:(16r12 + 1) MSB:false.  "(header at:19) + ((header at:20) * 256).   "

            height := header wordAt:(16r14 + 1) MSB:false. "(header at:21) + ((header at:22) * 256).   "

            inPlanes := header wordAt:(16r16 + 1) MSB:false.

            inDepth := header wordAt:(16r18 + 1) MSB:false.

            numColor := 1 bitShift:inDepth.

            rawMap := ByteArray uninitializedNew:(numColor * 3).

            aStream nextBytes:(numColor * 3) into:rawMap.

            fourBytesPerColorInfo := false.

            compression := 0.

            dataStart := header wordAt:(16r0A + 1) MSB:false

        ] ifFalse:[

            'WinIconReader [warning]: unknown format' infoPrintNL.

            ^ nil

        ].

    ].

    numColor ~~ 0 ifTrue:[

        "read the colormap"

        rMap := Array new:numColor.

        gMap := Array new:numColor.

        bMap := Array new:numColor.

        srcIndex := 1.

        1 to:numColor do:[:i |

            bMap at:i put:(rawMap at:srcIndex).

            srcIndex := srcIndex + 1.

            gMap at:i put:(rawMap at:srcIndex).

            srcIndex := srcIndex + 1.

            rMap at:i put:(rawMap at:srcIndex).

            srcIndex := srcIndex + 1.

            fourBytesPerColorInfo ifTrue:[

                srcIndex := srcIndex + 1.

            ]

        ].

    ].

    "/ check for valid compression

    compression ~~ 0 ifTrue:[

        "/ some compression

        compression == 1 ifTrue:[

            "/ RLE8 - must be depth-8

            inDepth ~~ 8 ifTrue:[

                'WinIconReader [warning]: RLE8 compression only allowed with depth8 images' infoPrintNL.

                ^ nil

            ].

        ].

        compression == 2 ifTrue:[