src/third_party/libwebp/enc/vp8enci.h

   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 1
  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.
 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 <= 0x0202
 575 void WebPMemoryWriterClear(WebPMemoryWriter* writer);
 576 #endif
 577
 578 #ifdef __cplusplus
 579 }    // extern "C"
 580 #endif
 581
 582 #endif  /* WEBP_ENC_VP8ENCI_H_ */