1/*
2 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
3 *
4 *  Use of this source code is governed by a BSD-style license
5 *  that can be found in the LICENSE file in the root of the source
6 *  tree. An additional intellectual property rights grant can be found
7 *  in the file PATENTS.  All contributing project authors may
8 *  be found in the AUTHORS file in the root of the source tree.
9 */
10
11#ifndef VP9_ENCODER_VP9_TOKENIZE_H_
12#define VP9_ENCODER_VP9_TOKENIZE_H_
13
14#include "vp9/common/vp9_entropy.h"
15
16#include "vp9/encoder/vp9_block.h"
17#include "vp9/encoder/vp9_treewriter.h"
18
19#ifdef __cplusplus
20extern "C" {
21#endif
22
23#define EOSB_TOKEN 127     // Not signalled, encoder only
24
25#if CONFIG_VP9_HIGHBITDEPTH
26  typedef int32_t EXTRABIT;
27#else
28  typedef int16_t EXTRABIT;
29#endif
30
31
32typedef struct {
33  int16_t token;
34  EXTRABIT extra;
35} TOKENVALUE;
36
37typedef struct {
38  const vpx_prob *context_tree;
39  EXTRABIT extra;
40  uint8_t token;
41  uint8_t skip_eob_node;
42} TOKENEXTRA;
43
44extern const vpx_tree_index vp9_coef_tree[];
45extern const vpx_tree_index vp9_coef_con_tree[];
46extern const struct vp9_token vp9_coef_encodings[];
47
48int vp9_is_skippable_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane);
49int vp9_has_high_freq_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane);
50
51struct VP9_COMP;
52struct ThreadData;
53
54void vp9_tokenize_sb(struct VP9_COMP *cpi, struct ThreadData *td,
55                     TOKENEXTRA **t, int dry_run, BLOCK_SIZE bsize);
56
57extern const int16_t *vp9_dct_value_cost_ptr;
58/* TODO: The Token field should be broken out into a separate char array to
59 *  improve cache locality, since it's needed for costing when the rest of the
60 *  fields are not.
61 */
62extern const TOKENVALUE *vp9_dct_value_tokens_ptr;
63extern const TOKENVALUE *vp9_dct_cat_lt_10_value_tokens;
64extern const int16_t vp9_cat6_low_cost[256];
65extern const int16_t vp9_cat6_high_cost[128];
66extern const int16_t vp9_cat6_high10_high_cost[512];
67extern const int16_t vp9_cat6_high12_high_cost[2048];
68static INLINE int16_t vp9_get_cost(int16_t token, EXTRABIT extrabits,
69                                   const int16_t *cat6_high_table) {
70  if (token != CATEGORY6_TOKEN)
71    return vp9_extra_bits[token].cost[extrabits];
72  return vp9_cat6_low_cost[extrabits & 0xff]
73      + cat6_high_table[extrabits >> 8];
74}
75
76#if CONFIG_VP9_HIGHBITDEPTH
77static INLINE const int16_t* vp9_get_high_cost_table(int bit_depth) {
78  return bit_depth == 8 ? vp9_cat6_high_cost
79      : (bit_depth == 10 ? vp9_cat6_high10_high_cost :
80         vp9_cat6_high12_high_cost);
81}
82#else
83static INLINE const int16_t* vp9_get_high_cost_table(int bit_depth) {
84  (void) bit_depth;
85  return vp9_cat6_high_cost;
86}
87#endif  // CONFIG_VP9_HIGHBITDEPTH
88
89static INLINE void vp9_get_token_extra(int v, int16_t *token, EXTRABIT *extra) {
90  if (v >= CAT6_MIN_VAL || v <= -CAT6_MIN_VAL) {
91    *token = CATEGORY6_TOKEN;
92    if (v >= CAT6_MIN_VAL)
93      *extra = 2 * v - 2 * CAT6_MIN_VAL;
94    else
95      *extra = -2 * v - 2 * CAT6_MIN_VAL + 1;
96    return;
97  }
98  *token = vp9_dct_cat_lt_10_value_tokens[v].token;
99  *extra = vp9_dct_cat_lt_10_value_tokens[v].extra;
100}
101static INLINE int16_t vp9_get_token(int v) {
102  if (v >= CAT6_MIN_VAL || v <= -CAT6_MIN_VAL)
103    return 10;
104  return vp9_dct_cat_lt_10_value_tokens[v].token;
105}
106
107
108#ifdef __cplusplus
109}  // extern "C"
110#endif
111
112#endif  // VP9_ENCODER_VP9_TOKENIZE_H_
113