]> git.saurik.com Git - wxWidgets.git/blob - src/cocoa/bitmap.mm
Ensure there is valid context for DrawRectangle
[wxWidgets.git] / src / cocoa / bitmap.mm
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: src/cocoa/bitmap.mm
3 // Purpose: wxBitmap
4 // Author: David Elliott
5 // Modified by:
6 // Created: 2003/07/19
7 // RCS-ID: $Id$
8 // Copyright: (c) 2003 David Elliott
9 // Licence: wxWindows licence
10 /////////////////////////////////////////////////////////////////////////////
11
12 #include "wx/wxprec.h"
13
14 #include "wx/bitmap.h"
15
16 #ifndef WX_PRECOMP
17 #include "wx/log.h"
18 #include "wx/utils.h"
19 #include "wx/palette.h"
20 #include "wx/icon.h"
21 #include "wx/colour.h"
22 #include "wx/image.h"
23 #endif //WX_PRECOMP
24
25 #include "wx/xpmdecod.h"
26 #include "wx/rawbmp.h"
27
28 #include "wx/cocoa/autorelease.h"
29 #include "wx/cocoa/string.h"
30 #include "wx/cocoa/ObjcRef.h"
31
32 #import <AppKit/NSBitmapImageRep.h>
33 #import <AppKit/NSGraphics.h>
34 #import <AppKit/NSImage.h>
35 #import <AppKit/NSColor.h>
36
37 IMPLEMENT_ABSTRACT_CLASS(wxBitmapHandler, wxBitmapHandlerBase)
38
39 // ========================================================================
40 // wxBitmapRefData
41 // ========================================================================
42
43 class wxBitmapRefData: public wxGDIRefData
44 {
45 friend class wxBitmap;
46 public:
47 wxBitmapRefData();
48 wxBitmapRefData( const wxBitmapRefData& data );
49 virtual ~wxBitmapRefData();
50
51 virtual bool IsOk() const { return m_ok; }
52
53 protected:
54 int m_width;
55 int m_height;
56 int m_depth;
57 bool m_ok;
58 int m_numColors;
59 wxPalette m_bitmapPalette;
60 int m_quality;
61 WX_NSBitmapImageRep m_cocoaNSBitmapImageRep;
62 wxMask *m_bitmapMask; // Optional mask
63 };
64
65 wxBitmapRefData::wxBitmapRefData()
66 {
67 m_ok = FALSE;
68 m_width = 0;
69 m_height = 0;
70 m_depth = 0;
71 m_quality = 0;
72 m_numColors = 0;
73 m_cocoaNSBitmapImageRep = nil;
74 m_bitmapMask = NULL;
75 }
76
77 wxBitmapRefData::wxBitmapRefData( const wxBitmapRefData& data)
78 {
79 wxAutoNSAutoreleasePool pool;
80
81 m_width = data.m_width;
82 m_height = data.m_height;
83 m_depth = data.m_depth;
84 m_ok = data.m_ok;
85 m_numColors = data.m_numColors;
86 m_bitmapPalette = data.m_bitmapPalette;
87 m_quality = data.m_quality;
88 m_cocoaNSBitmapImageRep = wxGCSafeRetain([[data.m_cocoaNSBitmapImageRep copyWithZone:nil] autorelease]);
89 m_bitmapMask = data.m_bitmapMask?new wxMask(*data.m_bitmapMask):NULL;
90 }
91
92 wxBitmapRefData::~wxBitmapRefData()
93 {
94 wxGCSafeRelease(m_cocoaNSBitmapImageRep);
95 m_cocoaNSBitmapImageRep = NULL;
96
97 delete m_bitmapMask;
98 m_bitmapMask = NULL;
99 }
100
101 // ========================================================================
102 // wxBitmap
103 // ========================================================================
104
105 #define M_BITMAPDATA ((wxBitmapRefData *)m_refData)
106
107 IMPLEMENT_DYNAMIC_CLASS(wxBitmap, wxGDIObject)
108
109 wxBitmap::wxBitmap()
110 {
111 m_refData = NULL;
112 }
113
114 wxBitmap::~wxBitmap()
115 {
116 }
117
118 wxBitmap::wxBitmap(const char bits[], int the_width, int the_height, int no_bits)
119 {
120 m_refData = new wxBitmapRefData;
121
122 M_BITMAPDATA->m_width = the_width ;
123 M_BITMAPDATA->m_height = the_height ;
124 M_BITMAPDATA->m_depth = no_bits ;
125 M_BITMAPDATA->m_numColors = 0;
126
127 /* TODO: create the bitmap from data */
128 }
129
130 wxBitmap::wxBitmap(NSImage* cocoaNSImage)
131 {
132 (void) Create(cocoaNSImage);
133 }
134
135 wxBitmap::wxBitmap(NSBitmapImageRep* cocoaNSBitmapImageRep)
136 {
137 (void) Create(cocoaNSBitmapImageRep);
138 }
139
140 wxBitmap::wxBitmap(const void* data, wxBitmapType type, int width, int height, int depth)
141 {
142 (void) Create(data, type, width, height, depth);
143 }
144
145 wxBitmap::wxBitmap(const wxString& filename, wxBitmapType type)
146 {
147 LoadFile(filename, type);
148 }
149
150 wxGDIRefData *wxBitmap::CreateGDIRefData() const
151 {
152 return new wxBitmapRefData;
153 }
154
155 wxGDIRefData *wxBitmap::CloneGDIRefData(const wxGDIRefData *data) const
156 {
157 return new wxBitmapRefData(*(wxBitmapRefData*)data);
158 }
159
160 WX_NSBitmapImageRep wxBitmap::GetNSBitmapImageRep()
161 {
162 if(!M_BITMAPDATA)
163 return NULL;
164 return M_BITMAPDATA->m_cocoaNSBitmapImageRep;
165 }
166
167 WX_NSImage wxBitmap::GetNSImage(bool useMask) const
168 {
169 if(!IsOk())
170 return nil;
171 NSImage *nsimage = [[[NSImage alloc]
172 initWithSize:NSMakeSize(GetWidth(), GetHeight())] autorelease];
173 if(!nsimage)
174 return nil;
175 [nsimage addRepresentation: M_BITMAPDATA->m_cocoaNSBitmapImageRep];
176 if(useMask && GetMask())
177 {
178 // Show before/after to prove that the bitmap itself is not changed
179 // even though we just composited onto the NSImage
180 wxLogTrace(wxTRACE_COCOA,wxT("Before: bpp=%d"),[M_BITMAPDATA->m_cocoaNSBitmapImageRep bitsPerPixel]);
181 NSImage *maskImage = [[NSImage alloc]
182 initWithSize:NSMakeSize(GetWidth(), GetHeight())];
183 [maskImage addRepresentation: GetMask()->GetNSBitmapImageRep()];
184 [nsimage lockFocus];
185 [maskImage compositeToPoint:NSZeroPoint operation:NSCompositeDestinationIn];
186 [nsimage unlockFocus];
187 [maskImage release];
188 wxLogTrace(wxTRACE_COCOA,wxT("After: bpp=%d"),[M_BITMAPDATA->m_cocoaNSBitmapImageRep bitsPerPixel]);
189 }
190 return nsimage;
191 }
192
193 void wxBitmap::SetNSBitmapImageRep(WX_NSBitmapImageRep bitmapImageRep)
194 {
195 if(!M_BITMAPDATA)
196 return;
197 // NOTE: No checking is done to make sure width/height agree
198 wxGCSafeRetain(bitmapImageRep);
199 wxGCSafeRelease(M_BITMAPDATA->m_cocoaNSBitmapImageRep);
200 M_BITMAPDATA->m_cocoaNSBitmapImageRep = bitmapImageRep;
201 }
202
203 void wxBitmap::SetWidth(int w)
204 {
205 if (!M_BITMAPDATA)
206 m_refData = new wxBitmapRefData;
207
208 M_BITMAPDATA->m_width = w;
209 }
210
211 void wxBitmap::SetHeight(int h)
212 {
213 if (!M_BITMAPDATA)
214 m_refData = new wxBitmapRefData;
215
216 M_BITMAPDATA->m_height = h;
217 }
218
219 void wxBitmap::SetDepth(int d)
220 {
221 if (!M_BITMAPDATA)
222 m_refData = new wxBitmapRefData;
223
224 M_BITMAPDATA->m_depth = d;
225 }
226
227 void wxBitmap::SetQuality(int q)
228 {
229 if (!M_BITMAPDATA)
230 m_refData = new wxBitmapRefData;
231
232 M_BITMAPDATA->m_quality = q;
233 }
234
235 void wxBitmap::SetOk(bool isOk)
236 {
237 if (!M_BITMAPDATA)
238 m_refData = new wxBitmapRefData;
239
240 M_BITMAPDATA->m_ok = isOk;
241 }
242
243 void wxBitmap::SetPalette(const wxPalette& palette)
244 {
245 if (!M_BITMAPDATA)
246 m_refData = new wxBitmapRefData;
247
248 M_BITMAPDATA->m_bitmapPalette = palette ;
249 }
250
251 void wxBitmap::SetMask(wxMask *mask)
252 {
253 if (!M_BITMAPDATA)
254 m_refData = new wxBitmapRefData;
255
256 M_BITMAPDATA->m_bitmapMask = mask ;
257 }
258
259 wxPalette* wxBitmap::GetPalette() const
260 {
261 if(!m_refData)
262 return NULL;
263 return &M_BITMAPDATA->m_bitmapPalette;
264 }
265
266 wxMask* wxBitmap::GetMask() const
267 {
268 if(!m_refData)
269 return NULL;
270 return M_BITMAPDATA->m_bitmapMask;
271 }
272
273 int wxBitmap::GetDepth() const
274 {
275 if(!m_refData)
276 return 0;
277 return M_BITMAPDATA->m_depth;
278 }
279
280 int wxBitmap::GetWidth() const
281 {
282 if(!m_refData)
283 return 0;
284 return M_BITMAPDATA->m_width;
285 }
286
287 int wxBitmap::GetHeight() const
288 {
289 if(!m_refData)
290 return 0;
291 return M_BITMAPDATA->m_height;
292 }
293
294 bool wxBitmap::Create(int w, int h, int d)
295 {
296 wxAutoNSAutoreleasePool pool;
297
298 UnRef();
299
300 m_refData = new wxBitmapRefData;
301
302 M_BITMAPDATA->m_width = w;
303 M_BITMAPDATA->m_height = h;
304 M_BITMAPDATA->m_depth = d;
305
306 /* TODO: create new bitmap */
307 M_BITMAPDATA->m_cocoaNSBitmapImageRep = wxGCSafeRetain([[[NSBitmapImageRep alloc]
308 initWithBitmapDataPlanes: NULL
309 pixelsWide: w
310 pixelsHigh: h
311 bitsPerSample: 8
312 samplesPerPixel: 3
313 hasAlpha: NO
314 isPlanar: NO
315 colorSpaceName: NSCalibratedRGBColorSpace
316 bytesPerRow: 0 // NOTE: Contrary to Apple documentation Mac OS
317 // 10.4 will add padding bytes when 0 is used here
318 bitsPerPixel: 0] autorelease]);
319
320 wxLogTrace(wxTRACE_COCOA,wxT("M_BITMAPDATA=%p NSBitmapImageRep bitmapData=%p"), M_BITMAPDATA, [M_BITMAPDATA->m_cocoaNSBitmapImageRep bitmapData]);
321 M_BITMAPDATA->m_ok = true;
322 M_BITMAPDATA->m_numColors = 0;
323 M_BITMAPDATA->m_quality = 0;
324 M_BITMAPDATA->m_bitmapMask = NULL;
325
326 return M_BITMAPDATA->m_ok;
327 }
328
329 bool wxBitmap::LoadFile(const wxString& filename, wxBitmapType type)
330 {
331 wxAutoNSAutoreleasePool pool;
332 UnRef();
333
334 m_refData = new wxBitmapRefData;
335
336 NSBitmapImageRep *imageRep = [NSBitmapImageRep
337 imageRepWithContentsOfFile:wxNSStringWithWxString(filename)];
338
339 if(imageRep)
340 {
341 M_BITMAPDATA->m_width = [imageRep pixelsWide];
342 M_BITMAPDATA->m_height = [imageRep pixelsHigh];
343 M_BITMAPDATA->m_depth = 24; // FIXME
344 M_BITMAPDATA->m_ok = true;
345 M_BITMAPDATA->m_numColors = 0;
346 M_BITMAPDATA->m_quality = 0;
347 M_BITMAPDATA->m_cocoaNSBitmapImageRep = wxGCSafeRetain(imageRep);
348 M_BITMAPDATA->m_bitmapMask = NULL;
349 return true;
350 }
351 wxImage image;
352 if(!image.LoadFile(filename,type))
353 return false;
354 if(!image.IsOk())
355 return false;
356 *this = wxBitmap(image);
357 return true;
358 }
359
360 bool wxBitmap::Create(NSImage* cocoaNSImage)
361 {
362 wxAutoNSAutoreleasePool pool;
363 NSBitmapImageRep *bitmapImageRep = [NSBitmapImageRep imageRepWithData:[cocoaNSImage TIFFRepresentation]];
364 return Create(bitmapImageRep);
365 }
366
367 bool wxBitmap::Create(NSBitmapImageRep *imageRep)
368 {
369 UnRef();
370 m_refData = new wxBitmapRefData;
371 if(imageRep != nil)
372 {
373 M_BITMAPDATA->m_width = [imageRep pixelsWide];
374 M_BITMAPDATA->m_height = [imageRep pixelsHigh];
375 M_BITMAPDATA->m_depth = [imageRep bitsPerPixel];
376 M_BITMAPDATA->m_ok = true;
377 M_BITMAPDATA->m_numColors = 0;
378 M_BITMAPDATA->m_quality = 0;
379 M_BITMAPDATA->m_cocoaNSBitmapImageRep = wxGCSafeRetain(imageRep);
380 M_BITMAPDATA->m_bitmapMask = NULL;
381 return true;
382 }
383 else
384 return false;
385 }
386
387 bool wxBitmap::Create(const void* data, wxBitmapType type, int width, int height, int depth)
388 {
389 UnRef();
390
391 m_refData = new wxBitmapRefData;
392
393 return false;
394 }
395
396 bool wxBitmap::SaveFile(const wxString& filename, wxBitmapType type, const wxPalette *palette) const
397 {
398 return false;
399 }
400
401 bool wxBitmap::CopyFromIcon(const wxIcon& icon)
402 {
403 // Pool here due to lack of one during wx init phase
404 wxAutoNSAutoreleasePool pool;
405
406 UnRef();
407 if(!icon.GetNSImage());
408 [icon.GetNSImage() lockFocus];
409 NSRect imageRect;
410 imageRect.origin.x = imageRect.origin.y = 0.0;
411 imageRect.size = [icon.GetNSImage() size];
412 NSBitmapImageRep *newBitmapRep = [[[NSBitmapImageRep alloc] initWithFocusedViewRect:imageRect] autorelease];
413 [icon.GetNSImage() unlockFocus];
414 if(!newBitmapRep)
415 return false;
416 m_refData = new wxBitmapRefData;
417 M_BITMAPDATA->m_cocoaNSBitmapImageRep = wxGCSafeRetain(newBitmapRep);
418 M_BITMAPDATA->m_width = [newBitmapRep pixelsWide];
419 M_BITMAPDATA->m_height = [newBitmapRep pixelsHigh];
420 M_BITMAPDATA->m_depth = [newBitmapRep bitsPerSample]*[newBitmapRep samplesPerPixel];
421 M_BITMAPDATA->m_ok = true;
422 M_BITMAPDATA->m_numColors = 0;
423 M_BITMAPDATA->m_quality = 0;
424 M_BITMAPDATA->m_bitmapMask = NULL;
425 return true;
426 }
427
428 wxBitmap wxBitmap::GetSubBitmap(const wxRect& rect) const
429 {
430 wxAutoNSAutoreleasePool pool;
431 if(!IsOk())
432 return wxNullBitmap;
433 NSImage *nsimage = GetNSImage(false);
434
435 [nsimage lockFocus];
436 NSRect imageRect = {{0,0}, [nsimage size]};
437 imageRect.origin.x = imageRect.size.width * rect.x / GetWidth();
438 imageRect.origin.y = imageRect.size.height * rect.y / GetHeight();
439 imageRect.size.width *= static_cast<CGFloat>(rect.width) / GetWidth();
440 imageRect.size.height *= static_cast<CGFloat>(rect.height) / GetHeight();
441
442 NSBitmapImageRep *newBitmapRep = [[NSBitmapImageRep alloc] initWithFocusedViewRect:imageRect];
443 [nsimage unlockFocus];
444
445 wxBitmap newBitmap(newBitmapRep);
446
447 return (newBitmap);
448 }
449
450 wxImage wxBitmap::ConvertToImage() const
451 {
452 wxAutoNSAutoreleasePool pool;
453 if(!IsOk())
454 return /*wxImage(5,5)*/wxNullImage;
455 NSImage *nsimage = GetNSImage(false /* don't use mask */);
456 wxImage newImage(M_BITMAPDATA->m_width,M_BITMAPDATA->m_height);
457 [nsimage lockFocus];
458 for(int i=0; i < M_BITMAPDATA->m_width; i++)
459 {
460 // Don't let the pool get too big as you'll notice we're creating
461 // two autoreleased NSColor objects with every iteration.
462 wxAutoNSAutoreleasePool loopPool;
463 for(int j=0; j < M_BITMAPDATA->m_height; j++)
464 {
465 NSColor *pixelColor = NSReadPixel(NSMakePoint(i,M_BITMAPDATA->m_height - j - 1));
466 NSColor *color = [pixelColor colorUsingColorSpaceName:NSCalibratedRGBColorSpace];
467 newImage.SetRGB(i,j,int([color redComponent]*255.0), int([color greenComponent]*255.0), int([color blueComponent]*255.0));
468 }
469 }
470 [nsimage unlockFocus];
471 return newImage;
472 }
473
474 bool wxBitmap::CreateFromImage(const wxImage& image, int depth)
475 {
476 wxAutoNSAutoreleasePool pool;
477 UnRef();
478
479 wxCHECK_MSG(image.IsOk(), false, wxT("invalid image"));
480 wxCHECK_MSG(depth == -1 || depth == 1, false, wxT("invalid bitmap depth"));
481
482 m_refData = new wxBitmapRefData();
483
484 M_BITMAPDATA->m_width = image.GetWidth();
485 M_BITMAPDATA->m_height = image.GetHeight();
486 NSBitmapImageRep *bitmapImage = [[[NSBitmapImageRep alloc]
487 initWithBitmapDataPlanes: NULL
488 pixelsWide: image.GetWidth()
489 pixelsHigh: image.GetHeight()
490 bitsPerSample: 8
491 samplesPerPixel: 3
492 hasAlpha: NO
493 isPlanar: NO
494 colorSpaceName: NSCalibratedRGBColorSpace
495 bytesPerRow: image.GetWidth()*3
496 bitsPerPixel: 0] autorelease];
497
498 // TODO: Specify bytesPerRow:0 and then use [bitmapImage bytesPerRow]
499 // so that the rows are aligned suitably for altivec by the OS (Tiger)
500 const int numBytes = image.GetWidth()*image.GetHeight()*3;
501 memcpy([bitmapImage bitmapData], image.GetData(), numBytes);
502 // TODO: Alpha and convert to desired depth
503 M_BITMAPDATA->m_depth = 24;
504 M_BITMAPDATA->m_ok = true;
505 M_BITMAPDATA->m_numColors = 0;
506 M_BITMAPDATA->m_quality = 0;
507 M_BITMAPDATA->m_cocoaNSBitmapImageRep = wxGCSafeRetain(bitmapImage);
508 M_BITMAPDATA->m_bitmapMask = new wxMask(*this,wxColour(image.GetMaskRed(),image.GetMaskGreen(),image.GetMaskBlue()));
509 return true;
510 }
511
512 void *wxBitmap::GetRawData(wxPixelDataBase& data, int bpp)
513 {
514 if(!IsOk())
515 return NULL;
516
517 NSBitmapImageRep *bitmapRep = M_BITMAPDATA->m_cocoaNSBitmapImageRep;
518 if(!bitmapRep)
519 return NULL;
520
521 if([bitmapRep bitsPerPixel]!=bpp)
522 {
523 wxFAIL_MSG( wxT("incorrect bitmap type in wxBitmap::GetRawData()") );
524 return NULL;
525 }
526 data.m_width = [bitmapRep pixelsWide];
527 data.m_height = [bitmapRep pixelsHigh];
528 data.m_stride = [bitmapRep bytesPerRow];
529 return [bitmapRep bitmapData];
530
531 // NOTE: It is up to the user to make sure they used the proper
532 // pixel format class that details what is actually inside the pixels
533 // We can only check to make sure that the total number of bits per
534 // pixel are being iterated over properly
535 // NSBitmapImageRep can contain grayscale or CMYK data and
536 // wxPixelDataBase doesn't really define the color format
537 }
538
539 void wxBitmap::UngetRawData(wxPixelDataBase& data)
540 { // TODO
541 }
542
543 // ========================================================================
544 // wxMask
545 // ========================================================================
546
547 IMPLEMENT_DYNAMIC_CLASS(wxMask, wxObject)
548
549 wxMask::wxMask()
550 {
551 m_cocoaNSBitmapImageRep = nil;
552 }
553
554 // Construct a mask from a bitmap and a colour indicating
555 // the transparent area
556 wxMask::wxMask(const wxBitmap& bitmap, const wxColour& colour)
557 {
558 m_cocoaNSBitmapImageRep = nil;
559 Create(bitmap, colour);
560 }
561
562 // Construct a mask from a bitmap and a palette index indicating
563 // the transparent area
564 wxMask::wxMask(const wxBitmap& bitmap, int paletteIndex)
565 {
566 m_cocoaNSBitmapImageRep = nil;
567
568 Create(bitmap, paletteIndex);
569 }
570
571 // Construct a mask from a mono bitmap (copies the bitmap).
572 wxMask::wxMask(const wxBitmap& bitmap)
573 {
574 m_cocoaNSBitmapImageRep = nil;
575
576 Create(bitmap);
577 }
578
579 // Copy constructor
580 wxMask::wxMask(const wxMask& src)
581 : wxObject(src)
582 , m_cocoaNSBitmapImageRep(wxGCSafeRetain(src.m_cocoaNSBitmapImageRep))
583 {
584 }
585
586 wxMask::~wxMask()
587 {
588 wxGCSafeRelease(m_cocoaNSBitmapImageRep);
589 }
590
591 // Create a mask from a mono bitmap (copies the bitmap).
592 bool wxMask::Create(const wxBitmap& bitmap)
593 {
594 // TODO
595 wxLogDebug(wxT("Cannot yet create a mask from a mono bitmap"));
596 return FALSE;
597 }
598
599 // Create a mask from a bitmap and a palette index indicating
600 // the transparent area
601 bool wxMask::Create(const wxBitmap& bitmap, int paletteIndex)
602 {
603 // TODO
604 wxLogDebug(wxT("Cannot yet create a mask from a palette bitmap"));
605 return FALSE;
606 }
607
608 template <typename PixelData>
609 static bool wxMask_CreateFromBitmapData(PixelData srcData, const wxColour& colour, unsigned char *dstData)
610 {
611 wxCHECK_MSG(dstData,false,wxT("Couldn't access mask data"));
612 typename PixelData::Iterator p(srcData);
613 const int nRows = srcData.GetHeight();
614 const int nCols = srcData.GetWidth();
615 // Total number of bytes per destination column
616 const int dstRowLength = (nCols+7)/8;
617 // Number of source columns that fit into a byte in the destination
618 const int width_aligned = nCols/8*8;
619 for(int y=0; y<nRows; ++y)
620 {
621 typename PixelData::Iterator rowStart(p);
622 unsigned char *dstRow = dstData + y*dstRowLength;
623 for(int x=0; x<width_aligned; x+=8)
624 {
625 unsigned char *dstByte = dstRow + x/8;
626 *dstByte = 0;
627 // Take source RGB, compare it with the wxColour
628 for(int j=0; j<8; ++j, ++p)
629 {
630 *dstByte +=
631 ( p.Red()!=colour.Red()
632 || p.Green()!=colour.Green()
633 || p.Blue()!=colour.Blue()
634 ) << (7-j);
635 }
636 }
637 // Handle the remaining 0-7 pixels in the row
638 unsigned char *dstByte = dstRow + width_aligned/8;
639 if(nCols%8>0)
640 *dstByte = 0;
641 for(int j=0; j<(nCols%8); ++j, ++p)
642 {
643 *dstByte +=
644 ( p.Red()!=colour.Red()
645 || p.Green()!=colour.Green()
646 || p.Blue()!=colour.Blue()
647 ) << (7-j);
648 }
649 p = rowStart;
650 p.OffsetY(srcData,1);
651 }
652 return true;
653 }
654
655 // Create a mask from a bitmap and a colour indicating
656 // the transparent area
657 bool wxMask::Create(const wxBitmap& bitmap, const wxColour& colour)
658 {
659 wxAutoNSAutoreleasePool pool;
660 if(!bitmap.IsOk())
661 return false;
662 int bmpWidth = bitmap.GetWidth();
663 int bmpHeight = bitmap.GetHeight();
664 int dstRowLength = (bmpWidth+7)/8;
665
666 // Create a bitmap image rep with 1-bit per pixel data representing
667 // the alpha channel padded such that rows end on byte boundaries
668 // Since NSBitmapImageRep doesn't have any sort of NSNullColorSpace
669 // we must have at least one channel of non-alpha data. In order to
670 // make our life easy, we use planar data which results in two
671 // separate arrays. We don't need to touch the first because it
672 // should never be used. The second is the 1-bit "alpha" data.
673 NSBitmapImageRep *maskRep = [[[NSBitmapImageRep alloc]
674 initWithBitmapDataPlanes:NULL pixelsWide:bmpWidth
675 pixelsHigh:bmpHeight bitsPerSample:1
676 samplesPerPixel:2 hasAlpha:YES isPlanar:YES
677 colorSpaceName:NSCalibratedWhiteColorSpace
678 bytesPerRow:dstRowLength bitsPerPixel:1] autorelease];
679 wxCHECK(maskRep,false);
680
681 // We need the source NSBitmapImageRep to detemine its pixel format
682 NSBitmapImageRep *srcBitmapRep = const_cast<wxBitmap&>(bitmap).GetNSBitmapImageRep();
683 wxCHECK_MSG(srcBitmapRep,false,wxT("Can't create mask for an uninitialized bitmap"));
684
685 // Get a pointer to the destination data
686 unsigned char *dstPlanes[5] = {NULL,NULL,NULL,NULL,NULL};
687 [maskRep getBitmapDataPlanes:dstPlanes];
688 unsigned char *dstData = dstPlanes[1];
689 // The wxImage format (which we use whenever we imported from wxImage)
690 if([srcBitmapRep bitsPerPixel]==24 && [srcBitmapRep bitsPerSample]==8 && [srcBitmapRep samplesPerPixel]==3 && [srcBitmapRep hasAlpha]==NO)
691 {
692 wxPixelData<wxBitmap,wxNativePixelFormat> pixelData(const_cast<wxBitmap&>(bitmap));
693 wxCHECK_MSG(wxMask_CreateFromBitmapData(pixelData, colour, dstData),
694 false, wxT("Unable to access raw data"));
695 }
696 // 32-bpp RGBx (x=throw away, no alpha)
697 else if([srcBitmapRep bitsPerPixel]==32 && [srcBitmapRep bitsPerSample]==8 && [srcBitmapRep samplesPerPixel]==3 && [srcBitmapRep hasAlpha]==NO)
698 {
699 typedef wxPixelFormat<unsigned char,32,0,1,2> PixelFormat;
700 wxPixelData<wxBitmap,PixelFormat> pixelData(const_cast<wxBitmap&>(bitmap));
701 wxCHECK_MSG(wxMask_CreateFromBitmapData(pixelData, colour, dstData),
702 false, wxT("Unable to access raw data"));
703 }
704 // 32-bpp RGBA
705 else if([srcBitmapRep bitsPerPixel]==32 && [srcBitmapRep bitsPerSample]==8 && [srcBitmapRep samplesPerPixel]==4 && [srcBitmapRep hasAlpha]==YES)
706 {
707 wxPixelData<wxBitmap,wxAlphaPixelFormat> pixelData(const_cast<wxBitmap&>(bitmap));
708 wxCHECK_MSG(wxMask_CreateFromBitmapData(pixelData, colour, dstData),
709 false, wxT("Unable to access raw data"));
710 }
711 else if([srcBitmapRep bitsPerPixel]==8 && [srcBitmapRep bitsPerSample]==8 && [srcBitmapRep samplesPerPixel]==1 && [srcBitmapRep hasAlpha]==NO)
712 // 8-bpp Grayscale, no alpha
713 { // Force all RGB to access the same grayscale component
714 typedef wxPixelFormat<unsigned char,8,0,0,0> PixelFormat;
715 wxPixelData<wxBitmap,PixelFormat> pixelData(const_cast<wxBitmap&>(bitmap));
716 wxCHECK_MSG(wxMask_CreateFromBitmapData(pixelData, colour, dstData),
717 false, wxT("Unable to access raw data"));
718 }
719 else
720 { wxCHECK_MSG(false,false,wxT("Unimplemented pixel format")); }
721
722 // maskRep was autoreleased in case we had to exit quickly
723 m_cocoaNSBitmapImageRep = wxGCSafeRetain(maskRep);
724 return true;
725 }