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 23extern "C" { 24#endif 25 26#define MAX_LAG_BUFFERS 25 27 28struct 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 38struct 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 */ 51struct 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 */ 63void 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 */ 81int vp9_lookahead_push(struct lookahead_ctx *ctx, YV12_BUFFER_CONFIG *src, 82 int64_t ts_start, int64_t ts_end, unsigned int flags); 83 84 85/**\brief Get the next source buffer to encode 86 * 87 * 88 * \param[in] ctx Pointer to the lookahead context 89 * \param[in] drain Flag indicating the buffer should be drained 90 * (return a buffer regardless of the current queue depth) 91 * 92 * \retval NULL, if drain set and queue is empty 93 * \retval NULL, if drain not set and queue not of the configured depth 94 */ 95struct lookahead_entry *vp9_lookahead_pop(struct lookahead_ctx *ctx, 96 int drain); 97 98 99/**\brief Get a future source buffer to encode 100 * 101 * \param[in] ctx Pointer to the lookahead context 102 * \param[in] index Index of the frame to be returned, 0 == next frame 103 * 104 * \retval NULL, if no buffer exists at the specified index 105 */ 106struct lookahead_entry *vp9_lookahead_peek(struct lookahead_ctx *ctx, 107 int index); 108 109 110/**\brief Get the number of frames currently in the lookahead queue 111 * 112 * \param[in] ctx Pointer to the lookahead context 113 */ 114unsigned int vp9_lookahead_depth(struct lookahead_ctx *ctx); 115 116#ifdef __cplusplus 117} // extern "C" 118#endif 119 120#endif // VP9_ENCODER_VP9_LOOKAHEAD_H_ 121