1
0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-11-21 10:55:51 +02:00

lavc/hevcdec: drop redundant HEVCContext.threads_{type,number}

They are useless duplicates of corresponding AVCodecContext fields.
This commit is contained in:
Anton Khirnov 2024-06-01 20:21:06 +02:00
parent b0c29a45dc
commit e4e9e1da15
5 changed files with 19 additions and 32 deletions

View File

@ -892,15 +892,15 @@ void ff_hevc_hls_filter(HEVCLocalContext *lc, const HEVCPPS *pps,
sao_filter_CTB(lc, s, pps, sps, x - ctb_size, y); sao_filter_CTB(lc, s, pps, sps, x - ctb_size, y);
if (y && x_end) { if (y && x_end) {
sao_filter_CTB(lc, s, pps, sps, x, y - ctb_size); sao_filter_CTB(lc, s, pps, sps, x, y - ctb_size);
if (s->threads_type & FF_THREAD_FRAME ) if (s->avctx->active_thread_type & FF_THREAD_FRAME )
ff_progress_frame_report(&s->cur_frame->tf, y); ff_progress_frame_report(&s->cur_frame->tf, y);
} }
if (x_end && y_end) { if (x_end && y_end) {
sao_filter_CTB(lc, s, pps, sps, x , y); sao_filter_CTB(lc, s, pps, sps, x , y);
if (s->threads_type & FF_THREAD_FRAME ) if (s->avctx->active_thread_type & FF_THREAD_FRAME )
ff_progress_frame_report(&s->cur_frame->tf, y + ctb_size); ff_progress_frame_report(&s->cur_frame->tf, y + ctb_size);
} }
} else if (s->threads_type & FF_THREAD_FRAME && x_end) } else if (s->avctx->active_thread_type & FF_THREAD_FRAME && x_end)
ff_progress_frame_report(&s->cur_frame->tf, y + ctb_size - 4); ff_progress_frame_report(&s->cur_frame->tf, y + ctb_size - 4);
} }

View File

@ -968,8 +968,6 @@ static int hls_slice_header(HEVCContext *s, GetBitContext *gb)
unsigned val = get_bits_long(gb, offset_len); unsigned val = get_bits_long(gb, offset_len);
sh->entry_point_offset[i] = val + 1; // +1; // +1 to get the size sh->entry_point_offset[i] = val + 1; // +1; // +1 to get the size
} }
if (s->threads_number > 1 && (pps->num_tile_rows > 1 || pps->num_tile_columns > 1))
s->threads_number = 1;
} }
} }
@ -1870,7 +1868,7 @@ static void chroma_mc_bi(HEVCLocalContext *lc,
static void hevc_await_progress(const HEVCContext *s, const HEVCFrame *ref, static void hevc_await_progress(const HEVCContext *s, const HEVCFrame *ref,
const Mv *mv, int y0, int height) const Mv *mv, int y0, int height)
{ {
if (s->threads_type == FF_THREAD_FRAME ) { if (s->avctx->active_thread_type == FF_THREAD_FRAME ) {
int y = FFMAX(0, (mv->y >> 2) + y0 + height + 9); int y = FFMAX(0, (mv->y >> 2) + y0 + height + 9);
ff_progress_frame_await(&ref->tf, y); ff_progress_frame_await(&ref->tf, y);
@ -2631,7 +2629,7 @@ static int hls_decode_entry(HEVCContext *s, GetBitContext *gb)
return ctb_addr_ts; return ctb_addr_ts;
} }
static int hls_decode_entry_wpp(AVCodecContext *avctxt, void *hevc_lclist, static int hls_decode_entry_wpp(AVCodecContext *avctx, void *hevc_lclist,
int job, int self_id) int job, int self_id)
{ {
HEVCLocalContext *lc = &((HEVCLocalContext*)hevc_lclist)[self_id]; HEVCLocalContext *lc = &((HEVCLocalContext*)hevc_lclist)[self_id];
@ -2643,7 +2641,7 @@ static int hls_decode_entry_wpp(AVCodecContext *avctxt, void *hevc_lclist,
int ctb_row = job; int ctb_row = job;
int ctb_addr_rs = s->sh.slice_ctb_addr_rs + ctb_row * ((sps->width + ctb_size - 1) >> sps->log2_ctb_size); int ctb_addr_rs = s->sh.slice_ctb_addr_rs + ctb_row * ((sps->width + ctb_size - 1) >> sps->log2_ctb_size);
int ctb_addr_ts = pps->ctb_addr_rs_to_ts[ctb_addr_rs]; int ctb_addr_ts = pps->ctb_addr_rs_to_ts[ctb_addr_rs];
int thread = ctb_row % s->threads_number; int thread = ctb_row % avctx->thread_count;
const uint8_t *data = s->data + s->sh.offset[ctb_row]; const uint8_t *data = s->data + s->sh.offset[ctb_row];
const size_t data_size = s->sh.size[ctb_row]; const size_t data_size = s->sh.size[ctb_row];
@ -2736,8 +2734,8 @@ static int hls_slice_data_wpp(HEVCContext *s, const H2645NAL *nal)
return AVERROR_INVALIDDATA; return AVERROR_INVALIDDATA;
} }
if (s->threads_number > s->nb_local_ctx) { if (s->avctx->thread_count > s->nb_local_ctx) {
HEVCLocalContext *tmp = av_malloc_array(s->threads_number, sizeof(*s->local_ctx)); HEVCLocalContext *tmp = av_malloc_array(s->avctx->thread_count, sizeof(*s->local_ctx));
if (!tmp) if (!tmp)
return AVERROR(ENOMEM); return AVERROR(ENOMEM);
@ -2746,7 +2744,7 @@ static int hls_slice_data_wpp(HEVCContext *s, const H2645NAL *nal)
av_free(s->local_ctx); av_free(s->local_ctx);
s->local_ctx = tmp; s->local_ctx = tmp;
for (unsigned i = s->nb_local_ctx; i < s->threads_number; i++) { for (unsigned i = s->nb_local_ctx; i < s->avctx->thread_count; i++) {
tmp = &s->local_ctx[i]; tmp = &s->local_ctx[i];
memset(tmp, 0, sizeof(*tmp)); memset(tmp, 0, sizeof(*tmp));
@ -2756,7 +2754,7 @@ static int hls_slice_data_wpp(HEVCContext *s, const H2645NAL *nal)
tmp->common_cabac_state = &s->cabac; tmp->common_cabac_state = &s->cabac;
} }
s->nb_local_ctx = s->threads_number; s->nb_local_ctx = s->avctx->thread_count;
} }
offset = s->sh.data_offset; offset = s->sh.data_offset;
@ -2795,7 +2793,7 @@ static int hls_slice_data_wpp(HEVCContext *s, const H2645NAL *nal)
s->data = data; s->data = data;
for (i = 1; i < s->threads_number; i++) { for (i = 1; i < s->nb_local_ctx; i++) {
s->local_ctx[i].first_qp_group = 1; s->local_ctx[i].first_qp_group = 1;
s->local_ctx[i].qp_y = s->local_ctx[0].qp_y; s->local_ctx[i].qp_y = s->local_ctx[0].qp_y;
} }
@ -3157,7 +3155,9 @@ static int decode_nal_unit(HEVCContext *s, const H2645NAL *nal)
goto fail; goto fail;
} }
if (s->threads_number > 1 && s->sh.num_entry_point_offsets > 0) if (s->avctx->active_thread_type == FF_THREAD_SLICE &&
s->sh.num_entry_point_offsets > 0 &&
s->pps->num_tile_rows == 1 && s->pps->num_tile_columns == 1)
ctb_addr_ts = hls_slice_data_wpp(s, nal); ctb_addr_ts = hls_slice_data_wpp(s, nal);
else else
ctb_addr_ts = hls_decode_entry(s, &gb); ctb_addr_ts = hls_decode_entry(s, &gb);
@ -3282,7 +3282,7 @@ static int decode_nal_units(HEVCContext *s, const uint8_t *buf, int length)
} }
fail: fail:
if (s->cur_frame && s->threads_type == FF_THREAD_FRAME) if (s->cur_frame && s->avctx->active_thread_type == FF_THREAD_FRAME)
ff_progress_frame_report(&s->cur_frame->tf, INT_MAX); ff_progress_frame_report(&s->cur_frame->tf, INT_MAX);
return ret; return ret;
@ -3610,9 +3610,6 @@ static int hevc_update_thread_context(AVCodecContext *dst,
s->is_nalff = s0->is_nalff; s->is_nalff = s0->is_nalff;
s->nal_length_size = s0->nal_length_size; s->nal_length_size = s0->nal_length_size;
s->threads_number = s0->threads_number;
s->threads_type = s0->threads_type;
s->film_grain_warning_shown = s0->film_grain_warning_shown; s->film_grain_warning_shown = s0->film_grain_warning_shown;
if (s0->eos) { if (s0->eos) {
@ -3660,17 +3657,10 @@ static av_cold int hevc_decode_init(AVCodecContext *avctx)
int ret; int ret;
if (avctx->active_thread_type & FF_THREAD_SLICE) { if (avctx->active_thread_type & FF_THREAD_SLICE) {
s->threads_number = avctx->thread_count;
ret = ff_slice_thread_init_progress(avctx); ret = ff_slice_thread_init_progress(avctx);
if (ret < 0) if (ret < 0)
return ret; return ret;
} else }
s->threads_number = 1;
if((avctx->active_thread_type & FF_THREAD_FRAME) && avctx->thread_count > 1)
s->threads_type = FF_THREAD_FRAME;
else
s->threads_type = FF_THREAD_SLICE;
ret = hevc_init_context(avctx); ret = hevc_init_context(avctx);
if (ret < 0) if (ret < 0)

View File

@ -453,9 +453,6 @@ typedef struct HEVCContext {
HEVCLocalContext *local_ctx; HEVCLocalContext *local_ctx;
unsigned nb_local_ctx; unsigned nb_local_ctx;
uint8_t threads_type;
uint8_t threads_number;
/** 1 if the independent slice segment header was successfully parsed */ /** 1 if the independent slice segment header was successfully parsed */
uint8_t slice_initialized; uint8_t slice_initialized;

View File

@ -248,7 +248,7 @@ static int temporal_luma_motion_vector(const HEVCContext *s, const HEVCSPS *sps,
x < sps->width) { x < sps->width) {
x &= ~15; x &= ~15;
y &= ~15; y &= ~15;
if (s->threads_type == FF_THREAD_FRAME) if (s->avctx->active_thread_type == FF_THREAD_FRAME)
ff_progress_frame_await(&ref->tf, y); ff_progress_frame_await(&ref->tf, y);
x_pu = x >> sps->log2_min_pu_size; x_pu = x >> sps->log2_min_pu_size;
y_pu = y >> sps->log2_min_pu_size; y_pu = y >> sps->log2_min_pu_size;
@ -262,7 +262,7 @@ static int temporal_luma_motion_vector(const HEVCContext *s, const HEVCSPS *sps,
y = y0 + (nPbH >> 1); y = y0 + (nPbH >> 1);
x &= ~15; x &= ~15;
y &= ~15; y &= ~15;
if (s->threads_type == FF_THREAD_FRAME) if (s->avctx->active_thread_type == FF_THREAD_FRAME)
ff_progress_frame_await(&ref->tf, y); ff_progress_frame_await(&ref->tf, y);
x_pu = x >> sps->log2_min_pu_size; x_pu = x >> sps->log2_min_pu_size;
y_pu = y >> sps->log2_min_pu_size; y_pu = y >> sps->log2_min_pu_size;

View File

@ -435,7 +435,7 @@ static HEVCFrame *generate_missing_ref(HEVCContext *s, int poc)
frame->sequence = HEVC_SEQUENCE_COUNTER_INVALID; frame->sequence = HEVC_SEQUENCE_COUNTER_INVALID;
frame->flags = 0; frame->flags = 0;
if (s->threads_type == FF_THREAD_FRAME) if (s->avctx->active_thread_type == FF_THREAD_FRAME)
ff_progress_frame_report(&frame->tf, INT_MAX); ff_progress_frame_report(&frame->tf, INT_MAX);
return frame; return frame;