X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=nageru%2Fpbo_frame_allocator.h;h=a7ae92e93a9bc0620af7b8606a143ff1686128ab;hb=a839022c035b3d9387feabc02843c166ac78b469;hp=eead7f4a32f979c8203a184f535862c6f1dab0e2;hpb=eeda8995329601f9f4e35047358400833eeae68e;p=nageru

diff --git a/nageru/pbo_frame_allocator.h b/nageru/pbo_frame_allocator.h
index eead7f4..a7ae92e 100644
--- a/nageru/pbo_frame_allocator.h
+++ b/nageru/pbo_frame_allocator.h
@@ -11,6 +11,9 @@
 #include <movit/ycbcr.h>
 
 #include "bmusb/bmusb.h"
+#include "mjpeg_encoder.h"
+
+class MJPEGEncoder;
 
 // An allocator that allocates straight into OpenGL pinned memory.
 // Meant for video frames only. We use a queue rather than a stack,
@@ -22,12 +25,15 @@ public:
 	PBOFrameAllocator(bmusb::PixelFormat pixel_format,
 	                  size_t frame_size,
 	                  GLuint width, GLuint height,
+	                  unsigned card_index,
+	                  MJPEGEncoder *mjpeg_encoder = nullptr,
 	                  size_t num_queued_frames = 16,
 	                  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;
+	Frame create_frame(size_t width, size_t height, size_t stride) override;
 	void release_frame(Frame frame) override;
 
 	struct Userdata {
@@ -52,12 +58,36 @@ public:
 		GLuint last_v210_width[2];  // PixelFormat_10BitYCbCr.
 		bool last_interlaced, last_has_signal, last_is_connected;
 		unsigned last_frame_rate_nom, last_frame_rate_den;
+		bool has_last_subtitle = false;
+		std::string last_subtitle;
+
+		// These are the source of the “data_copy” member in Frame,
+		// used for MJPEG encoding. There are three possibilities:
+		//
+		//  - MJPEG encoding is not active (at all, or for this specific
+		//    card). Then data_copy is nullptr, and what's in here
+		//    does not matter at all.
+		//  - We can encode directly into VA-API buffers (ie., VA-API
+		//    is active, and nothing strange happened wrt. strides);
+		//    then va_resources, va_resources_release and va_image
+		//    are fetched from MJPEGEncoder at create_frame() and released
+		//    back when the frame is uploaded (or would have been).
+		//    In this case, data_copy points into the mapped VAImage.
+		//  - If not, data_copy points to data_copy_malloc, and is copied
+		//    from there into VA-API buffers (by MJPEGEncoder) if needed.
+		enum { FROM_MALLOC, FROM_VA_API } data_copy_current_src;
+		uint8_t *data_copy_malloc;
+		MJPEGEncoder::VAResources va_resources;
+		MJPEGEncoder::ReleaseVAResources va_resources_release;
+		VAImage va_image;
 	};
 
 private:
 	void init_frame(size_t frame_idx, size_t frame_size, GLuint width, GLuint height, GLenum permissions, GLenum map_bits);
 	void destroy_frame(Frame *frame);
 
+	unsigned card_index;
+	MJPEGEncoder *mjpeg_encoder;
 	bmusb::PixelFormat pixel_format;
 	std::mutex freelist_mutex;
 	std::queue<Frame> freelist;