obs-source.c 103.1 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 19
#include <inttypes.h>

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

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

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

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

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

	return NULL;
}

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

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

91 92
	return signal_handler_add_array(source->context.signals,
			source_signals);
93 94
}

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

101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
static void allocate_audio_output_buffer(struct obs_source *source)
{
	size_t size = sizeof(float) *
		AUDIO_OUTPUT_FRAMES * MAX_AUDIO_CHANNELS * MAX_AUDIO_MIXES;
	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 已提交
117 118 119 120 121 122
static inline bool is_async_video_source(const struct obs_source *source)
{
	return (source->info.output_flags & OBS_SOURCE_ASYNC_VIDEO) ==
		OBS_SOURCE_ASYNC_VIDEO;
}

123 124 125 126 127 128 129 130 131 132
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 已提交
133 134
extern char *find_libobs_data_file(const char *file);

135
/* internal initialization */
136
bool obs_source_init(struct obs_source *source)
J
jp9000 已提交
137
{
J
jp9000 已提交
138 139
	pthread_mutexattr_t attr;

J
jp9000 已提交
140
	source->user_volume = 1.0f;
J
jp9000 已提交
141
	source->volume = 1.0f;
J
jp9000 已提交
142
	source->sync_offset = 0;
143
	pthread_mutex_init_value(&source->filter_mutex);
144
	pthread_mutex_init_value(&source->async_mutex);
145
	pthread_mutex_init_value(&source->audio_mutex);
146
	pthread_mutex_init_value(&source->audio_buf_mutex);
147
	pthread_mutex_init_value(&source->audio_cb_mutex);
148

J
jp9000 已提交
149 150 151 152 153
	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)
154
		return false;
155 156
	if (pthread_mutex_init(&source->audio_buf_mutex, NULL) != 0)
		return false;
J
jp9000 已提交
157 158
	if (pthread_mutex_init(&source->audio_actions_mutex, NULL) != 0)
		return false;
159 160
	if (pthread_mutex_init(&source->audio_cb_mutex, NULL) != 0)
		return false;
161 162
	if (pthread_mutex_init(&source->audio_mutex, NULL) != 0)
		return false;
163
	if (pthread_mutex_init(&source->async_mutex, NULL) != 0)
164
		return false;
J
jp9000 已提交
165

166
	if (is_audio_source(source) || is_composite_source(source))
167
		allocate_audio_output_buffer(source);
168

169 170 171 172 173 174 175 176
	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;
177
	source->audio_mixers = 0xFF;
178

179
	if (is_audio_source(source)) {
180 181 182 183 184 185 186 187 188 189 190
		pthread_mutex_lock(&obs->data.audio_sources_mutex);

		source->next_audio_source = obs->data.first_audio_source;
		source->prev_next_audio_source =
			&obs->data.first_audio_source;
		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);
191
	}
192

193 194 195
	obs_context_data_insert(&source->context,
			&obs->data.sources_mutex,
			&obs->data.first_source);
196
	return true;
J
jp9000 已提交
197 198
}

P
Palana 已提交
199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229
static bool obs_source_hotkey_mute(void *data,
		obs_hotkey_pair_id id, obs_hotkey_t *key, bool pressed)
{
	UNUSED_PARAMETER(id);
	UNUSED_PARAMETER(key);

	struct obs_source *source = data;

	if (!pressed || obs_source_muted(source)) return false;

	obs_source_set_muted(source, true);
	return true;
}

static bool obs_source_hotkey_unmute(void *data,
		obs_hotkey_pair_id id, obs_hotkey_t *key, bool pressed)
{
	UNUSED_PARAMETER(id);
	UNUSED_PARAMETER(key);

	struct obs_source *source = data;

	if (!pressed || !obs_source_muted(source)) return false;

	obs_source_set_muted(source, false);
	return true;
}

static void obs_source_hotkey_push_to_mute(void *data,
		obs_hotkey_id id, obs_hotkey_t *key, bool pressed)
{
J
jp9000 已提交
230 231 232 233 234 235
	struct audio_action action = {
		.timestamp = os_gettime_ns(),
		.type      = AUDIO_ACTION_PTM,
		.set       = pressed
	};

P
Palana 已提交
236 237 238 239 240
	UNUSED_PARAMETER(id);
	UNUSED_PARAMETER(key);

	struct obs_source *source = data;

J
jp9000 已提交
241 242 243 244 245
	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 已提交
246 247 248 249 250
}

static void obs_source_hotkey_push_to_talk(void *data,
		obs_hotkey_id id, obs_hotkey_t *key, bool pressed)
{
J
jp9000 已提交
251 252 253 254 255 256
	struct audio_action action = {
		.timestamp = os_gettime_ns(),
		.type      = AUDIO_ACTION_PTT,
		.set       = pressed
	};

P
Palana 已提交
257 258 259 260 261
	UNUSED_PARAMETER(id);
	UNUSED_PARAMETER(key);

	struct obs_source *source = data;

J
jp9000 已提交
262 263 264 265 266
	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 已提交
267 268 269 270
}

static void obs_source_init_audio_hotkeys(struct obs_source *source)
{
271 272
	if (!(source->info.output_flags & OBS_SOURCE_AUDIO) ||
	    source->info.type != OBS_SOURCE_TYPE_INPUT) {
P
Palana 已提交
273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292
		source->mute_unmute_key  = OBS_INVALID_HOTKEY_ID;
		source->push_to_talk_key = OBS_INVALID_HOTKEY_ID;
		return;
	}

	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);

	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);

	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);
}

J
jp9000 已提交
293 294 295
static obs_source_t *obs_source_create_internal(const char *id,
		const char *name, obs_data_t *settings,
		obs_data_t *hotkey_data, bool private)
J
jp9000 已提交
296
{
297
	struct obs_source *source = bzalloc(sizeof(struct obs_source));
J
jp9000 已提交
298

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

303 304 305 306
		source->info.id      = bstrdup(id);
		source->owns_info_id = true;
	} else {
		source->info = *info;
307 308 309 310 311 312 313

		/* 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)
			private = true;
314
	}
315

P
Palana 已提交
316 317 318 319
	source->mute_unmute_key  = OBS_INVALID_HOTKEY_PAIR_ID;
	source->push_to_mute_key = OBS_INVALID_HOTKEY_ID;
	source->push_to_talk_key = OBS_INVALID_HOTKEY_ID;

J
jp9000 已提交
320 321
	if (!obs_source_init_context(source, settings, name, hotkey_data,
				private))
322 323
		goto fail;

324 325
	if (info && info->get_defaults)
		info->get_defaults(source->context.settings);
J
jp9000 已提交
326

327
	if (!obs_source_init(source))
328 329
		goto fail;

J
jp9000 已提交
330 331
	if (!private)
		obs_source_init_audio_hotkeys(source);
P
Palana 已提交
332

333 334
	/* allow the source to be created even if creation fails so that the
	 * user's data doesn't become lost */
335 336 337
	if (info)
		source->context.data = info->create(source->context.settings,
				source);
338
	if (!source->context.data)
339
		blog(LOG_ERROR, "Failed to create source '%s'!", name);
340

341
	blog(LOG_DEBUG, "%ssource '%s' (%s) created",
J
jp9000 已提交
342
			private ? "private " : "", name, id);
343
	obs_source_dosignal(source, "source_create", NULL);
J
jp9000 已提交
344

345
	source->flags = source->default_flags;
346
	source->enabled = true;
J
jp9000 已提交
347
	return source;
348 349 350 351 352

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

J
jp9000 已提交
355 356 357 358 359 360 361 362 363 364 365 366 367
obs_source_t *obs_source_create(const char *id, const char *name,
		obs_data_t *settings, obs_data_t *hotkey_data)
{
	return obs_source_create_internal(id, name, settings, hotkey_data,
			false);
}

obs_source_t *obs_source_create_private(const char *id, const char *name,
		obs_data_t *settings)
{
	return obs_source_create_internal(id, name, settings, NULL, true);
}

368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407
static void duplicate_filters(obs_source_t *dst, obs_source_t *src,
		bool private)
{
	DARRAY(obs_source_t*) filters;

	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];
		obs_source_t *dst_filter = obs_source_duplicate(src_filter,
				src_filter->context.name, private);

		obs_source_filter_add(dst, dst_filter);
		obs_source_release(dst_filter);
		obs_source_release(src_filter);
	}

	da_free(filters);
}

obs_source_t *obs_source_duplicate(obs_source_t *source,
		const char *new_name, bool create_private)
{
	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;
	}

408 409 410 411 412
	if (source->info.type == OBS_SOURCE_TYPE_SCENE) {
		obs_scene_t *scene = obs_scene_from_source(source);
		obs_scene_t *new_scene = obs_scene_duplicate(scene, new_name,
				create_private ? OBS_SCENE_DUP_PRIVATE_COPY :
					OBS_SCENE_DUP_COPY);
413 414 415
		obs_source_t *new_source = obs_scene_get_source(new_scene);
		duplicate_filters(new_source, source, create_private);
		return new_source;
416 417
	}

418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439
	settings = obs_data_create();
	obs_data_apply(settings, source->context.settings);

	new_source = create_private ?
		obs_source_create_private(source->info.id, new_name, settings) :
		obs_source_create(source->info.id, new_name, settings, NULL);

	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;

	if (source->info.type != OBS_SOURCE_TYPE_FILTER)
		duplicate_filters(new_source, source, create_private);

	obs_data_release(settings);
	return new_source;
}

440 441
void obs_source_frame_init(struct obs_source_frame *frame,
		enum video_format format, uint32_t width, uint32_t height)
442
{
443
	struct video_frame vid_frame;
J
jp9000 已提交
444

445
	if (!obs_ptr_valid(frame, "obs_source_frame_init"))
J
jp9000 已提交
446 447
		return;

448
	video_frame_init(&vid_frame, format, width, height);
449 450 451
	frame->format = format;
	frame->width  = width;
	frame->height = height;
452

453 454 455
	for (size_t i = 0; i < MAX_AV_PLANES; i++) {
		frame->data[i]     = vid_frame.data[i];
		frame->linesize[i] = vid_frame.linesize[i];
456 457 458
	}
}

459 460 461 462 463 464
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);
}

465 466 467
static bool obs_source_filter_remove_refless(obs_source_t *source,
		obs_source_t *filter);

468
void obs_source_destroy(struct obs_source *source)
J
jp9000 已提交
469
{
470
	size_t i;
471

472
	if (!obs_source_valid(source, "obs_source_destroy"))
J
jp9000 已提交
473 474
		return;

J
jp9000 已提交
475 476 477
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
		obs_transition_clear(source);

478 479 480 481 482 483 484 485 486
	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);

487 488 489 490 491 492
	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]);

493 494
	obs_context_data_remove(&source->context);

495
	blog(LOG_DEBUG, "%ssource '%s' destroyed",
J
jp9000 已提交
496 497
			source->context.private ? "private " : "",
			source->context.name);
498

499
	obs_source_dosignal(source, "source_destroy", "destroy");
500

501
	if (source->context.data) {
502
		source->info.destroy(source->context.data);
503 504
		source->context.data = NULL;
	}
505

P
Palana 已提交
506 507 508 509
	obs_hotkey_unregister(source->push_to_talk_key);
	obs_hotkey_unregister(source->push_to_mute_key);
	obs_hotkey_pair_unregister(source->mute_unmute_key);

510
	for (i = 0; i < source->async_cache.num; i++)
511
		obs_source_frame_decref(source->async_cache.array[i].frame);
512

513
	gs_enter_context(obs->video.graphics);
514 515
	if (source->async_texrender)
		gs_texrender_destroy(source->async_texrender);
J
jp9000 已提交
516 517
	if (source->async_prev_texrender)
		gs_texrender_destroy(source->async_prev_texrender);
518 519
	if (source->async_texture)
		gs_texture_destroy(source->async_texture);
J
jp9000 已提交
520 521
	if (source->async_prev_texture)
		gs_texture_destroy(source->async_prev_texture);
522 523
	if (source->filter_texrender)
		gs_texrender_destroy(source->filter_texrender);
524
	gs_leave_context();
J
jp9000 已提交
525

J
jp9000 已提交
526
	for (i = 0; i < MAX_AV_PLANES; i++)
527
		bfree(source->audio_data.data[i]);
528 529
	for (i = 0; i < MAX_AUDIO_CHANNELS; i++)
		circlebuf_free(&source->audio_input_buf[i]);
530
	audio_resampler_destroy(source->resampler);
531
	bfree(source->audio_output_buf[0][0]);
532

J
jp9000 已提交
533 534 535
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
		obs_transition_free(source);

J
jp9000 已提交
536
	da_free(source->audio_actions);
537
	da_free(source->audio_cb_list);
538
	da_free(source->async_cache);
539
	da_free(source->async_frames);
540 541
	da_free(source->filters);
	pthread_mutex_destroy(&source->filter_mutex);
J
jp9000 已提交
542
	pthread_mutex_destroy(&source->audio_actions_mutex);
543
	pthread_mutex_destroy(&source->audio_buf_mutex);
544
	pthread_mutex_destroy(&source->audio_cb_mutex);
545
	pthread_mutex_destroy(&source->audio_mutex);
546
	pthread_mutex_destroy(&source->async_mutex);
547
	obs_context_data_free(&source->context);
548

549 550 551
	if (source->owns_info_id)
		bfree((void*)source->info.id);

552 553 554
	bfree(source);
}

555
void obs_source_addref(obs_source_t *source)
556
{
557 558 559 560
	if (!source)
		return;

	obs_ref_addref(&source->control->ref);
561 562
}

563
void obs_source_release(obs_source_t *source)
564
{
565 566 567 568 569 570
	if (!obs) {
		blog(LOG_WARNING, "Tried to release a source when the OBS "
		                  "core is shut down!");
		return;
	}

P
Palana 已提交
571 572
	if (!source)
		return;
573

574 575
	obs_weak_source_t *control = source->control;
	if (obs_ref_release(&control->ref)) {
P
Palana 已提交
576
		obs_source_destroy(source);
577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630
		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,
		obs_source_t *source)
{
	return weak && source && weak->source == source;
631 632
}

633
void obs_source_remove(obs_source_t *source)
634
{
635 636 637
	if (!obs_source_valid(source, "obs_source_remove"))
		return;

638 639
	if (!source->removed) {
		source->removed = true;
640
		obs_source_dosignal(source, "source_remove", "remove");
641
	}
642 643
}

644
bool obs_source_removed(const obs_source_t *source)
645
{
646 647
	return obs_source_valid(source, "obs_source_removed") ?
		source->removed : true;
J
jp9000 已提交
648 649
}

650
static inline obs_data_t *get_defaults(const struct obs_source_info *info)
J
jp9000 已提交
651
{
652
	obs_data_t *settings = obs_data_create();
653 654
	if (info->get_defaults)
		info->get_defaults(settings);
J
jp9000 已提交
655 656 657
	return settings;
}

658
obs_data_t *obs_source_settings(const char *id)
J
jp9000 已提交
659
{
660
	const struct obs_source_info *info = get_source_info(id);
J
jp9000 已提交
661
	return (info) ? get_defaults(info) : NULL;
J
jp9000 已提交
662 663
}

664
obs_data_t *obs_get_source_defaults(const char *id)
J
jp9000 已提交
665
{
666
	const struct obs_source_info *info = get_source_info(id);
J
jp9000 已提交
667 668 669
	return info ? get_defaults(info) : NULL;
}

670
obs_properties_t *obs_get_source_properties(const char *id)
J
jp9000 已提交
671
{
672
	const struct obs_source_info *info = get_source_info(id);
673
	if (info && info->get_properties) {
674 675
		obs_data_t       *defaults = get_defaults(info);
		obs_properties_t *properties;
J
jp9000 已提交
676

677
		properties = info->get_properties(NULL);
J
jp9000 已提交
678 679 680 681
		obs_properties_apply_settings(properties, defaults);
		obs_data_release(defaults);
		return properties;
	}
J
jp9000 已提交
682 683 684
	return NULL;
}

685 686 687 688 689 690 691 692 693 694 695 696
bool obs_is_source_configurable(const char *id)
{
	const struct obs_source_info *info = get_source_info(id);
	return info && info->get_properties;
}

bool obs_source_configurable(const obs_source_t *source)
{
	return data_valid(source, "obs_source_configurable") &&
		source->info.get_properties;
}

697
obs_properties_t *obs_source_properties(const obs_source_t *source)
698
{
J
jp9000 已提交
699 700 701 702
	if (!data_valid(source, "obs_source_properties"))
		return NULL;

	if (source->info.get_properties) {
703
		obs_properties_t *props;
704
		props = source->info.get_properties(source->context.data);
705
		obs_properties_apply_settings(props, source->context.settings);
J
jp9000 已提交
706 707 708
		return props;
	}

709 710 711
	return NULL;
}

712
uint32_t obs_source_get_output_flags(const obs_source_t *source)
J
jp9000 已提交
713
{
714 715
	return obs_source_valid(source, "obs_source_get_output_flags") ?
		source->info.output_flags : 0;
J
jp9000 已提交
716 717
}

718
uint32_t obs_get_source_output_flags(const char *id)
719
{
720
	const struct obs_source_info *info = get_source_info(id);
721 722 723
	return info ? info->output_flags : 0;
}

724
static void obs_source_deferred_update(obs_source_t *source)
725
{
726 727 728 729
	if (source->context.data && source->info.update)
		source->info.update(source->context.data,
				source->context.settings);

730 731 732
	source->defer_update = false;
}

733
void obs_source_update(obs_source_t *source, obs_data_t *settings)
J
jp9000 已提交
734
{
735 736
	if (!obs_source_valid(source, "obs_source_update"))
		return;
J
jp9000 已提交
737

738 739 740 741 742 743 744 745
	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,
				source->context.settings);
746
	}
J
jp9000 已提交
747 748
}

749 750
void obs_source_update_properties(obs_source_t *source)
{
751 752
	if (!obs_source_valid(source, "obs_source_update_properties"))
		return;
753

754
	obs_source_dosignal(source, NULL, "update_properties");
755 756
}

757
void obs_source_send_mouse_click(obs_source_t *source,
K
kc5nra 已提交
758 759 760 761
		const struct obs_mouse_event *event,
		int32_t type, bool mouse_up,
		uint32_t click_count)
{
762
	if (!obs_source_valid(source, "obs_source_send_mouse_click"))
K
kc5nra 已提交
763 764 765 766 767 768 769 770 771 772
		return;

	if (source->info.output_flags & OBS_SOURCE_INTERACTION) {
		if (source->info.mouse_click) {
			source->info.mouse_click(source->context.data,
					event, type, mouse_up, click_count);
		}
	}
}

773
void obs_source_send_mouse_move(obs_source_t *source,
K
kc5nra 已提交
774 775
		const struct obs_mouse_event *event, bool mouse_leave)
{
776
	if (!obs_source_valid(source, "obs_source_send_mouse_move"))
K
kc5nra 已提交
777 778 779 780 781 782 783 784 785 786
		return;

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

787
void obs_source_send_mouse_wheel(obs_source_t *source,
K
kc5nra 已提交
788 789
		const struct obs_mouse_event *event, int x_delta, int y_delta)
{
790
	if (!obs_source_valid(source, "obs_source_send_mouse_wheel"))
K
kc5nra 已提交
791 792 793 794 795 796 797 798 799 800
		return;

	if (source->info.output_flags & OBS_SOURCE_INTERACTION) {
		if (source->info.mouse_wheel) {
			source->info.mouse_wheel(source->context.data,
					event, x_delta, y_delta);
		}
	}
}

801
void obs_source_send_focus(obs_source_t *source, bool focus)
K
kc5nra 已提交
802
{
803
	if (!obs_source_valid(source, "obs_source_send_focus"))
K
kc5nra 已提交
804 805 806 807 808 809 810 811 812
		return;

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

813
void obs_source_send_key_click(obs_source_t *source,
K
kc5nra 已提交
814 815
		const struct obs_key_event *event, bool key_up)
{
816
	if (!obs_source_valid(source, "obs_source_send_key_click"))
K
kc5nra 已提交
817 818 819 820 821 822 823 824 825 826
		return;

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

827
static void activate_source(obs_source_t *source)
J
jp9000 已提交
828
{
829
	if (source->context.data && source->info.activate)
830
		source->info.activate(source->context.data);
831
	obs_source_dosignal(source, "source_activate", "activate");
J
jp9000 已提交
832 833
}

834
static void deactivate_source(obs_source_t *source)
J
jp9000 已提交
835
{
836
	if (source->context.data && source->info.deactivate)
837
		source->info.deactivate(source->context.data);
838
	obs_source_dosignal(source, "source_deactivate", "deactivate");
839
}
840

841
static void show_source(obs_source_t *source)
842
{
843
	if (source->context.data && source->info.show)
844
		source->info.show(source->context.data);
845
	obs_source_dosignal(source, "source_show", "show");
846 847
}

848
static void hide_source(obs_source_t *source)
849
{
850
	if (source->context.data && source->info.hide)
851
		source->info.hide(source->context.data);
852
	obs_source_dosignal(source, "source_hide", "hide");
853 854
}

855 856
static void activate_tree(obs_source_t *parent, obs_source_t *child,
		void *param)
857
{
858
	os_atomic_inc_long(&child->activate_refs);
J
jp9000 已提交
859 860 861

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
862 863
}

864
static void deactivate_tree(obs_source_t *parent, obs_source_t *child,
865 866
		void *param)
{
867
	os_atomic_dec_long(&child->activate_refs);
J
jp9000 已提交
868 869 870

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
871 872
}

873
static void show_tree(obs_source_t *parent, obs_source_t *child, void *param)
874
{
875
	os_atomic_inc_long(&child->show_refs);
876 877 878 879 880

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
}

881
static void hide_tree(obs_source_t *parent, obs_source_t *child, void *param)
882
{
883
	os_atomic_dec_long(&child->show_refs);
884 885 886 887 888

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
}

889
void obs_source_activate(obs_source_t *source, enum view_type type)
890
{
891 892
	if (!obs_source_valid(source, "obs_source_activate"))
		return;
893

894 895
	os_atomic_inc_long(&source->show_refs);
	obs_source_enum_active_tree(source, show_tree, NULL);
896 897

	if (type == MAIN_VIEW) {
898 899
		os_atomic_inc_long(&source->activate_refs);
		obs_source_enum_active_tree(source, activate_tree, NULL);
900 901 902
	}
}

903
void obs_source_deactivate(obs_source_t *source, enum view_type type)
904
{
905 906
	if (!obs_source_valid(source, "obs_source_deactivate"))
		return;
907

908 909
	if (os_atomic_load_long(&source->show_refs) > 0) {
		os_atomic_dec_long(&source->show_refs);
910
		obs_source_enum_active_tree(source, hide_tree, NULL);
911 912 913
	}

	if (type == MAIN_VIEW) {
914 915
		if (os_atomic_load_long(&source->activate_refs) > 0) {
			os_atomic_dec_long(&source->activate_refs);
916 917
			obs_source_enum_active_tree(source, deactivate_tree,
					NULL);
918
		}
919
	}
J
jp9000 已提交
920 921
}

922 923
static inline struct obs_source_frame *get_closest_frame(obs_source_t *source,
		uint64_t sys_time);
924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952
bool set_async_texture_size(struct obs_source *source,
		const struct obs_source_frame *frame);

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) {
			remove_async_frame(source,
					source->cur_async_frame);
			source->cur_async_frame = NULL;
		}

		source->cur_async_frame = get_closest_frame(source,
				sys_time);
	}

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

	if (source->cur_async_frame)
		source->async_update_texture = set_async_texture_size(source,
				source->cur_async_frame);
}
953

954
void obs_source_video_tick(obs_source_t *source, float seconds)
J
jp9000 已提交
955
{
956 957
	bool now_showing, now_active;

958 959
	if (!obs_source_valid(source, "obs_source_video_tick"))
		return;
J
jp9000 已提交
960

J
jp9000 已提交
961 962 963
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
		obs_transition_tick(source);

964 965
	if ((source->info.output_flags & OBS_SOURCE_ASYNC) != 0)
		async_tick(source);
966

967 968 969
	if (source->defer_update)
		obs_source_deferred_update(source);

J
jp9000 已提交
970 971
	/* reset the filter render texture information once every frame */
	if (source->filter_texrender)
972
		gs_texrender_reset(source->filter_texrender);
J
jp9000 已提交
973

974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997
	/* 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;
	}

998
	if (source->context.data && source->info.video_tick)
999
		source->info.video_tick(source->context.data, seconds);
1000 1001

	source->async_rendered = false;
J
jp9000 已提交
1002
	source->deinterlace_rendered = false;
J
jp9000 已提交
1003 1004
}

1005
/* unless the value is 3+ hours worth of frames, this won't overflow */
1006 1007
static inline uint64_t conv_frames_to_time(const size_t sample_rate,
		const size_t frames)
1008
{
1009
	return (uint64_t)frames * 1000000000ULL / (uint64_t)sample_rate;
1010 1011
}

J
jp9000 已提交
1012 1013 1014 1015 1016 1017 1018 1019
static inline size_t conv_time_to_frames(const size_t sample_rate,
		const uint64_t duration)
{
	return (size_t)(duration * (uint64_t)sample_rate / 1000000000ULL);
}

/* maximum buffer size */
#define MAX_BUF_SIZE        (1000 * AUDIO_OUTPUT_FRAMES * sizeof(float))
1020

1021 1022
static inline void reset_audio_timing(obs_source_t *source, uint64_t timestamp,
		uint64_t os_time)
1023 1024
{
	source->timing_set    = true;
1025
	source->timing_adjust = os_time - timestamp;
1026
}
1027

1028 1029 1030 1031 1032 1033 1034 1035
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,
					source->audio_input_buf[i].size);
	}

1036
	source->last_audio_input_buf_size = 0;
1037 1038 1039 1040
	source->audio_ts = os_time;
}

static void handle_ts_jump(obs_source_t *source, uint64_t expected,
1041
		uint64_t ts, uint64_t diff, uint64_t os_time)
1042
{
J
jp9000 已提交
1043
	blog(LOG_DEBUG, "Timestamp for source '%s' jumped by '%"PRIu64"', "
1044
	                "expected value %"PRIu64", input value %"PRIu64,
1045
	                source->context.name, diff, expected, ts);
1046

1047
	pthread_mutex_lock(&source->audio_buf_mutex);
1048
	reset_audio_timing(source, ts, os_time);
1049
	pthread_mutex_unlock(&source->audio_buf_mutex);
1050 1051
}

1052
static void source_signal_audio_data(obs_source_t *source,
1053
		struct audio_data *in, bool muted)
1054
{
1055
	pthread_mutex_lock(&source->audio_cb_mutex);
1056

1057 1058 1059 1060
	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);
	}
1061

1062
	pthread_mutex_unlock(&source->audio_cb_mutex);
1063 1064
}

1065 1066 1067 1068 1069
static inline uint64_t uint64_diff(uint64_t ts1, uint64_t ts2)
{
	return (ts1 < ts2) ?  (ts2 - ts1) : (ts1 - ts2);
}

1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098
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,
		const struct audio_data *in)
{
	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);

	buf_placement = get_buf_placement(audio,
			in->timestamp - source->audio_ts) * sizeof(float);

#if DEBUG_AUDIO == 1
	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);
#endif

J
jp9000 已提交
1099 1100 1101 1102 1103
	/* 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++) {
1104 1105
		circlebuf_place(&source->audio_input_buf[i], buf_placement,
				in->data[i], size);
J
jp9000 已提交
1106 1107 1108 1109
		circlebuf_pop_back(&source->audio_input_buf[i], NULL,
				source->audio_input_buf[i].size -
				(buf_placement + size));
	}
1110 1111

	source->last_audio_input_buf_size = 0;
1112 1113 1114 1115 1116 1117 1118
}

static inline void source_output_audio_push_back(obs_source_t *source,
		const struct audio_data *in)
{
	audio_t *audio = obs->audio.audio;
	size_t channels = audio_output_get_channels(audio);
J
jp9000 已提交
1119 1120 1121 1122 1123
	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;
1124 1125 1126

	for (size_t i = 0; i < channels; i++)
		circlebuf_push_back(&source->audio_input_buf[i],
J
jp9000 已提交
1127
				in->data[i], size);
1128 1129 1130 1131

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

1134 1135
static inline bool source_muted(obs_source_t *source, uint64_t os_time)
{
J
jp9000 已提交
1136
	if (source->push_to_mute_enabled && source->user_push_to_mute_pressed)
1137 1138 1139
		source->push_to_mute_stop_time = os_time +
			source->push_to_mute_delay * 1000000;

J
jp9000 已提交
1140
	if (source->push_to_talk_enabled && source->user_push_to_talk_pressed)
1141 1142 1143
		source->push_to_talk_stop_time = os_time +
			source->push_to_talk_delay * 1000000;

J
jp9000 已提交
1144
	bool push_to_mute_active = source->user_push_to_mute_pressed ||
1145
		os_time < source->push_to_mute_stop_time;
J
jp9000 已提交
1146
	bool push_to_talk_active = source->user_push_to_talk_pressed ||
1147 1148
		os_time < source->push_to_talk_stop_time;

J
jp9000 已提交
1149
	return !source->enabled || source->user_muted ||
1150 1151 1152 1153
			(source->push_to_mute_enabled && push_to_mute_active) ||
			(source->push_to_talk_enabled && !push_to_talk_active);
}

1154
static void source_output_audio_data(obs_source_t *source,
1155 1156
		const struct audio_data *data)
{
1157
	size_t sample_rate = audio_output_get_sample_rate(obs->audio.audio);
1158
	struct audio_data in = *data;
1159
	uint64_t diff;
1160
	uint64_t os_time = os_gettime_ns();
1161
	int64_t sync_offset;
1162 1163
	bool using_direct_ts = false;
	bool push_back = false;
1164

1165 1166 1167 1168 1169
	/* 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;
1170 1171
		using_direct_ts = true;
	}
1172

1173
	if (!source->timing_set) {
1174
		reset_audio_timing(source, in.timestamp, os_time);
1175

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

1179
		/* smooth audio if within threshold */
1180
		if (diff > MAX_TS_VAR && !using_direct_ts)
1181
			handle_ts_jump(source, source->next_audio_ts_min,
1182
					in.timestamp, diff, os_time);
1183
		else if (diff < TS_SMOOTHING_THRESHOLD)
1184
			in.timestamp = source->next_audio_ts_min;
1185 1186
	}

1187
	source->next_audio_ts_min = in.timestamp +
1188
		conv_frames_to_time(sample_rate, in.frames);
1189

1190
	in.timestamp += source->timing_adjust;
1191

1192 1193
	pthread_mutex_lock(&source->audio_buf_mutex);

J
jp9000 已提交
1194
	if (source->next_audio_sys_ts_min == in.timestamp) {
1195
		push_back = true;
1196 1197

	} else if (source->next_audio_sys_ts_min) {
J
jp9000 已提交
1198
		diff = uint64_diff(source->next_audio_sys_ts_min, in.timestamp);
1199 1200

		if (diff < TS_SMOOTHING_THRESHOLD) {
1201 1202 1203 1204 1205 1206 1207 1208
			push_back = true;

		/* This typically only happens if used with async video when
		 * 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. */
1209
		} else if (diff > MAX_TS_VAR) {
1210
			reset_audio_timing(source, data->timestamp,
1211
					os_time);
1212 1213
			in.timestamp = data->timestamp + source->timing_adjust;
		}
J
jp9000 已提交
1214 1215
	}

1216 1217
	sync_offset = source->sync_offset;
	in.timestamp += sync_offset;
1218 1219
	in.timestamp -= source->resample_offset;

1220
	source->next_audio_sys_ts_min = source->next_audio_ts_min +
1221 1222 1223 1224 1225 1226 1227
		source->timing_adjust;

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

1229
	if (push_back && source->audio_ts)
1230 1231 1232 1233 1234 1235
		source_output_audio_push_back(source, &in);
	else
		source_output_audio_place(source, &in);

	pthread_mutex_unlock(&source->audio_buf_mutex);

1236
	source_signal_audio_data(source, &in, source_muted(source, os_time));
1237 1238
}

1239 1240 1241 1242 1243 1244 1245 1246
enum convert_type {
	CONVERT_NONE,
	CONVERT_NV12,
	CONVERT_420,
	CONVERT_422_U,
	CONVERT_422_Y,
};

1247
static inline enum convert_type get_convert_type(enum video_format format)
1248
{
1249
	switch (format) {
1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260
	case VIDEO_FORMAT_I420:
		return CONVERT_420;
	case VIDEO_FORMAT_NV12:
		return CONVERT_NV12;

	case VIDEO_FORMAT_YVYU:
	case VIDEO_FORMAT_YUY2:
		return CONVERT_422_Y;
	case VIDEO_FORMAT_UYVY:
		return CONVERT_422_U;

J
jpk 已提交
1261
	case VIDEO_FORMAT_Y800:
J
jp9000 已提交
1262
	case VIDEO_FORMAT_I444:
1263
	case VIDEO_FORMAT_NONE:
1264 1265 1266 1267 1268 1269 1270 1271 1272
	case VIDEO_FORMAT_RGBA:
	case VIDEO_FORMAT_BGRA:
	case VIDEO_FORMAT_BGRX:
		return CONVERT_NONE;
	}

	return CONVERT_NONE;
}

1273
static inline bool set_packed422_sizes(struct obs_source *source,
1274
		const struct obs_source_frame *frame)
1275 1276
{
	source->async_convert_height = frame->height;
1277 1278 1279 1280 1281 1282
	source->async_convert_width  = frame->width / 2;
	source->async_texture_format = GS_BGRA;
	return true;
}

static inline bool set_planar420_sizes(struct obs_source *source,
1283
		const struct obs_source_frame *frame)
1284 1285 1286 1287 1288
{
	uint32_t size = frame->width * frame->height;
	size += size/2;

	source->async_convert_width   = frame->width;
1289
	source->async_convert_height  = size / frame->width;
1290
	source->async_texture_format  = GS_R8;
1291 1292
	source->async_plane_offset[0] = (int)(frame->data[1] - frame->data[0]);
	source->async_plane_offset[1] = (int)(frame->data[2] - frame->data[0]);
1293 1294 1295
	return true;
}

J
jp9000 已提交
1296
static inline bool set_nv12_sizes(struct obs_source *source,
1297
		const struct obs_source_frame *frame)
J
jp9000 已提交
1298 1299 1300 1301 1302
{
	uint32_t size = frame->width * frame->height;
	size += size/2;

	source->async_convert_width   = frame->width;
1303
	source->async_convert_height  = size / frame->width;
J
jp9000 已提交
1304
	source->async_texture_format  = GS_R8;
1305
	source->async_plane_offset[0] = (int)(frame->data[1] - frame->data[0]);
J
jp9000 已提交
1306 1307 1308
	return true;
}

1309
static inline bool init_gpu_conversion(struct obs_source *source,
1310
		const struct obs_source_frame *frame)
1311 1312 1313 1314 1315 1316 1317
{
	switch (get_convert_type(frame->format)) {
		case CONVERT_422_Y:
		case CONVERT_422_U:
			return set_packed422_sizes(source, frame);

		case CONVERT_420:
1318 1319 1320
			return set_planar420_sizes(source, frame);

		case CONVERT_NV12:
J
jp9000 已提交
1321
			return set_nv12_sizes(source, frame);
1322 1323 1324 1325 1326 1327 1328 1329 1330 1331
			break;

		case CONVERT_NONE:
			assert(false && "No conversion requested");
			break;

	}
	return false;
}

1332
bool set_async_texture_size(struct obs_source *source,
1333
		const struct obs_source_frame *frame)
1334
{
1335 1336
	enum convert_type cur = get_convert_type(frame->format);

J
jp9000 已提交
1337 1338 1339
	if (source->async_width  == frame->width  &&
	    source->async_height == frame->height &&
	    source->async_format == frame->format)
1340 1341
		return true;

J
jp9000 已提交
1342 1343 1344
	source->async_width  = frame->width;
	source->async_height = frame->height;
	source->async_format = frame->format;
1345

1346 1347
	gs_enter_context(obs->video.graphics);

1348
	gs_texture_destroy(source->async_texture);
J
jp9000 已提交
1349
	gs_texture_destroy(source->async_prev_texture);
1350
	gs_texrender_destroy(source->async_texrender);
J
jp9000 已提交
1351 1352 1353
	gs_texrender_destroy(source->async_prev_texrender);
	source->async_texture = NULL;
	source->async_prev_texture = NULL;
1354
	source->async_texrender = NULL;
J
jp9000 已提交
1355
	source->async_prev_texrender = NULL;
1356 1357 1358 1359

	if (cur != CONVERT_NONE && init_gpu_conversion(source, frame)) {
		source->async_gpu_conversion = true;

1360
		source->async_texrender =
1361
			gs_texrender_create(GS_BGRX, GS_ZS_NONE);
1362

1363
		source->async_texture = gs_texture_create(
1364 1365
				source->async_convert_width,
				source->async_convert_height,
1366 1367
				source->async_texture_format,
				1, NULL, GS_DYNAMIC);
1368 1369

	} else {
1370 1371
		enum gs_color_format format = convert_video_format(
				frame->format);
1372 1373
		source->async_gpu_conversion = false;

1374
		source->async_texture = gs_texture_create(
1375
				frame->width, frame->height,
1376
				format, 1, NULL, GS_DYNAMIC);
1377 1378
	}

J
jp9000 已提交
1379 1380 1381
	if (deinterlacing_enabled(source))
		set_deinterlace_texture_size(source);

1382 1383
	gs_leave_context();

1384
	return !!source->async_texture;
1385 1386
}

1387
static void upload_raw_frame(gs_texture_t *tex,
1388
		const struct obs_source_frame *frame)
1389 1390 1391 1392
{
	switch (get_convert_type(frame->format)) {
		case CONVERT_422_U:
		case CONVERT_422_Y:
1393
			gs_texture_set_image(tex, frame->data[0],
1394 1395 1396 1397
					frame->linesize[0], false);
			break;

		case CONVERT_420:
1398
			gs_texture_set_image(tex, frame->data[0],
1399 1400 1401 1402
					frame->width, false);
			break;

		case CONVERT_NV12:
J
jp9000 已提交
1403 1404
			gs_texture_set_image(tex, frame->data[0],
					frame->width, false);
1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416
			break;

		case CONVERT_NONE:
			assert(false && "No conversion requested");
			break;
	}
}

static const char *select_conversion_technique(enum video_format format)
{
	switch (format) {
		case VIDEO_FORMAT_UYVY:
1417
			return "UYVY_Reverse";
1418 1419 1420 1421 1422 1423 1424 1425

		case VIDEO_FORMAT_YUY2:
			return "YUY2_Reverse";

		case VIDEO_FORMAT_YVYU:
			return "YVYU_Reverse";

		case VIDEO_FORMAT_I420:
1426 1427 1428
			return "I420_Reverse";

		case VIDEO_FORMAT_NV12:
J
jp9000 已提交
1429
			return "NV12_Reverse";
1430 1431
			break;

J
jpk 已提交
1432
		case VIDEO_FORMAT_Y800:
1433 1434 1435 1436
		case VIDEO_FORMAT_BGRA:
		case VIDEO_FORMAT_BGRX:
		case VIDEO_FORMAT_RGBA:
		case VIDEO_FORMAT_NONE:
J
jp9000 已提交
1437
		case VIDEO_FORMAT_I444:
1438 1439 1440 1441 1442 1443
			assert(false && "No conversion requested");
			break;
	}
	return NULL;
}

1444
static inline void set_eparam(gs_effect_t *effect, const char *name, float val)
1445
{
1446
	gs_eparam_t *param = gs_effect_get_param_by_name(effect, name);
1447
	gs_effect_set_float(param, val);
1448 1449 1450
}

static bool update_async_texrender(struct obs_source *source,
1451 1452
		const struct obs_source_frame *frame,
		gs_texture_t *tex, gs_texrender_t *texrender)
1453
{
1454
	gs_texrender_reset(texrender);
1455 1456 1457 1458 1459 1460

	upload_raw_frame(tex, frame);

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

1461 1462 1463
	float convert_width  = (float)source->async_convert_width;
	float convert_height = (float)source->async_convert_height;

1464 1465
	gs_effect_t *conv = obs->video.conversion_effect;
	gs_technique_t *tech = gs_effect_get_technique(conv,
1466 1467
			select_conversion_technique(frame->format));

1468
	if (!gs_texrender_begin(texrender, cx, cy))
1469 1470
		return false;

1471 1472
	gs_technique_begin(tech);
	gs_technique_begin_pass(tech, 0);
1473

1474
	gs_effect_set_texture(gs_effect_get_param_by_name(conv, "image"), tex);
1475 1476 1477 1478
	set_eparam(conv, "width",  (float)cx);
	set_eparam(conv, "height", (float)cy);
	set_eparam(conv, "width_i",  1.0f / cx);
	set_eparam(conv, "height_i", 1.0f / cy);
1479
	set_eparam(conv, "width_d2",  cx * 0.5f);
1480
	set_eparam(conv, "height_d2", cy * 0.5f);
1481
	set_eparam(conv, "width_d2_i",  1.0f / (cx * 0.5f));
1482
	set_eparam(conv, "height_d2_i", 1.0f / (cy * 0.5f));
1483 1484 1485 1486 1487 1488 1489 1490 1491 1492
	set_eparam(conv, "input_width",  convert_width);
	set_eparam(conv, "input_height", convert_height);
	set_eparam(conv, "input_width_i",  1.0f / convert_width);
	set_eparam(conv, "input_height_i", 1.0f / convert_height);
	set_eparam(conv, "input_width_i_d2",  (1.0f / convert_width)  * 0.5f);
	set_eparam(conv, "input_height_i_d2", (1.0f / convert_height) * 0.5f);
	set_eparam(conv, "u_plane_offset",
			(float)source->async_plane_offset[0]);
	set_eparam(conv, "v_plane_offset",
			(float)source->async_plane_offset[1]);
1493 1494 1495 1496 1497

	gs_ortho(0.f, (float)cx, 0.f, (float)cy, -100.f, 100.f);

	gs_draw_sprite(tex, 0, cx, cy);

1498 1499
	gs_technique_end_pass(tech);
	gs_technique_end(tech);
1500

1501
	gs_texrender_end(texrender);
1502 1503 1504 1505

	return true;
}

1506 1507 1508
bool update_async_texture(struct obs_source *source,
		const struct obs_source_frame *frame,
		gs_texture_t *tex, gs_texrender_t *texrender)
1509
{
1510
	enum convert_type type      = get_convert_type(frame->format);
1511
	uint8_t           *ptr;
1512 1513
	uint32_t          linesize;

1514 1515
	source->async_flip       = frame->flip;
	source->async_full_range = frame->full_range;
1516 1517
	memcpy(source->async_color_matrix, frame->color_matrix,
			sizeof(frame->color_matrix));
1518 1519 1520 1521
	memcpy(source->async_color_range_min, frame->color_range_min,
			sizeof frame->color_range_min);
	memcpy(source->async_color_range_max, frame->color_range_max,
			sizeof frame->color_range_max);
1522

1523
	if (source->async_gpu_conversion && texrender)
1524
		return update_async_texrender(source, frame, tex, texrender);
1525

1526
	if (type == CONVERT_NONE) {
1527
		gs_texture_set_image(tex, frame->data[0], frame->linesize[0],
1528
				false);
1529 1530 1531
		return true;
	}

1532
	if (!gs_texture_map(tex, &ptr, &linesize))
1533 1534 1535
		return false;

	if (type == CONVERT_420)
J
jp9000 已提交
1536 1537 1538
		decompress_420((const uint8_t* const*)frame->data,
				frame->linesize,
				0, frame->height, ptr, linesize);
1539 1540

	else if (type == CONVERT_NV12)
J
jp9000 已提交
1541 1542 1543
		decompress_nv12((const uint8_t* const*)frame->data,
				frame->linesize,
				0, frame->height, ptr, linesize);
1544 1545

	else if (type == CONVERT_422_Y)
1546
		decompress_422(frame->data[0], frame->linesize[0],
J
jp9000 已提交
1547
				0, frame->height, ptr, linesize, true);
1548 1549

	else if (type == CONVERT_422_U)
1550
		decompress_422(frame->data[0], frame->linesize[0],
J
jp9000 已提交
1551
				0, frame->height, ptr, linesize, false);
1552

1553
	gs_texture_unmap(tex);
1554 1555 1556
	return true;
}

1557
static inline void obs_source_draw_texture(struct obs_source *source,
1558
		gs_effect_t *effect, float *color_matrix,
1559
		float const *color_range_min, float const *color_range_max)
1560
{
1561 1562
	gs_texture_t *tex = source->async_texture;
	gs_eparam_t  *param;
1563

1564 1565
	if (source->async_texrender)
		tex = gs_texrender_get_texture(source->async_texrender);
1566

P
Palana 已提交
1567
	if (color_range_min) {
1568
		size_t const size = sizeof(float) * 3;
1569 1570
		param = gs_effect_get_param_by_name(effect, "color_range_min");
		gs_effect_set_val(param, color_range_min, size);
P
Palana 已提交
1571
	}
1572

P
Palana 已提交
1573 1574
	if (color_range_max) {
		size_t const size = sizeof(float) * 3;
1575 1576
		param = gs_effect_get_param_by_name(effect, "color_range_max");
		gs_effect_set_val(param, color_range_max, size);
P
Palana 已提交
1577
	}
1578

P
Palana 已提交
1579
	if (color_matrix) {
1580 1581
		param = gs_effect_get_param_by_name(effect, "color_matrix");
		gs_effect_set_val(param, color_matrix, sizeof(float) * 16);
1582 1583
	}

1584 1585
	param = gs_effect_get_param_by_name(effect, "image");
	gs_effect_set_texture(param, tex);
1586

1587 1588
	gs_draw_sprite(tex, source->async_flip ? GS_FLIP_V : 0, 0, 0);
}
1589

1590 1591
static void obs_source_draw_async_texture(struct obs_source *source)
{
1592
	gs_effect_t    *effect        = gs_get_effect();
1593 1594 1595 1596
	bool           yuv           = format_is_yuv(source->async_format);
	bool           limited_range = yuv && !source->async_full_range;
	const char     *type         = yuv ? "DrawMatrix" : "Draw";
	bool           def_draw      = (!effect);
1597
	gs_technique_t *tech          = NULL;
1598 1599

	if (def_draw) {
1600
		effect = obs_get_base_effect(OBS_EFFECT_DEFAULT);
1601 1602 1603
		tech = gs_effect_get_technique(effect, type);
		gs_technique_begin(tech);
		gs_technique_begin_pass(tech, 0);
1604 1605 1606
	}

	obs_source_draw_texture(source, effect,
1607 1608 1609
			yuv ? source->async_color_matrix : NULL,
			limited_range ? source->async_color_range_min : NULL,
			limited_range ? source->async_color_range_max : NULL);
1610 1611

	if (def_draw) {
1612 1613
		gs_technique_end_pass(tech);
		gs_technique_end(tech);
1614
	}
1615 1616
}

1617
static void obs_source_update_async_video(obs_source_t *source)
1618
{
1619 1620 1621
	if (!source->async_rendered) {
		struct obs_source_frame *frame = obs_source_get_frame(source);

1622 1623 1624
		if (frame)
			frame = filter_async_video(source, frame);

1625 1626
		source->async_rendered = true;
		if (frame) {
1627 1628 1629 1630
			source->timing_adjust =
				os_gettime_ns() - frame->timestamp;
			source->timing_set = true;

1631
			if (source->async_update_texture) {
1632 1633
				update_async_texture(source, frame,
						source->async_texture,
1634
						source->async_texrender);
1635
				source->async_update_texture = false;
1636
			}
1637

1638 1639
			obs_source_release_frame(source, frame);
		}
1640
	}
1641
}
1642

1643 1644
static inline void obs_source_render_async_video(obs_source_t *source)
{
1645
	if (source->async_texture && source->async_active)
1646
		obs_source_draw_async_texture(source);
1647 1648
}

1649
static inline void obs_source_render_filters(obs_source_t *source)
1650 1651 1652 1653 1654 1655
{
	source->rendering_filter = true;
	obs_source_video_render(source->filters.array[0]);
	source->rendering_filter = false;
}

1656
static void obs_source_default_render(obs_source_t *source)
1657
{
1658
	gs_effect_t    *effect     = obs->video.default_effect;
1659
	gs_technique_t *tech       = gs_effect_get_technique(effect, "Draw");
1660
	size_t         passes, i;
1661

1662
	passes = gs_technique_begin(tech);
1663
	for (i = 0; i < passes; i++) {
1664
		gs_technique_begin_pass(tech, i);
1665 1666
		if (source->context.data)
			source->info.video_render(source->context.data, effect);
1667
		gs_technique_end_pass(tech);
1668
	}
1669
	gs_technique_end(tech);
1670 1671
}

1672
static inline void obs_source_main_render(obs_source_t *source)
1673
{
1674 1675
	uint32_t flags      = source->info.output_flags;
	bool custom_draw    = (flags & OBS_SOURCE_CUSTOM_DRAW) != 0;
1676 1677
	bool default_effect = !source->filter_parent &&
	                      source->filters.num == 0 &&
1678
	                      !custom_draw;
1679 1680

	if (default_effect)
1681
		obs_source_default_render(source);
1682
	else if (source->context.data)
1683
		source->info.video_render(source->context.data,
1684
				custom_draw ? NULL : gs_get_effect());
1685 1686
}

1687 1688
static bool ready_async_frame(obs_source_t *source, uint64_t sys_time);

1689
static inline void render_video(obs_source_t *source)
J
jp9000 已提交
1690
{
1691 1692
	if (source->info.type != OBS_SOURCE_TYPE_FILTER &&
	    (source->info.output_flags & OBS_SOURCE_VIDEO) == 0)
1693 1694
		return;

1695 1696
	if (source->info.type == OBS_SOURCE_TYPE_INPUT &&
	    (source->info.output_flags & OBS_SOURCE_ASYNC) != 0 &&
J
jp9000 已提交
1697 1698 1699
	    !source->rendering_filter) {
		if (deinterlacing_enabled(source))
			deinterlace_update_async_video(source);
1700
		obs_source_update_async_video(source);
J
jp9000 已提交
1701
	}
1702

1703
	if (!source->context.data || !source->enabled) {
1704 1705 1706 1707 1708
		if (source->filter_parent)
			obs_source_skip_video_filter(source);
		return;
	}

1709 1710
	if (source->filters.num && !source->rendering_filter)
		obs_source_render_filters(source);
1711

1712 1713 1714 1715
	else if (source->info.video_render)
		obs_source_main_render(source);

	else if (source->filter_target)
1716 1717
		obs_source_video_render(source->filter_target);

J
jp9000 已提交
1718 1719 1720
	else if (deinterlacing_enabled(source))
		deinterlace_render(source);

1721
	else
1722
		obs_source_render_async_video(source);
J
jp9000 已提交
1723 1724
}

1725 1726 1727 1728 1729 1730 1731 1732 1733 1734
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);
}

1735
static uint32_t get_base_width(const obs_source_t *source)
J
jp9000 已提交
1736
{
1737 1738
	bool is_filter = (source->info.type == OBS_SOURCE_TYPE_FILTER);

J
jp9000 已提交
1739 1740 1741 1742
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION) {
		return source->enabled ? source->transition_actual_cx : 0;

	} else if (source->info.get_width && (!is_filter || source->enabled)) {
1743
		return source->info.get_width(source->context.data);
1744 1745 1746 1747 1748

	} else if (source->info.type == OBS_SOURCE_TYPE_FILTER) {
		return get_base_width(source->filter_target);
	}

1749
	return source->async_active ? source->async_width : 0;
J
jp9000 已提交
1750 1751
}

1752
static uint32_t get_base_height(const obs_source_t *source)
J
jp9000 已提交
1753
{
1754 1755
	bool is_filter = (source->info.type == OBS_SOURCE_TYPE_FILTER);

J
jp9000 已提交
1756 1757 1758 1759
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION) {
		return source->enabled ? source->transition_actual_cy : 0;

	} else if (source->info.get_height && (!is_filter || source->enabled)) {
1760
		return source->info.get_height(source->context.data);
1761

1762
	} else if (is_filter) {
1763 1764 1765
		return get_base_height(source->filter_target);
	}

1766
	return source->async_active ? source->async_height : 0;
J
jp9000 已提交
1767 1768
}

1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800
static uint32_t get_recurse_width(obs_source_t *source)
{
	uint32_t width;

	pthread_mutex_lock(&source->filter_mutex);

	width = (source->filters.num) ?
		get_base_width(source->filters.array[0]) :
		get_base_width(source);

	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);

	height = (source->filters.num) ?
		get_base_height(source->filters.array[0]) :
		get_base_height(source);

	pthread_mutex_unlock(&source->filter_mutex);

	return height;
}

uint32_t obs_source_get_width(obs_source_t *source)
{
J
jp9000 已提交
1801 1802
	if (!data_valid(source, "obs_source_get_width"))
		return 0;
1803

1804
	return (source->info.type != OBS_SOURCE_TYPE_FILTER) ?
1805 1806 1807 1808 1809 1810
		get_recurse_width(source) :
	        get_base_width(source);
}

uint32_t obs_source_get_height(obs_source_t *source)
{
J
jp9000 已提交
1811 1812
	if (!data_valid(source, "obs_source_get_height"))
		return 0;
1813

1814
	return (source->info.type != OBS_SOURCE_TYPE_FILTER) ?
1815 1816 1817 1818
		get_recurse_height(source) :
		get_base_height(source);
}

1819 1820
uint32_t obs_source_get_base_width(obs_source_t *source)
{
J
jp9000 已提交
1821 1822
	if (!data_valid(source, "obs_source_get_base_width"))
		return 0;
1823 1824 1825 1826 1827 1828

	return get_base_width(source);
}

uint32_t obs_source_get_base_height(obs_source_t *source)
{
J
jp9000 已提交
1829 1830
	if (!data_valid(source, "obs_source_get_base_height"))
		return 0;
1831 1832 1833 1834

	return get_base_height(source);
}

1835
obs_source_t *obs_filter_get_parent(const obs_source_t *filter)
1836
{
1837 1838
	return obs_ptr_valid(filter, "obs_filter_get_parent") ?
		filter->filter_parent : NULL;
1839 1840
}

1841
obs_source_t *obs_filter_get_target(const obs_source_t *filter)
J
jp9000 已提交
1842
{
1843 1844
	return obs_ptr_valid(filter, "obs_filter_get_target") ?
		filter->filter_target : NULL;
J
jp9000 已提交
1845 1846
}

1847
void obs_source_filter_add(obs_source_t *source, obs_source_t *filter)
J
jp9000 已提交
1848
{
1849 1850
	struct calldata cd;
	uint8_t stack[128];
J
jp9000 已提交
1851

1852 1853 1854
	if (!obs_source_valid(source, "obs_source_filter_add"))
		return;
	if (!obs_ptr_valid(filter, "obs_source_filter_add"))
J
jp9000 已提交
1855 1856
		return;

1857 1858
	pthread_mutex_lock(&source->filter_mutex);

J
jp9000 已提交
1859
	if (da_find(source->filters, &filter, 0) != DARRAY_INVALID) {
J
jp9000 已提交
1860 1861
		blog(LOG_WARNING, "Tried to add a filter that was already "
		                  "present on the source");
1862
		pthread_mutex_unlock(&source->filter_mutex);
J
jp9000 已提交
1863 1864 1865
		return;
	}

1866 1867
	obs_source_addref(filter);

1868
	filter->filter_parent = source;
J
jp9000 已提交
1869 1870
	filter->filter_target = !source->filters.num ?
		source : source->filters.array[0];
1871

J
jp9000 已提交
1872
	da_insert(source->filters, 0, &filter);
1873 1874 1875

	pthread_mutex_unlock(&source->filter_mutex);

1876
	calldata_init_fixed(&cd, stack, sizeof(stack));
J
jp9000 已提交
1877 1878 1879 1880
	calldata_set_ptr(&cd, "source", source);
	calldata_set_ptr(&cd, "filter", filter);

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

	if (source && filter)
1883
		blog(LOG_DEBUG, "- filter '%s' (%s) added to source '%s'",
1884 1885
				filter->context.name, filter->info.id,
				source->context.name);
J
jp9000 已提交
1886 1887
}

1888 1889
static bool obs_source_filter_remove_refless(obs_source_t *source,
		obs_source_t *filter)
J
jp9000 已提交
1890
{
1891 1892
	struct calldata cd;
	uint8_t stack[128];
1893 1894 1895 1896 1897
	size_t idx;

	pthread_mutex_lock(&source->filter_mutex);

	idx = da_find(source->filters, &filter, 0);
1898 1899
	if (idx == DARRAY_INVALID) {
		pthread_mutex_unlock(&source->filter_mutex);
1900
		return false;
1901
	}
J
jp9000 已提交
1902 1903

	if (idx > 0) {
1904
		obs_source_t *prev = source->filters.array[idx-1];
J
jp9000 已提交
1905 1906 1907 1908
		prev->filter_target = filter->filter_target;
	}

	da_erase(source->filters, idx);
1909 1910 1911

	pthread_mutex_unlock(&source->filter_mutex);

1912
	calldata_init_fixed(&cd, stack, sizeof(stack));
J
jp9000 已提交
1913 1914 1915 1916 1917
	calldata_set_ptr(&cd, "source", source);
	calldata_set_ptr(&cd, "filter", filter);

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

1918
	if (source && filter)
1919
		blog(LOG_DEBUG, "- filter '%s' (%s) removed from source '%s'",
1920 1921 1922
				filter->context.name, filter->info.id,
				source->context.name);

1923 1924 1925 1926
	if (filter->info.filter_remove)
		filter->info.filter_remove(filter->context.data,
				filter->filter_parent);

1927
	filter->filter_parent = NULL;
J
jp9000 已提交
1928
	filter->filter_target = NULL;
1929 1930
	return true;
}
1931

1932 1933
void obs_source_filter_remove(obs_source_t *source, obs_source_t *filter)
{
1934 1935 1936 1937 1938
	if (!obs_source_valid(source, "obs_source_filter_remove"))
		return;
	if (!obs_ptr_valid(filter, "obs_source_filter_remove"))
		return;

1939 1940
	if (obs_source_filter_remove_refless(source, filter))
		obs_source_release(filter);
J
jp9000 已提交
1941 1942
}

1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981
static size_t find_next_filter(obs_source_t *source, obs_source_t *filter,
		size_t cur_idx)
{
	bool curAsync = (filter->info.output_flags & OBS_SOURCE_ASYNC) != 0;
	bool nextAsync;
	obs_source_t *next;

	if (cur_idx == source->filters.num-1)
		return DARRAY_INVALID;

	next = source->filters.array[cur_idx+1];
	nextAsync = (next->info.output_flags & OBS_SOURCE_ASYNC);

	if (nextAsync == curAsync)
		return cur_idx+1;
	else
		return find_next_filter(source, filter, cur_idx+1);
}

static size_t find_prev_filter(obs_source_t *source, obs_source_t *filter,
		size_t cur_idx)
{
	bool curAsync = (filter->info.output_flags & OBS_SOURCE_ASYNC) != 0;
	bool prevAsync;
	obs_source_t *prev;

	if (cur_idx == 0)
		return DARRAY_INVALID;

	prev = source->filters.array[cur_idx-1];
	prevAsync = (prev->info.output_flags & OBS_SOURCE_ASYNC);

	if (prevAsync == curAsync)
		return cur_idx-1;
	else
		return find_prev_filter(source, filter, cur_idx-1);
}

/* moves filters above/below matching filter types */
1982 1983
static bool move_filter_dir(obs_source_t *source,
		obs_source_t *filter, enum obs_order_movement movement)
J
jp9000 已提交
1984
{
1985
	size_t idx;
J
jp9000 已提交
1986 1987

	idx = da_find(source->filters, &filter, 0);
J
jp9000 已提交
1988
	if (idx == DARRAY_INVALID)
1989
		return false;
J
jp9000 已提交
1990

J
jp9000 已提交
1991
	if (movement == OBS_ORDER_MOVE_UP) {
1992 1993
		size_t next_id = find_next_filter(source, filter, idx);
		if (next_id == DARRAY_INVALID)
1994
			return false;
1995
		da_move_item(source->filters, idx, next_id);
J
jp9000 已提交
1996

J
jp9000 已提交
1997
	} else if (movement == OBS_ORDER_MOVE_DOWN) {
1998 1999
		size_t prev_id = find_prev_filter(source, filter, idx);
		if (prev_id == DARRAY_INVALID)
2000
			return false;
2001
		da_move_item(source->filters, idx, prev_id);
J
jp9000 已提交
2002

J
jp9000 已提交
2003
	} else if (movement == OBS_ORDER_MOVE_TOP) {
J
jp9000 已提交
2004
		if (idx == source->filters.num-1)
2005
			return false;
J
jp9000 已提交
2006 2007
		da_move_item(source->filters, idx, source->filters.num-1);

J
jp9000 已提交
2008
	} else if (movement == OBS_ORDER_MOVE_BOTTOM) {
J
jp9000 已提交
2009
		if (idx == 0)
2010
			return false;
J
jp9000 已提交
2011 2012 2013
		da_move_item(source->filters, idx, 0);
	}

2014
	/* reorder filter targets, not the nicest way of dealing with things */
2015
	for (size_t i = 0; i < source->filters.num; i++) {
2016
		obs_source_t *next_filter = (i == source->filters.num-1) ?
2017 2018
			source : source->filters.array[i + 1];

J
jp9000 已提交
2019 2020
		source->filters.array[i]->filter_target = next_filter;
	}
2021

2022 2023 2024 2025 2026 2027 2028
	return true;
}

void obs_source_filter_set_order(obs_source_t *source, obs_source_t *filter,
		enum obs_order_movement movement)
{
	bool success;
2029 2030 2031 2032

	if (!obs_source_valid(source, "obs_source_filter_set_order"))
		return;
	if (!obs_ptr_valid(filter, "obs_source_filter_set_order"))
2033 2034 2035 2036 2037 2038 2039 2040
		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 已提交
2041 2042
}

2043
obs_data_t *obs_source_get_settings(const obs_source_t *source)
J
jp9000 已提交
2044
{
2045 2046
	if (!obs_source_valid(source, "obs_source_get_settings"))
		return NULL;
J
jp9000 已提交
2047

2048 2049
	obs_data_addref(source->context.settings);
	return source->context.settings;
J
jp9000 已提交
2050 2051
}

2052
struct obs_source_frame *filter_async_video(obs_source_t *source,
2053
		struct obs_source_frame *in)
2054 2055
{
	size_t i;
2056 2057 2058

	pthread_mutex_lock(&source->filter_mutex);

2059 2060
	for (i = source->filters.num; i > 0; i--) {
		struct obs_source *filter = source->filters.array[i-1];
2061

2062 2063 2064
		if (!filter->enabled)
			continue;

2065
		if (filter->context.data && filter->info.filter_video) {
2066 2067
			in = filter->info.filter_video(filter->context.data,
					in);
2068
			if (!in)
2069
				break;
2070 2071 2072
		}
	}

2073 2074
	pthread_mutex_unlock(&source->filter_mutex);

2075 2076 2077
	return in;
}

2078 2079
static inline void copy_frame_data_line(struct obs_source_frame *dst,
		const struct obs_source_frame *src, uint32_t plane, uint32_t y)
2080
{
2081 2082 2083 2084
	uint32_t pos_src = y * src->linesize[plane];
	uint32_t pos_dst = y * dst->linesize[plane];
	uint32_t bytes = dst->linesize[plane] < src->linesize[plane] ?
		dst->linesize[plane] : src->linesize[plane];
2085 2086 2087 2088

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

2089 2090 2091
static inline void copy_frame_data_plane(struct obs_source_frame *dst,
		const struct obs_source_frame *src,
		uint32_t plane, uint32_t lines)
2092
{
2093
	if (dst->linesize[plane] != src->linesize[plane])
2094 2095 2096 2097
		for (uint32_t y = 0; y < lines; y++)
			copy_frame_data_line(dst, src, plane, y);
	else
		memcpy(dst->data[plane], src->data[plane],
2098
				dst->linesize[plane] * lines);
2099 2100
}

2101 2102 2103 2104 2105 2106 2107 2108 2109 2110
static void copy_frame_data_line_y800(uint32_t *dst, uint8_t *src, uint8_t *end)
{
	while (src < end) {
		register uint32_t val = *(src++);
		val |= (val << 8);
		val |= (val << 16);
		*(dst++) = val;
	}
}

J
jp9000 已提交
2111 2112 2113
static inline void copy_frame_data_y800(struct obs_source_frame *dst,
		const struct obs_source_frame *src)
{
2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132
	uint32_t *ptr_dst;
	uint8_t  *ptr_src;
	uint8_t  *src_end;

	if ((src->linesize[0] * 4) != dst->linesize[0]) {
		for (uint32_t cy = 0; cy < src->height; cy++) {
			ptr_dst = (uint32_t*)
				(dst->data[0] + cy * dst->linesize[0]);
			ptr_src = (src->data[0] + cy * src->linesize[0]);
			src_end = ptr_src + src->width;

			copy_frame_data_line_y800(ptr_dst, ptr_src, src_end);
		}
	} else {
		ptr_dst = (uint32_t*)dst->data[0];
		ptr_src = (uint8_t *)src->data[0];
		src_end = ptr_src + src->height * src->linesize[0];

		copy_frame_data_line_y800(ptr_dst, ptr_src, src_end);
J
jp9000 已提交
2133 2134 2135
	}
}

2136 2137
static void copy_frame_data(struct obs_source_frame *dst,
		const struct obs_source_frame *src)
2138 2139
{
	dst->flip         = src->flip;
2140
	dst->full_range   = src->full_range;
2141 2142
	dst->timestamp    = src->timestamp;
	memcpy(dst->color_matrix, src->color_matrix, sizeof(float) * 16);
2143 2144 2145 2146 2147
	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);
	}
2148

J
jp9000 已提交
2149
	switch (src->format) {
2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160
	case VIDEO_FORMAT_I420:
		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);
		break;

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

J
jp9000 已提交
2161 2162 2163 2164 2165 2166
	case VIDEO_FORMAT_I444:
		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;

2167 2168 2169 2170 2171 2172 2173 2174
	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:
		copy_frame_data_plane(dst, src, 0, dst->height);
2175
		break;
2176 2177 2178 2179

	case VIDEO_FORMAT_Y800:
		copy_frame_data_y800(dst, src);
		break;
2180 2181 2182
	}
}

2183 2184 2185 2186
static inline bool async_texture_changed(struct obs_source *source,
		const struct obs_source_frame *frame)
{
	enum convert_type prev, cur;
J
jp9000 已提交
2187
	prev = get_convert_type(source->async_cache_format);
2188 2189
	cur  = get_convert_type(frame->format);

J
jp9000 已提交
2190 2191
	return source->async_cache_width  != frame->width ||
	       source->async_cache_height != frame->height ||
2192 2193 2194 2195 2196 2197
	       prev != cur;
}

static inline void free_async_cache(struct obs_source *source)
{
	for (size_t i = 0; i < source->async_cache.num; i++)
2198
		obs_source_frame_decref(source->async_cache.array[i].frame);
2199 2200 2201

	da_resize(source->async_cache, 0);
	da_resize(source->async_frames, 0);
2202
	source->cur_async_frame = NULL;
J
jp9000 已提交
2203
	source->prev_async_frame = NULL;
2204 2205
}

2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222
#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);
			}
		}
	}
}

2223 2224
#define MAX_ASYNC_FRAMES 30

2225
static inline struct obs_source_frame *cache_video(struct obs_source *source,
2226
		const struct obs_source_frame *frame)
2227
{
2228 2229 2230 2231
	struct obs_source_frame *new_frame = NULL;

	pthread_mutex_lock(&source->async_mutex);

2232 2233 2234 2235 2236 2237 2238
	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;
	}

2239
	if (async_texture_changed(source, frame)) {
J
jp9000 已提交
2240 2241 2242 2243
		free_async_cache(source);
		source->async_cache_width  = frame->width;
		source->async_cache_height = frame->height;
		source->async_cache_format = frame->format;
2244 2245 2246 2247 2248 2249 2250
	}

	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;
			af->used = true;
2251
			af->unused_count = 0;
2252
			break;
2253 2254 2255
		}
	}

2256 2257
	clean_cache(source);

2258 2259
	if (!new_frame) {
		struct async_frame new_af;
2260
		enum video_format format = frame->format;
J
jp9000 已提交
2261

2262
		if (format == VIDEO_FORMAT_Y800)
J
jp9000 已提交
2263
			format = VIDEO_FORMAT_BGRX;
2264

J
jp9000 已提交
2265
		new_frame = obs_source_frame_create(format,
2266 2267 2268
				frame->width, frame->height);
		new_af.frame = new_frame;
		new_af.used = true;
2269
		new_af.unused_count = 0;
2270
		new_frame->refs = 1;
2271 2272 2273 2274

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

2275 2276
	os_atomic_inc_long(&new_frame->refs);

2277
	pthread_mutex_unlock(&source->async_mutex);
2278

2279
	copy_frame_data(new_frame, frame);
2280 2281 2282 2283 2284 2285

	if (os_atomic_dec_long(&new_frame->refs) == 0) {
		obs_source_frame_destroy(new_frame);
		new_frame = NULL;
	}

2286
	return new_frame;
2287 2288
}

2289
void obs_source_output_video(obs_source_t *source,
2290
		const struct obs_source_frame *frame)
2291
{
2292
	if (!obs_source_valid(source, "obs_source_output_video"))
J
jp9000 已提交
2293 2294
		return;

2295 2296 2297 2298 2299
	if (!frame) {
		source->async_active = false;
		return;
	}

2300 2301
	struct obs_source_frame *output = !!frame ?
		cache_video(source, frame) : NULL;
2302

2303 2304
	/* ------------------------------------------- */

2305
	if (output) {
2306 2307 2308
		pthread_mutex_lock(&source->async_mutex);
		da_push_back(source->async_frames, &output);
		pthread_mutex_unlock(&source->async_mutex);
2309
		source->async_active = true;
2310
	}
2311 2312
}

2313
static inline struct obs_audio_data *filter_async_audio(obs_source_t *source,
2314
		struct obs_audio_data *in)
2315 2316 2317 2318
{
	size_t i;
	for (i = source->filters.num; i > 0; i--) {
		struct obs_source *filter = source->filters.array[i-1];
2319

2320 2321 2322
		if (!filter->enabled)
			continue;

2323
		if (filter->context.data && filter->info.filter_audio) {
2324 2325
			in = filter->info.filter_audio(filter->context.data,
					in);
2326 2327 2328 2329 2330 2331 2332 2333
			if (!in)
				return NULL;
		}
	}

	return in;
}

2334
static inline void reset_resampler(obs_source_t *source,
2335
		const struct obs_source_audio *audio)
2336
{
J
jp9000 已提交
2337
	const struct audio_output_info *obs_info;
2338 2339
	struct resample_info output_info;

2340
	obs_info = audio_output_get_info(obs->audio.audio);
2341

2342 2343 2344 2345 2346 2347 2348 2349
	output_info.format           = obs_info->format;
	output_info.samples_per_sec  = obs_info->samples_per_sec;
	output_info.speakers         = obs_info->speakers;

	source->sample_info.format          = audio->format;
	source->sample_info.samples_per_sec = audio->samples_per_sec;
	source->sample_info.speakers        = audio->speakers;

2350 2351
	audio_resampler_destroy(source->resampler);
	source->resampler = NULL;
2352
	source->resample_offset = 0;
2353

2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368
	if (source->sample_info.samples_per_sec == obs_info->samples_per_sec &&
	    source->sample_info.format          == obs_info->format          &&
	    source->sample_info.speakers        == obs_info->speakers) {
		source->audio_failed = false;
		return;
	}

	source->resampler = audio_resampler_create(&output_info,
			&source->sample_info);

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

J
jp9000 已提交
2369
static void copy_audio_data(obs_source_t *source,
J
jp9000 已提交
2370
		const uint8_t *const data[], uint32_t frames, uint64_t ts)
2371
{
2372 2373
	size_t planes    = audio_output_get_planes(obs->audio.audio);
	size_t blocksize = audio_output_get_block_size(obs->audio.audio);
2374 2375
	size_t size      = (size_t)frames * blocksize;
	bool   resize    = source->audio_storage_size < size;
2376

J
jp9000 已提交
2377 2378
	source->audio_data.frames    = frames;
	source->audio_data.timestamp = ts;
2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391

	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;
2392 2393
}

2394 2395 2396
/* TODO: SSE optimization */
static void downmix_to_mono_planar(struct obs_source *source, uint32_t frames)
{
J
jp9000 已提交
2397
	size_t channels = audio_output_get_channels(obs->audio.audio);
2398 2399 2400
	const float channels_i = 1.0f / (float)channels;
	float **data = (float**)source->audio_data.data;

J
jp9000 已提交
2401
	for (size_t channel = 1; channel < channels; channel++) {
2402 2403 2404 2405 2406 2407 2408
		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 已提交
2409
	for (size_t channel = 1; channel < channels; channel++) {
2410 2411 2412 2413 2414
		for (uint32_t frame = 0; frame < frames; frame++)
			data[channel][frame] = data[0][frame];
	}
}

2415
/* resamples/remixes new audio to the designated main audio output format */
2416
static void process_audio(obs_source_t *source,
2417
		const struct obs_source_audio *audio)
2418
{
2419
	uint32_t frames = audio->frames;
2420
	bool mono_output;
2421

2422 2423 2424 2425 2426 2427 2428 2429 2430
	if (source->sample_info.samples_per_sec != audio->samples_per_sec ||
	    source->sample_info.format          != audio->format          ||
	    source->sample_info.speakers        != audio->speakers)
		reset_resampler(source, audio);

	if (source->audio_failed)
		return;

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

2433 2434 2435
		memset(output, 0, sizeof(output));

		audio_resampler_resample(source->resampler,
2436
				output, &frames, &source->resample_offset,
2437
				audio->data, audio->frames);
2438

J
jp9000 已提交
2439
		copy_audio_data(source, (const uint8_t *const *)output, frames,
2440
				audio->timestamp);
2441 2442 2443 2444
	} else {
		copy_audio_data(source, audio->data, audio->frames,
				audio->timestamp);
	}
2445

2446 2447 2448 2449
	mono_output = audio_output_get_channels(obs->audio.audio) == 1;

	if (!mono_output && (source->flags & OBS_SOURCE_FLAG_FORCE_MONO) != 0)
		downmix_to_mono_planar(source, frames);
2450 2451
}

2452
void obs_source_output_audio(obs_source_t *source,
2453
		const struct obs_source_audio *audio)
2454
{
2455
	struct obs_audio_data *output;
2456

2457 2458 2459
	if (!obs_source_valid(source, "obs_source_output_audio"))
		return;
	if (!obs_ptr_valid(audio, "obs_source_output_audio"))
J
jp9000 已提交
2460 2461
		return;

2462
	process_audio(source, audio);
2463 2464

	pthread_mutex_lock(&source->filter_mutex);
2465
	output = filter_async_audio(source, &source->audio_data);
2466 2467

	if (output) {
2468
		struct audio_data data;
J
jp9000 已提交
2469

2470 2471
		for (int i = 0; i < MAX_AV_PLANES; i++)
			data.data[i] = output->data[i];
2472

2473 2474
		data.frames    = output->frames;
		data.timestamp = output->timestamp;
2475

2476
		pthread_mutex_lock(&source->audio_mutex);
2477
		source_output_audio_data(source, &data);
2478 2479 2480 2481 2482 2483
		pthread_mutex_unlock(&source->audio_mutex);
	}

	pthread_mutex_unlock(&source->filter_mutex);
}

2484
void remove_async_frame(obs_source_t *source, struct obs_source_frame *frame)
2485
{
J
jp9000 已提交
2486 2487 2488
	if (frame)
		frame->prev_frame = false;

2489 2490 2491 2492 2493 2494 2495 2496 2497 2498
	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 已提交
2499 2500
/* #define DEBUG_ASYNC_FRAMES 1 */

2501
static bool ready_async_frame(obs_source_t *source, uint64_t sys_time)
2502
{
2503
	struct obs_source_frame *next_frame = source->async_frames.array[0];
2504
	struct obs_source_frame *frame      = NULL;
2505 2506 2507 2508
	uint64_t sys_offset = sys_time - source->last_sys_timestamp;
	uint64_t frame_time = next_frame->timestamp;
	uint64_t frame_offset = 0;

2509
	if ((source->flags & OBS_SOURCE_FLAG_UNBUFFERED) != 0) {
2510 2511
		while (source->async_frames.num > 1) {
			da_erase(source->async_frames, 0);
2512
			remove_async_frame(source, next_frame);
2513
			next_frame = source->async_frames.array[0];
J
jp9000 已提交
2514 2515 2516 2517 2518
		}

		return true;
	}

J
jp9000 已提交
2519 2520 2521 2522 2523 2524
#if DEBUG_ASYNC_FRAMES
	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,
2525
			(unsigned long)source->async_frames.num);
J
jp9000 已提交
2526 2527
#endif

2528 2529
	/* account for timestamp invalidation */
	if (frame_out_of_bounds(source, frame_time)) {
J
jp9000 已提交
2530 2531 2532
#if DEBUG_ASYNC_FRAMES
		blog(LOG_DEBUG, "timing jump");
#endif
2533
		source->last_frame_ts = next_frame->timestamp;
J
jp9000 已提交
2534
		return true;
2535 2536
	} else {
		frame_offset = frame_time - source->last_frame_ts;
J
jp9000 已提交
2537
		source->last_frame_ts += sys_offset;
2538 2539
	}

J
jp9000 已提交
2540 2541 2542 2543 2544 2545
	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 */
2546
		if ((source->last_frame_ts - next_frame->timestamp) < 2000000)
J
jp9000 已提交
2547 2548 2549
			break;

		if (frame)
2550
			da_erase(source->async_frames, 0);
J
jp9000 已提交
2551 2552 2553 2554 2555 2556 2557 2558 2559

#if DEBUG_ASYNC_FRAMES
		blog(LOG_DEBUG, "new frame, "
				"source->last_frame_ts: %llu, "
				"next_frame->timestamp: %llu",
				source->last_frame_ts,
				next_frame->timestamp);
#endif

2560
		remove_async_frame(source, frame);
2561

2562
		if (source->async_frames.num == 1)
2563 2564
			return true;

2565
		frame = next_frame;
2566
		next_frame = source->async_frames.array[1];
2567 2568

		/* more timestamp checking and compensating */
2569
		if ((next_frame->timestamp - frame_time) > MAX_TS_VAR) {
J
jp9000 已提交
2570 2571 2572
#if DEBUG_ASYNC_FRAMES
			blog(LOG_DEBUG, "timing jump");
#endif
2573 2574 2575 2576 2577 2578 2579 2580
			source->last_frame_ts =
				next_frame->timestamp - frame_offset;
		}

		frame_time   = next_frame->timestamp;
		frame_offset = frame_time - source->last_frame_ts;
	}

J
jp9000 已提交
2581 2582 2583 2584
#if DEBUG_ASYNC_FRAMES
	if (!frame)
		blog(LOG_DEBUG, "no frame!");
#endif
2585

2586 2587 2588
	return frame != NULL;
}

2589
static inline struct obs_source_frame *get_closest_frame(obs_source_t *source,
2590 2591
		uint64_t sys_time)
{
2592 2593 2594 2595
	if (!source->async_frames.num)
		return NULL;

	if (!source->last_frame_ts || ready_async_frame(source, sys_time)) {
2596 2597
		struct obs_source_frame *frame = source->async_frames.array[0];
		da_erase(source->async_frames, 0);
2598 2599 2600 2601

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

2602 2603 2604 2605
		return frame;
	}

	return NULL;
2606 2607
}

2608
/*
2609 2610
 * Ensures that cached frames are displayed on time.  If multiple frames
 * were cached between renders, then releases the unnecessary frames and uses
2611 2612
 * the frame with the closest timing to ensure sync.  Also ensures that timing
 * with audio is synchronized.
2613
 */
2614
struct obs_source_frame *obs_source_get_frame(obs_source_t *source)
J
jp9000 已提交
2615
{
2616
	struct obs_source_frame *frame = NULL;
2617

2618
	if (!obs_source_valid(source, "obs_source_get_frame"))
J
jp9000 已提交
2619 2620
		return NULL;

2621
	pthread_mutex_lock(&source->async_mutex);
2622

2623 2624
	frame = source->cur_async_frame;
	source->cur_async_frame = NULL;
J
jp9000 已提交
2625 2626

	if (frame) {
2627
		os_atomic_inc_long(&frame->refs);
2628 2629
	}

2630
	pthread_mutex_unlock(&source->async_mutex);
2631

2632
	return frame;
J
jp9000 已提交
2633 2634
}

2635
void obs_source_release_frame(obs_source_t *source,
2636
		struct obs_source_frame *frame)
J
jp9000 已提交
2637
{
2638
	if (!frame)
2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651
		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);
2652
	}
J
jp9000 已提交
2653
}
2654

2655
const char *obs_source_get_name(const obs_source_t *source)
2656
{
2657 2658
	return obs_source_valid(source, "obs_source_get_name") ?
		source->context.name : NULL;
2659 2660
}

2661
void obs_source_set_name(obs_source_t *source, const char *name)
2662
{
2663 2664
	if (!obs_source_valid(source, "obs_source_set_name"))
		return;
J
jp9000 已提交
2665

2666 2667
	if (!name || !*name || !source->context.name ||
			strcmp(name, source->context.name) != 0) {
J
jp9000 已提交
2668 2669 2670 2671 2672
		struct calldata data;
		char *prev_name = bstrdup(source->context.name);
		obs_context_data_setname(&source->context, name);

		calldata_init(&data);
2673 2674 2675
		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 已提交
2676 2677 2678
		if (!source->context.private)
			signal_handler_signal(obs->signals, "source_rename",
					&data);
J
jp9000 已提交
2679 2680 2681 2682
		signal_handler_signal(source->context.signals, "rename", &data);
		calldata_free(&data);
		bfree(prev_name);
	}
2683 2684
}

2685
enum obs_source_type obs_source_get_type(const obs_source_t *source)
2686
{
2687 2688
	return obs_source_valid(source, "obs_source_get_type") ?
		source->info.type : OBS_SOURCE_TYPE_INPUT;
J
jp9000 已提交
2689
}
J
jp9000 已提交
2690

2691
const char *obs_source_get_id(const obs_source_t *source)
J
jp9000 已提交
2692
{
2693 2694
	return obs_source_valid(source, "obs_source_get_id") ?
		source->info.id : NULL;
2695
}
2696

2697
static inline void render_filter_bypass(obs_source_t *target,
2698
		gs_effect_t *effect, const char *tech_name)
2699
{
2700
	gs_technique_t *tech    = gs_effect_get_technique(effect, tech_name);
2701 2702
	size_t      passes, i;

2703
	passes = gs_technique_begin(tech);
2704
	for (i = 0; i < passes; i++) {
2705
		gs_technique_begin_pass(tech, i);
2706
		obs_source_video_render(target);
2707
		gs_technique_end_pass(tech);
2708
	}
2709
	gs_technique_end(tech);
2710 2711
}

2712
static inline void render_filter_tex(gs_texture_t *tex, gs_effect_t *effect,
2713
		uint32_t width, uint32_t height, const char *tech_name)
2714
{
2715
	gs_technique_t *tech    = gs_effect_get_technique(effect, tech_name);
2716
	gs_eparam_t    *image   = gs_effect_get_param_by_name(effect, "image");
2717 2718
	size_t      passes, i;

2719
	gs_effect_set_texture(image, tex);
2720

2721
	passes = gs_technique_begin(tech);
2722
	for (i = 0; i < passes; i++) {
2723
		gs_technique_begin_pass(tech, i);
J
jp9000 已提交
2724
		gs_draw_sprite(tex, 0, width, height);
2725
		gs_technique_end_pass(tech);
2726
	}
2727
	gs_technique_end(tech);
2728 2729
}

2730 2731 2732 2733 2734 2735 2736 2737 2738 2739
static inline bool can_bypass(obs_source_t *target, obs_source_t *parent,
		uint32_t parent_flags,
		enum obs_allow_direct_render allow_direct)
{
	return (target == parent) &&
		(allow_direct == OBS_ALLOW_DIRECT_RENDERING) &&
		((parent_flags & OBS_SOURCE_CUSTOM_DRAW) == 0) &&
		((parent_flags & OBS_SOURCE_ASYNC) == 0);
}

2740
bool obs_source_process_filter_begin(obs_source_t *filter,
2741
		enum gs_color_format format,
2742
		enum obs_allow_direct_render allow_direct)
2743
{
2744
	obs_source_t *target, *parent;
J
jp9000 已提交
2745
	uint32_t     parent_flags;
J
jp9000 已提交
2746 2747
	int          cx, cy;

2748
	if (!obs_ptr_valid(filter, "obs_source_process_filter_begin"))
2749
		return false;
J
jp9000 已提交
2750

2751 2752
	target       = obs_filter_get_target(filter);
	parent       = obs_filter_get_parent(filter);
J
jp9000 已提交
2753 2754 2755 2756

	if (!target) {
		blog(LOG_INFO, "filter '%s' being processed with no target!",
				filter->context.name);
2757
		return false;
J
jp9000 已提交
2758 2759 2760 2761
	}
	if (!parent) {
		blog(LOG_INFO, "filter '%s' being processed with no parent!",
				filter->context.name);
2762
		return false;
J
jp9000 已提交
2763 2764
	}

J
jp9000 已提交
2765
	parent_flags = parent->info.output_flags;
J
jp9000 已提交
2766 2767
	cx           = get_base_width(target);
	cy           = get_base_height(target);
2768

2769 2770
	filter->allow_direct = allow_direct;

2771 2772 2773 2774
	/* 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 */
2775
	if (can_bypass(target, parent, parent_flags, allow_direct)) {
2776
		return true;
2777 2778
	}

2779 2780
	if (!cx || !cy) {
		obs_source_skip_video_filter(filter);
2781
		return false;
2782 2783
	}

J
jp9000 已提交
2784
	if (!filter->filter_texrender)
2785
		filter->filter_texrender = gs_texrender_create(format,
J
jp9000 已提交
2786 2787
				GS_ZS_NONE);

2788 2789 2790
	gs_blend_state_push();
	gs_blend_function(GS_BLEND_ONE, GS_BLEND_ZERO);

2791
	if (gs_texrender_begin(filter->filter_texrender, cx, cy)) {
2792 2793
		bool custom_draw = (parent_flags & OBS_SOURCE_CUSTOM_DRAW) != 0;
		bool async = (parent_flags & OBS_SOURCE_ASYNC) != 0;
2794 2795 2796 2797
		struct vec4 clear_color;

		vec4_zero(&clear_color);
		gs_clear(GS_CLEAR_COLOR, &clear_color, 0.0f, 0);
2798
		gs_ortho(0.0f, (float)cx, 0.0f, (float)cy, -100.0f, 100.0f);
2799

2800
		if (target == parent && !custom_draw && !async)
2801
			obs_source_default_render(target);
2802 2803
		else
			obs_source_video_render(target);
2804

2805
		gs_texrender_end(filter->filter_texrender);
2806
	}
2807 2808

	gs_blend_state_pop();
2809
	return true;
2810 2811
}

2812 2813 2814 2815 2816
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)
{
	obs_source_t *target, *parent;
	gs_texture_t *texture;
J
jp9000 已提交
2817
	uint32_t     parent_flags;
2818 2819 2820 2821 2822

	if (!filter) return;

	target       = obs_filter_get_target(filter);
	parent       = obs_filter_get_parent(filter);
J
jp9000 已提交
2823 2824 2825 2826

	if (!target || !parent)
		return;

2827 2828 2829 2830 2831 2832 2833 2834 2835 2836 2837 2838 2839
	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);
		render_filter_tex(texture, effect, width, height, tech);
	}
}


2840 2841 2842 2843 2844
void obs_source_process_filter_end(obs_source_t *filter, gs_effect_t *effect,
		uint32_t width, uint32_t height)
{
	obs_source_t *target, *parent;
	gs_texture_t *texture;
J
jp9000 已提交
2845
	uint32_t     parent_flags;
2846

2847 2848
	if (!obs_ptr_valid(filter, "obs_source_process_filter_end"))
		return;
2849

2850 2851 2852
	target       = obs_filter_get_target(filter);
	parent       = obs_filter_get_parent(filter);
	parent_flags = parent->info.output_flags;
2853

2854
	if (can_bypass(target, parent, parent_flags, filter->allow_direct)) {
2855
		render_filter_bypass(target, effect, "Draw");
2856 2857
	} else {
		texture = gs_texrender_get_texture(filter->filter_texrender);
2858
		if (texture)
2859 2860
			render_filter_tex(texture, effect, width, height,
					"Draw");
2861
	}
2862
}
2863

2864 2865 2866 2867 2868 2869
void obs_source_skip_video_filter(obs_source_t *filter)
{
	obs_source_t *target, *parent;
	bool custom_draw, async;
	uint32_t parent_flags;

2870 2871
	if (!obs_ptr_valid(filter, "obs_source_skip_video_filter"))
		return;
2872 2873 2874 2875 2876 2877 2878

	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 已提交
2879 2880
	if (target == parent) {
		if (!custom_draw && !async)
2881
			obs_source_default_render(target);
J
jp9000 已提交
2882 2883
		else if (target->info.video_render)
			obs_source_main_render(target);
J
jp9000 已提交
2884 2885
		else if (deinterlacing_enabled(target))
			deinterlace_render(target);
J
jp9000 已提交
2886 2887 2888 2889 2890 2891
		else
			obs_source_render_async_video(target);

	} else {
		obs_source_video_render(target);
	}
2892 2893
}

2894
signal_handler_t *obs_source_get_signal_handler(const obs_source_t *source)
2895
{
2896 2897
	return obs_source_valid(source, "obs_source_get_signal_handler") ?
		source->context.signals : NULL;
2898 2899
}

2900
proc_handler_t *obs_source_get_proc_handler(const obs_source_t *source)
2901
{
2902 2903
	return obs_source_valid(source, "obs_source_get_proc_handler") ?
		source->context.procs : NULL;
2904
}
J
jp9000 已提交
2905

2906
void obs_source_set_volume(obs_source_t *source, float volume)
J
jp9000 已提交
2907
{
2908
	if (obs_source_valid(source, "obs_source_set_volume")) {
J
jp9000 已提交
2909 2910 2911 2912 2913 2914
		struct audio_action action = {
			.timestamp = os_gettime_ns(),
			.type      = AUDIO_ACTION_VOL,
			.vol       = volume
		};

2915 2916 2917 2918
		struct calldata data;
		uint8_t stack[128];

		calldata_init_fixed(&data, stack, sizeof(stack));
2919 2920
		calldata_set_ptr(&data, "source", source);
		calldata_set_float(&data, "volume", volume);
J
jp9000 已提交
2921

2922
		signal_handler_signal(source->context.signals, "volume", &data);
J
jp9000 已提交
2923 2924 2925
		if (!source->context.private)
			signal_handler_signal(obs->signals, "source_volume",
					&data);
J
jp9000 已提交
2926

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

J
jp9000 已提交
2929 2930 2931 2932
		pthread_mutex_lock(&source->audio_actions_mutex);
		da_push_back(source->audio_actions, &action);
		pthread_mutex_unlock(&source->audio_actions_mutex);

J
jp9000 已提交
2933
		source->user_volume = volume;
J
jp9000 已提交
2934
	}
J
jp9000 已提交
2935 2936
}

2937
float obs_source_get_volume(const obs_source_t *source)
J
jp9000 已提交
2938
{
2939 2940
	return obs_source_valid(source, "obs_source_get_volume") ?
		source->user_volume : 0.0f;
J
jp9000 已提交
2941 2942
}

2943
void obs_source_set_sync_offset(obs_source_t *source, int64_t offset)
J
jp9000 已提交
2944
{
2945
	if (obs_source_valid(source, "obs_source_set_sync_offset")) {
2946 2947
		struct calldata data;
		uint8_t stack[128];
J
jp9000 已提交
2948

2949
		calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
2950 2951 2952 2953 2954 2955 2956 2957
		calldata_set_ptr(&data, "source", source);
		calldata_set_int(&data, "offset", offset);

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

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

2960
int64_t obs_source_get_sync_offset(const obs_source_t *source)
J
jp9000 已提交
2961
{
2962 2963
	return obs_source_valid(source, "obs_source_get_sync_offset") ?
		source->sync_offset : 0;
J
jp9000 已提交
2964
}
2965 2966 2967 2968 2969 2970

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

2971 2972
static void enum_source_active_tree_callback(obs_source_t *parent,
		obs_source_t *child, void *param)
2973 2974
{
	struct source_enum_data *data = param;
2975
	bool is_transition = child->info.type == OBS_SOURCE_TYPE_TRANSITION;
2976

2977 2978
	if (is_transition)
		obs_transition_enum_sources(child,
2979
				enum_source_active_tree_callback, param);
2980
	if (child->info.enum_active_sources) {
J
jp9000 已提交
2981
		if (child->context.data) {
2982
			child->info.enum_active_sources(child->context.data,
2983
					enum_source_active_tree_callback, data);
J
jp9000 已提交
2984
		}
2985 2986 2987 2988 2989
	}

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

2990
void obs_source_enum_active_sources(obs_source_t *source,
2991 2992 2993
		obs_source_enum_proc_t enum_callback,
		void *param)
{
J
jp9000 已提交
2994
	bool is_transition;
2995
	if (!data_valid(source, "obs_source_enum_active_sources"))
J
jp9000 已提交
2996
		return;
J
jp9000 已提交
2997 2998 2999

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

	obs_source_addref(source);

J
jp9000 已提交
3004 3005 3006 3007 3008
	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,
				enum_callback, param);
3009 3010 3011 3012

	obs_source_release(source);
}

3013
void obs_source_enum_active_tree(obs_source_t *source,
3014 3015 3016 3017
		obs_source_enum_proc_t enum_callback,
		void *param)
{
	struct source_enum_data data = {enum_callback, param};
J
jp9000 已提交
3018
	bool is_transition;
3019

3020
	if (!data_valid(source, "obs_source_enum_active_tree"))
J
jp9000 已提交
3021
		return;
J
jp9000 已提交
3022 3023 3024

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

	obs_source_addref(source);

J
jp9000 已提交
3029
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
3030 3031
		obs_transition_enum_sources(source,
				enum_source_active_tree_callback, &data);
J
jp9000 已提交
3032 3033
	if (source->info.enum_active_sources)
		source->info.enum_active_sources(source->context.data,
3034 3035 3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049 3050 3051 3052 3053 3054 3055 3056 3057 3058 3059 3060 3061 3062 3063 3064 3065 3066 3067 3068 3069 3070 3071 3072 3073 3074 3075 3076 3077 3078 3079 3080 3081 3082 3083 3084 3085 3086 3087 3088 3089 3090
				enum_source_active_tree_callback, &data);

	obs_source_release(source);
}

static void enum_source_full_tree_callback(obs_source_t *parent,
		obs_source_t *child, void *param)
{
	struct source_enum_data *data = param;
	bool is_transition = child->info.type == OBS_SOURCE_TYPE_TRANSITION;

	if (is_transition)
		obs_transition_enum_sources(child,
				enum_source_full_tree_callback, param);
	if (child->info.enum_all_sources) {
		if (child->context.data) {
			child->info.enum_active_sources(child->context.data,
					enum_source_full_tree_callback, data);
		}
	} else if (child->info.enum_active_sources) {
		if (child->context.data) {
			child->info.enum_active_sources(child->context.data,
					enum_source_full_tree_callback, data);
		}
	}

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

static void obs_source_enum_full_tree(obs_source_t *source,
		obs_source_enum_proc_t enum_callback,
		void *param)
{
	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)
		obs_transition_enum_sources(source,
				enum_source_full_tree_callback, &data);

	if (source->info.enum_all_sources) {
		source->info.enum_all_sources(source->context.data,
				enum_source_full_tree_callback, &data);

	} else if (source->info.enum_active_sources) {
		source->info.enum_active_sources(source->context.data,
				enum_source_full_tree_callback, &data);
	}
3091 3092 3093

	obs_source_release(source);
}
3094

J
jp9000 已提交
3095 3096 3097 3098 3099 3100 3101
struct descendant_info {
	bool exists;
	obs_source_t *target;
};

static void check_descendant(obs_source_t *parent, obs_source_t *child,
		void *param)
3102
{
J
jp9000 已提交
3103 3104 3105 3106 3107
	struct descendant_info *info = param;
	if (child == info->target || parent == info->target)
		info->exists = true;
}

3108
bool obs_source_add_active_child(obs_source_t *parent, obs_source_t *child)
J
jp9000 已提交
3109
{
3110
	struct descendant_info info = {false, parent};
3111

3112
	if (!obs_ptr_valid(parent, "obs_source_add_active_child"))
3113
		return false;
3114
	if (!obs_ptr_valid(child, "obs_source_add_active_child"))
3115 3116
		return false;
	if (parent == child) {
3117 3118
		blog(LOG_WARNING, "obs_source_add_active_child: "
				"parent == child");
3119 3120
		return false;
	}
J
jp9000 已提交
3121

3122
	obs_source_enum_active_tree(child, check_descendant, &info);
J
jp9000 已提交
3123 3124
	if (info.exists)
		return false;
3125

3126 3127 3128 3129 3130
	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 已提交
3131 3132

	return true;
3133 3134
}

3135
void obs_source_remove_active_child(obs_source_t *parent, obs_source_t *child)
3136
{
3137
	if (!obs_ptr_valid(parent, "obs_source_remove_active_child"))
3138
		return;
3139
	if (!obs_ptr_valid(child, "obs_source_remove_active_child"))
3140
		return;
3141

3142 3143 3144 3145 3146
	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);
	}
3147
}
J
jp9000 已提交
3148

3149
void obs_source_save(obs_source_t *source)
3150
{
J
jp9000 已提交
3151 3152 3153
	if (!data_valid(source, "obs_source_save"))
		return;

J
jp9000 已提交
3154 3155 3156 3157 3158
	obs_source_dosignal(source, "source_save", "save");

	if (source->info.save)
		source->info.save(source->context.data,
				source->context.settings);
3159 3160
}

3161
void obs_source_load(obs_source_t *source)
3162
{
J
jp9000 已提交
3163 3164
	if (!data_valid(source, "obs_source_load"))
		return;
J
jp9000 已提交
3165 3166 3167
	if (source->info.load)
		source->info.load(source->context.data,
				source->context.settings);
J
jp9000 已提交
3168

J
jp9000 已提交
3169
	obs_source_dosignal(source, "source_load", "load");
3170
}
J
jp9000 已提交
3171

J
jp9000 已提交
3172 3173
bool obs_source_active(const obs_source_t *source)
{
3174 3175
	return obs_source_valid(source, "obs_source_active") ?
		source->activate_refs != 0 : false;
J
jp9000 已提交
3176 3177
}

J
jp9000 已提交
3178 3179
bool obs_source_showing(const obs_source_t *source)
{
3180 3181
	return obs_source_valid(source, "obs_source_showing") ?
		source->show_refs != 0 : false;
J
jp9000 已提交
3182 3183
}

J
jp9000 已提交
3184 3185
static inline void signal_flags_updated(obs_source_t *source)
{
3186 3187
	struct calldata data;
	uint8_t stack[128];
J
jp9000 已提交
3188

3189
	calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
3190 3191 3192 3193 3194 3195 3196 3197
	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)
{
3198 3199
	if (!obs_source_valid(source, "obs_source_set_flags"))
		return;
J
jp9000 已提交
3200 3201 3202 3203 3204 3205 3206

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

3207 3208
void obs_source_set_default_flags(obs_source_t *source, uint32_t flags)
{
3209 3210
	if (!obs_source_valid(source, "obs_source_set_default_flags"))
		return;
3211 3212 3213 3214

	source->default_flags = flags;
}

J
jp9000 已提交
3215 3216
uint32_t obs_source_get_flags(const obs_source_t *source)
{
3217 3218
	return obs_source_valid(source, "obs_source_get_flags") ?
		source->flags : 0;
J
jp9000 已提交
3219
}
J
jp9000 已提交
3220

3221 3222
void obs_source_set_audio_mixers(obs_source_t *source, uint32_t mixers)
{
3223 3224
	struct calldata data;
	uint8_t stack[128];
3225

3226 3227 3228 3229
	if (!obs_source_valid(source, "obs_source_set_audio_mixers"))
		return;
	if ((source->info.output_flags & OBS_SOURCE_AUDIO) == 0)
		return;
3230

3231
	if (source->audio_mixers == mixers)
3232 3233
		return;

3234
	calldata_init_fixed(&data, stack, sizeof(stack));
3235 3236 3237 3238 3239 3240 3241
	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");

3242
	source->audio_mixers = mixers;
3243 3244 3245 3246
}

uint32_t obs_source_get_audio_mixers(const obs_source_t *source)
{
3247 3248 3249 3250
	if (!obs_source_valid(source, "obs_source_get_audio_mixers"))
		return 0;
	if ((source->info.output_flags & OBS_SOURCE_AUDIO) == 0)
		return 0;
3251

3252
	return source->audio_mixers;
3253 3254
}

J
jp9000 已提交
3255 3256 3257 3258
void obs_source_draw_set_color_matrix(const struct matrix4 *color_matrix,
		const struct vec3 *color_range_min,
		const struct vec3 *color_range_max)
{
3259 3260 3261 3262 3263 3264
	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 已提交
3265 3266 3267 3268 3269 3270
	gs_effect_t *effect = gs_get_effect();
	gs_eparam_t *matrix;
	gs_eparam_t *range_min;
	gs_eparam_t *range_max;

	if (!effect) {
3271 3272
		blog(LOG_WARNING, "obs_source_draw_set_color_matrix: no "
				"active effect!");
J
jp9000 已提交
3273 3274 3275
		return;
	}

3276
	if (!obs_ptr_valid(color_matrix, "obs_source_draw_set_color_matrix"))
J
jp9000 已提交
3277 3278 3279 3280 3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297 3298 3299 3300
		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);
	gs_effect_set_val(range_min, color_range_min, sizeof(float)*3);
	gs_effect_set_val(range_max, color_range_max, sizeof(float)*3);
}

void obs_source_draw(gs_texture_t *texture, int x, int y, uint32_t cx,
		uint32_t cy, bool flip)
{
	gs_effect_t *effect = gs_get_effect();
	bool change_pos = (x != 0 || y != 0);
	gs_eparam_t *image;

	if (!effect) {
3301
		blog(LOG_WARNING, "obs_source_draw: no active effect!");
J
jp9000 已提交
3302 3303 3304
		return;
	}

3305
	if (!obs_ptr_valid(texture, "obs_source_draw"))
J
jp9000 已提交
3306 3307 3308 3309 3310 3311 3312 3313 3314 3315 3316 3317 3318 3319 3320
		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();
}
3321

3322 3323
void obs_source_inc_showing(obs_source_t *source)
{
3324 3325
	if (obs_source_valid(source, "obs_source_inc_showing"))
		obs_source_activate(source, AUX_VIEW);
3326 3327 3328 3329
}

void obs_source_dec_showing(obs_source_t *source)
{
3330 3331
	if (obs_source_valid(source, "obs_source_dec_showing"))
		obs_source_deactivate(source, AUX_VIEW);
3332
}
3333 3334 3335 3336

void obs_source_enum_filters(obs_source_t *source,
		obs_source_enum_proc_t callback, void *param)
{
3337 3338 3339
	if (!obs_source_valid(source, "obs_source_enum_filters"))
		return;
	if (!obs_ptr_valid(callback, "obs_source_enum_filters"))
3340 3341 3342 3343
		return;

	pthread_mutex_lock(&source->filter_mutex);

3344 3345
	for (size_t i = source->filters.num; i > 0; i--) {
		struct obs_source *filter = source->filters.array[i - 1];
3346 3347 3348 3349 3350
		callback(source, filter, param);
	}

	pthread_mutex_unlock(&source->filter_mutex);
}
3351 3352 3353 3354 3355 3356

obs_source_t *obs_source_get_filter_by_name(obs_source_t *source,
		const char *name)
{
	obs_source_t *filter = NULL;

3357 3358 3359
	if (!obs_source_valid(source, "obs_source_get_filter_by_name"))
		return NULL;
	if (!obs_ptr_valid(name, "obs_source_get_filter_by_name"))
3360 3361 3362 3363 3364 3365 3366 3367 3368 3369 3370 3371 3372 3373 3374 3375 3376
		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;
}
3377 3378 3379

bool obs_source_enabled(const obs_source_t *source)
{
3380 3381
	return obs_source_valid(source, "obs_source_enabled") ?
		source->enabled : false;
3382 3383 3384 3385
}

void obs_source_set_enabled(obs_source_t *source, bool enabled)
{
3386 3387
	struct calldata data;
	uint8_t stack[128];
3388

3389
	if (!obs_source_valid(source, "obs_source_set_enabled"))
3390 3391 3392 3393
		return;

	source->enabled = enabled;

3394
	calldata_init_fixed(&data, stack, sizeof(stack));
3395 3396 3397 3398 3399
	calldata_set_ptr(&data, "source", source);
	calldata_set_bool(&data, "enabled", enabled);

	signal_handler_signal(source->context.signals, "enable", &data);
}
J
jp9000 已提交
3400 3401 3402

bool obs_source_muted(const obs_source_t *source)
{
3403
	return obs_source_valid(source, "obs_source_muted") ?
J
jp9000 已提交
3404
		source->user_muted : false;
J
jp9000 已提交
3405 3406 3407 3408
}

void obs_source_set_muted(obs_source_t *source, bool muted)
{
3409 3410
	struct calldata data;
	uint8_t stack[128];
J
jp9000 已提交
3411 3412 3413 3414 3415
	struct audio_action action = {
		.timestamp = os_gettime_ns(),
		.type      = AUDIO_ACTION_MUTE,
		.set       = muted
	};
J
jp9000 已提交
3416

3417
	if (!obs_source_valid(source, "obs_source_set_muted"))
J
jp9000 已提交
3418 3419
		return;

J
jp9000 已提交
3420
	source->user_muted = muted;
J
jp9000 已提交
3421

3422
	calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
3423 3424 3425 3426 3427
	calldata_set_ptr(&data, "source", source);
	calldata_set_bool(&data, "muted", muted);

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

J
jp9000 已提交
3428 3429 3430
	pthread_mutex_lock(&source->audio_actions_mutex);
	da_push_back(source->audio_actions, &action);
	pthread_mutex_unlock(&source->audio_actions_mutex);
J
jp9000 已提交
3431
}
P
Palana 已提交
3432 3433 3434 3435

static void source_signal_push_to_changed(obs_source_t *source,
		const char *signal, bool enabled)
{
3436 3437
	struct calldata data;
	uint8_t stack[128];
P
Palana 已提交
3438

3439
	calldata_init_fixed(&data, stack, sizeof(stack));
P
Palana 已提交
3440 3441 3442 3443 3444 3445 3446 3447 3448
	calldata_set_ptr (&data, "source",  source);
	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,
		const char *signal, uint64_t delay)
{
3449 3450
	struct calldata data;
	uint8_t stack[128];
P
Palana 已提交
3451

3452
	calldata_init_fixed(&data, stack, sizeof(stack));
P
Palana 已提交
3453 3454 3455 3456 3457 3458 3459 3460 3461
	calldata_set_ptr (&data, "source", source);
	calldata_set_bool(&data, "delay",  delay);

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

bool obs_source_push_to_mute_enabled(obs_source_t *source)
{
	bool enabled;
3462 3463
	if (!obs_source_valid(source, "obs_source_push_to_mute_enabled"))
		return false;
P
Palana 已提交
3464 3465 3466 3467 3468 3469 3470 3471 3472 3473

	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)
{
3474 3475
	if (!obs_source_valid(source, "obs_source_enable_push_to_mute"))
		return;
P
Palana 已提交
3476 3477 3478 3479 3480 3481 3482 3483 3484 3485 3486 3487 3488 3489 3490 3491 3492 3493 3494

	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",
				obs_source_get_name(source),
				enabled ? "enabled" : "disabled");

	source->push_to_mute_enabled = enabled;

	if (changed)
		source_signal_push_to_changed(source, "push_to_mute_changed",
				enabled);
	pthread_mutex_unlock(&source->audio_mutex);
}

uint64_t obs_source_get_push_to_mute_delay(obs_source_t *source)
{
	uint64_t delay;
3495 3496
	if (!obs_source_valid(source, "obs_source_get_push_to_mute_delay"))
		return 0;
P
Palana 已提交
3497 3498 3499 3500 3501 3502 3503 3504 3505 3506

	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)
{
3507 3508
	if (!obs_source_valid(source, "obs_source_set_push_to_mute_delay"))
		return;
P
Palana 已提交
3509 3510 3511 3512 3513 3514 3515 3516 3517 3518 3519

	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;
3520 3521
	if (!obs_source_valid(source, "obs_source_push_to_talk_enabled"))
		return false;
P
Palana 已提交
3522 3523 3524 3525 3526 3527 3528 3529 3530 3531

	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)
{
3532 3533
	if (!obs_source_valid(source, "obs_source_enable_push_to_talk"))
		return;
P
Palana 已提交
3534 3535 3536 3537 3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551 3552

	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",
				obs_source_get_name(source),
				enabled ? "enabled" : "disabled");

	source->push_to_talk_enabled = enabled;

	if (changed)
		source_signal_push_to_changed(source, "push_to_talk_changed",
				enabled);
	pthread_mutex_unlock(&source->audio_mutex);
}

uint64_t obs_source_get_push_to_talk_delay(obs_source_t *source)
{
	uint64_t delay;
3553 3554
	if (!obs_source_valid(source, "obs_source_get_push_to_talk_delay"))
		return 0;
P
Palana 已提交
3555 3556 3557 3558 3559 3560 3561 3562 3563 3564

	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)
{
3565 3566
	if (!obs_source_valid(source, "obs_source_set_push_to_talk_delay"))
		return;
P
Palana 已提交
3567 3568 3569 3570 3571 3572 3573

	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);
}
3574 3575 3576 3577 3578 3579

void *obs_source_get_type_data(obs_source_t *source)
{
	return obs_source_valid(source, "obs_source_get_type_data")
		? source->info.type_data : NULL;
}
3580

J
jp9000 已提交
3581 3582 3583 3584 3585 3586 3587 3588 3589 3590 3591 3592 3593 3594 3595 3596 3597 3598 3599 3600 3601 3602 3603 3604 3605 3606 3607 3608 3609 3610 3611 3612 3613 3614 3615 3616 3617 3618 3619 3620 3621 3622 3623 3624 3625 3626 3627 3628 3629 3630 3631 3632 3633 3634 3635 3636 3637 3638 3639 3640 3641 3642 3643 3644 3645 3646 3647 3648 3649 3650 3651 3652 3653 3654 3655 3656 3657 3658 3659 3660 3661 3662 3663 3664 3665 3666 3667 3668 3669 3670 3671 3672 3673 3674 3675 3676 3677 3678 3679 3680 3681 3682 3683 3684 3685 3686 3687 3688 3689 3690 3691 3692 3693 3694 3695 3696 3697 3698 3699 3700 3701 3702 3703 3704 3705 3706 3707 3708 3709 3710 3711 3712 3713 3714 3715 3716 3717 3718 3719 3720 3721 3722 3723 3724 3725 3726 3727 3728 3729 3730 3731 3732 3733 3734 3735 3736 3737 3738 3739 3740 3741 3742 3743 3744 3745 3746 3747 3748 3749 3750 3751 3752 3753 3754 3755 3756 3757 3758 3759 3760 3761 3762 3763 3764 3765 3766 3767 3768 3769 3770 3771 3772 3773 3774 3775 3776 3777 3778 3779
static float get_source_volume(obs_source_t *source, uint64_t os_time)
{
	if (source->push_to_mute_enabled && source->push_to_mute_pressed)
		source->push_to_mute_stop_time = os_time +
			source->push_to_mute_delay * 1000000;

	if (source->push_to_talk_enabled && source->push_to_talk_pressed)
		source->push_to_talk_stop_time = os_time +
			source->push_to_talk_delay * 1000000;

	bool push_to_mute_active = source->push_to_mute_pressed ||
		os_time < source->push_to_mute_stop_time;
	bool push_to_talk_active = source->push_to_talk_pressed ||
		os_time < source->push_to_talk_stop_time;

	bool muted = !source->enabled || source->muted ||
			(source->push_to_mute_enabled && push_to_mute_active) ||
			(source->push_to_talk_enabled && !push_to_talk_active);

	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,
		size_t channels, float vol)
{
	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,
		size_t channels, float *vol_data)
{
	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,
		const struct audio_action *action)
{
	switch (action->type) {
	case AUDIO_ACTION_VOL:
		source->volume = action->vol; break;
	case AUDIO_ACTION_MUTE:
		source->muted = action->set; break;
	case AUDIO_ACTION_PTT:
		source->push_to_talk_pressed = action->set; break;
	case AUDIO_ACTION_PTM:
		source->push_to_mute_pressed = action->set; break;
	}
}

static void apply_audio_actions(obs_source_t *source, size_t channels,
		size_t sample_rate)
{
	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;

		new_frame_num = conv_time_to_frames(sample_rate,
				timestamp - source->audio_ts);

		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,
		size_t channels, size_t sample_rate)
{
	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) {
		uint64_t duration = conv_frames_to_time(sample_rate,
				AUDIO_OUTPUT_FRAMES);

		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,
				AUDIO_OUTPUT_FRAMES * sizeof(float) *
				MAX_AUDIO_CHANNELS * MAX_AUDIO_MIXES);
		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,
		size_t channels, size_t sample_rate)
{
	struct obs_source_audio_mix audio_data;
	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.output[mix].data[ch] =
				source->audio_output_buf[mix][ch];
	}

	memset(audio_data.output[0].data[0], 0, AUDIO_OUTPUT_FRAMES *
			MAX_AUDIO_MIXES * channels * sizeof(float));

	success = source->info.audio_render(source->context.data, &ts,
			&audio_data, mixers, channels, sample_rate);
	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++) {
		if ((source->audio_mixers & (1 << mix)) == 0) {
			memset(source->audio_output_buf[mix][0], 0,
					sizeof(float) * AUDIO_OUTPUT_FRAMES *
					channels);
			continue;
		}
	}

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

static inline void process_audio_source_tick(obs_source_t *source,
		uint32_t mixers, size_t channels, size_t sample_rate,
		size_t size)
{
	pthread_mutex_lock(&source->audio_buf_mutex);

3780 3781 3782 3783 3784 3785
	if (source->audio_input_buf[0].size < size) {
		source->audio_pending = true;
		pthread_mutex_unlock(&source->audio_buf_mutex);
		return;
	}

J
jp9000 已提交
3786 3787 3788 3789 3790 3791 3792 3793 3794 3795 3796 3797 3798 3799 3800 3801 3802 3803 3804 3805 3806 3807 3808 3809 3810 3811 3812 3813 3814 3815 3816 3817 3818
	for (size_t ch = 0; ch < channels; ch++)
		circlebuf_peek_front(&source->audio_input_buf[ch],
				source->audio_output_buf[0][ch],
				size);

	pthread_mutex_unlock(&source->audio_buf_mutex);

	for (size_t mix = 1; 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) {
			memset(source->audio_output_buf[mix][0],
					0, size * channels);
			continue;
		}

		for (size_t ch = 0; ch < channels; ch++)
			memcpy(source->audio_output_buf[mix][ch],
					source->audio_output_buf[0][ch], size);
	}

	if ((source->audio_mixers & 1) == 0 || (mixers & 1) == 0)
		memset(source->audio_output_buf[0][0], 0,
				size * channels);

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

void obs_source_audio_render(obs_source_t *source, uint32_t mixers,
		size_t channels, size_t sample_rate, size_t size)
{
3819
	if (!source->audio_output_buf[0][0]) {
J
jp9000 已提交
3820 3821 3822 3823 3824 3825 3826 3827 3828
		source->audio_pending = true;
		return;
	}

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

3829
	if (!source->audio_ts) {
J
jp9000 已提交
3830 3831 3832 3833 3834 3835 3836 3837 3838
		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)
{
3839 3840 3841 3842 3843
	if (!obs_source_valid(source, "obs_source_audio_pending"))
		return true;

	return (is_composite_source(source) || is_audio_source(source)) ?
		source->audio_pending : true;
J
jp9000 已提交
3844 3845
}

3846 3847 3848 3849 3850 3851 3852 3853 3854 3855 3856 3857 3858 3859 3860 3861 3862 3863 3864 3865 3866
uint64_t obs_source_get_audio_timestamp(const obs_source_t *source)
{
	return obs_source_valid(source, "obs_source_get_audio_timestamp") ?
		source->audio_ts : 0;
}

void obs_source_get_audio_mix(const obs_source_t *source,
		struct obs_source_audio_mix *audio)
{
	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];
		}
	}
}
3867 3868 3869 3870 3871 3872 3873 3874 3875 3876 3877 3878 3879 3880 3881 3882 3883 3884 3885 3886 3887 3888 3889 3890 3891 3892

void obs_source_add_audio_capture_callback(obs_source_t *source,
		obs_source_audio_capture_t callback, void *param)
{
	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);
}

void obs_source_remove_audio_capture_callback(obs_source_t *source,
		obs_source_audio_capture_t callback, void *param)
{
	struct audio_cb_info info = {callback, param};

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

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