remove near duplicate function
[libav.git] / libavutil / fifo.c
CommitLineData
f5a478f6
RS
1/*
2 * A very simple circular buffer FIFO implementation
3 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
4 * Copyright (c) 2006 Roman Shaposhnik
5 *
b78e7197
DB
6 * This file is part of FFmpeg.
7 *
8 * FFmpeg is free software; you can redistribute it and/or
f5a478f6
RS
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
b78e7197 11 * version 2.1 of the License, or (at your option) any later version.
f5a478f6 12 *
b78e7197 13 * FFmpeg is distributed in the hope that it will be useful,
f5a478f6
RS
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
b78e7197 19 * License along with FFmpeg; if not, write to the Free Software
f5a478f6
RS
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 */
22#include "common.h"
23#include "fifo.h"
24
25int av_fifo_init(AVFifoBuffer *f, int size)
26{
27 f->buffer = av_malloc(size);
28 if (!f->buffer)
29 return -1;
30 f->end = f->buffer + size;
31 f->wptr = f->rptr = f->buffer;
32 return 0;
33}
34
35void av_fifo_free(AVFifoBuffer *f)
36{
37 av_free(f->buffer);
38}
39
40int av_fifo_size(AVFifoBuffer *f)
41{
42 int size = f->wptr - f->rptr;
43 if (size < 0)
44 size += f->end - f->buffer;
45 return size;
46}
47
48/**
49 * Get data from the fifo (returns -1 if not enough data).
50 */
51int av_fifo_read(AVFifoBuffer *f, uint8_t *buf, int buf_size)
52{
765d4f3b 53 return av_fifo_generic_read(f, buf_size, NULL, buf);
f5a478f6
RS
54}
55
56/**
57 * Resizes a FIFO.
58 */
59void av_fifo_realloc(AVFifoBuffer *f, unsigned int new_size) {
60 unsigned int old_size= f->end - f->buffer;
61
62 if(old_size < new_size){
63 uint8_t *old= f->buffer;
64
65 f->buffer= av_realloc(f->buffer, new_size);
66
67 f->rptr += f->buffer - old;
68 f->wptr += f->buffer - old;
69
70 if(f->wptr < f->rptr){
71 memmove(f->rptr + new_size - old_size, f->rptr, f->buffer + old_size - f->rptr);
72 f->rptr += new_size - old_size;
73 }
74 f->end= f->buffer + new_size;
75 }
76}
77
78void av_fifo_write(AVFifoBuffer *f, const uint8_t *buf, int size)
79{
f5a478f6 80 while (size > 0) {
870a12d1 81 int len = FFMIN(f->end - f->wptr, size);
f5a478f6
RS
82 memcpy(f->wptr, buf, len);
83 f->wptr += len;
84 if (f->wptr >= f->end)
85 f->wptr = f->buffer;
86 buf += len;
87 size -= len;
88 }
89}
90
91
92/* get data from the fifo (return -1 if not enough data) */
93int av_fifo_generic_read(AVFifoBuffer *f, int buf_size, void (*func)(void*, void*, int), void* dest)
94{
870a12d1 95 int size = av_fifo_size(f);
f5a478f6
RS
96
97 if (size < buf_size)
98 return -1;
99 while (buf_size > 0) {
870a12d1 100 int len = FFMIN(f->end - f->rptr, buf_size);
765d4f3b
MN
101 if(func) func(dest, f->rptr, len);
102 else{
103 memcpy(dest, f->rptr, len);
104 dest = (uint8_t*)dest + len;
105 }
f5a478f6
RS
106 f->rptr += len;
107 if (f->rptr >= f->end)
108 f->rptr = f->buffer;
109 buf_size -= len;
110 }
111 return 0;
112}
113
114/* discard data from the fifo */
115void av_fifo_drain(AVFifoBuffer *f, int size)
116{
117 f->rptr += size;
118 if (f->rptr >= f->end)
119 f->rptr -= f->end - f->buffer;
120}