+ if (cmap)
+ delete[] cmap;
+
+ image->SetMask(FALSE);
+
+ return stream.IsOk();
+}
+
+bool wxBMPHandler::LoadDib(wxImage *image, wxInputStream& stream,
+ bool verbose, bool IsBmp)
+{
+ wxUint16 aWord;
+ wxInt32 dbuf[4];
+ wxInt8 bbuf[4];
+ off_t offset;
+
+ offset = 0; // keep gcc quiet
+ if ( IsBmp )
+ {
+ // read the header off the .BMP format file
+
+ offset = stream.TellI();
+ if (offset == wxInvalidOffset) offset = 0;
+
+ stream.Read(bbuf, 2);
+ stream.Read(dbuf, 16);
+ }
+ else
+ {
+ stream.Read(dbuf, 4);
+ }
+ #if 0 // unused
+ wxInt32 size = wxINT32_SWAP_ON_BE(dbuf[0]);
+ #endif
+ offset = offset + wxINT32_SWAP_ON_BE(dbuf[2]);
+
+ stream.Read(dbuf, 4 * 2);
+ int width = (int)wxINT32_SWAP_ON_BE(dbuf[0]);
+ int height = (int)wxINT32_SWAP_ON_BE(dbuf[1]);
+ if ( !IsBmp)height = height / 2; // for icons divide by 2
+
+ if ( width > 32767 )
+ {
+ if (verbose)
+ wxLogError( _("DIB Header: Image width > 32767 pixels for file.") );
+ return FALSE;
+ }
+ if ( height > 32767 )
+ {
+ if (verbose)
+ wxLogError( _("DIB Header: Image height > 32767 pixels for file.") );
+ return FALSE;
+ }
+
+ stream.Read(&aWord, 2);
+ /*
+ TODO
+ int planes = (int)wxUINT16_SWAP_ON_BE( aWord );
+ */
+ stream.Read(&aWord, 2);
+ int bpp = (int)wxUINT16_SWAP_ON_BE(aWord);
+ if ( bpp != 1 && bpp != 4 && bpp != 8 && bpp != 16 && bpp != 24 && bpp != 32 )
+ {
+ if (verbose)
+ wxLogError( _("DIB Header: Unknown bitdepth in file.") );
+ return FALSE;
+ }
+
+ stream.Read(dbuf, 4 * 4);
+ int comp = (int)wxINT32_SWAP_ON_BE(dbuf[0]);
+ if ( comp != BI_RGB && comp != BI_RLE4 && comp != BI_RLE8 &&
+ comp != BI_BITFIELDS )
+ {
+ if (verbose)
+ wxLogError( _("DIB Header: Unknown encoding in file.") );
+ return FALSE;
+ }
+
+ stream.Read(dbuf, 4 * 2);
+ int ncolors = (int)wxINT32_SWAP_ON_BE( dbuf[0] );
+ if (ncolors == 0)
+ ncolors = 1 << bpp;
+ /* some more sanity checks */
+ if (((comp == BI_RLE4) && (bpp != 4)) ||
+ ((comp == BI_RLE8) && (bpp != 8)) ||
+ ((comp == BI_BITFIELDS) && (bpp != 16 && bpp != 32)))
+ {
+ if (verbose)
+ wxLogError( _("DIB Header: Encoding doesn't match bitdepth.") );
+ return FALSE;
+ }
+
+ //read DIB; this is the BMP image or the XOR part of an icon image
+ if ( !DoLoadDib(image, width, height, bpp, ncolors, comp, offset, stream,
+ verbose, IsBmp, TRUE) )
+ {
+ if (verbose)
+ wxLogError( _("Error in reading image DIB .") );
+ return FALSE;
+ }
+
+ if ( !IsBmp )
+ {
+ //read Icon mask which is monochrome
+ //there is no palette, so we will create one
+ wxImage mask;
+ if ( !DoLoadDib(&mask, width, height, 1, 2, BI_RGB, offset, stream,
+ verbose, IsBmp, FALSE) )
+ {
+ if (verbose)
+ wxLogError( _("ICO: Error in reading mask DIB.") );
+ return FALSE;
+ }
+ image->SetMaskFromImage(mask, 255, 255, 255);
+
+ }
+
+ return TRUE;
+}
+
+bool wxBMPHandler::LoadFile(wxImage *image, wxInputStream& stream,
+ bool verbose, int WXUNUSED(index))
+{
+ // Read a single DIB fom the file:
+ return LoadDib(image, stream, verbose, TRUE/*isBmp*/);
+}
+
+bool wxBMPHandler::DoCanRead(wxInputStream& stream)
+{
+ unsigned char hdr[2];
+
+ stream.Read(hdr, 2);
+ stream.SeekI(-2, wxFromCurrent);
+ return (hdr[0] == 'B' && hdr[1] == 'M');
+}
+
+
+#if wxUSE_ICO_CUR
+//-----------------------------------------------------------------------------
+// wxICOHandler
+//-----------------------------------------------------------------------------
+
+IMPLEMENT_DYNAMIC_CLASS(wxICOHandler, wxBMPHandler)
+
+struct ICONDIRENTRY
+{
+ wxUint8 bWidth; // Width of the image
+ wxUint8 bHeight; // Height of the image (times 2)
+ wxUint8 bColorCount; // Number of colors in image (0 if >=8bpp)
+ wxUint8 bReserved; // Reserved
+
+ // these two are different in icons and cursors:
+ // icon or cursor
+ wxUint16 wPlanes; // Color Planes or XHotSpot
+ wxUint16 wBitCount; // Bits per pixel or YHotSpot
+
+ wxUint32 dwBytesInRes; // how many bytes in this resource?
+ wxUint32 dwImageOffset; // where in the file is this image
+};
+
+struct ICONDIR
+{
+ wxUint16 idReserved; // Reserved
+ wxUint16 idType; // resource type (1 for icons, 2 for cursors)
+ wxUint16 idCount; // how many images?
+};
+
+
+bool wxICOHandler::SaveFile(wxImage *image,
+ wxOutputStream& stream,
+ bool verbose)
+
+{
+ bool bResult = FALSE;
+ //sanity check; icon must be less than 127 pixels high and 255 wide
+ if ( image->GetHeight () > 127 )
+ {
+ if ( verbose )
+ wxLogError(_("ICO: Image too tall for an icon."));
+ return FALSE;
+ }
+ if ( image->GetWidth () > 255 )
+ {
+ if ( verbose )
+ wxLogError(_("ICO: Image too wide for an icon."));
+ return FALSE;
+ }
+
+ int images = 1; // only generate one image
+
+ // VS: This is a hack of sort - since ICO and CUR files are almost
+ // identical, we have all the meat in wxICOHandler and check for
+ // the actual (handler) type when the code has to distinguish between
+ // the two formats
+ int type = (this->GetType() == wxBITMAP_TYPE_CUR) ? 2 : 1;
+
+ // write a header, (ICONDIR)
+ // Calculate the header size
+ wxUint32 offset = 3 * sizeof(wxUint16);
+
+ ICONDIR IconDir;
+ IconDir.idReserved = 0;
+ IconDir.idType = wxUINT16_SWAP_ON_BE(type);
+ IconDir.idCount = wxUINT16_SWAP_ON_BE(images);
+ stream.Write(&IconDir.idReserved, sizeof(IconDir.idReserved));
+ stream.Write(&IconDir.idType, sizeof(IconDir.idType));
+ stream.Write(&IconDir.idCount, sizeof(IconDir.idCount));
+ if ( !stream.IsOk() )
+ {
+ if ( verbose )
+ wxLogError(_("ICO: Error writing the image file!"));
+ return FALSE;
+ }
+
+ // for each iamage write a description ICONDIRENTRY:
+ ICONDIRENTRY icondirentry;
+ for (int i = 0; i < images; i++)
+ {
+ wxImage mask;
+
+ if ( image->HasMask() )
+ {
+ // make another image with black/white:
+ mask = image->ConvertToMono (image->GetMaskRed(), image->GetMaskGreen(), image->GetMaskBlue() );
+
+ // now we need to change the masked regions to black:
+ unsigned char r = image->GetMaskRed();
+ unsigned char g = image->GetMaskGreen();
+ unsigned char b = image->GetMaskBlue();
+ if ( (r != 0) || (g != 0) || (b != 0) )
+ {
+ // Go round and apply black to the masked bits:
+ int i, j;
+ for (i = 0; i < mask.GetWidth(); i++)
+ {
+ for (j = 0; j < mask.GetHeight(); j++)
+ {
+ if ((r == mask.GetRed(i, j)) &&
+ (g == mask.GetGreen(i, j))&&
+ (b == mask.GetBlue(i, j)) )
+ image->SetRGB(i, j, 0, 0, 0 );
+ }
+ }
+ }
+ }
+ else
+ {
+ // just make a black mask all over:
+ mask = image->Copy();
+ int i, j;
+ for (i = 0; i < mask.GetWidth(); i++)
+ for (j = 0; j < mask.GetHeight(); j++)
+ mask.SetRGB(i, j, 0, 0, 0 );
+ }
+ // Set the formats for image and mask
+ // (Windows never saves with more than 8 colors):
+ image->SetOption(wxIMAGE_OPTION_BMP_FORMAT, wxBMP_8BPP);
+
+ // monochome bitmap:
+ mask.SetOption(wxIMAGE_OPTION_BMP_FORMAT, wxBMP_1BPP_BW);
+ bool IsBmp = FALSE;
+ bool IsMask = FALSE;
+
+ //calculate size and offset of image and mask
+ wxCountingOutputStream cStream;
+ bResult = SaveDib(image, cStream, verbose, IsBmp, IsMask);
+ if ( !bResult )
+ {
+ if ( verbose )
+ wxLogError(_("ICO: Error writing the image file!"));
+ return FALSE;
+ }
+ IsMask = TRUE;
+
+ bResult = SaveDib(&mask, cStream, verbose, IsBmp, IsMask);
+ if ( !bResult )
+ {
+ if ( verbose )
+ wxLogError(_("ICO: Error writing the image file!"));
+ return FALSE;
+ }
+ wxUint32 Size = cStream.GetSize();
+
+ // wxCountingOutputStream::Ok() always returns TRUE for now and this
+ // "if" provokes VC++ warnings in optimized build
+#if 0
+ if ( !cStream.Ok() )
+ {
+ if ( verbose )
+ wxLogError(_("ICO: Error writing the image file!"));
+ return FALSE;
+ }
+#endif // 0
+
+ offset = offset + sizeof(ICONDIRENTRY);
+
+ icondirentry.bWidth = image->GetWidth();
+ icondirentry.bHeight = 2 * image->GetHeight();
+ icondirentry.bColorCount = 0;
+ icondirentry.bReserved = 0;
+ icondirentry.wPlanes = wxUINT16_SWAP_ON_BE(1);
+ icondirentry.wBitCount = wxUINT16_SWAP_ON_BE(wxBMP_8BPP);
+ if ( type == 2 /*CUR*/)
+ {
+ int hx = image->HasOption(wxIMAGE_OPTION_CUR_HOTSPOT_X) ?
+ image->GetOptionInt(wxIMAGE_OPTION_CUR_HOTSPOT_X) :
+ image->GetWidth() / 2;
+ int hy = image->HasOption(wxIMAGE_OPTION_CUR_HOTSPOT_Y) ?
+ image->GetOptionInt(wxIMAGE_OPTION_CUR_HOTSPOT_Y) :
+ image->GetHeight() / 2;
+
+ // actually write the values of the hot spot here:
+ icondirentry.wPlanes = wxUINT16_SWAP_ON_BE((wxUint16)hx);
+ icondirentry.wBitCount = wxUINT16_SWAP_ON_BE((wxUint16)hy);
+ }
+ icondirentry.dwBytesInRes = wxUINT32_SWAP_ON_BE(Size);
+ icondirentry.dwImageOffset = wxUINT32_SWAP_ON_BE(offset);
+
+ // increase size to allow for the data written:
+ offset += Size;
+
+ // write to stream:
+ stream.Write(&icondirentry.bWidth, sizeof(icondirentry.bWidth));
+ stream.Write(&icondirentry.bHeight, sizeof(icondirentry.bHeight));
+ stream.Write(&icondirentry.bColorCount, sizeof(icondirentry.bColorCount));
+ stream.Write(&icondirentry.bReserved, sizeof(icondirentry.bReserved));
+ stream.Write(&icondirentry.wPlanes, sizeof(icondirentry.wPlanes));
+ stream.Write(&icondirentry.wBitCount, sizeof(icondirentry.wBitCount));
+ stream.Write(&icondirentry.dwBytesInRes, sizeof(icondirentry.dwBytesInRes));
+ stream.Write(&icondirentry.dwImageOffset, sizeof(icondirentry.dwImageOffset));
+ if ( !stream.IsOk() )
+ {
+ if ( verbose )
+ wxLogError(_("ICO: Error writing the image file!"));
+ return FALSE;
+ }
+
+ // actually save it:
+ IsMask = FALSE;
+ bResult = SaveDib(image, stream, verbose, IsBmp, IsMask);
+ if ( !bResult )
+ {
+ if ( verbose )
+ wxLogError(_("ICO: Error writing the image file!"));
+ return FALSE;
+ }
+ IsMask = TRUE;
+
+ bResult = SaveDib(&mask, stream, verbose, IsBmp, IsMask);
+ if ( !bResult )
+ {
+ if ( verbose )
+ wxLogError(_("ICO: Error writing the image file!"));
+ return FALSE;
+ }
projects / wxWidgets.git / blobdiff