Michael Fabian 'Xaymar' Dirks
c5ff5d224d
Removes problems with memory leaks due to lost references and other issues.
66 lines
1.8 KiB
C++
66 lines
1.8 KiB
C++
// FFMPEG Video Encoder Integration for OBS Studio
|
|
// Copyright (C) 2018 - 2018 Michael Fabian Dirks
|
|
//
|
|
// This program is free software; you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation; either version 2 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program; if not, write to the Free Software
|
|
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
|
|
|
|
#pragma once
|
|
#include <deque>
|
|
#include <mutex>
|
|
|
|
extern "C" {
|
|
#pragma warning(push)
|
|
#pragma warning(disable : 4244)
|
|
#include <libavutil/frame.h>
|
|
#pragma warning(pop)
|
|
}
|
|
|
|
namespace ffmpeg {
|
|
class avframe_queue {
|
|
std::deque<std::shared_ptr<AVFrame>> frames;
|
|
std::mutex lock;
|
|
|
|
std::pair<uint32_t, uint32_t> resolution;
|
|
AVPixelFormat format = AV_PIX_FMT_NONE;
|
|
|
|
std::shared_ptr<AVFrame> create_frame();
|
|
|
|
public:
|
|
avframe_queue();
|
|
~avframe_queue();
|
|
|
|
void set_resolution(uint32_t width, uint32_t height);
|
|
void get_resolution(uint32_t& width, uint32_t& height);
|
|
uint32_t get_width();
|
|
uint32_t get_height();
|
|
|
|
void set_pixel_format(AVPixelFormat format);
|
|
AVPixelFormat get_pixel_format();
|
|
|
|
void precache(size_t count);
|
|
|
|
void clear();
|
|
|
|
void push(std::shared_ptr<AVFrame> frame);
|
|
|
|
std::shared_ptr<AVFrame> pop();
|
|
|
|
std::shared_ptr<AVFrame> pop_only();
|
|
|
|
bool empty();
|
|
|
|
size_t size();
|
|
};
|
|
} // namespace ffmpeg
|