| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. | 2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| 11 #ifndef VP9_DECODER_VP9_DECODER_H_ | 11 #ifndef VP9_DECODER_VP9_DECODER_H_ |
| 12 #define VP9_DECODER_VP9_DECODER_H_ | 12 #define VP9_DECODER_VP9_DECODER_H_ |
| 13 | 13 |
| 14 #include "./vpx_config.h" | 14 #include "./vpx_config.h" |
| 15 | 15 |
| 16 #include "vpx/vpx_codec.h" | 16 #include "vpx/vpx_codec.h" |
| 17 #include "vpx_scale/yv12config.h" | 17 #include "vpx_scale/yv12config.h" |
| 18 #include "vp9/common/vp9_loopfilter_thread.h" | 18 #include "vp9/common/vp9_loopfilter_thread.h" |
| 19 #include "vp9/common/vp9_onyxc_int.h" | 19 #include "vp9/common/vp9_onyxc_int.h" |
| 20 #include "vp9/common/vp9_ppflags.h" | 20 #include "vp9/common/vp9_ppflags.h" |
| 21 #include "vp9/common/vp9_thread.h" | 21 #include "vp9/common/vp9_thread.h" |
| 22 #include "vp9/decoder/vp9_dthread.h" |
| 22 #include "vp9/decoder/vp9_reader.h" | 23 #include "vp9/decoder/vp9_reader.h" |
| 23 | 24 |
| 24 #ifdef __cplusplus | 25 #ifdef __cplusplus |
| 25 extern "C" { | 26 extern "C" { |
| 26 #endif | 27 #endif |
| 27 | 28 |
| 28 // TODO(hkuang): combine this with TileWorkerData. | 29 // TODO(hkuang): combine this with TileWorkerData. |
| 29 typedef struct TileData { | 30 typedef struct TileData { |
| 30 VP9_COMMON *cm; | 31 VP9_COMMON *cm; |
| 31 vp9_reader bit_reader; | 32 vp9_reader bit_reader; |
| 32 DECLARE_ALIGNED(16, MACROBLOCKD, xd); | 33 DECLARE_ALIGNED(16, MACROBLOCKD, xd); |
| 33 } TileData; | 34 } TileData; |
| 34 | 35 |
| 35 typedef struct TileWorkerData { | 36 typedef struct TileWorkerData { |
| 36 VP9_COMMON *cm; | 37 struct VP9Decoder *pbi; |
| 37 vp9_reader bit_reader; | 38 vp9_reader bit_reader; |
| 38 DECLARE_ALIGNED(16, MACROBLOCKD, xd); | 39 DECLARE_ALIGNED(16, MACROBLOCKD, xd); |
| 39 struct vpx_internal_error_info error_info; | 40 struct vpx_internal_error_info error_info; |
| 40 } TileWorkerData; | 41 } TileWorkerData; |
| 41 | 42 |
| 42 typedef struct VP9Decoder { | 43 typedef struct VP9Decoder { |
| 43 DECLARE_ALIGNED(16, MACROBLOCKD, mb); | 44 DECLARE_ALIGNED(16, MACROBLOCKD, mb); |
| 44 | 45 |
| 45 DECLARE_ALIGNED(16, VP9_COMMON, common); | 46 DECLARE_ALIGNED(16, VP9_COMMON, common); |
| 46 | 47 |
| 47 int ready_for_new_data; | 48 int ready_for_new_data; |
| 48 | 49 |
| 49 int refresh_frame_flags; | 50 int refresh_frame_flags; |
| 50 | 51 |
| 51 int frame_parallel_decode; // frame-based threading. | 52 int frame_parallel_decode; // frame-based threading. |
| 52 | 53 |
| 54 // TODO(hkuang): Combine this with cur_buf in macroblockd as they are |
| 55 // the same. |
| 56 RefCntBuffer *cur_buf; // Current decoding frame buffer. |
| 57 RefCntBuffer *prev_buf; // Previous decoding frame buffer. |
| 58 |
| 59 VP9Worker *frame_worker_owner; // frame_worker that owns this pbi. |
| 53 VP9Worker lf_worker; | 60 VP9Worker lf_worker; |
| 54 VP9Worker *tile_workers; | 61 VP9Worker *tile_workers; |
| 55 TileWorkerData *tile_worker_data; | 62 TileWorkerData *tile_worker_data; |
| 56 TileInfo *tile_worker_info; | 63 TileInfo *tile_worker_info; |
| 57 int num_tile_workers; | 64 int num_tile_workers; |
| 58 | 65 |
| 59 TileData *tile_data; | 66 TileData *tile_data; |
| 60 int total_tiles; | 67 int total_tiles; |
| 61 | 68 |
| 62 VP9LfSync lf_row_sync; | 69 VP9LfSync lf_row_sync; |
| 63 | 70 |
| 64 vpx_decrypt_cb decrypt_cb; | 71 vpx_decrypt_cb decrypt_cb; |
| 65 void *decrypt_state; | 72 void *decrypt_state; |
| 66 | 73 |
| 67 int max_threads; | 74 int max_threads; |
| 68 int inv_tile_order; | 75 int inv_tile_order; |
| 69 int need_resync; // wait for key/intra-only frame | 76 int need_resync; // wait for key/intra-only frame. |
| 77 int hold_ref_buf; // hold the reference buffer. |
| 70 } VP9Decoder; | 78 } VP9Decoder; |
| 71 | 79 |
| 72 int vp9_receive_compressed_data(struct VP9Decoder *pbi, | 80 int vp9_receive_compressed_data(struct VP9Decoder *pbi, |
| 73 size_t size, const uint8_t **dest); | 81 size_t size, const uint8_t **dest); |
| 74 | 82 |
| 75 int vp9_get_raw_frame(struct VP9Decoder *pbi, YV12_BUFFER_CONFIG *sd, | 83 int vp9_get_raw_frame(struct VP9Decoder *pbi, YV12_BUFFER_CONFIG *sd, |
| 76 vp9_ppflags_t *flags); | 84 vp9_ppflags_t *flags); |
| 77 | 85 |
| 78 vpx_codec_err_t vp9_copy_reference_dec(struct VP9Decoder *pbi, | 86 vpx_codec_err_t vp9_copy_reference_dec(struct VP9Decoder *pbi, |
| 79 VP9_REFFRAME ref_frame_flag, | 87 VP9_REFFRAME ref_frame_flag, |
| 80 YV12_BUFFER_CONFIG *sd); | 88 YV12_BUFFER_CONFIG *sd); |
| 81 | 89 |
| 82 vpx_codec_err_t vp9_set_reference_dec(VP9_COMMON *cm, | 90 vpx_codec_err_t vp9_set_reference_dec(VP9_COMMON *cm, |
| 83 VP9_REFFRAME ref_frame_flag, | 91 VP9_REFFRAME ref_frame_flag, |
| 84 YV12_BUFFER_CONFIG *sd); | 92 YV12_BUFFER_CONFIG *sd); |
| 85 | 93 |
| 86 struct VP9Decoder *vp9_decoder_create(); | |
| 87 | |
| 88 void vp9_decoder_remove(struct VP9Decoder *pbi); | |
| 89 | |
| 90 static INLINE uint8_t read_marker(vpx_decrypt_cb decrypt_cb, | 94 static INLINE uint8_t read_marker(vpx_decrypt_cb decrypt_cb, |
| 91 void *decrypt_state, | 95 void *decrypt_state, |
| 92 const uint8_t *data) { | 96 const uint8_t *data) { |
| 93 if (decrypt_cb) { | 97 if (decrypt_cb) { |
| 94 uint8_t marker; | 98 uint8_t marker; |
| 95 decrypt_cb(decrypt_state, data, &marker, 1); | 99 decrypt_cb(decrypt_state, data, &marker, 1); |
| 96 return marker; | 100 return marker; |
| 97 } | 101 } |
| 98 return *data; | 102 return *data; |
| 99 } | 103 } |
| 100 | 104 |
| 101 // This function is exposed for use in tests, as well as the inlined function | 105 // This function is exposed for use in tests, as well as the inlined function |
| 102 // "read_marker". | 106 // "read_marker". |
| 103 vpx_codec_err_t vp9_parse_superframe_index(const uint8_t *data, | 107 vpx_codec_err_t vp9_parse_superframe_index(const uint8_t *data, |
| 104 size_t data_sz, | 108 size_t data_sz, |
| 105 uint32_t sizes[8], int *count, | 109 uint32_t sizes[8], int *count, |
| 106 vpx_decrypt_cb decrypt_cb, | 110 vpx_decrypt_cb decrypt_cb, |
| 107 void *decrypt_state); | 111 void *decrypt_state); |
| 108 | 112 |
| 113 struct VP9Decoder *vp9_decoder_create(BufferPool *const pool); |
| 114 |
| 115 void vp9_decoder_remove(struct VP9Decoder *pbi); |
| 116 |
| 117 static INLINE void decrease_ref_count(int idx, RefCntBuffer *const frame_bufs, |
| 118 BufferPool *const pool) { |
| 119 if (idx >= 0) { |
| 120 --frame_bufs[idx].ref_count; |
| 121 // A worker may only get a free framebuffer index when calling get_free_fb. |
| 122 // But the private buffer is not set up until finish decoding header. |
| 123 // So any error happens during decoding header, the frame_bufs will not |
| 124 // have valid priv buffer. |
| 125 if (frame_bufs[idx].ref_count == 0 && |
| 126 frame_bufs[idx].raw_frame_buffer.priv) { |
| 127 pool->release_fb_cb(pool->cb_priv, &frame_bufs[idx].raw_frame_buffer); |
| 128 } |
| 129 } |
| 130 } |
| 131 |
| 109 #ifdef __cplusplus | 132 #ifdef __cplusplus |
| 110 } // extern "C" | 133 } // extern "C" |
| 111 #endif | 134 #endif |
| 112 | 135 |
| 113 #endif // VP9_DECODER_VP9_DECODER_H_ | 136 #endif // VP9_DECODER_VP9_DECODER_H_ |
| OLD | NEW |