X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=h264encode.h;h=eadfca7f50a9e8bd19ddb011793dab46ac9f3c7e;hb=a1f1d6374a7c519c7a8b412578c9b88d64edddd4;hp=ec49fb71dde19f801b5a4f95f106c1cdc8089a04;hpb=02083dca89292d6ba95642cf8957f1cee8826313;p=nageru

diff --git a/h264encode.h b/h264encode.h
index ec49fb7..eadfca7 100644
--- a/h264encode.h
+++ b/h264encode.h
@@ -27,9 +27,12 @@
 #define _H264ENCODE_H
 
 extern "C" {
+#include <libavcodec/avcodec.h>
 #include <libavformat/avformat.h>
 }
 #include <epoxy/gl.h>
+#include <stdbool.h>
+#include <stdint.h>
 #include <atomic>
 #include <condition_variable>
 #include <map>
@@ -45,13 +48,14 @@ extern "C" {
 #include "ref_counted_frame.h"
 #include "ref_counted_gl_sync.h"
 
+class HTTPD;
 class QSurface;
 
 #define SURFACE_NUM 16 /* 16 surfaces for source YUV */
 
 class H264Encoder {
 public:
-	H264Encoder(QSurface *surface, int width, int height, const char *output_filename);
+	H264Encoder(QSurface *surface, int width, int height, HTTPD *httpd);
 	~H264Encoder();
 	//void add_frame(FrameAllocator::Frame frame, GLsync fence);
 
@@ -73,15 +77,22 @@ public:
 private:
 	struct storage_task {
 		unsigned long long display_order;
-		unsigned long long encode_order;
 		int frame_type;
 		std::vector<float> audio;
+		int64_t pts, dts;
+	};
+	struct PendingFrame {
+		RefCountedGLsync fence;
+		std::vector<RefCountedFrame> input_frames;
+		int64_t pts;
 	};
 
 	void copy_thread_func();
+	void encode_frame(PendingFrame frame, int encoding_frame_num, int display_frame_num, int gop_start_display_frame_num,
+	                  int frame_type, int64_t pts, int64_t dts);
 	void storage_task_thread();
 	void storage_task_enqueue(storage_task task);
-	int save_codeddata(storage_task task);
+	void save_codeddata(storage_task task);
 
 	std::thread copy_thread, storage_thread;
 
@@ -99,18 +110,12 @@ private:
 	//int ;
 	int current_storage_frame;
 
-	struct PendingFrame {
-		RefCountedGLsync fence;
-		std::vector<RefCountedFrame> input_frames;
-	};
 	std::map<int, PendingFrame> pending_video_frames;  // under frame_queue_mutex
 	std::map<int64_t, std::vector<float>> pending_audio_frames;  // under frame_queue_mutex
-	std::map<int, int64_t> timestamps;  // under frame_queue_mutex
 	QSurface *surface;
 
-	AVFormatContext *avctx;
-	AVStream *avstream_video;
-	AVStream *avstream_audio;
+	AVCodecContext *context_audio;
+	HTTPD *httpd;
 };
 
 #endif