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