]>
Commit | Line | Data |
---|---|---|
1 | ///////////////////////////////////////////////////////////////////////////// | |
2 | // Name: _bitmap.i | |
3 | // Purpose: SWIG interface for wxBitmap and wxMask | |
4 | // | |
5 | // Author: Robin Dunn | |
6 | // | |
7 | // Created: 7-July-1997 | |
8 | // RCS-ID: $Id$ | |
9 | // Copyright: (c) 2003 by Total Control Software | |
10 | // Licence: wxWindows license | |
11 | ///////////////////////////////////////////////////////////////////////////// | |
12 | ||
13 | // Not a %module | |
14 | ||
15 | %{ | |
16 | #include <wx/rawbmp.h> | |
17 | %} | |
18 | ||
19 | ||
20 | // Turn off the aquisition of the Global Interpreter Lock for this file | |
21 | %threadWrapperOff | |
22 | ||
23 | //--------------------------------------------------------------------------- | |
24 | ||
25 | %{ | |
26 | #include <wx/image.h> | |
27 | ||
28 | static char** ConvertListOfStrings(PyObject* listOfStrings) { | |
29 | char** cArray = NULL; | |
30 | int count; | |
31 | ||
32 | if (!PyList_Check(listOfStrings)) { | |
33 | PyErr_SetString(PyExc_TypeError, "Expected a list of strings."); | |
34 | return NULL; | |
35 | } | |
36 | count = PyList_Size(listOfStrings); | |
37 | cArray = new char*[count]; | |
38 | ||
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)); | |
42 | } | |
43 | return cArray; | |
44 | } | |
45 | ||
46 | %} | |
47 | ||
48 | //--------------------------------------------------------------------------- | |
49 | ||
50 | // TODO: When the API stabalizes and is available on other platforms, add | |
51 | // wrappers for the new wxBitmap, wxRawBitmap, wxDIB stuff... | |
52 | ||
53 | DocStr(wxBitmap, | |
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.", " | |
60 | ||
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. | |
65 | ||
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? | |
69 | ||
70 | :todo: Find a way to do very efficient PIL Image <--> wx.Bitmap | |
71 | converstions. | |
72 | ||
73 | :see: `wx.EmptyBitmap`, `wx.BitmapFromIcon`, `wx.BitmapFromImage`, | |
74 | `wx.BitmapFromXPMData`, `wx.BitmapFromBits`, `wx.BitmapFromBuffer`, | |
75 | `wx.BitmapFromBufferRGBA`, `wx.Image` | |
76 | "); | |
77 | ||
78 | ||
79 | MustHaveApp(wxBitmap); | |
80 | ||
81 | class wxBitmap : public wxGDIObject | |
82 | { | |
83 | public: | |
84 | DocCtorStr( | |
85 | wxBitmap(const wxString& name, wxBitmapType type=wxBITMAP_TYPE_ANY), | |
86 | "Loads a bitmap from a file.", | |
87 | " | |
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 | |
91 | loaded): | |
92 | ||
93 | * wx.BITMAP_TYPE_ANY | |
94 | * wx.BITMAP_TYPE_BMP | |
95 | * wx.BITMAP_TYPE_ICO | |
96 | * wx.BITMAP_TYPE_CUR | |
97 | * wx.BITMAP_TYPE_XBM | |
98 | * wx.BITMAP_TYPE_XPM | |
99 | * wx.BITMAP_TYPE_TIF | |
100 | * wx.BITMAP_TYPE_GIF | |
101 | * wx.BITMAP_TYPE_PNG | |
102 | * wx.BITMAP_TYPE_JPEG | |
103 | * wx.BITMAP_TYPE_PNM | |
104 | * wx.BITMAP_TYPE_PCX | |
105 | * wx.BITMAP_TYPE_PICT | |
106 | * wx.BITMAP_TYPE_ICON | |
107 | * wx.BITMAP_TYPE_ANI | |
108 | * wx.BITMAP_TYPE_IFF | |
109 | ||
110 | :see: Alternate constructors `wx.EmptyBitmap`, `wx.BitmapFromIcon`, | |
111 | `wx.BitmapFromImage`, `wx.BitmapFromXPMData`, `wx.BitmapFromBits`, | |
112 | `wx.BitmapFromBuffer`, `wx.BitmapFromBufferRGBA`, | |
113 | "); | |
114 | ||
115 | ~wxBitmap(); | |
116 | ||
117 | DocCtorStrName( | |
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.", "", | |
122 | EmptyBitmap); | |
123 | ||
124 | DocCtorStrName( | |
125 | wxBitmap(const wxIcon& icon), | |
126 | "Create a new bitmap from a `wx.Icon` object.", "", | |
127 | BitmapFromIcon); | |
128 | ||
129 | DocCtorStrName( | |
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.", "", | |
136 | BitmapFromImage); | |
137 | ||
138 | ||
139 | %extend { | |
140 | %RenameDocCtor( | |
141 | BitmapFromXPMData, | |
142 | "Construct a Bitmap from a list of strings formatted as XPM data.", "", | |
143 | wxBitmap(PyObject* listOfStrings)) | |
144 | { | |
145 | char** cArray = NULL; | |
146 | wxBitmap* bmp; | |
147 | ||
148 | cArray = ConvertListOfStrings(listOfStrings); | |
149 | if (! cArray) | |
150 | return NULL; | |
151 | bmp = new wxBitmap(cArray); | |
152 | delete [] cArray; | |
153 | return bmp; | |
154 | } | |
155 | ||
156 | ||
157 | %RenameDocCtor( | |
158 | BitmapFromBits, | |
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 )) | |
164 | { | |
165 | char* buf; | |
166 | Py_ssize_t length; | |
167 | PyString_AsStringAndSize(bits, &buf, &length); | |
168 | return new wxBitmap(buf, width, height, depth); | |
169 | } | |
170 | } | |
171 | ||
172 | ||
173 | // wxGDIImage methods | |
174 | #ifdef __WXMSW__ | |
175 | long GetHandle(); | |
176 | %extend { | |
177 | void SetHandle(long handle) { self->SetHandle((WXHANDLE)handle); } | |
178 | } | |
179 | #endif | |
180 | ||
181 | bool IsOk(); | |
182 | %pythoncode { Ok = IsOk } | |
183 | ||
184 | DocDeclStr( | |
185 | int , GetWidth(), | |
186 | "Gets the width of the bitmap in pixels.", ""); | |
187 | ||
188 | ||
189 | DocDeclStr( | |
190 | int , GetHeight(), | |
191 | "Gets the height of the bitmap in pixels.", ""); | |
192 | ||
193 | ||
194 | DocDeclStr( | |
195 | int , GetDepth(), | |
196 | "Gets the colour depth of the bitmap. A value of 1 indicates a | |
197 | monochrome bitmap.", ""); | |
198 | ||
199 | ||
200 | ||
201 | %extend { | |
202 | DocStr(GetSize, "Get the size of the bitmap.", ""); | |
203 | wxSize GetSize() { | |
204 | wxSize size(self->GetWidth(), self->GetHeight()); | |
205 | return size; | |
206 | } | |
207 | } | |
208 | ||
209 | ||
210 | DocDeclStr( | |
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.", ""); | |
215 | ||
216 | ||
217 | DocDeclStr( | |
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. | |
221 | ||
222 | :see: `SetMask`, `wx.Mask` | |
223 | ", ""); | |
224 | ||
225 | // MSW only? wxBitmap GetMaskBitmap() const; | |
226 | ||
227 | %disownarg(wxMask*); | |
228 | DocDeclStr( | |
229 | virtual void , SetMask(wxMask* mask), | |
230 | "Sets the mask for this bitmap. | |
231 | ||
232 | :see: `GetMask`, `wx.Mask` | |
233 | ", ""); | |
234 | %cleardisown(wxMask*); | |
235 | ||
236 | %extend { | |
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); | |
241 | self->SetMask(mask); | |
242 | } | |
243 | } | |
244 | ||
245 | ||
246 | DocDeclStr( | |
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 | |
250 | information.", ""); | |
251 | ||
252 | ||
253 | DocDeclStr( | |
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.", ""); | |
258 | ||
259 | ||
260 | DocDeclStr( | |
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.", ""); | |
264 | ||
265 | ||
266 | ||
267 | virtual wxPalette *GetPalette() const; | |
268 | #ifdef __WXMSW__ | |
269 | virtual void SetPalette(const wxPalette& palette); | |
270 | #endif | |
271 | ||
272 | ||
273 | virtual bool CopyFromIcon(const wxIcon& icon); | |
274 | ||
275 | DocDeclStr( | |
276 | virtual void , SetHeight(int height), | |
277 | "Set the height property (does not affect the existing bitmap data).", ""); | |
278 | ||
279 | ||
280 | DocDeclStr( | |
281 | virtual void , SetWidth(int width), | |
282 | "Set the width property (does not affect the existing bitmap data).", ""); | |
283 | ||
284 | ||
285 | DocDeclStr( | |
286 | virtual void , SetDepth(int depth), | |
287 | "Set the depth property (does not affect the existing bitmap data).", ""); | |
288 | ||
289 | ||
290 | %extend { | |
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); | |
295 | } | |
296 | } | |
297 | ||
298 | #ifdef __WXMSW__ | |
299 | bool CopyFromCursor(const wxCursor& cursor); | |
300 | ||
301 | // WXWIN_COMPATIBILITY_2_4 | |
302 | #if 0 | |
303 | int GetQuality(); | |
304 | void SetQuality(int q); | |
305 | %pythoncode { GetQuality = wx._deprecated(GetQuality) } | |
306 | %pythoncode { SetQuality = wx._deprecated(SetQuality) } | |
307 | #endif | |
308 | #endif | |
309 | ||
310 | %pythoncode { def __nonzero__(self): return self.IsOk() } | |
311 | ||
312 | %extend { | |
313 | bool __eq__(const wxBitmap* other) { return other ? (*self == *other) : false; } | |
314 | bool __ne__(const wxBitmap* other) { return other ? (*self != *other) : true; } | |
315 | } | |
316 | ||
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`"); | |
324 | ||
325 | }; | |
326 | ||
327 | ||
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. | |
332 | ||
333 | %{ | |
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. | |
337 | #ifdef __WXMSW__ | |
338 | #define wxPy_premultiply(p, a) ((p) * (a) / 0xff) | |
339 | #define wxPy_unpremultiply(p, a) ((a) ? ((p) * 0xff / (a)) : (p)) | |
340 | #else | |
341 | #define wxPy_premultiply(p, a) (p) | |
342 | #define wxPy_unpremultiply(p, a) (p) | |
343 | #endif | |
344 | %} | |
345 | ||
346 | ||
347 | %newobject _BitmapFromBufferAlpha; | |
348 | %newobject _BitmapFromBuffer; | |
349 | %inline %{ | |
350 | wxBitmap* _BitmapFromBufferAlpha(int width, int height, | |
351 | buffer data, int DATASIZE, | |
352 | buffer alpha, int ALPHASIZE) | |
353 | { | |
354 | if (DATASIZE != width*height*3) { | |
355 | wxPyErr_SetString(PyExc_ValueError, "Invalid data buffer size."); | |
356 | return NULL; | |
357 | } | |
358 | ||
359 | if (ALPHASIZE != width*height) { | |
360 | wxPyErr_SetString(PyExc_ValueError, "Invalid alpha buffer size."); | |
361 | return NULL; | |
362 | } | |
363 | ||
364 | wxBitmap* bmp = new wxBitmap(width, height, 32); | |
365 | wxAlphaPixelData pixData(*bmp, wxPoint(0,0), wxSize(width,height)); | |
366 | if (! pixData) { | |
367 | // raise an exception... | |
368 | wxPyErr_SetString(PyExc_RuntimeError, | |
369 | "Failed to gain raw access to bitmap data."); | |
370 | return NULL; | |
371 | } | |
372 | ||
373 | pixData.UseAlpha(); | |
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++) { | |
378 | byte a = *(alpha++); | |
379 | p.Red() = wxPy_premultiply(*(data++), a); | |
380 | p.Green() = wxPy_premultiply(*(data++), a); | |
381 | p.Blue() = wxPy_premultiply(*(data++), a); | |
382 | p.Alpha() = a; | |
383 | ++p; | |
384 | } | |
385 | p = rowStart; | |
386 | p.OffsetY(pixData, 1); | |
387 | } | |
388 | return bmp; | |
389 | } | |
390 | ||
391 | wxBitmap* _BitmapFromBuffer(int width, int height, buffer data, int DATASIZE) | |
392 | { | |
393 | if (DATASIZE != width*height*3) { | |
394 | wxPyErr_SetString(PyExc_ValueError, "Invalid data buffer size."); | |
395 | return NULL; | |
396 | } | |
397 | ||
398 | wxBitmap* bmp = new wxBitmap(width, height, 24); | |
399 | wxNativePixelData pixData(*bmp, wxPoint(0,0), wxSize(width,height)); | |
400 | if (! pixData) { | |
401 | // raise an exception... | |
402 | wxPyErr_SetString(PyExc_RuntimeError, | |
403 | "Failed to gain raw access to bitmap data."); | |
404 | return NULL; | |
405 | } | |
406 | ||
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++) { | |
411 | p.Red() = *(data++); | |
412 | p.Green() = *(data++); | |
413 | p.Blue() = *(data++); | |
414 | ++p; | |
415 | } | |
416 | p = rowStart; | |
417 | p.OffsetY(pixData, 1); | |
418 | } | |
419 | return bmp; | |
420 | } | |
421 | %} | |
422 | ||
423 | ||
424 | %pythoncode { | |
425 | def BitmapFromBuffer(width, height, dataBuffer, alphaBuffer=None): | |
426 | """ | |
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 | |
435 | themselves.) | |
436 | ||
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`. | |
444 | ||
445 | :see: `wx.Bitmap`, `wx.BitmapFromBufferRGBA`, `wx.NativePixelData`, | |
446 | `wx.AlphaPixelData`, `wx.ImageFromBuffer` | |
447 | """ | |
448 | if alphaBuffer is not None: | |
449 | return _gdi_._BitmapFromBufferAlpha(width, height, dataBuffer, alphaBuffer) | |
450 | else: | |
451 | return _gdi_._BitmapFromBuffer(width, height, dataBuffer) | |
452 | } | |
453 | ||
454 | ||
455 | ||
456 | %newobject _BitmapFromBufferRGBA; | |
457 | %inline %{ | |
458 | wxBitmap* _BitmapFromBufferRGBA(int width, int height, buffer data, int DATASIZE) | |
459 | { | |
460 | if (DATASIZE != width*height*4) { | |
461 | wxPyErr_SetString(PyExc_ValueError, "Invalid data buffer size."); | |
462 | return NULL; | |
463 | } | |
464 | ||
465 | wxBitmap* bmp = new wxBitmap(width, height, 32); | |
466 | wxAlphaPixelData pixData(*bmp, wxPoint(0,0), wxSize(width,height)); | |
467 | if (! pixData) { | |
468 | // raise an exception... | |
469 | wxPyErr_SetString(PyExc_RuntimeError, | |
470 | "Failed to gain raw access to bitmap data."); | |
471 | return NULL; | |
472 | } | |
473 | ||
474 | pixData.UseAlpha(); | |
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++) { | |
479 | byte a = data[3]; | |
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++; | |
484 | ++p; | |
485 | } | |
486 | p = rowStart; | |
487 | p.OffsetY(pixData, 1); | |
488 | } | |
489 | return bmp; | |
490 | } | |
491 | %} | |
492 | ||
493 | %pythoncode { | |
494 | def BitmapFromBufferRGBA(width, height, dataBuffer): | |
495 | """ | |
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.) | |
503 | ||
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`. | |
511 | ||
512 | :see: `wx.Bitmap`, `wx.BitmapFromBuffer`, `wx.NativePixelData`, | |
513 | `wx.AlphaPixelData`, `wx.ImageFromBuffer` | |
514 | """ | |
515 | return _gdi_._BitmapFromBufferRGBA(width, height, dataBuffer) | |
516 | } | |
517 | ||
518 | ||
519 | //--------------------------------------------------------------------------- | |
520 | ||
521 | class wxPixelDataBase | |
522 | { | |
523 | public: | |
524 | // origin of the rectangular region we represent | |
525 | wxPoint GetOrigin() const { return m_ptOrigin; } | |
526 | ||
527 | // width and height of the region we represent | |
528 | int GetWidth() const { return m_width; } | |
529 | int GetHeight() const { return m_height; } | |
530 | ||
531 | wxSize GetSize() const { return wxSize(m_width, m_height); } | |
532 | ||
533 | // the distance between two rows | |
534 | int GetRowStride() const { return m_stride; } | |
535 | ||
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`"); | |
541 | }; | |
542 | ||
543 | ||
544 | // Both wxNativePixelData and wxAlphaPixelData have the same interface, so | |
545 | // make a macro to declare them both. | |
546 | ||
547 | %define PIXELDATA(PixelData) | |
548 | %{ | |
549 | typedef PixelData##::Iterator PixelData##_Accessor; | |
550 | %} | |
551 | class PixelData##_Accessor; | |
552 | class PixelData : public wxPixelDataBase | |
553 | { | |
554 | public: | |
555 | %nokwargs PixelData; | |
556 | ||
557 | PixelData(wxBitmap& bmp); | |
558 | PixelData(wxBitmap& bmp, const wxRect& rect); | |
559 | PixelData(wxBitmap& bmp, const wxPoint& pt, const wxSize& sz); | |
560 | ||
561 | ~PixelData(); | |
562 | ||
563 | PixelData##_Accessor GetPixels() const; | |
564 | void UseAlpha(); | |
565 | ||
566 | %extend { | |
567 | bool __nonzero__() { return self->operator bool(); } | |
568 | } | |
569 | ||
570 | %pythoncode { | |
571 | def __iter__(self): | |
572 | """ | |
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 | |
575 | don't tell.) | |
576 | """ | |
577 | width = self.GetWidth() | |
578 | height = self.GetHeight() | |
579 | pixels = self.GetPixels() | |
580 | ||
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 | |
583 | # be called. | |
584 | class PixelFacade(object): | |
585 | def Get(self): | |
586 | return pixels.Get() | |
587 | def Set(self, *args, **kw): | |
588 | return pixels.Set(*args, **kw) | |
589 | def __str__(self): | |
590 | return str(self.Get()) | |
591 | def __repr__(self): | |
592 | return 'pixel(%d,%d): %s' % (x,y,self.Get()) | |
593 | X = property(lambda self: x) | |
594 | Y = property(lambda self: y) | |
595 | ||
596 | pf = PixelFacade() | |
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. | |
602 | yield pf | |
603 | pixels.nextPixel() | |
604 | pixels.MoveTo(self, 0, y) | |
605 | } | |
606 | ||
607 | %property(Pixels, GetPixels, doc="See `GetPixels`"); | |
608 | }; | |
609 | ||
610 | ||
611 | ||
612 | class PixelData##_Accessor | |
613 | { | |
614 | public: | |
615 | %nokwargs PixelData##_Accessor; | |
616 | ||
617 | PixelData##_Accessor(PixelData& data); | |
618 | PixelData##_Accessor(wxBitmap& bmp, PixelData& data); | |
619 | PixelData##_Accessor(); | |
620 | ||
621 | ~PixelData##_Accessor(); | |
622 | ||
623 | void Reset(const PixelData& data); | |
624 | bool IsOk() const; | |
625 | ||
626 | %extend { | |
627 | // PixelData##_Accessor& nextPixel() { return ++(*self); } | |
628 | void nextPixel() { ++(*self); } | |
629 | } | |
630 | ||
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); | |
635 | ||
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. | |
642 | ||
643 | // %extend { | |
644 | // byte _get_Red() { return self->Red(); } | |
645 | // byte _get_Green() { return self->Green(); } | |
646 | // byte _get_Blue() { return self->Blue(); } | |
647 | ||
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; } | |
651 | // } | |
652 | ||
653 | // %pythoncode { | |
654 | // Red = property(_get_Red, _set_Red) | |
655 | // Green = property(_get_Green, _set_Green) | |
656 | // Blue = property(_get_Blue, _set_Blue) | |
657 | // } | |
658 | }; | |
659 | %enddef | |
660 | ||
661 | ||
662 | %pythonAppend wxAlphaPixelData::wxAlphaPixelData "self.UseAlpha()" | |
663 | ||
664 | // Make the classes | |
665 | PIXELDATA(wxNativePixelData) | |
666 | PIXELDATA(wxAlphaPixelData) | |
667 | ||
668 | ||
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 | |
671 | // macro... | |
672 | ||
673 | %extend wxNativePixelData_Accessor { | |
674 | void Set(byte red, byte green, byte blue) { | |
675 | self->Red() = red; | |
676 | self->Green() = green; | |
677 | self->Blue() = blue; | |
678 | } | |
679 | ||
680 | PyObject* Get() { | |
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())); | |
685 | return rv; | |
686 | } | |
687 | } | |
688 | ||
689 | %extend wxAlphaPixelData_Accessor { | |
690 | // byte _get_Alpha() { return self->Alpha(); } | |
691 | // void _set_Alpha(byte val) { self->Alpha() = val; } | |
692 | ||
693 | // %pythoncode { | |
694 | // Alpha = property(_get_Alpha, _set_Alpha) | |
695 | // } | |
696 | ||
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; | |
702 | } | |
703 | ||
704 | PyObject* Get() { | |
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(); | |
710 | ||
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 )); | |
715 | return rv; | |
716 | } | |
717 | } | |
718 | ||
719 | ||
720 | //--------------------------------------------------------------------------- | |
721 | ||
722 | DocStr(wxMask, | |
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. | |
727 | ||
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 | |
731 | mask.", ""); | |
732 | ||
733 | MustHaveApp(wxMask); | |
734 | ||
735 | class wxMask : public wxObject { | |
736 | public: | |
737 | ||
738 | DocStr(wxMask, | |
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. | |
744 | ||
745 | :see: `wx.Bitmap`, `wx.Colour`", ""); | |
746 | ||
747 | %extend { | |
748 | wxMask(const wxBitmap& bitmap, const wxColour& colour = wxNullColour) { | |
749 | if ( !colour.IsOk() ) | |
750 | return new wxMask(bitmap, *wxBLACK); | |
751 | else | |
752 | return new wxMask(bitmap, colour); | |
753 | } | |
754 | } | |
755 | ||
756 | ~wxMask(); | |
757 | }; | |
758 | ||
759 | %pythoncode { MaskColour = wx._deprecated(Mask, "wx.MaskColour is deprecated, use `wx.Mask` instead.") } | |
760 | ||
761 | //--------------------------------------------------------------------------- | |
762 | //--------------------------------------------------------------------------- | |
763 | %threadWrapperOn |