| /* |
| * Copyright (c) 2021, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 3-Clause Clear License |
| * and the Alliance for Open Media Patent License 1.0. If the BSD 3-Clause Clear |
| * License was not distributed with this source code in the LICENSE file, you |
| * can obtain it at aomedia.org/license/software-license/bsd-3-c-c/. 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 |
| * aomedia.org/license/patent-license/. |
| */ |
| |
| /*!\file |
| * \brief Declares functions used in palette search. |
| */ |
| #ifndef AOM_AV1_ENCODER_PALETTE_H_ |
| #define AOM_AV1_ENCODER_PALETTE_H_ |
| |
| #include "av1/common/blockd.h" |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| struct AV1_COMP; |
| struct PICK_MODE_CONTEXT; |
| struct macroblock; |
| |
| /*!\cond */ |
| #define AV1_K_MEANS_RENAME(func, dim) func##_dim##dim |
| |
| void AV1_K_MEANS_RENAME(av1_calc_indices, 1)(const int *data, |
| const int *centroids, |
| uint8_t *indices, int n, int k); |
| void AV1_K_MEANS_RENAME(av1_calc_indices, 2)(const int *data, |
| const int *centroids, |
| uint8_t *indices, int n, int k); |
| void AV1_K_MEANS_RENAME(av1_k_means, 1)(const int *data, int *centroids, |
| uint8_t *indices, int n, int k, |
| int max_itr); |
| void AV1_K_MEANS_RENAME(av1_k_means, 2)(const int *data, int *centroids, |
| uint8_t *indices, int n, int k, |
| int max_itr); |
| /*!\endcond */ |
| |
| /*!\brief Extended Palette Coding Flags */ |
| typedef enum { |
| /*!\brief No extra palette flags. */ |
| PALETTE_NO_FLAGS, |
| |
| /*!\brief Identity row flag, all values are the same as first value*/ |
| PALETTE_IDENTIY_ROW, |
| |
| /*!\brief Line Copy flag, all values are the same as above line */ |
| PALETTE_LINE_COPY |
| } palette_flag_t; |
| |
| /*!\brief Calculates the cluster to which each data point belong. |
| * |
| * \ingroup palette_mode_search |
| * \param[in] data The data points whose cluster indices are |
| * to be computed. The data layout is |
| * NUM_DATA_POINTS X DATA_DIM. |
| * \param[in] centroids Pointer to the centroids. The data layout |
| * is NUM_CENTROIDS X DATA_DIM. |
| * \param[in] indices Pointer to store the computed indices. |
| * \param[in] n Number of data points. |
| * \param[in] k Number of clusters. |
| * \param[in] dim Data dimension. |
| * |
| * Returns nothing, but saves each data's cluster index in indices. |
| */ |
| static INLINE void av1_calc_indices(const int *data, const int *centroids, |
| uint8_t *indices, int n, int k, int dim) { |
| assert(n > 0); |
| assert(k > 0); |
| if (dim == 1) { |
| AV1_K_MEANS_RENAME(av1_calc_indices, 1)(data, centroids, indices, n, k); |
| } else if (dim == 2) { |
| AV1_K_MEANS_RENAME(av1_calc_indices, 2)(data, centroids, indices, n, k); |
| } else { |
| assert(0 && "Untemplated k means dimension"); |
| } |
| } |
| |
| /*!\brief Performs k-means cluster on the data. |
| * |
| * \ingroup palette_mode_search |
| * \param[in] data The data points to be clustered. The data |
| * layout is NUM_DATA_POINTS X DATA_DIM. |
| * \param[in] centroids Pointer to store the computed centroids. |
| * The data layout is |
| * NUM_CENTROIDS X DATA_DIM. |
| * \param[in] indices Pointer to store the computed indices. For |
| * each training data. |
| * \param[in] n Number of data points. |
| * \param[in] k Number of clusters. |
| * \param[in] dim Data dimension. |
| * \param[in] max_itr Maximum number of iterations to run. |
| * |
| * Returns nothing, but saves each cluster's centroid in centroids and |
| * each data's cluster index in indices. |
| * |
| * \attention The output centroids are rounded off to nearest integers. |
| */ |
| static INLINE void av1_k_means(const int *data, int *centroids, |
| uint8_t *indices, int n, int k, int dim, |
| int max_itr) { |
| assert(n > 0); |
| assert(k > 0); |
| if (dim == 1) { |
| AV1_K_MEANS_RENAME(av1_k_means, 1)(data, centroids, indices, n, k, max_itr); |
| } else if (dim == 2) { |
| AV1_K_MEANS_RENAME(av1_k_means, 2)(data, centroids, indices, n, k, max_itr); |
| } else { |
| assert(0 && "Untemplated k means dimension"); |
| } |
| } |
| |
| /*!\brief Removes duplicated centroid indices. |
| * |
| * \ingroup palette_mode_search |
| * \param[in] centroids A list of centroids index. |
| * \param[in] num_centroids Number of centroids. |
| * |
| * \return Returns the number of unique centroids and saves the unique centroids |
| * in beginning of the centroids array. |
| * |
| * \attention The centroids should be rounded to integers before calling this |
| * method. |
| */ |
| int av1_remove_duplicates(int *centroids, int num_centroids); |
| |
| /*!\brief Checks what colors are in the color cache. |
| * |
| * \ingroup palette_mode_search |
| * \param[in] color_cache A cache of colors. |
| * \param[in] n_cache Number of colors in the cache. |
| * \param[in] colors New base colors. |
| * \param[in] n_colors Number of new colors. |
| * \param[in] cache_color_found Stores what cached colors are presented in |
| * colors. |
| * \param[in] out_cache_colors Stores what colors are not in the cache. |
| * |
| * \return Returns the number of colors that are not in cache. In addition, |
| * records whether each cache color is presented in colors in cache_color_found, |
| * and stores and stores the out of cache colors in out_cache_colors. |
| */ |
| int av1_index_color_cache(const uint16_t *color_cache, int n_cache, |
| const uint16_t *colors, int n_colors, |
| uint8_t *cache_color_found, int *out_cache_colors); |
| |
| /*!\brief Gets the rate cost for each delta-encoding v palette. |
| * |
| * \ingroup palette_mode_search |
| * \param[in] pmi Struct that stores the palette mode info. |
| * \param[in] bit_depth Pixel bitdepth of the sequence. |
| * \param[in] zero_count Stores the number of zero deltas. |
| * \param[in] min_bits Minimum bits for the deltas. Sets to |
| * bit_depth - 4. |
| * |
| * \return Returns the number of bits used to transmit each v palette color |
| * delta and assigns zero_count with the number of deltas being 0. |
| */ |
| int av1_get_palette_delta_bits_v(const PALETTE_MODE_INFO *const pmi, |
| int bit_depth, int *zero_count, int *min_bits); |
| |
| /*!\brief Gets the rate cost for transmitting luma palette color values. |
| * |
| * \ingroup palette_mode_search |
| * \param[in] pmi Struct that stores the palette mode info. |
| * \param[in] color_cache Color cache presented at the decoder. |
| * \param[in] n_cache Number of colors in the cache. |
| * \param[in] bit_depth Pixel bitdepth of the sequence. |
| * |
| * \return Returns the rate needed to transmit the palette. Note that this does |
| * not include the cost of transmitted the color map. |
| */ |
| int av1_palette_color_cost_y(const PALETTE_MODE_INFO *const pmi, |
| const uint16_t *color_cache, int n_cache, |
| int bit_depth); |
| |
| /*!\brief Gets the rate cost for transmitting luma palette chroma values. |
| * |
| * \ingroup palette_mode_search |
| * \param[in] pmi Struct that stores the palette mode info. |
| * \param[in] color_cache Color cache presented at the decoder. |
| * \param[in] n_cache Number of colors in the cache. |
| * \param[in] bit_depth Pixel bitdepth of the sequence. |
| * |
| * \return Returns the rate needed to transmit the palette. Note that this does |
| * not include the cost of transmitted the color map. |
| */ |
| int av1_palette_color_cost_uv(const PALETTE_MODE_INFO *const pmi, |
| const uint16_t *color_cache, int n_cache, |
| int bit_depth); |
| |
| /*!\brief Search for the best palette in the luma plane. |
| * |
| * \ingroup palette_mode_search |
| * \callergraph |
| * This function is used in both inter and intra frame coding. |
| */ |
| void av1_rd_pick_palette_intra_sby( |
| const struct AV1_COMP *cpi, struct macroblock *x, BLOCK_SIZE bsize, |
| int dc_mode_cost, MB_MODE_INFO *best_mbmi, uint8_t *best_palette_color_map, |
| int64_t *best_rd, int64_t *best_model_rd, int *rate, int *rate_tokenonly, |
| int64_t *distortion, int *skippable, int *beat_best_rd, |
| struct PICK_MODE_CONTEXT *ctx, uint8_t *best_blk_skip, |
| TX_TYPE *tx_type_map); |
| |
| /*!\brief Search for the best palette in the chroma plane. |
| * |
| * \ingroup palette_mode_search |
| * \callergraph |
| * This function is used in both inter and intra frame coding. |
| */ |
| void av1_rd_pick_palette_intra_sbuv(const struct AV1_COMP *cpi, |
| struct macroblock *x, int dc_mode_cost, |
| uint8_t *best_palette_color_map, |
| MB_MODE_INFO *const best_mbmi, |
| int64_t *best_rd, int *rate, |
| int *rate_tokenonly, int64_t *distortion, |
| int *skippable); |
| |
| /*!\brief Resets palette color map for chroma channels. |
| */ |
| void av1_restore_uv_color_map(const struct AV1_COMP *cpi, struct macroblock *x); |
| |
| #ifdef __cplusplus |
| } // extern "C" |
| #endif |
| |
| #endif // AOM_AV1_ENCODER_PALETTE_H_ |