From 234cbf5dab247871183c842aacadeee4abb3501e Mon Sep 17 00:00:00 2001
From: Michael Fabian 'Xaymar' Dirks <info@xaymar.com>
Date: Wed, 7 Nov 2018 14:07:34 +0100
Subject: [PATCH] obs-tools: Various obs related tools

---
 source/obs-tools.cpp | 76 ++++++++++++++++++++++++++++++++++++++++++++
 source/obs-tools.hpp | 37 +++++++++++++++++++++
 2 files changed, 113 insertions(+)
 create mode 100644 source/obs-tools.cpp
 create mode 100644 source/obs-tools.hpp

diff --git a/source/obs-tools.cpp b/source/obs-tools.cpp
new file mode 100644
index 00000000..fa3a817b
--- /dev/null
+++ b/source/obs-tools.cpp
@@ -0,0 +1,76 @@
+/*
+ * Modern effects for a modern Streamer
+ * Copyright (C) 2018 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
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
+ */
+
+#include "obs-tools.hpp"
+#include <map>
+
+struct scs_searchdata {
+	obs_source_t* source;
+	bool          found = false;
+	std::map<obs_source_t*, bool> visited;
+};
+
+static bool scs_contains(scs_searchdata& sd, obs_source_t* source);
+
+static void scs_enum_active_cb(obs_source_t* parent, obs_source_t* child, void* searchdata)
+{
+	scs_searchdata& sd = reinterpret_cast<scs_searchdata&>(*reinterpret_cast<scs_searchdata*>(searchdata));
+	scs_contains(sd, child);
+}
+
+static bool scs_enum_items_cb(obs_scene_t* scene, obs_sceneitem_t* item, void* searchdata)
+{
+	scs_searchdata& sd     = reinterpret_cast<scs_searchdata&>(*reinterpret_cast<scs_searchdata*>(searchdata));
+	obs_source_t*   source = obs_sceneitem_get_source(item);
+	return scs_contains(sd, source);
+}
+
+static bool scs_contains(scs_searchdata& sd, obs_source_t* source)
+{
+	if (sd.visited.find(source) != sd.visited.end()) {
+		return false;
+	} else {
+		sd.visited.insert({source, true});
+	}
+
+	if (source == sd.source) {
+		sd.found = true;
+		return true;
+	} else {
+		if (strcmp(obs_source_get_id(source), "scene")) {
+			obs_scene_t* nscene = obs_scene_from_source(source);
+			obs_scene_enum_items(nscene, scs_enum_items_cb, &sd);
+		} else {
+			obs_source_enum_active_sources(source, scs_enum_active_cb, &sd);
+		}
+	}
+
+	if (sd.found) {
+		return false;
+	}
+	return true;
+}
+
+bool obs::tools::scene_contains_source(obs_scene_t* scene, obs_source_t* source)
+{
+	scs_searchdata sd;
+	sd.source = source;
+	obs_scene_enum_items(scene, scs_enum_items_cb, &sd);
+	return sd.found;
+}
diff --git a/source/obs-tools.hpp b/source/obs-tools.hpp
new file mode 100644
index 00000000..5f2a3997
--- /dev/null
+++ b/source/obs-tools.hpp
@@ -0,0 +1,37 @@
+/*
+ * Modern effects for a modern Streamer
+ * Copyright (C) 2018 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
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * 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_STREAM_EFFECTS_OBS_TOOLS_HPP
+#define OBS_STREAM_EFFECTS_OBS_TOOLS_HPP
+#pragma once
+
+#include <cinttypes>
+#include <string>
+
+extern "C" {
+#include "obs.h"
+}
+
+namespace obs {
+	namespace tools {
+		bool scene_contains_source(obs_scene_t* scene, obs_source_t* source);
+	}
+} // namespace obs
+
+#endif