/work/vvenc/source/Lib/CommonLib/Buffer.cpp
Line | Count | Source |
1 | | /* ----------------------------------------------------------------------------- |
2 | | The copyright in this software is being made available under the Clear BSD |
3 | | License, included below. No patent rights, trademark rights and/or |
4 | | other Intellectual Property Rights other than the copyrights concerning |
5 | | the Software are granted under this license. |
6 | | |
7 | | The Clear BSD License |
8 | | |
9 | | Copyright (c) 2019-2026, Fraunhofer-Gesellschaft zur Förderung der angewandten Forschung e.V. & The VVenC Authors. |
10 | | All rights reserved. |
11 | | |
12 | | Redistribution and use in source and binary forms, with or without modification, |
13 | | are permitted (subject to the limitations in the disclaimer below) provided that |
14 | | the following conditions are met: |
15 | | |
16 | | * Redistributions of source code must retain the above copyright notice, |
17 | | this list of conditions and the following disclaimer. |
18 | | |
19 | | * Redistributions in binary form must reproduce the above copyright |
20 | | notice, this list of conditions and the following disclaimer in the |
21 | | documentation and/or other materials provided with the distribution. |
22 | | |
23 | | * Neither the name of the copyright holder nor the names of its |
24 | | contributors may be used to endorse or promote products derived from this |
25 | | software without specific prior written permission. |
26 | | |
27 | | NO EXPRESS OR IMPLIED LICENSES TO ANY PARTY'S PATENT RIGHTS ARE GRANTED BY |
28 | | THIS LICENSE. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND |
29 | | CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
30 | | LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A |
31 | | PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR |
32 | | CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
33 | | EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
34 | | PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR |
35 | | BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER |
36 | | IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
37 | | ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
38 | | POSSIBILITY OF SUCH DAMAGE. |
39 | | |
40 | | |
41 | | ------------------------------------------------------------------------------------------- */ |
42 | | |
43 | | |
44 | | /** \file Buffer.cpp |
45 | | * \brief Low-overhead class describing 2D memory layout |
46 | | */ |
47 | | |
48 | | #define DONT_UNDEF_SIZE_AWARE_PER_EL_OP |
49 | | |
50 | | // unit needs to come first due to a forward declaration |
51 | | #include "Unit.h" |
52 | | #include "Slice.h" |
53 | | #include "InterpolationFilter.h" |
54 | | |
55 | | //! \ingroup CommonLib |
56 | | //! \{ |
57 | | |
58 | | namespace vvenc { |
59 | | |
60 | | void weightCiipCore( Pel* res, const Pel* src, const int numSamples, int numIntra ) |
61 | 0 | { |
62 | 0 | if( numIntra == 1 ) |
63 | 0 | { |
64 | 0 | for (int n = 0; n < numSamples; n+=2) |
65 | 0 | { |
66 | 0 | res[n ] = (res[n ] + src[n ] + 1) >> 1; |
67 | 0 | res[n+1] = (res[n+1] + src[n+1] + 1) >> 1; |
68 | 0 | } |
69 | 0 | } |
70 | 0 | else |
71 | 0 | { |
72 | 0 | const Pel* scale = numIntra ? src : res; |
73 | 0 | const Pel* unscale = numIntra ? res : src; |
74 | |
|
75 | 0 | for (int n = 0; n < numSamples; n+=2) |
76 | 0 | { |
77 | 0 | res[n ] = (unscale[n ] + 3*scale[n ] + 2) >> 2; |
78 | 0 | res[n+1] = (unscale[n+1] + 3*scale[n+1] + 2) >> 2; |
79 | 0 | } |
80 | 0 | } |
81 | 0 | } |
82 | | |
83 | | template< unsigned inputSize, unsigned outputSize > |
84 | | void mipMatrixMulCore( Pel* res, const Pel* input, const uint8_t* weight, const int maxVal, const int inputOffset, bool transpose ) |
85 | 257k | { |
86 | 257k | Pel buffer[ outputSize*outputSize]; |
87 | | |
88 | 257k | int sum = 0; |
89 | 2.32M | for( int i = 0; i < inputSize; i++ ) |
90 | 2.06M | { |
91 | 2.06M | sum += input[i]; |
92 | 2.06M | } |
93 | 257k | const int offset = (1 << (MIP_SHIFT_MATRIX - 1)) - MIP_OFFSET_MATRIX * sum + (inputOffset << MIP_SHIFT_MATRIX); |
94 | 257k | CHECK( inputSize != 4 * (inputSize >> 2), "Error, input size not divisible by four" ); |
95 | | |
96 | 257k | Pel* mat = transpose ? buffer : res; |
97 | 257k | unsigned posRes = 0; |
98 | 16.4M | for( unsigned n = 0; n < outputSize*outputSize; n++ ) |
99 | 16.1M | { |
100 | 16.1M | int tmp0 = input[0] * weight[0]; |
101 | 16.1M | int tmp1 = input[1] * weight[1]; |
102 | 16.1M | int tmp2 = input[2] * weight[2]; |
103 | 16.1M | int tmp3 = input[3] * weight[3]; |
104 | 16.1M | if( 8 == inputSize ) |
105 | 16.1M | { |
106 | 16.1M | tmp0 += input[4] * weight[4]; |
107 | 16.1M | tmp1 += input[5] * weight[5]; |
108 | 16.1M | tmp2 += input[6] * weight[6]; |
109 | 16.1M | tmp3 += input[7] * weight[7]; |
110 | 16.1M | } |
111 | 16.1M | mat[posRes++] = Clip3<int>( 0, maxVal, ((tmp0 + tmp1 + tmp2 + tmp3 + offset) >> MIP_SHIFT_MATRIX) ); |
112 | | |
113 | 16.1M | weight += inputSize; |
114 | 16.1M | } |
115 | | |
116 | 257k | if( transpose ) |
117 | 117k | { |
118 | 1.04M | for( int j = 0; j < outputSize; j++ ) |
119 | 924k | { |
120 | 8.26M | for( int i = 0; i < outputSize; i++ ) |
121 | 7.34M | { |
122 | 7.34M | res[j * outputSize + i] = buffer[i * outputSize + j]; |
123 | 7.34M | } |
124 | 924k | } |
125 | 117k | } |
126 | 257k | } Unexecuted instantiation: void vvenc::mipMatrixMulCore<4u, 4u>(short*, short const*, unsigned char const*, int, int, bool) void vvenc::mipMatrixMulCore<8u, 4u>(short*, short const*, unsigned char const*, int, int, bool) Line | Count | Source | 85 | 6.73k | { | 86 | 6.73k | Pel buffer[ outputSize*outputSize]; | 87 | | | 88 | 6.73k | int sum = 0; | 89 | 60.6k | for( int i = 0; i < inputSize; i++ ) | 90 | 53.8k | { | 91 | 53.8k | sum += input[i]; | 92 | 53.8k | } | 93 | 6.73k | const int offset = (1 << (MIP_SHIFT_MATRIX - 1)) - MIP_OFFSET_MATRIX * sum + (inputOffset << MIP_SHIFT_MATRIX); | 94 | 6.73k | CHECK( inputSize != 4 * (inputSize >> 2), "Error, input size not divisible by four" ); | 95 | | | 96 | 6.73k | Pel* mat = transpose ? buffer : res; | 97 | 6.73k | unsigned posRes = 0; | 98 | 114k | for( unsigned n = 0; n < outputSize*outputSize; n++ ) | 99 | 107k | { | 100 | 107k | int tmp0 = input[0] * weight[0]; | 101 | 107k | int tmp1 = input[1] * weight[1]; | 102 | 107k | int tmp2 = input[2] * weight[2]; | 103 | 107k | int tmp3 = input[3] * weight[3]; | 104 | 107k | if( 8 == inputSize ) | 105 | 107k | { | 106 | 107k | tmp0 += input[4] * weight[4]; | 107 | 107k | tmp1 += input[5] * weight[5]; | 108 | 107k | tmp2 += input[6] * weight[6]; | 109 | 107k | tmp3 += input[7] * weight[7]; | 110 | 107k | } | 111 | 107k | mat[posRes++] = Clip3<int>( 0, maxVal, ((tmp0 + tmp1 + tmp2 + tmp3 + offset) >> MIP_SHIFT_MATRIX) ); | 112 | | | 113 | 107k | weight += inputSize; | 114 | 107k | } | 115 | | | 116 | 6.73k | if( transpose ) | 117 | 3.36k | { | 118 | 16.8k | for( int j = 0; j < outputSize; j++ ) | 119 | 13.4k | { | 120 | 67.3k | for( int i = 0; i < outputSize; i++ ) | 121 | 53.8k | { | 122 | 53.8k | res[j * outputSize + i] = buffer[i * outputSize + j]; | 123 | 53.8k | } | 124 | 13.4k | } | 125 | 3.36k | } | 126 | 6.73k | } |
void vvenc::mipMatrixMulCore<8u, 8u>(short*, short const*, unsigned char const*, int, int, bool) Line | Count | Source | 85 | 251k | { | 86 | 251k | Pel buffer[ outputSize*outputSize]; | 87 | | | 88 | 251k | int sum = 0; | 89 | 2.26M | for( int i = 0; i < inputSize; i++ ) | 90 | 2.00M | { | 91 | 2.00M | sum += input[i]; | 92 | 2.00M | } | 93 | 251k | const int offset = (1 << (MIP_SHIFT_MATRIX - 1)) - MIP_OFFSET_MATRIX * sum + (inputOffset << MIP_SHIFT_MATRIX); | 94 | 251k | CHECK( inputSize != 4 * (inputSize >> 2), "Error, input size not divisible by four" ); | 95 | | | 96 | 251k | Pel* mat = transpose ? buffer : res; | 97 | 251k | unsigned posRes = 0; | 98 | 16.3M | for( unsigned n = 0; n < outputSize*outputSize; n++ ) | 99 | 16.0M | { | 100 | 16.0M | int tmp0 = input[0] * weight[0]; | 101 | 16.0M | int tmp1 = input[1] * weight[1]; | 102 | 16.0M | int tmp2 = input[2] * weight[2]; | 103 | 16.0M | int tmp3 = input[3] * weight[3]; | 104 | 16.0M | if( 8 == inputSize ) | 105 | 16.0M | { | 106 | 16.0M | tmp0 += input[4] * weight[4]; | 107 | 16.0M | tmp1 += input[5] * weight[5]; | 108 | 16.0M | tmp2 += input[6] * weight[6]; | 109 | 16.0M | tmp3 += input[7] * weight[7]; | 110 | 16.0M | } | 111 | 16.0M | mat[posRes++] = Clip3<int>( 0, maxVal, ((tmp0 + tmp1 + tmp2 + tmp3 + offset) >> MIP_SHIFT_MATRIX) ); | 112 | | | 113 | 16.0M | weight += inputSize; | 114 | 16.0M | } | 115 | | | 116 | 251k | if( transpose ) | 117 | 113k | { | 118 | 1.02M | for( int j = 0; j < outputSize; j++ ) | 119 | 910k | { | 120 | 8.19M | for( int i = 0; i < outputSize; i++ ) | 121 | 7.28M | { | 122 | 7.28M | res[j * outputSize + i] = buffer[i * outputSize + j]; | 123 | 7.28M | } | 124 | 910k | } | 125 | 113k | } | 126 | 251k | } |
|
127 | | |
128 | | template< typename T > |
129 | | void addAvgCore( const T* src1, int src1Stride, const T* src2, int src2Stride, T* dest, int dstStride, int width, int height, unsigned rshift, int offset, const ClpRng& clpRng ) |
130 | 0 | { |
131 | 0 | #define ADD_AVG_CORE_OP( ADDR ) dest[ADDR] = ClipPel( rightShiftU( ( src1[ADDR] + src2[ADDR] + offset ), rshift ), clpRng ) |
132 | 0 | #define ADD_AVG_CORE_INC \ |
133 | 0 | src1 += src1Stride; \ |
134 | 0 | src2 += src2Stride; \ |
135 | 0 | dest += dstStride; \ |
136 | 0 |
|
137 | 0 | SIZE_AWARE_PER_EL_OP( ADD_AVG_CORE_OP, ADD_AVG_CORE_INC ); |
138 | |
|
139 | 0 | #undef ADD_AVG_CORE_OP |
140 | 0 | #undef ADD_AVG_CORE_INC |
141 | 0 | } |
142 | | |
143 | | template<typename T> |
144 | | void addWeightedAvgCore( const T* src1, int src1Stride, const T* src2, int src2Stride, T* dest, int destStride, int width, int height, unsigned rshift, int offset, int w0, int w1, const ClpRng& clpRng ) |
145 | 0 | { |
146 | 0 | #define ADD_WGHT_AVG_OP( ADDR ) dest[ADDR] = ClipPel( rightShiftU( ( src1[ADDR]*w0 + src2[ADDR]*w1 + offset ), rshift ), clpRng ) |
147 | 0 | #define ADD_WGHT_AVG_INC \ |
148 | 0 | src1 += src1Stride; \ |
149 | 0 | src2 += src2Stride; \ |
150 | 0 | dest += destStride; \ |
151 | 0 |
|
152 | 0 | SIZE_AWARE_PER_EL_OP( ADD_WGHT_AVG_OP, ADD_WGHT_AVG_INC ); |
153 | |
|
154 | 0 | #undef ADD_WGHT_AVG_OP |
155 | 0 | #undef ADD_WGHT_AVG_INC |
156 | 0 | } |
157 | | |
158 | | template<typename T> |
159 | | void subsCore( const T* src0, int src0Stride, const T* src1, int src1Stride, T* dest, int destStride, int width, int height ) |
160 | 796k | { |
161 | 796k | #define SUBS_INC \ |
162 | 796k | dest += destStride; \ |
163 | 796k | src0 += src0Stride; \ |
164 | 796k | src1 += src1Stride; \ |
165 | 796k | |
166 | 373M | #define SUBS_OP( ADDR ) dest[ADDR] = src0[ADDR] - src1[ADDR] |
167 | | |
168 | 373M | SIZE_AWARE_PER_EL_OP( SUBS_OP, SUBS_INC ); |
169 | | |
170 | 796k | #undef SUBS_OP |
171 | 796k | #undef SUBS_INC |
172 | 796k | } |
173 | | |
174 | | void removeHighFreq(int16_t* dst, int dstStride, const int16_t* src, int srcStride, int width, int height) |
175 | 0 | { |
176 | 0 | #define REM_HF_INC \ |
177 | 0 | src += srcStride; \ |
178 | 0 | dst += dstStride; \ |
179 | 0 |
|
180 | 0 | #define REM_HF_OP( ADDR ) dst[ADDR] = 2 * dst[ADDR] - src[ADDR] |
181 | |
|
182 | 0 | SIZE_AWARE_PER_EL_OP(REM_HF_OP, REM_HF_INC); |
183 | |
|
184 | 0 | #undef REM_HF_INC |
185 | 0 | #undef REM_HF_OP |
186 | 0 | #undef REM_HF_OP_CLIP |
187 | 0 | } |
188 | | |
189 | | template<typename T> |
190 | | void reconstructCore( const T* src1, int src1Stride, const T* src2, int src2Stride, T* dest, int dstStride, int width, int height, const ClpRng& clpRng ) |
191 | 8.95k | { |
192 | 3.77M | #define RECO_CORE_OP( ADDR ) dest[ADDR] = ClipPel( src1[ADDR] + src2[ADDR], clpRng ) |
193 | 8.95k | #define RECO_CORE_INC \ |
194 | 8.95k | src1 += src1Stride; \ |
195 | 8.95k | src2 += src2Stride; \ |
196 | 8.95k | dest += dstStride; \ |
197 | 8.95k | |
198 | 3.77M | SIZE_AWARE_PER_EL_OP( RECO_CORE_OP, RECO_CORE_INC ); |
199 | | |
200 | 8.95k | #undef RECO_CORE_OP |
201 | 8.95k | #undef RECO_CORE_INC |
202 | 8.95k | } |
203 | | |
204 | | template<typename T> |
205 | | void recoCore( const T* src1, const T* src2, T* dest, int numSamples, const ClpRng& clpRng ) |
206 | 2.23M | { |
207 | 338M | for( int n = 0; n < numSamples; n+=2) |
208 | 336M | { |
209 | 336M | dest[n] = ClipPel( src1[n] + src2[n], clpRng ); |
210 | 336M | dest[n+1] = ClipPel( src1[n+1] + src2[n+1], clpRng ); |
211 | 336M | } |
212 | 2.23M | } |
213 | | |
214 | | template<typename T> |
215 | | void copyClipCore( const T* src, Pel* dst, int numSamples, const ClpRng& clpRng ) |
216 | 0 | { |
217 | 0 | for( int n = 0; n < numSamples; n+=2) |
218 | 0 | { |
219 | 0 | dst[n] = ClipPel( src[n] , clpRng ); |
220 | 0 | dst[n+1] = ClipPel( src[n+1] , clpRng ); |
221 | 0 | } |
222 | 0 | } |
223 | | |
224 | | template< typename T > |
225 | | void addAvgCore( const T* src1, const T* src2, T* dest, int numSamples, unsigned rshift, int offset, const ClpRng& clpRng ) |
226 | 0 | { |
227 | 0 | for( int n = 0; n < numSamples; n+=2) |
228 | 0 | { |
229 | 0 | dest[n] = ClipPel( rightShiftU( ( src1[n] + src2[n] + offset ), rshift ), clpRng ); |
230 | 0 | dest[n+1] = ClipPel( rightShiftU( ( src1[n+1] + src2[n+1] + offset ), rshift ), clpRng ); |
231 | 0 | } |
232 | 0 | } |
233 | | |
234 | | template< typename T > |
235 | | void roundGeoCore( const T* src, T* dest, const int numSamples, unsigned rshift, int offset, const ClpRng &clpRng) |
236 | 0 | { |
237 | 0 | for( int i = 0; i < numSamples; i+=2) |
238 | 0 | { |
239 | 0 | dest[i] = ClipPel(rightShiftU(src[i ] + offset, rshift), clpRng); |
240 | 0 | dest[i+1] = ClipPel(rightShiftU(src[i+1] + offset, rshift), clpRng); |
241 | 0 | } |
242 | 0 | } |
243 | | |
244 | | template<typename T> |
245 | | void linTfCore( const T* src, int srcStride, Pel* dst, int dstStride, int width, int height, int scale, unsigned shift, int offset, const ClpRng& clpRng, bool bClip ) |
246 | 203k | { |
247 | 203k | #define LINTF_CORE_INC \ |
248 | 203k | src += srcStride; \ |
249 | 203k | dst += dstStride; \ |
250 | 203k | |
251 | 203k | if( bClip ) |
252 | 203k | { |
253 | 41.0M | #define LINTF_CORE_OP( ADDR ) dst[ADDR] = ( Pel ) ClipPel( rightShiftU( scale * src[ADDR], shift ) + offset, clpRng ) |
254 | | |
255 | 41.0M | SIZE_AWARE_PER_EL_OP( LINTF_CORE_OP, LINTF_CORE_INC ); |
256 | | |
257 | 203k | #undef LINTF_CORE_OP |
258 | 203k | } |
259 | 0 | else |
260 | 0 | { |
261 | 0 | #define LINTF_CORE_OP( ADDR ) dst[ADDR] = ( Pel ) ( rightShiftU( scale * src[ADDR], shift ) + offset ) |
262 | |
|
263 | 0 | SIZE_AWARE_PER_EL_OP( LINTF_CORE_OP, LINTF_CORE_INC ); |
264 | |
|
265 | 0 | #undef LINTF_CORE_OP |
266 | 0 | } |
267 | 203k | #undef LINTF_CORE_INC |
268 | 203k | } |
269 | | |
270 | | template<typename T, int N> |
271 | | void transposeNxNCore( const Pel* src, int srcStride, Pel* dst, int dstStride ) |
272 | 7.24M | { |
273 | 63.6M | for( int i = 0; i < N; i++ ) |
274 | 56.3M | { |
275 | 500M | for( int j = 0; j < N; j++ ) |
276 | 444M | { |
277 | 444M | dst[j * dstStride] = src[j]; |
278 | 444M | } |
279 | | |
280 | 56.3M | dst++; |
281 | 56.3M | src += srcStride; |
282 | 56.3M | } |
283 | 7.24M | } void vvenc::transposeNxNCore<short, 4>(short const*, int, short*, int) Line | Count | Source | 272 | 396k | { | 273 | 1.98M | for( int i = 0; i < N; i++ ) | 274 | 1.58M | { | 275 | 7.92M | for( int j = 0; j < N; j++ ) | 276 | 6.33M | { | 277 | 6.33M | dst[j * dstStride] = src[j]; | 278 | 6.33M | } | 279 | | | 280 | 1.58M | dst++; | 281 | 1.58M | src += srcStride; | 282 | 1.58M | } | 283 | 396k | } |
void vvenc::transposeNxNCore<short, 8>(short const*, int, short*, int) Line | Count | Source | 272 | 6.84M | { | 273 | 61.6M | for( int i = 0; i < N; i++ ) | 274 | 54.7M | { | 275 | 493M | for( int j = 0; j < N; j++ ) | 276 | 438M | { | 277 | 438M | dst[j * dstStride] = src[j]; | 278 | 438M | } | 279 | | | 280 | 54.7M | dst++; | 281 | 54.7M | src += srcStride; | 282 | 54.7M | } | 283 | 6.84M | } |
|
284 | | |
285 | | template<typename T> |
286 | | void copyClipCore( const T* src, int srcStride, Pel* dst, int dstStride, int width, int height, const ClpRng& clpRng ) |
287 | 0 | { |
288 | 0 | #define RECO_OP( ADDR ) dst[ADDR] = ClipPel( src[ADDR], clpRng ) |
289 | 0 | #define RECO_INC \ |
290 | 0 | src += srcStride; \ |
291 | 0 | dst += dstStride; \ |
292 | 0 |
|
293 | 0 | SIZE_AWARE_PER_EL_OP( RECO_OP, RECO_INC ); |
294 | |
|
295 | 0 | #undef RECO_OP |
296 | 0 | #undef RECO_INC |
297 | 0 | } |
298 | | |
299 | | void copyBufferCore( const char* src, int srcStride, char* dst, int dstStride, int numBytes, int height) |
300 | 8.50M | { |
301 | 142M | for( int i = 0; i < height; i++, src += srcStride, dst += dstStride ) |
302 | 134M | { |
303 | 134M | memcpy( dst, src, numBytes ); |
304 | 134M | } |
305 | 8.50M | } |
306 | | |
307 | | void fillMapPtr_Core( void** ptrMap, const ptrdiff_t mapStride, int width, int height, void* val ) |
308 | 442k | { |
309 | 442k | if( width == mapStride ) |
310 | 289k | { |
311 | 289k | std::fill_n( ptrMap, width * height, val ); |
312 | 289k | } |
313 | 152k | else |
314 | 152k | { |
315 | 1.42M | while( height-- ) |
316 | 1.26M | { |
317 | 1.26M | std::fill_n( ptrMap, width, val ); |
318 | 1.26M | ptrMap += mapStride; |
319 | 1.26M | } |
320 | 152k | } |
321 | 442k | } |
322 | | |
323 | | uint64_t AvgHighPassCore( const int width, const int height, const Pel* pSrc, const int iSrcStride) |
324 | 10.8k | { |
325 | 10.8k | uint64_t saAct = 0; |
326 | 889k | for (int y = 1; y < height - 1; y++) |
327 | 878k | { |
328 | 82.9M | for (int x = 1; x < width - 1; x++) // center cols |
329 | 82.0M | { |
330 | 82.0M | const int s = 12 * (int) pSrc[x ] - 2 * ((int) pSrc[x-1] + (int) pSrc[x+1] + (int) pSrc[x -iSrcStride] + (int) pSrc[x +iSrcStride]) |
331 | 82.0M | - ((int) pSrc[x-1-iSrcStride] + (int) pSrc[x+1-iSrcStride] + (int) pSrc[x-1+iSrcStride] + (int) pSrc[x+1+iSrcStride]); |
332 | 82.0M | saAct += abs (s); |
333 | 82.0M | } |
334 | 878k | pSrc += iSrcStride; |
335 | 878k | } |
336 | 10.8k | return saAct; |
337 | 10.8k | } |
338 | | |
339 | | uint64_t HDHighPassCore (const int width, const int height,const Pel* pSrc,const Pel* pSM1,const int iSrcStride,const int iSM1Stride) |
340 | 0 | { |
341 | 0 | uint64_t taAct = 0; |
342 | 0 | for (int y = 1; y < height - 1; y++) |
343 | 0 | { |
344 | 0 | for (int x = 1; x < width - 1; x++) // cnt cols |
345 | 0 | { |
346 | 0 | const int t = (int) pSrc[x] - (int) pSM1[x]; |
347 | 0 | taAct += (1 + 3 * abs (t)) >> 1; |
348 | 0 | } |
349 | 0 | pSrc += iSrcStride; |
350 | 0 | pSM1 += iSM1Stride; |
351 | 0 | } |
352 | 0 | return taAct; |
353 | 0 | } |
354 | | |
355 | | uint64_t HDHighPass2Core (const int width, const int height,const Pel* pSrc,const Pel* pSM1,const Pel* pSM2,const int iSrcStride,const int iSM1Stride,const int iSM2Stride) |
356 | 0 | { |
357 | 0 | uint64_t taAct = 0; |
358 | 0 | for (int y = 1; y < height - 1; y++) |
359 | 0 | { |
360 | 0 | for (int x = 1; x < width - 1; x++) // cnt cols |
361 | 0 | { |
362 | 0 | const int t = (int) pSrc[x] - 2 * (int) pSM1[x] + (int) pSM2[x]; |
363 | 0 | taAct += abs (t); |
364 | 0 | } |
365 | 0 | pSrc += iSrcStride; |
366 | 0 | pSM1 += iSM1Stride; |
367 | 0 | pSM2 += iSM2Stride; |
368 | 0 | } |
369 | 0 | return taAct; |
370 | 0 | } |
371 | | uint64_t AvgHighPassWithDownsamplingCore( const int width, const int height, const Pel* pSrc, const int iSrcStride) |
372 | 0 | { |
373 | 0 | uint64_t saAct = 0; |
374 | 0 | pSrc -= iSrcStride; |
375 | 0 | pSrc -= iSrcStride; |
376 | 0 | for (int y = 2; y < height - 2; y += 2) |
377 | 0 | { |
378 | 0 | for (int x = 2; x < width - 2; x += 2) |
379 | 0 | { |
380 | 0 | const int f = 12 * ((int)pSrc[ y *iSrcStride + x ] + (int)pSrc[ y *iSrcStride + x+1] + (int)pSrc[(y+1)*iSrcStride + x ] + (int)pSrc[(y+1)*iSrcStride + x+1]) |
381 | 0 | - 3 * ((int)pSrc[(y-1)*iSrcStride + x ] + (int)pSrc[(y-1)*iSrcStride + x+1] + (int)pSrc[(y+2)*iSrcStride + x ] + (int)pSrc[(y+2)*iSrcStride + x+1]) |
382 | 0 | - 3 * ((int)pSrc[ y *iSrcStride + x-1] + (int)pSrc[ y *iSrcStride + x+2] + (int)pSrc[(y+1)*iSrcStride + x-1] + (int)pSrc[(y+1)*iSrcStride + x+2]) |
383 | 0 | - 2 * ((int)pSrc[(y-1)*iSrcStride + x-1] + (int)pSrc[(y-1)*iSrcStride + x+2] + (int)pSrc[(y+2)*iSrcStride + x-1] + (int)pSrc[(y+2)*iSrcStride + x+2]) |
384 | 0 | - ((int)pSrc[(y-2)*iSrcStride + x-1] + (int)pSrc[(y-2)*iSrcStride + x ] + (int)pSrc[(y-2)*iSrcStride + x+1] + (int)pSrc[(y-2)*iSrcStride + x+2] |
385 | 0 | + (int)pSrc[(y+3)*iSrcStride + x-1] + (int)pSrc[(y+3)*iSrcStride + x ] + (int)pSrc[(y+3)*iSrcStride + x+1] + (int)pSrc[(y+3)*iSrcStride + x+2] |
386 | 0 | + (int)pSrc[(y-1)*iSrcStride + x-2] + (int)pSrc[ y *iSrcStride + x-2] + (int)pSrc[(y+1)*iSrcStride + x-2] + (int)pSrc[(y+2)*iSrcStride + x-2] |
387 | 0 | + (int)pSrc[(y-1)*iSrcStride + x+3] + (int)pSrc[ y *iSrcStride + x+3] + (int)pSrc[(y+1)*iSrcStride + x+3] + (int)pSrc[(y+2)*iSrcStride + x+3]); |
388 | 0 | saAct += (uint64_t) abs(f); |
389 | 0 | } |
390 | 0 | } |
391 | 0 | return saAct; |
392 | 0 | } |
393 | | uint64_t AvgHighPassWithDownsamplingDiff1stCore (const int width, const int height, const Pel* pSrc,const Pel* pSrcM1, const int iSrcStride, const int iSrcM1Stride) |
394 | 0 | { |
395 | 0 | uint64_t taAct = 0; |
396 | 0 | pSrc -= iSrcStride; |
397 | 0 | pSrc -= iSrcStride; |
398 | 0 | pSrcM1-=iSrcM1Stride; |
399 | 0 | pSrcM1-=iSrcM1Stride; |
400 | |
|
401 | 0 | for (uint32_t y = 2; y < height-2; y += 2) |
402 | 0 | { |
403 | 0 | for (uint32_t x = 2; x < width-2; x += 2) |
404 | 0 | { |
405 | 0 | const int t = (int)pSrc [y*iSrcStride + x] + (int)pSrc [y*iSrcStride + x+1] + (int)pSrc [(y+1)*iSrcStride + x] + (int)pSrc [(y+1)*iSrcStride + x+1] |
406 | 0 | - ((int)pSrcM1[y*iSrcM1Stride + x] + (int)pSrcM1[y*iSrcM1Stride + x+1] + (int)pSrcM1[(y+1)*iSrcM1Stride + x] + (int)pSrcM1[(y+1)*iSrcM1Stride + x+1]); |
407 | 0 | taAct += (1 + 3 * abs (t)) >> 1; |
408 | 0 | } |
409 | 0 | } |
410 | 0 | return (taAct ); |
411 | 0 | } |
412 | | |
413 | | uint64_t AvgHighPassWithDownsamplingDiff2ndCore (const int width,const int height,const Pel* pSrc,const Pel* pSrcM1,const Pel* pSrcM2,const int iSrcStride,const int iSM1Stride,const int iSM2Stride) |
414 | 0 | { |
415 | 0 | uint64_t taAct = 0; |
416 | |
|
417 | 0 | pSrc -= iSrcStride; |
418 | 0 | pSrc -= iSrcStride; |
419 | 0 | pSrcM1-=iSM1Stride; |
420 | 0 | pSrcM1-=iSM1Stride; |
421 | 0 | pSrcM2-=iSM2Stride; |
422 | 0 | pSrcM2-=iSM2Stride; |
423 | |
|
424 | 0 | for (uint32_t y = 2; y < height-2; y += 2) |
425 | 0 | { |
426 | 0 | for (uint32_t x = 2; x < width-2; x += 2) |
427 | 0 | { |
428 | 0 | const int t = (int)pSrc [y*iSrcStride + x] + (int)pSrc [y*iSrcStride + x+1] + (int)pSrc [(y+1)*iSrcStride + x] + (int)pSrc [(y+1)*iSrcStride + x+1] |
429 | 0 | - 2 * ((int)pSrcM1[y*iSM1Stride + x] + (int)pSrcM1[y*iSM1Stride + x+1] + (int)pSrcM1[(y+1)*iSM1Stride + x] + (int)pSrcM1[(y+1)*iSM1Stride + x+1]) |
430 | 0 | + (int)pSrcM2[y*iSM2Stride + x] + (int)pSrcM2[y*iSM2Stride + x+1] + (int)pSrcM2[(y+1)*iSM2Stride + x] + (int)pSrcM2[(y+1)*iSM2Stride + x+1]; |
431 | 0 | taAct += (uint64_t) abs(t); |
432 | 0 | } |
433 | 0 | } |
434 | 0 | return (taAct); |
435 | 0 | } |
436 | | |
437 | | PelBufferOps::PelBufferOps() |
438 | 14 | { |
439 | 14 | addAvg = addAvgCore<Pel>; |
440 | 14 | reco = recoCore<Pel>; |
441 | 14 | copyClip = copyClipCore<Pel>; |
442 | 14 | roundGeo = roundGeoCore<Pel>; |
443 | | |
444 | 14 | addAvg4 = addAvgCore<Pel>; |
445 | 14 | addAvg8 = addAvgCore<Pel>; |
446 | 14 | addAvg16 = addAvgCore<Pel>; |
447 | | |
448 | 14 | sub4 = subsCore<Pel>; |
449 | 14 | sub8 = subsCore<Pel>; |
450 | | |
451 | 14 | wghtAvg4 = addWeightedAvgCore<Pel>; |
452 | 14 | wghtAvg8 = addWeightedAvgCore<Pel>; |
453 | | |
454 | 14 | copyClip4 = copyClipCore<Pel>; |
455 | 14 | copyClip8 = copyClipCore<Pel>; |
456 | | |
457 | 14 | reco4 = reconstructCore<Pel>; |
458 | 14 | reco8 = reconstructCore<Pel>; |
459 | | |
460 | 14 | linTf4 = linTfCore<Pel>; |
461 | 14 | linTf8 = linTfCore<Pel>; |
462 | | |
463 | 14 | copyBuffer = copyBufferCore; |
464 | | |
465 | 14 | removeHighFreq8 = removeHighFreq; |
466 | 14 | removeHighFreq4 = removeHighFreq; |
467 | | |
468 | 14 | transpose4x4 = transposeNxNCore<Pel,4>; |
469 | 14 | transpose8x8 = transposeNxNCore<Pel,8>; |
470 | 14 | mipMatrixMul_4_4 = mipMatrixMulCore<4,4>; |
471 | 14 | mipMatrixMul_8_4 = mipMatrixMulCore<8,4>; |
472 | 14 | mipMatrixMul_8_8 = mipMatrixMulCore<8,8>; |
473 | 14 | weightCiip = weightCiipCore; |
474 | 14 | roundIntVector = nullptr; |
475 | | |
476 | 14 | fillPtrMap = fillMapPtr_Core; |
477 | 14 | AvgHighPassWithDownsampling = AvgHighPassWithDownsamplingCore; |
478 | 14 | AvgHighPass = AvgHighPassCore; |
479 | 14 | AvgHighPassWithDownsamplingDiff1st = AvgHighPassWithDownsamplingDiff1stCore; |
480 | 14 | AvgHighPassWithDownsamplingDiff2nd = AvgHighPassWithDownsamplingDiff2ndCore; |
481 | 14 | HDHighPass = HDHighPassCore; |
482 | 14 | HDHighPass2 = HDHighPass2Core; |
483 | 14 | } |
484 | | |
485 | | void PelBufferOps::initPelBufOps( bool enableOpt ) |
486 | 0 | { |
487 | 0 | if( isInitSIMDDone ) |
488 | 0 | { |
489 | 0 | return; |
490 | 0 | } |
491 | 0 | isInitSIMDDone = true; |
492 | |
|
493 | 0 | if( enableOpt ) |
494 | 0 | { |
495 | 0 | #if ENABLE_SIMD_OPT_BUFFER |
496 | | # if defined( TARGET_SIMD_X86 ) |
497 | | g_pelBufOP.initPelBufOpsX86(); |
498 | | # endif |
499 | | # if defined( TARGET_SIMD_ARM ) |
500 | | g_pelBufOP.initPelBufOpsARM(); |
501 | | # endif |
502 | 0 | #endif // ENABLE_SIMD_OPT_BUFFER |
503 | 0 | } |
504 | 0 | } |
505 | | |
506 | | PelBufferOps g_pelBufOP = PelBufferOps(); |
507 | | |
508 | | template<> |
509 | | void AreaBuf<Pel>::addWeightedAvg(const AreaBuf<const Pel>& other1, const AreaBuf<const Pel>& other2, const ClpRng& clpRng, const int8_t BcwIdx) |
510 | 0 | { |
511 | 0 | const int8_t w0 = getBcwWeight( BcwIdx, REF_PIC_LIST_0 ); |
512 | 0 | const int8_t w1 = getBcwWeight( BcwIdx, REF_PIC_LIST_1 ); |
513 | 0 | const int8_t log2WeightBase = g_BcwLog2WeightBase; |
514 | 0 | const Pel* src0 = other1.buf; |
515 | 0 | const Pel* src2 = other2.buf; |
516 | 0 | Pel* dest = buf; |
517 | |
|
518 | 0 | const int src1Stride = other1.stride; |
519 | 0 | const int src2Stride = other2.stride; |
520 | 0 | const int destStride = stride; |
521 | 0 | const int clipbd = clpRng.bd; |
522 | 0 | const int shiftNum = std::max<int>( 2, ( IF_INTERNAL_PREC - clipbd ) ) + log2WeightBase; |
523 | 0 | const int offset = ( 1 << ( shiftNum - 1 ) ) + ( IF_INTERNAL_OFFS << log2WeightBase ); |
524 | |
|
525 | 0 | if( ( width & 7 ) == 0 ) |
526 | 0 | { |
527 | 0 | g_pelBufOP.wghtAvg8( src0, src1Stride, src2, src2Stride, dest, destStride, width, height, shiftNum, offset, w0, w1, clpRng ); |
528 | 0 | } |
529 | 0 | else if( ( width & 3 ) == 0 ) |
530 | 0 | { |
531 | 0 | g_pelBufOP.wghtAvg4( src0, src1Stride, src2, src2Stride, dest, destStride, width, height, shiftNum, offset, w0, w1, clpRng ); |
532 | 0 | } |
533 | 0 | else |
534 | 0 | { |
535 | 0 | #define WGHT_AVG_OP( ADDR ) dest[ADDR] = ClipPel( rightShiftU( ( src0[ADDR]*w0 + src2[ADDR]*w1 + offset ), shiftNum ), clpRng ) |
536 | 0 | #define WGHT_AVG_INC \ |
537 | 0 | src0 += src1Stride; \ |
538 | 0 | src2 += src2Stride; \ |
539 | 0 | dest += destStride; \ |
540 | 0 |
|
541 | 0 | SIZE_AWARE_PER_EL_OP( WGHT_AVG_OP, WGHT_AVG_INC ); |
542 | |
|
543 | 0 | #undef WGHT_AVG_OP |
544 | 0 | #undef WGHT_AVG_INC |
545 | 0 | } |
546 | 0 | } |
547 | | |
548 | | template<> |
549 | | void AreaBuf<Pel>::addAvg( const AreaBuf<const Pel>& other1, const AreaBuf<const Pel>& other2, const ClpRng& clpRng) |
550 | 0 | { |
551 | 0 | const Pel* src0 = other1.buf; |
552 | 0 | const Pel* src2 = other2.buf; |
553 | 0 | Pel* dest = buf; |
554 | |
|
555 | 0 | const unsigned src1Stride = other1.stride; |
556 | 0 | const unsigned src2Stride = other2.stride; |
557 | 0 | const unsigned destStride = stride; |
558 | 0 | const int clipbd = clpRng.bd; |
559 | 0 | const unsigned shiftNum = std::max<int>(2, (IF_INTERNAL_PREC - clipbd)) + 1; |
560 | 0 | const int offset = (1 << (shiftNum - 1)) + 2 * IF_INTERNAL_OFFS; |
561 | |
|
562 | 0 | #if ENABLE_SIMD_OPT_BUFFER |
563 | 0 | if( destStride == width ) |
564 | 0 | { |
565 | 0 | g_pelBufOP.addAvg(src0, src2, dest, width * height, shiftNum, offset, clpRng); |
566 | 0 | } |
567 | 0 | else if ((width & 15) == 0) |
568 | 0 | { |
569 | 0 | g_pelBufOP.addAvg16(src0, src1Stride, src2, src2Stride, dest, destStride, width, height, shiftNum, offset, clpRng); |
570 | 0 | } |
571 | 0 | else if( ( width & 7 ) == 0 ) |
572 | 0 | { |
573 | 0 | g_pelBufOP.addAvg8( src0, src1Stride, src2, src2Stride, dest, destStride, width, height, shiftNum, offset, clpRng ); |
574 | 0 | } |
575 | 0 | else if( ( width & 3 ) == 0 ) |
576 | 0 | { |
577 | 0 | g_pelBufOP.addAvg4( src0, src1Stride, src2, src2Stride, dest, destStride, width, height, shiftNum, offset, clpRng ); |
578 | 0 | } |
579 | 0 | else |
580 | 0 | #endif |
581 | 0 | { |
582 | 0 | #define ADD_AVG_OP( ADDR ) dest[ADDR] = ClipPel( rightShiftU( ( src0[ADDR] + src2[ADDR] + offset ), shiftNum ), clpRng ) |
583 | 0 | #define ADD_AVG_INC \ |
584 | 0 | src0 += src1Stride; \ |
585 | 0 | src2 += src2Stride; \ |
586 | 0 | dest += destStride; \ |
587 | 0 |
|
588 | 0 | SIZE_AWARE_PER_EL_OP( ADD_AVG_OP, ADD_AVG_INC ); |
589 | |
|
590 | 0 | #undef ADD_AVG_OP |
591 | 0 | #undef ADD_AVG_INC |
592 | 0 | } |
593 | 0 | } |
594 | | |
595 | | template<> |
596 | | void AreaBuf<Pel>::subtract( const AreaBuf<const Pel>& minuend, const AreaBuf<const Pel>& subtrahend ) |
597 | 796k | { |
598 | 796k | CHECKD( width != minuend.width, "Incompatible size" ); |
599 | 796k | CHECKD( height != minuend.height, "Incompatible size" ); |
600 | 796k | CHECKD( width != subtrahend.width, "Incompatible size"); |
601 | 796k | CHECKD( height != subtrahend.height, "Incompatible size"); |
602 | | |
603 | 796k | Pel* dest = buf; |
604 | 796k | const Pel* mins = minuend .buf; |
605 | 796k | const Pel* subs = subtrahend.buf; |
606 | | |
607 | | |
608 | 796k | #if ENABLE_SIMD_OPT_BUFFER |
609 | 796k | const unsigned destStride = stride; |
610 | 796k | const unsigned minsStride = minuend. stride; |
611 | 796k | const unsigned subsStride = subtrahend.stride; |
612 | | |
613 | 796k | if( ( width & 7 ) == 0 ) |
614 | 672k | { |
615 | 672k | g_pelBufOP.sub8( mins, minsStride, subs, subsStride, dest, destStride, width, height ); |
616 | 672k | } |
617 | 123k | else if( ( width & 3 ) == 0 ) |
618 | 123k | { |
619 | 123k | g_pelBufOP.sub4( mins, minsStride, subs, subsStride, dest, destStride, width, height ); |
620 | 123k | } |
621 | 0 | else |
622 | 0 | #endif |
623 | 0 | { |
624 | 0 | #define SUBS_INC \ |
625 | 0 | dest += stride; \ |
626 | 0 | mins += minuend .stride; \ |
627 | 0 | subs += subtrahend.stride; \ |
628 | 0 |
|
629 | 0 | #define SUBS_OP( ADDR ) dest[ADDR] = mins[ADDR] - subs[ADDR] |
630 | |
|
631 | 0 | SIZE_AWARE_PER_EL_OP( SUBS_OP, SUBS_INC ); |
632 | |
|
633 | 0 | #undef SUBS_OP |
634 | 0 | #undef SUBS_INC |
635 | 0 | } |
636 | 796k | } |
637 | | |
638 | | template<> |
639 | | void AreaBuf<const Pel>::calcVarianceSplit( const AreaBuf<const Pel>& Org, const uint32_t size, int& varh,int& varv) const |
640 | 0 | { |
641 | 0 | CHECK( Org.width != Org.height, "Incompatible size!" ); |
642 | 0 | int stride = Org.stride; |
643 | 0 | const Pel* src; |
644 | 0 | Pel data; |
645 | 0 | double variance=0; |
646 | 0 | double mean=0; |
647 | 0 | int64_t sum[4]={0,0,0,0}; |
648 | 0 | int64_t sum_sqr[4]={0,0,0,0}; |
649 | 0 | uint32_t halfsize =size>>1; |
650 | 0 | uint32_t off[4]={0,halfsize,size*halfsize,size*halfsize+halfsize}; |
651 | 0 | int n,x,y; |
652 | |
|
653 | 0 | for( n = 0; n < 4; n++) |
654 | 0 | { |
655 | 0 | src = Org.buf+off[n]; |
656 | 0 | for( y = 0; y < halfsize; y++) |
657 | 0 | { |
658 | 0 | for(x = 0; x < halfsize; x++) |
659 | 0 | { |
660 | 0 | data=src[y*stride+x]; |
661 | 0 | sum[n]+=data; |
662 | 0 | sum_sqr[n]+= data*data; |
663 | 0 | } |
664 | 0 | } |
665 | 0 | } |
666 | 0 | int num=size*(size>>1); |
667 | | // varhu |
668 | 0 | mean=(double)(sum[0]+sum[1])/(num); |
669 | 0 | variance = (double)(sum_sqr[0]+sum_sqr[1])/(num) - (mean*mean); |
670 | 0 | varh =(int)(variance+0.5); |
671 | | // varhl |
672 | 0 | mean=(double)(sum[2]+sum[3])/(num); |
673 | 0 | variance = (double)(sum_sqr[2]+sum_sqr[3])/(num) - (mean*mean); |
674 | 0 | varh +=(int)(variance+0.5); |
675 | | // varvl |
676 | 0 | mean=(double)(sum[0]+sum[2])/(num); |
677 | 0 | variance = (double)(sum_sqr[0]+sum_sqr[2])/(num) - (mean*mean); |
678 | 0 | varv =(int)(variance+0.5); |
679 | | // varvr |
680 | 0 | mean=(double)(sum[1]+sum[3])/(num); |
681 | 0 | variance = (double)(sum_sqr[1]+sum_sqr[3])/(num) - (mean*mean); |
682 | 0 | varv +=(int)(variance+0.5); |
683 | 0 | } |
684 | | |
685 | | template<> |
686 | | void AreaBuf<Pel>::copyClip( const AreaBuf<const Pel>& src, const ClpRng& clpRng ) |
687 | 0 | { |
688 | 0 | const Pel* srcp = src.buf; |
689 | 0 | Pel* dest = buf; |
690 | |
|
691 | 0 | const unsigned srcStride = src.stride; |
692 | 0 | const unsigned destStride = stride; |
693 | |
|
694 | 0 | if( destStride == width) |
695 | 0 | { |
696 | 0 | g_pelBufOP.copyClip(srcp, dest, width * height, clpRng); |
697 | 0 | } |
698 | 0 | else if ((width & 7) == 0) |
699 | 0 | { |
700 | 0 | g_pelBufOP.copyClip8(srcp, srcStride, dest, destStride, width, height, clpRng); |
701 | 0 | } |
702 | 0 | else if ((width & 3) == 0) |
703 | 0 | { |
704 | 0 | g_pelBufOP.copyClip4(srcp, srcStride, dest, destStride, width, height, clpRng); |
705 | 0 | } |
706 | 0 | else |
707 | 0 | { |
708 | 0 | for( int y = 0; y < height; y++ ) |
709 | 0 | { |
710 | 0 | dest[0] = ClipPel( srcp[0], clpRng); |
711 | 0 | dest[1] = ClipPel( srcp[1], clpRng); |
712 | 0 | srcp += srcStride; |
713 | 0 | dest += destStride; |
714 | 0 | } \ |
715 | 0 | } |
716 | 0 | } |
717 | | |
718 | | template<> |
719 | | void AreaBuf<Pel>::reconstruct( const AreaBuf<const Pel>& pred, const AreaBuf<const Pel>& resi, const ClpRng& clpRng ) |
720 | 2.24M | { |
721 | 2.24M | const Pel* src1 = pred.buf; |
722 | 2.24M | const Pel* src2 = resi.buf; |
723 | 2.24M | Pel* dest = buf; |
724 | | |
725 | 2.24M | const unsigned src1Stride = pred.stride; |
726 | 2.24M | const unsigned src2Stride = resi.stride; |
727 | 2.24M | const unsigned destStride = stride; |
728 | 2.24M | if( src2Stride == width ) |
729 | 2.23M | { |
730 | 2.23M | g_pelBufOP.reco( pred.buf, resi.buf, buf, width * height, clpRng ); |
731 | 2.23M | } |
732 | 8.95k | else if( ( width & 7 ) == 0 ) |
733 | 5.61k | { |
734 | 5.61k | g_pelBufOP.reco8( src1, src1Stride, src2, src2Stride, dest, destStride, width, height, clpRng ); |
735 | 5.61k | } |
736 | 3.34k | else if( ( width & 3 ) == 0 ) |
737 | 3.34k | { |
738 | 3.34k | g_pelBufOP.reco4( src1, src1Stride, src2, src2Stride, dest, destStride, width, height, clpRng ); |
739 | 3.34k | } |
740 | 0 | else if( ( width & 1 ) == 0 ) |
741 | 0 | { |
742 | 0 | for( int y = 0; y < height; y++ ) |
743 | 0 | { |
744 | 0 | dest[0] = ClipPel( src1[0] + src2[0], clpRng); |
745 | 0 | dest[1] = ClipPel( src1[1] + src2[1], clpRng); |
746 | 0 | src1 += src1Stride; |
747 | 0 | src2 += src2Stride; |
748 | 0 | dest += destStride; |
749 | 0 | } |
750 | 0 | } |
751 | 0 | else |
752 | 0 | { |
753 | 0 | CHECKD( width != 1, "Expecting width to be '1'!" ); |
754 | |
|
755 | 0 | for( int y = 0; y < height; y++ ) |
756 | 0 | { |
757 | 0 | dest[0] = ClipPel( src1[0] + src2[0], clpRng ); |
758 | |
|
759 | 0 | src1 += src1Stride; |
760 | 0 | src2 += src2Stride; |
761 | 0 | dest += destStride; |
762 | 0 | } |
763 | 0 | } |
764 | 2.24M | } |
765 | | |
766 | | template<> |
767 | | void AreaBuf<Pel>::linearTransform( const int scale, const unsigned shift, const int offset, bool bClip, const ClpRng& clpRng ) |
768 | 203k | { |
769 | 203k | const Pel* src = buf; |
770 | 203k | Pel* dst = buf; |
771 | | |
772 | 203k | if( stride == width) |
773 | 203k | { |
774 | 203k | if( width > 2 && height > 2 ) |
775 | 191k | { |
776 | 191k | g_pelBufOP.linTf8( src, stride<<2, dst, stride<<2, width<<2, height>>2, scale, shift, offset, clpRng, bClip ); |
777 | 191k | } |
778 | 12.2k | else |
779 | 12.2k | { |
780 | 12.2k | g_pelBufOP.linTf4( src, stride<<1, dst, stride<<1, width<<1, height>>1, scale, shift, offset, clpRng, bClip ); |
781 | 12.2k | } |
782 | 203k | } |
783 | 0 | else if( ( width & 7 ) == 0 ) |
784 | 0 | { |
785 | 0 | g_pelBufOP.linTf8( src, stride, dst, stride, width, height, scale, shift, offset, clpRng, bClip ); |
786 | 0 | } |
787 | 0 | else if( ( width & 3 ) == 0 ) |
788 | 0 | { |
789 | 0 | g_pelBufOP.linTf4( src, stride, dst, stride, width, height, scale, shift, offset, clpRng, bClip ); |
790 | 0 | } |
791 | 0 | else |
792 | 0 | { |
793 | 0 | if( bClip ) |
794 | 0 | { |
795 | 0 | for( int y = 0; y < height; y++ ) |
796 | 0 | { |
797 | 0 | dst[0] = ( Pel ) ClipPel( rightShiftU( scale * src[0], shift ) + offset, clpRng ); |
798 | 0 | dst[1] = ( Pel ) ClipPel( rightShiftU( scale * src[1], shift ) + offset, clpRng ); |
799 | 0 | src += stride; |
800 | 0 | dst += stride; |
801 | 0 | } |
802 | 0 | } |
803 | 0 | else |
804 | 0 | { |
805 | 0 | for( int y = 0; y < height; y++ ) |
806 | 0 | { |
807 | 0 | dst[0] = ( Pel ) ( rightShiftU( scale * src[0], shift ) + offset ); |
808 | 0 | dst[1] = ( Pel ) ( rightShiftU( scale * src[1], shift ) + offset ); |
809 | 0 | src += stride; |
810 | 0 | dst += stride; |
811 | 0 | } |
812 | 0 | } |
813 | 0 | } |
814 | 203k | } |
815 | | |
816 | | #if ENABLE_SIMD_OPT_BUFFER |
817 | | |
818 | | template<> |
819 | | void AreaBuf<Pel>::transposedFrom( const AreaBuf<const Pel>& other ) |
820 | 599k | { |
821 | 599k | CHECK( width != other.height || height != other.width, "Incompatible size" ); |
822 | | |
823 | 599k | if( ( ( width | height ) & 7 ) == 0 ) |
824 | 484k | { |
825 | 484k | const Pel* src = other.buf; |
826 | | |
827 | 1.97M | for( unsigned y = 0; y < other.height; y += 8 ) |
828 | 1.49M | { |
829 | 1.49M | Pel* dst = buf + y; |
830 | | |
831 | 8.34M | for( unsigned x = 0; x < other.width; x += 8 ) |
832 | 6.84M | { |
833 | 6.84M | g_pelBufOP.transpose8x8( &src[x], other.stride, dst, stride ); |
834 | | |
835 | 6.84M | dst += 8 * stride; |
836 | 6.84M | } |
837 | | |
838 | 1.49M | src += 8 * other.stride; |
839 | 1.49M | } |
840 | 484k | } |
841 | 115k | else if( ( ( width | height ) & 3 ) == 0 ) |
842 | 105k | { |
843 | 105k | const Pel* src = other.buf; |
844 | | |
845 | 310k | for( unsigned y = 0; y < other.height; y += 4 ) |
846 | 205k | { |
847 | 205k | Pel* dst = buf + y; |
848 | | |
849 | 601k | for( unsigned x = 0; x < other.width; x += 4 ) |
850 | 396k | { |
851 | 396k | g_pelBufOP.transpose4x4( &src[x], other.stride, dst, stride ); |
852 | | |
853 | 396k | dst += 4 * stride; |
854 | 396k | } |
855 | | |
856 | 205k | src += 4 * other.stride; |
857 | 205k | } |
858 | 105k | } |
859 | 9.93k | else |
860 | 9.93k | { |
861 | 9.93k | Pel* dst = buf; |
862 | 9.93k | const Pel* src = other.buf; |
863 | 9.93k | width = other.height; |
864 | 9.93k | height = other.width; |
865 | 9.93k | stride = stride < width ? width : stride; |
866 | | |
867 | 124k | for( unsigned y = 0; y < other.height; y++ ) |
868 | 114k | { |
869 | 344k | for( unsigned x = 0; x < other.width; x++ ) |
870 | 229k | { |
871 | 229k | dst[y + x*stride] = src[x + y * other.stride]; |
872 | 229k | } |
873 | 114k | } |
874 | 9.93k | } |
875 | 599k | } |
876 | | #endif |
877 | | |
878 | | template<> |
879 | | void AreaBuf<Pel>::weightCiip( const AreaBuf<const Pel>& intra, const int numIntra ) |
880 | 0 | { |
881 | 0 | CHECK(width == 2, "Width of 2 is not supported"); |
882 | 0 | g_pelBufOP.weightCiip( buf, intra.buf, width * height, numIntra ); |
883 | 0 | } |
884 | | |
885 | | template<> |
886 | | void AreaBuf<MotionInfo>::fill( const MotionInfo& val ) |
887 | 22.7k | { |
888 | 22.7k | if( width == stride ) |
889 | 22.7k | { |
890 | 22.7k | std::fill_n( buf, width * height, val ); |
891 | 22.7k | } |
892 | 0 | else |
893 | 0 | { |
894 | 0 | MotionInfo* dst = buf; |
895 | |
|
896 | 0 | for( int y = 0; y < height; y++, dst += stride ) |
897 | 0 | { |
898 | 0 | std::fill_n( dst, width, val ); |
899 | 0 | } |
900 | 0 | } |
901 | 22.7k | } |
902 | | |
903 | | PelStorage::PelStorage() |
904 | 4.08M | { |
905 | 16.3M | for( uint32_t i = 0; i < MAX_NUM_COMP; i++ ) |
906 | 12.2M | { |
907 | 12.2M | m_origin[i] = nullptr; |
908 | 12.2M | } |
909 | 4.08M | } |
910 | | |
911 | | PelStorage::~PelStorage() |
912 | 4.08M | { |
913 | 4.08M | destroy(); |
914 | 4.08M | } |
915 | | |
916 | | void PelStorage::create( const UnitArea& _UnitArea ) |
917 | 1.80M | { |
918 | 1.80M | create( _UnitArea.chromaFormat, _UnitArea.blocks[0] ); |
919 | 1.80M | m_maxArea = _UnitArea; |
920 | 1.80M | } |
921 | | |
922 | | void PelStorage::create( const ChromaFormat &_chromaFormat, const Area& _area ) |
923 | 3.59M | { |
924 | 3.59M | CHECK( !bufs.empty(), "Trying to re-create an already initialized buffer" ); |
925 | | |
926 | 3.59M | chromaFormat = _chromaFormat; |
927 | | |
928 | 3.59M | const uint32_t numComp = getNumberValidComponents( _chromaFormat ); |
929 | | |
930 | 3.59M | uint32_t bufSize = 0; |
931 | 12.8M | for( uint32_t i = 0; i < numComp; i++ ) |
932 | 9.23M | { |
933 | 9.23M | const ComponentID compID = ComponentID( i ); |
934 | 9.23M | const unsigned totalWidth = _area.width >> getComponentScaleX( compID, _chromaFormat ); |
935 | 9.23M | const unsigned totalHeight = _area.height >> getComponentScaleY( compID, _chromaFormat ); |
936 | | |
937 | 9.23M | const uint32_t area = totalWidth * totalHeight; |
938 | 9.23M | CHECK( !area, "Trying to create a buffer with zero area" ); |
939 | 9.23M | bufSize += area; |
940 | 9.23M | } |
941 | | |
942 | 3.59M | bufSize += 1; // for SIMD DMVR on the bottom right corner, which overreads the lines by 1 sample |
943 | | |
944 | | //allocate one buffer |
945 | 3.59M | m_origin[0] = ( Pel* ) xMalloc( Pel, bufSize ); |
946 | | |
947 | 3.59M | Pel* topLeft = m_origin[0]; |
948 | 12.8M | for( uint32_t i = 0; i < numComp; i++ ) |
949 | 9.23M | { |
950 | 9.23M | const ComponentID compID = ComponentID( i ); |
951 | 9.23M | const unsigned totalWidth = _area.width >> getComponentScaleX( compID, _chromaFormat ); |
952 | 9.23M | const unsigned totalHeight = _area.height >> getComponentScaleY( compID, _chromaFormat ); |
953 | 9.23M | const uint32_t area = totalWidth * totalHeight; |
954 | | |
955 | 9.23M | bufs.push_back( PelBuf( topLeft, totalWidth, totalWidth, totalHeight ) ); |
956 | 9.23M | topLeft += area; |
957 | 9.23M | } |
958 | | |
959 | 3.59M | m_maxArea = UnitArea( _chromaFormat, _area ); |
960 | 3.59M | } |
961 | | |
962 | | void PelStorage::create( const ChromaFormat &_chromaFormat, const Area& _area, const unsigned _maxCUSize, const unsigned _margin, const unsigned _alignment, const bool _scaleChromaMargin ) |
963 | 166k | { |
964 | 166k | CHECK( !bufs.empty(), "Trying to re-create an already initialized buffer" ); |
965 | | |
966 | 166k | chromaFormat = _chromaFormat; |
967 | | |
968 | 166k | const uint32_t numComp = getNumberValidComponents( _chromaFormat ); |
969 | | |
970 | 166k | unsigned extHeight = _area.height; |
971 | 166k | unsigned extWidth = _area.width; |
972 | | |
973 | 166k | if( _maxCUSize ) |
974 | 31.2k | { |
975 | 31.2k | extHeight = ( ( _area.height + _maxCUSize - 1 ) / _maxCUSize ) * _maxCUSize; |
976 | 31.2k | extWidth = ( ( _area.width + _maxCUSize - 1 ) / _maxCUSize ) * _maxCUSize; |
977 | 31.2k | } |
978 | | |
979 | 590k | for( uint32_t i = 0; i < numComp; i++ ) |
980 | 423k | { |
981 | 423k | const ComponentID compID = ComponentID( i ); |
982 | 423k | const unsigned scaleX = getComponentScaleX( compID, _chromaFormat ); |
983 | 423k | const unsigned scaleY = getComponentScaleY( compID, _chromaFormat ); |
984 | | |
985 | 423k | unsigned scaledHeight = extHeight >> scaleY; |
986 | 423k | unsigned scaledWidth = extWidth >> scaleX; |
987 | 423k | unsigned ymargin = _margin >> (_scaleChromaMargin?scaleY:0); |
988 | 423k | unsigned xmargin = _margin >> (_scaleChromaMargin?scaleX:0); |
989 | 423k | unsigned totalWidth = scaledWidth + 2*xmargin; |
990 | 423k | unsigned totalHeight = scaledHeight +2*ymargin; |
991 | | |
992 | 423k | if( _alignment ) |
993 | 237k | { |
994 | | // make sure buffer lines are align |
995 | 237k | CHECK( _alignment != MEMORY_ALIGN_DEF_SIZE, "Unsupported alignment" ); |
996 | 237k | totalWidth = ( ( totalWidth + _alignment - 1 ) / _alignment ) * _alignment; |
997 | 237k | } |
998 | 423k | uint32_t area = totalWidth * totalHeight; |
999 | 423k | CHECK( !area, "Trying to create a buffer with zero area" ); |
1000 | | |
1001 | 423k | m_origin[i] = ( Pel* ) xMalloc( Pel, area ); |
1002 | 423k | Pel* topLeft = m_origin[i] + totalWidth * ymargin + xmargin; |
1003 | 423k | bufs.push_back( PelBuf( topLeft, totalWidth, _area.width >> scaleX, _area.height >> scaleY ) ); |
1004 | 423k | } |
1005 | | |
1006 | 166k | m_maxArea = UnitArea( _chromaFormat, _area ); |
1007 | 166k | } |
1008 | | |
1009 | | void PelStorage::createFromBuf( PelUnitBuf buf ) |
1010 | 2.40k | { |
1011 | 2.40k | chromaFormat = buf.chromaFormat; |
1012 | | |
1013 | 2.40k | const uint32_t numCh = getNumberValidComponents( chromaFormat ); |
1014 | | |
1015 | 2.40k | bufs.resize(numCh); |
1016 | | |
1017 | 9.60k | for( uint32_t i = 0; i < numCh; i++ ) |
1018 | 7.20k | { |
1019 | 7.20k | PelBuf cPelBuf = buf.get( ComponentID( i ) ); |
1020 | 7.20k | bufs[i] = PelBuf( cPelBuf.bufAt( 0, 0 ), cPelBuf.stride, cPelBuf.width, cPelBuf.height ); |
1021 | 7.20k | } |
1022 | 2.40k | } |
1023 | | |
1024 | | void PelStorage::compactResize( const UnitArea& area ) |
1025 | 2.24M | { |
1026 | 2.24M | CHECK( bufs.size() < area.blocks.size(), "Cannot increase buffer size when compacting!" ); |
1027 | | |
1028 | 7.58M | for( uint32_t i = 0; i < area.blocks.size(); i++ ) |
1029 | 5.34M | { |
1030 | 5.34M | CHECK( m_maxArea.blocks[i].area() < area.blocks[i].area(), "Cannot increase buffer size when compacting!" ); |
1031 | | |
1032 | 5.34M | bufs[i].Size::operator=( area.blocks[i].size() ); |
1033 | 5.34M | bufs[i].stride = bufs[i].width; |
1034 | 5.34M | } |
1035 | 2.24M | } |
1036 | | |
1037 | | void PelStorage::takeOwnership( PelStorage& other ) |
1038 | 0 | { |
1039 | 0 | chromaFormat = other.chromaFormat; |
1040 | |
|
1041 | 0 | const uint32_t numCh = getNumberValidComponents( chromaFormat ); |
1042 | |
|
1043 | 0 | bufs.resize(numCh); |
1044 | |
|
1045 | 0 | for( uint32_t i = 0; i < numCh; i++ ) |
1046 | 0 | { |
1047 | 0 | PelBuf cPelBuf = other.get( ComponentID( i ) ); |
1048 | 0 | bufs[i] = PelBuf( cPelBuf.bufAt( 0, 0 ), cPelBuf.stride, cPelBuf.width, cPelBuf.height ); |
1049 | 0 | std::swap( m_origin[i], other.m_origin[i]); |
1050 | 0 | } |
1051 | |
|
1052 | 0 | m_maxArea = other.m_maxArea; |
1053 | |
|
1054 | 0 | other.destroy(); |
1055 | 0 | } |
1056 | | |
1057 | | |
1058 | | void PelStorage::swap( PelStorage& other ) |
1059 | 0 | { |
1060 | 0 | const uint32_t numCh = getNumberValidComponents( chromaFormat ); |
1061 | |
|
1062 | 0 | for( uint32_t i = 0; i < numCh; i++ ) |
1063 | 0 | { |
1064 | | // check this otherwise it would turn out to get very weird |
1065 | 0 | CHECK( chromaFormat != other.chromaFormat , "Incompatible formats" ); |
1066 | 0 | CHECK( get( ComponentID( i ) ) != other.get( ComponentID( i ) ) , "Incompatible formats" ); |
1067 | 0 | CHECK( get( ComponentID( i ) ).stride != other.get( ComponentID( i ) ).stride, "Incompatible formats" ); |
1068 | |
|
1069 | 0 | std::swap( bufs[i].buf, other.bufs[i].buf ); |
1070 | 0 | std::swap( bufs[i].stride, other.bufs[i].stride ); |
1071 | 0 | std::swap( m_origin[i], other.m_origin[i] ); |
1072 | 0 | } |
1073 | 0 | } |
1074 | | |
1075 | | void PelStorage::destroy() |
1076 | 7.82M | { |
1077 | 7.82M | chromaFormat = NUM_CHROMA_FORMAT; |
1078 | 31.3M | for( uint32_t i = 0; i < MAX_NUM_COMP; i++ ) |
1079 | 23.4M | { |
1080 | 23.4M | if( m_origin[i] ) |
1081 | 4.01M | { |
1082 | 4.01M | xFree( m_origin[i] ); |
1083 | 4.01M | m_origin[i] = nullptr; |
1084 | 4.01M | } |
1085 | 23.4M | } |
1086 | 7.82M | bufs.clear(); |
1087 | 7.82M | } |
1088 | | |
1089 | | PelBuf PelStorage::getBuf( const ComponentID CompID ) |
1090 | 16.0k | { |
1091 | 16.0k | return bufs[CompID]; |
1092 | 16.0k | } |
1093 | | |
1094 | | const CPelBuf PelStorage::getBuf( const ComponentID CompID ) const |
1095 | 0 | { |
1096 | 0 | return bufs[CompID]; |
1097 | 0 | } |
1098 | | |
1099 | | PelBuf PelStorage::getBuf( const CompArea& blk ) |
1100 | 20.9M | { |
1101 | 20.9M | const PelBuf& r = bufs[blk.compID]; |
1102 | 20.9M | return PelBuf( r.buf + rsAddr( blk, r.stride ), r.stride, blk ); |
1103 | 20.9M | } |
1104 | | |
1105 | | const CPelBuf PelStorage::getBuf( const CompArea& blk ) const |
1106 | 28.3k | { |
1107 | 28.3k | const PelBuf& r = bufs[blk.compID]; |
1108 | 28.3k | return CPelBuf( r.buf + rsAddr( blk, r.stride ), r.stride, blk ); |
1109 | 28.3k | } |
1110 | | |
1111 | | PelUnitBuf PelStorage::getBuf( const UnitArea& unit ) |
1112 | 3.84k | { |
1113 | 3.84k | return ( chromaFormat == CHROMA_400 ) ? PelUnitBuf( chromaFormat, getBuf( unit.Y() ) ) : PelUnitBuf( chromaFormat, getBuf( unit.Y() ), getBuf( unit.Cb() ), getBuf( unit.Cr() ) ); |
1114 | 3.84k | } |
1115 | | |
1116 | | const CPelUnitBuf PelStorage::getBuf( const UnitArea& unit ) const |
1117 | 0 | { |
1118 | 0 | return ( chromaFormat == CHROMA_400 ) ? CPelUnitBuf( chromaFormat, getBuf( unit.Y() ) ) : CPelUnitBuf( chromaFormat, getBuf( unit.Y() ), getBuf( unit.Cb() ), getBuf( unit.Cr() ) ); |
1119 | 0 | } |
1120 | | |
1121 | | PelUnitBuf PelStorage::getBuf(const int strY, const int strCb, const int strCr, const UnitArea& unit) |
1122 | 0 | { |
1123 | 0 | CHECKD( unit.Y().width > bufs[COMP_Y].width && unit.Y().height > bufs[COMP_Y].height, "unsuported request" ); |
1124 | 0 | CHECKD( strY > bufs[COMP_Y].stride, "unsuported request" ); |
1125 | 0 | CHECKD( strCb > bufs[COMP_Cb].stride, "unsuported request" ); |
1126 | 0 | CHECKD( strCr > bufs[COMP_Cr].stride, "unsuported request" ); |
1127 | 0 | return (chromaFormat == CHROMA_400) ? PelUnitBuf(chromaFormat, PelBuf( bufs[COMP_Y].buf, strY, unit.Y())) : PelUnitBuf(chromaFormat, PelBuf( bufs[COMP_Y].buf, strY, unit.Y()), PelBuf( bufs[COMP_Cb].buf, strCb, unit.Cb()), PelBuf( bufs[COMP_Cr].buf, strCr, unit.Cr())); |
1128 | 0 | } |
1129 | | |
1130 | | const CPelUnitBuf PelStorage::getBuf(const int strY, const int strCb, const int strCr, const UnitArea& unit) const |
1131 | 0 | { |
1132 | 0 | CHECKD( unit.Y().width > bufs[COMP_Y].width && unit.Y().height > bufs[COMP_Y].height, "unsuported request" ); |
1133 | 0 | CHECKD( strY > bufs[COMP_Y].stride, "unsuported request" ); |
1134 | 0 | CHECKD( strCb > bufs[COMP_Cb].stride, "unsuported request" ); |
1135 | 0 | CHECKD( strCr > bufs[COMP_Cr].stride, "unsuported request" ); |
1136 | 0 | return (chromaFormat == CHROMA_400) ? CPelUnitBuf(chromaFormat, CPelBuf( bufs[COMP_Y].buf, strY, unit.Y())) : CPelUnitBuf(chromaFormat, CPelBuf( bufs[COMP_Y].buf, strY, unit.Y()), CPelBuf( bufs[COMP_Cb].buf, strCb, unit.Cb()), CPelBuf( bufs[COMP_Cr].buf, strCr, unit.Cr())); |
1137 | 0 | } |
1138 | | |
1139 | | PelUnitBuf PelStorage::getBufPart(const UnitArea& unit) |
1140 | 0 | { |
1141 | 0 | CHECKD( unit.Y().width > bufs[COMP_Y].width && unit.Y().height > bufs[COMP_Y].height, "unsuported request" ); |
1142 | 0 | return (chromaFormat == CHROMA_400) ? PelUnitBuf(chromaFormat, PelBuf( bufs[COMP_Y].buf, bufs[COMP_Y].stride, unit.Y())) : PelUnitBuf(chromaFormat, PelBuf( bufs[COMP_Y].buf, bufs[COMP_Y].stride, unit.Y()), PelBuf( bufs[COMP_Cb].buf, bufs[COMP_Cb].stride, unit.Cb()), PelBuf( bufs[COMP_Cr].buf, bufs[COMP_Cr].stride, unit.Cr())); |
1143 | 0 | } |
1144 | | |
1145 | | const CPelUnitBuf PelStorage::getBufPart(const UnitArea& unit) const |
1146 | 0 | { |
1147 | 0 | CHECKD(unit.Y().width > bufs[COMP_Y].width && unit.Y().height > bufs[COMP_Y].height, "unsuported request"); |
1148 | 0 | return (chromaFormat == CHROMA_400) ? CPelUnitBuf(chromaFormat, CPelBuf(bufs[COMP_Y].buf, unit.Y().width, unit.Y())) : CPelUnitBuf(chromaFormat, CPelBuf(bufs[COMP_Y].buf, unit.Y().width, unit.Y()), CPelBuf(bufs[COMP_Cb].buf, unit.Cb().width, unit.Cb()), CPelBuf(bufs[COMP_Cr].buf, unit.Cr().width, unit.Cr())); |
1149 | 0 | } |
1150 | | |
1151 | | const CPelUnitBuf PelStorage::getCompactBuf(const UnitArea& unit) const |
1152 | 0 | { |
1153 | 0 | CHECKD( unit.Y().width > bufs[COMP_Y].width && unit.Y().height > bufs[COMP_Y].height, "unsuported request" ); |
1154 | |
|
1155 | 0 | PelUnitBuf ret; |
1156 | 0 | ret.chromaFormat = chromaFormat; |
1157 | 0 | ret.bufs.resize_noinit( chromaFormat == CHROMA_400 ? 1 : 3 ); |
1158 | | |
1159 | 0 | ret.Y ().buf = bufs[COMP_Y ].buf; ret.Y ().width = ret.Y ().stride = unit.Y ().width; ret.Y ().height = unit.Y ().height; |
1160 | 0 | if( chromaFormat != CHROMA_400 ) |
1161 | 0 | { |
1162 | 0 | ret.Cb().buf = bufs[COMP_Cb].buf; ret.Cb().width = ret.Cb().stride = unit.Cb().width; ret.Cb().height = unit.Cb().height; |
1163 | 0 | ret.Cr().buf = bufs[COMP_Cr].buf; ret.Cr().width = ret.Cr().stride = unit.Cr().width; ret.Cr().height = unit.Cr().height; |
1164 | 0 | } |
1165 | |
|
1166 | 0 | return ret; |
1167 | 0 | } |
1168 | | |
1169 | | PelUnitBuf PelStorage::getCompactBuf(const UnitArea& unit) |
1170 | 143k | { |
1171 | 143k | CHECKD( unit.Y().width > bufs[COMP_Y].width && unit.Y().height > bufs[COMP_Y].height, "unsuported request" ); |
1172 | | |
1173 | 143k | PelUnitBuf ret; |
1174 | 143k | ret.chromaFormat = chromaFormat; |
1175 | 143k | ret.bufs.resize_noinit( chromaFormat == CHROMA_400 ? 1 : 3 ); |
1176 | | |
1177 | 143k | ret.Y ().buf = bufs[COMP_Y ].buf; ret.Y ().width = ret.Y ().stride = unit.Y ().width; ret.Y ().height = unit.Y ().height; |
1178 | 143k | if( chromaFormat != CHROMA_400 ) |
1179 | 143k | { |
1180 | 143k | ret.Cb().buf = bufs[COMP_Cb].buf; ret.Cb().width = ret.Cb().stride = unit.Cb().width; ret.Cb().height = unit.Cb().height; |
1181 | 143k | ret.Cr().buf = bufs[COMP_Cr].buf; ret.Cr().width = ret.Cr().stride = unit.Cr().width; ret.Cr().height = unit.Cr().height; |
1182 | 143k | } |
1183 | | |
1184 | 143k | return ret; |
1185 | 143k | } |
1186 | | |
1187 | | const CPelBuf PelStorage::getCompactBuf(const CompArea& carea) const |
1188 | 0 | { |
1189 | 0 | return CPelBuf( bufs[carea.compID].buf, carea.width, carea); |
1190 | 0 | } |
1191 | | |
1192 | | PelBuf PelStorage::getCompactBuf(const CompArea& carea) |
1193 | 0 | { |
1194 | 0 | return PelBuf( bufs[carea.compID].buf, carea.width, carea); |
1195 | 0 | } |
1196 | | |
1197 | | void downsampleYuv(PelBuf& dest, const vvencYUVPlane& yuvPlaneIn, int downsampleStep) |
1198 | 0 | { |
1199 | 0 | const int widthd = dest.width; |
1200 | 0 | const int heightd = dest.height; |
1201 | 0 | int difStride = dest.stride - dest.width; |
1202 | |
|
1203 | 0 | const int16_t* src = yuvPlaneIn.ptr; |
1204 | 0 | const int instride = yuvPlaneIn.stride; |
1205 | 0 | const int width = yuvPlaneIn.width; |
1206 | 0 | int n = 0; |
1207 | 0 | for (int j = 0; j < heightd; j++) |
1208 | 0 | { |
1209 | 0 | int i = 0; |
1210 | 0 | for (i = 0; i < widthd; i++) |
1211 | 0 | { |
1212 | 0 | long int b = 0; |
1213 | 0 | for (int r = 0; r < downsampleStep; r++) |
1214 | 0 | { |
1215 | 0 | int posr = width * r; |
1216 | 0 | for (int n = 0; n < downsampleStep; n++) |
1217 | 0 | { |
1218 | 0 | b += src[posr + n]; |
1219 | 0 | } |
1220 | 0 | } |
1221 | 0 | src += downsampleStep; |
1222 | 0 | dest.buf[n] = (int16_t)((b + 2) / (downsampleStep << 1)); |
1223 | 0 | n++; |
1224 | 0 | } |
1225 | 0 | n += difStride; |
1226 | 0 | src = src - downsampleStep * i + width; |
1227 | |
|
1228 | 0 | src += (instride * (downsampleStep - 1)); |
1229 | 0 | } |
1230 | 0 | } |
1231 | | |
1232 | | void copyPadToPelUnitBuf( PelUnitBuf pelUnitBuf, const vvencYUVBuffer& yuvBuffer, const ChromaFormat& chFmt ) |
1233 | 1.20k | { |
1234 | 1.20k | CHECK( pelUnitBuf.bufs.size() == 0, "pelUnitBuf not initialized" ); |
1235 | 1.20k | pelUnitBuf.chromaFormat = chFmt; |
1236 | 1.20k | const int numComp = getNumberValidComponents( chFmt ); |
1237 | 4.80k | for ( int i = 0; i < numComp; i++ ) |
1238 | 3.60k | { |
1239 | 3.60k | const vvencYUVPlane& src = yuvBuffer.planes[ i ]; |
1240 | 3.60k | CHECK( src.ptr == nullptr, "yuvBuffer not setup" ); |
1241 | 3.60k | PelBuf& dest = pelUnitBuf.bufs[i]; |
1242 | 3.60k | CHECK( dest.buf == nullptr, "yuvBuffer not setup" ); |
1243 | | |
1244 | 3.60k | if (dest.width < src.width) |
1245 | 0 | { |
1246 | 0 | downsampleYuv(dest, src, 2); |
1247 | 0 | } |
1248 | 3.60k | else |
1249 | 3.60k | { |
1250 | 373k | for (int y = 0; y < src.height; y++) |
1251 | 369k | { |
1252 | 369k | ::memcpy(dest.buf + y * dest.stride, src.ptr + y * src.stride, src.width * sizeof(int16_t)); |
1253 | | |
1254 | | // pad right if required |
1255 | 369k | for (int x = src.width; x < dest.width; x++) |
1256 | 0 | { |
1257 | 0 | dest.buf[x + y * dest.stride] = dest.buf[src.width - 1 + y * dest.stride]; |
1258 | 0 | } |
1259 | 369k | } |
1260 | | |
1261 | | // pad bottom if required |
1262 | 3.60k | for (int y = src.height; y < dest.height; y++) |
1263 | 0 | { |
1264 | 0 | ::memcpy(dest.buf + y * dest.stride, dest.buf + (src.height - 1) * dest.stride, dest.width * sizeof(int16_t)); |
1265 | 0 | } |
1266 | 3.60k | } |
1267 | 3.60k | } |
1268 | 1.20k | } |
1269 | | |
1270 | | /* |
1271 | | void setupPelUnitBuf( const YUVBuffer& yuvBuffer, PelUnitBuf& pelUnitBuf, const ChromaFormat& chFmt ) |
1272 | | { |
1273 | | CHECK( pelUnitBuf.bufs.size() != 0, "pelUnitBuf already in use" ); |
1274 | | pelUnitBuf.chromaFormat = chFmt; |
1275 | | const int numComp = getNumberValidComponents( chFmt ); |
1276 | | for ( int i = 0; i < numComp; i++ ) |
1277 | | { |
1278 | | const YUVBuffer::Plane& yuvPlane = yuvBuffer.planes[ i ]; |
1279 | | CHECK( yuvPlane.ptr == nullptr, "yuvBuffer not setup" ); |
1280 | | PelBuf area( yuvPlane.ptr, yuvPlane.stride, yuvPlane.width, yuvPlane.height ); |
1281 | | pelUnitBuf.bufs.push_back( area ); |
1282 | | } |
1283 | | } |
1284 | | */ |
1285 | | void setupYuvBuffer ( const PelUnitBuf& pelUnitBuf, vvencYUVBuffer& yuvBuffer, const Window* confWindow ) |
1286 | 0 | { |
1287 | 0 | const ChromaFormat chFmt = pelUnitBuf.chromaFormat; |
1288 | 0 | const int numComp = getNumberValidComponents( chFmt ); |
1289 | 0 | for ( int i = 0; i < numComp; i++ ) |
1290 | 0 | { |
1291 | 0 | const ComponentID compId = ComponentID( i ); |
1292 | 0 | PelBuf area = pelUnitBuf.get( compId ); |
1293 | 0 | const int sx = getComponentScaleX( compId, chFmt ); |
1294 | 0 | const int sy = getComponentScaleY( compId, chFmt ); |
1295 | 0 | vvencYUVPlane& yuvPlane = yuvBuffer.planes[ i ]; |
1296 | 0 | CHECK( yuvPlane.ptr != nullptr, "yuvBuffer already in use" ); |
1297 | 0 | yuvPlane.ptr = area.bufAt( confWindow->winLeftOffset >> sx, confWindow->winTopOffset >> sy ); |
1298 | 0 | yuvPlane.width = ( ( area.width << sx ) - ( confWindow->winLeftOffset + confWindow->winRightOffset ) ) >> sx; |
1299 | 0 | yuvPlane.height = ( ( area.height << sy ) - ( confWindow->winTopOffset + confWindow->winBottomOffset ) ) >> sy; |
1300 | 0 | yuvPlane.stride = area.stride; |
1301 | 0 | } |
1302 | 0 | } |
1303 | | |
1304 | | } // namespace vvenc |
1305 | | |
1306 | | //! \} |
1307 | | |