1 // Hardware H.264 encoding via VAAPI. Heavily modified based on example
2 // code by Intel. Intel's original copyright and license is reproduced below:
4 // Copyright (c) 2007-2013 Intel Corporation. All Rights Reserved.
6 // Permission is hereby granted, free of charge, to any person obtaining a
7 // copy of this software and associated documentation files (the
8 // "Software"), to deal in the Software without restriction, including
9 // without limitation the rights to use, copy, modify, merge, publish,
10 // distribute, sub license, and/or sell copies of the Software, and to
11 // permit persons to whom the Software is furnished to do so, subject to
12 // the following conditions:
14 // The above copyright notice and this permission notice (including the
15 // next paragraph) shall be included in all copies or substantial portions
18 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
19 // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
21 // IN NO EVENT SHALL PRECISION INSIGHT AND/OR ITS SUPPLIERS BE LIABLE FOR
22 // ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23 // TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24 // SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
30 #include <libavformat/avformat.h>
34 #include <condition_variable>
44 #include "pbo_frame_allocator.h"
45 #include "ref_counted_gl_sync.h"
49 #define SURFACE_NUM 16 /* 16 surfaces for source YUV */
53 H264Encoder(QSurface *surface, int width, int height, const char *output_filename);
55 //void add_frame(FrameAllocator::Frame frame, GLsync fence);
61 GLuint y_tex, cbcr_tex;
68 bool begin_frame(GLuint *y_tex, GLuint *cbcr_tex);
69 void end_frame(RefCountedGLsync fence, const std::vector<FrameAllocator::Frame> &input_frames_to_release);
73 unsigned long long display_order;
74 unsigned long long encode_order;
78 void copy_thread_func();
79 void storage_task_thread();
80 void storage_task_enqueue(unsigned long long display_order, unsigned long long encode_order, int frame_type);
81 int save_codeddata(unsigned long long display_order, unsigned long long encode_order, int frame_type);
83 std::thread copy_thread, storage_thread;
85 std::mutex storage_task_queue_mutex;
86 std::condition_variable storage_task_queue_changed;
87 int srcsurface_status[SURFACE_NUM]; // protected by storage_task_queue_mutex
88 std::queue<storage_task> storage_task_queue; // protected by storage_task_queue_mutex
89 bool storage_thread_should_quit = false; // protected by storage_task_queue_mutex
91 std::mutex frame_queue_mutex;
92 std::condition_variable frame_queue_nonempty;
93 bool copy_thread_should_quit = false; // under frame_queue_mutex
95 //int frame_width, frame_height;
97 int current_storage_frame;
100 RefCountedGLsync fence;
101 std::vector<FrameAllocator::Frame> input_frames_to_release;
103 std::map<int, PendingFrame> pending_frames;
106 AVFormatContext *avctx;