1 /*
2 * Copyright (c) 2018, Alliance for Open Media. All rights reserved.
3 *
4 * This source code is subject to the terms of the BSD 2 Clause License and
5 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
6 * was not distributed with this source code in the LICENSE file, you can
7 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
8 * Media Patent License 1.0 was not distributed with this source code in the
9 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
10 */
11
12 #include <assert.h>
13 #include <stdlib.h>
14 #include <math.h>
15
16 #include "config/av1_rtcd.h"
17 #include "av1/encoder/dwt.h"
18
19 // Note: block length must be even for this implementation
analysis_53_row(int length,tran_low_t * x,tran_low_t * lowpass,tran_low_t * highpass)20 static void analysis_53_row(int length, tran_low_t *x, tran_low_t *lowpass,
21 tran_low_t *highpass) {
22 int n;
23 tran_low_t r, *a, *b;
24
25 n = length >> 1;
26 b = highpass;
27 a = lowpass;
28 while (--n) {
29 *a++ = (r = *x++) * 2;
30 *b++ = *x - ((r + x[1] + 1) >> 1);
31 x++;
32 }
33 *a = (r = *x++) * 2;
34 *b = *x - r;
35
36 n = length >> 1;
37 b = highpass;
38 a = lowpass;
39 r = *highpass;
40 while (n--) {
41 *a++ += (r + (*b) + 1) >> 1;
42 r = *b++;
43 }
44 }
45
analysis_53_col(int length,tran_low_t * x,tran_low_t * lowpass,tran_low_t * highpass)46 static void analysis_53_col(int length, tran_low_t *x, tran_low_t *lowpass,
47 tran_low_t *highpass) {
48 int n;
49 tran_low_t r, *a, *b;
50
51 n = length >> 1;
52 b = highpass;
53 a = lowpass;
54 while (--n) {
55 *a++ = (r = *x++);
56 *b++ = (((*x) * 2) - (r + x[1]) + 2) >> 2;
57 x++;
58 }
59 *a = (r = *x++);
60 *b = (*x - r + 1) >> 1;
61
62 n = length >> 1;
63 b = highpass;
64 a = lowpass;
65 r = *highpass;
66 while (n--) {
67 *a++ += (r + (*b) + 1) >> 1;
68 r = *b++;
69 }
70 }
71
dyadic_analyze_53_uint8_input(int levels,int width,int height,const uint8_t * x,int pitch_x,tran_low_t * c,int pitch_c,int dwt_scale_bits,int hbd)72 static void dyadic_analyze_53_uint8_input(int levels, int width, int height,
73 const uint8_t *x, int pitch_x,
74 tran_low_t *c, int pitch_c,
75 int dwt_scale_bits, int hbd) {
76 int lv, i, j, nh, nw, hh = height, hw = width;
77 tran_low_t buffer[2 * DWT_MAX_LENGTH];
78
79 if (hbd) {
80 const uint16_t *x16 = CONVERT_TO_SHORTPTR(x);
81 for (i = 0; i < height; i++) {
82 for (j = 0; j < width; j++) {
83 c[i * pitch_c + j] = x16[i * pitch_x + j] << dwt_scale_bits;
84 }
85 }
86 } else {
87 for (i = 0; i < height; i++) {
88 for (j = 0; j < width; j++) {
89 c[i * pitch_c + j] = x[i * pitch_x + j] << dwt_scale_bits;
90 }
91 }
92 }
93
94 for (lv = 0; lv < levels; lv++) {
95 nh = hh;
96 hh = (hh + 1) >> 1;
97 nw = hw;
98 hw = (hw + 1) >> 1;
99 if ((nh < 2) || (nw < 2)) return;
100 for (i = 0; i < nh; i++) {
101 memcpy(buffer, &c[i * pitch_c], nw * sizeof(tran_low_t));
102 analysis_53_row(nw, buffer, &c[i * pitch_c], &c[i * pitch_c] + hw);
103 }
104 for (j = 0; j < nw; j++) {
105 for (i = 0; i < nh; i++) buffer[i + nh] = c[i * pitch_c + j];
106 analysis_53_col(nh, buffer + nh, buffer, buffer + hh);
107 for (i = 0; i < nh; i++) c[i * pitch_c + j] = buffer[i];
108 }
109 }
110 }
111
av1_fdwt8x8_uint8_input_c(const uint8_t * input,tran_low_t * output,int stride,int hbd)112 void av1_fdwt8x8_uint8_input_c(const uint8_t *input, tran_low_t *output,
113 int stride, int hbd) {
114 dyadic_analyze_53_uint8_input(4, 8, 8, input, stride, output, 8, 2, hbd);
115 }
116
haar_ac_sad(const tran_low_t * output,int bw,int bh,int stride)117 static int haar_ac_sad(const tran_low_t *output, int bw, int bh, int stride) {
118 int acsad = 0;
119
120 for (int r = 0; r < bh; ++r)
121 for (int c = 0; c < bw; ++c) {
122 if (r >= bh / 2 || c >= bw / 2) acsad += abs(output[r * stride + c]);
123 }
124 return acsad;
125 }
126
haar_ac_sad_8x8_uint8_input(const uint8_t * input,int stride,int hbd)127 static int haar_ac_sad_8x8_uint8_input(const uint8_t *input, int stride,
128 int hbd) {
129 tran_low_t output[64];
130
131 av1_fdwt8x8_uint8_input(input, output, stride, hbd);
132 return haar_ac_sad(output, 8, 8, 8);
133 }
134
av1_haar_ac_sad_mxn_uint8_input(const uint8_t * input,int stride,int hbd,int num_8x8_rows,int num_8x8_cols)135 int64_t av1_haar_ac_sad_mxn_uint8_input(const uint8_t *input, int stride,
136 int hbd, int num_8x8_rows,
137 int num_8x8_cols) {
138 int64_t wavelet_energy = 0;
139 for (int r8 = 0; r8 < num_8x8_rows; ++r8) {
140 for (int c8 = 0; c8 < num_8x8_cols; ++c8) {
141 wavelet_energy += haar_ac_sad_8x8_uint8_input(
142 input + c8 * 8 + r8 * 8 * stride, stride, hbd);
143 }
144 }
145 return wavelet_energy;
146 }
147