Blame view

ffmpeg-4.2.2/tests/checkasm/jpeg2000dsp.c 3.62 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
  /*
   * Copyright (c) 2015 James Almer
   *
   * 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 "checkasm.h"
  #include "libavcodec/jpeg2000dsp.h"
  #include "libavutil/common.h"
  #include "libavutil/internal.h"
  #include "libavutil/intreadwrite.h"
  
  #define BUF_SIZE 512
  
  #define randomize_buffers()                 \
      do {                                    \
          int i;                              \
          for (i = 0; i < BUF_SIZE*3; i++)    \
              src[i] = rnd();                 \
      } while (0)
  
  #define randomize_buffers_float()                    \
      do {                                             \
          int i;                                       \
          for (i = 0; i < BUF_SIZE*3; i++)             \
              src[i] = (float)rnd() / (UINT_MAX >> 5); \
      } while (0)
  
  static void check_rct_int(void)
  {
      LOCAL_ALIGNED_32(int32_t, src, [BUF_SIZE*3]);
      LOCAL_ALIGNED_32(int32_t, ref, [BUF_SIZE*3]);
      LOCAL_ALIGNED_32(int32_t, new, [BUF_SIZE*3]);
      int32_t *ref0 = &ref[BUF_SIZE*0], *new0 = &new[BUF_SIZE*0];
      int32_t *ref1 = &ref[BUF_SIZE*1], *new1 = &new[BUF_SIZE*1];
      int32_t *ref2 = &ref[BUF_SIZE*2], *new2 = &new[BUF_SIZE*2];
  
      declare_func(void, void *src0, void *src1, void *src2, int csize);
  
      randomize_buffers();
      memcpy(ref, src, BUF_SIZE * 3 * sizeof(*src));
      memcpy(new, src, BUF_SIZE * 3 * sizeof(*src));
      call_ref(ref0, ref1, ref2, BUF_SIZE);
      call_new(new0, new1, new2, BUF_SIZE);
      if (memcmp(ref0, new0, BUF_SIZE * sizeof(*src)) ||
          memcmp(ref1, new1, BUF_SIZE * sizeof(*src)) ||
          memcmp(ref2, new2, BUF_SIZE * sizeof(*src)))
          fail();
      memcpy(new, src, BUF_SIZE * 3 * sizeof(*src));
      bench_new(new0, new1, new2, BUF_SIZE);
  }
  
  static void check_ict_float(void)
  {
      LOCAL_ALIGNED_32(float, src, [BUF_SIZE*3]);
      LOCAL_ALIGNED_32(float, ref, [BUF_SIZE*3]);
      LOCAL_ALIGNED_32(float, new, [BUF_SIZE*3]);
      float *ref0 = &ref[BUF_SIZE*0], *new0 = &new[BUF_SIZE*0];
      float *ref1 = &ref[BUF_SIZE*1], *new1 = &new[BUF_SIZE*1];
      float *ref2 = &ref[BUF_SIZE*2], *new2 = &new[BUF_SIZE*2];
  
      declare_func(void, void *src0, void *src1, void *src2, int csize);
  
      randomize_buffers_float();
      memcpy(ref, src, BUF_SIZE * 3 * sizeof(*src));
      memcpy(new, src, BUF_SIZE * 3 * sizeof(*src));
      call_ref(ref0, ref1, ref2, BUF_SIZE);
      call_new(new0, new1, new2, BUF_SIZE);
      if (!float_near_abs_eps_array(ref0, new0, 1.0e-5, BUF_SIZE) ||
          !float_near_abs_eps_array(ref1, new1, 1.0e-5, BUF_SIZE) ||
          !float_near_abs_eps_array(ref2, new2, 1.0e-5, BUF_SIZE))
          fail();
      memcpy(new, src, BUF_SIZE * 3 * sizeof(*src));
      bench_new(new0, new1, new2, BUF_SIZE);
  }
  
  void checkasm_check_jpeg2000dsp(void)
  {
      Jpeg2000DSPContext h;
  
      ff_jpeg2000dsp_init(&h);
  
      if (check_func(h.mct_decode[FF_DWT53], "jpeg2000_rct_int"))
          check_rct_int();
      if (check_func(h.mct_decode[FF_DWT97], "jpeg2000_ict_float"))
          check_ict_float();
  
      report("mct_decode");
  }