obs-source.c 123.5 KB
Newer Older
J
jp9000 已提交
1
/******************************************************************************
2
    Copyright (C) 2013-2014 by Hugh Bailey <obs.jim@gmail.com>
J
jp9000 已提交
3 4 5

    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
6
    the Free Software Foundation, either version 2 of the License, or
J
jp9000 已提交
7 8 9 10 11 12 13 14 15 16 17
    (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, see <http://www.gnu.org/licenses/>.
******************************************************************************/

18
#include <inttypes.h>
C
cg2121 已提交
19
#include <math.h>
20

21
#include "media-io/format-conversion.h"
22
#include "media-io/video-frame.h"
23
#include "media-io/audio-io.h"
J
jp9000 已提交
24
#include "util/threading.h"
25
#include "util/platform.h"
26
#include "callback/calldata.h"
27 28
#include "graphics/matrix3.h"
#include "graphics/vec3.h"
29

J
jp9000 已提交
30
#include "obs.h"
J
jp9000 已提交
31
#include "obs-internal.h"
J
jp9000 已提交
32

J
jp9000 已提交
33
static inline bool data_valid(const struct obs_source *source, const char *f)
34
{
35
	return obs_source_valid(source, f) && source->context.data;
36 37
}

J
jp9000 已提交
38 39 40 41 42
static inline bool deinterlacing_enabled(const struct obs_source *source)
{
	return source->deinterlace_mode != OBS_DEINTERLACE_MODE_DISABLE;
}

43
struct obs_source_info *get_source_info(const char *id)
J
jp9000 已提交
44
{
45 46
	for (size_t i = 0; i < obs->source_types.num; i++) {
		struct obs_source_info *info = &obs->source_types.array[i];
47
		if (strcmp(info->id, id) == 0)
J
jp9000 已提交
48 49 50 51 52 53
			return info;
	}

	return NULL;
}

54 55 56
static const char *source_signals[] = {
	"void destroy(ptr source)",
	"void remove(ptr source)",
J
jp9000 已提交
57 58
	"void save(ptr source)",
	"void load(ptr source)",
59 60 61 62
	"void activate(ptr source)",
	"void deactivate(ptr source)",
	"void show(ptr source)",
	"void hide(ptr source)",
J
jp9000 已提交
63
	"void mute(ptr source, bool muted)",
P
Palana 已提交
64 65 66 67
	"void push_to_mute_changed(ptr source, bool enabled)",
	"void push_to_mute_delay(ptr source, int delay)",
	"void push_to_talk_changed(ptr source, bool enabled)",
	"void push_to_talk_delay(ptr source, int delay)",
68
	"void enable(ptr source, bool enabled)",
J
jp9000 已提交
69
	"void rename(ptr source, string new_name, string prev_name)",
70
	"void volume(ptr source, in out float volume)",
71
	"void update_properties(ptr source)",
J
jp9000 已提交
72
	"void update_flags(ptr source, int flags)",
J
jp9000 已提交
73
	"void audio_sync(ptr source, int out int offset)",
74
	"void audio_mixers(ptr source, in out int mixers)",
J
jp9000 已提交
75 76
	"void filter_add(ptr source, ptr filter)",
	"void filter_remove(ptr source, ptr filter)",
77
	"void reorder_filters(ptr source)",
J
jp9000 已提交
78 79 80
	"void transition_start(ptr source)",
	"void transition_video_stop(ptr source)",
	"void transition_stop(ptr source)",
J
jp9000 已提交
81
	NULL,
82 83
};

J
jp9000 已提交
84 85 86
bool obs_source_init_context(struct obs_source *source, obs_data_t *settings,
			     const char *name, obs_data_t *hotkey_data,
			     bool private)
87
{
88
	if (!obs_context_data_init(&source->context, OBS_OBJ_TYPE_SOURCE,
J
jp9000 已提交
89
				   settings, name, hotkey_data, private))
90 91
		return false;

92
	return signal_handler_add_array(source->context.signals,
J
jp9000 已提交
93
					source_signals);
94 95
}

96
const char *obs_source_get_display_name(const char *id)
97
{
98
	const struct obs_source_info *info = get_source_info(id);
99
	return (info != NULL) ? info->get_name(info->type_data) : NULL;
100 101
}

102 103
static void allocate_audio_output_buffer(struct obs_source *source)
{
J
jp9000 已提交
104 105
	size_t size = sizeof(float) * AUDIO_OUTPUT_FRAMES * MAX_AUDIO_CHANNELS *
		      MAX_AUDIO_MIXES;
106 107 108 109 110 111 112 113 114 115 116 117
	float *ptr = bzalloc(size);

	for (size_t mix = 0; mix < MAX_AUDIO_MIXES; mix++) {
		size_t mix_pos = mix * AUDIO_OUTPUT_FRAMES * MAX_AUDIO_CHANNELS;

		for (size_t i = 0; i < MAX_AUDIO_CHANNELS; i++) {
			source->audio_output_buf[mix][i] =
				ptr + mix_pos + AUDIO_OUTPUT_FRAMES * i;
		}
	}
}

J
jp9000 已提交
118 119 120 121 122 123 124 125 126 127
static void allocate_audio_mix_buffer(struct obs_source *source)
{
	size_t size = sizeof(float) * AUDIO_OUTPUT_FRAMES * MAX_AUDIO_CHANNELS;
	float *ptr = bzalloc(size);

	for (size_t i = 0; i < MAX_AUDIO_CHANNELS; i++) {
		source->audio_mix_buf[i] = ptr + AUDIO_OUTPUT_FRAMES * i;
	}
}

J
jp9000 已提交
128 129 130
static inline bool is_async_video_source(const struct obs_source *source)
{
	return (source->info.output_flags & OBS_SOURCE_ASYNC_VIDEO) ==
J
jp9000 已提交
131
	       OBS_SOURCE_ASYNC_VIDEO;
J
jp9000 已提交
132 133
}

134 135 136 137 138 139 140 141 142 143
static inline bool is_audio_source(const struct obs_source *source)
{
	return source->info.output_flags & OBS_SOURCE_AUDIO;
}

static inline bool is_composite_source(const struct obs_source *source)
{
	return source->info.output_flags & OBS_SOURCE_COMPOSITE;
}

J
jp9000 已提交
144 145
extern char *find_libobs_data_file(const char *file);

146
/* internal initialization */
147
bool obs_source_init(struct obs_source *source)
J
jp9000 已提交
148
{
J
jp9000 已提交
149 150
	pthread_mutexattr_t attr;

J
jp9000 已提交
151
	source->user_volume = 1.0f;
J
jp9000 已提交
152
	source->volume = 1.0f;
J
jp9000 已提交
153
	source->sync_offset = 0;
C
cg2121 已提交
154
	source->balance = 0.5f;
155
	pthread_mutex_init_value(&source->filter_mutex);
156
	pthread_mutex_init_value(&source->async_mutex);
157
	pthread_mutex_init_value(&source->audio_mutex);
158
	pthread_mutex_init_value(&source->audio_buf_mutex);
159
	pthread_mutex_init_value(&source->audio_cb_mutex);
160

J
jp9000 已提交
161 162 163 164 165
	if (pthread_mutexattr_init(&attr) != 0)
		return false;
	if (pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE) != 0)
		return false;
	if (pthread_mutex_init(&source->filter_mutex, &attr) != 0)
166
		return false;
167 168
	if (pthread_mutex_init(&source->audio_buf_mutex, NULL) != 0)
		return false;
J
jp9000 已提交
169 170
	if (pthread_mutex_init(&source->audio_actions_mutex, NULL) != 0)
		return false;
171 172
	if (pthread_mutex_init(&source->audio_cb_mutex, NULL) != 0)
		return false;
173 174
	if (pthread_mutex_init(&source->audio_mutex, NULL) != 0)
		return false;
175
	if (pthread_mutex_init(&source->async_mutex, NULL) != 0)
176
		return false;
J
jp9000 已提交
177

178
	if (is_audio_source(source) || is_composite_source(source))
179
		allocate_audio_output_buffer(source);
J
jp9000 已提交
180 181
	if (source->info.audio_mix)
		allocate_audio_mix_buffer(source);
182

183 184 185 186 187 188 189 190
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION) {
		if (!obs_transition_init(source))
			return false;
	}

	source->control = bzalloc(sizeof(obs_weak_source_t));
	source->deinterlace_top_first = true;
	source->control->source = source;
191
	source->audio_mixers = 0xFF;
192

193
	if (is_audio_source(source)) {
194 195 196
		pthread_mutex_lock(&obs->data.audio_sources_mutex);

		source->next_audio_source = obs->data.first_audio_source;
J
jp9000 已提交
197
		source->prev_next_audio_source = &obs->data.first_audio_source;
198 199 200 201 202 203
		if (obs->data.first_audio_source)
			obs->data.first_audio_source->prev_next_audio_source =
				&source->next_audio_source;
		obs->data.first_audio_source = source;

		pthread_mutex_unlock(&obs->data.audio_sources_mutex);
204
	}
205

206 207
	source->private_settings = obs_data_create();

J
jp9000 已提交
208 209
	obs_context_data_insert(&source->context, &obs->data.sources_mutex,
				&obs->data.first_source);
210
	return true;
J
jp9000 已提交
211 212
}

J
jp9000 已提交
213 214
static bool obs_source_hotkey_mute(void *data, obs_hotkey_pair_id id,
				   obs_hotkey_t *key, bool pressed)
P
Palana 已提交
215 216 217 218 219 220
{
	UNUSED_PARAMETER(id);
	UNUSED_PARAMETER(key);

	struct obs_source *source = data;

J
jp9000 已提交
221 222
	if (!pressed || obs_source_muted(source))
		return false;
P
Palana 已提交
223 224 225 226 227

	obs_source_set_muted(source, true);
	return true;
}

J
jp9000 已提交
228 229
static bool obs_source_hotkey_unmute(void *data, obs_hotkey_pair_id id,
				     obs_hotkey_t *key, bool pressed)
P
Palana 已提交
230 231 232 233 234 235
{
	UNUSED_PARAMETER(id);
	UNUSED_PARAMETER(key);

	struct obs_source *source = data;

J
jp9000 已提交
236 237
	if (!pressed || !obs_source_muted(source))
		return false;
P
Palana 已提交
238 239 240 241 242

	obs_source_set_muted(source, false);
	return true;
}

J
jp9000 已提交
243 244
static void obs_source_hotkey_push_to_mute(void *data, obs_hotkey_id id,
					   obs_hotkey_t *key, bool pressed)
P
Palana 已提交
245
{
J
jp9000 已提交
246 247 248
	struct audio_action action = {.timestamp = os_gettime_ns(),
				      .type = AUDIO_ACTION_PTM,
				      .set = pressed};
J
jp9000 已提交
249

P
Palana 已提交
250 251 252 253 254
	UNUSED_PARAMETER(id);
	UNUSED_PARAMETER(key);

	struct obs_source *source = data;

J
jp9000 已提交
255 256 257 258 259
	pthread_mutex_lock(&source->audio_actions_mutex);
	da_push_back(source->audio_actions, &action);
	pthread_mutex_unlock(&source->audio_actions_mutex);

	source->user_push_to_mute_pressed = pressed;
P
Palana 已提交
260 261
}

J
jp9000 已提交
262 263
static void obs_source_hotkey_push_to_talk(void *data, obs_hotkey_id id,
					   obs_hotkey_t *key, bool pressed)
P
Palana 已提交
264
{
J
jp9000 已提交
265 266 267
	struct audio_action action = {.timestamp = os_gettime_ns(),
				      .type = AUDIO_ACTION_PTT,
				      .set = pressed};
J
jp9000 已提交
268

P
Palana 已提交
269 270 271 272 273
	UNUSED_PARAMETER(id);
	UNUSED_PARAMETER(key);

	struct obs_source *source = data;

J
jp9000 已提交
274 275 276 277 278
	pthread_mutex_lock(&source->audio_actions_mutex);
	da_push_back(source->audio_actions, &action);
	pthread_mutex_unlock(&source->audio_actions_mutex);

	source->user_push_to_talk_pressed = pressed;
P
Palana 已提交
279 280 281 282
}

static void obs_source_init_audio_hotkeys(struct obs_source *source)
{
283 284
	if (!(source->info.output_flags & OBS_SOURCE_AUDIO) ||
	    source->info.type != OBS_SOURCE_TYPE_INPUT) {
J
jp9000 已提交
285
		source->mute_unmute_key = OBS_INVALID_HOTKEY_ID;
P
Palana 已提交
286 287 288 289
		source->push_to_talk_key = OBS_INVALID_HOTKEY_ID;
		return;
	}

J
jp9000 已提交
290 291 292 293
	source->mute_unmute_key = obs_hotkey_pair_register_source(
		source, "libobs.mute", obs->hotkeys.mute, "libobs.unmute",
		obs->hotkeys.unmute, obs_source_hotkey_mute,
		obs_source_hotkey_unmute, source, source);
P
Palana 已提交
294

J
jp9000 已提交
295 296 297
	source->push_to_mute_key = obs_hotkey_register_source(
		source, "libobs.push-to-mute", obs->hotkeys.push_to_mute,
		obs_source_hotkey_push_to_mute, source);
P
Palana 已提交
298

J
jp9000 已提交
299 300 301
	source->push_to_talk_key = obs_hotkey_register_source(
		source, "libobs.push-to-talk", obs->hotkeys.push_to_talk,
		obs_source_hotkey_push_to_talk, source);
P
Palana 已提交
302 303
}

J
jp9000 已提交
304
static obs_source_t *obs_source_create_internal(const char *id,
J
jp9000 已提交
305 306 307 308
						const char *name,
						obs_data_t *settings,
						obs_data_t *hotkey_data,
						bool private)
J
jp9000 已提交
309
{
310
	struct obs_source *source = bzalloc(sizeof(struct obs_source));
J
jp9000 已提交
311

312
	const struct obs_source_info *info = get_source_info(id);
J
jp9000 已提交
313
	if (!info) {
P
Palana 已提交
314
		blog(LOG_ERROR, "Source ID '%s' not found", id);
J
jp9000 已提交
315

J
jp9000 已提交
316
		source->info.id = bstrdup(id);
317 318 319
		source->owns_info_id = true;
	} else {
		source->info = *info;
320 321 322 323 324 325

		/* Always mark filters as private so they aren't found by
		 * source enum/search functions.
		 *
		 * XXX: Fix design flaws with filters */
		if (info->type == OBS_SOURCE_TYPE_FILTER)
J
jp9000 已提交
326 327
		private
		= true;
328
	}
329

J
jp9000 已提交
330
	source->mute_unmute_key = OBS_INVALID_HOTKEY_PAIR_ID;
P
Palana 已提交
331 332 333
	source->push_to_mute_key = OBS_INVALID_HOTKEY_ID;
	source->push_to_talk_key = OBS_INVALID_HOTKEY_ID;

J
jp9000 已提交
334
	if (!obs_source_init_context(source, settings, name, hotkey_data,
J
jp9000 已提交
335
				     private))
336 337
		goto fail;

338
	if (info) {
339 340 341 342
		if (info->get_defaults) {
			info->get_defaults(source->context.settings);
		}
		if (info->get_defaults2) {
343
			info->get_defaults2(info->type_data,
J
jp9000 已提交
344
					    source->context.settings);
345
		}
346
	}
J
jp9000 已提交
347

348
	if (!obs_source_init(source))
349 350
		goto fail;

J
jp9000 已提交
351 352
	if (!private)
		obs_source_init_audio_hotkeys(source);
P
Palana 已提交
353

354 355
	/* allow the source to be created even if creation fails so that the
	 * user's data doesn't become lost */
J
jp9000 已提交
356
	if (info && info->create)
J
jp9000 已提交
357 358
		source->context.data =
			info->create(source->context.settings, source);
J
jp9000 已提交
359
	if (info->create && !source->context.data)
360
		blog(LOG_ERROR, "Failed to create source '%s'!", name);
361

J
jp9000 已提交
362 363
	blog(LOG_DEBUG, "%ssource '%s' (%s) created", private ? "private " : "",
	     name, id);
J
jp9000 已提交
364

365
	source->flags = source->default_flags;
366
	source->enabled = true;
367 368 369 370 371

	if (!private) {
		obs_source_dosignal(source, "source_create", NULL);
	}

J
jp9000 已提交
372
	return source;
373 374 375 376 377

fail:
	blog(LOG_ERROR, "obs_source_create failed");
	obs_source_destroy(source);
	return NULL;
J
jp9000 已提交
378 379
}

J
jp9000 已提交
380
obs_source_t *obs_source_create(const char *id, const char *name,
J
jp9000 已提交
381
				obs_data_t *settings, obs_data_t *hotkey_data)
J
jp9000 已提交
382 383
{
	return obs_source_create_internal(id, name, settings, hotkey_data,
J
jp9000 已提交
384
					  false);
J
jp9000 已提交
385 386 387
}

obs_source_t *obs_source_create_private(const char *id, const char *name,
J
jp9000 已提交
388
					obs_data_t *settings)
J
jp9000 已提交
389 390 391 392
{
	return obs_source_create_internal(id, name, settings, NULL, true);
}

393 394 395 396 397 398 399 400
static char *get_new_filter_name(obs_source_t *dst, const char *name)
{
	struct dstr new_name = {0};
	int inc = 0;

	dstr_copy(&new_name, name);

	for (;;) {
J
jp9000 已提交
401 402
		obs_source_t *existing_filter =
			obs_source_get_filter_by_name(dst, new_name.array);
403 404 405 406 407 408 409 410 411 412 413
		if (!existing_filter)
			break;

		obs_source_release(existing_filter);

		dstr_printf(&new_name, "%s %d", name, ++inc + 1);
	}

	return new_name.array;
}

414
static void duplicate_filters(obs_source_t *dst, obs_source_t *src,
J
jp9000 已提交
415
			      bool private)
416
{
J
jp9000 已提交
417
	DARRAY(obs_source_t *) filters;
418 419 420 421 422 423 424 425 426 427 428

	da_init(filters);

	pthread_mutex_lock(&src->filter_mutex);
	for (size_t i = 0; i < src->filters.num; i++)
		obs_source_addref(src->filters.array[i]);
	da_copy(filters, src->filters);
	pthread_mutex_unlock(&src->filter_mutex);

	for (size_t i = filters.num; i > 0; i--) {
		obs_source_t *src_filter = filters.array[i - 1];
J
jp9000 已提交
429 430
		char *new_name =
			get_new_filter_name(dst, src_filter->context.name);
431
		bool enabled = obs_source_enabled(src_filter);
432

J
jp9000 已提交
433 434
		obs_source_t *dst_filter =
			obs_source_duplicate(src_filter, new_name, private);
435
		obs_source_set_enabled(dst_filter, enabled);
436

437
		bfree(new_name);
438 439 440 441 442 443 444 445
		obs_source_filter_add(dst, dst_filter);
		obs_source_release(dst_filter);
		obs_source_release(src_filter);
	}

	da_free(filters);
}

446 447
void obs_source_copy_filters(obs_source_t *dst, obs_source_t *src)
{
448 449 450 451 452
	if (!obs_source_valid(dst, "obs_source_copy_filters"))
		return;
	if (!obs_source_valid(src, "obs_source_copy_filters"))
		return;

453
	duplicate_filters(dst, src, dst->context.private);
454 455
}

J
jp9000 已提交
456 457
extern obs_scene_t *obs_group_from_source(const obs_source_t *source);

J
jp9000 已提交
458 459
obs_source_t *obs_source_duplicate(obs_source_t *source, const char *new_name,
				   bool create_private)
460 461 462 463 464 465 466 467 468 469 470 471
{
	obs_source_t *new_source;
	obs_data_t *settings;

	if (!obs_source_valid(source, "obs_source_duplicate"))
		return NULL;

	if ((source->info.output_flags & OBS_SOURCE_DO_NOT_DUPLICATE) != 0) {
		obs_source_addref(source);
		return source;
	}

472 473
	if (source->info.type == OBS_SOURCE_TYPE_SCENE) {
		obs_scene_t *scene = obs_scene_from_source(source);
J
jp9000 已提交
474 475 476 477 478
		if (!scene)
			scene = obs_group_from_source(source);
		if (!scene)
			return NULL;

J
jp9000 已提交
479 480 481 482
		obs_scene_t *new_scene = obs_scene_duplicate(
			scene, new_name,
			create_private ? OBS_SCENE_DUP_PRIVATE_COPY
				       : OBS_SCENE_DUP_COPY);
483 484
		obs_source_t *new_source = obs_scene_get_source(new_scene);
		return new_source;
485 486
	}

487 488 489
	settings = obs_data_create();
	obs_data_apply(settings, source->context.settings);

J
jp9000 已提交
490 491 492 493 494
	new_source = create_private
			     ? obs_source_create_private(source->info.id,
							 new_name, settings)
			     : obs_source_create(source->info.id, new_name,
						 settings, NULL);
495 496 497 498 499 500 501 502 503

	new_source->audio_mixers = source->audio_mixers;
	new_source->sync_offset = source->sync_offset;
	new_source->user_volume = source->user_volume;
	new_source->user_muted = source->user_muted;
	new_source->volume = source->volume;
	new_source->muted = source->muted;
	new_source->flags = source->flags;

504 505
	obs_data_apply(new_source->private_settings, source->private_settings);

506 507 508 509 510 511 512
	if (source->info.type != OBS_SOURCE_TYPE_FILTER)
		duplicate_filters(new_source, source, create_private);

	obs_data_release(settings);
	return new_source;
}

513
void obs_source_frame_init(struct obs_source_frame *frame,
J
jp9000 已提交
514 515
			   enum video_format format, uint32_t width,
			   uint32_t height)
516
{
517
	struct video_frame vid_frame;
J
jp9000 已提交
518

519
	if (!obs_ptr_valid(frame, "obs_source_frame_init"))
J
jp9000 已提交
520 521
		return;

522
	video_frame_init(&vid_frame, format, width, height);
523
	frame->format = format;
J
jp9000 已提交
524
	frame->width = width;
525
	frame->height = height;
526

527
	for (size_t i = 0; i < MAX_AV_PLANES; i++) {
J
jp9000 已提交
528
		frame->data[i] = vid_frame.data[i];
529
		frame->linesize[i] = vid_frame.linesize[i];
530 531 532
	}
}

533 534 535 536 537 538
static inline void obs_source_frame_decref(struct obs_source_frame *frame)
{
	if (os_atomic_dec_long(&frame->refs) == 0)
		obs_source_frame_destroy(frame);
}

539
static bool obs_source_filter_remove_refless(obs_source_t *source,
J
jp9000 已提交
540
					     obs_source_t *filter);
541

542
void obs_source_destroy(struct obs_source *source)
J
jp9000 已提交
543
{
544
	size_t i;
545

546
	if (!obs_source_valid(source, "obs_source_destroy"))
J
jp9000 已提交
547 548
		return;

J
jp9000 已提交
549 550 551
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
		obs_transition_clear(source);

552 553 554 555 556 557 558 559 560
	pthread_mutex_lock(&obs->data.audio_sources_mutex);
	if (source->prev_next_audio_source) {
		*source->prev_next_audio_source = source->next_audio_source;
		if (source->next_audio_source)
			source->next_audio_source->prev_next_audio_source =
				source->prev_next_audio_source;
	}
	pthread_mutex_unlock(&obs->data.audio_sources_mutex);

561 562 563 564 565 566
	if (source->filter_parent)
		obs_source_filter_remove_refless(source->filter_parent, source);

	while (source->filters.num)
		obs_source_filter_remove(source, source->filters.array[0]);

567 568
	obs_context_data_remove(&source->context);

569
	blog(LOG_DEBUG, "%ssource '%s' destroyed",
J
jp9000 已提交
570
	     source->context.private ? "private " : "", source->context.name);
571

572
	obs_source_dosignal(source, "source_destroy", "destroy");
573

574
	if (source->context.data) {
575
		source->info.destroy(source->context.data);
576 577
		source->context.data = NULL;
	}
578

J
jp9000 已提交
579 580
	audio_monitor_destroy(source->monitor);

P
Palana 已提交
581 582 583 584
	obs_hotkey_unregister(source->push_to_talk_key);
	obs_hotkey_unregister(source->push_to_mute_key);
	obs_hotkey_pair_unregister(source->mute_unmute_key);

585
	for (i = 0; i < source->async_cache.num; i++)
586
		obs_source_frame_decref(source->async_cache.array[i].frame);
587

588
	gs_enter_context(obs->video.graphics);
589 590
	if (source->async_texrender)
		gs_texrender_destroy(source->async_texrender);
J
jp9000 已提交
591 592
	if (source->async_prev_texrender)
		gs_texrender_destroy(source->async_prev_texrender);
593 594 595 596
	for (size_t c = 0; c < MAX_AV_PLANES; c++) {
		gs_texture_destroy(source->async_textures[c]);
		gs_texture_destroy(source->async_prev_textures[c]);
	}
597 598
	if (source->filter_texrender)
		gs_texrender_destroy(source->filter_texrender);
599
	gs_leave_context();
J
jp9000 已提交
600

J
jp9000 已提交
601
	for (i = 0; i < MAX_AV_PLANES; i++)
602
		bfree(source->audio_data.data[i]);
603 604
	for (i = 0; i < MAX_AUDIO_CHANNELS; i++)
		circlebuf_free(&source->audio_input_buf[i]);
605
	audio_resampler_destroy(source->resampler);
606
	bfree(source->audio_output_buf[0][0]);
J
jp9000 已提交
607
	bfree(source->audio_mix_buf[0]);
608

609 610
	obs_source_frame_destroy(source->async_preload_frame);

J
jp9000 已提交
611 612 613
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
		obs_transition_free(source);

J
jp9000 已提交
614
	da_free(source->audio_actions);
615
	da_free(source->audio_cb_list);
616
	da_free(source->async_cache);
617
	da_free(source->async_frames);
618 619
	da_free(source->filters);
	pthread_mutex_destroy(&source->filter_mutex);
J
jp9000 已提交
620
	pthread_mutex_destroy(&source->audio_actions_mutex);
621
	pthread_mutex_destroy(&source->audio_buf_mutex);
622
	pthread_mutex_destroy(&source->audio_cb_mutex);
623
	pthread_mutex_destroy(&source->audio_mutex);
624
	pthread_mutex_destroy(&source->async_mutex);
625
	obs_data_release(source->private_settings);
626
	obs_context_data_free(&source->context);
627

628
	if (source->owns_info_id)
J
jp9000 已提交
629
		bfree((void *)source->info.id);
630

631 632 633
	bfree(source);
}

634
void obs_source_addref(obs_source_t *source)
635
{
636 637 638 639
	if (!source)
		return;

	obs_ref_addref(&source->control->ref);
640 641
}

642
void obs_source_release(obs_source_t *source)
643
{
644 645
	if (!obs) {
		blog(LOG_WARNING, "Tried to release a source when the OBS "
J
jp9000 已提交
646
				  "core is shut down!");
647 648 649
		return;
	}

P
Palana 已提交
650 651
	if (!source)
		return;
652

653 654
	obs_weak_source_t *control = source->control;
	if (obs_ref_release(&control->ref)) {
P
Palana 已提交
655
		obs_source_destroy(source);
656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706
		obs_weak_source_release(control);
	}
}

void obs_weak_source_addref(obs_weak_source_t *weak)
{
	if (!weak)
		return;

	obs_weak_ref_addref(&weak->ref);
}

void obs_weak_source_release(obs_weak_source_t *weak)
{
	if (!weak)
		return;

	if (obs_weak_ref_release(&weak->ref))
		bfree(weak);
}

obs_source_t *obs_source_get_ref(obs_source_t *source)
{
	if (!source)
		return NULL;

	return obs_weak_source_get_source(source->control);
}

obs_weak_source_t *obs_source_get_weak_source(obs_source_t *source)
{
	if (!source)
		return NULL;

	obs_weak_source_t *weak = source->control;
	obs_weak_source_addref(weak);
	return weak;
}

obs_source_t *obs_weak_source_get_source(obs_weak_source_t *weak)
{
	if (!weak)
		return NULL;

	if (obs_weak_ref_get_ref(&weak->ref))
		return weak->source;

	return NULL;
}

bool obs_weak_source_references_source(obs_weak_source_t *weak,
J
jp9000 已提交
707
				       obs_source_t *source)
708 709
{
	return weak && source && weak->source == source;
710 711
}

712
void obs_source_remove(obs_source_t *source)
713
{
714 715 716
	if (!obs_source_valid(source, "obs_source_remove"))
		return;

717 718
	if (!source->removed) {
		source->removed = true;
719
		obs_source_dosignal(source, "source_remove", "remove");
720
	}
721 722
}

723
bool obs_source_removed(const obs_source_t *source)
724
{
J
jp9000 已提交
725 726
	return obs_source_valid(source, "obs_source_removed") ? source->removed
							      : true;
J
jp9000 已提交
727 728
}

729
static inline obs_data_t *get_defaults(const struct obs_source_info *info)
J
jp9000 已提交
730
{
731
	obs_data_t *settings = obs_data_create();
732 733 734
	if (info->get_defaults2)
		info->get_defaults2(info->type_data, settings);
	else if (info->get_defaults)
735
		info->get_defaults(settings);
J
jp9000 已提交
736 737 738
	return settings;
}

739
obs_data_t *obs_source_settings(const char *id)
J
jp9000 已提交
740
{
741
	const struct obs_source_info *info = get_source_info(id);
J
jp9000 已提交
742
	return (info) ? get_defaults(info) : NULL;
J
jp9000 已提交
743 744
}

745
obs_data_t *obs_get_source_defaults(const char *id)
J
jp9000 已提交
746
{
747
	const struct obs_source_info *info = get_source_info(id);
J
jp9000 已提交
748 749 750
	return info ? get_defaults(info) : NULL;
}

751
obs_properties_t *obs_get_source_properties(const char *id)
J
jp9000 已提交
752
{
753
	const struct obs_source_info *info = get_source_info(id);
754
	if (info && (info->get_properties || info->get_properties2)) {
J
jp9000 已提交
755
		obs_data_t *defaults = get_defaults(info);
756 757 758 759 760 761
		obs_properties_t *props;

		if (info->get_properties2)
			props = info->get_properties2(NULL, info->type_data);
		else
			props = info->get_properties(NULL);
J
jp9000 已提交
762

763
		obs_properties_apply_settings(props, defaults);
J
jp9000 已提交
764
		obs_data_release(defaults);
765
		return props;
J
jp9000 已提交
766
	}
J
jp9000 已提交
767 768 769
	return NULL;
}

770 771 772
bool obs_is_source_configurable(const char *id)
{
	const struct obs_source_info *info = get_source_info(id);
773
	return info && (info->get_properties || info->get_properties2);
774 775 776 777 778
}

bool obs_source_configurable(const obs_source_t *source)
{
	return data_valid(source, "obs_source_configurable") &&
J
jp9000 已提交
779
	       (source->info.get_properties || source->info.get_properties2);
780 781
}

782
obs_properties_t *obs_source_properties(const obs_source_t *source)
783
{
J
jp9000 已提交
784 785 786
	if (!data_valid(source, "obs_source_properties"))
		return NULL;

787 788 789
	if (source->info.get_properties2) {
		obs_properties_t *props;
		props = source->info.get_properties2(source->context.data,
J
jp9000 已提交
790
						     source->info.type_data);
791 792 793 794
		obs_properties_apply_settings(props, source->context.settings);
		return props;

	} else if (source->info.get_properties) {
795
		obs_properties_t *props;
796
		props = source->info.get_properties(source->context.data);
797
		obs_properties_apply_settings(props, source->context.settings);
J
jp9000 已提交
798 799 800
		return props;
	}

801 802 803
	return NULL;
}

804
uint32_t obs_source_get_output_flags(const obs_source_t *source)
J
jp9000 已提交
805
{
J
jp9000 已提交
806 807 808
	return obs_source_valid(source, "obs_source_get_output_flags")
		       ? source->info.output_flags
		       : 0;
J
jp9000 已提交
809 810
}

811
uint32_t obs_get_source_output_flags(const char *id)
812
{
813
	const struct obs_source_info *info = get_source_info(id);
814 815 816
	return info ? info->output_flags : 0;
}

817
static void obs_source_deferred_update(obs_source_t *source)
818
{
819 820
	if (source->context.data && source->info.update)
		source->info.update(source->context.data,
J
jp9000 已提交
821
				    source->context.settings);
822

823 824 825
	source->defer_update = false;
}

826
void obs_source_update(obs_source_t *source, obs_data_t *settings)
J
jp9000 已提交
827
{
828 829
	if (!obs_source_valid(source, "obs_source_update"))
		return;
J
jp9000 已提交
830

831 832 833 834 835 836 837
	if (settings)
		obs_data_apply(source->context.settings, settings);

	if (source->info.output_flags & OBS_SOURCE_VIDEO) {
		source->defer_update = true;
	} else if (source->context.data && source->info.update) {
		source->info.update(source->context.data,
J
jp9000 已提交
838
				    source->context.settings);
839
	}
J
jp9000 已提交
840 841
}

842 843
void obs_source_update_properties(obs_source_t *source)
{
844 845
	if (!obs_source_valid(source, "obs_source_update_properties"))
		return;
846

847
	obs_source_dosignal(source, NULL, "update_properties");
848 849
}

850
void obs_source_send_mouse_click(obs_source_t *source,
J
jp9000 已提交
851 852 853
				 const struct obs_mouse_event *event,
				 int32_t type, bool mouse_up,
				 uint32_t click_count)
K
kc5nra 已提交
854
{
855
	if (!obs_source_valid(source, "obs_source_send_mouse_click"))
K
kc5nra 已提交
856 857 858 859
		return;

	if (source->info.output_flags & OBS_SOURCE_INTERACTION) {
		if (source->info.mouse_click) {
J
jp9000 已提交
860 861
			source->info.mouse_click(source->context.data, event,
						 type, mouse_up, click_count);
K
kc5nra 已提交
862 863 864 865
		}
	}
}

866
void obs_source_send_mouse_move(obs_source_t *source,
J
jp9000 已提交
867 868
				const struct obs_mouse_event *event,
				bool mouse_leave)
K
kc5nra 已提交
869
{
870
	if (!obs_source_valid(source, "obs_source_send_mouse_move"))
K
kc5nra 已提交
871 872 873 874
		return;

	if (source->info.output_flags & OBS_SOURCE_INTERACTION) {
		if (source->info.mouse_move) {
J
jp9000 已提交
875 876
			source->info.mouse_move(source->context.data, event,
						mouse_leave);
K
kc5nra 已提交
877 878 879 880
		}
	}
}

881
void obs_source_send_mouse_wheel(obs_source_t *source,
J
jp9000 已提交
882 883
				 const struct obs_mouse_event *event,
				 int x_delta, int y_delta)
K
kc5nra 已提交
884
{
885
	if (!obs_source_valid(source, "obs_source_send_mouse_wheel"))
K
kc5nra 已提交
886 887 888 889
		return;

	if (source->info.output_flags & OBS_SOURCE_INTERACTION) {
		if (source->info.mouse_wheel) {
J
jp9000 已提交
890 891
			source->info.mouse_wheel(source->context.data, event,
						 x_delta, y_delta);
K
kc5nra 已提交
892 893 894 895
		}
	}
}

896
void obs_source_send_focus(obs_source_t *source, bool focus)
K
kc5nra 已提交
897
{
898
	if (!obs_source_valid(source, "obs_source_send_focus"))
K
kc5nra 已提交
899 900 901 902 903 904 905 906 907
		return;

	if (source->info.output_flags & OBS_SOURCE_INTERACTION) {
		if (source->info.focus) {
			source->info.focus(source->context.data, focus);
		}
	}
}

908
void obs_source_send_key_click(obs_source_t *source,
J
jp9000 已提交
909
			       const struct obs_key_event *event, bool key_up)
K
kc5nra 已提交
910
{
911
	if (!obs_source_valid(source, "obs_source_send_key_click"))
K
kc5nra 已提交
912 913 914 915 916
		return;

	if (source->info.output_flags & OBS_SOURCE_INTERACTION) {
		if (source->info.key_click) {
			source->info.key_click(source->context.data, event,
J
jp9000 已提交
917
					       key_up);
K
kc5nra 已提交
918 919 920 921
		}
	}
}

922
static void activate_source(obs_source_t *source)
J
jp9000 已提交
923
{
924
	if (source->context.data && source->info.activate)
925
		source->info.activate(source->context.data);
926
	obs_source_dosignal(source, "source_activate", "activate");
J
jp9000 已提交
927 928
}

929
static void deactivate_source(obs_source_t *source)
J
jp9000 已提交
930
{
931
	if (source->context.data && source->info.deactivate)
932
		source->info.deactivate(source->context.data);
933
	obs_source_dosignal(source, "source_deactivate", "deactivate");
934
}
935

936
static void show_source(obs_source_t *source)
937
{
938
	if (source->context.data && source->info.show)
939
		source->info.show(source->context.data);
940
	obs_source_dosignal(source, "source_show", "show");
941 942
}

943
static void hide_source(obs_source_t *source)
944
{
945
	if (source->context.data && source->info.hide)
946
		source->info.hide(source->context.data);
947
	obs_source_dosignal(source, "source_hide", "hide");
948 949
}

950
static void activate_tree(obs_source_t *parent, obs_source_t *child,
J
jp9000 已提交
951
			  void *param)
952
{
953
	os_atomic_inc_long(&child->activate_refs);
J
jp9000 已提交
954 955 956

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
957 958
}

959
static void deactivate_tree(obs_source_t *parent, obs_source_t *child,
J
jp9000 已提交
960
			    void *param)
961
{
962
	os_atomic_dec_long(&child->activate_refs);
J
jp9000 已提交
963 964 965

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
966 967
}

968
static void show_tree(obs_source_t *parent, obs_source_t *child, void *param)
969
{
970
	os_atomic_inc_long(&child->show_refs);
971 972 973 974 975

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
}

976
static void hide_tree(obs_source_t *parent, obs_source_t *child, void *param)
977
{
978
	os_atomic_dec_long(&child->show_refs);
979 980 981 982 983

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
}

984
void obs_source_activate(obs_source_t *source, enum view_type type)
985
{
986 987
	if (!obs_source_valid(source, "obs_source_activate"))
		return;
988

989 990
	os_atomic_inc_long(&source->show_refs);
	obs_source_enum_active_tree(source, show_tree, NULL);
991 992

	if (type == MAIN_VIEW) {
993 994
		os_atomic_inc_long(&source->activate_refs);
		obs_source_enum_active_tree(source, activate_tree, NULL);
995 996 997
	}
}

998
void obs_source_deactivate(obs_source_t *source, enum view_type type)
999
{
1000 1001
	if (!obs_source_valid(source, "obs_source_deactivate"))
		return;
1002

1003 1004
	if (os_atomic_load_long(&source->show_refs) > 0) {
		os_atomic_dec_long(&source->show_refs);
1005
		obs_source_enum_active_tree(source, hide_tree, NULL);
1006 1007 1008
	}

	if (type == MAIN_VIEW) {
1009 1010
		if (os_atomic_load_long(&source->activate_refs) > 0) {
			os_atomic_dec_long(&source->activate_refs);
1011
			obs_source_enum_active_tree(source, deactivate_tree,
J
jp9000 已提交
1012
						    NULL);
1013
		}
1014
	}
J
jp9000 已提交
1015 1016
}

1017
static inline struct obs_source_frame *get_closest_frame(obs_source_t *source,
J
jp9000 已提交
1018
							 uint64_t sys_time);
1019
bool set_async_texture_size(struct obs_source *source,
J
jp9000 已提交
1020
			    const struct obs_source_frame *frame);
1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031

static void async_tick(obs_source_t *source)
{
	uint64_t sys_time = obs->video.video_time;

	pthread_mutex_lock(&source->async_mutex);

	if (deinterlacing_enabled(source)) {
		deinterlace_process_last_frame(source, sys_time);
	} else {
		if (source->cur_async_frame) {
J
jp9000 已提交
1032
			remove_async_frame(source, source->cur_async_frame);
1033 1034 1035
			source->cur_async_frame = NULL;
		}

J
jp9000 已提交
1036
		source->cur_async_frame = get_closest_frame(source, sys_time);
1037 1038 1039 1040 1041 1042
	}

	source->last_sys_timestamp = sys_time;
	pthread_mutex_unlock(&source->async_mutex);

	if (source->cur_async_frame)
J
jp9000 已提交
1043 1044
		source->async_update_texture =
			set_async_texture_size(source, source->cur_async_frame);
1045
}
1046

1047
void obs_source_video_tick(obs_source_t *source, float seconds)
J
jp9000 已提交
1048
{
1049 1050
	bool now_showing, now_active;

1051 1052
	if (!obs_source_valid(source, "obs_source_video_tick"))
		return;
J
jp9000 已提交
1053

J
jp9000 已提交
1054 1055 1056
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
		obs_transition_tick(source);

1057 1058
	if ((source->info.output_flags & OBS_SOURCE_ASYNC) != 0)
		async_tick(source);
1059

1060 1061 1062
	if (source->defer_update)
		obs_source_deferred_update(source);

J
jp9000 已提交
1063 1064
	/* reset the filter render texture information once every frame */
	if (source->filter_texrender)
1065
		gs_texrender_reset(source->filter_texrender);
J
jp9000 已提交
1066

1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090
	/* call show/hide if the reference changed */
	now_showing = !!source->show_refs;
	if (now_showing != source->showing) {
		if (now_showing) {
			show_source(source);
		} else {
			hide_source(source);
		}

		source->showing = now_showing;
	}

	/* call activate/deactivate if the reference changed */
	now_active = !!source->activate_refs;
	if (now_active != source->active) {
		if (now_active) {
			activate_source(source);
		} else {
			deactivate_source(source);
		}

		source->active = now_active;
	}

1091
	if (source->context.data && source->info.video_tick)
1092
		source->info.video_tick(source->context.data, seconds);
1093 1094

	source->async_rendered = false;
J
jp9000 已提交
1095
	source->deinterlace_rendered = false;
J
jp9000 已提交
1096 1097
}

1098
/* unless the value is 3+ hours worth of frames, this won't overflow */
1099
static inline uint64_t conv_frames_to_time(const size_t sample_rate,
J
jp9000 已提交
1100
					   const size_t frames)
1101
{
1102 1103
	if (!sample_rate)
		return 0;
J
jp9000 已提交
1104

1105
	return (uint64_t)frames * 1000000000ULL / (uint64_t)sample_rate;
1106 1107
}

J
jp9000 已提交
1108
static inline size_t conv_time_to_frames(const size_t sample_rate,
J
jp9000 已提交
1109
					 const uint64_t duration)
J
jp9000 已提交
1110 1111 1112 1113 1114
{
	return (size_t)(duration * (uint64_t)sample_rate / 1000000000ULL);
}

/* maximum buffer size */
J
jp9000 已提交
1115
#define MAX_BUF_SIZE (1000 * AUDIO_OUTPUT_FRAMES * sizeof(float))
1116

J
jp9000 已提交
1117 1118 1119 1120
/* time threshold in nanoseconds to ensure audio timing is as seamless as
 * possible */
#define TS_SMOOTHING_THRESHOLD 70000000ULL

1121
static inline void reset_audio_timing(obs_source_t *source, uint64_t timestamp,
J
jp9000 已提交
1122
				      uint64_t os_time)
1123
{
J
jp9000 已提交
1124
	source->timing_set = true;
1125
	source->timing_adjust = os_time - timestamp;
1126
}
1127

1128 1129 1130 1131 1132
static void reset_audio_data(obs_source_t *source, uint64_t os_time)
{
	for (size_t i = 0; i < MAX_AUDIO_CHANNELS; i++) {
		if (source->audio_input_buf[i].size)
			circlebuf_pop_front(&source->audio_input_buf[i], NULL,
J
jp9000 已提交
1133
					    source->audio_input_buf[i].size);
1134 1135
	}

1136
	source->last_audio_input_buf_size = 0;
1137
	source->audio_ts = os_time;
1138
	source->next_audio_sys_ts_min = os_time;
1139 1140
}

J
jp9000 已提交
1141 1142
static void handle_ts_jump(obs_source_t *source, uint64_t expected, uint64_t ts,
			   uint64_t diff, uint64_t os_time)
1143
{
J
jp9000 已提交
1144 1145 1146 1147
	blog(LOG_DEBUG,
	     "Timestamp for source '%s' jumped by '%" PRIu64 "', "
	     "expected value %" PRIu64 ", input value %" PRIu64,
	     source->context.name, diff, expected, ts);
1148

1149
	pthread_mutex_lock(&source->audio_buf_mutex);
1150
	reset_audio_timing(source, ts, os_time);
1151
	pthread_mutex_unlock(&source->audio_buf_mutex);
1152 1153
}

1154
static void source_signal_audio_data(obs_source_t *source,
J
jp9000 已提交
1155
				     const struct audio_data *in, bool muted)
1156
{
1157
	pthread_mutex_lock(&source->audio_cb_mutex);
1158

1159 1160 1161 1162
	for (size_t i = source->audio_cb_list.num; i > 0; i--) {
		struct audio_cb_info info = source->audio_cb_list.array[i - 1];
		info.callback(info.param, source, in, muted);
	}
1163

1164
	pthread_mutex_unlock(&source->audio_cb_mutex);
1165 1166
}

1167 1168
static inline uint64_t uint64_diff(uint64_t ts1, uint64_t ts2)
{
J
jp9000 已提交
1169
	return (ts1 < ts2) ? (ts2 - ts1) : (ts1 - ts2);
1170 1171
}

1172 1173 1174 1175 1176 1177 1178
static inline size_t get_buf_placement(audio_t *audio, uint64_t offset)
{
	uint32_t sample_rate = audio_output_get_sample_rate(audio);
	return (size_t)(offset * (uint64_t)sample_rate / 1000000000ULL);
}

static void source_output_audio_place(obs_source_t *source,
J
jp9000 已提交
1179
				      const struct audio_data *in)
1180 1181 1182 1183 1184 1185 1186 1187 1188
{
	audio_t *audio = obs->audio.audio;
	size_t buf_placement;
	size_t channels = audio_output_get_channels(audio);
	size_t size = in->frames * sizeof(float);

	if (!source->audio_ts || in->timestamp < source->audio_ts)
		reset_audio_data(source, in->timestamp);

J
jp9000 已提交
1189 1190 1191
	buf_placement =
		get_buf_placement(audio, in->timestamp - source->audio_ts) *
		sizeof(float);
1192 1193

#if DEBUG_AUDIO == 1
J
jp9000 已提交
1194 1195 1196 1197 1198
	blog(LOG_DEBUG,
	     "frames: %lu, size: %lu, placement: %lu, base_ts: %llu, ts: %llu",
	     (unsigned long)in->frames,
	     (unsigned long)source->audio_input_buf[0].size,
	     (unsigned long)buf_placement, source->audio_ts, in->timestamp);
1199 1200
#endif

J
jp9000 已提交
1201 1202 1203 1204 1205
	/* do not allow the circular buffers to become too big */
	if ((buf_placement + size) > MAX_BUF_SIZE)
		return;

	for (size_t i = 0; i < channels; i++) {
1206 1207
		circlebuf_place(&source->audio_input_buf[i], buf_placement,
				in->data[i], size);
J
jp9000 已提交
1208
		circlebuf_pop_back(&source->audio_input_buf[i], NULL,
J
jp9000 已提交
1209 1210
				   source->audio_input_buf[i].size -
					   (buf_placement + size));
J
jp9000 已提交
1211
	}
1212 1213

	source->last_audio_input_buf_size = 0;
1214 1215 1216
}

static inline void source_output_audio_push_back(obs_source_t *source,
J
jp9000 已提交
1217
						 const struct audio_data *in)
1218 1219 1220
{
	audio_t *audio = obs->audio.audio;
	size_t channels = audio_output_get_channels(audio);
J
jp9000 已提交
1221 1222 1223 1224 1225
	size_t size = in->frames * sizeof(float);

	/* do not allow the circular buffers to become too big */
	if ((source->audio_input_buf[0].size + size) > MAX_BUF_SIZE)
		return;
1226 1227

	for (size_t i = 0; i < channels; i++)
J
jp9000 已提交
1228 1229
		circlebuf_push_back(&source->audio_input_buf[i], in->data[i],
				    size);
1230 1231 1232 1233

	/* reset audio input buffer size to ensure that audio doesn't get
	 * perpetually cut */
	source->last_audio_input_buf_size = 0;
1234 1235
}

1236 1237
static inline bool source_muted(obs_source_t *source, uint64_t os_time)
{
J
jp9000 已提交
1238
	if (source->push_to_mute_enabled && source->user_push_to_mute_pressed)
J
jp9000 已提交
1239 1240
		source->push_to_mute_stop_time =
			os_time + source->push_to_mute_delay * 1000000;
1241

J
jp9000 已提交
1242
	if (source->push_to_talk_enabled && source->user_push_to_talk_pressed)
J
jp9000 已提交
1243 1244
		source->push_to_talk_stop_time =
			os_time + source->push_to_talk_delay * 1000000;
1245

J
jp9000 已提交
1246
	bool push_to_mute_active = source->user_push_to_mute_pressed ||
J
jp9000 已提交
1247
				   os_time < source->push_to_mute_stop_time;
J
jp9000 已提交
1248
	bool push_to_talk_active = source->user_push_to_talk_pressed ||
J
jp9000 已提交
1249
				   os_time < source->push_to_talk_stop_time;
1250

J
jp9000 已提交
1251
	return !source->enabled || source->user_muted ||
J
jp9000 已提交
1252 1253
	       (source->push_to_mute_enabled && push_to_mute_active) ||
	       (source->push_to_talk_enabled && !push_to_talk_active);
1254 1255
}

1256
static void source_output_audio_data(obs_source_t *source,
J
jp9000 已提交
1257
				     const struct audio_data *data)
1258
{
1259
	size_t sample_rate = audio_output_get_sample_rate(obs->audio.audio);
1260
	struct audio_data in = *data;
1261
	uint64_t diff;
1262
	uint64_t os_time = os_gettime_ns();
1263
	int64_t sync_offset;
1264 1265
	bool using_direct_ts = false;
	bool push_back = false;
1266

1267 1268 1269 1270 1271
	/* detects 'directly' set timestamps as long as they're within
	 * a certain threshold */
	if (uint64_diff(in.timestamp, os_time) < MAX_TS_VAR) {
		source->timing_adjust = 0;
		source->timing_set = true;
1272 1273
		using_direct_ts = true;
	}
1274

1275
	if (!source->timing_set) {
1276
		reset_audio_timing(source, in.timestamp, os_time);
1277

J
jp9000 已提交
1278
	} else if (source->next_audio_ts_min != 0) {
1279
		diff = uint64_diff(source->next_audio_ts_min, in.timestamp);
1280

1281
		/* smooth audio if within threshold */
1282
		if (diff > MAX_TS_VAR && !using_direct_ts)
1283
			handle_ts_jump(source, source->next_audio_ts_min,
J
jp9000 已提交
1284
				       in.timestamp, diff, os_time);
1285 1286 1287
		else if (diff < TS_SMOOTHING_THRESHOLD) {
			if (source->async_unbuffered && source->async_decoupled)
				source->timing_adjust = os_time - in.timestamp;
1288
			in.timestamp = source->next_audio_ts_min;
1289
		}
1290 1291
	}

J
jp9000 已提交
1292
	source->last_audio_ts = in.timestamp;
J
jp9000 已提交
1293 1294
	source->next_audio_ts_min =
		in.timestamp + conv_frames_to_time(sample_rate, in.frames);
1295

1296
	in.timestamp += source->timing_adjust;
1297

1298 1299
	pthread_mutex_lock(&source->audio_buf_mutex);

J
jp9000 已提交
1300
	if (source->next_audio_sys_ts_min == in.timestamp) {
1301
		push_back = true;
1302 1303

	} else if (source->next_audio_sys_ts_min) {
J
jp9000 已提交
1304
		diff = uint64_diff(source->next_audio_sys_ts_min, in.timestamp);
1305 1306

		if (diff < TS_SMOOTHING_THRESHOLD) {
1307 1308
			push_back = true;

J
jp9000 已提交
1309
			/* This typically only happens if used with async video when
1310 1311 1312 1313 1314
		 * audio/video start transitioning in to a timestamp jump.
		 * Audio will typically have a timestamp jump, and then video
		 * will have a timestamp jump.  If that case is encountered,
		 * just clear the audio data in that small window and force a
		 * resync.  This handles all cases rather than just looping. */
1315
		} else if (diff > MAX_TS_VAR) {
J
jp9000 已提交
1316
			reset_audio_timing(source, data->timestamp, os_time);
1317 1318
			in.timestamp = data->timestamp + source->timing_adjust;
		}
J
jp9000 已提交
1319 1320
	}

1321 1322
	sync_offset = source->sync_offset;
	in.timestamp += sync_offset;
1323 1324
	in.timestamp -= source->resample_offset;

J
jp9000 已提交
1325 1326
	source->next_audio_sys_ts_min =
		source->next_audio_ts_min + source->timing_adjust;
1327 1328 1329 1330 1331 1332

	if (source->last_sync_offset != sync_offset) {
		if (source->last_sync_offset)
			push_back = false;
		source->last_sync_offset = sync_offset;
	}
1333

J
jp9000 已提交
1334 1335 1336 1337 1338 1339
	if (source->monitoring_type != OBS_MONITORING_TYPE_MONITOR_ONLY) {
		if (push_back && source->audio_ts)
			source_output_audio_push_back(source, &in);
		else
			source_output_audio_place(source, &in);
	}
1340 1341 1342

	pthread_mutex_unlock(&source->audio_buf_mutex);

1343
	source_signal_audio_data(source, data, source_muted(source, os_time));
1344 1345
}

1346 1347 1348 1349
enum convert_type {
	CONVERT_NONE,
	CONVERT_NV12,
	CONVERT_420,
J
jpark37 已提交
1350
	CONVERT_420_A,
1351
	CONVERT_422,
J
jpark37 已提交
1352 1353
	CONVERT_422_A,
	CONVERT_422_PACK,
1354
	CONVERT_422_Y,
J
James Park 已提交
1355
	CONVERT_444,
J
jpark37 已提交
1356 1357
	CONVERT_444_A,
	CONVERT_444_A_PACK,
1358 1359
	CONVERT_800,
	CONVERT_RGB_LIMITED,
1360
	CONVERT_BGR3,
1361 1362
};

1363
static inline enum convert_type get_convert_type(enum video_format format,
J
jp9000 已提交
1364
						 bool full_range)
1365
{
1366
	switch (format) {
1367 1368 1369 1370
	case VIDEO_FORMAT_I420:
		return CONVERT_420;
	case VIDEO_FORMAT_NV12:
		return CONVERT_NV12;
J
James Park 已提交
1371 1372
	case VIDEO_FORMAT_I444:
		return CONVERT_444;
1373 1374
	case VIDEO_FORMAT_I422:
		return CONVERT_422;
1375 1376 1377 1378

	case VIDEO_FORMAT_YVYU:
	case VIDEO_FORMAT_YUY2:
	case VIDEO_FORMAT_UYVY:
J
jpark37 已提交
1379
		return CONVERT_422_PACK;
1380

J
jpk 已提交
1381
	case VIDEO_FORMAT_Y800:
1382 1383
		return CONVERT_800;

1384
	case VIDEO_FORMAT_NONE:
1385 1386 1387
	case VIDEO_FORMAT_RGBA:
	case VIDEO_FORMAT_BGRA:
	case VIDEO_FORMAT_BGRX:
1388
		return full_range ? CONVERT_NONE : CONVERT_RGB_LIMITED;
1389 1390 1391

	case VIDEO_FORMAT_BGR3:
		return CONVERT_BGR3;
J
jpark37 已提交
1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403

	case VIDEO_FORMAT_I40A:
		return CONVERT_420_A;

	case VIDEO_FORMAT_I42A:
		return CONVERT_422_A;

	case VIDEO_FORMAT_YUVA:
		return CONVERT_444_A;

	case VIDEO_FORMAT_AYUV:
		return CONVERT_444_A_PACK;
1404 1405 1406 1407 1408
	}

	return CONVERT_NONE;
}

1409
static inline bool set_packed422_sizes(struct obs_source *source,
J
jp9000 已提交
1410
				       const struct obs_source_frame *frame)
1411
{
1412 1413 1414 1415
	source->async_convert_width[0] = frame->width / 2;
	source->async_convert_height[0] = frame->height;
	source->async_texture_formats[0] = GS_BGRA;
	source->async_channel_count = 1;
1416 1417 1418
	return true;
}

J
jpark37 已提交
1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429
static inline bool
set_packed444_alpha_sizes(struct obs_source *source,
			  const struct obs_source_frame *frame)
{
	source->async_convert_width[0] = frame->width;
	source->async_convert_height[0] = frame->height;
	source->async_texture_formats[0] = GS_BGRA;
	source->async_channel_count = 1;
	return true;
}

J
James Park 已提交
1430
static inline bool set_planar444_sizes(struct obs_source *source,
J
jp9000 已提交
1431
				       const struct obs_source_frame *frame)
J
James Park 已提交
1432
{
1433 1434 1435 1436 1437 1438 1439 1440 1441 1442
	source->async_convert_width[0] = frame->width;
	source->async_convert_width[1] = frame->width;
	source->async_convert_width[2] = frame->width;
	source->async_convert_height[0] = frame->height;
	source->async_convert_height[1] = frame->height;
	source->async_convert_height[2] = frame->height;
	source->async_texture_formats[0] = GS_R8;
	source->async_texture_formats[1] = GS_R8;
	source->async_texture_formats[2] = GS_R8;
	source->async_channel_count = 3;
J
James Park 已提交
1443 1444 1445
	return true;
}

J
jpark37 已提交
1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465
static inline bool
set_planar444_alpha_sizes(struct obs_source *source,
			  const struct obs_source_frame *frame)
{
	source->async_convert_width[0] = frame->width;
	source->async_convert_width[1] = frame->width;
	source->async_convert_width[2] = frame->width;
	source->async_convert_width[3] = frame->width;
	source->async_convert_height[0] = frame->height;
	source->async_convert_height[1] = frame->height;
	source->async_convert_height[2] = frame->height;
	source->async_convert_height[3] = frame->height;
	source->async_texture_formats[0] = GS_R8;
	source->async_texture_formats[1] = GS_R8;
	source->async_texture_formats[2] = GS_R8;
	source->async_texture_formats[3] = GS_R8;
	source->async_channel_count = 4;
	return true;
}

1466
static inline bool set_planar420_sizes(struct obs_source *source,
J
jp9000 已提交
1467
				       const struct obs_source_frame *frame)
1468
{
1469 1470 1471 1472 1473 1474 1475 1476 1477 1478
	source->async_convert_width[0] = frame->width;
	source->async_convert_width[1] = frame->width / 2;
	source->async_convert_width[2] = frame->width / 2;
	source->async_convert_height[0] = frame->height;
	source->async_convert_height[1] = frame->height / 2;
	source->async_convert_height[2] = frame->height / 2;
	source->async_texture_formats[0] = GS_R8;
	source->async_texture_formats[1] = GS_R8;
	source->async_texture_formats[2] = GS_R8;
	source->async_channel_count = 3;
1479 1480 1481
	return true;
}

J
jpark37 已提交
1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501
static inline bool
set_planar420_alpha_sizes(struct obs_source *source,
			  const struct obs_source_frame *frame)
{
	source->async_convert_width[0] = frame->width;
	source->async_convert_width[1] = frame->width / 2;
	source->async_convert_width[2] = frame->width / 2;
	source->async_convert_width[3] = frame->width;
	source->async_convert_height[0] = frame->height;
	source->async_convert_height[1] = frame->height / 2;
	source->async_convert_height[2] = frame->height / 2;
	source->async_convert_height[3] = frame->height;
	source->async_texture_formats[0] = GS_R8;
	source->async_texture_formats[1] = GS_R8;
	source->async_texture_formats[2] = GS_R8;
	source->async_texture_formats[3] = GS_R8;
	source->async_channel_count = 4;
	return true;
}

1502 1503 1504
static inline bool set_planar422_sizes(struct obs_source *source,
				       const struct obs_source_frame *frame)
{
1505 1506 1507 1508 1509 1510 1511 1512 1513 1514
	source->async_convert_width[0] = frame->width;
	source->async_convert_width[1] = frame->width / 2;
	source->async_convert_width[2] = frame->width / 2;
	source->async_convert_height[0] = frame->height;
	source->async_convert_height[1] = frame->height;
	source->async_convert_height[2] = frame->height;
	source->async_texture_formats[0] = GS_R8;
	source->async_texture_formats[1] = GS_R8;
	source->async_texture_formats[2] = GS_R8;
	source->async_channel_count = 3;
1515 1516 1517
	return true;
}

J
jpark37 已提交
1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537
static inline bool
set_planar422_alpha_sizes(struct obs_source *source,
			  const struct obs_source_frame *frame)
{
	source->async_convert_width[0] = frame->width;
	source->async_convert_width[1] = frame->width / 2;
	source->async_convert_width[2] = frame->width / 2;
	source->async_convert_width[3] = frame->width;
	source->async_convert_height[0] = frame->height;
	source->async_convert_height[1] = frame->height;
	source->async_convert_height[2] = frame->height;
	source->async_convert_height[3] = frame->height;
	source->async_texture_formats[0] = GS_R8;
	source->async_texture_formats[1] = GS_R8;
	source->async_texture_formats[2] = GS_R8;
	source->async_texture_formats[3] = GS_R8;
	source->async_channel_count = 4;
	return true;
}

J
jp9000 已提交
1538
static inline bool set_nv12_sizes(struct obs_source *source,
J
jp9000 已提交
1539
				  const struct obs_source_frame *frame)
J
jp9000 已提交
1540
{
1541 1542 1543 1544 1545 1546 1547
	source->async_convert_width[0] = frame->width;
	source->async_convert_width[1] = frame->width / 2;
	source->async_convert_height[0] = frame->height;
	source->async_convert_height[1] = frame->height / 2;
	source->async_texture_formats[0] = GS_R8;
	source->async_texture_formats[1] = GS_R8G8;
	source->async_channel_count = 2;
J
jp9000 已提交
1548 1549 1550
	return true;
}

1551
static inline bool set_y800_sizes(struct obs_source *source,
J
jp9000 已提交
1552
				  const struct obs_source_frame *frame)
1553
{
1554 1555 1556 1557
	source->async_convert_width[0] = frame->width;
	source->async_convert_height[0] = frame->height;
	source->async_texture_formats[0] = GS_R8;
	source->async_channel_count = 1;
1558 1559 1560 1561
	return true;
}

static inline bool set_rgb_limited_sizes(struct obs_source *source,
J
jp9000 已提交
1562
					 const struct obs_source_frame *frame)
1563
{
1564 1565 1566 1567
	source->async_convert_width[0] = frame->width;
	source->async_convert_height[0] = frame->height;
	source->async_texture_formats[0] = convert_video_format(frame->format);
	source->async_channel_count = 1;
1568 1569 1570
	return true;
}

1571
static inline bool set_bgr3_sizes(struct obs_source *source,
J
jp9000 已提交
1572
				  const struct obs_source_frame *frame)
1573
{
1574 1575 1576 1577
	source->async_convert_width[0] = frame->width * 3;
	source->async_convert_height[0] = frame->height;
	source->async_texture_formats[0] = GS_R8;
	source->async_channel_count = 1;
1578 1579 1580
	return true;
}

1581
static inline bool init_gpu_conversion(struct obs_source *source,
J
jp9000 已提交
1582
				       const struct obs_source_frame *frame)
1583
{
1584
	switch (get_convert_type(frame->format, frame->full_range)) {
J
jpark37 已提交
1585
	case CONVERT_422_PACK:
J
jp9000 已提交
1586
		return set_packed422_sizes(source, frame);
1587

J
jp9000 已提交
1588 1589
	case CONVERT_420:
		return set_planar420_sizes(source, frame);
J
James Park 已提交
1590

1591 1592 1593
	case CONVERT_422:
		return set_planar422_sizes(source, frame);

J
jp9000 已提交
1594 1595
	case CONVERT_NV12:
		return set_nv12_sizes(source, frame);
1596

J
jp9000 已提交
1597 1598
	case CONVERT_444:
		return set_planar444_sizes(source, frame);
1599

J
jp9000 已提交
1600 1601
	case CONVERT_800:
		return set_y800_sizes(source, frame);
1602

J
jp9000 已提交
1603 1604
	case CONVERT_RGB_LIMITED:
		return set_rgb_limited_sizes(source, frame);
1605

J
jp9000 已提交
1606 1607
	case CONVERT_BGR3:
		return set_bgr3_sizes(source, frame);
1608

J
jpark37 已提交
1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620
	case CONVERT_420_A:
		return set_planar420_alpha_sizes(source, frame);

	case CONVERT_422_A:
		return set_planar422_alpha_sizes(source, frame);

	case CONVERT_444_A:
		return set_planar444_alpha_sizes(source, frame);

	case CONVERT_444_A_PACK:
		return set_packed444_alpha_sizes(source, frame);

J
jp9000 已提交
1621 1622 1623
	case CONVERT_NONE:
		assert(false && "No conversion requested");
		break;
1624 1625 1626 1627
	}
	return false;
}

1628
bool set_async_texture_size(struct obs_source *source,
J
jp9000 已提交
1629
			    const struct obs_source_frame *frame)
1630
{
J
jp9000 已提交
1631 1632
	enum convert_type cur =
		get_convert_type(frame->format, frame->full_range);
1633

J
jp9000 已提交
1634 1635 1636
	if (source->async_width == frame->width &&
	    source->async_height == frame->height &&
	    source->async_format == frame->format &&
1637
	    source->async_full_range == frame->full_range)
1638 1639
		return true;

J
jp9000 已提交
1640 1641 1642
	source->async_width = frame->width;
	source->async_height = frame->height;
	source->async_format = frame->format;
1643
	source->async_full_range = frame->full_range;
1644

1645 1646
	gs_enter_context(obs->video.graphics);

1647 1648 1649 1650 1651 1652 1653
	for (size_t c = 0; c < MAX_AV_PLANES; c++) {
		gs_texture_destroy(source->async_textures[c]);
		source->async_textures[c] = NULL;
		gs_texture_destroy(source->async_prev_textures[c]);
		source->async_prev_textures[c] = NULL;
	}

1654
	gs_texrender_destroy(source->async_texrender);
J
jp9000 已提交
1655
	gs_texrender_destroy(source->async_prev_texrender);
1656
	source->async_texrender = NULL;
J
jp9000 已提交
1657
	source->async_prev_texrender = NULL;
1658

1659 1660 1661 1662 1663
	const enum gs_color_format format = convert_video_format(frame->format);
	const bool async_gpu_conversion = (cur != CONVERT_NONE) &&
					  init_gpu_conversion(source, frame);
	source->async_gpu_conversion = async_gpu_conversion;
	if (async_gpu_conversion) {
1664
		source->async_texrender =
1665
			gs_texrender_create(format, GS_ZS_NONE);
1666

1667 1668 1669 1670 1671 1672
		for (int c = 0; c < source->async_channel_count; ++c)
			source->async_textures[c] = gs_texture_create(
				source->async_convert_width[c],
				source->async_convert_height[c],
				source->async_texture_formats[c], 1, NULL,
				GS_DYNAMIC);
1673
	} else {
1674 1675 1676
		source->async_textures[0] =
			gs_texture_create(frame->width, frame->height, format,
					  1, NULL, GS_DYNAMIC);
1677 1678
	}

J
jp9000 已提交
1679 1680 1681
	if (deinterlacing_enabled(source))
		set_deinterlace_texture_size(source);

1682 1683
	gs_leave_context();

1684
	return source->async_textures[0] != NULL;
1685 1686
}

1687
static void upload_raw_frame(gs_texture_t *tex[MAX_AV_PLANES],
J
jp9000 已提交
1688
			     const struct obs_source_frame *frame)
1689
{
1690
	switch (get_convert_type(frame->format, frame->full_range)) {
J
jpark37 已提交
1691
	case CONVERT_422_PACK:
J
jp9000 已提交
1692 1693 1694 1695
	case CONVERT_800:
	case CONVERT_RGB_LIMITED:
	case CONVERT_BGR3:
	case CONVERT_420:
1696
	case CONVERT_422:
J
jp9000 已提交
1697 1698
	case CONVERT_NV12:
	case CONVERT_444:
J
jpark37 已提交
1699 1700 1701 1702
	case CONVERT_420_A:
	case CONVERT_422_A:
	case CONVERT_444_A:
	case CONVERT_444_A_PACK:
1703 1704 1705 1706 1707
		for (size_t c = 0; c < MAX_AV_PLANES; c++) {
			if (tex[c])
				gs_texture_set_image(tex[c], frame->data[c],
						     frame->linesize[c], false);
		}
J
jp9000 已提交
1708
		break;
J
James Park 已提交
1709

J
jp9000 已提交
1710 1711 1712
	case CONVERT_NONE:
		assert(false && "No conversion requested");
		break;
1713 1714 1715
	}
}

1716
static const char *select_conversion_technique(enum video_format format,
J
jp9000 已提交
1717
					       bool full_range)
1718 1719
{
	switch (format) {
J
jp9000 已提交
1720 1721
	case VIDEO_FORMAT_UYVY:
		return "UYVY_Reverse";
1722

J
jp9000 已提交
1723 1724
	case VIDEO_FORMAT_YUY2:
		return "YUY2_Reverse";
1725

J
jp9000 已提交
1726 1727
	case VIDEO_FORMAT_YVYU:
		return "YVYU_Reverse";
1728

J
jp9000 已提交
1729 1730
	case VIDEO_FORMAT_I420:
		return "I420_Reverse";
1731

J
jp9000 已提交
1732 1733
	case VIDEO_FORMAT_NV12:
		return "NV12_Reverse";
J
James Park 已提交
1734

J
jp9000 已提交
1735 1736
	case VIDEO_FORMAT_I444:
		return "I444_Reverse";
1737

J
jp9000 已提交
1738 1739
	case VIDEO_FORMAT_Y800:
		return full_range ? "Y800_Full" : "Y800_Limited";
1740

J
jp9000 已提交
1741 1742
	case VIDEO_FORMAT_BGR3:
		return full_range ? "BGR3_Full" : "BGR3_Limited";
1743

1744 1745 1746
	case VIDEO_FORMAT_I422:
		return "I422_Reverse";

J
jpark37 已提交
1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758
	case VIDEO_FORMAT_I40A:
		return "I40A_Reverse";

	case VIDEO_FORMAT_I42A:
		return "I42A_Reverse";

	case VIDEO_FORMAT_YUVA:
		return "YUVA_Reverse";

	case VIDEO_FORMAT_AYUV:
		return "AYUV_Reverse";

J
jp9000 已提交
1759 1760 1761 1762 1763 1764 1765 1766 1767
	case VIDEO_FORMAT_BGRA:
	case VIDEO_FORMAT_BGRX:
	case VIDEO_FORMAT_RGBA:
	case VIDEO_FORMAT_NONE:
		if (full_range)
			assert(false && "No conversion requested");
		else
			return "RGB_Limited";
		break;
1768 1769 1770 1771
	}
	return NULL;
}

1772
static inline void set_eparam(gs_effect_t *effect, const char *name, float val)
1773
{
1774
	gs_eparam_t *param = gs_effect_get_param_by_name(effect, name);
1775
	gs_effect_set_float(param, val);
1776 1777
}

1778 1779 1780 1781 1782 1783
static inline void set_eparami(gs_effect_t *effect, const char *name, int val)
{
	gs_eparam_t *param = gs_effect_get_param_by_name(effect, name);
	gs_effect_set_int(param, val);
}

1784
static bool update_async_texrender(struct obs_source *source,
J
jp9000 已提交
1785
				   const struct obs_source_frame *frame,
1786 1787
				   gs_texture_t *tex[MAX_AV_PLANES],
				   gs_texrender_t *texrender)
1788
{
1789 1790
	GS_DEBUG_MARKER_BEGIN(GS_DEBUG_COLOR_CONVERT_FORMAT, "Convert Format");

1791
	gs_texrender_reset(texrender);
1792 1793 1794 1795 1796 1797

	upload_raw_frame(tex, frame);

	uint32_t cx = source->async_width;
	uint32_t cy = source->async_height;

1798
	gs_effect_t *conv = obs->video.conversion_effect;
J
jp9000 已提交
1799 1800
	const char *tech_name =
		select_conversion_technique(frame->format, frame->full_range);
1801
	gs_technique_t *tech = gs_effect_get_technique(conv, tech_name);
1802

1803
	const bool success = gs_texrender_begin(texrender, cx, cy);
1804

1805 1806
	if (success) {
		gs_enable_blending(false);
1807

1808 1809
		gs_technique_begin(tech);
		gs_technique_begin_pass(tech, 0);
1810

1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822
		if (tex[0])
			gs_effect_set_texture(
				gs_effect_get_param_by_name(conv, "image"),
				tex[0]);
		if (tex[1])
			gs_effect_set_texture(
				gs_effect_get_param_by_name(conv, "image1"),
				tex[1]);
		if (tex[2])
			gs_effect_set_texture(
				gs_effect_get_param_by_name(conv, "image2"),
				tex[2]);
J
jpark37 已提交
1823 1824 1825 1826
		if (tex[3])
			gs_effect_set_texture(
				gs_effect_get_param_by_name(conv, "image3"),
				tex[3]);
1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855
		set_eparam(conv, "width", (float)cx);
		set_eparam(conv, "height", (float)cy);
		set_eparam(conv, "width_d2", (float)cx * 0.5f);
		set_eparam(conv, "height_d2", (float)cy * 0.5f);
		set_eparam(conv, "width_x2_i", 0.5f / (float)cx);

		struct vec4 vec0, vec1, vec2;
		vec4_set(&vec0, frame->color_matrix[0], frame->color_matrix[1],
			 frame->color_matrix[2], frame->color_matrix[3]);
		vec4_set(&vec1, frame->color_matrix[4], frame->color_matrix[5],
			 frame->color_matrix[6], frame->color_matrix[7]);
		vec4_set(&vec2, frame->color_matrix[8], frame->color_matrix[9],
			 frame->color_matrix[10], frame->color_matrix[11]);
		gs_effect_set_vec4(
			gs_effect_get_param_by_name(conv, "color_vec0"), &vec0);
		gs_effect_set_vec4(
			gs_effect_get_param_by_name(conv, "color_vec1"), &vec1);
		gs_effect_set_vec4(
			gs_effect_get_param_by_name(conv, "color_vec2"), &vec2);
		if (!frame->full_range) {
			gs_eparam_t *min_param = gs_effect_get_param_by_name(
				conv, "color_range_min");
			gs_effect_set_val(min_param, frame->color_range_min,
					  sizeof(float) * 3);
			gs_eparam_t *max_param = gs_effect_get_param_by_name(
				conv, "color_range_max");
			gs_effect_set_val(max_param, frame->color_range_max,
					  sizeof(float) * 3);
		}
J
James Park 已提交
1856

1857
		gs_draw(GS_TRIS, 0, 3);
1858

1859 1860
		gs_technique_end_pass(tech);
		gs_technique_end(tech);
1861

1862
		gs_enable_blending(true);
1863

1864 1865
		gs_texrender_end(texrender);
	}
1866

1867
	GS_DEBUG_MARKER_END();
1868
	return success;
1869 1870
}

1871
bool update_async_texture(struct obs_source *source,
J
jp9000 已提交
1872 1873
			  const struct obs_source_frame *frame,
			  gs_texture_t *tex, gs_texrender_t *texrender)
1874 1875 1876 1877 1878 1879 1880 1881 1882 1883
{
	gs_texture_t *tex3[MAX_AV_PLANES] = {tex,  NULL, NULL, NULL,
					     NULL, NULL, NULL, NULL};
	return update_async_textures(source, frame, tex3, texrender);
}

bool update_async_textures(struct obs_source *source,
			   const struct obs_source_frame *frame,
			   gs_texture_t *tex[MAX_AV_PLANES],
			   gs_texrender_t *texrender)
1884
{
1885
	enum convert_type type;
1886

J
jp9000 已提交
1887
	source->async_flip = frame->flip;
1888

1889
	if (source->async_gpu_conversion && texrender)
1890
		return update_async_texrender(source, frame, tex, texrender);
1891

1892
	type = get_convert_type(frame->format, frame->full_range);
1893
	if (type == CONVERT_NONE) {
1894
		gs_texture_set_image(tex[0], frame->data[0], frame->linesize[0],
J
jp9000 已提交
1895
				     false);
1896 1897 1898
		return true;
	}

1899
	return false;
1900 1901
}

1902
static inline void obs_source_draw_texture(struct obs_source *source,
J
jp9000 已提交
1903
					   gs_effect_t *effect)
1904
{
1905
	gs_texture_t *tex = source->async_textures[0];
J
jp9000 已提交
1906
	gs_eparam_t *param;
1907

1908 1909
	if (source->async_texrender)
		tex = gs_texrender_get_texture(source->async_texrender);
1910

1911 1912
	param = gs_effect_get_param_by_name(effect, "image");
	gs_effect_set_texture(param, tex);
1913

1914 1915
	gs_draw_sprite(tex, source->async_flip ? GS_FLIP_V : 0, 0, 0);
}
1916

1917 1918
static void obs_source_draw_async_texture(struct obs_source *source)
{
J
jp9000 已提交
1919 1920 1921
	gs_effect_t *effect = gs_get_effect();
	bool def_draw = (!effect);
	gs_technique_t *tech = NULL;
1922 1923

	if (def_draw) {
1924
		effect = obs_get_base_effect(OBS_EFFECT_DEFAULT);
J
James Park 已提交
1925
		tech = gs_effect_get_technique(effect, "Draw");
1926 1927
		gs_technique_begin(tech);
		gs_technique_begin_pass(tech, 0);
1928 1929
	}

J
James Park 已提交
1930
	obs_source_draw_texture(source, effect);
1931 1932

	if (def_draw) {
1933 1934
		gs_technique_end_pass(tech);
		gs_technique_end(tech);
1935
	}
1936 1937
}

1938
static void obs_source_update_async_video(obs_source_t *source)
1939
{
1940 1941 1942
	if (!source->async_rendered) {
		struct obs_source_frame *frame = obs_source_get_frame(source);

1943 1944 1945
		if (frame)
			frame = filter_async_video(source, frame);

1946 1947
		source->async_rendered = true;
		if (frame) {
1948 1949
			if (!source->async_decoupled ||
			    !source->async_unbuffered) {
J
jp9000 已提交
1950 1951
				source->timing_adjust = obs->video.video_time -
							frame->timestamp;
1952 1953
				source->timing_set = true;
			}
1954

1955
			if (source->async_update_texture) {
1956 1957 1958
				update_async_textures(source, frame,
						      source->async_textures,
						      source->async_texrender);
1959
				source->async_update_texture = false;
1960
			}
1961

1962 1963
			obs_source_release_frame(source, frame);
		}
1964
	}
1965
}
1966

1967 1968
static inline void obs_source_render_async_video(obs_source_t *source)
{
1969
	if (source->async_textures[0] && source->async_active)
1970
		obs_source_draw_async_texture(source);
1971 1972
}

1973
static inline void obs_source_render_filters(obs_source_t *source)
1974
{
1975 1976 1977 1978 1979 1980 1981
	obs_source_t *first_filter;

	pthread_mutex_lock(&source->filter_mutex);
	first_filter = source->filters.array[0];
	obs_source_addref(first_filter);
	pthread_mutex_unlock(&source->filter_mutex);

1982
	source->rendering_filter = true;
1983
	obs_source_video_render(first_filter);
1984
	source->rendering_filter = false;
1985 1986

	obs_source_release(first_filter);
1987 1988
}

1989
void obs_source_default_render(obs_source_t *source)
1990
{
J
jp9000 已提交
1991 1992 1993
	gs_effect_t *effect = obs->video.default_effect;
	gs_technique_t *tech = gs_effect_get_technique(effect, "Draw");
	size_t passes, i;
1994

1995
	passes = gs_technique_begin(tech);
1996
	for (i = 0; i < passes; i++) {
1997
		gs_technique_begin_pass(tech, i);
1998 1999
		if (source->context.data)
			source->info.video_render(source->context.data, effect);
2000
		gs_technique_end_pass(tech);
2001
	}
2002
	gs_technique_end(tech);
2003 2004
}

2005
static inline void obs_source_main_render(obs_source_t *source)
2006
{
J
jp9000 已提交
2007 2008
	uint32_t flags = source->info.output_flags;
	bool custom_draw = (flags & OBS_SOURCE_CUSTOM_DRAW) != 0;
2009
	bool default_effect = !source->filter_parent &&
J
jp9000 已提交
2010
			      source->filters.num == 0 && !custom_draw;
2011 2012

	if (default_effect)
2013
		obs_source_default_render(source);
2014
	else if (source->context.data)
2015
		source->info.video_render(source->context.data,
J
jp9000 已提交
2016
					  custom_draw ? NULL : gs_get_effect());
2017 2018
}

2019 2020
static bool ready_async_frame(obs_source_t *source, uint64_t sys_time);

2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037 2038
#if GS_USE_DEBUG_MARKERS
static const char *get_type_format(enum obs_source_type type)
{
	switch (type) {
	case OBS_SOURCE_TYPE_INPUT:
		return "Input: %s";
	case OBS_SOURCE_TYPE_FILTER:
		return "Filter: %s";
	case OBS_SOURCE_TYPE_TRANSITION:
		return "Transition: %s";
	case OBS_SOURCE_TYPE_SCENE:
		return "Scene: %s";
	default:
		return "[Unknown]: %s";
	}
}
#endif

2039
static inline void render_video(obs_source_t *source)
J
jp9000 已提交
2040
{
2041
	if (source->info.type != OBS_SOURCE_TYPE_FILTER &&
2042 2043 2044
	    (source->info.output_flags & OBS_SOURCE_VIDEO) == 0) {
		if (source->filter_parent)
			obs_source_skip_video_filter(source);
2045
		return;
2046
	}
2047

2048 2049
	if (source->info.type == OBS_SOURCE_TYPE_INPUT &&
	    (source->info.output_flags & OBS_SOURCE_ASYNC) != 0 &&
J
jp9000 已提交
2050 2051 2052
	    !source->rendering_filter) {
		if (deinterlacing_enabled(source))
			deinterlace_update_async_video(source);
2053
		obs_source_update_async_video(source);
J
jp9000 已提交
2054
	}
2055

2056
	if (!source->context.data || !source->enabled) {
2057 2058 2059 2060 2061
		if (source->filter_parent)
			obs_source_skip_video_filter(source);
		return;
	}

2062
	GS_DEBUG_MARKER_BEGIN_FORMAT(GS_DEBUG_COLOR_SOURCE,
J
jp9000 已提交
2063 2064
				     get_type_format(source->info.type),
				     obs_source_get_name(source));
2065

2066 2067
	if (source->filters.num && !source->rendering_filter)
		obs_source_render_filters(source);
2068

2069 2070 2071 2072
	else if (source->info.video_render)
		obs_source_main_render(source);

	else if (source->filter_target)
2073 2074
		obs_source_video_render(source->filter_target);

J
jp9000 已提交
2075 2076 2077
	else if (deinterlacing_enabled(source))
		deinterlace_render(source);

2078
	else
2079
		obs_source_render_async_video(source);
2080 2081

	GS_DEBUG_MARKER_END();
J
jp9000 已提交
2082 2083
}

2084 2085 2086 2087 2088 2089 2090 2091 2092 2093
void obs_source_video_render(obs_source_t *source)
{
	if (!obs_source_valid(source, "obs_source_video_render"))
		return;

	obs_source_addref(source);
	render_video(source);
	obs_source_release(source);
}

2094
static uint32_t get_base_width(const obs_source_t *source)
J
jp9000 已提交
2095
{
2096
	bool is_filter = !!source->filter_parent;
2097
	bool func_valid = source->context.data && source->info.get_width;
2098

J
jp9000 已提交
2099 2100 2101
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION) {
		return source->enabled ? source->transition_actual_cx : 0;

2102
	} else if (func_valid && (!is_filter || source->enabled)) {
2103
		return source->info.get_width(source->context.data);
2104

2105
	} else if (is_filter) {
2106 2107 2108
		return get_base_width(source->filter_target);
	}

2109
	return source->async_active ? source->async_width : 0;
J
jp9000 已提交
2110 2111
}

2112
static uint32_t get_base_height(const obs_source_t *source)
J
jp9000 已提交
2113
{
2114
	bool is_filter = !!source->filter_parent;
2115
	bool func_valid = source->context.data && source->info.get_height;
2116

J
jp9000 已提交
2117 2118 2119
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION) {
		return source->enabled ? source->transition_actual_cy : 0;

2120
	} else if (func_valid && (!is_filter || source->enabled)) {
2121
		return source->info.get_height(source->context.data);
2122

2123
	} else if (is_filter) {
2124 2125 2126
		return get_base_height(source->filter_target);
	}

2127
	return source->async_active ? source->async_height : 0;
J
jp9000 已提交
2128 2129
}

2130 2131 2132 2133 2134 2135
static uint32_t get_recurse_width(obs_source_t *source)
{
	uint32_t width;

	pthread_mutex_lock(&source->filter_mutex);

J
jp9000 已提交
2136 2137
	width = (source->filters.num) ? get_base_width(source->filters.array[0])
				      : get_base_width(source);
2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149

	pthread_mutex_unlock(&source->filter_mutex);

	return width;
}

static uint32_t get_recurse_height(obs_source_t *source)
{
	uint32_t height;

	pthread_mutex_lock(&source->filter_mutex);

J
jp9000 已提交
2150 2151 2152
	height = (source->filters.num)
			 ? get_base_height(source->filters.array[0])
			 : get_base_height(source);
2153 2154 2155 2156 2157 2158 2159 2160

	pthread_mutex_unlock(&source->filter_mutex);

	return height;
}

uint32_t obs_source_get_width(obs_source_t *source)
{
J
jp9000 已提交
2161 2162
	if (!data_valid(source, "obs_source_get_width"))
		return 0;
2163

J
jp9000 已提交
2164 2165 2166
	return (source->info.type != OBS_SOURCE_TYPE_FILTER)
		       ? get_recurse_width(source)
		       : get_base_width(source);
2167 2168 2169 2170
}

uint32_t obs_source_get_height(obs_source_t *source)
{
J
jp9000 已提交
2171 2172
	if (!data_valid(source, "obs_source_get_height"))
		return 0;
2173

J
jp9000 已提交
2174 2175 2176
	return (source->info.type != OBS_SOURCE_TYPE_FILTER)
		       ? get_recurse_height(source)
		       : get_base_height(source);
2177 2178
}

2179 2180
uint32_t obs_source_get_base_width(obs_source_t *source)
{
J
jp9000 已提交
2181 2182
	if (!data_valid(source, "obs_source_get_base_width"))
		return 0;
2183 2184 2185 2186 2187 2188

	return get_base_width(source);
}

uint32_t obs_source_get_base_height(obs_source_t *source)
{
J
jp9000 已提交
2189 2190
	if (!data_valid(source, "obs_source_get_base_height"))
		return 0;
2191 2192 2193 2194

	return get_base_height(source);
}

2195
obs_source_t *obs_filter_get_parent(const obs_source_t *filter)
2196
{
J
jp9000 已提交
2197 2198 2199
	return obs_ptr_valid(filter, "obs_filter_get_parent")
		       ? filter->filter_parent
		       : NULL;
2200 2201
}

2202
obs_source_t *obs_filter_get_target(const obs_source_t *filter)
J
jp9000 已提交
2203
{
J
jp9000 已提交
2204 2205 2206
	return obs_ptr_valid(filter, "obs_filter_get_target")
		       ? filter->filter_target
		       : NULL;
J
jp9000 已提交
2207 2208
}

2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220
static bool filter_compatible(obs_source_t *source, obs_source_t *filter)
{
	uint32_t s_caps = source->info.output_flags;
	uint32_t f_caps = filter->info.output_flags;

	if ((f_caps & OBS_SOURCE_AUDIO) != 0 &&
	    (f_caps & OBS_SOURCE_VIDEO) == 0)
		f_caps &= ~OBS_SOURCE_ASYNC;

	return (s_caps & f_caps) == f_caps;
}

2221
void obs_source_filter_add(obs_source_t *source, obs_source_t *filter)
J
jp9000 已提交
2222
{
2223 2224
	struct calldata cd;
	uint8_t stack[128];
J
jp9000 已提交
2225

2226 2227 2228
	if (!obs_source_valid(source, "obs_source_filter_add"))
		return;
	if (!obs_ptr_valid(filter, "obs_source_filter_add"))
J
jp9000 已提交
2229 2230
		return;

2231 2232
	pthread_mutex_lock(&source->filter_mutex);

J
jp9000 已提交
2233
	if (da_find(source->filters, &filter, 0) != DARRAY_INVALID) {
J
jp9000 已提交
2234
		blog(LOG_WARNING, "Tried to add a filter that was already "
J
jp9000 已提交
2235
				  "present on the source");
2236
		pthread_mutex_unlock(&source->filter_mutex);
J
jp9000 已提交
2237 2238 2239
		return;
	}

2240 2241 2242 2243 2244
	if (!filter_compatible(source, filter)) {
		pthread_mutex_unlock(&source->filter_mutex);
		return;
	}

2245 2246
	obs_source_addref(filter);

2247
	filter->filter_parent = source;
J
jp9000 已提交
2248 2249
	filter->filter_target = !source->filters.num ? source
						     : source->filters.array[0];
2250

J
jp9000 已提交
2251
	da_insert(source->filters, 0, &filter);
2252 2253 2254

	pthread_mutex_unlock(&source->filter_mutex);

2255
	calldata_init_fixed(&cd, stack, sizeof(stack));
J
jp9000 已提交
2256 2257 2258 2259
	calldata_set_ptr(&cd, "source", source);
	calldata_set_ptr(&cd, "filter", filter);

	signal_handler_signal(source->context.signals, "filter_add", &cd);
2260

2261
	blog(LOG_DEBUG, "- filter '%s' (%s) added to source '%s'",
J
jp9000 已提交
2262
	     filter->context.name, filter->info.id, source->context.name);
J
jp9000 已提交
2263 2264
}

2265
static bool obs_source_filter_remove_refless(obs_source_t *source,
J
jp9000 已提交
2266
					     obs_source_t *filter)
J
jp9000 已提交
2267
{
2268 2269
	struct calldata cd;
	uint8_t stack[128];
2270 2271 2272 2273 2274
	size_t idx;

	pthread_mutex_lock(&source->filter_mutex);

	idx = da_find(source->filters, &filter, 0);
2275 2276
	if (idx == DARRAY_INVALID) {
		pthread_mutex_unlock(&source->filter_mutex);
2277
		return false;
2278
	}
J
jp9000 已提交
2279 2280

	if (idx > 0) {
J
jp9000 已提交
2281
		obs_source_t *prev = source->filters.array[idx - 1];
J
jp9000 已提交
2282 2283 2284 2285
		prev->filter_target = filter->filter_target;
	}

	da_erase(source->filters, idx);
2286 2287 2288

	pthread_mutex_unlock(&source->filter_mutex);

2289
	calldata_init_fixed(&cd, stack, sizeof(stack));
J
jp9000 已提交
2290 2291 2292 2293 2294
	calldata_set_ptr(&cd, "source", source);
	calldata_set_ptr(&cd, "filter", filter);

	signal_handler_signal(source->context.signals, "filter_remove", &cd);

2295
	blog(LOG_DEBUG, "- filter '%s' (%s) removed from source '%s'",
J
jp9000 已提交
2296
	     filter->context.name, filter->info.id, source->context.name);
2297

2298 2299
	if (filter->info.filter_remove)
		filter->info.filter_remove(filter->context.data,
J
jp9000 已提交
2300
					   filter->filter_parent);
2301

2302
	filter->filter_parent = NULL;
J
jp9000 已提交
2303
	filter->filter_target = NULL;
2304 2305
	return true;
}
2306

2307 2308
void obs_source_filter_remove(obs_source_t *source, obs_source_t *filter)
{
2309 2310 2311 2312 2313
	if (!obs_source_valid(source, "obs_source_filter_remove"))
		return;
	if (!obs_ptr_valid(filter, "obs_source_filter_remove"))
		return;

2314 2315
	if (obs_source_filter_remove_refless(source, filter))
		obs_source_release(filter);
J
jp9000 已提交
2316 2317
}

2318
static size_t find_next_filter(obs_source_t *source, obs_source_t *filter,
J
jp9000 已提交
2319
			       size_t cur_idx)
2320 2321 2322 2323 2324
{
	bool curAsync = (filter->info.output_flags & OBS_SOURCE_ASYNC) != 0;
	bool nextAsync;
	obs_source_t *next;

J
jp9000 已提交
2325
	if (cur_idx == source->filters.num - 1)
2326 2327
		return DARRAY_INVALID;

J
jp9000 已提交
2328
	next = source->filters.array[cur_idx + 1];
2329 2330 2331
	nextAsync = (next->info.output_flags & OBS_SOURCE_ASYNC);

	if (nextAsync == curAsync)
J
jp9000 已提交
2332
		return cur_idx + 1;
2333
	else
J
jp9000 已提交
2334
		return find_next_filter(source, filter, cur_idx + 1);
2335 2336 2337
}

static size_t find_prev_filter(obs_source_t *source, obs_source_t *filter,
J
jp9000 已提交
2338
			       size_t cur_idx)
2339 2340 2341 2342 2343 2344 2345 2346
{
	bool curAsync = (filter->info.output_flags & OBS_SOURCE_ASYNC) != 0;
	bool prevAsync;
	obs_source_t *prev;

	if (cur_idx == 0)
		return DARRAY_INVALID;

J
jp9000 已提交
2347
	prev = source->filters.array[cur_idx - 1];
2348 2349 2350
	prevAsync = (prev->info.output_flags & OBS_SOURCE_ASYNC);

	if (prevAsync == curAsync)
J
jp9000 已提交
2351
		return cur_idx - 1;
2352
	else
J
jp9000 已提交
2353
		return find_prev_filter(source, filter, cur_idx - 1);
2354 2355 2356
}

/* moves filters above/below matching filter types */
J
jp9000 已提交
2357 2358
static bool move_filter_dir(obs_source_t *source, obs_source_t *filter,
			    enum obs_order_movement movement)
J
jp9000 已提交
2359
{
2360
	size_t idx;
J
jp9000 已提交
2361 2362

	idx = da_find(source->filters, &filter, 0);
J
jp9000 已提交
2363
	if (idx == DARRAY_INVALID)
2364
		return false;
J
jp9000 已提交
2365

J
jp9000 已提交
2366
	if (movement == OBS_ORDER_MOVE_UP) {
2367 2368
		size_t next_id = find_next_filter(source, filter, idx);
		if (next_id == DARRAY_INVALID)
2369
			return false;
2370
		da_move_item(source->filters, idx, next_id);
J
jp9000 已提交
2371

J
jp9000 已提交
2372
	} else if (movement == OBS_ORDER_MOVE_DOWN) {
2373 2374
		size_t prev_id = find_prev_filter(source, filter, idx);
		if (prev_id == DARRAY_INVALID)
2375
			return false;
2376
		da_move_item(source->filters, idx, prev_id);
J
jp9000 已提交
2377

J
jp9000 已提交
2378
	} else if (movement == OBS_ORDER_MOVE_TOP) {
J
jp9000 已提交
2379
		if (idx == source->filters.num - 1)
2380
			return false;
J
jp9000 已提交
2381
		da_move_item(source->filters, idx, source->filters.num - 1);
J
jp9000 已提交
2382

J
jp9000 已提交
2383
	} else if (movement == OBS_ORDER_MOVE_BOTTOM) {
J
jp9000 已提交
2384
		if (idx == 0)
2385
			return false;
J
jp9000 已提交
2386 2387 2388
		da_move_item(source->filters, idx, 0);
	}

2389
	/* reorder filter targets, not the nicest way of dealing with things */
2390
	for (size_t i = 0; i < source->filters.num; i++) {
J
jp9000 已提交
2391 2392 2393 2394
		obs_source_t *next_filter =
			(i == source->filters.num - 1)
				? source
				: source->filters.array[i + 1];
2395

J
jp9000 已提交
2396 2397
		source->filters.array[i]->filter_target = next_filter;
	}
2398

2399 2400 2401 2402
	return true;
}

void obs_source_filter_set_order(obs_source_t *source, obs_source_t *filter,
J
jp9000 已提交
2403
				 enum obs_order_movement movement)
2404 2405
{
	bool success;
2406 2407 2408 2409

	if (!obs_source_valid(source, "obs_source_filter_set_order"))
		return;
	if (!obs_ptr_valid(filter, "obs_source_filter_set_order"))
2410 2411 2412 2413 2414 2415 2416 2417
		return;

	pthread_mutex_lock(&source->filter_mutex);
	success = move_filter_dir(source, filter, movement);
	pthread_mutex_unlock(&source->filter_mutex);

	if (success)
		obs_source_dosignal(source, NULL, "reorder_filters");
J
jp9000 已提交
2418 2419
}

2420
obs_data_t *obs_source_get_settings(const obs_source_t *source)
J
jp9000 已提交
2421
{
2422 2423
	if (!obs_source_valid(source, "obs_source_get_settings"))
		return NULL;
J
jp9000 已提交
2424

2425 2426
	obs_data_addref(source->context.settings);
	return source->context.settings;
J
jp9000 已提交
2427 2428
}

2429
struct obs_source_frame *filter_async_video(obs_source_t *source,
J
jp9000 已提交
2430
					    struct obs_source_frame *in)
2431 2432
{
	size_t i;
2433 2434 2435

	pthread_mutex_lock(&source->filter_mutex);

2436
	for (i = source->filters.num; i > 0; i--) {
J
jp9000 已提交
2437
		struct obs_source *filter = source->filters.array[i - 1];
2438

2439 2440 2441
		if (!filter->enabled)
			continue;

2442
		if (filter->context.data && filter->info.filter_video) {
2443
			in = filter->info.filter_video(filter->context.data,
J
jp9000 已提交
2444
						       in);
2445
			if (!in)
2446
				break;
2447 2448 2449
		}
	}

2450 2451
	pthread_mutex_unlock(&source->filter_mutex);

2452 2453 2454
	return in;
}

2455
static inline void copy_frame_data_line(struct obs_source_frame *dst,
J
jp9000 已提交
2456 2457
					const struct obs_source_frame *src,
					uint32_t plane, uint32_t y)
2458
{
2459 2460
	uint32_t pos_src = y * src->linesize[plane];
	uint32_t pos_dst = y * dst->linesize[plane];
J
jp9000 已提交
2461 2462 2463
	uint32_t bytes = dst->linesize[plane] < src->linesize[plane]
				 ? dst->linesize[plane]
				 : src->linesize[plane];
2464 2465 2466 2467

	memcpy(dst->data[plane] + pos_dst, src->data[plane] + pos_src, bytes);
}

2468
static inline void copy_frame_data_plane(struct obs_source_frame *dst,
J
jp9000 已提交
2469 2470
					 const struct obs_source_frame *src,
					 uint32_t plane, uint32_t lines)
2471
{
2472
	if (dst->linesize[plane] != src->linesize[plane])
2473 2474 2475 2476
		for (uint32_t y = 0; y < lines; y++)
			copy_frame_data_line(dst, src, plane, y);
	else
		memcpy(dst->data[plane], src->data[plane],
J
jp9000 已提交
2477
		       dst->linesize[plane] * lines);
2478 2479
}

2480
static void copy_frame_data(struct obs_source_frame *dst,
J
jp9000 已提交
2481
			    const struct obs_source_frame *src)
2482
{
J
jp9000 已提交
2483 2484 2485
	dst->flip = src->flip;
	dst->full_range = src->full_range;
	dst->timestamp = src->timestamp;
2486
	memcpy(dst->color_matrix, src->color_matrix, sizeof(float) * 16);
2487 2488 2489 2490 2491
	if (!dst->full_range) {
		size_t const size = sizeof(float) * 3;
		memcpy(dst->color_range_min, src->color_range_min, size);
		memcpy(dst->color_range_max, src->color_range_max, size);
	}
2492

J
jp9000 已提交
2493
	switch (src->format) {
2494 2495
	case VIDEO_FORMAT_I420:
		copy_frame_data_plane(dst, src, 0, dst->height);
J
jp9000 已提交
2496 2497
		copy_frame_data_plane(dst, src, 1, dst->height / 2);
		copy_frame_data_plane(dst, src, 2, dst->height / 2);
2498 2499 2500 2501
		break;

	case VIDEO_FORMAT_NV12:
		copy_frame_data_plane(dst, src, 0, dst->height);
J
jp9000 已提交
2502
		copy_frame_data_plane(dst, src, 1, dst->height / 2);
2503 2504
		break;

J
jp9000 已提交
2505
	case VIDEO_FORMAT_I444:
2506
	case VIDEO_FORMAT_I422:
J
jp9000 已提交
2507 2508 2509 2510 2511
		copy_frame_data_plane(dst, src, 0, dst->height);
		copy_frame_data_plane(dst, src, 1, dst->height);
		copy_frame_data_plane(dst, src, 2, dst->height);
		break;

2512 2513 2514 2515 2516 2517 2518
	case VIDEO_FORMAT_YVYU:
	case VIDEO_FORMAT_YUY2:
	case VIDEO_FORMAT_UYVY:
	case VIDEO_FORMAT_NONE:
	case VIDEO_FORMAT_RGBA:
	case VIDEO_FORMAT_BGRA:
	case VIDEO_FORMAT_BGRX:
2519
	case VIDEO_FORMAT_Y800:
2520
	case VIDEO_FORMAT_BGR3:
J
jpark37 已提交
2521
	case VIDEO_FORMAT_AYUV:
2522
		copy_frame_data_plane(dst, src, 0, dst->height);
2523
		break;
J
jpark37 已提交
2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538

	case VIDEO_FORMAT_I40A:
		copy_frame_data_plane(dst, src, 0, dst->height);
		copy_frame_data_plane(dst, src, 1, dst->height / 2);
		copy_frame_data_plane(dst, src, 2, dst->height / 2);
		copy_frame_data_plane(dst, src, 3, dst->height);
		break;

	case VIDEO_FORMAT_I42A:
	case VIDEO_FORMAT_YUVA:
		copy_frame_data_plane(dst, src, 0, dst->height);
		copy_frame_data_plane(dst, src, 1, dst->height);
		copy_frame_data_plane(dst, src, 2, dst->height);
		copy_frame_data_plane(dst, src, 3, dst->height);
		break;
2539 2540 2541
	}
}

B
Bas van Meel 已提交
2542
void obs_source_frame_copy(struct obs_source_frame *dst,
J
jp9000 已提交
2543
			   const struct obs_source_frame *src)
B
Bas van Meel 已提交
2544 2545 2546 2547
{
	copy_frame_data(dst, src);
}

2548
static inline bool async_texture_changed(struct obs_source *source,
J
jp9000 已提交
2549
					 const struct obs_source_frame *frame)
2550 2551
{
	enum convert_type prev, cur;
2552
	prev = get_convert_type(source->async_cache_format,
J
jp9000 已提交
2553 2554
				source->async_cache_full_range);
	cur = get_convert_type(frame->format, frame->full_range);
2555

J
jp9000 已提交
2556 2557
	return source->async_cache_width != frame->width ||
	       source->async_cache_height != frame->height || prev != cur;
2558 2559 2560 2561 2562
}

static inline void free_async_cache(struct obs_source *source)
{
	for (size_t i = 0; i < source->async_cache.num; i++)
2563
		obs_source_frame_decref(source->async_cache.array[i].frame);
2564 2565 2566

	da_resize(source->async_cache, 0);
	da_resize(source->async_frames, 0);
2567
	source->cur_async_frame = NULL;
J
jp9000 已提交
2568
	source->prev_async_frame = NULL;
2569 2570
}

2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587
#define MAX_UNUSED_FRAME_DURATION 5

/* frees frame allocations if they haven't been used for a specific period
 * of time */
static void clean_cache(obs_source_t *source)
{
	for (size_t i = source->async_cache.num; i > 0; i--) {
		struct async_frame *af = &source->async_cache.array[i - 1];
		if (!af->used) {
			if (++af->unused_count == MAX_UNUSED_FRAME_DURATION) {
				obs_source_frame_destroy(af->frame);
				da_erase(source->async_cache, i - 1);
			}
		}
	}
}

2588
#define MAX_ASYNC_FRAMES 30
2589
//if return value is not null then do (os_atomic_dec_long(&output->refs) == 0) && obs_source_frame_destroy(output)
J
jp9000 已提交
2590 2591
static inline struct obs_source_frame *
cache_video(struct obs_source *source, const struct obs_source_frame *frame)
2592
{
2593 2594 2595 2596
	struct obs_source_frame *new_frame = NULL;

	pthread_mutex_lock(&source->async_mutex);

2597 2598 2599 2600 2601 2602 2603
	if (source->async_frames.num >= MAX_ASYNC_FRAMES) {
		free_async_cache(source);
		source->last_frame_ts = 0;
		pthread_mutex_unlock(&source->async_mutex);
		return NULL;
	}

2604
	if (async_texture_changed(source, frame)) {
J
jp9000 已提交
2605
		free_async_cache(source);
J
jp9000 已提交
2606 2607
		source->async_cache_width = frame->width;
		source->async_cache_height = frame->height;
2608 2609
	}

2610 2611 2612 2613
	const enum video_format format = frame->format;
	source->async_cache_format = format;
	source->async_cache_full_range = frame->full_range;

2614 2615 2616 2617
	for (size_t i = 0; i < source->async_cache.num; i++) {
		struct async_frame *af = &source->async_cache.array[i];
		if (!af->used) {
			new_frame = af->frame;
2618
			new_frame->format = format;
2619
			af->used = true;
2620
			af->unused_count = 0;
2621
			break;
2622 2623 2624
		}
	}

2625 2626
	clean_cache(source);

2627 2628
	if (!new_frame) {
		struct async_frame new_af;
J
jp9000 已提交
2629

J
jp9000 已提交
2630 2631
		new_frame = obs_source_frame_create(format, frame->width,
						    frame->height);
2632 2633
		new_af.frame = new_frame;
		new_af.used = true;
2634
		new_af.unused_count = 0;
2635
		new_frame->refs = 1;
2636 2637 2638 2639

		da_push_back(source->async_cache, &new_af);
	}

2640 2641
	os_atomic_inc_long(&new_frame->refs);

2642
	pthread_mutex_unlock(&source->async_mutex);
2643

2644
	copy_frame_data(new_frame, frame);
2645

2646
	return new_frame;
2647 2648
}

J
jp9000 已提交
2649 2650 2651
static void
obs_source_output_video_internal(obs_source_t *source,
				 const struct obs_source_frame *frame)
2652
{
2653
	if (!obs_source_valid(source, "obs_source_output_video"))
J
jp9000 已提交
2654 2655
		return;

2656 2657 2658 2659 2660
	if (!frame) {
		source->async_active = false;
		return;
	}

J
jp9000 已提交
2661 2662
	struct obs_source_frame *output = !!frame ? cache_video(source, frame)
						  : NULL;
2663

2664
	/* ------------------------------------------- */
2665
	pthread_mutex_lock(&source->async_mutex);
2666
	if (output) {
2667 2668 2669 2670 2671 2672 2673
		if (os_atomic_dec_long(&output->refs) == 0) {
			obs_source_frame_destroy(output);
			output = NULL;
		} else {
			da_push_back(source->async_frames, &output);
			source->async_active = true;
		}
2674
	}
2675
	pthread_mutex_unlock(&source->async_mutex);
2676 2677
}

2678
void obs_source_output_video(obs_source_t *source,
J
jp9000 已提交
2679
			     const struct obs_source_frame *frame)
2680 2681 2682 2683 2684 2685 2686
{
	if (!frame) {
		obs_source_output_video_internal(source, NULL);
		return;
	}

	struct obs_source_frame new_frame = *frame;
J
jp9000 已提交
2687 2688
	new_frame.full_range =
		format_is_yuv(frame->format) ? new_frame.full_range : true;
2689 2690 2691 2692 2693

	obs_source_output_video_internal(source, &new_frame);
}

void obs_source_output_video2(obs_source_t *source,
J
jp9000 已提交
2694
			      const struct obs_source_frame2 *frame)
2695 2696 2697 2698 2699 2700 2701
{
	if (!frame) {
		obs_source_output_video_internal(source, NULL);
		return;
	}

	struct obs_source_frame new_frame;
J
jp9000 已提交
2702 2703
	enum video_range_type range =
		resolve_video_range(frame->format, frame->range);
2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715 2716 2717

	for (size_t i = 0; i < MAX_AV_PLANES; i++) {
		new_frame.data[i] = frame->data[i];
		new_frame.linesize[i] = frame->linesize[i];
	}

	new_frame.width = frame->width;
	new_frame.height = frame->height;
	new_frame.timestamp = frame->timestamp;
	new_frame.format = frame->format;
	new_frame.full_range = range == VIDEO_RANGE_FULL;
	new_frame.flip = frame->flip;

	memcpy(&new_frame.color_matrix, &frame->color_matrix,
J
jp9000 已提交
2718
	       sizeof(frame->color_matrix));
2719
	memcpy(&new_frame.color_range_min, &frame->color_range_min,
J
jp9000 已提交
2720
	       sizeof(frame->color_range_min));
2721
	memcpy(&new_frame.color_range_max, &frame->color_range_max,
J
jp9000 已提交
2722
	       sizeof(frame->color_range_max));
2723 2724 2725 2726

	obs_source_output_video_internal(source, &new_frame);
}

2727
static inline bool preload_frame_changed(obs_source_t *source,
J
jp9000 已提交
2728
					 const struct obs_source_frame *in)
2729 2730 2731 2732
{
	if (!source->async_preload_frame)
		return true;

J
jp9000 已提交
2733
	return in->width != source->async_preload_frame->width ||
2734 2735 2736 2737
	       in->height != source->async_preload_frame->height ||
	       in->format != source->async_preload_frame->format;
}

J
jp9000 已提交
2738 2739 2740
static void
obs_source_preload_video_internal(obs_source_t *source,
				  const struct obs_source_frame *frame)
2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751
{
	if (!obs_source_valid(source, "obs_source_preload_video"))
		return;
	if (!frame)
		return;

	obs_enter_graphics();

	if (preload_frame_changed(source, frame)) {
		obs_source_frame_destroy(source->async_preload_frame);
		source->async_preload_frame = obs_source_frame_create(
J
jp9000 已提交
2752
			frame->format, frame->width, frame->height);
2753 2754 2755 2756
	}

	copy_frame_data(source->async_preload_frame, frame);
	set_async_texture_size(source, source->async_preload_frame);
2757 2758
	update_async_textures(source, source->async_preload_frame,
			      source->async_textures, source->async_texrender);
2759 2760 2761 2762 2763 2764

	source->last_frame_ts = frame->timestamp;

	obs_leave_graphics();
}

2765
void obs_source_preload_video(obs_source_t *source,
J
jp9000 已提交
2766
			      const struct obs_source_frame *frame)
2767 2768 2769 2770 2771 2772 2773
{
	if (!frame) {
		obs_source_preload_video_internal(source, NULL);
		return;
	}

	struct obs_source_frame new_frame = *frame;
J
jp9000 已提交
2774 2775
	new_frame.full_range =
		format_is_yuv(frame->format) ? new_frame.full_range : true;
2776 2777 2778 2779 2780

	obs_source_preload_video_internal(source, &new_frame);
}

void obs_source_preload_video2(obs_source_t *source,
J
jp9000 已提交
2781
			       const struct obs_source_frame2 *frame)
2782 2783 2784 2785 2786 2787 2788
{
	if (!frame) {
		obs_source_preload_video_internal(source, NULL);
		return;
	}

	struct obs_source_frame new_frame;
J
jp9000 已提交
2789 2790
	enum video_range_type range =
		resolve_video_range(frame->format, frame->range);
2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804

	for (size_t i = 0; i < MAX_AV_PLANES; i++) {
		new_frame.data[i] = frame->data[i];
		new_frame.linesize[i] = frame->linesize[i];
	}

	new_frame.width = frame->width;
	new_frame.height = frame->height;
	new_frame.timestamp = frame->timestamp;
	new_frame.format = frame->format;
	new_frame.full_range = range == VIDEO_RANGE_FULL;
	new_frame.flip = frame->flip;

	memcpy(&new_frame.color_matrix, &frame->color_matrix,
J
jp9000 已提交
2805
	       sizeof(frame->color_matrix));
2806
	memcpy(&new_frame.color_range_min, &frame->color_range_min,
J
jp9000 已提交
2807
	       sizeof(frame->color_range_min));
2808
	memcpy(&new_frame.color_range_max, &frame->color_range_max,
J
jp9000 已提交
2809
	       sizeof(frame->color_range_max));
2810 2811 2812 2813

	obs_source_preload_video_internal(source, &new_frame);
}

2814 2815 2816 2817 2818 2819 2820 2821 2822 2823
void obs_source_show_preloaded_video(obs_source_t *source)
{
	uint64_t sys_ts;

	if (!obs_source_valid(source, "obs_source_show_preloaded_video"))
		return;

	source->async_active = true;

	pthread_mutex_lock(&source->audio_buf_mutex);
2824
	sys_ts = (source->monitoring_type != OBS_MONITORING_TYPE_MONITOR_ONLY)
J
jp9000 已提交
2825 2826
			 ? os_gettime_ns()
			 : 0;
2827 2828 2829 2830 2831
	reset_audio_timing(source, source->last_frame_ts, sys_ts);
	reset_audio_data(source, sys_ts);
	pthread_mutex_unlock(&source->audio_buf_mutex);
}

J
jp9000 已提交
2832 2833
static inline struct obs_audio_data *
filter_async_audio(obs_source_t *source, struct obs_audio_data *in)
2834 2835 2836
{
	size_t i;
	for (i = source->filters.num; i > 0; i--) {
J
jp9000 已提交
2837
		struct obs_source *filter = source->filters.array[i - 1];
2838

2839 2840 2841
		if (!filter->enabled)
			continue;

2842
		if (filter->context.data && filter->info.filter_audio) {
2843
			in = filter->info.filter_audio(filter->context.data,
J
jp9000 已提交
2844
						       in);
2845 2846 2847 2848 2849 2850 2851 2852
			if (!in)
				return NULL;
		}
	}

	return in;
}

2853
static inline void reset_resampler(obs_source_t *source,
J
jp9000 已提交
2854
				   const struct obs_source_audio *audio)
2855
{
J
jp9000 已提交
2856
	const struct audio_output_info *obs_info;
2857 2858
	struct resample_info output_info;

2859
	obs_info = audio_output_get_info(obs->audio.audio);
2860

J
jp9000 已提交
2861 2862 2863
	output_info.format = obs_info->format;
	output_info.samples_per_sec = obs_info->samples_per_sec;
	output_info.speakers = obs_info->speakers;
2864

J
jp9000 已提交
2865
	source->sample_info.format = audio->format;
2866
	source->sample_info.samples_per_sec = audio->samples_per_sec;
J
jp9000 已提交
2867
	source->sample_info.speakers = audio->speakers;
2868

2869 2870
	audio_resampler_destroy(source->resampler);
	source->resampler = NULL;
2871
	source->resample_offset = 0;
2872

2873
	if (source->sample_info.samples_per_sec == obs_info->samples_per_sec &&
J
jp9000 已提交
2874 2875
	    source->sample_info.format == obs_info->format &&
	    source->sample_info.speakers == obs_info->speakers) {
2876 2877 2878 2879
		source->audio_failed = false;
		return;
	}

J
jp9000 已提交
2880 2881
	source->resampler =
		audio_resampler_create(&output_info, &source->sample_info);
2882 2883 2884 2885 2886 2887

	source->audio_failed = source->resampler == NULL;
	if (source->resampler == NULL)
		blog(LOG_ERROR, "creation of resampler failed");
}

J
jp9000 已提交
2888 2889
static void copy_audio_data(obs_source_t *source, const uint8_t *const data[],
			    uint32_t frames, uint64_t ts)
2890
{
J
jp9000 已提交
2891
	size_t planes = audio_output_get_planes(obs->audio.audio);
2892
	size_t blocksize = audio_output_get_block_size(obs->audio.audio);
J
jp9000 已提交
2893 2894
	size_t size = (size_t)frames * blocksize;
	bool resize = source->audio_storage_size < size;
2895

J
jp9000 已提交
2896
	source->audio_data.frames = frames;
J
jp9000 已提交
2897
	source->audio_data.timestamp = ts;
2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908 2909 2910

	for (size_t i = 0; i < planes; i++) {
		/* ensure audio storage capacity */
		if (resize) {
			bfree(source->audio_data.data[i]);
			source->audio_data.data[i] = bmalloc(size);
		}

		memcpy(source->audio_data.data[i], data[i], size);
	}

	if (resize)
		source->audio_storage_size = size;
2911 2912
}

2913 2914 2915
/* TODO: SSE optimization */
static void downmix_to_mono_planar(struct obs_source *source, uint32_t frames)
{
J
jp9000 已提交
2916
	size_t channels = audio_output_get_channels(obs->audio.audio);
2917
	const float channels_i = 1.0f / (float)channels;
J
jp9000 已提交
2918
	float **data = (float **)source->audio_data.data;
2919

J
jp9000 已提交
2920
	for (size_t channel = 1; channel < channels; channel++) {
2921 2922 2923 2924 2925 2926 2927
		for (uint32_t frame = 0; frame < frames; frame++)
			data[0][frame] += data[channel][frame];
	}

	for (uint32_t frame = 0; frame < frames; frame++)
		data[0][frame] *= channels_i;

J
jp9000 已提交
2928
	for (size_t channel = 1; channel < channels; channel++) {
2929 2930 2931 2932 2933
		for (uint32_t frame = 0; frame < frames; frame++)
			data[channel][frame] = data[0][frame];
	}
}

C
cg2121 已提交
2934
static void process_audio_balancing(struct obs_source *source, uint32_t frames,
J
jp9000 已提交
2935
				    float balance, enum obs_balance_type type)
C
cg2121 已提交
2936
{
J
jp9000 已提交
2937
	float **data = (float **)source->audio_data.data;
C
cg2121 已提交
2938

J
jp9000 已提交
2939
	switch (type) {
C
cg2121 已提交
2940 2941 2942
	case OBS_BALANCE_TYPE_SINE_LAW:
		for (uint32_t frame = 0; frame < frames; frame++) {
			data[0][frame] = data[0][frame] *
J
jp9000 已提交
2943 2944 2945
					 sinf((1.0f - balance) * (M_PI / 2.0f));
			data[1][frame] =
				data[1][frame] * sinf(balance * (M_PI / 2.0f));
C
cg2121 已提交
2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958 2959 2960 2961 2962 2963 2964
		}
		break;
	case OBS_BALANCE_TYPE_SQUARE_LAW:
		for (uint32_t frame = 0; frame < frames; frame++) {
			data[0][frame] = data[0][frame] * sqrtf(1.0f - balance);
			data[1][frame] = data[1][frame] * sqrtf(balance);
		}
		break;
	case OBS_BALANCE_TYPE_LINEAR:
		for (uint32_t frame = 0; frame < frames; frame++) {
			data[0][frame] = data[0][frame] * (1.0f - balance);
			data[1][frame] = data[1][frame] * balance;
		}
		break;
	default:
		break;
	}
}

2965
/* resamples/remixes new audio to the designated main audio output format */
2966
static void process_audio(obs_source_t *source,
J
jp9000 已提交
2967
			  const struct obs_source_audio *audio)
2968
{
2969
	uint32_t frames = audio->frames;
2970
	bool mono_output;
2971

2972
	if (source->sample_info.samples_per_sec != audio->samples_per_sec ||
J
jp9000 已提交
2973 2974
	    source->sample_info.format != audio->format ||
	    source->sample_info.speakers != audio->speakers)
2975 2976 2977 2978 2979 2980
		reset_resampler(source, audio);

	if (source->audio_failed)
		return;

	if (source->resampler) {
J
jp9000 已提交
2981
		uint8_t *output[MAX_AV_PLANES];
2982

2983 2984
		memset(output, 0, sizeof(output));

J
jp9000 已提交
2985 2986 2987
		audio_resampler_resample(source->resampler, output, &frames,
					 &source->resample_offset, audio->data,
					 audio->frames);
2988

J
jp9000 已提交
2989
		copy_audio_data(source, (const uint8_t *const *)output, frames,
2990
				audio->timestamp);
2991 2992 2993 2994
	} else {
		copy_audio_data(source, audio->data, audio->frames,
				audio->timestamp);
	}
2995

2996 2997
	mono_output = audio_output_get_channels(obs->audio.audio) == 1;

2998 2999
	if (!mono_output && source->sample_info.speakers == SPEAKERS_STEREO &&
	    (source->balance > 0.51f || source->balance < 0.49f)) {
C
cg2121 已提交
3000
		process_audio_balancing(source, frames, source->balance,
J
jp9000 已提交
3001
					OBS_BALANCE_TYPE_SINE_LAW);
C
cg2121 已提交
3002 3003
	}

3004 3005
	if (!mono_output && (source->flags & OBS_SOURCE_FLAG_FORCE_MONO) != 0)
		downmix_to_mono_planar(source, frames);
3006 3007
}

3008
void obs_source_output_audio(obs_source_t *source,
J
jp9000 已提交
3009
			     const struct obs_source_audio *audio)
3010
{
3011
	struct obs_audio_data *output;
3012

3013 3014 3015
	if (!obs_source_valid(source, "obs_source_output_audio"))
		return;
	if (!obs_ptr_valid(audio, "obs_source_output_audio"))
J
jp9000 已提交
3016 3017
		return;

3018
	process_audio(source, audio);
3019 3020

	pthread_mutex_lock(&source->filter_mutex);
3021
	output = filter_async_audio(source, &source->audio_data);
3022 3023

	if (output) {
3024
		struct audio_data data;
J
jp9000 已提交
3025

3026 3027
		for (int i = 0; i < MAX_AV_PLANES; i++)
			data.data[i] = output->data[i];
3028

J
jp9000 已提交
3029
		data.frames = output->frames;
3030
		data.timestamp = output->timestamp;
3031

3032
		pthread_mutex_lock(&source->audio_mutex);
3033
		source_output_audio_data(source, &data);
3034 3035 3036 3037 3038 3039
		pthread_mutex_unlock(&source->audio_mutex);
	}

	pthread_mutex_unlock(&source->filter_mutex);
}

3040
void remove_async_frame(obs_source_t *source, struct obs_source_frame *frame)
3041
{
J
jp9000 已提交
3042 3043 3044
	if (frame)
		frame->prev_frame = false;

3045 3046 3047 3048 3049 3050 3051 3052 3053 3054
	for (size_t i = 0; i < source->async_cache.num; i++) {
		struct async_frame *f = &source->async_cache.array[i];

		if (f->frame == frame) {
			f->used = false;
			break;
		}
	}
}

J
jp9000 已提交
3055 3056
/* #define DEBUG_ASYNC_FRAMES 1 */

3057
static bool ready_async_frame(obs_source_t *source, uint64_t sys_time)
3058
{
3059
	struct obs_source_frame *next_frame = source->async_frames.array[0];
J
jp9000 已提交
3060
	struct obs_source_frame *frame = NULL;
3061 3062 3063 3064
	uint64_t sys_offset = sys_time - source->last_sys_timestamp;
	uint64_t frame_time = next_frame->timestamp;
	uint64_t frame_offset = 0;

3065
	if (source->async_unbuffered) {
3066 3067
		while (source->async_frames.num > 1) {
			da_erase(source->async_frames, 0);
3068
			remove_async_frame(source, next_frame);
3069
			next_frame = source->async_frames.array[0];
J
jp9000 已提交
3070 3071
		}

3072
		source->last_frame_ts = next_frame->timestamp;
J
jp9000 已提交
3073 3074 3075
		return true;
	}

J
jp9000 已提交
3076
#if DEBUG_ASYNC_FRAMES
J
jp9000 已提交
3077 3078 3079 3080 3081 3082 3083
	blog(LOG_DEBUG,
	     "source->last_frame_ts: %llu, frame_time: %llu, "
	     "sys_offset: %llu, frame_offset: %llu, "
	     "number of frames: %lu",
	     source->last_frame_ts, frame_time, sys_offset,
	     frame_time - source->last_frame_ts,
	     (unsigned long)source->async_frames.num);
J
jp9000 已提交
3084 3085
#endif

3086 3087
	/* account for timestamp invalidation */
	if (frame_out_of_bounds(source, frame_time)) {
J
jp9000 已提交
3088 3089 3090
#if DEBUG_ASYNC_FRAMES
		blog(LOG_DEBUG, "timing jump");
#endif
3091
		source->last_frame_ts = next_frame->timestamp;
J
jp9000 已提交
3092
		return true;
3093 3094
	} else {
		frame_offset = frame_time - source->last_frame_ts;
J
jp9000 已提交
3095
		source->last_frame_ts += sys_offset;
3096 3097
	}

J
jp9000 已提交
3098 3099 3100 3101 3102 3103
	while (source->last_frame_ts > next_frame->timestamp) {

		/* this tries to reduce the needless frame duplication, also
		 * helps smooth out async rendering to frame boundaries.  In
		 * other words, tries to keep the framerate as smooth as
		 * possible */
3104
		if ((source->last_frame_ts - next_frame->timestamp) < 2000000)
J
jp9000 已提交
3105 3106 3107
			break;

		if (frame)
3108
			da_erase(source->async_frames, 0);
J
jp9000 已提交
3109 3110

#if DEBUG_ASYNC_FRAMES
J
jp9000 已提交
3111 3112 3113 3114 3115
		blog(LOG_DEBUG,
		     "new frame, "
		     "source->last_frame_ts: %llu, "
		     "next_frame->timestamp: %llu",
		     source->last_frame_ts, next_frame->timestamp);
J
jp9000 已提交
3116 3117
#endif

3118
		remove_async_frame(source, frame);
3119

3120
		if (source->async_frames.num == 1)
3121 3122
			return true;

3123
		frame = next_frame;
3124
		next_frame = source->async_frames.array[1];
3125 3126

		/* more timestamp checking and compensating */
3127
		if ((next_frame->timestamp - frame_time) > MAX_TS_VAR) {
J
jp9000 已提交
3128 3129 3130
#if DEBUG_ASYNC_FRAMES
			blog(LOG_DEBUG, "timing jump");
#endif
3131 3132 3133 3134
			source->last_frame_ts =
				next_frame->timestamp - frame_offset;
		}

J
jp9000 已提交
3135
		frame_time = next_frame->timestamp;
3136 3137 3138
		frame_offset = frame_time - source->last_frame_ts;
	}

J
jp9000 已提交
3139 3140 3141 3142
#if DEBUG_ASYNC_FRAMES
	if (!frame)
		blog(LOG_DEBUG, "no frame!");
#endif
3143

3144 3145 3146
	return frame != NULL;
}

3147
static inline struct obs_source_frame *get_closest_frame(obs_source_t *source,
J
jp9000 已提交
3148
							 uint64_t sys_time)
3149
{
3150 3151 3152 3153
	if (!source->async_frames.num)
		return NULL;

	if (!source->last_frame_ts || ready_async_frame(source, sys_time)) {
3154 3155
		struct obs_source_frame *frame = source->async_frames.array[0];
		da_erase(source->async_frames, 0);
3156 3157 3158 3159

		if (!source->last_frame_ts)
			source->last_frame_ts = frame->timestamp;

3160 3161 3162 3163
		return frame;
	}

	return NULL;
3164 3165
}

3166
/*
3167 3168
 * Ensures that cached frames are displayed on time.  If multiple frames
 * were cached between renders, then releases the unnecessary frames and uses
3169 3170
 * the frame with the closest timing to ensure sync.  Also ensures that timing
 * with audio is synchronized.
3171
 */
3172
struct obs_source_frame *obs_source_get_frame(obs_source_t *source)
J
jp9000 已提交
3173
{
3174
	struct obs_source_frame *frame = NULL;
3175

3176
	if (!obs_source_valid(source, "obs_source_get_frame"))
J
jp9000 已提交
3177 3178
		return NULL;

3179
	pthread_mutex_lock(&source->async_mutex);
3180

3181 3182
	frame = source->cur_async_frame;
	source->cur_async_frame = NULL;
J
jp9000 已提交
3183 3184

	if (frame) {
3185
		os_atomic_inc_long(&frame->refs);
3186 3187
	}

3188
	pthread_mutex_unlock(&source->async_mutex);
3189

3190
	return frame;
J
jp9000 已提交
3191 3192
}

3193
void obs_source_release_frame(obs_source_t *source,
J
jp9000 已提交
3194
			      struct obs_source_frame *frame)
J
jp9000 已提交
3195
{
3196
	if (!frame)
3197 3198 3199 3200 3201 3202 3203 3204 3205 3206 3207 3208 3209
		return;

	if (!source) {
		obs_source_frame_destroy(frame);
	} else {
		pthread_mutex_lock(&source->async_mutex);

		if (os_atomic_dec_long(&frame->refs) == 0)
			obs_source_frame_destroy(frame);
		else
			remove_async_frame(source, frame);

		pthread_mutex_unlock(&source->async_mutex);
3210
	}
J
jp9000 已提交
3211
}
3212

3213
const char *obs_source_get_name(const obs_source_t *source)
3214
{
J
jp9000 已提交
3215 3216 3217
	return obs_source_valid(source, "obs_source_get_name")
		       ? source->context.name
		       : NULL;
3218 3219
}

3220
void obs_source_set_name(obs_source_t *source, const char *name)
3221
{
3222 3223
	if (!obs_source_valid(source, "obs_source_set_name"))
		return;
J
jp9000 已提交
3224

3225
	if (!name || !*name || !source->context.name ||
J
jp9000 已提交
3226
	    strcmp(name, source->context.name) != 0) {
J
jp9000 已提交
3227 3228 3229 3230 3231
		struct calldata data;
		char *prev_name = bstrdup(source->context.name);
		obs_context_data_setname(&source->context, name);

		calldata_init(&data);
3232 3233 3234
		calldata_set_ptr(&data, "source", source);
		calldata_set_string(&data, "new_name", source->context.name);
		calldata_set_string(&data, "prev_name", prev_name);
J
jp9000 已提交
3235 3236
		if (!source->context.private)
			signal_handler_signal(obs->signals, "source_rename",
J
jp9000 已提交
3237
					      &data);
J
jp9000 已提交
3238 3239 3240 3241
		signal_handler_signal(source->context.signals, "rename", &data);
		calldata_free(&data);
		bfree(prev_name);
	}
3242 3243
}

3244
enum obs_source_type obs_source_get_type(const obs_source_t *source)
3245
{
J
jp9000 已提交
3246 3247 3248
	return obs_source_valid(source, "obs_source_get_type")
		       ? source->info.type
		       : OBS_SOURCE_TYPE_INPUT;
J
jp9000 已提交
3249
}
J
jp9000 已提交
3250

3251
const char *obs_source_get_id(const obs_source_t *source)
J
jp9000 已提交
3252
{
J
jp9000 已提交
3253 3254
	return obs_source_valid(source, "obs_source_get_id") ? source->info.id
							     : NULL;
3255
}
3256

3257
static inline void render_filter_bypass(obs_source_t *target,
J
jp9000 已提交
3258 3259
					gs_effect_t *effect,
					const char *tech_name)
3260
{
J
jp9000 已提交
3261 3262
	gs_technique_t *tech = gs_effect_get_technique(effect, tech_name);
	size_t passes, i;
3263

3264
	passes = gs_technique_begin(tech);
3265
	for (i = 0; i < passes; i++) {
3266
		gs_technique_begin_pass(tech, i);
3267
		obs_source_video_render(target);
3268
		gs_technique_end_pass(tech);
3269
	}
3270
	gs_technique_end(tech);
3271 3272
}

3273
static inline void render_filter_tex(gs_texture_t *tex, gs_effect_t *effect,
J
jp9000 已提交
3274 3275
				     uint32_t width, uint32_t height,
				     const char *tech_name)
3276
{
J
jp9000 已提交
3277 3278 3279
	gs_technique_t *tech = gs_effect_get_technique(effect, tech_name);
	gs_eparam_t *image = gs_effect_get_param_by_name(effect, "image");
	size_t passes, i;
3280

3281
	gs_effect_set_texture(image, tex);
3282

3283
	passes = gs_technique_begin(tech);
3284
	for (i = 0; i < passes; i++) {
3285
		gs_technique_begin_pass(tech, i);
J
jp9000 已提交
3286
		gs_draw_sprite(tex, 0, width, height);
3287
		gs_technique_end_pass(tech);
3288
	}
3289
	gs_technique_end(tech);
3290 3291
}

3292
static inline bool can_bypass(obs_source_t *target, obs_source_t *parent,
J
jp9000 已提交
3293 3294
			      uint32_t parent_flags,
			      enum obs_allow_direct_render allow_direct)
3295 3296
{
	return (target == parent) &&
J
jp9000 已提交
3297 3298 3299
	       (allow_direct == OBS_ALLOW_DIRECT_RENDERING) &&
	       ((parent_flags & OBS_SOURCE_CUSTOM_DRAW) == 0) &&
	       ((parent_flags & OBS_SOURCE_ASYNC) == 0);
3300 3301
}

3302
bool obs_source_process_filter_begin(obs_source_t *filter,
J
jp9000 已提交
3303 3304
				     enum gs_color_format format,
				     enum obs_allow_direct_render allow_direct)
3305
{
3306
	obs_source_t *target, *parent;
J
jp9000 已提交
3307 3308
	uint32_t parent_flags;
	int cx, cy;
J
jp9000 已提交
3309

3310
	if (!obs_ptr_valid(filter, "obs_source_process_filter_begin"))
3311
		return false;
J
jp9000 已提交
3312

J
jp9000 已提交
3313 3314
	target = obs_filter_get_target(filter);
	parent = obs_filter_get_parent(filter);
J
jp9000 已提交
3315 3316 3317

	if (!target) {
		blog(LOG_INFO, "filter '%s' being processed with no target!",
J
jp9000 已提交
3318
		     filter->context.name);
3319
		return false;
J
jp9000 已提交
3320 3321 3322
	}
	if (!parent) {
		blog(LOG_INFO, "filter '%s' being processed with no parent!",
J
jp9000 已提交
3323
		     filter->context.name);
3324
		return false;
J
jp9000 已提交
3325 3326
	}

J
jp9000 已提交
3327
	parent_flags = parent->info.output_flags;
J
jp9000 已提交
3328 3329
	cx = get_base_width(target);
	cy = get_base_height(target);
3330

3331 3332
	filter->allow_direct = allow_direct;

3333 3334 3335 3336
	/* if the parent does not use any custom effects, and this is the last
	 * filter in the chain for the parent, then render the parent directly
	 * using the filter effect instead of rendering to texture to reduce
	 * the total number of passes */
3337
	if (can_bypass(target, parent, parent_flags, allow_direct)) {
3338
		return true;
3339 3340
	}

3341 3342
	if (!cx || !cy) {
		obs_source_skip_video_filter(filter);
3343
		return false;
3344 3345
	}

J
jp9000 已提交
3346
	if (!filter->filter_texrender)
J
jp9000 已提交
3347 3348
		filter->filter_texrender =
			gs_texrender_create(format, GS_ZS_NONE);
J
jp9000 已提交
3349

3350 3351 3352
	gs_blend_state_push();
	gs_blend_function(GS_BLEND_ONE, GS_BLEND_ZERO);

3353
	if (gs_texrender_begin(filter->filter_texrender, cx, cy)) {
3354 3355
		bool custom_draw = (parent_flags & OBS_SOURCE_CUSTOM_DRAW) != 0;
		bool async = (parent_flags & OBS_SOURCE_ASYNC) != 0;
3356 3357
		struct vec4 clear_color;

3358
		vec4_zero(&clear_color);
3359
		gs_clear(GS_CLEAR_COLOR, &clear_color, 0.0f, 0);
3360
		gs_ortho(0.0f, (float)cx, 0.0f, (float)cy, -100.0f, 100.0f);
3361

3362
		if (target == parent && !custom_draw && !async)
3363
			obs_source_default_render(target);
3364 3365
		else
			obs_source_video_render(target);
3366

3367
		gs_texrender_end(filter->filter_texrender);
3368
	}
3369 3370

	gs_blend_state_pop();
3371
	return true;
3372 3373
}

J
jp9000 已提交
3374 3375 3376
void obs_source_process_filter_tech_end(obs_source_t *filter,
					gs_effect_t *effect, uint32_t width,
					uint32_t height, const char *tech_name)
3377 3378 3379
{
	obs_source_t *target, *parent;
	gs_texture_t *texture;
J
jp9000 已提交
3380
	uint32_t parent_flags;
3381

J
jp9000 已提交
3382 3383
	if (!filter)
		return;
3384

J
jp9000 已提交
3385 3386
	target = obs_filter_get_target(filter);
	parent = obs_filter_get_parent(filter);
J
jp9000 已提交
3387 3388 3389 3390

	if (!target || !parent)
		return;

3391 3392 3393 3394 3395 3396 3397 3398
	parent_flags = parent->info.output_flags;

	const char *tech = tech_name ? tech_name : "Draw";

	if (can_bypass(target, parent, parent_flags, filter->allow_direct)) {
		render_filter_bypass(target, effect, tech);
	} else {
		texture = gs_texrender_get_texture(filter->filter_texrender);
3399 3400 3401
		if (texture) {
			render_filter_tex(texture, effect, width, height, tech);
		}
3402 3403 3404
	}
}

3405
void obs_source_process_filter_end(obs_source_t *filter, gs_effect_t *effect,
J
jp9000 已提交
3406
				   uint32_t width, uint32_t height)
3407
{
3408 3409
	if (!obs_ptr_valid(filter, "obs_source_process_filter_end"))
		return;
3410

3411 3412
	obs_source_process_filter_tech_end(filter, effect, width, height,
					   "Draw");
3413
}
3414

3415 3416 3417 3418 3419 3420
void obs_source_skip_video_filter(obs_source_t *filter)
{
	obs_source_t *target, *parent;
	bool custom_draw, async;
	uint32_t parent_flags;

3421 3422
	if (!obs_ptr_valid(filter, "obs_source_skip_video_filter"))
		return;
3423 3424 3425 3426 3427 3428 3429

	target = obs_filter_get_target(filter);
	parent = obs_filter_get_parent(filter);
	parent_flags = parent->info.output_flags;
	custom_draw = (parent_flags & OBS_SOURCE_CUSTOM_DRAW) != 0;
	async = (parent_flags & OBS_SOURCE_ASYNC) != 0;

J
jp9000 已提交
3430 3431
	if (target == parent) {
		if (!custom_draw && !async)
3432
			obs_source_default_render(target);
J
jp9000 已提交
3433 3434
		else if (target->info.video_render)
			obs_source_main_render(target);
J
jp9000 已提交
3435 3436
		else if (deinterlacing_enabled(target))
			deinterlace_render(target);
J
jp9000 已提交
3437 3438 3439 3440 3441 3442
		else
			obs_source_render_async_video(target);

	} else {
		obs_source_video_render(target);
	}
3443 3444
}

3445
signal_handler_t *obs_source_get_signal_handler(const obs_source_t *source)
3446
{
J
jp9000 已提交
3447 3448 3449
	return obs_source_valid(source, "obs_source_get_signal_handler")
		       ? source->context.signals
		       : NULL;
3450 3451
}

3452
proc_handler_t *obs_source_get_proc_handler(const obs_source_t *source)
3453
{
J
jp9000 已提交
3454 3455 3456
	return obs_source_valid(source, "obs_source_get_proc_handler")
		       ? source->context.procs
		       : NULL;
3457
}
J
jp9000 已提交
3458

3459
void obs_source_set_volume(obs_source_t *source, float volume)
J
jp9000 已提交
3460
{
3461
	if (obs_source_valid(source, "obs_source_set_volume")) {
J
jp9000 已提交
3462 3463 3464
		struct audio_action action = {.timestamp = os_gettime_ns(),
					      .type = AUDIO_ACTION_VOL,
					      .vol = volume};
J
jp9000 已提交
3465

3466 3467 3468 3469
		struct calldata data;
		uint8_t stack[128];

		calldata_init_fixed(&data, stack, sizeof(stack));
3470 3471
		calldata_set_ptr(&data, "source", source);
		calldata_set_float(&data, "volume", volume);
J
jp9000 已提交
3472

3473
		signal_handler_signal(source->context.signals, "volume", &data);
J
jp9000 已提交
3474 3475
		if (!source->context.private)
			signal_handler_signal(obs->signals, "source_volume",
J
jp9000 已提交
3476
					      &data);
J
jp9000 已提交
3477

3478
		volume = (float)calldata_float(&data, "volume");
J
jp9000 已提交
3479

J
jp9000 已提交
3480 3481 3482 3483
		pthread_mutex_lock(&source->audio_actions_mutex);
		da_push_back(source->audio_actions, &action);
		pthread_mutex_unlock(&source->audio_actions_mutex);

J
jp9000 已提交
3484
		source->user_volume = volume;
J
jp9000 已提交
3485
	}
J
jp9000 已提交
3486 3487
}

3488
float obs_source_get_volume(const obs_source_t *source)
J
jp9000 已提交
3489
{
J
jp9000 已提交
3490 3491 3492
	return obs_source_valid(source, "obs_source_get_volume")
		       ? source->user_volume
		       : 0.0f;
J
jp9000 已提交
3493 3494
}

3495
void obs_source_set_sync_offset(obs_source_t *source, int64_t offset)
J
jp9000 已提交
3496
{
3497
	if (obs_source_valid(source, "obs_source_set_sync_offset")) {
3498 3499
		struct calldata data;
		uint8_t stack[128];
J
jp9000 已提交
3500

3501
		calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
3502 3503 3504 3505
		calldata_set_ptr(&data, "source", source);
		calldata_set_int(&data, "offset", offset);

		signal_handler_signal(source->context.signals, "audio_sync",
J
jp9000 已提交
3506
				      &data);
J
jp9000 已提交
3507 3508 3509

		source->sync_offset = calldata_int(&data, "offset");
	}
J
jp9000 已提交
3510 3511
}

3512
int64_t obs_source_get_sync_offset(const obs_source_t *source)
J
jp9000 已提交
3513
{
J
jp9000 已提交
3514 3515 3516
	return obs_source_valid(source, "obs_source_get_sync_offset")
		       ? source->sync_offset
		       : 0;
J
jp9000 已提交
3517
}
3518 3519 3520 3521 3522 3523

struct source_enum_data {
	obs_source_enum_proc_t enum_callback;
	void *param;
};

3524
static void enum_source_active_tree_callback(obs_source_t *parent,
J
jp9000 已提交
3525
					     obs_source_t *child, void *param)
3526 3527
{
	struct source_enum_data *data = param;
3528
	bool is_transition = child->info.type == OBS_SOURCE_TYPE_TRANSITION;
3529

3530
	if (is_transition)
J
jp9000 已提交
3531 3532
		obs_transition_enum_sources(
			child, enum_source_active_tree_callback, param);
3533
	if (child->info.enum_active_sources) {
J
jp9000 已提交
3534
		if (child->context.data) {
J
jp9000 已提交
3535 3536 3537
			child->info.enum_active_sources(
				child->context.data,
				enum_source_active_tree_callback, data);
J
jp9000 已提交
3538
		}
3539 3540 3541 3542 3543
	}

	data->enum_callback(parent, child, data->param);
}

3544
void obs_source_enum_active_sources(obs_source_t *source,
J
jp9000 已提交
3545 3546
				    obs_source_enum_proc_t enum_callback,
				    void *param)
3547
{
J
jp9000 已提交
3548
	bool is_transition;
3549
	if (!data_valid(source, "obs_source_enum_active_sources"))
J
jp9000 已提交
3550
		return;
J
jp9000 已提交
3551 3552 3553

	is_transition = source->info.type == OBS_SOURCE_TYPE_TRANSITION;
	if (!is_transition && !source->info.enum_active_sources)
3554 3555 3556 3557
		return;

	obs_source_addref(source);

J
jp9000 已提交
3558 3559 3560 3561
	if (is_transition)
		obs_transition_enum_sources(source, enum_callback, param);
	if (source->info.enum_active_sources)
		source->info.enum_active_sources(source->context.data,
J
jp9000 已提交
3562
						 enum_callback, param);
3563 3564 3565 3566

	obs_source_release(source);
}

3567
void obs_source_enum_active_tree(obs_source_t *source,
J
jp9000 已提交
3568 3569
				 obs_source_enum_proc_t enum_callback,
				 void *param)
3570 3571
{
	struct source_enum_data data = {enum_callback, param};
J
jp9000 已提交
3572
	bool is_transition;
3573

3574
	if (!data_valid(source, "obs_source_enum_active_tree"))
J
jp9000 已提交
3575
		return;
J
jp9000 已提交
3576 3577 3578

	is_transition = source->info.type == OBS_SOURCE_TYPE_TRANSITION;
	if (!is_transition && !source->info.enum_active_sources)
3579 3580 3581 3582
		return;

	obs_source_addref(source);

J
jp9000 已提交
3583
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
J
jp9000 已提交
3584 3585
		obs_transition_enum_sources(
			source, enum_source_active_tree_callback, &data);
J
jp9000 已提交
3586
	if (source->info.enum_active_sources)
J
jp9000 已提交
3587 3588 3589
		source->info.enum_active_sources(
			source->context.data, enum_source_active_tree_callback,
			&data);
3590 3591 3592 3593 3594

	obs_source_release(source);
}

static void enum_source_full_tree_callback(obs_source_t *parent,
J
jp9000 已提交
3595
					   obs_source_t *child, void *param)
3596 3597 3598 3599 3600
{
	struct source_enum_data *data = param;
	bool is_transition = child->info.type == OBS_SOURCE_TYPE_TRANSITION;

	if (is_transition)
J
jp9000 已提交
3601 3602
		obs_transition_enum_sources(
			child, enum_source_full_tree_callback, param);
3603 3604
	if (child->info.enum_all_sources) {
		if (child->context.data) {
J
jp9000 已提交
3605 3606 3607
			child->info.enum_active_sources(
				child->context.data,
				enum_source_full_tree_callback, data);
3608 3609 3610
		}
	} else if (child->info.enum_active_sources) {
		if (child->context.data) {
J
jp9000 已提交
3611 3612 3613
			child->info.enum_active_sources(
				child->context.data,
				enum_source_full_tree_callback, data);
3614 3615 3616 3617 3618 3619 3620
		}
	}

	data->enum_callback(parent, child, data->param);
}

static void obs_source_enum_full_tree(obs_source_t *source,
J
jp9000 已提交
3621 3622
				      obs_source_enum_proc_t enum_callback,
				      void *param)
3623 3624 3625 3626 3627 3628 3629 3630 3631 3632 3633 3634 3635 3636
{
	struct source_enum_data data = {enum_callback, param};
	bool is_transition;

	if (!data_valid(source, "obs_source_enum_active_tree"))
		return;

	is_transition = source->info.type == OBS_SOURCE_TYPE_TRANSITION;
	if (!is_transition && !source->info.enum_active_sources)
		return;

	obs_source_addref(source);

	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
J
jp9000 已提交
3637 3638
		obs_transition_enum_sources(
			source, enum_source_full_tree_callback, &data);
3639 3640 3641

	if (source->info.enum_all_sources) {
		source->info.enum_all_sources(source->context.data,
J
jp9000 已提交
3642 3643
					      enum_source_full_tree_callback,
					      &data);
3644 3645 3646

	} else if (source->info.enum_active_sources) {
		source->info.enum_active_sources(source->context.data,
J
jp9000 已提交
3647 3648
						 enum_source_full_tree_callback,
						 &data);
3649
	}
3650 3651 3652

	obs_source_release(source);
}
3653

J
jp9000 已提交
3654 3655 3656 3657 3658 3659
struct descendant_info {
	bool exists;
	obs_source_t *target;
};

static void check_descendant(obs_source_t *parent, obs_source_t *child,
J
jp9000 已提交
3660
			     void *param)
3661
{
J
jp9000 已提交
3662 3663 3664 3665 3666
	struct descendant_info *info = param;
	if (child == info->target || parent == info->target)
		info->exists = true;
}

3667
bool obs_source_add_active_child(obs_source_t *parent, obs_source_t *child)
J
jp9000 已提交
3668
{
3669
	struct descendant_info info = {false, parent};
3670

3671
	if (!obs_ptr_valid(parent, "obs_source_add_active_child"))
3672
		return false;
3673
	if (!obs_ptr_valid(child, "obs_source_add_active_child"))
3674 3675
		return false;
	if (parent == child) {
3676
		blog(LOG_WARNING, "obs_source_add_active_child: "
J
jp9000 已提交
3677
				  "parent == child");
3678 3679
		return false;
	}
J
jp9000 已提交
3680

3681
	obs_source_enum_full_tree(child, check_descendant, &info);
J
jp9000 已提交
3682 3683
	if (info.exists)
		return false;
3684

3685 3686 3687 3688 3689
	for (int i = 0; i < parent->show_refs; i++) {
		enum view_type type;
		type = (i < parent->activate_refs) ? MAIN_VIEW : AUX_VIEW;
		obs_source_activate(child, type);
	}
J
jp9000 已提交
3690 3691

	return true;
3692 3693
}

3694
void obs_source_remove_active_child(obs_source_t *parent, obs_source_t *child)
3695
{
3696
	if (!obs_ptr_valid(parent, "obs_source_remove_active_child"))
3697
		return;
3698
	if (!obs_ptr_valid(child, "obs_source_remove_active_child"))
3699
		return;
3700

3701 3702 3703 3704 3705
	for (int i = 0; i < parent->show_refs; i++) {
		enum view_type type;
		type = (i < parent->activate_refs) ? MAIN_VIEW : AUX_VIEW;
		obs_source_deactivate(child, type);
	}
3706
}
J
jp9000 已提交
3707

3708
void obs_source_save(obs_source_t *source)
3709
{
J
jp9000 已提交
3710 3711 3712
	if (!data_valid(source, "obs_source_save"))
		return;

J
jp9000 已提交
3713 3714 3715 3716
	obs_source_dosignal(source, "source_save", "save");

	if (source->info.save)
		source->info.save(source->context.data,
J
jp9000 已提交
3717
				  source->context.settings);
3718 3719
}

3720
void obs_source_load(obs_source_t *source)
3721
{
J
jp9000 已提交
3722 3723
	if (!data_valid(source, "obs_source_load"))
		return;
J
jp9000 已提交
3724 3725
	if (source->info.load)
		source->info.load(source->context.data,
J
jp9000 已提交
3726
				  source->context.settings);
J
jp9000 已提交
3727

J
jp9000 已提交
3728
	obs_source_dosignal(source, "source_load", "load");
3729
}
J
jp9000 已提交
3730

J
jp9000 已提交
3731 3732
bool obs_source_active(const obs_source_t *source)
{
J
jp9000 已提交
3733 3734 3735
	return obs_source_valid(source, "obs_source_active")
		       ? source->activate_refs != 0
		       : false;
J
jp9000 已提交
3736 3737
}

J
jp9000 已提交
3738 3739
bool obs_source_showing(const obs_source_t *source)
{
J
jp9000 已提交
3740 3741 3742
	return obs_source_valid(source, "obs_source_showing")
		       ? source->show_refs != 0
		       : false;
J
jp9000 已提交
3743 3744
}

J
jp9000 已提交
3745 3746
static inline void signal_flags_updated(obs_source_t *source)
{
3747 3748
	struct calldata data;
	uint8_t stack[128];
J
jp9000 已提交
3749

3750
	calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
3751 3752 3753 3754 3755 3756 3757 3758
	calldata_set_ptr(&data, "source", source);
	calldata_set_int(&data, "flags", source->flags);

	signal_handler_signal(source->context.signals, "update_flags", &data);
}

void obs_source_set_flags(obs_source_t *source, uint32_t flags)
{
3759 3760
	if (!obs_source_valid(source, "obs_source_set_flags"))
		return;
J
jp9000 已提交
3761 3762 3763 3764 3765 3766 3767

	if (flags != source->flags) {
		source->flags = flags;
		signal_flags_updated(source);
	}
}

3768 3769
void obs_source_set_default_flags(obs_source_t *source, uint32_t flags)
{
3770 3771
	if (!obs_source_valid(source, "obs_source_set_default_flags"))
		return;
3772 3773 3774 3775

	source->default_flags = flags;
}

J
jp9000 已提交
3776 3777
uint32_t obs_source_get_flags(const obs_source_t *source)
{
J
jp9000 已提交
3778 3779
	return obs_source_valid(source, "obs_source_get_flags") ? source->flags
								: 0;
J
jp9000 已提交
3780
}
J
jp9000 已提交
3781

3782 3783
void obs_source_set_audio_mixers(obs_source_t *source, uint32_t mixers)
{
3784 3785
	struct calldata data;
	uint8_t stack[128];
3786

3787 3788 3789 3790
	if (!obs_source_valid(source, "obs_source_set_audio_mixers"))
		return;
	if ((source->info.output_flags & OBS_SOURCE_AUDIO) == 0)
		return;
3791

3792
	if (source->audio_mixers == mixers)
3793 3794
		return;

3795
	calldata_init_fixed(&data, stack, sizeof(stack));
3796 3797 3798 3799 3800 3801 3802
	calldata_set_ptr(&data, "source", source);
	calldata_set_int(&data, "mixers", mixers);

	signal_handler_signal(source->context.signals, "audio_mixers", &data);

	mixers = (uint32_t)calldata_int(&data, "mixers");

3803
	source->audio_mixers = mixers;
3804 3805 3806 3807
}

uint32_t obs_source_get_audio_mixers(const obs_source_t *source)
{
3808 3809 3810 3811
	if (!obs_source_valid(source, "obs_source_get_audio_mixers"))
		return 0;
	if ((source->info.output_flags & OBS_SOURCE_AUDIO) == 0)
		return 0;
3812

3813
	return source->audio_mixers;
3814 3815
}

J
jp9000 已提交
3816
void obs_source_draw_set_color_matrix(const struct matrix4 *color_matrix,
J
jp9000 已提交
3817 3818
				      const struct vec3 *color_range_min,
				      const struct vec3 *color_range_max)
J
jp9000 已提交
3819
{
3820 3821 3822 3823 3824 3825
	struct vec3 color_range_min_def;
	struct vec3 color_range_max_def;

	vec3_set(&color_range_min_def, 0.0f, 0.0f, 0.0f);
	vec3_set(&color_range_max_def, 1.0f, 1.0f, 1.0f);

J
jp9000 已提交
3826 3827 3828 3829 3830 3831
	gs_effect_t *effect = gs_get_effect();
	gs_eparam_t *matrix;
	gs_eparam_t *range_min;
	gs_eparam_t *range_max;

	if (!effect) {
3832
		blog(LOG_WARNING, "obs_source_draw_set_color_matrix: no "
J
jp9000 已提交
3833
				  "active effect!");
J
jp9000 已提交
3834 3835 3836
		return;
	}

3837
	if (!obs_ptr_valid(color_matrix, "obs_source_draw_set_color_matrix"))
J
jp9000 已提交
3838 3839 3840 3841 3842 3843 3844 3845 3846 3847 3848 3849
		return;

	if (!color_range_min)
		color_range_min = &color_range_min_def;
	if (!color_range_max)
		color_range_max = &color_range_max_def;

	matrix = gs_effect_get_param_by_name(effect, "color_matrix");
	range_min = gs_effect_get_param_by_name(effect, "color_range_min");
	range_max = gs_effect_get_param_by_name(effect, "color_range_max");

	gs_effect_set_matrix4(matrix, color_matrix);
J
jp9000 已提交
3850 3851
	gs_effect_set_val(range_min, color_range_min, sizeof(float) * 3);
	gs_effect_set_val(range_max, color_range_max, sizeof(float) * 3);
J
jp9000 已提交
3852 3853 3854
}

void obs_source_draw(gs_texture_t *texture, int x, int y, uint32_t cx,
J
jp9000 已提交
3855
		     uint32_t cy, bool flip)
J
jp9000 已提交
3856 3857 3858 3859 3860 3861
{
	gs_effect_t *effect = gs_get_effect();
	bool change_pos = (x != 0 || y != 0);
	gs_eparam_t *image;

	if (!effect) {
3862
		blog(LOG_WARNING, "obs_source_draw: no active effect!");
J
jp9000 已提交
3863 3864 3865
		return;
	}

3866
	if (!obs_ptr_valid(texture, "obs_source_draw"))
J
jp9000 已提交
3867 3868 3869 3870 3871 3872 3873 3874 3875 3876 3877 3878 3879 3880 3881
		return;

	image = gs_effect_get_param_by_name(effect, "image");
	gs_effect_set_texture(image, texture);

	if (change_pos) {
		gs_matrix_push();
		gs_matrix_translate3f((float)x, (float)y, 0.0f);
	}

	gs_draw_sprite(texture, flip ? GS_FLIP_V : 0, cx, cy);

	if (change_pos)
		gs_matrix_pop();
}
3882

3883 3884
void obs_source_inc_showing(obs_source_t *source)
{
3885 3886
	if (obs_source_valid(source, "obs_source_inc_showing"))
		obs_source_activate(source, AUX_VIEW);
3887 3888
}

3889 3890 3891 3892 3893 3894
void obs_source_inc_active(obs_source_t *source)
{
	if (obs_source_valid(source, "obs_source_inc_active"))
		obs_source_activate(source, MAIN_VIEW);
}

3895 3896
void obs_source_dec_showing(obs_source_t *source)
{
3897 3898
	if (obs_source_valid(source, "obs_source_dec_showing"))
		obs_source_deactivate(source, AUX_VIEW);
3899
}
3900

3901 3902 3903 3904 3905 3906
void obs_source_dec_active(obs_source_t *source)
{
	if (obs_source_valid(source, "obs_source_dec_active"))
		obs_source_deactivate(source, MAIN_VIEW);
}

3907
void obs_source_enum_filters(obs_source_t *source,
J
jp9000 已提交
3908
			     obs_source_enum_proc_t callback, void *param)
3909
{
3910 3911 3912
	if (!obs_source_valid(source, "obs_source_enum_filters"))
		return;
	if (!obs_ptr_valid(callback, "obs_source_enum_filters"))
3913 3914 3915 3916
		return;

	pthread_mutex_lock(&source->filter_mutex);

3917 3918
	for (size_t i = source->filters.num; i > 0; i--) {
		struct obs_source *filter = source->filters.array[i - 1];
3919 3920 3921 3922 3923
		callback(source, filter, param);
	}

	pthread_mutex_unlock(&source->filter_mutex);
}
3924 3925

obs_source_t *obs_source_get_filter_by_name(obs_source_t *source,
J
jp9000 已提交
3926
					    const char *name)
3927 3928 3929
{
	obs_source_t *filter = NULL;

3930 3931 3932
	if (!obs_source_valid(source, "obs_source_get_filter_by_name"))
		return NULL;
	if (!obs_ptr_valid(name, "obs_source_get_filter_by_name"))
3933 3934 3935 3936 3937 3938 3939 3940 3941 3942 3943 3944 3945 3946 3947 3948 3949
		return NULL;

	pthread_mutex_lock(&source->filter_mutex);

	for (size_t i = 0; i < source->filters.num; i++) {
		struct obs_source *cur_filter = source->filters.array[i];
		if (strcmp(cur_filter->context.name, name) == 0) {
			filter = cur_filter;
			obs_source_addref(filter);
			break;
		}
	}

	pthread_mutex_unlock(&source->filter_mutex);

	return filter;
}
3950 3951 3952

bool obs_source_enabled(const obs_source_t *source)
{
J
jp9000 已提交
3953 3954
	return obs_source_valid(source, "obs_source_enabled") ? source->enabled
							      : false;
3955 3956 3957 3958
}

void obs_source_set_enabled(obs_source_t *source, bool enabled)
{
3959 3960
	struct calldata data;
	uint8_t stack[128];
3961

3962
	if (!obs_source_valid(source, "obs_source_set_enabled"))
3963 3964 3965 3966
		return;

	source->enabled = enabled;

3967
	calldata_init_fixed(&data, stack, sizeof(stack));
3968 3969 3970 3971 3972
	calldata_set_ptr(&data, "source", source);
	calldata_set_bool(&data, "enabled", enabled);

	signal_handler_signal(source->context.signals, "enable", &data);
}
J
jp9000 已提交
3973 3974 3975

bool obs_source_muted(const obs_source_t *source)
{
J
jp9000 已提交
3976 3977
	return obs_source_valid(source, "obs_source_muted") ? source->user_muted
							    : false;
J
jp9000 已提交
3978 3979 3980 3981
}

void obs_source_set_muted(obs_source_t *source, bool muted)
{
3982 3983
	struct calldata data;
	uint8_t stack[128];
J
jp9000 已提交
3984 3985 3986
	struct audio_action action = {.timestamp = os_gettime_ns(),
				      .type = AUDIO_ACTION_MUTE,
				      .set = muted};
J
jp9000 已提交
3987

3988
	if (!obs_source_valid(source, "obs_source_set_muted"))
J
jp9000 已提交
3989 3990
		return;

J
jp9000 已提交
3991
	source->user_muted = muted;
J
jp9000 已提交
3992

3993
	calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
3994 3995 3996 3997 3998
	calldata_set_ptr(&data, "source", source);
	calldata_set_bool(&data, "muted", muted);

	signal_handler_signal(source->context.signals, "mute", &data);

J
jp9000 已提交
3999 4000 4001
	pthread_mutex_lock(&source->audio_actions_mutex);
	da_push_back(source->audio_actions, &action);
	pthread_mutex_unlock(&source->audio_actions_mutex);
J
jp9000 已提交
4002
}
P
Palana 已提交
4003 4004

static void source_signal_push_to_changed(obs_source_t *source,
J
jp9000 已提交
4005
					  const char *signal, bool enabled)
P
Palana 已提交
4006
{
4007 4008
	struct calldata data;
	uint8_t stack[128];
P
Palana 已提交
4009

4010
	calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
4011
	calldata_set_ptr(&data, "source", source);
P
Palana 已提交
4012 4013 4014 4015 4016 4017
	calldata_set_bool(&data, "enabled", enabled);

	signal_handler_signal(source->context.signals, signal, &data);
}

static void source_signal_push_to_delay(obs_source_t *source,
J
jp9000 已提交
4018
					const char *signal, uint64_t delay)
P
Palana 已提交
4019
{
4020 4021
	struct calldata data;
	uint8_t stack[128];
P
Palana 已提交
4022

4023
	calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
4024 4025
	calldata_set_ptr(&data, "source", source);
	calldata_set_bool(&data, "delay", delay);
P
Palana 已提交
4026 4027 4028 4029 4030 4031 4032

	signal_handler_signal(source->context.signals, signal, &data);
}

bool obs_source_push_to_mute_enabled(obs_source_t *source)
{
	bool enabled;
4033 4034
	if (!obs_source_valid(source, "obs_source_push_to_mute_enabled"))
		return false;
P
Palana 已提交
4035 4036 4037 4038 4039 4040 4041 4042 4043 4044

	pthread_mutex_lock(&source->audio_mutex);
	enabled = source->push_to_mute_enabled;
	pthread_mutex_unlock(&source->audio_mutex);

	return enabled;
}

void obs_source_enable_push_to_mute(obs_source_t *source, bool enabled)
{
4045 4046
	if (!obs_source_valid(source, "obs_source_enable_push_to_mute"))
		return;
P
Palana 已提交
4047 4048 4049 4050 4051

	pthread_mutex_lock(&source->audio_mutex);
	bool changed = source->push_to_mute_enabled != enabled;
	if (obs_source_get_output_flags(source) & OBS_SOURCE_AUDIO && changed)
		blog(LOG_INFO, "source '%s' %s push-to-mute",
J
jp9000 已提交
4052 4053
		     obs_source_get_name(source),
		     enabled ? "enabled" : "disabled");
P
Palana 已提交
4054 4055 4056 4057 4058

	source->push_to_mute_enabled = enabled;

	if (changed)
		source_signal_push_to_changed(source, "push_to_mute_changed",
J
jp9000 已提交
4059
					      enabled);
P
Palana 已提交
4060 4061 4062 4063 4064 4065
	pthread_mutex_unlock(&source->audio_mutex);
}

uint64_t obs_source_get_push_to_mute_delay(obs_source_t *source)
{
	uint64_t delay;
4066 4067
	if (!obs_source_valid(source, "obs_source_get_push_to_mute_delay"))
		return 0;
P
Palana 已提交
4068 4069 4070 4071 4072 4073 4074 4075 4076 4077

	pthread_mutex_lock(&source->audio_mutex);
	delay = source->push_to_mute_delay;
	pthread_mutex_unlock(&source->audio_mutex);

	return delay;
}

void obs_source_set_push_to_mute_delay(obs_source_t *source, uint64_t delay)
{
4078 4079
	if (!obs_source_valid(source, "obs_source_set_push_to_mute_delay"))
		return;
P
Palana 已提交
4080 4081 4082 4083 4084 4085 4086 4087 4088 4089 4090

	pthread_mutex_lock(&source->audio_mutex);
	source->push_to_mute_delay = delay;

	source_signal_push_to_delay(source, "push_to_mute_delay", delay);
	pthread_mutex_unlock(&source->audio_mutex);
}

bool obs_source_push_to_talk_enabled(obs_source_t *source)
{
	bool enabled;
4091 4092
	if (!obs_source_valid(source, "obs_source_push_to_talk_enabled"))
		return false;
P
Palana 已提交
4093 4094 4095 4096 4097 4098 4099 4100 4101 4102

	pthread_mutex_lock(&source->audio_mutex);
	enabled = source->push_to_talk_enabled;
	pthread_mutex_unlock(&source->audio_mutex);

	return enabled;
}

void obs_source_enable_push_to_talk(obs_source_t *source, bool enabled)
{
4103 4104
	if (!obs_source_valid(source, "obs_source_enable_push_to_talk"))
		return;
P
Palana 已提交
4105 4106 4107 4108 4109

	pthread_mutex_lock(&source->audio_mutex);
	bool changed = source->push_to_talk_enabled != enabled;
	if (obs_source_get_output_flags(source) & OBS_SOURCE_AUDIO && changed)
		blog(LOG_INFO, "source '%s' %s push-to-talk",
J
jp9000 已提交
4110 4111
		     obs_source_get_name(source),
		     enabled ? "enabled" : "disabled");
P
Palana 已提交
4112 4113 4114 4115 4116

	source->push_to_talk_enabled = enabled;

	if (changed)
		source_signal_push_to_changed(source, "push_to_talk_changed",
J
jp9000 已提交
4117
					      enabled);
P
Palana 已提交
4118 4119 4120 4121 4122 4123
	pthread_mutex_unlock(&source->audio_mutex);
}

uint64_t obs_source_get_push_to_talk_delay(obs_source_t *source)
{
	uint64_t delay;
4124 4125
	if (!obs_source_valid(source, "obs_source_get_push_to_talk_delay"))
		return 0;
P
Palana 已提交
4126 4127 4128 4129 4130 4131 4132 4133 4134 4135

	pthread_mutex_lock(&source->audio_mutex);
	delay = source->push_to_talk_delay;
	pthread_mutex_unlock(&source->audio_mutex);

	return delay;
}

void obs_source_set_push_to_talk_delay(obs_source_t *source, uint64_t delay)
{
4136 4137
	if (!obs_source_valid(source, "obs_source_set_push_to_talk_delay"))
		return;
P
Palana 已提交
4138 4139 4140 4141 4142 4143 4144

	pthread_mutex_lock(&source->audio_mutex);
	source->push_to_talk_delay = delay;

	source_signal_push_to_delay(source, "push_to_talk_delay", delay);
	pthread_mutex_unlock(&source->audio_mutex);
}
4145 4146 4147 4148

void *obs_source_get_type_data(obs_source_t *source)
{
	return obs_source_valid(source, "obs_source_get_type_data")
J
jp9000 已提交
4149 4150
		       ? source->info.type_data
		       : NULL;
4151
}
4152

J
jp9000 已提交
4153 4154 4155
static float get_source_volume(obs_source_t *source, uint64_t os_time)
{
	if (source->push_to_mute_enabled && source->push_to_mute_pressed)
J
jp9000 已提交
4156 4157
		source->push_to_mute_stop_time =
			os_time + source->push_to_mute_delay * 1000000;
J
jp9000 已提交
4158 4159

	if (source->push_to_talk_enabled && source->push_to_talk_pressed)
J
jp9000 已提交
4160 4161
		source->push_to_talk_stop_time =
			os_time + source->push_to_talk_delay * 1000000;
J
jp9000 已提交
4162 4163

	bool push_to_mute_active = source->push_to_mute_pressed ||
J
jp9000 已提交
4164
				   os_time < source->push_to_mute_stop_time;
J
jp9000 已提交
4165
	bool push_to_talk_active = source->push_to_talk_pressed ||
J
jp9000 已提交
4166
				   os_time < source->push_to_talk_stop_time;
J
jp9000 已提交
4167 4168

	bool muted = !source->enabled || source->muted ||
J
jp9000 已提交
4169 4170
		     (source->push_to_mute_enabled && push_to_mute_active) ||
		     (source->push_to_talk_enabled && !push_to_talk_active);
J
jp9000 已提交
4171 4172 4173 4174 4175 4176 4177 4178 4179 4180

	if (muted || close_float(source->volume, 0.0f, 0.0001f))
		return 0.0f;
	if (close_float(source->volume, 1.0f, 0.0001f))
		return 1.0f;

	return source->volume;
}

static inline void multiply_output_audio(obs_source_t *source, size_t mix,
J
jp9000 已提交
4181
					 size_t channels, float vol)
J
jp9000 已提交
4182 4183 4184 4185 4186 4187 4188 4189 4190
{
	register float *out = source->audio_output_buf[mix][0];
	register float *end = out + AUDIO_OUTPUT_FRAMES * channels;

	while (out < end)
		*(out++) *= vol;
}

static inline void multiply_vol_data(obs_source_t *source, size_t mix,
J
jp9000 已提交
4191
				     size_t channels, float *vol_data)
J
jp9000 已提交
4192 4193 4194 4195 4196 4197 4198 4199 4200 4201 4202 4203
{
	for (size_t ch = 0; ch < channels; ch++) {
		register float *out = source->audio_output_buf[mix][ch];
		register float *end = out + AUDIO_OUTPUT_FRAMES;
		register float *vol = vol_data;

		while (out < end)
			*(out++) *= *(vol++);
	}
}

static inline void apply_audio_action(obs_source_t *source,
J
jp9000 已提交
4204
				      const struct audio_action *action)
J
jp9000 已提交
4205 4206 4207
{
	switch (action->type) {
	case AUDIO_ACTION_VOL:
J
jp9000 已提交
4208 4209
		source->volume = action->vol;
		break;
J
jp9000 已提交
4210
	case AUDIO_ACTION_MUTE:
J
jp9000 已提交
4211 4212
		source->muted = action->set;
		break;
J
jp9000 已提交
4213
	case AUDIO_ACTION_PTT:
J
jp9000 已提交
4214 4215
		source->push_to_talk_pressed = action->set;
		break;
J
jp9000 已提交
4216
	case AUDIO_ACTION_PTM:
J
jp9000 已提交
4217 4218
		source->push_to_mute_pressed = action->set;
		break;
J
jp9000 已提交
4219 4220 4221 4222
	}
}

static void apply_audio_actions(obs_source_t *source, size_t channels,
J
jp9000 已提交
4223
				size_t sample_rate)
J
jp9000 已提交
4224 4225 4226 4227 4228 4229 4230 4231 4232 4233 4234 4235 4236 4237 4238
{
	float *vol_data = malloc(sizeof(float) * AUDIO_OUTPUT_FRAMES);
	float cur_vol = get_source_volume(source, source->audio_ts);
	size_t frame_num = 0;

	pthread_mutex_lock(&source->audio_actions_mutex);

	for (size_t i = 0; i < source->audio_actions.num; i++) {
		struct audio_action action = source->audio_actions.array[i];
		uint64_t timestamp = action.timestamp;
		size_t new_frame_num;

		if (timestamp < source->audio_ts)
			timestamp = source->audio_ts;

J
jp9000 已提交
4239 4240
		new_frame_num = conv_time_to_frames(
			sample_rate, timestamp - source->audio_ts);
J
jp9000 已提交
4241 4242 4243 4244 4245 4246 4247 4248 4249 4250 4251 4252 4253 4254 4255 4256 4257 4258 4259 4260 4261 4262 4263 4264 4265 4266 4267 4268 4269 4270

		if (new_frame_num >= AUDIO_OUTPUT_FRAMES)
			break;

		da_erase(source->audio_actions, i--);

		apply_audio_action(source, &action);

		if (new_frame_num > frame_num) {
			for (; frame_num < new_frame_num; frame_num++)
				vol_data[frame_num] = cur_vol;
		}

		cur_vol = get_source_volume(source, timestamp);
	}

	for (; frame_num < AUDIO_OUTPUT_FRAMES; frame_num++)
		vol_data[frame_num] = cur_vol;

	pthread_mutex_unlock(&source->audio_actions_mutex);

	for (size_t mix = 0; mix < MAX_AUDIO_MIXES; mix++) {
		if ((source->audio_mixers & (1 << mix)) != 0)
			multiply_vol_data(source, mix, channels, vol_data);
	}

	free(vol_data);
}

static void apply_audio_volume(obs_source_t *source, uint32_t mixers,
J
jp9000 已提交
4271
			       size_t channels, size_t sample_rate)
J
jp9000 已提交
4272 4273 4274 4275 4276 4277 4278 4279 4280 4281 4282 4283 4284 4285
{
	struct audio_action action;
	bool actions_pending;
	float vol;

	pthread_mutex_lock(&source->audio_actions_mutex);

	actions_pending = source->audio_actions.num > 0;
	if (actions_pending)
		action = source->audio_actions.array[0];

	pthread_mutex_unlock(&source->audio_actions_mutex);

	if (actions_pending) {
J
jp9000 已提交
4286 4287
		uint64_t duration =
			conv_frames_to_time(sample_rate, AUDIO_OUTPUT_FRAMES);
J
jp9000 已提交
4288 4289 4290 4291 4292 4293 4294 4295 4296 4297 4298 4299 4300

		if (action.timestamp < (source->audio_ts + duration)) {
			apply_audio_actions(source, channels, sample_rate);
			return;
		}
	}

	vol = get_source_volume(source, source->audio_ts);
	if (vol == 1.0f)
		return;

	if (vol == 0.0f || mixers == 0) {
		memset(source->audio_output_buf[0][0], 0,
J
jp9000 已提交
4301 4302
		       AUDIO_OUTPUT_FRAMES * sizeof(float) *
			       MAX_AUDIO_CHANNELS * MAX_AUDIO_MIXES);
J
jp9000 已提交
4303 4304 4305 4306 4307 4308 4309 4310 4311 4312 4313 4314
		return;
	}

	for (size_t mix = 0; mix < MAX_AUDIO_MIXES; mix++) {
		uint32_t mix_and_val = (1 << mix);
		if ((source->audio_mixers & mix_and_val) != 0 &&
		    (mixers & mix_and_val) != 0)
			multiply_output_audio(source, mix, channels, vol);
	}
}

static void custom_audio_render(obs_source_t *source, uint32_t mixers,
J
jp9000 已提交
4315
				size_t channels, size_t sample_rate)
J
jp9000 已提交
4316 4317 4318 4319 4320 4321
{
	struct obs_source_audio_mix audio_data;
	bool success;
	uint64_t ts;

	for (size_t mix = 0; mix < MAX_AUDIO_MIXES; mix++) {
4322
		for (size_t ch = 0; ch < channels; ch++) {
J
jp9000 已提交
4323 4324
			audio_data.output[mix].data[ch] =
				source->audio_output_buf[mix][ch];
4325
		}
J
jp9000 已提交
4326

4327 4328
		if ((source->audio_mixers & mixers & (1 << mix)) != 0) {
			memset(source->audio_output_buf[mix][0], 0,
J
jp9000 已提交
4329
			       sizeof(float) * AUDIO_OUTPUT_FRAMES * channels);
4330 4331
		}
	}
J
jp9000 已提交
4332 4333

	success = source->info.audio_render(source->context.data, &ts,
J
jp9000 已提交
4334 4335
					    &audio_data, mixers, channels,
					    sample_rate);
J
jp9000 已提交
4336 4337 4338 4339 4340 4341 4342
	source->audio_ts = success ? ts : 0;
	source->audio_pending = !success;

	if (!success || !source->audio_ts || !mixers)
		return;

	for (size_t mix = 0; mix < MAX_AUDIO_MIXES; mix++) {
4343 4344 4345 4346 4347 4348
		uint32_t mix_bit = 1 << mix;

		if ((mixers & mix_bit) == 0)
			continue;

		if ((source->audio_mixers & mix_bit) == 0) {
J
jp9000 已提交
4349
			memset(source->audio_output_buf[mix][0], 0,
J
jp9000 已提交
4350
			       sizeof(float) * AUDIO_OUTPUT_FRAMES * channels);
J
jp9000 已提交
4351 4352 4353 4354 4355 4356
		}
	}

	apply_audio_volume(source, mixers, channels, sample_rate);
}

J
jp9000 已提交
4357 4358 4359 4360 4361 4362 4363 4364 4365 4366 4367 4368 4369 4370 4371 4372 4373 4374 4375 4376 4377 4378 4379 4380 4381 4382 4383 4384 4385 4386 4387 4388 4389 4390 4391
static void audio_submix(obs_source_t *source, uint32_t mixers, size_t channels,
			 size_t sample_rate)
{
	struct audio_output_data audio_data;
	struct obs_source_audio audio = {0};
	bool success;
	uint64_t ts;

	for (size_t mix = 0; mix < MAX_AUDIO_MIXES; mix++) {
		for (size_t ch = 0; ch < channels; ch++) {
			audio_data.data[ch] = source->audio_mix_buf[ch];
		}

		memset(source->audio_mix_buf[0], 0,
		       sizeof(float) * AUDIO_OUTPUT_FRAMES * channels);
	}

	success = source->info.audio_mix(source->context.data, &ts, &audio_data,
					 channels, sample_rate);

	if (!success)
		return;

	for (size_t i = 0; i < channels; i++)
		audio.data[i] = (const uint8_t *)audio_data.data[i];

	audio.samples_per_sec = (uint32_t)sample_rate;
	audio.frames = AUDIO_OUTPUT_FRAMES;
	audio.format = AUDIO_FORMAT_FLOAT_PLANAR;
	audio.speakers = (enum speaker_layout)channels;
	audio.timestamp = ts;

	obs_source_output_audio(source, &audio);
}

J
jp9000 已提交
4392
static inline void process_audio_source_tick(obs_source_t *source,
J
jp9000 已提交
4393 4394
					     uint32_t mixers, size_t channels,
					     size_t sample_rate, size_t size)
J
jp9000 已提交
4395
{
J
jp9000 已提交
4396 4397
	bool audio_submix = !!(source->info.output_flags & OBS_SOURCE_SUBMIX);

J
jp9000 已提交
4398 4399
	pthread_mutex_lock(&source->audio_buf_mutex);

4400 4401 4402 4403 4404 4405
	if (source->audio_input_buf[0].size < size) {
		source->audio_pending = true;
		pthread_mutex_unlock(&source->audio_buf_mutex);
		return;
	}

J
jp9000 已提交
4406 4407
	for (size_t ch = 0; ch < channels; ch++)
		circlebuf_peek_front(&source->audio_input_buf[ch],
J
jp9000 已提交
4408
				     source->audio_output_buf[0][ch], size);
J
jp9000 已提交
4409 4410 4411 4412 4413 4414

	pthread_mutex_unlock(&source->audio_buf_mutex);

	for (size_t mix = 1; mix < MAX_AUDIO_MIXES; mix++) {
		uint32_t mix_and_val = (1 << mix);

J
jp9000 已提交
4415 4416 4417 4418 4419 4420 4421 4422
		if (audio_submix) {
			if (mix > 1)
				break;

			mixers = 1;
			mix_and_val = 1;
		}

J
jp9000 已提交
4423 4424
		if ((source->audio_mixers & mix_and_val) == 0 ||
		    (mixers & mix_and_val) == 0) {
J
jp9000 已提交
4425 4426
			memset(source->audio_output_buf[mix][0], 0,
			       size * channels);
J
jp9000 已提交
4427 4428 4429 4430 4431
			continue;
		}

		for (size_t ch = 0; ch < channels; ch++)
			memcpy(source->audio_output_buf[mix][ch],
J
jp9000 已提交
4432
			       source->audio_output_buf[0][ch], size);
J
jp9000 已提交
4433 4434
	}

J
jp9000 已提交
4435 4436 4437 4438 4439
	if (audio_submix) {
		source->audio_pending = false;
		return;
	}

J
jp9000 已提交
4440
	if ((source->audio_mixers & 1) == 0 || (mixers & 1) == 0)
J
jp9000 已提交
4441
		memset(source->audio_output_buf[0][0], 0, size * channels);
J
jp9000 已提交
4442 4443 4444 4445 4446 4447

	apply_audio_volume(source, mixers, channels, sample_rate);
	source->audio_pending = false;
}

void obs_source_audio_render(obs_source_t *source, uint32_t mixers,
J
jp9000 已提交
4448
			     size_t channels, size_t sample_rate, size_t size)
J
jp9000 已提交
4449
{
4450
	if (!source->audio_output_buf[0][0]) {
J
jp9000 已提交
4451 4452 4453 4454 4455 4456 4457 4458 4459
		source->audio_pending = true;
		return;
	}

	if (source->info.audio_render) {
		custom_audio_render(source, mixers, channels, sample_rate);
		return;
	}

J
jp9000 已提交
4460 4461 4462 4463
	if (source->info.audio_mix) {
		audio_submix(source, mixers, channels, sample_rate);
	}

4464
	if (!source->audio_ts) {
J
jp9000 已提交
4465 4466 4467 4468 4469 4470 4471 4472 4473
		source->audio_pending = true;
		return;
	}

	process_audio_source_tick(source, mixers, channels, sample_rate, size);
}

bool obs_source_audio_pending(const obs_source_t *source)
{
4474 4475 4476
	if (!obs_source_valid(source, "obs_source_audio_pending"))
		return true;

J
jp9000 已提交
4477 4478 4479
	return (is_composite_source(source) || is_audio_source(source))
		       ? source->audio_pending
		       : true;
J
jp9000 已提交
4480 4481
}

4482 4483
uint64_t obs_source_get_audio_timestamp(const obs_source_t *source)
{
J
jp9000 已提交
4484 4485 4486
	return obs_source_valid(source, "obs_source_get_audio_timestamp")
		       ? source->audio_ts
		       : 0;
4487 4488 4489
}

void obs_source_get_audio_mix(const obs_source_t *source,
J
jp9000 已提交
4490
			      struct obs_source_audio_mix *audio)
4491 4492 4493 4494 4495 4496 4497 4498 4499 4500 4501 4502 4503
{
	if (!obs_source_valid(source, "obs_source_get_audio_mix"))
		return;
	if (!obs_ptr_valid(audio, "audio"))
		return;

	for (size_t mix = 0; mix < MAX_AUDIO_MIXES; mix++) {
		for (size_t ch = 0; ch < MAX_AUDIO_CHANNELS; ch++) {
			audio->output[mix].data[ch] =
				source->audio_output_buf[mix][ch];
		}
	}
}
4504 4505

void obs_source_add_audio_capture_callback(obs_source_t *source,
J
jp9000 已提交
4506 4507
					   obs_source_audio_capture_t callback,
					   void *param)
4508 4509 4510 4511 4512 4513 4514 4515 4516 4517 4518
{
	struct audio_cb_info info = {callback, param};

	if (!obs_source_valid(source, "obs_source_add_audio_capture_callback"))
		return;

	pthread_mutex_lock(&source->audio_cb_mutex);
	da_push_back(source->audio_cb_list, &info);
	pthread_mutex_unlock(&source->audio_cb_mutex);
}

J
jp9000 已提交
4519 4520
void obs_source_remove_audio_capture_callback(
	obs_source_t *source, obs_source_audio_capture_t callback, void *param)
4521 4522 4523
{
	struct audio_cb_info info = {callback, param};

J
jp9000 已提交
4524 4525
	if (!obs_source_valid(source,
			      "obs_source_remove_audio_capture_callback"))
4526 4527 4528 4529 4530 4531
		return;

	pthread_mutex_lock(&source->audio_cb_mutex);
	da_erase_item(source->audio_cb_list, &info);
	pthread_mutex_unlock(&source->audio_cb_mutex);
}
J
jp9000 已提交
4532 4533

void obs_source_set_monitoring_type(obs_source_t *source,
J
jp9000 已提交
4534
				    enum obs_monitoring_type type)
J
jp9000 已提交
4535 4536 4537 4538 4539 4540 4541 4542 4543 4544 4545 4546 4547 4548 4549 4550 4551 4552 4553 4554 4555 4556 4557 4558
{
	bool was_on;
	bool now_on;

	if (!obs_source_valid(source, "obs_source_set_monitoring_type"))
		return;
	if (source->monitoring_type == type)
		return;

	was_on = source->monitoring_type != OBS_MONITORING_TYPE_NONE;
	now_on = type != OBS_MONITORING_TYPE_NONE;

	if (was_on != now_on) {
		if (!was_on) {
			source->monitor = audio_monitor_create(source);
		} else {
			audio_monitor_destroy(source->monitor);
			source->monitor = NULL;
		}
	}

	source->monitoring_type = type;
}

J
jp9000 已提交
4559 4560
enum obs_monitoring_type
obs_source_get_monitoring_type(const obs_source_t *source)
J
jp9000 已提交
4561
{
J
jp9000 已提交
4562 4563 4564
	return obs_source_valid(source, "obs_source_get_monitoring_type")
		       ? source->monitoring_type
		       : OBS_MONITORING_TYPE_NONE;
J
jp9000 已提交
4565
}
4566 4567 4568 4569 4570 4571 4572 4573 4574 4575 4576

void obs_source_set_async_unbuffered(obs_source_t *source, bool unbuffered)
{
	if (!obs_source_valid(source, "obs_source_set_async_unbuffered"))
		return;

	source->async_unbuffered = unbuffered;
}

bool obs_source_async_unbuffered(const obs_source_t *source)
{
J
jp9000 已提交
4577 4578 4579
	return obs_source_valid(source, "obs_source_async_unbuffered")
		       ? source->async_unbuffered
		       : false;
4580
}
4581 4582 4583 4584 4585 4586 4587 4588 4589

obs_data_t *obs_source_get_private_settings(obs_source_t *source)
{
	if (!obs_ptr_valid(source, "obs_source_get_private_settings"))
		return NULL;

	obs_data_addref(source->private_settings);
	return source->private_settings;
}
4590 4591 4592 4593 4594 4595 4596 4597 4598 4599 4600 4601 4602 4603 4604 4605 4606

void obs_source_set_async_decoupled(obs_source_t *source, bool decouple)
{
	if (!obs_ptr_valid(source, "obs_source_set_async_decoupled"))
		return;

	source->async_decoupled = decouple;
	if (decouple) {
		pthread_mutex_lock(&source->audio_buf_mutex);
		source->timing_set = false;
		reset_audio_data(source, 0);
		pthread_mutex_unlock(&source->audio_buf_mutex);
	}
}

bool obs_source_async_decoupled(const obs_source_t *source)
{
J
jp9000 已提交
4607 4608 4609
	return obs_source_valid(source, "obs_source_async_decoupled")
		       ? source->async_decoupled
		       : false;
4610
}
4611 4612 4613 4614 4615 4616 4617 4618 4619 4620 4621 4622 4623

/* hidden/undocumented export to allow source type redefinition for scripts */
EXPORT void obs_enable_source_type(const char *name, bool enable)
{
	struct obs_source_info *info = get_source_info(name);
	if (!info)
		return;

	if (enable)
		info->output_flags &= ~OBS_SOURCE_CAP_DISABLED;
	else
		info->output_flags |= OBS_SOURCE_CAP_DISABLED;
}
C
cg2121 已提交
4624 4625 4626 4627 4628 4629 4630 4631 4632 4633 4634 4635 4636 4637 4638 4639 4640 4641 4642

enum speaker_layout obs_source_get_speaker_layout(obs_source_t *source)
{
	if (!obs_source_valid(source, "obs_source_get_audio_channels"))
		return SPEAKERS_UNKNOWN;

	return source->sample_info.speakers;
}

void obs_source_set_balance_value(obs_source_t *source, float balance)
{
	if (!obs_source_valid(source, "obs_source_set_balance_value"))
		return;

	source->balance = balance;
}

float obs_source_get_balance_value(const obs_source_t *source)
{
J
jp9000 已提交
4643 4644 4645
	return obs_source_valid(source, "obs_source_get_balance_value")
		       ? source->balance
		       : 0.5f;
C
cg2121 已提交
4646
}