Spaces:
Runtime error
Runtime error
/* | |
* Copyright (c) 2008 Mans Rullgard <[email protected]> | |
* | |
* This file is part of FFmpeg. | |
* | |
* FFmpeg is free software; you can redistribute it and/or | |
* modify it under the terms of the GNU Lesser General Public | |
* License as published by the Free Software Foundation; either | |
* version 2.1 of the License, or (at your option) any later version. | |
* | |
* FFmpeg is distributed in the hope that it will be useful, | |
* but WITHOUT ANY WARRANTY; without even the implied warranty of | |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
* Lesser General Public License for more details. | |
* | |
* You should have received a copy of the GNU Lesser General Public | |
* License along with FFmpeg; if not, write to the Free Software | |
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
*/ | |
#include "config_components.h" | |
#include "libavutil/arm/asm.S" | |
/* chroma_mc8(uint8_t *dst, const uint8_t *src, ptrdiff_t stride, int h, int x, int y) */ | |
.macro h264_chroma_mc8 type, codec=h264 | |
function ff_\type\()_\codec\()_chroma_mc8_neon, =1 | |
push {r4-r7, lr} | |
ldrd r4, r5, [sp, #20] | |
.ifc \type,avg | |
mov lr, r0 | |
pld [r1] | |
pld [r1, r2] | |
.ifc \codec,rv40 | |
movrel r6, rv40bias | |
lsr r7, r5, #1 | |
add r6, r6, r7, lsl #3 | |
lsr r7, r4, #1 | |
add r6, r6, r7, lsl #1 | |
vld1.16 {d22[],d23[]}, [r6,:16] | |
.ifc \codec,vc1 | |
vmov.u16 q11, #28 | |
A muls r7, r4, r5 | |
T mul r7, r4, r5 | |
T cmp r7, #0 | |
rsb r6, r7, r5, lsl #3 | |
rsb r12, r7, r4, lsl #3 | |
sub r4, r7, r4, lsl #3 | |
sub r4, r4, r5, lsl #3 | |
add r4, r4, #64 | |
beq 2f | |
vdup.8 d0, r4 | |
vdup.8 d1, r12 | |
vld1.8 {d4, d5}, [r1], r2 | |
vdup.8 d2, r6 | |
vdup.8 d3, r7 | |
vext.8 d5, d4, d5, #1 | |
1: vld1.8 {d6, d7}, [r1], r2 | |
vmull.u8 q8, d4, d0 | |
vmlal.u8 q8, d5, d1 | |
vext.8 d7, d6, d7, #1 | |
vld1.8 {d4, d5}, [r1], r2 | |
vmlal.u8 q8, d6, d2 | |
pld [r1] | |
vext.8 d5, d4, d5, #1 | |
vmlal.u8 q8, d7, d3 | |
vmull.u8 q9, d6, d0 | |
subs r3, r3, #2 | |
vmlal.u8 q9, d7, d1 | |
vmlal.u8 q9, d4, d2 | |
vmlal.u8 q9, d5, d3 | |
pld [r1, r2] | |
.ifc \codec,h264 | |
vrshrn.u16 d16, q8, #6 | |
vrshrn.u16 d17, q9, #6 | |
vadd.u16 q8, q8, q11 | |
vadd.u16 q9, q9, q11 | |
vshrn.u16 d16, q8, #6 | |
vshrn.u16 d17, q9, #6 | |
.ifc \type,avg | |
vld1.8 {d20}, [lr,:64], r2 | |
vld1.8 {d21}, [lr,:64], r2 | |
vrhadd.u8 q8, q8, q10 | |
vst1.8 {d16}, [r0,:64], r2 | |
vst1.8 {d17}, [r0,:64], r2 | |
bgt 1b | |
pop {r4-r7, pc} | |
2: adds r12, r12, r6 | |
vdup.8 d0, r4 | |
beq 5f | |
tst r6, r6 | |
vdup.8 d1, r12 | |
beq 4f | |
vld1.8 {d4}, [r1], r2 | |
3: vld1.8 {d6}, [r1], r2 | |
vmull.u8 q8, d4, d0 | |
vmlal.u8 q8, d6, d1 | |
vld1.8 {d4}, [r1], r2 | |
vmull.u8 q9, d6, d0 | |
vmlal.u8 q9, d4, d1 | |
pld [r1] | |
.ifc \codec,h264 | |
vrshrn.u16 d16, q8, #6 | |
vrshrn.u16 d17, q9, #6 | |
vadd.u16 q8, q8, q11 | |
vadd.u16 q9, q9, q11 | |
vshrn.u16 d16, q8, #6 | |
vshrn.u16 d17, q9, #6 | |
pld [r1, r2] | |
.ifc \type,avg | |
vld1.8 {d20}, [lr,:64], r2 | |
vld1.8 {d21}, [lr,:64], r2 | |
vrhadd.u8 q8, q8, q10 | |
subs r3, r3, #2 | |
vst1.8 {d16}, [r0,:64], r2 | |
vst1.8 {d17}, [r0,:64], r2 | |
bgt 3b | |
pop {r4-r7, pc} | |
4: vld1.8 {d4, d5}, [r1], r2 | |
vld1.8 {d6, d7}, [r1], r2 | |
vext.8 d5, d4, d5, #1 | |
vext.8 d7, d6, d7, #1 | |
pld [r1] | |
subs r3, r3, #2 | |
vmull.u8 q8, d4, d0 | |
vmlal.u8 q8, d5, d1 | |
vmull.u8 q9, d6, d0 | |
vmlal.u8 q9, d7, d1 | |
pld [r1, r2] | |
.ifc \codec,h264 | |
vrshrn.u16 d16, q8, #6 | |
vrshrn.u16 d17, q9, #6 | |
vadd.u16 q8, q8, q11 | |
vadd.u16 q9, q9, q11 | |
vshrn.u16 d16, q8, #6 | |
vshrn.u16 d17, q9, #6 | |
.ifc \type,avg | |
vld1.8 {d20}, [lr,:64], r2 | |
vld1.8 {d21}, [lr,:64], r2 | |
vrhadd.u8 q8, q8, q10 | |
vst1.8 {d16}, [r0,:64], r2 | |
vst1.8 {d17}, [r0,:64], r2 | |
bgt 4b | |
pop {r4-r7, pc} | |
5: vld1.8 {d4}, [r1], r2 | |
vld1.8 {d5}, [r1], r2 | |
pld [r1] | |
subs r3, r3, #2 | |
vmull.u8 q8, d4, d0 | |
vmull.u8 q9, d5, d0 | |
pld [r1, r2] | |
.ifc \codec,h264 | |
vrshrn.u16 d16, q8, #6 | |
vrshrn.u16 d17, q9, #6 | |
vadd.u16 q8, q8, q11 | |
vadd.u16 q9, q9, q11 | |
vshrn.u16 d16, q8, #6 | |
vshrn.u16 d17, q9, #6 | |
.ifc \type,avg | |
vld1.8 {d20}, [lr,:64], r2 | |
vld1.8 {d21}, [lr,:64], r2 | |
vrhadd.u8 q8, q8, q10 | |
vst1.8 {d16}, [r0,:64], r2 | |
vst1.8 {d17}, [r0,:64], r2 | |
bgt 5b | |
pop {r4-r7, pc} | |
endfunc | |
.endm | |
/* chroma_mc4(uint8_t *dst, const uint8_t *src, ptrdiff_t stride, int h, int x, int y) */ | |
.macro h264_chroma_mc4 type, codec=h264 | |
function ff_\type\()_\codec\()_chroma_mc4_neon, =1 | |
push {r4-r7, lr} | |
ldrd r4, r5, [sp, #20] | |
.ifc \type,avg | |
mov lr, r0 | |
pld [r1] | |
pld [r1, r2] | |
.ifc \codec,rv40 | |
movrel r6, rv40bias | |
lsr r7, r5, #1 | |
add r6, r6, r7, lsl #3 | |
lsr r7, r4, #1 | |
add r6, r6, r7, lsl #1 | |
vld1.16 {d22[],d23[]}, [r6,:16] | |
.ifc \codec,vc1 | |
vmov.u16 q11, #28 | |
A muls r7, r4, r5 | |
T mul r7, r4, r5 | |
T cmp r7, #0 | |
rsb r6, r7, r5, lsl #3 | |
rsb r12, r7, r4, lsl #3 | |
sub r4, r7, r4, lsl #3 | |
sub r4, r4, r5, lsl #3 | |
add r4, r4, #64 | |
beq 2f | |
vdup.8 d0, r4 | |
vdup.8 d1, r12 | |
vld1.8 {d4}, [r1], r2 | |
vdup.8 d2, r6 | |
vdup.8 d3, r7 | |
vext.8 d5, d4, d5, #1 | |
vtrn.32 d4, d5 | |
vtrn.32 d0, d1 | |
vtrn.32 d2, d3 | |
1: vld1.8 {d6}, [r1], r2 | |
vext.8 d7, d6, d7, #1 | |
vtrn.32 d6, d7 | |
vmull.u8 q8, d4, d0 | |
vmlal.u8 q8, d6, d2 | |
vld1.8 {d4}, [r1], r2 | |
vext.8 d5, d4, d5, #1 | |
vtrn.32 d4, d5 | |
pld [r1] | |
vmull.u8 q9, d6, d0 | |
vmlal.u8 q9, d4, d2 | |
vadd.i16 d16, d16, d17 | |
vadd.i16 d17, d18, d19 | |
.ifc \codec,h264 | |
vrshrn.u16 d16, q8, #6 | |
vadd.u16 q8, q8, q11 | |
vshrn.u16 d16, q8, #6 | |
subs r3, r3, #2 | |
pld [r1, r2] | |
.ifc \type,avg | |
vld1.32 {d20[0]}, [lr,:32], r2 | |
vld1.32 {d20[1]}, [lr,:32], r2 | |
vrhadd.u8 d16, d16, d20 | |
vst1.32 {d16[0]}, [r0,:32], r2 | |
vst1.32 {d16[1]}, [r0,:32], r2 | |
bgt 1b | |
pop {r4-r7, pc} | |
2: adds r12, r12, r6 | |
vdup.8 d0, r4 | |
beq 5f | |
tst r6, r6 | |
vdup.8 d1, r12 | |
vtrn.32 d0, d1 | |
beq 4f | |
vext.32 d1, d0, d1, #1 | |
vld1.32 {d4[0]}, [r1], r2 | |
3: vld1.32 {d4[1]}, [r1], r2 | |
vmull.u8 q8, d4, d0 | |
vld1.32 {d4[0]}, [r1], r2 | |
vmull.u8 q9, d4, d1 | |
vadd.i16 d16, d16, d17 | |
vadd.i16 d17, d18, d19 | |
pld [r1] | |
.ifc \codec,h264 | |
vrshrn.u16 d16, q8, #6 | |
vadd.u16 q8, q8, q11 | |
vshrn.u16 d16, q8, #6 | |
.ifc \type,avg | |
vld1.32 {d20[0]}, [lr,:32], r2 | |
vld1.32 {d20[1]}, [lr,:32], r2 | |
vrhadd.u8 d16, d16, d20 | |
subs r3, r3, #2 | |
pld [r1, r2] | |
vst1.32 {d16[0]}, [r0,:32], r2 | |
vst1.32 {d16[1]}, [r0,:32], r2 | |
bgt 3b | |
pop {r4-r7, pc} | |
4: vld1.8 {d4}, [r1], r2 | |
vld1.8 {d6}, [r1], r2 | |
vext.8 d5, d4, d5, #1 | |
vext.8 d7, d6, d7, #1 | |
vtrn.32 d4, d5 | |
vtrn.32 d6, d7 | |
vmull.u8 q8, d4, d0 | |
vmull.u8 q9, d6, d0 | |
subs r3, r3, #2 | |
vadd.i16 d16, d16, d17 | |
vadd.i16 d17, d18, d19 | |
pld [r1] | |
.ifc \codec,h264 | |
vrshrn.u16 d16, q8, #6 | |
vadd.u16 q8, q8, q11 | |
vshrn.u16 d16, q8, #6 | |
.ifc \type,avg | |
vld1.32 {d20[0]}, [lr,:32], r2 | |
vld1.32 {d20[1]}, [lr,:32], r2 | |
vrhadd.u8 d16, d16, d20 | |
pld [r1] | |
vst1.32 {d16[0]}, [r0,:32], r2 | |
vst1.32 {d16[1]}, [r0,:32], r2 | |
bgt 4b | |
pop {r4-r7, pc} | |
5: vld1.32 {d4[0]}, [r1], r2 | |
vld1.32 {d4[1]}, [r1], r2 | |
vmull.u8 q8, d4, d0 | |
subs r3, r3, #2 | |
pld [r1] | |
.ifc \codec,h264 | |
vrshrn.u16 d16, q8, #6 | |
vadd.u16 q8, q8, q11 | |
vshrn.u16 d16, q8, #6 | |
.ifc \type,avg | |
vld1.32 {d20[0]}, [lr,:32], r2 | |
vld1.32 {d20[1]}, [lr,:32], r2 | |
vrhadd.u8 d16, d16, d20 | |
pld [r1] | |
vst1.32 {d16[0]}, [r0,:32], r2 | |
vst1.32 {d16[1]}, [r0,:32], r2 | |
bgt 5b | |
pop {r4-r7, pc} | |
endfunc | |
.endm | |
.macro h264_chroma_mc2 type | |
function ff_\type\()_h264_chroma_mc2_neon, =1 | |
push {r4-r6, lr} | |
ldr r4, [sp, #16] | |
ldr lr, [sp, #20] | |
pld [r1] | |
pld [r1, r2] | |
orrs r5, r4, lr | |
beq 2f | |
mul r5, r4, lr | |
rsb r6, r5, lr, lsl #3 | |
rsb r12, r5, r4, lsl #3 | |
sub r4, r5, r4, lsl #3 | |
sub r4, r4, lr, lsl #3 | |
add r4, r4, #64 | |
vdup.8 d0, r4 | |
vdup.8 d2, r12 | |
vdup.8 d1, r6 | |
vdup.8 d3, r5 | |
vtrn.16 q0, q1 | |
1: | |
vld1.32 {d4[0]}, [r1], r2 | |
vld1.32 {d4[1]}, [r1], r2 | |
vrev64.32 d5, d4 | |
vld1.32 {d5[1]}, [r1] | |
vext.8 q3, q2, q2, #1 | |
vtrn.16 q2, q3 | |
vmull.u8 q8, d4, d0 | |
vmlal.u8 q8, d5, d1 | |
.ifc \type,avg | |
vld1.16 {d18[0]}, [r0,:16], r2 | |
vld1.16 {d18[1]}, [r0,:16] | |
sub r0, r0, r2 | |
vtrn.32 d16, d17 | |
vadd.i16 d16, d16, d17 | |
vrshrn.u16 d16, q8, #6 | |
.ifc \type,avg | |
vrhadd.u8 d16, d16, d18 | |
vst1.16 {d16[0]}, [r0,:16], r2 | |
vst1.16 {d16[1]}, [r0,:16], r2 | |
subs r3, r3, #2 | |
bgt 1b | |
pop {r4-r6, pc} | |
2: | |
.ifc \type,put | |
ldrh_post r5, r1, r2 | |
strh_post r5, r0, r2 | |
ldrh_post r6, r1, r2 | |
strh_post r6, r0, r2 | |
vld1.16 {d16[0]}, [r1], r2 | |
vld1.16 {d16[1]}, [r1], r2 | |
vld1.16 {d18[0]}, [r0,:16], r2 | |
vld1.16 {d18[1]}, [r0,:16] | |
sub r0, r0, r2 | |
vrhadd.u8 d16, d16, d18 | |
vst1.16 {d16[0]}, [r0,:16], r2 | |
vst1.16 {d16[1]}, [r0,:16], r2 | |
subs r3, r3, #2 | |
bgt 2b | |
pop {r4-r6, pc} | |
endfunc | |
.endm | |
h264_chroma_mc8 put | |
h264_chroma_mc8 avg | |
h264_chroma_mc4 put | |
h264_chroma_mc4 avg | |
h264_chroma_mc2 put | |
h264_chroma_mc2 avg | |
#if CONFIG_RV40_DECODER | |
const rv40bias | |
.short 0, 16, 32, 16 | |
.short 32, 28, 32, 28 | |
.short 0, 32, 16, 32 | |
.short 32, 28, 32, 28 | |
endconst | |
h264_chroma_mc8 put, rv40 | |
h264_chroma_mc8 avg, rv40 | |
h264_chroma_mc4 put, rv40 | |
h264_chroma_mc4 avg, rv40 | |
#endif | |
#if CONFIG_VC1DSP | |
h264_chroma_mc8 put, vc1 | |
h264_chroma_mc8 avg, vc1 | |
h264_chroma_mc4 put, vc1 | |
h264_chroma_mc4 avg, vc1 | |
#endif | |