1 /* 2 * Copyright (c) 2016, Alliance for Open Media. All rights reserved 3 * 4 * This source code is subject to the terms of the BSD 2 Clause License and 5 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License 6 * was not distributed with this source code in the LICENSE file, you can 7 * obtain it at www.aomedia.org/license/software. If the Alliance for Open 8 * Media Patent License 1.0 was not distributed with this source code in the 9 * PATENTS file, you can obtain it at www.aomedia.org/license/patent. 10 */ 11 12 #ifndef AOM_AV1_ENCODER_AV1_QUANTIZE_H_ 13 #define AOM_AV1_ENCODER_AV1_QUANTIZE_H_ 14 15 #include "config/aom_config.h" 16 17 #include "av1/common/quant_common.h" 18 #include "av1/common/scan.h" 19 #include "av1/encoder/block.h" 20 21 #ifdef __cplusplus 22 extern "C" { 23 #endif 24 25 typedef struct QUANT_PARAM { 26 int log_scale; 27 TX_SIZE tx_size; 28 const qm_val_t *qmatrix; 29 const qm_val_t *iqmatrix; 30 int use_quant_b_adapt; 31 int use_optimize_b; 32 int xform_quant_idx; 33 } QUANT_PARAM; 34 35 typedef void (*AV1_QUANT_FACADE)(const tran_low_t *coeff_ptr, intptr_t n_coeffs, 36 const MACROBLOCK_PLANE *p, 37 tran_low_t *qcoeff_ptr, 38 tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr, 39 const SCAN_ORDER *sc, 40 const QUANT_PARAM *qparam); 41 42 // The QUANTS structure is used only for internal quantizer setup in 43 // av1_quantize.c. 44 // All of its fields use the same coefficient shift/scaling at TX. 45 typedef struct { 46 // 0: dc 1: ac 2-8: ac repeated to SIMD width 47 DECLARE_ALIGNED(16, int16_t, y_quant[QINDEX_RANGE][8]); 48 DECLARE_ALIGNED(16, int16_t, y_quant_shift[QINDEX_RANGE][8]); 49 DECLARE_ALIGNED(16, int16_t, y_zbin[QINDEX_RANGE][8]); 50 DECLARE_ALIGNED(16, int16_t, y_round[QINDEX_RANGE][8]); 51 52 // TODO(jingning): in progress of re-working the quantization. will decide 53 // if we want to deprecate the current use of y_quant. 54 DECLARE_ALIGNED(16, int16_t, y_quant_fp[QINDEX_RANGE][8]); 55 DECLARE_ALIGNED(16, int16_t, u_quant_fp[QINDEX_RANGE][8]); 56 DECLARE_ALIGNED(16, int16_t, v_quant_fp[QINDEX_RANGE][8]); 57 DECLARE_ALIGNED(16, int16_t, y_round_fp[QINDEX_RANGE][8]); 58 DECLARE_ALIGNED(16, int16_t, u_round_fp[QINDEX_RANGE][8]); 59 DECLARE_ALIGNED(16, int16_t, v_round_fp[QINDEX_RANGE][8]); 60 61 DECLARE_ALIGNED(16, int16_t, u_quant[QINDEX_RANGE][8]); 62 DECLARE_ALIGNED(16, int16_t, v_quant[QINDEX_RANGE][8]); 63 DECLARE_ALIGNED(16, int16_t, u_quant_shift[QINDEX_RANGE][8]); 64 DECLARE_ALIGNED(16, int16_t, v_quant_shift[QINDEX_RANGE][8]); 65 DECLARE_ALIGNED(16, int16_t, u_zbin[QINDEX_RANGE][8]); 66 DECLARE_ALIGNED(16, int16_t, v_zbin[QINDEX_RANGE][8]); 67 DECLARE_ALIGNED(16, int16_t, u_round[QINDEX_RANGE][8]); 68 DECLARE_ALIGNED(16, int16_t, v_round[QINDEX_RANGE][8]); 69 } QUANTS; 70 71 // The Dequants structure is used only for internal quantizer setup in 72 // av1_quantize.c. 73 // Fields are suffixed according to whether or not they're expressed in 74 // the same coefficient shift/precision as TX or a fixed Q3 format. 75 typedef struct { 76 DECLARE_ALIGNED(16, int16_t, 77 y_dequant_QTX[QINDEX_RANGE][8]); // 8: SIMD width 78 DECLARE_ALIGNED(16, int16_t, 79 u_dequant_QTX[QINDEX_RANGE][8]); // 8: SIMD width 80 DECLARE_ALIGNED(16, int16_t, 81 v_dequant_QTX[QINDEX_RANGE][8]); // 8: SIMD width 82 } Dequants; 83 84 typedef struct { 85 // Quantization parameters for internal quantizer setup. 86 QUANTS quants; 87 // Dequantization parameters for internal quantizer setup. 88 Dequants dequants; 89 } EncQuantDequantParams; 90 91 struct AV1_COMP; 92 struct AV1Common; 93 94 void av1_frame_init_quantizer(struct AV1_COMP *cpi); 95 96 void av1_init_plane_quantizers(const struct AV1_COMP *cpi, MACROBLOCK *x, 97 int segment_id, const int do_update); 98 99 void av1_build_quantizer(aom_bit_depth_t bit_depth, int y_dc_delta_q, 100 int u_dc_delta_q, int u_ac_delta_q, int v_dc_delta_q, 101 int v_ac_delta_q, QUANTS *const quants, 102 Dequants *const deq); 103 104 void av1_init_quantizer(EncQuantDequantParams *const enc_quant_dequant_params, 105 const CommonQuantParams *quant_params, 106 aom_bit_depth_t bit_depth); 107 108 void av1_set_quantizer(struct AV1Common *const cm, int min_qmlevel, 109 int max_qmlevel, int q, int enable_chroma_deltaq, 110 int enable_hdr_deltaq); 111 112 int av1_quantizer_to_qindex(int quantizer); 113 114 int av1_qindex_to_quantizer(int qindex); 115 116 void av1_quantize_skip(intptr_t n_coeffs, tran_low_t *qcoeff_ptr, 117 tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr); 118 119 /*!\brief Quantize transform coefficients without using qmatrix 120 * 121 * quant_ptr, dequant_ptr and round_ptr are size 2 arrays, 122 * where index 0 corresponds to dc coeff and index 1 corresponds to ac coeffs. 123 * 124 * \param[in] quant_ptr 16-bit fixed point representation of inverse 125 * quantize step size, i.e. 2^16/dequant 126 * \param[in] dequant_ptr quantize step size 127 * \param[in] round_ptr rounding 128 * \param[in] log_scale the relative log scale of the transform 129 * coefficients 130 * \param[in] scan scan[i] indicates the position of ith to-be-coded 131 * coefficient 132 * \param[in] coeff_count number of coefficients 133 * \param[out] qcoeff_ptr quantized coefficients 134 * \param[out] dqcoeff_ptr dequantized coefficients 135 * 136 * \return The last non-zero coefficient's scan index plus 1 137 */ 138 int av1_quantize_fp_no_qmatrix(const int16_t quant_ptr[2], 139 const int16_t dequant_ptr[2], 140 const int16_t round_ptr[2], int log_scale, 141 const int16_t *scan, int coeff_count, 142 const tran_low_t *coeff_ptr, 143 tran_low_t *qcoeff_ptr, tran_low_t *dqcoeff_ptr); 144 145 void av1_quantize_fp_facade(const tran_low_t *coeff_ptr, intptr_t n_coeffs, 146 const MACROBLOCK_PLANE *p, tran_low_t *qcoeff_ptr, 147 tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr, 148 const SCAN_ORDER *sc, const QUANT_PARAM *qparam); 149 150 void av1_quantize_b_facade(const tran_low_t *coeff_ptr, intptr_t n_coeffs, 151 const MACROBLOCK_PLANE *p, tran_low_t *qcoeff_ptr, 152 tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr, 153 const SCAN_ORDER *sc, const QUANT_PARAM *qparam); 154 155 void av1_quantize_dc_facade(const tran_low_t *coeff_ptr, intptr_t n_coeffs, 156 const MACROBLOCK_PLANE *p, tran_low_t *qcoeff_ptr, 157 tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr, 158 const SCAN_ORDER *sc, const QUANT_PARAM *qparam); 159 160 /*!\brief Update quantize parameters in MACROBLOCK 161 * 162 * \param[in] enc_quant_dequant_params This parameter cached the quantize and 163 * dequantize parameters for all q 164 * indices. 165 * \param[in] qindex Quantize index used for the current 166 * superblock. 167 * \param[out] x A superblock data structure for 168 * encoder. 169 */ 170 void av1_set_q_index(const EncQuantDequantParams *enc_quant_dequant_params, 171 int qindex, MACROBLOCK *x); 172 173 /*!\brief Update quantize matrix in MACROBLOCKD based on segment id 174 * 175 * \param[in] quant_params Quantize parameters used by encoder and decoder 176 * \param[in] segment_id Segment id. 177 * \param[out] xd A superblock data structure used by encoder and 178 * decoder. 179 */ 180 void av1_set_qmatrix(const CommonQuantParams *quant_params, int segment_id, 181 MACROBLOCKD *xd); 182 183 #if CONFIG_AV1_HIGHBITDEPTH 184 void av1_highbd_quantize_fp_facade(const tran_low_t *coeff_ptr, 185 intptr_t n_coeffs, const MACROBLOCK_PLANE *p, 186 tran_low_t *qcoeff_ptr, 187 tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr, 188 const SCAN_ORDER *sc, 189 const QUANT_PARAM *qparam); 190 191 void av1_highbd_quantize_b_facade(const tran_low_t *coeff_ptr, 192 intptr_t n_coeffs, const MACROBLOCK_PLANE *p, 193 tran_low_t *qcoeff_ptr, 194 tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr, 195 const SCAN_ORDER *sc, 196 const QUANT_PARAM *qparam); 197 198 void av1_highbd_quantize_dc_facade(const tran_low_t *coeff_ptr, 199 intptr_t n_coeffs, const MACROBLOCK_PLANE *p, 200 tran_low_t *qcoeff_ptr, 201 tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr, 202 const SCAN_ORDER *sc, 203 const QUANT_PARAM *qparam); 204 205 #endif 206 207 #ifdef __cplusplus 208 } // extern "C" 209 #endif 210 211 #endif // AOM_AV1_ENCODER_AV1_QUANTIZE_H_ 212