epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 1 | |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 2 | /* |
epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 3 | * Copyright 2007 The Android Open Source Project |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 4 | * |
epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 5 | * Use of this source code is governed by a BSD-style license that can be |
| 6 | * found in the LICENSE file. |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 7 | */ |
epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 8 | |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 9 | |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 10 | #include "bmpdecoderhelper.h" |
scroggo@google.com | dbf9f88 | 2013-08-21 15:01:48 +0000 | [diff] [blame] | 11 | #include "SkColorPriv.h" |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 12 | #include "SkImageDecoder.h" |
| 13 | #include "SkScaledBitmapSampler.h" |
| 14 | #include "SkStream.h" |
scroggo@google.com | dbf9f88 | 2013-08-21 15:01:48 +0000 | [diff] [blame] | 15 | #include "SkStreamHelpers.h" |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 16 | #include "SkTDArray.h" |
| 17 | |
| 18 | class SkBMPImageDecoder : public SkImageDecoder { |
| 19 | public: |
| 20 | SkBMPImageDecoder() {} |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 21 | |
commit-bot@chromium.org | a936e37 | 2013-03-14 14:42:18 +0000 | [diff] [blame] | 22 | virtual Format getFormat() const SK_OVERRIDE { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 23 | return kBMP_Format; |
| 24 | } |
| 25 | |
| 26 | protected: |
commit-bot@chromium.org | a936e37 | 2013-03-14 14:42:18 +0000 | [diff] [blame] | 27 | virtual bool onDecode(SkStream* stream, SkBitmap* bm, Mode mode) SK_OVERRIDE; |
| 28 | |
| 29 | private: |
| 30 | typedef SkImageDecoder INHERITED; |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 31 | }; |
| 32 | |
robertphillips@google.com | ec51cb8 | 2012-03-23 18:13:47 +0000 | [diff] [blame] | 33 | /////////////////////////////////////////////////////////////////////////////// |
| 34 | DEFINE_DECODER_CREATOR(BMPImageDecoder); |
| 35 | /////////////////////////////////////////////////////////////////////////////// |
| 36 | |
scroggo@google.com | b5571b3 | 2013-09-25 21:34:24 +0000 | [diff] [blame] | 37 | static bool is_bmp(SkStreamRewindable* stream) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 38 | static const char kBmpMagic[] = { 'B', 'M' }; |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 39 | |
djsollen@google.com | 5dd4502 | 2013-03-21 13:30:54 +0000 | [diff] [blame] | 40 | |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 41 | char buffer[sizeof(kBmpMagic)]; |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 42 | |
scroggo@google.com | 39edf4c | 2013-04-25 17:33:51 +0000 | [diff] [blame] | 43 | return stream->read(buffer, sizeof(kBmpMagic)) == sizeof(kBmpMagic) && |
| 44 | !memcmp(buffer, kBmpMagic, sizeof(kBmpMagic)); |
| 45 | } |
| 46 | |
scroggo@google.com | b5571b3 | 2013-09-25 21:34:24 +0000 | [diff] [blame] | 47 | static SkImageDecoder* sk_libbmp_dfactory(SkStreamRewindable* stream) { |
scroggo@google.com | 39edf4c | 2013-04-25 17:33:51 +0000 | [diff] [blame] | 48 | if (is_bmp(stream)) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 49 | return SkNEW(SkBMPImageDecoder); |
| 50 | } |
| 51 | return NULL; |
| 52 | } |
| 53 | |
mtklein@google.com | bd6343b | 2013-09-04 17:20:18 +0000 | [diff] [blame] | 54 | static SkImageDecoder_DecodeReg gReg(sk_libbmp_dfactory); |
reed@android.com | 00bf85a | 2009-01-22 13:04:56 +0000 | [diff] [blame] | 55 | |
scroggo@google.com | b5571b3 | 2013-09-25 21:34:24 +0000 | [diff] [blame] | 56 | static SkImageDecoder::Format get_format_bmp(SkStreamRewindable* stream) { |
scroggo@google.com | 39edf4c | 2013-04-25 17:33:51 +0000 | [diff] [blame] | 57 | if (is_bmp(stream)) { |
| 58 | return SkImageDecoder::kBMP_Format; |
| 59 | } |
| 60 | return SkImageDecoder::kUnknown_Format; |
| 61 | } |
| 62 | |
mtklein@google.com | bd6343b | 2013-09-04 17:20:18 +0000 | [diff] [blame] | 63 | static SkImageDecoder_FormatReg gFormatReg(get_format_bmp); |
scroggo@google.com | 39edf4c | 2013-04-25 17:33:51 +0000 | [diff] [blame] | 64 | |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 65 | /////////////////////////////////////////////////////////////////////////////// |
| 66 | |
| 67 | class SkBmpDecoderCallback : public image_codec::BmpDecoderCallback { |
| 68 | public: |
| 69 | // we don't copy the bitmap, just remember the pointer |
| 70 | SkBmpDecoderCallback(bool justBounds) : fJustBounds(justBounds) {} |
| 71 | |
| 72 | // override from BmpDecoderCallback |
| 73 | virtual uint8* SetSize(int width, int height) { |
| 74 | fWidth = width; |
| 75 | fHeight = height; |
| 76 | if (fJustBounds) { |
| 77 | return NULL; |
| 78 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 79 | |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 80 | fRGB.setCount(width * height * 3); // 3 == r, g, b |
| 81 | return fRGB.begin(); |
| 82 | } |
| 83 | |
| 84 | int width() const { return fWidth; } |
| 85 | int height() const { return fHeight; } |
commit-bot@chromium.org | aa537d4 | 2013-02-28 19:03:13 +0000 | [diff] [blame] | 86 | const uint8_t* rgb() const { return fRGB.begin(); } |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 87 | |
| 88 | private: |
| 89 | SkTDArray<uint8_t> fRGB; |
| 90 | int fWidth; |
| 91 | int fHeight; |
| 92 | bool fJustBounds; |
| 93 | }; |
| 94 | |
reed@android.com | 3f1f06a | 2010-03-03 21:04:12 +0000 | [diff] [blame] | 95 | bool SkBMPImageDecoder::onDecode(SkStream* stream, SkBitmap* bm, Mode mode) { |
scroggo@google.com | 2fd740f | 2013-08-21 14:59:03 +0000 | [diff] [blame] | 96 | // First read the entire stream, so that all of the data can be passed to |
| 97 | // the BmpDecoderHelper. |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 98 | |
scroggo@google.com | 2fd740f | 2013-08-21 14:59:03 +0000 | [diff] [blame] | 99 | // Allocated space used to hold the data. |
| 100 | SkAutoMalloc storage; |
scroggo@google.com | dbf9f88 | 2013-08-21 15:01:48 +0000 | [diff] [blame] | 101 | // Byte length of all of the data. |
| 102 | const size_t length = CopyStreamToStorage(&storage, stream); |
| 103 | if (0 == length) { |
| 104 | return 0; |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 105 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 106 | |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 107 | const bool justBounds = SkImageDecoder::kDecodeBounds_Mode == mode; |
| 108 | SkBmpDecoderCallback callback(justBounds); |
| 109 | |
| 110 | // Now decode the BMP into callback's rgb() array [r,g,b, r,g,b, ...] |
| 111 | { |
| 112 | image_codec::BmpDecoderHelper helper; |
| 113 | const int max_pixels = 16383*16383; // max width*height |
| 114 | if (!helper.DecodeImage((const char*)storage.get(), length, |
| 115 | max_pixels, &callback)) { |
| 116 | return false; |
| 117 | } |
| 118 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 119 | |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 120 | // we don't need this anymore, so free it now (before we try to allocate |
| 121 | // the bitmap's pixels) rather than waiting for its destructor |
| 122 | storage.free(); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 123 | |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 124 | int width = callback.width(); |
| 125 | int height = callback.height(); |
reed@android.com | 3f1f06a | 2010-03-03 21:04:12 +0000 | [diff] [blame] | 126 | SkBitmap::Config config = this->getPrefConfig(k32Bit_SrcDepth, false); |
| 127 | |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 128 | // only accept prefConfig if it makes sense for us |
reed@android.com | 3f1f06a | 2010-03-03 21:04:12 +0000 | [diff] [blame] | 129 | if (SkBitmap::kARGB_4444_Config != config && |
| 130 | SkBitmap::kRGB_565_Config != config) { |
| 131 | config = SkBitmap::kARGB_8888_Config; |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 132 | } |
| 133 | |
| 134 | SkScaledBitmapSampler sampler(width, height, getSampleSize()); |
| 135 | |
reed@google.com | 383a697 | 2013-10-21 14:00:07 +0000 | [diff] [blame] | 136 | bm->setConfig(config, sampler.scaledWidth(), sampler.scaledHeight(), 0, |
| 137 | kOpaque_SkAlphaType); |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 138 | |
scroggo@google.com | bc69ce9 | 2013-07-09 15:45:14 +0000 | [diff] [blame] | 139 | if (justBounds) { |
| 140 | return true; |
| 141 | } |
| 142 | |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 143 | if (!this->allocPixelRef(bm, NULL)) { |
| 144 | return false; |
| 145 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 146 | |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 147 | SkAutoLockPixels alp(*bm); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 148 | |
scroggo@google.com | 8d23924 | 2013-10-01 17:27:15 +0000 | [diff] [blame] | 149 | if (!sampler.begin(bm, SkScaledBitmapSampler::kRGB, *this)) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 150 | return false; |
| 151 | } |
| 152 | |
| 153 | const int srcRowBytes = width * 3; |
| 154 | const int dstHeight = sampler.scaledHeight(); |
| 155 | const uint8_t* srcRow = callback.rgb(); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 156 | |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 157 | srcRow += sampler.srcY0() * srcRowBytes; |
| 158 | for (int y = 0; y < dstHeight; y++) { |
| 159 | sampler.next(srcRow); |
| 160 | srcRow += sampler.srcDY() * srcRowBytes; |
| 161 | } |
| 162 | return true; |
| 163 | } |