epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2008 The Android Open Source Project |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license that can be |
| 5 | * found in the LICENSE file. |
| 6 | */ |
| 7 | |
scroggo@google.com | daaea2d | 2013-06-14 20:39:48 +0000 | [diff] [blame] | 8 | #include "SkCGUtils.h" |
reed@google.com | 2a7579d | 2012-11-07 18:30:18 +0000 | [diff] [blame] | 9 | #include "SkColorPriv.h" |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 10 | #include "SkImageDecoder.h" |
reed@android.com | b08eb2b | 2009-01-06 20:16:26 +0000 | [diff] [blame] | 11 | #include "SkImageEncoder.h" |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 12 | #include "SkMovie.h" |
| 13 | #include "SkStream.h" |
scroggo@google.com | 1a2e71d | 2013-08-28 19:13:13 +0000 | [diff] [blame] | 14 | #include "SkStreamHelpers.h" |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 15 | #include "SkTemplates.h" |
scroggo@google.com | daaea2d | 2013-06-14 20:39:48 +0000 | [diff] [blame] | 16 | #include "SkUnPreMultiply.h" |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 17 | |
yangsu@google.com | 900d877 | 2011-06-24 18:56:00 +0000 | [diff] [blame] | 18 | #ifdef SK_BUILD_FOR_MAC |
| 19 | #include <ApplicationServices/ApplicationServices.h> |
| 20 | #endif |
| 21 | |
| 22 | #ifdef SK_BUILD_FOR_IOS |
| 23 | #include <CoreGraphics/CoreGraphics.h> |
caryclark@google.com | 35f5ac9 | 2012-09-18 15:41:18 +0000 | [diff] [blame] | 24 | #include <ImageIO/ImageIO.h> |
caryclark@google.com | 594dd3c | 2012-09-24 19:33:57 +0000 | [diff] [blame] | 25 | #include <MobileCoreServices/MobileCoreServices.h> |
yangsu@google.com | 900d877 | 2011-06-24 18:56:00 +0000 | [diff] [blame] | 26 | #endif |
yangsu@google.com | c134f39 | 2011-06-23 22:27:30 +0000 | [diff] [blame] | 27 | |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 28 | static void malloc_release_proc(void* info, const void* data, size_t size) { |
| 29 | sk_free(info); |
| 30 | } |
| 31 | |
| 32 | static CGDataProviderRef SkStreamToDataProvider(SkStream* stream) { |
| 33 | // TODO: use callbacks, so we don't have to load all the data into RAM |
scroggo@google.com | 1a2e71d | 2013-08-28 19:13:13 +0000 | [diff] [blame] | 34 | SkAutoMalloc storage; |
| 35 | const size_t len = CopyStreamToStorage(&storage, stream); |
| 36 | void* data = storage.detach(); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 37 | |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 38 | return CGDataProviderCreateWithData(data, data, len, malloc_release_proc); |
| 39 | } |
| 40 | |
| 41 | static CGImageSourceRef SkStreamToCGImageSource(SkStream* stream) { |
| 42 | CGDataProviderRef data = SkStreamToDataProvider(stream); |
| 43 | CGImageSourceRef imageSrc = CGImageSourceCreateWithDataProvider(data, 0); |
| 44 | CGDataProviderRelease(data); |
| 45 | return imageSrc; |
| 46 | } |
| 47 | |
| 48 | class SkImageDecoder_CG : public SkImageDecoder { |
| 49 | protected: |
reed@android.com | 3f1f06a | 2010-03-03 21:04:12 +0000 | [diff] [blame] | 50 | virtual bool onDecode(SkStream* stream, SkBitmap* bm, Mode); |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 51 | }; |
| 52 | |
scroggo@google.com | daaea2d | 2013-06-14 20:39:48 +0000 | [diff] [blame] | 53 | // Returns an unpremultiplied version of color. It will have the same ordering and size as an |
| 54 | // SkPMColor, but the alpha will not be premultiplied. |
| 55 | static SkPMColor unpremultiply_pmcolor(SkPMColor color) { |
| 56 | U8CPU a = SkGetPackedA32(color); |
| 57 | const SkUnPreMultiply::Scale scale = SkUnPreMultiply::GetScale(a); |
| 58 | return SkPackARGB32NoCheck(a, |
| 59 | SkUnPreMultiply::ApplyScale(scale, SkGetPackedR32(color)), |
| 60 | SkUnPreMultiply::ApplyScale(scale, SkGetPackedG32(color)), |
| 61 | SkUnPreMultiply::ApplyScale(scale, SkGetPackedB32(color))); |
| 62 | } |
| 63 | |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 64 | #define BITMAP_INFO (kCGBitmapByteOrder32Big | kCGImageAlphaPremultipliedLast) |
| 65 | |
reed@android.com | 3f1f06a | 2010-03-03 21:04:12 +0000 | [diff] [blame] | 66 | bool SkImageDecoder_CG::onDecode(SkStream* stream, SkBitmap* bm, Mode mode) { |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 67 | CGImageSourceRef imageSrc = SkStreamToCGImageSource(stream); |
| 68 | |
| 69 | if (NULL == imageSrc) { |
| 70 | return false; |
| 71 | } |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 72 | SkAutoTCallVProc<const void, CFRelease> arsrc(imageSrc); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 73 | |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 74 | CGImageRef image = CGImageSourceCreateImageAtIndex(imageSrc, 0, NULL); |
| 75 | if (NULL == image) { |
| 76 | return false; |
| 77 | } |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 78 | SkAutoTCallVProc<CGImage, CGImageRelease> arimage(image); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 79 | |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 80 | const int width = CGImageGetWidth(image); |
| 81 | const int height = CGImageGetHeight(image); |
| 82 | bm->setConfig(SkBitmap::kARGB_8888_Config, width, height); |
| 83 | if (SkImageDecoder::kDecodeBounds_Mode == mode) { |
| 84 | return true; |
| 85 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 86 | |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 87 | if (!this->allocPixelRef(bm, NULL)) { |
| 88 | return false; |
| 89 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 90 | |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 91 | bm->lockPixels(); |
junov@google.com | dbfac8a | 2012-12-06 21:47:40 +0000 | [diff] [blame] | 92 | bm->eraseColor(SK_ColorTRANSPARENT); |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 93 | |
scroggo@google.com | daaea2d | 2013-06-14 20:39:48 +0000 | [diff] [blame] | 94 | CGColorSpaceRef cs = CGColorSpaceCreateDeviceRGB(); |
mike@reedtribe.org | 5bff6ec | 2012-12-15 12:35:17 +0000 | [diff] [blame] | 95 | CGContextRef cg = CGBitmapContextCreate(bm->getPixels(), width, height, 8, bm->rowBytes(), cs, BITMAP_INFO); |
scroggo@google.com | daaea2d | 2013-06-14 20:39:48 +0000 | [diff] [blame] | 96 | CFRelease(cs); |
| 97 | |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 98 | CGContextDrawImage(cg, CGRectMake(0, 0, width, height), image); |
| 99 | CGContextRelease(cg); |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 100 | |
reed@google.com | 52e91bf | 2012-11-07 22:06:55 +0000 | [diff] [blame] | 101 | CGImageAlphaInfo info = CGImageGetAlphaInfo(image); |
| 102 | switch (info) { |
| 103 | case kCGImageAlphaNone: |
| 104 | case kCGImageAlphaNoneSkipLast: |
| 105 | case kCGImageAlphaNoneSkipFirst: |
| 106 | SkASSERT(SkBitmap::ComputeIsOpaque(*bm)); |
reed@google.com | 383a697 | 2013-10-21 14:00:07 +0000 | [diff] [blame^] | 107 | bm->setAlphaType(kOpaque_SkAlphaType); |
reed@google.com | 52e91bf | 2012-11-07 22:06:55 +0000 | [diff] [blame] | 108 | break; |
| 109 | default: |
| 110 | // we don't know if we're opaque or not, so compute it. |
reed@google.com | 383a697 | 2013-10-21 14:00:07 +0000 | [diff] [blame^] | 111 | if (SkBitmap::ComputeIsOpaque(*bm)) { |
| 112 | bm->setAlphaType(kOpaque_SkAlphaType); |
| 113 | } |
reed@google.com | 52e91bf | 2012-11-07 22:06:55 +0000 | [diff] [blame] | 114 | } |
scroggo@google.com | daaea2d | 2013-06-14 20:39:48 +0000 | [diff] [blame] | 115 | if (!bm->isOpaque() && this->getRequireUnpremultipliedColors()) { |
| 116 | // CGBitmapContext does not support unpremultiplied, so the image has been premultiplied. |
| 117 | // Convert to unpremultiplied. |
| 118 | for (int i = 0; i < width; ++i) { |
| 119 | for (int j = 0; j < height; ++j) { |
| 120 | uint32_t* addr = bm->getAddr32(i, j); |
| 121 | *addr = unpremultiply_pmcolor(*addr); |
| 122 | } |
| 123 | } |
reed@google.com | 383a697 | 2013-10-21 14:00:07 +0000 | [diff] [blame^] | 124 | bm->setAlphaType(kUnpremul_SkAlphaType); |
scroggo@google.com | daaea2d | 2013-06-14 20:39:48 +0000 | [diff] [blame] | 125 | } |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 126 | bm->unlockPixels(); |
| 127 | return true; |
| 128 | } |
| 129 | |
| 130 | /////////////////////////////////////////////////////////////////////////////// |
| 131 | |
scroggo@google.com | b5571b3 | 2013-09-25 21:34:24 +0000 | [diff] [blame] | 132 | extern SkImageDecoder* image_decoder_from_stream(SkStreamRewindable*); |
scroggo@google.com | 4c6adf9 | 2013-04-17 21:07:55 +0000 | [diff] [blame] | 133 | |
scroggo@google.com | b5571b3 | 2013-09-25 21:34:24 +0000 | [diff] [blame] | 134 | SkImageDecoder* SkImageDecoder::Factory(SkStreamRewindable* stream) { |
scroggo@google.com | 4c6adf9 | 2013-04-17 21:07:55 +0000 | [diff] [blame] | 135 | SkImageDecoder* decoder = image_decoder_from_stream(stream); |
| 136 | if (NULL == decoder) { |
| 137 | // If no image decoder specific to the stream exists, use SkImageDecoder_CG. |
| 138 | return SkNEW(SkImageDecoder_CG); |
| 139 | } else { |
| 140 | return decoder; |
| 141 | } |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 142 | } |
| 143 | |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 144 | ///////////////////////////////////////////////////////////////////////// |
| 145 | |
scroggo@google.com | b5571b3 | 2013-09-25 21:34:24 +0000 | [diff] [blame] | 146 | SkMovie* SkMovie::DecodeStream(SkStreamRewindable* stream) { |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 147 | return NULL; |
| 148 | } |
| 149 | |
| 150 | ///////////////////////////////////////////////////////////////////////// |
| 151 | |
reed@android.com | 0ae6b24 | 2008-12-23 16:49:54 +0000 | [diff] [blame] | 152 | static size_t consumer_put(void* info, const void* buffer, size_t count) { |
| 153 | SkWStream* stream = reinterpret_cast<SkWStream*>(info); |
| 154 | return stream->write(buffer, count) ? count : 0; |
| 155 | } |
| 156 | |
| 157 | static void consumer_release(void* info) { |
| 158 | // we do nothing, since by design we don't "own" the stream (i.e. info) |
| 159 | } |
| 160 | |
| 161 | static CGDataConsumerRef SkStreamToCGDataConsumer(SkWStream* stream) { |
| 162 | CGDataConsumerCallbacks procs; |
| 163 | procs.putBytes = consumer_put; |
| 164 | procs.releaseConsumer = consumer_release; |
| 165 | // we don't own/reference the stream, so it our consumer must not live |
| 166 | // longer that our caller's ownership of the stream |
| 167 | return CGDataConsumerCreate(stream, &procs); |
| 168 | } |
| 169 | |
| 170 | static CGImageDestinationRef SkStreamToImageDestination(SkWStream* stream, |
| 171 | CFStringRef type) { |
| 172 | CGDataConsumerRef consumer = SkStreamToCGDataConsumer(stream); |
| 173 | if (NULL == consumer) { |
| 174 | return NULL; |
| 175 | } |
| 176 | SkAutoTCallVProc<const void, CFRelease> arconsumer(consumer); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 177 | |
reed@android.com | 0ae6b24 | 2008-12-23 16:49:54 +0000 | [diff] [blame] | 178 | return CGImageDestinationCreateWithDataConsumer(consumer, type, 1, NULL); |
| 179 | } |
| 180 | |
| 181 | class SkImageEncoder_CG : public SkImageEncoder { |
| 182 | public: |
| 183 | SkImageEncoder_CG(Type t) : fType(t) {} |
| 184 | |
| 185 | protected: |
| 186 | virtual bool onEncode(SkWStream* stream, const SkBitmap& bm, int quality); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 187 | |
reed@android.com | 0ae6b24 | 2008-12-23 16:49:54 +0000 | [diff] [blame] | 188 | private: |
| 189 | Type fType; |
| 190 | }; |
| 191 | |
reed@android.com | 0ae6b24 | 2008-12-23 16:49:54 +0000 | [diff] [blame] | 192 | /* Encode bitmaps via CGImageDestination. We setup a DataConsumer which writes |
| 193 | to our SkWStream. Since we don't reference/own the SkWStream, our consumer |
| 194 | must only live for the duration of the onEncode() method. |
| 195 | */ |
| 196 | bool SkImageEncoder_CG::onEncode(SkWStream* stream, const SkBitmap& bm, |
| 197 | int quality) { |
scroggo@google.com | 5a7c6be | 2012-10-04 21:46:08 +0000 | [diff] [blame] | 198 | // Used for converting a bitmap to 8888. |
| 199 | const SkBitmap* bmPtr = &bm; |
| 200 | SkBitmap bitmap8888; |
| 201 | |
reed@android.com | 0ae6b24 | 2008-12-23 16:49:54 +0000 | [diff] [blame] | 202 | CFStringRef type; |
| 203 | switch (fType) { |
scroggo@google.com | 4c6adf9 | 2013-04-17 21:07:55 +0000 | [diff] [blame] | 204 | case kICO_Type: |
| 205 | type = kUTTypeICO; |
| 206 | break; |
| 207 | case kBMP_Type: |
| 208 | type = kUTTypeBMP; |
| 209 | break; |
| 210 | case kGIF_Type: |
| 211 | type = kUTTypeGIF; |
| 212 | break; |
reed@android.com | 0ae6b24 | 2008-12-23 16:49:54 +0000 | [diff] [blame] | 213 | case kJPEG_Type: |
| 214 | type = kUTTypeJPEG; |
| 215 | break; |
| 216 | case kPNG_Type: |
scroggo@google.com | 5a7c6be | 2012-10-04 21:46:08 +0000 | [diff] [blame] | 217 | // PNG encoding an ARGB_4444 bitmap gives the following errors in GM: |
| 218 | // <Error>: CGImageDestinationAddImage image could not be converted to destination |
| 219 | // format. |
| 220 | // <Error>: CGImageDestinationFinalize image destination does not have enough images |
| 221 | // So instead we copy to 8888. |
| 222 | if (bm.getConfig() == SkBitmap::kARGB_4444_Config) { |
| 223 | bm.copyTo(&bitmap8888, SkBitmap::kARGB_8888_Config); |
| 224 | bmPtr = &bitmap8888; |
| 225 | } |
reed@android.com | 0ae6b24 | 2008-12-23 16:49:54 +0000 | [diff] [blame] | 226 | type = kUTTypePNG; |
| 227 | break; |
| 228 | default: |
| 229 | return false; |
| 230 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 231 | |
reed@android.com | 0ae6b24 | 2008-12-23 16:49:54 +0000 | [diff] [blame] | 232 | CGImageDestinationRef dst = SkStreamToImageDestination(stream, type); |
| 233 | if (NULL == dst) { |
| 234 | return false; |
| 235 | } |
| 236 | SkAutoTCallVProc<const void, CFRelease> ardst(dst); |
| 237 | |
scroggo@google.com | 5a7c6be | 2012-10-04 21:46:08 +0000 | [diff] [blame] | 238 | CGImageRef image = SkCreateCGImageRef(*bmPtr); |
reed@android.com | 0ae6b24 | 2008-12-23 16:49:54 +0000 | [diff] [blame] | 239 | if (NULL == image) { |
| 240 | return false; |
| 241 | } |
| 242 | SkAutoTCallVProc<CGImage, CGImageRelease> agimage(image); |
epoger@google.com | 0928c4a | 2012-01-31 15:14:08 +0000 | [diff] [blame] | 243 | |
| 244 | CGImageDestinationAddImage(dst, image, NULL); |
| 245 | return CGImageDestinationFinalize(dst); |
reed@android.com | 0ae6b24 | 2008-12-23 16:49:54 +0000 | [diff] [blame] | 246 | } |
| 247 | |
scroggo@google.com | 4c6adf9 | 2013-04-17 21:07:55 +0000 | [diff] [blame] | 248 | /////////////////////////////////////////////////////////////////////////////// |
| 249 | |
scroggo@google.com | 4c6adf9 | 2013-04-17 21:07:55 +0000 | [diff] [blame] | 250 | static SkImageEncoder* sk_imageencoder_cg_factory(SkImageEncoder::Type t) { |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 251 | switch (t) { |
scroggo@google.com | 4c6adf9 | 2013-04-17 21:07:55 +0000 | [diff] [blame] | 252 | case SkImageEncoder::kICO_Type: |
| 253 | case SkImageEncoder::kBMP_Type: |
| 254 | case SkImageEncoder::kGIF_Type: |
| 255 | case SkImageEncoder::kJPEG_Type: |
| 256 | case SkImageEncoder::kPNG_Type: |
reed@android.com | 0ae6b24 | 2008-12-23 16:49:54 +0000 | [diff] [blame] | 257 | break; |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 258 | default: |
| 259 | return NULL; |
| 260 | } |
reed@android.com | 0ae6b24 | 2008-12-23 16:49:54 +0000 | [diff] [blame] | 261 | return SkNEW_ARGS(SkImageEncoder_CG, (t)); |
reed@android.com | 0767e47 | 2008-12-23 16:06:51 +0000 | [diff] [blame] | 262 | } |
scroggo@google.com | 4c6adf9 | 2013-04-17 21:07:55 +0000 | [diff] [blame] | 263 | |
scroggo@google.com | b5571b3 | 2013-09-25 21:34:24 +0000 | [diff] [blame] | 264 | static SkImageEncoder_EncodeReg gEReg(sk_imageencoder_cg_factory); |
scroggo@google.com | 39edf4c | 2013-04-25 17:33:51 +0000 | [diff] [blame] | 265 | |
| 266 | struct FormatConversion { |
| 267 | CFStringRef fUTType; |
| 268 | SkImageDecoder::Format fFormat; |
| 269 | }; |
| 270 | |
| 271 | // Array of the types supported by the decoder. |
| 272 | static const FormatConversion gFormatConversions[] = { |
| 273 | { kUTTypeBMP, SkImageDecoder::kBMP_Format }, |
| 274 | { kUTTypeGIF, SkImageDecoder::kGIF_Format }, |
| 275 | { kUTTypeICO, SkImageDecoder::kICO_Format }, |
| 276 | { kUTTypeJPEG, SkImageDecoder::kJPEG_Format }, |
| 277 | // Also include JPEG2000 |
| 278 | { kUTTypeJPEG2000, SkImageDecoder::kJPEG_Format }, |
| 279 | { kUTTypePNG, SkImageDecoder::kPNG_Format }, |
| 280 | }; |
| 281 | |
| 282 | static SkImageDecoder::Format UTType_to_Format(const CFStringRef uttype) { |
| 283 | for (size_t i = 0; i < SK_ARRAY_COUNT(gFormatConversions); i++) { |
| 284 | if (CFStringCompare(uttype, gFormatConversions[i].fUTType, 0) == kCFCompareEqualTo) { |
| 285 | return gFormatConversions[i].fFormat; |
| 286 | } |
| 287 | } |
| 288 | return SkImageDecoder::kUnknown_Format; |
| 289 | } |
| 290 | |
scroggo@google.com | b5571b3 | 2013-09-25 21:34:24 +0000 | [diff] [blame] | 291 | static SkImageDecoder::Format get_format_cg(SkStreamRewindable* stream) { |
scroggo@google.com | 39edf4c | 2013-04-25 17:33:51 +0000 | [diff] [blame] | 292 | CGImageSourceRef imageSrc = SkStreamToCGImageSource(stream); |
| 293 | |
| 294 | if (NULL == imageSrc) { |
| 295 | return SkImageDecoder::kUnknown_Format; |
| 296 | } |
| 297 | |
| 298 | SkAutoTCallVProc<const void, CFRelease> arsrc(imageSrc); |
| 299 | const CFStringRef name = CGImageSourceGetType(imageSrc); |
| 300 | if (NULL == name) { |
| 301 | return SkImageDecoder::kUnknown_Format; |
| 302 | } |
| 303 | return UTType_to_Format(name); |
| 304 | } |
| 305 | |
scroggo@google.com | b5571b3 | 2013-09-25 21:34:24 +0000 | [diff] [blame] | 306 | static SkImageDecoder_FormatReg gFormatReg(get_format_cg); |