+
+ return ok;
+}
+
+bool wxGIFHandler::SaveFile(wxImage *image,
+ wxOutputStream& stream, bool verbose)
+{
+#if wxUSE_PALETTE
+ wxRGB pal[256];
+ int palCount;
+ int maskIndex;
+
+ return wxGIFHandler_GetPalette(*image, pal, &palCount, &maskIndex)
+ && DoSaveFile(*image, &stream, verbose, true /*first?*/, 0,
+ false /*loop?*/, pal, palCount, maskIndex)
+ && wxGIFHandler_WriteTerm(&stream);
+#else
+ wxUnusedVar(image);
+ wxUnusedVar(stream);
+ wxUnusedVar(verbose);
+ return false;
+#endif
+}
+
+bool wxGIFHandler::DoCanRead( wxInputStream& stream )
+{
+ wxGIFDecoder decod;
+ return decod.CanRead(stream);
+ // it's ok to modify the stream position here
+}
+
+int wxGIFHandler::DoGetImageCount( wxInputStream& stream )
+{
+ wxGIFDecoder decod;
+ wxGIFErrorCode error = decod.LoadGIF(stream);
+ if ( (error != wxGIF_OK) && (error != wxGIF_TRUNCATED) )
+ return -1;
+
+ // NOTE: this function modifies the current stream position but it's ok
+ // (see wxImageHandler::GetImageCount)
+
+ return decod.GetFrameCount();
+}
+
+bool wxGIFHandler::DoSaveFile(const wxImage& image, wxOutputStream *stream,
+ bool WXUNUSED(verbose), bool first, int delayMilliSecs, bool loop,
+ const wxRGB *pal, int palCount, int maskIndex)
+{
+ const unsigned long colorcount = image.CountColours(256+1);
+ bool ok = colorcount && (colorcount <= 256);
+ if (!ok)
+ {
+ return false;
+ }
+
+ int width = image.GetWidth();
+ int height = image.GetHeight();
+ int width_even = width + ((width % 2) ? 1 : 0);
+
+ if (first)
+ {
+ ok = wxGIFHandler_WriteHeader(stream, width, height, loop,
+ pal, palCount);
+ }
+
+ ok = ok
+ && wxGIFHandler_WriteComment(stream,
+ image.GetOption(wxIMAGE_OPTION_GIF_COMMENT))
+ && wxGIFHandler_WriteControl(stream, maskIndex, delayMilliSecs)
+ && wxGIFHandler_WriteByte(stream, GIF_MARKER_SEP)
+ && wxGIFHandler_WriteRect(stream, width, height);
+
+ // local palette
+ if (first)
+ {
+ // we already saved the (global) palette
+ ok = ok && wxGIFHandler_WriteZero(stream);
+ }
+ else
+ {
+ const int bpp = wxGIFHandler_BitSize(palCount);
+ wxUint8 b;
+
+ b = 0x80;
+ b |=(bpp - 1) << 5;
+ b |=(bpp - 1);
+ b &=~0x40; // clear interlaced
+
+ ok = ok && wxGIFHandler_WriteByte(stream, b)
+ && wxGIFHandler_WritePalette(stream, pal, palCount, bpp);
+ }
+
+ if (!ok)
+ {
+ return false;
+ }
+
+ if (!InitHashTable())
+ {
+ wxLogError(_("Couldn't initialize GIF hash table."));
+ return false;
+ }
+
+ const wxUint8 *src = image.GetData();
+ wxUint8 *eightBitData = new wxUint8[width];
+
+ SetupCompress(stream, 8);
+
+ m_pixelCount = height * width_even;
+ for (int y = 0; y < height; y++)
+ {
+ m_pixelCount -= width_even;
+ for (int x = 0; x < width; x++)
+ {
+ wxRGB rgb;
+ rgb.red = src[0];
+ rgb.green = src[1];
+ rgb.blue = src[2];
+ int index = wxGIFHandler_PaletteFind(rgb, pal, palCount);
+ wxASSERT(index != wxNOT_FOUND);
+ eightBitData[x] = (wxUint8)index;
+ src+=3;
+ }
+
+ ok = CompressLine(stream, eightBitData, width);
+ if (!ok)
+ {
+ break;
+ }
+ }
+
+ delete [] eightBitData;
+
+ wxDELETE(m_hashTable);
+
+ return ok;
+}
+
+bool wxGIFHandler::SaveAnimation(const wxImageArray& images,
+ wxOutputStream *stream, bool verbose, int delayMilliSecs)
+{
+#if wxUSE_PALETTE
+ bool ok = true;
+ size_t i;
+
+ wxSize size(0,0);
+ for (i = 0; (i < images.GetCount()) && ok; i++)
+ {
+ const wxImage& image = images.Item(i);
+ wxSize temp(image.GetWidth(), image.GetHeight());
+ ok = ok && image.HasPalette();
+ if (i)
+ {
+ ok = ok && (size == temp);
+ }
+ else
+ {
+ size = temp;
+ }
+ }
+
+ for (i = 0; (i < images.GetCount()) && ok; i++)
+ {
+ const wxImage& image = images.Item(i);
+
+ wxRGB pal[256];
+ int palCount;
+ int maskIndex;
+
+ ok = wxGIFHandler_GetPalette(image, pal, &palCount, &maskIndex)
+ && DoSaveFile(image, stream, verbose, i == 0 /*first?*/, delayMilliSecs,
+ true /*loop?*/, pal, palCount, maskIndex);
+ }
+
+ return ok && wxGIFHandler_WriteTerm(stream);
+#else
+ wxUnusedVar(images);
+ wxUnusedVar(stream);
+ wxUnusedVar(verbose);
+ wxUnusedVar(delayMilliSecs);
+
+ return false;
+#endif
+}
+
+bool wxGIFHandler::CompressOutput(wxOutputStream *stream, int code)
+{
+ if (code == FLUSH_OUTPUT)
+ {
+ while (m_crntShiftState > 0)
+ {
+ // Get rid of what is left in DWord, and flush it.
+ if (!wxGIFHandler_BufferedOutput(stream, m_LZBuf,
+ m_crntShiftDWord & 0xff))
+ {
+ return false;
+ }
+ m_crntShiftDWord >>= 8;
+ m_crntShiftState -= 8;
+ }
+ m_crntShiftState = 0; // For next time.
+ if (!wxGIFHandler_BufferedOutput(stream, m_LZBuf, FLUSH_OUTPUT))
+ {
+ return false;
+ }
+ }
+ else
+ {
+ m_crntShiftDWord |= ((long) code) << m_crntShiftState;
+ m_crntShiftState += m_runningBits;
+ while (m_crntShiftState >= 8)
+ {
+ // Dump out full bytes:
+ if (!wxGIFHandler_BufferedOutput(stream, m_LZBuf,
+ m_crntShiftDWord & 0xff))
+ {
+ return false;
+ }
+ m_crntShiftDWord >>= 8;
+ m_crntShiftState -= 8;
+ }
+ }
+
+ // If code can't fit into RunningBits bits, must raise its size. Note
+ // however that codes above LZ_MAX_CODE are used for special signaling.
+ if ( (m_runningCode >= m_maxCode1) && (code <= LZ_MAX_CODE))
+ {
+ m_maxCode1 = 1 << ++m_runningBits;
+ }
+ return true;
+}
+
+bool wxGIFHandler::SetupCompress(wxOutputStream *stream, int bpp)
+{
+ m_LZBuf[0] = 0; // Nothing was output yet.
+ m_clearCode = (1 << bpp);
+ m_EOFCode = m_clearCode + 1;
+ m_runningCode = m_EOFCode + 1;
+ m_runningBits = bpp + 1; // Number of bits per code.
+ m_maxCode1 = 1 << m_runningBits; // Max. code + 1.
+ m_crntCode = FIRST_CODE; // Signal that this is first one!
+ m_crntShiftState = 0; // No information in CrntShiftDWord.
+ m_crntShiftDWord = 0;
+
+ // Clear hash table and send Clear to make sure the decoder does the same.
+ ClearHashTable();
+
+ return wxGIFHandler_WriteByte(stream, (wxUint8)bpp)
+ && CompressOutput(stream, m_clearCode);
+}
+
+bool wxGIFHandler::CompressLine(wxOutputStream *stream,
+ const wxUint8 *line, int lineLen)
+{
+ int i = 0, crntCode, newCode;
+ unsigned long newKey;
+ wxUint8 pixel;
+ if (m_crntCode == FIRST_CODE) // It's first time!
+ crntCode = line[i++];
+ else
+ crntCode = m_crntCode; // Get last code in compression.
+
+ while (i < lineLen)
+ {
+ // Decode lineLen items.
+ pixel = line[i++]; // Get next pixel from stream.
+ // Form a new unique key to search hash table for the code combines
+ // crntCode as Prefix string with Pixel as postfix char.
+ newKey = (((unsigned long) crntCode) << 8) + pixel;
+ if ((newCode = ExistsHashTable(newKey)) >= 0)
+ {
+ // This Key is already there, or the string is old one, so
+ // simply take new code as our crntCode:
+ crntCode = newCode;
+ }
+ else
+ {
+ // Put it in hash table, output the prefix code, and make our
+ // crntCode equal to Pixel.
+ if (!CompressOutput(stream, crntCode))
+ {
+ return false;
+ }
+
+ crntCode = pixel;
+
+ // If however the HashTable is full, we send a clear first and
+ // Clear the hash table.
+ if (m_runningCode >= LZ_MAX_CODE)
+ {
+ // Time to do some clearance:
+ if (!CompressOutput(stream, m_clearCode))
+ {
+ return false;
+ }
+
+ m_runningCode = m_EOFCode + 1;
+ m_runningBits = 8 + 1;
+ m_maxCode1 = 1 << m_runningBits;
+ ClearHashTable();
+ }
+ else
+ {
+ // Put this unique key with its relative Code in hash table:
+ InsertHashTable(newKey, m_runningCode++);
+ }
+ }
+ }
+ // Preserve the current state of the compression algorithm:
+ m_crntCode = crntCode;
+ if (m_pixelCount == 0)
+ {
+ // We are done - output last Code and flush output buffers:
+ if (!CompressOutput(stream, crntCode)
+ || !CompressOutput(stream, m_EOFCode)
+ || !CompressOutput(stream, FLUSH_OUTPUT))
+ {
+ return false;
+ }
+ }
+
+ return true;
+}
+
+#endif // wxUSE_STREAMS
+
+bool wxGIFHandler::InitHashTable()
+{
+ if (!m_hashTable)
+ {
+ m_hashTable = new GifHashTableType();
+ }
+
+ if (!m_hashTable)
+ {
+ return false;
+ }
+
+ ClearHashTable();
+
+ return true;
+}
+
+void wxGIFHandler::ClearHashTable()
+{
+ int index = HT_SIZE;
+ wxUint32 *HTable = m_hashTable->HTable;
+
+ while (--index>=0)
+ {
+ HTable[index] = 0xfffffffful;
+ }
+}
+
+void wxGIFHandler::InsertHashTable(unsigned long key, int code)
+{
+ int hKey = wxGIFHandler_KeyItem(key);
+ wxUint32 *HTable = m_hashTable->HTable;
+
+ while (HT_GET_KEY(HTable[hKey]) != 0xFFFFFL)
+ {
+ hKey = (hKey + 1) & HT_KEY_MASK;
+ }
+ HTable[hKey] = HT_PUT_KEY(key) | HT_PUT_CODE(code);
+}
+
+
+int wxGIFHandler::ExistsHashTable(unsigned long key)
+{
+ int hKey = wxGIFHandler_KeyItem(key);
+ wxUint32 *HTable = m_hashTable->HTable, HTKey;
+
+ while ((HTKey = HT_GET_KEY(HTable[hKey])) != 0xFFFFFL)
+ {
+ if (key == HTKey)
+ {
+ return HT_GET_CODE(HTable[hKey]);
+ }
+ hKey = (hKey + 1) & HT_KEY_MASK;
+ }
+ return -1;
+}
+
+// ---------------------------------------------------------------------------
+// implementation of global private functions
+// ---------------------------------------------------------------------------
+
+int wxGIFHandler_KeyItem(unsigned long item)
+{
+ return ((item >> 12) ^ item) & HT_KEY_MASK;
+}
+
+#if wxUSE_STREAMS
+
+int wxGIFHandler_BitSize(int n)
+{
+ int i;
+ for (i = 1; i <= 8; i++)
+ {
+ if ((1 << i) >= n)
+ {
+ break;
+ }
+ }
+ return i;
+}
+
+#if wxUSE_PALETTE
+bool wxGIFHandler_GetPalette(const wxImage& image,
+ wxRGB *pal, int *pPalCount, int *pMaskIndex)
+{
+ if (!image.HasPalette())
+ {
+ return false;
+ }
+
+ const wxPalette& palette = image.GetPalette();
+ int palCount = palette.GetColoursCount();
+
+ for (int i = 0; i < palCount; ++i)
+ {
+ if (!palette.GetRGB(i, &pal[i].red, &pal[i].green, &pal[i].blue))
+ {
+ break;
+ }
+ }
+ if (image.HasMask())
+ {
+ wxRGB mask;
+
+ mask.red = image.GetMaskRed();
+ mask.green = image.GetMaskGreen();
+ mask.blue = image.GetMaskBlue();
+ *pMaskIndex = wxGIFHandler_PaletteFind(mask, pal, palCount);
+ if ( (*pMaskIndex == wxNOT_FOUND) && (palCount < 256))
+ {
+ *pMaskIndex = palCount;
+ pal[palCount++] = mask;
+ }
+ }
+ else
+ {
+ *pMaskIndex = wxNOT_FOUND;
+ }
+ *pPalCount = palCount;
+
+ return true;
+}
+#endif // wxUSE_PALETTE
+
+int wxGIFHandler_PaletteFind(const wxRGB& clr, const wxRGB *array, int count)
+{
+ for (int i = 0; i < count; i++)
+ {
+ if ( (clr.red == array[i].red)
+ && (clr.green == array[i].green)
+ && (clr.blue == array[i].blue))
+ {
+ return i;
+ }
+ }
+
+ return wxNOT_FOUND;
+}
+
+bool wxGIFHandler_Write(wxOutputStream *stream, const void *buf, size_t len)
+{
+ return (len == stream->Write(buf, len).LastWrite());
+}
+
+bool wxGIFHandler_WriteByte(wxOutputStream *stream, wxUint8 byte)
+{
+ return wxGIFHandler_Write(stream, &byte, sizeof(byte));
+}
+
+bool wxGIFHandler_WriteWord(wxOutputStream *stream, wxUint16 word)
+{
+ wxUint8 buf[2];
+
+ buf[0] = word & 0xff;
+ buf[1] = (word >> 8) & 0xff;
+ return wxGIFHandler_Write(stream, &buf, sizeof(buf));
+}
+
+bool wxGIFHandler_WriteHeader(wxOutputStream *stream, int width, int height,
+ bool loop, const wxRGB *pal, int palCount)
+{
+ const int bpp = wxGIFHandler_BitSize(palCount);
+ wxUint8 buf[3];
+
+ bool ok = wxGIFHandler_Write(stream, GIF89_HDR, sizeof(GIF89_HDR)-1)
+ && wxGIFHandler_WriteWord(stream, (wxUint16) width)
+ && wxGIFHandler_WriteWord(stream, (wxUint16) height);
+
+ buf[0] = 0x80;
+ buf[0] |=(bpp - 1) << 5;
+ buf[0] |=(bpp - 1);
+ buf[1] = 0; // background color == entry 0
+ buf[2] = 0; // aspect ratio 1:1
+ ok = ok && wxGIFHandler_Write(stream, buf, sizeof(buf))
+ && wxGIFHandler_WritePalette(stream, pal, palCount, bpp);
+
+ if (loop)
+ {
+ ok = ok && wxGIFHandler_WriteLoop(stream);
+ }
+
projects / wxWidgets.git / blobdiff