dthread.h 2.46 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
 */

12 13
#ifndef AV1_DECODER_DTHREAD_H_
#define AV1_DECODER_DTHREAD_H_
Jingning Han's avatar
Jingning Han committed
14

Adrian Grange's avatar
Adrian Grange committed
15 16 17
#include "./aom_config.h"
#include "aom_util/aom_thread.h"
#include "aom/internal/aom_codec_internal.h"
Jingning Han's avatar
Jingning Han committed
18

James Zern's avatar
James Zern committed
19 20 21 22
#ifdef __cplusplus
extern "C" {
#endif

23 24
struct AV1Common;
struct AV1Decoder;
Jingning Han's avatar
Jingning Han committed
25 26 27 28

// WorkerData for the FrameWorker thread. It contains all the information of
// the worker and decode structures for decoding a frame.
typedef struct FrameWorkerData {
29
  struct AV1Decoder *pbi;
Jingning Han's avatar
Jingning Han committed
30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
  const uint8_t *data;
  const uint8_t *data_end;
  size_t data_size;
  void *user_priv;
  int result;
  int worker_id;
  int received_frame;

  // scratch_buffer is used in frame parallel mode only.
  // It is used to make a copy of the compressed data.
  uint8_t *scratch_buffer;
  size_t scratch_buffer_size;

#if CONFIG_MULTITHREAD
  pthread_mutex_t stats_mutex;
  pthread_cond_t stats_cond;
#endif

  int frame_context_ready;  // Current frame's context is ready to read.
  int frame_decoded;        // Finished decoding current frame.
} FrameWorkerData;

Adrian Grange's avatar
Adrian Grange committed
52 53 54
void av1_frameworker_lock_stats(AVxWorker *const worker);
void av1_frameworker_unlock_stats(AVxWorker *const worker);
void av1_frameworker_signal_stats(AVxWorker *const worker);
Jingning Han's avatar
Jingning Han committed
55 56 57 58 59

// Wait until ref_buf has been decoded to row in real pixel unit.
// Note: worker may already finish decoding ref_buf and release it in order to
// start decoding next frame. So need to check whether worker is still decoding
// ref_buf.
Adrian Grange's avatar
Adrian Grange committed
60
void av1_frameworker_wait(AVxWorker *const worker, RefCntBuffer *const ref_buf,
clang-format's avatar
clang-format committed
61
                           int row);
Jingning Han's avatar
Jingning Han committed
62 63 64

// FrameWorker broadcasts its decoding progress so other workers that are
// waiting on it can resume decoding.
65
void av1_frameworker_broadcast(RefCntBuffer *const buf, int row);
Jingning Han's avatar
Jingning Han committed
66 67

// Copy necessary decoding context from src worker to dst worker.
Adrian Grange's avatar
Adrian Grange committed
68 69
void av1_frameworker_copy_context(AVxWorker *const dst_worker,
                                   AVxWorker *const src_worker);
Jingning Han's avatar
Jingning Han committed
70

James Zern's avatar
James Zern committed
71
#ifdef __cplusplus
clang-format's avatar
clang-format committed
72
}  // extern "C"
James Zern's avatar
James Zern committed
73 74
#endif

75
#endif  // AV1_DECODER_DTHREAD_H_