X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=quicksync_encoder.h;h=16fd3d179cd048a2d6b29e447d3209a673ecebca;hb=1fc8b5d92fce7ac0b7e87ed3087e393774fd4b3c;hp=495ddae48d998c00f1a927515c5330d0f0a44033;hpb=90d0745245f83999e88bc50cad57be3e4a1c7439;p=nageru

diff --git a/quicksync_encoder.h b/quicksync_encoder.h
index 495ddae..16fd3d1 100644
--- a/quicksync_encoder.h
+++ b/quicksync_encoder.h
@@ -27,35 +27,48 @@
 #define _H264ENCODE_H
 
 #include <epoxy/gl.h>
+#include <stdbool.h>
 #include <stdint.h>
 #include <atomic>
+#include <functional>
 #include <memory>
 #include <string>
 #include <vector>
 
-#include "mux.h"
+extern "C" {
+#include <libavformat/avformat.h>
+}
+
 #include "ref_counted_frame.h"
 #include "ref_counted_gl_sync.h"
 
-class QuickSyncEncoderImpl;
+class AudioEncoder;
+class DiskSpaceEstimator;
+class Mux;
 class QSurface;
+class QuickSyncEncoderImpl;
+class RefCountedFrame;
+class X264Encoder;
+
+namespace movit {
+class ResourcePool;
+}  // namespace movit
 
 // This is just a pimpl, because including anything X11-related in a .h file
 // tends to trip up Qt. All the real logic is in QuickSyncEncoderImpl, defined in the
 // .cpp file.
 class QuickSyncEncoder {
 public:
-        QuickSyncEncoder(QSurface *surface, const std::string &va_display, int width, int height, Mux *stream_mux);
+        QuickSyncEncoder(const std::string &filename, movit::ResourcePool *resource_pool, QSurface *surface, const std::string &va_display, int width, int height, AVOutputFormat *oformat, X264Encoder *x264_encoder, DiskSpaceEstimator *disk_space_estimator);
         ~QuickSyncEncoder();
 
+	void set_stream_mux(Mux *mux);  // Does not take ownership. Must be called unless x264 is used for the stream.
 	void add_audio(int64_t pts, std::vector<float> audio);
 	bool begin_frame(GLuint *y_tex, GLuint *cbcr_tex);
 	RefCountedGLsync end_frame(int64_t pts, int64_t duration, const std::vector<RefCountedFrame> &input_frames);
-	void shutdown();  // Blocking.
-
-	// You can only have one going at the same time.
-	void open_output_file(const std::string &filename);
-	void close_output_file();
+	void shutdown();  // Blocking. Does not require an OpenGL context.
+	void release_gl_resources();  // Requires an OpenGL context. Must be run after shutdown.
+	int64_t global_delay() const;  // So we never get negative dts.
 
 private:
 	std::unique_ptr<QuickSyncEncoderImpl> impl;