Coverage Report

Created: 2026-07-25 07:52

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/src/libvpx/vp8/encoder/vp8_quantize.c
Line
Count
Source
1
/*
2
 *  Copyright (c) 2010 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
#include <math.h>
12
13
#include "./vpx_config.h"
14
#include "vpx_ports/bitops.h"
15
#include "vpx_mem/vpx_mem.h"
16
17
#include "onyx_int.h"
18
#include "vp8/encoder/quantize.h"
19
#include "vp8/common/quant_common.h"
20
21
0
void vp8_fast_quantize_b_c(BLOCK *b, BLOCKD *d) {
22
0
  int i, rc, eob;
23
0
  int x, y, z, sz;
24
0
  short *coeff_ptr = b->coeff;
25
0
  short *round_ptr = b->round;
26
0
  short *quant_ptr = b->quant_fast;
27
0
  short *qcoeff_ptr = d->qcoeff;
28
0
  short *dqcoeff_ptr = d->dqcoeff;
29
0
  short *dequant_ptr = d->dequant;
30
31
0
  eob = -1;
32
0
  for (i = 0; i < 16; ++i) {
33
0
    rc = vp8_default_zig_zag1d[i];
34
0
    z = coeff_ptr[rc];
35
36
0
    sz = (z >> 31);    /* sign of z */
37
0
    x = (z ^ sz) - sz; /* x = abs(z) */
38
39
0
    y = ((x + round_ptr[rc]) * quant_ptr[rc]) >> 16; /* quantize (x) */
40
0
    x = (y ^ sz) - sz;                               /* get the sign back */
41
0
    qcoeff_ptr[rc] = x;                              /* write to destination */
42
0
    dqcoeff_ptr[rc] = x * dequant_ptr[rc];           /* dequantized value */
43
44
0
    if (y) {
45
0
      eob = i; /* last nonzero coeffs */
46
0
    }
47
0
  }
48
0
  *d->eob = (char)(eob + 1);
49
0
}
50
51
0
void vp8_regular_quantize_b_c(BLOCK *b, BLOCKD *d) {
52
0
  int i, rc, eob;
53
0
  int zbin;
54
0
  int x, y, z, sz;
55
0
  short *zbin_boost_ptr = b->zrun_zbin_boost;
56
0
  short *coeff_ptr = b->coeff;
57
0
  short *zbin_ptr = b->zbin;
58
0
  short *round_ptr = b->round;
59
0
  short *quant_ptr = b->quant;
60
0
  short *quant_shift_ptr = b->quant_shift;
61
0
  short *qcoeff_ptr = d->qcoeff;
62
0
  short *dqcoeff_ptr = d->dqcoeff;
63
0
  short *dequant_ptr = d->dequant;
64
0
  short zbin_oq_value = b->zbin_extra;
65
66
0
  memset(qcoeff_ptr, 0, 32);
67
0
  memset(dqcoeff_ptr, 0, 32);
68
69
0
  eob = -1;
70
71
0
  for (i = 0; i < 16; ++i) {
72
0
    rc = vp8_default_zig_zag1d[i];
73
0
    z = coeff_ptr[rc];
74
75
0
    zbin = zbin_ptr[rc] + *zbin_boost_ptr + zbin_oq_value;
76
77
0
    zbin_boost_ptr++;
78
0
    sz = (z >> 31);    /* sign of z */
79
0
    x = (z ^ sz) - sz; /* x = abs(z) */
80
81
0
    if (x >= zbin) {
82
0
      x += round_ptr[rc];
83
0
      y = ((((x * quant_ptr[rc]) >> 16) + x) * quant_shift_ptr[rc]) >>
84
0
          16;                                /* quantize (x) */
85
0
      x = (y ^ sz) - sz;                     /* get the sign back */
86
0
      qcoeff_ptr[rc] = x;                    /* write to destination */
87
0
      dqcoeff_ptr[rc] = x * dequant_ptr[rc]; /* dequantized value */
88
89
0
      if (y) {
90
0
        eob = i;                             /* last nonzero coeffs */
91
0
        zbin_boost_ptr = b->zrun_zbin_boost; /* reset zero runlength */
92
0
      }
93
0
    }
94
0
  }
95
96
0
  *d->eob = (char)(eob + 1);
97
0
}
98
99
858k
void vp8_quantize_mby(MACROBLOCK *x) {
100
858k
  int i;
101
858k
  int has_2nd_order = (x->e_mbd.mode_info_context->mbmi.mode != B_PRED &&
102
858k
                       x->e_mbd.mode_info_context->mbmi.mode != SPLITMV);
103
104
14.5M
  for (i = 0; i < 16; ++i) x->quantize_b(&x->block[i], &x->e_mbd.block[i]);
105
106
858k
  if (has_2nd_order) x->quantize_b(&x->block[24], &x->e_mbd.block[24]);
107
858k
}
108
109
643k
void vp8_quantize_mb(MACROBLOCK *x) {
110
643k
  int i;
111
643k
  int has_2nd_order = (x->e_mbd.mode_info_context->mbmi.mode != B_PRED &&
112
643k
                       x->e_mbd.mode_info_context->mbmi.mode != SPLITMV);
113
114
16.5M
  for (i = 0; i < 24 + has_2nd_order; ++i) {
115
15.9M
    x->quantize_b(&x->block[i], &x->e_mbd.block[i]);
116
15.9M
  }
117
643k
}
118
119
8.84M
void vp8_quantize_mbuv(MACROBLOCK *x) {
120
8.84M
  int i;
121
122
79.5M
  for (i = 16; i < 24; ++i) x->quantize_b(&x->block[i], &x->e_mbd.block[i]);
123
8.84M
}
124
125
static const int qrounding_factors[129] = {
126
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48,
127
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48,
128
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48,
129
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48,
130
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48,
131
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48,
132
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48
133
};
134
135
static const int qzbin_factors[129] = {
136
  84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84,
137
  84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84,
138
  84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 80, 80, 80, 80, 80, 80, 80, 80, 80,
139
  80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80,
140
  80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80,
141
  80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80,
142
  80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80
143
};
144
145
static const int qrounding_factors_y2[129] = {
146
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48,
147
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48,
148
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48,
149
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48,
150
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48,
151
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48,
152
  48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48
153
};
154
155
static const int qzbin_factors_y2[129] = {
156
  84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84,
157
  84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 84,
158
  84, 84, 84, 84, 84, 84, 84, 84, 84, 84, 80, 80, 80, 80, 80, 80, 80, 80, 80,
159
  80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80,
160
  80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80,
161
  80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80,
162
  80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 80
163
};
164
165
static void invert_quant(int improved_quant, short *quant, short *shift,
166
10.0M
                         short d) {
167
10.0M
  if (improved_quant) {
168
8.22M
    unsigned int t;
169
8.22M
    int l, m;
170
8.22M
    t = (unsigned int)d;
171
8.22M
    l = get_msb(t);
172
8.22M
    m = 1 + (1 << (16 + l)) / d;
173
8.22M
    *quant = (short)(m - (1 << 16));
174
8.22M
    *shift = l;
175
    /* use multiplication and constant shift by 16 */
176
8.22M
    *shift = 1 << (16 - *shift);
177
8.22M
  } else {
178
1.79M
    *quant = (1 << 16) / d;
179
1.79M
    *shift = 0;
180
1.79M
  }
181
10.0M
}
182
183
13.0k
void vp8cx_init_quantizer(VP8_COMP *cpi) {
184
13.0k
  int i;
185
13.0k
  int quant_val;
186
13.0k
  int Q;
187
188
13.0k
  int zbin_boost[16] = { 0,  0,  8,  10, 12, 14, 16, 20,
189
13.0k
                         24, 28, 32, 36, 40, 44, 44, 44 };
190
191
1.68M
  for (Q = 0; Q < QINDEX_RANGE; ++Q) {
192
    /* dc values */
193
1.66M
    quant_val = vp8_dc_quant(Q, cpi->common.y1dc_delta_q);
194
1.66M
    cpi->Y1quant_fast[Q][0] = (1 << 16) / quant_val;
195
1.66M
    invert_quant(cpi->sf.improved_quant, cpi->Y1quant[Q] + 0,
196
1.66M
                 cpi->Y1quant_shift[Q] + 0, quant_val);
197
1.66M
    cpi->Y1zbin[Q][0] = ((qzbin_factors[Q] * quant_val) + 64) >> 7;
198
1.66M
    cpi->Y1round[Q][0] = (qrounding_factors[Q] * quant_val) >> 7;
199
1.66M
    cpi->common.Y1dequant[Q][0] = quant_val;
200
1.66M
    cpi->zrun_zbin_boost_y1[Q][0] = (quant_val * zbin_boost[0]) >> 7;
201
202
1.66M
    quant_val = vp8_dc2quant(Q, cpi->common.y2dc_delta_q);
203
1.66M
    cpi->Y2quant_fast[Q][0] = (1 << 16) / quant_val;
204
1.66M
    invert_quant(cpi->sf.improved_quant, cpi->Y2quant[Q] + 0,
205
1.66M
                 cpi->Y2quant_shift[Q] + 0, quant_val);
206
1.66M
    cpi->Y2zbin[Q][0] = ((qzbin_factors_y2[Q] * quant_val) + 64) >> 7;
207
1.66M
    cpi->Y2round[Q][0] = (qrounding_factors_y2[Q] * quant_val) >> 7;
208
1.66M
    cpi->common.Y2dequant[Q][0] = quant_val;
209
1.66M
    cpi->zrun_zbin_boost_y2[Q][0] = (quant_val * zbin_boost[0]) >> 7;
210
211
1.66M
    quant_val = vp8_dc_uv_quant(Q, cpi->common.uvdc_delta_q);
212
1.66M
    cpi->UVquant_fast[Q][0] = (1 << 16) / quant_val;
213
1.66M
    invert_quant(cpi->sf.improved_quant, cpi->UVquant[Q] + 0,
214
1.66M
                 cpi->UVquant_shift[Q] + 0, quant_val);
215
1.66M
    cpi->UVzbin[Q][0] = ((qzbin_factors[Q] * quant_val) + 64) >> 7;
216
1.66M
    cpi->UVround[Q][0] = (qrounding_factors[Q] * quant_val) >> 7;
217
1.66M
    cpi->common.UVdequant[Q][0] = quant_val;
218
1.66M
    cpi->zrun_zbin_boost_uv[Q][0] = (quant_val * zbin_boost[0]) >> 7;
219
220
    /* all the ac values = ; */
221
1.66M
    quant_val = vp8_ac_yquant(Q);
222
1.66M
    cpi->Y1quant_fast[Q][1] = (1 << 16) / quant_val;
223
1.66M
    invert_quant(cpi->sf.improved_quant, cpi->Y1quant[Q] + 1,
224
1.66M
                 cpi->Y1quant_shift[Q] + 1, quant_val);
225
1.66M
    cpi->Y1zbin[Q][1] = ((qzbin_factors[Q] * quant_val) + 64) >> 7;
226
1.66M
    cpi->Y1round[Q][1] = (qrounding_factors[Q] * quant_val) >> 7;
227
1.66M
    cpi->common.Y1dequant[Q][1] = quant_val;
228
1.66M
    cpi->zrun_zbin_boost_y1[Q][1] = (quant_val * zbin_boost[1]) >> 7;
229
230
1.66M
    quant_val = vp8_ac2quant(Q, cpi->common.y2ac_delta_q);
231
1.66M
    cpi->Y2quant_fast[Q][1] = (1 << 16) / quant_val;
232
1.66M
    invert_quant(cpi->sf.improved_quant, cpi->Y2quant[Q] + 1,
233
1.66M
                 cpi->Y2quant_shift[Q] + 1, quant_val);
234
1.66M
    cpi->Y2zbin[Q][1] = ((qzbin_factors_y2[Q] * quant_val) + 64) >> 7;
235
1.66M
    cpi->Y2round[Q][1] = (qrounding_factors_y2[Q] * quant_val) >> 7;
236
1.66M
    cpi->common.Y2dequant[Q][1] = quant_val;
237
1.66M
    cpi->zrun_zbin_boost_y2[Q][1] = (quant_val * zbin_boost[1]) >> 7;
238
239
1.66M
    quant_val = vp8_ac_uv_quant(Q, cpi->common.uvac_delta_q);
240
1.66M
    cpi->UVquant_fast[Q][1] = (1 << 16) / quant_val;
241
1.66M
    invert_quant(cpi->sf.improved_quant, cpi->UVquant[Q] + 1,
242
1.66M
                 cpi->UVquant_shift[Q] + 1, quant_val);
243
1.66M
    cpi->UVzbin[Q][1] = ((qzbin_factors[Q] * quant_val) + 64) >> 7;
244
1.66M
    cpi->UVround[Q][1] = (qrounding_factors[Q] * quant_val) >> 7;
245
1.66M
    cpi->common.UVdequant[Q][1] = quant_val;
246
1.66M
    cpi->zrun_zbin_boost_uv[Q][1] = (quant_val * zbin_boost[1]) >> 7;
247
248
25.0M
    for (i = 2; i < 16; ++i) {
249
23.3M
      cpi->Y1quant_fast[Q][i] = cpi->Y1quant_fast[Q][1];
250
23.3M
      cpi->Y1quant[Q][i] = cpi->Y1quant[Q][1];
251
23.3M
      cpi->Y1quant_shift[Q][i] = cpi->Y1quant_shift[Q][1];
252
23.3M
      cpi->Y1zbin[Q][i] = cpi->Y1zbin[Q][1];
253
23.3M
      cpi->Y1round[Q][i] = cpi->Y1round[Q][1];
254
23.3M
      cpi->zrun_zbin_boost_y1[Q][i] =
255
23.3M
          (cpi->common.Y1dequant[Q][1] * zbin_boost[i]) >> 7;
256
257
23.3M
      cpi->Y2quant_fast[Q][i] = cpi->Y2quant_fast[Q][1];
258
23.3M
      cpi->Y2quant[Q][i] = cpi->Y2quant[Q][1];
259
23.3M
      cpi->Y2quant_shift[Q][i] = cpi->Y2quant_shift[Q][1];
260
23.3M
      cpi->Y2zbin[Q][i] = cpi->Y2zbin[Q][1];
261
23.3M
      cpi->Y2round[Q][i] = cpi->Y2round[Q][1];
262
23.3M
      cpi->zrun_zbin_boost_y2[Q][i] =
263
23.3M
          (cpi->common.Y2dequant[Q][1] * zbin_boost[i]) >> 7;
264
265
23.3M
      cpi->UVquant_fast[Q][i] = cpi->UVquant_fast[Q][1];
266
23.3M
      cpi->UVquant[Q][i] = cpi->UVquant[Q][1];
267
23.3M
      cpi->UVquant_shift[Q][i] = cpi->UVquant_shift[Q][1];
268
23.3M
      cpi->UVzbin[Q][i] = cpi->UVzbin[Q][1];
269
23.3M
      cpi->UVround[Q][i] = cpi->UVround[Q][1];
270
23.3M
      cpi->zrun_zbin_boost_uv[Q][i] =
271
23.3M
          (cpi->common.UVdequant[Q][1] * zbin_boost[i]) >> 7;
272
23.3M
    }
273
1.66M
  }
274
13.0k
}
275
276
#define ZBIN_EXTRA_Y                                                \
277
777k
  ((cpi->common.Y1dequant[QIndex][1] *                              \
278
777k
    (x->zbin_over_quant + x->zbin_mode_boost + x->act_zbin_adj)) >> \
279
777k
   7)
280
281
#define ZBIN_EXTRA_UV                                               \
282
777k
  ((cpi->common.UVdequant[QIndex][1] *                              \
283
777k
    (x->zbin_over_quant + x->zbin_mode_boost + x->act_zbin_adj)) >> \
284
777k
   7)
285
286
#define ZBIN_EXTRA_Y2                                                     \
287
777k
  ((cpi->common.Y2dequant[QIndex][1] *                                    \
288
777k
    ((x->zbin_over_quant / 2) + x->zbin_mode_boost + x->act_zbin_adj)) >> \
289
777k
   7)
290
291
133k
void vp8cx_mb_init_quantizer(VP8_COMP *cpi, MACROBLOCK *x, int ok_to_skip) {
292
133k
  int i;
293
133k
  int QIndex;
294
133k
  MACROBLOCKD *xd = &x->e_mbd;
295
133k
  int zbin_extra;
296
297
  /* Select the baseline MB Q index. */
298
133k
  if (xd->segmentation_enabled) {
299
    /* Abs Value */
300
0
    if (xd->mb_segment_abs_delta == SEGMENT_ABSDATA) {
301
0
      QIndex = xd->segment_feature_data[MB_LVL_ALT_Q]
302
0
                                       [xd->mode_info_context->mbmi.segment_id];
303
      /* Delta Value */
304
0
    } else {
305
0
      QIndex = cpi->common.base_qindex +
306
0
               xd->segment_feature_data[MB_LVL_ALT_Q]
307
0
                                       [xd->mode_info_context->mbmi.segment_id];
308
      /* Clamp to valid range */
309
0
      QIndex = (QIndex >= 0) ? ((QIndex <= MAXQ) ? QIndex : MAXQ) : 0;
310
0
    }
311
133k
  } else {
312
133k
    QIndex = cpi->common.base_qindex;
313
133k
  }
314
315
  /* This initialization should be called at least once. Use ok_to_skip to
316
   * decide if it is ok to skip.
317
   * Before encoding a frame, this function is always called with ok_to_skip
318
   * =0, which means no skiping of calculations. The "last" values are
319
   * initialized at that time.
320
   */
321
133k
  if (!ok_to_skip || QIndex != x->q_index) {
322
133k
    xd->dequant_y1_dc[0] = 1;
323
133k
    xd->dequant_y1[0] = cpi->common.Y1dequant[QIndex][0];
324
133k
    xd->dequant_y2[0] = cpi->common.Y2dequant[QIndex][0];
325
133k
    xd->dequant_uv[0] = cpi->common.UVdequant[QIndex][0];
326
327
2.12M
    for (i = 1; i < 16; ++i) {
328
1.99M
      xd->dequant_y1_dc[i] = xd->dequant_y1[i] =
329
1.99M
          cpi->common.Y1dequant[QIndex][1];
330
1.99M
      xd->dequant_y2[i] = cpi->common.Y2dequant[QIndex][1];
331
1.99M
      xd->dequant_uv[i] = cpi->common.UVdequant[QIndex][1];
332
1.99M
    }
333
133k
#if 1
334
    /*TODO:  Remove dequant from BLOCKD.  This is a temporary solution until
335
     * the quantizer code uses a passed in pointer to the dequant constants.
336
     * This will also require modifications to the x86 and neon assembly.
337
     * */
338
2.26M
    for (i = 0; i < 16; ++i) x->e_mbd.block[i].dequant = xd->dequant_y1;
339
1.19M
    for (i = 16; i < 24; ++i) x->e_mbd.block[i].dequant = xd->dequant_uv;
340
133k
    x->e_mbd.block[24].dequant = xd->dequant_y2;
341
133k
#endif
342
343
    /* Y */
344
133k
    zbin_extra = ZBIN_EXTRA_Y;
345
346
2.26M
    for (i = 0; i < 16; ++i) {
347
2.12M
      x->block[i].quant = cpi->Y1quant[QIndex];
348
2.12M
      x->block[i].quant_fast = cpi->Y1quant_fast[QIndex];
349
2.12M
      x->block[i].quant_shift = cpi->Y1quant_shift[QIndex];
350
2.12M
      x->block[i].zbin = cpi->Y1zbin[QIndex];
351
2.12M
      x->block[i].round = cpi->Y1round[QIndex];
352
2.12M
      x->block[i].zrun_zbin_boost = cpi->zrun_zbin_boost_y1[QIndex];
353
2.12M
      x->block[i].zbin_extra = (short)zbin_extra;
354
2.12M
    }
355
356
    /* UV */
357
133k
    zbin_extra = ZBIN_EXTRA_UV;
358
359
1.19M
    for (i = 16; i < 24; ++i) {
360
1.06M
      x->block[i].quant = cpi->UVquant[QIndex];
361
1.06M
      x->block[i].quant_fast = cpi->UVquant_fast[QIndex];
362
1.06M
      x->block[i].quant_shift = cpi->UVquant_shift[QIndex];
363
1.06M
      x->block[i].zbin = cpi->UVzbin[QIndex];
364
1.06M
      x->block[i].round = cpi->UVround[QIndex];
365
1.06M
      x->block[i].zrun_zbin_boost = cpi->zrun_zbin_boost_uv[QIndex];
366
1.06M
      x->block[i].zbin_extra = (short)zbin_extra;
367
1.06M
    }
368
369
    /* Y2 */
370
133k
    zbin_extra = ZBIN_EXTRA_Y2;
371
372
133k
    x->block[24].quant_fast = cpi->Y2quant_fast[QIndex];
373
133k
    x->block[24].quant = cpi->Y2quant[QIndex];
374
133k
    x->block[24].quant_shift = cpi->Y2quant_shift[QIndex];
375
133k
    x->block[24].zbin = cpi->Y2zbin[QIndex];
376
133k
    x->block[24].round = cpi->Y2round[QIndex];
377
133k
    x->block[24].zrun_zbin_boost = cpi->zrun_zbin_boost_y2[QIndex];
378
133k
    x->block[24].zbin_extra = (short)zbin_extra;
379
380
    /* save this macroblock QIndex for vp8_update_zbin_extra() */
381
133k
    x->q_index = QIndex;
382
383
133k
    x->last_zbin_over_quant = x->zbin_over_quant;
384
133k
    x->last_zbin_mode_boost = x->zbin_mode_boost;
385
133k
    x->last_act_zbin_adj = x->act_zbin_adj;
386
387
133k
  } else if (x->last_zbin_over_quant != x->zbin_over_quant ||
388
0
             x->last_zbin_mode_boost != x->zbin_mode_boost ||
389
0
             x->last_act_zbin_adj != x->act_zbin_adj) {
390
    /* Y */
391
0
    zbin_extra = ZBIN_EXTRA_Y;
392
393
0
    for (i = 0; i < 16; ++i) x->block[i].zbin_extra = (short)zbin_extra;
394
395
    /* UV */
396
0
    zbin_extra = ZBIN_EXTRA_UV;
397
398
0
    for (i = 16; i < 24; ++i) x->block[i].zbin_extra = (short)zbin_extra;
399
400
    /* Y2 */
401
0
    zbin_extra = ZBIN_EXTRA_Y2;
402
0
    x->block[24].zbin_extra = (short)zbin_extra;
403
404
0
    x->last_zbin_over_quant = x->zbin_over_quant;
405
0
    x->last_zbin_mode_boost = x->zbin_mode_boost;
406
0
    x->last_act_zbin_adj = x->act_zbin_adj;
407
0
  }
408
133k
}
409
410
644k
void vp8_update_zbin_extra(VP8_COMP *cpi, MACROBLOCK *x) {
411
644k
  int i;
412
644k
  int QIndex = x->q_index;
413
644k
  int zbin_extra;
414
415
  /* Y */
416
644k
  zbin_extra = ZBIN_EXTRA_Y;
417
418
10.9M
  for (i = 0; i < 16; ++i) x->block[i].zbin_extra = (short)zbin_extra;
419
420
  /* UV */
421
644k
  zbin_extra = ZBIN_EXTRA_UV;
422
423
5.80M
  for (i = 16; i < 24; ++i) x->block[i].zbin_extra = (short)zbin_extra;
424
425
  /* Y2 */
426
644k
  zbin_extra = ZBIN_EXTRA_Y2;
427
644k
  x->block[24].zbin_extra = (short)zbin_extra;
428
644k
}
429
#undef ZBIN_EXTRA_Y
430
#undef ZBIN_EXTRA_UV
431
#undef ZBIN_EXTRA_Y2
432
433
133k
void vp8cx_frame_init_quantizer(VP8_COMP *cpi) {
434
  /* Clear Zbin mode boost for default case */
435
133k
  cpi->mb.zbin_mode_boost = 0;
436
437
  /* MB level quantizer setup */
438
133k
  vp8cx_mb_init_quantizer(cpi, &cpi->mb, 0);
439
133k
}
440
441
133k
void vp8_set_quantizer(struct VP8_COMP *cpi, int Q) {
442
133k
  VP8_COMMON *cm = &cpi->common;
443
133k
  MACROBLOCKD *mbd = &cpi->mb.e_mbd;
444
133k
  int update = 0;
445
133k
  int new_delta_q;
446
133k
  int new_uv_delta_q;
447
133k
  cm->base_qindex = Q;
448
449
  /* if any of the delta_q values are changing update flag has to be set */
450
  /* currently only y2dc_delta_q may change */
451
452
133k
  cm->y1dc_delta_q = 0;
453
133k
  cm->y2ac_delta_q = 0;
454
455
133k
  if (Q < 4) {
456
0
    new_delta_q = 4 - Q;
457
133k
  } else {
458
133k
    new_delta_q = 0;
459
133k
  }
460
461
133k
  update |= cm->y2dc_delta_q != new_delta_q;
462
133k
  cm->y2dc_delta_q = new_delta_q;
463
464
133k
  new_uv_delta_q = 0;
465
  // For screen content, lower the q value for UV channel. For now, select
466
  // conservative delta; same delta for dc and ac, and decrease it with lower
467
  // Q, and set to 0 below some threshold. May want to condition this in
468
  // future on the variance/energy in UV channel.
469
133k
  if (cpi->oxcf.screen_content_mode && Q > 40) {
470
0
    new_uv_delta_q = -(int)(0.15 * Q);
471
    // Check range: magnitude of delta is 4 bits.
472
0
    if (new_uv_delta_q < -15) {
473
0
      new_uv_delta_q = -15;
474
0
    }
475
0
  }
476
133k
  update |= cm->uvdc_delta_q != new_uv_delta_q;
477
133k
  cm->uvdc_delta_q = new_uv_delta_q;
478
133k
  cm->uvac_delta_q = new_uv_delta_q;
479
480
  /* Set Segment specific quatizers */
481
133k
  mbd->segment_feature_data[MB_LVL_ALT_Q][0] =
482
133k
      cpi->segment_feature_data[MB_LVL_ALT_Q][0];
483
133k
  mbd->segment_feature_data[MB_LVL_ALT_Q][1] =
484
133k
      cpi->segment_feature_data[MB_LVL_ALT_Q][1];
485
133k
  mbd->segment_feature_data[MB_LVL_ALT_Q][2] =
486
133k
      cpi->segment_feature_data[MB_LVL_ALT_Q][2];
487
133k
  mbd->segment_feature_data[MB_LVL_ALT_Q][3] =
488
133k
      cpi->segment_feature_data[MB_LVL_ALT_Q][3];
489
490
  /* quantizer has to be reinitialized for any delta_q changes */
491
133k
  if (update) vp8cx_init_quantizer(cpi);
492
133k
}