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 AV1_COMMON_QUANT_COMMON_H_
13 #define AV1_COMMON_QUANT_COMMON_H_
14
15 #include "aom/aom_codec.h"
16 #include "av1/common/seg_common.h"
17 #include "av1/common/enums.h"
18 #include "av1/common/entropy.h"
19
20 #ifdef __cplusplus
21 extern "C" {
22 #endif
23
24 #define MINQ 0
25 #define MAXQ 255
26 #define QINDEX_RANGE (MAXQ - MINQ + 1)
27 #define QINDEX_BITS 8
28 #if CONFIG_AOM_QM
29 // Total number of QM sets stored
30 #define QM_LEVEL_BITS 4
31 #define NUM_QM_LEVELS (1 << QM_LEVEL_BITS)
32 /* Range of QMS is between first and last value, with offset applied to inter
33 * blocks*/
34 #define DEFAULT_QM_FIRST 5
35 #define DEFAULT_QM_LAST 9
36 #define DEFAULT_QM_INTER_OFFSET 0
37 #endif
38
39 struct AV1Common;
40
41 int16_t av1_dc_quant(int qindex, int delta, aom_bit_depth_t bit_depth);
42 int16_t av1_ac_quant(int qindex, int delta, aom_bit_depth_t bit_depth);
43 int16_t av1_qindex_from_ac(int ac, aom_bit_depth_t bit_depth);
44
45 int av1_get_qindex(const struct segmentation *seg, int segment_id,
46 int base_qindex);
47 #if CONFIG_AOM_QM
48 // Reduce the large number of quantizers to a smaller number of levels for which
49 // different matrices may be defined
aom_get_qmlevel(int qindex,int first,int last)50 static INLINE int aom_get_qmlevel(int qindex, int first, int last) {
51 return first + (qindex * (last + 1 - first)) / QINDEX_RANGE;
52 }
53 void aom_qm_init(struct AV1Common *cm);
54 const qm_val_t *aom_iqmatrix(struct AV1Common *cm, int qindex, int comp,
55 TX_SIZE tx_size, int is_intra);
56 const qm_val_t *aom_qmatrix(struct AV1Common *cm, int qindex, int comp,
57 TX_SIZE tx_size, int is_intra);
58 #endif
59
60 #if CONFIG_NEW_QUANT
61
62 #define QUANT_PROFILES 4
63 #define QUANT_RANGES 2
64 #define NUQ_KNOTS 3
65
66 typedef tran_low_t dequant_val_type_nuq[NUQ_KNOTS + 1];
67 typedef tran_low_t cuml_bins_type_nuq[NUQ_KNOTS];
68 void av1_get_dequant_val_nuq(int q, int band, tran_low_t *dq,
69 tran_low_t *cuml_bins, int dq_off_index);
70 tran_low_t av1_dequant_abscoeff_nuq(int v, int q, const tran_low_t *dq);
71 tran_low_t av1_dequant_coeff_nuq(int v, int q, const tran_low_t *dq);
72
qindex_to_qrange(int qindex)73 static INLINE int qindex_to_qrange(int qindex) {
74 return (qindex < 140 ? 1 : 0);
75 }
76
get_dq_profile_from_ctx(int qindex,int q_ctx,int is_inter,PLANE_TYPE plane_type)77 static INLINE int get_dq_profile_from_ctx(int qindex, int q_ctx, int is_inter,
78 PLANE_TYPE plane_type) {
79 // intra/inter, Y/UV, ctx, qrange
80 static const int
81 def_dq_profile_lookup[REF_TYPES][PLANE_TYPES][COEFF_CONTEXTS0]
82 [QUANT_RANGES] = {
83 {
84 // intra
85 { { 2, 1 }, { 2, 1 }, { 2, 1 } }, // Y
86 { { 3, 1 }, { 3, 1 }, { 3, 1 } }, // UV
87 },
88 {
89 // inter
90 { { 3, 1 }, { 2, 1 }, { 2, 1 } }, // Y
91 { { 3, 1 }, { 3, 1 }, { 3, 1 } }, // UV
92 },
93 };
94 if (!qindex) return 0; // lossless
95 return def_dq_profile_lookup[is_inter][plane_type][q_ctx]
96 [qindex_to_qrange(qindex)];
97 }
98 #endif // CONFIG_NEW_QUANT
99
100 #if CONFIG_PVQ
101 extern const int OD_QM8_Q4_FLAT[];
102 extern const int OD_QM8_Q4_HVS[];
103 #endif
104
105 #ifdef __cplusplus
106 } // extern "C"
107 #endif
108
109 #endif // AV1_COMMON_QUANT_COMMON_H_
110