diff options
author | Matt A. Tobin <email@mattatobin.com> | 2020-04-07 23:30:51 -0400 |
---|---|---|
committer | wolfbeast <mcwerewolf@wolfbeast.com> | 2020-04-14 13:26:42 +0200 |
commit | 277f2116b6660e9bbe7f5d67524be57eceb49b8b (patch) | |
tree | 4595f7cc71418f71b9a97dfaeb03a30aa60f336a /third_party/aom/av1/encoder/tokenize.c | |
parent | d270404436f6e84ffa3b92af537ac721bf10d66e (diff) | |
download | UXP-277f2116b6660e9bbe7f5d67524be57eceb49b8b.tar UXP-277f2116b6660e9bbe7f5d67524be57eceb49b8b.tar.gz UXP-277f2116b6660e9bbe7f5d67524be57eceb49b8b.tar.lz UXP-277f2116b6660e9bbe7f5d67524be57eceb49b8b.tar.xz UXP-277f2116b6660e9bbe7f5d67524be57eceb49b8b.zip |
Move aom source to a sub-directory under media/libaom
There is no damned reason to treat this differently than any other media lib given its license and there never was.
Diffstat (limited to 'third_party/aom/av1/encoder/tokenize.c')
-rw-r--r-- | third_party/aom/av1/encoder/tokenize.c | 248 |
1 files changed, 0 insertions, 248 deletions
diff --git a/third_party/aom/av1/encoder/tokenize.c b/third_party/aom/av1/encoder/tokenize.c deleted file mode 100644 index 16a6a9a35..000000000 --- a/third_party/aom/av1/encoder/tokenize.c +++ /dev/null @@ -1,248 +0,0 @@ -/* - * 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. - */ - -#include <assert.h> -#include <math.h> -#include <stdio.h> -#include <string.h> - -#include "aom_mem/aom_mem.h" - -#include "av1/common/entropy.h" -#include "av1/common/pred_common.h" -#include "av1/common/scan.h" -#include "av1/common/seg_common.h" - -#include "av1/encoder/cost.h" -#include "av1/encoder/encoder.h" -#include "av1/encoder/encodetxb.h" -#include "av1/encoder/rdopt.h" -#include "av1/encoder/tokenize.h" - -static int cost_and_tokenize_map(Av1ColorMapParam *param, TOKENEXTRA **t, - int plane, int calc_rate, int allow_update_cdf, - FRAME_COUNTS *counts) { - const uint8_t *const color_map = param->color_map; - MapCdf map_cdf = param->map_cdf; - ColorCost color_cost = param->color_cost; - const int plane_block_width = param->plane_width; - const int rows = param->rows; - const int cols = param->cols; - const int n = param->n_colors; - const int palette_size_idx = n - PALETTE_MIN_SIZE; - int this_rate = 0; - uint8_t color_order[PALETTE_MAX_SIZE]; - - (void)plane; - (void)counts; - - for (int k = 1; k < rows + cols - 1; ++k) { - for (int j = AOMMIN(k, cols - 1); j >= AOMMAX(0, k - rows + 1); --j) { - int i = k - j; - int color_new_idx; - const int color_ctx = av1_get_palette_color_index_context( - color_map, plane_block_width, i, j, n, color_order, &color_new_idx); - assert(color_new_idx >= 0 && color_new_idx < n); - if (calc_rate) { - this_rate += (*color_cost)[palette_size_idx][color_ctx][color_new_idx]; - } else { - (*t)->token = color_new_idx; - (*t)->color_map_cdf = map_cdf[palette_size_idx][color_ctx]; - ++(*t); - if (allow_update_cdf) - update_cdf(map_cdf[palette_size_idx][color_ctx], color_new_idx, n); -#if CONFIG_ENTROPY_STATS - if (plane) { - ++counts->palette_uv_color_index[palette_size_idx][color_ctx] - [color_new_idx]; - } else { - ++counts->palette_y_color_index[palette_size_idx][color_ctx] - [color_new_idx]; - } -#endif - } - } - } - if (calc_rate) return this_rate; - return 0; -} - -static void get_palette_params(const MACROBLOCK *const x, int plane, - BLOCK_SIZE bsize, Av1ColorMapParam *params) { - const MACROBLOCKD *const xd = &x->e_mbd; - const MB_MODE_INFO *const mbmi = xd->mi[0]; - const PALETTE_MODE_INFO *const pmi = &mbmi->palette_mode_info; - params->color_map = xd->plane[plane].color_index_map; - params->map_cdf = plane ? xd->tile_ctx->palette_uv_color_index_cdf - : xd->tile_ctx->palette_y_color_index_cdf; - params->color_cost = - plane ? &x->palette_uv_color_cost : &x->palette_y_color_cost; - params->n_colors = pmi->palette_size[plane]; - av1_get_block_dimensions(bsize, plane, xd, ¶ms->plane_width, NULL, - ¶ms->rows, ¶ms->cols); -} - -static void get_color_map_params(const MACROBLOCK *const x, int plane, - BLOCK_SIZE bsize, TX_SIZE tx_size, - COLOR_MAP_TYPE type, - Av1ColorMapParam *params) { - (void)tx_size; - memset(params, 0, sizeof(*params)); - switch (type) { - case PALETTE_MAP: get_palette_params(x, plane, bsize, params); break; - default: assert(0 && "Invalid color map type"); return; - } -} - -int av1_cost_color_map(const MACROBLOCK *const x, int plane, BLOCK_SIZE bsize, - TX_SIZE tx_size, COLOR_MAP_TYPE type) { - assert(plane == 0 || plane == 1); - Av1ColorMapParam color_map_params; - get_color_map_params(x, plane, bsize, tx_size, type, &color_map_params); - return cost_and_tokenize_map(&color_map_params, NULL, plane, 1, 0, NULL); -} - -void av1_tokenize_color_map(const MACROBLOCK *const x, int plane, - TOKENEXTRA **t, BLOCK_SIZE bsize, TX_SIZE tx_size, - COLOR_MAP_TYPE type, int allow_update_cdf, - FRAME_COUNTS *counts) { - assert(plane == 0 || plane == 1); - Av1ColorMapParam color_map_params; - get_color_map_params(x, plane, bsize, tx_size, type, &color_map_params); - // The first color index does not use context or entropy. - (*t)->token = color_map_params.color_map[0]; - (*t)->color_map_cdf = NULL; - ++(*t); - cost_and_tokenize_map(&color_map_params, t, plane, 0, allow_update_cdf, - counts); -} - -void tokenize_vartx(ThreadData *td, TOKENEXTRA **t, RUN_TYPE dry_run, - TX_SIZE tx_size, BLOCK_SIZE plane_bsize, int blk_row, - int blk_col, int block, int plane, void *arg) { - MACROBLOCK *const x = &td->mb; - MACROBLOCKD *const xd = &x->e_mbd; - MB_MODE_INFO *const mbmi = xd->mi[0]; - const struct macroblockd_plane *const pd = &xd->plane[plane]; - const int max_blocks_high = max_block_high(xd, plane_bsize, plane); - const int max_blocks_wide = max_block_wide(xd, plane_bsize, plane); - - if (blk_row >= max_blocks_high || blk_col >= max_blocks_wide) return; - - const TX_SIZE plane_tx_size = - plane ? av1_get_max_uv_txsize(mbmi->sb_type, pd->subsampling_x, - pd->subsampling_y) - : mbmi->inter_tx_size[av1_get_txb_size_index(plane_bsize, blk_row, - blk_col)]; - - if (tx_size == plane_tx_size || plane) { - plane_bsize = get_plane_block_size(mbmi->sb_type, pd->subsampling_x, - pd->subsampling_y); - if (!dry_run) { - av1_update_and_record_txb_context(plane, block, blk_row, blk_col, - plane_bsize, tx_size, arg); - } else if (dry_run == DRY_RUN_NORMAL) { - av1_update_txb_context_b(plane, block, blk_row, blk_col, plane_bsize, - tx_size, arg); - } else { - printf("DRY_RUN_COSTCOEFFS is not supported yet\n"); - assert(0); - } - } else { - // Half the block size in transform block unit. - const TX_SIZE sub_txs = sub_tx_size_map[tx_size]; - const int bsw = tx_size_wide_unit[sub_txs]; - const int bsh = tx_size_high_unit[sub_txs]; - const int step = bsw * bsh; - - assert(bsw > 0 && bsh > 0); - - for (int row = 0; row < tx_size_high_unit[tx_size]; row += bsh) { - for (int col = 0; col < tx_size_wide_unit[tx_size]; col += bsw) { - const int offsetr = blk_row + row; - const int offsetc = blk_col + col; - - if (offsetr >= max_blocks_high || offsetc >= max_blocks_wide) continue; - - tokenize_vartx(td, t, dry_run, sub_txs, plane_bsize, offsetr, offsetc, - block, plane, arg); - block += step; - } - } - } -} - -void av1_tokenize_sb_vartx(const AV1_COMP *cpi, ThreadData *td, TOKENEXTRA **t, - RUN_TYPE dry_run, int mi_row, int mi_col, - BLOCK_SIZE bsize, int *rate, - uint8_t allow_update_cdf) { - const AV1_COMMON *const cm = &cpi->common; - const int num_planes = av1_num_planes(cm); - MACROBLOCK *const x = &td->mb; - MACROBLOCKD *const xd = &x->e_mbd; - MB_MODE_INFO *const mbmi = xd->mi[0]; - (void)t; - struct tokenize_b_args arg = { cpi, td, t, 0, allow_update_cdf }; - if (mi_row >= cm->mi_rows || mi_col >= cm->mi_cols) return; - - if (mbmi->skip) { - av1_reset_skip_context(xd, mi_row, mi_col, bsize, num_planes); - return; - } - - for (int plane = 0; plane < num_planes; ++plane) { - if (!is_chroma_reference(mi_row, mi_col, bsize, - xd->plane[plane].subsampling_x, - xd->plane[plane].subsampling_y)) { - continue; - } - const struct macroblockd_plane *const pd = &xd->plane[plane]; - const BLOCK_SIZE bsizec = - scale_chroma_bsize(bsize, pd->subsampling_x, pd->subsampling_y); - const BLOCK_SIZE plane_bsize = - get_plane_block_size(bsizec, pd->subsampling_x, pd->subsampling_y); - const int mi_width = block_size_wide[plane_bsize] >> tx_size_wide_log2[0]; - const int mi_height = block_size_high[plane_bsize] >> tx_size_high_log2[0]; - const TX_SIZE max_tx_size = get_vartx_max_txsize(xd, plane_bsize, plane); - const BLOCK_SIZE txb_size = txsize_to_bsize[max_tx_size]; - int bw = block_size_wide[txb_size] >> tx_size_wide_log2[0]; - int bh = block_size_high[txb_size] >> tx_size_high_log2[0]; - int idx, idy; - int block = 0; - int step = tx_size_wide_unit[max_tx_size] * tx_size_high_unit[max_tx_size]; - - const BLOCK_SIZE max_unit_bsize = - get_plane_block_size(BLOCK_64X64, pd->subsampling_x, pd->subsampling_y); - int mu_blocks_wide = - block_size_wide[max_unit_bsize] >> tx_size_wide_log2[0]; - int mu_blocks_high = - block_size_high[max_unit_bsize] >> tx_size_high_log2[0]; - - mu_blocks_wide = AOMMIN(mi_width, mu_blocks_wide); - mu_blocks_high = AOMMIN(mi_height, mu_blocks_high); - - for (idy = 0; idy < mi_height; idy += mu_blocks_high) { - for (idx = 0; idx < mi_width; idx += mu_blocks_wide) { - int blk_row, blk_col; - const int unit_height = AOMMIN(mu_blocks_high + idy, mi_height); - const int unit_width = AOMMIN(mu_blocks_wide + idx, mi_width); - for (blk_row = idy; blk_row < unit_height; blk_row += bh) { - for (blk_col = idx; blk_col < unit_width; blk_col += bw) { - tokenize_vartx(td, t, dry_run, max_tx_size, plane_bsize, blk_row, - blk_col, block, plane, &arg); - block += step; - } - } - } - } - } - if (rate) *rate += arg.this_rate; -} |