upsampling_neon.c revision 0406ce1417f76f2034833414dcecc9f56253640c
11e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// Copyright 2011 Google Inc. All Rights Reserved. 21e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// 30406ce1417f76f2034833414dcecc9f56253640cVikas Arora// Use of this source code is governed by a BSD-style license 40406ce1417f76f2034833414dcecc9f56253640cVikas Arora// that can be found in the COPYING file in the root of the source 50406ce1417f76f2034833414dcecc9f56253640cVikas Arora// tree. An additional intellectual property rights grant can be found 60406ce1417f76f2034833414dcecc9f56253640cVikas Arora// in the file PATENTS. All contributing project authors may 70406ce1417f76f2034833414dcecc9f56253640cVikas Arora// be found in the AUTHORS file in the root of the source tree. 81e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// ----------------------------------------------------------------------------- 91e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// 101e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// NEON version of YUV to RGB upsampling functions. 111e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// 121e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// Author: mans@mansr.com (Mans Rullgard) 131e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// Based on SSE code by: somnath@google.com (Somnath Banerjee) 141e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 151e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#include "./dsp.h" 161e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 171e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#if defined(__cplusplus) || defined(c_plusplus) 181e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroraextern "C" { 191e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#endif 201e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 211e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#if defined(WEBP_USE_NEON) 221e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 231e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#include <assert.h> 241e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#include <arm_neon.h> 251e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#include <string.h> 261e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#include "./yuv.h" 271e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 281e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#ifdef FANCY_UPSAMPLING 291e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 301e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// Loads 9 pixels each from rows r1 and r2 and generates 16 pixels. 311e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define UPSAMPLE_16PIXELS(r1, r2, out) { \ 321e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint8x8_t a = vld1_u8(r1); \ 331e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint8x8_t b = vld1_u8(r1 + 1); \ 341e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint8x8_t c = vld1_u8(r2); \ 351e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint8x8_t d = vld1_u8(r2 + 1); \ 361e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 371e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint16x8_t al = vshll_n_u8(a, 1); \ 381e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint16x8_t bl = vshll_n_u8(b, 1); \ 391e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint16x8_t cl = vshll_n_u8(c, 1); \ 401e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint16x8_t dl = vshll_n_u8(d, 1); \ 411e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 421e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint8x8_t diag1, diag2; \ 431e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint16x8_t sl; \ 441e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 451e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora /* a + b + c + d */ \ 461e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora sl = vaddl_u8(a, b); \ 471e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora sl = vaddw_u8(sl, c); \ 481e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora sl = vaddw_u8(sl, d); \ 491e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 501e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora al = vaddq_u16(sl, al); /* 3a + b + c + d */ \ 511e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora bl = vaddq_u16(sl, bl); /* a + 3b + c + d */ \ 521e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 531e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora al = vaddq_u16(al, dl); /* 3a + b + c + 3d */ \ 541e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora bl = vaddq_u16(bl, cl); /* a + 3b + 3c + d */ \ 551e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 561e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora diag2 = vshrn_n_u16(al, 3); \ 571e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora diag1 = vshrn_n_u16(bl, 3); \ 581e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 591e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora a = vrhadd_u8(a, diag1); \ 601e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora b = vrhadd_u8(b, diag2); \ 611e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora c = vrhadd_u8(c, diag2); \ 621e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora d = vrhadd_u8(d, diag1); \ 631e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 641e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora { \ 651e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const uint8x8x2_t a_b = {{ a, b }}; \ 661e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const uint8x8x2_t c_d = {{ c, d }}; \ 671e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vst2_u8(out, a_b); \ 681e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vst2_u8(out + 32, c_d); \ 691e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora } \ 701e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} 711e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 721e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// Turn the macro into a function for reducing code-size when non-critical 731e7bf8805bd030c19924a5306837ecd72c295751Vikas Arorastatic void Upsample16Pixels(const uint8_t *r1, const uint8_t *r2, 741e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint8_t *out) { 751e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora UPSAMPLE_16PIXELS(r1, r2, out); 761e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} 771e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 781e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define UPSAMPLE_LAST_BLOCK(tb, bb, num_pixels, out) { \ 791e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint8_t r1[9], r2[9]; \ 801e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora memcpy(r1, (tb), (num_pixels)); \ 811e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora memcpy(r2, (bb), (num_pixels)); \ 821e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora /* replicate last byte */ \ 831e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora memset(r1 + (num_pixels), r1[(num_pixels) - 1], 9 - (num_pixels)); \ 841e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora memset(r2 + (num_pixels), r2[(num_pixels) - 1], 9 - (num_pixels)); \ 851e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora Upsample16Pixels(r1, r2, out); \ 861e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} 871e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 881e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define CY 76283 891e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define CVR 89858 901e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define CUG 22014 911e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define CVG 45773 921e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define CUB 113618 931e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 941e7bf8805bd030c19924a5306837ecd72c295751Vikas Arorastatic const int16_t coef[4] = { CVR / 4, CUG, CVG / 2, CUB / 4 }; 951e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 961e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define CONVERT8(FMT, XSTEP, N, src_y, src_uv, out, cur_x) { \ 971e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int i; \ 981e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora for (i = 0; i < N; i += 8) { \ 991e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int off = ((cur_x) + i) * XSTEP; \ 1001e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint8x8_t y = vld1_u8(src_y + (cur_x) + i); \ 1011e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint8x8_t u = vld1_u8((src_uv) + i); \ 1021e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint8x8_t v = vld1_u8((src_uv) + i + 16); \ 1031e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int16x8_t yy = vreinterpretq_s16_u16(vsubl_u8(y, u16)); \ 1041e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int16x8_t uu = vreinterpretq_s16_u16(vsubl_u8(u, u128)); \ 1051e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int16x8_t vv = vreinterpretq_s16_u16(vsubl_u8(v, u128)); \ 1061e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 1071e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int16x8_t ud = vshlq_n_s16(uu, 1); \ 1081e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int16x8_t vd = vshlq_n_s16(vv, 1); \ 1091e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 1101e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t vrl = vqdmlal_lane_s16(vshll_n_s16(vget_low_s16(vv), 1), \ 1111e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vget_low_s16(vd), cf16, 0); \ 1121e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t vrh = vqdmlal_lane_s16(vshll_n_s16(vget_high_s16(vv), 1), \ 1131e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vget_high_s16(vd), cf16, 0); \ 1141e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int16x8_t vr = vcombine_s16(vrshrn_n_s32(vrl, 16), \ 1151e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vrshrn_n_s32(vrh, 16)); \ 1161e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 1171e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t vl = vmovl_s16(vget_low_s16(vv)); \ 1181e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t vh = vmovl_s16(vget_high_s16(vv)); \ 1191e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t ugl = vmlal_lane_s16(vl, vget_low_s16(uu), cf16, 1); \ 1201e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t ugh = vmlal_lane_s16(vh, vget_high_s16(uu), cf16, 1); \ 1211e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t gcl = vqdmlal_lane_s16(ugl, vget_low_s16(vv), cf16, 2); \ 1221e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t gch = vqdmlal_lane_s16(ugh, vget_high_s16(vv), cf16, 2); \ 1231e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int16x8_t gc = vcombine_s16(vrshrn_n_s32(gcl, 16), \ 1241e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vrshrn_n_s32(gch, 16)); \ 1251e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 1261e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t ubl = vqdmlal_lane_s16(vshll_n_s16(vget_low_s16(uu), 1), \ 1271e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vget_low_s16(ud), cf16, 3); \ 1281e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t ubh = vqdmlal_lane_s16(vshll_n_s16(vget_high_s16(uu), 1), \ 1291e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vget_high_s16(ud), cf16, 3); \ 1301e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int16x8_t ub = vcombine_s16(vrshrn_n_s32(ubl, 16), \ 1311e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vrshrn_n_s32(ubh, 16)); \ 1321e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 1331e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t rl = vaddl_s16(vget_low_s16(yy), vget_low_s16(vr)); \ 1341e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t rh = vaddl_s16(vget_high_s16(yy), vget_high_s16(vr)); \ 1351e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t gl = vsubl_s16(vget_low_s16(yy), vget_low_s16(gc)); \ 1361e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t gh = vsubl_s16(vget_high_s16(yy), vget_high_s16(gc)); \ 1371e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t bl = vaddl_s16(vget_low_s16(yy), vget_low_s16(ub)); \ 1381e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int32x4_t bh = vaddl_s16(vget_high_s16(yy), vget_high_s16(ub)); \ 1391e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 1401e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora rl = vmulq_lane_s32(rl, cf32, 0); \ 1411e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora rh = vmulq_lane_s32(rh, cf32, 0); \ 1421e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora gl = vmulq_lane_s32(gl, cf32, 0); \ 1431e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora gh = vmulq_lane_s32(gh, cf32, 0); \ 1441e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora bl = vmulq_lane_s32(bl, cf32, 0); \ 1451e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora bh = vmulq_lane_s32(bh, cf32, 0); \ 1461e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 1471e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora y = vqmovun_s16(vcombine_s16(vrshrn_n_s32(rl, 16), \ 1481e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vrshrn_n_s32(rh, 16))); \ 1491e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora u = vqmovun_s16(vcombine_s16(vrshrn_n_s32(gl, 16), \ 1501e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vrshrn_n_s32(gh, 16))); \ 1511e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora v = vqmovun_s16(vcombine_s16(vrshrn_n_s32(bl, 16), \ 1521e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vrshrn_n_s32(bh, 16))); \ 1531e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora STR_ ## FMT(out + off, y, u, v); \ 1541e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora } \ 1551e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} 1561e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 1571e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define v255 vmov_n_u8(255) 1581e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 1591e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define STR_Rgb(out, r, g, b) do { \ 1601e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const uint8x8x3_t r_g_b = {{ r, g, b }}; \ 1611e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vst3_u8(out, r_g_b); \ 1621e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} while (0) 1631e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 1641e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define STR_Bgr(out, r, g, b) do { \ 1651e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const uint8x8x3_t b_g_r = {{ b, g, r }}; \ 1661e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vst3_u8(out, b_g_r); \ 1671e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} while (0) 1681e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 1691e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define STR_Rgba(out, r, g, b) do { \ 1701e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const uint8x8x4_t r_g_b_v255 = {{ r, g, b, v255 }}; \ 1711e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vst4_u8(out, r_g_b_v255); \ 1721e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} while (0) 1731e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 1741e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define STR_Bgra(out, r, g, b) do { \ 1751e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const uint8x8x4_t b_g_r_v255 = {{ b, g, r, v255 }}; \ 1761e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora vst4_u8(out, b_g_r_v255); \ 1771e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} while (0) 1781e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 1791e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define CONVERT1(FMT, XSTEP, N, src_y, src_uv, rgb, cur_x) { \ 1801e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int i; \ 1811e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora for (i = 0; i < N; i++) { \ 1821e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int off = ((cur_x) + i) * XSTEP; \ 1831e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int y = src_y[(cur_x) + i]; \ 1841e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int u = (src_uv)[i]; \ 1851e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int v = (src_uv)[i + 16]; \ 1861e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora VP8YuvTo ## FMT(y, u, v, rgb + off); \ 1871e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora } \ 1881e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} 1891e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 1901e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define CONVERT2RGB_8(FMT, XSTEP, top_y, bottom_y, uv, \ 1911e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora top_dst, bottom_dst, cur_x, len) { \ 1921e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora if (top_y) { \ 1931e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora CONVERT8(FMT, XSTEP, len, top_y, uv, top_dst, cur_x) \ 1941e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora } \ 1951e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora if (bottom_y) { \ 1961e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora CONVERT8(FMT, XSTEP, len, bottom_y, (uv) + 32, bottom_dst, cur_x) \ 1971e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora } \ 1981e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} 1991e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 2001e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define CONVERT2RGB_1(FMT, XSTEP, top_y, bottom_y, uv, \ 2011e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora top_dst, bottom_dst, cur_x, len) { \ 2021e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora if (top_y) { \ 2031e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora CONVERT1(FMT, XSTEP, len, top_y, uv, top_dst, cur_x); \ 2041e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora } \ 2051e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora if (bottom_y) { \ 2061e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora CONVERT1(FMT, XSTEP, len, bottom_y, (uv) + 32, bottom_dst, cur_x); \ 2071e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora } \ 2081e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} 2091e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 2101e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#define NEON_UPSAMPLE_FUNC(FUNC_NAME, FMT, XSTEP) \ 2111e7bf8805bd030c19924a5306837ecd72c295751Vikas Arorastatic void FUNC_NAME(const uint8_t *top_y, const uint8_t *bottom_y, \ 2121e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const uint8_t *top_u, const uint8_t *top_v, \ 2131e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const uint8_t *cur_u, const uint8_t *cur_v, \ 2141e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint8_t *top_dst, uint8_t *bottom_dst, int len) { \ 2151e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int block; \ 2161e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora /* 16 byte aligned array to cache reconstructed u and v */ \ 2171e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint8_t uv_buf[2 * 32 + 15]; \ 2181e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint8_t *const r_uv = (uint8_t*)((uintptr_t)(uv_buf + 15) & ~15); \ 2191e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const int uv_len = (len + 1) >> 1; \ 2201e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora /* 9 pixels must be read-able for each block */ \ 2211e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const int num_blocks = (uv_len - 1) >> 3; \ 2221e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const int leftover = uv_len - num_blocks * 8; \ 2231e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const int last_pos = 1 + 16 * num_blocks; \ 2241e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 2251e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const int u_diag = ((top_u[0] + cur_u[0]) >> 1) + 1; \ 2261e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const int v_diag = ((top_v[0] + cur_v[0]) >> 1) + 1; \ 2271e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 2281e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const int16x4_t cf16 = vld1_s16(coef); \ 2291e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const int32x2_t cf32 = vmov_n_s32(CY); \ 2301e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const uint8x8_t u16 = vmov_n_u8(16); \ 2311e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const uint8x8_t u128 = vmov_n_u8(128); \ 2321e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 2331e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora /* Treat the first pixel in regular way */ \ 2341e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora if (top_y) { \ 2351e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const int u0 = (top_u[0] + u_diag) >> 1; \ 2361e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const int v0 = (top_v[0] + v_diag) >> 1; \ 2371e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora VP8YuvTo ## FMT(top_y[0], u0, v0, top_dst); \ 2381e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora } \ 2391e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora if (bottom_y) { \ 2401e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const int u0 = (cur_u[0] + u_diag) >> 1; \ 2411e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const int v0 = (cur_v[0] + v_diag) >> 1; \ 2421e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora VP8YuvTo ## FMT(bottom_y[0], u0, v0, bottom_dst); \ 2431e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora } \ 2441e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 2451e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora for (block = 0; block < num_blocks; ++block) { \ 2461e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora UPSAMPLE_16PIXELS(top_u, cur_u, r_uv); \ 2471e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora UPSAMPLE_16PIXELS(top_v, cur_v, r_uv + 16); \ 2481e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora CONVERT2RGB_8(FMT, XSTEP, top_y, bottom_y, r_uv, \ 2491e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora top_dst, bottom_dst, 16 * block + 1, 16); \ 2501e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora top_u += 8; \ 2511e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora cur_u += 8; \ 2521e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora top_v += 8; \ 2531e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora cur_v += 8; \ 2541e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora } \ 2551e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora \ 2561e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora UPSAMPLE_LAST_BLOCK(top_u, cur_u, leftover, r_uv); \ 2571e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora UPSAMPLE_LAST_BLOCK(top_v, cur_v, leftover, r_uv + 16); \ 2581e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora CONVERT2RGB_1(FMT, XSTEP, top_y, bottom_y, r_uv, \ 2591e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora top_dst, bottom_dst, last_pos, len - last_pos); \ 2601e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} 2611e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 2621e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// NEON variants of the fancy upsampler. 2631e7bf8805bd030c19924a5306837ecd72c295751Vikas AroraNEON_UPSAMPLE_FUNC(UpsampleRgbLinePairNEON, Rgb, 3) 2641e7bf8805bd030c19924a5306837ecd72c295751Vikas AroraNEON_UPSAMPLE_FUNC(UpsampleBgrLinePairNEON, Bgr, 3) 2651e7bf8805bd030c19924a5306837ecd72c295751Vikas AroraNEON_UPSAMPLE_FUNC(UpsampleRgbaLinePairNEON, Rgba, 4) 2661e7bf8805bd030c19924a5306837ecd72c295751Vikas AroraNEON_UPSAMPLE_FUNC(UpsampleBgraLinePairNEON, Bgra, 4) 2671e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 2681e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#endif // FANCY_UPSAMPLING 2691e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 2701e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#endif // WEBP_USE_NEON 2711e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 2721e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora//------------------------------------------------------------------------------ 2731e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 2741e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroraextern WebPUpsampleLinePairFunc WebPUpsamplers[/* MODE_LAST */]; 2751e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 2761e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroravoid WebPInitUpsamplersNEON(void) { 2771e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#if defined(WEBP_USE_NEON) 2781e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora WebPUpsamplers[MODE_RGB] = UpsampleRgbLinePairNEON; 2791e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora WebPUpsamplers[MODE_RGBA] = UpsampleRgbaLinePairNEON; 2801e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora WebPUpsamplers[MODE_BGR] = UpsampleBgrLinePairNEON; 2811e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora WebPUpsamplers[MODE_BGRA] = UpsampleBgraLinePairNEON; 2821e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#endif // WEBP_USE_NEON 2831e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} 2841e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 2851e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroravoid WebPInitPremultiplyNEON(void) { 2861e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#if defined(WEBP_USE_NEON) 2871e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora WebPUpsamplers[MODE_rgbA] = UpsampleRgbaLinePairNEON; 2881e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora WebPUpsamplers[MODE_bgrA] = UpsampleBgraLinePairNEON; 2891e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#endif // WEBP_USE_NEON 2901e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} 2911e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 2921e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#if defined(__cplusplus) || defined(c_plusplus) 2931e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} // extern "C" 2941e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#endif 295