avcodec/h264: keep track of which frames used gray references
Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
This commit is contained in:
parent
e4337606e1
commit
29f6c9b04d
@ -96,6 +96,7 @@ static void h264_copy_picture_params(H264Picture *dst, const H264Picture *src)
|
|||||||
dst->field_picture = src->field_picture;
|
dst->field_picture = src->field_picture;
|
||||||
dst->reference = src->reference;
|
dst->reference = src->reference;
|
||||||
dst->recovered = src->recovered;
|
dst->recovered = src->recovered;
|
||||||
|
dst->gray = src->gray;
|
||||||
dst->invalid_gap = src->invalid_gap;
|
dst->invalid_gap = src->invalid_gap;
|
||||||
dst->sei_recovery_frame_cnt = src->sei_recovery_frame_cnt;
|
dst->sei_recovery_frame_cnt = src->sei_recovery_frame_cnt;
|
||||||
dst->mb_width = src->mb_width;
|
dst->mb_width = src->mb_width;
|
||||||
|
@ -457,6 +457,7 @@ int ff_h264_update_thread_context(AVCodecContext *dst,
|
|||||||
h->poc.prev_frame_num = h->poc.frame_num;
|
h->poc.prev_frame_num = h->poc.frame_num;
|
||||||
|
|
||||||
h->recovery_frame = h1->recovery_frame;
|
h->recovery_frame = h1->recovery_frame;
|
||||||
|
h->non_gray = h1->non_gray;
|
||||||
|
|
||||||
return err;
|
return err;
|
||||||
}
|
}
|
||||||
@ -1544,11 +1545,15 @@ static int h264_field_start(H264Context *h, const H264SliceContext *sl,
|
|||||||
if (ret < 0)
|
if (ret < 0)
|
||||||
return ret;
|
return ret;
|
||||||
h->short_ref[0]->poc = prev->poc + 2U;
|
h->short_ref[0]->poc = prev->poc + 2U;
|
||||||
|
h->short_ref[0]->gray = prev->gray;
|
||||||
ff_thread_report_progress(&h->short_ref[0]->tf, INT_MAX, 0);
|
ff_thread_report_progress(&h->short_ref[0]->tf, INT_MAX, 0);
|
||||||
if (h->short_ref[0]->field_picture)
|
if (h->short_ref[0]->field_picture)
|
||||||
ff_thread_report_progress(&h->short_ref[0]->tf, INT_MAX, 1);
|
ff_thread_report_progress(&h->short_ref[0]->tf, INT_MAX, 1);
|
||||||
} else if (!h->frame_recovered && !h->avctx->hwaccel)
|
} else if (!h->frame_recovered) {
|
||||||
color_frame(h->short_ref[0]->f, c);
|
if (!h->avctx->hwaccel)
|
||||||
|
color_frame(h->short_ref[0]->f, c);
|
||||||
|
h->short_ref[0]->gray = 1;
|
||||||
|
}
|
||||||
h->short_ref[0]->frame_num = h->poc.prev_frame_num;
|
h->short_ref[0]->frame_num = h->poc.prev_frame_num;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -2007,6 +2012,19 @@ static int h264_slice_init(H264Context *h, H264SliceContext *sl,
|
|||||||
(sl->ref_list[j][i].reference & 3);
|
(sl->ref_list[j][i].reference & 3);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (sl->slice_type_nos == AV_PICTURE_TYPE_I) {
|
||||||
|
h->cur_pic_ptr->gray = 0;
|
||||||
|
h->non_gray = 1;
|
||||||
|
} else {
|
||||||
|
int gray = 0;
|
||||||
|
for (j = 0; j < sl->list_count; j++) {
|
||||||
|
for (i = 0; i < sl->ref_count[j]; i++) {
|
||||||
|
gray |= sl->ref_list[j][i].parent->gray;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
h->cur_pic_ptr->gray = gray;
|
||||||
|
}
|
||||||
|
|
||||||
if (h->avctx->debug & FF_DEBUG_PICT_INFO) {
|
if (h->avctx->debug & FF_DEBUG_PICT_INFO) {
|
||||||
av_log(h->avctx, AV_LOG_DEBUG,
|
av_log(h->avctx, AV_LOG_DEBUG,
|
||||||
"slice:%d %c mb:%d %c%s%s frame:%d poc:%d/%d ref:%d/%d qp:%d loop:%d:%d:%d weight:%d%s %s\n",
|
"slice:%d %c mb:%d %c%s%s frame:%d poc:%d/%d ref:%d/%d qp:%d loop:%d:%d:%d weight:%d%s %s\n",
|
||||||
|
@ -492,6 +492,7 @@ static void h264_decode_flush(AVCodecContext *avctx)
|
|||||||
ff_h264_unref_picture(&h->cur_pic);
|
ff_h264_unref_picture(&h->cur_pic);
|
||||||
|
|
||||||
h->mb_y = 0;
|
h->mb_y = 0;
|
||||||
|
h->non_gray = 0;
|
||||||
|
|
||||||
ff_h264_free_tables(h);
|
ff_h264_free_tables(h);
|
||||||
h->context_initialized = 0;
|
h->context_initialized = 0;
|
||||||
|
@ -154,6 +154,8 @@ typedef struct H264Picture {
|
|||||||
|
|
||||||
/// RefStruct reference; its pointee is shared between decoding threads.
|
/// RefStruct reference; its pointee is shared between decoding threads.
|
||||||
atomic_int *decode_error_flags;
|
atomic_int *decode_error_flags;
|
||||||
|
|
||||||
|
int gray;
|
||||||
} H264Picture;
|
} H264Picture;
|
||||||
|
|
||||||
typedef struct H264Ref {
|
typedef struct H264Ref {
|
||||||
@ -567,6 +569,8 @@ typedef struct H264Context {
|
|||||||
struct FFRefStructPool *ref_index_pool;
|
struct FFRefStructPool *ref_index_pool;
|
||||||
struct FFRefStructPool *decode_error_flags_pool;
|
struct FFRefStructPool *decode_error_flags_pool;
|
||||||
int ref2frm[MAX_SLICES][2][64]; ///< reference to frame number lists, used in the loop filter, the first 2 are for -2,-1
|
int ref2frm[MAX_SLICES][2][64]; ///< reference to frame number lists, used in the loop filter, the first 2 are for -2,-1
|
||||||
|
|
||||||
|
int non_gray; ///< Did we encounter a intra frame after a gray gap frame
|
||||||
} H264Context;
|
} H264Context;
|
||||||
|
|
||||||
extern const uint16_t ff_h264_mb_sizes[4];
|
extern const uint16_t ff_h264_mb_sizes[4];
|
||||||
|
Loading…
x
Reference in New Issue
Block a user