cuda: clear the frame queue on flush

xiongbinsh · Mar 28, 2014 · 1f7c0fc · 1f7c0fc
1 parent 53ce6b8
commit 1f7c0fc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/QtAV/AVDecoder.h b/src/QtAV/AVDecoder.h
@@ -45,7 +45,7 @@ class Q_AV_EXPORT AVDecoder
     virtual bool open();
     virtual bool close();
     bool isOpen() const;
-    void flush();
+    virtual void flush();
     void setCodecContext(AVCodecContext* codecCtx); //protected
     AVCodecContext* codecContext() const;
     // force a codec

diff --git a/src/VideoDecoderCUDA.cpp b/src/VideoDecoderCUDA.cpp
@@ -74,6 +74,7 @@ class VideoDecoderCUDA : public VideoDecoder
 public:
     VideoDecoderCUDA();
     virtual ~VideoDecoderCUDA();
+    virtual void flush();
     virtual bool prepare();
     virtual bool decode(const QByteArray &encoded);
     virtual VideoFrame frame();
@@ -460,6 +461,12 @@ VideoDecoderCUDA::~VideoDecoderCUDA()
 {
 }
 
+void VideoDecoderCUDA::flush()
+{
+    DPTR_D(VideoDecoderCUDA);
+    d.frame_queue.clear();
+}
+
 bool VideoDecoderCUDA::prepare()
 {
     //TODO: destroy decoder