f53df7da64
Code submissions have continually suffered from formatting inconsistencies that constantly have to be addressed. Using clang-format simplifies this by making code formatting more consistent, and allows automation of the code formatting so that maintainers can focus more on the code itself instead of code formatting.
307 lines
7.3 KiB
C++
307 lines
7.3 KiB
C++
#include <obs-module.h>
|
|
#include <obs-frontend-api.h>
|
|
#include <QMainWindow>
|
|
#include <QAction>
|
|
#include <util/util.hpp>
|
|
#include <util/platform.h>
|
|
#include <media-io/video-io.h>
|
|
#include <media-io/video-frame.h>
|
|
#include "DecklinkOutputUI.h"
|
|
#include "../../../plugins/decklink/const.h"
|
|
|
|
OBS_DECLARE_MODULE()
|
|
OBS_MODULE_USE_DEFAULT_LOCALE("decklink-output-ui", "en-US")
|
|
|
|
DecklinkOutputUI *doUI;
|
|
|
|
bool main_output_running = false;
|
|
bool preview_output_running = false;
|
|
|
|
obs_output_t *output;
|
|
|
|
struct preview_output {
|
|
bool enabled;
|
|
obs_source_t *current_source;
|
|
obs_output_t *output;
|
|
|
|
video_t *video_queue;
|
|
gs_texrender_t *texrender;
|
|
gs_stagesurf_t *stagesurface;
|
|
uint8_t *video_data;
|
|
uint32_t video_linesize;
|
|
|
|
obs_video_info ovi;
|
|
};
|
|
|
|
static struct preview_output context = {0};
|
|
|
|
OBSData load_settings()
|
|
{
|
|
BPtr<char> path = obs_module_get_config_path(
|
|
obs_current_module(), "decklinkOutputProps.json");
|
|
BPtr<char> jsonData = os_quick_read_utf8_file(path);
|
|
if (!!jsonData) {
|
|
obs_data_t *data = obs_data_create_from_json(jsonData);
|
|
OBSData dataRet(data);
|
|
obs_data_release(data);
|
|
|
|
return dataRet;
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
void output_start()
|
|
{
|
|
if (!main_output_running) {
|
|
OBSData settings = load_settings();
|
|
|
|
if (settings != nullptr) {
|
|
output = obs_output_create("decklink_output",
|
|
"decklink_output", settings,
|
|
NULL);
|
|
|
|
obs_output_start(output);
|
|
obs_data_release(settings);
|
|
|
|
main_output_running = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
void output_stop()
|
|
{
|
|
if (main_output_running) {
|
|
obs_output_stop(output);
|
|
obs_output_release(output);
|
|
main_output_running = false;
|
|
}
|
|
}
|
|
|
|
OBSData load_preview_settings()
|
|
{
|
|
BPtr<char> path = obs_module_get_config_path(
|
|
obs_current_module(), "decklinkPreviewOutputProps.json");
|
|
BPtr<char> jsonData = os_quick_read_utf8_file(path);
|
|
if (!!jsonData) {
|
|
obs_data_t *data = obs_data_create_from_json(jsonData);
|
|
OBSData dataRet(data);
|
|
obs_data_release(data);
|
|
|
|
return dataRet;
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
void on_preview_scene_changed(enum obs_frontend_event event, void *param);
|
|
void render_preview_source(void *param, uint32_t cx, uint32_t cy);
|
|
|
|
void preview_output_start()
|
|
{
|
|
if (!preview_output_running) {
|
|
OBSData settings = load_preview_settings();
|
|
|
|
if (settings != nullptr) {
|
|
context.output = obs_output_create(
|
|
"decklink_output", "decklink_preview_output",
|
|
settings, NULL);
|
|
|
|
obs_get_video_info(&context.ovi);
|
|
|
|
uint32_t width = context.ovi.base_width;
|
|
uint32_t height = context.ovi.base_height;
|
|
|
|
obs_enter_graphics();
|
|
context.texrender =
|
|
gs_texrender_create(GS_BGRA, GS_ZS_NONE);
|
|
context.stagesurface =
|
|
gs_stagesurface_create(width, height, GS_BGRA);
|
|
obs_leave_graphics();
|
|
|
|
const video_output_info *mainVOI =
|
|
video_output_get_info(obs_get_video());
|
|
|
|
video_output_info vi = {0};
|
|
vi.format = VIDEO_FORMAT_BGRA;
|
|
vi.width = width;
|
|
vi.height = height;
|
|
vi.fps_den = context.ovi.fps_den;
|
|
vi.fps_num = context.ovi.fps_num;
|
|
vi.cache_size = 16;
|
|
vi.colorspace = mainVOI->colorspace;
|
|
vi.range = mainVOI->range;
|
|
vi.name = "decklink_preview_output";
|
|
|
|
video_output_open(&context.video_queue, &vi);
|
|
|
|
obs_frontend_add_event_callback(
|
|
on_preview_scene_changed, &context);
|
|
if (obs_frontend_preview_program_mode_active()) {
|
|
context.current_source =
|
|
obs_frontend_get_current_preview_scene();
|
|
} else {
|
|
context.current_source =
|
|
obs_frontend_get_current_scene();
|
|
}
|
|
obs_add_main_render_callback(render_preview_source,
|
|
&context);
|
|
|
|
obs_output_set_media(context.output,
|
|
context.video_queue,
|
|
obs_get_audio());
|
|
obs_output_start(context.output);
|
|
|
|
preview_output_running = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
void preview_output_stop()
|
|
{
|
|
if (preview_output_running) {
|
|
obs_output_stop(context.output);
|
|
video_output_stop(context.video_queue);
|
|
|
|
obs_remove_main_render_callback(render_preview_source,
|
|
&context);
|
|
obs_frontend_remove_event_callback(on_preview_scene_changed,
|
|
&context);
|
|
|
|
obs_source_release(context.current_source);
|
|
|
|
obs_enter_graphics();
|
|
gs_stagesurface_destroy(context.stagesurface);
|
|
gs_texrender_destroy(context.texrender);
|
|
obs_leave_graphics();
|
|
|
|
video_output_close(context.video_queue);
|
|
|
|
preview_output_running = false;
|
|
}
|
|
}
|
|
|
|
void on_preview_scene_changed(enum obs_frontend_event event, void *param)
|
|
{
|
|
auto ctx = (struct preview_output *)param;
|
|
switch (event) {
|
|
case OBS_FRONTEND_EVENT_STUDIO_MODE_ENABLED:
|
|
case OBS_FRONTEND_EVENT_PREVIEW_SCENE_CHANGED:
|
|
obs_source_release(ctx->current_source);
|
|
ctx->current_source = obs_frontend_get_current_preview_scene();
|
|
break;
|
|
case OBS_FRONTEND_EVENT_STUDIO_MODE_DISABLED:
|
|
obs_source_release(ctx->current_source);
|
|
ctx->current_source = obs_frontend_get_current_scene();
|
|
break;
|
|
case OBS_FRONTEND_EVENT_SCENE_CHANGED:
|
|
if (!obs_frontend_preview_program_mode_active()) {
|
|
obs_source_release(ctx->current_source);
|
|
ctx->current_source = obs_frontend_get_current_scene();
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
void render_preview_source(void *param, uint32_t cx, uint32_t cy)
|
|
{
|
|
auto ctx = (struct preview_output *)param;
|
|
|
|
if (!ctx->current_source)
|
|
return;
|
|
|
|
uint32_t width = obs_source_get_base_width(ctx->current_source);
|
|
uint32_t height = obs_source_get_base_height(ctx->current_source);
|
|
|
|
gs_texrender_reset(ctx->texrender);
|
|
|
|
if (gs_texrender_begin(ctx->texrender, width, height)) {
|
|
struct vec4 background;
|
|
vec4_zero(&background);
|
|
|
|
gs_clear(GS_CLEAR_COLOR, &background, 0.0f, 0);
|
|
gs_ortho(0.0f, (float)width, 0.0f, (float)height, -100.0f,
|
|
100.0f);
|
|
|
|
gs_blend_state_push();
|
|
gs_blend_function(GS_BLEND_ONE, GS_BLEND_ZERO);
|
|
|
|
obs_source_video_render(ctx->current_source);
|
|
|
|
gs_blend_state_pop();
|
|
gs_texrender_end(ctx->texrender);
|
|
|
|
struct video_frame output_frame;
|
|
if (video_output_lock_frame(ctx->video_queue, &output_frame, 1,
|
|
os_gettime_ns())) {
|
|
gs_stage_texture(
|
|
ctx->stagesurface,
|
|
gs_texrender_get_texture(ctx->texrender));
|
|
|
|
if (gs_stagesurface_map(ctx->stagesurface,
|
|
&ctx->video_data,
|
|
&ctx->video_linesize)) {
|
|
uint32_t linesize = output_frame.linesize[0];
|
|
for (uint32_t i = 0; i < ctx->ovi.base_height;
|
|
i++) {
|
|
uint32_t dst_offset = linesize * i;
|
|
uint32_t src_offset =
|
|
ctx->video_linesize * i;
|
|
memcpy(output_frame.data[0] +
|
|
dst_offset,
|
|
ctx->video_data + src_offset,
|
|
linesize);
|
|
}
|
|
|
|
gs_stagesurface_unmap(ctx->stagesurface);
|
|
ctx->video_data = nullptr;
|
|
}
|
|
|
|
video_output_unlock_frame(ctx->video_queue);
|
|
}
|
|
}
|
|
}
|
|
|
|
void addOutputUI(void)
|
|
{
|
|
QAction *action = (QAction *)obs_frontend_add_tools_menu_qaction(
|
|
obs_module_text("Decklink Output"));
|
|
|
|
QMainWindow *window = (QMainWindow *)obs_frontend_get_main_window();
|
|
|
|
obs_frontend_push_ui_translation(obs_module_get_string);
|
|
doUI = new DecklinkOutputUI(window);
|
|
obs_frontend_pop_ui_translation();
|
|
|
|
auto cb = []() { doUI->ShowHideDialog(); };
|
|
|
|
action->connect(action, &QAction::triggered, cb);
|
|
}
|
|
|
|
static void OBSEvent(enum obs_frontend_event event, void *)
|
|
{
|
|
if (event == OBS_FRONTEND_EVENT_FINISHED_LOADING) {
|
|
OBSData settings = load_settings();
|
|
|
|
if (settings && obs_data_get_bool(settings, "auto_start"))
|
|
output_start();
|
|
|
|
OBSData previewSettings = load_preview_settings();
|
|
|
|
if (previewSettings &&
|
|
obs_data_get_bool(previewSettings, "auto_start"))
|
|
preview_output_start();
|
|
}
|
|
}
|
|
|
|
bool obs_module_load(void)
|
|
{
|
|
addOutputUI();
|
|
|
|
obs_frontend_add_event_callback(OBSEvent, nullptr);
|
|
|
|
return true;
|
|
}
|