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.
281 lines
11 KiB
281 lines
11 KiB
/* |
|
* software YUV to RGB converter |
|
* |
|
* Copyright (C) 2001-2007 Michael Niedermayer |
|
* Copyright (C) 2009-2010 Konstantin Shishkov |
|
* |
|
* MMX/MMXEXT template stuff (needed for fast movntq support), |
|
* 1,4,8bpp support and context / deglobalize stuff |
|
* by Michael Niedermayer (michaelni@gmx.at) |
|
* |
|
* 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 <stdio.h> |
|
#include <stdlib.h> |
|
#include <inttypes.h> |
|
|
|
#include "config.h" |
|
#include "libswscale/rgb2rgb.h" |
|
#include "libswscale/swscale.h" |
|
#include "libswscale/swscale_internal.h" |
|
#include "libavutil/attributes.h" |
|
#include "libavutil/x86/asm.h" |
|
#include "libavutil/x86/cpu.h" |
|
#include "libavutil/cpu.h" |
|
|
|
#if HAVE_X86ASM |
|
|
|
#define YUV2RGB_LOOP(depth) \ |
|
h_size = (c->opts.dst_w + 7) & ~7; \ |
|
if (h_size * depth > FFABS(dstStride[0])) \ |
|
h_size -= 8; \ |
|
\ |
|
vshift = c->opts.src_format != AV_PIX_FMT_YUV422P; \ |
|
\ |
|
for (y = 0; y < srcSliceH; y++) { \ |
|
uint8_t *image = dst[0] + (y + srcSliceY) * dstStride[0]; \ |
|
const uint8_t *py = src[0] + y * srcStride[0]; \ |
|
const uint8_t *pu = src[1] + (y >> vshift) * srcStride[1]; \ |
|
const uint8_t *pv = src[2] + (y >> vshift) * srcStride[2]; \ |
|
x86_reg index = -h_size / 2; \ |
|
|
|
extern void ff_yuv_420_rgb24_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index, |
|
const uint8_t *pv_index, const uint64_t *pointer_c_dither, |
|
const uint8_t *py_2index); |
|
extern void ff_yuv_420_bgr24_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index, |
|
const uint8_t *pv_index, const uint64_t *pointer_c_dither, |
|
const uint8_t *py_2index); |
|
|
|
extern void ff_yuv_420_rgb15_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index, |
|
const uint8_t *pv_index, const uint64_t *pointer_c_dither, |
|
const uint8_t *py_2index); |
|
extern void ff_yuv_420_rgb16_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index, |
|
const uint8_t *pv_index, const uint64_t *pointer_c_dither, |
|
const uint8_t *py_2index); |
|
extern void ff_yuv_420_rgb32_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index, |
|
const uint8_t *pv_index, const uint64_t *pointer_c_dither, |
|
const uint8_t *py_2index); |
|
extern void ff_yuv_420_bgr32_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index, |
|
const uint8_t *pv_index, const uint64_t *pointer_c_dither, |
|
const uint8_t *py_2index); |
|
extern void ff_yuva_420_rgb32_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index, |
|
const uint8_t *pv_index, const uint64_t *pointer_c_dither, |
|
const uint8_t *py_2index, const uint8_t *pa_2index); |
|
extern void ff_yuva_420_bgr32_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index, |
|
const uint8_t *pv_index, const uint64_t *pointer_c_dither, |
|
const uint8_t *py_2index, const uint8_t *pa_2index); |
|
#if ARCH_X86_64 |
|
extern void ff_yuv_420_gbrp24_ssse3(x86_reg index, uint8_t *image, uint8_t *dst_b, uint8_t *dst_r, |
|
const uint8_t *pu_index, const uint8_t *pv_index, |
|
const uint64_t *pointer_c_dither, |
|
const uint8_t *py_2index); |
|
#endif |
|
|
|
static inline int yuv420_rgb15_ssse3(SwsInternal *c, const uint8_t *const src[], |
|
const int srcStride[], |
|
int srcSliceY, int srcSliceH, |
|
uint8_t *const dst[], const int dstStride[]) |
|
{ |
|
int y, h_size, vshift; |
|
|
|
YUV2RGB_LOOP(2) |
|
|
|
c->blueDither = ff_dither8[y & 1]; |
|
c->greenDither = ff_dither8[y & 1]; |
|
c->redDither = ff_dither8[(y + 1) & 1]; |
|
|
|
ff_yuv_420_rgb15_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index); |
|
} |
|
return srcSliceH; |
|
} |
|
|
|
static inline int yuv420_rgb16_ssse3(SwsInternal *c, const uint8_t *const src[], |
|
const int srcStride[], |
|
int srcSliceY, int srcSliceH, |
|
uint8_t *const dst[], const int dstStride[]) |
|
{ |
|
int y, h_size, vshift; |
|
|
|
YUV2RGB_LOOP(2) |
|
|
|
c->blueDither = ff_dither8[y & 1]; |
|
c->greenDither = ff_dither4[y & 1]; |
|
c->redDither = ff_dither8[(y + 1) & 1]; |
|
|
|
ff_yuv_420_rgb16_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index); |
|
} |
|
return srcSliceH; |
|
} |
|
|
|
static inline int yuv420_rgb32_ssse3(SwsInternal *c, const uint8_t *const src[], |
|
const int srcStride[], |
|
int srcSliceY, int srcSliceH, |
|
uint8_t *const dst[], const int dstStride[]) |
|
{ |
|
int y, h_size, vshift; |
|
|
|
YUV2RGB_LOOP(4) |
|
|
|
ff_yuv_420_rgb32_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index); |
|
} |
|
return srcSliceH; |
|
} |
|
|
|
static inline int yuv420_bgr32_ssse3(SwsInternal *c, const uint8_t *const src[], |
|
const int srcStride[], |
|
int srcSliceY, int srcSliceH, |
|
uint8_t *const dst[], const int dstStride[]) |
|
{ |
|
int y, h_size, vshift; |
|
|
|
YUV2RGB_LOOP(4) |
|
|
|
ff_yuv_420_bgr32_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index); |
|
} |
|
return srcSliceH; |
|
} |
|
|
|
static inline int yuva420_rgb32_ssse3(SwsInternal *c, const uint8_t *const src[], |
|
const int srcStride[], |
|
int srcSliceY, int srcSliceH, |
|
uint8_t *const dst[], const int dstStride[]) |
|
{ |
|
int y, h_size, vshift; |
|
YUV2RGB_LOOP(4) |
|
|
|
const uint8_t *pa = src[3] + y * srcStride[3]; |
|
ff_yuva_420_rgb32_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index, pa - 2 * index); |
|
} |
|
return srcSliceH; |
|
} |
|
|
|
static inline int yuva420_bgr32_ssse3(SwsInternal *c, const uint8_t *const src[], |
|
const int srcStride[], |
|
int srcSliceY, int srcSliceH, |
|
uint8_t *const dst[], const int dstStride[]) |
|
{ |
|
int y, h_size, vshift; |
|
|
|
YUV2RGB_LOOP(4) |
|
|
|
const uint8_t *pa = src[3] + y * srcStride[3]; |
|
ff_yuva_420_bgr32_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index, pa - 2 * index); |
|
} |
|
return srcSliceH; |
|
} |
|
|
|
static inline int yuv420_rgb24_ssse3(SwsInternal *c, const uint8_t *const src[], |
|
const int srcStride[], |
|
int srcSliceY, int srcSliceH, |
|
uint8_t *const dst[], const int dstStride[]) |
|
{ |
|
int y, h_size, vshift; |
|
|
|
YUV2RGB_LOOP(3) |
|
|
|
ff_yuv_420_rgb24_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index); |
|
} |
|
return srcSliceH; |
|
} |
|
|
|
static inline int yuv420_bgr24_ssse3(SwsInternal *c, const uint8_t *const src[], |
|
const int srcStride[], |
|
int srcSliceY, int srcSliceH, |
|
uint8_t *const dst[], const int dstStride[]) |
|
{ |
|
int y, h_size, vshift; |
|
|
|
YUV2RGB_LOOP(3) |
|
|
|
ff_yuv_420_bgr24_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index); |
|
} |
|
return srcSliceH; |
|
} |
|
|
|
#if ARCH_X86_64 |
|
static inline int yuv420_gbrp_ssse3(SwsInternal *c, const uint8_t *const src[], |
|
const int srcStride[], |
|
int srcSliceY, int srcSliceH, |
|
uint8_t *const dst[], const int dstStride[]) |
|
{ |
|
int y, h_size, vshift; |
|
|
|
h_size = (c->opts.dst_w + 7) & ~7; |
|
if (h_size * 3 > FFABS(dstStride[0])) |
|
h_size -= 8; |
|
|
|
vshift = c->opts.src_format != AV_PIX_FMT_YUV422P; |
|
|
|
for (y = 0; y < srcSliceH; y++) { |
|
uint8_t *dst_g = dst[0] + (y + srcSliceY) * dstStride[0]; |
|
uint8_t *dst_b = dst[1] + (y + srcSliceY) * dstStride[1]; |
|
uint8_t *dst_r = dst[2] + (y + srcSliceY) * dstStride[2]; |
|
const uint8_t *py = src[0] + y * srcStride[0]; |
|
const uint8_t *pu = src[1] + (y >> vshift) * srcStride[1]; |
|
const uint8_t *pv = src[2] + (y >> vshift) * srcStride[2]; |
|
x86_reg index = -h_size / 2; |
|
|
|
ff_yuv_420_gbrp24_ssse3(index, dst_g, dst_b, dst_r, pu - index, pv - index, &(c->redDither), py - 2 * index); |
|
} |
|
return srcSliceH; |
|
} |
|
#endif |
|
|
|
#endif /* HAVE_X86ASM */ |
|
|
|
av_cold SwsFunc ff_yuv2rgb_init_x86(SwsInternal *c) |
|
{ |
|
#if HAVE_X86ASM |
|
int cpu_flags = av_get_cpu_flags(); |
|
|
|
if (EXTERNAL_SSSE3(cpu_flags)) { |
|
switch (c->opts.dst_format) { |
|
case AV_PIX_FMT_RGB32: |
|
if (c->opts.src_format == AV_PIX_FMT_YUVA420P) { |
|
#if CONFIG_SWSCALE_ALPHA |
|
return yuva420_rgb32_ssse3; |
|
#endif |
|
break; |
|
} else |
|
return yuv420_rgb32_ssse3; |
|
case AV_PIX_FMT_BGR32: |
|
if (c->opts.src_format == AV_PIX_FMT_YUVA420P) { |
|
#if CONFIG_SWSCALE_ALPHA |
|
return yuva420_bgr32_ssse3; |
|
#endif |
|
break; |
|
} else |
|
return yuv420_bgr32_ssse3; |
|
case AV_PIX_FMT_RGB24: |
|
return yuv420_rgb24_ssse3; |
|
case AV_PIX_FMT_BGR24: |
|
return yuv420_bgr24_ssse3; |
|
case AV_PIX_FMT_RGB565: |
|
return yuv420_rgb16_ssse3; |
|
case AV_PIX_FMT_RGB555: |
|
return yuv420_rgb15_ssse3; |
|
#if ARCH_X86_64 |
|
case AV_PIX_FMT_GBRP: |
|
return yuv420_gbrp_ssse3; |
|
#endif |
|
} |
|
} |
|
|
|
#endif /* HAVE_X86ASM */ |
|
return NULL; |
|
}
|
|
|