1/*
2 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
3 *
4 *  Use of this source code is governed by a BSD-style license
5 *  that can be found in the LICENSE file in the root of the source
6 *  tree. An additional intellectual property rights grant can be found
7 *  in the file PATENTS.  All contributing project authors may
8 *  be found in the AUTHORS file in the root of the source tree.
9 */
10
11#ifndef VP9_DECODER_VP9_DECODER_H_
12#define VP9_DECODER_VP9_DECODER_H_
13
14#include "./vpx_config.h"
15
16#include "vpx/vpx_codec.h"
17#include "vpx_scale/yv12config.h"
18
19#include "vp9/common/vp9_onyxc_int.h"
20#include "vp9/common/vp9_ppflags.h"
21#include "vp9/common/vp9_thread.h"
22
23#include "vp9/decoder/vp9_dthread.h"
24
25#ifdef __cplusplus
26extern "C" {
27#endif
28
29// TODO(hkuang): combine this with TileWorkerData.
30typedef struct TileData {
31  VP9_COMMON *cm;
32  vp9_reader bit_reader;
33  DECLARE_ALIGNED(16, MACROBLOCKD, xd);
34} TileData;
35
36typedef struct VP9Decoder {
37  DECLARE_ALIGNED(16, MACROBLOCKD, mb);
38
39  DECLARE_ALIGNED(16, VP9_COMMON, common);
40
41  int ready_for_new_data;
42
43  int refresh_frame_flags;
44
45  int frame_parallel_decode;  // frame-based threading.
46
47  VP9Worker lf_worker;
48  VP9Worker *tile_workers;
49  int num_tile_workers;
50
51  TileData *tile_data;
52  int total_tiles;
53
54  VP9LfSync lf_row_sync;
55
56  vpx_decrypt_cb decrypt_cb;
57  void *decrypt_state;
58
59  int max_threads;
60  int inv_tile_order;
61  int need_resync;  // wait for key/intra-only frame
62} VP9Decoder;
63
64int vp9_receive_compressed_data(struct VP9Decoder *pbi,
65                                size_t size, const uint8_t **dest);
66
67int vp9_get_raw_frame(struct VP9Decoder *pbi, YV12_BUFFER_CONFIG *sd,
68                      vp9_ppflags_t *flags);
69
70vpx_codec_err_t vp9_copy_reference_dec(struct VP9Decoder *pbi,
71                                       VP9_REFFRAME ref_frame_flag,
72                                       YV12_BUFFER_CONFIG *sd);
73
74vpx_codec_err_t vp9_set_reference_dec(VP9_COMMON *cm,
75                                      VP9_REFFRAME ref_frame_flag,
76                                      YV12_BUFFER_CONFIG *sd);
77
78struct VP9Decoder *vp9_decoder_create();
79
80void vp9_decoder_remove(struct VP9Decoder *pbi);
81
82static INLINE uint8_t read_marker(vpx_decrypt_cb decrypt_cb,
83                                  void *decrypt_state,
84                                  const uint8_t *data) {
85  if (decrypt_cb) {
86    uint8_t marker;
87    decrypt_cb(decrypt_state, data, &marker, 1);
88    return marker;
89  }
90  return *data;
91}
92
93// This function is exposed for use in tests, as well as the inlined function
94// "read_marker".
95vpx_codec_err_t vp9_parse_superframe_index(const uint8_t *data,
96                                           size_t data_sz,
97                                           uint32_t sizes[8], int *count,
98                                           vpx_decrypt_cb decrypt_cb,
99                                           void *decrypt_state);
100
101#ifdef __cplusplus
102}  // extern "C"
103#endif
104
105#endif  // VP9_DECODER_VP9_DECODER_H_
106