77bba8d30a
PVQ replaces the scalar quantizer and coefficient coding with a new
design originally developed in Daala. It currently depends on the
Daala entropy coder although it could be adapted to work with another
entropy coder if needed:
./configure --enable-experimental --enable-daala_ec --enable-pvq
The version of PVQ in this commit is adapted from the following
revision of Daala:
fb51c1ade6
More information about PVQ:
- https://people.xiph.org/~jm/daala/pvq_demo/
- https://jmvalin.ca/papers/spie_pvq.pdf
The following files are copied as-is from Daala with minimal
adaptations, therefore we disable clang-format on those files
to make it easier to synchronize the AV1 and Daala codebases in the future:
av1/common/generic_code.c
av1/common/generic_code.h
av1/common/laplace_tables.c
av1/common/partition.c
av1/common/partition.h
av1/common/pvq.c
av1/common/pvq.h
av1/common/state.c
av1/common/state.h
av1/common/zigzag.h
av1/common/zigzag16.c
av1/common/zigzag32.c
av1/common/zigzag4.c
av1/common/zigzag64.c
av1/common/zigzag8.c
av1/decoder/decint.h
av1/decoder/generic_decoder.c
av1/decoder/laplace_decoder.c
av1/decoder/pvq_decoder.c
av1/decoder/pvq_decoder.h
av1/encoder/daala_compat_enc.c
av1/encoder/encint.h
av1/encoder/generic_encoder.c
av1/encoder/laplace_encoder.c
av1/encoder/pvq_encoder.c
av1/encoder/pvq_encoder.h
Known issues:
- Lossless mode is not supported, '--lossless=1' will give the same result as
'--end-usage=q --cq-level=1'.
- High bit depth is not supported by PVQ.
Change-Id: I1ae0d6517b87f4c1ccea944b2e12dc906979f25e
104 lines
3.6 KiB
C
104 lines
3.6 KiB
C
/*
|
|
* Copyright (c) 2001-2016, Alliance for Open Media. All rights reserved
|
|
*
|
|
* This source code is subject to the terms of the BSD 2 Clause License and
|
|
* the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
|
|
* was not distributed with this source code in the LICENSE file, you can
|
|
* obtain it at www.aomedia.org/license/software. If the Alliance for Open
|
|
* Media Patent License 1.0 was not distributed with this source code in the
|
|
* PATENTS file, you can obtain it at www.aomedia.org/license/patent.
|
|
*/
|
|
|
|
#if !defined(_entenc_H)
|
|
#define _entenc_H (1)
|
|
#include <stddef.h>
|
|
#include "aom_dsp/entcode.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
typedef struct od_ec_enc od_ec_enc;
|
|
|
|
#define OD_MEASURE_EC_OVERHEAD (0)
|
|
|
|
/*The entropy encoder context.*/
|
|
struct od_ec_enc {
|
|
/*Buffered output.
|
|
This contains only the raw bits until the final call to od_ec_enc_done(),
|
|
where all the arithmetic-coded data gets prepended to it.*/
|
|
unsigned char *buf;
|
|
/*The size of the buffer.*/
|
|
uint32_t storage;
|
|
/*The offset at which the last byte containing raw bits was written.*/
|
|
uint32_t end_offs;
|
|
/*Bits that will be read from/written at the end.*/
|
|
od_ec_window end_window;
|
|
/*Number of valid bits in end_window.*/
|
|
int nend_bits;
|
|
/*A buffer for output bytes with their associated carry flags.*/
|
|
uint16_t *precarry_buf;
|
|
/*The size of the pre-carry buffer.*/
|
|
uint32_t precarry_storage;
|
|
/*The offset at which the next entropy-coded byte will be written.*/
|
|
uint32_t offs;
|
|
/*The low end of the current range.*/
|
|
od_ec_window low;
|
|
/*The number of values in the current range.*/
|
|
uint16_t rng;
|
|
/*The number of bits of data in the current value.*/
|
|
int16_t cnt;
|
|
/*Nonzero if an error occurred.*/
|
|
int error;
|
|
#if OD_MEASURE_EC_OVERHEAD
|
|
double entropy;
|
|
int nb_symbols;
|
|
#endif
|
|
};
|
|
|
|
/*See entenc.c for further documentation.*/
|
|
|
|
void od_ec_enc_init(od_ec_enc *enc, uint32_t size) OD_ARG_NONNULL(1);
|
|
void od_ec_enc_reset(od_ec_enc *enc) OD_ARG_NONNULL(1);
|
|
void od_ec_enc_clear(od_ec_enc *enc) OD_ARG_NONNULL(1);
|
|
|
|
void od_ec_encode_bool(od_ec_enc *enc, int val, unsigned fz, unsigned _ft)
|
|
OD_ARG_NONNULL(1);
|
|
void od_ec_encode_bool_q15(od_ec_enc *enc, int val, unsigned fz_q15)
|
|
OD_ARG_NONNULL(1);
|
|
void od_ec_encode_cdf(od_ec_enc *enc, int s, const uint16_t *cdf, int nsyms)
|
|
OD_ARG_NONNULL(1) OD_ARG_NONNULL(3);
|
|
void od_ec_encode_cdf_q15(od_ec_enc *enc, int s, const uint16_t *cdf, int nsyms)
|
|
OD_ARG_NONNULL(1) OD_ARG_NONNULL(3);
|
|
void od_ec_encode_cdf_unscaled(od_ec_enc *enc, int s, const uint16_t *cdf,
|
|
int nsyms) OD_ARG_NONNULL(1) OD_ARG_NONNULL(3);
|
|
void od_ec_encode_cdf_unscaled_dyadic(od_ec_enc *enc, int s,
|
|
const uint16_t *cdf, int nsyms,
|
|
unsigned ftb) OD_ARG_NONNULL(1)
|
|
OD_ARG_NONNULL(3);
|
|
|
|
void od_ec_enc_uint(od_ec_enc *enc, uint32_t fl, uint32_t ft) OD_ARG_NONNULL(1);
|
|
|
|
void od_ec_enc_bits(od_ec_enc *enc, uint32_t fl, unsigned ftb)
|
|
OD_ARG_NONNULL(1);
|
|
|
|
void od_ec_enc_patch_initial_bits(od_ec_enc *enc, unsigned val, int nbits)
|
|
OD_ARG_NONNULL(1);
|
|
OD_WARN_UNUSED_RESULT unsigned char *od_ec_enc_done(od_ec_enc *enc,
|
|
uint32_t *nbytes)
|
|
OD_ARG_NONNULL(1) OD_ARG_NONNULL(2);
|
|
|
|
OD_WARN_UNUSED_RESULT int od_ec_enc_tell(const od_ec_enc *enc)
|
|
OD_ARG_NONNULL(1);
|
|
OD_WARN_UNUSED_RESULT uint32_t od_ec_enc_tell_frac(const od_ec_enc *enc)
|
|
OD_ARG_NONNULL(1);
|
|
|
|
void od_ec_enc_checkpoint(od_ec_enc *dst, const od_ec_enc *src);
|
|
void od_ec_enc_rollback(od_ec_enc *dst, const od_ec_enc *src);
|
|
|
|
#ifdef __cplusplus
|
|
} // extern "C"
|
|
#endif
|
|
|
|
#endif
|