mirror of https://github.com/FFmpeg/FFmpeg.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
208 lines
6.5 KiB
208 lines
6.5 KiB
/* |
|
* Copyright (c) 2023 Institue of Software Chinese Academy of Sciences (ISCAS). |
|
* Copyright (c) 2024 Rémi Denis-Courmont. |
|
* |
|
* 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/riscv/asm.S" |
|
|
|
func ff_vc1_inv_trans_8x8_dc_rvv, zve64x |
|
lh t2, (a2) |
|
vsetivli zero, 8, e8, mf2, ta, ma |
|
vlse64.v v0, (a0), a1 |
|
sh1add t2, t2, t2 |
|
addi t2, t2, 1 |
|
srai t2, t2, 1 |
|
sh1add t2, t2, t2 |
|
addi t2, t2, 16 |
|
srai t2, t2, 5 |
|
li t0, 8*8 |
|
vsetvli zero, t0, e16, m8, ta, ma |
|
vzext.vf2 v8, v0 |
|
vadd.vx v8, v8, t2 |
|
vmax.vx v8, v8, zero |
|
vsetvli zero, zero, e8, m4, ta, ma |
|
vnclipu.wi v0, v8, 0 |
|
vsetivli zero, 8, e8, mf2, ta, ma |
|
vsse64.v v0, (a0), a1 |
|
ret |
|
endfunc |
|
|
|
func ff_vc1_inv_trans_4x8_dc_rvv, zve32x |
|
lh t2, (a2) |
|
vsetivli zero, 8, e8, mf2, ta, ma |
|
vlse32.v v0, (a0), a1 |
|
slli t1, t2, 4 |
|
add t2, t2, t1 |
|
addi t2, t2, 4 |
|
srai t2, t2, 3 |
|
sh1add t2, t2, t2 |
|
slli t2, t2, 2 |
|
addi t2, t2, 64 |
|
srai t2, t2, 7 |
|
li t0, 4*8 |
|
vsetvli zero, t0, e16, m4, ta, ma |
|
vzext.vf2 v4, v0 |
|
vadd.vx v4, v4, t2 |
|
vmax.vx v4, v4, zero |
|
vsetvli zero, zero, e8, m2, ta, ma |
|
vnclipu.wi v0, v4, 0 |
|
vsetivli zero, 8, e8, mf2, ta, ma |
|
vsse32.v v0, (a0), a1 |
|
ret |
|
endfunc |
|
|
|
func ff_vc1_inv_trans_8x4_dc_rvv, zve64x |
|
lh t2, (a2) |
|
vsetivli zero, 4, e8, mf4, ta, ma |
|
vlse64.v v0, (a0), a1 |
|
sh1add t2, t2, t2 |
|
addi t2, t2, 1 |
|
srai t2, t2, 1 |
|
slli t1, t2, 4 |
|
add t2, t2, t1 |
|
addi t2, t2, 64 |
|
srai t2, t2, 7 |
|
li t0, 8*4 |
|
vsetvli zero, t0, e16, m4, ta, ma |
|
vzext.vf2 v4, v0 |
|
vadd.vx v4, v4, t2 |
|
vmax.vx v4, v4, zero |
|
vsetvli zero, zero, e8, m2, ta, ma |
|
vnclipu.wi v0, v4, 0 |
|
vsetivli zero, 4, e8, mf4, ta, ma |
|
vsse64.v v0, (a0), a1 |
|
ret |
|
endfunc |
|
|
|
func ff_vc1_inv_trans_4x4_dc_rvv, zve32x |
|
lh t2, (a2) |
|
vsetivli zero, 4, e8, mf4, ta, ma |
|
vlse32.v v0, (a0), a1 |
|
slli t1, t2, 4 |
|
add t2, t2, t1 |
|
addi t2, t2, 4 |
|
srai t2, t2, 3 |
|
slli t1, t2, 4 |
|
add t2, t2, t1 |
|
addi t2, t2, 64 |
|
srai t2, t2, 7 |
|
vsetivli zero, 4*4, e16, m2, ta, ma |
|
vzext.vf2 v2, v0 |
|
vadd.vx v2, v2, t2 |
|
vmax.vx v2, v2, zero |
|
vsetvli zero, zero, e8, m1, ta, ma |
|
vnclipu.wi v0, v2, 0 |
|
vsetivli zero, 4, e8, mf4, ta, ma |
|
vsse32.v v0, (a0), a1 |
|
ret |
|
endfunc |
|
|
|
.macro mspel_op op pos n1 n2 |
|
add t1, \pos, a2 |
|
v\op\()e8.v v\n1, (\pos) |
|
sh1add \pos, a2, \pos |
|
v\op\()e8.v v\n2, (t1) |
|
.endm |
|
|
|
.macro mspel_op_all op pos a1 a2 a3 a4 a5 a6 a7 a8 a9 a10 a11 a12 a13 a14 a15 a16 |
|
mspel_op \op \pos \a1 \a2 |
|
mspel_op \op \pos \a3 \a4 |
|
mspel_op \op \pos \a5 \a6 |
|
mspel_op \op \pos \a7 \a8 |
|
mspel_op \op \pos \a9 \a10 |
|
mspel_op \op \pos \a11 \a12 |
|
mspel_op \op \pos \a13 \a14 |
|
mspel_op \op \pos \a15 \a16 |
|
.endm |
|
|
|
func ff_avg_pixels16x16_rvv, zve32x |
|
li t0, 16 |
|
vsetivli zero, 16, e8, m1, ta, ma |
|
j 1f |
|
endfunc |
|
|
|
func ff_avg_pixels8x8_rvv, zve32x |
|
li t0, 8 |
|
vsetivli zero, 8, e8, mf2, ta, ma |
|
1: |
|
csrwi vxrm, 0 |
|
2: |
|
vle8.v v16, (a1) |
|
addi t0, t0, -1 |
|
vle8.v v8, (a0) |
|
add a1, a1, a2 |
|
vaaddu.vv v16, v16, v8 |
|
vse8.v v16, (a0) |
|
add a0, a0, a2 |
|
bnez t0, 2b |
|
|
|
ret |
|
endfunc |
|
|
|
func ff_vc1_unescape_buffer_rvv, zve32x |
|
vsetivli zero, 2, e8, m1, ta, ma |
|
vmv.v.i v8, -1 |
|
li t4, 1 |
|
vmv.v.i v12, -1 |
|
li t3, -1 |
|
mv t5, a2 |
|
blez a1, 3f |
|
1: |
|
vsetvli t0, a1, e8, m4, ta, ma |
|
vle8.v v16, (a0) |
|
vslideup.vi v8, v16, 2 |
|
addi t0, t0, -1 # we cannot fully process the last element |
|
vslideup.vi v12, v16, 1 |
|
vslide1down.vx v20, v16, t3 |
|
vsetvli zero, t0, e8, m4, ta, ma |
|
vmseq.vi v0, v8, 0 |
|
vmseq.vi v1, v12, 0 |
|
vmseq.vi v2, v16, 3 |
|
vmand.mm v0, v0, v1 |
|
vmsltu.vi v3, v20, 4 |
|
vmand.mm v0, v0, v2 |
|
vmand.mm v0, v0, v3 |
|
vfirst.m t2, v0 |
|
bgez t2, 4f # found an escape byte? |
|
|
|
vse8.v v16, (a2) |
|
addi t2, t0, -2 |
|
add a2, a2, t0 |
|
2: |
|
vslidedown.vx v8, v16, t2 |
|
sub a1, a1, t0 |
|
vslidedown.vi v12, v8, 1 |
|
add a0, a0, t0 |
|
bgtu a1, t4, 1b // size > 1 |
|
|
|
lb t0, (a0) |
|
sb t0, (a2) # copy last byte (cannot be escaped) |
|
addi a2, a2, 1 |
|
3: |
|
sub a0, a2, t5 |
|
ret |
|
4: |
|
vsetvli zero, t2, e8, m4, ta, ma |
|
vse8.v v16, (a2) |
|
addi t0, t2, 1 |
|
add a2, a2, t2 |
|
addi t2, t2, -1 |
|
vsetvli zero, t0, e8, m4, ta, ma |
|
j 2b |
|
endfunc
|
|
|