summaryrefslogtreecommitdiffstats
path: root/third_party/aom/av1/encoder/k_means_template.h
diff options
context:
space:
mode:
authorMatt A. Tobin <email@mattatobin.com>2020-04-07 23:30:51 -0400
committerMatt A. Tobin <email@mattatobin.com>2020-04-07 23:30:51 -0400
commit5545a8983ff0ef1fb52e64aef8e66fa9b13c1cbb (patch)
tree45d55e3e5e73c4255c4d71258d9be5b2d004d28f /third_party/aom/av1/encoder/k_means_template.h
parent50f1986697a7412e4160976fa5e11217b4ef1f44 (diff)
downloadUXP-5545a8983ff0ef1fb52e64aef8e66fa9b13c1cbb.tar
UXP-5545a8983ff0ef1fb52e64aef8e66fa9b13c1cbb.tar.gz
UXP-5545a8983ff0ef1fb52e64aef8e66fa9b13c1cbb.tar.lz
UXP-5545a8983ff0ef1fb52e64aef8e66fa9b13c1cbb.tar.xz
UXP-5545a8983ff0ef1fb52e64aef8e66fa9b13c1cbb.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/k_means_template.h')
-rw-r--r--third_party/aom/av1/encoder/k_means_template.h123
1 files changed, 0 insertions, 123 deletions
diff --git a/third_party/aom/av1/encoder/k_means_template.h b/third_party/aom/av1/encoder/k_means_template.h
deleted file mode 100644
index 9e526b88b..000000000
--- a/third_party/aom/av1/encoder/k_means_template.h
+++ /dev/null
@@ -1,123 +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 <stdint.h>
-#include <string.h>
-
-#include "av1/encoder/palette.h"
-#include "av1/encoder/random.h"
-
-#ifndef AV1_K_MEANS_DIM
-#error "This template requires AV1_K_MEANS_DIM to be defined"
-#endif
-
-#define RENAME_(x, y) AV1_K_MEANS_RENAME(x, y)
-#define RENAME(x) RENAME_(x, AV1_K_MEANS_DIM)
-
-static int RENAME(calc_dist)(const int *p1, const int *p2) {
- int dist = 0;
- for (int i = 0; i < AV1_K_MEANS_DIM; ++i) {
- const int diff = p1[i] - p2[i];
- dist += diff * diff;
- }
- return dist;
-}
-
-void RENAME(av1_calc_indices)(const int *data, const int *centroids,
- uint8_t *indices, int n, int k) {
- for (int i = 0; i < n; ++i) {
- int min_dist = RENAME(calc_dist)(data + i * AV1_K_MEANS_DIM, centroids);
- indices[i] = 0;
- for (int j = 1; j < k; ++j) {
- const int this_dist = RENAME(calc_dist)(data + i * AV1_K_MEANS_DIM,
- centroids + j * AV1_K_MEANS_DIM);
- if (this_dist < min_dist) {
- min_dist = this_dist;
- indices[i] = j;
- }
- }
- }
-}
-
-static void RENAME(calc_centroids)(const int *data, int *centroids,
- const uint8_t *indices, int n, int k) {
- int i, j;
- int count[PALETTE_MAX_SIZE] = { 0 };
- unsigned int rand_state = (unsigned int)data[0];
- assert(n <= 32768);
- memset(centroids, 0, sizeof(centroids[0]) * k * AV1_K_MEANS_DIM);
-
- for (i = 0; i < n; ++i) {
- const int index = indices[i];
- assert(index < k);
- ++count[index];
- for (j = 0; j < AV1_K_MEANS_DIM; ++j) {
- centroids[index * AV1_K_MEANS_DIM + j] += data[i * AV1_K_MEANS_DIM + j];
- }
- }
-
- for (i = 0; i < k; ++i) {
- if (count[i] == 0) {
- memcpy(centroids + i * AV1_K_MEANS_DIM,
- data + (lcg_rand16(&rand_state) % n) * AV1_K_MEANS_DIM,
- sizeof(centroids[0]) * AV1_K_MEANS_DIM);
- } else {
- for (j = 0; j < AV1_K_MEANS_DIM; ++j) {
- centroids[i * AV1_K_MEANS_DIM + j] =
- DIVIDE_AND_ROUND(centroids[i * AV1_K_MEANS_DIM + j], count[i]);
- }
- }
- }
-}
-
-static int64_t RENAME(calc_total_dist)(const int *data, const int *centroids,
- const uint8_t *indices, int n, int k) {
- int64_t dist = 0;
- (void)k;
- for (int i = 0; i < n; ++i) {
- dist += RENAME(calc_dist)(data + i * AV1_K_MEANS_DIM,
- centroids + indices[i] * AV1_K_MEANS_DIM);
- }
- return dist;
-}
-
-void RENAME(av1_k_means)(const int *data, int *centroids, uint8_t *indices,
- int n, int k, int max_itr) {
- int pre_centroids[2 * PALETTE_MAX_SIZE];
- uint8_t pre_indices[MAX_SB_SQUARE];
-
- RENAME(av1_calc_indices)(data, centroids, indices, n, k);
- int64_t this_dist = RENAME(calc_total_dist)(data, centroids, indices, n, k);
-
- for (int i = 0; i < max_itr; ++i) {
- const int64_t pre_dist = this_dist;
- memcpy(pre_centroids, centroids,
- sizeof(pre_centroids[0]) * k * AV1_K_MEANS_DIM);
- memcpy(pre_indices, indices, sizeof(pre_indices[0]) * n);
-
- RENAME(calc_centroids)(data, centroids, indices, n, k);
- RENAME(av1_calc_indices)(data, centroids, indices, n, k);
- this_dist = RENAME(calc_total_dist)(data, centroids, indices, n, k);
-
- if (this_dist > pre_dist) {
- memcpy(centroids, pre_centroids,
- sizeof(pre_centroids[0]) * k * AV1_K_MEANS_DIM);
- memcpy(indices, pre_indices, sizeof(pre_indices[0]) * n);
- break;
- }
- if (!memcmp(centroids, pre_centroids,
- sizeof(pre_centroids[0]) * k * AV1_K_MEANS_DIM))
- break;
- }
-}
-#undef RENAME_
-#undef RENAME