vp9_reader.h 2.56 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
/*
2
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
John Koleszar's avatar
John Koleszar committed
3
 *
4
 *  Use of this source code is governed by a BSD-style license
5 6
 *  that can be found in the LICENSE file in the root of the source
 *  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.
John Koleszar's avatar
John Koleszar committed
9 10
 */

11 12
#ifndef VP9_DECODER_VP9_READER_H_
#define VP9_DECODER_VP9_READER_H_
13

14 15
#include <stddef.h>
#include <limits.h>
Dmitry Kovalev's avatar
Dmitry Kovalev committed
16

17
#include "./vpx_config.h"
John Koleszar's avatar
John Koleszar committed
18
#include "vpx_ports/mem.h"
19
#include "vpx/vp8dx.h"
20
#include "vpx/vpx_integer.h"
John Koleszar's avatar
John Koleszar committed
21

22
#include "vp9/common/vp9_prob.h"
23

24 25 26 27
#ifdef __cplusplus
extern "C" {
#endif

28 29 30 31
typedef size_t BD_VALUE;

#define BD_VALUE_SIZE ((int)sizeof(BD_VALUE) * CHAR_BIT)

John Koleszar's avatar
John Koleszar committed
32
typedef struct {
33 34
  const uint8_t *buffer_end;
  const uint8_t *buffer;
35
  uint8_t clear_buffer[sizeof(BD_VALUE) + 1];
36
  BD_VALUE value;
37 38
  int count;
  unsigned int range;
39 40
  vpx_decrypt_cb decrypt_cb;
  void *decrypt_state;
41
} vp9_reader;
John Koleszar's avatar
John Koleszar committed
42

43 44 45 46 47
int vp9_reader_init(vp9_reader *r,
                    const uint8_t *buffer,
                    size_t size,
                    vpx_decrypt_cb decrypt_cb,
                    void *decrypt_state);
Johann's avatar
Johann committed
48

49
void vp9_reader_fill(vp9_reader *r);
50

51 52
int vp9_reader_has_error(vp9_reader *r);

53
const uint8_t *vp9_reader_find_end(vp9_reader *r);
54

55
static int vp9_read(vp9_reader *r, int prob) {
John Koleszar's avatar
John Koleszar committed
56
  unsigned int bit = 0;
57 58
  BD_VALUE value;
  BD_VALUE bigsplit;
John Koleszar's avatar
John Koleszar committed
59 60
  int count;
  unsigned int range;
61
  unsigned int split = (r->range * prob + (256 - prob)) >> CHAR_BIT;
John Koleszar's avatar
John Koleszar committed
62

63 64
  if (r->count < 0)
    vp9_reader_fill(r);
John Koleszar's avatar
John Koleszar committed
65

66 67
  value = r->value;
  count = r->count;
John Koleszar's avatar
John Koleszar committed
68

69
  bigsplit = (BD_VALUE)split << (BD_VALUE_SIZE - CHAR_BIT);
John Koleszar's avatar
John Koleszar committed
70 71 72 73

  range = split;

  if (value >= bigsplit) {
74
    range = r->range - split;
John Koleszar's avatar
John Koleszar committed
75 76 77 78 79
    value = value - bigsplit;
    bit = 1;
  }

  {
80
    register unsigned int shift = vp9_norm[range];
John Koleszar's avatar
John Koleszar committed
81 82 83 84
    range <<= shift;
    value <<= shift;
    count -= shift;
  }
85 86 87
  r->value = value;
  r->count = count;
  r->range = range;
John Koleszar's avatar
John Koleszar committed
88 89

  return bit;
John Koleszar's avatar
John Koleszar committed
90 91
}

92
static int vp9_read_bit(vp9_reader *r) {
93 94 95
  return vp9_read(r, 128);  // vp9_prob_half
}

96 97
static int vp9_read_literal(vp9_reader *r, int bits) {
  int literal = 0, bit;
John Koleszar's avatar
John Koleszar committed
98

99
  for (bit = bits - 1; bit >= 0; bit--)
100
    literal |= vp9_read_bit(r) << bit;
John Koleszar's avatar
John Koleszar committed
101

102
  return literal;
John Koleszar's avatar
John Koleszar committed
103
}
104

105 106 107 108 109 110 111 112 113
static int vp9_read_tree(vp9_reader *r, const vp9_tree_index *tree,
                         const vp9_prob *probs) {
  vp9_tree_index i = 0;

  while ((i = tree[i + vp9_read(r, probs[i >> 1])]) > 0)
    continue;

  return -i;
}
114

115 116 117 118
#ifdef __cplusplus
}  // extern "C"
#endif

119
#endif  // VP9_DECODER_VP9_READER_H_