1b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian/* 2b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * Copyright (c) 2014 The WebM project authors. All Rights Reserved. 3b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * 4b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * Use of this source code is governed by a BSD-style license 5b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * that can be found in the LICENSE file in the root of the source 6b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * tree. An additional intellectual property rights grant can be found 7b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * in the file PATENTS. All contributing project authors may 8b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * be found in the AUTHORS file in the root of the source tree. 9b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian */ 10b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 11b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian#include <arm_neon.h> 12b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 13b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanianstatic const int16_t cospi8sqrt2minus1 = 20091; 14b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanianstatic const int16_t sinpi8sqrt2 = 35468; 15b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 16b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanianvoid vp8_dequant_idct_add_neon( 17b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian int16_t *input, 18b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian int16_t *dq, 19b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian unsigned char *dst, 20b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian int stride) { 21b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian unsigned char *dst0; 22b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian int32x2_t d14, d15; 23b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian int16x4_t d2, d3, d4, d5, d10, d11, d12, d13; 24b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian int16x8_t q1, q2, q3, q4, q5, q6; 25b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian int16x8_t qEmpty = vdupq_n_s16(0); 26b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian int32x2x2_t d2tmp0, d2tmp1; 27b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian int16x4x2_t d2tmp2, d2tmp3; 28b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 29b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d14 = d15 = vdup_n_s32(0); 30b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 31b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian // load input 32b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q3 = vld1q_s16(input); 33b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vst1q_s16(input, qEmpty); 34b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian input += 8; 35b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q4 = vld1q_s16(input); 36b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vst1q_s16(input, qEmpty); 37b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 38b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian // load dq 39b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q5 = vld1q_s16(dq); 40b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian dq += 8; 41b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q6 = vld1q_s16(dq); 42b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 43b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian // load src from dst 44b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian dst0 = dst; 45b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d14 = vld1_lane_s32((const int32_t *)dst0, d14, 0); 46b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian dst0 += stride; 47b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d14 = vld1_lane_s32((const int32_t *)dst0, d14, 1); 48b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian dst0 += stride; 49b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d15 = vld1_lane_s32((const int32_t *)dst0, d15, 0); 50b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian dst0 += stride; 51b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d15 = vld1_lane_s32((const int32_t *)dst0, d15, 1); 52b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 53b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q1 = vreinterpretq_s16_u16(vmulq_u16(vreinterpretq_u16_s16(q3), 54b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vreinterpretq_u16_s16(q5))); 55b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q2 = vreinterpretq_s16_u16(vmulq_u16(vreinterpretq_u16_s16(q4), 56b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vreinterpretq_u16_s16(q6))); 57b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 58b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d12 = vqadd_s16(vget_low_s16(q1), vget_low_s16(q2)); 59b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d13 = vqsub_s16(vget_low_s16(q1), vget_low_s16(q2)); 60b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 61b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q2 = vcombine_s16(vget_high_s16(q1), vget_high_s16(q2)); 62b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 63b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q3 = vqdmulhq_n_s16(q2, sinpi8sqrt2); 64b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q4 = vqdmulhq_n_s16(q2, cospi8sqrt2minus1); 65b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 66b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q3 = vshrq_n_s16(q3, 1); 67b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q4 = vshrq_n_s16(q4, 1); 68b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 69b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q3 = vqaddq_s16(q3, q2); 70b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q4 = vqaddq_s16(q4, q2); 71b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 72b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d10 = vqsub_s16(vget_low_s16(q3), vget_high_s16(q4)); 73b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d11 = vqadd_s16(vget_high_s16(q3), vget_low_s16(q4)); 74b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 75b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d2 = vqadd_s16(d12, d11); 76b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d3 = vqadd_s16(d13, d10); 77b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d4 = vqsub_s16(d13, d10); 78b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d5 = vqsub_s16(d12, d11); 79b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 80b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d2tmp0 = vtrn_s32(vreinterpret_s32_s16(d2), vreinterpret_s32_s16(d4)); 81b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d2tmp1 = vtrn_s32(vreinterpret_s32_s16(d3), vreinterpret_s32_s16(d5)); 82b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d2tmp2 = vtrn_s16(vreinterpret_s16_s32(d2tmp0.val[0]), 83b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vreinterpret_s16_s32(d2tmp1.val[0])); 84b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d2tmp3 = vtrn_s16(vreinterpret_s16_s32(d2tmp0.val[1]), 85b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vreinterpret_s16_s32(d2tmp1.val[1])); 86b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 87b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian // loop 2 88b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q2 = vcombine_s16(d2tmp2.val[1], d2tmp3.val[1]); 89b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 90b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q3 = vqdmulhq_n_s16(q2, sinpi8sqrt2); 91b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q4 = vqdmulhq_n_s16(q2, cospi8sqrt2minus1); 92b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 93b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d12 = vqadd_s16(d2tmp2.val[0], d2tmp3.val[0]); 94b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d13 = vqsub_s16(d2tmp2.val[0], d2tmp3.val[0]); 95b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 96b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q3 = vshrq_n_s16(q3, 1); 97b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q4 = vshrq_n_s16(q4, 1); 98b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 99b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q3 = vqaddq_s16(q3, q2); 100b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q4 = vqaddq_s16(q4, q2); 101b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 102b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d10 = vqsub_s16(vget_low_s16(q3), vget_high_s16(q4)); 103b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d11 = vqadd_s16(vget_high_s16(q3), vget_low_s16(q4)); 104b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 105b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d2 = vqadd_s16(d12, d11); 106b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d3 = vqadd_s16(d13, d10); 107b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d4 = vqsub_s16(d13, d10); 108b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d5 = vqsub_s16(d12, d11); 109b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 110b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d2 = vrshr_n_s16(d2, 3); 111b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d3 = vrshr_n_s16(d3, 3); 112b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d4 = vrshr_n_s16(d4, 3); 113b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d5 = vrshr_n_s16(d5, 3); 114b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 115b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d2tmp0 = vtrn_s32(vreinterpret_s32_s16(d2), vreinterpret_s32_s16(d4)); 116b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d2tmp1 = vtrn_s32(vreinterpret_s32_s16(d3), vreinterpret_s32_s16(d5)); 117b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d2tmp2 = vtrn_s16(vreinterpret_s16_s32(d2tmp0.val[0]), 118b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vreinterpret_s16_s32(d2tmp1.val[0])); 119b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d2tmp3 = vtrn_s16(vreinterpret_s16_s32(d2tmp0.val[1]), 120b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vreinterpret_s16_s32(d2tmp1.val[1])); 121b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 122b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q1 = vcombine_s16(d2tmp2.val[0], d2tmp2.val[1]); 123b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q2 = vcombine_s16(d2tmp3.val[0], d2tmp3.val[1]); 124b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 125b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q1 = vreinterpretq_s16_u16(vaddw_u8(vreinterpretq_u16_s16(q1), 126b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vreinterpret_u8_s32(d14))); 127b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian q2 = vreinterpretq_s16_u16(vaddw_u8(vreinterpretq_u16_s16(q2), 128b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vreinterpret_u8_s32(d15))); 129b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 130b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d14 = vreinterpret_s32_u8(vqmovun_s16(q1)); 131b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian d15 = vreinterpret_s32_u8(vqmovun_s16(q2)); 132b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 133b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian dst0 = dst; 134b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vst1_lane_s32((int32_t *)dst0, d14, 0); 135b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian dst0 += stride; 136b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vst1_lane_s32((int32_t *)dst0, d14, 1); 137b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian dst0 += stride; 138b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vst1_lane_s32((int32_t *)dst0, d15, 0); 139b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian dst0 += stride; 140b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian vst1_lane_s32((int32_t *)dst0, d15, 1); 141b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian return; 142b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian} 143