xaymar/obs-ffmpeg-encoder
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 14 Wiki Activity

plugin: Implement support for initializers and finalizers

Browse Source
This commit is contained in:
Michael Fabian 'Xaymar' Dirks
2019-07-07 13:29:06 +02:00
parent 924c64cd37
commit 9ce8f39efc
2 changed files with 49 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/plugin.cpp
+42 -30
View File
@@ -31,44 +31,56 @@ extern "C" {
#pragma warning(pop)
}
// Initializers and finalizers.
std::list<std::function<void()>> obsffmpeg::initializers;
std::list<std::function<void()>> obsffmpeg::finalizers;
static std::map<AVCodec*, std::shared_ptr<encoder::generic_factory>> generic_factories;
MODULE_EXPORT bool obs_module_load(void)
{
try {
avcodec_register_all();
try {
// Initialize avcodec.
avcodec_register_all();
// Register all codecs.
AVCodec* cdc = nullptr;
while ((cdc = av_codec_next(cdc)) != nullptr) {
if (!av_codec_is_encoder(cdc))
continue;
if ((cdc->type == AVMediaType::AVMEDIA_TYPE_AUDIO)
|| (cdc->type == AVMediaType::AVMEDIA_TYPE_VIDEO)) {
auto ptr = std::make_shared<encoder::generic_factory>(cdc);
ptr->register_encoder();
generic_factories.emplace(cdc, ptr);
}
}
obsffmpeg::encoder::prores_aw::initialize();
return true;
} catch (std::exception ex) {
PLOG_ERROR("Exception during initalization: %s.", ex.what());
} catch (...) {
PLOG_ERROR("Unrecognized exception during initalization.");
// Run all initializers.
for (auto func : obsffmpeg::initializers) {
func();
}
// Register all codecs.
AVCodec* cdc = nullptr;
while ((cdc = av_codec_next(cdc)) != nullptr) {
if (!av_codec_is_encoder(cdc))
continue;
if ((cdc->type == AVMediaType::AVMEDIA_TYPE_AUDIO) || (cdc->type == AVMediaType::AVMEDIA_TYPE_VIDEO)) {
auto ptr = std::make_shared<encoder::generic_factory>(cdc);
ptr->register_encoder();
generic_factories.emplace(cdc, ptr);
}
}
obsffmpeg::encoder::prores_aw::initialize();
return true;
} catch (std::exception ex) {
PLOG_ERROR("Exception during initalization: %s.", ex.what());
return false;
} catch (...) {
PLOG_ERROR("Unrecognized exception during initalization.");
return false;
}
MODULE_EXPORT void obs_module_unload(void)
{
try {
obsffmpeg::encoder::prores_aw::finalize();
} catch (std::exception ex) {
PLOG_ERROR("Exception during finalizing: %s.", ex.what());
} catch (...) {
PLOG_ERROR("Unrecognized exception during finalizing.");
try {
obsffmpeg::encoder::prores_aw::finalize();
// Run all finalizers.
for (auto func : obsffmpeg::finalizers) {
func();
}
} catch (std::exception ex) {
PLOG_ERROR("Exception during finalizing: %s.", ex.what());
} catch (...) {
PLOG_ERROR("Unrecognized exception during finalizing.");
}
source/plugin.hpp
+7 -5
View File
@@ -1,5 +1,5 @@
// FFMPEG Video Encoder Integration for OBS Studio
// Copyright (C) 2018 - 2018 Michael Fabian Dirks
// Copyright (C) 2018 - 2019 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
@@ -15,16 +15,18 @@
// along with this program; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
#ifndef OBS_FFMPEG_PLUGIN_HPP
#define OBS_FFMPEG_PLUGIN_HPP
#pragma once
#include <functional>
#include <list>
#include <memory>
#include <obs-module.h>
namespace obsffmpeg {
extern std::list<std::function<void()>> initializers;
extern std::list<std::function<void()>> finalizers;
} // namespace obsffmpeg
#endif OBS_FFMPEG_PLUGIN_HPP
MODULE_EXPORT bool obs_module_load(void);
MODULE_EXPORT void obs_module_unload(void);