blob: 5a1b39ac90d3b36e96e5736917ddd88630da11ce [file] [log] [blame]
/*
* Copyright 2007 The Android Open Source Project
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
// Author: cevans@google.com (Chris Evans)
#include "bmpdecoderhelper.h"
namespace image_codec {
static const int kBmpHeaderSize = 14;
static const int kBmpInfoSize = 40;
static const int kBmpOS2InfoSize = 12;
static const int kMaxDim = SHRT_MAX / 2;
bool BmpDecoderHelper::DecodeImage(const char* p,
size_t len,
int max_pixels,
BmpDecoderCallback* callback) {
data_ = reinterpret_cast<const uint8*>(p);
pos_ = 0;
len_ = len;
inverted_ = true;
// Parse the header structure.
if (len < kBmpHeaderSize + 4) {
return false;
}
GetShort(); // Signature.
GetInt(); // Size.
GetInt(); // Reserved.
int offset = GetInt();
// Parse the info structure.
int infoSize = GetInt();
if (infoSize != kBmpOS2InfoSize && infoSize < kBmpInfoSize) {
return false;
}
int cols = 0;
int comp = 0;
int colLen = 4;
if (infoSize >= kBmpInfoSize) {
if (len < kBmpHeaderSize + kBmpInfoSize) {
return false;
}
width_ = GetInt();
height_ = GetInt();
GetShort(); // Planes.
bpp_ = GetShort();
comp = GetInt();
GetInt(); // Size.
GetInt(); // XPPM.
GetInt(); // YPPM.
cols = GetInt();
GetInt(); // Important colours.
} else {
if (len < kBmpHeaderSize + kBmpOS2InfoSize) {
return false;
}
colLen = 3;
width_ = GetShort();
height_ = GetShort();
GetShort(); // Planes.
bpp_ = GetShort();
}
if (height_ < 0) {
height_ = -height_;
inverted_ = false;
}
if (width_ <= 0 || width_ > kMaxDim || height_ <= 0 || height_ > kMaxDim) {
return false;
}
if (width_ * height_ > max_pixels) {
return false;
}
if (cols < 0 || cols > 256) {
return false;
}
// Allocate then read in the colour map.
if (cols == 0 && bpp_ <= 8) {
cols = 1 << bpp_;
}
if (bpp_ <= 8 || cols > 0) {
uint8* colBuf = new uint8[256 * 3];
memset(colBuf, '\0', 256 * 3);
colTab_.reset(colBuf);
}
if (cols > 0) {
if (pos_ + (cols * colLen) > len_) {
return false;
}
for (int i = 0; i < cols; ++i) {
int base = i * 3;
colTab_[base + 2] = GetByte();
colTab_[base + 1] = GetByte();
colTab_[base] = GetByte();
if (colLen == 4) {
GetByte();
}
}
}
// Read in the compression data if necessary.
redBits_ = 0x7c00;
greenBits_ = 0x03e0;
blueBits_ = 0x001f;
bool rle = false;
if (comp == 1 || comp == 2) {
rle = true;
} else if (comp == 3) {
if (pos_ + 12 > len_) {
return false;
}
redBits_ = GetInt() & 0xffff;
greenBits_ = GetInt() & 0xffff;
blueBits_ = GetInt() & 0xffff;
}
redShiftRight_ = CalcShiftRight(redBits_);
greenShiftRight_ = CalcShiftRight(greenBits_);
blueShiftRight_ = CalcShiftRight(blueBits_);
redShiftLeft_ = CalcShiftLeft(redBits_);
greenShiftLeft_ = CalcShiftLeft(greenBits_);
blueShiftLeft_ = CalcShiftLeft(blueBits_);
rowPad_ = 0;
pixelPad_ = 0;
int rowLen;
if (bpp_ == 32) {
rowLen = width_ * 4;
pixelPad_ = 1;
} else if (bpp_ == 24) {
rowLen = width_ * 3;
} else if (bpp_ == 16) {
rowLen = width_ * 2;
} else if (bpp_ == 8) {
rowLen = width_;
} else if (bpp_ == 4) {
rowLen = width_ / 2;
if (width_ & 1) {
rowLen++;
}
} else if (bpp_ == 1) {
rowLen = width_ / 8;
if (width_ & 7) {
rowLen++;
}
} else {
return false;
}
// Round the rowLen up to a multiple of 4.
if (rowLen % 4 != 0) {
rowPad_ = 4 - (rowLen % 4);
rowLen += rowPad_;
}
if (offset > 0 && (size_t)offset > pos_ && (size_t)offset < len_) {
pos_ = offset;
}
// Deliberately off-by-one; a load of BMPs seem to have their last byte
// missing.
if (!rle && (pos_ + (rowLen * height_) > len_ + 1)) {
return false;
}
output_ = callback->SetSize(width_, height_);
if (NULL == output_) {
return true; // meaning we succeeded, but they want us to stop now
}
if (rle && (bpp_ == 4 || bpp_ == 8)) {
DoRLEDecode();
} else {
DoStandardDecode();
}
return true;
}
void BmpDecoderHelper::DoRLEDecode() {
static const uint8 RLE_ESCAPE = 0;
static const uint8 RLE_EOL = 0;
static const uint8 RLE_EOF = 1;
static const uint8 RLE_DELTA = 2;
int x = 0;
int y = height_ - 1;
while (pos_ + 1 < len_) {
uint8 cmd = GetByte();
if (cmd != RLE_ESCAPE) {
uint8 pixels = GetByte();
int num = 0;
uint8 col = pixels;
while (cmd-- && x < width_) {
if (bpp_ == 4) {
if (num & 1) {
col = pixels & 0xf;
} else {
col = pixels >> 4;
}
}
PutPixel(x++, y, col);
num++;
}
} else {
cmd = GetByte();
if (cmd == RLE_EOF) {
return;
} else if (cmd == RLE_EOL) {
x = 0;
y--;
if (y < 0) {
return;
}
} else if (cmd == RLE_DELTA) {
if (pos_ + 1 < len_) {
uint8 dx = GetByte();
uint8 dy = GetByte();
x += dx;
if (x > width_) {
x = width_;
}
y -= dy;
if (y < 0) {
return;
}
}
} else {
int num = 0;
int bytesRead = 0;
uint8 val = 0;
while (cmd-- && pos_ < len_) {
if (bpp_ == 8 || !(num & 1)) {
val = GetByte();
bytesRead++;
}
uint8 col = val;
if (bpp_ == 4) {
if (num & 1) {
col = col & 0xf;
} else {
col >>= 4;
}
}
if (x < width_) {
PutPixel(x++, y, col);
}
num++;
}
// All pixel runs must be an even number of bytes - skip a byte if we
// read an odd number.
if ((bytesRead & 1) && pos_ < len_) {
GetByte();
}
}
}
}
}
void BmpDecoderHelper::PutPixel(int x, int y, uint8 col) {
CHECK(x >= 0 && x < width_);
CHECK(y >= 0 && y < height_);
if (!inverted_) {
y = height_ - (y + 1);
}
int base = ((y * width_) + x) * 3;
int colBase = col * 3;
output_[base] = colTab_[colBase];
output_[base + 1] = colTab_[colBase + 1];
output_[base + 2] = colTab_[colBase + 2];
}
void BmpDecoderHelper::DoStandardDecode() {
int row = 0;
uint8 currVal = 0;
for (int h = height_ - 1; h >= 0; h--, row++) {
int realH = h;
if (!inverted_) {
realH = height_ - (h + 1);
}
uint8* line = output_ + (3 * width_ * realH);
for (int w = 0; w < width_; w++) {
if (bpp_ >= 24) {
line[2] = GetByte();
line[1] = GetByte();
line[0] = GetByte();
} else if (bpp_ == 16) {
uint32 val = GetShort();
line[0] = ((val & redBits_) >> redShiftRight_) << redShiftLeft_;
line[1] = ((val & greenBits_) >> greenShiftRight_) << greenShiftLeft_;
line[2] = ((val & blueBits_) >> blueShiftRight_) << blueShiftLeft_;
} else if (bpp_ <= 8) {
uint8 col;
if (bpp_ == 8) {
col = GetByte();
} else if (bpp_ == 4) {
if ((w % 2) == 0) {
currVal = GetByte();
col = currVal >> 4;
} else {
col = currVal & 0xf;
}
} else {
if ((w % 8) == 0) {
currVal = GetByte();
}
int bit = w & 7;
col = ((currVal >> (7 - bit)) & 1);
}
int base = col * 3;
line[0] = colTab_[base];
line[1] = colTab_[base + 1];
line[2] = colTab_[base + 2];
}
line += 3;
for (int i = 0; i < pixelPad_; ++i) {
GetByte();
}
}
for (int i = 0; i < rowPad_; ++i) {
GetByte();
}
}
}
int BmpDecoderHelper::GetInt() {
uint8 b1 = GetByte();
uint8 b2 = GetByte();
uint8 b3 = GetByte();
uint8 b4 = GetByte();
return b1 | (b2 << 8) | (b3 << 16) | (b4 << 24);
}
int BmpDecoderHelper::GetShort() {
uint8 b1 = GetByte();
uint8 b2 = GetByte();
return b1 | (b2 << 8);
}
uint8 BmpDecoderHelper::GetByte() {
CHECK(pos_ <= len_);
// We deliberately allow this off-by-one access to cater for BMPs with their
// last byte missing.
if (pos_ == len_) {
return 0;
}
return data_[pos_++];
}
int BmpDecoderHelper::CalcShiftRight(uint32 mask) {
int ret = 0;
while (mask != 0 && !(mask & 1)) {
mask >>= 1;
ret++;
}
return ret;
}
int BmpDecoderHelper::CalcShiftLeft(uint32 mask) {
int ret = 0;
while (mask != 0 && !(mask & 1)) {
mask >>= 1;
}
while (mask != 0 && !(mask & 0x80)) {
mask <<= 1;
ret++;
}
return ret;
}
} // namespace image_codec