123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172 |
- /*
- * Copyright (C) 2022 Loongson Technology Corporation Limited
- * Contributed by Hao Chen(chenhao@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 "swscale_loongarch.h"
- #include "libswscale/swscale_internal.h"
- #include "libswscale/rgb2rgb.h"
- #include "libavutil/loongarch/cpu.h"
- av_cold void ff_sws_init_range_convert_loongarch(SwsContext *c)
- {
- int cpu_flags = av_get_cpu_flags();
- if (have_lsx(cpu_flags)) {
- if (c->srcRange != c->dstRange && !isAnyRGB(c->dstFormat)) {
- if (c->dstBpc <= 14) {
- if (c->srcRange) {
- c->lumConvertRange = lumRangeFromJpeg_lsx;
- c->chrConvertRange = chrRangeFromJpeg_lsx;
- } else {
- c->lumConvertRange = lumRangeToJpeg_lsx;
- c->chrConvertRange = chrRangeToJpeg_lsx;
- }
- }
- }
- }
- #if HAVE_LASX
- if (have_lasx(cpu_flags)) {
- if (c->srcRange != c->dstRange && !isAnyRGB(c->dstFormat)) {
- if (c->dstBpc <= 14) {
- if (c->srcRange) {
- c->lumConvertRange = lumRangeFromJpeg_lasx;
- c->chrConvertRange = chrRangeFromJpeg_lasx;
- } else {
- c->lumConvertRange = lumRangeToJpeg_lasx;
- c->chrConvertRange = chrRangeToJpeg_lasx;
- }
- }
- }
- }
- #endif // #if HAVE_LASX
- }
- av_cold void ff_sws_init_swscale_loongarch(SwsContext *c)
- {
- int cpu_flags = av_get_cpu_flags();
- if (have_lsx(cpu_flags)) {
- ff_sws_init_output_lsx(c, &c->yuv2plane1, &c->yuv2planeX,
- &c->yuv2nv12cX, &c->yuv2packed1,
- &c->yuv2packed2, &c->yuv2packedX, &c->yuv2anyX);
- ff_sws_init_input_lsx(c);
- if (c->srcBpc == 8) {
- if (c->dstBpc <= 14) {
- c->hyScale = c->hcScale = ff_hscale_8_to_15_lsx;
- } else {
- c->hyScale = c->hcScale = ff_hscale_8_to_19_lsx;
- }
- } else {
- c->hyScale = c->hcScale = c->dstBpc > 14 ? ff_hscale_16_to_19_lsx
- : ff_hscale_16_to_15_lsx;
- }
- }
- #if HAVE_LASX
- if (have_lasx(cpu_flags)) {
- ff_sws_init_output_lasx(c, &c->yuv2plane1, &c->yuv2planeX,
- &c->yuv2nv12cX, &c->yuv2packed1,
- &c->yuv2packed2, &c->yuv2packedX, &c->yuv2anyX);
- ff_sws_init_input_lasx(c);
- if (c->srcBpc == 8) {
- if (c->dstBpc <= 14) {
- c->hyScale = c->hcScale = ff_hscale_8_to_15_lasx;
- } else {
- c->hyScale = c->hcScale = ff_hscale_8_to_19_lasx;
- }
- } else {
- c->hyScale = c->hcScale = c->dstBpc > 14 ? ff_hscale_16_to_19_lasx
- : ff_hscale_16_to_15_lasx;
- }
- }
- #endif // #if HAVE_LASX
- ff_sws_init_range_convert_loongarch(c);
- }
- av_cold void rgb2rgb_init_loongarch(void)
- {
- #if HAVE_LASX
- int cpu_flags = av_get_cpu_flags();
- if (have_lasx(cpu_flags))
- interleaveBytes = ff_interleave_bytes_lasx;
- #endif // #if HAVE_LASX
- }
- av_cold SwsFunc ff_yuv2rgb_init_loongarch(SwsContext *c)
- {
- int cpu_flags = av_get_cpu_flags();
- #if HAVE_LASX
- if (have_lasx(cpu_flags)) {
- switch (c->dstFormat) {
- case AV_PIX_FMT_RGB24:
- return yuv420_rgb24_lasx;
- case AV_PIX_FMT_BGR24:
- return yuv420_bgr24_lasx;
- case AV_PIX_FMT_RGBA:
- if (CONFIG_SWSCALE_ALPHA && isALPHA(c->srcFormat)) {
- break;
- } else
- return yuv420_rgba32_lasx;
- case AV_PIX_FMT_ARGB:
- if (CONFIG_SWSCALE_ALPHA && isALPHA(c->srcFormat)) {
- break;
- } else
- return yuv420_argb32_lasx;
- case AV_PIX_FMT_BGRA:
- if (CONFIG_SWSCALE_ALPHA && isALPHA(c->srcFormat)) {
- break;
- } else
- return yuv420_bgra32_lasx;
- case AV_PIX_FMT_ABGR:
- if (CONFIG_SWSCALE_ALPHA && isALPHA(c->srcFormat)) {
- break;
- } else
- return yuv420_abgr32_lasx;
- }
- }
- #endif // #if HAVE_LASX
- if (have_lsx(cpu_flags)) {
- switch (c->dstFormat) {
- case AV_PIX_FMT_RGB24:
- return yuv420_rgb24_lsx;
- case AV_PIX_FMT_BGR24:
- return yuv420_bgr24_lsx;
- case AV_PIX_FMT_RGBA:
- if (CONFIG_SWSCALE_ALPHA && isALPHA(c->srcFormat)) {
- break;
- } else
- return yuv420_rgba32_lsx;
- case AV_PIX_FMT_ARGB:
- if (CONFIG_SWSCALE_ALPHA && isALPHA(c->srcFormat)) {
- break;
- } else
- return yuv420_argb32_lsx;
- case AV_PIX_FMT_BGRA:
- if (CONFIG_SWSCALE_ALPHA && isALPHA(c->srcFormat)) {
- break;
- } else
- return yuv420_bgra32_lsx;
- case AV_PIX_FMT_ABGR:
- if (CONFIG_SWSCALE_ALPHA && isALPHA(c->srcFormat)) {
- break;
- } else
- return yuv420_abgr32_lsx;
- }
- }
- return NULL;
- }
|