Spaces:
Runtime error
Runtime error
| /* | |
| * ARM NEON optimised DSP functions | |
| * Copyright (c) 2008 Mans Rullgard <[email protected]> | |
| * Copyright (c) 2013 Janne Grunau <[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 "libavutil/aarch64/asm.S" | |
| .macro pixels16 rnd=1, avg=0 | |
| \avg | |
| mov x12, x0 | |
| 1: ld1 {v0.16B}, [x1], x2 | |
| ld1 {v1.16B}, [x1], x2 | |
| ld1 {v2.16B}, [x1], x2 | |
| ld1 {v3.16B}, [x1], x2 | |
| \avg | |
| ld1 {v4.16B}, [x12], x2 | |
| urhadd v0.16B, v0.16B, v4.16B | |
| ld1 {v5.16B}, [x12], x2 | |
| urhadd v1.16B, v1.16B, v5.16B | |
| ld1 {v6.16B}, [x12], x2 | |
| urhadd v2.16B, v2.16B, v6.16B | |
| ld1 {v7.16B}, [x12], x2 | |
| urhadd v3.16B, v3.16B, v7.16B | |
| subs w3, w3, #4 | |
| st1 {v0.16B}, [x0], x2 | |
| st1 {v1.16B}, [x0], x2 | |
| st1 {v2.16B}, [x0], x2 | |
| st1 {v3.16B}, [x0], x2 | |
| b.ne 1b | |
| ret | |
| .endm | |
| .macro pixels16_x2 rnd=1, avg=0 | |
| 1: ld1 {v0.16B, v1.16B}, [x1], x2 | |
| ld1 {v2.16B, v3.16B}, [x1], x2 | |
| subs w3, w3, #2 | |
| ext v1.16B, v0.16B, v1.16B, #1 | |
| avg v0.16B, v0.16B, v1.16B | |
| ext v3.16B, v2.16B, v3.16B, #1 | |
| avg v2.16B, v2.16B, v3.16B | |
| \avg | |
| ld1 {v1.16B}, [x0], x2 | |
| ld1 {v3.16B}, [x0] | |
| urhadd v0.16B, v0.16B, v1.16B | |
| urhadd v2.16B, v2.16B, v3.16B | |
| sub x0, x0, x2 | |
| st1 {v0.16B}, [x0], x2 | |
| st1 {v2.16B}, [x0], x2 | |
| b.ne 1b | |
| ret | |
| .endm | |
| .macro pixels16_y2 rnd=1, avg=0 | |
| sub w3, w3, #2 | |
| ld1 {v0.16B}, [x1], x2 | |
| ld1 {v1.16B}, [x1], x2 | |
| 1: subs w3, w3, #2 | |
| avg v2.16B, v0.16B, v1.16B | |
| ld1 {v0.16B}, [x1], x2 | |
| avg v3.16B, v0.16B, v1.16B | |
| ld1 {v1.16B}, [x1], x2 | |
| \avg | |
| ld1 {v4.16B}, [x0], x2 | |
| ld1 {v5.16B}, [x0] | |
| urhadd v2.16B, v2.16B, v4.16B | |
| urhadd v3.16B, v3.16B, v5.16B | |
| sub x0, x0, x2 | |
| st1 {v2.16B}, [x0], x2 | |
| st1 {v3.16B}, [x0], x2 | |
| b.ne 1b | |
| avg v2.16B, v0.16B, v1.16B | |
| ld1 {v0.16B}, [x1], x2 | |
| avg v3.16B, v0.16B, v1.16B | |
| \avg | |
| ld1 {v4.16B}, [x0], x2 | |
| ld1 {v5.16B}, [x0] | |
| urhadd v2.16B, v2.16B, v4.16B | |
| urhadd v3.16B, v3.16B, v5.16B | |
| sub x0, x0, x2 | |
| st1 {v2.16B}, [x0], x2 | |
| st1 {v3.16B}, [x0], x2 | |
| ret | |
| .endm | |
| .macro pixels16_xy2 rnd=1, avg=0 | |
| sub w3, w3, #2 | |
| ld1 {v0.16B, v1.16B}, [x1], x2 | |
| ld1 {v4.16B, v5.16B}, [x1], x2 | |
| NRND movi v26.8H, #1 | |
| ext v1.16B, v0.16B, v1.16B, #1 | |
| ext v5.16B, v4.16B, v5.16B, #1 | |
| uaddl v16.8H, v0.8B, v1.8B | |
| uaddl2 v20.8H, v0.16B, v1.16B | |
| uaddl v18.8H, v4.8B, v5.8B | |
| uaddl2 v22.8H, v4.16B, v5.16B | |
| 1: subs w3, w3, #2 | |
| ld1 {v0.16B, v1.16B}, [x1], x2 | |
| add v24.8H, v16.8H, v18.8H | |
| NRND add v24.8H, v24.8H, v26.8H | |
| ext v30.16B, v0.16B, v1.16B, #1 | |
| add v1.8H, v20.8H, v22.8H | |
| mshrn v28.8B, v24.8H, #2 | |
| NRND add v1.8H, v1.8H, v26.8H | |
| mshrn2 v28.16B, v1.8H, #2 | |
| \avg | |
| ld1 {v16.16B}, [x0] | |
| urhadd v28.16B, v28.16B, v16.16B | |
| uaddl v16.8H, v0.8B, v30.8B | |
| ld1 {v2.16B, v3.16B}, [x1], x2 | |
| uaddl2 v20.8H, v0.16B, v30.16B | |
| st1 {v28.16B}, [x0], x2 | |
| add v24.8H, v16.8H, v18.8H | |
| NRND add v24.8H, v24.8H, v26.8H | |
| ext v3.16B, v2.16B, v3.16B, #1 | |
| add v0.8H, v20.8H, v22.8H | |
| mshrn v30.8B, v24.8H, #2 | |
| NRND add v0.8H, v0.8H, v26.8H | |
| mshrn2 v30.16B, v0.8H, #2 | |
| \avg | |
| ld1 {v18.16B}, [x0] | |
| urhadd v30.16B, v30.16B, v18.16B | |
| uaddl v18.8H, v2.8B, v3.8B | |
| uaddl2 v22.8H, v2.16B, v3.16B | |
| st1 {v30.16B}, [x0], x2 | |
| b.gt 1b | |
| ld1 {v0.16B, v1.16B}, [x1], x2 | |
| add v24.8H, v16.8H, v18.8H | |
| NRND add v24.8H, v24.8H, v26.8H | |
| ext v30.16B, v0.16B, v1.16B, #1 | |
| add v1.8H, v20.8H, v22.8H | |
| mshrn v28.8B, v24.8H, #2 | |
| NRND add v1.8H, v1.8H, v26.8H | |
| mshrn2 v28.16B, v1.8H, #2 | |
| \avg | |
| ld1 {v16.16B}, [x0] | |
| urhadd v28.16B, v28.16B, v16.16B | |
| uaddl v16.8H, v0.8B, v30.8B | |
| uaddl2 v20.8H, v0.16B, v30.16B | |
| st1 {v28.16B}, [x0], x2 | |
| add v24.8H, v16.8H, v18.8H | |
| NRND add v24.8H, v24.8H, v26.8H | |
| add v0.8H, v20.8H, v22.8H | |
| mshrn v30.8B, v24.8H, #2 | |
| NRND add v0.8H, v0.8H, v26.8H | |
| mshrn2 v30.16B, v0.8H, #2 | |
| \avg | |
| ld1 {v18.16B}, [x0] | |
| urhadd v30.16B, v30.16B, v18.16B | |
| st1 {v30.16B}, [x0], x2 | |
| ret | |
| .endm | |
| .macro pixels8 rnd=1, avg=0 | |
| 1: ld1 {v0.8B}, [x1], x2 | |
| ld1 {v1.8B}, [x1], x2 | |
| ld1 {v2.8B}, [x1], x2 | |
| ld1 {v3.8B}, [x1], x2 | |
| \avg | |
| ld1 {v4.8B}, [x0], x2 | |
| urhadd v0.8B, v0.8B, v4.8B | |
| ld1 {v5.8B}, [x0], x2 | |
| urhadd v1.8B, v1.8B, v5.8B | |
| ld1 {v6.8B}, [x0], x2 | |
| urhadd v2.8B, v2.8B, v6.8B | |
| ld1 {v7.8B}, [x0], x2 | |
| urhadd v3.8B, v3.8B, v7.8B | |
| sub x0, x0, x2, lsl #2 | |
| subs w3, w3, #4 | |
| st1 {v0.8B}, [x0], x2 | |
| st1 {v1.8B}, [x0], x2 | |
| st1 {v2.8B}, [x0], x2 | |
| st1 {v3.8B}, [x0], x2 | |
| b.ne 1b | |
| ret | |
| .endm | |
| .macro pixels8_x2 rnd=1, avg=0 | |
| 1: ld1 {v0.8B, v1.8B}, [x1], x2 | |
| ext v1.8B, v0.8B, v1.8B, #1 | |
| ld1 {v2.8B, v3.8B}, [x1], x2 | |
| ext v3.8B, v2.8B, v3.8B, #1 | |
| subs w3, w3, #2 | |
| avg v0.8B, v0.8B, v1.8B | |
| avg v2.8B, v2.8B, v3.8B | |
| \avg | |
| ld1 {v4.8B}, [x0], x2 | |
| ld1 {v5.8B}, [x0] | |
| urhadd v0.8B, v0.8B, v4.8B | |
| urhadd v2.8B, v2.8B, v5.8B | |
| sub x0, x0, x2 | |
| st1 {v0.8B}, [x0], x2 | |
| st1 {v2.8B}, [x0], x2 | |
| b.ne 1b | |
| ret | |
| .endm | |
| .macro pixels8_y2 rnd=1, avg=0 | |
| sub w3, w3, #2 | |
| ld1 {v0.8B}, [x1], x2 | |
| ld1 {v1.8B}, [x1], x2 | |
| 1: subs w3, w3, #2 | |
| avg v4.8B, v0.8B, v1.8B | |
| ld1 {v0.8B}, [x1], x2 | |
| avg v5.8B, v0.8B, v1.8B | |
| ld1 {v1.8B}, [x1], x2 | |
| \avg | |
| ld1 {v2.8B}, [x0], x2 | |
| ld1 {v3.8B}, [x0] | |
| urhadd v4.8B, v4.8B, v2.8B | |
| urhadd v5.8B, v5.8B, v3.8B | |
| sub x0, x0, x2 | |
| st1 {v4.8B}, [x0], x2 | |
| st1 {v5.8B}, [x0], x2 | |
| b.ne 1b | |
| avg v4.8B, v0.8B, v1.8B | |
| ld1 {v0.8B}, [x1], x2 | |
| avg v5.8B, v0.8B, v1.8B | |
| \avg | |
| ld1 {v2.8B}, [x0], x2 | |
| ld1 {v3.8B}, [x0] | |
| urhadd v4.8B, v4.8B, v2.8B | |
| urhadd v5.8B, v5.8B, v3.8B | |
| sub x0, x0, x2 | |
| st1 {v4.8B}, [x0], x2 | |
| st1 {v5.8B}, [x0], x2 | |
| ret | |
| .endm | |
| .macro pixels8_xy2 rnd=1, avg=0 | |
| sub w3, w3, #2 | |
| ld1 {v0.16B}, [x1], x2 | |
| ld1 {v1.16B}, [x1], x2 | |
| NRND movi v19.8H, #1 | |
| ext v4.16B, v0.16B, v4.16B, #1 | |
| ext v6.16B, v1.16B, v6.16B, #1 | |
| uaddl v16.8H, v0.8B, v4.8B | |
| uaddl v17.8H, v1.8B, v6.8B | |
| 1: subs w3, w3, #2 | |
| ld1 {v0.16B}, [x1], x2 | |
| add v18.8H, v16.8H, v17.8H | |
| ext v4.16B, v0.16B, v4.16B, #1 | |
| NRND add v18.8H, v18.8H, v19.8H | |
| uaddl v16.8H, v0.8B, v4.8B | |
| mshrn v5.8B, v18.8H, #2 | |
| ld1 {v1.16B}, [x1], x2 | |
| add v18.8H, v16.8H, v17.8H | |
| \avg | |
| ld1 {v7.8B}, [x0] | |
| urhadd v5.8B, v5.8B, v7.8B | |
| NRND add v18.8H, v18.8H, v19.8H | |
| st1 {v5.8B}, [x0], x2 | |
| mshrn v7.8B, v18.8H, #2 | |
| \avg | |
| ld1 {v5.8B}, [x0] | |
| urhadd v7.8B, v7.8B, v5.8B | |
| ext v6.16B, v1.16B, v6.16B, #1 | |
| uaddl v17.8H, v1.8B, v6.8B | |
| st1 {v7.8B}, [x0], x2 | |
| b.gt 1b | |
| ld1 {v0.16B}, [x1], x2 | |
| add v18.8H, v16.8H, v17.8H | |
| ext v4.16B, v0.16B, v4.16B, #1 | |
| NRND add v18.8H, v18.8H, v19.8H | |
| uaddl v16.8H, v0.8B, v4.8B | |
| mshrn v5.8B, v18.8H, #2 | |
| add v18.8H, v16.8H, v17.8H | |
| \avg | |
| ld1 {v7.8B}, [x0] | |
| urhadd v5.8B, v5.8B, v7.8B | |
| NRND add v18.8H, v18.8H, v19.8H | |
| st1 {v5.8B}, [x0], x2 | |
| mshrn v7.8B, v18.8H, #2 | |
| \avg | |
| ld1 {v5.8B}, [x0] | |
| urhadd v7.8B, v7.8B, v5.8B | |
| st1 {v7.8B}, [x0], x2 | |
| ret | |
| .endm | |
| .macro pixfunc pfx, name, suf, rnd=1, avg=0 | |
| \rnd | |
| avg rd, rn, rm | |
| urhadd \rd, \rn, \rm | |
| mshrn rd, rn, rm | |
| rshrn \rd, \rn, \rm | |
| mshrn2 rd, rn, rm | |
| rshrn2 \rd, \rn, \rm | |
| NRND insn:vararg | |
| avg rd, rn, rm | |
| uhadd \rd, \rn, \rm | |
| mshrn rd, rn, rm | |
| shrn \rd, \rn, \rm | |
| mshrn2 rd, rn, rm | |
| shrn2 \rd, \rn, \rm | |
| NRND insn:vararg | |
| \insn | |
| function ff_\pfx\name\suf\()_neon, =1 | |
| \name \rnd, \avg | |
| endfunc | |
| .purgem avg | |
| .purgem mshrn | |
| .purgem mshrn2 | |
| .purgem NRND | |
| .endm | |
| .macro pixfunc2 pfx, name, avg=0 | |
| pixfunc \pfx, \name, rnd=1, avg=\avg | |
| pixfunc \pfx, \name, _no_rnd, rnd=0, avg=\avg | |
| .endm | |
| function ff_put_h264_qpel16_mc00_neon, =1 | |
| mov w3, #16 | |
| endfunc | |
| pixfunc put_, pixels16, avg=0 | |
| pixfunc2 put_, pixels16_x2, avg=0 | |
| pixfunc2 put_, pixels16_y2, avg=0 | |
| pixfunc2 put_, pixels16_xy2, avg=0 | |
| function ff_avg_h264_qpel16_mc00_neon, =1 | |
| mov w3, #16 | |
| endfunc | |
| pixfunc avg_, pixels16, avg=1 | |
| pixfunc2 avg_, pixels16_x2, avg=1 | |
| pixfunc2 avg_, pixels16_y2, avg=1 | |
| pixfunc2 avg_, pixels16_xy2, avg=1 | |
| function ff_put_h264_qpel8_mc00_neon, =1 | |
| mov w3, #8 | |
| endfunc | |
| pixfunc put_, pixels8, avg=0 | |
| pixfunc2 put_, pixels8_x2, avg=0 | |
| pixfunc2 put_, pixels8_y2, avg=0 | |
| pixfunc2 put_, pixels8_xy2, avg=0 | |
| function ff_avg_h264_qpel8_mc00_neon, =1 | |
| mov w3, #8 | |
| endfunc | |
| pixfunc avg_, pixels8, avg=1 | |
| pixfunc avg_, pixels8_x2, avg=1 | |
| pixfunc avg_, pixels8_y2, avg=1 | |
| pixfunc avg_, pixels8_xy2, avg=1 | |