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