1 | // Copyright 2011 Google Inc. All Rights Reserved. |
2 | // |
3 | // Use of this source code is governed by a BSD-style license |
4 | // that can be found in the COPYING file in the root of the source |
5 | // tree. An additional intellectual property rights grant can be found |
6 | // in the file PATENTS. All contributing project authors may |
7 | // be found in the AUTHORS file in the root of the source tree. |
8 | // ----------------------------------------------------------------------------- |
9 | // |
10 | // Speed-critical functions. |
11 | // |
12 | // Author: Skal (pascal.massimino@gmail.com) |
13 | |
14 | #ifndef WEBP_DSP_DSP_H_ |
15 | #define WEBP_DSP_DSP_H_ |
16 | |
17 | #ifdef HAVE_CONFIG_H |
18 | #include "../webp/config.h" |
19 | #endif |
20 | |
21 | #include "../webp/types.h" |
22 | |
23 | #ifdef __cplusplus |
24 | extern "C" { |
25 | #endif |
26 | |
27 | #define BPS 32 // this is the common stride for enc/dec |
28 | |
29 | //------------------------------------------------------------------------------ |
30 | // CPU detection |
31 | |
32 | #if defined(__GNUC__) |
33 | # define LOCAL_GCC_VERSION ((__GNUC__ << 8) | __GNUC_MINOR__) |
34 | # define LOCAL_GCC_PREREQ(maj, min) \ |
35 | (LOCAL_GCC_VERSION >= (((maj) << 8) | (min))) |
36 | #else |
37 | # define LOCAL_GCC_VERSION 0 |
38 | # define LOCAL_GCC_PREREQ(maj, min) 0 |
39 | #endif |
40 | |
41 | #ifndef __has_builtin |
42 | # define __has_builtin(x) 0 |
43 | #endif |
44 | |
45 | #if defined(_MSC_VER) && _MSC_VER > 1310 && \ |
46 | (defined(_M_X64) || defined(_M_IX86)) |
47 | #define WEBP_MSC_SSE2 // Visual C++ SSE2 targets |
48 | #endif |
49 | |
50 | #if defined(_MSC_VER) && _MSC_VER >= 1500 && \ |
51 | (defined(_M_X64) || defined(_M_IX86)) |
52 | #define WEBP_MSC_SSE41 // Visual C++ SSE4.1 targets |
53 | #endif |
54 | |
55 | // WEBP_HAVE_* are used to indicate the presence of the instruction set in dsp |
56 | // files without intrinsics, allowing the corresponding Init() to be called. |
57 | // Files containing intrinsics will need to be built targeting the instruction |
58 | // set so should succeed on one of the earlier tests. |
59 | #if defined(__SSE2__) || defined(WEBP_MSC_SSE2) || defined(WEBP_HAVE_SSE2) |
60 | #define WEBP_USE_SSE2 |
61 | #endif |
62 | |
63 | #if defined(__SSE4_1__) || defined(WEBP_MSC_SSE41) || defined(WEBP_HAVE_SSE41) |
64 | #define WEBP_USE_SSE41 |
65 | #endif |
66 | |
67 | #if defined(__AVX2__) || defined(WEBP_HAVE_AVX2) |
68 | #define WEBP_USE_AVX2 |
69 | #endif |
70 | |
71 | #if defined(__ANDROID__) && defined(__ARM_ARCH_7A__) |
72 | #define WEBP_ANDROID_NEON // Android targets that might support NEON |
73 | #endif |
74 | |
75 | // The intrinsics currently cause compiler errors with arm-nacl-gcc and the |
76 | // inline assembly would need to be modified for use with Native Client. |
77 | #if (defined(__ARM_NEON__) || defined(WEBP_ANDROID_NEON) || \ |
78 | defined(__aarch64__) || defined(WEBP_HAVE_NEON)) && \ |
79 | !defined(__native_client__) |
80 | #define WEBP_USE_NEON |
81 | #endif |
82 | |
83 | #if defined(_MSC_VER) && _MSC_VER >= 1700 && defined(_M_ARM) |
84 | #define WEBP_USE_NEON |
85 | #define WEBP_USE_INTRINSICS |
86 | #endif |
87 | |
88 | #if defined(__mips__) && !defined(__mips64) && \ |
89 | defined(__mips_isa_rev) && (__mips_isa_rev >= 1) && (__mips_isa_rev < 6) |
90 | #define WEBP_USE_MIPS32 |
91 | #if (__mips_isa_rev >= 2) |
92 | #define WEBP_USE_MIPS32_R2 |
93 | #if defined(__mips_dspr2) || (__mips_dsp_rev >= 2) |
94 | #define WEBP_USE_MIPS_DSP_R2 |
95 | #endif |
96 | #endif |
97 | #endif |
98 | |
99 | #if defined(__mips_msa) && defined(__mips_isa_rev) && (__mips_isa_rev >= 5) |
100 | #define WEBP_USE_MSA |
101 | #endif |
102 | |
103 | // This macro prevents thread_sanitizer from reporting known concurrent writes. |
104 | #define WEBP_TSAN_IGNORE_FUNCTION |
105 | #if defined(__has_feature) |
106 | #if __has_feature(thread_sanitizer) |
107 | #undef WEBP_TSAN_IGNORE_FUNCTION |
108 | #define WEBP_TSAN_IGNORE_FUNCTION __attribute__((no_sanitize_thread)) |
109 | #endif |
110 | #endif |
111 | |
112 | #define WEBP_UBSAN_IGNORE_UNDEF |
113 | #define WEBP_UBSAN_IGNORE_UNSIGNED_OVERFLOW |
114 | #if defined(__clang__) && defined(__has_attribute) |
115 | #if __has_attribute(no_sanitize) |
116 | // This macro prevents the undefined behavior sanitizer from reporting |
117 | // failures. This is only meant to silence unaligned loads on platforms that |
118 | // are known to support them. |
119 | #undef WEBP_UBSAN_IGNORE_UNDEF |
120 | #define WEBP_UBSAN_IGNORE_UNDEF \ |
121 | __attribute__((no_sanitize("undefined"))) |
122 | |
123 | // This macro prevents the undefined behavior sanitizer from reporting |
124 | // failures related to unsigned integer overflows. This is only meant to |
125 | // silence cases where this well defined behavior is expected. |
126 | #undef WEBP_UBSAN_IGNORE_UNSIGNED_OVERFLOW |
127 | #define WEBP_UBSAN_IGNORE_UNSIGNED_OVERFLOW \ |
128 | __attribute__((no_sanitize("unsigned-integer-overflow"))) |
129 | #endif |
130 | #endif |
131 | |
132 | typedef enum { |
133 | kSSE2, |
134 | kSSE3, |
135 | kSlowSSSE3, // special feature for slow SSSE3 architectures |
136 | kSSE4_1, |
137 | kAVX, |
138 | kAVX2, |
139 | kNEON, |
140 | kMIPS32, |
141 | kMIPSdspR2, |
142 | kMSA |
143 | } CPUFeature; |
144 | // returns true if the CPU supports the feature. |
145 | typedef int (*VP8CPUInfo)(CPUFeature feature); |
146 | WEBP_EXTERN(VP8CPUInfo) VP8GetCPUInfo; |
147 | |
148 | //------------------------------------------------------------------------------ |
149 | // Init stub generator |
150 | |
151 | // Defines an init function stub to ensure each module exposes a symbol, |
152 | // avoiding a compiler warning. |
153 | #define WEBP_DSP_INIT_STUB(func) \ |
154 | extern void func(void); \ |
155 | WEBP_TSAN_IGNORE_FUNCTION void func(void) {} |
156 | |
157 | //------------------------------------------------------------------------------ |
158 | // Encoding |
159 | |
160 | // Transforms |
161 | // VP8Idct: Does one of two inverse transforms. If do_two is set, the transforms |
162 | // will be done for (ref, in, dst) and (ref + 4, in + 16, dst + 4). |
163 | typedef void (*VP8Idct)(const uint8_t* ref, const int16_t* in, uint8_t* dst, |
164 | int do_two); |
165 | typedef void (*VP8Fdct)(const uint8_t* src, const uint8_t* ref, int16_t* out); |
166 | typedef void (*VP8WHT)(const int16_t* in, int16_t* out); |
167 | extern VP8Idct VP8ITransform; |
168 | extern VP8Fdct VP8FTransform; |
169 | extern VP8Fdct VP8FTransform2; // performs two transforms at a time |
170 | extern VP8WHT VP8FTransformWHT; |
171 | // Predictions |
172 | // *dst is the destination block. *top and *left can be NULL. |
173 | typedef void (*VP8IntraPreds)(uint8_t *dst, const uint8_t* left, |
174 | const uint8_t* top); |
175 | typedef void (*VP8Intra4Preds)(uint8_t *dst, const uint8_t* top); |
176 | extern VP8Intra4Preds VP8EncPredLuma4; |
177 | extern VP8IntraPreds VP8EncPredLuma16; |
178 | extern VP8IntraPreds VP8EncPredChroma8; |
179 | |
180 | typedef int (*VP8Metric)(const uint8_t* pix, const uint8_t* ref); |
181 | extern VP8Metric VP8SSE16x16, VP8SSE16x8, VP8SSE8x8, VP8SSE4x4; |
182 | typedef int (*VP8WMetric)(const uint8_t* pix, const uint8_t* ref, |
183 | const uint16_t* const weights); |
184 | // The weights for VP8TDisto4x4 and VP8TDisto16x16 contain a row-major |
185 | // 4 by 4 symmetric matrix. |
186 | extern VP8WMetric VP8TDisto4x4, VP8TDisto16x16; |
187 | |
188 | // Compute the average (DC) of four 4x4 blocks. |
189 | // Each sub-4x4 block #i sum is stored in dc[i]. |
190 | typedef void (*VP8MeanMetric)(const uint8_t* ref, uint32_t dc[4]); |
191 | extern VP8MeanMetric VP8Mean16x4; |
192 | |
193 | typedef void (*VP8BlockCopy)(const uint8_t* src, uint8_t* dst); |
194 | extern VP8BlockCopy VP8Copy4x4; |
195 | extern VP8BlockCopy VP8Copy16x8; |
196 | // Quantization |
197 | struct VP8Matrix; // forward declaration |
198 | typedef int (*VP8QuantizeBlock)(int16_t in[16], int16_t out[16], |
199 | const struct VP8Matrix* const mtx); |
200 | // Same as VP8QuantizeBlock, but quantizes two consecutive blocks. |
201 | typedef int (*VP8Quantize2Blocks)(int16_t in[32], int16_t out[32], |
202 | const struct VP8Matrix* const mtx); |
203 | |
204 | extern VP8QuantizeBlock VP8EncQuantizeBlock; |
205 | extern VP8Quantize2Blocks VP8EncQuantize2Blocks; |
206 | |
207 | // specific to 2nd transform: |
208 | typedef int (*VP8QuantizeBlockWHT)(int16_t in[16], int16_t out[16], |
209 | const struct VP8Matrix* const mtx); |
210 | extern VP8QuantizeBlockWHT VP8EncQuantizeBlockWHT; |
211 | |
212 | extern const int VP8DspScan[16 + 4 + 4]; |
213 | |
214 | // Collect histogram for susceptibility calculation. |
215 | #define MAX_COEFF_THRESH 31 // size of histogram used by CollectHistogram. |
216 | typedef struct { |
217 | // We only need to store max_value and last_non_zero, not the distribution. |
218 | int max_value; |
219 | int last_non_zero; |
220 | } VP8Histogram; |
221 | typedef void (*VP8CHisto)(const uint8_t* ref, const uint8_t* pred, |
222 | int start_block, int end_block, |
223 | VP8Histogram* const histo); |
224 | extern VP8CHisto VP8CollectHistogram; |
225 | // General-purpose util function to help VP8CollectHistogram(). |
226 | void VP8SetHistogramData(const int distribution[MAX_COEFF_THRESH + 1], |
227 | VP8Histogram* const histo); |
228 | |
229 | // must be called before using any of the above |
230 | void VP8EncDspInit(void); |
231 | |
232 | //------------------------------------------------------------------------------ |
233 | // cost functions (encoding) |
234 | |
235 | extern const uint16_t VP8EntropyCost[256]; // 8bit fixed-point log(p) |
236 | // approximate cost per level: |
237 | extern const uint16_t VP8LevelFixedCosts[2047 /*MAX_LEVEL*/ + 1]; |
238 | extern const uint8_t VP8EncBands[16 + 1]; |
239 | |
240 | struct VP8Residual; |
241 | typedef void (*VP8SetResidualCoeffsFunc)(const int16_t* const coeffs, |
242 | struct VP8Residual* const res); |
243 | extern VP8SetResidualCoeffsFunc VP8SetResidualCoeffs; |
244 | |
245 | // Cost calculation function. |
246 | typedef int (*VP8GetResidualCostFunc)(int ctx0, |
247 | const struct VP8Residual* const res); |
248 | extern VP8GetResidualCostFunc VP8GetResidualCost; |
249 | |
250 | // must be called before anything using the above |
251 | void VP8EncDspCostInit(void); |
252 | |
253 | //------------------------------------------------------------------------------ |
254 | // SSIM / PSNR utils |
255 | |
256 | // struct for accumulating statistical moments |
257 | typedef struct { |
258 | uint32_t w; // sum(w_i) : sum of weights |
259 | uint32_t xm, ym; // sum(w_i * x_i), sum(w_i * y_i) |
260 | uint32_t xxm, xym, yym; // sum(w_i * x_i * x_i), etc. |
261 | } VP8DistoStats; |
262 | |
263 | // Compute the final SSIM value |
264 | // The non-clipped version assumes stats->w = (2 * VP8_SSIM_KERNEL + 1)^2. |
265 | double VP8SSIMFromStats(const VP8DistoStats* const stats); |
266 | double VP8SSIMFromStatsClipped(const VP8DistoStats* const stats); |
267 | |
268 | #define VP8_SSIM_KERNEL 3 // total size of the kernel: 2 * VP8_SSIM_KERNEL + 1 |
269 | typedef double (*VP8SSIMGetClippedFunc)(const uint8_t* src1, int stride1, |
270 | const uint8_t* src2, int stride2, |
271 | int xo, int yo, // center position |
272 | int W, int H); // plane dimension |
273 | |
274 | // This version is called with the guarantee that you can load 8 bytes and |
275 | // 8 rows at offset src1 and src2 |
276 | typedef double (*VP8SSIMGetFunc)(const uint8_t* src1, int stride1, |
277 | const uint8_t* src2, int stride2); |
278 | |
279 | extern VP8SSIMGetFunc VP8SSIMGet; // unclipped / unchecked |
280 | extern VP8SSIMGetClippedFunc VP8SSIMGetClipped; // with clipping |
281 | |
282 | typedef uint32_t (*VP8AccumulateSSEFunc)(const uint8_t* src1, |
283 | const uint8_t* src2, int len); |
284 | extern VP8AccumulateSSEFunc VP8AccumulateSSE; |
285 | |
286 | // must be called before using any of the above directly |
287 | void VP8SSIMDspInit(void); |
288 | |
289 | //------------------------------------------------------------------------------ |
290 | // Decoding |
291 | |
292 | typedef void (*VP8DecIdct)(const int16_t* coeffs, uint8_t* dst); |
293 | // when doing two transforms, coeffs is actually int16_t[2][16]. |
294 | typedef void (*VP8DecIdct2)(const int16_t* coeffs, uint8_t* dst, int do_two); |
295 | extern VP8DecIdct2 VP8Transform; |
296 | extern VP8DecIdct VP8TransformAC3; |
297 | extern VP8DecIdct VP8TransformUV; |
298 | extern VP8DecIdct VP8TransformDC; |
299 | extern VP8DecIdct VP8TransformDCUV; |
300 | extern VP8WHT VP8TransformWHT; |
301 | |
302 | // *dst is the destination block, with stride BPS. Boundary samples are |
303 | // assumed accessible when needed. |
304 | typedef void (*VP8PredFunc)(uint8_t* dst); |
305 | extern VP8PredFunc VP8PredLuma16[/* NUM_B_DC_MODES */]; |
306 | extern VP8PredFunc VP8PredChroma8[/* NUM_B_DC_MODES */]; |
307 | extern VP8PredFunc VP8PredLuma4[/* NUM_BMODES */]; |
308 | |
309 | // clipping tables (for filtering) |
310 | extern const int8_t* const VP8ksclip1; // clips [-1020, 1020] to [-128, 127] |
311 | extern const int8_t* const VP8ksclip2; // clips [-112, 112] to [-16, 15] |
312 | extern const uint8_t* const VP8kclip1; // clips [-255,511] to [0,255] |
313 | extern const uint8_t* const VP8kabs0; // abs(x) for x in [-255,255] |
314 | // must be called first |
315 | void VP8InitClipTables(void); |
316 | |
317 | // simple filter (only for luma) |
318 | typedef void (*VP8SimpleFilterFunc)(uint8_t* p, int stride, int thresh); |
319 | extern VP8SimpleFilterFunc VP8SimpleVFilter16; |
320 | extern VP8SimpleFilterFunc VP8SimpleHFilter16; |
321 | extern VP8SimpleFilterFunc VP8SimpleVFilter16i; // filter 3 inner edges |
322 | extern VP8SimpleFilterFunc VP8SimpleHFilter16i; |
323 | |
324 | // regular filter (on both macroblock edges and inner edges) |
325 | typedef void (*VP8LumaFilterFunc)(uint8_t* luma, int stride, |
326 | int thresh, int ithresh, int hev_t); |
327 | typedef void (*VP8ChromaFilterFunc)(uint8_t* u, uint8_t* v, int stride, |
328 | int thresh, int ithresh, int hev_t); |
329 | // on outer edge |
330 | extern VP8LumaFilterFunc VP8VFilter16; |
331 | extern VP8LumaFilterFunc VP8HFilter16; |
332 | extern VP8ChromaFilterFunc VP8VFilter8; |
333 | extern VP8ChromaFilterFunc VP8HFilter8; |
334 | |
335 | // on inner edge |
336 | extern VP8LumaFilterFunc VP8VFilter16i; // filtering 3 inner edges altogether |
337 | extern VP8LumaFilterFunc VP8HFilter16i; |
338 | extern VP8ChromaFilterFunc VP8VFilter8i; // filtering u and v altogether |
339 | extern VP8ChromaFilterFunc VP8HFilter8i; |
340 | |
341 | // Dithering. Combines dithering values (centered around 128) with dst[], |
342 | // according to: dst[] = clip(dst[] + (((dither[]-128) + 8) >> 4) |
343 | #define VP8_DITHER_DESCALE 4 |
344 | #define VP8_DITHER_DESCALE_ROUNDER (1 << (VP8_DITHER_DESCALE - 1)) |
345 | #define VP8_DITHER_AMP_BITS 7 |
346 | #define VP8_DITHER_AMP_CENTER (1 << VP8_DITHER_AMP_BITS) |
347 | extern void (*VP8DitherCombine8x8)(const uint8_t* dither, uint8_t* dst, |
348 | int dst_stride); |
349 | |
350 | // must be called before anything using the above |
351 | void VP8DspInit(void); |
352 | |
353 | //------------------------------------------------------------------------------ |
354 | // WebP I/O |
355 | |
356 | #define FANCY_UPSAMPLING // undefined to remove fancy upsampling support |
357 | |
358 | // Convert a pair of y/u/v lines together to the output rgb/a colorspace. |
359 | // bottom_y can be NULL if only one line of output is needed (at top/bottom). |
360 | typedef void (*WebPUpsampleLinePairFunc)( |
361 | const uint8_t* top_y, const uint8_t* bottom_y, |
362 | const uint8_t* top_u, const uint8_t* top_v, |
363 | const uint8_t* cur_u, const uint8_t* cur_v, |
364 | uint8_t* top_dst, uint8_t* bottom_dst, int len); |
365 | |
366 | #ifdef FANCY_UPSAMPLING |
367 | |
368 | // Fancy upsampling functions to convert YUV to RGB(A) modes |
369 | extern WebPUpsampleLinePairFunc WebPUpsamplers[/* MODE_LAST */]; |
370 | |
371 | #endif // FANCY_UPSAMPLING |
372 | |
373 | // Per-row point-sampling methods. |
374 | typedef void (*WebPSamplerRowFunc)(const uint8_t* y, |
375 | const uint8_t* u, const uint8_t* v, |
376 | uint8_t* dst, int len); |
377 | // Generic function to apply 'WebPSamplerRowFunc' to the whole plane: |
378 | void WebPSamplerProcessPlane(const uint8_t* y, int y_stride, |
379 | const uint8_t* u, const uint8_t* v, int uv_stride, |
380 | uint8_t* dst, int dst_stride, |
381 | int width, int height, WebPSamplerRowFunc func); |
382 | |
383 | // Sampling functions to convert rows of YUV to RGB(A) |
384 | extern WebPSamplerRowFunc WebPSamplers[/* MODE_LAST */]; |
385 | |
386 | // General function for converting two lines of ARGB or RGBA. |
387 | // 'alpha_is_last' should be true if 0xff000000 is stored in memory as |
388 | // as 0x00, 0x00, 0x00, 0xff (little endian). |
389 | WebPUpsampleLinePairFunc WebPGetLinePairConverter(int alpha_is_last); |
390 | |
391 | // YUV444->RGB converters |
392 | typedef void (*WebPYUV444Converter)(const uint8_t* y, |
393 | const uint8_t* u, const uint8_t* v, |
394 | uint8_t* dst, int len); |
395 | |
396 | extern WebPYUV444Converter WebPYUV444Converters[/* MODE_LAST */]; |
397 | |
398 | // Must be called before using the WebPUpsamplers[] (and for premultiplied |
399 | // colorspaces like rgbA, rgbA4444, etc) |
400 | void WebPInitUpsamplers(void); |
401 | // Must be called before using WebPSamplers[] |
402 | void WebPInitSamplers(void); |
403 | // Must be called before using WebPYUV444Converters[] |
404 | void WebPInitYUV444Converters(void); |
405 | |
406 | //------------------------------------------------------------------------------ |
407 | // ARGB -> YUV converters |
408 | |
409 | // Convert ARGB samples to luma Y. |
410 | extern void (*WebPConvertARGBToY)(const uint32_t* argb, uint8_t* y, int width); |
411 | // Convert ARGB samples to U/V with downsampling. do_store should be '1' for |
412 | // even lines and '0' for odd ones. 'src_width' is the original width, not |
413 | // the U/V one. |
414 | extern void (*WebPConvertARGBToUV)(const uint32_t* argb, uint8_t* u, uint8_t* v, |
415 | int src_width, int do_store); |
416 | |
417 | // Convert a row of accumulated (four-values) of rgba32 toward U/V |
418 | extern void (*WebPConvertRGBA32ToUV)(const uint16_t* rgb, |
419 | uint8_t* u, uint8_t* v, int width); |
420 | |
421 | // Convert RGB or BGR to Y |
422 | extern void (*WebPConvertRGB24ToY)(const uint8_t* rgb, uint8_t* y, int width); |
423 | extern void (*WebPConvertBGR24ToY)(const uint8_t* bgr, uint8_t* y, int width); |
424 | |
425 | // used for plain-C fallback. |
426 | extern void WebPConvertARGBToUV_C(const uint32_t* argb, uint8_t* u, uint8_t* v, |
427 | int src_width, int do_store); |
428 | extern void WebPConvertRGBA32ToUV_C(const uint16_t* rgb, |
429 | uint8_t* u, uint8_t* v, int width); |
430 | |
431 | // utilities for accurate RGB->YUV conversion |
432 | extern uint64_t (*WebPSharpYUVUpdateY)(const uint16_t* src, const uint16_t* ref, |
433 | uint16_t* dst, int len); |
434 | extern void (*WebPSharpYUVUpdateRGB)(const int16_t* src, const int16_t* ref, |
435 | int16_t* dst, int len); |
436 | extern void (*WebPSharpYUVFilterRow)(const int16_t* A, const int16_t* B, |
437 | int len, |
438 | const uint16_t* best_y, uint16_t* out); |
439 | |
440 | // Must be called before using the above. |
441 | void WebPInitConvertARGBToYUV(void); |
442 | |
443 | //------------------------------------------------------------------------------ |
444 | // Rescaler |
445 | |
446 | struct WebPRescaler; |
447 | |
448 | // Import a row of data and save its contribution in the rescaler. |
449 | // 'channel' denotes the channel number to be imported. 'Expand' corresponds to |
450 | // the wrk->x_expand case. Otherwise, 'Shrink' is to be used. |
451 | typedef void (*WebPRescalerImportRowFunc)(struct WebPRescaler* const wrk, |
452 | const uint8_t* src); |
453 | |
454 | extern WebPRescalerImportRowFunc WebPRescalerImportRowExpand; |
455 | extern WebPRescalerImportRowFunc WebPRescalerImportRowShrink; |
456 | |
457 | // Export one row (starting at x_out position) from rescaler. |
458 | // 'Expand' corresponds to the wrk->y_expand case. |
459 | // Otherwise 'Shrink' is to be used |
460 | typedef void (*WebPRescalerExportRowFunc)(struct WebPRescaler* const wrk); |
461 | extern WebPRescalerExportRowFunc WebPRescalerExportRowExpand; |
462 | extern WebPRescalerExportRowFunc WebPRescalerExportRowShrink; |
463 | |
464 | // Plain-C implementation, as fall-back. |
465 | extern void WebPRescalerImportRowExpandC(struct WebPRescaler* const wrk, |
466 | const uint8_t* src); |
467 | extern void WebPRescalerImportRowShrinkC(struct WebPRescaler* const wrk, |
468 | const uint8_t* src); |
469 | extern void WebPRescalerExportRowExpandC(struct WebPRescaler* const wrk); |
470 | extern void WebPRescalerExportRowShrinkC(struct WebPRescaler* const wrk); |
471 | |
472 | // Main entry calls: |
473 | extern void WebPRescalerImportRow(struct WebPRescaler* const wrk, |
474 | const uint8_t* src); |
475 | // Export one row (starting at x_out position) from rescaler. |
476 | extern void WebPRescalerExportRow(struct WebPRescaler* const wrk); |
477 | |
478 | // Must be called first before using the above. |
479 | void WebPRescalerDspInit(void); |
480 | |
481 | //------------------------------------------------------------------------------ |
482 | // Utilities for processing transparent channel. |
483 | |
484 | // Apply alpha pre-multiply on an rgba, bgra or argb plane of size w * h. |
485 | // alpha_first should be 0 for argb, 1 for rgba or bgra (where alpha is last). |
486 | extern void (*WebPApplyAlphaMultiply)( |
487 | uint8_t* rgba, int alpha_first, int w, int h, int stride); |
488 | |
489 | // Same, buf specifically for RGBA4444 format |
490 | extern void (*WebPApplyAlphaMultiply4444)( |
491 | uint8_t* rgba4444, int w, int h, int stride); |
492 | |
493 | // Dispatch the values from alpha[] plane to the ARGB destination 'dst'. |
494 | // Returns true if alpha[] plane has non-trivial values different from 0xff. |
495 | extern int (*WebPDispatchAlpha)(const uint8_t* alpha, int alpha_stride, |
496 | int width, int height, |
497 | uint8_t* dst, int dst_stride); |
498 | |
499 | // Transfer packed 8b alpha[] values to green channel in dst[], zero'ing the |
500 | // A/R/B values. 'dst_stride' is the stride for dst[] in uint32_t units. |
501 | extern void (*WebPDispatchAlphaToGreen)(const uint8_t* alpha, int alpha_stride, |
502 | int width, int height, |
503 | uint32_t* dst, int dst_stride); |
504 | |
505 | // Extract the alpha values from 32b values in argb[] and pack them into alpha[] |
506 | // (this is the opposite of WebPDispatchAlpha). |
507 | // Returns true if there's only trivial 0xff alpha values. |
508 | extern int (*)(const uint8_t* argb, int argb_stride, |
509 | int width, int height, |
510 | uint8_t* alpha, int alpha_stride); |
511 | |
512 | // Extract the green values from 32b values in argb[] and pack them into alpha[] |
513 | // (this is the opposite of WebPDispatchAlphaToGreen). |
514 | extern void (*)(const uint32_t* argb, uint8_t* alpha, int size); |
515 | |
516 | // Pre-Multiply operation transforms x into x * A / 255 (where x=Y,R,G or B). |
517 | // Un-Multiply operation transforms x into x * 255 / A. |
518 | |
519 | // Pre-Multiply or Un-Multiply (if 'inverse' is true) argb values in a row. |
520 | extern void (*WebPMultARGBRow)(uint32_t* const ptr, int width, int inverse); |
521 | |
522 | // Same a WebPMultARGBRow(), but for several rows. |
523 | void WebPMultARGBRows(uint8_t* ptr, int stride, int width, int num_rows, |
524 | int inverse); |
525 | |
526 | // Same for a row of single values, with side alpha values. |
527 | extern void (*WebPMultRow)(uint8_t* const ptr, const uint8_t* const alpha, |
528 | int width, int inverse); |
529 | |
530 | // Same a WebPMultRow(), but for several 'num_rows' rows. |
531 | void WebPMultRows(uint8_t* ptr, int stride, |
532 | const uint8_t* alpha, int alpha_stride, |
533 | int width, int num_rows, int inverse); |
534 | |
535 | // Plain-C versions, used as fallback by some implementations. |
536 | void WebPMultRowC(uint8_t* const ptr, const uint8_t* const alpha, |
537 | int width, int inverse); |
538 | void WebPMultARGBRowC(uint32_t* const ptr, int width, int inverse); |
539 | |
540 | // To be called first before using the above. |
541 | void WebPInitAlphaProcessing(void); |
542 | |
543 | // ARGB packing function: a/r/g/b input is rgba or bgra order. |
544 | extern void (*VP8PackARGB)(const uint8_t* a, const uint8_t* r, |
545 | const uint8_t* g, const uint8_t* b, int len, |
546 | uint32_t* out); |
547 | |
548 | // RGB packing function. 'step' can be 3 or 4. r/g/b input is rgb or bgr order. |
549 | extern void (*VP8PackRGB)(const uint8_t* r, const uint8_t* g, const uint8_t* b, |
550 | int len, int step, uint32_t* out); |
551 | |
552 | // To be called first before using the above. |
553 | void VP8EncDspARGBInit(void); |
554 | |
555 | //------------------------------------------------------------------------------ |
556 | // Filter functions |
557 | |
558 | typedef enum { // Filter types. |
559 | WEBP_FILTER_NONE = 0, |
560 | WEBP_FILTER_HORIZONTAL, |
561 | WEBP_FILTER_VERTICAL, |
562 | WEBP_FILTER_GRADIENT, |
563 | WEBP_FILTER_LAST = WEBP_FILTER_GRADIENT + 1, // end marker |
564 | WEBP_FILTER_BEST, // meta-types |
565 | WEBP_FILTER_FAST |
566 | } WEBP_FILTER_TYPE; |
567 | |
568 | typedef void (*WebPFilterFunc)(const uint8_t* in, int width, int height, |
569 | int stride, uint8_t* out); |
570 | // In-place un-filtering. |
571 | // Warning! 'prev_line' pointer can be equal to 'cur_line' or 'preds'. |
572 | typedef void (*WebPUnfilterFunc)(const uint8_t* prev_line, const uint8_t* preds, |
573 | uint8_t* cur_line, int width); |
574 | |
575 | // Filter the given data using the given predictor. |
576 | // 'in' corresponds to a 2-dimensional pixel array of size (stride * height) |
577 | // in raster order. |
578 | // 'stride' is number of bytes per scan line (with possible padding). |
579 | // 'out' should be pre-allocated. |
580 | extern WebPFilterFunc WebPFilters[WEBP_FILTER_LAST]; |
581 | |
582 | // In-place reconstruct the original data from the given filtered data. |
583 | // The reconstruction will be done for 'num_rows' rows starting from 'row' |
584 | // (assuming rows upto 'row - 1' are already reconstructed). |
585 | extern WebPUnfilterFunc WebPUnfilters[WEBP_FILTER_LAST]; |
586 | |
587 | // To be called first before using the above. |
588 | void VP8FiltersInit(void); |
589 | |
590 | #ifdef __cplusplus |
591 | } // extern "C" |
592 | #endif |
593 | |
594 | #endif /* WEBP_DSP_DSP_H_ */ |
595 | |