Merge commit 'aa45b90804ab21175b8c116bd8e5eb4b4e85fbcb' into release/0.8
[ffmpeg:ffmpeg.git] / libavcodec / bytestream.h
1 /*
2  * Bytestream functions
3  * copyright (c) 2006 Baptiste Coudurier <baptiste.coudurier@free.fr>
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #ifndef AVCODEC_BYTESTREAM_H
23 #define AVCODEC_BYTESTREAM_H
24
25 #include <string.h>
26 #include "libavutil/common.h"
27 #include "libavutil/intreadwrite.h"
28
29 typedef struct {
30     const uint8_t *buffer, *buffer_end;
31 } GetByteContext;
32
33 #define DEF_T(type, name, bytes, read, write)                             \
34 static av_always_inline type bytestream_get_ ## name(const uint8_t **b){\
35     (*b) += bytes;\
36     return read(*b - bytes);\
37 }\
38 static av_always_inline void bytestream_put_ ##name(uint8_t **b, const type value){\
39     write(*b, value);\
40     (*b) += bytes;\
41 }\
42 static av_always_inline type bytestream2_get_ ## name(GetByteContext *g)\
43 {\
44     if (g->buffer_end - g->buffer < bytes)\
45         return 0;\
46     return bytestream_get_ ## name(&g->buffer);\
47 }\
48 static av_always_inline type bytestream2_peek_ ## name(GetByteContext *g)\
49 {\
50     if (g->buffer_end - g->buffer < bytes)\
51         return 0;\
52     return read(g->buffer);\
53 }
54
55 #define DEF(name, bytes, read, write) \
56     DEF_T(unsigned int, name, bytes, read, write)
57 #define DEF64(name, bytes, read, write) \
58     DEF_T(uint64_t, name, bytes, read, write)
59
60 DEF64(le64, 8, AV_RL64, AV_WL64)
61 DEF  (le32, 4, AV_RL32, AV_WL32)
62 DEF  (le24, 3, AV_RL24, AV_WL24)
63 DEF  (le16, 2, AV_RL16, AV_WL16)
64 DEF64(be64, 8, AV_RB64, AV_WB64)
65 DEF  (be32, 4, AV_RB32, AV_WB32)
66 DEF  (be24, 3, AV_RB24, AV_WB24)
67 DEF  (be16, 2, AV_RB16, AV_WB16)
68 DEF  (byte, 1, AV_RB8 , AV_WB8 )
69
70 #undef DEF
71 #undef DEF64
72 #undef DEF_T
73
74 static av_always_inline void bytestream2_init(GetByteContext *g,
75                                               const uint8_t *buf, int buf_size)
76 {
77     g->buffer =  buf;
78     g->buffer_end = buf + buf_size;
79 }
80
81 static av_always_inline unsigned int bytestream2_get_bytes_left(GetByteContext *g)
82 {
83     return g->buffer_end - g->buffer;
84 }
85
86 static av_always_inline void bytestream2_skip(GetByteContext *g,
87                                               unsigned int size)
88 {
89     g->buffer += FFMIN(g->buffer_end - g->buffer, size);
90 }
91
92 static av_always_inline unsigned int bytestream2_get_buffer(GetByteContext *g,
93                                                             uint8_t *dst,
94                                                             unsigned int size)
95 {
96     int size2 = FFMIN(g->buffer_end - g->buffer, size);
97     memcpy(dst, g->buffer, size2);
98     g->buffer += size2;
99     return size2;
100 }
101
102 static av_always_inline unsigned int bytestream_get_buffer(const uint8_t **b, uint8_t *dst, unsigned int size)
103 {
104     memcpy(dst, *b, size);
105     (*b) += size;
106     return size;
107 }
108
109 static av_always_inline void bytestream_put_buffer(uint8_t **b, const uint8_t *src, unsigned int size)
110 {
111     memcpy(*b, src, size);
112     (*b) += size;
113 }
114
115 #endif /* AVCODEC_BYTESTREAM_H */