| Index: source/libvpx/vp9/decoder/vp9_dthread.c
|
| diff --git a/source/libvpx/vp9/decoder/vp9_dthread.c b/source/libvpx/vp9/decoder/vp9_dthread.c
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..7aa8888488e259ba391b81ee95523bd5b405e7bf
|
| --- /dev/null
|
| +++ b/source/libvpx/vp9/decoder/vp9_dthread.c
|
| @@ -0,0 +1,192 @@
|
| +/*
|
| + * Copyright (c) 2014 The WebM project authors. All Rights Reserved.
|
| + *
|
| + * Use of this source code is governed by a BSD-style license
|
| + * that can be found in the LICENSE file in the root of the source
|
| + * tree. An additional intellectual property rights grant can be found
|
| + * in the file PATENTS. All contributing project authors may
|
| + * be found in the AUTHORS file in the root of the source tree.
|
| + */
|
| +
|
| +#include "./vpx_config.h"
|
| +#include "vpx_mem/vpx_mem.h"
|
| +#include "vp9/common/vp9_reconinter.h"
|
| +#include "vp9/decoder/vp9_dthread.h"
|
| +#include "vp9/decoder/vp9_decoder.h"
|
| +
|
| +// #define DEBUG_THREAD
|
| +
|
| +// TODO(hkuang): Clean up all the #ifdef in this file.
|
| +void vp9_frameworker_lock_stats(VP9Worker *const worker) {
|
| +#if CONFIG_MULTITHREAD
|
| + FrameWorkerData *const worker_data = worker->data1;
|
| + pthread_mutex_lock(&worker_data->stats_mutex);
|
| +#else
|
| + (void)worker;
|
| +#endif
|
| +}
|
| +
|
| +void vp9_frameworker_unlock_stats(VP9Worker *const worker) {
|
| +#if CONFIG_MULTITHREAD
|
| + FrameWorkerData *const worker_data = worker->data1;
|
| + pthread_mutex_unlock(&worker_data->stats_mutex);
|
| +#else
|
| + (void)worker;
|
| +#endif
|
| +}
|
| +
|
| +void vp9_frameworker_signal_stats(VP9Worker *const worker) {
|
| +#if CONFIG_MULTITHREAD
|
| + FrameWorkerData *const worker_data = worker->data1;
|
| + // TODO(hkuang): Investigate using broadcast or signal.
|
| + pthread_cond_signal(&worker_data->stats_cond);
|
| +#else
|
| + (void)worker;
|
| +#endif
|
| +}
|
| +
|
| +// TODO(hkuang): Remove worker parameter as it is only used in debug code.
|
| +void vp9_frameworker_wait(VP9Worker *const worker, RefCntBuffer *const ref_buf,
|
| + int row) {
|
| +#if CONFIG_MULTITHREAD
|
| + if (!ref_buf)
|
| + return;
|
| +
|
| + // Enabling the following line of code will get harmless tsan error but
|
| + // will get best performance.
|
| + // if (ref_buf->row >= row && ref_buf->buf.corrupted != 1) return;
|
| +
|
| + {
|
| + // Find the worker thread that owns the reference frame. If the reference
|
| + // frame has been fully decoded, it may not have owner.
|
| + VP9Worker *const ref_worker = ref_buf->frame_worker_owner;
|
| + FrameWorkerData *const ref_worker_data =
|
| + (FrameWorkerData *)ref_worker->data1;
|
| + const VP9Decoder *const pbi = ref_worker_data->pbi;
|
| +
|
| +#ifdef DEBUG_THREAD
|
| + {
|
| + FrameWorkerData *const worker_data = (FrameWorkerData *)worker->data1;
|
| + printf("%d %p worker is waiting for %d %p worker (%d) ref %d \r\n",
|
| + worker_data->worker_id, worker, ref_worker_data->worker_id,
|
| + ref_buf->frame_worker_owner, row, ref_buf->row);
|
| + }
|
| +#endif
|
| +
|
| + vp9_frameworker_lock_stats(ref_worker);
|
| + while (ref_buf->row < row && pbi->cur_buf == ref_buf &&
|
| + ref_buf->buf.corrupted != 1) {
|
| + pthread_cond_wait(&ref_worker_data->stats_cond,
|
| + &ref_worker_data->stats_mutex);
|
| + }
|
| +
|
| + if (ref_buf->buf.corrupted == 1) {
|
| + FrameWorkerData *const worker_data = (FrameWorkerData *)worker->data1;
|
| + vp9_frameworker_unlock_stats(ref_worker);
|
| + vpx_internal_error(&worker_data->pbi->common.error,
|
| + VPX_CODEC_CORRUPT_FRAME,
|
| + "Worker %p failed to decode frame", worker);
|
| + }
|
| + vp9_frameworker_unlock_stats(ref_worker);
|
| + }
|
| +#else
|
| + (void)worker;
|
| + (void)ref_buf;
|
| + (void)row;
|
| + (void)ref_buf;
|
| +#endif // CONFIG_MULTITHREAD
|
| +}
|
| +
|
| +void vp9_frameworker_broadcast(RefCntBuffer *const buf, int row) {
|
| +#if CONFIG_MULTITHREAD
|
| + VP9Worker *worker = buf->frame_worker_owner;
|
| +
|
| +#ifdef DEBUG_THREAD
|
| + {
|
| + FrameWorkerData *const worker_data = (FrameWorkerData *)worker->data1;
|
| + printf("%d %p worker decode to (%d) \r\n", worker_data->worker_id,
|
| + buf->frame_worker_owner, row);
|
| + }
|
| +#endif
|
| +
|
| + vp9_frameworker_lock_stats(worker);
|
| + buf->row = row;
|
| + vp9_frameworker_signal_stats(worker);
|
| + vp9_frameworker_unlock_stats(worker);
|
| +#else
|
| + (void)buf;
|
| + (void)row;
|
| +#endif // CONFIG_MULTITHREAD
|
| +}
|
| +
|
| +void vp9_frameworker_copy_context(VP9Worker *const dst_worker,
|
| + VP9Worker *const src_worker) {
|
| +#if CONFIG_MULTITHREAD
|
| + FrameWorkerData *const src_worker_data = (FrameWorkerData *)src_worker->data1;
|
| + FrameWorkerData *const dst_worker_data = (FrameWorkerData *)dst_worker->data1;
|
| + VP9_COMMON *const src_cm = &src_worker_data->pbi->common;
|
| + VP9_COMMON *const dst_cm = &dst_worker_data->pbi->common;
|
| + int i;
|
| +
|
| + // Wait until source frame's context is ready.
|
| + vp9_frameworker_lock_stats(src_worker);
|
| + while (!src_worker_data->frame_context_ready) {
|
| + pthread_cond_wait(&src_worker_data->stats_cond,
|
| + &src_worker_data->stats_mutex);
|
| + }
|
| +
|
| + // src worker may have already finished decoding a frame and swapped the mi.
|
| + // TODO(hkuang): Remove following code after implenment no ModeInfo decoding.
|
| + if (src_worker_data->frame_decoded) {
|
| + dst_cm->prev_mip = src_cm->prev_mip;
|
| + dst_cm->prev_mi = src_cm->prev_mi;
|
| + } else {
|
| + dst_cm->prev_mip = src_cm->mip;
|
| + dst_cm->prev_mi = src_cm->mi;
|
| + }
|
| +
|
| + dst_cm->last_frame_seg_map = src_cm->seg.enabled ?
|
| + src_cm->current_frame_seg_map : src_cm->last_frame_seg_map;
|
| + dst_worker_data->pbi->need_resync = src_worker_data->pbi->need_resync;
|
| + vp9_frameworker_unlock_stats(src_worker);
|
| +
|
| + dst_worker_data->pbi->prev_buf =
|
| + src_worker_data->pbi->common.show_existing_frame ?
|
| + NULL : src_worker_data->pbi->cur_buf;
|
| +
|
| + dst_cm->prev_frame = src_cm->show_existing_frame ?
|
| + src_cm->prev_frame : src_cm->cur_frame;
|
| + dst_cm->last_width = !src_cm->show_existing_frame ?
|
| + src_cm->width : src_cm->last_width;
|
| + dst_cm->last_height = !src_cm->show_existing_frame ?
|
| + src_cm->height : src_cm->last_height;
|
| + dst_cm->display_width = src_cm->display_width;
|
| + dst_cm->display_height = src_cm->display_height;
|
| + dst_cm->subsampling_x = src_cm->subsampling_x;
|
| + dst_cm->subsampling_y = src_cm->subsampling_y;
|
| + dst_cm->last_show_frame = !src_cm->show_existing_frame ?
|
| + src_cm->show_frame : src_cm->last_show_frame;
|
| + dst_cm->last_frame_type = src_cm->last_frame_type;
|
| + dst_cm->frame_type = src_cm->frame_type;
|
| + dst_cm->y_dc_delta_q = src_cm->y_dc_delta_q;
|
| + dst_cm->uv_dc_delta_q = src_cm->uv_dc_delta_q;
|
| + dst_cm->uv_ac_delta_q = src_cm->uv_ac_delta_q;
|
| + dst_cm->base_qindex = src_cm->base_qindex;
|
| +
|
| + for (i = 0; i < REF_FRAMES; ++i)
|
| + dst_cm->ref_frame_map[i] = src_cm->next_ref_frame_map[i];
|
| +
|
| + memcpy(dst_cm->lf_info.lfthr, src_cm->lf_info.lfthr,
|
| + (MAX_LOOP_FILTER + 1) * sizeof(loop_filter_thresh));
|
| + dst_cm->lf.last_sharpness_level = src_cm->lf.sharpness_level;
|
| + dst_cm->lf.filter_level = src_cm->lf.filter_level;
|
| + memcpy(dst_cm->lf.ref_deltas, src_cm->lf.ref_deltas, MAX_REF_LF_DELTAS);
|
| + memcpy(dst_cm->lf.mode_deltas, src_cm->lf.mode_deltas, MAX_MODE_LF_DELTAS);
|
| + dst_cm->seg = src_cm->seg;
|
| + memcpy(dst_cm->frame_contexts, src_cm->frame_contexts,
|
| + FRAME_CONTEXTS * sizeof(dst_cm->frame_contexts[0]));
|
| +#else
|
| + (void) dst_worker;
|
| + (void) src_worker;
|
| +#endif // CONFIG_MULTITHREAD
|
| +}
|
|
|