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.
388 lines
12 KiB
388 lines
12 KiB
/* |
|
* Loongson LSX optimized swscale |
|
* |
|
* Copyright (c) 2023 Loongson Technology Corporation Limited |
|
* Contributed by Lu Wang <wanglu@loongson.cn> |
|
* |
|
* 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 "libavcodec/loongarch/loongson_asm.S" |
|
|
|
/* static void yuv2planeX_8_lsx(const int16_t *filter, int filterSize, |
|
* const int16_t **src, uint8_t *dest, int dstW, |
|
* const uint8_t *dither, int offset) |
|
*/ |
|
function yuv2planeX_8_lsx |
|
addi.w t1, a6, 1 |
|
addi.w t2, a6, 2 |
|
addi.w t3, a6, 3 |
|
addi.w t4, a6, 4 |
|
addi.w t5, a6, 5 |
|
addi.w t6, a6, 6 |
|
addi.w t7, a6, 7 |
|
andi t0, a6, 7 |
|
andi t1, t1, 7 |
|
andi t2, t2, 7 |
|
andi t3, t3, 7 |
|
andi t4, t4, 7 |
|
andi t5, t5, 7 |
|
andi t6, t6, 7 |
|
andi t7, t7, 7 |
|
ldx.bu t0, a5, t0 |
|
ldx.bu t1, a5, t1 |
|
ldx.bu t2, a5, t2 |
|
ldx.bu t3, a5, t3 |
|
ldx.bu t4, a5, t4 |
|
ldx.bu t5, a5, t5 |
|
ldx.bu t6, a5, t6 |
|
ldx.bu t7, a5, t7 |
|
vreplgr2vr.w vr0, t0 |
|
vreplgr2vr.w vr1, t1 |
|
vreplgr2vr.w vr2, t2 |
|
vreplgr2vr.w vr3, t3 |
|
vreplgr2vr.w vr4, t4 |
|
vreplgr2vr.w vr5, t5 |
|
vreplgr2vr.w vr6, t6 |
|
vreplgr2vr.w vr7, t7 |
|
vilvl.w vr0, vr2, vr0 |
|
vilvl.w vr4, vr6, vr4 |
|
vilvl.w vr1, vr3, vr1 |
|
vilvl.w vr5, vr7, vr5 |
|
vilvl.d vr12, vr4, vr0 |
|
vilvl.d vr13, vr5, vr1 |
|
li.w t5, 0 |
|
li.w t8, 8 |
|
bge a4, t8, .WIDTH8 |
|
blt zero, a4, .WIDTH |
|
b .END |
|
|
|
.WIDTH8: |
|
li.d t1, 0 |
|
li.d t4, 0 |
|
vslli.w vr2, vr12, 12 |
|
vslli.w vr3, vr13, 12 |
|
move t3, a0 |
|
|
|
.FILTERSIZE8: |
|
ldx.d t2, a2, t1 |
|
vldx vr4, t2, t5 |
|
vldrepl.h vr5, t3, 0 |
|
vmaddwev.w.h vr2, vr4, vr5 |
|
vmaddwod.w.h vr3, vr4, vr5 |
|
addi.d t1, t1, 8 |
|
addi.d t3, t3, 2 |
|
addi.d t4, t4, 1 |
|
blt t4, a1, .FILTERSIZE8 |
|
vsrai.w vr2, vr2, 19 |
|
vsrai.w vr3, vr3, 19 |
|
vclip255.w vr2, vr2 |
|
vclip255.w vr3, vr3 |
|
vpickev.h vr2, vr3, vr2 |
|
vpickev.b vr2, vr2, vr2 |
|
vbsrl.v vr3, vr2, 4 |
|
vilvl.b vr2, vr3, vr2 |
|
fst.d f2, a3, 0 |
|
addi.d t5, t5, 16 |
|
addi.d a4, a4, -8 |
|
addi.d a3, a3, 8 |
|
bge a4, t8, .WIDTH8 |
|
blt zero, a4, .WIDTH |
|
b .END |
|
|
|
.WIDTH: |
|
li.d t1, 0 |
|
li.d t4, 0 |
|
vslli.w vr2, vr12, 12 |
|
vslli.w vr3, vr13, 12 |
|
.FILTERSIZE: |
|
ldx.d t2, a2, t1 |
|
vldx vr4, t2, t5 |
|
vldrepl.h vr5, a0, 0 |
|
vmaddwev.w.h vr2, vr4, vr5 |
|
vmaddwod.w.h vr3, vr4, vr5 |
|
addi.d t1, t1, 8 |
|
addi.d a0, a0, 2 |
|
addi.d t4, t4, 1 |
|
blt t4, a1, .FILTERSIZE |
|
vsrai.w vr2, vr2, 19 |
|
vsrai.w vr3, vr3, 19 |
|
vclip255.w vr2, vr2 |
|
vclip255.w vr3, vr3 |
|
vpickev.h vr2, vr3, vr2 |
|
vpickev.b vr2, vr2, vr2 |
|
vbsrl.v vr3, vr2, 4 |
|
vilvl.b vr2, vr3, vr2 |
|
|
|
.DEST: |
|
vstelm.b vr2, a3, 0, 0 |
|
vbsrl.v vr2, vr2, 1 |
|
addi.d a4, a4, -1 |
|
addi.d a3, a3, 1 |
|
blt zero, a4, .DEST |
|
.END: |
|
endfunc |
|
|
|
/* |
|
* void yuv2plane1_8_lsx(const int16_t *src, uint8_t *dest, int dstW, |
|
* const uint8_t *dither, int offset) |
|
*/ |
|
function yuv2plane1_8_lsx |
|
addi.w t1, a4, 1 |
|
addi.w t2, a4, 2 |
|
addi.w t3, a4, 3 |
|
addi.w t4, a4, 4 |
|
addi.w t5, a4, 5 |
|
addi.w t6, a4, 6 |
|
addi.w t7, a4, 7 |
|
andi t0, a4, 7 |
|
andi t1, t1, 7 |
|
andi t2, t2, 7 |
|
andi t3, t3, 7 |
|
andi t4, t4, 7 |
|
andi t5, t5, 7 |
|
andi t6, t6, 7 |
|
andi t7, t7, 7 |
|
ldx.bu t0, a3, t0 |
|
ldx.bu t1, a3, t1 |
|
ldx.bu t2, a3, t2 |
|
ldx.bu t3, a3, t3 |
|
ldx.bu t4, a3, t4 |
|
ldx.bu t5, a3, t5 |
|
ldx.bu t6, a3, t6 |
|
ldx.bu t7, a3, t7 |
|
vinsgr2vr.h vr1, t0, 0 |
|
vinsgr2vr.h vr1, t1, 1 |
|
vinsgr2vr.h vr1, t2, 2 |
|
vinsgr2vr.h vr1, t3, 3 |
|
vinsgr2vr.h vr1, t4, 4 |
|
vinsgr2vr.h vr1, t5, 5 |
|
vinsgr2vr.h vr1, t6, 6 |
|
vinsgr2vr.h vr1, t7, 7 |
|
vsub.h vr0, vr0, vr0 |
|
vilvl.h vr2, vr0, vr1 |
|
vilvh.h vr3, vr0, vr1 |
|
|
|
andi t8, a2, 7 |
|
srli.d a2, a2, 3 |
|
beqz a2, 2f |
|
1: |
|
vld vr1, a0, 0 |
|
addi.d a0, a0, 16 |
|
vshuf4i.d vr0, vr1, 8 |
|
vexth.w.h vr4, vr0 |
|
vexth.w.h vr5, vr1 |
|
|
|
vadd.w vr4, vr2, vr4 |
|
vadd.w vr5, vr3, vr5 |
|
vsrai.w vr4, vr4, 7 |
|
vsrai.w vr5, vr5, 7 |
|
vclip255.w vr4, vr4 |
|
vclip255.w vr5, vr5 |
|
vpickev.h vr1, vr5, vr4 |
|
vpickev.b vr1, vr1, vr1 |
|
fst.d f1, a1, 0 |
|
addi.d a1, a1, 8 |
|
addi.d a2, a2, -1 |
|
bnez a2, 1b |
|
2: |
|
beqz t8, 4f |
|
3: |
|
add.w a4, a4, t8 |
|
addi.w t1, a4, 1 |
|
addi.w t2, a4, 2 |
|
addi.w t3, a4, 3 |
|
addi.w t4, a4, 4 |
|
addi.w t5, a4, 5 |
|
addi.w t6, a4, 6 |
|
addi.w t7, a4, 7 |
|
andi t0, a4, 7 |
|
andi t1, t1, 7 |
|
andi t2, t2, 7 |
|
andi t3, t3, 7 |
|
andi t4, t4, 7 |
|
andi t5, t5, 7 |
|
andi t6, t6, 7 |
|
andi t7, t7, 7 |
|
ldx.bu t0, a3, t0 |
|
ldx.bu t1, a3, t1 |
|
ldx.bu t2, a3, t2 |
|
ldx.bu t3, a3, t3 |
|
ldx.bu t4, a3, t4 |
|
ldx.bu t5, a3, t5 |
|
ldx.bu t6, a3, t6 |
|
ldx.bu t7, a3, t7 |
|
vinsgr2vr.h vr1, t0, 0 |
|
vinsgr2vr.h vr1, t1, 1 |
|
vinsgr2vr.h vr1, t2, 2 |
|
vinsgr2vr.h vr1, t3, 3 |
|
vinsgr2vr.h vr1, t4, 4 |
|
vinsgr2vr.h vr1, t5, 5 |
|
vinsgr2vr.h vr1, t6, 6 |
|
vinsgr2vr.h vr1, t7, 7 |
|
vsub.h vr0, vr0, vr0 |
|
vilvl.h vr2, vr0, vr1 |
|
vilvh.h vr3, vr0, vr1 |
|
|
|
addi.d a0, a0, -16 |
|
add.d a0, a0, t8 |
|
add.d a0, a0, t8 |
|
addi.d a1, a1, -8 |
|
add.d a1, a1, t8 |
|
|
|
vld vr1, a0, 0 |
|
vshuf4i.d vr0, vr1, 8 |
|
vexth.w.h vr4, vr0 |
|
vexth.w.h vr5, vr1 |
|
|
|
vadd.w vr4, vr2, vr4 |
|
vadd.w vr5, vr3, vr5 |
|
vsrai.w vr4, vr4, 7 |
|
vsrai.w vr5, vr5, 7 |
|
vclip255.w vr4, vr4 |
|
vclip255.w vr5, vr5 |
|
vpickev.h vr1, vr5, vr4 |
|
vpickev.b vr1, vr1, vr1 |
|
fst.d f1, a1, 0 |
|
4: |
|
endfunc |
|
|
|
function yuv2plane1_8_lasx |
|
addi.w t1, a4, 1 |
|
addi.w t2, a4, 2 |
|
addi.w t3, a4, 3 |
|
addi.w t4, a4, 4 |
|
addi.w t5, a4, 5 |
|
addi.w t6, a4, 6 |
|
addi.w t7, a4, 7 |
|
andi t0, a4, 7 |
|
andi t1, t1, 7 |
|
andi t2, t2, 7 |
|
andi t3, t3, 7 |
|
andi t4, t4, 7 |
|
andi t5, t5, 7 |
|
andi t6, t6, 7 |
|
andi t7, t7, 7 |
|
ldx.bu t0, a3, t0 |
|
ldx.bu t1, a3, t1 |
|
ldx.bu t2, a3, t2 |
|
ldx.bu t3, a3, t3 |
|
ldx.bu t4, a3, t4 |
|
ldx.bu t5, a3, t5 |
|
ldx.bu t6, a3, t6 |
|
ldx.bu t7, a3, t7 |
|
vinsgr2vr.h vr1, t0, 0 |
|
vinsgr2vr.h vr1, t1, 1 |
|
vinsgr2vr.h vr1, t2, 2 |
|
vinsgr2vr.h vr1, t3, 3 |
|
vinsgr2vr.h vr1, t4, 4 |
|
vinsgr2vr.h vr1, t5, 5 |
|
vinsgr2vr.h vr1, t6, 6 |
|
vinsgr2vr.h vr1, t7, 7 |
|
xvpermi.q xr1, xr1, 0 |
|
xvsub.h xr0, xr0, xr0 |
|
xvilvl.h xr2, xr0, xr1 |
|
xvilvh.h xr3, xr0, xr1 |
|
|
|
andi t8, a2, 15 |
|
srli.d a2, a2, 4 |
|
beqz a2, 2f |
|
1: |
|
xvld xr1, a0, 0 |
|
addi.d a0, a0, 32 |
|
xvpermi.d xr0, xr1, 0xa0 |
|
xvexth.w.h xr4, xr0 |
|
xvexth.w.h xr5, xr1 |
|
|
|
xvadd.w xr4, xr2, xr4 |
|
xvadd.w xr5, xr3, xr5 |
|
xvsrai.w xr4, xr4, 7 |
|
xvsrai.w xr5, xr5, 7 |
|
xvclip255.w xr4, xr4 |
|
xvclip255.w xr5, xr5 |
|
xvpickev.h xr1, xr5, xr4 |
|
xvpickev.b xr0, xr1, xr1 |
|
xvpermi.q xr1, xr0, 1 |
|
fst.d f0, a1, 0 |
|
fst.d f1, a1, 8 |
|
addi.d a1, a1, 16 |
|
addi.d a2, a2, -1 |
|
bnez a2, 1b |
|
2: |
|
beqz t8, 4f |
|
3: |
|
add.w a4, a4, t8 |
|
addi.w t1, a4, 1 |
|
addi.w t2, a4, 2 |
|
addi.w t3, a4, 3 |
|
addi.w t4, a4, 4 |
|
addi.w t5, a4, 5 |
|
addi.w t6, a4, 6 |
|
addi.w t7, a4, 7 |
|
andi t0, a4, 7 |
|
andi t1, t1, 7 |
|
andi t2, t2, 7 |
|
andi t3, t3, 7 |
|
andi t4, t4, 7 |
|
andi t5, t5, 7 |
|
andi t6, t6, 7 |
|
andi t7, t7, 7 |
|
ldx.bu t0, a3, t0 |
|
ldx.bu t1, a3, t1 |
|
ldx.bu t2, a3, t2 |
|
ldx.bu t3, a3, t3 |
|
ldx.bu t4, a3, t4 |
|
ldx.bu t5, a3, t5 |
|
ldx.bu t6, a3, t6 |
|
ldx.bu t7, a3, t7 |
|
vinsgr2vr.h vr1, t0, 0 |
|
vinsgr2vr.h vr1, t1, 1 |
|
vinsgr2vr.h vr1, t2, 2 |
|
vinsgr2vr.h vr1, t3, 3 |
|
vinsgr2vr.h vr1, t4, 4 |
|
vinsgr2vr.h vr1, t5, 5 |
|
vinsgr2vr.h vr1, t6, 6 |
|
vinsgr2vr.h vr1, t7, 7 |
|
xvpermi.q xr1, xr1, 0 |
|
xvsub.h xr0, xr0, xr0 |
|
xvilvl.h xr2, xr0, xr1 |
|
xvilvh.h xr3, xr0, xr1 |
|
|
|
addi.d a0, a0, -32 |
|
add.d a0, a0, t8 |
|
add.d a0, a0, t8 |
|
addi.d a1, a1, -16 |
|
add.d a1, a1, t8 |
|
|
|
xvld xr1, a0, 0 |
|
xvpermi.d xr0, xr1, 0xa0 |
|
xvexth.w.h xr4, xr0 |
|
xvexth.w.h xr5, xr1 |
|
|
|
xvadd.w xr4, xr2, xr4 |
|
xvadd.w xr5, xr3, xr5 |
|
xvsrai.w xr4, xr4, 7 |
|
xvsrai.w xr5, xr5, 7 |
|
xvclip255.w xr4, xr4 |
|
xvclip255.w xr5, xr5 |
|
xvpickev.h xr1, xr5, xr4 |
|
xvpickev.b xr0, xr1, xr1 |
|
xvpermi.q xr1, xr0, 1 |
|
fst.d f0, a1, 0 |
|
fst.d f1, a1, 8 |
|
4: |
|
endfunc
|
|
|