2018-05-06 08:54:49 +00:00
|
|
|
/*
|
|
|
|
* Copyright (c) 2018 Clément Bœsch <u pkh me>
|
|
|
|
*
|
|
|
|
* 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"
|
|
|
|
|
|
|
|
// acc_sum_store(ABCD) = {X+A, X+A+B, X+A+B+C, X+A+B+C+D}
|
|
|
|
.macro acc_sum_store x, xb
|
2023-10-17 11:16:24 +00:00
|
|
|
dup v24.4s, v24.s[3] // ...X -> XXXX
|
|
|
|
ext v25.16b, v26.16b, \xb, #12 // ext(0000,ABCD,12)=0ABC
|
|
|
|
add v24.4s, v24.4s, \x // XXXX+ABCD={X+A,X+B,X+C,X+D}
|
|
|
|
add v24.4s, v24.4s, v25.4s // {X+A,X+B+A,X+C+B,X+D+C} (+0ABC)
|
|
|
|
ext v25.16b, v26.16b, v25.16b, #12 // ext(0000,0ABC,12)=00AB
|
|
|
|
add v24.4s, v24.4s, v25.4s // {X+A,X+B+A,X+C+B+A,X+D+C+B} (+00AB)
|
|
|
|
ext v25.16b, v26.16b, v25.16b, #12 // ext(0000,00AB,12)=000A
|
|
|
|
add v24.4s, v24.4s, v25.4s // {X+A,X+B+A,X+C+B+A,X+D+C+B+A} (+000A)
|
|
|
|
st1 {v24.4s}, [x0], #16 // write 4x32-bit final values
|
2018-05-06 08:54:49 +00:00
|
|
|
.endm
|
|
|
|
|
|
|
|
function ff_compute_safe_ssd_integral_image_neon, export=1
|
2023-10-17 11:16:24 +00:00
|
|
|
movi v26.4s, #0 // used as zero for the "rotations" in acc_sum_store
|
2023-10-17 11:24:47 +00:00
|
|
|
sub x3, x3, w6, uxtw // s1 padding (s1_linesize - w)
|
|
|
|
sub x5, x5, w6, uxtw // s2 padding (s2_linesize - w)
|
|
|
|
sub x9, x0, w1, uxtw #2 // dst_top
|
|
|
|
sub x1, x1, w6, uxtw // dst padding (dst_linesize_32 - w)
|
2018-05-06 08:54:49 +00:00
|
|
|
lsl x1, x1, #2 // dst padding expressed in bytes
|
|
|
|
1: mov w10, w6 // width copy for each line
|
|
|
|
sub x0, x0, #16 // beginning of the dst line minus 4 sums
|
|
|
|
sub x8, x9, #4 // dst_top-1
|
2023-10-17 11:16:24 +00:00
|
|
|
ld1 {v24.4s}, [x0], #16 // load ...X (contextual last sums)
|
|
|
|
2: ld1 {v0.16b}, [x2], #16 // s1[x + 0..15]
|
|
|
|
ld1 {v1.16b}, [x4], #16 // s2[x + 0..15]
|
|
|
|
ld1 {v16.4s,v17.4s}, [x8], #32 // dst_top[x + 0..7 - 1]
|
|
|
|
usubl v2.8h, v0.8b, v1.8b // d[x + 0..7] = s1[x + 0..7] - s2[x + 0..7]
|
|
|
|
usubl2 v3.8h, v0.16b, v1.16b // d[x + 8..15] = s1[x + 8..15] - s2[x + 8..15]
|
|
|
|
ld1 {v18.4s,v19.4s}, [x8], #32 // dst_top[x + 8..15 - 1]
|
|
|
|
smull v4.4s, v2.4h, v2.4h // d[x + 0..3]^2
|
|
|
|
smull2 v5.4s, v2.8h, v2.8h // d[x + 4..7]^2
|
|
|
|
ld1 {v20.4s,v21.4s}, [x9], #32 // dst_top[x + 0..7]
|
|
|
|
smull v6.4s, v3.4h, v3.4h // d[x + 8..11]^2
|
|
|
|
smull2 v7.4s, v3.8h, v3.8h // d[x + 12..15]^2
|
|
|
|
ld1 {v22.4s,v23.4s}, [x9], #32 // dst_top[x + 8..15]
|
|
|
|
sub v0.4s, v20.4s, v16.4s // dst_top[x + 0..3] - dst_top[x + 0..3 - 1]
|
|
|
|
sub v1.4s, v21.4s, v17.4s // dst_top[x + 4..7] - dst_top[x + 4..7 - 1]
|
|
|
|
add v0.4s, v0.4s, v4.4s // + d[x + 0..3]^2
|
|
|
|
add v1.4s, v1.4s, v5.4s // + d[x + 4..7]^2
|
|
|
|
sub v2.4s, v22.4s, v18.4s // dst_top[x + 8..11] - dst_top[x + 8..11 - 1]
|
|
|
|
sub v3.4s, v23.4s, v19.4s // dst_top[x + 12..15] - dst_top[x + 12..15 - 1]
|
|
|
|
add v2.4s, v2.4s, v6.4s // + d[x + 8..11]^2
|
|
|
|
add v3.4s, v3.4s, v7.4s // + d[x + 12..15]^2
|
|
|
|
acc_sum_store v0.4s, v0.16b // accumulate and store dst[ 0..3]
|
|
|
|
acc_sum_store v1.4s, v1.16b // accumulate and store dst[ 4..7]
|
|
|
|
acc_sum_store v2.4s, v2.16b // accumulate and store dst[ 8..11]
|
|
|
|
acc_sum_store v3.4s, v3.16b // accumulate and store dst[12..15]
|
2018-05-06 08:54:49 +00:00
|
|
|
subs w10, w10, #16 // width dec
|
|
|
|
b.ne 2b // loop til next line
|
|
|
|
add x2, x2, x3 // skip to next line (s1)
|
|
|
|
add x4, x4, x5 // skip to next line (s2)
|
|
|
|
add x0, x0, x1 // skip to next line (dst)
|
|
|
|
add x9, x9, x1 // skip to next line (dst_top)
|
|
|
|
subs w7, w7, #1 // height dec
|
|
|
|
b.ne 1b
|
|
|
|
ret
|
|
|
|
endfunc
|