Blame view

ffmpeg-4.2.2/tests/checkasm/vf_hflip.c 2.32 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
  /*
   * This file is part of FFmpeg.
   *
   * FFmpeg is free software; you can redistribute it and/or modify
   * it under the terms of the GNU General Public License as published by
   * the Free Software Foundation; either version 2 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 General Public License for more details.
   *
   * You should have received a copy of the GNU 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 <string.h>
  #include "checkasm.h"
  #include "libavfilter/hflip.h"
  #include "libavutil/intreadwrite.h"
  
  #define WIDTH 256
  #define WIDTH_PADDED 256 + 32
  
  #define randomize_buffers(buf, size)      \
      do {                                  \
          int j;                            \
          uint8_t *tmp_buf = (uint8_t *)buf;\
          for (j = 0; j < size; j++)        \
              tmp_buf[j] = rnd() & 0xFF;    \
      } while (0)
  
  static void check_hflip(int step, const char * report_name){
      LOCAL_ALIGNED_32(uint8_t, src,     [WIDTH_PADDED]);
      LOCAL_ALIGNED_32(uint8_t, dst_ref, [WIDTH_PADDED]);
      LOCAL_ALIGNED_32(uint8_t, dst_new, [WIDTH_PADDED]);
      int w = WIDTH;
      int i;
      int step_array[4] = {1, 1, 1, 1};
      FlipContext s;
  
      declare_func(void, const uint8_t *src, uint8_t *dst, int w);
  
      memset(src,     0, WIDTH_PADDED);
      memset(dst_ref, 0, WIDTH_PADDED);
      memset(dst_new, 0, WIDTH_PADDED);
      randomize_buffers(src, WIDTH_PADDED);
  
      if (step == 2) {
          w /= 2;
          for (i = 0; i < 4; i++)
              step_array[i] = step;
      }
  
      ff_hflip_init(&s, step_array, 4);
  
      if (check_func(s.flip_line[0], "hflip_%s", report_name)) {
          for (i = 1; i < w; i++) {
              call_ref(src + (w - 1) * step, dst_ref, i);
              call_new(src + (w - 1) * step, dst_new, i);
              if (memcmp(dst_ref, dst_new, i * step))
                  fail();
          }
          bench_new(src + (w - 1) * step, dst_new, w);
      }
  }
  void checkasm_check_vf_hflip(void)
  {
      check_hflip(1, "byte");
      report("hflip_byte");
  
      check_hflip(2, "short");
      report("hflip_short");
  }