15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright 2011 Google Inc. All Rights Reserved.
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
3eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch// Use of this source code is governed by a BSD-style license
4eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch// that can be found in the COPYING file in the root of the source
5eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch// tree. An additional intellectual property rights grant can be found
6eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch// in the file PATENTS. All contributing project authors may
7eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch// be found in the AUTHORS file in the root of the source tree.
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// -----------------------------------------------------------------------------
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//   WebP encoder: internal header.
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Author: Skal (pascal.massimino@gmail.com)
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef WEBP_ENC_VP8ENCI_H_
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define WEBP_ENC_VP8ENCI_H_
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string.h>     // for memcpy()
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "../webp/encode.h"
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "../dsp/dsp.h"
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "../utils/bit_writer.h"
212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "../utils/thread.h"
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
235d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#ifdef __cplusplus
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern "C" {
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//------------------------------------------------------------------------------
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Various defines and enums
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// version numbers
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define ENC_MAJ_VERSION 0
325d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#define ENC_MIN_VERSION 4
335f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)#define ENC_REV_VERSION 1
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// intra prediction modes
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)enum { B_DC_PRED = 0,   // 4x4 modes
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       B_TM_PRED = 1,
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       B_VE_PRED = 2,
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       B_HE_PRED = 3,
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       B_RD_PRED = 4,
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       B_VR_PRED = 5,
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       B_LD_PRED = 6,
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       B_VL_PRED = 7,
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       B_HD_PRED = 8,
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       B_HU_PRED = 9,
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       NUM_BMODES = B_HU_PRED + 1 - B_DC_PRED,  // = 10
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       // Luma16 or UV modes
495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       DC_PRED = B_DC_PRED, V_PRED = B_VE_PRED,
502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)       H_PRED = B_HE_PRED, TM_PRED = B_TM_PRED,
512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)       NUM_PRED_MODES = 4
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)     };
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)enum { NUM_MB_SEGMENTS = 4,
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       MAX_NUM_PARTITIONS = 8,
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       NUM_TYPES = 4,   // 0: i16-AC,  1: i16-DC,  2:chroma-AC,  3:i4-AC
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       NUM_BANDS = 8,
585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       NUM_CTX = 3,
595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       NUM_PROBAS = 11,
60c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)       MAX_LF_LEVELS = 64,       // Maximum loop filter level
61c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)       MAX_VARIABLE_LEVEL = 67,  // last (inclusive) level with variable cost
62c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)       MAX_LEVEL = 2047          // max level (note: max codable is 2047 + 67)
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)     };
645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)typedef enum {   // Rate-distortion optimization levels
662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  RD_OPT_NONE        = 0,  // no rd-opt
672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  RD_OPT_BASIC       = 1,  // basic scoring (no trellis)
682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  RD_OPT_TRELLIS     = 2,  // perform trellis-quant on the final decision only
692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  RD_OPT_TRELLIS_ALL = 3   // trellis-quant for every scoring (much slower)
702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} VP8RDLevel;
712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// YUV-cache parameters. Cache is 16-pixels wide.
735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The original or reconstructed samples can be accessed using VP8Scan[]
745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The predicted blocks can be accessed using offsets to yuv_p_ and
755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the arrays VP8*ModeOffsets[];
765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         +----+      YUV Samples area. See VP8Scan[] for accessing the blocks.
775d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)//  Y_OFF  |YYYY| <- original samples  ('yuv_in_')
785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//  U_OFF  |UUVV| V_OFF  (=U_OFF + 8)
825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |UUVV|
835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         +----+
845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//  Y_OFF  |YYYY| <- compressed/decoded samples  ('yuv_out_')
855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|    There are two buffers like this ('yuv_out_'/'yuv_out2_')
865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//  U_OFF  |UUVV| V_OFF
895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |UUVV|
905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//          x2 (for yuv_out2_)
915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         +----+     Prediction area ('yuv_p_', size = PRED_SIZE)
925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// I16DC16 |YYYY|  Intra16 predictions (16x16 block each)
935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// I16TM16 |YYYY|
975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// I16VE16 |YYYY|
1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// I16HE16 |YYYY|
1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY|
1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         +----+  Chroma U/V predictions (16x8 block each)
1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// C8DC8   |UUVV|
1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |UUVV|
1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// C8TM8   |UUVV|
1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |UUVV|
1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// C8VE8   |UUVV|
1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |UUVV|
1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// C8HE8   |UUVV|
1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |UUVV|
1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         +----+  Intra 4x4 predictions (4x4 block each)
1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY| I4DC4 I4TM4 I4VE4 I4HE4
1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YYYY| I4RD4 I4VR4 I4LD4 I4VL4
1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         |YY..| I4HD4 I4HU4 I4TMP
1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         +----+
1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define BPS       16   // this is the common stride
1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define Y_SIZE   (BPS * 16)
1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define UV_SIZE  (BPS * 8)
1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define YUV_SIZE (Y_SIZE + UV_SIZE)
1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define PRED_SIZE (6 * 16 * BPS + 12 * BPS)
1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define Y_OFF    (0)
1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define U_OFF    (Y_SIZE)
1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define V_OFF    (U_OFF + 8)
1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define ALIGN_CST 15
1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define DO_ALIGN(PTR) ((uintptr_t)((PTR) + ALIGN_CST) & ~ALIGN_CST)
1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1335f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)extern const int VP8Scan[16];           // in quant.c
1345f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)extern const int VP8UVModeOffsets[4];   // in analyze.c
1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern const int VP8I16ModeOffsets[4];
1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern const int VP8I4ModeOffsets[NUM_BMODES];
1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Layout of prediction blocks
1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// intra 16x16
1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I16DC16 (0 * 16 * BPS)
1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I16TM16 (1 * 16 * BPS)
1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I16VE16 (2 * 16 * BPS)
1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I16HE16 (3 * 16 * BPS)
1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// chroma 8x8, two U/V blocks side by side (hence: 16x8 each)
1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define C8DC8 (4 * 16 * BPS)
1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define C8TM8 (4 * 16 * BPS + 8 * BPS)
1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define C8VE8 (5 * 16 * BPS)
1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define C8HE8 (5 * 16 * BPS + 8 * BPS)
1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// intra 4x4
1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I4DC4 (6 * 16 * BPS +  0)
1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I4TM4 (6 * 16 * BPS +  4)
1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I4VE4 (6 * 16 * BPS +  8)
1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I4HE4 (6 * 16 * BPS + 12)
1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I4RD4 (6 * 16 * BPS + 4 * BPS +  0)
1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I4VR4 (6 * 16 * BPS + 4 * BPS +  4)
1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I4LD4 (6 * 16 * BPS + 4 * BPS +  8)
1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I4VL4 (6 * 16 * BPS + 4 * BPS + 12)
1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I4HD4 (6 * 16 * BPS + 8 * BPS +  0)
1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I4HU4 (6 * 16 * BPS + 8 * BPS +  4)
1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define I4TMP (6 * 16 * BPS + 8 * BPS +  8)
1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef int64_t score_t;     // type used for scores, rate, distortion
1635f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)// Note that MAX_COST is not the maximum allowed by sizeof(score_t),
1645f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)// in order to allow overflowing computations.
1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define MAX_COST ((score_t)0x7fffffffffffffLL)
1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define QFIX 17
1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define BIAS(b)  ((b) << (QFIX - 8))
1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Fun fact: this is the _only_ line where we're actually being lossy and
1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// discarding bits.
1715f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)static WEBP_INLINE int QUANTDIV(uint32_t n, uint32_t iQ, uint32_t B) {
1725f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  return (int)((n * iQ + B) >> QFIX);
1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// size of histogram used by CollectHistogram.
1762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define MAX_COEFF_THRESH   31
1772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)typedef struct VP8Histogram VP8Histogram;
1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)struct VP8Histogram {
1792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // TODO(skal): we only need to store the max_value and last_non_zero actually.
1802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  int distribution[MAX_COEFF_THRESH + 1];
1812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)};
1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Uncomment the following to remove token-buffer code:
1842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// #define DISABLE_TOKEN_BUFFER
1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//------------------------------------------------------------------------------
1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Headers
1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)typedef uint32_t proba_t;   // 16b + 16b
1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef uint8_t ProbaArray[NUM_CTX][NUM_PROBAS];
1912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)typedef proba_t StatsArray[NUM_CTX][NUM_PROBAS];
1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef uint16_t CostArray[NUM_CTX][MAX_VARIABLE_LEVEL + 1];
1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef double LFStats[NUM_MB_SEGMENTS][MAX_LF_LEVELS];  // filter stats
1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef struct VP8Encoder VP8Encoder;
1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// segment features
1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef struct {
1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int num_segments_;      // Actual number of segments. 1 segment only = unused.
2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int update_map_;        // whether to update the segment map or not.
2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                          // must be 0 if there's only 1 segment.
2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int size_;              // bit-cost for transmitting the segment map
2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} VP8SegmentHeader;
2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Struct collecting all frame-persistent probabilities.
2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef struct {
2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint8_t segments_[3];     // probabilities for segment tree
2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint8_t skip_proba_;      // final probability of being skipped.
2095f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  ProbaArray coeffs_[NUM_TYPES][NUM_BANDS];      // 1056 bytes
2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  StatsArray stats_[NUM_TYPES][NUM_BANDS];       // 4224 bytes
2115f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  CostArray level_cost_[NUM_TYPES][NUM_BANDS];   // 13056 bytes
2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  int dirty_;               // if true, need to call VP8CalculateLevelCosts()
2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int use_skip_proba_;      // Note: we always use skip_proba for now.
2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int nb_skip_;             // number of skipped blocks
2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} VP8Proba;
2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Filter parameters. Not actually used in the code (we don't perform
2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the in-loop filtering), but filled from user's config
2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef struct {
2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int simple_;             // filtering type: 0=complex, 1=simple
2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int level_;              // base filter level [0..63]
2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int sharpness_;          // [0..7]
2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int i4x4_lf_delta_;      // delta filter level for i4x4 relative to i16x16
2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} VP8FilterHeader;
2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//------------------------------------------------------------------------------
2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Informations about the macroblocks.
2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef struct {
2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // block type
2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  unsigned int type_:2;     // 0=i4x4, 1=i16x16
2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  unsigned int uv_mode_:2;
2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  unsigned int skip_:1;
2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  unsigned int segment_:2;
2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint8_t alpha_;      // quantization-susceptibility
2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} VP8MBInfo;
2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef struct VP8Matrix {
2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint16_t q_[16];        // quantizer steps
2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint16_t iq_[16];       // reciprocals, fixed point.
2415f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  uint32_t bias_[16];     // rounding bias
2425f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  uint32_t zthresh_[16];  // value below which a coefficient is zeroed
2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint16_t sharpen_[16];  // frequency boosters for slight sharpening
2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} VP8Matrix;
2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef struct {
2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  VP8Matrix y1_, y2_, uv_;  // quantization matrices
2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int alpha_;      // quant-susceptibility, range [-127,127]. Zero is neutral.
2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                   // Lower values indicate a lower risk of blurriness.
2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int beta_;       // filter-susceptibility, range [0,255].
2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int quant_;      // final segment quantizer.
2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int fstrength_;  // final in-loop filtering strength
2535d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  int max_edge_;   // max edge delta (for filtering strength)
2545d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  int min_disto_;  // minimum distortion required to trigger filtering record
2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // reactivities
2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int lambda_i16_, lambda_i4_, lambda_uv_;
2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int lambda_mode_, lambda_trellis_, tlambda_;
2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int lambda_trellis_i16_, lambda_trellis_i4_, lambda_trellis_uv_;
2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} VP8SegmentInfo;
2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2615d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// Handy transient struct to accumulate score and info during RD-optimization
2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// and mode evaluation.
2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef struct {
2645d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  score_t D, SD;              // Distortion, spectral distortion
2655d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  score_t H, R, score;        // header bits, rate, score.
2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int16_t y_dc_levels[16];    // Quantized levels for luma-DC, luma-AC, chroma.
2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int16_t y_ac_levels[16][16];
2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int16_t uv_levels[4 + 4][16];
2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int mode_i16;               // mode number for intra16 prediction
2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint8_t modes_i4[16];       // mode numbers for intra4 predictions
2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int mode_uv;                // mode number of chroma prediction
2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint32_t nz;                // non-zero blocks
2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} VP8ModeScore;
2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Iterator structure to iterate through macroblocks, pointing to the
2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// right neighbouring data (samples, predictions, contexts, ...)
2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef struct {
2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int x_, y_;                      // current macroblock
2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int y_stride_, uv_stride_;       // respective strides
2805d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  uint8_t*      yuv_in_;           // input samples
2815d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  uint8_t*      yuv_out_;          // output samples
2825d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  uint8_t*      yuv_out2_;         // secondary buffer swapped with yuv_out_.
2835d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  uint8_t*      yuv_p_;            // scratch buffer for prediction
2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  VP8Encoder*   enc_;              // back-pointer
2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  VP8MBInfo*    mb_;               // current macroblock
2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  VP8BitWriter* bw_;               // current bit-writer
2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint8_t*      preds_;            // intra mode predictors (4x4 blocks)
2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint32_t*     nz_;               // non-zero pattern
2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint8_t       i4_boundary_[37];  // 32+5 boundary samples needed by intra4x4
2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint8_t*      i4_top_;           // pointer to the current top boundary sample
2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int           i4_;               // current intra4x4 mode being tested
2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int           top_nz_[9];        // top-non-zero context.
2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int           left_nz_[9];       // left-non-zero. left_nz[8] is independent.
2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint64_t      bit_count_[4][3];  // bit counters for coded levels.
2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint64_t      luma_bits_;        // macroblock bit-cost for luma
2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint64_t      uv_bits_;          // macroblock bit-cost for chroma
2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  LFStats*      lf_stats_;         // filter stats (borrowed from enc_)
2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int           do_trellis_;       // if true, perform extra level optimisation
2995d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  int           count_down_;       // number of mb still to be processed
3005d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  int           count_down0_;      // starting counter value (for progress)
3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int           percent0_;         // saved initial progress percent
3025d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
3035d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  uint8_t* y_left_;    // left luma samples (addressable from index -1 to 15).
3045d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  uint8_t* u_left_;    // left u samples (addressable from index -1 to 7)
3055d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  uint8_t* v_left_;    // left v samples (addressable from index -1 to 7)
3065d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
3075d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  uint8_t* y_top_;     // top luma samples at position 'x_'
3085d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  uint8_t* uv_top_;    // top u/v samples at position 'x_', packed as 16 bytes
3095d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
3105d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  // memory for storing y/u/v_left_ and yuv_in_/out_*
3115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  uint8_t yuv_left_mem_[17 + 16 + 16 + 8 + ALIGN_CST];     // memory for *_left_
3125d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  uint8_t yuv_mem_[3 * YUV_SIZE + PRED_SIZE + ALIGN_CST];  // memory for yuv_*
3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} VP8EncIterator;
3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // in iterator.c
3165d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// must be called first
3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8IteratorInit(VP8Encoder* const enc, VP8EncIterator* const it);
3185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// restart a scan
3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8IteratorReset(VP8EncIterator* const it);
3205d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// reset iterator position to row 'y'
3215d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void VP8IteratorSetRow(VP8EncIterator* const it, int y);
3225d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// set count down (=number of iterations to go)
3235d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void VP8IteratorSetCountDown(VP8EncIterator* const it, int count_down);
3245d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// return true if iteration is finished
3255d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)int VP8IteratorIsDone(const VP8EncIterator* const it);
3265d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// Import uncompressed samples from source.
3275d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// If tmp_32 is not NULL, import boundary samples too.
3285d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// tmp_32 is a 32-bytes scratch buffer that must be aligned in memory.
3295d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void VP8IteratorImport(VP8EncIterator* const it, uint8_t* tmp_32);
3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// export decimated samples
3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8IteratorExport(const VP8EncIterator* const it);
3325d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// go to next macroblock. Returns false if not finished.
3335d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)int VP8IteratorNext(VP8EncIterator* const it);
3345d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// save the yuv_out_ boundary values to top_/left_ arrays for next iterations.
3355d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void VP8IteratorSaveBoundary(VP8EncIterator* const it);
3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Report progression based on macroblock rows. Return 0 for user-abort request.
3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int VP8IteratorProgress(const VP8EncIterator* const it,
3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                        int final_delta_percent);
3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Intra4x4 iterations
3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8IteratorStartI4(VP8EncIterator* const it);
3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// returns true if not done.
3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int VP8IteratorRotateI4(VP8EncIterator* const it,
3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                        const uint8_t* const yuv_out);
3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Non-zero context setup/teardown
3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8IteratorNzToBytes(VP8EncIterator* const it);
3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8IteratorBytesToNz(VP8EncIterator* const it);
3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Helper functions to set mode properties
3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8SetIntra16Mode(const VP8EncIterator* const it, int mode);
3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8SetIntra4Mode(const VP8EncIterator* const it, const uint8_t* modes);
3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8SetIntraUVMode(const VP8EncIterator* const it, int mode);
3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8SetSkip(const VP8EncIterator* const it, int skip);
3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8SetSegment(const VP8EncIterator* const it, int segment);
3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//------------------------------------------------------------------------------
3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Paginated token buffer
3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)typedef struct VP8Tokens VP8Tokens;  // struct details in token.c
3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef struct {
3622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#if !defined(DISABLE_TOKEN_BUFFER)
3632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  VP8Tokens* pages_;        // first page
3642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  VP8Tokens** last_page_;   // last page
3652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  uint16_t* tokens_;        // set to (*last_page_)->tokens_
3665f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  int left_;                // how many free tokens left before the page is full
3675f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  int page_size_;           // number of tokens per page
3682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif
3692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  int error_;         // true in case of malloc error
3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} VP8TBuffer;
3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3725f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)// initialize an empty buffer
3735f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)void VP8TBufferInit(VP8TBuffer* const b, int page_size);
3742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void VP8TBufferClear(VP8TBuffer* const b);   // de-allocate pages memory
3752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
3762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#if !defined(DISABLE_TOKEN_BUFFER)
3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Finalizes bitstream when probabilities are known.
3792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Deletes the allocated token memory if final_pass is true.
3802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)int VP8EmitTokens(VP8TBuffer* const b, VP8BitWriter* const bw,
3812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                  const uint8_t* const probas, int final_pass);
3822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
3832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// record the coding of coefficients without knowing the probabilities yet
3842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)int VP8RecordCoeffTokens(int ctx, int coeff_type, int first, int last,
3852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                         const int16_t* const coeffs,
3862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                         VP8TBuffer* const tokens);
3872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
3885d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// Estimate the final coded size given a set of 'probas'.
3895d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)size_t VP8EstimateTokenSize(VP8TBuffer* const b, const uint8_t* const probas);
3905d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
3912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// unused for now
3922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void VP8TokenToStats(const VP8TBuffer* const b, proba_t* const stats);
3932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
3942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif  // !DISABLE_TOKEN_BUFFER
3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//------------------------------------------------------------------------------
3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// VP8Encoder
3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)struct VP8Encoder {
4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const WebPConfig* config_;    // user configuration and parameters
4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  WebPPicture* pic_;            // input / output picture
4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // headers
4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  VP8FilterHeader   filter_hdr_;     // filtering information
4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  VP8SegmentHeader  segment_hdr_;    // segment information
4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int profile_;                      // VP8's profile, deduced from Config.
4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // dimension, in macroblock units.
4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int mb_w_, mb_h_;
4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int preds_w_;   // stride of the *preds_ prediction plane (=4*mb_w + 1)
4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // number of partitions (1, 2, 4 or 8 = MAX_NUM_PARTITIONS)
4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int num_parts_;
4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // per-partition boolean decoders.
4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  VP8BitWriter bw_;                         // part0
4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  VP8BitWriter parts_[MAX_NUM_PARTITIONS];  // token partitions
4192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  VP8TBuffer tokens_;                       // token buffer
4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int percent_;                             // for progress
4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // transparency blob
4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int has_alpha_;
4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint8_t* alpha_data_;       // non-NULL if transparency is present
4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint32_t alpha_data_size_;
4272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  WebPWorker alpha_worker_;
4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // quantization info (one set of DC/AC dequant factor per segment)
4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  VP8SegmentInfo dqm_[NUM_MB_SEGMENTS];
4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int base_quant_;                 // nominal quantizer value. Only used
4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                   // for relative coding of segments' quant.
4332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  int alpha_;                      // global susceptibility (<=> complexity)
4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int uv_alpha_;                   // U/V quantization susceptibility
4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // global offset of quantizers, shared by all segments
4365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int dq_y1_dc_;
4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int dq_y2_dc_, dq_y2_ac_;
4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int dq_uv_dc_, dq_uv_ac_;
4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // probabilities and statistics
4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  VP8Proba proba_;
4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint64_t sse_[4];        // sum of Y/U/V/A squared errors for all macroblocks
4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint64_t sse_count_;     // pixel count for the sse_[] stats
4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int      coded_size_;
4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int      residual_bytes_[3][4];
4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int      block_count_[3];
4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // quality/speed settings
4492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  int method_;               // 0=fastest, 6=best/slowest.
4502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  VP8RDLevel rd_opt_level_;  // Deduced from method_.
4512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  int max_i4_header_bits_;   // partition #0 safeness factor
4522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  int thread_level_;         // derived from config->thread_level
4532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  int do_search_;            // derived from config->target_XXX
4542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  int use_tokens_;           // if true, use token buffer
4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Memory
4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  VP8MBInfo* mb_info_;   // contextual macroblock infos (mb_w_ + 1)
4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint8_t*   preds_;     // predictions modes: (4*mb_w+1) * (4*mb_h+1)
4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint32_t*  nz_;        // non-zero bit context: mb_w+1
4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint8_t   *y_top_;     // top luma samples.
4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint8_t   *uv_top_;    // top u/v samples.
4625d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)                         // U and V are packed into 16 bytes (8 U + 8 V)
4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  LFStats   *lf_stats_;  // autofilter stats (if NULL, autofilter is off)
4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)};
4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//------------------------------------------------------------------------------
4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// internal functions. Not public.
4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // in tree.c
4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern const uint8_t VP8CoeffsProba0[NUM_TYPES][NUM_BANDS][NUM_CTX][NUM_PROBAS];
4715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern const uint8_t
4725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    VP8CoeffsUpdateProba[NUM_TYPES][NUM_BANDS][NUM_CTX][NUM_PROBAS];
4735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Reset the token probabilities to their initial (default) values
4745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8DefaultProbas(VP8Encoder* const enc);
4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Write the token probabilities
4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8WriteProbas(VP8BitWriter* const bw, const VP8Proba* const probas);
4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Writes the partition #0 modes (that is: all intra modes)
4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8CodeIntraModes(VP8Encoder* const enc);
4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // in syntax.c
4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Generates the final bitstream by coding the partition0 and headers,
4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// and appending an assembly of all the pre-coded token partitions.
4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Return true if everything is ok.
4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int VP8EncWrite(VP8Encoder* const enc);
4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Release memory allocated for bit-writing in VP8EncLoop & seq.
4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8EncFreeBitWriters(VP8Encoder* const enc);
4875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // in frame.c
4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern const uint8_t VP8EncBands[16 + 1];
4902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)extern const uint8_t VP8Cat3[];
4912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)extern const uint8_t VP8Cat4[];
4922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)extern const uint8_t VP8Cat5[];
4932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)extern const uint8_t VP8Cat6[];
4942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Form all the four Intra16x16 predictions in the yuv_p_ cache
4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8MakeLuma16Preds(const VP8EncIterator* const it);
4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Form all the four Chroma8x8 predictions in the yuv_p_ cache
4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8MakeChroma8Preds(const VP8EncIterator* const it);
4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Form all the ten Intra4x4 predictions in the yuv_p_ cache
5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// for the 4x4 block it->i4_
5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8MakeIntra4Preds(const VP8EncIterator* const it);
5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Rate calculation
5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int VP8GetCostLuma16(VP8EncIterator* const it, const VP8ModeScore* const rd);
5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int VP8GetCostLuma4(VP8EncIterator* const it, const int16_t levels[16]);
5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int VP8GetCostUV(VP8EncIterator* const it, const VP8ModeScore* const rd);
5062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Main coding calls
5075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int VP8EncLoop(VP8Encoder* const enc);
5082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)int VP8EncTokenLoop(VP8Encoder* const enc);
5095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // in webpenc.c
5115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Assign an error code to a picture. Return false for convenience.
5125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int WebPEncodingSetError(const WebPPicture* const pic, WebPEncodingError error);
5135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int WebPReportProgress(const WebPPicture* const pic,
5145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       int percent, int* const percent_store);
5155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // in analysis.c
5175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Main analysis loop. Decides the segmentations and complexity.
5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Assigns a first guess for Intra16 and uvmode_ prediction modes.
5195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int VP8EncAnalyze(VP8Encoder* const enc);
5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // in quant.c
5225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Sets up segment's quantization values, base_quant_ and filter strengths.
5235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8SetSegmentParams(VP8Encoder* const enc, float quality);
5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Pick best modes and fills the levels. Returns true if skipped.
5252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)int VP8Decimate(VP8EncIterator* const it, VP8ModeScore* const rd,
5262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                VP8RDLevel rd_opt);
5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // in alpha.c
5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8EncInitAlpha(VP8Encoder* const enc);    // initialize alpha compression
5302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)int VP8EncStartAlpha(VP8Encoder* const enc);    // start alpha coding process
5315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int VP8EncFinishAlpha(VP8Encoder* const enc);   // finalize compressed data
5322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)int VP8EncDeleteAlpha(VP8Encoder* const enc);   // delete compressed data
5335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // in filter.c
5355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// SSIM utils
5375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef struct {
5385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  double w, xm, ym, xxm, xym, yym;
5395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} DistoStats;
5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8SSIMAddStats(const DistoStats* const src, DistoStats* const dst);
5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8SSIMAccumulatePlane(const uint8_t* src1, int stride1,
5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                            const uint8_t* src2, int stride2,
5435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                            int W, int H, DistoStats* const stats);
5445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)double VP8SSIMGet(const DistoStats* const stats);
5455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)double VP8SSIMGetSquaredError(const DistoStats* const stats);
5465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// autofilter
5485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8InitFilter(VP8EncIterator* const it);
5495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8StoreFilterStats(VP8EncIterator* const it);
5505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VP8AdjustFilterStrength(VP8EncIterator* const it);
5515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5525d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// returns the approximate filtering strength needed to smooth a edge
5535d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// step of 'delta', given a sharpness parameter 'sharpness'.
5545d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)int VP8FilterStrengthFromDelta(int sharpness, int delta);
5555d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
5565f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  // misc utils for picture_*.c:
5575f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)
5585f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)// Remove reference to the ARGB/YUVA buffer (doesn't free anything).
5595f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)void WebPPictureResetBuffers(WebPPicture* const picture);
5605f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)
5615f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)// Allocates ARGB buffer of given dimension (previous one is always free'd).
5625f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)// Preserves the YUV(A) buffer. Returns false in case of error (invalid param,
5635f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)// out-of-memory).
5645f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)int WebPPictureAllocARGB(WebPPicture* const picture, int width, int height);
5655f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)
5665f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)// Allocates YUVA buffer of given dimension (previous one is always free'd).
5675f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)// Uses picture->csp to determine whether an alpha buffer is needed.
5685f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)// Preserves the ARGB buffer.
5695f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)// Returns false in case of error (invalid param, out-of-memory).
5705f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)int WebPPictureAllocYUVA(WebPPicture* const picture, int width, int height);
5715f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)
5725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//------------------------------------------------------------------------------
5735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5745f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)#if WEBP_ENCODER_ABI_VERSION <= 0x0202
5755f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)void WebPMemoryWriterClear(WebPMemoryWriter* writer);
5765f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)#endif
5775f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)
5785d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#ifdef __cplusplus
5795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}    // extern "C"
5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
5815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif  /* WEBP_ENC_VP8ENCI_H_ */
583