ans.h 1.61 KB
Newer Older
1
/*
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
3
 *
4
5
6
7
8
9
 * 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.
10
11
 */

Alex Converse's avatar
Alex Converse committed
12
13
#ifndef AOM_DSP_ANS_H_
#define AOM_DSP_ANS_H_
14
// Constants, types and utilities for Asymmetric Numeral Systems
15
16
// http://arxiv.org/abs/1311.2540v2

Alex Converse's avatar
Alex Converse committed
17
#include <assert.h>
Yaowu Xu's avatar
Yaowu Xu committed
18
19
#include "./aom_config.h"
#include "aom/aom_integer.h"
20
#include "aom_dsp/prob.h"
21
22
23
24
25
26

#ifdef __cplusplus
extern "C" {
#endif  // __cplusplus

typedef uint8_t AnsP8;
27
28
#define ANS_P8_PRECISION 256u
#define ANS_P8_SHIFT 8
29
typedef uint16_t AnsP10;
30
31
#define ANS_P10_PRECISION 1024u
#define RANS_PROB_BITS 10
32

33
#define RANS_PRECISION ANS_P10_PRECISION
34

35
36
37
#define L_BASE (ANS_P10_PRECISION * 4)  // L_BASE % precision must be 0
#define IO_BASE 256
// Range I = { L_BASE, L_BASE + 1, ..., L_BASE * IO_BASE - 1 }
38

39
// This is now just a boring cdf. It starts with an explicit zero.
Alex Converse's avatar
Alex Converse committed
40
// TODO(aconverse): Remove starting zero.
41
typedef uint16_t rans_lut[16];
42

43
void aom_rans_build_cdf_from_pdf(const AnsP10 token_probs[], rans_lut cdf_tab);
44

45
46
void aom_rans_merge_prob8_pdf(AnsP10 *const out_pdf, const AnsP8 node_prob,
                              const AnsP10 *const src_pdf, int in_syms);
47
48
49
#ifdef __cplusplus
}  // extern "C"
#endif  // __cplusplus
Alex Converse's avatar
Alex Converse committed
50
#endif  // AOM_DSP_ANS_H_