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
87 lines
3.2 KiB
C
87 lines
3.2 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.
|
|
*/
|
|
|
|
/* clang-format off */
|
|
|
|
#if !defined(_generic_code_H)
|
|
# define _generic_code_H
|
|
|
|
# include "aom_dsp/entdec.h"
|
|
# include "aom_dsp/entenc.h"
|
|
|
|
# define GENERIC_TABLES 12
|
|
|
|
#if OD_ACCOUNTING
|
|
# define generic_decode(dec, model, max, ex_q16, integration, str) generic_decode_(dec, model, max, ex_q16, integration, str)
|
|
# define od_decode_cdf_adapt_q15(ec, cdf, n, count, rate, str) od_decode_cdf_adapt_q15_(ec, cdf, n, count, rate, str)
|
|
# define od_decode_cdf_adapt(ec, cdf, n, increment, str) od_decode_cdf_adapt_(ec, cdf, n, increment, str)
|
|
#else
|
|
# define generic_decode(dec, model, max, ex_q16, integration, str) generic_decode_(dec, model, max, ex_q16, integration)
|
|
# define od_decode_cdf_adapt_q15(ec, cdf, n, count, rate, str) od_decode_cdf_adapt_q15_(ec, cdf, n, count, rate)
|
|
# define od_decode_cdf_adapt(ec, cdf, n, increment, str) od_decode_cdf_adapt_(ec, cdf, n, increment)
|
|
#endif
|
|
|
|
typedef struct {
|
|
/** cdf for multiple expectations of x */
|
|
uint16_t cdf[GENERIC_TABLES][16];
|
|
/** Frequency increment for learning the cdfs */
|
|
int increment;
|
|
} generic_encoder;
|
|
|
|
#define OD_IIR_DIADIC(y, x, shift) ((y) += ((x) - (y)) >> (shift))
|
|
|
|
void generic_model_init(generic_encoder *model);
|
|
|
|
#define OD_CDFS_INIT(cdf, val) od_cdf_init(&cdf[0][0],\
|
|
sizeof(cdf)/sizeof(cdf[0]), sizeof(cdf[0])/sizeof(cdf[0][0]), val, val)
|
|
|
|
#define OD_CDFS_INIT_FIRST(cdf, val, first) od_cdf_init(&cdf[0][0],\
|
|
sizeof(cdf)/sizeof(cdf[0]), sizeof(cdf[0])/sizeof(cdf[0][0]), val, first)
|
|
|
|
#define OD_SINGLE_CDF_INIT(cdf, val) od_cdf_init(cdf,\
|
|
1, sizeof(cdf)/sizeof(cdf[0]), val, val)
|
|
|
|
#define OD_SINGLE_CDF_INIT_FIRST(cdf, val, first) od_cdf_init(cdf,\
|
|
1, sizeof(cdf)/sizeof(cdf[0]), val, first)
|
|
|
|
void od_cdf_init(uint16_t *cdf, int ncdfs, int nsyms, int val, int first);
|
|
|
|
void od_cdf_adapt_q15(int val, uint16_t *cdf, int n, int *count, int rate);
|
|
|
|
void od_encode_cdf_adapt_q15(od_ec_enc *ec, int val, uint16_t *cdf, int n,
|
|
int *count, int rate);
|
|
|
|
void od_encode_cdf_adapt(od_ec_enc *ec, int val, uint16_t *cdf, int n,
|
|
int increment);
|
|
|
|
int od_decode_cdf_adapt_(od_ec_dec *ec, uint16_t *cdf, int n,
|
|
int increment OD_ACC_STR);
|
|
|
|
void generic_encode(od_ec_enc *enc, generic_encoder *model, int x, int max,
|
|
int *ex_q16, int integration);
|
|
double generic_encode_cost(generic_encoder *model, int x, int max,
|
|
int *ex_q16);
|
|
|
|
double od_encode_cdf_cost(int val, uint16_t *cdf, int n);
|
|
|
|
int od_decode_cdf_adapt_q15_(od_ec_dec *ec, uint16_t *cdf, int n,
|
|
int *count, int rate OD_ACC_STR);
|
|
|
|
int generic_decode_(od_ec_dec *dec, generic_encoder *model, int max,
|
|
int *ex_q16, int integration OD_ACC_STR);
|
|
|
|
int log_ex(int ex_q16);
|
|
|
|
void generic_model_update(generic_encoder *model, int *ex_q16, int x, int xs,
|
|
int id, int integration);
|
|
|
|
#endif
|