* Supports: BGR24 (RGB 24bpp)
*/
+#include "config_components.h"
+
#include <stdio.h>
#include <stdlib.h>
#include "libavutil/pixdesc.h"
#include "avcodec.h"
#include "bytestream.h"
-#include "internal.h"
+#include "codec_internal.h"
#include "lcl.h"
#include "thread.h"
#if CONFIG_ZLIB_DECODER
+#include "zlib_wrapper.h"
#include <zlib.h>
#endif
// Decompression buffer
unsigned char* decomp_buf;
#if CONFIG_ZLIB_DECODER
- z_stream zstream;
+ FFZStream zstream;
#endif
} LclDecContext;
static int zlib_decomp(AVCodecContext *avctx, const uint8_t *src, int src_len, int offset, int expected)
{
LclDecContext *c = avctx->priv_data;
- int zret = inflateReset(&c->zstream);
+ z_stream *const zstream = &c->zstream.zstream;
+ int zret = inflateReset(zstream);
if (zret != Z_OK) {
av_log(avctx, AV_LOG_ERROR, "Inflate reset error: %d\n", zret);
return AVERROR_UNKNOWN;
}
- c->zstream.next_in = src;
- c->zstream.avail_in = src_len;
- c->zstream.next_out = c->decomp_buf + offset;
- c->zstream.avail_out = c->decomp_size - offset;
- zret = inflate(&c->zstream, Z_FINISH);
+ zstream->next_in = src;
+ zstream->avail_in = src_len;
+ zstream->next_out = c->decomp_buf + offset;
+ zstream->avail_out = c->decomp_size - offset;
+ zret = inflate(zstream, Z_FINISH);
if (zret != Z_OK && zret != Z_STREAM_END) {
av_log(avctx, AV_LOG_ERROR, "Inflate error: %d\n", zret);
return AVERROR_UNKNOWN;
}
- if (expected != (unsigned int)c->zstream.total_out) {
+ if (expected != (unsigned int)zstream->total_out) {
av_log(avctx, AV_LOG_ERROR, "Decoded size differs (%d != %lu)\n",
- expected, c->zstream.total_out);
+ expected, zstream->total_out);
+ if (expected > (unsigned int)zstream->total_out)
+ return (unsigned int)zstream->total_out;
return AVERROR_UNKNOWN;
}
- return c->zstream.total_out;
+ return zstream->total_out;
}
#endif
-static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
+static int decode_frame(AVCodecContext *avctx, AVFrame *frame,
+ int *got_frame, AVPacket *avpkt)
{
- AVFrame *frame = data;
- ThreadFrame tframe = { .f = data };
const uint8_t *buf = avpkt->data;
int buf_size = avpkt->size;
LclDecContext * const c = avctx->priv_data;
- unsigned int pixel_ptr;
+ ptrdiff_t pixel_ptr;
int row, col;
unsigned char *encoded = avpkt->data, *outptr;
uint8_t *y_out, *u_out, *v_out;
- unsigned int width = avctx->width; // Real image width
- unsigned int height = avctx->height; // Real image height
+ int width = avctx->width; // Real image width
+ int height = avctx->height; // Real image height
unsigned int mszh_dlen;
unsigned char yq, y1q, uq, vq;
int uqvq, ret;
unsigned int mthread_inlen, mthread_outlen;
unsigned int len = buf_size;
- int linesize;
+ int linesize, offset;
- if ((ret = ff_thread_get_buffer(avctx, &tframe, 0)) < 0)
+ if ((ret = ff_thread_get_buffer(avctx, frame, 0)) < 0)
return ret;
outptr = frame->data[0]; // Output image pointer
if (c->decomp_size != mszh_dlen) {
av_log(avctx, AV_LOG_ERROR, "Decoded size differs (%d != %d)\n",
c->decomp_size, mszh_dlen);
- return AVERROR_INVALIDDATA;
+ if (c->decomp_size != mszh_dlen &&
+ c->decomp_size != mszh_dlen + 2) // YUV420 306x306 is missing 2 bytes
+ return AVERROR_INVALIDDATA;
}
encoded = c->decomp_buf;
len = mszh_dlen;
break;
case COMP_MSZH_NOCOMP: {
int bppx2;
+ int aligned_width = width;
switch (c->imgtype) {
case IMGTYPE_YUV111:
case IMGTYPE_RGB24:
bppx2 = 6;
break;
case IMGTYPE_YUV422:
+ aligned_width &= ~3;
case IMGTYPE_YUV211:
bppx2 = 4;
break;
case IMGTYPE_YUV411:
+ aligned_width &= ~3;
case IMGTYPE_YUV420:
bppx2 = 3;
break;
bppx2 = 0; // will error out below
break;
}
- if (len < ((width * height * bppx2) >> 1))
+ if (len < ((aligned_width * height * bppx2) >> 1))
return AVERROR_INVALIDDATA;
break;
}
ret = zlib_decomp(avctx, buf + 8 + mthread_inlen, len - 8 - mthread_inlen,
mthread_outlen, mthread_outlen);
if (ret < 0) return ret;
+ len = c->decomp_size;
} else {
int ret = zlib_decomp(avctx, buf, len, 0, c->decomp_size);
if (ret < 0) return ret;
+ len = ret;
}
encoded = c->decomp_buf;
- len = c->decomp_size;
break;
#endif
default:
}
break;
case IMGTYPE_YUV422:
+ pixel_ptr = 0;
for (row = 0; row < height; row++) {
- pixel_ptr = row * width * 2;
yq = uq = vq =0;
for (col = 0; col < width/4; col++) {
encoded[pixel_ptr] = yq -= encoded[pixel_ptr];
}
break;
case IMGTYPE_YUV411:
+ pixel_ptr = 0;
for (row = 0; row < height; row++) {
- pixel_ptr = row * width / 2 * 3;
yq = uq = vq =0;
for (col = 0; col < width/4; col++) {
encoded[pixel_ptr] = yq -= encoded[pixel_ptr];
/* Convert colorspace */
y_out = frame->data[0] + (height - 1) * frame->linesize[0];
- u_out = frame->data[1] + (height - 1) * frame->linesize[1];
- v_out = frame->data[2] + (height - 1) * frame->linesize[2];
+ offset = (height - 1) * frame->linesize[1];
+ u_out = FF_PTR_ADD(frame->data[1], offset);
+ offset = (height - 1) * frame->linesize[2];
+ v_out = FF_PTR_ADD(frame->data[2], offset);
switch (c->imgtype) {
case IMGTYPE_YUV111:
for (row = 0; row < height; row++) {
v_out[ col >> 1 ] = *encoded++ + 128;
v_out[(col >> 1) + 1] = *encoded++ + 128;
}
+ if (col && col < width) {
+ u_out[ col >> 1 ] = u_out[(col>>1) - 1];
+ v_out[ col >> 1 ] = v_out[(col>>1) - 1];
+ }
+
y_out -= frame->linesize[0];
u_out -= frame->linesize[1];
v_out -= frame->linesize[2];
u_out[col >> 2] = *encoded++ + 128;
v_out[col >> 2] = *encoded++ + 128;
}
+ if (col && col < width) {
+ u_out[col >> 2] = u_out[(col>>2) - 1];
+ v_out[col >> 2] = v_out[(col>>2) - 1];
+ }
y_out -= frame->linesize[0];
u_out -= frame->linesize[1];
v_out -= frame->linesize[2];
return AVERROR_INVALIDDATA;
}
- frame->key_frame = 1;
+ frame->flags |= AV_FRAME_FLAG_KEY;
frame->pict_type = AV_PICTURE_TYPE_I;
*got_frame = 1;
FFALIGN(avctx->height, 4);
unsigned int max_decomp_size;
int subsample_h, subsample_v;
+ int partial_h_supported = 0;
if (avctx->extradata_size < 8) {
av_log(avctx, AV_LOG_ERROR, "Extradata size too small.\n");
av_log(avctx, AV_LOG_DEBUG, "Image type is YUV 1:1:1.\n");
break;
case IMGTYPE_YUV422:
- c->decomp_size = basesize * 2;
+ c->decomp_size = (avctx->width & ~3) * avctx->height * 2;
max_decomp_size = max_basesize * 2;
avctx->pix_fmt = AV_PIX_FMT_YUV422P;
av_log(avctx, AV_LOG_DEBUG, "Image type is YUV 4:2:2.\n");
- if (avctx->width % 4) {
- avpriv_request_sample(avctx, "Unsupported dimensions");
- return AVERROR_INVALIDDATA;
- }
+ partial_h_supported = 1;
break;
case IMGTYPE_RGB24:
- c->decomp_size = basesize * 3;
+ c->decomp_size = FFALIGN(avctx->width*3, 4) * avctx->height;
max_decomp_size = max_basesize * 3;
avctx->pix_fmt = AV_PIX_FMT_BGR24;
av_log(avctx, AV_LOG_DEBUG, "Image type is RGB 24.\n");
break;
case IMGTYPE_YUV411:
- c->decomp_size = basesize / 2 * 3;
+ c->decomp_size = (avctx->width & ~3) * avctx->height / 2 * 3;
max_decomp_size = max_basesize / 2 * 3;
avctx->pix_fmt = AV_PIX_FMT_YUV411P;
av_log(avctx, AV_LOG_DEBUG, "Image type is YUV 4:1:1.\n");
+ partial_h_supported = 1;
break;
case IMGTYPE_YUV211:
c->decomp_size = basesize * 2;
}
av_pix_fmt_get_chroma_sub_sample(avctx->pix_fmt, &subsample_h, &subsample_v);
- if (avctx->width % (1<<subsample_h) || avctx->height % (1<<subsample_v)) {
+ if ((avctx->width % (1<<subsample_h) && !partial_h_supported) || avctx->height % (1<<subsample_v)) {
avpriv_request_sample(avctx, "Unsupported dimensions");
return AVERROR_INVALIDDATA;
}
/* If needed init zlib */
#if CONFIG_ZLIB_DECODER
- if (avctx->codec_id == AV_CODEC_ID_ZLIB) {
- int zret;
- c->zstream.zalloc = Z_NULL;
- c->zstream.zfree = Z_NULL;
- c->zstream.opaque = Z_NULL;
- zret = inflateInit(&c->zstream);
- if (zret != Z_OK) {
- av_log(avctx, AV_LOG_ERROR, "Inflate init error: %d\n", zret);
- av_freep(&c->decomp_buf);
- return AVERROR_UNKNOWN;
- }
- }
+ if (avctx->codec_id == AV_CODEC_ID_ZLIB)
+ return ff_inflate_init(&c->zstream, avctx);
#endif
return 0;
av_freep(&c->decomp_buf);
#if CONFIG_ZLIB_DECODER
- if (avctx->codec_id == AV_CODEC_ID_ZLIB)
- inflateEnd(&c->zstream);
+ ff_inflate_end(&c->zstream);
#endif
return 0;
}
#if CONFIG_MSZH_DECODER
-AVCodec ff_mszh_decoder = {
- .name = "mszh",
- .long_name = NULL_IF_CONFIG_SMALL("LCL (LossLess Codec Library) MSZH"),
- .type = AVMEDIA_TYPE_VIDEO,
- .id = AV_CODEC_ID_MSZH,
+const FFCodec ff_mszh_decoder = {
+ .p.name = "mszh",
+ CODEC_LONG_NAME("LCL (LossLess Codec Library) MSZH"),
+ .p.type = AVMEDIA_TYPE_VIDEO,
+ .p.id = AV_CODEC_ID_MSZH,
+ .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_FRAME_THREADS,
.priv_data_size = sizeof(LclDecContext),
.init = decode_init,
.close = decode_end,
- .decode = decode_frame,
- .capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_FRAME_THREADS,
- .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
+ FF_CODEC_DECODE_CB(decode_frame),
+ .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
};
#endif
#if CONFIG_ZLIB_DECODER
-AVCodec ff_zlib_decoder = {
- .name = "zlib",
- .long_name = NULL_IF_CONFIG_SMALL("LCL (LossLess Codec Library) ZLIB"),
- .type = AVMEDIA_TYPE_VIDEO,
- .id = AV_CODEC_ID_ZLIB,
+const FFCodec ff_zlib_decoder = {
+ .p.name = "zlib",
+ CODEC_LONG_NAME("LCL (LossLess Codec Library) ZLIB"),
+ .p.type = AVMEDIA_TYPE_VIDEO,
+ .p.id = AV_CODEC_ID_ZLIB,
+ .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_FRAME_THREADS,
.priv_data_size = sizeof(LclDecContext),
.init = decode_init,
.close = decode_end,
- .decode = decode_frame,
- .capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_FRAME_THREADS,
- .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
+ FF_CODEC_DECODE_CB(decode_frame),
+ .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
};
#endif