]> git.sesse.net Git - nageru/blob - nageru/quicksync_encoder.h
Fix some leftovers in warning messages.
[nageru] / nageru / quicksync_encoder.h
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.
5 //
6 // Heavily modified based on example code by Intel. Intel's original copyright
7 // and license is reproduced below:
8 //
9 // Copyright (c) 2007-2013 Intel Corporation. All Rights Reserved.
10 //
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:
18 // 
19 // The above copyright notice and this permission notice (including the
20 // next paragraph) shall be included in all copies or substantial portions
21 // of the Software.
22 // 
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.
30
31 #ifndef _H264ENCODE_H
32 #define _H264ENCODE_H
33
34 #include <epoxy/gl.h>
35 #include <movit/image_format.h>
36 #include <stdbool.h>
37 #include <stdint.h>
38 #include <memory>
39 #include <string>
40 #include <vector>
41
42 extern "C" {
43 #include <libavformat/avformat.h>
44 }
45
46 #include "shared/ref_counted_gl_sync.h"
47
48 class DiskSpaceEstimator;
49 class Mux;
50 class QSurface;
51 class QuickSyncEncoderImpl;
52 class RefCountedFrame;
53 class X264Encoder;
54
55 namespace movit {
56 class ResourcePool;
57 }  // namespace movit
58
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.
62 //
63 // This class is _not_ thread-safe, except where mentioned.
64 class QuickSyncEncoder {
65 public:
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);
67         ~QuickSyncEncoder();
68
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.
72
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.
80
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();
85
86 private:
87         std::unique_ptr<QuickSyncEncoderImpl> impl;
88 };
89
90 #endif