blob: 0677b7bc304399b4aa6ffebd56992d16c99252fd [file] [log] [blame]
epoger@google.comec3ed6a2011-07-28 14:26:00 +00001
2/*
3 * Copyright 2011 Google Inc.
4 *
5 * Use of this source code is governed by a BSD-style license that can be
6 * found in the LICENSE file.
7 */
reed@android.com0d55f1e2008-12-18 19:26:11 +00008#include "SkCGUtils.h"
reed@android.com758b1292008-12-18 17:54:12 +00009#include "SkBitmap.h"
reed@android.coma545a552009-06-29 17:07:19 +000010#include "SkColorPriv.h"
reed@android.com758b1292008-12-18 17:54:12 +000011
reed@android.com2b26cac2008-12-22 02:33:11 +000012static void SkBitmap_ReleaseInfo(void* info, const void* pixelData, size_t size) {
reed@android.com758b1292008-12-18 17:54:12 +000013 SkBitmap* bitmap = reinterpret_cast<SkBitmap*>(info);
14 delete bitmap;
15}
16
rmistry@google.comd6176b02012-08-23 18:14:13 +000017static bool getBitmapInfo(const SkBitmap& bm,
reed@google.com9c16bc02011-06-28 21:52:34 +000018 size_t* bitsPerComponent,
19 CGBitmapInfo* info,
20 bool* upscaleTo32) {
21 if (upscaleTo32) {
22 *upscaleTo32 = false;
23 }
rmistry@google.comd6176b02012-08-23 18:14:13 +000024
reed@android.com758b1292008-12-18 17:54:12 +000025 switch (bm.config()) {
reed@android.com32a42492009-07-10 03:33:52 +000026 case SkBitmap::kRGB_565_Config:
reed@google.com9c16bc02011-06-28 21:52:34 +000027 if (upscaleTo32) {
28 *upscaleTo32 = true;
29 }
reed@android.com32a42492009-07-10 03:33:52 +000030 // fall through
reed@android.com758b1292008-12-18 17:54:12 +000031 case SkBitmap::kARGB_8888_Config:
32 *bitsPerComponent = 8;
commit-bot@chromium.org5dc14c12013-07-15 13:00:45 +000033#if SK_PMCOLOR_BYTE_ORDER(R,G,B,A)
scroggo@google.com426648e2012-11-12 16:18:43 +000034 *info = kCGBitmapByteOrder32Big;
35 if (bm.isOpaque()) {
36 *info |= kCGImageAlphaNoneSkipLast;
37 } else {
38 *info |= kCGImageAlphaPremultipliedLast;
39 }
commit-bot@chromium.org5dc14c12013-07-15 13:00:45 +000040#elif SK_PMCOLOR_BYTE_ORDER(B,G,R,A)
reed@android.com8ede4922009-06-22 20:04:33 +000041 // Matches the CGBitmapInfo that Apple recommends for best
42 // performance, used by google chrome.
scroggo@google.com426648e2012-11-12 16:18:43 +000043 *info = kCGBitmapByteOrder32Little;
44 if (bm.isOpaque()) {
45 *info |= kCGImageAlphaNoneSkipFirst;
46 } else {
47 *info |= kCGImageAlphaPremultipliedFirst;
48 }
reed@android.com8ede4922009-06-22 20:04:33 +000049#else
reed@google.com9c16bc02011-06-28 21:52:34 +000050 // ...add more formats as required...
reed@android.com8ede4922009-06-22 20:04:33 +000051#warning Cannot convert SkBitmap to CGImageRef with these shiftmasks. \
reed@google.com9c16bc02011-06-28 21:52:34 +000052This will probably not work.
reed@android.com8ede4922009-06-22 20:04:33 +000053 // Legacy behavior. Perhaps turn this into an error at some
54 // point.
scroggo@google.com426648e2012-11-12 16:18:43 +000055 *info = kCGBitmapByteOrder32Big;
56 if (bm.isOpaque()) {
57 *info |= kCGImageAlphaNoneSkipLast;
58 } else {
59 *info |= kCGImageAlphaPremultipliedLast;
60 }
reed@android.com8ede4922009-06-22 20:04:33 +000061#endif
reed@android.com758b1292008-12-18 17:54:12 +000062 break;
reed@android.com32a42492009-07-10 03:33:52 +000063#if 0
reed@android.com0680d6c2008-12-19 19:46:15 +000064 case SkBitmap::kRGB_565_Config:
65 // doesn't see quite right. Are they thinking 1555?
66 *bitsPerComponent = 5;
scroggo@google.com426648e2012-11-12 16:18:43 +000067 *info = kCGBitmapByteOrder16Little | kCGImageAlphaNone;
reed@android.com0680d6c2008-12-19 19:46:15 +000068 break;
reed@android.com32a42492009-07-10 03:33:52 +000069#endif
reed@android.com0680d6c2008-12-19 19:46:15 +000070 case SkBitmap::kARGB_4444_Config:
71 *bitsPerComponent = 4;
scroggo@google.com426648e2012-11-12 16:18:43 +000072 *info = kCGBitmapByteOrder16Little;
73 if (bm.isOpaque()) {
74 *info |= kCGImageAlphaNoneSkipLast;
75 } else {
76 *info |= kCGImageAlphaPremultipliedLast;
77 }
reed@android.com0680d6c2008-12-19 19:46:15 +000078 break;
reed@android.com758b1292008-12-18 17:54:12 +000079 default:
reed@google.com9c16bc02011-06-28 21:52:34 +000080 return false;
81 }
82 return true;
83}
84
85static SkBitmap* prepareForImageRef(const SkBitmap& bm,
86 size_t* bitsPerComponent,
87 CGBitmapInfo* info) {
88 bool upscaleTo32;
89 if (!getBitmapInfo(bm, bitsPerComponent, info, &upscaleTo32)) {
90 return NULL;
reed@android.com758b1292008-12-18 17:54:12 +000091 }
92
reed@android.com32a42492009-07-10 03:33:52 +000093 SkBitmap* copy;
94 if (upscaleTo32) {
95 copy = new SkBitmap;
96 // here we make a ceep copy of the pixels, since CG won't take our
97 // 565 directly
98 bm.copyTo(copy, SkBitmap::kARGB_8888_Config);
99 } else {
100 copy = new SkBitmap(bm);
101 }
102 return copy;
reed@android.com758b1292008-12-18 17:54:12 +0000103}
104
reed@android.com38669c12011-01-03 13:48:50 +0000105CGImageRef SkCreateCGImageRefWithColorspace(const SkBitmap& bm,
106 CGColorSpaceRef colorSpace) {
107 size_t bitsPerComponent SK_INIT_TO_AVOID_WARNING;
108 CGBitmapInfo info SK_INIT_TO_AVOID_WARNING;
reed@android.com758b1292008-12-18 17:54:12 +0000109
110 SkBitmap* bitmap = prepareForImageRef(bm, &bitsPerComponent, &info);
111 if (NULL == bitmap) {
112 return NULL;
113 }
114
115 const int w = bitmap->width();
116 const int h = bitmap->height();
117 const size_t s = bitmap->getSize();
118
reed@android.com758b1292008-12-18 17:54:12 +0000119 // our provider "owns" the bitmap*, and will take care of deleting it
rmistry@google.comd6176b02012-08-23 18:14:13 +0000120 // we initially lock it, so we can access the pixels. The bitmap will be deleted in the release
121 // proc, which will in turn unlock the pixels
122 bitmap->lockPixels();
reed@android.com2b26cac2008-12-22 02:33:11 +0000123 CGDataProviderRef dataRef = CGDataProviderCreateWithData(bitmap, bitmap->getPixels(), s,
rmistry@google.comd6176b02012-08-23 18:14:13 +0000124 SkBitmap_ReleaseInfo);
reed@android.com2b26cac2008-12-22 02:33:11 +0000125
reed@android.com38669c12011-01-03 13:48:50 +0000126 bool releaseColorSpace = false;
127 if (NULL == colorSpace) {
reed@google.comc280d112011-01-05 16:07:35 +0000128 colorSpace = CGColorSpaceCreateDeviceRGB();
reed@android.com38669c12011-01-03 13:48:50 +0000129 releaseColorSpace = true;
130 }
131
reed@android.com758b1292008-12-18 17:54:12 +0000132 CGImageRef ref = CGImageCreate(w, h, bitsPerComponent,
133 bitmap->bytesPerPixel() * 8,
reed@android.com38669c12011-01-03 13:48:50 +0000134 bitmap->rowBytes(), colorSpace, info, dataRef,
reed@android.com758b1292008-12-18 17:54:12 +0000135 NULL, false, kCGRenderingIntentDefault);
reed@android.com38669c12011-01-03 13:48:50 +0000136
137 if (releaseColorSpace) {
138 CGColorSpaceRelease(colorSpace);
139 }
reed@android.com758b1292008-12-18 17:54:12 +0000140 CGDataProviderRelease(dataRef);
141 return ref;
142}
143
reed@android.comf2b98d62010-12-20 18:26:13 +0000144void SkCGDrawBitmap(CGContextRef cg, const SkBitmap& bm, float x, float y) {
145 CGImageRef img = SkCreateCGImageRef(bm);
146
147 if (img) {
148 CGRect r = CGRectMake(0, 0, bm.width(), bm.height());
reed@google.com62f46592011-01-05 15:49:43 +0000149
reed@android.comf2b98d62010-12-20 18:26:13 +0000150 CGContextSaveGState(cg);
151 CGContextTranslateCTM(cg, x, r.size.height + y);
152 CGContextScaleCTM(cg, 1, -1);
reed@google.com62f46592011-01-05 15:49:43 +0000153
reed@android.comf2b98d62010-12-20 18:26:13 +0000154 CGContextDrawImage(cg, r, img);
reed@google.com62f46592011-01-05 15:49:43 +0000155
reed@android.comf2b98d62010-12-20 18:26:13 +0000156 CGContextRestoreGState(cg);
reed@google.com62f46592011-01-05 15:49:43 +0000157
reed@android.comf2b98d62010-12-20 18:26:13 +0000158 CGImageRelease(img);
159 }
160}
161
reed@google.com292ade62011-06-28 20:54:03 +0000162///////////////////////////////////////////////////////////////////////////////
reed@android.comf2b98d62010-12-20 18:26:13 +0000163
reed@google.com292ade62011-06-28 20:54:03 +0000164#include "SkStream.h"
165
166class SkAutoPDFRelease {
167public:
168 SkAutoPDFRelease(CGPDFDocumentRef doc) : fDoc(doc) {}
169 ~SkAutoPDFRelease() {
170 if (fDoc) {
171 CGPDFDocumentRelease(fDoc);
172 }
173 }
174private:
175 CGPDFDocumentRef fDoc;
176};
commit-bot@chromium.orge61a86c2013-11-18 16:03:59 +0000177#define SkAutoPDFRelease(...) SK_REQUIRE_LOCAL_VAR(SkAutoPDFRelease)
reed@google.com292ade62011-06-28 20:54:03 +0000178
179static void CGDataProviderReleaseData_FromMalloc(void*, const void* data,
180 size_t size) {
181 sk_free((void*)data);
182}
183
184bool SkPDFDocumentToBitmap(SkStream* stream, SkBitmap* output) {
185 size_t size = stream->getLength();
186 void* ptr = sk_malloc_throw(size);
187 stream->read(ptr, size);
188 CGDataProviderRef data = CGDataProviderCreateWithData(NULL, ptr, size,
189 CGDataProviderReleaseData_FromMalloc);
190 if (NULL == data) {
191 return false;
192 }
rmistry@google.comd6176b02012-08-23 18:14:13 +0000193
reed@google.com292ade62011-06-28 20:54:03 +0000194 CGPDFDocumentRef pdf = CGPDFDocumentCreateWithProvider(data);
195 CGDataProviderRelease(data);
196 if (NULL == pdf) {
197 return false;
198 }
199 SkAutoPDFRelease releaseMe(pdf);
200
201 CGPDFPageRef page = CGPDFDocumentGetPage(pdf, 1);
202 if (NULL == page) {
203 return false;
204 }
rmistry@google.comd6176b02012-08-23 18:14:13 +0000205
reed@google.com292ade62011-06-28 20:54:03 +0000206 CGRect bounds = CGPDFPageGetBoxRect(page, kCGPDFMediaBox);
rmistry@google.comd6176b02012-08-23 18:14:13 +0000207
reed@google.com292ade62011-06-28 20:54:03 +0000208 int w = (int)CGRectGetWidth(bounds);
209 int h = (int)CGRectGetHeight(bounds);
rmistry@google.comd6176b02012-08-23 18:14:13 +0000210
reed@google.com292ade62011-06-28 20:54:03 +0000211 SkBitmap bitmap;
212 bitmap.setConfig(SkBitmap::kARGB_8888_Config, w, h);
213 bitmap.allocPixels();
214 bitmap.eraseColor(SK_ColorWHITE);
reed@google.com9c16bc02011-06-28 21:52:34 +0000215
216 size_t bitsPerComponent;
217 CGBitmapInfo info;
rmistry@google.comd6176b02012-08-23 18:14:13 +0000218 getBitmapInfo(bitmap, &bitsPerComponent, &info, NULL);
reed@google.comcd88d7c2011-06-29 20:12:38 +0000219
220 CGColorSpaceRef cs = CGColorSpaceCreateDeviceRGB();
221 CGContextRef ctx = CGBitmapContextCreate(bitmap.getPixels(), w, h,
222 bitsPerComponent, bitmap.rowBytes(),
223 cs, info);
224 CGColorSpaceRelease(cs);
225
reed@google.com292ade62011-06-28 20:54:03 +0000226 if (ctx) {
227 CGContextDrawPDFPage(ctx, page);
228 CGContextRelease(ctx);
229 }
230
231 output->swap(bitmap);
232 return true;
233}