57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
//go:build !disable_format_aac && cgo
|
|
|
|
package packetizer
|
|
|
|
import (
|
|
"git.gammaspectra.live/S.O.N.G/Kirika/audio/format/aac"
|
|
"git.gammaspectra.live/S.O.N.G/Kirika/audio/format/flac"
|
|
"git.gammaspectra.live/S.O.N.G/Kirika/test"
|
|
"io"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestPacketizeADTS(t *testing.T) {
|
|
t.Parallel()
|
|
fp, err := os.Open(test.TestSingleSample24)
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
defer fp.Close()
|
|
source, err := flac.NewFormat().Open(fp)
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
|
|
reader, writer := io.Pipe()
|
|
|
|
go func() {
|
|
defer writer.Close()
|
|
err = aac.NewFormat().Encode(source, writer, nil)
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
}()
|
|
|
|
pipePacketizer := NewAdtsPacketizer(reader)
|
|
packetCount := 0
|
|
packetBytes := 0
|
|
for {
|
|
packet := pipePacketizer.GetPacket()
|
|
if packet == nil {
|
|
break
|
|
}
|
|
packetCount++
|
|
packetBytes += len(packet.GetData())
|
|
}
|
|
if packetCount != 16920 {
|
|
t.Errorf("Wrong Packet Count %d != %d", packetCount, 16920)
|
|
}
|
|
if packetBytes != 6436973 {
|
|
t.Errorf("Wrong Packet Bytes %d != %d", packetBytes, 6436973)
|
|
}
|
|
}
|