| AOMedia AV1 Codec

palette.h
Go to the documentation of this file.
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
15#ifndef AOM_AV1_ENCODER_PALETTE_H_
16#define AOM_AV1_ENCODER_PALETTE_H_
17
18#include "av1/common/blockd.h"
19
20#ifdef __cplusplus
21 extern "C" {
22#endif
23
24 struct AV1_COMP;
25 struct PICK_MODE_CONTEXT;
26 struct macroblock;
27
29#define AV1_K_MEANS_RENAME(func, dim) func##_dim##dim
30
31 void AV1_K_MEANS_RENAME(av1_k_means, 1)(const int16_t *data, int16_t *centroids,
32 uint8_t *indices, int n, int k,
33 int max_itr);
34 void AV1_K_MEANS_RENAME(av1_k_means, 2)(const int16_t *data, int16_t *centroids,
35 uint8_t *indices, int n, int k,
36 int max_itr);
54 static inline void av1_calc_indices(const int16_t *data,
55 const int16_t *centroids, uint8_t *indices,
56 int n, int k, int dim) {
57 assert(n > 0);
58 assert(k > 0);
59 if (dim == 1) {
60 av1_calc_indices_dim1(data, centroids, indices, /*total_dist=*/NULL, n, k);
61 } else if (dim == 2) {
62 av1_calc_indices_dim2(data, centroids, indices, /*total_dist=*/NULL, n, k);
63 } else {
64 assert(0 && "Untemplated k means dimension");
65 }
66}
67
88 static inline void av1_k_means(const int16_t *data, int16_t *centroids,
89 uint8_t *indices, int n, int k, int dim,
90 int max_itr) {
91 assert(n > 0);
92 assert(k > 0);
93 if (dim == 1) {
94 AV1_K_MEANS_RENAME(av1_k_means, 1)(data, centroids, indices, n, k, max_itr);
95 } else if (dim == 2) {
96 AV1_K_MEANS_RENAME(av1_k_means, 2)(data, centroids, indices, n, k, max_itr);
97 } else {
98 assert(0 && "Untemplated k means dimension");
99 }
100}
101
117 int av1_index_color_cache(const uint16_t *color_cache, int n_cache,
118 const uint16_t *colors, int n_colors,
119 uint8_t *cache_color_found, int *out_cache_colors);
120
133 int av1_get_palette_delta_bits_v(const PALETTE_MODE_INFO *const pmi,
134 int bit_depth, int *zero_count, int *min_bits);
135
147 int av1_palette_color_cost_y(const PALETTE_MODE_INFO *const pmi,
148 const uint16_t *color_cache, int n_cache,
149 int bit_depth);
150
162 int av1_palette_color_cost_uv(const PALETTE_MODE_INFO *const pmi,
163 const uint16_t *color_cache, int n_cache,
164 int bit_depth);
165
172 void av1_rd_pick_palette_intra_sby(
173 const struct AV1_COMP *cpi, struct macroblock *x, BLOCK_SIZE bsize,
174 int dc_mode_cost, MB_MODE_INFO *best_mbmi, uint8_t *best_palette_color_map,
175 int64_t *best_rd, int *rate, int *rate_tokenonly, int64_t *distortion,
176 uint8_t *skippable, int *beat_best_rd, struct PICK_MODE_CONTEXT *ctx,
177 uint8_t *best_blk_skip, uint8_t *tx_type_map);
178
185 void av1_rd_pick_palette_intra_sbuv(const struct AV1_COMP *cpi,
186 struct macroblock *x, int dc_mode_cost,
187 uint8_t *best_palette_color_map,
188 MB_MODE_INFO *const best_mbmi,
189 int64_t *best_rd, int *rate,
190 int *rate_tokenonly, int64_t *distortion,
191 uint8_t *skippable);
192
195 void av1_restore_uv_color_map(const struct AV1_COMP *cpi, struct macroblock *x);
196
197#ifdef __cplusplus
198} // extern "C"
199#endif
200
201#endif // AOM_AV1_ENCODER_PALETTE_H_
av1_palette_color_cost_y
int av1_palette_color_cost_y(const PALETTE_MODE_INFO *const pmi, const uint16_t *color_cache, int n_cache, int bit_depth)
Gets the rate cost for transmitting luma palette color values.
Definition palette.c:137
av1_index_color_cache
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)
Checks what colors are in the color cache.
Definition palette.c:90
av1_rd_pick_palette_intra_sbuv
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, uint8_t *skippable)
Search for the best palette in the chroma plane.
av1_get_palette_delta_bits_v
int av1_get_palette_delta_bits_v(const PALETTE_MODE_INFO *const pmi, int bit_depth, int *zero_count, int *min_bits)
Gets the rate cost for each delta-encoding v palette.
Definition palette.c:118
av1_k_means
static void av1_k_means(const int16_t *data, int16_t *centroids, uint8_t *indices, int n, int k, int dim, int max_itr)
Performs k-means cluster on the data.
Definition palette.h:88
av1_calc_indices
static void av1_calc_indices(const int16_t *data, const int16_t *centroids, uint8_t *indices, int n, int k, int dim)
Calculates the cluster to which each data point belong.
Definition palette.h:54
av1_palette_color_cost_uv
int av1_palette_color_cost_uv(const PALETTE_MODE_INFO *const pmi, const uint16_t *color_cache, int n_cache, int bit_depth)
Gets the rate cost for transmitting luma palette chroma values.
Definition palette.c:151
av1_rd_pick_palette_intra_sby
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, int *rate, int *rate_tokenonly, int64_t *distortion, uint8_t *skippable, int *beat_best_rd, struct PICK_MODE_CONTEXT *ctx, uint8_t *best_blk_skip, uint8_t *tx_type_map)
Search for the best palette in the luma plane.
av1_restore_uv_color_map
void av1_restore_uv_color_map(const struct AV1_COMP *cpi, struct macroblock *x)
Resets palette color map for chroma channels.
AV1_COMP
Top level encoder structure.
Definition encoder.h:2870
MB_MODE_INFO
Stores the prediction/txfm mode of the current coding block.
Definition blockd.h:222
macroblock
Encoder's parameters related to the current coding block.
Definition block.h:878