9fa62f2a |
/* |
76ebb18f |
* BMP image format decoder |
9fa62f2a |
* Copyright (c) 2005 Mans Rullgard
* |
b78e7197 |
* This file is part of FFmpeg.
*
* FFmpeg is free software; you can redistribute it and/or |
9fa62f2a |
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either |
b78e7197 |
* version 2.1 of the License, or (at your option) any later version. |
9fa62f2a |
* |
b78e7197 |
* FFmpeg is distributed in the hope that it will be useful, |
9fa62f2a |
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public |
b78e7197 |
* License along with FFmpeg; if not, write to the Free Software |
5509bffa |
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
9fa62f2a |
*/
#include "avcodec.h" |
e96cc09d |
#include "bytestream.h" |
76ebb18f |
#include "bmp.h" |
594d4d5d |
#include "internal.h" |
fca506df |
#include "msrledec.h" |
9fa62f2a |
|
115329f1 |
static int bmp_decode_frame(AVCodecContext *avctx, |
df9b9567 |
void *data, int *got_frame, |
7a00bbad |
AVPacket *avpkt) |
9fa62f2a |
{ |
7a00bbad |
const uint8_t *buf = avpkt->data; |
3eab6007 |
int buf_size = avpkt->size; |
759001c5 |
AVFrame *p = data; |
9fa62f2a |
unsigned int fsize, hsize;
int width, height;
unsigned int depth; |
3e997aa4 |
BiCompression comp; |
9fa62f2a |
unsigned int ihsize; |
abcc2354 |
int i, j, n, linesize, ret; |
5270cb39 |
uint32_t rgb[3] = {0}; |
bd36ec55 |
uint32_t alpha = 0; |
9fa62f2a |
uint8_t *ptr;
int dsize; |
4bae195f |
const uint8_t *buf0 = buf; |
992f71e9 |
GetByteContext gb; |
9fa62f2a |
|
3eab6007 |
if (buf_size < 14) { |
9fa62f2a |
av_log(avctx, AV_LOG_ERROR, "buf size too small (%d)\n", buf_size); |
abcc2354 |
return AVERROR_INVALIDDATA; |
9fa62f2a |
}
|
3eab6007 |
if (bytestream_get_byte(&buf) != 'B' ||
bytestream_get_byte(&buf) != 'M') { |
9fa62f2a |
av_log(avctx, AV_LOG_ERROR, "bad magic number\n"); |
abcc2354 |
return AVERROR_INVALIDDATA; |
9fa62f2a |
}
|
e96cc09d |
fsize = bytestream_get_le32(&buf); |
3eab6007 |
if (buf_size < fsize) { |
42402a55 |
av_log(avctx, AV_LOG_ERROR, "not enough data (%d < %d), trying to decode anyway\n", |
9fa62f2a |
buf_size, fsize); |
42402a55 |
fsize = buf_size; |
9fa62f2a |
}
|
e96cc09d |
buf += 2; /* reserved1 */
buf += 2; /* reserved2 */ |
9fa62f2a |
|
3eab6007 |
hsize = bytestream_get_le32(&buf); /* header size */
ihsize = bytestream_get_le32(&buf); /* more header size */
if (ihsize + 14 > hsize) { |
9fa62f2a |
av_log(avctx, AV_LOG_ERROR, "invalid header size %d\n", hsize); |
abcc2354 |
return AVERROR_INVALIDDATA; |
9fa62f2a |
}
|
c5b2fe16 |
/* sometimes file size is set to some headers size, set a real size in that case */ |
3eab6007 |
if (fsize == 14 || fsize == ihsize + 14) |
c5b2fe16 |
fsize = buf_size - 2;
|
3eab6007 |
if (fsize <= hsize) { |
c5b2fe16 |
av_log(avctx, AV_LOG_ERROR, "declared file size is less than header size (%d < %d)\n",
fsize, hsize); |
abcc2354 |
return AVERROR_INVALIDDATA; |
c5b2fe16 |
}
|
3eab6007 |
switch (ihsize) { |
9608e3a1 |
case 40: // windib
case 56: // windib v3 |
4a14e666 |
case 64: // OS/2 v2
case 108: // windib v4
case 124: // windib v5 |
3eab6007 |
width = bytestream_get_le32(&buf); |
6f7b915a |
height = bytestream_get_le32(&buf); |
4a14e666 |
break;
case 12: // OS/2 v1 |
b65213c0 |
width = bytestream_get_le16(&buf);
height = bytestream_get_le16(&buf); |
4a14e666 |
break;
default: |
3a9a15c4 |
av_log(avctx, AV_LOG_ERROR, "unsupported BMP file, patch welcome\n"); |
abcc2354 |
return AVERROR_PATCHWELCOME; |
b65213c0 |
} |
9fa62f2a |
|
3eab6007 |
/* planes */
if (bytestream_get_le16(&buf) != 1) { |
9fa62f2a |
av_log(avctx, AV_LOG_ERROR, "invalid BMP header\n"); |
abcc2354 |
return AVERROR_INVALIDDATA; |
9fa62f2a |
}
|
e96cc09d |
depth = bytestream_get_le16(&buf); |
9fa62f2a |
|
313b40ef |
if (ihsize >= 40) |
e96cc09d |
comp = bytestream_get_le32(&buf); |
9fa62f2a |
else
comp = BMP_RGB;
|
3eab6007 |
if (comp != BMP_RGB && comp != BMP_BITFIELDS && comp != BMP_RLE4 &&
comp != BMP_RLE8) { |
9fa62f2a |
av_log(avctx, AV_LOG_ERROR, "BMP coding %d not supported\n", comp); |
abcc2354 |
return AVERROR_INVALIDDATA; |
9fa62f2a |
}
|
3eab6007 |
if (comp == BMP_BITFIELDS) { |
e96cc09d |
buf += 20;
rgb[0] = bytestream_get_le32(&buf);
rgb[1] = bytestream_get_le32(&buf);
rgb[2] = bytestream_get_le32(&buf); |
313b40ef |
alpha = bytestream_get_le32(&buf); |
9fa62f2a |
}
|
3eab6007 |
avctx->width = width;
avctx->height = height > 0 ? height : -height; |
9fa62f2a |
|
716d413c |
avctx->pix_fmt = AV_PIX_FMT_NONE; |
9fa62f2a |
|
3eab6007 |
switch (depth) { |
9fa62f2a |
case 32: |
3eab6007 |
if (comp == BMP_BITFIELDS) { |
7c4b3975 |
if (rgb[0] == 0xFF000000 && rgb[1] == 0x00FF0000 && rgb[2] == 0x0000FF00) |
ac627b3d |
avctx->pix_fmt = alpha ? AV_PIX_FMT_ABGR : AV_PIX_FMT_0BGR; |
7c4b3975 |
else if (rgb[0] == 0x00FF0000 && rgb[1] == 0x0000FF00 && rgb[2] == 0x000000FF) |
ac627b3d |
avctx->pix_fmt = alpha ? AV_PIX_FMT_BGRA : AV_PIX_FMT_BGR0; |
7c4b3975 |
else if (rgb[0] == 0x0000FF00 && rgb[1] == 0x00FF0000 && rgb[2] == 0xFF000000) |
ac627b3d |
avctx->pix_fmt = alpha ? AV_PIX_FMT_ARGB : AV_PIX_FMT_0RGB; |
7c4b3975 |
else if (rgb[0] == 0x000000FF && rgb[1] == 0x0000FF00 && rgb[2] == 0x00FF0000) |
ac627b3d |
avctx->pix_fmt = alpha ? AV_PIX_FMT_RGBA : AV_PIX_FMT_RGB0; |
7c4b3975 |
else {
av_log(avctx, AV_LOG_ERROR, "Unknown bitfields %0X %0X %0X\n", rgb[0], rgb[1], rgb[2]);
return AVERROR(EINVAL); |
9fa62f2a |
}
} else { |
ac627b3d |
avctx->pix_fmt = AV_PIX_FMT_BGRA; |
9fa62f2a |
}
break;
case 24: |
716d413c |
avctx->pix_fmt = AV_PIX_FMT_BGR24; |
9fa62f2a |
break;
case 16: |
3eab6007 |
if (comp == BMP_RGB) |
716d413c |
avctx->pix_fmt = AV_PIX_FMT_RGB555; |
a6839c4e |
else if (comp == BMP_BITFIELDS) {
if (rgb[0] == 0xF800 && rgb[1] == 0x07E0 && rgb[2] == 0x001F) |
716d413c |
avctx->pix_fmt = AV_PIX_FMT_RGB565; |
a6839c4e |
else if (rgb[0] == 0x7C00 && rgb[1] == 0x03E0 && rgb[2] == 0x001F) |
716d413c |
avctx->pix_fmt = AV_PIX_FMT_RGB555; |
a6839c4e |
else if (rgb[0] == 0x0F00 && rgb[1] == 0x00F0 && rgb[2] == 0x000F) |
716d413c |
avctx->pix_fmt = AV_PIX_FMT_RGB444; |
a6839c4e |
else {
av_log(avctx, AV_LOG_ERROR, "Unknown bitfields %0X %0X %0X\n", rgb[0], rgb[1], rgb[2]);
return AVERROR(EINVAL);
}
} |
15501c32 |
break;
case 8: |
3eab6007 |
if (hsize - ihsize - 14 > 0) |
716d413c |
avctx->pix_fmt = AV_PIX_FMT_PAL8; |
15501c32 |
else |
716d413c |
avctx->pix_fmt = AV_PIX_FMT_GRAY8; |
15501c32 |
break; |
0e609d74 |
case 1: |
15501c32 |
case 4: |
3eab6007 |
if (hsize - ihsize - 14 > 0) { |
716d413c |
avctx->pix_fmt = AV_PIX_FMT_PAL8; |
3eab6007 |
} else { |
0e609d74 |
av_log(avctx, AV_LOG_ERROR, "Unknown palette for %d-colour BMP\n", 1<<depth); |
abcc2354 |
return AVERROR_INVALIDDATA; |
15501c32 |
}
break; |
9fa62f2a |
default:
av_log(avctx, AV_LOG_ERROR, "depth %d not supported\n", depth); |
abcc2354 |
return AVERROR_INVALIDDATA; |
9fa62f2a |
}
|
3eab6007 |
if (avctx->pix_fmt == AV_PIX_FMT_NONE) { |
9fa62f2a |
av_log(avctx, AV_LOG_ERROR, "unsupported pixel format\n"); |
abcc2354 |
return AVERROR_INVALIDDATA; |
9fa62f2a |
}
|
1ec94b0f |
if ((ret = ff_get_buffer(avctx, p, 0)) < 0) |
abcc2354 |
return ret; |
ce5e49b0 |
p->pict_type = AV_PICTURE_TYPE_I; |
9fa62f2a |
p->key_frame = 1;
|
3eab6007 |
buf = buf0 + hsize; |
9fa62f2a |
dsize = buf_size - hsize;
|
743311a1 |
/* Line size in file multiple of 4 */ |
6f2054c5 |
n = ((avctx->width * depth + 31) / 8) & ~3; |
9fa62f2a |
|
3eab6007 |
if (n * avctx->height > dsize && comp != BMP_RLE4 && comp != BMP_RLE8) { |
9fa62f2a |
av_log(avctx, AV_LOG_ERROR, "not enough data (%d < %d)\n",
dsize, n * avctx->height); |
abcc2354 |
return AVERROR_INVALIDDATA; |
9fa62f2a |
}
|
fca506df |
// RLE may skip decoding some picture areas, so blank picture before decoding |
3eab6007 |
if (comp == BMP_RLE4 || comp == BMP_RLE8) |
fca506df |
memset(p->data[0], 0, avctx->height * p->linesize[0]);
|
3eab6007 |
if (height > 0) {
ptr = p->data[0] + (avctx->height - 1) * p->linesize[0]; |
9fa62f2a |
linesize = -p->linesize[0];
} else { |
3eab6007 |
ptr = p->data[0]; |
9fa62f2a |
linesize = p->linesize[0];
}
|
3eab6007 |
if (avctx->pix_fmt == AV_PIX_FMT_PAL8) { |
22da339f |
int colors = 1 << depth; |
8b78c296 |
memset(p->data[1], 0, 1024);
|
3eab6007 |
if (ihsize >= 36) { |
22da339f |
int t;
buf = buf0 + 46; |
3eab6007 |
t = bytestream_get_le32(&buf);
if (t < 0 || t > (1 << depth)) { |
22da339f |
av_log(avctx, AV_LOG_ERROR, "Incorrect number of colors - %X for bitdepth %d\n", t, depth); |
3eab6007 |
} else if (t) { |
22da339f |
colors = t;
}
} |
15501c32 |
buf = buf0 + 14 + ihsize; //palette location |
3eab6007 |
// OS/2 bitmap, 3 bytes per palette entry
if ((hsize-ihsize-14) < (colors << 2)) { |
633f9974 |
if ((hsize-ihsize-14) < colors * 3) { |
a9bbf59b |
av_log(avctx, AV_LOG_ERROR, "palette doesn't fit in packet\n"); |
633f9974 |
return AVERROR_INVALIDDATA;
} |
3eab6007 |
for (i = 0; i < colors; i++) |
2fed05f5 |
((uint32_t*)p->data[1])[i] = (0xFFU<<24) | bytestream_get_le24(&buf); |
3eab6007 |
} else {
for (i = 0; i < colors; i++) |
3a15051a |
((uint32_t*)p->data[1])[i] = 0xFFU << 24 | bytestream_get_le32(&buf); |
15501c32 |
}
buf = buf0 + hsize;
} |
3eab6007 |
if (comp == BMP_RLE4 || comp == BMP_RLE8) { |
5f39992e |
if (comp == BMP_RLE8 && height < 0) { |
3eab6007 |
p->data[0] += p->linesize[0] * (avctx->height - 1); |
5073cca4 |
p->linesize[0] = -p->linesize[0];
} |
992f71e9 |
bytestream2_init(&gb, buf, dsize);
ff_msrle_decode(avctx, (AVPicture*)p, depth, &gb); |
3eab6007 |
if (height < 0) {
p->data[0] += p->linesize[0] * (avctx->height - 1); |
5073cca4 |
p->linesize[0] = -p->linesize[0];
} |
3eab6007 |
} else {
switch (depth) { |
b94a631f |
case 1: |
1e92d58e |
for (i = 0; i < avctx->height; i++) { |
0e609d74 |
int j; |
1e92d58e |
for (j = 0; j < n; j++) { |
0e609d74 |
ptr[j*8+0] = buf[j] >> 7;
ptr[j*8+1] = (buf[j] >> 6) & 1;
ptr[j*8+2] = (buf[j] >> 5) & 1;
ptr[j*8+3] = (buf[j] >> 4) & 1;
ptr[j*8+4] = (buf[j] >> 3) & 1;
ptr[j*8+5] = (buf[j] >> 2) & 1;
ptr[j*8+6] = (buf[j] >> 1) & 1;
ptr[j*8+7] = buf[j] & 1;
}
buf += n;
ptr += linesize;
}
break; |
6b78a77a |
case 8:
case 24: |
7c4b3975 |
case 32: |
3eab6007 |
for (i = 0; i < avctx->height; i++) { |
b94a631f |
memcpy(ptr, buf, n);
buf += n;
ptr += linesize; |
15501c32 |
} |
b94a631f |
break;
case 4: |
3eab6007 |
for (i = 0; i < avctx->height; i++) { |
b94a631f |
int j; |
3eab6007 |
for (j = 0; j < n; j++) { |
b94a631f |
ptr[j*2+0] = (buf[j] >> 4) & 0xF;
ptr[j*2+1] = buf[j] & 0xF;
}
buf += n;
ptr += linesize;
}
break;
case 16: |
3eab6007 |
for (i = 0; i < avctx->height; i++) { |
b94a631f |
const uint16_t *src = (const uint16_t *) buf; |
3eab6007 |
uint16_t *dst = (uint16_t *) ptr; |
9fa62f2a |
|
3eab6007 |
for (j = 0; j < avctx->width; j++) |
8fc0162a |
*dst++ = av_le2ne16(*src++); |
9fa62f2a |
|
b94a631f |
buf += n;
ptr += linesize; |
9fa62f2a |
} |
b94a631f |
break;
default:
av_log(avctx, AV_LOG_ERROR, "BMP decoder is broken\n"); |
abcc2354 |
return AVERROR_INVALIDDATA; |
9fa62f2a |
} |
fca506df |
} |
9fa62f2a |
|
df9b9567 |
*got_frame = 1; |
9fa62f2a |
return buf_size;
}
|
e7e2df27 |
AVCodec ff_bmp_decoder = { |
ec6402b7 |
.name = "bmp", |
b2bed932 |
.long_name = NULL_IF_CONFIG_SMALL("BMP (Windows and OS/2 bitmap)"), |
ec6402b7 |
.type = AVMEDIA_TYPE_VIDEO, |
36ef5369 |
.id = AV_CODEC_ID_BMP, |
ec6402b7 |
.decode = bmp_decode_frame,
.capabilities = CODEC_CAP_DR1, |
9fa62f2a |
}; |