upsampling_neon.c revision eb525c5499e34cc9c4b825d6d9e75bb07cc06ace
12a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Copyright 2011 Google Inc. All Rights Reserved. 22a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// 3eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch// Use of this source code is governed by a BSD-style license 4eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch// that can be found in the COPYING file in the root of the source 5eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch// tree. An additional intellectual property rights grant can be found 6eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch// in the file PATENTS. All contributing project authors may 7eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch// be found in the AUTHORS file in the root of the source tree. 82a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// ----------------------------------------------------------------------------- 92a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// 102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// NEON version of YUV to RGB upsampling functions. 112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// 122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Author: mans@mansr.com (Mans Rullgard) 132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Based on SSE code by: somnath@google.com (Somnath Banerjee) 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "./dsp.h" 162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#if defined(__cplusplus) || defined(c_plusplus) 182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)extern "C" { 192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif 202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#if defined(WEBP_USE_NEON) 222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include <assert.h> 242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include <arm_neon.h> 252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include <string.h> 262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "./yuv.h" 272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#ifdef FANCY_UPSAMPLING 292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Loads 9 pixels each from rows r1 and r2 and generates 16 pixels. 312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define UPSAMPLE_16PIXELS(r1, r2, out) { \ 322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint8x8_t a = vld1_u8(r1); \ 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint8x8_t b = vld1_u8(r1 + 1); \ 342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint8x8_t c = vld1_u8(r2); \ 352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint8x8_t d = vld1_u8(r2 + 1); \ 362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint16x8_t al = vshll_n_u8(a, 1); \ 382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint16x8_t bl = vshll_n_u8(b, 1); \ 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint16x8_t cl = vshll_n_u8(c, 1); \ 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint16x8_t dl = vshll_n_u8(d, 1); \ 412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint8x8_t diag1, diag2; \ 432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint16x8_t sl; \ 442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) /* a + b + c + d */ \ 462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sl = vaddl_u8(a, b); \ 472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sl = vaddw_u8(sl, c); \ 482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sl = vaddw_u8(sl, d); \ 492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) al = vaddq_u16(sl, al); /* 3a + b + c + d */ \ 512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bl = vaddq_u16(sl, bl); /* a + 3b + c + d */ \ 522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) al = vaddq_u16(al, dl); /* 3a + b + c + 3d */ \ 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bl = vaddq_u16(bl, cl); /* a + 3b + 3c + d */ \ 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) diag2 = vshrn_n_u16(al, 3); \ 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) diag1 = vshrn_n_u16(bl, 3); \ 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) a = vrhadd_u8(a, diag1); \ 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) b = vrhadd_u8(b, diag2); \ 612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c = vrhadd_u8(c, diag2); \ 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) d = vrhadd_u8(d, diag1); \ 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) { \ 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const uint8x8x2_t a_b = {{ a, b }}; \ 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const uint8x8x2_t c_d = {{ c, d }}; \ 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vst2_u8(out, a_b); \ 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vst2_u8(out + 32, c_d); \ 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } \ 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Turn the macro into a function for reducing code-size when non-critical 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)static void Upsample16Pixels(const uint8_t *r1, const uint8_t *r2, 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint8_t *out) { 752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UPSAMPLE_16PIXELS(r1, r2, out); 762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define UPSAMPLE_LAST_BLOCK(tb, bb, num_pixels, out) { \ 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint8_t r1[9], r2[9]; \ 802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) memcpy(r1, (tb), (num_pixels)); \ 812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) memcpy(r2, (bb), (num_pixels)); \ 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) /* replicate last byte */ \ 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) memset(r1 + (num_pixels), r1[(num_pixels) - 1], 9 - (num_pixels)); \ 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) memset(r2 + (num_pixels), r2[(num_pixels) - 1], 9 - (num_pixels)); \ 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Upsample16Pixels(r1, r2, out); \ 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define CY 76283 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define CVR 89858 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define CUG 22014 912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define CVG 45773 922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define CUB 113618 932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)static const int16_t coef[4] = { CVR / 4, CUG, CVG / 2, CUB / 4 }; 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define CONVERT8(FMT, XSTEP, N, src_y, src_uv, out, cur_x) { \ 972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int i; \ 982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (i = 0; i < N; i += 8) { \ 992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int off = ((cur_x) + i) * XSTEP; \ 1002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint8x8_t y = vld1_u8(src_y + (cur_x) + i); \ 1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint8x8_t u = vld1_u8((src_uv) + i); \ 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint8x8_t v = vld1_u8((src_uv) + i + 16); \ 1032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int16x8_t yy = vreinterpretq_s16_u16(vsubl_u8(y, u16)); \ 1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int16x8_t uu = vreinterpretq_s16_u16(vsubl_u8(u, u128)); \ 1052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int16x8_t vv = vreinterpretq_s16_u16(vsubl_u8(v, u128)); \ 1062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int16x8_t ud = vshlq_n_s16(uu, 1); \ 1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int16x8_t vd = vshlq_n_s16(vv, 1); \ 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 1102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t vrl = vqdmlal_lane_s16(vshll_n_s16(vget_low_s16(vv), 1), \ 1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vget_low_s16(vd), cf16, 0); \ 1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t vrh = vqdmlal_lane_s16(vshll_n_s16(vget_high_s16(vv), 1), \ 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vget_high_s16(vd), cf16, 0); \ 1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int16x8_t vr = vcombine_s16(vrshrn_n_s32(vrl, 16), \ 1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vrshrn_n_s32(vrh, 16)); \ 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t vl = vmovl_s16(vget_low_s16(vv)); \ 1182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t vh = vmovl_s16(vget_high_s16(vv)); \ 1192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t ugl = vmlal_lane_s16(vl, vget_low_s16(uu), cf16, 1); \ 1202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t ugh = vmlal_lane_s16(vh, vget_high_s16(uu), cf16, 1); \ 1212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t gcl = vqdmlal_lane_s16(ugl, vget_low_s16(vv), cf16, 2); \ 1222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t gch = vqdmlal_lane_s16(ugh, vget_high_s16(vv), cf16, 2); \ 1232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int16x8_t gc = vcombine_s16(vrshrn_n_s32(gcl, 16), \ 1242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vrshrn_n_s32(gch, 16)); \ 1252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 1262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t ubl = vqdmlal_lane_s16(vshll_n_s16(vget_low_s16(uu), 1), \ 1272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vget_low_s16(ud), cf16, 3); \ 1282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t ubh = vqdmlal_lane_s16(vshll_n_s16(vget_high_s16(uu), 1), \ 1292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vget_high_s16(ud), cf16, 3); \ 1302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int16x8_t ub = vcombine_s16(vrshrn_n_s32(ubl, 16), \ 1312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vrshrn_n_s32(ubh, 16)); \ 1322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 1332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t rl = vaddl_s16(vget_low_s16(yy), vget_low_s16(vr)); \ 1342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t rh = vaddl_s16(vget_high_s16(yy), vget_high_s16(vr)); \ 1352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t gl = vsubl_s16(vget_low_s16(yy), vget_low_s16(gc)); \ 1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t gh = vsubl_s16(vget_high_s16(yy), vget_high_s16(gc)); \ 1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t bl = vaddl_s16(vget_low_s16(yy), vget_low_s16(ub)); \ 1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int32x4_t bh = vaddl_s16(vget_high_s16(yy), vget_high_s16(ub)); \ 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) rl = vmulq_lane_s32(rl, cf32, 0); \ 1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) rh = vmulq_lane_s32(rh, cf32, 0); \ 1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) gl = vmulq_lane_s32(gl, cf32, 0); \ 1432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) gh = vmulq_lane_s32(gh, cf32, 0); \ 1442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bl = vmulq_lane_s32(bl, cf32, 0); \ 1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bh = vmulq_lane_s32(bh, cf32, 0); \ 1462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 1472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) y = vqmovun_s16(vcombine_s16(vrshrn_n_s32(rl, 16), \ 1482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vrshrn_n_s32(rh, 16))); \ 1492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) u = vqmovun_s16(vcombine_s16(vrshrn_n_s32(gl, 16), \ 1502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vrshrn_n_s32(gh, 16))); \ 1512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) v = vqmovun_s16(vcombine_s16(vrshrn_n_s32(bl, 16), \ 1522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vrshrn_n_s32(bh, 16))); \ 1532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) STR_ ## FMT(out + off, y, u, v); \ 1542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } \ 1552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define v255 vmov_n_u8(255) 1582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define STR_Rgb(out, r, g, b) do { \ 1602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const uint8x8x3_t r_g_b = {{ r, g, b }}; \ 1612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vst3_u8(out, r_g_b); \ 1622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} while (0) 1632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define STR_Bgr(out, r, g, b) do { \ 1652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const uint8x8x3_t b_g_r = {{ b, g, r }}; \ 1662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vst3_u8(out, b_g_r); \ 1672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} while (0) 1682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define STR_Rgba(out, r, g, b) do { \ 1702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const uint8x8x4_t r_g_b_v255 = {{ r, g, b, v255 }}; \ 1712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vst4_u8(out, r_g_b_v255); \ 1722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} while (0) 1732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define STR_Bgra(out, r, g, b) do { \ 1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const uint8x8x4_t b_g_r_v255 = {{ b, g, r, v255 }}; \ 1762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) vst4_u8(out, b_g_r_v255); \ 1772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} while (0) 1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define CONVERT1(FMT, XSTEP, N, src_y, src_uv, rgb, cur_x) { \ 1802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int i; \ 1812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (i = 0; i < N; i++) { \ 1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int off = ((cur_x) + i) * XSTEP; \ 1832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int y = src_y[(cur_x) + i]; \ 1842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int u = (src_uv)[i]; \ 1852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int v = (src_uv)[i + 16]; \ 1862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) VP8YuvTo ## FMT(y, u, v, rgb + off); \ 1872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } \ 1882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define CONVERT2RGB_8(FMT, XSTEP, top_y, bottom_y, uv, \ 1912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) top_dst, bottom_dst, cur_x, len) { \ 1922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (top_y) { \ 1932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CONVERT8(FMT, XSTEP, len, top_y, uv, top_dst, cur_x) \ 1942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } \ 1952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (bottom_y) { \ 1962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CONVERT8(FMT, XSTEP, len, bottom_y, (uv) + 32, bottom_dst, cur_x) \ 1972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } \ 1982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define CONVERT2RGB_1(FMT, XSTEP, top_y, bottom_y, uv, \ 2012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) top_dst, bottom_dst, cur_x, len) { \ 2022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (top_y) { \ 2032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CONVERT1(FMT, XSTEP, len, top_y, uv, top_dst, cur_x); \ 2042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } \ 2052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (bottom_y) { \ 2062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CONVERT1(FMT, XSTEP, len, bottom_y, (uv) + 32, bottom_dst, cur_x); \ 2072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } \ 2082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define NEON_UPSAMPLE_FUNC(FUNC_NAME, FMT, XSTEP) \ 2112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)static void FUNC_NAME(const uint8_t *top_y, const uint8_t *bottom_y, \ 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const uint8_t *top_u, const uint8_t *top_v, \ 2132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const uint8_t *cur_u, const uint8_t *cur_v, \ 2142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint8_t *top_dst, uint8_t *bottom_dst, int len) { \ 2152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int block; \ 2162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) /* 16 byte aligned array to cache reconstructed u and v */ \ 2172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint8_t uv_buf[2 * 32 + 15]; \ 2182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) uint8_t *const r_uv = (uint8_t*)((uintptr_t)(uv_buf + 15) & ~15); \ 2192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int uv_len = (len + 1) >> 1; \ 2202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) /* 9 pixels must be read-able for each block */ \ 2212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int num_blocks = (uv_len - 1) >> 3; \ 2222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int leftover = uv_len - num_blocks * 8; \ 2232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int last_pos = 1 + 16 * num_blocks; \ 2242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 2252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int u_diag = ((top_u[0] + cur_u[0]) >> 1) + 1; \ 2262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int v_diag = ((top_v[0] + cur_v[0]) >> 1) + 1; \ 2272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 2282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int16x4_t cf16 = vld1_s16(coef); \ 2292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int32x2_t cf32 = vmov_n_s32(CY); \ 2302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const uint8x8_t u16 = vmov_n_u8(16); \ 2312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const uint8x8_t u128 = vmov_n_u8(128); \ 2322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 2332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) /* Treat the first pixel in regular way */ \ 2342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (top_y) { \ 2352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int u0 = (top_u[0] + u_diag) >> 1; \ 2362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int v0 = (top_v[0] + v_diag) >> 1; \ 2372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) VP8YuvTo ## FMT(top_y[0], u0, v0, top_dst); \ 2382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } \ 2392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (bottom_y) { \ 2402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int u0 = (cur_u[0] + u_diag) >> 1; \ 2412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int v0 = (cur_v[0] + v_diag) >> 1; \ 2422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) VP8YuvTo ## FMT(bottom_y[0], u0, v0, bottom_dst); \ 2432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } \ 2442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 2452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (block = 0; block < num_blocks; ++block) { \ 2462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UPSAMPLE_16PIXELS(top_u, cur_u, r_uv); \ 2472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UPSAMPLE_16PIXELS(top_v, cur_v, r_uv + 16); \ 2482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CONVERT2RGB_8(FMT, XSTEP, top_y, bottom_y, r_uv, \ 2492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) top_dst, bottom_dst, 16 * block + 1, 16); \ 2502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) top_u += 8; \ 2512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cur_u += 8; \ 2522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) top_v += 8; \ 2532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cur_v += 8; \ 2542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } \ 2552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) \ 2562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UPSAMPLE_LAST_BLOCK(top_u, cur_u, leftover, r_uv); \ 2572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UPSAMPLE_LAST_BLOCK(top_v, cur_v, leftover, r_uv + 16); \ 2582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CONVERT2RGB_1(FMT, XSTEP, top_y, bottom_y, r_uv, \ 2592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) top_dst, bottom_dst, last_pos, len - last_pos); \ 2602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// NEON variants of the fancy upsampler. 2632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)NEON_UPSAMPLE_FUNC(UpsampleRgbLinePairNEON, Rgb, 3) 2642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)NEON_UPSAMPLE_FUNC(UpsampleBgrLinePairNEON, Bgr, 3) 2652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)NEON_UPSAMPLE_FUNC(UpsampleRgbaLinePairNEON, Rgba, 4) 2662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)NEON_UPSAMPLE_FUNC(UpsampleBgraLinePairNEON, Bgra, 4) 2672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif // FANCY_UPSAMPLING 2692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif // WEBP_USE_NEON 2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)//------------------------------------------------------------------------------ 2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)extern WebPUpsampleLinePairFunc WebPUpsamplers[/* MODE_LAST */]; 2752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void WebPInitUpsamplersNEON(void) { 2772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#if defined(WEBP_USE_NEON) 2782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) WebPUpsamplers[MODE_RGB] = UpsampleRgbLinePairNEON; 2792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) WebPUpsamplers[MODE_RGBA] = UpsampleRgbaLinePairNEON; 2802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) WebPUpsamplers[MODE_BGR] = UpsampleBgrLinePairNEON; 2812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) WebPUpsamplers[MODE_BGRA] = UpsampleBgraLinePairNEON; 2822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif // WEBP_USE_NEON 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void WebPInitPremultiplyNEON(void) { 2862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#if defined(WEBP_USE_NEON) 2872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) WebPUpsamplers[MODE_rgbA] = UpsampleRgbaLinePairNEON; 2882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) WebPUpsamplers[MODE_bgrA] = UpsampleBgraLinePairNEON; 2892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif // WEBP_USE_NEON 2902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#if defined(__cplusplus) || defined(c_plusplus) 2932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} // extern "C" 2942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif 295