Commit | Line | Data |
---|---|---|
2ba45a60 DM |
1 | /* |
2 | * Copyright (c) 2012 Justin Ruggles <justin.ruggles@gmail.com> | |
3 | * | |
4 | * This file is part of FFmpeg. | |
5 | * | |
6 | * FFmpeg is free software; you can redistribute it and/or | |
7 | * modify it under the terms of the GNU Lesser General Public | |
8 | * License as published by the Free Software Foundation; either | |
9 | * version 2.1 of the License, or (at your option) any later version. | |
10 | * | |
11 | * FFmpeg is distributed in the hope that it will be useful, | |
12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
14 | * Lesser General Public License for more details. | |
15 | * | |
16 | * You should have received a copy of the GNU Lesser General Public | |
17 | * License along with FFmpeg; if not, write to the Free Software | |
18 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
19 | */ | |
20 | ||
21 | #include "config.h" | |
22 | #include "libavutil/cpu.h" | |
23 | #include "libavutil/x86/cpu.h" | |
24 | #include "libavresample/dither.h" | |
25 | ||
26 | void ff_quantize_sse2(int16_t *dst, const float *src, float *dither, int len); | |
27 | ||
28 | void ff_dither_int_to_float_rectangular_sse2(float *dst, int *src, int len); | |
29 | void ff_dither_int_to_float_rectangular_avx(float *dst, int *src, int len); | |
30 | ||
31 | void ff_dither_int_to_float_triangular_sse2(float *dst, int *src0, int len); | |
32 | void ff_dither_int_to_float_triangular_avx(float *dst, int *src0, int len); | |
33 | ||
34 | av_cold void ff_dither_init_x86(DitherDSPContext *ddsp, | |
35 | enum AVResampleDitherMethod method) | |
36 | { | |
37 | int cpu_flags = av_get_cpu_flags(); | |
38 | ||
39 | if (EXTERNAL_SSE2(cpu_flags)) { | |
40 | ddsp->quantize = ff_quantize_sse2; | |
41 | ddsp->ptr_align = 16; | |
42 | ddsp->samples_align = 8; | |
43 | } | |
44 | ||
45 | if (method == AV_RESAMPLE_DITHER_RECTANGULAR) { | |
46 | if (EXTERNAL_SSE2(cpu_flags)) { | |
47 | ddsp->dither_int_to_float = ff_dither_int_to_float_rectangular_sse2; | |
48 | } | |
49 | if (EXTERNAL_AVX(cpu_flags)) { | |
50 | ddsp->dither_int_to_float = ff_dither_int_to_float_rectangular_avx; | |
51 | } | |
52 | } else { | |
53 | if (EXTERNAL_SSE2(cpu_flags)) { | |
54 | ddsp->dither_int_to_float = ff_dither_int_to_float_triangular_sse2; | |
55 | } | |
56 | if (EXTERNAL_AVX(cpu_flags)) { | |
57 | ddsp->dither_int_to_float = ff_dither_int_to_float_triangular_avx; | |
58 | } | |
59 | } | |
60 | } |