obs-StreamFX/source/obs/obs-source-tracker.hpp
Michael Fabian 'Xaymar' Dirks e8ec23c4d4 obs/source-tracker: Fix leaked source references
This functionality broke at some point in the past without anyone noticing, resulting in most dropdowns that rely on this functionality being blank.

Fixes #1025
2023-04-05 18:59:03 +02:00

82 lines
3 KiB
C++

/*
* Modern effects for a modern Streamer
* Copyright (C) 2017-2023 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
*/
#pragma once
#include "common.hpp"
#include "obs/obs-weak-source.hpp"
#include "warning-disable.hpp"
#include <functional>
#include <map>
#include <mutex>
#include "warning-enable.hpp"
namespace streamfx::obs {
class source_tracker {
std::map<std::string, ::streamfx::obs::weak_source> _sources;
std::mutex _mutex;
public:
// Callback function for enumerating sources.
//
// @param std::string Name of the Source
// @param obs_source_t* Source
// @return true to abort enumeration, false to keep going.
typedef std::function<bool(std::string, ::streamfx::obs::source)> enumerate_cb_t;
// Filter function for enumerating sources.
//
// @param std::string Name of the Source
// @param obs_source_t* Source
// @return true to skip, false to pass along.
typedef std::function<bool(std::string, ::streamfx::obs::source)> filter_cb_t;
protected:
source_tracker();
public:
~source_tracker();
//! Enumerate all tracked sources
//
// @param enumerate_cb The function called for each tracked source.
// @param filter_cb Filter function to narrow down results.
void enumerate(enumerate_cb_t enumerate_cb, filter_cb_t filter_cb = nullptr);
protected:
void insert_source(obs_source_t* source);
void remove_source(obs_source_t* source);
void rename_source(std::string_view old_name, std::string_view new_name, obs_source_t* source);
public:
static bool filter_sources(std::string name, ::streamfx::obs::source source);
static bool filter_audio_sources(std::string name, ::streamfx::obs::source source);
static bool filter_video_sources(std::string name, ::streamfx::obs::source source);
static bool filter_transitions(std::string name, ::streamfx::obs::source source);
static bool filter_scenes(std::string name, ::streamfx::obs::source source);
private:
static void source_create_handler(void* ptr, calldata_t* data) noexcept;
static void source_destroy_handler(void* ptr, calldata_t* data) noexcept;
static void source_rename_handler(void* ptr, calldata_t* data) noexcept;
public: // Singleton
static std::shared_ptr<streamfx::obs::source_tracker> get();
};
} // namespace streamfx::obs