Add code for non-transcoded video streams (#277)

* Add code for non-transcoded video streams * Missed reset_frame_flag!
JuliaIO · Nov 22, 2020 · 78276d7 · 78276d7
1 parent 7e05dc3
commit 78276d7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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