mirror of https://git.ffmpeg.org/ffmpeg.git
120 lines
4.4 KiB
C
120 lines
4.4 KiB
C
/*
|
|
* Copyright (c) 2016 Ronald S. Bultje <rsbultje@gmail.com>
|
|
*
|
|
* 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/x86/cpu.h"
|
|
|
|
#include "libavfilter/colorspacedsp.h"
|
|
|
|
#define decl_yuv2yuv_fn(t) \
|
|
void ff_yuv2yuv_##t##_sse2(uint8_t *yuv_out[3], const ptrdiff_t yuv_out_stride[3], \
|
|
uint8_t *yuv_in[3], const ptrdiff_t yuv_in_stride[3], \
|
|
int w, int h, const int16_t yuv2yuv_coeffs[3][3][8], \
|
|
const int16_t yuv_offset[2][8])
|
|
|
|
#define decl_yuv2yuv_fns(ss) \
|
|
decl_yuv2yuv_fn(ss##p8to8); \
|
|
decl_yuv2yuv_fn(ss##p10to8); \
|
|
decl_yuv2yuv_fn(ss##p12to8); \
|
|
decl_yuv2yuv_fn(ss##p8to10); \
|
|
decl_yuv2yuv_fn(ss##p10to10); \
|
|
decl_yuv2yuv_fn(ss##p12to10); \
|
|
decl_yuv2yuv_fn(ss##p8to12); \
|
|
decl_yuv2yuv_fn(ss##p10to12); \
|
|
decl_yuv2yuv_fn(ss##p12to12)
|
|
|
|
decl_yuv2yuv_fns(420);
|
|
decl_yuv2yuv_fns(422);
|
|
decl_yuv2yuv_fns(444);
|
|
|
|
#define decl_yuv2rgb_fn(t) \
|
|
void ff_yuv2rgb_##t##_sse2(int16_t *rgb_out[3], ptrdiff_t rgb_stride, \
|
|
uint8_t *yuv_in[3], const ptrdiff_t yuv_stride[3], \
|
|
int w, int h, const int16_t coeff[3][3][8], \
|
|
const int16_t yuv_offset[8])
|
|
|
|
#define decl_yuv2rgb_fns(ss) \
|
|
decl_yuv2rgb_fn(ss##p8); \
|
|
decl_yuv2rgb_fn(ss##p10); \
|
|
decl_yuv2rgb_fn(ss##p12)
|
|
|
|
decl_yuv2rgb_fns(420);
|
|
decl_yuv2rgb_fns(422);
|
|
decl_yuv2rgb_fns(444);
|
|
|
|
#define decl_rgb2yuv_fn(t) \
|
|
void ff_rgb2yuv_##t##_sse2(uint8_t *yuv_out[3], const ptrdiff_t yuv_stride[3], \
|
|
int16_t *rgb_in[3], ptrdiff_t rgb_stride, \
|
|
int w, int h, const int16_t coeff[3][3][8], \
|
|
const int16_t yuv_offset[8])
|
|
|
|
#define decl_rgb2yuv_fns(ss) \
|
|
decl_rgb2yuv_fn(ss##p8); \
|
|
decl_rgb2yuv_fn(ss##p10); \
|
|
decl_rgb2yuv_fn(ss##p12)
|
|
|
|
decl_rgb2yuv_fns(420);
|
|
decl_rgb2yuv_fns(422);
|
|
decl_rgb2yuv_fns(444);
|
|
|
|
void ff_multiply3x3_sse2(int16_t *data[3], ptrdiff_t stride, int w, int h,
|
|
const int16_t coeff[3][3][8]);
|
|
|
|
void ff_colorspacedsp_x86_init(ColorSpaceDSPContext *dsp)
|
|
{
|
|
int cpu_flags = av_get_cpu_flags();
|
|
|
|
if (ARCH_X86_64 && EXTERNAL_SSE2(cpu_flags)) {
|
|
#define assign_yuv2yuv_fns(ss) \
|
|
dsp->yuv2yuv[BPP_8 ][BPP_8 ][SS_##ss] = ff_yuv2yuv_##ss##p8to8_sse2; \
|
|
dsp->yuv2yuv[BPP_8 ][BPP_10][SS_##ss] = ff_yuv2yuv_##ss##p8to10_sse2; \
|
|
dsp->yuv2yuv[BPP_8 ][BPP_12][SS_##ss] = ff_yuv2yuv_##ss##p8to12_sse2; \
|
|
dsp->yuv2yuv[BPP_10][BPP_8 ][SS_##ss] = ff_yuv2yuv_##ss##p10to8_sse2; \
|
|
dsp->yuv2yuv[BPP_10][BPP_10][SS_##ss] = ff_yuv2yuv_##ss##p10to10_sse2; \
|
|
dsp->yuv2yuv[BPP_10][BPP_12][SS_##ss] = ff_yuv2yuv_##ss##p10to12_sse2; \
|
|
dsp->yuv2yuv[BPP_12][BPP_8 ][SS_##ss] = ff_yuv2yuv_##ss##p12to8_sse2; \
|
|
dsp->yuv2yuv[BPP_12][BPP_10][SS_##ss] = ff_yuv2yuv_##ss##p12to10_sse2; \
|
|
dsp->yuv2yuv[BPP_12][BPP_12][SS_##ss] = ff_yuv2yuv_##ss##p12to12_sse2
|
|
|
|
assign_yuv2yuv_fns(420);
|
|
assign_yuv2yuv_fns(422);
|
|
assign_yuv2yuv_fns(444);
|
|
|
|
#define assign_yuv2rgb_fns(ss) \
|
|
dsp->yuv2rgb[BPP_8 ][SS_##ss] = ff_yuv2rgb_##ss##p8_sse2; \
|
|
dsp->yuv2rgb[BPP_10][SS_##ss] = ff_yuv2rgb_##ss##p10_sse2; \
|
|
dsp->yuv2rgb[BPP_12][SS_##ss] = ff_yuv2rgb_##ss##p12_sse2
|
|
|
|
assign_yuv2rgb_fns(420);
|
|
assign_yuv2rgb_fns(422);
|
|
assign_yuv2rgb_fns(444);
|
|
|
|
#define assign_rgb2yuv_fns(ss) \
|
|
dsp->rgb2yuv[BPP_8 ][SS_##ss] = ff_rgb2yuv_##ss##p8_sse2; \
|
|
dsp->rgb2yuv[BPP_10][SS_##ss] = ff_rgb2yuv_##ss##p10_sse2; \
|
|
dsp->rgb2yuv[BPP_12][SS_##ss] = ff_rgb2yuv_##ss##p12_sse2
|
|
|
|
assign_rgb2yuv_fns(420);
|
|
assign_rgb2yuv_fns(422);
|
|
assign_rgb2yuv_fns(444);
|
|
|
|
dsp->multiply3x3 = ff_multiply3x3_sse2;
|
|
}
|
|
}
|