---
libavcodec/bytestream.h | 109 +++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 109 insertions(+), 0 deletions(-)
diff --git a/libavcodec/bytestream.h b/libavcodec/bytestream.h
index 503598a..95d4b41 100644
--- a/libavcodec/bytestream.h
+++ b/libavcodec/bytestream.h
@@ -30,6 +30,11 @@ typedef struct {
const uint8_t *buffer, *buffer_end, *buffer_start;
} GetByteContext;
+typedef struct {
+ uint8_t *buffer, *buffer_end, *buffer_start;
+ int eof;
+} PutByteContext;
+
#define DEF_T(type, name, bytes, read, write) \
static av_always_inline type bytestream_get_ ## name(const uint8_t **b){\
(*b) += bytes;\
@@ -39,6 +44,13 @@ static av_always_inline void bytestream_put_ ##name(uint8_t
**b, const type valu
write(*b, value);\
(*b) += bytes;\
}\
+static av_always_inline void bytestream2_put_ ## name(PutByteContext *p, const
type value){\
+ if (!p->eof && (p->buffer_end - p->buffer >= bytes)) {\
+ write(p->buffer, value);\
+ p->buffer += bytes;\
+ } else\
+ p->eof = 1;\
+}\
static av_always_inline type bytestream2_get_ ## name ## u(GetByteContext *g)\
{\
return bytestream_get_ ## name(&g->buffer);\
@@ -119,22 +131,53 @@ static av_always_inline void
bytestream2_init(GetByteContext *g,
g->buffer_end = buf + buf_size;
}
+static av_always_inline void bytestream2_init_writer(PutByteContext *p,
+ uint8_t *buf, int
buf_size)
+{
+ p->buffer = buf;
+ p->buffer_start = buf;
+ p->buffer_end = buf + buf_size;
+ p->eof = 0;
+}
+
static av_always_inline unsigned int bytestream2_get_bytes_left(GetByteContext
*g)
{
return g->buffer_end - g->buffer;
}
+static av_always_inline unsigned int
bytestream2_get_bytes_left_p(PutByteContext *p)
+{
+ return p->buffer_end - p->buffer;
+}
+
static av_always_inline void bytestream2_skip(GetByteContext *g,
unsigned int size)
{
g->buffer += FFMIN(g->buffer_end - g->buffer, size);
}
+static av_always_inline void bytestream2_skip_p(PutByteContext *p,
+ unsigned int size)
+{
+ while (size--) {
+ if (p->eof || (p->buffer_end - p->buffer < 1)) {
+ p->eof = 1;
+ break;
+ }
+ p->buffer++;
+ }
+}
+
static av_always_inline int bytestream2_tell(GetByteContext *g)
{
return (int)(g->buffer - g->buffer_start);
}
+static av_always_inline int bytestream2_tell_p(PutByteContext *p)
+{
+ return (int)(p->buffer - p->buffer_start);
+}
+
static av_always_inline int bytestream2_seek(GetByteContext *g, int offset,
int whence)
{
@@ -158,6 +201,55 @@ static av_always_inline int
bytestream2_seek(GetByteContext *g, int offset,
return bytestream2_tell(g);
}
+static av_always_inline int bytestream2_seek_p(PutByteContext *p, int offset,
+ int whence)
+{
+ switch (whence) {
+ case SEEK_CUR:
+ offset = av_clip(offset, -(p->buffer - p->buffer_start),
+ p->buffer_end - p->buffer);
+ while (offset--) {
+ if (p->eof || (p->buffer_end - p->buffer < 1)) {
+ p->eof = 1;
+ break;
+ }
+ p->buffer += offset;
+ }
+ break;
+ case SEEK_END: {
+ int temp_offset;
+ offset = av_clip(offset, -(p->buffer_end - p->buffer_start), 0);
+ temp_offset = offset;
+ while (offset++) {
+ if (p->eof || (p->buffer - p->buffer_start < 1)) {
+ p->eof = 1;
+ break;
+ }
+ temp_offset--;
+ }
+ p->buffer = p->buffer_end + temp_offset;
+ break;
+ }
+ case SEEK_SET: {
+ int temp_offset;
+ offset = av_clip(offset, 0, p->buffer_end - p->buffer_start);
+ temp_offset = offset;
+ while (offset--) {
+ if (p->eof || (p->buffer_end - (p->buffer_start + temp_offset) <
1)) {
+ p->eof = 1;
+ break;
+ }
+ temp_offset++;
+ }
+ p->buffer = p->buffer_start + temp_offset;
+ break;
+ }
+ default:
+ return AVERROR(EINVAL);
+ }
+ return bytestream2_tell_p(p);
+}
+
static av_always_inline unsigned int bytestream2_get_buffer(GetByteContext *g,
uint8_t *dst,
unsigned int size)
@@ -168,6 +260,23 @@ static av_always_inline unsigned int
bytestream2_get_buffer(GetByteContext *g,
return size2;
}
+static av_always_inline void bytestream2_put_buffer(PutByteContext *p, const
uint8_t *src, unsigned int size)
+{
+ while (size--) {
+ if (p->eof || (p->buffer_end - p->buffer < 1)) {
+ p->eof = 1;
+ break;
+ }
+ memcpy(p->buffer, src, size);
+ p->buffer += size;
+ }
+}
+
+static av_always_inline unsigned int bytestream2_get_eof(PutByteContext *p)
+{
+ return p->eof;
+}
+
static av_always_inline unsigned int bytestream_get_buffer(const uint8_t **b,
uint8_t *dst, unsigned int size)
{
memcpy(dst, *b, size);
--
1.7.1
_______________________________________________
libav-devel mailing list
[email protected]
https://lists.libav.org/mailman/listinfo/libav-devel
