Michael Fabian 'Xaymar' Dirks
71440ed3c5
Breaks more than intended, and this way is better anyway as the error that used to show up gave no indication as to what went wrong. With this reverted it now shows that starting encoding failed.
130 lines
3.8 KiB
C++
130 lines
3.8 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 <vector>
|
|
#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 unsupported_gpu_exception : public std::runtime_error {
|
|
public:
|
|
unsupported_gpu_exception(const std::string& reason) : runtime_error(reason) {}
|
|
};
|
|
|
|
struct encoder_info {
|
|
std::string uid;
|
|
std::string codec;
|
|
std::string readable_name;
|
|
obs_encoder_info oei;
|
|
};
|
|
|
|
class encoder_factory {
|
|
encoder_info info;
|
|
encoder_info info_fallback;
|
|
const AVCodec* avcodec_ptr;
|
|
|
|
public:
|
|
encoder_factory(const AVCodec* codec);
|
|
virtual ~encoder_factory();
|
|
|
|
void register_encoder();
|
|
|
|
void get_defaults(obs_data_t* settings);
|
|
|
|
void get_properties(obs_properties_t* props);
|
|
|
|
const AVCodec* get_avcodec();
|
|
|
|
const encoder_info& get_info();
|
|
|
|
const encoder_info& get_fallback();
|
|
};
|
|
|
|
class encoder {
|
|
obs_encoder_t* _self;
|
|
encoder_factory* _factory;
|
|
|
|
const AVCodec* _codec;
|
|
AVCodecContext* _context;
|
|
|
|
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, bool is_texture_encode = false);
|
|
virtual ~encoder();
|
|
|
|
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
|