quant_common.h 3.76 KB
Newer Older
Jingning Han's avatar
Jingning Han committed
1
/*
Yaowu Xu's avatar
Yaowu Xu committed
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
Jingning Han's avatar
Jingning Han committed
3
 *
Yaowu Xu's avatar
Yaowu Xu committed
4
5
6
7
8
9
 * 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.
Jingning Han's avatar
Jingning Han committed
10
11
 */

Yaowu Xu's avatar
Yaowu Xu committed
12
13
#ifndef AV1_COMMON_QUANT_COMMON_H_
#define AV1_COMMON_QUANT_COMMON_H_
Jingning Han's avatar
Jingning Han committed
14

Yaowu Xu's avatar
Yaowu Xu committed
15
#include "aom/aom_codec.h"
16
17
#include "av1/common/seg_common.h"
#include "av1/common/enums.h"
18
#include "av1/common/entropy.h"
Jingning Han's avatar
Jingning Han committed
19
20
21
22
23
24
25
26
27

#ifdef __cplusplus
extern "C" {
#endif

#define MINQ 0
#define MAXQ 255
#define QINDEX_RANGE (MAXQ - MINQ + 1)
#define QINDEX_BITS 8
28
29
30
31
#if CONFIG_AOM_QM
// Total number of QM sets stored
#define QM_LEVEL_BITS 4
#define NUM_QM_LEVELS (1 << QM_LEVEL_BITS)
32
33
34
35
/* Range of QMS is between first and last value, with offset applied to inter
 * blocks*/
#define DEFAULT_QM_FIRST 5
#define DEFAULT_QM_LAST 9
36
37
#endif

Yaowu Xu's avatar
Yaowu Xu committed
38
struct AV1Common;
Jingning Han's avatar
Jingning Han committed
39

40
41
42
43
44
int16_t av1_dc_quant_Q3(int qindex, int delta, aom_bit_depth_t bit_depth);
int16_t av1_ac_quant_Q3(int qindex, int delta, aom_bit_depth_t bit_depth);
int16_t av1_dc_quant_QTX(int qindex, int delta, aom_bit_depth_t bit_depth);
int16_t av1_ac_quant_QTX(int qindex, int delta, aom_bit_depth_t bit_depth);
int16_t av1_qindex_from_ac_Q3(int ac_Q3, aom_bit_depth_t bit_depth);
Jingning Han's avatar
Jingning Han committed
45

Yaowu Xu's avatar
Yaowu Xu committed
46
47
int av1_get_qindex(const struct segmentation *seg, int segment_id,
                   int base_qindex);
48
49
50
#if CONFIG_AOM_QM
// Reduce the large number of quantizers to a smaller number of levels for which
// different matrices may be defined
51
static INLINE int aom_get_qmlevel(int qindex, int first, int last) {
David Barker's avatar
David Barker committed
52
  return first + (qindex * (last + 1 - first)) / QINDEX_RANGE;
53
}
Yaowu Xu's avatar
Yaowu Xu committed
54
55
void aom_qm_init(struct AV1Common *cm);
qm_val_t *aom_iqmatrix(struct AV1Common *cm, int qindex, int comp,
56
                       TX_SIZE tx_size);
Yaowu Xu's avatar
Yaowu Xu committed
57
qm_val_t *aom_qmatrix(struct AV1Common *cm, int qindex, int comp,
58
                      TX_SIZE tx_size);
59
#endif  // CONFIG_AOM_QM
Jingning Han's avatar
Jingning Han committed
60

61
#if CONFIG_NEW_QUANT
62

63
#define QUANT_PROFILES ((DQ_TYPES - 1) * 8 + 1)
clang-format's avatar
clang-format committed
64
#define QUANT_RANGES 2
65
#define NUQ_KNOTS 1
66

67
typedef tran_low_t dequant_val_type_nuq[NUQ_KNOTS];
68
typedef tran_low_t cuml_bins_type_nuq[NUQ_KNOTS];
69
void av1_get_dequant_val_nuq(int q, int is_ac_coeff, tran_low_t *dq,
Yaowu Xu's avatar
Yaowu Xu committed
70
71
72
                             tran_low_t *cuml_bins, int dq_off_index);
tran_low_t av1_dequant_abscoeff_nuq(int v, int q, const tran_low_t *dq);
tran_low_t av1_dequant_coeff_nuq(int v, int q, const tran_low_t *dq);
73

74
75
76
77
static INLINE int qindex_to_qrange(int qindex) {
  return (qindex < 140 ? 1 : 0);
}

78
static INLINE int get_dq_profile(DqType dqtype, int qindex, int is_inter,
79
                                 PLANE_TYPE plane_type) {
80
  // intra/inter, Y/UV, ctx, qrange
81
82
83
  static const int
      dq_profile_lookup[DQ_TYPES][REF_TYPES][PLANE_TYPES][QUANT_RANGES] = {
        {
84
85
            { { 0, 0 }, { 0, 0 } },  // intra: Y, UV
            { { 0, 0 }, { 0, 0 } },  // inter: Y, UV
86
87
        },
        {
88
89
90
91
92
93
94
95
96
97
            { { 1, 2 }, { 3, 4 } },  // intra: Y, UV
            { { 5, 6 }, { 7, 8 } },  // inter: Y, UV
        },
        {
            { { 9, 10 }, { 11, 12 } },   // intra: Y, UV
            { { 13, 14 }, { 15, 16 } },  // inter: Y, UV
        },
        {
            { { 17, 18 }, { 19, 20 } },  // intra: Y, UV
            { { 21, 22 }, { 23, 24 } },  // inter: Y, UV
98
99
        },
      };
100
  if (!qindex) return 0;  // lossless
101
  if (!dqtype) return 0;  // DQ_MULT
102
103
  return dq_profile_lookup[dqtype][is_inter][plane_type]
                          [qindex_to_qrange(qindex)];
104
}
105
106
#endif  // CONFIG_NEW_QUANT

Jingning Han's avatar
Jingning Han committed
107
108
109
110
#ifdef __cplusplus
}  // extern "C"
#endif

Yaowu Xu's avatar
Yaowu Xu committed
111
#endif  // AV1_COMMON_QUANT_COMMON_H_