| /* |
| * 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_ENCODEMB_H_ |
| #define AV1_ENCODER_ENCODEMB_H_ |
| |
| #include "config/aom_config.h" |
| |
| #include "av1/common/onyxc_int.h" |
| #include "av1/common/txb_common.h" |
| #include "av1/encoder/block.h" |
| #include "av1/encoder/tokenize.h" |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| struct optimize_ctx { |
| ENTROPY_CONTEXT ta[MAX_MB_PLANE][MAX_MIB_SIZE]; |
| ENTROPY_CONTEXT tl[MAX_MB_PLANE][MAX_MIB_SIZE]; |
| }; |
| |
| struct encode_b_args { |
| const struct AV1_COMP *cpi; |
| MACROBLOCK *x; |
| struct optimize_ctx *ctx; |
| int8_t *skip; |
| ENTROPY_CONTEXT *ta; |
| ENTROPY_CONTEXT *tl; |
| int8_t enable_optimize_b; |
| }; |
| |
| typedef enum AV1_XFORM_QUANT { |
| AV1_XFORM_QUANT_FP = 0, |
| AV1_XFORM_QUANT_B = 1, |
| AV1_XFORM_QUANT_DC = 2, |
| AV1_XFORM_QUANT_SKIP_QUANT, |
| AV1_XFORM_QUANT_TYPES, |
| } AV1_XFORM_QUANT; |
| |
| void av1_encode_sb(const struct AV1_COMP *cpi, MACROBLOCK *x, BLOCK_SIZE bsize, |
| int mi_row, int mi_col, RUN_TYPE dry_run); |
| void av1_encode_sby_pass1(AV1_COMMON *cm, MACROBLOCK *x, BLOCK_SIZE bsize); |
| void av1_xform_quant(const AV1_COMMON *cm, MACROBLOCK *x, int plane, int block, |
| int blk_row, int blk_col, BLOCK_SIZE plane_bsize, |
| TX_SIZE tx_size, TX_TYPE tx_type, |
| AV1_XFORM_QUANT xform_quant_idx); |
| |
| int av1_optimize_b(const struct AV1_COMP *cpi, MACROBLOCK *mb, int plane, |
| int block, TX_SIZE tx_size, TX_TYPE tx_type, |
| const TXB_CTX *const txb_ctx, int fast_mode, int *rate_cost); |
| |
| void av1_subtract_txb(MACROBLOCK *x, int plane, BLOCK_SIZE plane_bsize, |
| int blk_col, int blk_row, TX_SIZE tx_size); |
| |
| void av1_subtract_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane); |
| |
| static INLINE void av1_set_txb_context(MACROBLOCK *x, int plane, int block, |
| TX_SIZE tx_size, ENTROPY_CONTEXT *a, |
| ENTROPY_CONTEXT *l) { |
| const uint8_t ctx = x->plane[plane].txb_entropy_ctx[block]; |
| memset(a, ctx, tx_size_wide_unit[tx_size] * sizeof(*a)); |
| memset(l, ctx, tx_size_high_unit[tx_size] * sizeof(*l)); |
| } |
| |
| void av1_encode_block_intra(int plane, int block, int blk_row, int blk_col, |
| BLOCK_SIZE plane_bsize, TX_SIZE tx_size, void *arg); |
| |
| void av1_encode_intra_block_plane(const struct AV1_COMP *cpi, MACROBLOCK *x, |
| BLOCK_SIZE bsize, int plane, |
| int enable_optimize_b, int mi_row, |
| int mi_col); |
| |
| #ifdef __cplusplus |
| } // extern "C" |
| #endif |
| |
| #endif // AV1_ENCODER_ENCODEMB_H_ |