Commit 06798c6e authored by Michael Niedermayer's avatar Michael Niedermayer

escape130: Fix y_base handling, this fixes some artifacts, vissible in most videos.

Signed-off-by: 's avatarMichael Niedermayer <michaelni@gmx.at>
parent 8ff3d1cc
...@@ -26,6 +26,7 @@ ...@@ -26,6 +26,7 @@
typedef struct Escape130Context { typedef struct Escape130Context {
AVFrame frame; AVFrame frame;
uint8_t *bases;
} Escape130Context; } Escape130Context;
static int can_safely_read(GetBitContext* gb, int bits) { static int can_safely_read(GetBitContext* gb, int bits) {
...@@ -39,8 +40,11 @@ static int can_safely_read(GetBitContext* gb, int bits) { ...@@ -39,8 +40,11 @@ static int can_safely_read(GetBitContext* gb, int bits) {
*/ */
static av_cold int escape130_decode_init(AVCodecContext *avctx) static av_cold int escape130_decode_init(AVCodecContext *avctx)
{ {
Escape130Context *s = avctx->priv_data;
avctx->pix_fmt = PIX_FMT_YUV420P; avctx->pix_fmt = PIX_FMT_YUV420P;
s->bases= av_malloc(avctx->width * avctx->height /4);
return 0; return 0;
} }
...@@ -51,6 +55,8 @@ static av_cold int escape130_decode_close(AVCodecContext *avctx) ...@@ -51,6 +55,8 @@ static av_cold int escape130_decode_close(AVCodecContext *avctx)
if (s->frame.data[0]) if (s->frame.data[0])
avctx->release_buffer(avctx, &s->frame); avctx->release_buffer(avctx, &s->frame);
av_freep(&s->bases);
return 0; return 0;
} }
...@@ -108,6 +114,8 @@ static int escape130_decode_frame(AVCodecContext *avctx, ...@@ -108,6 +114,8 @@ static int escape130_decode_frame(AVCodecContext *avctx,
block_index, row_index = 0; block_index, row_index = 0;
unsigned y[4] = {0}, cb = 16, cr = 16; unsigned y[4] = {0}, cb = 16, cr = 16;
unsigned skip = -1; unsigned skip = -1;
unsigned y_base;
uint8_t *yb= s->bases;
AVFrame new_frame = { { 0 } }; AVFrame new_frame = { { 0 } };
...@@ -154,17 +162,18 @@ static int escape130_decode_frame(AVCodecContext *avctx, ...@@ -154,17 +162,18 @@ static int escape130_decode_frame(AVCodecContext *avctx,
y[1] = old_y[1] / 4; y[1] = old_y[1] / 4;
y[2] = old_y[old_y_stride] / 4; y[2] = old_y[old_y_stride] / 4;
y[3] = old_y[old_y_stride+1] / 4; y[3] = old_y[old_y_stride+1] / 4;
y_base= yb[0];
cb = old_cb[0] / 8; cb = old_cb[0] / 8;
cr = old_cr[0] / 8; cr = old_cr[0] / 8;
} else { } else {
y[0] = y[1] = y[2] = y[3] = 0; y_base=y[0] = y[1] = y[2] = y[3] = 0;
cb = cr = 16; cb = cr = 16;
} }
} else { } else {
if (get_bits1(&gb)) { if (get_bits1(&gb)) {
unsigned sign_selector = get_bits(&gb, 6); unsigned sign_selector = get_bits(&gb, 6);
unsigned difference_selector = get_bits(&gb, 2); unsigned difference_selector = get_bits(&gb, 2);
unsigned y_base = 2 * get_bits(&gb, 5); y_base = 2 * get_bits(&gb, 5);
static const uint8_t offset_table[] = {2, 4, 10, 20}; static const uint8_t offset_table[] = {2, 4, 10, 20};
static const int8_t sign_table[64][4] = static const int8_t sign_table[64][4] =
{ {0, 0, 0, 0}, { {0, 0, 0, 0},
...@@ -230,15 +239,14 @@ static int escape130_decode_frame(AVCodecContext *avctx, ...@@ -230,15 +239,14 @@ static int escape130_decode_frame(AVCodecContext *avctx,
} }
} else if (get_bits1(&gb)) { } else if (get_bits1(&gb)) {
if (get_bits1(&gb)) { if (get_bits1(&gb)) {
unsigned y_base = get_bits(&gb, 6); y_base = get_bits(&gb, 6);
for (i = 0; i < 4; i++)
y[i] = y_base;
} else { } else {
unsigned adjust_index = get_bits(&gb, 3); unsigned adjust_index = get_bits(&gb, 3);
static const int8_t adjust[] = {-4, -3, -2, -1, 1, 2, 3, 4}; static const int8_t adjust[] = {-4, -3, -2, -1, 1, 2, 3, 4};
for (i = 0; i < 4; i++) y_base = (y_base + adjust[adjust_index]) & 63;
y[i] = av_clip(y[i] + adjust[adjust_index], 0, 63);
} }
for (i = 0; i < 4; i++)
y[i] = y_base;
} }
if (get_bits1(&gb)) { if (get_bits1(&gb)) {
...@@ -255,6 +263,7 @@ static int escape130_decode_frame(AVCodecContext *avctx, ...@@ -255,6 +263,7 @@ static int escape130_decode_frame(AVCodecContext *avctx,
} }
} }
} }
*yb++= y_base;
new_y[0] = y[0] * 4; new_y[0] = y[0] * 4;
new_y[1] = y[1] * 4; new_y[1] = y[1] * 4;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment