diff options
Diffstat (limited to 'libavfilter/x86/colorspacedsp_init.c')
-rw-r--r-- | libavfilter/x86/colorspacedsp_init.c | 119 |
1 files changed, 119 insertions, 0 deletions
diff --git a/libavfilter/x86/colorspacedsp_init.c b/libavfilter/x86/colorspacedsp_init.c new file mode 100644 index 0000000000..b5006ac295 --- /dev/null +++ b/libavfilter/x86/colorspacedsp_init.c @@ -0,0 +1,119 @@ +/* + * 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; + } +} |