+}
+
+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);
+ }
+
+ return ok;
+}
+
+bool wxGIFHandler_WriteRect(wxOutputStream *stream, int width, int height)
+{
+ return wxGIFHandler_WriteWord(stream, 0) // left
+ && wxGIFHandler_WriteWord(stream, 0) // top
+ && wxGIFHandler_WriteWord(stream, (wxUint16) width)
+ && wxGIFHandler_WriteWord(stream, (wxUint16) height);
+}
+
+#if wxUSE_PALETTE
+bool wxGIFHandler_WriteTerm(wxOutputStream *stream)
+{
+ return wxGIFHandler_WriteByte(stream, GIF_MARKER_ENDOFDATA);
+}
+#endif
+
+bool wxGIFHandler_WriteZero(wxOutputStream *stream)
+{
+ return wxGIFHandler_WriteByte(stream, 0);
+}
+
+bool wxGIFHandler_WritePalette(wxOutputStream *stream,
+ const wxRGB *array, size_t count, int bpp)
+{
+ wxUint8 buf[3];
+ for (int i = 0; (i < (1 << bpp)); i++)
+ {
+ if (i < (int)count)
+ {
+ buf[0] = array[i].red;
+ buf[1] = array[i].green;
+ buf[2] = array[i].blue;
+ }
+ else
+ {
+ buf[0] = buf[1] = buf[2] = 0;
+ }
+
+ if ( !wxGIFHandler_Write(stream, buf, sizeof(buf)) )
+ {
+ return false;
+ }
+ }
+
+ return true;
+}
+
+bool wxGIFHandler_WriteControl(wxOutputStream *stream,
+ int maskIndex, int delayMilliSecs)
+{
+ wxUint8 buf[8];
+
+ buf[0] = GIF_MARKER_EXT; // extension marker
+ buf[1] = GIF_MARKER_EXT_GRAPHICS_CONTROL;
+ buf[2] = 4; // length of block
+ buf[3] = (maskIndex != wxNOT_FOUND) ? 1 : 0; // has transparency
+ buf[4] = delayMilliSecs / 10; // delay time
+ buf[5] = 0;
+ buf[6] = (maskIndex != wxNOT_FOUND) ? (wxUint8) maskIndex : 0;
+ buf[7] = 0;
+ return wxGIFHandler_Write(stream, buf, sizeof(buf));
+}
+
+bool wxGIFHandler_WriteComment(wxOutputStream *stream, const wxString& comment)
+{
+ if ( comment.empty() )
+ {
+ return true;
+ }
+
+ // Write comment header.
+ wxUint8 buf[2];
+ buf[0] = GIF_MARKER_EXT;
+ buf[1] = GIF_MARKER_EXT_COMMENT;
+ if ( !wxGIFHandler_Write(stream, buf, sizeof(buf)) )
+ {
+ return false;
+ }
+
+ /*
+ If comment is longer than 255 bytes write it in blocks of maximum 255
+ bytes each.
+ */
+ wxCharBuffer text( comment.mb_str() );
+
+ size_t pos = 0, fullLength = text.length();
+
+ do
+ {
+ size_t blockLength = wxMin(fullLength - pos, 255);
+
+ if ( !wxGIFHandler_WriteByte(stream, (wxUint8) blockLength)
+ || !wxGIFHandler_Write(stream, &text.data()[pos], blockLength) )
+ {
+ return false;
+ }
+
+ pos += blockLength;
+ }while (pos < fullLength);
+
+
+ // Write comment footer.
+ return wxGIFHandler_WriteZero(stream);
+}
+
+bool wxGIFHandler_WriteLoop(wxOutputStream *stream)
+{
+ wxUint8 buf[4];
+ const int loopcount = 0; // infinite
+
+ buf[0] = GIF_MARKER_EXT;
+ buf[1] = GIF_MARKER_EXT_APP;
+ buf[2] = 0x0B;
+ bool ok = wxGIFHandler_Write(stream, buf, 3)
+ && wxGIFHandler_Write(stream, NETSCAPE_LOOP, sizeof(NETSCAPE_LOOP)-1);
+
+ buf[0] = 3;
+ buf[1] = 1;
+ buf[2] = loopcount & 0xFF;
+ buf[3] = loopcount >> 8;
+
+ return ok && wxGIFHandler_Write(stream, buf, 4)
+ && wxGIFHandler_WriteZero(stream);
+}
+
+bool wxGIFHandler_BufferedOutput(wxOutputStream *stream, wxUint8 *buf, int c)
+{
+ bool ok = true;
+
+ if (c == FLUSH_OUTPUT)
+ {
+ // Flush everything out.
+ if (buf[0])
+ {
+ ok = wxGIFHandler_Write(stream, buf, buf[0]+1);
+ }
+ // Mark end of compressed data, by an empty block (see GIF doc):
+ wxGIFHandler_WriteZero(stream);
+ }
+ else
+ {
+ if (buf[0] == 255)
+ {
+ // Dump out this buffer - it is full:
+ ok = wxGIFHandler_Write(stream, buf, buf[0] + 1);
+ buf[0] = 0;
+ }
+ buf[++buf[0]] = c;
+ }
+
+ return ok;
+}
+
+#endif // wxUSE_STREAMS
+
+#endif // wxUSE_IMAGE && wxUSE_GIF
projects / wxWidgets.git / blobdiff