/src/aom/av1/encoder/encodemb.h
Line | Count | Source (jump to first uncovered line) |
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_ENCODEMB_H_ |
13 | | #define AOM_AV1_ENCODER_ENCODEMB_H_ |
14 | | |
15 | | #include "config/aom_config.h" |
16 | | |
17 | | #include "av1/common/av1_common_int.h" |
18 | | #include "av1/common/txb_common.h" |
19 | | #include "av1/encoder/av1_quantize.h" |
20 | | #include "av1/encoder/block.h" |
21 | | #include "av1/encoder/tokenize.h" |
22 | | #ifdef __cplusplus |
23 | | extern "C" { |
24 | | #endif |
25 | | |
26 | | enum { |
27 | | AV1_XFORM_QUANT_FP = 0, |
28 | | AV1_XFORM_QUANT_B = 1, |
29 | | AV1_XFORM_QUANT_DC = 2, |
30 | | AV1_XFORM_QUANT_SKIP_QUANT, |
31 | | AV1_XFORM_QUANT_TYPES, |
32 | | } UENUM1BYTE(AV1_XFORM_QUANT); |
33 | | |
34 | | // TODO(any): Merge OPT_TYPe and TRELLLIS_OPT_TYPE |
35 | | // Available optimization types to optimize the quantized coefficients. |
36 | | enum { |
37 | | NONE_OPT = 0, // No optimization. |
38 | | TRELLIS_OPT = 1, // Trellis optimization. See `av1_optimize_b()`. |
39 | | DROPOUT_OPT = 2, // Dropout optimization. See `av1_dropout_qcoeff()`. |
40 | | TRELLIS_DROPOUT_OPT = 3 // Perform dropout after trellis optimization. |
41 | | } UENUM1BYTE(OPT_TYPE); |
42 | | |
43 | | enum { |
44 | | NO_TRELLIS_OPT, // No trellis optimization |
45 | | FULL_TRELLIS_OPT, // Trellis optimization in all stages |
46 | | FINAL_PASS_TRELLIS_OPT, // Trellis optimization in only the final encode pass |
47 | | NO_ESTIMATE_YRD_TRELLIS_OPT // Disable trellis in estimate_yrd_for_sb |
48 | | } UENUM1BYTE(TRELLIS_OPT_TYPE); |
49 | | |
50 | | struct optimize_ctx { |
51 | | ENTROPY_CONTEXT ta[MAX_MB_PLANE][MAX_MIB_SIZE]; |
52 | | ENTROPY_CONTEXT tl[MAX_MB_PLANE][MAX_MIB_SIZE]; |
53 | | }; |
54 | | |
55 | | struct encode_b_args { |
56 | | const struct AV1_COMP *cpi; |
57 | | MACROBLOCK *x; |
58 | | struct optimize_ctx *ctx; |
59 | | ENTROPY_CONTEXT *ta; |
60 | | ENTROPY_CONTEXT *tl; |
61 | | RUN_TYPE dry_run; |
62 | | TRELLIS_OPT_TYPE enable_optimize_b; |
63 | | }; |
64 | | |
65 | | void av1_encode_sb(const struct AV1_COMP *cpi, MACROBLOCK *x, BLOCK_SIZE bsize, |
66 | | RUN_TYPE dry_run); |
67 | | |
68 | | void av1_foreach_transformed_block_in_plane( |
69 | | const MACROBLOCKD *const xd, BLOCK_SIZE plane_bsize, int plane, |
70 | | foreach_transformed_block_visitor visit, void *arg); |
71 | | |
72 | | void av1_encode_sby_pass1(struct AV1_COMP *cpi, MACROBLOCK *x, |
73 | | BLOCK_SIZE bsize); |
74 | | |
75 | | void av1_setup_xform(const AV1_COMMON *cm, MACROBLOCK *x, TX_SIZE tx_size, |
76 | | TX_TYPE tx_type, TxfmParam *txfm_param); |
77 | | void av1_setup_quant(TX_SIZE tx_size, int use_optimize_b, int xform_quant_idx, |
78 | | int use_quant_b_adapt, QUANT_PARAM *qparam); |
79 | | void av1_setup_qmatrix(const CommonQuantParams *quant_params, |
80 | | const MACROBLOCKD *xd, int plane, TX_SIZE tx_size, |
81 | | TX_TYPE tx_type, QUANT_PARAM *qparam); |
82 | | |
83 | | void av1_xform_dc_only(MACROBLOCK *x, int plane, int block, |
84 | | TxfmParam *txfm_param, int64_t per_px_mean); |
85 | | |
86 | | void av1_xform_quant(MACROBLOCK *x, int plane, int block, int blk_row, |
87 | | int blk_col, BLOCK_SIZE plane_bsize, TxfmParam *txfm_param, |
88 | | const QUANT_PARAM *qparam); |
89 | | |
90 | | void av1_xform(MACROBLOCK *x, int plane, int block, int blk_row, int blk_col, |
91 | | BLOCK_SIZE plane_bsize, TxfmParam *txfm_param); |
92 | | |
93 | | void av1_quant(MACROBLOCK *x, int plane, int block, TxfmParam *txfm_param, |
94 | | const QUANT_PARAM *qparam); |
95 | | |
96 | | int av1_optimize_b(const struct AV1_COMP *cpi, MACROBLOCK *mb, int plane, |
97 | | int block, TX_SIZE tx_size, TX_TYPE tx_type, |
98 | | const TXB_CTX *const txb_ctx, int *rate_cost); |
99 | | |
100 | | // This function can be used as (i) a further optimization to reduce the |
101 | | // redundancy of quantized coefficients (a.k.a., `qcoeff`) after trellis |
102 | | // optimization, or (ii) an alternative to trellis optimization in high-speed |
103 | | // compression mode (e.g., real-time mode under speed-6) due to its LOW time |
104 | | // complexity. The rational behind is to drop out the may-be redundant quantized |
105 | | // coefficient which is among a bunch of zeros. NOTE: This algorithm is not as |
106 | | // accurate as trellis optimization since the hyper-parameters are hard-coded |
107 | | // instead of dynamic search. More adaptive logic may improve the performance. |
108 | | // This function should be applied to all or partical block cells. |
109 | | // Inputs: |
110 | | // mb: Pointer to the MACROBLOCK to perform dropout on. |
111 | | // plane: Index of the plane to which the target block belongs. |
112 | | // block: Index of the target block. |
113 | | // tx_size: Transform size of the target block. |
114 | | // tx_type: Transform type of the target block. This field is particularly |
115 | | // used to find out the scan order of the block. |
116 | | // qindex: Quantization index used for target block. In general, all blocks |
117 | | // in a same plane share the same quantization index. This field is |
118 | | // particularly used to determine how many zeros should be used to |
119 | | // drop out a coefficient. |
120 | | // Returns: |
121 | | // Nothing will be returned, but `qcoeff`, `dqcoeff`, `eob`, as well as |
122 | | // `txb_entropy_ctx`, which `mb` points to, may be modified by this function. |
123 | | void av1_dropout_qcoeff(MACROBLOCK *mb, int plane, int block, TX_SIZE tx_size, |
124 | | TX_TYPE tx_type, int qindex); |
125 | | // Same as above, with the number of zeroes needed before/after a coeff to drop |
126 | | // it explicitly passed in, instead of being derived from qindex. |
127 | | void av1_dropout_qcoeff_num(MACROBLOCK *mb, int plane, int block, |
128 | | TX_SIZE tx_size, TX_TYPE tx_type, |
129 | | int dropout_num_before, int dropout_num_after); |
130 | | |
131 | | void av1_subtract_block(BitDepthInfo bd_info, int rows, int cols, int16_t *diff, |
132 | | ptrdiff_t diff_stride, const uint8_t *src8, |
133 | | ptrdiff_t src_stride, const uint8_t *pred8, |
134 | | ptrdiff_t pred_stride); |
135 | | |
136 | | void av1_subtract_txb(MACROBLOCK *x, int plane, BLOCK_SIZE plane_bsize, |
137 | | int blk_col, int blk_row, TX_SIZE tx_size); |
138 | | |
139 | | void av1_subtract_plane(MACROBLOCK *x, BLOCK_SIZE plane_bsize, int plane); |
140 | | |
141 | | static inline void av1_set_txb_context(MACROBLOCK *x, int plane, int block, |
142 | | TX_SIZE tx_size, ENTROPY_CONTEXT *a, |
143 | 0 | ENTROPY_CONTEXT *l) { |
144 | 0 | const uint8_t ctx = x->plane[plane].txb_entropy_ctx[block]; |
145 | 0 | memset(a, ctx, tx_size_wide_unit[tx_size] * sizeof(*a)); |
146 | 0 | memset(l, ctx, tx_size_high_unit[tx_size] * sizeof(*l)); |
147 | 0 | } Unexecuted instantiation: av1_cx_iface.c:av1_set_txb_context Unexecuted instantiation: allintra_vis.c:av1_set_txb_context Unexecuted instantiation: av1_quantize.c:av1_set_txb_context Unexecuted instantiation: bitstream.c:av1_set_txb_context Unexecuted instantiation: context_tree.c:av1_set_txb_context Unexecuted instantiation: encodeframe.c:av1_set_txb_context Unexecuted instantiation: encodeframe_utils.c:av1_set_txb_context Unexecuted instantiation: encodemb.c:av1_set_txb_context Unexecuted instantiation: encodemv.c:av1_set_txb_context Unexecuted instantiation: encoder.c:av1_set_txb_context Unexecuted instantiation: encoder_utils.c:av1_set_txb_context Unexecuted instantiation: encodetxb.c:av1_set_txb_context Unexecuted instantiation: ethread.c:av1_set_txb_context Unexecuted instantiation: firstpass.c:av1_set_txb_context Unexecuted instantiation: global_motion_facade.c:av1_set_txb_context Unexecuted instantiation: level.c:av1_set_txb_context Unexecuted instantiation: lookahead.c:av1_set_txb_context Unexecuted instantiation: mcomp.c:av1_set_txb_context Unexecuted instantiation: mv_prec.c:av1_set_txb_context Unexecuted instantiation: palette.c:av1_set_txb_context Unexecuted instantiation: partition_search.c:av1_set_txb_context Unexecuted instantiation: partition_strategy.c:av1_set_txb_context Unexecuted instantiation: pass2_strategy.c:av1_set_txb_context Unexecuted instantiation: pickcdef.c:av1_set_txb_context Unexecuted instantiation: picklpf.c:av1_set_txb_context Unexecuted instantiation: pickrst.c:av1_set_txb_context Unexecuted instantiation: ratectrl.c:av1_set_txb_context Unexecuted instantiation: rd.c:av1_set_txb_context Unexecuted instantiation: rdopt.c:av1_set_txb_context Unexecuted instantiation: nonrd_pickmode.c:av1_set_txb_context Unexecuted instantiation: nonrd_opt.c:av1_set_txb_context Unexecuted instantiation: segmentation.c:av1_set_txb_context Unexecuted instantiation: speed_features.c:av1_set_txb_context Unexecuted instantiation: superres_scale.c:av1_set_txb_context Unexecuted instantiation: svc_layercontext.c:av1_set_txb_context Unexecuted instantiation: temporal_filter.c:av1_set_txb_context Unexecuted instantiation: tokenize.c:av1_set_txb_context Unexecuted instantiation: tpl_model.c:av1_set_txb_context Unexecuted instantiation: tx_search.c:av1_set_txb_context Unexecuted instantiation: txb_rdopt.c:av1_set_txb_context Unexecuted instantiation: intra_mode_search.c:av1_set_txb_context Unexecuted instantiation: var_based_part.c:av1_set_txb_context Unexecuted instantiation: av1_noise_estimate.c:av1_set_txb_context Unexecuted instantiation: aq_complexity.c:av1_set_txb_context Unexecuted instantiation: aq_cyclicrefresh.c:av1_set_txb_context Unexecuted instantiation: aq_variance.c:av1_set_txb_context Unexecuted instantiation: compound_type.c:av1_set_txb_context Unexecuted instantiation: encode_strategy.c:av1_set_txb_context Unexecuted instantiation: global_motion.c:av1_set_txb_context Unexecuted instantiation: gop_structure.c:av1_set_txb_context Unexecuted instantiation: interp_search.c:av1_set_txb_context Unexecuted instantiation: motion_search_facade.c:av1_set_txb_context |
148 | | |
149 | | void av1_encode_intra_block_plane(const struct AV1_COMP *cpi, MACROBLOCK *x, |
150 | | BLOCK_SIZE bsize, int plane, RUN_TYPE dry_run, |
151 | | TRELLIS_OPT_TYPE enable_optimize_b); |
152 | | |
153 | | static inline int is_trellis_used(TRELLIS_OPT_TYPE optimize_b, |
154 | 0 | RUN_TYPE dry_run) { |
155 | 0 | if (optimize_b == NO_TRELLIS_OPT) return false; |
156 | 0 | if (optimize_b == FINAL_PASS_TRELLIS_OPT && dry_run != OUTPUT_ENABLED) |
157 | 0 | return false; |
158 | 0 | return true; |
159 | 0 | } Unexecuted instantiation: av1_cx_iface.c:is_trellis_used Unexecuted instantiation: allintra_vis.c:is_trellis_used Unexecuted instantiation: av1_quantize.c:is_trellis_used Unexecuted instantiation: bitstream.c:is_trellis_used Unexecuted instantiation: context_tree.c:is_trellis_used Unexecuted instantiation: encodeframe.c:is_trellis_used Unexecuted instantiation: encodeframe_utils.c:is_trellis_used Unexecuted instantiation: encodemb.c:is_trellis_used Unexecuted instantiation: encodemv.c:is_trellis_used Unexecuted instantiation: encoder.c:is_trellis_used Unexecuted instantiation: encoder_utils.c:is_trellis_used Unexecuted instantiation: encodetxb.c:is_trellis_used Unexecuted instantiation: ethread.c:is_trellis_used Unexecuted instantiation: firstpass.c:is_trellis_used Unexecuted instantiation: global_motion_facade.c:is_trellis_used Unexecuted instantiation: level.c:is_trellis_used Unexecuted instantiation: lookahead.c:is_trellis_used Unexecuted instantiation: mcomp.c:is_trellis_used Unexecuted instantiation: mv_prec.c:is_trellis_used Unexecuted instantiation: palette.c:is_trellis_used Unexecuted instantiation: partition_search.c:is_trellis_used Unexecuted instantiation: partition_strategy.c:is_trellis_used Unexecuted instantiation: pass2_strategy.c:is_trellis_used Unexecuted instantiation: pickcdef.c:is_trellis_used Unexecuted instantiation: picklpf.c:is_trellis_used Unexecuted instantiation: pickrst.c:is_trellis_used Unexecuted instantiation: ratectrl.c:is_trellis_used Unexecuted instantiation: rd.c:is_trellis_used Unexecuted instantiation: rdopt.c:is_trellis_used Unexecuted instantiation: nonrd_pickmode.c:is_trellis_used Unexecuted instantiation: nonrd_opt.c:is_trellis_used Unexecuted instantiation: segmentation.c:is_trellis_used Unexecuted instantiation: speed_features.c:is_trellis_used Unexecuted instantiation: superres_scale.c:is_trellis_used Unexecuted instantiation: svc_layercontext.c:is_trellis_used Unexecuted instantiation: temporal_filter.c:is_trellis_used Unexecuted instantiation: tokenize.c:is_trellis_used Unexecuted instantiation: tpl_model.c:is_trellis_used Unexecuted instantiation: tx_search.c:is_trellis_used Unexecuted instantiation: txb_rdopt.c:is_trellis_used Unexecuted instantiation: intra_mode_search.c:is_trellis_used Unexecuted instantiation: var_based_part.c:is_trellis_used Unexecuted instantiation: av1_noise_estimate.c:is_trellis_used Unexecuted instantiation: aq_complexity.c:is_trellis_used Unexecuted instantiation: aq_cyclicrefresh.c:is_trellis_used Unexecuted instantiation: aq_variance.c:is_trellis_used Unexecuted instantiation: compound_type.c:is_trellis_used Unexecuted instantiation: encode_strategy.c:is_trellis_used Unexecuted instantiation: global_motion.c:is_trellis_used Unexecuted instantiation: gop_structure.c:is_trellis_used Unexecuted instantiation: interp_search.c:is_trellis_used Unexecuted instantiation: motion_search_facade.c:is_trellis_used |
160 | | |
161 | | // Scaling terms (precision of 12 bits) to perform tx-size specific |
162 | | // normalization that is used in DCT_DCT forward transform. |
163 | | // For transform blocks of 1:2 and 2:1 - sqrt(2) normalization is used |
164 | | // For transform blocks of 1:4 and 4:1 - factor of 2 is used |
165 | | // For transform blocks TX_8x8 and below - an additional factor of 2 is used |
166 | | // For transform blocks max(width,height)=64 - currently not supported |
167 | | |
168 | | static const uint16_t dc_coeff_scale[TX_SIZES_ALL] = { |
169 | | 1024, 2048, 4096, 4096, 0, 1448, 1448, 2896, 2896, 2896, |
170 | | 2896, 0, 0, 2048, 2048, 4096, 4096, 0, 0 |
171 | | }; |
172 | | |
173 | | #ifdef __cplusplus |
174 | | } // extern "C" |
175 | | #endif |
176 | | |
177 | | #endif // AOM_AV1_ENCODER_ENCODEMB_H_ |