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.
115 lines
3.2 KiB
115 lines
3.2 KiB
/* |
|
* Copyright (c) 2024 Institue of Software Chinese Academy of Sciences (ISCAS). |
|
* |
|
* 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" |
|
|
|
.macro avgdc size |
|
vwredsumu.vs v16, v8, v16 |
|
vsetivli zero, 1, e16, m1, ta, ma |
|
vmv.x.s t1, v16 |
|
addi t1, t1, 1 << (\size - 1) |
|
srai t1, t1, \size |
|
.endm |
|
|
|
.macro getdc type size |
|
.ifc \type,top |
|
vmv.v.x v16, zero |
|
vle8.v v8, (a3) |
|
avgdc \size |
|
.elseif \type == left |
|
vmv.v.x v16, zero |
|
vle8.v v8, (a2) |
|
avgdc \size |
|
.elseif \type == dc |
|
vmv.v.x v16, zero |
|
vle8.v v8, (a2) |
|
vwredsumu.vs v16, v8, v16 |
|
vle8.v v8, (a3) |
|
avgdc \size |
|
.else |
|
li t1, \type |
|
.endif |
|
.endm |
|
|
|
.macro dc_e32 type size n restore |
|
.ifc \size,32 |
|
li t0, 32 |
|
vsetvli zero, t0, e8, m2, ta, ma |
|
.else |
|
vsetivli zero, 16, e8, m1, ta, ma |
|
.endif |
|
getdc \type \n |
|
|
|
.if \restore == 1 && \size == 32 |
|
vsetvli zero, t0, e8, m2, ta, ma |
|
.elseif \restore == 1 && \size == 16 |
|
vsetivli zero, 16, e8, m1, ta, ma |
|
.endif |
|
vmv.v.x v0, t1 |
|
|
|
.rept \size |
|
vse8.v v0, (a0) |
|
add a0, a0, a1 |
|
.endr |
|
|
|
ret |
|
.endm |
|
|
|
.macro dc_e64 type size n restore |
|
vsetivli zero, 8, e8, mf2, ta, ma |
|
getdc \type \n |
|
|
|
li t0, 64 |
|
vsetvli zero, t0, e8, m4, ta, ma |
|
vmv.v.x v0, t1 |
|
vsetivli zero, 8, e8, mf2, ta, ma |
|
vsse64.v v0, (a0), a1 |
|
|
|
ret |
|
.endm |
|
|
|
.macro func_dc name size type n restore ext |
|
func ff_\()\name\()_\()\size\()x\size\()_rvv, \ext |
|
.ifc \size,8 |
|
dc_e64 \type \size \n \restore |
|
.else |
|
dc_e32 \type \size \n \restore |
|
.endif |
|
endfunc |
|
.endm |
|
|
|
func_dc dc_127 32 127 0 0 zve32x |
|
func_dc dc_127 16 127 0 0 zve32x |
|
func_dc dc_127 8 127 0 0 zve64x |
|
func_dc dc_128 32 128 0 0 zve32x |
|
func_dc dc_128 16 128 0 0 zve32x |
|
func_dc dc_128 8 128 0 0 zve64x |
|
func_dc dc_129 32 129 0 0 zve32x |
|
func_dc dc_129 16 129 0 0 zve32x |
|
func_dc dc_129 8 129 0 0 zve64x |
|
func_dc dc 32 dc 6 1 zve32x |
|
func_dc dc 16 dc 5 1 zve32x |
|
func_dc dc 8 dc 4 0 zve64x |
|
func_dc dc_left 32 left 5 1 zve32x |
|
func_dc dc_left 16 left 4 1 zve32x |
|
func_dc dc_left 8 left 3 0 zve64x |
|
func_dc dc_top 32 top 5 1 zve32x |
|
func_dc dc_top 16 top 4 1 zve32x |
|
func_dc dc_top 8 top 3 0 zve64x
|
|
|