mirror of https://bitbucket.org/ausocean/av.git
138 lines
3.3 KiB
Go
138 lines
3.3 KiB
Go
/*
|
|
NAME
|
|
parser_test.go
|
|
|
|
DESCRIPTION
|
|
See Readme.md
|
|
|
|
AUTHOR
|
|
Saxon Nelson-Milton <saxon@ausocean.org>
|
|
|
|
LICENSE
|
|
parser_test.go is Copyright (C) 2017 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
|
|
along with revid in gpl.txt. If not, see http://www.gnu.org/licenses.
|
|
*/
|
|
|
|
package parser
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"strconv"
|
|
"testing"
|
|
)
|
|
|
|
const (
|
|
mjpegInputFileName = "testInput/testInput.avi"
|
|
h264InputFileName = "../../test/test-data/av/input/betterInput.h264"
|
|
)
|
|
|
|
func TestH264Parser(t *testing.T) {
|
|
log.SetOutput(os.Stderr)
|
|
log.Println("Opening input file!")
|
|
inputFile, err := os.Open(h264InputFileName)
|
|
if err != nil {
|
|
t.Errorf("Should not have got error opening file!")
|
|
return
|
|
}
|
|
|
|
log.Println("Getting file stats!")
|
|
stats, err := inputFile.Stat()
|
|
if err != nil {
|
|
t.Errorf("Could not get input file stats!")
|
|
return
|
|
}
|
|
|
|
log.Println("Creating space for file data and reading!")
|
|
data := make([]byte, stats.Size())
|
|
_, err = inputFile.Read(data)
|
|
if err != nil {
|
|
t.Errorf("Should not have got read error!")
|
|
return
|
|
}
|
|
|
|
// Create 'parser' and start it up
|
|
log.Println("Creating parser!")
|
|
parser := NewH264Parser()
|
|
parser.SetOutputChan(make(chan []byte, 10000))
|
|
parser.Start()
|
|
|
|
count := 0
|
|
for i := range data {
|
|
// If we have a whole frame, write it to a file for inspection otherwise do
|
|
// nothing
|
|
select {
|
|
case parser.InputChan() <- data[i]:
|
|
case frame:=<-parser.OutputChan():
|
|
outputFile, err := os.Create("testOutput/" + strconv.Itoa(count) + ".h264_frame")
|
|
if err != nil {
|
|
t.Errorf("Should not have got error creating output file!")
|
|
return
|
|
}
|
|
outputFile.Write(frame)
|
|
outputFile.Close()
|
|
count++
|
|
if count > 4 {
|
|
return
|
|
}
|
|
default:
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/*
|
|
func TestMJPEGParser(t *testing.T) {
|
|
log.Println("Opening input file!")
|
|
// Open the input file
|
|
inputFile, err := os.Open(testInputFileName)
|
|
if err != nil {
|
|
t.Errorf("Should not have got error opening file!")
|
|
}
|
|
log.Println("Getting file stats!")
|
|
stats, err := inputFile.Stat()
|
|
if err != nil {
|
|
t.Errorf("Could not get input file stats!")
|
|
return
|
|
}
|
|
log.Println("Creating space for file data!")
|
|
data := make([]byte, stats.Size())
|
|
_, err = inputFile.Read(data)
|
|
if err != nil {
|
|
t.Errorf("Should not have got read error!")
|
|
return
|
|
}
|
|
log.Println("Creating parser!")
|
|
parser := NewMJPEGParser(len(data) + 1)
|
|
parser.SetOutputChan(make(chan []byte, 10000))
|
|
parser.Start()
|
|
log.Printf("len(data): %v\n", len(data))
|
|
for i := range data {
|
|
parser.GetInputChan() <- data[i]
|
|
}
|
|
log.Println("Writing jpegs to files!")
|
|
for i := 0; len(parser.GetOutputChan()) > 0; i++ {
|
|
// Open a new output file
|
|
outputFile, err := os.Create("testOutput/image" + strconv.Itoa(i) + ".jpeg")
|
|
if err != nil {
|
|
t.Errorf("Should not have got error creating output file!")
|
|
return
|
|
}
|
|
outputFile.Write(<-parser.GetOutputChan())
|
|
outputFile.Close()
|
|
}
|
|
}
|
|
*/
|