1/*
2 *  Copyright (c) 2011 The LibYuv project authors. All Rights Reserved.
3 *
4 *  Use of this source code is governed by a BSD-style license
5 *  that can be found in the LICENSE file in the root of the source
6 *  tree. An additional intellectual property rights grant can be found
7 *  in the file PATENTS.  All contributing project authors may
8 *  be found in the AUTHORS file in the root of the source tree.
9 */
10
11#include "libyuv/scale.h"
12#include "unit_test.h"
13#include <stdlib.h>
14#include <time.h>
15
16using namespace libyuv;
17
18#define align_buffer_16(var, size) \
19  uint8 *var; \
20  uint8 *var##_mem; \
21  var##_mem = reinterpret_cast<uint8*>(calloc(size+15, sizeof(uint8))); \
22  var = reinterpret_cast<uint8*> \
23        ((reinterpret_cast<intptr_t>(var##_mem) + 15) & (~0x0f));
24
25#define free_aligned_buffer_16(var) \
26  free(var##_mem);  \
27  var = 0;
28
29TEST_F(libyuvTest, ScaleDownBy4) {
30  int b = 128;
31  int src_width = 1280;
32  int src_height = 720;
33  int src_width_uv = (src_width + 1) >> 1;
34  int src_height_uv = (src_height + 1) >> 1;
35
36  int src_y_plane_size = (src_width + (2 * b)) * (src_height + (2 * b));
37  int src_uv_plane_size = (src_width_uv + (2 * b)) * (src_height_uv + (2 * b));
38
39  int src_stride_y = 2 * b + src_width;
40  int src_stride_uv = 2 * b + src_width_uv;
41
42  align_buffer_16(src_y, src_y_plane_size)
43  align_buffer_16(src_u, src_uv_plane_size)
44  align_buffer_16(src_v, src_uv_plane_size)
45
46  int dst_width = src_width >> 2;
47  int dst_height = src_height >> 2;
48
49  int dst_width_uv = (dst_width + 1) >> 1;
50  int dst_height_uv = (dst_height + 1) >> 1;
51
52  int dst_y_plane_size = (dst_width + (2 * b)) * (dst_height + (2 * b));
53  int dst_uv_plane_size = (dst_width_uv + (2 * b)) * (dst_height_uv + (2 * b));
54
55  int dst_stride_y = 2 * b + dst_width;
56  int dst_stride_uv = 2 * b + dst_width_uv;
57
58  align_buffer_16(dst_y, dst_y_plane_size)
59  align_buffer_16(dst_u, dst_uv_plane_size)
60  align_buffer_16(dst_v, dst_uv_plane_size)
61
62  // create an image with random data reoccurring in 4x4 grid.  When the image
63  // is filtered all the values should be the same.
64  srandom(time(NULL));
65
66  uint8 block_data[16];
67
68  int i, j;
69
70  // Pulling 16 random numbers there is an infinitesimally small
71  //  chance that they are all 0.  Then the output will be all 0.
72  //  Output buffer is filled with 0, want to make sure that after the
73  //  filtering something went into the output buffer.
74  //  Avoid this by setting one of the values to 128.  Also set the
75  //  random data to at least 1 for when point sampling to prevent
76  //  output all being 0.
77  block_data[0] = 128;
78
79  for (i = 1; i < 16; i++)
80    block_data[i] = (random() & 0xfe) + 1;
81
82  for (i = b; i < (src_height + b); i += 4) {
83    for (j = b; j < (src_width + b); j += 4) {
84      uint8 *ptr = src_y + (i * src_stride_y) + j;
85      int k, l;
86      for (k = 0; k < 4; ++k)
87        for (l = 0; l < 4; ++l)
88          ptr[k + src_stride_y * l] = block_data[k + 4 * l];
89    }
90  }
91
92  for (i = 1; i < 16; i++)
93    block_data[i] = (random() & 0xfe) + 1;
94
95  for (i = b; i < (src_height_uv + b); i += 4) {
96    for (j = b; j < (src_width_uv + b); j += 4) {
97      uint8 *ptru = src_u + (i * src_stride_uv) + j;
98      uint8 *ptrv = src_v + (i * src_stride_uv) + j;
99      int k, l;
100      for (k = 0; k < 4; ++k)
101        for (l = 0; l < 4; ++l) {
102          ptru[k + src_stride_uv * l] = block_data[k + 4 * l];
103          ptrv[k + src_stride_uv * l] = block_data[k + 4 * l];
104        }
105    }
106  }
107
108  int f;
109  int err = 0;
110
111  // currently three filter modes, defined as FilterMode in scale.h
112  for (f = 0; f < 3; ++f) {
113    I420Scale(src_y + (src_stride_y * b) + b, src_stride_y,
114              src_u + (src_stride_uv * b) + b, src_stride_uv,
115              src_v + (src_stride_uv * b) + b, src_stride_uv,
116              src_width, src_height,
117              dst_y + (dst_stride_y * b) + b, dst_stride_y,
118              dst_u + (dst_stride_uv * b) + b, dst_stride_uv,
119              dst_v + (dst_stride_uv * b) + b, dst_stride_uv,
120              dst_width, dst_height,
121              static_cast<FilterMode>(f));
122
123    int value = dst_y[(dst_stride_y * b) + b];
124
125    // catch the case that the output buffer is all 0
126    if (value == 0)
127      ++err;
128
129    for (i = b; i < (dst_height + b); ++i) {
130      for (j = b; j < (dst_width + b); ++j) {
131        if (value != dst_y[(i * dst_stride_y) + j])
132          ++err;
133      }
134    }
135
136    value = dst_u[(dst_stride_uv * b) + b];
137
138    if (value == 0)
139      ++err;
140
141    for (i = b; i < (dst_height_uv + b); ++i) {
142      for (j = b; j < (dst_width_uv + b); ++j) {
143        if (value != dst_u[(i * dst_stride_uv) + j])
144          ++err;
145        if (value != dst_v[(i * dst_stride_uv) + j])
146          ++err;
147      }
148    }
149  }
150
151  free_aligned_buffer_16(src_y)
152  free_aligned_buffer_16(src_u)
153  free_aligned_buffer_16(src_v)
154  free_aligned_buffer_16(dst_y)
155  free_aligned_buffer_16(dst_u)
156  free_aligned_buffer_16(dst_v)
157
158  EXPECT_EQ(0, err);
159}
160