add wxSize overloads to wxBitmap ctors and to wxBitmap::Create
[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:     wxWidgets 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(!Ok())
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.Ok())
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(!Ok())
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(!Ok())
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.Ok(), 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(!Ok())
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( _T("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.Ok())
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 }