From 41985d40c652b4065848204e42fac6706e15955a Mon Sep 17 00:00:00 2001
From: Rostislav Pehlivanov <atomnuker@gmail.com>
Date: Tue, 1 Mar 2016 13:04:40 +0000
Subject: [PATCH] vc2enc: simplify slice cost caching

The fact that now all quantization indices costs are cached justifies
storing 20 more integers in a structure already allocated on heap.

Signed-off-by: Rostislav Pehlivanov <atomnuker@gmail.com>
---
 libavcodec/vc2enc.c | 44 +++++++++++++-------------------------------
 1 file changed, 13 insertions(+), 31 deletions(-)

diff --git a/libavcodec/vc2enc.c b/libavcodec/vc2enc.c
index 5724d65af7..2d875d13d3 100644
--- a/libavcodec/vc2enc.c
+++ b/libavcodec/vc2enc.c
@@ -36,9 +36,6 @@
  * (COEF_LUT_TAB*MAX_QUANT_INDEX) since the sign is appended during encoding */
 #define COEF_LUT_TAB 2048
 
-/* Per slice quantization bit cost cache */
-#define SLICE_CACHED_QUANTIZERS 30
-
 /* Decides the cutoff point in # of slices to distribute the leftover bytes */
 #define SLICE_REDIST_TOTAL 150
 
@@ -67,15 +64,9 @@ typedef struct Plane {
     ptrdiff_t coef_stride;
 } Plane;
 
-typedef struct BitCostCache {
-    int bits;
-    int quantizer;
-} BitCostCache;
-
 typedef struct SliceArgs {
     PutBitContext pb;
-    BitCostCache cache[SLICE_CACHED_QUANTIZERS];
-    int cached_results;
+    int cache[MAX_QUANT_INDEX];
     void *ctx;
     int x;
     int y;
@@ -584,18 +575,15 @@ static void encode_subband(VC2EncContext *s, PutBitContext *pb, int sx, int sy,
     }
 }
 
-static int count_hq_slice(VC2EncContext *s, BitCostCache *cache,
-                          int *cached_results, int slice_x, int slice_y,
-                          int quant_idx)
+static int count_hq_slice(VC2EncContext *s, int *cache,
+                          int slice_x, int slice_y, int quant_idx)
 {
-    int i, x, y;
+    int x, y;
     uint8_t quants[MAX_DWT_LEVELS][4];
     int bits = 0, p, level, orientation;
 
-    if (cache && *cached_results)
-        for (i = 0; i < *cached_results; i++)
-            if (cache[i].quantizer == quant_idx)
-                return cache[i].bits;
+    if (cache && cache[quant_idx])
+        return cache[quant_idx];
 
     bits += 8*s->prefix_bytes;
     bits += 8; /* quant_idx */
@@ -646,11 +634,8 @@ static int count_hq_slice(VC2EncContext *s, BitCostCache *cache,
         bits += pad_c*8;
     }
 
-    if (cache) {
-        cache[*cached_results].quantizer = quant_idx;
-        cache[*cached_results].bits = bits;
-        *cached_results = FFMIN(*cached_results + 1, SLICE_CACHED_QUANTIZERS);
-    }
+    if (cache)
+        cache[quant_idx] = bits;
 
     return bits;
 }
@@ -668,14 +653,12 @@ static int rate_control(AVCodecContext *avctx, void *arg)
     int quant = slice_dat->quant_idx, range = quant/5;
     const int top = slice_dat->bits_ceil;
     const int bottom = slice_dat->bits_floor;
-    int bits = count_hq_slice(s, slice_dat->cache, &slice_dat->cached_results,
-                              sx, sy, quant);
+    int bits = count_hq_slice(s, slice_dat->cache, sx, sy, quant);
     range -= range & 1; /* Make it an even number */
     while ((bits > top) || (bits < bottom)) {
         range *= bits > top ? +1 : -1;
         quant = av_clip(quant + range, 0, s->q_ceil);
-        bits = count_hq_slice(s, slice_dat->cache, &slice_dat->cached_results,
-                              sx, sy, quant);
+        bits = count_hq_slice(s, slice_dat->cache, sx, sy, quant);
         range = av_clip(range/2, 1, s->q_ceil);
         if (quant_buf[1] == quant) {
             quant = bits_last < bits ? quant_buf[0] : quant;
@@ -704,9 +687,9 @@ static void calc_slice_sizes(VC2EncContext *s)
             args->ctx = s;
             args->x = slice_x;
             args->y = slice_y;
-            args->cached_results = 0;
             args->bits_ceil = s->slice_max_bytes << 3;
             args->bits_floor = s->slice_min_bytes << 3;
+            memset(args->cache, 0, MAX_QUANT_INDEX*sizeof(*args->cache));
         }
     }
 
@@ -808,8 +791,7 @@ static int encode_slices(VC2EncContext *s)
             args = top_loc[i];
             prev_bytes = args->bytes;
             new_idx = av_clip(args->quant_idx - 1, 0, s->q_ceil);
-            bits = count_hq_slice(s, args->cache, &args->cached_results,
-                                  args->x, args->y, new_idx);
+            bits = count_hq_slice(s, args->cache, args->x, args->y, new_idx);
             bytes = FFALIGN((bits >> 3), s->size_scaler) + 4 + s->prefix_bytes;
             diff = bytes - prev_bytes;
             if ((bytes_left - diff) >= 0) {
@@ -1050,7 +1032,7 @@ static int minimum_frame_bits(VC2EncContext *s)
     s->size_scaler = 64;
     for (slice_y = 0; slice_y < s->num_y; slice_y++) {
         for (slice_x = 0; slice_x < s->num_x; slice_x++) {
-            bits += count_hq_slice(s, NULL, NULL, slice_x, slice_y, s->q_ceil);
+            bits += count_hq_slice(s, NULL, slice_x, slice_y, s->q_ceil);
         }
     }
     return bits;