1 /* 2 * Copyright (c) 2011 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_ENCODER_VP9_LOOKAHEAD_H_ 12 #define VP9_ENCODER_VP9_LOOKAHEAD_H_ 13 14 #include "vpx_scale/yv12config.h" 15 #include "vpx/vpx_integer.h" 16 17 #if CONFIG_SPATIAL_SVC 18 #include "vpx/vp8cx.h" 19 #include "vpx/vpx_encoder.h" 20 #endif 21 22 #ifdef __cplusplus 23 extern "C" { 24 #endif 25 26 #define MAX_LAG_BUFFERS 25 27 28 struct lookahead_entry { 29 YV12_BUFFER_CONFIG img; 30 int64_t ts_start; 31 int64_t ts_end; 32 unsigned int flags; 33 }; 34 35 // The max of past frames we want to keep in the queue. 36 #define MAX_PRE_FRAMES 1 37 38 struct lookahead_ctx { 39 unsigned int max_sz; /* Absolute size of the queue */ 40 unsigned int sz; /* Number of buffers currently in the queue */ 41 unsigned int read_idx; /* Read index */ 42 unsigned int write_idx; /* Write index */ 43 struct lookahead_entry *buf; /* Buffer list */ 44 }; 45 46 /**\brief Initializes the lookahead stage 47 * 48 * The lookahead stage is a queue of frame buffers on which some analysis 49 * may be done when buffers are enqueued. 50 */ 51 struct lookahead_ctx *vp9_lookahead_init(unsigned int width, 52 unsigned int height, 53 unsigned int subsampling_x, 54 unsigned int subsampling_y, 55 #if CONFIG_VP9_HIGHBITDEPTH 56 int use_highbitdepth, 57 #endif 58 unsigned int depth); 59 60 61 /**\brief Destroys the lookahead stage 62 */ 63 void vp9_lookahead_destroy(struct lookahead_ctx *ctx); 64 65 66 /**\brief Enqueue a source buffer 67 * 68 * This function will copy the source image into a new framebuffer with 69 * the expected stride/border. 70 * 71 * If active_map is non-NULL and there is only one frame in the queue, then copy 72 * only active macroblocks. 73 * 74 * \param[in] ctx Pointer to the lookahead context 75 * \param[in] src Pointer to the image to enqueue 76 * \param[in] ts_start Timestamp for the start of this frame 77 * \param[in] ts_end Timestamp for the end of this frame 78 * \param[in] flags Flags set on this frame 79 * \param[in] active_map Map that specifies which macroblock is active 80 */ 81 int vp9_lookahead_push(struct lookahead_ctx *ctx, YV12_BUFFER_CONFIG *src, 82 int64_t ts_start, int64_t ts_end, 83 #if CONFIG_VP9_HIGHBITDEPTH 84 int use_highbitdepth, 85 #endif 86 unsigned int flags); 87 88 89 /**\brief Get the next source buffer to encode 90 * 91 * 92 * \param[in] ctx Pointer to the lookahead context 93 * \param[in] drain Flag indicating the buffer should be drained 94 * (return a buffer regardless of the current queue depth) 95 * 96 * \retval NULL, if drain set and queue is empty 97 * \retval NULL, if drain not set and queue not of the configured depth 98 */ 99 struct lookahead_entry *vp9_lookahead_pop(struct lookahead_ctx *ctx, 100 int drain); 101 102 103 /**\brief Get a future source buffer to encode 104 * 105 * \param[in] ctx Pointer to the lookahead context 106 * \param[in] index Index of the frame to be returned, 0 == next frame 107 * 108 * \retval NULL, if no buffer exists at the specified index 109 */ 110 struct lookahead_entry *vp9_lookahead_peek(struct lookahead_ctx *ctx, 111 int index); 112 113 114 /**\brief Get the number of frames currently in the lookahead queue 115 * 116 * \param[in] ctx Pointer to the lookahead context 117 */ 118 unsigned int vp9_lookahead_depth(struct lookahead_ctx *ctx); 119 120 #ifdef __cplusplus 121 } // extern "C" 122 #endif 123 124 #endif // VP9_ENCODER_VP9_LOOKAHEAD_H_ 125