зеркало из https://github.com/mozilla/gecko-dev.git
1302 строки
49 KiB
C++
1302 строки
49 KiB
C++
/*
|
|
* Copyright 2011 The LibYuv 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 "libyuv/rotate.h"
|
|
|
|
#include "libyuv/cpu_id.h"
|
|
#include "libyuv/convert.h"
|
|
#include "libyuv/planar_functions.h"
|
|
#include "libyuv/row.h"
|
|
|
|
#ifdef __cplusplus
|
|
namespace libyuv {
|
|
extern "C" {
|
|
#endif
|
|
|
|
#if !defined(LIBYUV_DISABLE_X86) && \
|
|
(defined(_M_IX86) || defined(__x86_64__) || defined(__i386__))
|
|
#if defined(__APPLE__) && defined(__i386__)
|
|
#define DECLARE_FUNCTION(name) \
|
|
".text \n" \
|
|
".private_extern _" #name " \n" \
|
|
".align 4,0x90 \n" \
|
|
"_" #name ": \n"
|
|
#elif defined(__MINGW32__) || defined(__CYGWIN__) && defined(__i386__)
|
|
#define DECLARE_FUNCTION(name) \
|
|
".text \n" \
|
|
".align 4,0x90 \n" \
|
|
"_" #name ": \n"
|
|
#else
|
|
#define DECLARE_FUNCTION(name) \
|
|
".text \n" \
|
|
".align 4,0x90 \n" \
|
|
#name ": \n"
|
|
#endif
|
|
#endif
|
|
|
|
#if !defined(LIBYUV_DISABLE_NEON) && !defined(__native_client__) && \
|
|
(defined(__ARM_NEON__) || defined(LIBYUV_NEON))
|
|
#define HAS_MIRRORROW_NEON
|
|
void MirrorRow_NEON(const uint8* src, uint8* dst, int width);
|
|
#define HAS_MIRRORROW_UV_NEON
|
|
void MirrorUVRow_NEON(const uint8* src, uint8* dst_a, uint8* dst_b, int width);
|
|
#define HAS_TRANSPOSE_WX8_NEON
|
|
void TransposeWx8_NEON(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride, int width);
|
|
#define HAS_TRANSPOSE_UVWX8_NEON
|
|
void TransposeUVWx8_NEON(const uint8* src, int src_stride,
|
|
uint8* dst_a, int dst_stride_a,
|
|
uint8* dst_b, int dst_stride_b,
|
|
int width);
|
|
#endif // defined(__ARM_NEON__)
|
|
|
|
#if !defined(LIBYUV_DISABLE_MIPS) && !defined(__native_client__) && \
|
|
defined(__mips__) && \
|
|
defined(__mips_dsp) && (__mips_dsp_rev >= 2)
|
|
#define HAS_TRANSPOSE_WX8_MIPS_DSPR2
|
|
void TransposeWx8_MIPS_DSPR2(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride, int width);
|
|
|
|
void TransposeWx8_FAST_MIPS_DSPR2(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride, int width);
|
|
#define HAS_TRANSPOSE_UVWx8_MIPS_DSPR2
|
|
void TransposeUVWx8_MIPS_DSPR2(const uint8* src, int src_stride,
|
|
uint8* dst_a, int dst_stride_a,
|
|
uint8* dst_b, int dst_stride_b,
|
|
int width);
|
|
#endif // defined(__mips__)
|
|
|
|
#if !defined(LIBYUV_DISABLE_X86) && \
|
|
defined(_M_IX86) && defined(_MSC_VER)
|
|
#define HAS_TRANSPOSE_WX8_SSSE3
|
|
__declspec(naked)
|
|
static void TransposeWx8_SSSE3(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride, int width) {
|
|
__asm {
|
|
push edi
|
|
push esi
|
|
push ebp
|
|
mov eax, [esp + 12 + 4] // src
|
|
mov edi, [esp + 12 + 8] // src_stride
|
|
mov edx, [esp + 12 + 12] // dst
|
|
mov esi, [esp + 12 + 16] // dst_stride
|
|
mov ecx, [esp + 12 + 20] // width
|
|
|
|
// Read in the data from the source pointer.
|
|
// First round of bit swap.
|
|
align 4
|
|
convertloop:
|
|
movq xmm0, qword ptr [eax]
|
|
lea ebp, [eax + 8]
|
|
movq xmm1, qword ptr [eax + edi]
|
|
lea eax, [eax + 2 * edi]
|
|
punpcklbw xmm0, xmm1
|
|
movq xmm2, qword ptr [eax]
|
|
movdqa xmm1, xmm0
|
|
palignr xmm1, xmm1, 8
|
|
movq xmm3, qword ptr [eax + edi]
|
|
lea eax, [eax + 2 * edi]
|
|
punpcklbw xmm2, xmm3
|
|
movdqa xmm3, xmm2
|
|
movq xmm4, qword ptr [eax]
|
|
palignr xmm3, xmm3, 8
|
|
movq xmm5, qword ptr [eax + edi]
|
|
punpcklbw xmm4, xmm5
|
|
lea eax, [eax + 2 * edi]
|
|
movdqa xmm5, xmm4
|
|
movq xmm6, qword ptr [eax]
|
|
palignr xmm5, xmm5, 8
|
|
movq xmm7, qword ptr [eax + edi]
|
|
punpcklbw xmm6, xmm7
|
|
mov eax, ebp
|
|
movdqa xmm7, xmm6
|
|
palignr xmm7, xmm7, 8
|
|
// Second round of bit swap.
|
|
punpcklwd xmm0, xmm2
|
|
punpcklwd xmm1, xmm3
|
|
movdqa xmm2, xmm0
|
|
movdqa xmm3, xmm1
|
|
palignr xmm2, xmm2, 8
|
|
palignr xmm3, xmm3, 8
|
|
punpcklwd xmm4, xmm6
|
|
punpcklwd xmm5, xmm7
|
|
movdqa xmm6, xmm4
|
|
movdqa xmm7, xmm5
|
|
palignr xmm6, xmm6, 8
|
|
palignr xmm7, xmm7, 8
|
|
// Third round of bit swap.
|
|
// Write to the destination pointer.
|
|
punpckldq xmm0, xmm4
|
|
movq qword ptr [edx], xmm0
|
|
movdqa xmm4, xmm0
|
|
palignr xmm4, xmm4, 8
|
|
movq qword ptr [edx + esi], xmm4
|
|
lea edx, [edx + 2 * esi]
|
|
punpckldq xmm2, xmm6
|
|
movdqa xmm6, xmm2
|
|
palignr xmm6, xmm6, 8
|
|
movq qword ptr [edx], xmm2
|
|
punpckldq xmm1, xmm5
|
|
movq qword ptr [edx + esi], xmm6
|
|
lea edx, [edx + 2 * esi]
|
|
movdqa xmm5, xmm1
|
|
movq qword ptr [edx], xmm1
|
|
palignr xmm5, xmm5, 8
|
|
punpckldq xmm3, xmm7
|
|
movq qword ptr [edx + esi], xmm5
|
|
lea edx, [edx + 2 * esi]
|
|
movq qword ptr [edx], xmm3
|
|
movdqa xmm7, xmm3
|
|
palignr xmm7, xmm7, 8
|
|
sub ecx, 8
|
|
movq qword ptr [edx + esi], xmm7
|
|
lea edx, [edx + 2 * esi]
|
|
jg convertloop
|
|
|
|
pop ebp
|
|
pop esi
|
|
pop edi
|
|
ret
|
|
}
|
|
}
|
|
|
|
#define HAS_TRANSPOSE_UVWX8_SSE2
|
|
__declspec(naked)
|
|
static void TransposeUVWx8_SSE2(const uint8* src, int src_stride,
|
|
uint8* dst_a, int dst_stride_a,
|
|
uint8* dst_b, int dst_stride_b,
|
|
int w) {
|
|
__asm {
|
|
push ebx
|
|
push esi
|
|
push edi
|
|
push ebp
|
|
mov eax, [esp + 16 + 4] // src
|
|
mov edi, [esp + 16 + 8] // src_stride
|
|
mov edx, [esp + 16 + 12] // dst_a
|
|
mov esi, [esp + 16 + 16] // dst_stride_a
|
|
mov ebx, [esp + 16 + 20] // dst_b
|
|
mov ebp, [esp + 16 + 24] // dst_stride_b
|
|
mov ecx, esp
|
|
sub esp, 4 + 16
|
|
and esp, ~15
|
|
mov [esp + 16], ecx
|
|
mov ecx, [ecx + 16 + 28] // w
|
|
|
|
align 4
|
|
convertloop:
|
|
// Read in the data from the source pointer.
|
|
// First round of bit swap.
|
|
movdqa xmm0, [eax]
|
|
movdqa xmm1, [eax + edi]
|
|
lea eax, [eax + 2 * edi]
|
|
movdqa xmm7, xmm0 // use xmm7 as temp register.
|
|
punpcklbw xmm0, xmm1
|
|
punpckhbw xmm7, xmm1
|
|
movdqa xmm1, xmm7
|
|
movdqa xmm2, [eax]
|
|
movdqa xmm3, [eax + edi]
|
|
lea eax, [eax + 2 * edi]
|
|
movdqa xmm7, xmm2
|
|
punpcklbw xmm2, xmm3
|
|
punpckhbw xmm7, xmm3
|
|
movdqa xmm3, xmm7
|
|
movdqa xmm4, [eax]
|
|
movdqa xmm5, [eax + edi]
|
|
lea eax, [eax + 2 * edi]
|
|
movdqa xmm7, xmm4
|
|
punpcklbw xmm4, xmm5
|
|
punpckhbw xmm7, xmm5
|
|
movdqa xmm5, xmm7
|
|
movdqa xmm6, [eax]
|
|
movdqa xmm7, [eax + edi]
|
|
lea eax, [eax + 2 * edi]
|
|
movdqa [esp], xmm5 // backup xmm5
|
|
neg edi
|
|
movdqa xmm5, xmm6 // use xmm5 as temp register.
|
|
punpcklbw xmm6, xmm7
|
|
punpckhbw xmm5, xmm7
|
|
movdqa xmm7, xmm5
|
|
lea eax, [eax + 8 * edi + 16]
|
|
neg edi
|
|
// Second round of bit swap.
|
|
movdqa xmm5, xmm0
|
|
punpcklwd xmm0, xmm2
|
|
punpckhwd xmm5, xmm2
|
|
movdqa xmm2, xmm5
|
|
movdqa xmm5, xmm1
|
|
punpcklwd xmm1, xmm3
|
|
punpckhwd xmm5, xmm3
|
|
movdqa xmm3, xmm5
|
|
movdqa xmm5, xmm4
|
|
punpcklwd xmm4, xmm6
|
|
punpckhwd xmm5, xmm6
|
|
movdqa xmm6, xmm5
|
|
movdqa xmm5, [esp] // restore xmm5
|
|
movdqa [esp], xmm6 // backup xmm6
|
|
movdqa xmm6, xmm5 // use xmm6 as temp register.
|
|
punpcklwd xmm5, xmm7
|
|
punpckhwd xmm6, xmm7
|
|
movdqa xmm7, xmm6
|
|
// Third round of bit swap.
|
|
// Write to the destination pointer.
|
|
movdqa xmm6, xmm0
|
|
punpckldq xmm0, xmm4
|
|
punpckhdq xmm6, xmm4
|
|
movdqa xmm4, xmm6
|
|
movdqa xmm6, [esp] // restore xmm6
|
|
movlpd qword ptr [edx], xmm0
|
|
movhpd qword ptr [ebx], xmm0
|
|
movlpd qword ptr [edx + esi], xmm4
|
|
lea edx, [edx + 2 * esi]
|
|
movhpd qword ptr [ebx + ebp], xmm4
|
|
lea ebx, [ebx + 2 * ebp]
|
|
movdqa xmm0, xmm2 // use xmm0 as the temp register.
|
|
punpckldq xmm2, xmm6
|
|
movlpd qword ptr [edx], xmm2
|
|
movhpd qword ptr [ebx], xmm2
|
|
punpckhdq xmm0, xmm6
|
|
movlpd qword ptr [edx + esi], xmm0
|
|
lea edx, [edx + 2 * esi]
|
|
movhpd qword ptr [ebx + ebp], xmm0
|
|
lea ebx, [ebx + 2 * ebp]
|
|
movdqa xmm0, xmm1 // use xmm0 as the temp register.
|
|
punpckldq xmm1, xmm5
|
|
movlpd qword ptr [edx], xmm1
|
|
movhpd qword ptr [ebx], xmm1
|
|
punpckhdq xmm0, xmm5
|
|
movlpd qword ptr [edx + esi], xmm0
|
|
lea edx, [edx + 2 * esi]
|
|
movhpd qword ptr [ebx + ebp], xmm0
|
|
lea ebx, [ebx + 2 * ebp]
|
|
movdqa xmm0, xmm3 // use xmm0 as the temp register.
|
|
punpckldq xmm3, xmm7
|
|
movlpd qword ptr [edx], xmm3
|
|
movhpd qword ptr [ebx], xmm3
|
|
punpckhdq xmm0, xmm7
|
|
sub ecx, 8
|
|
movlpd qword ptr [edx + esi], xmm0
|
|
lea edx, [edx + 2 * esi]
|
|
movhpd qword ptr [ebx + ebp], xmm0
|
|
lea ebx, [ebx + 2 * ebp]
|
|
jg convertloop
|
|
|
|
mov esp, [esp + 16]
|
|
pop ebp
|
|
pop edi
|
|
pop esi
|
|
pop ebx
|
|
ret
|
|
}
|
|
}
|
|
#elif !defined(LIBYUV_DISABLE_X86) && \
|
|
(defined(__i386__) || (defined(__x86_64__) && !defined(__native_client__)))
|
|
#define HAS_TRANSPOSE_WX8_SSSE3
|
|
static void TransposeWx8_SSSE3(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride, int width) {
|
|
asm volatile (
|
|
// Read in the data from the source pointer.
|
|
// First round of bit swap.
|
|
".p2align 2 \n"
|
|
"1: \n"
|
|
"movq (%0),%%xmm0 \n"
|
|
"movq (%0,%3),%%xmm1 \n"
|
|
"lea (%0,%3,2),%0 \n"
|
|
"punpcklbw %%xmm1,%%xmm0 \n"
|
|
"movq (%0),%%xmm2 \n"
|
|
"movdqa %%xmm0,%%xmm1 \n"
|
|
"palignr $0x8,%%xmm1,%%xmm1 \n"
|
|
"movq (%0,%3),%%xmm3 \n"
|
|
"lea (%0,%3,2),%0 \n"
|
|
"punpcklbw %%xmm3,%%xmm2 \n"
|
|
"movdqa %%xmm2,%%xmm3 \n"
|
|
"movq (%0),%%xmm4 \n"
|
|
"palignr $0x8,%%xmm3,%%xmm3 \n"
|
|
"movq (%0,%3),%%xmm5 \n"
|
|
"lea (%0,%3,2),%0 \n"
|
|
"punpcklbw %%xmm5,%%xmm4 \n"
|
|
"movdqa %%xmm4,%%xmm5 \n"
|
|
"movq (%0),%%xmm6 \n"
|
|
"palignr $0x8,%%xmm5,%%xmm5 \n"
|
|
"movq (%0,%3),%%xmm7 \n"
|
|
"lea (%0,%3,2),%0 \n"
|
|
"punpcklbw %%xmm7,%%xmm6 \n"
|
|
"neg %3 \n"
|
|
"movdqa %%xmm6,%%xmm7 \n"
|
|
"lea 0x8(%0,%3,8),%0 \n"
|
|
"palignr $0x8,%%xmm7,%%xmm7 \n"
|
|
"neg %3 \n"
|
|
// Second round of bit swap.
|
|
"punpcklwd %%xmm2,%%xmm0 \n"
|
|
"punpcklwd %%xmm3,%%xmm1 \n"
|
|
"movdqa %%xmm0,%%xmm2 \n"
|
|
"movdqa %%xmm1,%%xmm3 \n"
|
|
"palignr $0x8,%%xmm2,%%xmm2 \n"
|
|
"palignr $0x8,%%xmm3,%%xmm3 \n"
|
|
"punpcklwd %%xmm6,%%xmm4 \n"
|
|
"punpcklwd %%xmm7,%%xmm5 \n"
|
|
"movdqa %%xmm4,%%xmm6 \n"
|
|
"movdqa %%xmm5,%%xmm7 \n"
|
|
"palignr $0x8,%%xmm6,%%xmm6 \n"
|
|
"palignr $0x8,%%xmm7,%%xmm7 \n"
|
|
// Third round of bit swap.
|
|
// Write to the destination pointer.
|
|
"punpckldq %%xmm4,%%xmm0 \n"
|
|
"movq %%xmm0,(%1) \n"
|
|
"movdqa %%xmm0,%%xmm4 \n"
|
|
"palignr $0x8,%%xmm4,%%xmm4 \n"
|
|
"movq %%xmm4,(%1,%4) \n"
|
|
"lea (%1,%4,2),%1 \n"
|
|
"punpckldq %%xmm6,%%xmm2 \n"
|
|
"movdqa %%xmm2,%%xmm6 \n"
|
|
"movq %%xmm2,(%1) \n"
|
|
"palignr $0x8,%%xmm6,%%xmm6 \n"
|
|
"punpckldq %%xmm5,%%xmm1 \n"
|
|
"movq %%xmm6,(%1,%4) \n"
|
|
"lea (%1,%4,2),%1 \n"
|
|
"movdqa %%xmm1,%%xmm5 \n"
|
|
"movq %%xmm1,(%1) \n"
|
|
"palignr $0x8,%%xmm5,%%xmm5 \n"
|
|
"movq %%xmm5,(%1,%4) \n"
|
|
"lea (%1,%4,2),%1 \n"
|
|
"punpckldq %%xmm7,%%xmm3 \n"
|
|
"movq %%xmm3,(%1) \n"
|
|
"movdqa %%xmm3,%%xmm7 \n"
|
|
"palignr $0x8,%%xmm7,%%xmm7 \n"
|
|
"sub $0x8,%2 \n"
|
|
"movq %%xmm7,(%1,%4) \n"
|
|
"lea (%1,%4,2),%1 \n"
|
|
"jg 1b \n"
|
|
: "+r"(src), // %0
|
|
"+r"(dst), // %1
|
|
"+r"(width) // %2
|
|
: "r"((intptr_t)(src_stride)), // %3
|
|
"r"((intptr_t)(dst_stride)) // %4
|
|
: "memory", "cc"
|
|
#if defined(__SSE2__)
|
|
, "xmm0", "xmm1", "xmm2", "xmm3", "xmm4", "xmm5", "xmm6", "xmm7"
|
|
#endif
|
|
);
|
|
}
|
|
|
|
#if !defined(LIBYUV_DISABLE_X86) && defined(__i386__)
|
|
#define HAS_TRANSPOSE_UVWX8_SSE2
|
|
extern "C" void TransposeUVWx8_SSE2(const uint8* src, int src_stride,
|
|
uint8* dst_a, int dst_stride_a,
|
|
uint8* dst_b, int dst_stride_b,
|
|
int w);
|
|
asm (
|
|
DECLARE_FUNCTION(TransposeUVWx8_SSE2)
|
|
"push %ebx \n"
|
|
"push %esi \n"
|
|
"push %edi \n"
|
|
"push %ebp \n"
|
|
"mov 0x14(%esp),%eax \n"
|
|
"mov 0x18(%esp),%edi \n"
|
|
"mov 0x1c(%esp),%edx \n"
|
|
"mov 0x20(%esp),%esi \n"
|
|
"mov 0x24(%esp),%ebx \n"
|
|
"mov 0x28(%esp),%ebp \n"
|
|
"mov %esp,%ecx \n"
|
|
"sub $0x14,%esp \n"
|
|
"and $0xfffffff0,%esp \n"
|
|
"mov %ecx,0x10(%esp) \n"
|
|
"mov 0x2c(%ecx),%ecx \n"
|
|
|
|
"1: \n"
|
|
"movdqa (%eax),%xmm0 \n"
|
|
"movdqa (%eax,%edi,1),%xmm1 \n"
|
|
"lea (%eax,%edi,2),%eax \n"
|
|
"movdqa %xmm0,%xmm7 \n"
|
|
"punpcklbw %xmm1,%xmm0 \n"
|
|
"punpckhbw %xmm1,%xmm7 \n"
|
|
"movdqa %xmm7,%xmm1 \n"
|
|
"movdqa (%eax),%xmm2 \n"
|
|
"movdqa (%eax,%edi,1),%xmm3 \n"
|
|
"lea (%eax,%edi,2),%eax \n"
|
|
"movdqa %xmm2,%xmm7 \n"
|
|
"punpcklbw %xmm3,%xmm2 \n"
|
|
"punpckhbw %xmm3,%xmm7 \n"
|
|
"movdqa %xmm7,%xmm3 \n"
|
|
"movdqa (%eax),%xmm4 \n"
|
|
"movdqa (%eax,%edi,1),%xmm5 \n"
|
|
"lea (%eax,%edi,2),%eax \n"
|
|
"movdqa %xmm4,%xmm7 \n"
|
|
"punpcklbw %xmm5,%xmm4 \n"
|
|
"punpckhbw %xmm5,%xmm7 \n"
|
|
"movdqa %xmm7,%xmm5 \n"
|
|
"movdqa (%eax),%xmm6 \n"
|
|
"movdqa (%eax,%edi,1),%xmm7 \n"
|
|
"lea (%eax,%edi,2),%eax \n"
|
|
"movdqa %xmm5,(%esp) \n"
|
|
"neg %edi \n"
|
|
"movdqa %xmm6,%xmm5 \n"
|
|
"punpcklbw %xmm7,%xmm6 \n"
|
|
"punpckhbw %xmm7,%xmm5 \n"
|
|
"movdqa %xmm5,%xmm7 \n"
|
|
"lea 0x10(%eax,%edi,8),%eax \n"
|
|
"neg %edi \n"
|
|
"movdqa %xmm0,%xmm5 \n"
|
|
"punpcklwd %xmm2,%xmm0 \n"
|
|
"punpckhwd %xmm2,%xmm5 \n"
|
|
"movdqa %xmm5,%xmm2 \n"
|
|
"movdqa %xmm1,%xmm5 \n"
|
|
"punpcklwd %xmm3,%xmm1 \n"
|
|
"punpckhwd %xmm3,%xmm5 \n"
|
|
"movdqa %xmm5,%xmm3 \n"
|
|
"movdqa %xmm4,%xmm5 \n"
|
|
"punpcklwd %xmm6,%xmm4 \n"
|
|
"punpckhwd %xmm6,%xmm5 \n"
|
|
"movdqa %xmm5,%xmm6 \n"
|
|
"movdqa (%esp),%xmm5 \n"
|
|
"movdqa %xmm6,(%esp) \n"
|
|
"movdqa %xmm5,%xmm6 \n"
|
|
"punpcklwd %xmm7,%xmm5 \n"
|
|
"punpckhwd %xmm7,%xmm6 \n"
|
|
"movdqa %xmm6,%xmm7 \n"
|
|
"movdqa %xmm0,%xmm6 \n"
|
|
"punpckldq %xmm4,%xmm0 \n"
|
|
"punpckhdq %xmm4,%xmm6 \n"
|
|
"movdqa %xmm6,%xmm4 \n"
|
|
"movdqa (%esp),%xmm6 \n"
|
|
"movlpd %xmm0,(%edx) \n"
|
|
"movhpd %xmm0,(%ebx) \n"
|
|
"movlpd %xmm4,(%edx,%esi,1) \n"
|
|
"lea (%edx,%esi,2),%edx \n"
|
|
"movhpd %xmm4,(%ebx,%ebp,1) \n"
|
|
"lea (%ebx,%ebp,2),%ebx \n"
|
|
"movdqa %xmm2,%xmm0 \n"
|
|
"punpckldq %xmm6,%xmm2 \n"
|
|
"movlpd %xmm2,(%edx) \n"
|
|
"movhpd %xmm2,(%ebx) \n"
|
|
"punpckhdq %xmm6,%xmm0 \n"
|
|
"movlpd %xmm0,(%edx,%esi,1) \n"
|
|
"lea (%edx,%esi,2),%edx \n"
|
|
"movhpd %xmm0,(%ebx,%ebp,1) \n"
|
|
"lea (%ebx,%ebp,2),%ebx \n"
|
|
"movdqa %xmm1,%xmm0 \n"
|
|
"punpckldq %xmm5,%xmm1 \n"
|
|
"movlpd %xmm1,(%edx) \n"
|
|
"movhpd %xmm1,(%ebx) \n"
|
|
"punpckhdq %xmm5,%xmm0 \n"
|
|
"movlpd %xmm0,(%edx,%esi,1) \n"
|
|
"lea (%edx,%esi,2),%edx \n"
|
|
"movhpd %xmm0,(%ebx,%ebp,1) \n"
|
|
"lea (%ebx,%ebp,2),%ebx \n"
|
|
"movdqa %xmm3,%xmm0 \n"
|
|
"punpckldq %xmm7,%xmm3 \n"
|
|
"movlpd %xmm3,(%edx) \n"
|
|
"movhpd %xmm3,(%ebx) \n"
|
|
"punpckhdq %xmm7,%xmm0 \n"
|
|
"sub $0x8,%ecx \n"
|
|
"movlpd %xmm0,(%edx,%esi,1) \n"
|
|
"lea (%edx,%esi,2),%edx \n"
|
|
"movhpd %xmm0,(%ebx,%ebp,1) \n"
|
|
"lea (%ebx,%ebp,2),%ebx \n"
|
|
"jg 1b \n"
|
|
"mov 0x10(%esp),%esp \n"
|
|
"pop %ebp \n"
|
|
"pop %edi \n"
|
|
"pop %esi \n"
|
|
"pop %ebx \n"
|
|
#if defined(__native_client__)
|
|
"pop %ecx \n"
|
|
"and $0xffffffe0,%ecx \n"
|
|
"jmp *%ecx \n"
|
|
#else
|
|
"ret \n"
|
|
#endif
|
|
);
|
|
#elif !defined(LIBYUV_DISABLE_X86) && !defined(__native_client__) && \
|
|
defined(__x86_64__)
|
|
// 64 bit version has enough registers to do 16x8 to 8x16 at a time.
|
|
#define HAS_TRANSPOSE_WX8_FAST_SSSE3
|
|
static void TransposeWx8_FAST_SSSE3(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride, int width) {
|
|
asm volatile (
|
|
// Read in the data from the source pointer.
|
|
// First round of bit swap.
|
|
".p2align 2 \n"
|
|
"1: \n"
|
|
"movdqa (%0),%%xmm0 \n"
|
|
"movdqa (%0,%3),%%xmm1 \n"
|
|
"lea (%0,%3,2),%0 \n"
|
|
"movdqa %%xmm0,%%xmm8 \n"
|
|
"punpcklbw %%xmm1,%%xmm0 \n"
|
|
"punpckhbw %%xmm1,%%xmm8 \n"
|
|
"movdqa (%0),%%xmm2 \n"
|
|
"movdqa %%xmm0,%%xmm1 \n"
|
|
"movdqa %%xmm8,%%xmm9 \n"
|
|
"palignr $0x8,%%xmm1,%%xmm1 \n"
|
|
"palignr $0x8,%%xmm9,%%xmm9 \n"
|
|
"movdqa (%0,%3),%%xmm3 \n"
|
|
"lea (%0,%3,2),%0 \n"
|
|
"movdqa %%xmm2,%%xmm10 \n"
|
|
"punpcklbw %%xmm3,%%xmm2 \n"
|
|
"punpckhbw %%xmm3,%%xmm10 \n"
|
|
"movdqa %%xmm2,%%xmm3 \n"
|
|
"movdqa %%xmm10,%%xmm11 \n"
|
|
"movdqa (%0),%%xmm4 \n"
|
|
"palignr $0x8,%%xmm3,%%xmm3 \n"
|
|
"palignr $0x8,%%xmm11,%%xmm11 \n"
|
|
"movdqa (%0,%3),%%xmm5 \n"
|
|
"lea (%0,%3,2),%0 \n"
|
|
"movdqa %%xmm4,%%xmm12 \n"
|
|
"punpcklbw %%xmm5,%%xmm4 \n"
|
|
"punpckhbw %%xmm5,%%xmm12 \n"
|
|
"movdqa %%xmm4,%%xmm5 \n"
|
|
"movdqa %%xmm12,%%xmm13 \n"
|
|
"movdqa (%0),%%xmm6 \n"
|
|
"palignr $0x8,%%xmm5,%%xmm5 \n"
|
|
"palignr $0x8,%%xmm13,%%xmm13 \n"
|
|
"movdqa (%0,%3),%%xmm7 \n"
|
|
"lea (%0,%3,2),%0 \n"
|
|
"movdqa %%xmm6,%%xmm14 \n"
|
|
"punpcklbw %%xmm7,%%xmm6 \n"
|
|
"punpckhbw %%xmm7,%%xmm14 \n"
|
|
"neg %3 \n"
|
|
"movdqa %%xmm6,%%xmm7 \n"
|
|
"movdqa %%xmm14,%%xmm15 \n"
|
|
"lea 0x10(%0,%3,8),%0 \n"
|
|
"palignr $0x8,%%xmm7,%%xmm7 \n"
|
|
"palignr $0x8,%%xmm15,%%xmm15 \n"
|
|
"neg %3 \n"
|
|
// Second round of bit swap.
|
|
"punpcklwd %%xmm2,%%xmm0 \n"
|
|
"punpcklwd %%xmm3,%%xmm1 \n"
|
|
"movdqa %%xmm0,%%xmm2 \n"
|
|
"movdqa %%xmm1,%%xmm3 \n"
|
|
"palignr $0x8,%%xmm2,%%xmm2 \n"
|
|
"palignr $0x8,%%xmm3,%%xmm3 \n"
|
|
"punpcklwd %%xmm6,%%xmm4 \n"
|
|
"punpcklwd %%xmm7,%%xmm5 \n"
|
|
"movdqa %%xmm4,%%xmm6 \n"
|
|
"movdqa %%xmm5,%%xmm7 \n"
|
|
"palignr $0x8,%%xmm6,%%xmm6 \n"
|
|
"palignr $0x8,%%xmm7,%%xmm7 \n"
|
|
"punpcklwd %%xmm10,%%xmm8 \n"
|
|
"punpcklwd %%xmm11,%%xmm9 \n"
|
|
"movdqa %%xmm8,%%xmm10 \n"
|
|
"movdqa %%xmm9,%%xmm11 \n"
|
|
"palignr $0x8,%%xmm10,%%xmm10 \n"
|
|
"palignr $0x8,%%xmm11,%%xmm11 \n"
|
|
"punpcklwd %%xmm14,%%xmm12 \n"
|
|
"punpcklwd %%xmm15,%%xmm13 \n"
|
|
"movdqa %%xmm12,%%xmm14 \n"
|
|
"movdqa %%xmm13,%%xmm15 \n"
|
|
"palignr $0x8,%%xmm14,%%xmm14 \n"
|
|
"palignr $0x8,%%xmm15,%%xmm15 \n"
|
|
// Third round of bit swap.
|
|
// Write to the destination pointer.
|
|
"punpckldq %%xmm4,%%xmm0 \n"
|
|
"movq %%xmm0,(%1) \n"
|
|
"movdqa %%xmm0,%%xmm4 \n"
|
|
"palignr $0x8,%%xmm4,%%xmm4 \n"
|
|
"movq %%xmm4,(%1,%4) \n"
|
|
"lea (%1,%4,2),%1 \n"
|
|
"punpckldq %%xmm6,%%xmm2 \n"
|
|
"movdqa %%xmm2,%%xmm6 \n"
|
|
"movq %%xmm2,(%1) \n"
|
|
"palignr $0x8,%%xmm6,%%xmm6 \n"
|
|
"punpckldq %%xmm5,%%xmm1 \n"
|
|
"movq %%xmm6,(%1,%4) \n"
|
|
"lea (%1,%4,2),%1 \n"
|
|
"movdqa %%xmm1,%%xmm5 \n"
|
|
"movq %%xmm1,(%1) \n"
|
|
"palignr $0x8,%%xmm5,%%xmm5 \n"
|
|
"movq %%xmm5,(%1,%4) \n"
|
|
"lea (%1,%4,2),%1 \n"
|
|
"punpckldq %%xmm7,%%xmm3 \n"
|
|
"movq %%xmm3,(%1) \n"
|
|
"movdqa %%xmm3,%%xmm7 \n"
|
|
"palignr $0x8,%%xmm7,%%xmm7 \n"
|
|
"movq %%xmm7,(%1,%4) \n"
|
|
"lea (%1,%4,2),%1 \n"
|
|
"punpckldq %%xmm12,%%xmm8 \n"
|
|
"movq %%xmm8,(%1) \n"
|
|
"movdqa %%xmm8,%%xmm12 \n"
|
|
"palignr $0x8,%%xmm12,%%xmm12 \n"
|
|
"movq %%xmm12,(%1,%4) \n"
|
|
"lea (%1,%4,2),%1 \n"
|
|
"punpckldq %%xmm14,%%xmm10 \n"
|
|
"movdqa %%xmm10,%%xmm14 \n"
|
|
"movq %%xmm10,(%1) \n"
|
|
"palignr $0x8,%%xmm14,%%xmm14 \n"
|
|
"punpckldq %%xmm13,%%xmm9 \n"
|
|
"movq %%xmm14,(%1,%4) \n"
|
|
"lea (%1,%4,2),%1 \n"
|
|
"movdqa %%xmm9,%%xmm13 \n"
|
|
"movq %%xmm9,(%1) \n"
|
|
"palignr $0x8,%%xmm13,%%xmm13 \n"
|
|
"movq %%xmm13,(%1,%4) \n"
|
|
"lea (%1,%4,2),%1 \n"
|
|
"punpckldq %%xmm15,%%xmm11 \n"
|
|
"movq %%xmm11,(%1) \n"
|
|
"movdqa %%xmm11,%%xmm15 \n"
|
|
"palignr $0x8,%%xmm15,%%xmm15 \n"
|
|
"sub $0x10,%2 \n"
|
|
"movq %%xmm15,(%1,%4) \n"
|
|
"lea (%1,%4,2),%1 \n"
|
|
"jg 1b \n"
|
|
: "+r"(src), // %0
|
|
"+r"(dst), // %1
|
|
"+r"(width) // %2
|
|
: "r"((intptr_t)(src_stride)), // %3
|
|
"r"((intptr_t)(dst_stride)) // %4
|
|
: "memory", "cc",
|
|
"xmm0", "xmm1", "xmm2", "xmm3", "xmm4", "xmm5", "xmm6", "xmm7",
|
|
"xmm8", "xmm9", "xmm10", "xmm11", "xmm12", "xmm13", "xmm14", "xmm15"
|
|
);
|
|
}
|
|
|
|
#define HAS_TRANSPOSE_UVWX8_SSE2
|
|
static void TransposeUVWx8_SSE2(const uint8* src, int src_stride,
|
|
uint8* dst_a, int dst_stride_a,
|
|
uint8* dst_b, int dst_stride_b,
|
|
int w) {
|
|
asm volatile (
|
|
// Read in the data from the source pointer.
|
|
// First round of bit swap.
|
|
".p2align 2 \n"
|
|
"1: \n"
|
|
"movdqa (%0),%%xmm0 \n"
|
|
"movdqa (%0,%4),%%xmm1 \n"
|
|
"lea (%0,%4,2),%0 \n"
|
|
"movdqa %%xmm0,%%xmm8 \n"
|
|
"punpcklbw %%xmm1,%%xmm0 \n"
|
|
"punpckhbw %%xmm1,%%xmm8 \n"
|
|
"movdqa %%xmm8,%%xmm1 \n"
|
|
"movdqa (%0),%%xmm2 \n"
|
|
"movdqa (%0,%4),%%xmm3 \n"
|
|
"lea (%0,%4,2),%0 \n"
|
|
"movdqa %%xmm2,%%xmm8 \n"
|
|
"punpcklbw %%xmm3,%%xmm2 \n"
|
|
"punpckhbw %%xmm3,%%xmm8 \n"
|
|
"movdqa %%xmm8,%%xmm3 \n"
|
|
"movdqa (%0),%%xmm4 \n"
|
|
"movdqa (%0,%4),%%xmm5 \n"
|
|
"lea (%0,%4,2),%0 \n"
|
|
"movdqa %%xmm4,%%xmm8 \n"
|
|
"punpcklbw %%xmm5,%%xmm4 \n"
|
|
"punpckhbw %%xmm5,%%xmm8 \n"
|
|
"movdqa %%xmm8,%%xmm5 \n"
|
|
"movdqa (%0),%%xmm6 \n"
|
|
"movdqa (%0,%4),%%xmm7 \n"
|
|
"lea (%0,%4,2),%0 \n"
|
|
"movdqa %%xmm6,%%xmm8 \n"
|
|
"punpcklbw %%xmm7,%%xmm6 \n"
|
|
"neg %4 \n"
|
|
"lea 0x10(%0,%4,8),%0 \n"
|
|
"punpckhbw %%xmm7,%%xmm8 \n"
|
|
"movdqa %%xmm8,%%xmm7 \n"
|
|
"neg %4 \n"
|
|
// Second round of bit swap.
|
|
"movdqa %%xmm0,%%xmm8 \n"
|
|
"movdqa %%xmm1,%%xmm9 \n"
|
|
"punpckhwd %%xmm2,%%xmm8 \n"
|
|
"punpckhwd %%xmm3,%%xmm9 \n"
|
|
"punpcklwd %%xmm2,%%xmm0 \n"
|
|
"punpcklwd %%xmm3,%%xmm1 \n"
|
|
"movdqa %%xmm8,%%xmm2 \n"
|
|
"movdqa %%xmm9,%%xmm3 \n"
|
|
"movdqa %%xmm4,%%xmm8 \n"
|
|
"movdqa %%xmm5,%%xmm9 \n"
|
|
"punpckhwd %%xmm6,%%xmm8 \n"
|
|
"punpckhwd %%xmm7,%%xmm9 \n"
|
|
"punpcklwd %%xmm6,%%xmm4 \n"
|
|
"punpcklwd %%xmm7,%%xmm5 \n"
|
|
"movdqa %%xmm8,%%xmm6 \n"
|
|
"movdqa %%xmm9,%%xmm7 \n"
|
|
// Third round of bit swap.
|
|
// Write to the destination pointer.
|
|
"movdqa %%xmm0,%%xmm8 \n"
|
|
"punpckldq %%xmm4,%%xmm0 \n"
|
|
"movlpd %%xmm0,(%1) \n" // Write back U channel
|
|
"movhpd %%xmm0,(%2) \n" // Write back V channel
|
|
"punpckhdq %%xmm4,%%xmm8 \n"
|
|
"movlpd %%xmm8,(%1,%5) \n"
|
|
"lea (%1,%5,2),%1 \n"
|
|
"movhpd %%xmm8,(%2,%6) \n"
|
|
"lea (%2,%6,2),%2 \n"
|
|
"movdqa %%xmm2,%%xmm8 \n"
|
|
"punpckldq %%xmm6,%%xmm2 \n"
|
|
"movlpd %%xmm2,(%1) \n"
|
|
"movhpd %%xmm2,(%2) \n"
|
|
"punpckhdq %%xmm6,%%xmm8 \n"
|
|
"movlpd %%xmm8,(%1,%5) \n"
|
|
"lea (%1,%5,2),%1 \n"
|
|
"movhpd %%xmm8,(%2,%6) \n"
|
|
"lea (%2,%6,2),%2 \n"
|
|
"movdqa %%xmm1,%%xmm8 \n"
|
|
"punpckldq %%xmm5,%%xmm1 \n"
|
|
"movlpd %%xmm1,(%1) \n"
|
|
"movhpd %%xmm1,(%2) \n"
|
|
"punpckhdq %%xmm5,%%xmm8 \n"
|
|
"movlpd %%xmm8,(%1,%5) \n"
|
|
"lea (%1,%5,2),%1 \n"
|
|
"movhpd %%xmm8,(%2,%6) \n"
|
|
"lea (%2,%6,2),%2 \n"
|
|
"movdqa %%xmm3,%%xmm8 \n"
|
|
"punpckldq %%xmm7,%%xmm3 \n"
|
|
"movlpd %%xmm3,(%1) \n"
|
|
"movhpd %%xmm3,(%2) \n"
|
|
"punpckhdq %%xmm7,%%xmm8 \n"
|
|
"sub $0x8,%3 \n"
|
|
"movlpd %%xmm8,(%1,%5) \n"
|
|
"lea (%1,%5,2),%1 \n"
|
|
"movhpd %%xmm8,(%2,%6) \n"
|
|
"lea (%2,%6,2),%2 \n"
|
|
"jg 1b \n"
|
|
: "+r"(src), // %0
|
|
"+r"(dst_a), // %1
|
|
"+r"(dst_b), // %2
|
|
"+r"(w) // %3
|
|
: "r"((intptr_t)(src_stride)), // %4
|
|
"r"((intptr_t)(dst_stride_a)), // %5
|
|
"r"((intptr_t)(dst_stride_b)) // %6
|
|
: "memory", "cc",
|
|
"xmm0", "xmm1", "xmm2", "xmm3", "xmm4", "xmm5", "xmm6", "xmm7",
|
|
"xmm8", "xmm9"
|
|
);
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
static void TransposeWx8_C(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride,
|
|
int width) {
|
|
int i;
|
|
for (i = 0; i < width; ++i) {
|
|
dst[0] = src[0 * src_stride];
|
|
dst[1] = src[1 * src_stride];
|
|
dst[2] = src[2 * src_stride];
|
|
dst[3] = src[3 * src_stride];
|
|
dst[4] = src[4 * src_stride];
|
|
dst[5] = src[5 * src_stride];
|
|
dst[6] = src[6 * src_stride];
|
|
dst[7] = src[7 * src_stride];
|
|
++src;
|
|
dst += dst_stride;
|
|
}
|
|
}
|
|
|
|
static void TransposeWxH_C(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride,
|
|
int width, int height) {
|
|
int i;
|
|
for (i = 0; i < width; ++i) {
|
|
int j;
|
|
for (j = 0; j < height; ++j) {
|
|
dst[i * dst_stride + j] = src[j * src_stride + i];
|
|
}
|
|
}
|
|
}
|
|
|
|
LIBYUV_API
|
|
void TransposePlane(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride,
|
|
int width, int height) {
|
|
int i = height;
|
|
void (*TransposeWx8)(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride,
|
|
int width) = TransposeWx8_C;
|
|
#if defined(HAS_TRANSPOSE_WX8_NEON)
|
|
if (TestCpuFlag(kCpuHasNEON)) {
|
|
TransposeWx8 = TransposeWx8_NEON;
|
|
}
|
|
#endif
|
|
#if defined(HAS_TRANSPOSE_WX8_SSSE3)
|
|
if (TestCpuFlag(kCpuHasSSSE3) && IS_ALIGNED(width, 8)) {
|
|
TransposeWx8 = TransposeWx8_SSSE3;
|
|
}
|
|
#endif
|
|
#if defined(HAS_TRANSPOSE_WX8_FAST_SSSE3)
|
|
if (TestCpuFlag(kCpuHasSSSE3) &&
|
|
IS_ALIGNED(width, 16) &&
|
|
IS_ALIGNED(src, 16) && IS_ALIGNED(src_stride, 16)) {
|
|
TransposeWx8 = TransposeWx8_FAST_SSSE3;
|
|
}
|
|
#endif
|
|
#if defined(HAS_TRANSPOSE_WX8_MIPS_DSPR2)
|
|
if (TestCpuFlag(kCpuHasMIPS_DSPR2)) {
|
|
if (IS_ALIGNED(width, 4) &&
|
|
IS_ALIGNED(src, 4) && IS_ALIGNED(src_stride, 4)) {
|
|
TransposeWx8 = TransposeWx8_FAST_MIPS_DSPR2;
|
|
} else {
|
|
TransposeWx8 = TransposeWx8_MIPS_DSPR2;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
// Work across the source in 8x8 tiles
|
|
while (i >= 8) {
|
|
TransposeWx8(src, src_stride, dst, dst_stride, width);
|
|
src += 8 * src_stride; // Go down 8 rows.
|
|
dst += 8; // Move over 8 columns.
|
|
i -= 8;
|
|
}
|
|
|
|
TransposeWxH_C(src, src_stride, dst, dst_stride, width, i);
|
|
}
|
|
|
|
LIBYUV_API
|
|
void RotatePlane90(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride,
|
|
int width, int height) {
|
|
// Rotate by 90 is a transpose with the source read
|
|
// from bottom to top. So set the source pointer to the end
|
|
// of the buffer and flip the sign of the source stride.
|
|
src += src_stride * (height - 1);
|
|
src_stride = -src_stride;
|
|
TransposePlane(src, src_stride, dst, dst_stride, width, height);
|
|
}
|
|
|
|
LIBYUV_API
|
|
void RotatePlane270(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride,
|
|
int width, int height) {
|
|
// Rotate by 270 is a transpose with the destination written
|
|
// from bottom to top. So set the destination pointer to the end
|
|
// of the buffer and flip the sign of the destination stride.
|
|
dst += dst_stride * (width - 1);
|
|
dst_stride = -dst_stride;
|
|
TransposePlane(src, src_stride, dst, dst_stride, width, height);
|
|
}
|
|
|
|
LIBYUV_API
|
|
void RotatePlane180(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride,
|
|
int width, int height) {
|
|
// Swap first and last row and mirror the content. Uses a temporary row.
|
|
align_buffer_64(row, width);
|
|
const uint8* src_bot = src + src_stride * (height - 1);
|
|
uint8* dst_bot = dst + dst_stride * (height - 1);
|
|
int half_height = (height + 1) >> 1;
|
|
int y;
|
|
void (*MirrorRow)(const uint8* src, uint8* dst, int width) = MirrorRow_C;
|
|
void (*CopyRow)(const uint8* src, uint8* dst, int width) = CopyRow_C;
|
|
#if defined(HAS_MIRRORROW_NEON)
|
|
if (TestCpuFlag(kCpuHasNEON) && IS_ALIGNED(width, 16)) {
|
|
MirrorRow = MirrorRow_NEON;
|
|
}
|
|
#endif
|
|
#if defined(HAS_MIRRORROW_SSE2)
|
|
if (TestCpuFlag(kCpuHasSSE2) && IS_ALIGNED(width, 16) &&
|
|
IS_ALIGNED(src, 16) && IS_ALIGNED(src_stride, 16) &&
|
|
IS_ALIGNED(dst, 16) && IS_ALIGNED(dst_stride, 16)) {
|
|
MirrorRow = MirrorRow_SSE2;
|
|
}
|
|
#endif
|
|
#if defined(HAS_MIRRORROW_SSSE3)
|
|
if (TestCpuFlag(kCpuHasSSSE3) && IS_ALIGNED(width, 16) &&
|
|
IS_ALIGNED(src, 16) && IS_ALIGNED(src_stride, 16) &&
|
|
IS_ALIGNED(dst, 16) && IS_ALIGNED(dst_stride, 16)) {
|
|
MirrorRow = MirrorRow_SSSE3;
|
|
}
|
|
#endif
|
|
#if defined(HAS_MIRRORROW_AVX2)
|
|
if (TestCpuFlag(kCpuHasAVX2) && IS_ALIGNED(width, 32)) {
|
|
MirrorRow = MirrorRow_AVX2;
|
|
}
|
|
#endif
|
|
#if defined(HAS_MIRRORROW_MIPS_DSPR2)
|
|
if (TestCpuFlag(kCpuHasMIPS_DSPR2) &&
|
|
IS_ALIGNED(src, 4) && IS_ALIGNED(src_stride, 4) &&
|
|
IS_ALIGNED(dst, 4) && IS_ALIGNED(dst_stride, 4)) {
|
|
MirrorRow = MirrorRow_MIPS_DSPR2;
|
|
}
|
|
#endif
|
|
#if defined(HAS_COPYROW_NEON)
|
|
if (TestCpuFlag(kCpuHasNEON) && IS_ALIGNED(width, 32)) {
|
|
CopyRow = CopyRow_NEON;
|
|
}
|
|
#endif
|
|
#if defined(HAS_COPYROW_X86)
|
|
if (TestCpuFlag(kCpuHasX86) && IS_ALIGNED(width, 4)) {
|
|
CopyRow = CopyRow_X86;
|
|
}
|
|
#endif
|
|
#if defined(HAS_COPYROW_SSE2)
|
|
if (TestCpuFlag(kCpuHasSSE2) && IS_ALIGNED(width, 32) &&
|
|
IS_ALIGNED(src, 16) && IS_ALIGNED(src_stride, 16) &&
|
|
IS_ALIGNED(dst, 16) && IS_ALIGNED(dst_stride, 16)) {
|
|
CopyRow = CopyRow_SSE2;
|
|
}
|
|
#endif
|
|
#if defined(HAS_COPYROW_ERMS)
|
|
if (TestCpuFlag(kCpuHasERMS)) {
|
|
CopyRow = CopyRow_ERMS;
|
|
}
|
|
#endif
|
|
#if defined(HAS_COPYROW_MIPS)
|
|
if (TestCpuFlag(kCpuHasMIPS)) {
|
|
CopyRow = CopyRow_MIPS;
|
|
}
|
|
#endif
|
|
|
|
// Odd height will harmlessly mirror the middle row twice.
|
|
for (y = 0; y < half_height; ++y) {
|
|
MirrorRow(src, row, width); // Mirror first row into a buffer
|
|
src += src_stride;
|
|
MirrorRow(src_bot, dst, width); // Mirror last row into first row
|
|
dst += dst_stride;
|
|
CopyRow(row, dst_bot, width); // Copy first mirrored row into last
|
|
src_bot -= src_stride;
|
|
dst_bot -= dst_stride;
|
|
}
|
|
free_aligned_buffer_64(row);
|
|
}
|
|
|
|
static void TransposeUVWx8_C(const uint8* src, int src_stride,
|
|
uint8* dst_a, int dst_stride_a,
|
|
uint8* dst_b, int dst_stride_b,
|
|
int width) {
|
|
int i;
|
|
for (i = 0; i < width; ++i) {
|
|
dst_a[0] = src[0 * src_stride + 0];
|
|
dst_b[0] = src[0 * src_stride + 1];
|
|
dst_a[1] = src[1 * src_stride + 0];
|
|
dst_b[1] = src[1 * src_stride + 1];
|
|
dst_a[2] = src[2 * src_stride + 0];
|
|
dst_b[2] = src[2 * src_stride + 1];
|
|
dst_a[3] = src[3 * src_stride + 0];
|
|
dst_b[3] = src[3 * src_stride + 1];
|
|
dst_a[4] = src[4 * src_stride + 0];
|
|
dst_b[4] = src[4 * src_stride + 1];
|
|
dst_a[5] = src[5 * src_stride + 0];
|
|
dst_b[5] = src[5 * src_stride + 1];
|
|
dst_a[6] = src[6 * src_stride + 0];
|
|
dst_b[6] = src[6 * src_stride + 1];
|
|
dst_a[7] = src[7 * src_stride + 0];
|
|
dst_b[7] = src[7 * src_stride + 1];
|
|
src += 2;
|
|
dst_a += dst_stride_a;
|
|
dst_b += dst_stride_b;
|
|
}
|
|
}
|
|
|
|
static void TransposeUVWxH_C(const uint8* src, int src_stride,
|
|
uint8* dst_a, int dst_stride_a,
|
|
uint8* dst_b, int dst_stride_b,
|
|
int width, int height) {
|
|
int i;
|
|
for (i = 0; i < width * 2; i += 2) {
|
|
int j;
|
|
for (j = 0; j < height; ++j) {
|
|
dst_a[j + ((i >> 1) * dst_stride_a)] = src[i + (j * src_stride)];
|
|
dst_b[j + ((i >> 1) * dst_stride_b)] = src[i + (j * src_stride) + 1];
|
|
}
|
|
}
|
|
}
|
|
|
|
LIBYUV_API
|
|
void TransposeUV(const uint8* src, int src_stride,
|
|
uint8* dst_a, int dst_stride_a,
|
|
uint8* dst_b, int dst_stride_b,
|
|
int width, int height) {
|
|
int i = height;
|
|
void (*TransposeUVWx8)(const uint8* src, int src_stride,
|
|
uint8* dst_a, int dst_stride_a,
|
|
uint8* dst_b, int dst_stride_b,
|
|
int width) = TransposeUVWx8_C;
|
|
#if defined(HAS_TRANSPOSE_UVWX8_NEON)
|
|
if (TestCpuFlag(kCpuHasNEON)) {
|
|
TransposeUVWx8 = TransposeUVWx8_NEON;
|
|
}
|
|
#elif defined(HAS_TRANSPOSE_UVWX8_SSE2)
|
|
if (TestCpuFlag(kCpuHasSSE2) &&
|
|
IS_ALIGNED(width, 8) &&
|
|
IS_ALIGNED(src, 16) && IS_ALIGNED(src_stride, 16)) {
|
|
TransposeUVWx8 = TransposeUVWx8_SSE2;
|
|
}
|
|
#elif defined(HAS_TRANSPOSE_UVWx8_MIPS_DSPR2)
|
|
if (TestCpuFlag(kCpuHasMIPS_DSPR2) && IS_ALIGNED(width, 2) &&
|
|
IS_ALIGNED(src, 4) && IS_ALIGNED(src_stride, 4)) {
|
|
TransposeUVWx8 = TransposeUVWx8_MIPS_DSPR2;
|
|
}
|
|
#endif
|
|
|
|
// Work through the source in 8x8 tiles.
|
|
while (i >= 8) {
|
|
TransposeUVWx8(src, src_stride,
|
|
dst_a, dst_stride_a,
|
|
dst_b, dst_stride_b,
|
|
width);
|
|
src += 8 * src_stride; // Go down 8 rows.
|
|
dst_a += 8; // Move over 8 columns.
|
|
dst_b += 8; // Move over 8 columns.
|
|
i -= 8;
|
|
}
|
|
|
|
TransposeUVWxH_C(src, src_stride,
|
|
dst_a, dst_stride_a,
|
|
dst_b, dst_stride_b,
|
|
width, i);
|
|
}
|
|
|
|
LIBYUV_API
|
|
void RotateUV90(const uint8* src, int src_stride,
|
|
uint8* dst_a, int dst_stride_a,
|
|
uint8* dst_b, int dst_stride_b,
|
|
int width, int height) {
|
|
src += src_stride * (height - 1);
|
|
src_stride = -src_stride;
|
|
|
|
TransposeUV(src, src_stride,
|
|
dst_a, dst_stride_a,
|
|
dst_b, dst_stride_b,
|
|
width, height);
|
|
}
|
|
|
|
LIBYUV_API
|
|
void RotateUV270(const uint8* src, int src_stride,
|
|
uint8* dst_a, int dst_stride_a,
|
|
uint8* dst_b, int dst_stride_b,
|
|
int width, int height) {
|
|
dst_a += dst_stride_a * (width - 1);
|
|
dst_b += dst_stride_b * (width - 1);
|
|
dst_stride_a = -dst_stride_a;
|
|
dst_stride_b = -dst_stride_b;
|
|
|
|
TransposeUV(src, src_stride,
|
|
dst_a, dst_stride_a,
|
|
dst_b, dst_stride_b,
|
|
width, height);
|
|
}
|
|
|
|
// Rotate 180 is a horizontal and vertical flip.
|
|
LIBYUV_API
|
|
void RotateUV180(const uint8* src, int src_stride,
|
|
uint8* dst_a, int dst_stride_a,
|
|
uint8* dst_b, int dst_stride_b,
|
|
int width, int height) {
|
|
int i;
|
|
void (*MirrorRowUV)(const uint8* src, uint8* dst_u, uint8* dst_v, int width) =
|
|
MirrorUVRow_C;
|
|
#if defined(HAS_MIRRORUVROW_NEON)
|
|
if (TestCpuFlag(kCpuHasNEON) && IS_ALIGNED(width, 8)) {
|
|
MirrorRowUV = MirrorUVRow_NEON;
|
|
}
|
|
#elif defined(HAS_MIRRORROW_UV_SSSE3)
|
|
if (TestCpuFlag(kCpuHasSSSE3) && IS_ALIGNED(width, 16) &&
|
|
IS_ALIGNED(src, 16) && IS_ALIGNED(src_stride, 16)) {
|
|
MirrorRowUV = MirrorUVRow_SSSE3;
|
|
}
|
|
#elif defined(HAS_MIRRORUVROW_MIPS_DSPR2)
|
|
if (TestCpuFlag(kCpuHasMIPS_DSPR2) &&
|
|
IS_ALIGNED(src, 4) && IS_ALIGNED(src_stride, 4)) {
|
|
MirrorRowUV = MirrorUVRow_MIPS_DSPR2;
|
|
}
|
|
#endif
|
|
|
|
dst_a += dst_stride_a * (height - 1);
|
|
dst_b += dst_stride_b * (height - 1);
|
|
|
|
for (i = 0; i < height; ++i) {
|
|
MirrorRowUV(src, dst_a, dst_b, width);
|
|
src += src_stride;
|
|
dst_a -= dst_stride_a;
|
|
dst_b -= dst_stride_b;
|
|
}
|
|
}
|
|
|
|
LIBYUV_API
|
|
int RotatePlane(const uint8* src, int src_stride,
|
|
uint8* dst, int dst_stride,
|
|
int width, int height,
|
|
enum RotationMode mode) {
|
|
if (!src || width <= 0 || height == 0 || !dst) {
|
|
return -1;
|
|
}
|
|
|
|
// Negative height means invert the image.
|
|
if (height < 0) {
|
|
height = -height;
|
|
src = src + (height - 1) * src_stride;
|
|
src_stride = -src_stride;
|
|
}
|
|
|
|
switch (mode) {
|
|
case kRotate0:
|
|
// copy frame
|
|
CopyPlane(src, src_stride,
|
|
dst, dst_stride,
|
|
width, height);
|
|
return 0;
|
|
case kRotate90:
|
|
RotatePlane90(src, src_stride,
|
|
dst, dst_stride,
|
|
width, height);
|
|
return 0;
|
|
case kRotate270:
|
|
RotatePlane270(src, src_stride,
|
|
dst, dst_stride,
|
|
width, height);
|
|
return 0;
|
|
case kRotate180:
|
|
RotatePlane180(src, src_stride,
|
|
dst, dst_stride,
|
|
width, height);
|
|
return 0;
|
|
default:
|
|
break;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
LIBYUV_API
|
|
int I420Rotate(const uint8* src_y, int src_stride_y,
|
|
const uint8* src_u, int src_stride_u,
|
|
const uint8* src_v, int src_stride_v,
|
|
uint8* dst_y, int dst_stride_y,
|
|
uint8* dst_u, int dst_stride_u,
|
|
uint8* dst_v, int dst_stride_v,
|
|
int width, int height,
|
|
enum RotationMode mode) {
|
|
int halfwidth = (width + 1) >> 1;
|
|
int halfheight = (height + 1) >> 1;
|
|
if (!src_y || !src_u || !src_v || width <= 0 || height == 0 ||
|
|
!dst_y || !dst_u || !dst_v) {
|
|
return -1;
|
|
}
|
|
|
|
// Negative height means invert the image.
|
|
if (height < 0) {
|
|
height = -height;
|
|
halfheight = (height + 1) >> 1;
|
|
src_y = src_y + (height - 1) * src_stride_y;
|
|
src_u = src_u + (halfheight - 1) * src_stride_u;
|
|
src_v = src_v + (halfheight - 1) * src_stride_v;
|
|
src_stride_y = -src_stride_y;
|
|
src_stride_u = -src_stride_u;
|
|
src_stride_v = -src_stride_v;
|
|
}
|
|
|
|
switch (mode) {
|
|
case kRotate0:
|
|
// copy frame
|
|
return I420Copy(src_y, src_stride_y,
|
|
src_u, src_stride_u,
|
|
src_v, src_stride_v,
|
|
dst_y, dst_stride_y,
|
|
dst_u, dst_stride_u,
|
|
dst_v, dst_stride_v,
|
|
width, height);
|
|
case kRotate90:
|
|
RotatePlane90(src_y, src_stride_y,
|
|
dst_y, dst_stride_y,
|
|
width, height);
|
|
RotatePlane90(src_u, src_stride_u,
|
|
dst_u, dst_stride_u,
|
|
halfwidth, halfheight);
|
|
RotatePlane90(src_v, src_stride_v,
|
|
dst_v, dst_stride_v,
|
|
halfwidth, halfheight);
|
|
return 0;
|
|
case kRotate270:
|
|
RotatePlane270(src_y, src_stride_y,
|
|
dst_y, dst_stride_y,
|
|
width, height);
|
|
RotatePlane270(src_u, src_stride_u,
|
|
dst_u, dst_stride_u,
|
|
halfwidth, halfheight);
|
|
RotatePlane270(src_v, src_stride_v,
|
|
dst_v, dst_stride_v,
|
|
halfwidth, halfheight);
|
|
return 0;
|
|
case kRotate180:
|
|
RotatePlane180(src_y, src_stride_y,
|
|
dst_y, dst_stride_y,
|
|
width, height);
|
|
RotatePlane180(src_u, src_stride_u,
|
|
dst_u, dst_stride_u,
|
|
halfwidth, halfheight);
|
|
RotatePlane180(src_v, src_stride_v,
|
|
dst_v, dst_stride_v,
|
|
halfwidth, halfheight);
|
|
return 0;
|
|
default:
|
|
break;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
LIBYUV_API
|
|
int NV12ToI420Rotate(const uint8* src_y, int src_stride_y,
|
|
const uint8* src_uv, int src_stride_uv,
|
|
uint8* dst_y, int dst_stride_y,
|
|
uint8* dst_u, int dst_stride_u,
|
|
uint8* dst_v, int dst_stride_v,
|
|
int width, int height,
|
|
enum RotationMode mode) {
|
|
int halfwidth = (width + 1) >> 1;
|
|
int halfheight = (height + 1) >> 1;
|
|
if (!src_y || !src_uv || width <= 0 || height == 0 ||
|
|
!dst_y || !dst_u || !dst_v) {
|
|
return -1;
|
|
}
|
|
|
|
// Negative height means invert the image.
|
|
if (height < 0) {
|
|
height = -height;
|
|
halfheight = (height + 1) >> 1;
|
|
src_y = src_y + (height - 1) * src_stride_y;
|
|
src_uv = src_uv + (halfheight - 1) * src_stride_uv;
|
|
src_stride_y = -src_stride_y;
|
|
src_stride_uv = -src_stride_uv;
|
|
}
|
|
|
|
switch (mode) {
|
|
case kRotate0:
|
|
// copy frame
|
|
return NV12ToI420(src_y, src_stride_y,
|
|
src_uv, src_stride_uv,
|
|
dst_y, dst_stride_y,
|
|
dst_u, dst_stride_u,
|
|
dst_v, dst_stride_v,
|
|
width, height);
|
|
case kRotate90:
|
|
RotatePlane90(src_y, src_stride_y,
|
|
dst_y, dst_stride_y,
|
|
width, height);
|
|
RotateUV90(src_uv, src_stride_uv,
|
|
dst_u, dst_stride_u,
|
|
dst_v, dst_stride_v,
|
|
halfwidth, halfheight);
|
|
return 0;
|
|
case kRotate270:
|
|
RotatePlane270(src_y, src_stride_y,
|
|
dst_y, dst_stride_y,
|
|
width, height);
|
|
RotateUV270(src_uv, src_stride_uv,
|
|
dst_u, dst_stride_u,
|
|
dst_v, dst_stride_v,
|
|
halfwidth, halfheight);
|
|
return 0;
|
|
case kRotate180:
|
|
RotatePlane180(src_y, src_stride_y,
|
|
dst_y, dst_stride_y,
|
|
width, height);
|
|
RotateUV180(src_uv, src_stride_uv,
|
|
dst_u, dst_stride_u,
|
|
dst_v, dst_stride_v,
|
|
halfwidth, halfheight);
|
|
return 0;
|
|
default:
|
|
break;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
} // extern "C"
|
|
} // namespace libyuv
|
|
#endif
|