Blame view

ffmpeg-4.2.2/libavcodec/ppc/svq1enc_altivec.c 2.46 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
  /*
   * Copyright (c) 2007 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 "config.h"
  
  #include <stdint.h>
  
  #include "libavutil/attributes.h"
  #include "libavutil/cpu.h"
  #include "libavutil/ppc/cpu.h"
  #include "libavutil/ppc/util_altivec.h"
  
  #include "libavcodec/svq1enc.h"
  
  #if HAVE_ALTIVEC
  static int ssd_int8_vs_int16_altivec(const int8_t *pix1, const int16_t *pix2,
                                       intptr_t size)
  {
      int i, size16 = size >> 4;
      vector signed char vpix1;
      vector signed short vpix2, vdiff, vpix1l, vpix1h;
      union {
          vector signed int vscore;
          int32_t score[4];
      } u = { .vscore = vec_splat_s32(0) };
  
      while (size16) {
          // score += (pix1[i] - pix2[i]) * (pix1[i] - pix2[i]);
          // load pix1 and the first batch of pix2
  
          vpix1 = vec_unaligned_load(pix1);
          vpix2 = vec_unaligned_load(pix2);
          pix2 += 8;
          // unpack
          vpix1h = vec_unpackh(vpix1);
          vdiff  = vec_sub(vpix1h, vpix2);
          vpix1l = vec_unpackl(vpix1);
          // load another batch from pix2
          vpix2    = vec_unaligned_load(pix2);
          u.vscore = vec_msum(vdiff, vdiff, u.vscore);
          vdiff    = vec_sub(vpix1l, vpix2);
          u.vscore = vec_msum(vdiff, vdiff, u.vscore);
          pix1    += 16;
          pix2    += 8;
          size16--;
      }
      u.vscore = vec_sums(u.vscore, vec_splat_s32(0));
  
      size %= 16;
      for (i = 0; i < size; i++)
          u.score[3] += (pix1[i] - pix2[i]) * (pix1[i] - pix2[i]);
  
      return u.score[3];
  }
  #endif /* HAVE_ALTIVEC */
  
  av_cold void ff_svq1enc_init_ppc(SVQ1EncContext *c)
  {
  #if HAVE_ALTIVEC
      if (!PPC_ALTIVEC(av_get_cpu_flags()))
          return;
  
      c->ssd_int8_vs_int16 = ssd_int8_vs_int16_altivec;
  #endif /* HAVE_ALTIVEC */
  }