1 // Hardware H.264 encoding via VAAPI. Also orchestrates the H.264 encoding
2 // in general; this is unfortunate, and probably needs a cleanup. In particular,
3 // even if you don't actually use Quick Sync for anything, this class
4 // (or actually, QuickSyncEncoderImpl) still takes on a pretty central role.
6 // Heavily modified based on example code by Intel. Intel's original copyright
7 // and license is reproduced below:
9 // Copyright (c) 2007-2013 Intel Corporation. All Rights Reserved.
11 // Permission is hereby granted, free of charge, to any person obtaining a
12 // copy of this software and associated documentation files (the
13 // "Software"), to deal in the Software without restriction, including
14 // without limitation the rights to use, copy, modify, merge, publish,
15 // distribute, sub license, and/or sell copies of the Software, and to
16 // permit persons to whom the Software is furnished to do so, subject to
17 // the following conditions:
19 // The above copyright notice and this permission notice (including the
20 // next paragraph) shall be included in all copies or substantial portions
23 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
24 // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
25 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
26 // IN NO EVENT SHALL PRECISION INSIGHT AND/OR ITS SUPPLIERS BE LIABLE FOR
27 // ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
28 // TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
29 // SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
35 #include <movit/image_format.h>
43 #include <libavformat/avformat.h>
46 #include "shared/ref_counted_gl_sync.h"
48 class DiskSpaceEstimator;
51 class QuickSyncEncoderImpl;
52 class RefCountedFrame;
59 // This is just a pimpl, because including anything X11-related in a .h file
60 // tends to trip up Qt. All the real logic is in QuickSyncEncoderImpl,
61 // defined in quicksync_encoder_impl.h.
63 // This class is _not_ thread-safe, except where mentioned.
64 class QuickSyncEncoder {
66 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);
69 void set_stream_mux(Mux *mux); // Does not take ownership. Must be called unless x264 is used for the stream.
70 void add_audio(int64_t pts, std::vector<float> audio); // Thread-safe.
71 bool is_zerocopy() const; // Thread-safe.
73 // See VideoEncoder::begin_frame().
74 bool begin_frame(int64_t pts, int64_t duration, movit::YCbCrLumaCoefficients ycbcr_coefficients, const std::vector<RefCountedFrame> &input_frames, GLuint *y_tex, GLuint *cbcr_tex);
75 RefCountedGLsync end_frame();
76 void shutdown(); // Blocking. Does not require an OpenGL context.
77 void close_file(); // Does not require an OpenGL context. Must be run after shutdown.
78 void release_gl_resources(); // Requires an OpenGL context. Must be run after shutdown.
79 int64_t global_delay() const; // So we never get negative dts.
81 // Tries to autodetect a device with a usable VA-API H.264 encoder.
82 // Tries first the default X11 display, then every /dev/dri/renderD* node in turn.
83 // Dies if none could be found.
84 static std::string get_usable_va_display();
87 std::unique_ptr<QuickSyncEncoderImpl> impl;