1 /////////////////////////////////////////////////////////////////////////////
3 // Purpose: SWIG interface for wxBitmap and wxMask
7 // Created: 7-July-1997
9 // Copyright: (c) 2003 by Total Control Software
10 // Licence: wxWindows license
11 /////////////////////////////////////////////////////////////////////////////
16 #include <wx/rawbmp.h>
20 // Turn off the aquisition of the Global Interpreter Lock for this file
23 //---------------------------------------------------------------------------
28 static char** ConvertListOfStrings(PyObject* listOfStrings) {
32 if (!PyList_Check(listOfStrings)) {
33 PyErr_SetString(PyExc_TypeError, "Expected a list of strings.");
36 count = PyList_Size(listOfStrings);
37 cArray = new char*[count];
39 for(int x=0; x<count; x++) {
40 // TODO: Need some validation and error checking here
41 cArray[x] = PyString_AsString(PyList_GET_ITEM(listOfStrings, x));
48 //---------------------------------------------------------------------------
50 // TODO: When the API stabalizes and is available on other platforms, add
51 // wrappers for the new wxBitmap, wxRawBitmap, wxDIB stuff...
54 "The wx.Bitmap class encapsulates the concept of a platform-dependent
55 bitmap. It can be either monochrome or colour, and either loaded from
56 a file or created dynamically. A bitmap can be selected into a memory
57 device context (instance of `wx.MemoryDC`). This enables the bitmap to
58 be copied to a window or memory device context using `wx.DC.Blit`, or
59 to be used as a drawing surface.", "
61 The BMP and XMP image file formats are supported on all platforms by
62 wx.Bitmap. Other formats are automatically loaded by `wx.Image` and
63 converted to a wx.Bitmap, so any image file format supported by
64 `wx.Image` can be used.
66 :todo: Add wrappers and support for raw bitmap data access. Can this
67 be be put into Python without losing the speed benefits of the
68 teplates and iterators in rawbmp.h?
70 :todo: Find a way to do very efficient PIL Image <--> wx.Bitmap
73 :see: `wx.EmptyBitmap`, `wx.BitmapFromIcon`, `wx.BitmapFromImage`,
74 `wx.BitmapFromXPMData`, `wx.BitmapFromBits`, `wx.BitmapFromBuffer`,
75 `wx.BitmapFromBufferRGBA`, `wx.Image`
79 MustHaveApp(wxBitmap);
81 class wxBitmap : public wxGDIObject
85 wxBitmap(const wxString& name, wxBitmapType type=wxBITMAP_TYPE_ANY),
86 "Loads a bitmap from a file.",
88 :param name: Name of the file to load the bitmap from.
89 :param type: The type of image to expect. Can be one of the following
90 constants (assuming that the neccessary `wx.Image` handlers are
102 * wx.BITMAP_TYPE_JPEG
105 * wx.BITMAP_TYPE_PICT
106 * wx.BITMAP_TYPE_ICON
110 :see: Alternate constructors `wx.EmptyBitmap`, `wx.BitmapFromIcon`,
111 `wx.BitmapFromImage`, `wx.BitmapFromXPMData`, `wx.BitmapFromBits`,
112 `wx.BitmapFromBuffer`, `wx.BitmapFromBufferRGBA`,
118 wxBitmap(int width, int height, int depth=-1),
119 "Creates a new bitmap of the given size. A depth of -1 indicates the
120 depth of the current screen or visual. Some platforms only support 1
121 for monochrome and -1 for the current display depth.", "",
125 wxBitmap(const wxIcon& icon),
126 "Create a new bitmap from a `wx.Icon` object.", "",
130 wxBitmap(const wxImage& image, int depth=-1),
131 "Creates bitmap object from a `wx.Image`. This has to be done to
132 actually display a `wx.Image` as you cannot draw an image directly on
133 a window. The resulting bitmap will use the provided colour depth (or
134 that of the current screen colour depth if depth is -1) which entails
135 that a colour reduction may have to take place.", "",
142 "Construct a Bitmap from a list of strings formatted as XPM data.", "",
143 wxBitmap(PyObject* listOfStrings))
145 char** cArray = NULL;
148 cArray = ConvertListOfStrings(listOfStrings);
151 bmp = new wxBitmap(cArray);
159 "Creates a bitmap from an array of bits. You should only use this
160 function for monochrome bitmaps (depth 1) in portable programs: in
161 this case the bits parameter should contain an XBM image. For other
162 bit depths, the behaviour is platform dependent.", "",
163 wxBitmap(PyObject* bits, int width, int height, int depth=1 ))
167 PyString_AsStringAndSize(bits, &buf, &length);
168 return new wxBitmap(buf, width, height, depth);
173 // wxGDIImage methods
177 void SetHandle(long handle) { self->SetHandle((WXHANDLE)handle); }
182 %pythoncode { Ok = IsOk }
186 "Gets the width of the bitmap in pixels.", "");
191 "Gets the height of the bitmap in pixels.", "");
196 "Gets the colour depth of the bitmap. A value of 1 indicates a
197 monochrome bitmap.", "");
202 DocStr(GetSize, "Get the size of the bitmap.", "");
204 wxSize size(self->GetWidth(), self->GetHeight());
211 virtual wxImage , ConvertToImage() const,
212 "Creates a platform-independent image from a platform-dependent
213 bitmap. This preserves mask information so that bitmaps and images can
214 be converted back and forth without loss in that respect.", "");
218 virtual wxMask* , GetMask() const,
219 "Gets the associated mask (if any) which may have been loaded from a
220 file or explpicitly set for the bitmap.
222 :see: `SetMask`, `wx.Mask`
225 // MSW only? wxBitmap GetMaskBitmap() const;
229 virtual void , SetMask(wxMask* mask),
230 "Sets the mask for this bitmap.
232 :see: `GetMask`, `wx.Mask`
234 %cleardisown(wxMask*);
237 DocStr(SetMaskColour,
238 "Create a Mask based on a specified colour in the Bitmap.", "");
239 void SetMaskColour(const wxColour& colour) {
240 wxMask *mask = new wxMask(*self, colour);
247 virtual wxBitmap , GetSubBitmap(const wxRect& rect) const,
248 "Returns a sub-bitmap of the current one as long as the rect belongs
249 entirely to the bitmap. This function preserves bit depth and mask
254 virtual bool , SaveFile(const wxString &name, wxBitmapType type,
255 wxPalette *palette = NULL),
256 "Saves a bitmap in the named file. See `__init__` for a description of
257 the ``type`` parameter.", "");
261 virtual bool , LoadFile(const wxString &name, wxBitmapType type),
262 "Loads a bitmap from a file. See `__init__` for a description of the
263 ``type`` parameter.", "");
267 virtual wxPalette *GetPalette() const;
269 virtual void SetPalette(const wxPalette& palette);
273 virtual bool CopyFromIcon(const wxIcon& icon);
276 virtual void , SetHeight(int height),
277 "Set the height property (does not affect the existing bitmap data).", "");
281 virtual void , SetWidth(int width),
282 "Set the width property (does not affect the existing bitmap data).", "");
286 virtual void , SetDepth(int depth),
287 "Set the depth property (does not affect the existing bitmap data).", "");
291 DocStr(SetSize, "Set the bitmap size (does not affect the existing bitmap data).", "");
292 void SetSize(const wxSize& size) {
293 self->SetWidth(size.x);
294 self->SetHeight(size.y);
299 bool CopyFromCursor(const wxCursor& cursor);
301 // WXWIN_COMPATIBILITY_2_4
304 void SetQuality(int q);
305 %pythoncode { GetQuality = wx._deprecated(GetQuality) }
306 %pythoncode { SetQuality = wx._deprecated(SetQuality) }
310 %pythoncode { def __nonzero__(self): return self.IsOk() }
313 bool __eq__(const wxBitmap* other) { return other ? (*self == *other) : false; }
314 bool __ne__(const wxBitmap* other) { return other ? (*self != *other) : true; }
317 %property(Depth, GetDepth, SetDepth, doc="See `GetDepth` and `SetDepth`");
318 %property(Height, GetHeight, SetHeight, doc="See `GetHeight` and `SetHeight`");
319 %property(Mask, GetMask, SetMask, doc="See `GetMask` and `SetMask`");
320 %property(Palette, GetPalette, doc="See `GetPalette`");
321 %property(Size, GetSize, SetSize, doc="See `GetSize` and `SetSize`");
322 %property(SubBitmap, GetSubBitmap, doc="See `GetSubBitmap`");
323 %property(Width, GetWidth, SetWidth, doc="See `GetWidth` and `SetWidth`");
328 //---------------------------------------------------------------------------
329 // Factory functions for creating wxBitmaps from Python buffer objects. They
330 // use the Abstract Pixel API to be able to set RGB and A bytes directly into
331 // the wxBitmap's pixel buffer.
334 // See http://tinyurl.com/e5adr for what premultiplying alpha means. It
335 // appears to me that the other platforms are already doing it, so I'll just
336 // automatically do it for wxMSW here.
338 #define wxPy_premultiply(p, a) ((p) * (a) / 0xff)
339 #define wxPy_unpremultiply(p, a) ((a) ? ((p) * 0xff / (a)) : (p))
341 #define wxPy_premultiply(p, a) (p)
342 #define wxPy_unpremultiply(p, a) (p)
347 %newobject _BitmapFromBufferAlpha;
348 %newobject _BitmapFromBuffer;
350 wxBitmap* _BitmapFromBufferAlpha(int width, int height,
351 buffer data, int DATASIZE,
352 buffer alpha, int ALPHASIZE)
354 if (DATASIZE != width*height*3) {
355 wxPyErr_SetString(PyExc_ValueError, "Invalid data buffer size.");
359 if (ALPHASIZE != width*height) {
360 wxPyErr_SetString(PyExc_ValueError, "Invalid alpha buffer size.");
364 wxBitmap* bmp = new wxBitmap(width, height, 32);
365 wxAlphaPixelData pixData(*bmp, wxPoint(0,0), wxSize(width,height));
367 // raise an exception...
368 wxPyErr_SetString(PyExc_RuntimeError,
369 "Failed to gain raw access to bitmap data.");
374 wxAlphaPixelData::Iterator p(pixData);
375 for (int y=0; y<height; y++) {
376 wxAlphaPixelData::Iterator rowStart = p;
377 for (int x=0; x<width; x++) {
379 p.Red() = wxPy_premultiply(*(data++), a);
380 p.Green() = wxPy_premultiply(*(data++), a);
381 p.Blue() = wxPy_premultiply(*(data++), a);
386 p.OffsetY(pixData, 1);
391 wxBitmap* _BitmapFromBuffer(int width, int height, buffer data, int DATASIZE)
393 if (DATASIZE != width*height*3) {
394 wxPyErr_SetString(PyExc_ValueError, "Invalid data buffer size.");
398 wxBitmap* bmp = new wxBitmap(width, height, 24);
399 wxNativePixelData pixData(*bmp, wxPoint(0,0), wxSize(width,height));
401 // raise an exception...
402 wxPyErr_SetString(PyExc_RuntimeError,
403 "Failed to gain raw access to bitmap data.");
407 wxNativePixelData::Iterator p(pixData);
408 for (int y=0; y<height; y++) {
409 wxNativePixelData::Iterator rowStart = p;
410 for (int x=0; x<width; x++) {
412 p.Green() = *(data++);
413 p.Blue() = *(data++);
417 p.OffsetY(pixData, 1);
425 def BitmapFromBuffer(width, height, dataBuffer, alphaBuffer=None):
427 Creates a `wx.Bitmap` from the data in dataBuffer. The dataBuffer
428 parameter must be a Python object that implements the buffer
429 interface, such as a string, array, etc. The dataBuffer object is
430 expected to contain a series of RGB bytes and be width*height*3
431 bytes long. A buffer object can optionally be supplied for the
432 image's alpha channel data, and it is expected to be width*height
433 bytes long. On Windows the RGB values are 'premultiplied' by the
434 alpha values. (The other platforms do the multiplication
437 Unlike `wx.ImageFromBuffer` the bitmap created with this function
438 does not share the memory buffer with the buffer object. This is
439 because the native pixel buffer format varies on different
440 platforms, and so instead an efficient as possible copy of the
441 data is made from the buffer objects to the bitmap's native pixel
442 buffer. For direct access to a bitmap's pixel buffer see
443 `wx.NativePixelData` and `wx.AlphaPixelData`.
445 :see: `wx.Bitmap`, `wx.BitmapFromBufferRGBA`, `wx.NativePixelData`,
446 `wx.AlphaPixelData`, `wx.ImageFromBuffer`
448 if alphaBuffer is not None:
449 return _gdi_._BitmapFromBufferAlpha(width, height, dataBuffer, alphaBuffer)
451 return _gdi_._BitmapFromBuffer(width, height, dataBuffer)
456 %newobject _BitmapFromBufferRGBA;
458 wxBitmap* _BitmapFromBufferRGBA(int width, int height, buffer data, int DATASIZE)
460 if (DATASIZE != width*height*4) {
461 wxPyErr_SetString(PyExc_ValueError, "Invalid data buffer size.");
465 wxBitmap* bmp = new wxBitmap(width, height, 32);
466 wxAlphaPixelData pixData(*bmp, wxPoint(0,0), wxSize(width,height));
468 // raise an exception...
469 wxPyErr_SetString(PyExc_RuntimeError,
470 "Failed to gain raw access to bitmap data.");
475 wxAlphaPixelData::Iterator p(pixData);
476 for (int y=0; y<height; y++) {
477 wxAlphaPixelData::Iterator rowStart = p;
478 for (int x=0; x<width; x++) {
480 p.Red() = wxPy_premultiply(*(data++), a);
481 p.Green() = wxPy_premultiply(*(data++), a);
482 p.Blue() = wxPy_premultiply(*(data++), a);
483 p.Alpha() = a; data++;
487 p.OffsetY(pixData, 1);
494 def BitmapFromBufferRGBA(width, height, dataBuffer):
496 Creates a `wx.Bitmap` from the data in dataBuffer. The dataBuffer
497 parameter must be a Python object that implements the buffer
498 interface, such as a string, array, etc. The dataBuffer object is
499 expected to contain a series of RGBA bytes (red, green, blue and
500 alpha) and be width*height*4 bytes long. On Windows the RGB
501 values are 'premultiplied' by the alpha values. (The other
502 platforms do the multiplication themselves.)
504 Unlike `wx.ImageFromBuffer` the bitmap created with this function
505 does not share the memory buffer with the buffer object. This is
506 because the native pixel buffer format varies on different
507 platforms, and so instead an efficient as possible copy of the
508 data is made from the buffer object to the bitmap's native pixel
509 buffer. For direct access to a bitmap's pixel buffer see
510 `wx.NativePixelData` and `wx.AlphaPixelData`.
512 :see: `wx.Bitmap`, `wx.BitmapFromBuffer`, `wx.NativePixelData`,
513 `wx.AlphaPixelData`, `wx.ImageFromBuffer`
515 return _gdi_._BitmapFromBufferRGBA(width, height, dataBuffer)
519 //---------------------------------------------------------------------------
521 class wxPixelDataBase
524 // origin of the rectangular region we represent
525 wxPoint GetOrigin() const { return m_ptOrigin; }
527 // width and height of the region we represent
528 int GetWidth() const { return m_width; }
529 int GetHeight() const { return m_height; }
531 wxSize GetSize() const { return wxSize(m_width, m_height); }
533 // the distance between two rows
534 int GetRowStride() const { return m_stride; }
536 %property(Height, GetHeight, doc="See `GetHeight`");
537 %property(Origin, GetOrigin, doc="See `GetOrigin`");
538 %property(RowStride, GetRowStride, doc="See `GetRowStride`");
539 %property(Size, GetSize, doc="See `GetSize`");
540 %property(Width, GetWidth, doc="See `GetWidth`");
544 // Both wxNativePixelData and wxAlphaPixelData have the same interface, so
545 // make a macro to declare them both.
547 %define PIXELDATA(PixelData)
549 typedef PixelData##::Iterator PixelData##_Accessor;
551 class PixelData##_Accessor;
552 class PixelData : public wxPixelDataBase
557 PixelData(wxBitmap& bmp);
558 PixelData(wxBitmap& bmp, const wxRect& rect);
559 PixelData(wxBitmap& bmp, const wxPoint& pt, const wxSize& sz);
563 PixelData##_Accessor GetPixels() const;
567 bool __nonzero__() { return self->operator bool(); }
573 Create and return an iterator object for this pixel data
574 object. (It's really a generator but I won't tell if you
577 width = self.GetWidth()
578 height = self.GetHeight()
579 pixels = self.GetPixels()
581 # This class is a facade over the pixels object (using the one
582 # in the enclosing scope) that only allows Get() and Set() to
584 class PixelFacade(object):
587 def Set(self, *args, **kw):
588 return pixels.Set(*args, **kw)
590 return str(self.Get())
592 return 'pixel(%d,%d): %s' % (x,y,self.Get())
593 X = property(lambda self: x)
594 Y = property(lambda self: y)
597 for y in xrange(height):
598 for x in xrange(width):
599 # We always generate the same pf instance, but it
600 # accesses the pixels object which we use to iterate
601 # over the pixel buffer.
604 pixels.MoveTo(self, 0, y)
607 %property(Pixels, GetPixels, doc="See `GetPixels`");
612 class PixelData##_Accessor
615 %nokwargs PixelData##_Accessor;
617 PixelData##_Accessor(PixelData& data);
618 PixelData##_Accessor(wxBitmap& bmp, PixelData& data);
619 PixelData##_Accessor();
621 ~PixelData##_Accessor();
623 void Reset(const PixelData& data);
627 // PixelData##_Accessor& nextPixel() { return ++(*self); }
628 void nextPixel() { ++(*self); }
631 void Offset(const PixelData& data, int x, int y);
632 void OffsetX(const PixelData& data, int x);
633 void OffsetY(const PixelData& data, int y);
634 void MoveTo(const PixelData& data, int x, int y);
636 // NOTE: For now I'm not wrapping the Red, Green, Blue and Alpha
637 // functions because I can't hide the premultiplying needed on wxMSW
638 // if only the individual components are wrapped, plus it would mean 3
639 // or 4 trips per pixel from Python to C++ instead of just one.
640 // Instead I've added the Set and Get functions and put the
641 // premultiplying in there.
644 // byte _get_Red() { return self->Red(); }
645 // byte _get_Green() { return self->Green(); }
646 // byte _get_Blue() { return self->Blue(); }
648 // void _set_Red(byte val) { self->Red() = val; }
649 // void _set_Green(byte val) { self->Green() = val; }
650 // void _set_Blue(byte val) { self->Blue() = val; }
654 // Red = property(_get_Red, _set_Red)
655 // Green = property(_get_Green, _set_Green)
656 // Blue = property(_get_Blue, _set_Blue)
662 %pythonAppend wxAlphaPixelData::wxAlphaPixelData "self.UseAlpha()"
665 PIXELDATA(wxNativePixelData)
666 PIXELDATA(wxAlphaPixelData)
669 // Add in a few things that are different between the wxNativePixelData and
670 // wxAlphaPixelData and the iterator classes and so are not included in our
673 %extend wxNativePixelData_Accessor {
674 void Set(byte red, byte green, byte blue) {
676 self->Green() = green;
681 PyObject* rv = PyTuple_New(3);
682 PyTuple_SetItem(rv, 0, PyInt_FromLong(self->Red()));
683 PyTuple_SetItem(rv, 1, PyInt_FromLong(self->Green()));
684 PyTuple_SetItem(rv, 2, PyInt_FromLong(self->Blue()));
689 %extend wxAlphaPixelData_Accessor {
690 // byte _get_Alpha() { return self->Alpha(); }
691 // void _set_Alpha(byte val) { self->Alpha() = val; }
694 // Alpha = property(_get_Alpha, _set_Alpha)
697 void Set(byte red, byte green, byte blue, byte alpha) {
698 self->Red() = wxPy_premultiply(red, alpha);
699 self->Green() = wxPy_premultiply(green, alpha);
700 self->Blue() = wxPy_premultiply(blue, alpha);
701 self->Alpha() = alpha;
705 PyObject* rv = PyTuple_New(4);
706 int red = self->Red();
707 int green = self->Green();
708 int blue = self->Blue();
709 int alpha = self->Alpha();
711 PyTuple_SetItem(rv, 0, PyInt_FromLong( wxPy_unpremultiply(red, alpha) ));
712 PyTuple_SetItem(rv, 1, PyInt_FromLong( wxPy_unpremultiply(green, alpha) ));
713 PyTuple_SetItem(rv, 2, PyInt_FromLong( wxPy_unpremultiply(blue, alpha) ));
714 PyTuple_SetItem(rv, 3, PyInt_FromLong( alpha ));
720 //---------------------------------------------------------------------------
723 "This class encapsulates a monochrome mask bitmap, where the masked
724 area is black and the unmasked area is white. When associated with a
725 bitmap and drawn in a device context, the unmasked area of the bitmap
726 will be drawn, and the masked area will not be drawn.
728 A mask may be associated with a `wx.Bitmap`. It is used in
729 `wx.DC.DrawBitmap` or `wx.DC.Blit` when the source device context is a
730 `wx.MemoryDC` with a `wx.Bitmap` selected into it that contains a
735 class wxMask : public wxObject {
739 "Constructs a mask from a `wx.Bitmap` and a `wx.Colour` in that bitmap
740 that indicates the transparent portions of the mask. In other words,
741 the pixels in ``bitmap`` that match ``colour`` will be the transparent
742 portions of the mask. If no ``colour`` or an invalid ``colour`` is
743 passed then BLACK is used.
745 :see: `wx.Bitmap`, `wx.Colour`", "");
748 wxMask(const wxBitmap& bitmap, const wxColour& colour = wxNullColour) {
749 if ( !colour.IsOk() )
750 return new wxMask(bitmap, *wxBLACK);
752 return new wxMask(bitmap, colour);
759 %pythoncode { MaskColour = wx._deprecated(Mask, "wx.MaskColour is deprecated, use `wx.Mask` instead.") }
761 //---------------------------------------------------------------------------
762 //---------------------------------------------------------------------------