Index: source/libvpx/vp8/encoder/arm/neon/fastquantizeb_neon.c |
=================================================================== |
--- source/libvpx/vp8/encoder/arm/neon/fastquantizeb_neon.c (revision 0) |
+++ source/libvpx/vp8/encoder/arm/neon/fastquantizeb_neon.c (working copy) |
@@ -0,0 +1,89 @@ |
+/* |
+ * 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> |
+#include "vp8/encoder/block.h" |
+ |
+static const uint16_t inv_zig_zag[16] = { |
+ 1, 2, 6, 7, |
+ 3, 5, 8, 13, |
+ 4, 9, 12, 14, |
+ 10, 11, 15, 16 |
+}; |
+ |
+void vp8_fast_quantize_b_neon(BLOCK *b, BLOCKD *d) { |
+ const int16x8_t one_q = vdupq_n_s16(-1), |
+ z0 = vld1q_s16(b->coeff), |
+ z1 = vld1q_s16(b->coeff + 8), |
+ round0 = vld1q_s16(b->round), |
+ round1 = vld1q_s16(b->round + 8), |
+ quant0 = vld1q_s16(b->quant_fast), |
+ quant1 = vld1q_s16(b->quant_fast + 8), |
+ dequant0 = vld1q_s16(d->dequant), |
+ dequant1 = vld1q_s16(d->dequant + 8); |
+ const uint16x8_t zig_zag0 = vld1q_u16(inv_zig_zag), |
+ zig_zag1 = vld1q_u16(inv_zig_zag + 8); |
+ int16x8_t x0, x1, sz0, sz1, y0, y1; |
+ uint16x8_t eob0, eob1; |
+ uint16x4_t eob_d16; |
+ uint32x2_t eob_d32; |
+ uint32x4_t eob_q32; |
+ |
+ /* sign of z: z >> 15 */ |
+ sz0 = vshrq_n_s16(z0, 15); |
+ sz1 = vshrq_n_s16(z1, 15); |
+ |
+ /* x = abs(z) */ |
+ x0 = vabsq_s16(z0); |
+ x1 = vabsq_s16(z1); |
+ |
+ /* x += round */ |
+ x0 = vaddq_s16(x0, round0); |
+ x1 = vaddq_s16(x1, round1); |
+ |
+ /* y = 2 * (x * quant) >> 16 */ |
+ y0 = vqdmulhq_s16(x0, quant0); |
+ y1 = vqdmulhq_s16(x1, quant1); |
+ |
+ /* Compensate for doubling in vqdmulhq */ |
+ y0 = vshrq_n_s16(y0, 1); |
+ y1 = vshrq_n_s16(y1, 1); |
+ |
+ /* Restore sign bit */ |
+ y0 = veorq_s16(y0, sz0); |
+ y1 = veorq_s16(y1, sz1); |
+ x0 = vsubq_s16(y0, sz0); |
+ x1 = vsubq_s16(y1, sz1); |
+ |
+ /* find non-zero elements */ |
+ eob0 = vtstq_s16(x0, one_q); |
+ eob1 = vtstq_s16(x1, one_q); |
+ |
+ /* mask zig zag */ |
+ eob0 = vandq_u16(eob0, zig_zag0); |
+ eob1 = vandq_u16(eob1, zig_zag1); |
+ |
+ /* select the largest value */ |
+ eob0 = vmaxq_u16(eob0, eob1); |
+ eob_d16 = vmax_u16(vget_low_u16(eob0), vget_high_u16(eob0)); |
+ eob_q32 = vmovl_u16(eob_d16); |
+ eob_d32 = vmax_u32(vget_low_u32(eob_q32), vget_high_u32(eob_q32)); |
+ eob_d32 = vpmax_u32(eob_d32, eob_d32); |
+ |
+ /* qcoeff = x */ |
+ vst1q_s16(d->qcoeff, x0); |
+ vst1q_s16(d->qcoeff + 8, x1); |
+ |
+ /* dqcoeff = x * dequant */ |
+ vst1q_s16(d->dqcoeff, vmulq_s16(dequant0, x0)); |
+ vst1q_s16(d->dqcoeff + 8, vmulq_s16(dequant1, x1)); |
+ |
+ vst1_lane_s8((int8_t *)d->eob, vreinterpret_s8_u32(eob_d32), 0); |
+} |