From 78276d706a63879867080ef6d49e48a1fceae490 Mon Sep 17 00:00:00 2001
From: David Barton <davebarton@cityinthesky.co.uk>
Date: Sun, 22 Nov 2020 16:12:36 +0000
Subject: [PATCH] Add code for non-transcoded video streams (#277)

* Add code for non-transcoded video streams

* Missed reset_frame_flag!
---
 src/avio.jl | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/avio.jl b/src/avio.jl
index b5308105..8dfb7459 100644
--- a/src/avio.jl
+++ b/src/avio.jl
@@ -405,7 +405,7 @@ function retrieve(r::VideoReader{NO_TRANSCODE}) # false=don't transcode
 
     # TODO: set actual dimensions ?
     buf_sz = avpicture_get_size(r.format, r.width, r.height)
-    buf = Array(UInt8, buf_sz)
+    buf = Array{UInt8}(undef, buf_sz)
 
     retrieve!(r, buf)
 end
@@ -476,6 +476,12 @@ function retrieve!(r::VideoReader{NO_TRANSCODE}, buf::VidArray{T}) where T <: Ei
     if !bufsize_check(r, buf)
         error("Buffer is the wrong size")
     end
+
+    unsafe_copyto!(pointer(buf), r.aVideoFrame[1].data[1], sizeof(buf))
+
+    reset_frame_flag!(r)
+
+    return buf
 end
 
 # Utility functions