Add audio FIFO

audio_fifo.go

@@ -0,0 +1,46 @@
+package astiav
+
+//#cgo pkg-config: libavutil
+//#include <libavutil/audio_fifo.h>
+//#include <stdlib.h>
+import "C"
+import "unsafe"
+
+type AudioFifo struct {
+	c *C.struct_AVAudioFifo
+}
+
+func newAudioFifoFromC(c *C.struct_AVAudioFifo) *AudioFifo {
+	if c == nil {
+		return nil
+	}
+	return &AudioFifo{c: c}
+}
+
+func AllocAudioFifo(sampleFmt SampleFormat, channels int, nbSamples int) *AudioFifo {
+	return newAudioFifoFromC(C.av_audio_fifo_alloc(C.enum_AVSampleFormat(sampleFmt), C.int(channels), C.int(nbSamples)))
+}
+
+func (a *AudioFifo) AudioFifoRealloc(nbSamples int) int {
+	return int(C.av_audio_fifo_realloc((*C.struct_AVAudioFifo)(a.c), C.int(nbSamples)))
+}
+
+func (a *AudioFifo) AudioFifoSize() int {
+	return int(C.av_audio_fifo_size((*C.struct_AVAudioFifo)(a.c)))
+}
+
+func (a *AudioFifo) AudioFifoSpace() int {
+	return int(C.av_audio_fifo_space((*C.struct_AVAudioFifo)(a.c)))
+}
+
+func (a *AudioFifo) AudioFifoWrite(data **uint8, nbSamples int) int {
+	return int(C.av_audio_fifo_write((*C.struct_AVAudioFifo)(a.c), (*unsafe.Pointer)(unsafe.Pointer(data)), C.int(nbSamples)))
+}
+
+func (a *AudioFifo) AudioFifoRead(data **uint8, nbSamples int) int {
+	return int(C.av_audio_fifo_read((*C.struct_AVAudioFifo)(a.c), (*unsafe.Pointer)(unsafe.Pointer(data)), C.int(nbSamples)))
+}
+
+func (a *AudioFifo) AudioFifoFree() {
+	C.av_audio_fifo_free((*C.struct_AVAudioFifo)(a.c))
+}

audio_fifo_test.go

@@ -0,0 +1,36 @@
+package astiav
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestAudioFIFO(t *testing.T) {
+	audioFifo := AllocAudioFifo(
+		SampleFormatFltp,
+		2,
+		960)
+
+	writeSamples := 1024
+	readSamples := 120
+	writeFrame := AllocFrame()
+	writeFrame.SetNbSamples(writeSamples)
+	writeFrame.SetChannelLayout(ChannelLayoutStereo)
+	writeFrame.SetSampleFormat(SampleFormatFltp)
+	writeFrame.SetSampleRate(48000)
+	writeFrame.AllocBuffer(0)
+
+	readFrame := AllocFrame()
+	readFrame.SetNbSamples(readSamples)
+	readFrame.SetChannelLayout(ChannelLayoutStereo)
+	readFrame.SetSampleFormat(SampleFormatFltp)
+	readFrame.SetSampleRate(48000)
+	readFrame.AllocBuffer(0)
+
+	written := audioFifo.AudioFifoWrite(writeFrame.DataPtr(), writeFrame.NbSamples())
+	require.Equal(t, writeSamples, written)
+	read := audioFifo.AudioFifoRead(readFrame.DataPtr(), readFrame.NbSamples())
+	require.Equal(t, readSamples, read)
+	require.Equal(t, audioFifo.AudioFifoSize(), writeSamples-readSamples)
+}

frame.go

@@ -217,3 +217,7 @@ func (f *Frame) MoveRef(src *Frame) {
 func (f *Frame) UnsafePointer() unsafe.Pointer {
 	return unsafe.Pointer(f.c)
 }
+
+func (f *Frame) DataPtr() **uint8 {
+	return (**uint8)(unsafe.Pointer(&f.c.data[0]))
+}