1 /////////////////////////////////////////////////////////////////////////////
2 // Name: src/common/imagbmp.cpp
3 // Purpose: wxImage BMP,ICO and CUR handlers
4 // Author: Robert Roebling, Chris Elliott
6 // Copyright: (c) Robert Roebling, Chris Elliott
7 // Licence: wxWindows licence
8 /////////////////////////////////////////////////////////////////////////////
10 // For compilers that support precompilation, includes "wx.h".
11 #include "wx/wxprec.h"
19 #include "wx/imagbmp.h"
23 #include "wx/msw/wrapwin.h"
27 #include "wx/bitmap.h"
28 #include "wx/palette.h"
32 #include "wx/filefn.h"
33 #include "wx/wfstream.h"
34 #include "wx/quantize.h"
35 #include "wx/anidecod.h"
46 //-----------------------------------------------------------------------------
48 //-----------------------------------------------------------------------------
50 IMPLEMENT_DYNAMIC_CLASS(wxBMPHandler
,wxImageHandler
)
67 #define BI_BITFIELDS 3
70 #define poffset (line * width * 3 + column * 3)
72 bool wxBMPHandler::SaveFile(wxImage
*image
,
73 wxOutputStream
& stream
,
76 return SaveDib(image
, stream
, verbose
, true/*IsBmp*/, false/*IsMask*/);
79 bool wxBMPHandler::SaveDib(wxImage
*image
,
80 wxOutputStream
& stream
,
86 wxCHECK_MSG( image
, false, _T("invalid pointer in wxBMPHandler::SaveFile") );
91 wxLogError(_("BMP: Couldn't save invalid image."));
95 // get the format of the BMP file to save, else use 24bpp
96 unsigned format
= wxBMP_24BPP
;
97 if ( image
->HasOption(wxIMAGE_OPTION_BMP_FORMAT
) )
98 format
= image
->GetOptionInt(wxIMAGE_OPTION_BMP_FORMAT
);
100 wxUint16 bpp
; // # of bits per pixel
101 int palette_size
; // # of color map entries, ie. 2^bpp colors
103 // set the bpp and appropriate palette_size, and do additional checks
104 if ( (format
== wxBMP_1BPP
) || (format
== wxBMP_1BPP_BW
) )
109 else if ( format
== wxBMP_4BPP
)
114 else if ( (format
== wxBMP_8BPP
) || (format
== wxBMP_8BPP_GREY
) ||
115 (format
== wxBMP_8BPP_RED
) || (format
== wxBMP_8BPP_PALETTE
) )
117 // need to set a wxPalette to use this, HOW TO CHECK IF VALID, SIZE?
118 if ((format
== wxBMP_8BPP_PALETTE
)
120 && !image
->HasPalette()
121 #endif // wxUSE_PALETTE
125 wxLogError(_("BMP: wxImage doesn't have own wxPalette."));
131 else // you get 24bpp
133 format
= wxBMP_24BPP
;
138 unsigned width
= image
->GetWidth();
139 unsigned row_padding
= (4 - int(width
*bpp
/8.0) % 4) % 4; // # bytes to pad to dword
140 unsigned row_width
= int(width
* bpp
/8.0) + row_padding
; // # of bytes per row
145 wxUint16 magic
; // format magic, always 'BM'
146 wxUint32 filesize
; // total file size, inc. headers
147 wxUint32 reserved
; // for future use
148 wxUint32 data_offset
; // image data offset in the file
151 wxUint32 bih_size
; // 2nd part's size
152 wxUint32 width
, height
; // bitmap's dimensions
153 wxUint16 planes
; // num of planes
154 wxUint16 bpp
; // bits per pixel
155 wxUint32 compression
; // compression method
156 wxUint32 size_of_bmp
; // size of the bitmap
157 wxUint32 h_res
, v_res
; // image resolution in dpi
158 wxUint32 num_clrs
; // number of colors used
159 wxUint32 num_signif_clrs
;// number of significant colors
162 wxUint32 hdr_size
= 14/*BitmapHeader*/ + 40/*BitmapInfoHeader*/;
164 hdr
.magic
= wxUINT16_SWAP_ON_BE(0x4D42/*'BM'*/);
165 hdr
.filesize
= wxUINT32_SWAP_ON_BE( hdr_size
+ palette_size
*4 +
166 row_width
* image
->GetHeight() );
168 hdr
.data_offset
= wxUINT32_SWAP_ON_BE(hdr_size
+ palette_size
*4);
170 hdr
.bih_size
= wxUINT32_SWAP_ON_BE(hdr_size
- 14);
171 hdr
.width
= wxUINT32_SWAP_ON_BE(image
->GetWidth());
174 hdr
.height
= wxUINT32_SWAP_ON_BE(image
->GetHeight());
178 hdr
.height
= wxUINT32_SWAP_ON_BE(2 * image
->GetHeight());
180 hdr
.planes
= wxUINT16_SWAP_ON_BE(1); // always 1 plane
181 hdr
.bpp
= wxUINT16_SWAP_ON_BE(bpp
);
182 hdr
.compression
= 0; // RGB uncompressed
183 hdr
.size_of_bmp
= wxUINT32_SWAP_ON_BE(row_width
* image
->GetHeight());
184 hdr
.h_res
= hdr
.v_res
= wxUINT32_SWAP_ON_BE(72); // 72dpi is standard
185 hdr
.num_clrs
= wxUINT32_SWAP_ON_BE(palette_size
); // # colors in colormap
186 hdr
.num_signif_clrs
= 0; // all colors are significant
190 if (// VS: looks ugly but compilers tend to do ugly things with structs,
191 // like aligning hdr.filesize's ofset to dword :(
192 // VZ: we should add padding then...
193 !stream
.Write(&hdr
.magic
, 2) ||
194 !stream
.Write(&hdr
.filesize
, 4) ||
195 !stream
.Write(&hdr
.reserved
, 4) ||
196 !stream
.Write(&hdr
.data_offset
, 4)
200 wxLogError(_("BMP: Couldn't write the file (Bitmap) header."));
207 !stream
.Write(&hdr
.bih_size
, 4) ||
208 !stream
.Write(&hdr
.width
, 4) ||
209 !stream
.Write(&hdr
.height
, 4) ||
210 !stream
.Write(&hdr
.planes
, 2) ||
211 !stream
.Write(&hdr
.bpp
, 2) ||
212 !stream
.Write(&hdr
.compression
, 4) ||
213 !stream
.Write(&hdr
.size_of_bmp
, 4) ||
214 !stream
.Write(&hdr
.h_res
, 4) ||
215 !stream
.Write(&hdr
.v_res
, 4) ||
216 !stream
.Write(&hdr
.num_clrs
, 4) ||
217 !stream
.Write(&hdr
.num_signif_clrs
, 4)
221 wxLogError(_("BMP: Couldn't write the file (BitmapInfo) header."));
226 wxPalette
*palette
= NULL
; // entries for quantized images
227 wxUint8
*rgbquad
= NULL
; // for the RGBQUAD bytes for the colormap
228 wxImage
*q_image
= NULL
; // destination for quantized image
230 // if <24bpp use quantization to reduce colors for *some* of the formats
231 if ( (format
== wxBMP_1BPP
) || (format
== wxBMP_4BPP
) ||
232 (format
== wxBMP_8BPP
) || (format
== wxBMP_8BPP_PALETTE
) )
234 // make a new palette and quantize the image
235 if (format
!= wxBMP_8BPP_PALETTE
)
237 q_image
= new wxImage();
239 // I get a delete error using Quantize when desired colors > 236
240 int quantize
= ((palette_size
> 236) ? 236 : palette_size
);
241 // fill the destination too, it gives much nicer 4bpp images
242 wxQuantize::Quantize( *image
, *q_image
, &palette
, quantize
, 0,
243 wxQUANTIZE_FILL_DESTINATION_IMAGE
);
248 palette
= new wxPalette(image
->GetPalette());
249 #endif // wxUSE_PALETTE
253 unsigned char r
, g
, b
;
254 rgbquad
= new wxUint8
[palette_size
*4];
256 for (i
= 0; i
< palette_size
; i
++)
259 if ( !palette
->GetRGB(i
, &r
, &g
, &b
) )
260 #endif // wxUSE_PALETTE
269 // make a 256 entry greyscale colormap or 2 entry black & white
270 else if ( (format
== wxBMP_8BPP_GREY
) || (format
== wxBMP_8BPP_RED
) ||
271 (format
== wxBMP_1BPP_BW
) )
273 rgbquad
= new wxUint8
[palette_size
*4];
275 for ( int i
= 0; i
< palette_size
; i
++ )
277 // if 1BPP_BW then the value should be either 0 or 255
278 wxUint8 c
= (wxUint8
)((i
> 0) && (format
== wxBMP_1BPP_BW
) ? 255 : i
);
287 // if the colormap was made, then it needs to be written
292 if ( !stream
.Write(rgbquad
, palette_size
*4) )
295 wxLogError(_("BMP: Couldn't write RGB color map."));
299 #endif // wxUSE_PALETTE
307 // pointer to the image data, use quantized if available
308 wxUint8
*data
= (wxUint8
*) image
->GetData();
309 if (q_image
) if (q_image
->Ok()) data
= (wxUint8
*) q_image
->GetData();
311 wxUint8
*buffer
= new wxUint8
[row_width
];
312 memset(buffer
, 0, row_width
);
316 for (y
= image
->GetHeight() -1; y
>= 0; y
--)
318 if ( format
== wxBMP_24BPP
) // 3 bytes per pixel red,green,blue
320 for ( x
= 0; x
< width
; x
++ )
322 pixel
= 3*(y
*width
+ x
);
324 buffer
[3*x
] = data
[pixel
+2];
325 buffer
[3*x
+ 1] = data
[pixel
+1];
326 buffer
[3*x
+ 2] = data
[pixel
];
329 else if ((format
== wxBMP_8BPP
) || // 1 byte per pixel in color
330 (format
== wxBMP_8BPP_PALETTE
))
332 for (x
= 0; x
< width
; x
++)
334 pixel
= 3*(y
*width
+ x
);
336 buffer
[x
] = (wxUint8
)palette
->GetPixel( data
[pixel
],
340 // FIXME: what should this be? use some std palette maybe?
342 #endif // wxUSE_PALETTE
345 else if ( format
== wxBMP_8BPP_GREY
) // 1 byte per pix, rgb ave to grey
347 for (x
= 0; x
< width
; x
++)
349 pixel
= 3*(y
*width
+ x
);
350 buffer
[x
] = (wxUint8
)(.299*data
[pixel
] +
355 else if ( format
== wxBMP_8BPP_RED
) // 1 byte per pixel, red as greys
357 for (x
= 0; x
< width
; x
++)
359 buffer
[x
] = (wxUint8
)data
[3*(y
*width
+ x
)];
362 else if ( format
== wxBMP_4BPP
) // 4 bpp in color
364 for (x
= 0; x
< width
; x
+=2)
366 pixel
= 3*(y
*width
+ x
);
368 // fill buffer, ignore if > width
370 buffer
[x
/2] = (wxUint8
)(
371 ((wxUint8
)palette
->GetPixel(data
[pixel
],
373 data
[pixel
+2]) << 4) |
376 : ((wxUint8
)palette
->GetPixel(data
[pixel
+3],
380 // FIXME: what should this be? use some std palette maybe?
382 #endif // wxUSE_PALETTE
385 else if ( format
== wxBMP_1BPP
) // 1 bpp in "color"
387 for (x
= 0; x
< width
; x
+=8)
389 pixel
= 3*(y
*width
+ x
);
392 buffer
[x
/8] = (wxUint8
)(
393 ((wxUint8
)palette
->GetPixel(data
[pixel
], data
[pixel
+1], data
[pixel
+2]) << 7) |
394 (((x
+1) > width
) ? 0 : ((wxUint8
)palette
->GetPixel(data
[pixel
+3], data
[pixel
+4], data
[pixel
+5]) << 6)) |
395 (((x
+2) > width
) ? 0 : ((wxUint8
)palette
->GetPixel(data
[pixel
+6], data
[pixel
+7], data
[pixel
+8]) << 5)) |
396 (((x
+3) > width
) ? 0 : ((wxUint8
)palette
->GetPixel(data
[pixel
+9], data
[pixel
+10], data
[pixel
+11]) << 4)) |
397 (((x
+4) > width
) ? 0 : ((wxUint8
)palette
->GetPixel(data
[pixel
+12], data
[pixel
+13], data
[pixel
+14]) << 3)) |
398 (((x
+5) > width
) ? 0 : ((wxUint8
)palette
->GetPixel(data
[pixel
+15], data
[pixel
+16], data
[pixel
+17]) << 2)) |
399 (((x
+6) > width
) ? 0 : ((wxUint8
)palette
->GetPixel(data
[pixel
+18], data
[pixel
+19], data
[pixel
+20]) << 1)) |
400 (((x
+7) > width
) ? 0 : ((wxUint8
)palette
->GetPixel(data
[pixel
+21], data
[pixel
+22], data
[pixel
+23]) )) );
402 // FIXME: what should this be? use some std palette maybe?
404 #endif // wxUSE_PALETTE
407 else if ( format
== wxBMP_1BPP_BW
) // 1 bpp B&W colormap from red color ONLY
409 for (x
= 0; x
< width
; x
+=8)
411 pixel
= 3*(y
*width
+ x
);
413 buffer
[x
/8] = (wxUint8
)(
414 (((wxUint8
)(data
[pixel
] /128.)) << 7) |
415 (((x
+1) > width
) ? 0 : (((wxUint8
)(data
[pixel
+3] /128.)) << 6)) |
416 (((x
+2) > width
) ? 0 : (((wxUint8
)(data
[pixel
+6] /128.)) << 5)) |
417 (((x
+3) > width
) ? 0 : (((wxUint8
)(data
[pixel
+9] /128.)) << 4)) |
418 (((x
+4) > width
) ? 0 : (((wxUint8
)(data
[pixel
+12]/128.)) << 3)) |
419 (((x
+5) > width
) ? 0 : (((wxUint8
)(data
[pixel
+15]/128.)) << 2)) |
420 (((x
+6) > width
) ? 0 : (((wxUint8
)(data
[pixel
+18]/128.)) << 1)) |
421 (((x
+7) > width
) ? 0 : (((wxUint8
)(data
[pixel
+21]/128.)) )) );
425 if ( !stream
.Write(buffer
, row_width
) )
428 wxLogError(_("BMP: Couldn't write data."));
432 #endif // wxUSE_PALETTE
440 #endif // wxUSE_PALETTE
449 unsigned char r
, g
, b
;
452 bool wxBMPHandler::DoLoadDib(wxImage
* image
, int width
, int height
,
453 int bpp
, int ncolors
, int comp
,
454 wxFileOffset bmpOffset
, wxInputStream
& stream
,
455 bool verbose
, bool IsBmp
, bool hasPalette
)
457 wxInt32 aDword
, rmask
= 0, gmask
= 0, bmask
= 0;
458 int rshift
= 0, gshift
= 0, bshift
= 0;
459 int rbits
= 0, gbits
= 0, bbits
= 0;
465 // allocate space for palette if needed:
470 cmap
= new _cmap
[ncolors
];
474 wxLogError(_("BMP: Couldn't allocate memory."));
481 // destroy existing here instead of:
483 image
->Create(width
, height
);
485 unsigned char *ptr
= image
->GetData();
490 wxLogError( _("BMP: Couldn't allocate memory.") );
496 // Reading the palette, if it exists:
497 if ( bpp
< 16 && ncolors
!= 0 )
499 unsigned char* r
= new unsigned char[ncolors
];
500 unsigned char* g
= new unsigned char[ncolors
];
501 unsigned char* b
= new unsigned char[ncolors
];
502 for (int j
= 0; j
< ncolors
; j
++)
506 stream
.Read(bbuf
, 4);
517 //used in reading .ico file mask
520 b
[j
] = cmap
[j
].b
= ( j
? 255 : 0 );
525 // Set the palette for the wxImage
526 image
->SetPalette(wxPalette(ncolors
, r
, g
, b
));
527 #endif // wxUSE_PALETTE
533 else if ( bpp
== 16 || bpp
== 32 )
535 if ( comp
== BI_BITFIELDS
)
538 stream
.Read(dbuf
, 4 * 3);
539 rmask
= wxINT32_SWAP_ON_BE(dbuf
[0]);
540 gmask
= wxINT32_SWAP_ON_BE(dbuf
[1]);
541 bmask
= wxINT32_SWAP_ON_BE(dbuf
[2]);
542 // find shift amount (Least significant bit of mask)
543 for (bit
= bpp
-1; bit
>=0; bit
--)
545 if (bmask
& (1 << bit
))
547 if (gmask
& (1 << bit
))
549 if (rmask
& (1 << bit
))
552 // Find number of bits in mask (MSB-LSB+1)
553 for (bit
= 0; bit
< bpp
; bit
++)
555 if (bmask
& (1 << bit
))
556 bbits
= bit
-bshift
+1;
557 if (gmask
& (1 << bit
))
558 gbits
= bit
-gshift
+1;
559 if (rmask
& (1 << bit
))
560 rbits
= bit
-rshift
+1;
563 else if ( bpp
== 16 )
575 else if ( bpp
== 32 )
590 * Reading the image data
593 stream
.SeekI(bmpOffset
); // else icon, just carry on
595 unsigned char *data
= ptr
;
597 /* set the whole image to the background color */
598 if ( bpp
< 16 && (comp
== BI_RLE4
|| comp
== BI_RLE8
) )
600 for (int i
= 0; i
< width
* height
; i
++)
609 int linesize
= ((width
* bpp
+ 31) / 32) * 4;
611 /* BMPs are stored upside down */
612 for ( int line
= (height
- 1); line
>= 0; line
-- )
615 for ( int column
= 0; column
< width
; )
620 aByte
= stream
.GetC();
623 for (int bit
= 0; bit
< 8 && column
< width
; bit
++)
625 int index
= ((aByte
& (0x80 >> bit
)) ? 1 : 0);
626 ptr
[poffset
] = cmap
[index
].r
;
627 ptr
[poffset
+ 1] = cmap
[index
].g
;
628 ptr
[poffset
+ 2] = cmap
[index
].b
;
634 if ( comp
== BI_RLE4
)
638 aByte
= stream
.GetC();
646 else if ( aByte
== 1 )
651 else if ( aByte
== 2 )
653 aByte
= stream
.GetC();
655 linepos
= column
* bpp
/ 4;
656 aByte
= stream
.GetC();
657 line
-= aByte
; // upside down
661 int absolute
= aByte
;
664 for (int k
= 0; k
< absolute
; k
++)
669 aByte
= stream
.GetC();
670 nibble
[0] = (wxUint8
)( (aByte
& 0xF0) >> 4 ) ;
671 nibble
[1] = (wxUint8
)( aByte
& 0x0F ) ;
673 ptr
[poffset
] = cmap
[nibble
[k%2
]].r
;
674 ptr
[poffset
+ 1] = cmap
[nibble
[k%2
]].g
;
675 ptr
[poffset
+ 2] = cmap
[nibble
[k%2
]].b
;
680 if ( readBytes
& 0x01 )
681 aByte
= stream
.GetC();
687 nibble
[0] = (wxUint8
)( (aByte
& 0xF0) >> 4 ) ;
688 nibble
[1] = (wxUint8
)( aByte
& 0x0F ) ;
690 for ( int l
= 0; l
< first
&& column
< width
; l
++ )
692 ptr
[poffset
] = cmap
[nibble
[l%2
]].r
;
693 ptr
[poffset
+ 1] = cmap
[nibble
[l%2
]].g
;
694 ptr
[poffset
+ 2] = cmap
[nibble
[l%2
]].b
;
703 for (int nibble
= 0; nibble
< 2 && column
< width
; nibble
++)
705 int index
= ((aByte
& (0xF0 >> nibble
* 4)) >> (!nibble
* 4));
708 ptr
[poffset
] = cmap
[index
].r
;
709 ptr
[poffset
+ 1] = cmap
[index
].g
;
710 ptr
[poffset
+ 2] = cmap
[index
].b
;
717 if ( comp
== BI_RLE8
)
721 aByte
= stream
.GetC();
726 /* column = width; */
728 else if ( aByte
== 1 )
733 else if ( aByte
== 2 )
735 aByte
= stream
.GetC();
737 linepos
= column
* bpp
/ 8;
738 aByte
= stream
.GetC();
743 int absolute
= aByte
;
744 for (int k
= 0; k
< absolute
; k
++)
747 aByte
= stream
.GetC();
748 ptr
[poffset
] = cmap
[aByte
].r
;
749 ptr
[poffset
+ 1] = cmap
[aByte
].g
;
750 ptr
[poffset
+ 2] = cmap
[aByte
].b
;
753 if ( absolute
& 0x01 )
754 aByte
= stream
.GetC();
759 for ( int l
= 0; l
< first
&& column
< width
; l
++ )
761 ptr
[poffset
] = cmap
[aByte
].r
;
762 ptr
[poffset
+ 1] = cmap
[aByte
].g
;
763 ptr
[poffset
+ 2] = cmap
[aByte
].b
;
771 ptr
[poffset
] = cmap
[aByte
].r
;
772 ptr
[poffset
+ 1] = cmap
[aByte
].g
;
773 ptr
[poffset
+ 2] = cmap
[aByte
].b
;
775 // linepos += size; seems to be wrong, RR
779 else if ( bpp
== 24 )
781 stream
.Read(bbuf
, 3);
783 ptr
[poffset
] = (unsigned char)bbuf
[2];
784 ptr
[poffset
+ 1] = (unsigned char)bbuf
[1];
785 ptr
[poffset
+ 2] = (unsigned char)bbuf
[0];
788 else if ( bpp
== 16 )
791 stream
.Read(&aWord
, 2);
792 aWord
= wxUINT16_SWAP_ON_BE(aWord
);
794 /* use the masks and calculated amonut of shift
795 to retrieve the color data out of the word. Then
796 shift it left by (8 - number of bits) such that
797 the image has the proper dynamic range */
798 temp
= (unsigned char)((aWord
& rmask
) >> rshift
<< (8-rbits
));
800 temp
= (unsigned char)((aWord
& gmask
) >> gshift
<< (8-gbits
));
801 ptr
[poffset
+ 1] = temp
;
802 temp
= (unsigned char)((aWord
& bmask
) >> bshift
<< (8-bbits
));
803 ptr
[poffset
+ 2] = temp
;
809 stream
.Read(&aDword
, 4);
810 aDword
= wxINT32_SWAP_ON_BE(aDword
);
812 temp
= (unsigned char)((aDword
& rmask
) >> rshift
);
814 temp
= (unsigned char)((aDword
& gmask
) >> gshift
);
815 ptr
[poffset
+ 1] = temp
;
816 temp
= (unsigned char)((aDword
& bmask
) >> bshift
);
817 ptr
[poffset
+ 2] = temp
;
821 while ( (linepos
< linesize
) && (comp
!= 1) && (comp
!= 2) )
823 stream
.Read(&aByte
, 1);
832 image
->SetMask(false);
834 const wxStreamError err
= stream
.GetLastError();
835 return err
== wxSTREAM_NO_ERROR
|| err
== wxSTREAM_EOF
;
838 bool wxBMPHandler::LoadDib(wxImage
*image
, wxInputStream
& stream
,
839 bool verbose
, bool IsBmp
)
845 wxFileOffset offset
= 0; // keep gcc quiet
848 // read the header off the .BMP format file
850 offset
= stream
.TellI();
851 if (offset
== wxInvalidOffset
)
854 stream
.Read(bbuf
, 2);
855 stream
.Read(dbuf
, 16);
859 stream
.Read(dbuf
, 4);
862 wxInt32 size
= wxINT32_SWAP_ON_BE(dbuf
[0]);
864 offset
= offset
+ wxINT32_SWAP_ON_BE(dbuf
[2]);
866 stream
.Read(dbuf
, 4 * 2);
867 int width
= wxINT32_SWAP_ON_BE((int)dbuf
[0]);
868 int height
= wxINT32_SWAP_ON_BE((int)dbuf
[1]);
869 if ( !IsBmp
)height
= height
/ 2; // for icons divide by 2
874 wxLogError( _("DIB Header: Image width > 32767 pixels for file.") );
877 if ( height
> 32767 )
880 wxLogError( _("DIB Header: Image height > 32767 pixels for file.") );
884 stream
.Read(&aWord
, 2);
887 int planes = (int)wxUINT16_SWAP_ON_BE( aWord );
889 stream
.Read(&aWord
, 2);
890 int bpp
= wxUINT16_SWAP_ON_BE((int)aWord
);
891 if ( bpp
!= 1 && bpp
!= 4 && bpp
!= 8 && bpp
!= 16 && bpp
!= 24 && bpp
!= 32 )
894 wxLogError( _("DIB Header: Unknown bitdepth in file.") );
898 stream
.Read(dbuf
, 4 * 4);
899 int comp
= wxINT32_SWAP_ON_BE((int)dbuf
[0]);
900 if ( comp
!= BI_RGB
&& comp
!= BI_RLE4
&& comp
!= BI_RLE8
&&
901 comp
!= BI_BITFIELDS
)
904 wxLogError( _("DIB Header: Unknown encoding in file.") );
908 stream
.Read(dbuf
, 4 * 2);
909 int ncolors
= wxINT32_SWAP_ON_BE( (int)dbuf
[0] );
912 /* some more sanity checks */
913 if (((comp
== BI_RLE4
) && (bpp
!= 4)) ||
914 ((comp
== BI_RLE8
) && (bpp
!= 8)) ||
915 ((comp
== BI_BITFIELDS
) && (bpp
!= 16 && bpp
!= 32)))
918 wxLogError( _("DIB Header: Encoding doesn't match bitdepth.") );
922 //read DIB; this is the BMP image or the XOR part of an icon image
923 if ( !DoLoadDib(image
, width
, height
, bpp
, ncolors
, comp
, offset
, stream
,
924 verbose
, IsBmp
, true) )
927 wxLogError( _("Error in reading image DIB.") );
933 //read Icon mask which is monochrome
934 //there is no palette, so we will create one
936 if ( !DoLoadDib(&mask
, width
, height
, 1, 2, BI_RGB
, offset
, stream
,
937 verbose
, IsBmp
, false) )
940 wxLogError( _("ICO: Error in reading mask DIB.") );
943 image
->SetMaskFromImage(mask
, 255, 255, 255);
950 bool wxBMPHandler::LoadFile(wxImage
*image
, wxInputStream
& stream
,
951 bool verbose
, int WXUNUSED(index
))
953 // Read a single DIB fom the file:
954 return LoadDib(image
, stream
, verbose
, true/*isBmp*/);
957 bool wxBMPHandler::DoCanRead(wxInputStream
& stream
)
959 unsigned char hdr
[2];
961 if ( !stream
.Read(hdr
, WXSIZEOF(hdr
)) )
964 // do we have the BMP file signature?
965 return hdr
[0] == 'B' && hdr
[1] == 'M';
968 #endif // wxUSE_STREAMS
972 //-----------------------------------------------------------------------------
974 //-----------------------------------------------------------------------------
976 IMPLEMENT_DYNAMIC_CLASS(wxICOHandler
, wxBMPHandler
)
982 wxUint8 bWidth
; // Width of the image
983 wxUint8 bHeight
; // Height of the image (times 2)
984 wxUint8 bColorCount
; // Number of colors in image (0 if >=8bpp)
985 wxUint8 bReserved
; // Reserved
987 // these two are different in icons and cursors:
989 wxUint16 wPlanes
; // Color Planes or XHotSpot
990 wxUint16 wBitCount
; // Bits per pixel or YHotSpot
992 wxUint32 dwBytesInRes
; // how many bytes in this resource?
993 wxUint32 dwImageOffset
; // where in the file is this image
998 wxUint16 idReserved
; // Reserved
999 wxUint16 idType
; // resource type (1 for icons, 2 for cursors)
1000 wxUint16 idCount
; // how many images?
1004 bool wxICOHandler::SaveFile(wxImage
*image
,
1005 wxOutputStream
& stream
,
1009 //sanity check; icon must be less than 127 pixels high and 255 wide
1010 if ( image
->GetHeight () > 127 )
1013 wxLogError(_("ICO: Image too tall for an icon."));
1016 if ( image
->GetWidth () > 255 )
1019 wxLogError(_("ICO: Image too wide for an icon."));
1023 const int images
= 1; // only generate one image
1025 // VS: This is a hack of sort - since ICO and CUR files are almost
1026 // identical, we have all the meat in wxICOHandler and check for
1027 // the actual (handler) type when the code has to distinguish between
1029 int type
= (this->GetType() == wxBITMAP_TYPE_CUR
) ? 2 : 1;
1031 // write a header, (ICONDIR)
1032 // Calculate the header size
1033 wxUint32 offset
= 3 * sizeof(wxUint16
);
1036 IconDir
.idReserved
= 0;
1037 IconDir
.idType
= wxUINT16_SWAP_ON_BE((wxUint16
)type
);
1038 IconDir
.idCount
= wxUINT16_SWAP_ON_BE((wxUint16
)images
);
1039 stream
.Write(&IconDir
.idReserved
, sizeof(IconDir
.idReserved
));
1040 stream
.Write(&IconDir
.idType
, sizeof(IconDir
.idType
));
1041 stream
.Write(&IconDir
.idCount
, sizeof(IconDir
.idCount
));
1042 if ( !stream
.IsOk() )
1045 wxLogError(_("ICO: Error writing the image file!"));
1049 // for each iamage write a description ICONDIRENTRY:
1050 ICONDIRENTRY icondirentry
;
1051 for (int img
= 0; img
< images
; img
++)
1055 if ( image
->HasMask() )
1057 // make another image with black/white:
1058 mask
= image
->ConvertToMono (image
->GetMaskRed(), image
->GetMaskGreen(), image
->GetMaskBlue() );
1060 // now we need to change the masked regions to black:
1061 unsigned char r
= image
->GetMaskRed();
1062 unsigned char g
= image
->GetMaskGreen();
1063 unsigned char b
= image
->GetMaskBlue();
1064 if ( (r
!= 0) || (g
!= 0) || (b
!= 0) )
1066 // Go round and apply black to the masked bits:
1068 for (i
= 0; i
< mask
.GetWidth(); i
++)
1070 for (j
= 0; j
< mask
.GetHeight(); j
++)
1072 if ((r
== mask
.GetRed(i
, j
)) &&
1073 (g
== mask
.GetGreen(i
, j
))&&
1074 (b
== mask
.GetBlue(i
, j
)) )
1075 image
->SetRGB(i
, j
, 0, 0, 0 );
1082 // just make a black mask all over:
1083 mask
= image
->Copy();
1085 for (i
= 0; i
< mask
.GetWidth(); i
++)
1086 for (j
= 0; j
< mask
.GetHeight(); j
++)
1087 mask
.SetRGB(i
, j
, 0, 0, 0 );
1089 // Set the formats for image and mask
1090 // (Windows never saves with more than 8 colors):
1091 image
->SetOption(wxIMAGE_OPTION_BMP_FORMAT
, wxBMP_8BPP
);
1093 // monochome bitmap:
1094 mask
.SetOption(wxIMAGE_OPTION_BMP_FORMAT
, wxBMP_1BPP_BW
);
1096 bool IsMask
= false;
1098 //calculate size and offset of image and mask
1099 wxCountingOutputStream cStream
;
1100 bool bResult
= SaveDib(image
, cStream
, verbose
, IsBmp
, IsMask
);
1104 wxLogError(_("ICO: Error writing the image file!"));
1109 bResult
= SaveDib(&mask
, cStream
, verbose
, IsBmp
, IsMask
);
1113 wxLogError(_("ICO: Error writing the image file!"));
1116 wxUint32 Size
= cStream
.GetSize();
1118 // wxCountingOutputStream::IsOk() always returns true for now and this
1119 // "if" provokes VC++ warnings in optimized build
1121 if ( !cStream
.Ok() )
1124 wxLogError(_("ICO: Error writing the image file!"));
1129 offset
= offset
+ sizeof(ICONDIRENTRY
);
1131 icondirentry
.bWidth
= (wxUint8
)image
->GetWidth();
1132 icondirentry
.bHeight
= (wxUint8
)(2 * image
->GetHeight());
1133 icondirentry
.bColorCount
= 0;
1134 icondirentry
.bReserved
= 0;
1135 icondirentry
.wPlanes
= wxUINT16_SWAP_ON_BE(1);
1136 icondirentry
.wBitCount
= wxUINT16_SWAP_ON_BE(wxBMP_8BPP
);
1137 if ( type
== 2 /*CUR*/)
1139 int hx
= image
->HasOption(wxIMAGE_OPTION_CUR_HOTSPOT_X
) ?
1140 image
->GetOptionInt(wxIMAGE_OPTION_CUR_HOTSPOT_X
) :
1141 image
->GetWidth() / 2;
1142 int hy
= image
->HasOption(wxIMAGE_OPTION_CUR_HOTSPOT_Y
) ?
1143 image
->GetOptionInt(wxIMAGE_OPTION_CUR_HOTSPOT_Y
) :
1144 image
->GetHeight() / 2;
1146 // actually write the values of the hot spot here:
1147 icondirentry
.wPlanes
= wxUINT16_SWAP_ON_BE((wxUint16
)hx
);
1148 icondirentry
.wBitCount
= wxUINT16_SWAP_ON_BE((wxUint16
)hy
);
1150 icondirentry
.dwBytesInRes
= wxUINT32_SWAP_ON_BE(Size
);
1151 icondirentry
.dwImageOffset
= wxUINT32_SWAP_ON_BE(offset
);
1153 // increase size to allow for the data written:
1157 stream
.Write(&icondirentry
.bWidth
, sizeof(icondirentry
.bWidth
));
1158 stream
.Write(&icondirentry
.bHeight
, sizeof(icondirentry
.bHeight
));
1159 stream
.Write(&icondirentry
.bColorCount
, sizeof(icondirentry
.bColorCount
));
1160 stream
.Write(&icondirentry
.bReserved
, sizeof(icondirentry
.bReserved
));
1161 stream
.Write(&icondirentry
.wPlanes
, sizeof(icondirentry
.wPlanes
));
1162 stream
.Write(&icondirentry
.wBitCount
, sizeof(icondirentry
.wBitCount
));
1163 stream
.Write(&icondirentry
.dwBytesInRes
, sizeof(icondirentry
.dwBytesInRes
));
1164 stream
.Write(&icondirentry
.dwImageOffset
, sizeof(icondirentry
.dwImageOffset
));
1165 if ( !stream
.IsOk() )
1168 wxLogError(_("ICO: Error writing the image file!"));
1172 // actually save it:
1174 bResult
= SaveDib(image
, stream
, verbose
, IsBmp
, IsMask
);
1178 wxLogError(_("ICO: Error writing the image file!"));
1183 bResult
= SaveDib(&mask
, stream
, verbose
, IsBmp
, IsMask
);
1187 wxLogError(_("ICO: Error writing the image file!"));
1191 } // end of for loop
1196 bool wxICOHandler::LoadFile(wxImage
*image
, wxInputStream
& stream
,
1197 bool verbose
, int index
)
1200 return DoLoadFile(image
, stream
, verbose
, index
);
1203 bool wxICOHandler::DoLoadFile(wxImage
*image
, wxInputStream
& stream
,
1204 bool WXUNUSED(verbose
), int index
)
1206 bool bResult
wxDUMMY_INITIALIZE(false);
1211 wxFileOffset iPos
= stream
.TellI();
1212 stream
.Read(&IconDir
, sizeof(IconDir
));
1213 wxUint16 nIcons
= wxUINT16_SWAP_ON_BE(IconDir
.idCount
);
1214 // nType is 1 for Icons, 2 for Cursors:
1215 wxUint16 nType
= wxUINT16_SWAP_ON_BE(IconDir
.idType
);
1217 // loop round the icons and choose the best one:
1218 ICONDIRENTRY
*pIconDirEntry
= new ICONDIRENTRY
[nIcons
];
1219 ICONDIRENTRY
*pCurrentEntry
= pIconDirEntry
;
1222 int iSel
= wxNOT_FOUND
;
1224 for (int i
= 0; i
< nIcons
; i
++ )
1226 stream
.Read(pCurrentEntry
, sizeof(ICONDIRENTRY
));
1227 // bHeight and bColorCount are wxUint8
1228 if ( pCurrentEntry
->bWidth
>= wMax
)
1230 // see if we have more colors, ==0 indicates > 8bpp:
1231 if ( pCurrentEntry
->bColorCount
== 0 )
1232 pCurrentEntry
->bColorCount
= 255;
1233 if ( pCurrentEntry
->bColorCount
>= colmax
)
1236 wMax
= pCurrentEntry
->bWidth
;
1237 colmax
= pCurrentEntry
->bColorCount
;
1245 // VS: Note that we *have* to run the loop above even if index != -1, because
1246 // it reads ICONDIRENTRies.
1250 if ( iSel
== wxNOT_FOUND
|| iSel
< 0 || iSel
>= nIcons
)
1252 wxLogError(_("ICO: Invalid icon index."));
1257 // seek to selected icon:
1258 pCurrentEntry
= pIconDirEntry
+ iSel
;
1259 stream
.SeekI(iPos
+ wxUINT32_SWAP_ON_BE(pCurrentEntry
->dwImageOffset
), wxFromStart
);
1260 bResult
= LoadDib(image
, stream
, true, IsBmp
);
1261 bool bIsCursorType
= (this->GetType() == wxBITMAP_TYPE_CUR
) || (this->GetType() == wxBITMAP_TYPE_ANI
);
1262 if ( bResult
&& bIsCursorType
&& nType
== 2 )
1264 // it is a cursor, so let's set the hotspot:
1265 image
->SetOption(wxIMAGE_OPTION_CUR_HOTSPOT_X
, wxUINT16_SWAP_ON_BE(pCurrentEntry
->wPlanes
));
1266 image
->SetOption(wxIMAGE_OPTION_CUR_HOTSPOT_Y
, wxUINT16_SWAP_ON_BE(pCurrentEntry
->wBitCount
));
1269 delete[] pIconDirEntry
;
1273 int wxICOHandler::GetImageCount(wxInputStream
& stream
)
1276 wxFileOffset iPos
= stream
.TellI();
1278 stream
.Read(&IconDir
, sizeof(IconDir
));
1279 wxUint16 nIcons
= wxUINT16_SWAP_ON_BE(IconDir
.idCount
);
1284 bool wxICOHandler::DoCanRead(wxInputStream
& stream
)
1287 unsigned char hdr
[4];
1288 if ( !stream
.Read(hdr
, WXSIZEOF(hdr
)) )
1291 // hdr[2] is one for an icon and two for a cursor
1292 return hdr
[0] == '\0' && hdr
[1] == '\0' && hdr
[2] == '\1' && hdr
[3] == '\0';
1295 #endif // wxUSE_STREAMS
1298 //-----------------------------------------------------------------------------
1300 //-----------------------------------------------------------------------------
1302 IMPLEMENT_DYNAMIC_CLASS(wxCURHandler
, wxICOHandler
)
1306 bool wxCURHandler::DoCanRead(wxInputStream
& stream
)
1309 unsigned char hdr
[4];
1310 if ( !stream
.Read(hdr
, WXSIZEOF(hdr
)) )
1313 // hdr[2] is one for an icon and two for a cursor
1314 return hdr
[0] == '\0' && hdr
[1] == '\0' && hdr
[2] == '\2' && hdr
[3] == '\0';
1317 #endif // wxUSE_STREAMS
1319 //-----------------------------------------------------------------------------
1321 //-----------------------------------------------------------------------------
1323 IMPLEMENT_DYNAMIC_CLASS(wxANIHandler
, wxCURHandler
)
1327 bool wxANIHandler::LoadFile(wxImage
*image
, wxInputStream
& stream
,
1328 bool WXUNUSED(verbose
), int index
)
1330 wxANIDecoder decoder
;
1331 if (!decoder
.Load(stream
))
1334 return decoder
.ConvertToImage(index
!= -1 ? (size_t)index
: 0, image
);
1337 bool wxANIHandler::DoCanRead(wxInputStream
& stream
)
1340 return decod
.CanRead(stream
);
1343 int wxANIHandler::GetImageCount(wxInputStream
& stream
)
1345 wxANIDecoder decoder
;
1346 if (!decoder
.Load(stream
))
1349 return decoder
.GetFrameCount();
1352 #endif // wxUSE_STREAMS
1354 #endif // wxUSE_ICO_CUR
1356 #endif // wxUSE_IMAGE