#define _H264ENCODE_H
extern "C" {
+#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
}
-#include <epoxy/egl.h>
+#include <epoxy/gl.h>
+#include <stdbool.h>
+#include <stdint.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/bmusb.h"
#include "context.h"
+#include "pbo_frame_allocator.h"
+#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);
};
void
#endif
+ void add_audio(int64_t pts, std::vector<float> audio); // Needs to come before end_frame() of same pts.
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, int64_t pts, const std::vector<RefCountedFrame> &input_frames);
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;
};
void copy_thread_func();
void storage_task_thread();
- void storage_task_enqueue(unsigned long long display_order, unsigned long long encode_order, int frame_type);
- int save_codeddata(unsigned long long display_order, unsigned long long encode_order, int frame_type);
+ void storage_task_enqueue(storage_task task);
+ int save_codeddata(storage_task task);
std::thread copy_thread, storage_thread;
struct PendingFrame {
RefCountedGLsync fence;
- std::vector<FrameAllocator::Frame> input_frames_to_release;
+ std::vector<RefCountedFrame> input_frames;
+ int64_t pts;
};
- std::map<int, PendingFrame> pending_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
QSurface *surface;
- AVFormatContext *avctx;
- AVStream *avstream;
+ AVCodecContext *context_audio;
+ HTTPD *httpd;
};
#endif