// Licence: wxWindows licence
/////////////////////////////////////////////////////////////////////////////
-/*
- We don't put pragma implement in this file because it is already present in
- src/common/image.cpp
-*/
+#ifdef __GNUG__
+#pragma implementation "imagpcx.h"
+#endif
// For compilers that support precompilation, includes "wx.h".
#include "wx/wxprec.h"
#if wxUSE_STREAMS && wxUSE_PCX
-#include "wx/image.h"
+#include "wx/imagpcx.h"
#include "wx/wfstream.h"
#include "wx/module.h"
#include "wx/log.h"
+#include "wx/intl.h"
+#include "wx/hash.h"
+#include "wx/list.h"
+#include "wx/object.h"
//-----------------------------------------------------------------------------
-// PCX decoding
+// RLE encoding and decoding
//-----------------------------------------------------------------------------
-void RLEencode(unsigned char *WXUNUSED(p), unsigned int WXUNUSED(size), wxOutputStream& WXUNUSED(s))
+void RLEencode(unsigned char *p, unsigned int size, wxOutputStream& s)
{
+ unsigned int data, last, cont;
+
+ // Write 'size' bytes. The PCX official specs say there will be
+ // a decoding break at the end of each scanline, so in order to
+ // force this decoding break use this function to write, at most,
+ // _one_ complete scanline at a time.
+
+ last = (unsigned char) *(p++);
+ cont = 1;
+ size--;
+
+ while (size-- > 0)
+ {
+ data = (unsigned char) *(p++);
+
+ // Up to 63 bytes with the same value can be stored using a
+ // single { cont, value } pair.
+ //
+ if ((data == last) && (cont < 63))
+ {
+ cont++;
+ }
+ else
+ {
+ // Need to write a 'counter' byte?
+ //
+ if ((cont > 1) || ((last & 0xC0) == 0xC0))
+ s.PutC((char) (cont | 0xC0));
+
+ s.PutC((char) last);
+ last = data;
+ cont = 1;
+ }
+ }
+
+ // Write the last one and return;
+ //
+ if ((cont > 1) || ((last & 0xC0) == 0xC0))
+ s.PutC((char) (cont | 0xC0));
+
+ s.PutC((char) last);
}
void RLEdecode(unsigned char *p, unsigned int size, wxInputStream& s)
}
-/* PCX header */
+//-----------------------------------------------------------------------------
+// PCX reading and saving
+//-----------------------------------------------------------------------------
+
+// PCX header
+#define HDR_MANUFACTURER 0
#define HDR_VERSION 1
#define HDR_ENCODING 2
#define HDR_BITSPERPIXEL 3
#define HDR_YMAX 10
#define HDR_NPLANES 65
#define HDR_BYTESPERLINE 66
+#define HDR_PALETTEINFO 68
-/* image formats */
-#define IMAGE_8BIT 0 // 8 bpp, 1 plane (8 bit)
-#define IMAGE_24BIT 1 // 8 bpp, 3 planes (24 bit)
+// image formats
+enum {
+ wxPCX_8BIT, // 8 bpp, 1 plane (8 bit)
+ wxPCX_24BIT // 8 bpp, 3 planes (24 bit)
+};
-/* error codes */
-#define E_OK 0 // everything was OK
-#define E_FORMATO 1 // error in pcx file format
-#define E_MEMORIA 2 // error allocating memory
-#define E_VERSION 3 // error in pcx version number
+// error codes
+enum {
+ wxPCX_OK = 0, // everything was OK
+ wxPCX_INVFORMAT = 1, // error in pcx file format
+ wxPCX_MEMERR = 2, // error allocating memory
+ wxPCX_VERERR = 3 // error in pcx version number
+};
// ReadPCX:
// Loads a PCX file into the wxImage object pointed by image.
-// Returns E_OK on success, or an error code otherwise (see
-// above for error codes)
+// Returns wxPCX_OK on success, or an error code otherwise
+// (see above for error codes)
//
int ReadPCX(wxImage *image, wxInputStream& stream)
{
int nplanes; // number of planes
int encoding; // is the image RLE encoded?
int format; // image format (8 bit, 24 bit)
- unsigned int i;
- off_t pos;
+ unsigned int i, j;
// Read PCX header and check the version number (it must
// be at least 5 or higher for 8 bit and 24 bit images).
//
stream.Read(hdr, 128);
- if (hdr[HDR_VERSION] < 5) return E_VERSION;
+ if (hdr[HDR_VERSION] < 5) return wxPCX_VERERR;
// Extract all image info from the PCX header.
//
// 8 bits (8 bpp, 1 plane) and 24 bits (8 bpp, 3 planes).
//
if ((nplanes == 3) && (bitsperpixel == 8))
- format = IMAGE_24BIT;
+ format = wxPCX_24BIT;
else if ((nplanes == 1) && (bitsperpixel == 8))
- format = IMAGE_8BIT;
+ format = wxPCX_8BIT;
else
- return E_FORMATO;
+ return wxPCX_INVFORMAT;
- // If the image is of type IMAGE_8BIT, then there is a
- // palette at the end of the file. Read it now before
- // proceeding.
- //
- if (format == IMAGE_8BIT)
- {
- pos = stream.TellI();
- stream.SeekI(-769, wxFromEnd);
-
- if (stream.GetC() != 12)
- return E_FORMATO;
+ // If the image is of type wxPCX_8BIT, then there is
+ // a palette at the end of the image data. If we were
+ // working with a file, we could seek at the end to the
+ // end (SeekI(-769, wxFromEnd) and read the palette
+ // before proceeding. Unfortunately, this would prevent
+ // loading several PCXs in a single stream, so we can't
+ // do it. Thus, 8-bit images will have to be decoded in
+ // two passes: one to read and decode the image data,
+ // and another to replace 'colour indexes' with RGB
+ // values.
- stream.Read(pal, 768);
- stream.SeekI(pos, wxFromStart);
- }
- // Allocate memory for a scanline and resize the image.
+ // Resize the image and allocate memory for a scanline.
//
image->Create(width, height);
if (!image->Ok())
- return E_MEMORIA;
+ return wxPCX_MEMERR;
if ((p = (unsigned char *) malloc(bytesperline * nplanes)) == NULL)
- return E_MEMORIA;
+ return wxPCX_MEMERR;
// Now start reading the file, line by line, and store
// the data in the format required by wxImage.
//
dst = image->GetData();
- for (; height; height--)
+ for (j = height; j; j--)
{
if (encoding)
RLEdecode(p, bytesperline * nplanes, stream);
switch (format)
{
- case IMAGE_8BIT:
+ case wxPCX_8BIT:
{
for (i = 0; i < width; i++)
{
- *(dst++) = pal[ 3 * (p[i]) ];
- *(dst++) = pal[ 3 * (p[i]) + 1];
- *(dst++) = pal[ 3 * (p[i]) + 2];
+ // first pass, just store the colour index
+ *dst = p[i];
+ dst += 3;
}
break;
}
- case IMAGE_24BIT:
+ case wxPCX_24BIT:
{
for (i = 0; i < width; i++)
{
free(p);
- return E_OK;
+ // For 8 bit images, we read the palette, and then do a second
+ // pass replacing indexes with their RGB values;
+ //
+ if (format == wxPCX_8BIT)
+ {
+ unsigned char index;
+
+ if (stream.GetC() != 12)
+ return wxPCX_INVFORMAT;
+
+ stream.Read(pal, 768);
+
+ p = image->GetData();
+ for (unsigned long k = height * width; k; k--)
+ {
+ index = *p;
+ *(p++) = pal[3 * index];
+ *(p++) = pal[3 * index + 1];
+ *(p++) = pal[3 * index + 2];
+ }
+ }
+
+ return wxPCX_OK;
}
+// SavePCX:
+// Saves a PCX file into the wxImage object pointed by image.
+// Returns wxPCX_OK on success, or an error code otherwise
+// (see above for error codes). Currently, always saves images
+// in 24 bit format. XXX
+//
+int SavePCX(wxImage *image, wxOutputStream& stream)
+{
+ unsigned char hdr[128]; // PCX header
+ unsigned char pal[768]; // palette for 8 bit images
+ unsigned char *p; // space to store one scanline
+ unsigned char *src; // pointer into wxImage data
+ unsigned int width, height; // size of the image
+ unsigned int bytesperline; // bytes per line (each plane)
+ int nplanes = 3; // number of planes
+ int format = wxPCX_24BIT; // image format (8 bit, 24 bit)
+ wxHashTable h(wxKEY_INTEGER); // image histogram
+ unsigned long key; // key in the hashtable
+ unsigned int i;
+
+ // See if we can save as 8 bit.
+ //
+ if (image->CountColours(256) <= 256)
+ {
+ image->ComputeHistogram(h);
+ format = wxPCX_8BIT;
+ nplanes = 1;
+ }
+
+ // Get image dimensions, calculate bytesperline (must be even,
+ // according to PCX specs) and allocate space for one complete
+ // scanline.
+ //
+ if (!image->Ok())
+ return wxPCX_INVFORMAT;
+
+ width = image->GetWidth();
+ height = image->GetHeight();
+ bytesperline = width;
+ if (bytesperline % 2)
+ bytesperline++;
+
+ if ((p = (unsigned char *) malloc(bytesperline * nplanes)) == NULL)
+ return wxPCX_MEMERR;
+
+ // Build header data and write it to the stream. Initially,
+ // set all bytes to zero (most values default to zero).
+ //
+ memset(hdr, 0, sizeof(hdr));
+
+ hdr[HDR_MANUFACTURER] = 10;
+ hdr[HDR_VERSION] = 5;
+ hdr[HDR_ENCODING] = 1;
+ hdr[HDR_NPLANES] = nplanes;
+ hdr[HDR_BITSPERPIXEL] = 8;
+ hdr[HDR_BYTESPERLINE] = bytesperline % 256;
+ hdr[HDR_BYTESPERLINE + 1] = bytesperline / 256;
+ hdr[HDR_XMAX] = (width - 1) % 256;
+ hdr[HDR_XMAX + 1] = (width - 1) / 256;
+ hdr[HDR_YMAX] = (height - 1) % 256;
+ hdr[HDR_YMAX + 1] = (height - 1) / 256;
+ hdr[HDR_PALETTEINFO] = 1;
+
+ stream.Write(hdr, 128);
+
+ // Encode image data line by line and write it to the stream
+ //
+ src = image->GetData();
+
+ for (; height; height--)
+ {
+ switch (format)
+ {
+ case wxPCX_8BIT:
+ {
+ unsigned char r, g, b;
+ wxHNode *hnode;
+
+ for (i = 0; i < width; i++)
+ {
+ r = *(src++);
+ g = *(src++);
+ b = *(src++);
+ key = (r << 16) | (g << 8) | b;
+
+ hnode = (wxHNode *) h.Get(key);
+ p[i] = (unsigned char)hnode->index;
+ }
+ break;
+ }
+ case wxPCX_24BIT:
+ {
+ for (i = 0; i < width; i++)
+ {
+ p[i] = *(src++);
+ p[i + bytesperline] = *(src++);
+ p[i + 2 * bytesperline] = *(src++);
+ }
+ break;
+ }
+ }
+
+ RLEencode(p, bytesperline * nplanes, stream);
+ }
+
+ free(p);
+
+ // For 8 bit images, build the palette and write it to the stream
+ //
+ if (format == wxPCX_8BIT)
+ {
+ wxNode *node;
+ wxHNode *hnode;
+
+ // zero unused colours
+ memset(pal, 0, sizeof(pal));
+
+ h.BeginFind();
+ while ((node = h.Next()) != NULL)
+ {
+ key = node->GetKeyInteger();
+ hnode = (wxHNode *) node->GetData();
+
+ pal[3 * hnode->index] = (unsigned char)(key >> 16);
+ pal[3 * hnode->index + 1] = (unsigned char)(key >> 8);
+ pal[3 * hnode->index + 2] = (unsigned char)(key);
+ delete hnode;
+ }
+
+ stream.PutC(12);
+ stream.Write(pal, 768);
+ }
+
+ return wxPCX_OK;
+}
//-----------------------------------------------------------------------------
// wxPCXHandler
//-----------------------------------------------------------------------------
-#if !USE_SHARED_LIBRARIES
IMPLEMENT_DYNAMIC_CLASS(wxPCXHandler,wxImageHandler)
-#endif
bool wxPCXHandler::LoadFile( wxImage *image, wxInputStream& stream, bool verbose, int WXUNUSED(index) )
{
image->Destroy();
- if ((error = ReadPCX(image, stream)) != E_OK)
+ if ((error = ReadPCX(image, stream)) != wxPCX_OK)
{
if (verbose)
{
switch (error)
{
- case E_FORMATO: wxLogError(_("PCX: Image format unsupported.")); break;
- case E_MEMORIA: wxLogError(_("PCX: Couldn't allocate memory.")); break;
- case E_VERSION: wxLogError(_("PCX: Wrong version number.")); break;
- default: wxLogError(_("PCX: Unknown error."));
+ case wxPCX_INVFORMAT: wxLogError(_("wxPCXHandler: image format unsupported")); break;
+ case wxPCX_MEMERR: wxLogError(_("wxPCXHandler: couldn't allocate memory")); break;
+ case wxPCX_VERERR: wxLogError(_("wxPCXHandler: version number too low")); break;
+ default: wxLogError(_("wxPCXHandler: unknown error !!!"));
}
}
image->Destroy();
return TRUE;
}
-bool wxPCXHandler::SaveFile( wxImage *WXUNUSED(image), wxOutputStream& WXUNUSED(stream), bool verbose )
+bool wxPCXHandler::SaveFile( wxImage *image, wxOutputStream& stream, bool verbose )
{
- wxFAIL_MSG(wxT("wxPCXHandler::SaveFile still not implemented"));
+ int error;
+
+ if ((error = SavePCX(image, stream)) != wxPCX_OK)
+ {
+ if (verbose)
+ {
+ switch (error)
+ {
+ case wxPCX_INVFORMAT: wxLogError(_("wxPCXHandler: invalid image")); break;
+ case wxPCX_MEMERR: wxLogError(_("wxPCXHandler: couldn't allocate memory")); break;
+ default: wxLogError(_("wxPCXHandler: unknown error !!!"));
+ }
+ }
+ }
- return FALSE;
+ return (error == wxPCX_OK);
}
bool wxPCXHandler::DoCanRead( wxInputStream& stream )
projects / wxWidgets.git / blobdiff