X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=pbo_frame_allocator.h;h=aedae3b99cc89d5d8085c95694d73ea95a49d458;hb=a98732df9454028ddaa54a9d07b5d7513767bfe9;hp=825d01f5034429db4a0f8a1baa2e67a31f3f3cbe;hpb=d147175aaa3e5abfe271940c4518e9ef8dbbc932;p=nageru

diff --git a/pbo_frame_allocator.h b/pbo_frame_allocator.h
index 825d01f..aedae3b 100644
--- a/pbo_frame_allocator.h
+++ b/pbo_frame_allocator.h
@@ -2,10 +2,12 @@
 #define _PBO_FRAME_ALLOCATOR 1
 
 #include <epoxy/gl.h>
+#include <stddef.h>
+#include <memory>
 #include <mutex>
 #include <queue>
 
-#include "bmusb.h"
+#include "bmusb/bmusb.h"
 
 // An allocator that allocates straight into OpenGL pinned memory.
 // Meant for video frames only. We use a queue rather than a stack,
@@ -15,13 +17,20 @@ public:
 	// Note: You need to have an OpenGL context when calling
 	// the constructor.
 	PBOFrameAllocator(size_t frame_size,
+	                  GLuint width, GLuint height,
 	                  size_t num_queued_frames = 16,  // FIXME: should be 6
 	                  GLenum buffer = GL_PIXEL_UNPACK_BUFFER_ARB,
 	                  GLenum permissions = GL_MAP_WRITE_BIT,
 	                  GLenum map_bits = GL_MAP_FLUSH_EXPLICIT_BIT);
 	~PBOFrameAllocator() override;
-        Frame alloc_frame() override;
-        void release_frame(Frame frame) override;
+	Frame alloc_frame() override;
+	void release_frame(Frame frame) override;
+
+	struct Userdata {
+		GLuint pbo;
+		GLuint tex_y, tex_cbcr;
+		GLuint last_width, last_height;
+	};
 
 private:
 	size_t frame_size;
@@ -29,6 +38,7 @@ private:
 	std::mutex freelist_mutex;
 	std::queue<Frame> freelist;
 	GLenum buffer;
+	std::unique_ptr<Userdata[]> userdata;
 };
 
 #endif  // !defined(_PBO_FRAME_ALLOCATOR)