| @@ -0,0 +1,102 @@ |
| 1 | +/* |
| 2 | + * Microsoft Paint (MSP) version 2 decoder |
| 3 | + * Copyright (c) 2020 Peter Ross (pross@xvid.org) |
| 4 | + * |
| 5 | + * This file is part of FFmpeg. |
| 6 | + * |
| 7 | + * FFmpeg is free software; you can redistribute it and/or |
| 8 | + * modify it under the terms of the GNU Lesser General Public |
| 9 | + * License as published by the Free Software Foundation; either |
| 10 | + * version 2.1 of the License, or (at your option) any later version. |
| 11 | + * |
| 12 | + * FFmpeg is distributed in the hope that it will be useful, |
| 13 | + * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 14 | + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 15 | + * Lesser General Public License for more details. |
| 16 | + * |
| 17 | + * You should have received a copy of the GNU Lesser General Public |
| 18 | + * License along with FFmpeg; if not, write to the Free Software |
| 19 | + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| 20 | + */ |
| 21 | + |
| 22 | +/** |
| 23 | + * @file |
| 24 | + * Microsoft Paint (MSP) version 2 decoder |
| 25 | + */ |
| 26 | + |
| 27 | +#include "avcodec.h" |
| 28 | +#include "bytestream.h" |
| 29 | +#include "internal.h" |
| 30 | + |
| 31 | +static int msp2_decode_frame(AVCodecContext *avctx, |
| 32 | + void *data, int *got_frame, |
| 33 | + AVPacket *avpkt) |
| 34 | +{ |
| 35 | + const uint8_t *buf = avpkt->data; |
| 36 | + int buf_size = avpkt->size; |
| 37 | + AVFrame *p = data; |
| 38 | + int ret; |
| 39 | + unsigned int x, y, width = (avctx->width + 7) / 8; |
| 40 | + GetByteContext idx, gb; |
| 41 | + |
| 42 | + if (buf_size <= 2 * avctx->height) |
| 43 | + return AVERROR_INVALIDDATA; |
| 44 | + |
| 45 | + avctx->pix_fmt = AV_PIX_FMT_MONOBLACK; |
| 46 | + |
| 47 | + if ((ret = ff_get_buffer(avctx, p, 0)) < 0) |
| 48 | + return ret; |
| 49 | + |
| 50 | + p->pict_type = AV_PICTURE_TYPE_I; |
| 51 | + p->key_frame = 1; |
| 52 | + |
| 53 | + bytestream2_init(&idx, buf, 2 * avctx->height); |
| 54 | + buf += 2 * avctx->height; |
| 55 | + buf_size -= 2 * avctx->height; |
| 56 | + |
| 57 | + for (y = 0; y < avctx->height; y++) { |
| 58 | + unsigned int pkt_size = bytestream2_get_le16(&idx); |
| 59 | + if (!pkt_size) { |
| 60 | + memset(p->data[0] + y * p->linesize[0], 0xFF, width); |
| 61 | + continue; |
| 62 | + } |
| 63 | + |
| 64 | + if (pkt_size > buf_size) { |
| 65 | + av_log(avctx, AV_LOG_WARNING, "image probably corrupt\n"); |
| 66 | + pkt_size = buf_size; |
| 67 | + } |
| 68 | + |
| 69 | + bytestream2_init(&gb, buf, pkt_size); |
| 70 | + x = 0; |
| 71 | + while (bytestream2_get_bytes_left(&gb) && x < width) { |
| 72 | + int size = bytestream2_get_byte(&gb); |
| 73 | + if (size) { |
| 74 | + memcpy(p->data[0] + y * p->linesize[0] + x, gb.buffer, FFMIN(size, width - x)); |
| 75 | + bytestream2_skip(&gb, size); |
| 76 | + } else { |
| 77 | + int value; |
| 78 | + size = bytestream2_get_byte(&gb); |
| 79 | + if (!size) |
| 80 | + avpriv_request_sample(avctx, "escape value"); |
| 81 | + value = bytestream2_get_byte(&gb); |
| 82 | + memset(p->data[0] + y * p->linesize[0] + x, value, FFMIN(size, width - x)); |
| 83 | + } |
| 84 | + x += size; |
| 85 | + } |
| 86 | + |
| 87 | + buf += pkt_size; |
| 88 | + buf_size -= pkt_size; |
| 89 | + } |
| 90 | + |
| 91 | + *got_frame = 1; |
| 92 | + return buf_size; |
| 93 | +} |
| 94 | + |
| 95 | +AVCodec ff_msp2_decoder = { |
| 96 | + .name = "msp2", |
| 97 | + .long_name = NULL_IF_CONFIG_SMALL("Microsoft Paint (MSP) version 2"), |
| 98 | + .type = AVMEDIA_TYPE_VIDEO, |
| 99 | + .id = AV_CODEC_ID_MSP2, |
| 100 | + .decode = msp2_decode_frame, |
| 101 | + .capabilities = AV_CODEC_CAP_DR1, |
| 102 | +}; |