--- /dev/null
+// Hardware H.264 encoding via VAAPI. Also orchestrates the H.264 encoding
+// in general; this is unfortunate, and probably needs a cleanup. In particular,
+// even if you don't actually use Quick Sync for anything, this class
+// (or actually, QuickSyncEncoderImpl) still takes on a pretty central role.
+//
+// Heavily modified based on example code by Intel. Intel's original copyright
+// and license is reproduced below:
+//
+// Copyright (c) 2007-2013 Intel Corporation. All Rights Reserved.
+//
+// Permission is hereby granted, free of charge, to any person obtaining a
+// copy of this software and associated documentation files (the
+// "Software"), to deal in the Software without restriction, including
+// without limitation the rights to use, copy, modify, merge, publish,
+// distribute, sub license, and/or sell copies of the Software, and to
+// permit persons to whom the Software is furnished to do so, subject to
+// the following conditions:
+//
+// The above copyright notice and this permission notice (including the
+// next paragraph) shall be included in all copies or substantial portions
+// of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
+// IN NO EVENT SHALL PRECISION INSIGHT AND/OR ITS SUPPLIERS BE LIABLE FOR
+// ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+// TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+// SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+#ifndef _H264ENCODE_H
+#define _H264ENCODE_H
+
+#include <epoxy/gl.h>
+#include <movit/image_format.h>
+#include <stdbool.h>
+#include <stdint.h>
+#include <memory>
+#include <string>
+#include <vector>
+
+extern "C" {
+#include <libavformat/avformat.h>
+}
+
+#include "ref_counted_gl_sync.h"
+
+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 quicksync_encoder_impl.h.
+//
+// This class is _not_ thread-safe, except where mentioned.
+class QuickSyncEncoder {
+public:
+ 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); // Thread-safe.
+ bool is_zerocopy() const; // Thread-safe.
+
+ // See VideoEncoder::begin_frame().
+ 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);
+ RefCountedGLsync end_frame();
+ void shutdown(); // Blocking. Does not require an OpenGL context.
+ void close_file(); // Does not require an OpenGL context. Must be run after shutdown.
+ void release_gl_resources(); // Requires an OpenGL context. Must be run after shutdown.
+ int64_t global_delay() const; // So we never get negative dts.
+
+ // Tries to autodetect a device with a usable VA-API H.264 encoder.
+ // Tries first the default X11 display, then every /dev/dri/renderD* node in turn.
+ // Dies if none could be found.
+ static std::string get_usable_va_display();
+
+private:
+ std::unique_ptr<QuickSyncEncoderImpl> impl;
+};
+
+#endif
projects / nageru / blobdiff