• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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 //   WebP encoder: internal header.
11 //
12 // Author: Skal (pascal.massimino@gmail.com)
13 
14 #ifndef WEBP_ENC_VP8ENCI_H_
15 #define WEBP_ENC_VP8ENCI_H_
16 
17 #include <string.h>     // for memcpy()
18 #include "webp/encode.h"
19 #include "../dsp/dsp.h"
20 #include "../utils/bit_writer.h"
21 #include "../utils/thread.h"
22 
23 #ifdef __cplusplus
24 extern "C" {
25 #endif
26 
27 //------------------------------------------------------------------------------
28 // Various defines and enums
29 
30 // version numbers
31 #define ENC_MAJ_VERSION 0
32 #define ENC_MIN_VERSION 4
33 #define ENC_REV_VERSION 2
34 
35 // intra prediction modes
36 enum { B_DC_PRED = 0,   // 4x4 modes
37        B_TM_PRED = 1,
38        B_VE_PRED = 2,
39        B_HE_PRED = 3,
40        B_RD_PRED = 4,
41        B_VR_PRED = 5,
42        B_LD_PRED = 6,
43        B_VL_PRED = 7,
44        B_HD_PRED = 8,
45        B_HU_PRED = 9,
46        NUM_BMODES = B_HU_PRED + 1 - B_DC_PRED,  // = 10
47 
48        // Luma16 or UV modes
49        DC_PRED = B_DC_PRED, V_PRED = B_VE_PRED,
50        H_PRED = B_HE_PRED, TM_PRED = B_TM_PRED,
51        NUM_PRED_MODES = 4
52      };
53 
54 enum { NUM_MB_SEGMENTS = 4,
55        MAX_NUM_PARTITIONS = 8,
56        NUM_TYPES = 4,   // 0: i16-AC,  1: i16-DC,  2:chroma-AC,  3:i4-AC
57        NUM_BANDS = 8,
58        NUM_CTX = 3,
59        NUM_PROBAS = 11,
60        MAX_LF_LEVELS = 64,       // Maximum loop filter level
61        MAX_VARIABLE_LEVEL = 67,  // last (inclusive) level with variable cost
62        MAX_LEVEL = 2047          // max level (note: max codable is 2047 + 67)
63      };
64 
65 typedef enum {   // Rate-distortion optimization levels
66   RD_OPT_NONE        = 0,  // no rd-opt
67   RD_OPT_BASIC       = 1,  // basic scoring (no trellis)
68   RD_OPT_TRELLIS     = 2,  // perform trellis-quant on the final decision only
69   RD_OPT_TRELLIS_ALL = 3   // trellis-quant for every scoring (much slower)
70 } VP8RDLevel;
71 
72 // YUV-cache parameters. Cache is 16-pixels wide.
73 // The original or reconstructed samples can be accessed using VP8Scan[]
74 // The predicted blocks can be accessed using offsets to yuv_p_ and
75 // the arrays VP8*ModeOffsets[];
76 //         +----+      YUV Samples area. See VP8Scan[] for accessing the blocks.
77 //  Y_OFF  |YYYY| <- original samples  ('yuv_in_')
78 //         |YYYY|
79 //         |YYYY|
80 //         |YYYY|
81 //  U_OFF  |UUVV| V_OFF  (=U_OFF + 8)
82 //         |UUVV|
83 //         +----+
84 //  Y_OFF  |YYYY| <- compressed/decoded samples  ('yuv_out_')
85 //         |YYYY|    There are two buffers like this ('yuv_out_'/'yuv_out2_')
86 //         |YYYY|
87 //         |YYYY|
88 //  U_OFF  |UUVV| V_OFF
89 //         |UUVV|
90 //          x2 (for yuv_out2_)
91 //         +----+     Prediction area ('yuv_p_', size = PRED_SIZE)
92 // I16DC16 |YYYY|  Intra16 predictions (16x16 block each)
93 //         |YYYY|
94 //         |YYYY|
95 //         |YYYY|
96 // I16TM16 |YYYY|
97 //         |YYYY|
98 //         |YYYY|
99 //         |YYYY|
100 // I16VE16 |YYYY|
101 //         |YYYY|
102 //         |YYYY|
103 //         |YYYY|
104 // I16HE16 |YYYY|
105 //         |YYYY|
106 //         |YYYY|
107 //         |YYYY|
108 //         +----+  Chroma U/V predictions (16x8 block each)
109 // C8DC8   |UUVV|
110 //         |UUVV|
111 // C8TM8   |UUVV|
112 //         |UUVV|
113 // C8VE8   |UUVV|
114 //         |UUVV|
115 // C8HE8   |UUVV|
116 //         |UUVV|
117 //         +----+  Intra 4x4 predictions (4x4 block each)
118 //         |YYYY| I4DC4 I4TM4 I4VE4 I4HE4
119 //         |YYYY| I4RD4 I4VR4 I4LD4 I4VL4
120 //         |YY..| I4HD4 I4HU4 I4TMP
121 //         +----+
122 #define BPS       16   // this is the common stride
123 #define Y_SIZE   (BPS * 16)
124 #define UV_SIZE  (BPS * 8)
125 #define YUV_SIZE (Y_SIZE + UV_SIZE)
126 #define PRED_SIZE (6 * 16 * BPS + 12 * BPS)
127 #define Y_OFF    (0)
128 #define U_OFF    (Y_SIZE)
129 #define V_OFF    (U_OFF + 8)
130 #define ALIGN_CST 15
131 #define DO_ALIGN(PTR) ((uintptr_t)((PTR) + ALIGN_CST) & ~ALIGN_CST)
132 
133 extern const int VP8Scan[16];           // in quant.c
134 extern const int VP8UVModeOffsets[4];   // in analyze.c
135 extern const int VP8I16ModeOffsets[4];
136 extern const int VP8I4ModeOffsets[NUM_BMODES];
137 
138 // Layout of prediction blocks
139 // intra 16x16
140 #define I16DC16 (0 * 16 * BPS)
141 #define I16TM16 (1 * 16 * BPS)
142 #define I16VE16 (2 * 16 * BPS)
143 #define I16HE16 (3 * 16 * BPS)
144 // chroma 8x8, two U/V blocks side by side (hence: 16x8 each)
145 #define C8DC8 (4 * 16 * BPS)
146 #define C8TM8 (4 * 16 * BPS + 8 * BPS)
147 #define C8VE8 (5 * 16 * BPS)
148 #define C8HE8 (5 * 16 * BPS + 8 * BPS)
149 // intra 4x4
150 #define I4DC4 (6 * 16 * BPS +  0)
151 #define I4TM4 (6 * 16 * BPS +  4)
152 #define I4VE4 (6 * 16 * BPS +  8)
153 #define I4HE4 (6 * 16 * BPS + 12)
154 #define I4RD4 (6 * 16 * BPS + 4 * BPS +  0)
155 #define I4VR4 (6 * 16 * BPS + 4 * BPS +  4)
156 #define I4LD4 (6 * 16 * BPS + 4 * BPS +  8)
157 #define I4VL4 (6 * 16 * BPS + 4 * BPS + 12)
158 #define I4HD4 (6 * 16 * BPS + 8 * BPS +  0)
159 #define I4HU4 (6 * 16 * BPS + 8 * BPS +  4)
160 #define I4TMP (6 * 16 * BPS + 8 * BPS +  8)
161 
162 typedef int64_t score_t;     // type used for scores, rate, distortion
163 // Note that MAX_COST is not the maximum allowed by sizeof(score_t),
164 // in order to allow overflowing computations.
165 #define MAX_COST ((score_t)0x7fffffffffffffLL)
166 
167 #define QFIX 17
168 #define BIAS(b)  ((b) << (QFIX - 8))
169 // Fun fact: this is the _only_ line where we're actually being lossy and
170 // discarding bits.
QUANTDIV(uint32_t n,uint32_t iQ,uint32_t B)171 static WEBP_INLINE int QUANTDIV(uint32_t n, uint32_t iQ, uint32_t B) {
172   return (int)((n * iQ + B) >> QFIX);
173 }
174 
175 // size of histogram used by CollectHistogram.
176 #define MAX_COEFF_THRESH   31
177 typedef struct VP8Histogram VP8Histogram;
178 struct VP8Histogram {
179   // TODO(skal): we only need to store the max_value and last_non_zero actually.
180   int distribution[MAX_COEFF_THRESH + 1];
181 };
182 
183 // Uncomment the following to remove token-buffer code:
184 // #define DISABLE_TOKEN_BUFFER
185 
186 //------------------------------------------------------------------------------
187 // Headers
188 
189 typedef uint32_t proba_t;   // 16b + 16b
190 typedef uint8_t ProbaArray[NUM_CTX][NUM_PROBAS];
191 typedef proba_t StatsArray[NUM_CTX][NUM_PROBAS];
192 typedef uint16_t CostArray[NUM_CTX][MAX_VARIABLE_LEVEL + 1];
193 typedef double LFStats[NUM_MB_SEGMENTS][MAX_LF_LEVELS];  // filter stats
194 
195 typedef struct VP8Encoder VP8Encoder;
196 
197 // segment features
198 typedef struct {
199   int num_segments_;      // Actual number of segments. 1 segment only = unused.
200   int update_map_;        // whether to update the segment map or not.
201                           // must be 0 if there's only 1 segment.
202   int size_;              // bit-cost for transmitting the segment map
203 } VP8SegmentHeader;
204 
205 // Struct collecting all frame-persistent probabilities.
206 typedef struct {
207   uint8_t segments_[3];     // probabilities for segment tree
208   uint8_t skip_proba_;      // final probability of being skipped.
209   ProbaArray coeffs_[NUM_TYPES][NUM_BANDS];      // 1056 bytes
210   StatsArray stats_[NUM_TYPES][NUM_BANDS];       // 4224 bytes
211   CostArray level_cost_[NUM_TYPES][NUM_BANDS];   // 13056 bytes
212   int dirty_;               // if true, need to call VP8CalculateLevelCosts()
213   int use_skip_proba_;      // Note: we always use skip_proba for now.
214   int nb_skip_;             // number of skipped blocks
215 } VP8Proba;
216 
217 // Filter parameters. Not actually used in the code (we don't perform
218 // the in-loop filtering), but filled from user's config
219 typedef struct {
220   int simple_;             // filtering type: 0=complex, 1=simple
221   int level_;              // base filter level [0..63]
222   int sharpness_;          // [0..7]
223   int i4x4_lf_delta_;      // delta filter level for i4x4 relative to i16x16
224 } VP8FilterHeader;
225 
226 //------------------------------------------------------------------------------
227 // Informations about the macroblocks.
228 
229 typedef struct {
230   // block type
231   unsigned int type_:2;     // 0=i4x4, 1=i16x16
232   unsigned int uv_mode_:2;
233   unsigned int skip_:1;
234   unsigned int segment_:2;
235   uint8_t alpha_;      // quantization-susceptibility
236 } VP8MBInfo;
237 
238 typedef struct VP8Matrix {
239   uint16_t q_[16];        // quantizer steps
240   uint16_t iq_[16];       // reciprocals, fixed point.
241   uint32_t bias_[16];     // rounding bias
242   uint32_t zthresh_[16];  // value below which a coefficient is zeroed
243   uint16_t sharpen_[16];  // frequency boosters for slight sharpening
244 } VP8Matrix;
245 
246 typedef struct {
247   VP8Matrix y1_, y2_, uv_;  // quantization matrices
248   int alpha_;      // quant-susceptibility, range [-127,127]. Zero is neutral.
249                    // Lower values indicate a lower risk of blurriness.
250   int beta_;       // filter-susceptibility, range [0,255].
251   int quant_;      // final segment quantizer.
252   int fstrength_;  // final in-loop filtering strength
253   int max_edge_;   // max edge delta (for filtering strength)
254   int min_disto_;  // minimum distortion required to trigger filtering record
255   // reactivities
256   int lambda_i16_, lambda_i4_, lambda_uv_;
257   int lambda_mode_, lambda_trellis_, tlambda_;
258   int lambda_trellis_i16_, lambda_trellis_i4_, lambda_trellis_uv_;
259 } VP8SegmentInfo;
260 
261 // Handy transient struct to accumulate score and info during RD-optimization
262 // and mode evaluation.
263 typedef struct {
264   score_t D, SD;              // Distortion, spectral distortion
265   score_t H, R, score;        // header bits, rate, score.
266   int16_t y_dc_levels[16];    // Quantized levels for luma-DC, luma-AC, chroma.
267   int16_t y_ac_levels[16][16];
268   int16_t uv_levels[4 + 4][16];
269   int mode_i16;               // mode number for intra16 prediction
270   uint8_t modes_i4[16];       // mode numbers for intra4 predictions
271   int mode_uv;                // mode number of chroma prediction
272   uint32_t nz;                // non-zero blocks
273 } VP8ModeScore;
274 
275 // Iterator structure to iterate through macroblocks, pointing to the
276 // right neighbouring data (samples, predictions, contexts, ...)
277 typedef struct {
278   int x_, y_;                      // current macroblock
279   int y_stride_, uv_stride_;       // respective strides
280   uint8_t*      yuv_in_;           // input samples
281   uint8_t*      yuv_out_;          // output samples
282   uint8_t*      yuv_out2_;         // secondary buffer swapped with yuv_out_.
283   uint8_t*      yuv_p_;            // scratch buffer for prediction
284   VP8Encoder*   enc_;              // back-pointer
285   VP8MBInfo*    mb_;               // current macroblock
286   VP8BitWriter* bw_;               // current bit-writer
287   uint8_t*      preds_;            // intra mode predictors (4x4 blocks)
288   uint32_t*     nz_;               // non-zero pattern
289   uint8_t       i4_boundary_[37];  // 32+5 boundary samples needed by intra4x4
290   uint8_t*      i4_top_;           // pointer to the current top boundary sample
291   int           i4_;               // current intra4x4 mode being tested
292   int           top_nz_[9];        // top-non-zero context.
293   int           left_nz_[9];       // left-non-zero. left_nz[8] is independent.
294   uint64_t      bit_count_[4][3];  // bit counters for coded levels.
295   uint64_t      luma_bits_;        // macroblock bit-cost for luma
296   uint64_t      uv_bits_;          // macroblock bit-cost for chroma
297   LFStats*      lf_stats_;         // filter stats (borrowed from enc_)
298   int           do_trellis_;       // if true, perform extra level optimisation
299   int           count_down_;       // number of mb still to be processed
300   int           count_down0_;      // starting counter value (for progress)
301   int           percent0_;         // saved initial progress percent
302 
303   uint8_t* y_left_;    // left luma samples (addressable from index -1 to 15).
304   uint8_t* u_left_;    // left u samples (addressable from index -1 to 7)
305   uint8_t* v_left_;    // left v samples (addressable from index -1 to 7)
306 
307   uint8_t* y_top_;     // top luma samples at position 'x_'
308   uint8_t* uv_top_;    // top u/v samples at position 'x_', packed as 16 bytes
309 
310   // memory for storing y/u/v_left_ and yuv_in_/out_*
311   uint8_t yuv_left_mem_[17 + 16 + 16 + 8 + ALIGN_CST];     // memory for *_left_
312   uint8_t yuv_mem_[3 * YUV_SIZE + PRED_SIZE + ALIGN_CST];  // memory for yuv_*
313 } VP8EncIterator;
314 
315   // in iterator.c
316 // must be called first
317 void VP8IteratorInit(VP8Encoder* const enc, VP8EncIterator* const it);
318 // restart a scan
319 void VP8IteratorReset(VP8EncIterator* const it);
320 // reset iterator position to row 'y'
321 void VP8IteratorSetRow(VP8EncIterator* const it, int y);
322 // set count down (=number of iterations to go)
323 void VP8IteratorSetCountDown(VP8EncIterator* const it, int count_down);
324 // return true if iteration is finished
325 int VP8IteratorIsDone(const VP8EncIterator* const it);
326 // Import uncompressed samples from source.
327 // If tmp_32 is not NULL, import boundary samples too.
328 // tmp_32 is a 32-bytes scratch buffer that must be aligned in memory.
329 void VP8IteratorImport(VP8EncIterator* const it, uint8_t* tmp_32);
330 // export decimated samples
331 void VP8IteratorExport(const VP8EncIterator* const it);
332 // go to next macroblock. Returns false if not finished.
333 int VP8IteratorNext(VP8EncIterator* const it);
334 // save the yuv_out_ boundary values to top_/left_ arrays for next iterations.
335 void VP8IteratorSaveBoundary(VP8EncIterator* const it);
336 // Report progression based on macroblock rows. Return 0 for user-abort request.
337 int VP8IteratorProgress(const VP8EncIterator* const it,
338                         int final_delta_percent);
339 // Intra4x4 iterations
340 void VP8IteratorStartI4(VP8EncIterator* const it);
341 // returns true if not done.
342 int VP8IteratorRotateI4(VP8EncIterator* const it,
343                         const uint8_t* const yuv_out);
344 
345 // Non-zero context setup/teardown
346 void VP8IteratorNzToBytes(VP8EncIterator* const it);
347 void VP8IteratorBytesToNz(VP8EncIterator* const it);
348 
349 // Helper functions to set mode properties
350 void VP8SetIntra16Mode(const VP8EncIterator* const it, int mode);
351 void VP8SetIntra4Mode(const VP8EncIterator* const it, const uint8_t* modes);
352 void VP8SetIntraUVMode(const VP8EncIterator* const it, int mode);
353 void VP8SetSkip(const VP8EncIterator* const it, int skip);
354 void VP8SetSegment(const VP8EncIterator* const it, int segment);
355 
356 //------------------------------------------------------------------------------
357 // Paginated token buffer
358 
359 typedef struct VP8Tokens VP8Tokens;  // struct details in token.c
360 
361 typedef struct {
362 #if !defined(DISABLE_TOKEN_BUFFER)
363   VP8Tokens* pages_;        // first page
364   VP8Tokens** last_page_;   // last page
365   uint16_t* tokens_;        // set to (*last_page_)->tokens_
366   int left_;                // how many free tokens left before the page is full
367   int page_size_;           // number of tokens per page
368 #endif
369   int error_;         // true in case of malloc error
370 } VP8TBuffer;
371 
372 // initialize an empty buffer
373 void VP8TBufferInit(VP8TBuffer* const b, int page_size);
374 void VP8TBufferClear(VP8TBuffer* const b);   // de-allocate pages memory
375 
376 #if !defined(DISABLE_TOKEN_BUFFER)
377 
378 // Finalizes bitstream when probabilities are known.
379 // Deletes the allocated token memory if final_pass is true.
380 int VP8EmitTokens(VP8TBuffer* const b, VP8BitWriter* const bw,
381                   const uint8_t* const probas, int final_pass);
382 
383 // record the coding of coefficients without knowing the probabilities yet
384 int VP8RecordCoeffTokens(int ctx, int coeff_type, int first, int last,
385                          const int16_t* const coeffs,
386                          VP8TBuffer* const tokens);
387 
388 // Estimate the final coded size given a set of 'probas'.
389 size_t VP8EstimateTokenSize(VP8TBuffer* const b, const uint8_t* const probas);
390 
391 // unused for now
392 void VP8TokenToStats(const VP8TBuffer* const b, proba_t* const stats);
393 
394 #endif  // !DISABLE_TOKEN_BUFFER
395 
396 //------------------------------------------------------------------------------
397 // VP8Encoder
398 
399 struct VP8Encoder {
400   const WebPConfig* config_;    // user configuration and parameters
401   WebPPicture* pic_;            // input / output picture
402 
403   // headers
404   VP8FilterHeader   filter_hdr_;     // filtering information
405   VP8SegmentHeader  segment_hdr_;    // segment information
406 
407   int profile_;                      // VP8's profile, deduced from Config.
408 
409   // dimension, in macroblock units.
410   int mb_w_, mb_h_;
411   int preds_w_;   // stride of the *preds_ prediction plane (=4*mb_w + 1)
412 
413   // number of partitions (1, 2, 4 or 8 = MAX_NUM_PARTITIONS)
414   int num_parts_;
415 
416   // per-partition boolean decoders.
417   VP8BitWriter bw_;                         // part0
418   VP8BitWriter parts_[MAX_NUM_PARTITIONS];  // token partitions
419   VP8TBuffer tokens_;                       // token buffer
420 
421   int percent_;                             // for progress
422 
423   // transparency blob
424   int has_alpha_;
425   uint8_t* alpha_data_;       // non-NULL if transparency is present
426   uint32_t alpha_data_size_;
427   WebPWorker alpha_worker_;
428 
429   // quantization info (one set of DC/AC dequant factor per segment)
430   VP8SegmentInfo dqm_[NUM_MB_SEGMENTS];
431   int base_quant_;                 // nominal quantizer value. Only used
432                                    // for relative coding of segments' quant.
433   int alpha_;                      // global susceptibility (<=> complexity)
434   int uv_alpha_;                   // U/V quantization susceptibility
435   // global offset of quantizers, shared by all segments
436   int dq_y1_dc_;
437   int dq_y2_dc_, dq_y2_ac_;
438   int dq_uv_dc_, dq_uv_ac_;
439 
440   // probabilities and statistics
441   VP8Proba proba_;
442   uint64_t sse_[4];        // sum of Y/U/V/A squared errors for all macroblocks
443   uint64_t sse_count_;     // pixel count for the sse_[] stats
444   int      coded_size_;
445   int      residual_bytes_[3][4];
446   int      block_count_[3];
447 
448   // quality/speed settings
449   int method_;               // 0=fastest, 6=best/slowest.
450   VP8RDLevel rd_opt_level_;  // Deduced from method_.
451   int max_i4_header_bits_;   // partition #0 safeness factor
452   int thread_level_;         // derived from config->thread_level
453   int do_search_;            // derived from config->target_XXX
454   int use_tokens_;           // if true, use token buffer
455 
456   // Memory
457   VP8MBInfo* mb_info_;   // contextual macroblock infos (mb_w_ + 1)
458   uint8_t*   preds_;     // predictions modes: (4*mb_w+1) * (4*mb_h+1)
459   uint32_t*  nz_;        // non-zero bit context: mb_w+1
460   uint8_t*   y_top_;     // top luma samples.
461   uint8_t*   uv_top_;    // top u/v samples.
462                          // U and V are packed into 16 bytes (8 U + 8 V)
463   LFStats*   lf_stats_;  // autofilter stats (if NULL, autofilter is off)
464 };
465 
466 //------------------------------------------------------------------------------
467 // internal functions. Not public.
468 
469   // in tree.c
470 extern const uint8_t VP8CoeffsProba0[NUM_TYPES][NUM_BANDS][NUM_CTX][NUM_PROBAS];
471 extern const uint8_t
472     VP8CoeffsUpdateProba[NUM_TYPES][NUM_BANDS][NUM_CTX][NUM_PROBAS];
473 // Reset the token probabilities to their initial (default) values
474 void VP8DefaultProbas(VP8Encoder* const enc);
475 // Write the token probabilities
476 void VP8WriteProbas(VP8BitWriter* const bw, const VP8Proba* const probas);
477 // Writes the partition #0 modes (that is: all intra modes)
478 void VP8CodeIntraModes(VP8Encoder* const enc);
479 
480   // in syntax.c
481 // Generates the final bitstream by coding the partition0 and headers,
482 // and appending an assembly of all the pre-coded token partitions.
483 // Return true if everything is ok.
484 int VP8EncWrite(VP8Encoder* const enc);
485 // Release memory allocated for bit-writing in VP8EncLoop & seq.
486 void VP8EncFreeBitWriters(VP8Encoder* const enc);
487 
488   // in frame.c
489 extern const uint8_t VP8EncBands[16 + 1];
490 extern const uint8_t VP8Cat3[];
491 extern const uint8_t VP8Cat4[];
492 extern const uint8_t VP8Cat5[];
493 extern const uint8_t VP8Cat6[];
494 
495 // Form all the four Intra16x16 predictions in the yuv_p_ cache
496 void VP8MakeLuma16Preds(const VP8EncIterator* const it);
497 // Form all the four Chroma8x8 predictions in the yuv_p_ cache
498 void VP8MakeChroma8Preds(const VP8EncIterator* const it);
499 // Form all the ten Intra4x4 predictions in the yuv_p_ cache
500 // for the 4x4 block it->i4_
501 void VP8MakeIntra4Preds(const VP8EncIterator* const it);
502 // Rate calculation
503 int VP8GetCostLuma16(VP8EncIterator* const it, const VP8ModeScore* const rd);
504 int VP8GetCostLuma4(VP8EncIterator* const it, const int16_t levels[16]);
505 int VP8GetCostUV(VP8EncIterator* const it, const VP8ModeScore* const rd);
506 // Main coding calls
507 int VP8EncLoop(VP8Encoder* const enc);
508 int VP8EncTokenLoop(VP8Encoder* const enc);
509 
510   // in webpenc.c
511 // Assign an error code to a picture. Return false for convenience.
512 int WebPEncodingSetError(const WebPPicture* const pic, WebPEncodingError error);
513 int WebPReportProgress(const WebPPicture* const pic,
514                        int percent, int* const percent_store);
515 
516   // in analysis.c
517 // Main analysis loop. Decides the segmentations and complexity.
518 // Assigns a first guess for Intra16 and uvmode_ prediction modes.
519 int VP8EncAnalyze(VP8Encoder* const enc);
520 
521   // in quant.c
522 // Sets up segment's quantization values, base_quant_ and filter strengths.
523 void VP8SetSegmentParams(VP8Encoder* const enc, float quality);
524 // Pick best modes and fills the levels. Returns true if skipped.
525 int VP8Decimate(VP8EncIterator* const it, VP8ModeScore* const rd,
526                 VP8RDLevel rd_opt);
527 
528   // in alpha.c
529 void VP8EncInitAlpha(VP8Encoder* const enc);    // initialize alpha compression
530 int VP8EncStartAlpha(VP8Encoder* const enc);    // start alpha coding process
531 int VP8EncFinishAlpha(VP8Encoder* const enc);   // finalize compressed data
532 int VP8EncDeleteAlpha(VP8Encoder* const enc);   // delete compressed data
533 
534   // in filter.c
535 
536 // SSIM utils
537 typedef struct {
538   double w, xm, ym, xxm, xym, yym;
539 } DistoStats;
540 void VP8SSIMAddStats(const DistoStats* const src, DistoStats* const dst);
541 void VP8SSIMAccumulatePlane(const uint8_t* src1, int stride1,
542                             const uint8_t* src2, int stride2,
543                             int W, int H, DistoStats* const stats);
544 double VP8SSIMGet(const DistoStats* const stats);
545 double VP8SSIMGetSquaredError(const DistoStats* const stats);
546 
547 // autofilter
548 void VP8InitFilter(VP8EncIterator* const it);
549 void VP8StoreFilterStats(VP8EncIterator* const it);
550 void VP8AdjustFilterStrength(VP8EncIterator* const it);
551 
552 // returns the approximate filtering strength needed to smooth a edge
553 // step of 'delta', given a sharpness parameter 'sharpness'.
554 int VP8FilterStrengthFromDelta(int sharpness, int delta);
555 
556   // misc utils for picture_*.c:
557 
558 // Remove reference to the ARGB/YUVA buffer (doesn't free anything).
559 void WebPPictureResetBuffers(WebPPicture* const picture);
560 
561 // Allocates ARGB buffer of given dimension (previous one is always free'd).
562 // Preserves the YUV(A) buffer. Returns false in case of error (invalid param,
563 // out-of-memory).
564 int WebPPictureAllocARGB(WebPPicture* const picture, int width, int height);
565 
566 // Allocates YUVA buffer of given dimension (previous one is always free'd).
567 // Uses picture->csp to determine whether an alpha buffer is needed.
568 // Preserves the ARGB buffer.
569 // Returns false in case of error (invalid param, out-of-memory).
570 int WebPPictureAllocYUVA(WebPPicture* const picture, int width, int height);
571 
572 //------------------------------------------------------------------------------
573 
574 #if WEBP_ENCODER_ABI_VERSION <= 0x0203
575 void WebPMemoryWriterClear(WebPMemoryWriter* writer);
576 #endif
577 
578 #ifdef __cplusplus
579 }    // extern "C"
580 #endif
581 
582 #endif  /* WEBP_ENC_VP8ENCI_H_ */
583