mirror of https://bitbucket.org/ausocean/av.git
container/mts: removed audio_test.go
This commit is contained in:
parent
dbec11f3b4
commit
a805dc13a6
|
@ -1,144 +0,0 @@
|
|||
/*
|
||||
NAME
|
||||
audio_test.go
|
||||
|
||||
AUTHOR
|
||||
Trek Hopton <trek@ausocean.org>
|
||||
|
||||
LICENSE
|
||||
audio_test.go is Copyright (C) 2017-2019 the Australian Ocean Lab (AusOcean)
|
||||
|
||||
It is free software: you can redistribute it and/or modify them
|
||||
under the terms of the GNU General Public License as published by the
|
||||
Free Software Foundation, either version 3 of the License, or (at your
|
||||
option) any later version.
|
||||
|
||||
It is distributed in the hope that it will be useful, but WITHOUT
|
||||
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||||
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
||||
for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License in gpl.txt.
|
||||
If not, see http://www.gnu.org/licenses.
|
||||
*/
|
||||
|
||||
package mts
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"testing"
|
||||
|
||||
"github.com/Comcast/gots/packet"
|
||||
"github.com/Comcast/gots/pes"
|
||||
|
||||
"bitbucket.org/ausocean/av/container/mts/meta"
|
||||
)
|
||||
|
||||
type nopCloser struct{ io.Writer }
|
||||
|
||||
func (nopCloser) Close() error { return nil }
|
||||
|
||||
// TestEncodePcm tests the mpegts encoder's ability to encode pcm audio data.
|
||||
// It reads and encodes input pcm data into mpegts, then decodes the mpegts and compares the result to the input pcm.
|
||||
func TestEncodePcm(t *testing.T) {
|
||||
Meta = meta.New()
|
||||
|
||||
var buf bytes.Buffer
|
||||
sampleRate := 48000
|
||||
sampleSize := 2
|
||||
blockSize := 16000
|
||||
writeFreq := float64(sampleRate*sampleSize) / float64(blockSize)
|
||||
e := NewEncoder(nopCloser{&buf}, writeFreq, Audio)
|
||||
|
||||
inPath := "../../../test/test-data/av/input/sweep_400Hz_20000Hz_-3dBFS_5s_48khz.pcm"
|
||||
inPcm, err := ioutil.ReadFile(inPath)
|
||||
if err != nil {
|
||||
t.Errorf("unable to read file: %v", err)
|
||||
}
|
||||
|
||||
// Break pcm into blocks and encode to mts and get the resulting bytes.
|
||||
for i := 0; i < len(inPcm); i += blockSize {
|
||||
if len(inPcm)-i < blockSize {
|
||||
block := inPcm[i:]
|
||||
_, err = e.Write(block)
|
||||
if err != nil {
|
||||
t.Errorf("unable to write block: %v", err)
|
||||
}
|
||||
} else {
|
||||
block := inPcm[i : i+blockSize]
|
||||
_, err = e.Write(block)
|
||||
if err != nil {
|
||||
t.Errorf("unable to write block: %v", err)
|
||||
}
|
||||
}
|
||||
}
|
||||
clip := buf.Bytes()
|
||||
|
||||
// Get the first MTS packet to check
|
||||
var pkt packet.Packet
|
||||
pesPacket := make([]byte, 0, blockSize)
|
||||
got := make([]byte, 0, len(inPcm))
|
||||
i := 0
|
||||
if i+PacketSize <= len(clip) {
|
||||
copy(pkt[:], clip[i:i+PacketSize])
|
||||
}
|
||||
|
||||
// Loop through MTS packets until all the audio data from PES packets has been retrieved
|
||||
for i+PacketSize <= len(clip) {
|
||||
|
||||
// Check MTS packet
|
||||
if !(pkt.PID() == audioPid) {
|
||||
i += PacketSize
|
||||
if i+PacketSize <= len(clip) {
|
||||
copy(pkt[:], clip[i:i+PacketSize])
|
||||
}
|
||||
continue
|
||||
}
|
||||
if !pkt.PayloadUnitStartIndicator() {
|
||||
i += PacketSize
|
||||
if i+PacketSize <= len(clip) {
|
||||
copy(pkt[:], clip[i:i+PacketSize])
|
||||
}
|
||||
} else {
|
||||
// Copy the first MTS payload
|
||||
payload, err := pkt.Payload()
|
||||
if err != nil {
|
||||
t.Errorf("unable to get MTS payload: %v", err)
|
||||
}
|
||||
pesPacket = append(pesPacket, payload...)
|
||||
|
||||
i += PacketSize
|
||||
if i+PacketSize <= len(clip) {
|
||||
copy(pkt[:], clip[i:i+PacketSize])
|
||||
}
|
||||
|
||||
// Copy the rest of the MTS payloads that are part of the same PES packet
|
||||
for (!pkt.PayloadUnitStartIndicator()) && i+PacketSize <= len(clip) {
|
||||
payload, err = pkt.Payload()
|
||||
if err != nil {
|
||||
t.Errorf("unable to get MTS payload: %v", err)
|
||||
}
|
||||
pesPacket = append(pesPacket, payload...)
|
||||
|
||||
i += PacketSize
|
||||
if i+PacketSize <= len(clip) {
|
||||
copy(pkt[:], clip[i:i+PacketSize])
|
||||
}
|
||||
}
|
||||
}
|
||||
// Get the audio data from the current PES packet
|
||||
pesHeader, err := pes.NewPESHeader(pesPacket)
|
||||
if err != nil {
|
||||
t.Errorf("unable to read PES packet: %v", err)
|
||||
}
|
||||
got = append(got, pesHeader.Data()...)
|
||||
pesPacket = pesPacket[:0]
|
||||
}
|
||||
|
||||
// Compare data from MTS with original data.
|
||||
if !bytes.Equal(got, inPcm) {
|
||||
t.Error("data decoded from mts did not match input data")
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue