1 /////////////////////////////////////////////////////////////////////////////
2 // Name: src/cocoa/bitmap.mm
4 // Author: David Elliott
8 // Copyright: (c) 2003 David Elliott
9 // Licence: wxWidgets licence
10 /////////////////////////////////////////////////////////////////////////////
12 #include "wx/wxprec.h"
14 #include "wx/bitmap.h"
19 #include "wx/palette.h"
21 #include "wx/colour.h"
25 #include "wx/xpmdecod.h"
26 #include "wx/rawbmp.h"
28 #include "wx/cocoa/autorelease.h"
29 #include "wx/cocoa/string.h"
31 #import <AppKit/NSBitmapImageRep.h>
32 #import <AppKit/NSGraphics.h>
33 #import <AppKit/NSImage.h>
34 #import <AppKit/NSColor.h>
36 IMPLEMENT_ABSTRACT_CLASS(wxBitmapHandler, wxBitmapHandlerBase)
38 // ========================================================================
40 // ========================================================================
41 class wxBitmapRefData: public wxGDIRefData
43 friend class wxBitmap;
46 wxBitmapRefData( const wxBitmapRefData& data );
47 virtual ~wxBitmapRefData();
55 wxPalette m_bitmapPalette;
57 WX_NSBitmapImageRep m_cocoaNSBitmapImageRep;
58 wxMask *m_bitmapMask; // Optional mask
61 #define M_BITMAPDATA ((wxBitmapRefData *)m_refData)
63 wxBitmapRefData::wxBitmapRefData()
71 m_cocoaNSBitmapImageRep = nil;
75 wxBitmapRefData::wxBitmapRefData( const wxBitmapRefData& data)
77 m_width = data.m_width;
78 m_height = data.m_height;
79 m_depth = data.m_depth;
81 m_numColors = data.m_numColors;
82 m_bitmapPalette = data.m_bitmapPalette;
83 m_quality = data.m_quality;
84 m_cocoaNSBitmapImageRep = [data.m_cocoaNSBitmapImageRep copyWithZone:nil];
85 m_bitmapMask = data.m_bitmapMask?new wxMask(*data.m_bitmapMask):NULL;
88 wxBitmapRefData::~wxBitmapRefData()
90 [m_cocoaNSBitmapImageRep release];
91 m_cocoaNSBitmapImageRep = NULL;
97 // ========================================================================
99 // ========================================================================
100 IMPLEMENT_DYNAMIC_CLASS(wxBitmap, wxGDIObject)
107 wxBitmap::~wxBitmap()
111 wxBitmap::wxBitmap(const char bits[], int the_width, int the_height, int no_bits)
113 m_refData = new wxBitmapRefData;
115 M_BITMAPDATA->m_width = the_width ;
116 M_BITMAPDATA->m_height = the_height ;
117 M_BITMAPDATA->m_depth = no_bits ;
118 M_BITMAPDATA->m_numColors = 0;
120 /* TODO: create the bitmap from data */
123 wxBitmap::wxBitmap(int w, int h, int d)
125 (void)Create(w, h, d);
128 wxBitmap::wxBitmap(const void* data, wxBitmapType type, int width, int height, int depth)
130 (void) Create(data, type, width, height, depth);
133 wxBitmap::wxBitmap(const wxString& filename, wxBitmapType type)
135 LoadFile(filename, type);
138 wxObjectRefData *wxBitmap::CreateRefData() const
140 return new wxBitmapRefData;
143 wxObjectRefData *wxBitmap::CloneRefData(const wxObjectRefData *data) const
145 return new wxBitmapRefData(*(wxBitmapRefData*)data);
148 WX_NSBitmapImageRep wxBitmap::GetNSBitmapImageRep()
152 return M_BITMAPDATA->m_cocoaNSBitmapImageRep;
155 WX_NSImage wxBitmap::GetNSImage(bool useMask) const
159 NSImage *nsimage = [[[NSImage alloc]
160 initWithSize:NSMakeSize(GetWidth(), GetHeight())] autorelease];
163 [nsimage addRepresentation: M_BITMAPDATA->m_cocoaNSBitmapImageRep];
164 if(useMask && GetMask())
166 // Show before/after to prove that the bitmap itself is not changed
167 // even though we just composited onto the NSImage
168 wxLogTrace(wxTRACE_COCOA,wxT("Before: bpp=%d"),[M_BITMAPDATA->m_cocoaNSBitmapImageRep bitsPerPixel]);
169 NSImage *maskImage = [[NSImage alloc]
170 initWithSize:NSMakeSize(GetWidth(), GetHeight())];
171 [maskImage addRepresentation: GetMask()->GetNSBitmapImageRep()];
173 [maskImage compositeToPoint:NSZeroPoint operation:NSCompositeDestinationIn];
174 [nsimage unlockFocus];
176 wxLogTrace(wxTRACE_COCOA,wxT("After: bpp=%d"),[M_BITMAPDATA->m_cocoaNSBitmapImageRep bitsPerPixel]);
181 void wxBitmap::SetNSBitmapImageRep(WX_NSBitmapImageRep bitmapImageRep)
185 // NOTE: No checking is done to make sure width/height agree
186 [bitmapImageRep retain];
187 [M_BITMAPDATA->m_cocoaNSBitmapImageRep release];
188 M_BITMAPDATA->m_cocoaNSBitmapImageRep = bitmapImageRep;
191 void wxBitmap::SetWidth(int w)
194 m_refData = new wxBitmapRefData;
196 M_BITMAPDATA->m_width = w;
199 void wxBitmap::SetHeight(int h)
202 m_refData = new wxBitmapRefData;
204 M_BITMAPDATA->m_height = h;
207 void wxBitmap::SetDepth(int d)
210 m_refData = new wxBitmapRefData;
212 M_BITMAPDATA->m_depth = d;
215 void wxBitmap::SetQuality(int q)
218 m_refData = new wxBitmapRefData;
220 M_BITMAPDATA->m_quality = q;
223 void wxBitmap::SetOk(bool isOk)
226 m_refData = new wxBitmapRefData;
228 M_BITMAPDATA->m_ok = isOk;
231 void wxBitmap::SetPalette(const wxPalette& palette)
234 m_refData = new wxBitmapRefData;
236 M_BITMAPDATA->m_bitmapPalette = palette ;
239 void wxBitmap::SetMask(wxMask *mask)
242 m_refData = new wxBitmapRefData;
244 M_BITMAPDATA->m_bitmapMask = mask ;
247 bool wxBitmap::Ok() const
249 return m_refData && M_BITMAPDATA->m_ok;
252 wxPalette* wxBitmap::GetPalette() const
256 return &M_BITMAPDATA->m_bitmapPalette;
259 wxMask* wxBitmap::GetMask() const
263 return M_BITMAPDATA->m_bitmapMask;
266 int wxBitmap::GetDepth() const
270 return M_BITMAPDATA->m_depth;
273 int wxBitmap::GetWidth() const
277 return M_BITMAPDATA->m_width;
280 int wxBitmap::GetHeight() const
284 return M_BITMAPDATA->m_height;
287 bool wxBitmap::Create(int w, int h, int d)
291 m_refData = new wxBitmapRefData;
293 M_BITMAPDATA->m_width = w;
294 M_BITMAPDATA->m_height = h;
295 M_BITMAPDATA->m_depth = d;
297 /* TODO: create new bitmap */
298 M_BITMAPDATA->m_cocoaNSBitmapImageRep = [[NSBitmapImageRep alloc]
299 initWithBitmapDataPlanes: NULL
306 colorSpaceName: NSCalibratedRGBColorSpace
307 bytesPerRow: 0 // NOTE: Contrary to Apple documentation Mac OS
308 // 10.4 will add padding bytes when 0 is used here
311 wxLogTrace(wxTRACE_COCOA,wxT("M_BITMAPDATA=%p NSBitmapImageRep bitmapData=%p"), M_BITMAPDATA, [M_BITMAPDATA->m_cocoaNSBitmapImageRep bitmapData]);
312 M_BITMAPDATA->m_ok = true;
313 M_BITMAPDATA->m_numColors = 0;
314 M_BITMAPDATA->m_quality = 0;
315 M_BITMAPDATA->m_bitmapMask = NULL;
317 return M_BITMAPDATA->m_ok;
320 bool wxBitmap::LoadFile(const wxString& filename, wxBitmapType type)
322 wxAutoNSAutoreleasePool pool;
325 m_refData = new wxBitmapRefData;
327 NSBitmapImageRep *imageRep = [NSBitmapImageRep
328 imageRepWithContentsOfFile:wxNSStringWithWxString(filename)];
332 M_BITMAPDATA->m_width = [imageRep pixelsWide];
333 M_BITMAPDATA->m_height = [imageRep pixelsHigh];
334 M_BITMAPDATA->m_depth = 24; // FIXME
335 M_BITMAPDATA->m_ok = true;
336 M_BITMAPDATA->m_numColors = 0;
337 M_BITMAPDATA->m_quality = 0;
338 M_BITMAPDATA->m_cocoaNSBitmapImageRep = [imageRep retain];
339 M_BITMAPDATA->m_bitmapMask = NULL;
343 if(!image.LoadFile(filename,type))
347 *this = wxBitmap(image);
351 bool wxBitmap::Create(const void* data, wxBitmapType type, int width, int height, int depth)
355 m_refData = new wxBitmapRefData;
360 bool wxBitmap::SaveFile(const wxString& filename, wxBitmapType type, const wxPalette *palette) const
365 bool wxBitmap::CopyFromIcon(const wxIcon& icon)
368 if(!icon.GetNSImage());
369 [icon.GetNSImage() lockFocus];
371 imageRect.origin.x = imageRect.origin.y = 0.0;
372 imageRect.size = [icon.GetNSImage() size];
373 NSBitmapImageRep *newBitmapRep = [[NSBitmapImageRep alloc] initWithFocusedViewRect:imageRect];
374 [icon.GetNSImage() unlockFocus];
377 m_refData = new wxBitmapRefData;
378 M_BITMAPDATA->m_cocoaNSBitmapImageRep = newBitmapRep;
379 M_BITMAPDATA->m_width = [newBitmapRep pixelsWide];
380 M_BITMAPDATA->m_height = [newBitmapRep pixelsHigh];
381 M_BITMAPDATA->m_depth = [newBitmapRep bitsPerSample]*[newBitmapRep samplesPerPixel];
382 M_BITMAPDATA->m_ok = true;
383 M_BITMAPDATA->m_numColors = 0;
384 M_BITMAPDATA->m_quality = 0;
385 M_BITMAPDATA->m_bitmapMask = NULL;
389 wxBitmap wxBitmap::GetSubBitmap(wxRect const&) const
394 wxImage wxBitmap::ConvertToImage() const
396 wxAutoNSAutoreleasePool pool;
398 return /*wxImage(5,5)*/wxNullImage;
399 NSImage *nsimage = GetNSImage(false /* don't use mask */);
400 wxImage newImage(M_BITMAPDATA->m_width,M_BITMAPDATA->m_height);
402 for(int i=0; i < M_BITMAPDATA->m_width; i++)
404 // Don't let the pool get too big as you'll notice we're creating
405 // two autoreleased NSColor objects with every iteration.
406 wxAutoNSAutoreleasePool loopPool;
407 for(int j=0; j < M_BITMAPDATA->m_height; j++)
409 NSColor *pixelColor = NSReadPixel(NSMakePoint(i,M_BITMAPDATA->m_height - j - 1));
410 NSColor *color = [pixelColor colorUsingColorSpaceName:NSCalibratedRGBColorSpace];
411 newImage.SetRGB(i,j,int([color redComponent]*255.0), int([color greenComponent]*255.0), int([color blueComponent]*255.0));
414 [nsimage unlockFocus];
418 bool wxBitmap::CreateFromImage(const wxImage& image, int depth)
422 wxCHECK_MSG(image.Ok(), false, wxT("invalid image"));
423 wxCHECK_MSG(depth == -1 || depth == 1, false, wxT("invalid bitmap depth"));
425 m_refData = new wxBitmapRefData();
427 M_BITMAPDATA->m_width = image.GetWidth();
428 M_BITMAPDATA->m_height = image.GetHeight();
429 NSBitmapImageRep *bitmapImage = [[NSBitmapImageRep alloc]
430 initWithBitmapDataPlanes: NULL
431 pixelsWide: image.GetWidth()
432 pixelsHigh: image.GetHeight()
437 colorSpaceName: NSCalibratedRGBColorSpace
438 bytesPerRow: image.GetWidth()*3
441 // TODO: Specify bytesPerRow:0 and then use [bitmapImage bytesPerRow]
442 // so that the rows are aligned suitably for altivec by the OS (Tiger)
443 const int numBytes = image.GetWidth()*image.GetHeight()*3;
444 memcpy([bitmapImage bitmapData], image.GetData(), numBytes);
445 // TODO: Alpha and convert to desired depth
446 M_BITMAPDATA->m_depth = 24;
447 M_BITMAPDATA->m_ok = true;
448 M_BITMAPDATA->m_numColors = 0;
449 M_BITMAPDATA->m_quality = 0;
450 M_BITMAPDATA->m_cocoaNSBitmapImageRep = bitmapImage;
451 M_BITMAPDATA->m_bitmapMask = new wxMask(*this,wxColour(image.GetMaskRed(),image.GetMaskGreen(),image.GetMaskBlue()));
455 void *wxBitmap::GetRawData(wxPixelDataBase& data, int bpp)
460 NSBitmapImageRep *bitmapRep = M_BITMAPDATA->m_cocoaNSBitmapImageRep;
464 if([bitmapRep bitsPerPixel]!=bpp)
466 wxFAIL_MSG( _T("incorrect bitmap type in wxBitmap::GetRawData()") );
469 data.m_width = [bitmapRep pixelsWide];
470 data.m_height = [bitmapRep pixelsHigh];
471 data.m_stride = [bitmapRep bytesPerRow];
472 return [bitmapRep bitmapData];
474 // NOTE: It is up to the user to make sure they used the proper
475 // pixel format class that details what is actually inside the pixels
476 // We can only check to make sure that the total number of bits per
477 // pixel are being iterated over properly
478 // NSBitmapImageRep can contain grayscale or CMYK data and
479 // wxPixelDataBase doesn't really define the color format
482 void wxBitmap::UngetRawData(wxPixelDataBase& data)
486 void wxBitmap::UseAlpha()
490 // ========================================================================
492 // ========================================================================
494 IMPLEMENT_DYNAMIC_CLASS(wxMask, wxObject)
498 m_cocoaNSBitmapImageRep = nil;
501 // Construct a mask from a bitmap and a colour indicating
502 // the transparent area
503 wxMask::wxMask(const wxBitmap& bitmap, const wxColour& colour)
505 m_cocoaNSBitmapImageRep = nil;
506 Create(bitmap, colour);
509 // Construct a mask from a bitmap and a palette index indicating
510 // the transparent area
511 wxMask::wxMask(const wxBitmap& bitmap, int paletteIndex)
513 m_cocoaNSBitmapImageRep = nil;
515 Create(bitmap, paletteIndex);
518 // Construct a mask from a mono bitmap (copies the bitmap).
519 wxMask::wxMask(const wxBitmap& bitmap)
521 m_cocoaNSBitmapImageRep = nil;
528 [m_cocoaNSBitmapImageRep release];
531 // Create a mask from a mono bitmap (copies the bitmap).
532 bool wxMask::Create(const wxBitmap& bitmap)
538 // Create a mask from a bitmap and a palette index indicating
539 // the transparent area
540 bool wxMask::Create(const wxBitmap& bitmap, int paletteIndex)
546 template <typename PixelData>
547 static bool wxMask_CreateFromBitmapData(PixelData srcData, const wxColour& colour, unsigned char *dstData)
549 wxCHECK_MSG(dstData,false,wxT("Couldn't access mask data"));
550 typename PixelData::Iterator p(srcData);
551 const int nRows = srcData.GetHeight();
552 const int nCols = srcData.GetWidth();
553 // Total number of bytes per destination column
554 const int dstRowLength = (nCols+7)/8;
555 // Number of source columns that fit into a byte in the destination
556 const int width_aligned = nCols/8*8;
557 for(int y=0; y<nRows; ++y)
559 typename PixelData::Iterator rowStart(p);
560 unsigned char *dstRow = dstData + y*dstRowLength;
561 for(int x=0; x<width_aligned; x+=8)
563 unsigned char *dstByte = dstRow + x/8;
565 // Take source RGB, compare it with the wxColour
566 for(int j=0; j<8; ++j, ++p)
569 ( p.Red()!=colour.Red()
570 || p.Green()!=colour.Green()
571 || p.Blue()!=colour.Blue()
575 // Handle the remaining 0-7 pixels in the row
576 unsigned char *dstByte = dstRow + width_aligned/8;
579 for(int j=0; j<(nCols%8); ++j, ++p)
582 ( p.Red()!=colour.Red()
583 || p.Green()!=colour.Green()
584 || p.Blue()!=colour.Blue()
588 p.OffsetY(srcData,1);
593 // Create a mask from a bitmap and a colour indicating
594 // the transparent area
595 bool wxMask::Create(const wxBitmap& bitmap, const wxColour& colour)
597 wxAutoNSAutoreleasePool pool;
600 int bmpWidth = bitmap.GetWidth();
601 int bmpHeight = bitmap.GetHeight();
602 int dstRowLength = (bmpWidth+7)/8;
604 // Create a bitmap image rep with 1-bit per pixel data representing
605 // the alpha channel padded such that rows end on byte boundaries
606 // Since NSBitmapImageRep doesn't have any sort of NSNullColorSpace
607 // we must have at least one channel of non-alpha data. In order to
608 // make our life easy, we use planar data which results in two
609 // separate arrays. We don't need to touch the first because it
610 // should never be used. The second is the 1-bit "alpha" data.
611 NSBitmapImageRep *maskRep = [[[NSBitmapImageRep alloc]
612 initWithBitmapDataPlanes:NULL pixelsWide:bmpWidth
613 pixelsHigh:bmpHeight bitsPerSample:1
614 samplesPerPixel:2 hasAlpha:YES isPlanar:YES
615 colorSpaceName:NSCalibratedWhiteColorSpace
616 bytesPerRow:dstRowLength bitsPerPixel:1] autorelease];
617 wxCHECK(maskRep,false);
619 // We need the source NSBitmapImageRep to detemine its pixel format
620 NSBitmapImageRep *srcBitmapRep = const_cast<wxBitmap&>(bitmap).GetNSBitmapImageRep();
621 wxCHECK_MSG(srcBitmapRep,false,wxT("Can't create mask for an uninitialized bitmap"));
623 // Get a pointer to the destination data
624 unsigned char *dstPlanes[5] = {NULL,NULL,NULL,NULL,NULL};
625 [maskRep getBitmapDataPlanes:dstPlanes];
626 unsigned char *dstData = dstPlanes[1];
627 // The wxImage format (which we use whenever we imported from wxImage)
628 if([srcBitmapRep bitsPerPixel]==24 && [srcBitmapRep bitsPerSample]==8 && [srcBitmapRep samplesPerPixel]==3 && [srcBitmapRep hasAlpha]==NO)
630 wxPixelData<wxBitmap,wxNativePixelFormat> pixelData(const_cast<wxBitmap&>(bitmap));
631 wxCHECK_MSG(wxMask_CreateFromBitmapData(pixelData, colour, dstData),
632 false, wxT("Unable to access raw data"));
634 // 32-bpp RGBx (x=throw away, no alpha)
635 else if([srcBitmapRep bitsPerPixel]==32 && [srcBitmapRep bitsPerSample]==8 && [srcBitmapRep samplesPerPixel]==3 && [srcBitmapRep hasAlpha]==NO)
637 typedef wxPixelFormat<unsigned char,32,0,1,2> PixelFormat;
638 wxPixelData<wxBitmap,PixelFormat> pixelData(const_cast<wxBitmap&>(bitmap));
639 wxCHECK_MSG(wxMask_CreateFromBitmapData(pixelData, colour, dstData),
640 false, wxT("Unable to access raw data"));
643 else if([srcBitmapRep bitsPerPixel]==32 && [srcBitmapRep bitsPerSample]==8 && [srcBitmapRep samplesPerPixel]==4 && [srcBitmapRep hasAlpha]==YES)
645 wxPixelData<wxBitmap,wxAlphaPixelFormat> pixelData(const_cast<wxBitmap&>(bitmap));
646 wxCHECK_MSG(wxMask_CreateFromBitmapData(pixelData, colour, dstData),
647 false, wxT("Unable to access raw data"));
650 { wxCHECK_MSG(false,false,wxT("Unimplemented pixel format")); }
652 // maskRep was autoreleased in case we had to exit quickly
653 m_cocoaNSBitmapImageRep = [maskRep retain];