Added getters for structs

This commit is contained in:
Alex Eidt 2022-04-16 10:45:14 -07:00
parent fb2c0b201d
commit 4fef48055a
6 changed files with 257 additions and 126 deletions

View file

@ -25,8 +25,8 @@ func TestVideoMetaData(t *testing.T) {
assertEquals(video.duration, 3.366667)
assertEquals(video.fps, float64(30))
assertEquals(video.codec, "h264")
assertEquals(video.audio_codec, "aac")
assertEquals(video.pix_fmt, "yuv420p")
assertEquals(video.audioCodec, "aac")
assertEquals(video.pixfmt, "yuv420p")
assertEquals(len(video.framebuffer), 0)
if video.pipe != nil {
@ -63,17 +63,17 @@ func TestVideoWriting(t *testing.T) {
testWriting := func(input, output string, audio bool) {
video := NewVideo(input)
options := Options{
fps: video.fps,
bitrate: video.bitrate,
codec: video.codec,
FPS: video.FPS(),
Bitrate: video.Bitrate(),
Codec: video.Codec(),
}
if audio {
options.audio = input
options.Audio = input
}
writer := NewVideoWriter(output, video.width, video.height, &options)
for video.Read() {
writer.Write(video.framebuffer)
writer.Write(video.FrameBuffer())
}
writer.Close()
@ -86,6 +86,30 @@ func TestVideoWriting(t *testing.T) {
fmt.Println("Video Writing (without Audio) Test Passed")
}
func TestCameraIO(t *testing.T) {
webcam := NewCamera(0)
options := Options{FPS: webcam.FPS()}
writer := NewVideoWriter("test/camera.mp4", webcam.width, webcam.height, &options)
count := 0
for webcam.Read() {
frame := webcam.FrameBuffer()
writer.Write(frame)
count++
if count > 100 {
break
}
}
webcam.Close()
writer.Close()
//os.Remove("test/camera.mp4")
fmt.Println("Camera IO Test Passed")
}
func TestFFprobe(t *testing.T) {
koalaVideo := ffprobe("test/koala.mp4", "v")
assertEquals(koalaVideo["width"], "480")