198 строки
8.5 KiB
C
198 строки
8.5 KiB
C
/*
|
|
* Copyright (c) 2016, Alliance for Open Media. All rights reserved
|
|
*
|
|
* This source code is subject to the terms of the BSD 2 Clause License and
|
|
* the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
|
|
* was not distributed with this source code in the LICENSE file, you can
|
|
* obtain it at www.aomedia.org/license/software. If the Alliance for Open
|
|
* Media Patent License 1.0 was not distributed with this source code in the
|
|
* PATENTS file, you can obtain it at www.aomedia.org/license/patent.
|
|
*/
|
|
|
|
#ifndef AV1_ENCODER_QUANTIZE_H_
|
|
#define AV1_ENCODER_QUANTIZE_H_
|
|
|
|
#include "./aom_config.h"
|
|
#include "av1/common/quant_common.h"
|
|
#include "av1/common/scan.h"
|
|
#include "av1/encoder/block.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
typedef struct QUANT_PARAM {
|
|
int log_scale;
|
|
#if CONFIG_NEW_QUANT
|
|
TX_SIZE tx_size;
|
|
int dq;
|
|
#endif // CONFIG_NEW_QUANT
|
|
#if CONFIG_AOM_QM
|
|
const qm_val_t *qmatrix;
|
|
const qm_val_t *iqmatrix;
|
|
#endif // CONFIG_AOM_QM
|
|
} QUANT_PARAM;
|
|
|
|
typedef void (*AV1_QUANT_FACADE)(const tran_low_t *coeff_ptr, intptr_t n_coeffs,
|
|
const MACROBLOCK_PLANE *p,
|
|
tran_low_t *qcoeff_ptr,
|
|
const MACROBLOCKD_PLANE *pd,
|
|
tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr,
|
|
const SCAN_ORDER *sc,
|
|
const QUANT_PARAM *qparam);
|
|
|
|
typedef struct {
|
|
#if CONFIG_NEW_QUANT
|
|
DECLARE_ALIGNED(
|
|
16, tran_low_t,
|
|
y_cuml_bins_nuq[QUANT_PROFILES][QINDEX_RANGE][COEF_BANDS][NUQ_KNOTS]);
|
|
DECLARE_ALIGNED(
|
|
16, tran_low_t,
|
|
uv_cuml_bins_nuq[QUANT_PROFILES][QINDEX_RANGE][COEF_BANDS][NUQ_KNOTS]);
|
|
#endif // CONFIG_NEW_QUANT
|
|
// 0: dc 1: ac 2-8: ac repeated to SIMD width
|
|
DECLARE_ALIGNED(16, int16_t, y_quant[QINDEX_RANGE][8]);
|
|
DECLARE_ALIGNED(16, int16_t, y_quant_shift[QINDEX_RANGE][8]);
|
|
DECLARE_ALIGNED(16, int16_t, y_zbin[QINDEX_RANGE][8]);
|
|
DECLARE_ALIGNED(16, int16_t, y_round[QINDEX_RANGE][8]);
|
|
|
|
// TODO(jingning): in progress of re-working the quantization. will decide
|
|
// if we want to deprecate the current use of y_quant.
|
|
DECLARE_ALIGNED(16, int16_t, y_quant_fp[QINDEX_RANGE][8]);
|
|
DECLARE_ALIGNED(16, int16_t, uv_quant_fp[QINDEX_RANGE][8]);
|
|
DECLARE_ALIGNED(16, int16_t, y_round_fp[QINDEX_RANGE][8]);
|
|
DECLARE_ALIGNED(16, int16_t, uv_round_fp[QINDEX_RANGE][8]);
|
|
|
|
DECLARE_ALIGNED(16, int16_t, uv_quant[QINDEX_RANGE][8]);
|
|
DECLARE_ALIGNED(16, int16_t, uv_quant_shift[QINDEX_RANGE][8]);
|
|
DECLARE_ALIGNED(16, int16_t, uv_zbin[QINDEX_RANGE][8]);
|
|
DECLARE_ALIGNED(16, int16_t, uv_round[QINDEX_RANGE][8]);
|
|
} QUANTS;
|
|
|
|
typedef struct {
|
|
DECLARE_ALIGNED(16, int16_t, y_dequant[QINDEX_RANGE][8]); // 8: SIMD width
|
|
DECLARE_ALIGNED(16, int16_t, uv_dequant[QINDEX_RANGE][8]); // 8: SIMD width
|
|
#if CONFIG_NEW_QUANT
|
|
DECLARE_ALIGNED(16, dequant_val_type_nuq,
|
|
y_dequant_val_nuq[QUANT_PROFILES][QINDEX_RANGE][COEF_BANDS]);
|
|
DECLARE_ALIGNED(16, dequant_val_type_nuq,
|
|
uv_dequant_val_nuq[QUANT_PROFILES][QINDEX_RANGE][COEF_BANDS]);
|
|
#endif // CONFIG_NEW_QUANT
|
|
} Dequants;
|
|
|
|
struct AV1_COMP;
|
|
struct AV1Common;
|
|
|
|
void av1_frame_init_quantizer(struct AV1_COMP *cpi);
|
|
|
|
void av1_init_plane_quantizers(const struct AV1_COMP *cpi, MACROBLOCK *x,
|
|
int segment_id);
|
|
|
|
void av1_build_quantizer(aom_bit_depth_t bit_depth, int y_dc_delta_q,
|
|
int uv_dc_delta_q, int uv_ac_delta_q,
|
|
QUANTS *const quants, Dequants *const deq);
|
|
|
|
void av1_init_quantizer(struct AV1_COMP *cpi);
|
|
|
|
void av1_set_quantizer(struct AV1Common *cm, int q);
|
|
|
|
int av1_quantizer_to_qindex(int quantizer);
|
|
|
|
int av1_qindex_to_quantizer(int qindex);
|
|
|
|
void av1_quantize_skip(intptr_t n_coeffs, tran_low_t *qcoeff_ptr,
|
|
tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr);
|
|
|
|
void av1_quantize_fp_facade(const tran_low_t *coeff_ptr, intptr_t n_coeffs,
|
|
const MACROBLOCK_PLANE *p, tran_low_t *qcoeff_ptr,
|
|
const MACROBLOCKD_PLANE *pd,
|
|
tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr,
|
|
const SCAN_ORDER *sc, const QUANT_PARAM *qparam);
|
|
|
|
void av1_quantize_b_facade(const tran_low_t *coeff_ptr, intptr_t n_coeffs,
|
|
const MACROBLOCK_PLANE *p, tran_low_t *qcoeff_ptr,
|
|
const MACROBLOCKD_PLANE *pd, tran_low_t *dqcoeff_ptr,
|
|
uint16_t *eob_ptr, const SCAN_ORDER *sc,
|
|
const QUANT_PARAM *qparam);
|
|
|
|
void av1_quantize_dc_facade(const tran_low_t *coeff_ptr, intptr_t n_coeffs,
|
|
const MACROBLOCK_PLANE *p, tran_low_t *qcoeff_ptr,
|
|
const MACROBLOCKD_PLANE *pd,
|
|
tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr,
|
|
const SCAN_ORDER *sc, const QUANT_PARAM *qparam);
|
|
|
|
#if CONFIG_NEW_QUANT
|
|
void av1_quantize_fp_nuq_facade(const tran_low_t *coeff_ptr, intptr_t n_coeffs,
|
|
const MACROBLOCK_PLANE *p,
|
|
tran_low_t *qcoeff_ptr,
|
|
const MACROBLOCKD_PLANE *pd,
|
|
tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr,
|
|
const SCAN_ORDER *sc,
|
|
const QUANT_PARAM *qparam);
|
|
|
|
void av1_quantize_b_nuq_facade(const tran_low_t *coeff_ptr, intptr_t n_coeffs,
|
|
const MACROBLOCK_PLANE *p,
|
|
tran_low_t *qcoeff_ptr,
|
|
const MACROBLOCKD_PLANE *pd,
|
|
tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr,
|
|
const SCAN_ORDER *sc, const QUANT_PARAM *qparam);
|
|
|
|
void av1_quantize_dc_nuq_facade(const tran_low_t *coeff_ptr, intptr_t n_coeffs,
|
|
const MACROBLOCK_PLANE *p,
|
|
tran_low_t *qcoeff_ptr,
|
|
const MACROBLOCKD_PLANE *pd,
|
|
tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr,
|
|
const SCAN_ORDER *sc,
|
|
const QUANT_PARAM *qparam);
|
|
#endif // CONFIG_NEW_QUANT
|
|
|
|
void av1_highbd_quantize_fp_facade(const tran_low_t *coeff_ptr,
|
|
intptr_t n_coeffs, const MACROBLOCK_PLANE *p,
|
|
tran_low_t *qcoeff_ptr,
|
|
const MACROBLOCKD_PLANE *pd,
|
|
tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr,
|
|
const SCAN_ORDER *sc,
|
|
const QUANT_PARAM *qparam);
|
|
|
|
void av1_highbd_quantize_b_facade(const tran_low_t *coeff_ptr,
|
|
intptr_t n_coeffs, const MACROBLOCK_PLANE *p,
|
|
tran_low_t *qcoeff_ptr,
|
|
const MACROBLOCKD_PLANE *pd,
|
|
tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr,
|
|
const SCAN_ORDER *sc,
|
|
const QUANT_PARAM *qparam);
|
|
|
|
void av1_highbd_quantize_dc_facade(const tran_low_t *coeff_ptr,
|
|
intptr_t n_coeffs, const MACROBLOCK_PLANE *p,
|
|
tran_low_t *qcoeff_ptr,
|
|
const MACROBLOCKD_PLANE *pd,
|
|
tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr,
|
|
const SCAN_ORDER *sc,
|
|
const QUANT_PARAM *qparam);
|
|
|
|
#if CONFIG_NEW_QUANT
|
|
void av1_highbd_quantize_fp_nuq_facade(
|
|
const tran_low_t *coeff_ptr, intptr_t n_coeffs, const MACROBLOCK_PLANE *p,
|
|
tran_low_t *qcoeff_ptr, const MACROBLOCKD_PLANE *pd,
|
|
tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr, const SCAN_ORDER *sc,
|
|
const QUANT_PARAM *qparam);
|
|
|
|
void av1_highbd_quantize_b_nuq_facade(
|
|
const tran_low_t *coeff_ptr, intptr_t n_coeffs, const MACROBLOCK_PLANE *p,
|
|
tran_low_t *qcoeff_ptr, const MACROBLOCKD_PLANE *pd,
|
|
tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr, const SCAN_ORDER *sc,
|
|
const QUANT_PARAM *qparam);
|
|
|
|
void av1_highbd_quantize_dc_nuq_facade(
|
|
const tran_low_t *coeff_ptr, intptr_t n_coeffs, const MACROBLOCK_PLANE *p,
|
|
tran_low_t *qcoeff_ptr, const MACROBLOCKD_PLANE *pd,
|
|
tran_low_t *dqcoeff_ptr, uint16_t *eob_ptr, const SCAN_ORDER *sc,
|
|
const QUANT_PARAM *qparam);
|
|
#endif // CONFIG_NEW_QUANT
|
|
|
|
#ifdef __cplusplus
|
|
} // extern "C"
|
|
#endif
|
|
|
|
#endif // AV1_ENCODER_QUANTIZE_H_
|