wxPython/src/image.i

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        image.i
   3 // Purpose:     SWIG interface file for wxImage, wxImageHandler, etc.
   4 //
   5 // Author:      Robin Dunn
   6 //
   7 // Created:     28-Apr-1999
   8 // RCS-ID:      $Id$
   9 // Copyright:   (c) 1998 by Total Control Software
  10 // Licence:     wxWindows license
  11 /////////////////////////////////////////////////////////////////////////////
  12
  13
  14 %module image
  15
  16 %{
  17 #include "helpers.h"
  18 #include "pyistream.h"
  19 #include <wx/image.h>
  20 %}
  21
  22 //----------------------------------------------------------------------
  23
  24 %include typemaps.i
  25 %include my_typemaps.i
  26
  27 // Import some definitions of other classes, etc.
  28 %import _defs.i
  29 %import misc.i
  30 %import gdi.i
  31 %import streams.i
  32
  33 //---------------------------------------------------------------------------
  34
  35 class wxImageHandler : public wxObject {
  36 public:
  37     // wxImageHandler();    Abstract Base Class
  38     wxString GetName();
  39     wxString GetExtension();
  40     long GetType();
  41     wxString GetMimeType();
  42
  43     //bool LoadFile(wxImage* image, wxInputStream& stream);
  44     //bool SaveFile(wxImage* image, wxOutputStream& stream);
  45     //virtual int GetImageCount( wxInputStream& stream );
  46     //bool CanRead( wxInputStream& stream );
  47
  48     bool CanRead( const wxString& name );
  49
  50     void SetName(const wxString& name);
  51     void SetExtension(const wxString& extension);
  52     void SetType(long type);
  53     void SetMimeType(const wxString& mimetype);
  54 };
  55
  56 //---------------------------------------------------------------------------
  57
  58 class wxPNGHandler : public wxImageHandler {
  59 public:
  60     wxPNGHandler();
  61 };
  62
  63
  64 class wxJPEGHandler : public wxImageHandler {
  65 public:
  66     wxJPEGHandler();
  67 };
  68
  69
  70 class wxBMPHandler : public wxImageHandler {
  71 public:
  72     wxBMPHandler();
  73 };
  74
  75 class wxICOHandler : public wxBMPHandler {
  76 public:
  77     wxICOHandler();
  78 };
  79
  80 class wxCURHandler : public wxICOHandler {
  81 public:
  82     wxCURHandler();
  83 };
  84
  85 class wxANIHandler : public wxCURHandler {
  86 public:
  87     wxANIHandler();
  88 };
  89
  90 class wxGIFHandler : public wxImageHandler {
  91 public:
  92     wxGIFHandler();
  93 };
  94
  95 class wxPNMHandler : public wxImageHandler {
  96 public:
  97     wxPNMHandler();
  98 };
  99
 100 class wxPCXHandler : public wxImageHandler {
 101 public:
 102     wxPCXHandler();
 103 };
 104
 105 class wxTIFFHandler : public wxImageHandler {
 106 public:
 107     wxTIFFHandler();
 108 };
 109
 110
 111
 112 //---------------------------------------------------------------------------
 113
 114 class wxImage : public wxObject {
 115 public:
 116     wxImage( const wxString& name, long type = wxBITMAP_TYPE_ANY, int index = -1 );
 117     ~wxImage();
 118
 119     void Create( int width, int height );
 120     void Destroy();
 121
 122     wxImage Scale( int width, int height );
 123     wxImage& Rescale(int width, int height);
 124
 125     void SetRGB( int x, int y, unsigned char r, unsigned char g, unsigned char b );
 126     unsigned char GetRed( int x, int y );
 127     unsigned char GetGreen( int x, int y );
 128     unsigned char GetBlue( int x, int y );
 129
 130         // find first colour that is not used in the image and has higher
 131     // RGB values than <startR,startG,startB>
 132     bool FindFirstUnusedColour( byte *OUTPUT, byte *OUTPUT, byte *OUTPUT,
 133                                 byte startR = 0, byte startG = 0, byte startB = 0 ) const;
 134
 135     // Set image's mask to the area of 'mask' that has <mr,mg,mb> colour
 136     bool SetMaskFromImage(const wxImage & mask,
 137                           byte mr, byte mg, byte mb);
 138
 139 //      void DoFloodFill (wxCoord x, wxCoord y,
 140 //          const wxBrush & fillBrush,
 141 //          const wxColour& testColour,
 142 //          int style = wxFLOOD_SURFACE,
 143 //          int LogicalFunction = wxCOPY /* currently unused */ ) ;
 144
 145     static bool CanRead( const wxString& name );
 146     static int GetImageCount( const wxString& name, long type = wxBITMAP_TYPE_ANY );
 147
 148     bool LoadFile( const wxString& name, long type = wxBITMAP_TYPE_ANY, int index = -1 );
 149     %name(LoadMimeFile)bool LoadFile( const wxString& name, const wxString& mimetype, int index = -1 );
 150
 151     bool SaveFile( const wxString& name, int type );
 152     %name(SaveMimeFile)bool SaveFile( const wxString& name, const wxString& mimetype );
 153
 154     %name(CanReadStream) static bool CanRead( wxInputStream& stream );
 155     %name(LoadStream) bool LoadFile( wxInputStream& stream, long type = wxBITMAP_TYPE_ANY, int index = -1 );
 156     %name(LoadMimeStream) bool LoadFile( wxInputStream& stream, const wxString& mimetype, int index = -1 );
 157
 158     bool Ok();
 159     int GetWidth();
 160     int GetHeight();
 161
 162     wxImage GetSubImage(const wxRect& rect);
 163     wxImage Copy();
 164     void Paste( const wxImage &image, int x, int y );
 165
 166     //unsigned char *GetData();
 167     //void SetData( unsigned char *data );
 168
 169     %addmethods {
 170         PyObject* GetDataBuffer() {
 171             unsigned char* data = self->GetData();
 172             int len = self->GetWidth() * self->GetHeight() * 3;
 173             return PyBuffer_FromReadWriteMemory(data, len);
 174         }
 175
 176         PyObject* GetData() {
 177             unsigned char* data = self->GetData();
 178             int len = self->GetWidth() * self->GetHeight() * 3;
 179             return PyString_FromStringAndSize((char*)data, len);
 180         }
 181
 182         void SetData(PyObject* data) {
 183             unsigned char* dataPtr;
 184
 185             if (! PyString_Check(data)) {
 186                 PyErr_SetString(PyExc_TypeError, "Expected string object");
 187                 return /* NULL */ ;
 188             }
 189
 190             size_t len = self->GetWidth() * self->GetHeight() * 3;
 191             dataPtr = (unsigned char*) malloc(len);
 192             memcpy(dataPtr, PyString_AsString(data), len);
 193             self->SetData(dataPtr);
 194         }
 195     }
 196
 197     void SetMaskColour( unsigned char r, unsigned char g, unsigned char b );
 198     unsigned char GetMaskRed();
 199     unsigned char GetMaskGreen();
 200     unsigned char GetMaskBlue();
 201     void SetMask( bool mask = TRUE );
 202     bool HasMask();
 203
 204     wxImage Rotate(double angle, const wxPoint & centre_of_rotation,
 205                    bool interpolating = TRUE, wxPoint * offset_after_rotation = NULL) const ;
 206     wxImage Rotate90( bool clockwise = TRUE ) ;
 207     wxImage Mirror( bool horizontally = TRUE ) ;
 208
 209     void Replace( unsigned char r1, unsigned char g1, unsigned char b1,
 210                   unsigned char r2, unsigned char g2, unsigned char b2 );
 211
 212     // convert to monochrome image (<r,g,b> will be replaced by white, everything else by black)
 213     wxImage ConvertToMono( unsigned char r, unsigned char g, unsigned char b ) const;
 214
 215     void SetOption(const wxString& name, const wxString& value);
 216     %name(SetOptionInt)void SetOption(const wxString& name, int value);
 217     wxString GetOption(const wxString& name) const;
 218     int GetOptionInt(const wxString& name) const;
 219     bool HasOption(const wxString& name) const;
 220
 221     unsigned long CountColours( unsigned long stopafter = (unsigned long) -1 );
 222     // TODO: unsigned long ComputeHistogram( wxHashTable &h );
 223
 224     static void AddHandler( wxImageHandler *handler );
 225     static void InsertHandler( wxImageHandler *handler );
 226     static bool RemoveHandler( const wxString& name );
 227
 228
 229     %addmethods {
 230         wxBitmap ConvertToBitmap() {
 231             wxBitmap bitmap(*self);
 232             return bitmap;
 233         }
 234
 235         wxBitmap ConvertToMonoBitmap( unsigned char red,
 236                                       unsigned char green,
 237                                       unsigned char blue ) {
 238             wxImage mono = self->ConvertToMono( red, green, blue );
 239             wxBitmap bitmap( mono, 1 );
 240             return bitmap;
 241         }
 242     }
 243 };
 244
 245
 246 // Alternate constructors
 247 %new wxImage* wxEmptyImage(int width=0, int height=0);
 248 %new wxImage* wxImageFromMime(const wxString& name, const wxString& mimetype, int index = -1);
 249 %new wxImage* wxImageFromBitmap(const wxBitmap &bitmap);
 250 %new wxImage* wxImageFromData(int width, int height, unsigned char* data);
 251 %new wxImage* wxImageFromStream(wxInputStream& stream, long type = wxBITMAP_TYPE_ANY, int index = -1);
 252 %new wxImage* wxImageFromStreamMime(wxInputStream& stream, const wxString& mimetype, int index = -1 );
 253
 254 %{
 255     wxImage* wxEmptyImage(int width=0, int height=0) {
 256         if (width == 0 && height == 0)
 257             return new wxImage;
 258         else
 259             return new wxImage(width, height);
 260     }
 261
 262
 263     wxImage* wxImageFromMime(const wxString& name, const wxString& mimetype, int index) {
 264         return new wxImage(name, mimetype, index);
 265     }
 266
 267
 268     wxImage* wxImageFromBitmap(const wxBitmap &bitmap) {
 269         return new wxImage(bitmap.ConvertToImage());
 270     }
 271
 272
 273     wxImage* wxImageFromData(int width, int height, unsigned char* data) {
 274         // Copy the source data so the wxImage can clean it up later
 275         unsigned char* copy = (unsigned char*)malloc(width*height*3);
 276         if (copy == NULL) {
 277             PyErr_NoMemory();
 278             return NULL;
 279         }
 280         memcpy(copy, data, width*height*3);
 281         return new wxImage(width, height, copy, FALSE);
 282     }
 283
 284
 285     wxImage* wxImageFromStream(wxInputStream& stream,
 286                                long type = wxBITMAP_TYPE_ANY, int index = -1) {
 287         return new wxImage(stream, type, index);
 288     }
 289
 290
 291     wxImage* wxImageFromStreamMime(wxInputStream& stream,
 292                                    const wxString& mimetype, int index = -1 ) {
 293         return new wxImage(stream, mimetype, index);
 294     }
 295 %}
 296
 297
 298
 299 void wxInitAllImageHandlers();
 300
 301
 302 %readonly
 303 %{
 304 #if 0
 305 %}
 306
 307 extern wxImage    wxNullImage;
 308
 309 %readwrite
 310 %{
 311 #endif
 312 %}
 313
 314
 315
 316 //---------------------------------------------------------------------------
 317 // This one is here to avoid circular imports
 318
 319 %new wxBitmap* wxBitmapFromImage(const wxImage& img, int depth=-1);
 320
 321 %{
 322     wxBitmap* wxBitmapFromImage(const wxImage& img, int depth=-1) {
 323         return new wxBitmap(img, depth);
 324     }
 325
 326 %}
 327
 328
 329 //---------------------------------------------------------------------------