mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-23 12:43:46 +02:00
Merge commit '1877712c586df2261f2806f45388c77592b89d1e'
* commit '1877712c586df2261f2806f45388c77592b89d1e': svq3: move mb_{x,y,xy} to SVQ3Context Merged-by: Derek Buitenhuis <derek.buitenhuis@gmail.com>
This commit is contained in:
commit
c18535399d
@ -93,6 +93,9 @@ typedef struct SVQ3Context {
|
||||
int h_edge_pos;
|
||||
int v_edge_pos;
|
||||
int last_frame_output;
|
||||
|
||||
int mb_x, mb_y;
|
||||
int mb_xy;
|
||||
} SVQ3Context;
|
||||
|
||||
#define FULLPEL_MODE 1
|
||||
@ -392,11 +395,11 @@ static inline int svq3_mc_dir(SVQ3Context *s, int size, int mode,
|
||||
|
||||
for (i = 0; i < 16; i += part_height)
|
||||
for (j = 0; j < 16; j += part_width) {
|
||||
const int b_xy = (4 * sl->mb_x + (j >> 2)) +
|
||||
(4 * sl->mb_y + (i >> 2)) * h->b_stride;
|
||||
const int b_xy = (4 * s->mb_x + (j >> 2)) +
|
||||
(4 * s->mb_y + (i >> 2)) * h->b_stride;
|
||||
int dxy;
|
||||
x = 16 * sl->mb_x + j;
|
||||
y = 16 * sl->mb_y + i;
|
||||
x = 16 * s->mb_x + j;
|
||||
y = 16 * s->mb_y + i;
|
||||
k = (j >> 2 & 1) + (i >> 1 & 2) +
|
||||
(j >> 1 & 4) + (i & 8);
|
||||
|
||||
@ -533,7 +536,7 @@ static av_always_inline void hl_decode_mb_predict_luma(SVQ3Context *s,
|
||||
int nnz, tr;
|
||||
if (dir == DIAG_DOWN_LEFT_PRED || dir == VERT_LEFT_PRED) {
|
||||
const int topright_avail = (sl->topright_samples_available << i) & 0x8000;
|
||||
av_assert2(sl->mb_y || linesize <= block_offset[i]);
|
||||
av_assert2(s->mb_y || linesize <= block_offset[i]);
|
||||
if (!topright_avail) {
|
||||
tr = ptr[3 - linesize] * 0x01010101u;
|
||||
topright = (uint8_t *)&tr;
|
||||
@ -556,9 +559,9 @@ static av_always_inline void hl_decode_mb_predict_luma(SVQ3Context *s,
|
||||
|
||||
static void hl_decode_mb(SVQ3Context *s, const H264Context *h, H264SliceContext *sl)
|
||||
{
|
||||
const int mb_x = sl->mb_x;
|
||||
const int mb_y = sl->mb_y;
|
||||
const int mb_xy = sl->mb_xy;
|
||||
const int mb_x = s->mb_x;
|
||||
const int mb_y = s->mb_y;
|
||||
const int mb_xy = s->mb_xy;
|
||||
const int mb_type = h->cur_pic.mb_type[mb_xy];
|
||||
uint8_t *dest_y, *dest_cb, *dest_cr;
|
||||
int linesize, uvlinesize;
|
||||
@ -570,8 +573,8 @@ static void hl_decode_mb(SVQ3Context *s, const H264Context *h, H264SliceContext
|
||||
dest_cb = h->cur_pic.f->data[1] + mb_x * 8 + mb_y * sl->uvlinesize * block_h;
|
||||
dest_cr = h->cur_pic.f->data[2] + mb_x * 8 + mb_y * sl->uvlinesize * block_h;
|
||||
|
||||
s->vdsp.prefetch(dest_y + (sl->mb_x & 3) * 4 * sl->linesize + 64, sl->linesize, 4);
|
||||
s->vdsp.prefetch(dest_cb + (sl->mb_x & 7) * sl->uvlinesize + 64, dest_cr - dest_cb, 2);
|
||||
s->vdsp.prefetch(dest_y + (s->mb_x & 3) * 4 * sl->linesize + 64, sl->linesize, 4);
|
||||
s->vdsp.prefetch(dest_cb + (s->mb_x & 7) * sl->uvlinesize + 64, dest_cr - dest_cb, 2);
|
||||
|
||||
h->list_counts[mb_xy] = sl->list_count;
|
||||
|
||||
@ -612,21 +615,21 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type)
|
||||
int cbp = 0;
|
||||
uint32_t vlc;
|
||||
int8_t *top, *left;
|
||||
const int mb_xy = sl->mb_xy;
|
||||
const int b_xy = 4 * sl->mb_x + 4 * sl->mb_y * h->b_stride;
|
||||
const int mb_xy = s->mb_xy;
|
||||
const int b_xy = 4 * s->mb_x + 4 * s->mb_y * h->b_stride;
|
||||
|
||||
sl->top_samples_available = (sl->mb_y == 0) ? 0x33FF : 0xFFFF;
|
||||
sl->left_samples_available = (sl->mb_x == 0) ? 0x5F5F : 0xFFFF;
|
||||
sl->top_samples_available = (s->mb_y == 0) ? 0x33FF : 0xFFFF;
|
||||
sl->left_samples_available = (s->mb_x == 0) ? 0x5F5F : 0xFFFF;
|
||||
sl->topright_samples_available = 0xFFFF;
|
||||
|
||||
if (mb_type == 0) { /* SKIP */
|
||||
if (h->pict_type == AV_PICTURE_TYPE_P ||
|
||||
s->next_pic->mb_type[mb_xy] == -1) {
|
||||
svq3_mc_dir_part(s, 16 * sl->mb_x, 16 * sl->mb_y, 16, 16,
|
||||
svq3_mc_dir_part(s, 16 * s->mb_x, 16 * s->mb_y, 16, 16,
|
||||
0, 0, 0, 0, 0, 0);
|
||||
|
||||
if (h->pict_type == AV_PICTURE_TYPE_B)
|
||||
svq3_mc_dir_part(s, 16 * sl->mb_x, 16 * sl->mb_y, 16, 16,
|
||||
svq3_mc_dir_part(s, 16 * s->mb_x, 16 * s->mb_y, 16, 16,
|
||||
0, 0, 0, 0, 1, 1);
|
||||
|
||||
mb_type = MB_TYPE_SKIP;
|
||||
@ -658,7 +661,7 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type)
|
||||
*/
|
||||
|
||||
for (m = 0; m < 2; m++) {
|
||||
if (sl->mb_x > 0 && sl->intra4x4_pred_mode[h->mb2br_xy[mb_xy - 1] + 6] != -1) {
|
||||
if (s->mb_x > 0 && sl->intra4x4_pred_mode[h->mb2br_xy[mb_xy - 1] + 6] != -1) {
|
||||
for (i = 0; i < 4; i++)
|
||||
AV_COPY32(sl->mv_cache[m][scan8[0] - 1 + i * 8],
|
||||
h->cur_pic.motion_val[m][b_xy - 1 + i * h->b_stride]);
|
||||
@ -666,14 +669,14 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type)
|
||||
for (i = 0; i < 4; i++)
|
||||
AV_ZERO32(sl->mv_cache[m][scan8[0] - 1 + i * 8]);
|
||||
}
|
||||
if (sl->mb_y > 0) {
|
||||
if (s->mb_y > 0) {
|
||||
memcpy(sl->mv_cache[m][scan8[0] - 1 * 8],
|
||||
h->cur_pic.motion_val[m][b_xy - h->b_stride],
|
||||
4 * 2 * sizeof(int16_t));
|
||||
memset(&sl->ref_cache[m][scan8[0] - 1 * 8],
|
||||
(sl->intra4x4_pred_mode[h->mb2br_xy[mb_xy - h->mb_stride]] == -1) ? PART_NOT_AVAILABLE : 1, 4);
|
||||
|
||||
if (sl->mb_x < h->mb_width - 1) {
|
||||
if (s->mb_x < h->mb_width - 1) {
|
||||
AV_COPY32(sl->mv_cache[m][scan8[0] + 4 - 1 * 8],
|
||||
h->cur_pic.motion_val[m][b_xy - h->b_stride + 4]);
|
||||
sl->ref_cache[m][scan8[0] + 4 - 1 * 8] =
|
||||
@ -681,7 +684,7 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type)
|
||||
sl->intra4x4_pred_mode[h->mb2br_xy[mb_xy - h->mb_stride]] == -1) ? PART_NOT_AVAILABLE : 1;
|
||||
} else
|
||||
sl->ref_cache[m][scan8[0] + 4 - 1 * 8] = PART_NOT_AVAILABLE;
|
||||
if (sl->mb_x > 0) {
|
||||
if (s->mb_x > 0) {
|
||||
AV_COPY32(sl->mv_cache[m][scan8[0] - 1 - 1 * 8],
|
||||
h->cur_pic.motion_val[m][b_xy - h->b_stride - 1]);
|
||||
sl->ref_cache[m][scan8[0] - 1 - 1 * 8] =
|
||||
@ -721,19 +724,19 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type)
|
||||
|
||||
mb_type = MB_TYPE_16x16;
|
||||
} else if (mb_type == 8 || mb_type == 33) { /* INTRA4x4 */
|
||||
int8_t *i4x4 = sl->intra4x4_pred_mode + h->mb2br_xy[sl->mb_xy];
|
||||
int8_t *i4x4 = sl->intra4x4_pred_mode + h->mb2br_xy[s->mb_xy];
|
||||
int8_t *i4x4_cache = sl->intra4x4_pred_mode_cache;
|
||||
|
||||
memset(sl->intra4x4_pred_mode_cache, -1, 8 * 5 * sizeof(int8_t));
|
||||
|
||||
if (mb_type == 8) {
|
||||
if (sl->mb_x > 0) {
|
||||
if (s->mb_x > 0) {
|
||||
for (i = 0; i < 4; i++)
|
||||
sl->intra4x4_pred_mode_cache[scan8[0] - 1 + i * 8] = sl->intra4x4_pred_mode[h->mb2br_xy[mb_xy - 1] + 6 - i];
|
||||
if (sl->intra4x4_pred_mode_cache[scan8[0] - 1] == -1)
|
||||
sl->left_samples_available = 0x5F5F;
|
||||
}
|
||||
if (sl->mb_y > 0) {
|
||||
if (s->mb_y > 0) {
|
||||
sl->intra4x4_pred_mode_cache[4 + 8 * 0] = sl->intra4x4_pred_mode[h->mb2br_xy[mb_xy - h->mb_stride] + 0];
|
||||
sl->intra4x4_pred_mode_cache[5 + 8 * 0] = sl->intra4x4_pred_mode[h->mb2br_xy[mb_xy - h->mb_stride] + 1];
|
||||
sl->intra4x4_pred_mode_cache[6 + 8 * 0] = sl->intra4x4_pred_mode[h->mb2br_xy[mb_xy - h->mb_stride] + 2];
|
||||
@ -777,8 +780,8 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type)
|
||||
if (mb_type == 8) {
|
||||
ff_h264_check_intra4x4_pred_mode(h, sl);
|
||||
|
||||
sl->top_samples_available = (sl->mb_y == 0) ? 0x33FF : 0xFFFF;
|
||||
sl->left_samples_available = (sl->mb_x == 0) ? 0x5F5F : 0xFFFF;
|
||||
sl->top_samples_available = (s->mb_y == 0) ? 0x33FF : 0xFFFF;
|
||||
sl->left_samples_available = (s->mb_x == 0) ? 0x5F5F : 0xFFFF;
|
||||
} else {
|
||||
for (i = 0; i < 4; i++)
|
||||
memset(&sl->intra4x4_pred_mode_cache[scan8[0] + 8 * i], DC_128_PRED, 4);
|
||||
@ -906,7 +909,7 @@ static int svq3_decode_slice_header(AVCodecContext *avctx)
|
||||
SVQ3Context *s = avctx->priv_data;
|
||||
H264Context *h = &s->h;
|
||||
H264SliceContext *sl = &h->slice_ctx[0];
|
||||
const int mb_xy = sl->mb_xy;
|
||||
const int mb_xy = s->mb_xy;
|
||||
int i, header;
|
||||
unsigned slice_id;
|
||||
|
||||
@ -959,7 +962,7 @@ static int svq3_decode_slice_header(AVCodecContext *avctx)
|
||||
if ((header & 0x9F) == 2) {
|
||||
i = (h->mb_num < 64) ? 6 : (1 + av_log2(h->mb_num - 1));
|
||||
sl->mb_skip_run = get_bits(&h->gb, i) -
|
||||
(sl->mb_y * h->mb_width + sl->mb_x);
|
||||
(s->mb_y * h->mb_width + s->mb_x);
|
||||
} else {
|
||||
skip_bits1(&h->gb);
|
||||
sl->mb_skip_run = 0;
|
||||
@ -982,17 +985,17 @@ static int svq3_decode_slice_header(AVCodecContext *avctx)
|
||||
return AVERROR_INVALIDDATA;
|
||||
|
||||
/* reset intra predictors and invalidate motion vector references */
|
||||
if (sl->mb_x > 0) {
|
||||
if (s->mb_x > 0) {
|
||||
memset(sl->intra4x4_pred_mode + h->mb2br_xy[mb_xy - 1] + 3,
|
||||
-1, 4 * sizeof(int8_t));
|
||||
memset(sl->intra4x4_pred_mode + h->mb2br_xy[mb_xy - sl->mb_x],
|
||||
-1, 8 * sizeof(int8_t) * sl->mb_x);
|
||||
memset(sl->intra4x4_pred_mode + h->mb2br_xy[mb_xy - s->mb_x],
|
||||
-1, 8 * sizeof(int8_t) * s->mb_x);
|
||||
}
|
||||
if (sl->mb_y > 0) {
|
||||
if (s->mb_y > 0) {
|
||||
memset(sl->intra4x4_pred_mode + h->mb2br_xy[mb_xy - h->mb_stride],
|
||||
-1, 8 * sizeof(int8_t) * (h->mb_width - sl->mb_x));
|
||||
-1, 8 * sizeof(int8_t) * (h->mb_width - s->mb_x));
|
||||
|
||||
if (sl->mb_x > 0)
|
||||
if (s->mb_x > 0)
|
||||
sl->intra4x4_pred_mode[h->mb2br_xy[mb_xy - h->mb_stride - 1] + 3] = -1;
|
||||
}
|
||||
|
||||
@ -1436,10 +1439,10 @@ static int svq3_decode_frame(AVCodecContext *avctx, void *data,
|
||||
}
|
||||
}
|
||||
|
||||
for (sl->mb_y = 0; sl->mb_y < h->mb_height; sl->mb_y++) {
|
||||
for (sl->mb_x = 0; sl->mb_x < h->mb_width; sl->mb_x++) {
|
||||
for (s->mb_y = 0; s->mb_y < h->mb_height; s->mb_y++) {
|
||||
for (s->mb_x = 0; s->mb_x < h->mb_width; s->mb_x++) {
|
||||
unsigned mb_type;
|
||||
sl->mb_xy = sl->mb_x + sl->mb_y * h->mb_stride;
|
||||
s->mb_xy = s->mb_x + s->mb_y * h->mb_stride;
|
||||
|
||||
if ((get_bits_left(&h->gb)) <= 7) {
|
||||
if (((get_bits_count(&h->gb) & 7) == 0 ||
|
||||
@ -1459,7 +1462,7 @@ static int svq3_decode_frame(AVCodecContext *avctx, void *data,
|
||||
mb_type += 4;
|
||||
if (mb_type > 33 || svq3_decode_mb(s, mb_type)) {
|
||||
av_log(h->avctx, AV_LOG_ERROR,
|
||||
"error while decoding MB %d %d\n", sl->mb_x, sl->mb_y);
|
||||
"error while decoding MB %d %d\n", s->mb_x, s->mb_y);
|
||||
return -1;
|
||||
}
|
||||
|
||||
@ -1467,13 +1470,13 @@ static int svq3_decode_frame(AVCodecContext *avctx, void *data,
|
||||
hl_decode_mb(s, h, &h->slice_ctx[0]);
|
||||
|
||||
if (h->pict_type != AV_PICTURE_TYPE_B && !h->low_delay)
|
||||
h->cur_pic.mb_type[sl->mb_x + sl->mb_y * h->mb_stride] =
|
||||
h->cur_pic.mb_type[s->mb_x + s->mb_y * h->mb_stride] =
|
||||
(h->pict_type == AV_PICTURE_TYPE_P && mb_type < 8) ? (mb_type - 1) : -1;
|
||||
}
|
||||
|
||||
ff_draw_horiz_band(avctx, s->cur_pic->f,
|
||||
s->last_pic->f->data[0] ? s->last_pic->f : NULL,
|
||||
16 * sl->mb_y, 16, h->picture_structure, 0,
|
||||
16 * s->mb_y, 16, h->picture_structure, 0,
|
||||
h->low_delay);
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user