partition_search.h (4050B)
1 /* 2 * Copyright (c) 2020, 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_PARTITION_SEARCH_H_ 13 #define AOM_AV1_ENCODER_PARTITION_SEARCH_H_ 14 15 #include "config/aom_config.h" 16 17 #include "av1/encoder/block.h" 18 #include "av1/encoder/encoder.h" 19 #include "av1/encoder/encodeframe.h" 20 #include "av1/encoder/tokenize.h" 21 22 void av1_set_offsets_without_segment_id(const AV1_COMP *const cpi, 23 const TileInfo *const tile, 24 MACROBLOCK *const x, int mi_row, 25 int mi_col, BLOCK_SIZE bsize); 26 void av1_set_offsets(const AV1_COMP *const cpi, const TileInfo *const tile, 27 MACROBLOCK *const x, int mi_row, int mi_col, 28 BLOCK_SIZE bsize); 29 void av1_rd_use_partition(AV1_COMP *cpi, ThreadData *td, TileDataEnc *tile_data, 30 MB_MODE_INFO **mib, TokenExtra **tp, int mi_row, 31 int mi_col, BLOCK_SIZE bsize, int *rate, 32 int64_t *dist, int do_recon, PC_TREE *pc_tree); 33 void av1_nonrd_use_partition(AV1_COMP *cpi, ThreadData *td, 34 TileDataEnc *tile_data, MB_MODE_INFO **mib, 35 TokenExtra **tp, int mi_row, int mi_col, 36 BLOCK_SIZE bsize, PC_TREE *pc_tree); 37 #if CONFIG_RT_ML_PARTITIONING 38 void av1_nonrd_pick_partition(AV1_COMP *cpi, ThreadData *td, 39 TileDataEnc *tile_data, TokenExtra **tp, 40 int mi_row, int mi_col, BLOCK_SIZE bsize, 41 RD_STATS *rd_cost, int do_recon, int64_t best_rd, 42 PC_TREE *pc_tree); 43 #endif 44 45 #if CONFIG_PARTITION_SEARCH_ORDER 46 void av1_reset_part_sf(PARTITION_SPEED_FEATURES *part_sf); 47 void av1_reset_sf_for_ext_part(AV1_COMP *const cpi); 48 bool av1_rd_partition_search(AV1_COMP *const cpi, ThreadData *td, 49 TileDataEnc *tile_data, TokenExtra **tp, 50 SIMPLE_MOTION_DATA_TREE *sms_root, int mi_row, 51 int mi_col, BLOCK_SIZE bsize, 52 RD_STATS *best_rd_cost); 53 #endif 54 55 bool av1_rd_pick_partition(AV1_COMP *const cpi, ThreadData *td, 56 TileDataEnc *tile_data, TokenExtra **tp, int mi_row, 57 int mi_col, BLOCK_SIZE bsize, RD_STATS *rd_cost, 58 RD_STATS best_rdc, PC_TREE *pc_tree, 59 SIMPLE_MOTION_DATA_TREE *sms_tree, int64_t *none_rd, 60 SB_MULTI_PASS_MODE multi_pass_mode, 61 RD_RECT_PART_WIN_INFO *rect_part_win_info); 62 63 static inline void set_cb_offsets(uint16_t *cb_offset, 64 const uint16_t cb_offset_y, 65 const uint16_t cb_offset_uv) { 66 cb_offset[PLANE_TYPE_Y] = cb_offset_y; 67 cb_offset[PLANE_TYPE_UV] = cb_offset_uv; 68 } 69 70 static inline void update_cb_offsets(MACROBLOCK *x, const BLOCK_SIZE bsize, 71 const int subsampling_x, 72 const int subsampling_y) { 73 x->cb_offset[PLANE_TYPE_Y] += block_size_wide[bsize] * block_size_high[bsize]; 74 if (x->e_mbd.is_chroma_ref) { 75 const BLOCK_SIZE plane_bsize = 76 get_plane_block_size(bsize, subsampling_x, subsampling_y); 77 assert(plane_bsize != BLOCK_INVALID); 78 x->cb_offset[PLANE_TYPE_UV] += 79 block_size_wide[plane_bsize] * block_size_high[plane_bsize]; 80 } 81 } 82 83 #endif // AOM_AV1_ENCODER_PARTITION_SEARCH_H_