Blame view

3rdparty/ffmpeg-4.4.4/libavcodec/x86/ac3dsp_init.c 4.31 KB
09c2d08c   Hu Chunming   arm交付版
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
  /*
   * x86-optimized AC-3 DSP functions
   * Copyright (c) 2011 Justin Ruggles
   *
   * 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/attributes.h"
  #include "libavutil/mem.h"
  #include "libavutil/x86/asm.h"
  #include "libavutil/x86/cpu.h"
  #include "libavcodec/ac3.h"
  #include "libavcodec/ac3dsp.h"
  
  void ff_ac3_exponent_min_mmx   (uint8_t *exp, int num_reuse_blocks, int nb_coefs);
  void ff_ac3_exponent_min_mmxext(uint8_t *exp, int num_reuse_blocks, int nb_coefs);
  void ff_ac3_exponent_min_sse2  (uint8_t *exp, int num_reuse_blocks, int nb_coefs);
  
  void ff_float_to_fixed24_3dnow(int32_t *dst, const float *src, unsigned int len);
  void ff_float_to_fixed24_sse  (int32_t *dst, const float *src, unsigned int len);
  void ff_float_to_fixed24_sse2 (int32_t *dst, const float *src, unsigned int len);
  
  int ff_ac3_compute_mantissa_size_sse2(uint16_t mant_cnt[6][16]);
  
  void ff_ac3_extract_exponents_sse2 (uint8_t *exp, int32_t *coef, int nb_coefs);
  void ff_ac3_extract_exponents_ssse3(uint8_t *exp, int32_t *coef, int nb_coefs);
  
  av_cold void ff_ac3dsp_init_x86(AC3DSPContext *c, int bit_exact)
  {
      int cpu_flags = av_get_cpu_flags();
  
      if (EXTERNAL_MMX(cpu_flags)) {
          c->ac3_exponent_min = ff_ac3_exponent_min_mmx;
      }
      if (EXTERNAL_AMD3DNOW(cpu_flags)) {
          if (!bit_exact) {
              c->float_to_fixed24 = ff_float_to_fixed24_3dnow;
          }
      }
      if (EXTERNAL_MMXEXT(cpu_flags)) {
          c->ac3_exponent_min = ff_ac3_exponent_min_mmxext;
      }
      if (EXTERNAL_SSE(cpu_flags)) {
          c->float_to_fixed24 = ff_float_to_fixed24_sse;
      }
      if (EXTERNAL_SSE2(cpu_flags)) {
          c->ac3_exponent_min = ff_ac3_exponent_min_sse2;
          c->float_to_fixed24 = ff_float_to_fixed24_sse2;
          c->compute_mantissa_size = ff_ac3_compute_mantissa_size_sse2;
          c->extract_exponents = ff_ac3_extract_exponents_sse2;
      }
  
      if (EXTERNAL_SSSE3(cpu_flags)) {
          if (!(cpu_flags & AV_CPU_FLAG_ATOM))
              c->extract_exponents = ff_ac3_extract_exponents_ssse3;
      }
  }
  
  #define DOWNMIX_FUNC_OPT(ch, opt)                                       \
  void ff_ac3_downmix_ ## ch ## _to_1_ ## opt(float **samples,            \
                                              float **matrix, int len);   \
  void ff_ac3_downmix_ ## ch ## _to_2_ ## opt(float **samples,            \
                                              float **matrix, int len);
  
  #define DOWNMIX_FUNCS(opt)   \
      DOWNMIX_FUNC_OPT(3, opt) \
      DOWNMIX_FUNC_OPT(4, opt) \
      DOWNMIX_FUNC_OPT(5, opt) \
      DOWNMIX_FUNC_OPT(6, opt)
  
  DOWNMIX_FUNCS(sse)
  DOWNMIX_FUNCS(avx)
  DOWNMIX_FUNCS(fma3)
  
  void ff_ac3dsp_set_downmix_x86(AC3DSPContext *c)
  {
      int cpu_flags = av_get_cpu_flags();
  
  #define SET_DOWNMIX(ch, suf, SUF)                                       \
      if (ch == c->in_channels) {                                         \
          if (EXTERNAL_ ## SUF (cpu_flags)) {                             \
              if (c->out_channels == 1)                                   \
                  c->downmix = ff_ac3_downmix_ ## ch ## _to_1_ ## suf;    \
              else                                                        \
                  c->downmix = ff_ac3_downmix_ ## ch ## _to_2_ ## suf;    \
          }                                                               \
      }
  
  #define SET_DOWNMIX_ALL(suf, SUF)                   \
      SET_DOWNMIX(3, suf, SUF)                        \
      SET_DOWNMIX(4, suf, SUF)                        \
      SET_DOWNMIX(5, suf, SUF)                        \
      SET_DOWNMIX(6, suf, SUF)
  
      SET_DOWNMIX_ALL(sse,  SSE)
      if (!(cpu_flags & AV_CPU_FLAG_AVXSLOW)) {
          SET_DOWNMIX_ALL(avx,  AVX)
          SET_DOWNMIX_ALL(fma3, FMA3)
      }
  }