Add audio FIFO

audio_fifo.go

@@ -0,0 +1,52 @@
+package astiav
+
+//#include <libavutil/audio_fifo.h>
+//#include <stdlib.h>
+import "C"
+import "unsafe"
+
+// https://github.com/FFmpeg/FFmpeg/blob/n7.0/libavutil/audio_fifo.c#L37
+type AudioFifo struct {
+	c *C.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) Realloc(nbSamples int) error {
+	return newError(C.av_audio_fifo_realloc(a.c, C.int(nbSamples)))
+}
+
+func (a *AudioFifo) Size() int {
+	return int(C.av_audio_fifo_size(a.c))
+}
+
+func (a *AudioFifo) Space() int {
+	return int(C.av_audio_fifo_space(a.c))
+}
+
+func (a *AudioFifo) Write(f *Frame) (int, error) {
+	frameRawData := (**uint8)(unsafe.Pointer(&f.c.data[0]))
+	ret := int(C.av_audio_fifo_write(a.c, (*unsafe.Pointer)(unsafe.Pointer(frameRawData)), C.int(f.NbSamples())))
+	if ret < 0 {
+		return ret, newError(C.int(ret))
+	}
+	return ret, nil
+}
+
+func (a *AudioFifo) Read(f *Frame) int {
+	frameRawData := (**uint8)(unsafe.Pointer(&f.c.data[0]))
+	return int(C.av_audio_fifo_read(a.c, (*unsafe.Pointer)(unsafe.Pointer(frameRawData)), C.int(f.NbSamples())))
+}
+
+func (a *AudioFifo) Free() {
+	C.av_audio_fifo_free(a.c)
+}

audio_fifo_test.go

@@ -0,0 +1,44 @@
+package astiav
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestAudioFIFO(t *testing.T) {
+	af := AllocAudioFifo(
+		SampleFormatFltp,
+		2,
+		960)
+	defer af.Free()
+	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, err := af.Write(writeFrame)
+	require.Equal(t, err, nil)
+	require.Equal(t, writeSamples, written)
+	read := af.Read(readFrame)
+	require.Equal(t, readSamples, read)
+	require.Equal(t, af.Size(), writeSamples-readSamples)
+	reallocSamples := 3000
+	err = af.Realloc(reallocSamples)
+	require.Equal(t, err, nil)
+	expectedAfSize := writeSamples - readSamples
+	require.Equal(t, af.Space(), reallocSamples-expectedAfSize)
+	// It still has the same amount of data
+	require.Equal(t, af.Size(), expectedAfSize)
+}

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]))
+}