44 #include "libavutil/mem.h"
45 #include "libavutil/pixdesc.h"
51 #if CONFIG_ZLIB_DECODER
71 #if CONFIG_ZLIB_DECODER
81 static unsigned int mszh_decomp(
const unsigned char * srcptr,
int srclen,
unsigned char * destptr,
unsigned int destsize)
83 unsigned char *destptr_bak = destptr;
84 unsigned char *destptr_end = destptr + destsize;
85 const unsigned char *srcptr_end = srcptr + srclen;
86 unsigned mask = *srcptr++;
87 unsigned maskbit = 0x80;
89 while (srcptr < srcptr_end && destptr < destptr_end) {
90 if (!(mask & maskbit)) {
91 memcpy(destptr, srcptr, 4);
95 unsigned ofs = bytestream_get_le16(&srcptr);
96 unsigned cnt = (ofs >> 11) + 1;
98 ofs =
FFMIN(ofs, destptr - destptr_bak);
100 cnt =
FFMIN(cnt, destptr_end - destptr);
106 memset(destptr, 0, cnt);
114 if (destptr_end - destptr < 32 || srcptr_end - srcptr < 32)
break;
115 memcpy(destptr, srcptr, 32);
124 return destptr - destptr_bak;
128 #if CONFIG_ZLIB_DECODER
139 int zret = inflateReset(&c->zstream);
144 c->zstream.next_in = (
uint8_t *)src;
145 c->zstream.avail_in = src_len;
148 zret = inflate(&c->zstream, Z_FINISH);
149 if (zret != Z_OK && zret != Z_STREAM_END) {
153 if (expected != (
unsigned int)c->zstream.total_out) {
155 expected, c->zstream.total_out);
158 return c->zstream.total_out;
171 int buf_size = avpkt->
size;
173 unsigned char *encoded = (
unsigned char *)buf;
174 unsigned int pixel_ptr;
176 unsigned char *outptr;
177 uint8_t *y_out, *u_out, *v_out;
180 unsigned int mszh_dlen;
181 unsigned char yq, y1q, uq, vq;
183 unsigned int mthread_inlen, mthread_outlen;
184 unsigned int len = buf_size;
206 mthread_inlen =
AV_RL32(encoded);
211 mthread_inlen =
FFMIN(mthread_inlen, len - 8);
212 mthread_outlen =
AV_RL32(encoded+4);
215 if (mthread_outlen != mszh_dlen) {
217 mthread_outlen, mszh_dlen);
220 mszh_dlen =
mszh_decomp(encoded + 8 + mthread_inlen, len - 8 - mthread_inlen,
222 if (mthread_outlen != mszh_dlen) {
224 mthread_outlen, mszh_dlen);
259 if (len < ((width * height * bppx2) >> 1))
268 #if CONFIG_ZLIB_DECODER
274 len == width * height * 3) {
282 mthread_inlen =
AV_RL32(encoded);
283 mthread_inlen =
FFMIN(mthread_inlen, len - 8);
284 mthread_outlen =
AV_RL32(encoded+4);
286 ret = zlib_decomp(avctx, encoded + 8, mthread_inlen, 0, mthread_outlen);
287 if (ret < 0)
return ret;
288 ret = zlib_decomp(avctx, encoded + 8 + mthread_inlen, len - 8 - mthread_inlen,
289 mthread_outlen, mthread_outlen);
290 if (ret < 0)
return ret;
292 int ret = zlib_decomp(avctx, encoded, len, 0, c->
decomp_size);
293 if (ret < 0)
return ret;
300 av_log(avctx,
AV_LOG_ERROR,
"BUG! Unknown codec in frame decoder compression switch.\n");
310 for (row = 0; row <
height; row++) {
311 pixel_ptr = row * width * 3;
312 yq = encoded[pixel_ptr++];
313 uqvq =
AV_RL16(encoded+pixel_ptr);
315 for (col = 1; col <
width; col++) {
316 encoded[pixel_ptr] = yq -= encoded[pixel_ptr];
317 uqvq -=
AV_RL16(encoded+pixel_ptr+1);
318 AV_WL16(encoded+pixel_ptr+1, uqvq);
324 for (row = 0; row <
height; row++) {
325 pixel_ptr = row * width * 2;
327 for (col = 0; col < width/4; col++) {
328 encoded[pixel_ptr] = yq -= encoded[pixel_ptr];
329 encoded[pixel_ptr+1] = yq -= encoded[pixel_ptr+1];
330 encoded[pixel_ptr+2] = yq -= encoded[pixel_ptr+2];
331 encoded[pixel_ptr+3] = yq -= encoded[pixel_ptr+3];
332 encoded[pixel_ptr+4] = uq -= encoded[pixel_ptr+4];
333 encoded[pixel_ptr+5] = uq -= encoded[pixel_ptr+5];
334 encoded[pixel_ptr+6] = vq -= encoded[pixel_ptr+6];
335 encoded[pixel_ptr+7] = vq -= encoded[pixel_ptr+7];
341 for (row = 0; row <
height; row++) {
342 pixel_ptr = row * width / 2 * 3;
344 for (col = 0; col < width/4; col++) {
345 encoded[pixel_ptr] = yq -= encoded[pixel_ptr];
346 encoded[pixel_ptr+1] = yq -= encoded[pixel_ptr+1];
347 encoded[pixel_ptr+2] = yq -= encoded[pixel_ptr+2];
348 encoded[pixel_ptr+3] = yq -= encoded[pixel_ptr+3];
349 encoded[pixel_ptr+4] = uq -= encoded[pixel_ptr+4];
350 encoded[pixel_ptr+5] = vq -= encoded[pixel_ptr+5];
356 for (row = 0; row <
height; row++) {
357 pixel_ptr = row * width * 2;
359 for (col = 0; col < width/2; col++) {
360 encoded[pixel_ptr] = yq -= encoded[pixel_ptr];
361 encoded[pixel_ptr+1] = yq -= encoded[pixel_ptr+1];
362 encoded[pixel_ptr+2] = uq -= encoded[pixel_ptr+2];
363 encoded[pixel_ptr+3] = vq -= encoded[pixel_ptr+3];
369 for (row = 0; row < height/2; row++) {
370 pixel_ptr = row * width * 3;
371 yq = y1q = uq = vq =0;
372 for (col = 0; col < width/2; col++) {
373 encoded[pixel_ptr] = yq -= encoded[pixel_ptr];
374 encoded[pixel_ptr+1] = yq -= encoded[pixel_ptr+1];
375 encoded[pixel_ptr+2] = y1q -= encoded[pixel_ptr+2];
376 encoded[pixel_ptr+3] = y1q -= encoded[pixel_ptr+3];
377 encoded[pixel_ptr+4] = uq -= encoded[pixel_ptr+4];
378 encoded[pixel_ptr+5] = vq -= encoded[pixel_ptr+5];
395 for (row = 0; row <
height; row++) {
396 for (col = 0; col <
width; col++) {
397 y_out[col] = *encoded++;
398 u_out[col] = *encoded++ + 128;
399 v_out[col] = *encoded++ + 128;
407 for (row = 0; row <
height; row++) {
408 for (col = 0; col < width - 3; col += 4) {
409 memcpy(y_out + col, encoded, 4);
411 u_out[ col >> 1 ] = *encoded++ + 128;
412 u_out[(col >> 1) + 1] = *encoded++ + 128;
413 v_out[ col >> 1 ] = *encoded++ + 128;
414 v_out[(col >> 1) + 1] = *encoded++ + 128;
422 for (row = height - 1; row >= 0; row--) {
424 memcpy(outptr + pixel_ptr, encoded, 3 * width);
425 encoded += 3 *
width;
429 for (row = 0; row <
height; row++) {
430 for (col = 0; col < width - 3; col += 4) {
431 memcpy(y_out + col, encoded, 4);
433 u_out[col >> 2] = *encoded++ + 128;
434 v_out[col >> 2] = *encoded++ + 128;
442 for (row = 0; row <
height; row++) {
443 for (col = 0; col < width - 1; col += 2) {
444 memcpy(y_out + col, encoded, 2);
446 u_out[col >> 1] = *encoded++ + 128;
447 v_out[col >> 1] = *encoded++ + 128;
457 for (row = 0; row < height - 1; row += 2) {
458 for (col = 0; col < width - 1; col += 2) {
459 memcpy(y_out + col, encoded, 2);
461 memcpy(y_out + col - c->
pic.
linesize[0], encoded, 2);
463 u_out[col >> 1] = *encoded++ + 128;
464 v_out[col >> 1] = *encoded++ + 128;
491 unsigned int basesize = avctx->
width * avctx->
height;
494 unsigned int max_decomp_size;
495 int subsample_h, subsample_v;
506 av_log(avctx,
AV_LOG_ERROR,
"Codec id and codec type mismatch. This should not happen.\n");
513 max_decomp_size = max_basesize * 3;
519 max_decomp_size = max_basesize * 2;
522 if (avctx->
width % 4) {
528 max_decomp_size = max_basesize * 3;
534 max_decomp_size = max_basesize / 2 * 3;
540 max_decomp_size = max_basesize * 2;
546 max_decomp_size = max_basesize / 2 * 3;
556 if (avctx->
width % (1<<subsample_h) || avctx->
height % (1<<subsample_v)) {
577 #if CONFIG_ZLIB_DECODER
623 #if CONFIG_ZLIB_DECODER
626 c->zstream.zalloc = Z_NULL;
627 c->zstream.zfree = Z_NULL;
628 c->zstream.opaque = Z_NULL;
629 zret = inflateInit(&c->zstream);
653 #if CONFIG_ZLIB_DECODER
655 inflateEnd(&c->zstream);
661 #if CONFIG_MSZH_DECODER
675 #if CONFIG_ZLIB_DECODER