palette.h 4.57 KB
Newer Older
hui su's avatar
hui su committed
1
/*
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
hui su's avatar
hui su committed
3
 *
4
5
6
7
8
9
 * 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.
hui su's avatar
hui su committed
10
11
 */

Yaowu Xu's avatar
Yaowu Xu committed
12
13
#ifndef AV1_ENCODER_PALETTE_H_
#define AV1_ENCODER_PALETTE_H_
hui su's avatar
hui su committed
14

15
#include "av1/common/blockd.h"
hui su's avatar
hui su committed
16
17
18
19
20

#ifdef __cplusplus
extern "C" {
#endif

21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#define AV1_K_MEANS_RENAME(func, dim) func##_dim##dim

void AV1_K_MEANS_RENAME(av1_calc_indices, 1)(const float *data,
                                             const float *centroids,
                                             uint8_t *indices, int n, int k);
void AV1_K_MEANS_RENAME(av1_calc_indices, 2)(const float *data,
                                             const float *centroids,
                                             uint8_t *indices, int n, int k);
void AV1_K_MEANS_RENAME(av1_k_means, 1)(const float *data, float *centroids,
                                        uint8_t *indices, int n, int k,
                                        int max_itr);
void AV1_K_MEANS_RENAME(av1_k_means, 2)(const float *data, float *centroids,
                                        uint8_t *indices, int n, int k,
                                        int max_itr);

Yaowu Xu's avatar
Yaowu Xu committed
36
37
// Given 'n' 'data' points and 'k' 'centroids' each of dimension 'dim',
// calculate the centroid 'indices' for the data points.
38
39
40
41
42
43
44
45
46
47
static INLINE void av1_calc_indices(const float *data, const float *centroids,
                                    uint8_t *indices, int n, int k, int dim) {
  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");
  }
}
48

Yaowu Xu's avatar
Yaowu Xu committed
49
50
51
// Given 'n' 'data' points and an initial guess of 'k' 'centroids' each of
// dimension 'dim', runs up to 'max_itr' iterations of k-means algorithm to get
// updated 'centroids' and the centroid 'indices' for elements in 'data'.
52
// Note: the output centroids are rounded off to nearest integers.
53
54
55
56
57
58
59
60
61
62
63
static INLINE void av1_k_means(const float *data, float *centroids,
                               uint8_t *indices, int n, int k, int dim,
                               int max_itr) {
  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");
  }
}
64
65
66

// Given a list of centroids, returns the unique number of centroids 'k', and
// puts these unique centroids in first 'k' indices of 'centroids' array.
67
68
// Ideally, the centroids should be rounded to integers before calling this
// method.
Yaowu Xu's avatar
Yaowu Xu committed
69
int av1_remove_duplicates(float *centroids, int num_centroids);
70

hui su's avatar
hui su committed
71
#if CONFIG_PALETTE_DELTA_ENCODING
72
73
74
// Given a color cache and a set of base colors, find if each cache color is
// present in the base colors, record the binary results in "cache_color_found".
// Record the colors that are not in the color cache in "out_cache_colors".
75
76
int av1_index_color_cache(const uint16_t *color_cache, int n_cache,
                          const uint16_t *colors, int n_colors,
77
                          uint8_t *cache_color_found, int *out_cache_colors);
hui su's avatar
hui su committed
78
79
80
81
82
83
84
85

// Return the number of bits used to transmit each v palette color delta;
// assign 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);
#endif  // CONFIG_PALETTE_DELTA_ENCODING

// Return the rate cost for transmitting luma palette color values.
86
87
88
89
90
int av1_palette_color_cost_y(const PALETTE_MODE_INFO *const pmi,
#if CONFIG_PALETTE_DELTA_ENCODING
                             uint16_t *color_cache, int n_cache,
#endif  // CONFIG_PALETTE_DELTA_ENCODING
                             int bit_depth);
hui su's avatar
hui su committed
91
92
93

// Return the rate cost for transmitting chroma palette color values.
int av1_palette_color_cost_uv(const PALETTE_MODE_INFO *const pmi,
94
95
96
#if CONFIG_PALETTE_DELTA_ENCODING
                              uint16_t *color_cache, int n_cache,
#endif  // CONFIG_PALETTE_DELTA_ENCODING
hui su's avatar
hui su committed
97
98
                              int bit_depth);

hui su's avatar
hui su committed
99
100
101
102
#ifdef __cplusplus
}  // extern "C"
#endif

Yaowu Xu's avatar
Yaowu Xu committed
103
#endif /* AV1_ENCODER_PALETTE_H_ */