Michael Fabian 'Xaymar' Dirks
5c8939b4a8
Fixes a crash due to reading/writing out of bounds when the resolution, color and framerate end up different than initially expected during the creation of the encoder. While this does change the error message that appears when the encoder can't be initialized, it is better than outright crashing OBS Studio.
125 lines
3.6 KiB
C++
125 lines
3.6 KiB
C++
// FFMPEG Video Encoder Integration for OBS Studio
|
|
// Copyright (c) 2019 Michael Fabian Dirks <info@xaymar.com>
|
|
//
|
|
// 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, sublicense, 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 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 NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS 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.
|
|
|
|
#pragma once
|
|
|
|
#include <condition_variable>
|
|
#include <mutex>
|
|
#include <thread>
|
|
#include "ffmpeg/avframe-queue.hpp"
|
|
#include "ffmpeg/swscale.hpp"
|
|
|
|
extern "C" {
|
|
#include <obs-properties.h>
|
|
#include <obs.h>
|
|
#pragma warning(push)
|
|
#pragma warning(disable : 4244)
|
|
#include <libavcodec/avcodec.h>
|
|
#include <libavutil/frame.h>
|
|
#pragma warning(pop)
|
|
}
|
|
|
|
namespace obsffmpeg {
|
|
class encoder_factory {
|
|
struct info {
|
|
std::string uid;
|
|
std::string codec;
|
|
std::string readable_name;
|
|
obs_encoder_info oei;
|
|
} info;
|
|
const AVCodec* avcodec_ptr;
|
|
|
|
public:
|
|
encoder_factory(const AVCodec* codec);
|
|
virtual ~encoder_factory();
|
|
|
|
void register_encoder();
|
|
|
|
const char* get_name();
|
|
|
|
void get_defaults(obs_data_t* settings);
|
|
|
|
void get_properties(obs_properties_t* props);
|
|
|
|
const AVCodec* get_avcodec();
|
|
};
|
|
|
|
class encoder {
|
|
obs_encoder_t* _self;
|
|
encoder_factory* _factory;
|
|
|
|
const AVCodec* _codec;
|
|
AVCodecContext* _context;
|
|
|
|
bool _initialized;
|
|
std::pair<size_t, size_t> _resolution;
|
|
AVPixelFormat _format;
|
|
|
|
ffmpeg::avframe_queue _frame_queue;
|
|
ffmpeg::avframe_queue _frame_queue_used;
|
|
ffmpeg::swscale _swscale;
|
|
AVPacket _current_packet;
|
|
|
|
int64_t _lag_in_frames;
|
|
int64_t _count_send_frames;
|
|
|
|
// Extra Data
|
|
bool _have_first_frame;
|
|
std::vector<uint8_t> _extra_data;
|
|
std::vector<uint8_t> _sei_data;
|
|
|
|
public:
|
|
encoder(obs_data_t* settings, obs_encoder_t* encoder);
|
|
virtual ~encoder();
|
|
|
|
bool initialize();
|
|
|
|
public: // OBS API
|
|
// Shared
|
|
void get_properties(obs_properties_t* props);
|
|
|
|
bool update(obs_data_t* settings);
|
|
|
|
// Audio only
|
|
void get_audio_info(struct audio_convert_info* info);
|
|
|
|
size_t get_frame_size();
|
|
|
|
bool audio_encode(struct encoder_frame* frame, struct encoder_packet* packet, bool* received_packet);
|
|
|
|
// Video only
|
|
void get_video_info(struct video_scale_info* info);
|
|
|
|
bool get_sei_data(uint8_t** sei_data, size_t* size);
|
|
|
|
bool get_extra_data(uint8_t** extra_data, size_t* size);
|
|
|
|
bool video_encode(struct encoder_frame* frame, struct encoder_packet* packet, bool* received_packet);
|
|
|
|
bool video_encode_texture(uint32_t handle, int64_t pts, uint64_t lock_key, uint64_t* next_key,
|
|
struct encoder_packet* packet, bool* received_packet);
|
|
|
|
int receive_packet(bool* received_packet, struct encoder_packet* packet);
|
|
|
|
int send_frame(std::shared_ptr<AVFrame> frame);
|
|
};
|
|
} // namespace obsffmpeg
|