Blame view

ffmpeg-4.2.2/libavutil/ppc/float_dsp_vsx.c 3.66 KB
aac5773f   hucm   功能基本完成,接口待打磨
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
115
116
117
  /*
   * Copyright (c) 2015 Luca Barbato <lu_zero@gentoo.org>
   *
   * 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 "util_altivec.h"
  #include "float_dsp_vsx.h"
  
  void ff_vector_fmul_vsx(float *dst,
                          const float *src0, const float *src1,
                          int len)
  {
      int i;
      vec_f d0, d1, zero = (vec_f)vec_splat_u32(0);
      for (i = 0; i < len - 7; i += 8) {
          d0 = vec_vsx_ld( 0, src0 + i);
          d1 = vec_vsx_ld(16, src0 + i);
          d0 = vec_madd(d0, vec_vsx_ld( 0, src1 + i), zero);
          d1 = vec_madd(d1, vec_vsx_ld(16, src1 + i), zero);
          vec_vsx_st(d0,  0, dst + i);
          vec_vsx_st(d1, 16, dst + i);
      }
  }
  
  void ff_vector_fmul_window_vsx(float *dst, const float *src0,
                                 const float *src1, const float *win,
                                 int len)
  {
      vec_f zero, t0, t1, s0, s1, wi, wj;
      const vec_u8 reverse = vcprm(3, 2, 1, 0);
      int i, j;
  
      dst  += len;
      win  += len;
      src0 += len;
  
      zero = (vec_f)vec_splat_u32(0);
  
      for (i = -len * 4, j = len * 4 - 16; i < 0; i += 16, j -= 16) {
          s0 = vec_vsx_ld(i, src0);
          s1 = vec_vsx_ld(j, src1);
          wi = vec_vsx_ld(i, win);
          wj = vec_vsx_ld(j, win);
  
          s1 = vec_perm(s1, s1, reverse);
          wj = vec_perm(wj, wj, reverse);
  
          t0 = vec_madd(s0, wj, zero);
          t0 = vec_nmsub(s1, wi, t0);
          t1 = vec_madd(s0, wi, zero);
          t1 = vec_madd(s1, wj, t1);
          t1 = vec_perm(t1, t1, reverse);
  
          vec_vsx_st(t0, i, dst);
          vec_vsx_st(t1, j, dst);
      }
  }
  
  void ff_vector_fmul_add_vsx(float *dst, const float *src0,
                              const float *src1, const float *src2,
                              int len)
  {
      int i;
      vec_f d, s0, s1, s2;
  
      for (i = 0; i < len - 3; i += 4) {
          s0 = vec_vsx_ld(0, src0 + i);
          s1 = vec_vsx_ld(0, src1 + i);
          s2 = vec_vsx_ld(0, src2 + i);
          d = vec_madd(s0, s1, s2);
          vec_vsx_st(d, 0, dst + i);
      }
  }
  
  void ff_vector_fmul_reverse_vsx(float *dst, const float *src0,
                                  const float *src1, int len)
  {
      int i;
      vec_f d, s0, s1, h0, l0, s2, s3;
      vec_f zero = (vec_f)vec_splat_u32(0);
  
      src1 += len - 4;
      for (i = 0; i < len - 7; i += 8) {
          s1 = vec_vsx_ld(0, src1 - i);              // [a,b,c,d]
          s0 = vec_vsx_ld(0, src0 + i);
          l0 = vec_mergel(s1, s1);               // [c,c,d,d]
          s3 = vec_vsx_ld(-16, src1 - i);
          h0 = vec_mergeh(s1, s1);               // [a,a,b,b]
          s2 = vec_vsx_ld(16, src0 + i);
          s1 = vec_mergeh(vec_mergel(l0, h0),    // [d,b,d,b]
                          vec_mergeh(l0, h0));   // [c,a,c,a]
          // [d,c,b,a]
          l0 = vec_mergel(s3, s3);
          d = vec_madd(s0, s1, zero);
          h0 = vec_mergeh(s3, s3);
          vec_vsx_st(d, 0, dst + i);
          s3 = vec_mergeh(vec_mergel(l0, h0),
                          vec_mergeh(l0, h0));
          d = vec_madd(s2, s3, zero);
          vec_vsx_st(d, 16, dst + i);
      }
  }