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             // wxImage takes ownership of dataPtr...
 195         }
 196     }
 197
 198     void SetMaskColour( unsigned char r, unsigned char g, unsigned char b );
 199     unsigned char GetMaskRed();
 200     unsigned char GetMaskGreen();
 201     unsigned char GetMaskBlue();
 202     void SetMask( bool mask = TRUE );
 203     bool HasMask();
 204
 205     wxImage Rotate(double angle, const wxPoint & centre_of_rotation,
 206                    bool interpolating = TRUE, wxPoint * offset_after_rotation = NULL) const ;
 207     wxImage Rotate90( bool clockwise = TRUE ) ;
 208     wxImage Mirror( bool horizontally = TRUE ) ;
 209
 210     void Replace( unsigned char r1, unsigned char g1, unsigned char b1,
 211                   unsigned char r2, unsigned char g2, unsigned char b2 );
 212
 213     // convert to monochrome image (<r,g,b> will be replaced by white, everything else by black)
 214     wxImage ConvertToMono( unsigned char r, unsigned char g, unsigned char b ) const;
 215
 216     void SetOption(const wxString& name, const wxString& value);
 217     %name(SetOptionInt)void SetOption(const wxString& name, int value);
 218     wxString GetOption(const wxString& name) const;
 219     int GetOptionInt(const wxString& name) const;
 220     bool HasOption(const wxString& name) const;
 221
 222     unsigned long CountColours( unsigned long stopafter = (unsigned long) -1 );
 223     // TODO: unsigned long ComputeHistogram( wxHashTable &h );
 224
 225     static void AddHandler( wxImageHandler *handler );
 226     static void InsertHandler( wxImageHandler *handler );
 227     static bool RemoveHandler( const wxString& name );
 228
 229
 230     %addmethods {
 231         wxBitmap ConvertToBitmap() {
 232             wxBitmap bitmap(*self);
 233             return bitmap;
 234         }
 235
 236         wxBitmap ConvertToMonoBitmap( unsigned char red,
 237                                       unsigned char green,
 238                                       unsigned char blue ) {
 239             wxImage mono = self->ConvertToMono( red, green, blue );
 240             wxBitmap bitmap( mono, 1 );
 241             return bitmap;
 242         }
 243     }
 244 };
 245
 246
 247 // Alternate constructors
 248 %new wxImage* wxEmptyImage(int width=0, int height=0);
 249 %new wxImage* wxImageFromMime(const wxString& name, const wxString& mimetype, int index = -1);
 250 %new wxImage* wxImageFromBitmap(const wxBitmap &bitmap);
 251 %new wxImage* wxImageFromData(int width, int height, unsigned char* data);
 252 %new wxImage* wxImageFromStream(wxInputStream& stream, long type = wxBITMAP_TYPE_ANY, int index = -1);
 253 %new wxImage* wxImageFromStreamMime(wxInputStream& stream, const wxString& mimetype, int index = -1 );
 254
 255 %{
 256     wxImage* wxEmptyImage(int width=0, int height=0) {
 257         if (width == 0 && height == 0)
 258             return new wxImage;
 259         else
 260             return new wxImage(width, height);
 261     }
 262
 263
 264     wxImage* wxImageFromMime(const wxString& name, const wxString& mimetype, int index) {
 265         return new wxImage(name, mimetype, index);
 266     }
 267
 268
 269     wxImage* wxImageFromBitmap(const wxBitmap &bitmap) {
 270         return new wxImage(bitmap.ConvertToImage());
 271     }
 272
 273
 274     wxImage* wxImageFromData(int width, int height, unsigned char* data) {
 275         // Copy the source data so the wxImage can clean it up later
 276         unsigned char* copy = (unsigned char*)malloc(width*height*3);
 277         if (copy == NULL) {
 278             PyErr_NoMemory();
 279             return NULL;
 280         }
 281         memcpy(copy, data, width*height*3);
 282         return new wxImage(width, height, copy, FALSE);
 283     }
 284
 285
 286     wxImage* wxImageFromStream(wxInputStream& stream,
 287                                long type = wxBITMAP_TYPE_ANY, int index = -1) {
 288         return new wxImage(stream, type, index);
 289     }
 290
 291
 292     wxImage* wxImageFromStreamMime(wxInputStream& stream,
 293                                    const wxString& mimetype, int index = -1 ) {
 294         return new wxImage(stream, mimetype, index);
 295     }
 296 %}
 297
 298
 299
 300 void wxInitAllImageHandlers();
 301
 302
 303 %readonly
 304 %{
 305 #if 0
 306 %}
 307
 308 extern wxImage    wxNullImage;
 309
 310 %readwrite
 311 %{
 312 #endif
 313 %}
 314
 315
 316
 317 //---------------------------------------------------------------------------
 318 // This one is here to avoid circular imports
 319
 320 %new wxBitmap* wxBitmapFromImage(const wxImage& img, int depth=-1);
 321
 322 %{
 323     wxBitmap* wxBitmapFromImage(const wxImage& img, int depth=-1) {
 324         return new wxBitmap(img, depth);
 325     }
 326
 327 %}
 328
 329
 330 //---------------------------------------------------------------------------