3 * Copyright (c) 2012 Justin Ruggles <justin.ruggles@gmail.com>
5 * This file is part of FFmpeg.
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.
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.
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
28 #include "audio_fifo.h"
32 #include "samplefmt.h"
35 AVFifoBuffer
**buf
; /**< single buffer for interleaved, per-channel buffers for planar */
36 int nb_buffers
; /**< number of buffers */
37 int nb_samples
; /**< number of samples currently in the FIFO */
38 int allocated_samples
; /**< current allocated size, in samples */
40 int channels
; /**< number of channels */
41 enum AVSampleFormat sample_fmt
; /**< sample format */
42 int sample_size
; /**< size, in bytes, of one sample in a buffer */
45 void av_audio_fifo_free(AVAudioFifo
*af
)
50 for (i
= 0; i
< af
->nb_buffers
; i
++) {
52 av_fifo_free(af
->buf
[i
]);
60 AVAudioFifo
*av_audio_fifo_alloc(enum AVSampleFormat sample_fmt
, int channels
,
66 /* get channel buffer size (also validates parameters) */
67 if (av_samples_get_buffer_size(&buf_size
, channels
, nb_samples
, sample_fmt
, 1) < 0)
70 af
= av_mallocz(sizeof(*af
));
74 af
->channels
= channels
;
75 af
->sample_fmt
= sample_fmt
;
76 af
->sample_size
= buf_size
/ nb_samples
;
77 af
->nb_buffers
= av_sample_fmt_is_planar(sample_fmt
) ? channels
: 1;
79 af
->buf
= av_mallocz_array(af
->nb_buffers
, sizeof(*af
->buf
));
83 for (i
= 0; i
< af
->nb_buffers
; i
++) {
84 af
->buf
[i
] = av_fifo_alloc(buf_size
);
88 af
->allocated_samples
= nb_samples
;
93 av_audio_fifo_free(af
);
97 int av_audio_fifo_realloc(AVAudioFifo
*af
, int nb_samples
)
101 if ((ret
= av_samples_get_buffer_size(&buf_size
, af
->channels
, nb_samples
,
102 af
->sample_fmt
, 1)) < 0)
105 for (i
= 0; i
< af
->nb_buffers
; i
++) {
106 if ((ret
= av_fifo_realloc2(af
->buf
[i
], buf_size
)) < 0)
109 af
->allocated_samples
= nb_samples
;
113 int av_audio_fifo_write(AVAudioFifo
*af
, void **data
, int nb_samples
)
117 /* automatically reallocate buffers if needed */
118 if (av_audio_fifo_space(af
) < nb_samples
) {
119 int current_size
= av_audio_fifo_size(af
);
120 /* check for integer overflow in new size calculation */
121 if (INT_MAX
/ 2 - current_size
< nb_samples
)
122 return AVERROR(EINVAL
);
123 /* reallocate buffers */
124 if ((ret
= av_audio_fifo_realloc(af
, 2 * (current_size
+ nb_samples
))) < 0)
128 size
= nb_samples
* af
->sample_size
;
129 for (i
= 0; i
< af
->nb_buffers
; i
++) {
130 ret
= av_fifo_generic_write(af
->buf
[i
], data
[i
], size
, NULL
);
134 af
->nb_samples
+= nb_samples
;
139 int av_audio_fifo_read(AVAudioFifo
*af
, void **data
, int nb_samples
)
144 return AVERROR(EINVAL
);
145 nb_samples
= FFMIN(nb_samples
, af
->nb_samples
);
149 size
= nb_samples
* af
->sample_size
;
150 for (i
= 0; i
< af
->nb_buffers
; i
++) {
151 if ((ret
= av_fifo_generic_read(af
->buf
[i
], data
[i
], size
, NULL
)) < 0)
154 af
->nb_samples
-= nb_samples
;
159 int av_audio_fifo_drain(AVAudioFifo
*af
, int nb_samples
)
164 return AVERROR(EINVAL
);
165 nb_samples
= FFMIN(nb_samples
, af
->nb_samples
);
168 size
= nb_samples
* af
->sample_size
;
169 for (i
= 0; i
< af
->nb_buffers
; i
++)
170 av_fifo_drain(af
->buf
[i
], size
);
171 af
->nb_samples
-= nb_samples
;
176 void av_audio_fifo_reset(AVAudioFifo
*af
)
180 for (i
= 0; i
< af
->nb_buffers
; i
++)
181 av_fifo_reset(af
->buf
[i
]);
186 int av_audio_fifo_size(AVAudioFifo
*af
)
188 return af
->nb_samples
;
191 int av_audio_fifo_space(AVAudioFifo
*af
)
193 return af
->allocated_samples
- af
->nb_samples
;