1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
|
/*
* Copyright (c) 2017, 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 AOM_AV1_ENCODER_ENCODETXB_H_
#define AOM_AV1_ENCODER_ENCODETXB_H_
#include "config/aom_config.h"
#include "av1/common/av1_common_int.h"
#include "av1/common/blockd.h"
#include "av1/common/txb_common.h"
#include "av1/encoder/block.h"
#include "av1/encoder/encoder.h"
#include "aom_dsp/bitwriter.h"
#ifdef __cplusplus
extern "C" {
#endif
/*!\cond */
#define TXB_SKIP_CTX_MASK 15
#define DC_SIGN_CTX_SHIFT 4
#define DC_SIGN_CTX_MASK 3
int av1_get_eob_pos_token(const int eob, int *const extra);
/*!\endcond */
/*!\brief Allocate the memory resources for all the macro blocks in the current
* coding frame.
* \ingroup coefficient_coding
*
* Each macro block will need a \ref CB_COEFF_BUFFER to store information for
* rate-distortion optimization and entropy coding of transform coefficients.
*
* \param[in] cpi Top-level encoder structure
*/
void av1_alloc_txb_buf(AV1_COMP *cpi);
/*!\brief Free the memory resources for all the macro blocks in the current
* coding frame.
* \ingroup coefficient_coding
*
* See \ref av1_alloc_txb_buf and \ref CB_COEFF_BUFFER for more details.
*
* \param[in] cpi Top-level encoder structure
*/
void av1_free_txb_buf(AV1_COMP *cpi);
/*!\brief Write quantized coefficients in a transform block into bitstream using
* entropy coding.
*
* \ingroup coefficient_coding
*
* This function will write the quantized coefficients in a transform block into
* the bitstream using entropy coding.
*
* The coding steps are as follows.
*
* 1) Code the end of block position "eob", which is the scan index of the
* last non-zero coefficient plus one.
*
* 2) Code the lower magnitude level (<= COEFF_BASE_RANGE + NUM_BASE_LEVELS)
* for each coefficient in reversed scan order.
*
* 3) Code the sign and higher magnitude level
* (> COEFF_BASE_RANGE + NUM_BASE_LEVELS) in forward scan order.
*
* \param[in] cm Top-level structure shared by encoder and
* decoder
* \param[in] x Pointer to structure holding the data for the
current encoding macroblock
* \param[in] w Entropy coding write pointer
* \param[in] blk_row The row index of the current transform block
* in the macroblock. Each unit has 4 pixels in y plane
* \param[in] blk_col The col index of the current transform block
* in the macroblock. Each unit has 4 pixels in y plane
* \param[in] plane The index of the current plane
* \param[in] block The index of the current transform block in the
* macroblock. It's defined by number of 4x4 units that have been coded before
* the currernt transform block
* \param[in] tx_size The given transform size
*/
void av1_write_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCK *const x,
aom_writer *w, int blk_row, int blk_col, int plane,
int block, TX_SIZE tx_size);
/*!\brief Write quantized coefficients of all transform blocks in an intra
* macroblock into the bitstream using entropy coding.
*
* \ingroup coefficient_coding
*
* All transform blocks in the intra macroblock share the same transform size.
*
* This function use \ref av1_write_coeffs_txb() to code each transform block in
* raster order.
*
* \param[in] cm Top-level structure shared by encoder and
* decoder
* \param[in] x Pointer to structure holding the data for the
current encoding macroblock
* \param[in] w Entropy coding write pointer
* \param[in] bsize Block size of the current macroblock
*/
void av1_write_intra_coeffs_mb(const AV1_COMMON *const cm, MACROBLOCK *x,
aom_writer *w, BLOCK_SIZE bsize);
/*!\brief Pack the context info of the current transform block into an uint8_t.
* \ingroup coefficient_coding
*
* This context info will be collected and consolidated by its neighbor
* transform blocks for coding transform block skip flag (tx_skip) and
* the sign of DC coefficient (dc_sign).
*
* \param[in] qcoeff Buffer of quantized coefficients
* \param[in] scan_order Coding order of coefficients in the transform
* block
* \param[in] eob The scan index of last non-zero coefficient plus
* one
*/
uint8_t av1_get_txb_entropy_context(const tran_low_t *qcoeff,
const SCAN_ORDER *scan_order, int eob);
/*!\brief Update the probability model (cdf) and the entropy context related to
* coefficient coding for all transform blocks in the intra macroblock.
*
* \ingroup coefficient_coding
*
* This function will go through each transform block in the intra macorblock
* and call \ref av1_update_and_record_txb_context to update the probability
* model and entropy context properly.
*
* \param[in] cpi Top-level encoder structure
* \param[in] td Top-level multithreading structure
* \param[in] dry_run Whether this is a dry run.
* \param[in] bsize Block size of the current macroblock
* \param[in] allow_update_cdf Allowed to update probability model (cdf) or
* not.
*/
void av1_update_intra_mb_txb_context(const AV1_COMP *cpi, ThreadData *td,
RUN_TYPE dry_run, BLOCK_SIZE bsize,
uint8_t allow_update_cdf);
/*!\brief Update the probability model (cdf) and the entropy context related to
* coefficient coding for a transform block.
*
* \ingroup coefficient_coding
*
* There are regular mode and dry run for this funtion.
*
* Regular mode:
*
* The probability model (cdf) for each coding symbol in the
* transform block will be updated.
*
* The entropy context of this transform block will be updated.
*
* Dry run:
*
* The probability model update will be skipped.
*
* The entropy context of this transform block will be updated.
*
* \param[in] plane The index of the current plane.
* \param[in] block The index of the current transform block in the
* macroblock. It's defined by number of 4x4 units that have been coded before
* the currernt transform block.
* \param[in] blk_row The row index of the current transform block
* in the macroblock. Each unit has 4 pixels in y plane.
* \param[in] blk_col The col index of the current transform block
* in the macroblock. Each unit has 4 pixels in y plane.
* \param[in] plane_bsize Block size for this plane. When the video source
* uses chroma subsampling, the block size of UV planes will be smaller than the
* block size of Y plane.
* \param[in] tx_size The given transform size.
* \param[in] arg This parameter will be translated into
* tokenize_b_args, in which RUN_TYPE indicates using regular mode or dry run.
*/
void av1_update_and_record_txb_context(int plane, int block, int blk_row,
int blk_col, BLOCK_SIZE plane_bsize,
TX_SIZE tx_size, void *arg);
/*!\brief Update the entropy context related to coefficient coding for a
* transform block.
*
* \ingroup coefficient_coding
*
* There are regular mode and dry run for this function.
*
* Regular mode:
*
* The entropy context of this transform block will be updated.
*
* Dry run:
*
* The probability model update will be skipped.
*
* The entropy context of this transform block will be updated.
*
* \param[in] plane The index of the current plane.
* \param[in] block The index of the current transform block in the
* macroblock. It's defined by number of 4x4 units that have been coded before
* the currernt transform block.
* \param[in] blk_row The row index of the current transform block
* in the macroblock. Each unit has 4 pixels in y plane.
* \param[in] blk_col The col index of the current transform block
* in the macroblock. Each unit has 4 pixels in y plane.
* \param[in] plane_bsize Block size for this plane. When the video source
* uses chroma subsampling, the block size of UV planes will be smaller than the
* block size of Y plane.
* \param[in] tx_size The given transform size.
* \param[in] arg This parameter will be translated into
* tokenize_b_args, in which RUN_TYPE indicates using regular mode or dry run.
*/
void av1_record_txb_context(int plane, int block, int blk_row, int blk_col,
BLOCK_SIZE plane_bsize, TX_SIZE tx_size, void *arg);
/*!\brief Get the corresponding \ref CB_COEFF_BUFFER of the current macro block.
*
* \ingroup coefficient_coding
*
* The macroblock's location is described by mi_row and mi_col, row and column
* mi indexes in the coding frame.
*
* Each mi unit is a 4x4 pixel block.
*
* \param[in] cpi Top-level encoder structure.
* \param[in] mi_row Row mi index of the current transform block
* in the frame.
* \param[in] mi_col Column mi index of the current transform
* block in the frame.
* \return CB_COEFF_BUFFER* Pointer of \ref CB_COEFF_BUFFER associated
* to this macroblock.
*/
CB_COEFF_BUFFER *av1_get_cb_coeff_buffer(const struct AV1_COMP *cpi, int mi_row,
int mi_col);
/*!\brief Returns the entropy cost associated with skipping the current
* transform block.
*
* \ingroup coefficient_coding
*
* \param[in] coeff_costs Table of entropy cost for coefficient coding.
* \param[in] txb_ctx Context info for entropy coding transform block
* skip flag (tx_skip) and the sign of DC coefficient (dc_sign).
* \param[in] plane The index of the current plane
* \param[in] tx_size The transform size
*/
static INLINE int av1_cost_skip_txb(const CoeffCosts *coeff_costs,
const TXB_CTX *const txb_ctx, int plane,
TX_SIZE tx_size) {
const TX_SIZE txs_ctx = get_txsize_entropy_ctx(tx_size);
const PLANE_TYPE plane_type = get_plane_type(plane);
const LV_MAP_COEFF_COST *const coeff_costs_ =
&coeff_costs->coeff_costs[txs_ctx][plane_type];
return coeff_costs_->txb_skip_cost[txb_ctx->txb_skip_ctx][1];
}
/*!\cond */
// These numbers are empirically obtained.
static const int plane_rd_mult[REF_TYPES][PLANE_TYPES] = {
{ 17, 13 },
{ 16, 10 },
};
/*!\endcond */
#ifdef __cplusplus
}
#endif
#endif // AOM_AV1_ENCODER_ENCODETXB_H_
|
