mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
104 lines
3.3 KiB
104 lines
3.3 KiB
// Copyright 2022 Google Inc. All Rights Reserved. |
|
// |
|
// Use of this source code is governed by a BSD-style license |
|
// that can be found in the COPYING file in the root of the source |
|
// tree. An additional intellectual property rights grant can be found |
|
// in the file PATENTS. All contributing project authors may |
|
// be found in the AUTHORS file in the root of the source tree. |
|
// ----------------------------------------------------------------------------- |
|
// |
|
// Speed-critical functions for Sharp YUV. |
|
// |
|
// Author: Skal (pascal.massimino@gmail.com) |
|
|
|
#include "sharpyuv/sharpyuv_dsp.h" |
|
|
|
#include <assert.h> |
|
#include <stdlib.h> |
|
|
|
#include "sharpyuv/sharpyuv_cpu.h" |
|
|
|
//----------------------------------------------------------------------------- |
|
|
|
#if !WEBP_NEON_OMIT_C_CODE |
|
static uint16_t clip(int v, int max) { |
|
return (v < 0) ? 0 : (v > max) ? max : (uint16_t)v; |
|
} |
|
|
|
static uint64_t SharpYuvUpdateY_C(const uint16_t* ref, const uint16_t* src, |
|
uint16_t* dst, int len, int bit_depth) { |
|
uint64_t diff = 0; |
|
int i; |
|
const int max_y = (1 << bit_depth) - 1; |
|
for (i = 0; i < len; ++i) { |
|
const int diff_y = ref[i] - src[i]; |
|
const int new_y = (int)dst[i] + diff_y; |
|
dst[i] = clip(new_y, max_y); |
|
diff += (uint64_t)abs(diff_y); |
|
} |
|
return diff; |
|
} |
|
|
|
static void SharpYuvUpdateRGB_C(const int16_t* ref, const int16_t* src, |
|
int16_t* dst, int len) { |
|
int i; |
|
for (i = 0; i < len; ++i) { |
|
const int diff_uv = ref[i] - src[i]; |
|
dst[i] += diff_uv; |
|
} |
|
} |
|
|
|
static void SharpYuvFilterRow_C(const int16_t* A, const int16_t* B, int len, |
|
const uint16_t* best_y, uint16_t* out, |
|
int bit_depth) { |
|
int i; |
|
const int max_y = (1 << bit_depth) - 1; |
|
for (i = 0; i < len; ++i, ++A, ++B) { |
|
const int v0 = (A[0] * 9 + A[1] * 3 + B[0] * 3 + B[1] + 8) >> 4; |
|
const int v1 = (A[1] * 9 + A[0] * 3 + B[1] * 3 + B[0] + 8) >> 4; |
|
out[2 * i + 0] = clip(best_y[2 * i + 0] + v0, max_y); |
|
out[2 * i + 1] = clip(best_y[2 * i + 1] + v1, max_y); |
|
} |
|
} |
|
#endif // !WEBP_NEON_OMIT_C_CODE |
|
|
|
//----------------------------------------------------------------------------- |
|
|
|
uint64_t (*SharpYuvUpdateY)(const uint16_t* src, const uint16_t* ref, |
|
uint16_t* dst, int len, int bit_depth); |
|
void (*SharpYuvUpdateRGB)(const int16_t* src, const int16_t* ref, int16_t* dst, |
|
int len); |
|
void (*SharpYuvFilterRow)(const int16_t* A, const int16_t* B, int len, |
|
const uint16_t* best_y, uint16_t* out, |
|
int bit_depth); |
|
|
|
extern VP8CPUInfo SharpYuvGetCPUInfo; |
|
extern void InitSharpYuvSSE2(void); |
|
extern void InitSharpYuvNEON(void); |
|
|
|
void SharpYuvInitDsp(void) { |
|
#if !WEBP_NEON_OMIT_C_CODE |
|
SharpYuvUpdateY = SharpYuvUpdateY_C; |
|
SharpYuvUpdateRGB = SharpYuvUpdateRGB_C; |
|
SharpYuvFilterRow = SharpYuvFilterRow_C; |
|
#endif |
|
|
|
if (SharpYuvGetCPUInfo != NULL) { |
|
#if defined(WEBP_HAVE_SSE2) |
|
if (SharpYuvGetCPUInfo(kSSE2)) { |
|
InitSharpYuvSSE2(); |
|
} |
|
#endif // WEBP_HAVE_SSE2 |
|
} |
|
|
|
#if defined(WEBP_HAVE_NEON) |
|
if (WEBP_NEON_OMIT_C_CODE || |
|
(SharpYuvGetCPUInfo != NULL && SharpYuvGetCPUInfo(kNEON))) { |
|
InitSharpYuvNEON(); |
|
} |
|
#endif // WEBP_HAVE_NEON |
|
|
|
assert(SharpYuvUpdateY != NULL); |
|
assert(SharpYuvUpdateRGB != NULL); |
|
assert(SharpYuvFilterRow != NULL); |
|
}
|
|
|