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 #ifndef VP9_ENCODER_VP9_QUANTIZE_H_
12 #define VP9_ENCODER_VP9_QUANTIZE_H_
13 
14 #include "./vpx_config.h"
15 #include "vp9/encoder/vp9_block.h"
16 
17 #ifdef __cplusplus
18 extern "C" {
19 #endif
20 
21 typedef struct {
22   DECLARE_ALIGNED(16, int16_t, y_quant[QINDEX_RANGE][8]);
23   DECLARE_ALIGNED(16, int16_t, y_quant_shift[QINDEX_RANGE][8]);
24   DECLARE_ALIGNED(16, int16_t, y_zbin[QINDEX_RANGE][8]);
25   DECLARE_ALIGNED(16, int16_t, y_round[QINDEX_RANGE][8]);
26 
27   // TODO(jingning): in progress of re-working the quantization. will decide
28   // if we want to deprecate the current use of y_quant.
29   DECLARE_ALIGNED(16, int16_t, y_quant_fp[QINDEX_RANGE][8]);
30   DECLARE_ALIGNED(16, int16_t, uv_quant_fp[QINDEX_RANGE][8]);
31   DECLARE_ALIGNED(16, int16_t, y_round_fp[QINDEX_RANGE][8]);
32   DECLARE_ALIGNED(16, int16_t, uv_round_fp[QINDEX_RANGE][8]);
33 
34   DECLARE_ALIGNED(16, int16_t, uv_quant[QINDEX_RANGE][8]);
35   DECLARE_ALIGNED(16, int16_t, uv_quant_shift[QINDEX_RANGE][8]);
36   DECLARE_ALIGNED(16, int16_t, uv_zbin[QINDEX_RANGE][8]);
37   DECLARE_ALIGNED(16, int16_t, uv_round[QINDEX_RANGE][8]);
38 } QUANTS;
39 
40 void vp9_quantize_dc(const int16_t *coeff_ptr, int skip_block,
41                      const int16_t *round_ptr, const int16_t quant_ptr,
42                      int16_t *qcoeff_ptr, int16_t *dqcoeff_ptr,
43                      const int16_t dequant_ptr, uint16_t *eob_ptr);
44 void vp9_quantize_dc_32x32(const int16_t *coeff_ptr, int skip_block,
45                            const int16_t *round_ptr, const int16_t quant_ptr,
46                            int16_t *qcoeff_ptr, int16_t *dqcoeff_ptr,
47                            const int16_t dequant_ptr, uint16_t *eob_ptr);
48 void vp9_regular_quantize_b_4x4(MACROBLOCK *x, int plane, int block,
49                                 const int16_t *scan, const int16_t *iscan);
50 
51 struct VP9_COMP;
52 struct VP9Common;
53 
54 void vp9_frame_init_quantizer(struct VP9_COMP *cpi);
55 
56 void vp9_update_zbin_extra(struct VP9_COMP *cpi, MACROBLOCK *x);
57 
58 void vp9_init_plane_quantizers(struct VP9_COMP *cpi, MACROBLOCK *x);
59 
60 void vp9_init_quantizer(struct VP9_COMP *cpi);
61 
62 void vp9_set_quantizer(struct VP9Common *cm, int q);
63 
64 int vp9_quantizer_to_qindex(int quantizer);
65 
66 int vp9_qindex_to_quantizer(int qindex);
67 
68 #ifdef __cplusplus
69 }  // extern "C"
70 #endif
71 
72 #endif  // VP9_ENCODER_VP9_QUANTIZE_H_
73