aef1559206e2f0cbacb99c22b7e841c51d51cd0b
[speexdsp.git] / libspeex / buffer.c
1 /* Copyright (C) 2007 Jean-Marc Valin
2       
3    File: buffer.c
4    This is a very simple ring buffer implementation. It is not thread-safe
5    so you need to do your own locking.
6
7    Redistribution and use in source and binary forms, with or without
8    modification, are permitted provided that the following conditions are
9    met:
10
11    1. Redistributions of source code must retain the above copyright notice,
12    this list of conditions and the following disclaimer.
13
14    2. Redistributions in binary form must reproduce the above copyright
15    notice, this list of conditions and the following disclaimer in the
16    documentation and/or other materials provided with the distribution.
17
18    3. The name of the author may not be used to endorse or promote products
19    derived from this software without specific prior written permission.
20
21    THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
22    IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
23    OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
24    DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
25    INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
26    (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
27    SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28    HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
29    STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
30    ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
31    POSSIBILITY OF SUCH DAMAGE.
32 */
33
34 #ifdef HAVE_CONFIG_H
35 #include "config.h"
36 #endif
37
38
39 #include "os_support.h"
40 #include "misc.h"
41 #include <speex/speex_buffer.h>
42
43 struct SpeexBuffer_ {
44    void *data;
45    int   size;
46    int   read_ptr;
47    int   write_ptr;
48    int   available;
49 };
50
51 SpeexBuffer *speex_buffer_init(int size)
52 {
53    SpeexBuffer *st = speex_alloc(sizeof(SpeexBuffer));
54    st->data = speex_alloc(size);
55    st->size = size;
56    st->read_ptr = 0;
57    st->write_ptr = 0;
58    st->available = 0;
59    return st;
60 }
61
62 void speex_buffer_destroy(SpeexBuffer *st)
63 {
64    speex_free(st->data);
65    speex_free(st);
66 }
67
68 int speex_buffer_write(SpeexBuffer *st, void *data, int len)
69 {
70    int end;
71    int end1;
72    if (len > st->size)
73    {
74       data += len-st->size;
75       len = st->size;
76    }
77    end = st->write_ptr + len;
78    end1 = end;
79    if (end1 > st->size)
80       end1 = st->size;
81    speex_move(st->data + st->write_ptr, data, end1 - st->write_ptr);
82    if (end > st->size)
83    {
84       end -= st->size;
85       speex_move(st->data, data+end1 - st->write_ptr, end);
86    }
87    st->available += len;
88    if (st->available > st->size)
89    {
90       st->available = st->size;
91       st->read_ptr = st->write_ptr;
92    }
93    st->write_ptr += len;
94    if (st->write_ptr > st->size)
95       st->write_ptr -= st->size;
96    return len;
97 }
98
99 int speex_buffer_writezeros(SpeexBuffer *st, int len)
100 {
101    /* This is almost the same as for speex_buffer_write() but using 
102    speex_memset() instead of speex_move(). Update accordingly. */
103       int end;
104    int end1;
105    if (len > st->size)
106    {
107       len = st->size;
108    }
109    end = st->write_ptr + len;
110    end1 = end;
111    if (end1 > st->size)
112       end1 = st->size;
113    speex_memset(st->data + st->write_ptr, 0, end1 - st->write_ptr);
114    if (end > st->size)
115    {
116       end -= st->size;
117       speex_memset(st->data, 0, end);
118    }
119    st->available += len;
120    if (st->available > st->size)
121    {
122       st->available = st->size;
123       st->read_ptr = st->write_ptr;
124    }
125    st->write_ptr += len;
126    if (st->write_ptr > st->size)
127       st->write_ptr -= st->size;
128    return len;
129 }
130
131 int speex_buffer_read(SpeexBuffer *st, void *data, int len)
132 {
133    int end, end1;
134    if (len > st->available)
135    {
136       speex_memset(data+st->available, 0, st->size-st->available);
137       len = st->available;
138    }
139    end = st->read_ptr + len;
140    end1 = end;
141    if (end1 > st->size)
142       end1 = st->size;
143    speex_move(data, st->data + st->read_ptr, end1 - st->read_ptr);
144
145    if (end > st->size)
146    {
147       end -= st->size;
148       speex_move(data+end1 - st->read_ptr, st->data, end);
149    }
150    st->available -= len;
151    st->read_ptr += len;
152    if (st->read_ptr > st->size)
153       st->read_ptr -= st->size;
154    return len;
155 }
156
157 int speex_buffer_get_available(SpeexBuffer *st)
158 {
159    return st->available;
160 }
161
162 int speex_buffer_resize(SpeexBuffer *st, int len)
163 {
164    int old_len = st->size;
165    if (len > old_len)
166    {
167       st->data = speex_realloc(st->data, len);
168       /* FIXME: move data/pointers properly for growing the buffer */
169    } else {
170       /* FIXME: move data/pointers properly for shrinking the buffer */
171       st->data = speex_realloc(st->data, len);
172    }
173    return len;
174 }