xref: /aosp_15_r20/external/libvpx/vp9/encoder/vp9_resize.c (revision fb1b10ab9aebc7c7068eedab379b749d7e3900be)
1 /*
2  *  Copyright (c) 2014 The WebM 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 <assert.h>
12 #include <limits.h>
13 #include <math.h>
14 #include <stdio.h>
15 #include <stdlib.h>
16 #include <string.h>
17 
18 #include "./vpx_config.h"
19 #if CONFIG_VP9_HIGHBITDEPTH
20 #include "vpx_dsp/vpx_dsp_common.h"
21 #endif  // CONFIG_VP9_HIGHBITDEPTH
22 #include "vpx_ports/mem.h"
23 #include "vp9/common/vp9_common.h"
24 #include "vp9/encoder/vp9_resize.h"
25 
26 #define FILTER_BITS 7
27 
28 #define INTERP_TAPS 8
29 #define SUBPEL_BITS 5
30 #define SUBPEL_MASK ((1 << SUBPEL_BITS) - 1)
31 #define INTERP_PRECISION_BITS 32
32 
33 typedef int16_t interp_kernel[INTERP_TAPS];
34 
35 // Filters for interpolation (0.5-band) - note this also filters integer pels.
36 static const interp_kernel filteredinterp_filters500[(1 << SUBPEL_BITS)] = {
37   { -3, 0, 35, 64, 35, 0, -3, 0 },    { -3, -1, 34, 64, 36, 1, -3, 0 },
38   { -3, -1, 32, 64, 38, 1, -3, 0 },   { -2, -2, 31, 63, 39, 2, -3, 0 },
39   { -2, -2, 29, 63, 41, 2, -3, 0 },   { -2, -2, 28, 63, 42, 3, -4, 0 },
40   { -2, -3, 27, 63, 43, 4, -4, 0 },   { -2, -3, 25, 62, 45, 5, -4, 0 },
41   { -2, -3, 24, 62, 46, 5, -4, 0 },   { -2, -3, 23, 61, 47, 6, -4, 0 },
42   { -2, -3, 21, 60, 49, 7, -4, 0 },   { -1, -4, 20, 60, 50, 8, -4, -1 },
43   { -1, -4, 19, 59, 51, 9, -4, -1 },  { -1, -4, 17, 58, 52, 10, -4, 0 },
44   { -1, -4, 16, 57, 53, 12, -4, -1 }, { -1, -4, 15, 56, 54, 13, -4, -1 },
45   { -1, -4, 14, 55, 55, 14, -4, -1 }, { -1, -4, 13, 54, 56, 15, -4, -1 },
46   { -1, -4, 12, 53, 57, 16, -4, -1 }, { 0, -4, 10, 52, 58, 17, -4, -1 },
47   { -1, -4, 9, 51, 59, 19, -4, -1 },  { -1, -4, 8, 50, 60, 20, -4, -1 },
48   { 0, -4, 7, 49, 60, 21, -3, -2 },   { 0, -4, 6, 47, 61, 23, -3, -2 },
49   { 0, -4, 5, 46, 62, 24, -3, -2 },   { 0, -4, 5, 45, 62, 25, -3, -2 },
50   { 0, -4, 4, 43, 63, 27, -3, -2 },   { 0, -4, 3, 42, 63, 28, -2, -2 },
51   { 0, -3, 2, 41, 63, 29, -2, -2 },   { 0, -3, 2, 39, 63, 31, -2, -2 },
52   { 0, -3, 1, 38, 64, 32, -1, -3 },   { 0, -3, 1, 36, 64, 34, -1, -3 }
53 };
54 
55 // Filters for interpolation (0.625-band) - note this also filters integer pels.
56 static const interp_kernel filteredinterp_filters625[(1 << SUBPEL_BITS)] = {
57   { -1, -8, 33, 80, 33, -8, -1, 0 }, { -1, -8, 30, 80, 35, -8, -1, 1 },
58   { -1, -8, 28, 80, 37, -7, -2, 1 }, { 0, -8, 26, 79, 39, -7, -2, 1 },
59   { 0, -8, 24, 79, 41, -7, -2, 1 },  { 0, -8, 22, 78, 43, -6, -2, 1 },
60   { 0, -8, 20, 78, 45, -5, -3, 1 },  { 0, -8, 18, 77, 48, -5, -3, 1 },
61   { 0, -8, 16, 76, 50, -4, -3, 1 },  { 0, -8, 15, 75, 52, -3, -4, 1 },
62   { 0, -7, 13, 74, 54, -3, -4, 1 },  { 0, -7, 11, 73, 56, -2, -4, 1 },
63   { 0, -7, 10, 71, 58, -1, -4, 1 },  { 1, -7, 8, 70, 60, 0, -5, 1 },
64   { 1, -6, 6, 68, 62, 1, -5, 1 },    { 1, -6, 5, 67, 63, 2, -5, 1 },
65   { 1, -6, 4, 65, 65, 4, -6, 1 },    { 1, -5, 2, 63, 67, 5, -6, 1 },
66   { 1, -5, 1, 62, 68, 6, -6, 1 },    { 1, -5, 0, 60, 70, 8, -7, 1 },
67   { 1, -4, -1, 58, 71, 10, -7, 0 },  { 1, -4, -2, 56, 73, 11, -7, 0 },
68   { 1, -4, -3, 54, 74, 13, -7, 0 },  { 1, -4, -3, 52, 75, 15, -8, 0 },
69   { 1, -3, -4, 50, 76, 16, -8, 0 },  { 1, -3, -5, 48, 77, 18, -8, 0 },
70   { 1, -3, -5, 45, 78, 20, -8, 0 },  { 1, -2, -6, 43, 78, 22, -8, 0 },
71   { 1, -2, -7, 41, 79, 24, -8, 0 },  { 1, -2, -7, 39, 79, 26, -8, 0 },
72   { 1, -2, -7, 37, 80, 28, -8, -1 }, { 1, -1, -8, 35, 80, 30, -8, -1 },
73 };
74 
75 // Filters for interpolation (0.75-band) - note this also filters integer pels.
76 static const interp_kernel filteredinterp_filters750[(1 << SUBPEL_BITS)] = {
77   { 2, -11, 25, 96, 25, -11, 2, 0 }, { 2, -11, 22, 96, 28, -11, 2, 0 },
78   { 2, -10, 19, 95, 31, -11, 2, 0 }, { 2, -10, 17, 95, 34, -12, 2, 0 },
79   { 2, -9, 14, 94, 37, -12, 2, 0 },  { 2, -8, 12, 93, 40, -12, 1, 0 },
80   { 2, -8, 9, 92, 43, -12, 1, 1 },   { 2, -7, 7, 91, 46, -12, 1, 0 },
81   { 2, -7, 5, 90, 49, -12, 1, 0 },   { 2, -6, 3, 88, 52, -12, 0, 1 },
82   { 2, -5, 1, 86, 55, -12, 0, 1 },   { 2, -5, -1, 84, 58, -11, 0, 1 },
83   { 2, -4, -2, 82, 61, -11, -1, 1 }, { 2, -4, -4, 80, 64, -10, -1, 1 },
84   { 1, -3, -5, 77, 67, -9, -1, 1 },  { 1, -3, -6, 75, 70, -8, -2, 1 },
85   { 1, -2, -7, 72, 72, -7, -2, 1 },  { 1, -2, -8, 70, 75, -6, -3, 1 },
86   { 1, -1, -9, 67, 77, -5, -3, 1 },  { 1, -1, -10, 64, 80, -4, -4, 2 },
87   { 1, -1, -11, 61, 82, -2, -4, 2 }, { 1, 0, -11, 58, 84, -1, -5, 2 },
88   { 1, 0, -12, 55, 86, 1, -5, 2 },   { 1, 0, -12, 52, 88, 3, -6, 2 },
89   { 0, 1, -12, 49, 90, 5, -7, 2 },   { 0, 1, -12, 46, 91, 7, -7, 2 },
90   { 1, 1, -12, 43, 92, 9, -8, 2 },   { 0, 1, -12, 40, 93, 12, -8, 2 },
91   { 0, 2, -12, 37, 94, 14, -9, 2 },  { 0, 2, -12, 34, 95, 17, -10, 2 },
92   { 0, 2, -11, 31, 95, 19, -10, 2 }, { 0, 2, -11, 28, 96, 22, -11, 2 }
93 };
94 
95 // Filters for interpolation (0.875-band) - note this also filters integer pels.
96 static const interp_kernel filteredinterp_filters875[(1 << SUBPEL_BITS)] = {
97   { 3, -8, 13, 112, 13, -8, 3, 0 },   { 3, -7, 10, 112, 17, -9, 3, -1 },
98   { 2, -6, 7, 111, 21, -9, 3, -1 },   { 2, -5, 4, 111, 24, -10, 3, -1 },
99   { 2, -4, 1, 110, 28, -11, 3, -1 },  { 1, -3, -1, 108, 32, -12, 4, -1 },
100   { 1, -2, -3, 106, 36, -13, 4, -1 }, { 1, -1, -6, 105, 40, -14, 4, -1 },
101   { 1, -1, -7, 102, 44, -14, 4, -1 }, { 1, 0, -9, 100, 48, -15, 4, -1 },
102   { 1, 1, -11, 97, 53, -16, 4, -1 },  { 0, 1, -12, 95, 57, -16, 4, -1 },
103   { 0, 2, -13, 91, 61, -16, 4, -1 },  { 0, 2, -14, 88, 65, -16, 4, -1 },
104   { 0, 3, -15, 84, 69, -17, 4, 0 },   { 0, 3, -16, 81, 73, -16, 3, 0 },
105   { 0, 3, -16, 77, 77, -16, 3, 0 },   { 0, 3, -16, 73, 81, -16, 3, 0 },
106   { 0, 4, -17, 69, 84, -15, 3, 0 },   { -1, 4, -16, 65, 88, -14, 2, 0 },
107   { -1, 4, -16, 61, 91, -13, 2, 0 },  { -1, 4, -16, 57, 95, -12, 1, 0 },
108   { -1, 4, -16, 53, 97, -11, 1, 1 },  { -1, 4, -15, 48, 100, -9, 0, 1 },
109   { -1, 4, -14, 44, 102, -7, -1, 1 }, { -1, 4, -14, 40, 105, -6, -1, 1 },
110   { -1, 4, -13, 36, 106, -3, -2, 1 }, { -1, 4, -12, 32, 108, -1, -3, 1 },
111   { -1, 3, -11, 28, 110, 1, -4, 2 },  { -1, 3, -10, 24, 111, 4, -5, 2 },
112   { -1, 3, -9, 21, 111, 7, -6, 2 },   { -1, 3, -9, 17, 112, 10, -7, 3 }
113 };
114 
115 // Filters for interpolation (full-band) - no filtering for integer pixels
116 static const interp_kernel filteredinterp_filters1000[(1 << SUBPEL_BITS)] = {
117   { 0, 0, 0, 128, 0, 0, 0, 0 },        { 0, 1, -3, 128, 3, -1, 0, 0 },
118   { -1, 2, -6, 127, 7, -2, 1, 0 },     { -1, 3, -9, 126, 12, -4, 1, 0 },
119   { -1, 4, -12, 125, 16, -5, 1, 0 },   { -1, 4, -14, 123, 20, -6, 2, 0 },
120   { -1, 5, -15, 120, 25, -8, 2, 0 },   { -1, 5, -17, 118, 30, -9, 3, -1 },
121   { -1, 6, -18, 114, 35, -10, 3, -1 }, { -1, 6, -19, 111, 41, -12, 3, -1 },
122   { -1, 6, -20, 107, 46, -13, 4, -1 }, { -1, 6, -21, 103, 52, -14, 4, -1 },
123   { -1, 6, -21, 99, 57, -16, 5, -1 },  { -1, 6, -21, 94, 63, -17, 5, -1 },
124   { -1, 6, -20, 89, 68, -18, 5, -1 },  { -1, 6, -20, 84, 73, -19, 6, -1 },
125   { -1, 6, -20, 79, 79, -20, 6, -1 },  { -1, 6, -19, 73, 84, -20, 6, -1 },
126   { -1, 5, -18, 68, 89, -20, 6, -1 },  { -1, 5, -17, 63, 94, -21, 6, -1 },
127   { -1, 5, -16, 57, 99, -21, 6, -1 },  { -1, 4, -14, 52, 103, -21, 6, -1 },
128   { -1, 4, -13, 46, 107, -20, 6, -1 }, { -1, 3, -12, 41, 111, -19, 6, -1 },
129   { -1, 3, -10, 35, 114, -18, 6, -1 }, { -1, 3, -9, 30, 118, -17, 5, -1 },
130   { 0, 2, -8, 25, 120, -15, 5, -1 },   { 0, 2, -6, 20, 123, -14, 4, -1 },
131   { 0, 1, -5, 16, 125, -12, 4, -1 },   { 0, 1, -4, 12, 126, -9, 3, -1 },
132   { 0, 1, -2, 7, 127, -6, 2, -1 },     { 0, 0, -1, 3, 128, -3, 1, 0 }
133 };
134 
135 // Filters for factor of 2 downsampling.
136 static const int16_t vp9_down2_symeven_half_filter[] = { 56, 12, -3, -1 };
137 static const int16_t vp9_down2_symodd_half_filter[] = { 64, 35, 0, -3 };
138 
choose_interp_filter(int inlength,int outlength)139 static const interp_kernel *choose_interp_filter(int inlength, int outlength) {
140   int outlength16 = outlength * 16;
141   if (outlength16 >= inlength * 16)
142     return filteredinterp_filters1000;
143   else if (outlength16 >= inlength * 13)
144     return filteredinterp_filters875;
145   else if (outlength16 >= inlength * 11)
146     return filteredinterp_filters750;
147   else if (outlength16 >= inlength * 9)
148     return filteredinterp_filters625;
149   else
150     return filteredinterp_filters500;
151 }
152 
interpolate(const uint8_t * const input,int inlength,uint8_t * output,int outlength)153 static void interpolate(const uint8_t *const input, int inlength,
154                         uint8_t *output, int outlength) {
155   const int64_t delta =
156       (((uint64_t)inlength << 32) + outlength / 2) / outlength;
157   const int64_t offset =
158       inlength > outlength
159           ? (((int64_t)(inlength - outlength) << 31) + outlength / 2) /
160                 outlength
161           : -(((int64_t)(outlength - inlength) << 31) + outlength / 2) /
162                 outlength;
163   uint8_t *optr = output;
164   int x, x1, x2, sum, k, int_pel, sub_pel;
165   int64_t y;
166 
167   const interp_kernel *interp_filters =
168       choose_interp_filter(inlength, outlength);
169 
170   x = 0;
171   y = offset;
172   while ((y >> INTERP_PRECISION_BITS) < (INTERP_TAPS / 2 - 1)) {
173     x++;
174     y += delta;
175   }
176   x1 = x;
177   x = outlength - 1;
178   y = delta * x + offset;
179   while ((y >> INTERP_PRECISION_BITS) + (int64_t)(INTERP_TAPS / 2) >=
180          inlength) {
181     x--;
182     y -= delta;
183   }
184   x2 = x;
185   if (x1 > x2) {
186     for (x = 0, y = offset; x < outlength; ++x, y += delta) {
187       const int16_t *filter;
188       int_pel = y >> INTERP_PRECISION_BITS;
189       sub_pel = (y >> (INTERP_PRECISION_BITS - SUBPEL_BITS)) & SUBPEL_MASK;
190       filter = interp_filters[sub_pel];
191       sum = 0;
192       for (k = 0; k < INTERP_TAPS; ++k) {
193         const int pk = int_pel - INTERP_TAPS / 2 + 1 + k;
194         sum += filter[k] *
195                input[(pk < 0 ? 0 : (pk >= inlength ? inlength - 1 : pk))];
196       }
197       *optr++ = clip_pixel(ROUND_POWER_OF_TWO(sum, FILTER_BITS));
198     }
199   } else {
200     // Initial part.
201     for (x = 0, y = offset; x < x1; ++x, y += delta) {
202       const int16_t *filter;
203       int_pel = y >> INTERP_PRECISION_BITS;
204       sub_pel = (y >> (INTERP_PRECISION_BITS - SUBPEL_BITS)) & SUBPEL_MASK;
205       filter = interp_filters[sub_pel];
206       sum = 0;
207       for (k = 0; k < INTERP_TAPS; ++k)
208         sum += filter[k] * input[(int_pel - INTERP_TAPS / 2 + 1 + k < 0
209                                       ? 0
210                                       : int_pel - INTERP_TAPS / 2 + 1 + k)];
211       *optr++ = clip_pixel(ROUND_POWER_OF_TWO(sum, FILTER_BITS));
212     }
213     // Middle part.
214     for (; x <= x2; ++x, y += delta) {
215       const int16_t *filter;
216       int_pel = y >> INTERP_PRECISION_BITS;
217       sub_pel = (y >> (INTERP_PRECISION_BITS - SUBPEL_BITS)) & SUBPEL_MASK;
218       filter = interp_filters[sub_pel];
219       sum = 0;
220       for (k = 0; k < INTERP_TAPS; ++k)
221         sum += filter[k] * input[int_pel - INTERP_TAPS / 2 + 1 + k];
222       *optr++ = clip_pixel(ROUND_POWER_OF_TWO(sum, FILTER_BITS));
223     }
224     // End part.
225     for (; x < outlength; ++x, y += delta) {
226       const int16_t *filter;
227       int_pel = y >> INTERP_PRECISION_BITS;
228       sub_pel = (y >> (INTERP_PRECISION_BITS - SUBPEL_BITS)) & SUBPEL_MASK;
229       filter = interp_filters[sub_pel];
230       sum = 0;
231       for (k = 0; k < INTERP_TAPS; ++k)
232         sum += filter[k] * input[(int_pel - INTERP_TAPS / 2 + 1 + k >= inlength
233                                       ? inlength - 1
234                                       : int_pel - INTERP_TAPS / 2 + 1 + k)];
235       *optr++ = clip_pixel(ROUND_POWER_OF_TWO(sum, FILTER_BITS));
236     }
237   }
238 }
239 
down2_symeven(const uint8_t * const input,int length,uint8_t * output)240 static void down2_symeven(const uint8_t *const input, int length,
241                           uint8_t *output) {
242   // Actual filter len = 2 * filter_len_half.
243   const int16_t *filter = vp9_down2_symeven_half_filter;
244   const int filter_len_half = sizeof(vp9_down2_symeven_half_filter) / 2;
245   int i, j;
246   uint8_t *optr = output;
247   int l1 = filter_len_half;
248   int l2 = (length - filter_len_half);
249   l1 += (l1 & 1);
250   l2 += (l2 & 1);
251   if (l1 > l2) {
252     // Short input length.
253     for (i = 0; i < length; i += 2) {
254       int sum = (1 << (FILTER_BITS - 1));
255       for (j = 0; j < filter_len_half; ++j) {
256         sum += (input[(i - j < 0 ? 0 : i - j)] +
257                 input[(i + 1 + j >= length ? length - 1 : i + 1 + j)]) *
258                filter[j];
259       }
260       sum >>= FILTER_BITS;
261       *optr++ = clip_pixel(sum);
262     }
263   } else {
264     // Initial part.
265     for (i = 0; i < l1; i += 2) {
266       int sum = (1 << (FILTER_BITS - 1));
267       for (j = 0; j < filter_len_half; ++j) {
268         sum += (input[(i - j < 0 ? 0 : i - j)] + input[i + 1 + j]) * filter[j];
269       }
270       sum >>= FILTER_BITS;
271       *optr++ = clip_pixel(sum);
272     }
273     // Middle part.
274     for (; i < l2; i += 2) {
275       int sum = (1 << (FILTER_BITS - 1));
276       for (j = 0; j < filter_len_half; ++j) {
277         sum += (input[i - j] + input[i + 1 + j]) * filter[j];
278       }
279       sum >>= FILTER_BITS;
280       *optr++ = clip_pixel(sum);
281     }
282     // End part.
283     for (; i < length; i += 2) {
284       int sum = (1 << (FILTER_BITS - 1));
285       for (j = 0; j < filter_len_half; ++j) {
286         sum += (input[i - j] +
287                 input[(i + 1 + j >= length ? length - 1 : i + 1 + j)]) *
288                filter[j];
289       }
290       sum >>= FILTER_BITS;
291       *optr++ = clip_pixel(sum);
292     }
293   }
294 }
295 
down2_symodd(const uint8_t * const input,int length,uint8_t * output)296 static void down2_symodd(const uint8_t *const input, int length,
297                          uint8_t *output) {
298   // Actual filter len = 2 * filter_len_half - 1.
299   const int16_t *filter = vp9_down2_symodd_half_filter;
300   const int filter_len_half = sizeof(vp9_down2_symodd_half_filter) / 2;
301   int i, j;
302   uint8_t *optr = output;
303   int l1 = filter_len_half - 1;
304   int l2 = (length - filter_len_half + 1);
305   l1 += (l1 & 1);
306   l2 += (l2 & 1);
307   if (l1 > l2) {
308     // Short input length.
309     for (i = 0; i < length; i += 2) {
310       int sum = (1 << (FILTER_BITS - 1)) + input[i] * filter[0];
311       for (j = 1; j < filter_len_half; ++j) {
312         sum += (input[(i - j < 0 ? 0 : i - j)] +
313                 input[(i + j >= length ? length - 1 : i + j)]) *
314                filter[j];
315       }
316       sum >>= FILTER_BITS;
317       *optr++ = clip_pixel(sum);
318     }
319   } else {
320     // Initial part.
321     for (i = 0; i < l1; i += 2) {
322       int sum = (1 << (FILTER_BITS - 1)) + input[i] * filter[0];
323       for (j = 1; j < filter_len_half; ++j) {
324         sum += (input[(i - j < 0 ? 0 : i - j)] + input[i + j]) * filter[j];
325       }
326       sum >>= FILTER_BITS;
327       *optr++ = clip_pixel(sum);
328     }
329     // Middle part.
330     for (; i < l2; i += 2) {
331       int sum = (1 << (FILTER_BITS - 1)) + input[i] * filter[0];
332       for (j = 1; j < filter_len_half; ++j) {
333         sum += (input[i - j] + input[i + j]) * filter[j];
334       }
335       sum >>= FILTER_BITS;
336       *optr++ = clip_pixel(sum);
337     }
338     // End part.
339     for (; i < length; i += 2) {
340       int sum = (1 << (FILTER_BITS - 1)) + input[i] * filter[0];
341       for (j = 1; j < filter_len_half; ++j) {
342         sum += (input[i - j] + input[(i + j >= length ? length - 1 : i + j)]) *
343                filter[j];
344       }
345       sum >>= FILTER_BITS;
346       *optr++ = clip_pixel(sum);
347     }
348   }
349 }
350 
get_down2_length(int length,int steps)351 static int get_down2_length(int length, int steps) {
352   int s;
353   for (s = 0; s < steps; ++s) length = (length + 1) >> 1;
354   return length;
355 }
356 
get_down2_steps(int in_length,int out_length)357 static int get_down2_steps(int in_length, int out_length) {
358   int steps = 0;
359   int proj_in_length;
360   while ((proj_in_length = get_down2_length(in_length, 1)) >= out_length) {
361     ++steps;
362     in_length = proj_in_length;
363     if (in_length == 1) {
364       // Special case: we break because any further calls to get_down2_length()
365       // with be with length == 1, which return 1, resulting in an infinite
366       // loop.
367       break;
368     }
369   }
370   return steps;
371 }
372 
resize_multistep(const uint8_t * const input,int length,uint8_t * output,int olength,uint8_t * otmp)373 static void resize_multistep(const uint8_t *const input, int length,
374                              uint8_t *output, int olength, uint8_t *otmp) {
375   int steps;
376   if (length == olength) {
377     memcpy(output, input, sizeof(output[0]) * length);
378     return;
379   }
380   steps = get_down2_steps(length, olength);
381 
382   if (steps > 0) {
383     int s;
384     uint8_t *out = NULL;
385     uint8_t *otmp2;
386     int filteredlength = length;
387 
388     assert(otmp != NULL);
389     otmp2 = otmp + get_down2_length(length, 1);
390     for (s = 0; s < steps; ++s) {
391       const int proj_filteredlength = get_down2_length(filteredlength, 1);
392       const uint8_t *const in = (s == 0 ? input : out);
393       if (s == steps - 1 && proj_filteredlength == olength)
394         out = output;
395       else
396         out = (s & 1 ? otmp2 : otmp);
397       if (filteredlength & 1)
398         down2_symodd(in, filteredlength, out);
399       else
400         down2_symeven(in, filteredlength, out);
401       filteredlength = proj_filteredlength;
402     }
403     if (filteredlength != olength) {
404       interpolate(out, filteredlength, output, olength);
405     }
406   } else {
407     interpolate(input, length, output, olength);
408   }
409 }
410 
fill_col_to_arr(uint8_t * img,int stride,int len,uint8_t * arr)411 static void fill_col_to_arr(uint8_t *img, int stride, int len, uint8_t *arr) {
412   int i;
413   uint8_t *iptr = img;
414   uint8_t *aptr = arr;
415   for (i = 0; i < len; ++i, iptr += stride) {
416     *aptr++ = *iptr;
417   }
418 }
419 
fill_arr_to_col(uint8_t * img,int stride,int len,uint8_t * arr)420 static void fill_arr_to_col(uint8_t *img, int stride, int len, uint8_t *arr) {
421   int i;
422   uint8_t *iptr = img;
423   uint8_t *aptr = arr;
424   for (i = 0; i < len; ++i, iptr += stride) {
425     *iptr = *aptr++;
426   }
427 }
428 
vp9_resize_plane(const uint8_t * const input,int height,int width,int in_stride,uint8_t * output,int height2,int width2,int out_stride)429 void vp9_resize_plane(const uint8_t *const input, int height, int width,
430                       int in_stride, uint8_t *output, int height2, int width2,
431                       int out_stride) {
432   int i;
433   uint8_t *intbuf = (uint8_t *)calloc(width2 * height, sizeof(*intbuf));
434   uint8_t *tmpbuf =
435       (uint8_t *)calloc(width < height ? height : width, sizeof(*tmpbuf));
436   uint8_t *arrbuf = (uint8_t *)calloc(height, sizeof(*arrbuf));
437   uint8_t *arrbuf2 = (uint8_t *)calloc(height2, sizeof(*arrbuf2));
438   if (intbuf == NULL || tmpbuf == NULL || arrbuf == NULL || arrbuf2 == NULL)
439     goto Error;
440   assert(width > 0);
441   assert(height > 0);
442   assert(width2 > 0);
443   assert(height2 > 0);
444   for (i = 0; i < height; ++i)
445     resize_multistep(input + in_stride * i, width, intbuf + width2 * i, width2,
446                      tmpbuf);
447   for (i = 0; i < width2; ++i) {
448     fill_col_to_arr(intbuf + i, width2, height, arrbuf);
449     resize_multistep(arrbuf, height, arrbuf2, height2, tmpbuf);
450     fill_arr_to_col(output + i, out_stride, height2, arrbuf2);
451   }
452 
453 Error:
454   free(intbuf);
455   free(tmpbuf);
456   free(arrbuf);
457   free(arrbuf2);
458 }
459 
460 #if CONFIG_VP9_HIGHBITDEPTH
highbd_interpolate(const uint16_t * const input,int inlength,uint16_t * output,int outlength,int bd)461 static void highbd_interpolate(const uint16_t *const input, int inlength,
462                                uint16_t *output, int outlength, int bd) {
463   const int64_t delta =
464       (((uint64_t)inlength << 32) + outlength / 2) / outlength;
465   const int64_t offset =
466       inlength > outlength
467           ? (((int64_t)(inlength - outlength) << 31) + outlength / 2) /
468                 outlength
469           : -(((int64_t)(outlength - inlength) << 31) + outlength / 2) /
470                 outlength;
471   uint16_t *optr = output;
472   int x, x1, x2, sum, k, int_pel, sub_pel;
473   int64_t y;
474 
475   const interp_kernel *interp_filters =
476       choose_interp_filter(inlength, outlength);
477 
478   x = 0;
479   y = offset;
480   while ((y >> INTERP_PRECISION_BITS) < (INTERP_TAPS / 2 - 1)) {
481     x++;
482     y += delta;
483   }
484   x1 = x;
485   x = outlength - 1;
486   y = delta * x + offset;
487   while ((y >> INTERP_PRECISION_BITS) + (int64_t)(INTERP_TAPS / 2) >=
488          inlength) {
489     x--;
490     y -= delta;
491   }
492   x2 = x;
493   if (x1 > x2) {
494     for (x = 0, y = offset; x < outlength; ++x, y += delta) {
495       const int16_t *filter;
496       int_pel = y >> INTERP_PRECISION_BITS;
497       sub_pel = (y >> (INTERP_PRECISION_BITS - SUBPEL_BITS)) & SUBPEL_MASK;
498       filter = interp_filters[sub_pel];
499       sum = 0;
500       for (k = 0; k < INTERP_TAPS; ++k) {
501         const int pk = int_pel - INTERP_TAPS / 2 + 1 + k;
502         sum += filter[k] *
503                input[(pk < 0 ? 0 : (pk >= inlength ? inlength - 1 : pk))];
504       }
505       *optr++ = clip_pixel_highbd(ROUND_POWER_OF_TWO(sum, FILTER_BITS), bd);
506     }
507   } else {
508     // Initial part.
509     for (x = 0, y = offset; x < x1; ++x, y += delta) {
510       const int16_t *filter;
511       int_pel = y >> INTERP_PRECISION_BITS;
512       sub_pel = (y >> (INTERP_PRECISION_BITS - SUBPEL_BITS)) & SUBPEL_MASK;
513       filter = interp_filters[sub_pel];
514       sum = 0;
515       for (k = 0; k < INTERP_TAPS; ++k) {
516         assert(int_pel - INTERP_TAPS / 2 + 1 + k < inlength);
517         sum += filter[k] * input[(int_pel - INTERP_TAPS / 2 + 1 + k < 0
518                                       ? 0
519                                       : int_pel - INTERP_TAPS / 2 + 1 + k)];
520       }
521       *optr++ = clip_pixel_highbd(ROUND_POWER_OF_TWO(sum, FILTER_BITS), bd);
522     }
523     // Middle part.
524     for (; x <= x2; ++x, y += delta) {
525       const int16_t *filter;
526       int_pel = y >> INTERP_PRECISION_BITS;
527       sub_pel = (y >> (INTERP_PRECISION_BITS - SUBPEL_BITS)) & SUBPEL_MASK;
528       filter = interp_filters[sub_pel];
529       sum = 0;
530       for (k = 0; k < INTERP_TAPS; ++k)
531         sum += filter[k] * input[int_pel - INTERP_TAPS / 2 + 1 + k];
532       *optr++ = clip_pixel_highbd(ROUND_POWER_OF_TWO(sum, FILTER_BITS), bd);
533     }
534     // End part.
535     for (; x < outlength; ++x, y += delta) {
536       const int16_t *filter;
537       int_pel = y >> INTERP_PRECISION_BITS;
538       sub_pel = (y >> (INTERP_PRECISION_BITS - SUBPEL_BITS)) & SUBPEL_MASK;
539       filter = interp_filters[sub_pel];
540       sum = 0;
541       for (k = 0; k < INTERP_TAPS; ++k)
542         sum += filter[k] * input[(int_pel - INTERP_TAPS / 2 + 1 + k >= inlength
543                                       ? inlength - 1
544                                       : int_pel - INTERP_TAPS / 2 + 1 + k)];
545       *optr++ = clip_pixel_highbd(ROUND_POWER_OF_TWO(sum, FILTER_BITS), bd);
546     }
547   }
548 }
549 
highbd_down2_symeven(const uint16_t * const input,int length,uint16_t * output,int bd)550 static void highbd_down2_symeven(const uint16_t *const input, int length,
551                                  uint16_t *output, int bd) {
552   // Actual filter len = 2 * filter_len_half.
553   static const int16_t *filter = vp9_down2_symeven_half_filter;
554   const int filter_len_half = sizeof(vp9_down2_symeven_half_filter) / 2;
555   int i, j;
556   uint16_t *optr = output;
557   int l1 = filter_len_half;
558   int l2 = (length - filter_len_half);
559   l1 += (l1 & 1);
560   l2 += (l2 & 1);
561   if (l1 > l2) {
562     // Short input length.
563     for (i = 0; i < length; i += 2) {
564       int sum = (1 << (FILTER_BITS - 1));
565       for (j = 0; j < filter_len_half; ++j) {
566         sum += (input[(i - j < 0 ? 0 : i - j)] +
567                 input[(i + 1 + j >= length ? length - 1 : i + 1 + j)]) *
568                filter[j];
569       }
570       sum >>= FILTER_BITS;
571       *optr++ = clip_pixel_highbd(sum, bd);
572     }
573   } else {
574     // Initial part.
575     for (i = 0; i < l1; i += 2) {
576       int sum = (1 << (FILTER_BITS - 1));
577       for (j = 0; j < filter_len_half; ++j) {
578         sum += (input[(i - j < 0 ? 0 : i - j)] + input[i + 1 + j]) * filter[j];
579       }
580       sum >>= FILTER_BITS;
581       *optr++ = clip_pixel_highbd(sum, bd);
582     }
583     // Middle part.
584     for (; i < l2; i += 2) {
585       int sum = (1 << (FILTER_BITS - 1));
586       for (j = 0; j < filter_len_half; ++j) {
587         sum += (input[i - j] + input[i + 1 + j]) * filter[j];
588       }
589       sum >>= FILTER_BITS;
590       *optr++ = clip_pixel_highbd(sum, bd);
591     }
592     // End part.
593     for (; i < length; i += 2) {
594       int sum = (1 << (FILTER_BITS - 1));
595       for (j = 0; j < filter_len_half; ++j) {
596         sum += (input[i - j] +
597                 input[(i + 1 + j >= length ? length - 1 : i + 1 + j)]) *
598                filter[j];
599       }
600       sum >>= FILTER_BITS;
601       *optr++ = clip_pixel_highbd(sum, bd);
602     }
603   }
604 }
605 
highbd_down2_symodd(const uint16_t * const input,int length,uint16_t * output,int bd)606 static void highbd_down2_symodd(const uint16_t *const input, int length,
607                                 uint16_t *output, int bd) {
608   // Actual filter len = 2 * filter_len_half - 1.
609   static const int16_t *filter = vp9_down2_symodd_half_filter;
610   const int filter_len_half = sizeof(vp9_down2_symodd_half_filter) / 2;
611   int i, j;
612   uint16_t *optr = output;
613   int l1 = filter_len_half - 1;
614   int l2 = (length - filter_len_half + 1);
615   l1 += (l1 & 1);
616   l2 += (l2 & 1);
617   if (l1 > l2) {
618     // Short input length.
619     for (i = 0; i < length; i += 2) {
620       int sum = (1 << (FILTER_BITS - 1)) + input[i] * filter[0];
621       for (j = 1; j < filter_len_half; ++j) {
622         sum += (input[(i - j < 0 ? 0 : i - j)] +
623                 input[(i + j >= length ? length - 1 : i + j)]) *
624                filter[j];
625       }
626       sum >>= FILTER_BITS;
627       *optr++ = clip_pixel_highbd(sum, bd);
628     }
629   } else {
630     // Initial part.
631     for (i = 0; i < l1; i += 2) {
632       int sum = (1 << (FILTER_BITS - 1)) + input[i] * filter[0];
633       for (j = 1; j < filter_len_half; ++j) {
634         sum += (input[(i - j < 0 ? 0 : i - j)] + input[i + j]) * filter[j];
635       }
636       sum >>= FILTER_BITS;
637       *optr++ = clip_pixel_highbd(sum, bd);
638     }
639     // Middle part.
640     for (; i < l2; i += 2) {
641       int sum = (1 << (FILTER_BITS - 1)) + input[i] * filter[0];
642       for (j = 1; j < filter_len_half; ++j) {
643         sum += (input[i - j] + input[i + j]) * filter[j];
644       }
645       sum >>= FILTER_BITS;
646       *optr++ = clip_pixel_highbd(sum, bd);
647     }
648     // End part.
649     for (; i < length; i += 2) {
650       int sum = (1 << (FILTER_BITS - 1)) + input[i] * filter[0];
651       for (j = 1; j < filter_len_half; ++j) {
652         sum += (input[i - j] + input[(i + j >= length ? length - 1 : i + j)]) *
653                filter[j];
654       }
655       sum >>= FILTER_BITS;
656       *optr++ = clip_pixel_highbd(sum, bd);
657     }
658   }
659 }
660 
highbd_resize_multistep(const uint16_t * const input,int length,uint16_t * output,int olength,uint16_t * otmp,int bd)661 static void highbd_resize_multistep(const uint16_t *const input, int length,
662                                     uint16_t *output, int olength,
663                                     uint16_t *otmp, int bd) {
664   int steps;
665   if (length == olength) {
666     memcpy(output, input, sizeof(output[0]) * length);
667     return;
668   }
669   steps = get_down2_steps(length, olength);
670 
671   if (steps > 0) {
672     int s;
673     uint16_t *out = NULL;
674     uint16_t *otmp2;
675     int filteredlength = length;
676 
677     assert(otmp != NULL);
678     otmp2 = otmp + get_down2_length(length, 1);
679     for (s = 0; s < steps; ++s) {
680       const int proj_filteredlength = get_down2_length(filteredlength, 1);
681       const uint16_t *const in = (s == 0 ? input : out);
682       if (s == steps - 1 && proj_filteredlength == olength)
683         out = output;
684       else
685         out = (s & 1 ? otmp2 : otmp);
686       if (filteredlength & 1)
687         highbd_down2_symodd(in, filteredlength, out, bd);
688       else
689         highbd_down2_symeven(in, filteredlength, out, bd);
690       filteredlength = proj_filteredlength;
691     }
692     if (filteredlength != olength) {
693       highbd_interpolate(out, filteredlength, output, olength, bd);
694     }
695   } else {
696     highbd_interpolate(input, length, output, olength, bd);
697   }
698 }
699 
highbd_fill_col_to_arr(uint16_t * img,int stride,int len,uint16_t * arr)700 static void highbd_fill_col_to_arr(uint16_t *img, int stride, int len,
701                                    uint16_t *arr) {
702   int i;
703   uint16_t *iptr = img;
704   uint16_t *aptr = arr;
705   for (i = 0; i < len; ++i, iptr += stride) {
706     *aptr++ = *iptr;
707   }
708 }
709 
highbd_fill_arr_to_col(uint16_t * img,int stride,int len,uint16_t * arr)710 static void highbd_fill_arr_to_col(uint16_t *img, int stride, int len,
711                                    uint16_t *arr) {
712   int i;
713   uint16_t *iptr = img;
714   uint16_t *aptr = arr;
715   for (i = 0; i < len; ++i, iptr += stride) {
716     *iptr = *aptr++;
717   }
718 }
719 
vp9_highbd_resize_plane(const uint8_t * const input,int height,int width,int in_stride,uint8_t * output,int height2,int width2,int out_stride,int bd)720 void vp9_highbd_resize_plane(const uint8_t *const input, int height, int width,
721                              int in_stride, uint8_t *output, int height2,
722                              int width2, int out_stride, int bd) {
723   int i;
724   uint16_t *intbuf = (uint16_t *)malloc(sizeof(uint16_t) * width2 * height);
725   uint16_t *tmpbuf =
726       (uint16_t *)malloc(sizeof(uint16_t) * (width < height ? height : width));
727   uint16_t *arrbuf = (uint16_t *)malloc(sizeof(uint16_t) * height);
728   uint16_t *arrbuf2 = (uint16_t *)malloc(sizeof(uint16_t) * height2);
729   if (intbuf == NULL || tmpbuf == NULL || arrbuf == NULL || arrbuf2 == NULL)
730     goto Error;
731   assert(width > 0);
732   assert(height > 0);
733   assert(width2 > 0);
734   assert(height2 > 0);
735   for (i = 0; i < height; ++i) {
736     highbd_resize_multistep(CONVERT_TO_SHORTPTR(input + in_stride * i), width,
737                             intbuf + width2 * i, width2, tmpbuf, bd);
738   }
739   for (i = 0; i < width2; ++i) {
740     highbd_fill_col_to_arr(intbuf + i, width2, height, arrbuf);
741     highbd_resize_multistep(arrbuf, height, arrbuf2, height2, tmpbuf, bd);
742     highbd_fill_arr_to_col(CONVERT_TO_SHORTPTR(output + i), out_stride, height2,
743                            arrbuf2);
744   }
745 
746 Error:
747   free(intbuf);
748   free(tmpbuf);
749   free(arrbuf);
750   free(arrbuf2);
751 }
752 #endif  // CONFIG_VP9_HIGHBITDEPTH
753