SceneSwitcher/advanced-scene-switcher.cpp
2017-10-29 00:28:47 +02:00

1031 lines
34 KiB
C++

#include <obs-frontend-api.h>
#include <obs-module.h>
#include <obs.hpp>
#include <util/util.hpp>
#include <QMainWindow>
#include <QMessageBox>
#include <QAction>
#include <QDir>
#include <QFileDialog>
#include <QTextStream>
#include <QTimer>
#include <condition_variable>
#include <chrono>
#include <string>
#include <vector>
#include <thread>
#include <regex>
#include <mutex>
#include <fstream>
#include "switcher-data-structs.hpp"
#include "utility.hpp"
#include "advanced-scene-switcher.hpp"
using namespace std;
SwitcherData* switcher = nullptr;
SceneSwitcher::SceneSwitcher(QWidget* parent)
: QDialog(parent)
, ui(new Ui_SceneSwitcher)
{
ui->setupUi(this);
lock_guard<mutex> lock(switcher->m);
switcher->Prune();
BPtr<char*> scenes = obs_frontend_get_scene_names();
char** temp = scenes;
while (*temp)
{
const char* name = *temp;
ui->scenes->addItem(name);
ui->noMatchSwitchScene->addItem(name);
ui->screenRegionScenes->addItem(name);
ui->pauseScenesScenes->addItem(name);
ui->sceneRoundTripScenes1->addItem(name);
ui->sceneRoundTripScenes2->addItem(name);
ui->autoStopScenes->addItem(name);
ui->transitionsScene1->addItem(name);
ui->transitionsScene2->addItem(name);
ui->defaultTransitionsScene->addItem(name);
ui->executableScenes->addItem(name);
ui->idleScenes->addItem(name);
temp++;
}
obs_frontend_source_list* transitions = new obs_frontend_source_list();
obs_frontend_get_transitions(transitions);
for (size_t i = 0; i < transitions->sources.num; i++)
{
const char* name = obs_source_get_name(transitions->sources.array[i]);
ui->transitions->addItem(name);
ui->screenRegionsTransitions->addItem(name);
ui->sceneRoundTripTransitions->addItem(name);
ui->transitionsTransitions->addItem(name);
ui->defaultTransitionsTransitions->addItem(name);
ui->executableTransitions->addItem(name);
ui->idleTransitions->addItem(name);
}
obs_frontend_source_list_free(transitions);
if (switcher->switchIfNotMatching)
ui->noMatchSwitch->setChecked(true);
else
ui->noMatchDontSwitch->setChecked(true);
ui->noMatchSwitchScene->setCurrentText(GetWeakSourceName(switcher->nonMatchingScene).c_str());
ui->checkInterval->setValue(switcher->interval);
vector<string> windows;
GetWindowList(windows);
for (string& window : windows)
{
ui->windows->addItem(window.c_str());
ui->ignoreWindowsWindows->addItem(window.c_str());
ui->pauseWindowsWindows->addItem(window.c_str());
ui->ignoreIdleWindowsWindows->addItem(window.c_str());
}
QStringList processes;
GetProcessList(processes);
for (QString& process : processes)
ui->executable->addItem(process);
for (auto& s : switcher->executableSwitches)
{
string sceneName = GetWeakSourceName(s.mScene);
string transitionName = GetWeakSourceName(s.mTransition);
QString text = MakeSwitchNameExecutable(
sceneName.c_str(), s.mExe, transitionName.c_str(), s.mInFocus);
QListWidgetItem* item = new QListWidgetItem(text, ui->executables);
item->setData(Qt::UserRole, s.mExe);
}
for (auto& s : switcher->switches)
{
string sceneName = GetWeakSourceName(s.scene);
string transitionName = GetWeakSourceName(s.transition);
QString text = MakeSwitchName(
sceneName.c_str(), s.window.c_str(), transitionName.c_str(), s.fullscreen);
QListWidgetItem* item = new QListWidgetItem(text, ui->switches);
item->setData(Qt::UserRole, s.window.c_str());
}
for (auto& s : switcher->screenRegionSwitches)
{
string sceneName = GetWeakSourceName(s.scene);
string transitionName = GetWeakSourceName(s.transition);
QString text = MakeScreenRegionSwitchName(
sceneName.c_str(), transitionName.c_str(), s.minX, s.minY, s.maxX, s.maxY);
QListWidgetItem* item = new QListWidgetItem(text, ui->screenRegions);
item->setData(Qt::UserRole, s.regionStr.c_str());
}
ui->autoStopSceneCheckBox->setChecked(switcher->autoStopEnable);
ui->autoStopScenes->setCurrentText(GetWeakSourceName(switcher->autoStopScene).c_str());
if (ui->autoStopSceneCheckBox->checkState())
{
ui->autoStopScenes->setDisabled(false);
}
else
{
ui->autoStopScenes->setDisabled(true);
}
for (auto& scene : switcher->pauseScenesSwitches)
{
string sceneName = GetWeakSourceName(scene);
QString text = QString::fromStdString(sceneName);
QListWidgetItem* item = new QListWidgetItem(text, ui->pauseScenes);
item->setData(Qt::UserRole, text);
}
for (auto& window : switcher->pauseWindowsSwitches)
{
QString text = QString::fromStdString(window);
QListWidgetItem* item = new QListWidgetItem(text, ui->pauseWindows);
item->setData(Qt::UserRole, text);
}
for (auto& window : switcher->ignoreWindowsSwitches)
{
QString text = QString::fromStdString(window);
QListWidgetItem* item = new QListWidgetItem(text, ui->ignoreWindows);
item->setData(Qt::UserRole, text);
}
int smallestDelay = switcher->interval;
for (auto& s : switcher->sceneRoundTripSwitches)
{
string sceneName1 = GetWeakSourceName(s.scene1);
string sceneName2 = GetWeakSourceName(s.scene2);
string transitionName = GetWeakSourceName(s.transition);
QString text = MakeSceneRoundTripSwitchName(
sceneName1.c_str(), sceneName2.c_str(), transitionName.c_str(), s.delay);
QListWidgetItem* item = new QListWidgetItem(text, ui->sceneRoundTrips);
item->setData(Qt::UserRole, text);
if (s.delay * 1000 < smallestDelay)
smallestDelay = s.delay * 1000;
}
(smallestDelay < switcher->interval) ? ui->intervalWarning : ui->intervalWarning->setVisible(false);
for (auto& s : switcher->sceneTransitions)
{
string sceneName1 = GetWeakSourceName(s.scene1);
string sceneName2 = GetWeakSourceName(s.scene2);
string transitionName = GetWeakSourceName(s.transition);
QString text = MakeSceneTransitionName(
sceneName1.c_str(), sceneName2.c_str(), transitionName.c_str());
QListWidgetItem* item = new QListWidgetItem(text, ui->sceneTransitions);
item->setData(Qt::UserRole, text);
}
//(transitionDurationLongerThanInterval(switcher->interval)) ? ui->transitionWarning->setVisible(true) : ui->transitionWarning->setVisible(false);
for (auto& s : switcher->defaultSceneTransitions)
{
string sceneName = GetWeakSourceName(s.scene);
string transitionName = GetWeakSourceName(s.transition);
QString text = MakeDefaultSceneTransitionName(
sceneName.c_str(), transitionName.c_str());
QListWidgetItem* item = new QListWidgetItem(text, ui->defaultTransitions);
item->setData(Qt::UserRole, text);
}
for (auto& window : switcher->ignoreIdleWindows)
{
QString text = QString::fromStdString(window);
QListWidgetItem* item = new QListWidgetItem(text, ui->ignoreIdleWindows);
item->setData(Qt::UserRole, text);
}
ui->idleCheckBox->setChecked(switcher->idleData.idleEnable);
ui->idleScenes->setCurrentText(GetWeakSourceName(switcher->idleData.scene).c_str());
ui->idleTransitions->setCurrentText(GetWeakSourceName(switcher->idleData.transition).c_str());
ui->idleSpinBox->setValue(switcher->idleData.time);
if (ui->idleCheckBox->checkState())
{
ui->idleScenes->setDisabled(false);
ui->idleSpinBox->setDisabled(false);
ui->idleTransitions->setDisabled(false);
}
else
{
ui->idleScenes->setDisabled(true);
ui->idleSpinBox->setDisabled(true);
ui->idleTransitions->setDisabled(true);
}
ui->readPathLineEdit->setText(QString::fromStdString(switcher->fileIO.readPath.c_str()));
ui->readFileCheckBox->setChecked(switcher->fileIO.readEnabled);
ui->writePathLineEdit->setText(QString::fromStdString(switcher->fileIO.writePath.c_str()));
if (ui->readFileCheckBox->checkState())
{
ui->browseButton_2->setDisabled(false);
ui->readPathLineEdit->setDisabled(false);
}
else
{
ui->browseButton_2->setDisabled(true);
ui->readPathLineEdit->setDisabled(true);
}
if (switcher->th.joinable())
SetStarted();
else
SetStopped();
loading = false;
// screen region cursor position
QTimer* screenRegionTimer = new QTimer(this);
connect(screenRegionTimer, SIGNAL(timeout()), this, SLOT(updateScreenRegionCursorPos()));
screenRegionTimer->start(1000);
for (string s : switcher->functionNamesByPriority)
{
QString text(s.c_str());
QListWidgetItem* item = new QListWidgetItem(text, ui->priorityList);
item->setData(Qt::UserRole, text);
}
}
static void SaveSceneSwitcher(obs_data_t* save_data, bool saving, void*)
{
if (saving)
{
lock_guard<mutex> lock(switcher->m);
obs_data_t* obj = obs_data_create();
obs_data_array_t* array = obs_data_array_create();
obs_data_array_t* screenRegionArray = obs_data_array_create();
obs_data_array_t* pauseScenesArray = obs_data_array_create();
obs_data_array_t* pauseWindowsArray = obs_data_array_create();
obs_data_array_t* ignoreWindowsArray = obs_data_array_create();
obs_data_array_t* sceneRoundTripArray = obs_data_array_create();
obs_data_array_t* sceneTransitionsArray = obs_data_array_create();
obs_data_array_t* defaultTransitionsArray = obs_data_array_create();
obs_data_array_t* ignoreIdleWindowsArray = obs_data_array_create();
switcher->Prune();
for (SceneSwitch& s : switcher->switches)
{
obs_data_t* array_obj = obs_data_create();
obs_source_t* source = obs_weak_source_get_source(s.scene);
obs_source_t* transition = obs_weak_source_get_source(s.transition);
if (source && transition)
{
const char* sceneName = obs_source_get_name(source);
const char* transitionName = obs_source_get_name(transition);
obs_data_set_string(array_obj, "scene", sceneName);
obs_data_set_string(array_obj, "transition", transitionName);
obs_data_set_string(array_obj, "window_title", s.window.c_str());
obs_data_set_bool(array_obj, "fullscreen", s.fullscreen);
obs_data_array_push_back(array, array_obj);
obs_source_release(source);
obs_source_release(transition);
}
obs_data_release(array_obj);
}
for (ScreenRegionSwitch& s : switcher->screenRegionSwitches)
{
obs_data_t* array_obj = obs_data_create();
obs_source_t* source = obs_weak_source_get_source(s.scene);
obs_source_t* transition = obs_weak_source_get_source(s.transition);
if (source && transition)
{
const char* sceneName = obs_source_get_name(source);
const char* transitionName = obs_source_get_name(transition);
obs_data_set_string(array_obj, "screenRegionScene", sceneName);
obs_data_set_string(array_obj, "transition", transitionName);
obs_data_set_int(array_obj, "minX", s.minX);
obs_data_set_int(array_obj, "minY", s.minY);
obs_data_set_int(array_obj, "maxX", s.maxX);
obs_data_set_int(array_obj, "maxY", s.maxY);
obs_data_set_string(array_obj, "screenRegionStr", s.regionStr.c_str());
obs_data_array_push_back(screenRegionArray, array_obj);
obs_source_release(source);
obs_source_release(transition);
}
obs_data_release(array_obj);
}
for (OBSWeakSource& scene : switcher->pauseScenesSwitches)
{
obs_data_t* array_obj = obs_data_create();
obs_source_t* source = obs_weak_source_get_source(scene);
if (source)
{
const char* n = obs_source_get_name(source);
obs_data_set_string(array_obj, "pauseScene", n);
obs_data_array_push_back(pauseScenesArray, array_obj);
obs_source_release(source);
}
obs_data_release(array_obj);
}
for (string& window : switcher->pauseWindowsSwitches)
{
obs_data_t* array_obj = obs_data_create();
obs_data_set_string(array_obj, "pauseWindow", window.c_str());
obs_data_array_push_back(pauseWindowsArray, array_obj);
obs_data_release(array_obj);
}
for (string& window : switcher->ignoreWindowsSwitches)
{
obs_data_t* array_obj = obs_data_create();
obs_data_set_string(array_obj, "ignoreWindow", window.c_str());
obs_data_array_push_back(ignoreWindowsArray, array_obj);
obs_data_release(array_obj);
}
for (SceneRoundTripSwitch& s : switcher->sceneRoundTripSwitches)
{
obs_data_t* array_obj = obs_data_create();
obs_source_t* source1 = obs_weak_source_get_source(s.scene1);
obs_source_t* source2 = obs_weak_source_get_source(s.scene2);
obs_source_t* transition = obs_weak_source_get_source(s.transition);
if (source1 && source2 && transition)
{
const char* sceneName1 = obs_source_get_name(source1);
const char* sceneName2 = obs_source_get_name(source2);
const char* transitionName = obs_source_get_name(transition);
obs_data_set_string(array_obj, "sceneRoundTripScene1", sceneName1);
obs_data_set_string(array_obj, "sceneRoundTripScene2", sceneName2);
obs_data_set_string(array_obj, "transition", transitionName);
obs_data_set_int(array_obj, "sceneRoundTripDelay", s.delay);
obs_data_set_string(array_obj, "sceneRoundTripStr", s.sceneRoundTripStr.c_str());
obs_data_array_push_back(sceneRoundTripArray, array_obj);
obs_source_release(source1);
obs_source_release(source2);
obs_source_release(transition);
}
obs_data_release(array_obj);
}
for (SceneTransition& s : switcher->sceneTransitions)
{
obs_data_t* array_obj = obs_data_create();
obs_source_t* source1 = obs_weak_source_get_source(s.scene1);
obs_source_t* source2 = obs_weak_source_get_source(s.scene2);
obs_source_t* transition = obs_weak_source_get_source(s.transition);
if (source1 && source2 && transition)
{
const char* sceneName1 = obs_source_get_name(source1);
const char* sceneName2 = obs_source_get_name(source2);
const char* transitionName = obs_source_get_name(transition);
obs_data_set_string(array_obj, "Scene1", sceneName1);
obs_data_set_string(array_obj, "Scene2", sceneName2);
obs_data_set_string(array_obj, "transition", transitionName);
obs_data_set_string(array_obj, "Str", s.sceneTransitionStr.c_str());
obs_data_array_push_back(sceneTransitionsArray, array_obj);
obs_source_release(source1);
obs_source_release(source2);
obs_source_release(transition);
}
obs_data_release(array_obj);
}
for (DefaultSceneTransition& s : switcher->defaultSceneTransitions)
{
obs_data_t* array_obj = obs_data_create();
obs_source_t* source = obs_weak_source_get_source(s.scene);
obs_source_t* transition = obs_weak_source_get_source(s.transition);
if (source && transition)
{
const char* sceneName = obs_source_get_name(source);
const char* transitionName = obs_source_get_name(transition);
obs_data_set_string(array_obj, "Scene", sceneName);
obs_data_set_string(array_obj, "transition", transitionName);
obs_data_set_string(array_obj, "Str", s.sceneTransitionStr.c_str());
obs_data_array_push_back(defaultTransitionsArray, array_obj);
obs_source_release(source);
obs_source_release(transition);
}
obs_data_release(array_obj);
}
obs_data_array_t* executableArray = obs_data_array_create();
for (ExecutableSceneSwitch& s : switcher->executableSwitches)
{
obs_data_t* array_obj = obs_data_create();
obs_source_t* source = obs_weak_source_get_source(s.mScene);
obs_source_t* transition = obs_weak_source_get_source(s.mTransition);
if (source && transition)
{
const char* sceneName = obs_source_get_name(source);
const char* transitionName = obs_source_get_name(transition);
obs_data_set_string(array_obj, "scene", sceneName);
obs_data_set_string(array_obj, "transition", transitionName);
obs_data_set_string(array_obj, "exefile", s.mExe.toUtf8());
obs_data_set_bool(array_obj, "infocus", s.mInFocus);
obs_data_array_push_back(executableArray, array_obj);
obs_source_release(source);
obs_source_release(transition);
}
obs_data_release(array_obj);
}
for (string& window : switcher->ignoreIdleWindows)
{
obs_data_t* array_obj = obs_data_create();
obs_data_set_string(array_obj, "window", window.c_str());
obs_data_array_push_back(ignoreIdleWindowsArray, array_obj);
obs_data_release(array_obj);
}
string nonMatchingSceneName = GetWeakSourceName(switcher->nonMatchingScene);
obs_data_set_int(obj, "interval", switcher->interval);
obs_data_set_string(obj, "non_matching_scene", nonMatchingSceneName.c_str());
obs_data_set_bool(obj, "switch_if_not_matching", switcher->switchIfNotMatching);
obs_data_set_bool(obj, "active", switcher->th.joinable());
obs_data_set_array(obj, "switches", array);
obs_data_set_array(obj, "screenRegion", screenRegionArray);
obs_data_set_array(obj, "pauseScenes", pauseScenesArray);
obs_data_set_array(obj, "pauseWindows", pauseWindowsArray);
obs_data_set_array(obj, "ignoreWindows", ignoreWindowsArray);
obs_data_set_array(obj, "sceneRoundTrip", sceneRoundTripArray);
obs_data_set_array(obj, "sceneTransitions", sceneTransitionsArray);
obs_data_set_array(obj, "defaultTransitions", defaultTransitionsArray);
obs_data_set_array(obj, "executableSwitches", executableArray);
obs_data_set_array(obj, "ignoreIdleWindows", ignoreIdleWindowsArray);
string autoStopSceneName = GetWeakSourceName(switcher->autoStopScene);
obs_data_set_bool(obj, "autoStopEnable", switcher->autoStopEnable);
obs_data_set_string(obj, "autoStopSceneName", autoStopSceneName.c_str());
string idleSceneName = GetWeakSourceName(switcher->idleData.scene);
string idleTransitionName = GetWeakSourceName(switcher->idleData.transition);
obs_data_set_bool(obj, "idleEnable", switcher->idleData.idleEnable);
obs_data_set_string(obj, "idleSceneName", idleSceneName.c_str());
obs_data_set_string(obj, "idleTransitionName", idleTransitionName.c_str());
obs_data_set_int(obj, "idleTime", switcher->idleData.time);
obs_data_set_bool(obj, "readEnabled", switcher->fileIO.readEnabled);
obs_data_set_string(obj, "readPath", switcher->fileIO.readPath.c_str());
obs_data_set_bool(obj, "writeEnabled", switcher->fileIO.writeEnabled);
obs_data_set_string(obj, "writePath", switcher->fileIO.writePath.c_str());
obs_data_set_string(obj, "priority0", switcher->functionNamesByPriority[0].c_str());
obs_data_set_string(obj, "priority1", switcher->functionNamesByPriority[1].c_str());
obs_data_set_string(obj, "priority2", switcher->functionNamesByPriority[2].c_str());
obs_data_set_string(obj, "priority3", switcher->functionNamesByPriority[3].c_str());
obs_data_set_string(obj, "priority4", switcher->functionNamesByPriority[4].c_str());
obs_data_set_string(obj, "priority5", switcher->functionNamesByPriority[5].c_str());
obs_data_set_obj(save_data, "advanced-scene-switcher", obj);
obs_data_array_release(array);
obs_data_array_release(screenRegionArray);
obs_data_array_release(pauseScenesArray);
obs_data_array_release(pauseWindowsArray);
obs_data_array_release(ignoreWindowsArray);
obs_data_array_release(sceneRoundTripArray);
obs_data_array_release(sceneTransitionsArray);
obs_data_array_release(defaultTransitionsArray);
obs_data_array_release(executableArray);
obs_data_array_release(ignoreIdleWindowsArray);
obs_data_release(obj);
}
else
{
switcher->m.lock();
obs_data_t* obj = obs_data_get_obj(save_data, "advanced-scene-switcher");
obs_data_array_t* array = obs_data_get_array(obj, "switches");
obs_data_array_t* screenRegionArray = obs_data_get_array(obj, "screenRegion");
obs_data_array_t* pauseScenesArray = obs_data_get_array(obj, "pauseScenes");
obs_data_array_t* pauseWindowsArray = obs_data_get_array(obj, "pauseWindows");
obs_data_array_t* ignoreWindowsArray = obs_data_get_array(obj, "ignoreWindows");
obs_data_array_t* sceneRoundTripArray = obs_data_get_array(obj, "sceneRoundTrip");
obs_data_array_t* sceneTransitionsArray = obs_data_get_array(obj, "sceneTransitions");
obs_data_array_t* defaultTransitionsArray = obs_data_get_array(obj, "defaultTransitions");
obs_data_array_t* executableArray = obs_data_get_array(obj, "executableSwitches");
obs_data_array_t* ignoreIdleWindowsArray = obs_data_get_array(obj, "ignoreIdleWindows");
if (!obj)
obj = obs_data_create();
obs_data_set_default_int(obj, "interval", DEFAULT_INTERVAL);
switcher->interval = obs_data_get_int(obj, "interval");
switcher->switchIfNotMatching = obs_data_get_bool(obj, "switch_if_not_matching");
string nonMatchingScene = obs_data_get_string(obj, "non_matching_scene");
bool active = obs_data_get_bool(obj, "active");
switcher->nonMatchingScene = GetWeakSourceByName(nonMatchingScene.c_str());
switcher->switches.clear();
size_t count = obs_data_array_count(array);
for (size_t i = 0; i < count; i++)
{
obs_data_t* array_obj = obs_data_array_item(array, i);
const char* scene = obs_data_get_string(array_obj, "scene");
const char* transition = obs_data_get_string(array_obj, "transition");
const char* window = obs_data_get_string(array_obj, "window_title");
bool fullscreen = obs_data_get_bool(array_obj, "fullscreen");
switcher->switches.emplace_back(GetWeakSourceByName(scene), window,
GetWeakTransitionByName(transition), fullscreen);
obs_data_release(array_obj);
}
switcher->screenRegionSwitches.clear();
count = obs_data_array_count(screenRegionArray);
for (size_t i = 0; i < count; i++)
{
obs_data_t* array_obj = obs_data_array_item(screenRegionArray, i);
const char* scene = obs_data_get_string(array_obj, "screenRegionScene");
const char* transition = obs_data_get_string(array_obj, "transition");
int minX = obs_data_get_int(array_obj, "minX");
int minY = obs_data_get_int(array_obj, "minY");
int maxX = obs_data_get_int(array_obj, "maxX");
int maxY = obs_data_get_int(array_obj, "maxY");
string regionStr = obs_data_get_string(array_obj, "screenRegionStr");
switcher->screenRegionSwitches.emplace_back(GetWeakSourceByName(scene),
GetWeakTransitionByName(transition), minX, minY, maxX, maxY, regionStr);
obs_data_release(array_obj);
}
switcher->pauseScenesSwitches.clear();
count = obs_data_array_count(pauseScenesArray);
for (size_t i = 0; i < count; i++)
{
obs_data_t* array_obj = obs_data_array_item(pauseScenesArray, i);
const char* scene = obs_data_get_string(array_obj, "pauseScene");
switcher->pauseScenesSwitches.emplace_back(GetWeakSourceByName(scene));
obs_data_release(array_obj);
}
switcher->pauseWindowsSwitches.clear();
count = obs_data_array_count(pauseWindowsArray);
for (size_t i = 0; i < count; i++)
{
obs_data_t* array_obj = obs_data_array_item(pauseWindowsArray, i);
const char* window = obs_data_get_string(array_obj, "pauseWindow");
switcher->pauseWindowsSwitches.emplace_back(window);
obs_data_release(array_obj);
}
obs_data_array_release(pauseWindowsArray);
switcher->ignoreWindowsSwitches.clear();
count = obs_data_array_count(ignoreWindowsArray);
for (size_t i = 0; i < count; i++)
{
obs_data_t* array_obj = obs_data_array_item(ignoreWindowsArray, i);
const char* window = obs_data_get_string(array_obj, "ignoreWindow");
switcher->ignoreWindowsSwitches.emplace_back(window);
obs_data_release(array_obj);
}
switcher->sceneRoundTripSwitches.clear();
count = obs_data_array_count(sceneRoundTripArray);
for (size_t i = 0; i < count; i++)
{
obs_data_t* array_obj = obs_data_array_item(sceneRoundTripArray, i);
const char* scene1 = obs_data_get_string(array_obj, "sceneRoundTripScene1");
const char* scene2 = obs_data_get_string(array_obj, "sceneRoundTripScene2");
const char* transition = obs_data_get_string(array_obj, "transition");
int delay = obs_data_get_int(array_obj, "sceneRoundTripDelay");
const char* sceneRoundTripStr = obs_data_get_string(array_obj, "sceneRoundTripStr");
switcher->sceneRoundTripSwitches.emplace_back(GetWeakSourceByName(scene1),
GetWeakSourceByName(scene2), GetWeakTransitionByName(transition), delay,
sceneRoundTripStr);
obs_data_release(array_obj);
}
switcher->sceneTransitions.clear();
count = obs_data_array_count(sceneTransitionsArray);
for (size_t i = 0; i < count; i++)
{
obs_data_t* array_obj = obs_data_array_item(sceneTransitionsArray, i);
const char* scene1 = obs_data_get_string(array_obj, "Scene1");
const char* scene2 = obs_data_get_string(array_obj, "Scene2");
const char* transition = obs_data_get_string(array_obj, "transition");
const char* sceneTransitionsStr = obs_data_get_string(array_obj, "Str");
switcher->sceneTransitions.emplace_back(GetWeakSourceByName(scene1),
GetWeakSourceByName(scene2), GetWeakTransitionByName(transition),
sceneTransitionsStr);
obs_data_release(array_obj);
}
switcher->defaultSceneTransitions.clear();
count = obs_data_array_count(defaultTransitionsArray);
for (size_t i = 0; i < count; i++)
{
obs_data_t* array_obj = obs_data_array_item(defaultTransitionsArray, i);
const char* scene = obs_data_get_string(array_obj, "Scene");
const char* transition = obs_data_get_string(array_obj, "transition");
const char* sceneTransitionsStr = obs_data_get_string(array_obj, "Str");
switcher->defaultSceneTransitions.emplace_back(GetWeakSourceByName(scene),
GetWeakTransitionByName(transition),
sceneTransitionsStr);
obs_data_release(array_obj);
}
switcher->executableSwitches.clear();
count = obs_data_array_count(executableArray);
for (size_t i = 0; i < count; i++)
{
obs_data_t* array_obj = obs_data_array_item(executableArray, i);
const char* scene = obs_data_get_string(array_obj, "scene");
const char* transition = obs_data_get_string(array_obj, "transition");
const char* exe = obs_data_get_string(array_obj, "exefile");
bool infocus = obs_data_get_bool(array_obj, "infocus");
switcher->executableSwitches.emplace_back(
GetWeakSourceByName(scene), GetWeakTransitionByName(transition), exe, infocus);
obs_data_release(array_obj);
}
switcher->ignoreIdleWindows.clear();
count = obs_data_array_count(ignoreIdleWindowsArray);
for (size_t i = 0; i < count; i++)
{
obs_data_t* array_obj = obs_data_array_item(ignoreIdleWindowsArray, i);
const char* window = obs_data_get_string(array_obj, "window");
switcher->ignoreIdleWindows.emplace_back(window);
obs_data_release(array_obj);
}
string autoStopScene = obs_data_get_string(obj, "autoStopSceneName");
switcher->autoStopEnable = obs_data_get_bool(obj, "autoStopEnable");
switcher->autoStopScene = GetWeakSourceByName(autoStopScene.c_str());
string idleSceneName = obs_data_get_string(obj, "idleSceneName");
string idleTransitionName = obs_data_get_string(obj, "idleTransitionName");
switcher->idleData.scene = GetWeakSourceByName(idleSceneName.c_str());
switcher->idleData.transition = GetWeakTransitionByName(idleTransitionName.c_str());
switcher->idleData.idleEnable = obs_data_get_bool(obj, "idleEnable");
switcher->idleData.time = obs_data_get_int(obj, "idleTime");
switcher->fileIO.readEnabled = obs_data_get_bool(obj, "readEnabled");
switcher->fileIO.readPath = obs_data_get_string(obj, "readPath");
switcher->fileIO.writeEnabled = obs_data_get_bool(obj, "writeEnabled");
switcher->fileIO.writePath = obs_data_get_string(obj, "writePath");
obs_data_set_default_string(obj, "priority0", DEFAULT_PRIORITY_0);
obs_data_set_default_string(obj, "priority1", DEFAULT_PRIORITY_1);
obs_data_set_default_string(obj, "priority2", DEFAULT_PRIORITY_2);
obs_data_set_default_string(obj, "priority3", DEFAULT_PRIORITY_3);
obs_data_set_default_string(obj, "priority4", DEFAULT_PRIORITY_4);
obs_data_set_default_string(obj, "priority5", DEFAULT_PRIORITY_5);
switcher->functionNamesByPriority.clear();
switcher->functionNamesByPriority.push_back(obs_data_get_string(obj, "priority0"));
switcher->functionNamesByPriority.push_back(obs_data_get_string(obj, "priority1"));
switcher->functionNamesByPriority.push_back(obs_data_get_string(obj, "priority2"));
switcher->functionNamesByPriority.push_back(obs_data_get_string(obj, "priority3"));
switcher->functionNamesByPriority.push_back(obs_data_get_string(obj, "priority4"));
switcher->functionNamesByPriority.push_back(obs_data_get_string(obj, "priority5"));
if (!switcher->prioFuncsValid())
{
switcher->functionNamesByPriority.clear();
switcher->functionNamesByPriority.push_back(DEFAULT_PRIORITY_0);
switcher->functionNamesByPriority.push_back(DEFAULT_PRIORITY_1);
switcher->functionNamesByPriority.push_back(DEFAULT_PRIORITY_2);
switcher->functionNamesByPriority.push_back(DEFAULT_PRIORITY_3);
switcher->functionNamesByPriority.push_back(DEFAULT_PRIORITY_4);
switcher->functionNamesByPriority.push_back(DEFAULT_PRIORITY_5);
}
obs_data_array_release(array);
obs_data_array_release(screenRegionArray);
obs_data_array_release(pauseScenesArray);
obs_data_array_release(ignoreWindowsArray);
obs_data_array_release(sceneRoundTripArray);
obs_data_array_release(sceneTransitionsArray);
obs_data_array_release(defaultTransitionsArray);
obs_data_array_release(executableArray);
obs_data_array_release(ignoreIdleWindowsArray);
obs_data_release(obj);
switcher->m.unlock();
if (active)
switcher->Start();
else
switcher->Stop();
}
}
bool SwitcherData::sceneChangedDuringWait(){
bool r = false;
obs_source_t* currentSource = obs_frontend_get_current_scene();
string curName = (obs_source_get_name(currentSource));
obs_source_release(currentSource);
if (!waitSceneName.empty() && curName != waitSceneName)
r = true;
waitSceneName = "";
return r;
}
obs_weak_source_t* getNextTransition(obs_weak_source_t* scene1, obs_weak_source_t* scene2);
void switchScene(OBSWeakSource scene, OBSWeakSource transition)
{
obs_source_t* source = obs_weak_source_get_source(scene);
obs_source_t* currentSource = obs_frontend_get_current_scene();
if (source && source != currentSource)
{
obs_weak_source_t* currentScene = obs_source_get_weak_source(currentSource);
obs_weak_source_t* nextTransitionWs = getNextTransition(currentScene, scene);
obs_weak_source_release(currentScene);
if (nextTransitionWs)
{
obs_source_t* nextTransition = obs_weak_source_get_source(nextTransitionWs);
//lock.unlock();
//transitionCv.wait(transitionLock, transitionActiveCheck);
//lock.lock();
obs_frontend_set_current_transition(nextTransition);
obs_source_release(nextTransition);
}
else if (transition)
{
obs_source_t* nextTransition = obs_weak_source_get_source(transition);
//lock.unlock();
//transitionCv.wait(transitionLock, transitionActiveCheck);
//lock.lock();
obs_frontend_set_current_transition(nextTransition);
obs_source_release(nextTransition);
}
obs_frontend_set_current_scene(source);
obs_weak_source_release(nextTransitionWs);
}
obs_source_release(currentSource);
obs_source_release(source);
}
void SwitcherData::Thread()
{
while (true)
{
startLoop:
unique_lock<mutex> lock(m);
unique_lock<mutex> transitionLock(transitionMutex);
bool match = false;
OBSWeakSource scene;
OBSWeakSource transition;
chrono::milliseconds duration(interval);
switcher->Prune();
//sleep for a bit
cv.wait_for(lock, duration);
if (switcher->stop)
{
break;
}
setDefaultSceneTransitions(lock);
if (autoStopEnable)
{
autoStopStreamAndRecording();
}
if (checkPause())
{
continue;
}
//do callbacks 'n stuff in future ...
for (string switchFuncName : functionNamesByPriority)
{
if (switchFuncName == READ_FILE_FUNC)
{
checkSwitchInfoFromFile(match, scene, transition);
}
if (switchFuncName == IDLE_FUNC)
{
checkIdleSwitch(match, scene, transition);
}
if (switchFuncName == EXE_FUNC)
{
checkExeSwitch(match, scene, transition);
}
if (switchFuncName == SCREEN_REGION_FUNC)
{
checkScreenRegionSwitch(match, scene, transition);
}
if (switchFuncName == WINDOW_TITLE_FUNC)
{
checkWindowTitleSwitch(match, scene, transition);
}
if (switchFuncName == ROUND_TRIP_FUNC)
{
checkSceneRoundTrip(match, scene, transition, lock);
if (sceneChangedDuringWait()) //scene might have changed during the sleep
goto startLoop;
}
if (switcher->stop)
{
goto endLoop;
}
if (match)
{
break;
}
}
//switch if no match?
if (!match && switchIfNotMatching && nonMatchingScene)
{
match = true;
scene = nonMatchingScene;
transition = nullptr;
}
if (match)
{
switchScene(scene, transition);
}
}
endLoop:
;
}
void SwitcherData::Start()
{
if (!th.joinable())
{
stop = false;
switcher->th = thread([]()
{
switcher->Thread();
});
}
}
void SwitcherData::Stop()
{
if (th.joinable())
{
switcher->stop = true;
switcher->stop = true;
cv.notify_one();
transitionCv.notify_one();
th.join();
cv.notify_one();
}
}
extern "C" void FreeSceneSwitcher()
{
delete switcher;
switcher = nullptr;
}
//why is OBS_FRONTEND_EVENT_SCENE_CHANGED fired after OBS_FRONTEND_EVENT_TRANSITION_STOPPED?
static void OBSEvent(enum obs_frontend_event event, void* switcher)
{
switch (event){
case OBS_FRONTEND_EVENT_EXIT:
FreeSceneSwitcher();
break;
case OBS_FRONTEND_EVENT_SCENE_CHANGED:
{
SwitcherData* s = (SwitcherData*)switcher;
//wakeup if waiting on timer if scene already changed
lock_guard<mutex> lock(s->m);
if (s->sceneChangedDuringWait())
s->cv.notify_one();
break;
}
case OBS_FRONTEND_EVENT_TRANSITION_STOPPED:
{
break;
}
case OBS_FRONTEND_EVENT_TRANSITION_CHANGED:
{
break;
}
}
}
void startStopHotkeyFunc(void* data, obs_hotkey_id id, obs_hotkey_t* hotkey, bool pressed);
void loadKeybinding(obs_hotkey_id hotkeyId);
extern "C" void InitSceneSwitcher()
{
QAction* action
= (QAction*)obs_frontend_add_tools_menu_qaction(obs_module_text("Advanced Scene Switcher"));
switcher = new SwitcherData;
auto cb = []()
{
obs_frontend_push_ui_translation(obs_module_get_string);
QMainWindow* window = (QMainWindow*)obs_frontend_get_main_window();
SceneSwitcher ss(window);
ss.exec();
obs_frontend_pop_ui_translation();
};
obs_frontend_add_save_callback(SaveSceneSwitcher, nullptr);
obs_frontend_add_event_callback(OBSEvent, switcher);
action->connect(action, &QAction::triggered, cb);
char* path = obs_module_config_path("");
QDir dir(path);
if (!dir.exists())
{
dir.mkpath(".");
}
bfree(path);
obs_hotkey_id pauseHotkeyId = obs_hotkey_register_frontend("startStopSwitcherHotkey",
"Toggle Start/Stop for the Advanced Scene Switcher", startStopHotkeyFunc, NULL);
loadKeybinding(pauseHotkeyId);
}