270 lines
9.4 KiB
C
270 lines
9.4 KiB
C
// Copyright 2014 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.
|
|
// -----------------------------------------------------------------------------
|
|
//
|
|
// NEON variant of methods for lossless decoder
|
|
//
|
|
// Author: Skal (pascal.massimino@gmail.com)
|
|
|
|
#include "./dsp.h"
|
|
|
|
#if defined(WEBP_USE_NEON)
|
|
|
|
#include <arm_neon.h>
|
|
|
|
#include "./lossless.h"
|
|
#include "./neon.h"
|
|
|
|
//------------------------------------------------------------------------------
|
|
// Colorspace conversion functions
|
|
|
|
#if !defined(WORK_AROUND_GCC)
|
|
// gcc 4.6.0 had some trouble (NDK-r9) with this code. We only use it for
|
|
// gcc-4.8.x at least.
|
|
static void ConvertBGRAToRGBA(const uint32_t* src,
|
|
int num_pixels, uint8_t* dst) {
|
|
const uint32_t* const end = src + (num_pixels & ~15);
|
|
for (; src < end; src += 16) {
|
|
uint8x16x4_t pixel = vld4q_u8((uint8_t*)src);
|
|
// swap B and R. (VSWP d0,d2 has no intrinsics equivalent!)
|
|
const uint8x16_t tmp = pixel.val[0];
|
|
pixel.val[0] = pixel.val[2];
|
|
pixel.val[2] = tmp;
|
|
vst4q_u8(dst, pixel);
|
|
dst += 64;
|
|
}
|
|
VP8LConvertBGRAToRGBA_C(src, num_pixels & 15, dst); // left-overs
|
|
}
|
|
|
|
static void ConvertBGRAToBGR(const uint32_t* src,
|
|
int num_pixels, uint8_t* dst) {
|
|
const uint32_t* const end = src + (num_pixels & ~15);
|
|
for (; src < end; src += 16) {
|
|
const uint8x16x4_t pixel = vld4q_u8((uint8_t*)src);
|
|
const uint8x16x3_t tmp = { { pixel.val[0], pixel.val[1], pixel.val[2] } };
|
|
vst3q_u8(dst, tmp);
|
|
dst += 48;
|
|
}
|
|
VP8LConvertBGRAToBGR_C(src, num_pixels & 15, dst); // left-overs
|
|
}
|
|
|
|
static void ConvertBGRAToRGB(const uint32_t* src,
|
|
int num_pixels, uint8_t* dst) {
|
|
const uint32_t* const end = src + (num_pixels & ~15);
|
|
for (; src < end; src += 16) {
|
|
const uint8x16x4_t pixel = vld4q_u8((uint8_t*)src);
|
|
const uint8x16x3_t tmp = { { pixel.val[2], pixel.val[1], pixel.val[0] } };
|
|
vst3q_u8(dst, tmp);
|
|
dst += 48;
|
|
}
|
|
VP8LConvertBGRAToRGB_C(src, num_pixels & 15, dst); // left-overs
|
|
}
|
|
|
|
#else // WORK_AROUND_GCC
|
|
|
|
// gcc-4.6.0 fallback
|
|
|
|
static const uint8_t kRGBAShuffle[8] = { 2, 1, 0, 3, 6, 5, 4, 7 };
|
|
|
|
static void ConvertBGRAToRGBA(const uint32_t* src,
|
|
int num_pixels, uint8_t* dst) {
|
|
const uint32_t* const end = src + (num_pixels & ~1);
|
|
const uint8x8_t shuffle = vld1_u8(kRGBAShuffle);
|
|
for (; src < end; src += 2) {
|
|
const uint8x8_t pixels = vld1_u8((uint8_t*)src);
|
|
vst1_u8(dst, vtbl1_u8(pixels, shuffle));
|
|
dst += 8;
|
|
}
|
|
VP8LConvertBGRAToRGBA_C(src, num_pixels & 1, dst); // left-overs
|
|
}
|
|
|
|
static const uint8_t kBGRShuffle[3][8] = {
|
|
{ 0, 1, 2, 4, 5, 6, 8, 9 },
|
|
{ 10, 12, 13, 14, 16, 17, 18, 20 },
|
|
{ 21, 22, 24, 25, 26, 28, 29, 30 }
|
|
};
|
|
|
|
static void ConvertBGRAToBGR(const uint32_t* src,
|
|
int num_pixels, uint8_t* dst) {
|
|
const uint32_t* const end = src + (num_pixels & ~7);
|
|
const uint8x8_t shuffle0 = vld1_u8(kBGRShuffle[0]);
|
|
const uint8x8_t shuffle1 = vld1_u8(kBGRShuffle[1]);
|
|
const uint8x8_t shuffle2 = vld1_u8(kBGRShuffle[2]);
|
|
for (; src < end; src += 8) {
|
|
uint8x8x4_t pixels;
|
|
INIT_VECTOR4(pixels,
|
|
vld1_u8((const uint8_t*)(src + 0)),
|
|
vld1_u8((const uint8_t*)(src + 2)),
|
|
vld1_u8((const uint8_t*)(src + 4)),
|
|
vld1_u8((const uint8_t*)(src + 6)));
|
|
vst1_u8(dst + 0, vtbl4_u8(pixels, shuffle0));
|
|
vst1_u8(dst + 8, vtbl4_u8(pixels, shuffle1));
|
|
vst1_u8(dst + 16, vtbl4_u8(pixels, shuffle2));
|
|
dst += 8 * 3;
|
|
}
|
|
VP8LConvertBGRAToBGR_C(src, num_pixels & 7, dst); // left-overs
|
|
}
|
|
|
|
static const uint8_t kRGBShuffle[3][8] = {
|
|
{ 2, 1, 0, 6, 5, 4, 10, 9 },
|
|
{ 8, 14, 13, 12, 18, 17, 16, 22 },
|
|
{ 21, 20, 26, 25, 24, 30, 29, 28 }
|
|
};
|
|
|
|
static void ConvertBGRAToRGB(const uint32_t* src,
|
|
int num_pixels, uint8_t* dst) {
|
|
const uint32_t* const end = src + (num_pixels & ~7);
|
|
const uint8x8_t shuffle0 = vld1_u8(kRGBShuffle[0]);
|
|
const uint8x8_t shuffle1 = vld1_u8(kRGBShuffle[1]);
|
|
const uint8x8_t shuffle2 = vld1_u8(kRGBShuffle[2]);
|
|
for (; src < end; src += 8) {
|
|
uint8x8x4_t pixels;
|
|
INIT_VECTOR4(pixels,
|
|
vld1_u8((const uint8_t*)(src + 0)),
|
|
vld1_u8((const uint8_t*)(src + 2)),
|
|
vld1_u8((const uint8_t*)(src + 4)),
|
|
vld1_u8((const uint8_t*)(src + 6)));
|
|
vst1_u8(dst + 0, vtbl4_u8(pixels, shuffle0));
|
|
vst1_u8(dst + 8, vtbl4_u8(pixels, shuffle1));
|
|
vst1_u8(dst + 16, vtbl4_u8(pixels, shuffle2));
|
|
dst += 8 * 3;
|
|
}
|
|
VP8LConvertBGRAToRGB_C(src, num_pixels & 7, dst); // left-overs
|
|
}
|
|
|
|
#endif // !WORK_AROUND_GCC
|
|
|
|
//------------------------------------------------------------------------------
|
|
// Subtract-Green Transform
|
|
|
|
// vtbl?_u8 are marked unavailable for iOS arm64 with Xcode < 6.3, use
|
|
// non-standard versions there.
|
|
#if defined(__APPLE__) && defined(__aarch64__) && \
|
|
defined(__apple_build_version__) && (__apple_build_version__< 6020037)
|
|
#define USE_VTBLQ
|
|
#endif
|
|
|
|
#ifdef USE_VTBLQ
|
|
// 255 = byte will be zeroed
|
|
static const uint8_t kGreenShuffle[16] = {
|
|
1, 255, 1, 255, 5, 255, 5, 255, 9, 255, 9, 255, 13, 255, 13, 255
|
|
};
|
|
|
|
static WEBP_INLINE uint8x16_t DoGreenShuffle(const uint8x16_t argb,
|
|
const uint8x16_t shuffle) {
|
|
return vcombine_u8(vtbl1q_u8(argb, vget_low_u8(shuffle)),
|
|
vtbl1q_u8(argb, vget_high_u8(shuffle)));
|
|
}
|
|
#else // !USE_VTBLQ
|
|
// 255 = byte will be zeroed
|
|
static const uint8_t kGreenShuffle[8] = { 1, 255, 1, 255, 5, 255, 5, 255 };
|
|
|
|
static WEBP_INLINE uint8x16_t DoGreenShuffle(const uint8x16_t argb,
|
|
const uint8x8_t shuffle) {
|
|
return vcombine_u8(vtbl1_u8(vget_low_u8(argb), shuffle),
|
|
vtbl1_u8(vget_high_u8(argb), shuffle));
|
|
}
|
|
#endif // USE_VTBLQ
|
|
|
|
static void AddGreenToBlueAndRed(uint32_t* argb_data, int num_pixels) {
|
|
const uint32_t* const end = argb_data + (num_pixels & ~3);
|
|
#ifdef USE_VTBLQ
|
|
const uint8x16_t shuffle = vld1q_u8(kGreenShuffle);
|
|
#else
|
|
const uint8x8_t shuffle = vld1_u8(kGreenShuffle);
|
|
#endif
|
|
for (; argb_data < end; argb_data += 4) {
|
|
const uint8x16_t argb = vld1q_u8((uint8_t*)argb_data);
|
|
const uint8x16_t greens = DoGreenShuffle(argb, shuffle);
|
|
vst1q_u8((uint8_t*)argb_data, vaddq_u8(argb, greens));
|
|
}
|
|
// fallthrough and finish off with plain-C
|
|
VP8LAddGreenToBlueAndRed_C(argb_data, num_pixels & 3);
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
// Color Transform
|
|
|
|
static void TransformColorInverse(const VP8LMultipliers* const m,
|
|
uint32_t* argb_data, int num_pixels) {
|
|
// sign-extended multiplying constants, pre-shifted by 6.
|
|
#define CST(X) (((int16_t)(m->X << 8)) >> 6)
|
|
const int16_t rb[8] = {
|
|
CST(green_to_blue_), CST(green_to_red_),
|
|
CST(green_to_blue_), CST(green_to_red_),
|
|
CST(green_to_blue_), CST(green_to_red_),
|
|
CST(green_to_blue_), CST(green_to_red_)
|
|
};
|
|
const int16x8_t mults_rb = vld1q_s16(rb);
|
|
const int16_t b2[8] = {
|
|
0, CST(red_to_blue_), 0, CST(red_to_blue_),
|
|
0, CST(red_to_blue_), 0, CST(red_to_blue_),
|
|
};
|
|
const int16x8_t mults_b2 = vld1q_s16(b2);
|
|
#undef CST
|
|
#ifdef USE_VTBLQ
|
|
static const uint8_t kg0g0[16] = {
|
|
255, 1, 255, 1, 255, 5, 255, 5, 255, 9, 255, 9, 255, 13, 255, 13
|
|
};
|
|
const uint8x16_t shuffle = vld1q_u8(kg0g0);
|
|
#else
|
|
static const uint8_t k0g0g[8] = { 255, 1, 255, 1, 255, 5, 255, 5 };
|
|
const uint8x8_t shuffle = vld1_u8(k0g0g);
|
|
#endif
|
|
const uint32x4_t mask_ag = vdupq_n_u32(0xff00ff00u);
|
|
int i;
|
|
for (i = 0; i + 4 <= num_pixels; i += 4) {
|
|
const uint8x16_t in = vld1q_u8((uint8_t*)(argb_data + i));
|
|
const uint32x4_t a0g0 = vandq_u32(vreinterpretq_u32_u8(in), mask_ag);
|
|
// 0 g 0 g
|
|
const uint8x16_t greens = DoGreenShuffle(in, shuffle);
|
|
// x dr x db1
|
|
const int16x8_t A = vqdmulhq_s16(vreinterpretq_s16_u8(greens), mults_rb);
|
|
// x r' x b'
|
|
const int8x16_t B = vaddq_s8(vreinterpretq_s8_u8(in),
|
|
vreinterpretq_s8_s16(A));
|
|
// r' 0 b' 0
|
|
const int16x8_t C = vshlq_n_s16(vreinterpretq_s16_s8(B), 8);
|
|
// x db2 0 0
|
|
const int16x8_t D = vqdmulhq_s16(C, mults_b2);
|
|
// 0 x db2 0
|
|
const uint32x4_t E = vshrq_n_u32(vreinterpretq_u32_s16(D), 8);
|
|
// r' x b'' 0
|
|
const int8x16_t F = vaddq_s8(vreinterpretq_s8_u32(E),
|
|
vreinterpretq_s8_s16(C));
|
|
// 0 r' 0 b''
|
|
const uint16x8_t G = vshrq_n_u16(vreinterpretq_u16_s8(F), 8);
|
|
const uint32x4_t out = vorrq_u32(vreinterpretq_u32_u16(G), a0g0);
|
|
vst1q_u32(argb_data + i, out);
|
|
}
|
|
// Fall-back to C-version for left-overs.
|
|
VP8LTransformColorInverse_C(m, argb_data + i, num_pixels - i);
|
|
}
|
|
|
|
#undef USE_VTBLQ
|
|
|
|
//------------------------------------------------------------------------------
|
|
// Entry point
|
|
|
|
extern void VP8LDspInitNEON(void);
|
|
|
|
WEBP_TSAN_IGNORE_FUNCTION void VP8LDspInitNEON(void) {
|
|
VP8LConvertBGRAToRGBA = ConvertBGRAToRGBA;
|
|
VP8LConvertBGRAToBGR = ConvertBGRAToBGR;
|
|
VP8LConvertBGRAToRGB = ConvertBGRAToRGB;
|
|
|
|
VP8LAddGreenToBlueAndRed = AddGreenToBlueAndRed;
|
|
VP8LTransformColorInverse = TransformColorInverse;
|
|
}
|
|
|
|
#else // !WEBP_USE_NEON
|
|
|
|
WEBP_DSP_INIT_STUB(VP8LDspInitNEON)
|
|
|
|
#endif // WEBP_USE_NEON
|