1 // A wrapper around x264, to encode video in higher quality than Quick Sync
2 // can give us. We maintain a queue of uncompressed Y'CbCr frames (of 50 frames,
3 // so a little under 100 MB at 720p), then have a separate thread pull out
4 // those threads as fast as we can to give it to x264 for encoding.
6 // TODO: We use x264's “speedcontrol” patch if available, so that quality is
7 // automatically scaled up or down to content and available CPU time.
9 // The encoding threads are niced down because mixing is more important than
10 // encoding; if we lose frames in mixing, we'll lose frames to disk _and_
11 // to the stream, as where if we lose frames in encoding, we'll lose frames
12 // to the stream only, so the latter is strictly better. More importantly,
13 // this allows speedcontrol (when implemented) to do its thing without
14 // disturbing the mixer.
17 #define _X264ENCODE_H 1
22 #include <condition_variable>
30 #include <libavformat/avformat.h>
34 class X264SpeedControl;
38 X264Encoder(AVOutputFormat *oformat); // Does not take ownership.
40 // Called after the last frame. Will block; once this returns,
41 // the last data is flushed.
44 // Must be called before first frame. Does not take ownership.
45 void set_mux(Mux *mux) { this->mux = mux; }
47 // <data> is taken to be raw NV12 data of WIDTHxHEIGHT resolution.
49 void add_frame(int64_t pts, int64_t duration, const uint8_t *data);
51 std::string get_global_headers() const { return global_headers; }
55 int64_t pts, duration;
58 void encoder_thread_func();
60 void encode_frame(QueuedFrame qf);
62 // One big memory chunk of all 50 (or whatever) frames, allocated in
63 // the constructor. All data functions just use pointers into this
65 std::unique_ptr<uint8_t[]> frame_pool;
68 bool wants_global_headers;
70 std::string global_headers;
71 std::string buffered_sei; // Will be output before first frame, if any.
73 std::thread encoder_thread;
74 std::atomic<bool> should_quit{false};
76 std::unique_ptr<X264SpeedControl> speed_control;
78 // Protects everything below it.
81 // Frames that are not being encoded or waiting to be encoded,
82 // so that add_frame() can use new ones.
83 std::queue<uint8_t *> free_frames;
85 // Frames that are waiting to be encoded (ie., add_frame() has been
86 // called, but they are not picked up for encoding yet).
87 std::queue<QueuedFrame> queued_frames;
89 // Whenever the state of <queued_frames> changes.
90 std::condition_variable queued_frames_nonempty;
93 #endif // !defined(_X264ENCODE_H)