OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. | 2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #ifndef VP9_ENCODER_VP9_WRITER_H_ | 11 #ifndef VP9_ENCODER_VP9_WRITER_H_ |
12 #define VP9_ENCODER_VP9_WRITER_H_ | 12 #define VP9_ENCODER_VP9_WRITER_H_ |
13 | 13 |
14 #include "vpx_ports/mem.h" | 14 #include "vpx_ports/mem.h" |
15 | 15 |
16 #include "vp9/common/vp9_prob.h" | 16 #include "vp9/common/vp9_prob.h" |
17 | 17 |
18 #ifdef __cplusplus | 18 #ifdef __cplusplus |
19 extern "C" { | 19 extern "C" { |
20 #endif | 20 #endif |
21 | 21 |
22 typedef struct { | 22 typedef struct { |
23 unsigned int lowvalue; | 23 unsigned int lowvalue; |
24 unsigned int range; | 24 unsigned int range; |
25 unsigned int value; | |
26 int count; | 25 int count; |
27 unsigned int pos; | 26 unsigned int pos; |
28 uint8_t *buffer; | 27 uint8_t *buffer; |
29 | |
30 // Variables used to track bit costs without outputing to the bitstream | |
31 unsigned int measure_cost; | |
32 uint64_t bit_counter; | |
33 } vp9_writer; | 28 } vp9_writer; |
34 | 29 |
35 void vp9_start_encode(vp9_writer *bc, uint8_t *buffer); | 30 void vp9_start_encode(vp9_writer *bc, uint8_t *buffer); |
36 void vp9_stop_encode(vp9_writer *bc); | 31 void vp9_stop_encode(vp9_writer *bc); |
37 | 32 |
38 static void vp9_write(vp9_writer *br, int bit, int probability) { | 33 static INLINE void vp9_write(vp9_writer *br, int bit, int probability) { |
39 unsigned int split; | 34 unsigned int split; |
40 int count = br->count; | 35 int count = br->count; |
41 unsigned int range = br->range; | 36 unsigned int range = br->range; |
42 unsigned int lowvalue = br->lowvalue; | 37 unsigned int lowvalue = br->lowvalue; |
43 register unsigned int shift; | 38 register unsigned int shift; |
44 | 39 |
45 split = 1 + (((range - 1) * probability) >> 8); | 40 split = 1 + (((range - 1) * probability) >> 8); |
46 | 41 |
47 range = split; | 42 range = split; |
48 | 43 |
(...skipping 27 matching lines...) Expand all Loading... |
76 lowvalue &= 0xffffff; | 71 lowvalue &= 0xffffff; |
77 count -= 8; | 72 count -= 8; |
78 } | 73 } |
79 | 74 |
80 lowvalue <<= shift; | 75 lowvalue <<= shift; |
81 br->count = count; | 76 br->count = count; |
82 br->lowvalue = lowvalue; | 77 br->lowvalue = lowvalue; |
83 br->range = range; | 78 br->range = range; |
84 } | 79 } |
85 | 80 |
86 static void vp9_write_bit(vp9_writer *w, int bit) { | 81 static INLINE void vp9_write_bit(vp9_writer *w, int bit) { |
87 vp9_write(w, bit, 128); // vp9_prob_half | 82 vp9_write(w, bit, 128); // vp9_prob_half |
88 } | 83 } |
89 | 84 |
90 static void vp9_write_literal(vp9_writer *w, int data, int bits) { | 85 static INLINE void vp9_write_literal(vp9_writer *w, int data, int bits) { |
91 int bit; | 86 int bit; |
92 | 87 |
93 for (bit = bits - 1; bit >= 0; bit--) | 88 for (bit = bits - 1; bit >= 0; bit--) |
94 vp9_write_bit(w, 1 & (data >> bit)); | 89 vp9_write_bit(w, 1 & (data >> bit)); |
95 } | 90 } |
96 | 91 |
97 #define vp9_write_prob(w, v) vp9_write_literal((w), (v), 8) | 92 #define vp9_write_prob(w, v) vp9_write_literal((w), (v), 8) |
98 | 93 |
99 #ifdef __cplusplus | 94 #ifdef __cplusplus |
100 } // extern "C" | 95 } // extern "C" |
101 #endif | 96 #endif |
102 | 97 |
103 #endif // VP9_ENCODER_VP9_WRITER_H_ | 98 #endif // VP9_ENCODER_VP9_WRITER_H_ |
OLD | NEW |