1a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora// Copyright 2011 Google Inc. All Rights Reserved. 27c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// 30406ce1417f76f2034833414dcecc9f56253640cVikas Arora// Use of this source code is governed by a BSD-style license 40406ce1417f76f2034833414dcecc9f56253640cVikas Arora// that can be found in the COPYING file in the root of the source 50406ce1417f76f2034833414dcecc9f56253640cVikas Arora// tree. An additional intellectual property rights grant can be found 60406ce1417f76f2034833414dcecc9f56253640cVikas Arora// in the file PATENTS. All contributing project authors may 70406ce1417f76f2034833414dcecc9f56253640cVikas Arora// be found in the AUTHORS file in the root of the source tree. 87c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// ----------------------------------------------------------------------------- 97c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// 107c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// WebP encoder: internal header. 117c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// 127c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Author: Skal (pascal.massimino@gmail.com) 137c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 14a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zern#ifndef WEBP_ENC_VP8I_ENC_H_ 15a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zern#define WEBP_ENC_VP8I_ENC_H_ 167c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 17a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora#include <string.h> // for memcpy() 18a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zern#include "src/dec/common_dec.h" 19a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zern#include "src/dsp/dsp.h" 20a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zern#include "src/utils/bit_writer_utils.h" 21a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zern#include "src/utils/thread_utils.h" 22a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zern#include "src/utils/utils.h" 23a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zern#include "src/webp/encode.h" 247c8da7ce66017295a65ec028084b90800be377f8James Zern 258b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora#ifdef __cplusplus 267c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroraextern "C" { 277c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora#endif 287c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 29a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 307c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Various defines and enums 317c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 327c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// version numbers 337c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora#define ENC_MAJ_VERSION 0 34fa39824bb690c5806358871f46940d0450973d8aJames Zern#define ENC_MIN_VERSION 6 35a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zern#define ENC_REV_VERSION 1 367c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 377c8da7ce66017295a65ec028084b90800be377f8James Zernenum { MAX_LF_LEVELS = 64, // Maximum loop filter level 380406ce1417f76f2034833414dcecc9f56253640cVikas Arora MAX_VARIABLE_LEVEL = 67, // last (inclusive) level with variable cost 390406ce1417f76f2034833414dcecc9f56253640cVikas Arora MAX_LEVEL = 2047 // max level (note: max codable is 2047 + 67) 407c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora }; 417c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 421e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroratypedef enum { // Rate-distortion optimization levels 431e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora RD_OPT_NONE = 0, // no rd-opt 441e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora RD_OPT_BASIC = 1, // basic scoring (no trellis) 451e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora RD_OPT_TRELLIS = 2, // perform trellis-quant on the final decision only 461e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora RD_OPT_TRELLIS_ALL = 3 // trellis-quant for every scoring (much slower) 471e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora} VP8RDLevel; 481e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 497c8da7ce66017295a65ec028084b90800be377f8James Zern// YUV-cache parameters. Cache is 32-bytes wide (= one cacheline). 507c8da7ce66017295a65ec028084b90800be377f8James Zern// The original or reconstructed samples can be accessed using VP8Scan[]. 517c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// The predicted blocks can be accessed using offsets to yuv_p_ and 527c8da7ce66017295a65ec028084b90800be377f8James Zern// the arrays VP8*ModeOffsets[]. 537c8da7ce66017295a65ec028084b90800be377f8James Zern// * YUV Samples area (yuv_in_/yuv_out_/yuv_out2_) 547c8da7ce66017295a65ec028084b90800be377f8James Zern// (see VP8Scan[] for accessing the blocks, along with 557c8da7ce66017295a65ec028084b90800be377f8James Zern// Y_OFF_ENC/U_OFF_ENC/V_OFF_ENC): 567c8da7ce66017295a65ec028084b90800be377f8James Zern// +----+----+ 577c8da7ce66017295a65ec028084b90800be377f8James Zern// Y_OFF_ENC |YYYY|UUVV| 587c8da7ce66017295a65ec028084b90800be377f8James Zern// U_OFF_ENC |YYYY|UUVV| 597c8da7ce66017295a65ec028084b90800be377f8James Zern// V_OFF_ENC |YYYY|....| <- 25% wasted U/V area 607c8da7ce66017295a65ec028084b90800be377f8James Zern// |YYYY|....| 617c8da7ce66017295a65ec028084b90800be377f8James Zern// +----+----+ 627c8da7ce66017295a65ec028084b90800be377f8James Zern// * Prediction area ('yuv_p_', size = PRED_SIZE_ENC) 637c8da7ce66017295a65ec028084b90800be377f8James Zern// Intra16 predictions (16x16 block each, two per row): 647c8da7ce66017295a65ec028084b90800be377f8James Zern// |I16DC16|I16TM16| 657c8da7ce66017295a65ec028084b90800be377f8James Zern// |I16VE16|I16HE16| 667c8da7ce66017295a65ec028084b90800be377f8James Zern// Chroma U/V predictions (16x8 block each, two per row): 677c8da7ce66017295a65ec028084b90800be377f8James Zern// |C8DC8|C8TM8| 687c8da7ce66017295a65ec028084b90800be377f8James Zern// |C8VE8|C8HE8| 697c8da7ce66017295a65ec028084b90800be377f8James Zern// Intra 4x4 predictions (4x4 block each) 707c8da7ce66017295a65ec028084b90800be377f8James Zern// |I4DC4 I4TM4 I4VE4 I4HE4|I4RD4 I4VR4 I4LD4 I4VL4| 717c8da7ce66017295a65ec028084b90800be377f8James Zern// |I4HD4 I4HU4 I4TMP .....|.......................| <- ~31% wasted 727c8da7ce66017295a65ec028084b90800be377f8James Zern#define YUV_SIZE_ENC (BPS * 16) 737c8da7ce66017295a65ec028084b90800be377f8James Zern#define PRED_SIZE_ENC (32 * BPS + 16 * BPS + 8 * BPS) // I16+Chroma+I4 preds 747c8da7ce66017295a65ec028084b90800be377f8James Zern#define Y_OFF_ENC (0) 757c8da7ce66017295a65ec028084b90800be377f8James Zern#define U_OFF_ENC (16) 767c8da7ce66017295a65ec028084b90800be377f8James Zern#define V_OFF_ENC (16 + 8) 777c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 78a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zernextern const uint16_t VP8Scan[16]; 79a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zernextern const uint16_t VP8UVModeOffsets[4]; 80a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zernextern const uint16_t VP8I16ModeOffsets[4]; 81a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zernextern const uint16_t VP8I4ModeOffsets[NUM_BMODES]; 827c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 837c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Layout of prediction blocks 847c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// intra 16x16 857c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora#define I16DC16 (0 * 16 * BPS) 867c8da7ce66017295a65ec028084b90800be377f8James Zern#define I16TM16 (I16DC16 + 16) 877c8da7ce66017295a65ec028084b90800be377f8James Zern#define I16VE16 (1 * 16 * BPS) 887c8da7ce66017295a65ec028084b90800be377f8James Zern#define I16HE16 (I16VE16 + 16) 897c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// chroma 8x8, two U/V blocks side by side (hence: 16x8 each) 907c8da7ce66017295a65ec028084b90800be377f8James Zern#define C8DC8 (2 * 16 * BPS) 917c8da7ce66017295a65ec028084b90800be377f8James Zern#define C8TM8 (C8DC8 + 1 * 16) 927c8da7ce66017295a65ec028084b90800be377f8James Zern#define C8VE8 (2 * 16 * BPS + 8 * BPS) 937c8da7ce66017295a65ec028084b90800be377f8James Zern#define C8HE8 (C8VE8 + 1 * 16) 947c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// intra 4x4 957c8da7ce66017295a65ec028084b90800be377f8James Zern#define I4DC4 (3 * 16 * BPS + 0) 967c8da7ce66017295a65ec028084b90800be377f8James Zern#define I4TM4 (I4DC4 + 4) 977c8da7ce66017295a65ec028084b90800be377f8James Zern#define I4VE4 (I4DC4 + 8) 987c8da7ce66017295a65ec028084b90800be377f8James Zern#define I4HE4 (I4DC4 + 12) 997c8da7ce66017295a65ec028084b90800be377f8James Zern#define I4RD4 (I4DC4 + 16) 1007c8da7ce66017295a65ec028084b90800be377f8James Zern#define I4VR4 (I4DC4 + 20) 1017c8da7ce66017295a65ec028084b90800be377f8James Zern#define I4LD4 (I4DC4 + 24) 1027c8da7ce66017295a65ec028084b90800be377f8James Zern#define I4VL4 (I4DC4 + 28) 1037c8da7ce66017295a65ec028084b90800be377f8James Zern#define I4HD4 (3 * 16 * BPS + 4 * BPS) 1047c8da7ce66017295a65ec028084b90800be377f8James Zern#define I4HU4 (I4HD4 + 4) 1057c8da7ce66017295a65ec028084b90800be377f8James Zern#define I4TMP (I4HD4 + 8) 1067c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1077c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroratypedef int64_t score_t; // type used for scores, rate, distortion 10833f74dabbc7920a65ed435d7417987589febdc16Vikas Arora// Note that MAX_COST is not the maximum allowed by sizeof(score_t), 10933f74dabbc7920a65ed435d7417987589febdc16Vikas Arora// in order to allow overflowing computations. 1107c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora#define MAX_COST ((score_t)0x7fffffffffffffLL) 1117c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1127c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora#define QFIX 17 1137c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora#define BIAS(b) ((b) << (QFIX - 8)) 1147c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Fun fact: this is the _only_ line where we're actually being lossy and 1157c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// discarding bits. 11633f74dabbc7920a65ed435d7417987589febdc16Vikas Arorastatic WEBP_INLINE int QUANTDIV(uint32_t n, uint32_t iQ, uint32_t B) { 11733f74dabbc7920a65ed435d7417987589febdc16Vikas Arora return (int)((n * iQ + B) >> QFIX); 1187c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 1191e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 1201e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// Uncomment the following to remove token-buffer code: 1211e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// #define DISABLE_TOKEN_BUFFER 1227c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 123a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 1247c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Headers 1257c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 126a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroratypedef uint32_t proba_t; // 16b + 16b 1277c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroratypedef uint8_t ProbaArray[NUM_CTX][NUM_PROBAS]; 128a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroratypedef proba_t StatsArray[NUM_CTX][NUM_PROBAS]; 1297c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroratypedef uint16_t CostArray[NUM_CTX][MAX_VARIABLE_LEVEL + 1]; 1307c8da7ce66017295a65ec028084b90800be377f8James Zerntypedef const uint16_t* (*CostArrayPtr)[NUM_CTX]; // for easy casting 1317c8da7ce66017295a65ec028084b90800be377f8James Zerntypedef const uint16_t* CostArrayMap[16][NUM_CTX]; 1327c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroratypedef double LFStats[NUM_MB_SEGMENTS][MAX_LF_LEVELS]; // filter stats 1337c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1347c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroratypedef struct VP8Encoder VP8Encoder; 1357c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1367c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// segment features 1377c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroratypedef struct { 1387c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int num_segments_; // Actual number of segments. 1 segment only = unused. 1397c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int update_map_; // whether to update the segment map or not. 1407c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // must be 0 if there's only 1 segment. 1417c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int size_; // bit-cost for transmitting the segment map 1427c8da7ce66017295a65ec028084b90800be377f8James Zern} VP8EncSegmentHeader; 1437c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1447c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Struct collecting all frame-persistent probabilities. 1457c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroratypedef struct { 1467c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint8_t segments_[3]; // probabilities for segment tree 1477c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint8_t skip_proba_; // final probability of being skipped. 14833f74dabbc7920a65ed435d7417987589febdc16Vikas Arora ProbaArray coeffs_[NUM_TYPES][NUM_BANDS]; // 1056 bytes 149a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora StatsArray stats_[NUM_TYPES][NUM_BANDS]; // 4224 bytes 15033f74dabbc7920a65ed435d7417987589febdc16Vikas Arora CostArray level_cost_[NUM_TYPES][NUM_BANDS]; // 13056 bytes 1517c8da7ce66017295a65ec028084b90800be377f8James Zern CostArrayMap remapped_costs_[NUM_TYPES]; // 1536 bytes 152a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora int dirty_; // if true, need to call VP8CalculateLevelCosts() 1537c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int use_skip_proba_; // Note: we always use skip_proba for now. 15488fe2b83c4b9232cd08729556fd0485d6a6a92cdVikas Arora int nb_skip_; // number of skipped blocks 1557c8da7ce66017295a65ec028084b90800be377f8James Zern} VP8EncProba; 1567c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1577c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Filter parameters. Not actually used in the code (we don't perform 1587c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// the in-loop filtering), but filled from user's config 1597c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroratypedef struct { 1607c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int simple_; // filtering type: 0=complex, 1=simple 1617c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int level_; // base filter level [0..63] 1627c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int sharpness_; // [0..7] 1637c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int i4x4_lf_delta_; // delta filter level for i4x4 relative to i16x16 1647c8da7ce66017295a65ec028084b90800be377f8James Zern} VP8EncFilterHeader; 1657c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 166a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 1677c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Informations about the macroblocks. 1687c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1697c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroratypedef struct { 1707c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // block type 171a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora unsigned int type_:2; // 0=i4x4, 1=i16x16 172a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora unsigned int uv_mode_:2; 173a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora unsigned int skip_:1; 174a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora unsigned int segment_:2; 1757c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint8_t alpha_; // quantization-susceptibility 1767c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} VP8MBInfo; 1777c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 178a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroratypedef struct VP8Matrix { 1797c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint16_t q_[16]; // quantizer steps 1807c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint16_t iq_[16]; // reciprocals, fixed point. 18133f74dabbc7920a65ed435d7417987589febdc16Vikas Arora uint32_t bias_[16]; // rounding bias 18233f74dabbc7920a65ed435d7417987589febdc16Vikas Arora uint32_t zthresh_[16]; // value below which a coefficient is zeroed 1837c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint16_t sharpen_[16]; // frequency boosters for slight sharpening 1847c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} VP8Matrix; 1857c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1867c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroratypedef struct { 1877c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8Matrix y1_, y2_, uv_; // quantization matrices 1887c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int alpha_; // quant-susceptibility, range [-127,127]. Zero is neutral. 1897c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // Lower values indicate a lower risk of blurriness. 1907c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int beta_; // filter-susceptibility, range [0,255]. 1917c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int quant_; // final segment quantizer. 1927c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int fstrength_; // final in-loop filtering strength 1938b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora int max_edge_; // max edge delta (for filtering strength) 1948b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora int min_disto_; // minimum distortion required to trigger filtering record 1957c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // reactivities 1967c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int lambda_i16_, lambda_i4_, lambda_uv_; 1977c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int lambda_mode_, lambda_trellis_, tlambda_; 1987c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int lambda_trellis_i16_, lambda_trellis_i4_, lambda_trellis_uv_; 1990912efc2528d03c59d45dd9bdc9ff9ec800a3fc1James Zern 2000912efc2528d03c59d45dd9bdc9ff9ec800a3fc1James Zern // lambda values for distortion-based evaluation 2010912efc2528d03c59d45dd9bdc9ff9ec800a3fc1James Zern score_t i4_penalty_; // penalty for using Intra4 2027c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} VP8SegmentInfo; 2037c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 2048b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// Handy transient struct to accumulate score and info during RD-optimization 2057c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// and mode evaluation. 2067c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroratypedef struct { 2078b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora score_t D, SD; // Distortion, spectral distortion 2088b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora score_t H, R, score; // header bits, rate, score. 2097c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int16_t y_dc_levels[16]; // Quantized levels for luma-DC, luma-AC, chroma. 2107c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int16_t y_ac_levels[16][16]; 2117c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int16_t uv_levels[4 + 4][16]; 2127c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int mode_i16; // mode number for intra16 prediction 213a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora uint8_t modes_i4[16]; // mode numbers for intra4 predictions 2147c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int mode_uv; // mode number of chroma prediction 2157c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint32_t nz; // non-zero blocks 2167c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} VP8ModeScore; 2177c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 2187c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Iterator structure to iterate through macroblocks, pointing to the 2197c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// right neighbouring data (samples, predictions, contexts, ...) 2207c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroratypedef struct { 2217c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int x_, y_; // current macroblock 2228b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora uint8_t* yuv_in_; // input samples 2238b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora uint8_t* yuv_out_; // output samples 2248b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora uint8_t* yuv_out2_; // secondary buffer swapped with yuv_out_. 2258b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora uint8_t* yuv_p_; // scratch buffer for prediction 2267c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8Encoder* enc_; // back-pointer 2277c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8MBInfo* mb_; // current macroblock 2287c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8BitWriter* bw_; // current bit-writer 2297c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint8_t* preds_; // intra mode predictors (4x4 blocks) 2307c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint32_t* nz_; // non-zero pattern 2317c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint8_t i4_boundary_[37]; // 32+5 boundary samples needed by intra4x4 232466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora uint8_t* i4_top_; // pointer to the current top boundary sample 2337c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int i4_; // current intra4x4 mode being tested 2347c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int top_nz_[9]; // top-non-zero context. 2357c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int left_nz_[9]; // left-non-zero. left_nz[8] is independent. 2367c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint64_t bit_count_[4][3]; // bit counters for coded levels. 2377c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint64_t luma_bits_; // macroblock bit-cost for luma 2387c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint64_t uv_bits_; // macroblock bit-cost for chroma 2397c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora LFStats* lf_stats_; // filter stats (borrowed from enc_) 2407c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int do_trellis_; // if true, perform extra level optimisation 2418b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora int count_down_; // number of mb still to be processed 2428b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora int count_down0_; // starting counter value (for progress) 243a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora int percent0_; // saved initial progress percent 2448b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora 2458b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora uint8_t* y_left_; // left luma samples (addressable from index -1 to 15). 2468b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora uint8_t* u_left_; // left u samples (addressable from index -1 to 7) 2478b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora uint8_t* v_left_; // left v samples (addressable from index -1 to 7) 2488b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora 2498b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora uint8_t* y_top_; // top luma samples at position 'x_' 2508b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora uint8_t* uv_top_; // top u/v samples at position 'x_', packed as 16 bytes 2518b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora 2527c8da7ce66017295a65ec028084b90800be377f8James Zern // memory for storing y/u/v_left_ 2537c8da7ce66017295a65ec028084b90800be377f8James Zern uint8_t yuv_left_mem_[17 + 16 + 16 + 8 + WEBP_ALIGN_CST]; 2547c8da7ce66017295a65ec028084b90800be377f8James Zern // memory for yuv_* 2557c8da7ce66017295a65ec028084b90800be377f8James Zern uint8_t yuv_mem_[3 * YUV_SIZE_ENC + PRED_SIZE_ENC + WEBP_ALIGN_CST]; 2567c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} VP8EncIterator; 2577c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 2587c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // in iterator.c 2598b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// must be called first 2607c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8IteratorInit(VP8Encoder* const enc, VP8EncIterator* const it); 2618b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// restart a scan 2627c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8IteratorReset(VP8EncIterator* const it); 2638b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// reset iterator position to row 'y' 2648b720228d581a84fd173b6dcb2fa295b59db489aVikas Aroravoid VP8IteratorSetRow(VP8EncIterator* const it, int y); 2658b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// set count down (=number of iterations to go) 2668b720228d581a84fd173b6dcb2fa295b59db489aVikas Aroravoid VP8IteratorSetCountDown(VP8EncIterator* const it, int count_down); 2678b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// return true if iteration is finished 2688b720228d581a84fd173b6dcb2fa295b59db489aVikas Aroraint VP8IteratorIsDone(const VP8EncIterator* const it); 2698b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// Import uncompressed samples from source. 2708b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// If tmp_32 is not NULL, import boundary samples too. 2718b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// tmp_32 is a 32-bytes scratch buffer that must be aligned in memory. 2728b720228d581a84fd173b6dcb2fa295b59db489aVikas Aroravoid VP8IteratorImport(VP8EncIterator* const it, uint8_t* tmp_32); 2737c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// export decimated samples 2747c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8IteratorExport(const VP8EncIterator* const it); 2758b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// go to next macroblock. Returns false if not finished. 2768b720228d581a84fd173b6dcb2fa295b59db489aVikas Aroraint VP8IteratorNext(VP8EncIterator* const it); 2778b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// save the yuv_out_ boundary values to top_/left_ arrays for next iterations. 2788b720228d581a84fd173b6dcb2fa295b59db489aVikas Aroravoid VP8IteratorSaveBoundary(VP8EncIterator* const it); 279a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora// Report progression based on macroblock rows. Return 0 for user-abort request. 280a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroraint VP8IteratorProgress(const VP8EncIterator* const it, 281a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora int final_delta_percent); 2827c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Intra4x4 iterations 2837c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8IteratorStartI4(VP8EncIterator* const it); 2847c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// returns true if not done. 2857c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroraint VP8IteratorRotateI4(VP8EncIterator* const it, 2867c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const uint8_t* const yuv_out); 2877c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 2887c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Non-zero context setup/teardown 2897c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8IteratorNzToBytes(VP8EncIterator* const it); 2907c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8IteratorBytesToNz(VP8EncIterator* const it); 2917c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 2927c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Helper functions to set mode properties 2937c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8SetIntra16Mode(const VP8EncIterator* const it, int mode); 294a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroravoid VP8SetIntra4Mode(const VP8EncIterator* const it, const uint8_t* modes); 2957c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8SetIntraUVMode(const VP8EncIterator* const it, int mode); 2967c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8SetSkip(const VP8EncIterator* const it, int skip); 2977c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8SetSegment(const VP8EncIterator* const it, int segment); 2987c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 299a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 300a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora// Paginated token buffer 301a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora 3021e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroratypedef struct VP8Tokens VP8Tokens; // struct details in token.c 303a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora 304a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroratypedef struct { 3051e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#if !defined(DISABLE_TOKEN_BUFFER) 3061e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora VP8Tokens* pages_; // first page 3071e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora VP8Tokens** last_page_; // last page 3081e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora uint16_t* tokens_; // set to (*last_page_)->tokens_ 30933f74dabbc7920a65ed435d7417987589febdc16Vikas Arora int left_; // how many free tokens left before the page is full 31033f74dabbc7920a65ed435d7417987589febdc16Vikas Arora int page_size_; // number of tokens per page 3111e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#endif 3121e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int error_; // true in case of malloc error 313a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora} VP8TBuffer; 314a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora 31533f74dabbc7920a65ed435d7417987589febdc16Vikas Arora// initialize an empty buffer 31633f74dabbc7920a65ed435d7417987589febdc16Vikas Aroravoid VP8TBufferInit(VP8TBuffer* const b, int page_size); 3171e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroravoid VP8TBufferClear(VP8TBuffer* const b); // de-allocate pages memory 318a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora 3191e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#if !defined(DISABLE_TOKEN_BUFFER) 3201e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 3211e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// Finalizes bitstream when probabilities are known. 3221e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// Deletes the allocated token memory if final_pass is true. 3231e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroraint VP8EmitTokens(VP8TBuffer* const b, VP8BitWriter* const bw, 3241e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const uint8_t* const probas, int final_pass); 3251e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 3261e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// record the coding of coefficients without knowing the probabilities yet 32798a63a77eb8652c81d64b5b7c3d8a347111807caJames Zernint VP8RecordCoeffTokens(int ctx, const struct VP8Residual* const res, 3281e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora VP8TBuffer* const tokens); 3291e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 3308b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// Estimate the final coded size given a set of 'probas'. 3318b720228d581a84fd173b6dcb2fa295b59db489aVikas Arorasize_t VP8EstimateTokenSize(VP8TBuffer* const b, const uint8_t* const probas); 3328b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora 3331e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#endif // !DISABLE_TOKEN_BUFFER 334a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora 335a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 3367c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// VP8Encoder 3377c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3387c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastruct VP8Encoder { 3397c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const WebPConfig* config_; // user configuration and parameters 3407c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora WebPPicture* pic_; // input / output picture 3417c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3427c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // headers 3437c8da7ce66017295a65ec028084b90800be377f8James Zern VP8EncFilterHeader filter_hdr_; // filtering information 3447c8da7ce66017295a65ec028084b90800be377f8James Zern VP8EncSegmentHeader segment_hdr_; // segment information 3457c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3467c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int profile_; // VP8's profile, deduced from Config. 3477c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3487c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // dimension, in macroblock units. 3497c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int mb_w_, mb_h_; 3507c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int preds_w_; // stride of the *preds_ prediction plane (=4*mb_w + 1) 3517c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3527c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // number of partitions (1, 2, 4 or 8 = MAX_NUM_PARTITIONS) 3537c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int num_parts_; 3547c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3557c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // per-partition boolean decoders. 3567c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8BitWriter bw_; // part0 3577c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8BitWriter parts_[MAX_NUM_PARTITIONS]; // token partitions 3581e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora VP8TBuffer tokens_; // token buffer 3597c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 360a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora int percent_; // for progress 361a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora 362466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // transparency blob 363466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int has_alpha_; 364466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora uint8_t* alpha_data_; // non-NULL if transparency is present 365a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora uint32_t alpha_data_size_; 3661e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora WebPWorker alpha_worker_; 367466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 3687c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // quantization info (one set of DC/AC dequant factor per segment) 3697c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8SegmentInfo dqm_[NUM_MB_SEGMENTS]; 3707c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int base_quant_; // nominal quantizer value. Only used 3717c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // for relative coding of segments' quant. 3721e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int alpha_; // global susceptibility (<=> complexity) 3737c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int uv_alpha_; // U/V quantization susceptibility 3747c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // global offset of quantizers, shared by all segments 3757c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int dq_y1_dc_; 3767c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int dq_y2_dc_, dq_y2_ac_; 3777c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int dq_uv_dc_, dq_uv_ac_; 3787c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3797c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // probabilities and statistics 3807c8da7ce66017295a65ec028084b90800be377f8James Zern VP8EncProba proba_; 3817c8da7ce66017295a65ec028084b90800be377f8James Zern uint64_t sse_[4]; // sum of Y/U/V/A squared errors for all macroblocks 3827c8da7ce66017295a65ec028084b90800be377f8James Zern uint64_t sse_count_; // pixel count for the sse_[] stats 3837c8da7ce66017295a65ec028084b90800be377f8James Zern int coded_size_; 3847c8da7ce66017295a65ec028084b90800be377f8James Zern int residual_bytes_[3][4]; 3857c8da7ce66017295a65ec028084b90800be377f8James Zern int block_count_[3]; 3867c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3877c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // quality/speed settings 3881e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int method_; // 0=fastest, 6=best/slowest. 3891e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora VP8RDLevel rd_opt_level_; // Deduced from method_. 3901e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int max_i4_header_bits_; // partition #0 safeness factor 3910912efc2528d03c59d45dd9bdc9ff9ec800a3fc1James Zern int mb_header_limit_; // rough limit for header bits per MB 3921e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int thread_level_; // derived from config->thread_level 3931e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int do_search_; // derived from config->target_XXX 3941e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora int use_tokens_; // if true, use token buffer 3957c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3967c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // Memory 3977c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8MBInfo* mb_info_; // contextual macroblock infos (mb_w_ + 1) 3987c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint8_t* preds_; // predictions modes: (4*mb_w+1) * (4*mb_h+1) 3997c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint32_t* nz_; // non-zero bit context: mb_w+1 4008c098653157979e397d3954fc2ea0ee43bae6ab2Vikas Arora uint8_t* y_top_; // top luma samples. 4018c098653157979e397d3954fc2ea0ee43bae6ab2Vikas Arora uint8_t* uv_top_; // top u/v samples. 4028b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora // U and V are packed into 16 bytes (8 U + 8 V) 4038c098653157979e397d3954fc2ea0ee43bae6ab2Vikas Arora LFStats* lf_stats_; // autofilter stats (if NULL, autofilter is off) 4047c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora}; 4057c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 406a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 4077c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// internal functions. Not public. 4087c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 4097c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // in tree.c 4107c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroraextern const uint8_t VP8CoeffsProba0[NUM_TYPES][NUM_BANDS][NUM_CTX][NUM_PROBAS]; 4117c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroraextern const uint8_t 4127c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8CoeffsUpdateProba[NUM_TYPES][NUM_BANDS][NUM_CTX][NUM_PROBAS]; 4137c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Reset the token probabilities to their initial (default) values 4147c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8DefaultProbas(VP8Encoder* const enc); 4157c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Write the token probabilities 4167c8da7ce66017295a65ec028084b90800be377f8James Zernvoid VP8WriteProbas(VP8BitWriter* const bw, const VP8EncProba* const probas); 4177c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Writes the partition #0 modes (that is: all intra modes) 4187c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8CodeIntraModes(VP8Encoder* const enc); 4197c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 4207c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // in syntax.c 4217c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Generates the final bitstream by coding the partition0 and headers, 4227c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// and appending an assembly of all the pre-coded token partitions. 4237c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Return true if everything is ok. 4247c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroraint VP8EncWrite(VP8Encoder* const enc); 425a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora// Release memory allocated for bit-writing in VP8EncLoop & seq. 426a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroravoid VP8EncFreeBitWriters(VP8Encoder* const enc); 4277c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 4287c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // in frame.c 4291e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroraextern const uint8_t VP8Cat3[]; 4301e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroraextern const uint8_t VP8Cat4[]; 4311e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroraextern const uint8_t VP8Cat5[]; 4321e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroraextern const uint8_t VP8Cat6[]; 4331e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 4347c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Form all the four Intra16x16 predictions in the yuv_p_ cache 4357c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8MakeLuma16Preds(const VP8EncIterator* const it); 4367c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Form all the four Chroma8x8 predictions in the yuv_p_ cache 4377c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8MakeChroma8Preds(const VP8EncIterator* const it); 4387c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Form all the ten Intra4x4 predictions in the yuv_p_ cache 4397c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// for the 4x4 block it->i4_ 4407c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8MakeIntra4Preds(const VP8EncIterator* const it); 4417c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Rate calculation 4427c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroraint VP8GetCostLuma16(VP8EncIterator* const it, const VP8ModeScore* const rd); 4437c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroraint VP8GetCostLuma4(VP8EncIterator* const it, const int16_t levels[16]); 4447c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroraint VP8GetCostUV(VP8EncIterator* const it, const VP8ModeScore* const rd); 4451e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// Main coding calls 4467c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroraint VP8EncLoop(VP8Encoder* const enc); 4471e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroraint VP8EncTokenLoop(VP8Encoder* const enc); 4487c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 449466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // in webpenc.c 450466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Assign an error code to a picture. Return false for convenience. 451a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroraint WebPEncodingSetError(const WebPPicture* const pic, WebPEncodingError error); 452a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroraint WebPReportProgress(const WebPPicture* const pic, 453a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora int percent, int* const percent_store); 454a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora 4557c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // in analysis.c 4567c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Main analysis loop. Decides the segmentations and complexity. 4577c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Assigns a first guess for Intra16 and uvmode_ prediction modes. 4587c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroraint VP8EncAnalyze(VP8Encoder* const enc); 4597c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 4607c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // in quant.c 4617c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Sets up segment's quantization values, base_quant_ and filter strengths. 4627c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8SetSegmentParams(VP8Encoder* const enc, float quality); 4637c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Pick best modes and fills the levels. Returns true if skipped. 4641e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroraint VP8Decimate(VP8EncIterator* const it, VP8ModeScore* const rd, 4651e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora VP8RDLevel rd_opt); 4667c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 467466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // in alpha.c 468a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroravoid VP8EncInitAlpha(VP8Encoder* const enc); // initialize alpha compression 4691e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroraint VP8EncStartAlpha(VP8Encoder* const enc); // start alpha coding process 470a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroraint VP8EncFinishAlpha(VP8Encoder* const enc); // finalize compressed data 4711e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroraint VP8EncDeleteAlpha(VP8Encoder* const enc); // delete compressed data 472466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 473a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora// autofilter 474a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroravoid VP8InitFilter(VP8EncIterator* const it); 475a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroravoid VP8StoreFilterStats(VP8EncIterator* const it); 476a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroravoid VP8AdjustFilterStrength(VP8EncIterator* const it); 477a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora 4788b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// returns the approximate filtering strength needed to smooth a edge 4798b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora// step of 'delta', given a sharpness parameter 'sharpness'. 4808b720228d581a84fd173b6dcb2fa295b59db489aVikas Aroraint VP8FilterStrengthFromDelta(int sharpness, int delta); 4818b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora 48233f74dabbc7920a65ed435d7417987589febdc16Vikas Arora // misc utils for picture_*.c: 48333f74dabbc7920a65ed435d7417987589febdc16Vikas Arora 48433f74dabbc7920a65ed435d7417987589febdc16Vikas Arora// Remove reference to the ARGB/YUVA buffer (doesn't free anything). 48533f74dabbc7920a65ed435d7417987589febdc16Vikas Aroravoid WebPPictureResetBuffers(WebPPicture* const picture); 48633f74dabbc7920a65ed435d7417987589febdc16Vikas Arora 48733f74dabbc7920a65ed435d7417987589febdc16Vikas Arora// Allocates ARGB buffer of given dimension (previous one is always free'd). 48833f74dabbc7920a65ed435d7417987589febdc16Vikas Arora// Preserves the YUV(A) buffer. Returns false in case of error (invalid param, 48933f74dabbc7920a65ed435d7417987589febdc16Vikas Arora// out-of-memory). 49033f74dabbc7920a65ed435d7417987589febdc16Vikas Aroraint WebPPictureAllocARGB(WebPPicture* const picture, int width, int height); 49133f74dabbc7920a65ed435d7417987589febdc16Vikas Arora 49233f74dabbc7920a65ed435d7417987589febdc16Vikas Arora// Allocates YUVA buffer of given dimension (previous one is always free'd). 49333f74dabbc7920a65ed435d7417987589febdc16Vikas Arora// Uses picture->csp to determine whether an alpha buffer is needed. 49433f74dabbc7920a65ed435d7417987589febdc16Vikas Arora// Preserves the ARGB buffer. 49533f74dabbc7920a65ed435d7417987589febdc16Vikas Arora// Returns false in case of error (invalid param, out-of-memory). 49633f74dabbc7920a65ed435d7417987589febdc16Vikas Aroraint WebPPictureAllocYUVA(WebPPicture* const picture, int width, int height); 49733f74dabbc7920a65ed435d7417987589febdc16Vikas Arora 4987c8da7ce66017295a65ec028084b90800be377f8James Zern// Clean-up the RGB samples under fully transparent area, to help lossless 4997c8da7ce66017295a65ec028084b90800be377f8James Zern// compressibility (no guarantee, though). Assumes that pic->use_argb is true. 5007c8da7ce66017295a65ec028084b90800be377f8James Zernvoid WebPCleanupTransparentAreaLossless(WebPPicture* const pic); 5017c8da7ce66017295a65ec028084b90800be377f8James Zern 502a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 5037c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 5048b720228d581a84fd173b6dcb2fa295b59db489aVikas Arora#ifdef __cplusplus 5057c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} // extern "C" 5067c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora#endif 5077c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 508a187300ff9a8a7c10b1fb2ec84223fdd14e6d47bJames Zern#endif /* WEBP_ENC_VP8I_ENC_H_ */ 509