dthread.c 6.52 KB
Newer Older
Jingning Han's avatar
Jingning Han committed
1
/*
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
Jingning Han's avatar
Jingning Han committed
3
 *
4 5 6 7 8 9
 * This source code is subject to the terms of the BSD 2 Clause License and
 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
 * was not distributed with this source code in the LICENSE file, you can
 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
 * Media Patent License 1.0 was not distributed with this source code in the
 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
Jingning Han's avatar
Jingning Han committed
10 11
 */

Adrian Grange's avatar
Adrian Grange committed
12 13
#include "./aom_config.h"
#include "aom_mem/aom_mem.h"
Yaowu Xu's avatar
Yaowu Xu committed
14 15 16
#include "av1/common/reconinter.h"
#include "av1/decoder/dthread.h"
#include "av1/decoder/decoder.h"
Jingning Han's avatar
Jingning Han committed
17 18 19 20

// #define DEBUG_THREAD

// TODO(hkuang): Clean up all the #ifdef in this file.
21
void av1_frameworker_lock_stats(VPxWorker *const worker) {
Jingning Han's avatar
Jingning Han committed
22 23 24 25 26 27 28 29
#if CONFIG_MULTITHREAD
  FrameWorkerData *const worker_data = worker->data1;
  pthread_mutex_lock(&worker_data->stats_mutex);
#else
  (void)worker;
#endif
}

30
void av1_frameworker_unlock_stats(VPxWorker *const worker) {
Jingning Han's avatar
Jingning Han committed
31 32 33 34 35 36 37 38
#if CONFIG_MULTITHREAD
  FrameWorkerData *const worker_data = worker->data1;
  pthread_mutex_unlock(&worker_data->stats_mutex);
#else
  (void)worker;
#endif
}

39
void av1_frameworker_signal_stats(VPxWorker *const worker) {
Jingning Han's avatar
Jingning Han committed
40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62
#if CONFIG_MULTITHREAD
  FrameWorkerData *const worker_data = worker->data1;

// TODO(hkuang): Fix the pthread_cond_broadcast in windows wrapper.
#if defined(_WIN32) && !HAVE_PTHREAD_H
  pthread_cond_signal(&worker_data->stats_cond);
#else
  pthread_cond_broadcast(&worker_data->stats_cond);
#endif

#else
  (void)worker;
#endif
}

// This macro prevents thread_sanitizer from reporting known concurrent writes.
#if defined(__has_feature)
#if __has_feature(thread_sanitizer)
#define BUILDING_WITH_TSAN
#endif
#endif

// TODO(hkuang): Remove worker parameter as it is only used in debug code.
63
void av1_frameworker_wait(VPxWorker *const worker, RefCntBuffer *const ref_buf,
clang-format's avatar
clang-format committed
64
                           int row) {
Jingning Han's avatar
Jingning Han committed
65
#if CONFIG_MULTITHREAD
clang-format's avatar
clang-format committed
66
  if (!ref_buf) return;
Jingning Han's avatar
Jingning Han committed
67 68 69 70 71 72 73 74 75 76 77 78 79

#ifndef BUILDING_WITH_TSAN
  // The following line of code will get harmless tsan error but it is the key
  // to get best performance.
  if (ref_buf->row >= row && ref_buf->buf.corrupted != 1) return;
#endif

  {
    // Find the worker thread that owns the reference frame. If the reference
    // frame has been fully decoded, it may not have owner.
    VPxWorker *const ref_worker = ref_buf->frame_worker_owner;
    FrameWorkerData *const ref_worker_data =
        (FrameWorkerData *)ref_worker->data1;
80
    const AV1Decoder *const pbi = ref_worker_data->pbi;
Jingning Han's avatar
Jingning Han committed
81 82 83 84 85 86 87 88 89 90

#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

91
    av1_frameworker_lock_stats(ref_worker);
Jingning Han's avatar
Jingning Han committed
92 93 94 95 96 97 98 99
    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;
100
      av1_frameworker_unlock_stats(ref_worker);
Adrian Grange's avatar
Adrian Grange committed
101
      aom_internal_error(&worker_data->pbi->common.error,
Jingning Han's avatar
Jingning Han committed
102 103 104
                         VPX_CODEC_CORRUPT_FRAME,
                         "Worker %p failed to decode frame", worker);
    }
105
    av1_frameworker_unlock_stats(ref_worker);
Jingning Han's avatar
Jingning Han committed
106 107 108 109 110 111 112 113 114
  }
#else
  (void)worker;
  (void)ref_buf;
  (void)row;
  (void)ref_buf;
#endif  // CONFIG_MULTITHREAD
}

115
void av1_frameworker_broadcast(RefCntBuffer *const buf, int row) {
Jingning Han's avatar
Jingning Han committed
116 117 118 119 120 121 122 123 124 125 126
#if CONFIG_MULTITHREAD
  VPxWorker *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

127
  av1_frameworker_lock_stats(worker);
Jingning Han's avatar
Jingning Han committed
128
  buf->row = row;
129 130
  av1_frameworker_signal_stats(worker);
  av1_frameworker_unlock_stats(worker);
Jingning Han's avatar
Jingning Han committed
131 132 133 134 135 136
#else
  (void)buf;
  (void)row;
#endif  // CONFIG_MULTITHREAD
}

137
void av1_frameworker_copy_context(VPxWorker *const dst_worker,
clang-format's avatar
clang-format committed
138
                                   VPxWorker *const src_worker) {
Jingning Han's avatar
Jingning Han committed
139 140 141
#if CONFIG_MULTITHREAD
  FrameWorkerData *const src_worker_data = (FrameWorkerData *)src_worker->data1;
  FrameWorkerData *const dst_worker_data = (FrameWorkerData *)dst_worker->data1;
142 143
  AV1_COMMON *const src_cm = &src_worker_data->pbi->common;
  AV1_COMMON *const dst_cm = &dst_worker_data->pbi->common;
Jingning Han's avatar
Jingning Han committed
144 145 146
  int i;

  // Wait until source frame's context is ready.
147
  av1_frameworker_lock_stats(src_worker);
Jingning Han's avatar
Jingning Han committed
148 149
  while (!src_worker_data->frame_context_ready) {
    pthread_cond_wait(&src_worker_data->stats_cond,
clang-format's avatar
clang-format committed
150
                      &src_worker_data->stats_mutex);
Jingning Han's avatar
Jingning Han committed
151 152
  }

clang-format's avatar
clang-format committed
153 154 155
  dst_cm->last_frame_seg_map = src_cm->seg.enabled
                                   ? src_cm->current_frame_seg_map
                                   : src_cm->last_frame_seg_map;
Jingning Han's avatar
Jingning Han committed
156
  dst_worker_data->pbi->need_resync = src_worker_data->pbi->need_resync;
157
  av1_frameworker_unlock_stats(src_worker);
Jingning Han's avatar
Jingning Han committed
158 159

  dst_cm->bit_depth = src_cm->bit_depth;
160
#if CONFIG_AOM_HIGHBITDEPTH
Jingning Han's avatar
Jingning Han committed
161 162
  dst_cm->use_highbitdepth = src_cm->use_highbitdepth;
#endif
clang-format's avatar
clang-format committed
163 164 165 166 167 168
  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;
Jingning Han's avatar
Jingning Han committed
169 170 171
  dst_cm->subsampling_x = src_cm->subsampling_x;
  dst_cm->subsampling_y = src_cm->subsampling_y;
  dst_cm->frame_type = src_cm->frame_type;
clang-format's avatar
clang-format committed
172 173 174
  dst_cm->last_show_frame = !src_cm->show_existing_frame
                                ? src_cm->show_frame
                                : src_cm->last_show_frame;
Jingning Han's avatar
Jingning Han committed
175 176 177 178 179 180 181
  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;
182
  memcpy(dst_cm->lf.ref_deltas, src_cm->lf.ref_deltas, MAX_REF_FRAMES);
Jingning Han's avatar
Jingning Han committed
183 184 185 186 187
  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
clang-format's avatar
clang-format committed
188 189
  (void)dst_worker;
  (void)src_worker;
Jingning Han's avatar
Jingning Han committed
190 191
#endif  // CONFIG_MULTITHREAD
}