Refcount the input frames directly instead of trying to free them after-the-fact...

[nageru] / h264encode.h

diff --git a/h264encode.h b/h264encode.h
index b14ca6f4690643d6281be45c0dbd27e19b49f3a5..9525d802c8875a8908e30dfa319b246087ac825c 100644 (file)
--- a/h264encode.h
+++ b/h264encode.h
@@ -29,18 +29,24 @@
 extern "C" {
 #include <libavformat/avformat.h>
 }
-#include <epoxy/egl.h>
+#include <epoxy/gl.h>
 #include <atomic>
+#include <condition_variable>
 #include <map>
 #include <memory>
 #include <mutex>
+#include <queue>
 #include <thread>
-#include <condition_variable>
+#include <vector>
 
-#include "pbo_frame_allocator.h"
+#include "bmusb.h"
 #include "context.h"
+#include "pbo_frame_allocator.h"
+#include "ref_counted_frame.h"
 #include "ref_counted_gl_sync.h"
 
+class QSurface;
+
 #define SURFACE_NUM 16 /* 16 surfaces for source YUV */
 
 class H264Encoder {
@@ -61,7 +67,7 @@ public:
        void 
 #endif
        bool begin_frame(GLuint *y_tex, GLuint *cbcr_tex);
-       void end_frame(RefCountedGLsync fence, const std::vector<FrameAllocator::Frame> &input_frames_to_release);
+       void end_frame(RefCountedGLsync fence, const std::vector<RefCountedFrame> &input_frames);
 
 private:
        struct storage_task {
@@ -93,7 +99,7 @@ private:
 
        struct PendingFrame {
                RefCountedGLsync fence;
-               std::vector<FrameAllocator::Frame> input_frames_to_release;
+               std::vector<RefCountedFrame> input_frames;
        };
        std::map<int, PendingFrame> pending_frames;
        QSurface *surface;