src/common/imagxpm.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        imagxpm.cpp
   3 // Purpose:     wxXPMHandler
   4 // Author:      Vaclav Slavik, Robert Roebling
   5 // RCS-ID:      $Id$
   6 // Copyright:   (c) 2001 Vaclav Slavik
   7 // Licence:     wxWindows licence
   8 /////////////////////////////////////////////////////////////////////////////
   9
  10 /*
  11
  12 This file is partially based on source code of ImageMagick by John Cristy. Its
  13 license is as follows:
  14
  15 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
  16 %                                                                             %
  17 %                                                                             %
  18 %                                                                             %
  19 %                            X   X  PPPP   M   M                              %
  20 %                             X X   P   P  MM MM                              %
  21 %                              X    PPPP   M M M                              %
  22 %                             X X   P      M   M                              %
  23 %                            X   X  P      M   M                              %
  24 %                                                                             %
  25 %                                                                             %
  26 %                    Read/Write ImageMagick Image Format.                     %
  27 %                                                                             %
  28 %                                                                             %
  29 %                              Software Design                                %
  30 %                                John Cristy                                  %
  31 %                                 July 1992                                   %
  32 %                                                                             %
  33 %                                                                             %
  34 %  Copyright (C) 2001 ImageMagick Studio, a non-profit organization dedicated %
  35 %  to making software imaging solutions freely available.                     %
  36 %                                                                             %
  37 %  Permission is hereby granted, free of charge, to any person obtaining a    %
  38 %  copy of this software and associated documentation files ("ImageMagick"),  %
  39 %  to deal in ImageMagick without restriction, including without limitation   %
  40 %  the rights to use, copy, modify, merge, publish, distribute, sublicense,   %
  41 %  and/or sell copies of ImageMagick, and to permit persons to whom the       %
  42 %  ImageMagick is furnished to do so, subject to the following conditions:    %
  43 %                                                                             %
  44 %  The above copyright notice and this permission notice shall be included in %
  45 %  all copies or substantial portions of ImageMagick.                         %
  46 %                                                                             %
  47 %  The software is provided "as is", without warranty of any kind, express or %
  48 %  implied, including but not limited to the warranties of merchantability,   %
  49 %  fitness for a particular purpose and noninfringement.  In no event shall   %
  50 %  ImageMagick Studio be liable for any claim, damages or other liability,    %
  51 %  whether in an action of contract, tort or otherwise, arising from, out of  %
  52 %  or in connection with ImageMagick or the use or other dealings in          %
  53 %  ImageMagick.                                                               %
  54 %                                                                             %
  55 %  Except as contained in this notice, the name of the ImageMagick Studio     %
  56 %  shall not be used in advertising or otherwise to promote the sale, use or  %
  57 %  other dealings in ImageMagick without prior written authorization from the %
  58 %  ImageMagick Studio.                                                        %
  59 %                                                                             %
  60 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
  61 %
  62 %
  63 */
  64
  65 #ifdef __GNUG__
  66 #pragma implementation "imagxpm.h"
  67 #endif
  68
  69 // For compilers that support precompilation, includes "wx.h".
  70 #include "wx/wxprec.h"
  71
  72 #ifdef __BORLANDC__
  73 #  pragma hdrstop
  74 #endif
  75
  76 #ifndef WX_PRECOMP
  77 #  include "wx/defs.h"
  78 #endif
  79
  80 #if wxUSE_XPM
  81
  82 #include "wx/imagxpm.h"
  83 #include "wx/wfstream.h"
  84 #include "wx/log.h"
  85 #include "wx/intl.h"
  86 #include "wx/utils.h"
  87 #include "wx/xpmdecod.h"
  88
  89 IMPLEMENT_DYNAMIC_CLASS(wxXPMHandler,wxImageHandler)
  90
  91 //-----------------------------------------------------------------------------
  92 // wxXPMHandler
  93 //-----------------------------------------------------------------------------
  94
  95 #if wxUSE_STREAMS
  96
  97 bool wxXPMHandler::LoadFile(wxImage *image,
  98                             wxInputStream& stream,
  99                             bool WXUNUSED(verbose), int WXUNUSED(index))
 100 {
 101     wxXPMDecoder decoder;
 102
 103     wxImage img = decoder.ReadFile(stream);
 104     if ( !img.Ok() )
 105         return FALSE;
 106     *image = img;
 107     return TRUE;
 108 }
 109
 110 bool wxXPMHandler::SaveFile(wxImage * image,
 111                             wxOutputStream& stream, bool WXUNUSED(verbose))
 112 {
 113     wxString tmp;
 114     char tmp_c;
 115
 116     // 1. count colours:
 117     #define MaxCixels  92
 118     static const char Cixel[MaxCixels+1] =
 119                          " .XoO+@#$%&*=-;:>,<1234567890qwertyuipasdfghjk"
 120                          "lzxcvbnmMNBVCZASDFGHJKLPIUYTREWQ!~^/()_`'][{}|";
 121     int chars_per_pixel;
 122     int cols;
 123     int i, j, k;
 124
 125     cols = image->CountColours();
 126     chars_per_pixel = 1;
 127     for ( k = MaxCixels; cols > k; k *= MaxCixels)
 128         chars_per_pixel++;
 129
 130     // 2. write the header:
 131     char tmpbuf[200];
 132     // VS: 200b is safe upper bound for anything produced by sprintf below
 133     //     (101 bytes the string, neither %i can expand into more than 10 chars)
 134     sprintf(tmpbuf,
 135                "/* XPM */\n"
 136                "static char *xpm_data[] = {\n"
 137                "/* columns rows colors chars-per-pixel */\n"
 138                "\"%i %i %i %i\",\n",
 139                image->GetWidth(), image->GetHeight(), cols, chars_per_pixel);
 140     stream.Write(tmpbuf, strlen(tmpbuf));
 141
 142     // 3. create color symbols table:
 143     wxHashTable table(wxKEY_INTEGER);
 144     image->ComputeHistogram(table);
 145
 146     char *symbols_data = new char[cols * (chars_per_pixel+1)];
 147     char **symbols = new char*[cols];
 148
 149     // 2a. find mask colour:
 150     long mask_key = -1;
 151     if (image->HasMask())
 152         mask_key = (image->GetMaskRed() << 16) |
 153                    (image->GetMaskGreen() << 8) | image->GetMaskBlue();
 154
 155     // 2b. generate colour table:
 156     table.BeginFind();
 157     wxNode *node = NULL;
 158     while ((node = table.Next()) != NULL)
 159     {
 160         wxHNode *hnode = (wxHNode*) node->GetData();
 161         long index = hnode->index;
 162         symbols[index] = symbols_data + index * (chars_per_pixel+1);
 163         char *sym = symbols[index];
 164
 165         k = index % MaxCixels;
 166         sym[0] = Cixel[k];
 167         for (j = 1; j < chars_per_pixel; j++)
 168         {
 169             k = ((index - k) / MaxCixels) % MaxCixels;
 170             sym[j] = Cixel[k];
 171         }
 172         sym[j] = '\0';
 173
 174         long key = node->GetKeyInteger();
 175
 176         if (key == 0)
 177             tmp.Printf(wxT("\"%s c Black\",\n"), sym);
 178         else if (key == mask_key)
 179             tmp.Printf(wxT("\"%s c None\",\n"), sym);
 180         else
 181             tmp.Printf(wxT("\"%s c #%s%s%s\",\n"), sym,
 182                        wxDecToHex((unsigned char)(key >> 16)).c_str(),
 183                        wxDecToHex((unsigned char)(key >> 8)).c_str(),
 184                        wxDecToHex((unsigned char)(key)).c_str());
 185         stream.Write(tmp.mb_str(), tmp.Length());
 186     }
 187
 188     tmp = wxT("/* pixels */\n");
 189     stream.Write(tmp.mb_str(), tmp.Length());
 190
 191     unsigned char *data = image->GetData();
 192     for (j = 0; j < image->GetHeight(); j++)
 193     {
 194         tmp_c = '\"'; stream.Write(&tmp_c, 1);
 195         for (i = 0; i < image->GetWidth(); i++, data += 3)
 196         {
 197             unsigned long key = (data[0] << 16) | (data[1] << 8) | (data[2]);
 198             wxHNode *hnode = (wxHNode*) table.Get(key);
 199             stream.Write(symbols[hnode->index], chars_per_pixel);
 200         }
 201         tmp_c = '\"'; stream.Write(&tmp_c, 1);
 202         if ( j + 1 < image->GetHeight() )
 203         {
 204             tmp_c = ','; stream.Write(&tmp_c, 1);
 205         }
 206         tmp_c = '\n'; stream.Write(&tmp_c, 1);
 207     }
 208     tmp = wxT("};\n");
 209     stream.Write(tmp.mb_str(), 3);
 210
 211     // Clean up:
 212     delete[] symbols;
 213     delete[] symbols_data;
 214
 215     // FIXME: it will be better to use macros-based wxHashTable & DeleteContents(TRUE)
 216     table.BeginFind();
 217     while ((node = table.Next()) != NULL)
 218     {
 219         delete (wxHNode *) node->GetData();
 220     }
 221
 222     return TRUE;
 223 }
 224
 225 bool wxXPMHandler::DoCanRead(wxInputStream& stream)
 226 {
 227     wxXPMDecoder decoder;
 228     return decoder.CanRead(stream);
 229 }
 230
 231 #endif  // wxUSE_STREAMS
 232
 233 #endif // wxUSE_XPM