videoio(ffmpeg): fix handling of AVERROR_EOF

decoder should be properly flushed after that
5 years ago · f1dd5e49c5
parent c6b60e219d
commit f1dd5e49c5
1 changed files with 13 additions and 2 deletions
--- a/modules/videoio/src/cap_ffmpeg_impl.hpp
+++ b/modules/videoio/src/cap_ffmpeg_impl.hpp
@ -1215,9 +1215,20 @@ bool CvCapture_FFMPEG::grabFrame()
 #endif

        int ret = av_read_frame(ic, &packet);
-        if (ret == AVERROR(EAGAIN)) continue;

-        /* else if (ret < 0) break; */
+        if (ret == AVERROR(EAGAIN))
+            continue;
+
+        if (ret == AVERROR_EOF)
+        {
+            if (rawMode)
+                break;
+
+            // flush cached frames from video decoder
+            packet.data = NULL;
+            packet.size = 0;
+            packet.stream_index = video_stream;
+        }

        if( packet.stream_index != video_stream )
        {