X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=libavcodec%2Fqtrle.c;h=0c74798226c132b2f9737e1d6362035a8bed397b;hb=284e65d64ea91dc172dea1b843d162b22ed48730;hp=84482bac7bea21313d515282b974a01d755dc320;hpb=04deea9ad25299579dbbe0e4103e8141b165a666;p=ffmpeg diff --git a/libavcodec/qtrle.c b/libavcodec/qtrle.c index 84482bac7be..0c74798226c 100644 --- a/libavcodec/qtrle.c +++ b/libavcodec/qtrle.c @@ -2,25 +2,25 @@ * Quicktime Animation (RLE) Video Decoder * Copyright (C) 2004 the ffmpeg project * - * This file is part of FFmpeg. + * This file is part of Libav. * - * FFmpeg is free software; you can redistribute it and/or + * Libav is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * - * FFmpeg is distributed in the hope that it will be useful, + * Libav is distributed in the hope that it will be useful, * 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 - * License along with FFmpeg; if not, write to the Free Software + * License along with Libav; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ /** - * @file qtrle.c + * @file * QT RLE Video Decoder by Mike Melanson (melanson@pcisys.net) * For more information about the QT RLE format, visit: * http://www.pcisys.net/~melanson/codecs/ @@ -34,20 +34,19 @@ #include #include #include -#include +#include "libavutil/intreadwrite.h" #include "avcodec.h" -#include "dsputil.h" typedef struct QtrleContext { AVCodecContext *avctx; - DSPContext dsp; AVFrame frame; const unsigned char *buf; int size; + uint32_t pal[256]; } QtrleContext; #define CHECK_STREAM_PTR(n) \ @@ -64,98 +63,107 @@ typedef struct QtrleContext { return; \ } \ -static void qtrle_decode_1bpp(QtrleContext *s) +static void qtrle_decode_1bpp(QtrleContext *s, int stream_ptr, int row_ptr, int lines_to_change) { -} + int rle_code; + int pixel_ptr = 0; + int row_inc = s->frame.linesize[0]; + unsigned char pi0, pi1; /* 2 8-pixel values */ + unsigned char *rgb = s->frame.data[0]; + int pixel_limit = s->frame.linesize[0] * s->avctx->height; + int skip; -static void qtrle_decode_2bpp(QtrleContext *s) -{ + while (lines_to_change) { + CHECK_STREAM_PTR(2); + skip = s->buf[stream_ptr++]; + rle_code = (signed char)s->buf[stream_ptr++]; + if (rle_code == 0) + break; + if(skip & 0x80) { + lines_to_change--; + row_ptr += row_inc; + pixel_ptr = row_ptr + 2 * (skip & 0x7f); + } else + pixel_ptr += 2 * skip; + CHECK_PIXEL_PTR(0); /* make sure pixel_ptr is positive */ + + if (rle_code < 0) { + /* decode the run length code */ + rle_code = -rle_code; + /* get the next 2 bytes from the stream, treat them as groups + * of 8 pixels, and output them rle_code times */ + CHECK_STREAM_PTR(2); + pi0 = s->buf[stream_ptr++]; + pi1 = s->buf[stream_ptr++]; + CHECK_PIXEL_PTR(rle_code * 2); + + while (rle_code--) { + rgb[pixel_ptr++] = pi0; + rgb[pixel_ptr++] = pi1; + } + } else { + /* copy the same pixel directly to output 2 times */ + rle_code *= 2; + CHECK_STREAM_PTR(rle_code); + CHECK_PIXEL_PTR(rle_code); + + while (rle_code--) + rgb[pixel_ptr++] = s->buf[stream_ptr++]; + } + } } -static void qtrle_decode_4bpp(QtrleContext *s) +static inline void qtrle_decode_2n4bpp(QtrleContext *s, int stream_ptr, + int row_ptr, int lines_to_change, int bpp) { - int stream_ptr; - int header; - int start_line; - int lines_to_change; - int rle_code; - int row_ptr, pixel_ptr; + int rle_code, i; + int pixel_ptr; int row_inc = s->frame.linesize[0]; - unsigned char pi1, pi2, pi3, pi4, pi5, pi6, pi7, pi8; /* 8 palette indices */ + unsigned char pi[16]; /* 16 palette indices */ unsigned char *rgb = s->frame.data[0]; int pixel_limit = s->frame.linesize[0] * s->avctx->height; + int num_pixels = (bpp == 4) ? 8 : 16; - /* check if this frame is even supposed to change */ - if (s->size < 8) - return; - - /* start after the chunk size */ - stream_ptr = 4; - - /* fetch the header */ - CHECK_STREAM_PTR(2); - header = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 2; - - /* if a header is present, fetch additional decoding parameters */ - if (header & 0x0008) { - CHECK_STREAM_PTR(8); - start_line = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 4; - lines_to_change = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 4; - } else { - start_line = 0; - lines_to_change = s->avctx->height; - } - - row_ptr = row_inc * start_line; while (lines_to_change--) { CHECK_STREAM_PTR(2); - pixel_ptr = row_ptr + (8 * (s->buf[stream_ptr++] - 1)); + pixel_ptr = row_ptr + (num_pixels * (s->buf[stream_ptr++] - 1)); while ((rle_code = (signed char)s->buf[stream_ptr++]) != -1) { if (rle_code == 0) { /* there's another skip code in the stream */ CHECK_STREAM_PTR(1); - pixel_ptr += (8 * (s->buf[stream_ptr++] - 1)); + pixel_ptr += (num_pixels * (s->buf[stream_ptr++] - 1)); CHECK_PIXEL_PTR(0); /* make sure pixel_ptr is positive */ } else if (rle_code < 0) { /* decode the run length code */ rle_code = -rle_code; /* get the next 4 bytes from the stream, treat them as palette - * indices, and output them rle_code times */ + * indexes, and output them rle_code times */ CHECK_STREAM_PTR(4); - pi1 = ((s->buf[stream_ptr]) >> 4) & 0x0f; - pi2 = (s->buf[stream_ptr++]) & 0x0f; - pi3 = ((s->buf[stream_ptr]) >> 4) & 0x0f; - pi4 = (s->buf[stream_ptr++]) & 0x0f; - pi5 = ((s->buf[stream_ptr]) >> 4) & 0x0f; - pi6 = (s->buf[stream_ptr++]) & 0x0f; - pi7 = ((s->buf[stream_ptr]) >> 4) & 0x0f; - pi8 = (s->buf[stream_ptr++]) & 0x0f; - - CHECK_PIXEL_PTR(rle_code * 8); - + for (i = num_pixels-1; i >= 0; i--) { + pi[num_pixels-1-i] = (s->buf[stream_ptr] >> ((i*bpp) & 0x07)) & ((1<>2)-1)) == 0); + } + CHECK_PIXEL_PTR(rle_code * num_pixels); while (rle_code--) { - rgb[pixel_ptr++] = pi1; - rgb[pixel_ptr++] = pi2; - rgb[pixel_ptr++] = pi3; - rgb[pixel_ptr++] = pi4; - rgb[pixel_ptr++] = pi5; - rgb[pixel_ptr++] = pi6; - rgb[pixel_ptr++] = pi7; - rgb[pixel_ptr++] = pi8; + for (i = 0; i < num_pixels; i++) + rgb[pixel_ptr++] = pi[i]; } } else { /* copy the same pixel directly to output 4 times */ rle_code *= 4; CHECK_STREAM_PTR(rle_code); - CHECK_PIXEL_PTR(rle_code*2); - + CHECK_PIXEL_PTR(rle_code*(num_pixels>>2)); while (rle_code--) { - rgb[pixel_ptr++] = ((s->buf[stream_ptr]) >> 4) & 0x0f; - rgb[pixel_ptr++] = (s->buf[stream_ptr++]) & 0x0f; + if(bpp == 4) { + rgb[pixel_ptr++] = ((s->buf[stream_ptr]) >> 4) & 0x0f; + rgb[pixel_ptr++] = (s->buf[stream_ptr++]) & 0x0f; + } else { + rgb[pixel_ptr++] = ((s->buf[stream_ptr]) >> 6) & 0x03; + rgb[pixel_ptr++] = ((s->buf[stream_ptr]) >> 4) & 0x03; + rgb[pixel_ptr++] = ((s->buf[stream_ptr]) >> 2) & 0x03; + rgb[pixel_ptr++] = (s->buf[stream_ptr++]) & 0x03; + } } } } @@ -163,44 +171,15 @@ static void qtrle_decode_4bpp(QtrleContext *s) } } -static void qtrle_decode_8bpp(QtrleContext *s) +static void qtrle_decode_8bpp(QtrleContext *s, int stream_ptr, int row_ptr, int lines_to_change) { - int stream_ptr; - int header; - int start_line; - int lines_to_change; int rle_code; - int row_ptr, pixel_ptr; + int pixel_ptr; int row_inc = s->frame.linesize[0]; - unsigned char pi1, pi2, pi3, pi4; /* 4 palette indices */ + unsigned char pi1, pi2, pi3, pi4; /* 4 palette indexes */ unsigned char *rgb = s->frame.data[0]; int pixel_limit = s->frame.linesize[0] * s->avctx->height; - /* check if this frame is even supposed to change */ - if (s->size < 8) - return; - - /* start after the chunk size */ - stream_ptr = 4; - - /* fetch the header */ - CHECK_STREAM_PTR(2); - header = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 2; - - /* if a header is present, fetch additional decoding parameters */ - if (header & 0x0008) { - CHECK_STREAM_PTR(8); - start_line = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 4; - lines_to_change = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 4; - } else { - start_line = 0; - lines_to_change = s->avctx->height; - } - - row_ptr = row_inc * start_line; while (lines_to_change--) { CHECK_STREAM_PTR(2); pixel_ptr = row_ptr + (4 * (s->buf[stream_ptr++] - 1)); @@ -215,7 +194,7 @@ static void qtrle_decode_8bpp(QtrleContext *s) /* decode the run length code */ rle_code = -rle_code; /* get the next 4 bytes from the stream, treat them as palette - * indices, and output them rle_code times */ + * indexes, and output them rle_code times */ CHECK_STREAM_PTR(4); pi1 = s->buf[stream_ptr++]; pi2 = s->buf[stream_ptr++]; @@ -245,44 +224,15 @@ static void qtrle_decode_8bpp(QtrleContext *s) } } -static void qtrle_decode_16bpp(QtrleContext *s) +static void qtrle_decode_16bpp(QtrleContext *s, int stream_ptr, int row_ptr, int lines_to_change) { - int stream_ptr; - int header; - int start_line; - int lines_to_change; int rle_code; - int row_ptr, pixel_ptr; + int pixel_ptr; int row_inc = s->frame.linesize[0]; unsigned short rgb16; unsigned char *rgb = s->frame.data[0]; int pixel_limit = s->frame.linesize[0] * s->avctx->height; - /* check if this frame is even supposed to change */ - if (s->size < 8) - return; - - /* start after the chunk size */ - stream_ptr = 4; - - /* fetch the header */ - CHECK_STREAM_PTR(2); - header = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 2; - - /* if a header is present, fetch additional decoding parameters */ - if (header & 0x0008) { - CHECK_STREAM_PTR(8); - start_line = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 4; - lines_to_change = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 4; - } else { - start_line = 0; - lines_to_change = s->avctx->height; - } - - row_ptr = row_inc * start_line; while (lines_to_change--) { CHECK_STREAM_PTR(2); pixel_ptr = row_ptr + (s->buf[stream_ptr++] - 1) * 2; @@ -323,44 +273,15 @@ static void qtrle_decode_16bpp(QtrleContext *s) } } -static void qtrle_decode_24bpp(QtrleContext *s) +static void qtrle_decode_24bpp(QtrleContext *s, int stream_ptr, int row_ptr, int lines_to_change) { - int stream_ptr; - int header; - int start_line; - int lines_to_change; int rle_code; - int row_ptr, pixel_ptr; + int pixel_ptr; int row_inc = s->frame.linesize[0]; unsigned char r, g, b; unsigned char *rgb = s->frame.data[0]; int pixel_limit = s->frame.linesize[0] * s->avctx->height; - /* check if this frame is even supposed to change */ - if (s->size < 8) - return; - - /* start after the chunk size */ - stream_ptr = 4; - - /* fetch the header */ - CHECK_STREAM_PTR(2); - header = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 2; - - /* if a header is present, fetch additional decoding parameters */ - if (header & 0x0008) { - CHECK_STREAM_PTR(8); - start_line = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 4; - lines_to_change = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 4; - } else { - start_line = 0; - lines_to_change = s->avctx->height; - } - - row_ptr = row_inc * start_line; while (lines_to_change--) { CHECK_STREAM_PTR(2); pixel_ptr = row_ptr + (s->buf[stream_ptr++] - 1) * 3; @@ -402,45 +323,15 @@ static void qtrle_decode_24bpp(QtrleContext *s) } } -static void qtrle_decode_32bpp(QtrleContext *s) +static void qtrle_decode_32bpp(QtrleContext *s, int stream_ptr, int row_ptr, int lines_to_change) { - int stream_ptr; - int header; - int start_line; - int lines_to_change; int rle_code; - int row_ptr, pixel_ptr; + int pixel_ptr; int row_inc = s->frame.linesize[0]; - unsigned char a, r, g, b; unsigned int argb; unsigned char *rgb = s->frame.data[0]; int pixel_limit = s->frame.linesize[0] * s->avctx->height; - /* check if this frame is even supposed to change */ - if (s->size < 8) - return; - - /* start after the chunk size */ - stream_ptr = 4; - - /* fetch the header */ - CHECK_STREAM_PTR(2); - header = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 2; - - /* if a header is present, fetch additional decoding parameters */ - if (header & 0x0008) { - CHECK_STREAM_PTR(8); - start_line = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 4; - lines_to_change = AV_RB16(&s->buf[stream_ptr]); - stream_ptr += 4; - } else { - start_line = 0; - lines_to_change = s->avctx->height; - } - - row_ptr = row_inc * start_line; while (lines_to_change--) { CHECK_STREAM_PTR(2); pixel_ptr = row_ptr + (s->buf[stream_ptr++] - 1) * 4; @@ -455,16 +346,13 @@ static void qtrle_decode_32bpp(QtrleContext *s) /* decode the run length code */ rle_code = -rle_code; CHECK_STREAM_PTR(4); - a = s->buf[stream_ptr++]; - r = s->buf[stream_ptr++]; - g = s->buf[stream_ptr++]; - b = s->buf[stream_ptr++]; - argb = (a << 24) | (r << 16) | (g << 8) | (b << 0); + argb = AV_RB32(s->buf + stream_ptr); + stream_ptr += 4; CHECK_PIXEL_PTR(rle_code * 4); while (rle_code--) { - *(unsigned int *)(&rgb[pixel_ptr]) = argb; + AV_WN32A(rgb + pixel_ptr, argb); pixel_ptr += 4; } } else { @@ -473,13 +361,10 @@ static void qtrle_decode_32bpp(QtrleContext *s) /* copy pixels directly to output */ while (rle_code--) { - a = s->buf[stream_ptr++]; - r = s->buf[stream_ptr++]; - g = s->buf[stream_ptr++]; - b = s->buf[stream_ptr++]; - argb = (a << 24) | (r << 16) | (g << 8) | (b << 0); - *(unsigned int *)(&rgb[pixel_ptr]) = argb; - pixel_ptr += 4; + argb = AV_RB32(s->buf + stream_ptr); + AV_WN32A(rgb + pixel_ptr, argb); + stream_ptr += 4; + pixel_ptr += 4; } } } @@ -487,17 +372,20 @@ static void qtrle_decode_32bpp(QtrleContext *s) } } -static int qtrle_decode_init(AVCodecContext *avctx) +static av_cold int qtrle_decode_init(AVCodecContext *avctx) { QtrleContext *s = avctx->priv_data; s->avctx = avctx; - switch (avctx->bits_per_sample) { + switch (avctx->bits_per_coded_sample) { case 1: + case 33: + avctx->pix_fmt = PIX_FMT_MONOWHITE; + break; + case 2: case 4: case 8: - case 33: case 34: case 36: case 40: @@ -518,10 +406,9 @@ static int qtrle_decode_init(AVCodecContext *avctx) default: av_log (avctx, AV_LOG_ERROR, "Unsupported colorspace: %d bits/sample?\n", - avctx->bits_per_sample); + avctx->bits_per_coded_sample); break; } - dsputil_init(&s->dsp, avctx); s->frame.data[0] = NULL; @@ -530,9 +417,14 @@ static int qtrle_decode_init(AVCodecContext *avctx) static int qtrle_decode_frame(AVCodecContext *avctx, void *data, int *data_size, - const uint8_t *buf, int buf_size) + AVPacket *avpkt) { + const uint8_t *buf = avpkt->data; + int buf_size = avpkt->size; QtrleContext *s = avctx->priv_data; + int header, start_line; + int stream_ptr, height, row_ptr; + int has_palette = 0; s->buf = buf; s->size = buf_size; @@ -545,57 +437,86 @@ static int qtrle_decode_frame(AVCodecContext *avctx, return -1; } - switch (avctx->bits_per_sample) { + /* check if this frame is even supposed to change */ + if (s->size < 8) + goto done; + + /* start after the chunk size */ + stream_ptr = 4; + + /* fetch the header */ + header = AV_RB16(&s->buf[stream_ptr]); + stream_ptr += 2; + + /* if a header is present, fetch additional decoding parameters */ + if (header & 0x0008) { + if(s->size < 14) + goto done; + start_line = AV_RB16(&s->buf[stream_ptr]); + stream_ptr += 4; + height = AV_RB16(&s->buf[stream_ptr]); + stream_ptr += 4; + } else { + start_line = 0; + height = s->avctx->height; + } + row_ptr = s->frame.linesize[0] * start_line; + + switch (avctx->bits_per_coded_sample) { case 1: case 33: - qtrle_decode_1bpp(s); + qtrle_decode_1bpp(s, stream_ptr, row_ptr, height); break; case 2: case 34: - qtrle_decode_2bpp(s); + qtrle_decode_2n4bpp(s, stream_ptr, row_ptr, height, 2); + has_palette = 1; break; case 4: case 36: - qtrle_decode_4bpp(s); - /* make the palette available on the way out */ - memcpy(s->frame.data[1], s->avctx->palctrl->palette, AVPALETTE_SIZE); - if (s->avctx->palctrl->palette_changed) { - s->frame.palette_has_changed = 1; - s->avctx->palctrl->palette_changed = 0; - } + qtrle_decode_2n4bpp(s, stream_ptr, row_ptr, height, 4); + has_palette = 1; break; case 8: case 40: - qtrle_decode_8bpp(s); - /* make the palette available on the way out */ - memcpy(s->frame.data[1], s->avctx->palctrl->palette, AVPALETTE_SIZE); - if (s->avctx->palctrl->palette_changed) { - s->frame.palette_has_changed = 1; - s->avctx->palctrl->palette_changed = 0; - } + qtrle_decode_8bpp(s, stream_ptr, row_ptr, height); + has_palette = 1; break; case 16: - qtrle_decode_16bpp(s); + qtrle_decode_16bpp(s, stream_ptr, row_ptr, height); break; case 24: - qtrle_decode_24bpp(s); + qtrle_decode_24bpp(s, stream_ptr, row_ptr, height); break; case 32: - qtrle_decode_32bpp(s); + qtrle_decode_32bpp(s, stream_ptr, row_ptr, height); break; default: av_log (s->avctx, AV_LOG_ERROR, "Unsupported colorspace: %d bits/sample?\n", - avctx->bits_per_sample); + avctx->bits_per_coded_sample); break; } + if(has_palette) { + const uint8_t *pal = av_packet_get_side_data(avpkt, AV_PKT_DATA_PALETTE, NULL); + + if (pal) { + s->frame.palette_has_changed = 1; + memcpy(s->pal, pal, AVPALETTE_SIZE); + } + + /* make the palette available on the way out */ + memcpy(s->frame.data[1], s->pal, AVPALETTE_SIZE); + } + +done: *data_size = sizeof(AVFrame); *(AVFrame*)data = s->frame; @@ -603,7 +524,7 @@ static int qtrle_decode_frame(AVCodecContext *avctx, return buf_size; } -static int qtrle_decode_end(AVCodecContext *avctx) +static av_cold int qtrle_decode_end(AVCodecContext *avctx) { QtrleContext *s = avctx->priv_data; @@ -613,15 +534,15 @@ static int qtrle_decode_end(AVCodecContext *avctx) return 0; } -AVCodec qtrle_decoder = { - "qtrle", - CODEC_TYPE_VIDEO, - CODEC_ID_QTRLE, - sizeof(QtrleContext), - qtrle_decode_init, - NULL, - qtrle_decode_end, - qtrle_decode_frame, - CODEC_CAP_DR1, +AVCodec ff_qtrle_decoder = { + .name = "qtrle", + .type = AVMEDIA_TYPE_VIDEO, + .id = CODEC_ID_QTRLE, + .priv_data_size = sizeof(QtrleContext), + .init = qtrle_decode_init, + .close = qtrle_decode_end, + .decode = qtrle_decode_frame, + .capabilities = CODEC_CAP_DR1, + .long_name = NULL_IF_CONFIG_SMALL("QuickTime Animation (RLE) video"), };