Index: source/libvpx/vp9/encoder/vp9_lookahead.c |
=================================================================== |
--- source/libvpx/vp9/encoder/vp9_lookahead.c (revision 0) |
+++ source/libvpx/vp9/encoder/vp9_lookahead.c (revision 0) |
@@ -0,0 +1,191 @@ |
+/* |
+ * Copyright (c) 2011 The WebM project authors. All Rights Reserved. |
+ * |
+ * Use of this source code is governed by a BSD-style license |
+ * that can be found in the LICENSE file in the root of the source |
+ * tree. An additional intellectual property rights grant can be found |
+ * in the file PATENTS. All contributing project authors may |
+ * be found in the AUTHORS file in the root of the source tree. |
+ */ |
+#include <assert.h> |
+#include <stdlib.h> |
+#include "vpx_config.h" |
+#include "vp9/encoder/vp9_lookahead.h" |
+#include "vp9/common/vp9_extend.h" |
+ |
+#define MAX_LAG_BUFFERS 25 |
+ |
+struct lookahead_ctx { |
+ unsigned int max_sz; /* Absolute size of the queue */ |
+ unsigned int sz; /* Number of buffers currently in the queue */ |
+ unsigned int read_idx; /* Read index */ |
+ unsigned int write_idx; /* Write index */ |
+ struct lookahead_entry *buf; /* Buffer list */ |
+}; |
+ |
+ |
+/* Return the buffer at the given absolute index and increment the index */ |
+static struct lookahead_entry * |
+pop(struct lookahead_ctx *ctx, |
+ unsigned int *idx) { |
+ unsigned int index = *idx; |
+ struct lookahead_entry *buf = ctx->buf + index; |
+ |
+ assert(index < ctx->max_sz); |
+ if (++index >= ctx->max_sz) |
+ index -= ctx->max_sz; |
+ *idx = index; |
+ return buf; |
+} |
+ |
+ |
+void |
+vp9_lookahead_destroy(struct lookahead_ctx *ctx) { |
+ if (ctx) { |
+ if (ctx->buf) { |
+ unsigned int i; |
+ |
+ for (i = 0; i < ctx->max_sz; i++) |
+ vp8_yv12_de_alloc_frame_buffer(&ctx->buf[i].img); |
+ free(ctx->buf); |
+ } |
+ free(ctx); |
+ } |
+} |
+ |
+ |
+struct lookahead_ctx * |
+vp9_lookahead_init(unsigned int width, |
+ unsigned int height, |
+ unsigned int depth) { |
+ struct lookahead_ctx *ctx = NULL; |
+ |
+ /* Clamp the lookahead queue depth */ |
+ if (depth < 1) |
+ depth = 1; |
+ else if (depth > MAX_LAG_BUFFERS) |
+ depth = MAX_LAG_BUFFERS; |
+ |
+ /* Align the buffer dimensions */ |
+ width = (width + 15) &~15; |
+ height = (height + 15) &~15; |
+ |
+ /* Allocate the lookahead structures */ |
+ ctx = calloc(1, sizeof(*ctx)); |
+ if (ctx) { |
+ unsigned int i; |
+ ctx->max_sz = depth; |
+ ctx->buf = calloc(depth, sizeof(*ctx->buf)); |
+ if (!ctx->buf) |
+ goto bail; |
+ for (i = 0; i < depth; i++) |
+ if (vp8_yv12_alloc_frame_buffer(&ctx->buf[i].img, |
+ width, height, VP9BORDERINPIXELS)) |
+ goto bail; |
+ } |
+ return ctx; |
+bail: |
+ vp9_lookahead_destroy(ctx); |
+ return NULL; |
+} |
+ |
+ |
+int |
+vp9_lookahead_push(struct lookahead_ctx *ctx, |
+ YV12_BUFFER_CONFIG *src, |
+ int64_t ts_start, |
+ int64_t ts_end, |
+ unsigned int flags, |
+ unsigned char *active_map) { |
+ struct lookahead_entry *buf; |
+ int row, col, active_end; |
+ int mb_rows = (src->y_height + 15) >> 4; |
+ int mb_cols = (src->y_width + 15) >> 4; |
+ |
+ if (ctx->sz + 1 > ctx->max_sz) |
+ return 1; |
+ ctx->sz++; |
+ buf = pop(ctx, &ctx->write_idx); |
+ |
+ // Only do this partial copy if the following conditions are all met: |
+ // 1. Lookahead queue has has size of 1. |
+ // 2. Active map is provided. |
+ // 3. This is not a key frame, golden nor altref frame. |
+ if (ctx->max_sz == 1 && active_map && !flags) { |
+ for (row = 0; row < mb_rows; ++row) { |
+ col = 0; |
+ |
+ while (1) { |
+ // Find the first active macroblock in this row. |
+ for (; col < mb_cols; ++col) { |
+ if (active_map[col]) |
+ break; |
+ } |
+ |
+ // No more active macroblock in this row. |
+ if (col == mb_cols) |
+ break; |
+ |
+ // Find the end of active region in this row. |
+ active_end = col; |
+ |
+ for (; active_end < mb_cols; ++active_end) { |
+ if (!active_map[active_end]) |
+ break; |
+ } |
+ |
+ // Only copy this active region. |
+ vp9_copy_and_extend_frame_with_rect(src, &buf->img, |
+ row << 4, |
+ col << 4, 16, |
+ (active_end - col) << 4); |
+ |
+ // Start again from the end of this active region. |
+ col = active_end; |
+ } |
+ |
+ active_map += mb_cols; |
+ } |
+ } else { |
+ vp9_copy_and_extend_frame(src, &buf->img); |
+ } |
+ buf->ts_start = ts_start; |
+ buf->ts_end = ts_end; |
+ buf->flags = flags; |
+ return 0; |
+} |
+ |
+ |
+struct lookahead_entry * |
+vp9_lookahead_pop(struct lookahead_ctx *ctx, |
+ int drain) { |
+ struct lookahead_entry *buf = NULL; |
+ |
+ if (ctx->sz && (drain || ctx->sz == ctx->max_sz)) { |
+ buf = pop(ctx, &ctx->read_idx); |
+ ctx->sz--; |
+ } |
+ return buf; |
+} |
+ |
+ |
+struct lookahead_entry * |
+vp9_lookahead_peek(struct lookahead_ctx *ctx, |
+ int index) { |
+ struct lookahead_entry *buf = NULL; |
+ |
+ assert(index < ctx->max_sz); |
+ if (index < (int)ctx->sz) { |
+ index += ctx->read_idx; |
+ if (index >= (int)ctx->max_sz) |
+ index -= ctx->max_sz; |
+ buf = ctx->buf + index; |
+ } |
+ return buf; |
+} |
+ |
+ |
+unsigned int |
+vp9_lookahead_depth(struct lookahead_ctx *ctx) { |
+ return ctx->sz; |
+} |
Property changes on: source/libvpx/vp9/encoder/vp9_lookahead.c |
___________________________________________________________________ |
Added: svn:eol-style |
+ LF |