143 lines
4.5 KiB
C
143 lines
4.5 KiB
C
|
/*
|
||
|
|
* Copyright (c) 2014 The WebM project authors. All Rights Reserved.
|
||
|
|
*
|
||
|
|
* Use of this source code is governed by a BSD-style license
|
||
|
|
* that can be found in the LICENSE 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.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <arm_neon.h>
|
||
|
|
|
||
|
|
static const int16_t cospi8sqrt2minus1 = 20091;
|
||
|
|
static const int16_t sinpi8sqrt2 = 35468;
|
||
|
|
|
||
|
|
void vp8_dequant_idct_add_neon(
|
||
|
|
int16_t *input,
|
||
|
|
int16_t *dq,
|
||
|
|
unsigned char *dst,
|
||
|
|
int stride) {
|
||
|
|
unsigned char *dst0;
|
||
|
|
int32x2_t d14, d15;
|
||
|
|
int16x4_t d2, d3, d4, d5, d10, d11, d12, d13;
|
||
|
|
int16x8_t q1, q2, q3, q4, q5, q6;
|
||
|
|
int16x8_t qEmpty = vdupq_n_s16(0);
|
||
|
|
int32x2x2_t d2tmp0, d2tmp1;
|
||
|
|
int16x4x2_t d2tmp2, d2tmp3;
|
||
|
|
|
||
|
|
d14 = d15 = vdup_n_s32(0);
|
||
|
|
|
||
|
|
// load input
|
||
|
|
q3 = vld1q_s16(input);
|
||
|
|
vst1q_s16(input, qEmpty);
|
||
|
|
input += 8;
|
||
|
|
q4 = vld1q_s16(input);
|
||
|
|
vst1q_s16(input, qEmpty);
|
||
|
|
|
||
|
|
// load dq
|
||
|
|
q5 = vld1q_s16(dq);
|
||
|
|
dq += 8;
|
||
|
|
q6 = vld1q_s16(dq);
|
||
|
|
|
||
|
|
// load src from dst
|
||
|
|
dst0 = dst;
|
||
|
|
d14 = vld1_lane_s32((const int32_t *)dst0, d14, 0);
|
||
|
|
dst0 += stride;
|
||
|
|
d14 = vld1_lane_s32((const int32_t *)dst0, d14, 1);
|
||
|
|
dst0 += stride;
|
||
|
|
d15 = vld1_lane_s32((const int32_t *)dst0, d15, 0);
|
||
|
|
dst0 += stride;
|
||
|
|
d15 = vld1_lane_s32((const int32_t *)dst0, d15, 1);
|
||
|
|
|
||
|
|
q1 = vreinterpretq_s16_u16(vmulq_u16(vreinterpretq_u16_s16(q3),
|
||
|
|
vreinterpretq_u16_s16(q5)));
|
||
|
|
q2 = vreinterpretq_s16_u16(vmulq_u16(vreinterpretq_u16_s16(q4),
|
||
|
|
vreinterpretq_u16_s16(q6)));
|
||
|
|
|
||
|
|
d12 = vqadd_s16(vget_low_s16(q1), vget_low_s16(q2));
|
||
|
|
d13 = vqsub_s16(vget_low_s16(q1), vget_low_s16(q2));
|
||
|
|
|
||
|
|
q2 = vcombine_s16(vget_high_s16(q1), vget_high_s16(q2));
|
||
|
|
|
||
|
|
q3 = vqdmulhq_n_s16(q2, sinpi8sqrt2);
|
||
|
|
q4 = vqdmulhq_n_s16(q2, cospi8sqrt2minus1);
|
||
|
|
|
||
|
|
q3 = vshrq_n_s16(q3, 1);
|
||
|
|
q4 = vshrq_n_s16(q4, 1);
|
||
|
|
|
||
|
|
q3 = vqaddq_s16(q3, q2);
|
||
|
|
q4 = vqaddq_s16(q4, q2);
|
||
|
|
|
||
|
|
d10 = vqsub_s16(vget_low_s16(q3), vget_high_s16(q4));
|
||
|
|
d11 = vqadd_s16(vget_high_s16(q3), vget_low_s16(q4));
|
||
|
|
|
||
|
|
d2 = vqadd_s16(d12, d11);
|
||
|
|
d3 = vqadd_s16(d13, d10);
|
||
|
|
d4 = vqsub_s16(d13, d10);
|
||
|
|
d5 = vqsub_s16(d12, d11);
|
||
|
|
|
||
|
|
d2tmp0 = vtrn_s32(vreinterpret_s32_s16(d2), vreinterpret_s32_s16(d4));
|
||
|
|
d2tmp1 = vtrn_s32(vreinterpret_s32_s16(d3), vreinterpret_s32_s16(d5));
|
||
|
|
d2tmp2 = vtrn_s16(vreinterpret_s16_s32(d2tmp0.val[0]),
|
||
|
|
vreinterpret_s16_s32(d2tmp1.val[0]));
|
||
|
|
d2tmp3 = vtrn_s16(vreinterpret_s16_s32(d2tmp0.val[1]),
|
||
|
|
vreinterpret_s16_s32(d2tmp1.val[1]));
|
||
|
|
|
||
|
|
// loop 2
|
||
|
|
q2 = vcombine_s16(d2tmp2.val[1], d2tmp3.val[1]);
|
||
|
|
|
||
|
|
q3 = vqdmulhq_n_s16(q2, sinpi8sqrt2);
|
||
|
|
q4 = vqdmulhq_n_s16(q2, cospi8sqrt2minus1);
|
||
|
|
|
||
|
|
d12 = vqadd_s16(d2tmp2.val[0], d2tmp3.val[0]);
|
||
|
|
d13 = vqsub_s16(d2tmp2.val[0], d2tmp3.val[0]);
|
||
|
|
|
||
|
|
q3 = vshrq_n_s16(q3, 1);
|
||
|
|
q4 = vshrq_n_s16(q4, 1);
|
||
|
|
|
||
|
|
q3 = vqaddq_s16(q3, q2);
|
||
|
|
q4 = vqaddq_s16(q4, q2);
|
||
|
|
|
||
|
|
d10 = vqsub_s16(vget_low_s16(q3), vget_high_s16(q4));
|
||
|
|
d11 = vqadd_s16(vget_high_s16(q3), vget_low_s16(q4));
|
||
|
|
|
||
|
|
d2 = vqadd_s16(d12, d11);
|
||
|
|
d3 = vqadd_s16(d13, d10);
|
||
|
|
d4 = vqsub_s16(d13, d10);
|
||
|
|
d5 = vqsub_s16(d12, d11);
|
||
|
|
|
||
|
|
d2 = vrshr_n_s16(d2, 3);
|
||
|
|
d3 = vrshr_n_s16(d3, 3);
|
||
|
|
d4 = vrshr_n_s16(d4, 3);
|
||
|
|
d5 = vrshr_n_s16(d5, 3);
|
||
|
|
|
||
|
|
d2tmp0 = vtrn_s32(vreinterpret_s32_s16(d2), vreinterpret_s32_s16(d4));
|
||
|
|
d2tmp1 = vtrn_s32(vreinterpret_s32_s16(d3), vreinterpret_s32_s16(d5));
|
||
|
|
d2tmp2 = vtrn_s16(vreinterpret_s16_s32(d2tmp0.val[0]),
|
||
|
|
vreinterpret_s16_s32(d2tmp1.val[0]));
|
||
|
|
d2tmp3 = vtrn_s16(vreinterpret_s16_s32(d2tmp0.val[1]),
|
||
|
|
vreinterpret_s16_s32(d2tmp1.val[1]));
|
||
|
|
|
||
|
|
q1 = vcombine_s16(d2tmp2.val[0], d2tmp2.val[1]);
|
||
|
|
q2 = vcombine_s16(d2tmp3.val[0], d2tmp3.val[1]);
|
||
|
|
|
||
|
|
q1 = vreinterpretq_s16_u16(vaddw_u8(vreinterpretq_u16_s16(q1),
|
||
|
|
vreinterpret_u8_s32(d14)));
|
||
|
|
q2 = vreinterpretq_s16_u16(vaddw_u8(vreinterpretq_u16_s16(q2),
|
||
|
|
vreinterpret_u8_s32(d15)));
|
||
|
|
|
||
|
|
d14 = vreinterpret_s32_u8(vqmovun_s16(q1));
|
||
|
|
d15 = vreinterpret_s32_u8(vqmovun_s16(q2));
|
||
|
|
|
||
|
|
dst0 = dst;
|
||
|
|
vst1_lane_s32((int32_t *)dst0, d14, 0);
|
||
|
|
dst0 += stride;
|
||
|
|
vst1_lane_s32((int32_t *)dst0, d14, 1);
|
||
|
|
dst0 += stride;
|
||
|
|
vst1_lane_s32((int32_t *)dst0, d15, 0);
|
||
|
|
dst0 += stride;
|
||
|
|
vst1_lane_s32((int32_t *)dst0, d15, 1);
|
||
|
|
return;
|
||
|
|
}
|