Commit 2dc954e0 authored by Hendrik Leppkes's avatar Hendrik Leppkes

Merge commit '44d16df4'

* commit '44d16df4':
  h264_parser: eliminate H264SliceContext usage
Merged-by: 's avatarHendrik Leppkes <h.leppkes@gmail.com>
parents d970f7ba 44d16df4
...@@ -139,35 +139,35 @@ found: ...@@ -139,35 +139,35 @@ found:
return i - (state & 5) - 5 * (state > 7); return i - (state & 5) - 5 * (state > 7);
} }
static int scan_mmco_reset(AVCodecParserContext *s) static int scan_mmco_reset(AVCodecParserContext *s, GetBitContext *gb)
{ {
H264PredWeightTable pwt;
int slice_type_nos = s->pict_type & 3;
H264ParseContext *p = s->priv_data; H264ParseContext *p = s->priv_data;
H264Context *h = &p->h; H264Context *h = &p->h;
H264SliceContext *sl = &h->slice_ctx[0];
int list_count, ref_count[2]; int list_count, ref_count[2];
sl->slice_type_nos = s->pict_type & 3;
if (h->pps.redundant_pic_cnt_present) if (h->pps.redundant_pic_cnt_present)
get_ue_golomb(&sl->gb); // redundant_pic_count get_ue_golomb(gb); // redundant_pic_count
if (sl->slice_type_nos == AV_PICTURE_TYPE_B) if (slice_type_nos == AV_PICTURE_TYPE_B)
get_bits1(&sl->gb); // direct_spatial_mv_pred get_bits1(gb); // direct_spatial_mv_pred
if (ff_h264_parse_ref_count(&list_count, ref_count, &sl->gb, &h->pps, if (ff_h264_parse_ref_count(&list_count, ref_count, gb, &h->pps,
sl->slice_type_nos, h->picture_structure, h->avctx) < 0) slice_type_nos, h->picture_structure, h->avctx) < 0)
return AVERROR_INVALIDDATA; return AVERROR_INVALIDDATA;
if (sl->slice_type_nos != AV_PICTURE_TYPE_I) { if (slice_type_nos != AV_PICTURE_TYPE_I) {
int list; int list;
for (list = 0; list < list_count; list++) { for (list = 0; list < list_count; list++) {
if (get_bits1(&sl->gb)) { if (get_bits1(gb)) {
int index; int index;
for (index = 0; ; index++) { for (index = 0; ; index++) {
unsigned int reordering_of_pic_nums_idc = get_ue_golomb_31(&sl->gb); unsigned int reordering_of_pic_nums_idc = get_ue_golomb_31(gb);
if (reordering_of_pic_nums_idc < 3) if (reordering_of_pic_nums_idc < 3)
get_ue_golomb_long(&sl->gb); get_ue_golomb_long(gb);
else if (reordering_of_pic_nums_idc > 3) { else if (reordering_of_pic_nums_idc > 3) {
av_log(h->avctx, AV_LOG_ERROR, av_log(h->avctx, AV_LOG_ERROR,
"illegal reordering_of_pic_nums_idc %d\n", "illegal reordering_of_pic_nums_idc %d\n",
...@@ -186,15 +186,15 @@ static int scan_mmco_reset(AVCodecParserContext *s) ...@@ -186,15 +186,15 @@ static int scan_mmco_reset(AVCodecParserContext *s)
} }
} }
if ((h->pps.weighted_pred && sl->slice_type_nos == AV_PICTURE_TYPE_P) || if ((h->pps.weighted_pred && slice_type_nos == AV_PICTURE_TYPE_P) ||
(h->pps.weighted_bipred_idc == 1 && sl->slice_type_nos == AV_PICTURE_TYPE_B)) (h->pps.weighted_bipred_idc == 1 && slice_type_nos == AV_PICTURE_TYPE_B))
ff_h264_pred_weight_table(&sl->gb, &h->sps, ref_count, sl->slice_type_nos, ff_h264_pred_weight_table(gb, &h->sps, ref_count, slice_type_nos,
&sl->pwt); &pwt);
if (get_bits1(&sl->gb)) { // adaptive_ref_pic_marking_mode_flag if (get_bits1(gb)) { // adaptive_ref_pic_marking_mode_flag
int i; int i;
for (i = 0; i < MAX_MMCO_COUNT; i++) { for (i = 0; i < MAX_MMCO_COUNT; i++) {
MMCOOpcode opcode = get_ue_golomb_31(&sl->gb); MMCOOpcode opcode = get_ue_golomb_31(gb);
if (opcode > (unsigned) MMCO_LONG) { if (opcode > (unsigned) MMCO_LONG) {
av_log(h->avctx, AV_LOG_ERROR, av_log(h->avctx, AV_LOG_ERROR,
"illegal memory management control operation %d\n", "illegal memory management control operation %d\n",
...@@ -207,10 +207,10 @@ static int scan_mmco_reset(AVCodecParserContext *s) ...@@ -207,10 +207,10 @@ static int scan_mmco_reset(AVCodecParserContext *s)
return 1; return 1;
if (opcode == MMCO_SHORT2UNUSED || opcode == MMCO_SHORT2LONG) if (opcode == MMCO_SHORT2UNUSED || opcode == MMCO_SHORT2LONG)
get_ue_golomb_long(&sl->gb); // difference_of_pic_nums_minus1 get_ue_golomb_long(gb); // difference_of_pic_nums_minus1
if (opcode == MMCO_SHORT2LONG || opcode == MMCO_LONG2UNUSED || if (opcode == MMCO_SHORT2LONG || opcode == MMCO_LONG2UNUSED ||
opcode == MMCO_LONG || opcode == MMCO_SET_MAX_LONG) opcode == MMCO_LONG || opcode == MMCO_SET_MAX_LONG)
get_ue_golomb_31(&sl->gb); get_ue_golomb_31(gb);
} }
} }
...@@ -231,7 +231,6 @@ static inline int parse_nal_units(AVCodecParserContext *s, ...@@ -231,7 +231,6 @@ static inline int parse_nal_units(AVCodecParserContext *s,
{ {
H264ParseContext *p = s->priv_data; H264ParseContext *p = s->priv_data;
H264Context *h = &p->h; H264Context *h = &p->h;
H264SliceContext *sl = &h->slice_ctx[0];
H2645NAL nal = { NULL }; H2645NAL nal = { NULL };
int buf_index, next_avc; int buf_index, next_avc;
unsigned int pps_id; unsigned int pps_id;
...@@ -325,15 +324,14 @@ static inline int parse_nal_units(AVCodecParserContext *s, ...@@ -325,15 +324,14 @@ static inline int parse_nal_units(AVCodecParserContext *s,
h->prev_poc_lsb = 0; h->prev_poc_lsb = 0;
/* fall through */ /* fall through */
case NAL_SLICE: case NAL_SLICE:
sl->gb = nal.gb; get_ue_golomb_long(&nal.gb); // skip first_mb_in_slice
get_ue_golomb_long(&sl->gb); // skip first_mb_in_slice slice_type = get_ue_golomb_31(&nal.gb);
slice_type = get_ue_golomb_31(&sl->gb);
s->pict_type = ff_h264_golomb_to_pict_type[slice_type % 5]; s->pict_type = ff_h264_golomb_to_pict_type[slice_type % 5];
if (h->sei_recovery_frame_cnt >= 0) { if (h->sei_recovery_frame_cnt >= 0) {
/* key frame, since recovery_frame_cnt is set */ /* key frame, since recovery_frame_cnt is set */
s->key_frame = 1; s->key_frame = 1;
} }
pps_id = get_ue_golomb(&sl->gb); pps_id = get_ue_golomb(&nal.gb);
if (pps_id >= MAX_PPS_COUNT) { if (pps_id >= MAX_PPS_COUNT) {
av_log(h->avctx, AV_LOG_ERROR, av_log(h->avctx, AV_LOG_ERROR,
"pps_id %u out of range\n", pps_id); "pps_id %u out of range\n", pps_id);
...@@ -351,7 +349,7 @@ static inline int parse_nal_units(AVCodecParserContext *s, ...@@ -351,7 +349,7 @@ static inline int parse_nal_units(AVCodecParserContext *s,
goto fail; goto fail;
} }
h->sps = *h->sps_buffers[h->pps.sps_id]; h->sps = *h->sps_buffers[h->pps.sps_id];
h->frame_num = get_bits(&sl->gb, h->sps.log2_max_frame_num); h->frame_num = get_bits(&nal.gb, h->sps.log2_max_frame_num);
if(h->sps.ref_frame_count <= 1 && h->pps.ref_count[0] <= 1 && s->pict_type == AV_PICTURE_TYPE_I) if(h->sps.ref_frame_count <= 1 && h->pps.ref_count[0] <= 1 && s->pict_type == AV_PICTURE_TYPE_I)
s->key_frame = 1; s->key_frame = 1;
...@@ -391,30 +389,30 @@ static inline int parse_nal_units(AVCodecParserContext *s, ...@@ -391,30 +389,30 @@ static inline int parse_nal_units(AVCodecParserContext *s,
if (h->sps.frame_mbs_only_flag) { if (h->sps.frame_mbs_only_flag) {
h->picture_structure = PICT_FRAME; h->picture_structure = PICT_FRAME;
} else { } else {
if (get_bits1(&sl->gb)) { // field_pic_flag if (get_bits1(&nal.gb)) { // field_pic_flag
h->picture_structure = PICT_TOP_FIELD + get_bits1(&sl->gb); // bottom_field_flag h->picture_structure = PICT_TOP_FIELD + get_bits1(&nal.gb); // bottom_field_flag
} else { } else {
h->picture_structure = PICT_FRAME; h->picture_structure = PICT_FRAME;
} }
} }
if (h->nal_unit_type == NAL_IDR_SLICE) if (h->nal_unit_type == NAL_IDR_SLICE)
get_ue_golomb_long(&sl->gb); /* idr_pic_id */ get_ue_golomb_long(&nal.gb); /* idr_pic_id */
if (h->sps.poc_type == 0) { if (h->sps.poc_type == 0) {
h->poc_lsb = get_bits(&sl->gb, h->sps.log2_max_poc_lsb); h->poc_lsb = get_bits(&nal.gb, h->sps.log2_max_poc_lsb);
if (h->pps.pic_order_present == 1 && if (h->pps.pic_order_present == 1 &&
h->picture_structure == PICT_FRAME) h->picture_structure == PICT_FRAME)
h->delta_poc_bottom = get_se_golomb(&sl->gb); h->delta_poc_bottom = get_se_golomb(&nal.gb);
} }
if (h->sps.poc_type == 1 && if (h->sps.poc_type == 1 &&
!h->sps.delta_pic_order_always_zero_flag) { !h->sps.delta_pic_order_always_zero_flag) {
h->delta_poc[0] = get_se_golomb(&sl->gb); h->delta_poc[0] = get_se_golomb(&nal.gb);
if (h->pps.pic_order_present == 1 && if (h->pps.pic_order_present == 1 &&
h->picture_structure == PICT_FRAME) h->picture_structure == PICT_FRAME)
h->delta_poc[1] = get_se_golomb(&sl->gb); h->delta_poc[1] = get_se_golomb(&nal.gb);
} }
/* Decode POC of this picture. /* Decode POC of this picture.
...@@ -427,7 +425,7 @@ static inline int parse_nal_units(AVCodecParserContext *s, ...@@ -427,7 +425,7 @@ static inline int parse_nal_units(AVCodecParserContext *s,
* Maybe, we should parse all undisposable non-IDR slice of this * Maybe, we should parse all undisposable non-IDR slice of this
* picture until encountering MMCO_RESET in a slice of it. */ * picture until encountering MMCO_RESET in a slice of it. */
if (h->nal_ref_idc && h->nal_unit_type != NAL_IDR_SLICE) { if (h->nal_ref_idc && h->nal_unit_type != NAL_IDR_SLICE) {
got_reset = scan_mmco_reset(s); got_reset = scan_mmco_reset(s, &nal.gb);
if (got_reset < 0) if (got_reset < 0)
goto fail; goto fail;
} }
......
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