vp9_decoder.h 2.99 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
/*
2
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
John Koleszar's avatar
John Koleszar committed
3
 *
4
 *  Use of this source code is governed by a BSD-style license
5
6
 *  that can be found in the LICENSE file in the root of the source
 *  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.
John Koleszar's avatar
John Koleszar committed
9
10
 */

11
12
#ifndef VP9_DECODER_VP9_DECODER_H_
#define VP9_DECODER_VP9_DECODER_H_
John Koleszar's avatar
John Koleszar committed
13

14
15
16
#include "./vpx_config.h"

#include "vpx/vpx_codec.h"
John Koleszar's avatar
John Koleszar committed
17
#include "vpx_scale/yv12config.h"
18
19

#include "vp9/common/vp9_onyxc_int.h"
20
#include "vp9/common/vp9_ppflags.h"
hkuang's avatar
hkuang committed
21
#include "vp9/common/vp9_thread.h"
22
23

#include "vp9/decoder/vp9_dthread.h"
John Koleszar's avatar
John Koleszar committed
24

25
26
27
28
#ifdef __cplusplus
extern "C" {
#endif

29
30
31
32
33
34
35
// TODO(hkuang): combine this with TileWorkerData.
typedef struct TileData {
  VP9_COMMON *cm;
  vp9_reader bit_reader;
  DECLARE_ALIGNED(16, MACROBLOCKD, xd);
} TileData;

36
typedef struct VP9Decoder {
37
38
39
40
41
42
43
44
  DECLARE_ALIGNED(16, MACROBLOCKD, mb);

  DECLARE_ALIGNED(16, VP9_COMMON, common);

  int ready_for_new_data;

  int refresh_frame_flags;

45
46
  int frame_parallel_decode;  // frame-based threading.

47
48
  VP9Worker lf_worker;
  VP9Worker *tile_workers;
49
50
  TileWorkerData *tile_worker_data;
  TileInfo *tile_worker_info;
51
52
  int num_tile_workers;

53
54
55
  TileData *tile_data;
  int total_tiles;

56
  VP9LfSync lf_row_sync;
57
58
59

  vpx_decrypt_cb decrypt_cb;
  void *decrypt_state;
Dmitry Kovalev's avatar
Dmitry Kovalev committed
60
61
62

  int max_threads;
  int inv_tile_order;
63
  int need_resync;  // wait for key/intra-only frame
64
} VP9Decoder;
65

66
int vp9_receive_compressed_data(struct VP9Decoder *pbi,
67
                                size_t size, const uint8_t **dest);
68

69
int vp9_get_raw_frame(struct VP9Decoder *pbi, YV12_BUFFER_CONFIG *sd,
70
71
                      vp9_ppflags_t *flags);

72
vpx_codec_err_t vp9_copy_reference_dec(struct VP9Decoder *pbi,
73
74
75
                                       VP9_REFFRAME ref_frame_flag,
                                       YV12_BUFFER_CONFIG *sd);

76
vpx_codec_err_t vp9_set_reference_dec(VP9_COMMON *cm,
77
78
79
                                      VP9_REFFRAME ref_frame_flag,
                                      YV12_BUFFER_CONFIG *sd);

Dmitry Kovalev's avatar
Dmitry Kovalev committed
80
struct VP9Decoder *vp9_decoder_create();
81

82
void vp9_decoder_remove(struct VP9Decoder *pbi);
John Koleszar's avatar
John Koleszar committed
83

84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
static INLINE uint8_t read_marker(vpx_decrypt_cb decrypt_cb,
                                  void *decrypt_state,
                                  const uint8_t *data) {
  if (decrypt_cb) {
    uint8_t marker;
    decrypt_cb(decrypt_state, data, &marker, 1);
    return marker;
  }
  return *data;
}

// This function is exposed for use in tests, as well as the inlined function
// "read_marker".
vpx_codec_err_t vp9_parse_superframe_index(const uint8_t *data,
                                           size_t data_sz,
                                           uint32_t sizes[8], int *count,
                                           vpx_decrypt_cb decrypt_cb,
                                           void *decrypt_state);

John Koleszar's avatar
John Koleszar committed
103
#ifdef __cplusplus
104
}  // extern "C"
John Koleszar's avatar
John Koleszar committed
105
106
#endif

107
#endif  // VP9_DECODER_VP9_DECODER_H_