obs-source.c 100.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 177 178
	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;
	source->audio_mixers = 0xF;

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 924
static inline struct obs_source_frame *get_closest_frame(obs_source_t *source,
		uint64_t sys_time);

925
void obs_source_video_tick(obs_source_t *source, float seconds)
J
jp9000 已提交
926
{
927 928
	bool now_showing, now_active;

929 930
	if (!obs_source_valid(source, "obs_source_video_tick"))
		return;
J
jp9000 已提交
931

J
jp9000 已提交
932 933 934
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
		obs_transition_tick(source);

935
	if ((source->info.output_flags & OBS_SOURCE_ASYNC) != 0) {
936
		uint64_t sys_time = obs->video.video_time;
937 938

		pthread_mutex_lock(&source->async_mutex);
J
jp9000 已提交
939 940 941 942 943 944 945 946 947 948 949 950

		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);
951 952 953 954 955 956
		}

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

957 958 959
	if (source->defer_update)
		obs_source_deferred_update(source);

J
jp9000 已提交
960 961
	/* reset the filter render texture information once every frame */
	if (source->filter_texrender)
962
		gs_texrender_reset(source->filter_texrender);
J
jp9000 已提交
963

964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987
	/* 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;
	}

988
	if (source->context.data && source->info.video_tick)
989
		source->info.video_tick(source->context.data, seconds);
990 991

	source->async_rendered = false;
J
jp9000 已提交
992
	source->deinterlace_rendered = false;
J
jp9000 已提交
993 994
}

995
/* unless the value is 3+ hours worth of frames, this won't overflow */
996 997
static inline uint64_t conv_frames_to_time(const size_t sample_rate,
		const size_t frames)
998
{
999
	return (uint64_t)frames * 1000000000ULL / (uint64_t)sample_rate;
1000 1001
}

J
jp9000 已提交
1002 1003 1004 1005 1006 1007 1008 1009
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))
1010

1011 1012
static inline void reset_audio_timing(obs_source_t *source, uint64_t timestamp,
		uint64_t os_time)
1013 1014
{
	source->timing_set    = true;
1015
	source->timing_adjust = os_time - timestamp;
1016
}
1017

1018 1019 1020 1021 1022 1023 1024 1025
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);
	}

1026
	source->last_audio_input_buf_size = 0;
1027 1028 1029 1030
	source->audio_ts = os_time;
}

static void handle_ts_jump(obs_source_t *source, uint64_t expected,
1031
		uint64_t ts, uint64_t diff, uint64_t os_time)
1032
{
J
jp9000 已提交
1033
	blog(LOG_DEBUG, "Timestamp for source '%s' jumped by '%"PRIu64"', "
1034
	                "expected value %"PRIu64", input value %"PRIu64,
1035
	                source->context.name, diff, expected, ts);
1036

1037
	pthread_mutex_lock(&source->audio_buf_mutex);
1038
	reset_audio_timing(source, ts, os_time);
1039
	pthread_mutex_unlock(&source->audio_buf_mutex);
1040 1041
}

1042
static void source_signal_audio_data(obs_source_t *source,
1043
		struct audio_data *in, bool muted)
1044
{
1045
	pthread_mutex_lock(&source->audio_cb_mutex);
1046

1047 1048 1049 1050
	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);
	}
1051

1052
	pthread_mutex_unlock(&source->audio_cb_mutex);
1053 1054
}

1055 1056 1057 1058 1059
static inline uint64_t uint64_diff(uint64_t ts1, uint64_t ts2)
{
	return (ts1 < ts2) ?  (ts2 - ts1) : (ts1 - ts2);
}

1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088
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 已提交
1089 1090 1091 1092 1093
	/* 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++) {
1094 1095
		circlebuf_place(&source->audio_input_buf[i], buf_placement,
				in->data[i], size);
J
jp9000 已提交
1096 1097 1098 1099
		circlebuf_pop_back(&source->audio_input_buf[i], NULL,
				source->audio_input_buf[i].size -
				(buf_placement + size));
	}
1100 1101

	source->last_audio_input_buf_size = 0;
1102 1103 1104 1105 1106 1107 1108
}

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 已提交
1109 1110 1111 1112 1113
	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;
1114 1115 1116

	for (size_t i = 0; i < channels; i++)
		circlebuf_push_back(&source->audio_input_buf[i],
J
jp9000 已提交
1117
				in->data[i], size);
1118 1119 1120 1121

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

1124 1125
static inline bool source_muted(obs_source_t *source, uint64_t os_time)
{
J
jp9000 已提交
1126
	if (source->push_to_mute_enabled && source->user_push_to_mute_pressed)
1127 1128 1129
		source->push_to_mute_stop_time = os_time +
			source->push_to_mute_delay * 1000000;

J
jp9000 已提交
1130
	if (source->push_to_talk_enabled && source->user_push_to_talk_pressed)
1131 1132 1133
		source->push_to_talk_stop_time = os_time +
			source->push_to_talk_delay * 1000000;

J
jp9000 已提交
1134
	bool push_to_mute_active = source->user_push_to_mute_pressed ||
1135
		os_time < source->push_to_mute_stop_time;
J
jp9000 已提交
1136
	bool push_to_talk_active = source->user_push_to_talk_pressed ||
1137 1138
		os_time < source->push_to_talk_stop_time;

J
jp9000 已提交
1139
	return !source->enabled || source->user_muted ||
1140 1141 1142 1143
			(source->push_to_mute_enabled && push_to_mute_active) ||
			(source->push_to_talk_enabled && !push_to_talk_active);
}

1144
static void source_output_audio_data(obs_source_t *source,
1145 1146
		const struct audio_data *data)
{
1147
	size_t sample_rate = audio_output_get_sample_rate(obs->audio.audio);
1148
	struct audio_data in = *data;
1149
	uint64_t diff;
1150
	uint64_t os_time = os_gettime_ns();
1151
	int64_t sync_offset;
1152 1153
	bool using_direct_ts = false;
	bool push_back = false;
1154

1155 1156 1157 1158 1159
	/* 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;
1160 1161
		using_direct_ts = true;
	}
1162

1163
	if (!source->timing_set) {
1164
		reset_audio_timing(source, in.timestamp, os_time);
1165

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

1169
		/* smooth audio if within threshold */
1170
		if (diff > MAX_TS_VAR && !using_direct_ts)
1171
			handle_ts_jump(source, source->next_audio_ts_min,
1172
					in.timestamp, diff, os_time);
1173
		else if (diff < TS_SMOOTHING_THRESHOLD)
1174
			in.timestamp = source->next_audio_ts_min;
1175 1176
	}

1177
	source->next_audio_ts_min = in.timestamp +
1178
		conv_frames_to_time(sample_rate, in.frames);
1179

1180
	in.timestamp += source->timing_adjust;
1181

1182 1183
	pthread_mutex_lock(&source->audio_buf_mutex);

J
jp9000 已提交
1184
	if (source->next_audio_sys_ts_min == in.timestamp) {
1185
		push_back = true;
1186 1187

	} else if (source->next_audio_sys_ts_min) {
J
jp9000 已提交
1188
		diff = uint64_diff(source->next_audio_sys_ts_min, in.timestamp);
1189 1190

		if (diff < TS_SMOOTHING_THRESHOLD) {
1191 1192 1193 1194 1195 1196 1197 1198
			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. */
1199
		} else if (diff > MAX_TS_VAR) {
1200
			reset_audio_timing(source, data->timestamp,
1201
					os_time);
1202 1203
			in.timestamp = data->timestamp + source->timing_adjust;
		}
J
jp9000 已提交
1204 1205
	}

1206 1207
	sync_offset = source->sync_offset;
	in.timestamp += sync_offset;
1208 1209
	in.timestamp -= source->resample_offset;

1210
	source->next_audio_sys_ts_min = source->next_audio_ts_min +
1211 1212 1213 1214 1215 1216 1217
		source->timing_adjust;

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

1219
	if (push_back && source->audio_ts)
1220 1221 1222 1223 1224 1225
		source_output_audio_push_back(source, &in);
	else
		source_output_audio_place(source, &in);

	pthread_mutex_unlock(&source->audio_buf_mutex);

1226
	source_signal_audio_data(source, &in, source_muted(source, os_time));
1227 1228
}

1229 1230 1231 1232 1233 1234 1235 1236
enum convert_type {
	CONVERT_NONE,
	CONVERT_NV12,
	CONVERT_420,
	CONVERT_422_U,
	CONVERT_422_Y,
};

1237
static inline enum convert_type get_convert_type(enum video_format format)
1238
{
1239
	switch (format) {
1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250
	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 已提交
1251
	case VIDEO_FORMAT_Y800:
J
jp9000 已提交
1252
	case VIDEO_FORMAT_I444:
1253
	case VIDEO_FORMAT_NONE:
1254 1255 1256 1257 1258 1259 1260 1261 1262
	case VIDEO_FORMAT_RGBA:
	case VIDEO_FORMAT_BGRA:
	case VIDEO_FORMAT_BGRX:
		return CONVERT_NONE;
	}

	return CONVERT_NONE;
}

1263
static inline bool set_packed422_sizes(struct obs_source *source,
1264
		const struct obs_source_frame *frame)
1265 1266
{
	source->async_convert_height = frame->height;
1267 1268 1269 1270 1271 1272
	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,
1273
		const struct obs_source_frame *frame)
1274 1275 1276 1277 1278
{
	uint32_t size = frame->width * frame->height;
	size += size/2;

	source->async_convert_width   = frame->width;
1279
	source->async_convert_height  = size / frame->width;
1280
	source->async_texture_format  = GS_R8;
1281 1282
	source->async_plane_offset[0] = (int)(frame->data[1] - frame->data[0]);
	source->async_plane_offset[1] = (int)(frame->data[2] - frame->data[0]);
1283 1284 1285
	return true;
}

J
jp9000 已提交
1286
static inline bool set_nv12_sizes(struct obs_source *source,
1287
		const struct obs_source_frame *frame)
J
jp9000 已提交
1288 1289 1290 1291 1292
{
	uint32_t size = frame->width * frame->height;
	size += size/2;

	source->async_convert_width   = frame->width;
1293
	source->async_convert_height  = size / frame->width;
J
jp9000 已提交
1294
	source->async_texture_format  = GS_R8;
1295
	source->async_plane_offset[0] = (int)(frame->data[1] - frame->data[0]);
J
jp9000 已提交
1296 1297 1298
	return true;
}

1299
static inline bool init_gpu_conversion(struct obs_source *source,
1300
		const struct obs_source_frame *frame)
1301 1302 1303 1304 1305 1306 1307
{
	switch (get_convert_type(frame->format)) {
		case CONVERT_422_Y:
		case CONVERT_422_U:
			return set_packed422_sizes(source, frame);

		case CONVERT_420:
1308 1309 1310
			return set_planar420_sizes(source, frame);

		case CONVERT_NV12:
J
jp9000 已提交
1311
			return set_nv12_sizes(source, frame);
1312 1313 1314 1315 1316 1317 1318 1319 1320 1321
			break;

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

	}
	return false;
}

1322
bool set_async_texture_size(struct obs_source *source,
1323
		const struct obs_source_frame *frame)
1324
{
1325 1326
	enum convert_type cur = get_convert_type(frame->format);

J
jp9000 已提交
1327 1328 1329
	if (source->async_width  == frame->width  &&
	    source->async_height == frame->height &&
	    source->async_format == frame->format)
1330 1331
		return true;

J
jp9000 已提交
1332 1333 1334
	source->async_width  = frame->width;
	source->async_height = frame->height;
	source->async_format = frame->format;
1335

1336
	gs_texture_destroy(source->async_texture);
J
jp9000 已提交
1337
	gs_texture_destroy(source->async_prev_texture);
1338
	gs_texrender_destroy(source->async_texrender);
J
jp9000 已提交
1339 1340 1341
	gs_texrender_destroy(source->async_prev_texrender);
	source->async_texture = NULL;
	source->async_prev_texture = NULL;
1342
	source->async_texrender = NULL;
J
jp9000 已提交
1343
	source->async_prev_texrender = NULL;
1344 1345 1346 1347

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

1348
		source->async_texrender =
1349
			gs_texrender_create(GS_BGRX, GS_ZS_NONE);
1350

1351
		source->async_texture = gs_texture_create(
1352 1353
				source->async_convert_width,
				source->async_convert_height,
1354 1355
				source->async_texture_format,
				1, NULL, GS_DYNAMIC);
1356 1357

	} else {
1358 1359
		enum gs_color_format format = convert_video_format(
				frame->format);
1360 1361
		source->async_gpu_conversion = false;

1362
		source->async_texture = gs_texture_create(
1363
				frame->width, frame->height,
1364
				format, 1, NULL, GS_DYNAMIC);
1365 1366
	}

J
jp9000 已提交
1367 1368 1369
	if (deinterlacing_enabled(source))
		set_deinterlace_texture_size(source);

1370
	return !!source->async_texture;
1371 1372
}

1373
static void upload_raw_frame(gs_texture_t *tex,
1374
		const struct obs_source_frame *frame)
1375 1376 1377 1378
{
	switch (get_convert_type(frame->format)) {
		case CONVERT_422_U:
		case CONVERT_422_Y:
1379
			gs_texture_set_image(tex, frame->data[0],
1380 1381 1382 1383
					frame->linesize[0], false);
			break;

		case CONVERT_420:
1384
			gs_texture_set_image(tex, frame->data[0],
1385 1386 1387 1388
					frame->width, false);
			break;

		case CONVERT_NV12:
J
jp9000 已提交
1389 1390
			gs_texture_set_image(tex, frame->data[0],
					frame->width, false);
1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402
			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:
1403
			return "UYVY_Reverse";
1404 1405 1406 1407 1408 1409 1410 1411

		case VIDEO_FORMAT_YUY2:
			return "YUY2_Reverse";

		case VIDEO_FORMAT_YVYU:
			return "YVYU_Reverse";

		case VIDEO_FORMAT_I420:
1412 1413 1414
			return "I420_Reverse";

		case VIDEO_FORMAT_NV12:
J
jp9000 已提交
1415
			return "NV12_Reverse";
1416 1417
			break;

J
jpk 已提交
1418
		case VIDEO_FORMAT_Y800:
1419 1420 1421 1422
		case VIDEO_FORMAT_BGRA:
		case VIDEO_FORMAT_BGRX:
		case VIDEO_FORMAT_RGBA:
		case VIDEO_FORMAT_NONE:
J
jp9000 已提交
1423
		case VIDEO_FORMAT_I444:
1424 1425 1426 1427 1428 1429
			assert(false && "No conversion requested");
			break;
	}
	return NULL;
}

1430
static inline void set_eparam(gs_effect_t *effect, const char *name, float val)
1431
{
1432
	gs_eparam_t *param = gs_effect_get_param_by_name(effect, name);
1433
	gs_effect_set_float(param, val);
1434 1435 1436
}

static bool update_async_texrender(struct obs_source *source,
1437 1438
		const struct obs_source_frame *frame,
		gs_texture_t *tex, gs_texrender_t *texrender)
1439
{
1440
	gs_texrender_reset(texrender);
1441 1442 1443 1444 1445 1446

	upload_raw_frame(tex, frame);

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

1447 1448 1449
	float convert_width  = (float)source->async_convert_width;
	float convert_height = (float)source->async_convert_height;

1450 1451
	gs_effect_t *conv = obs->video.conversion_effect;
	gs_technique_t *tech = gs_effect_get_technique(conv,
1452 1453
			select_conversion_technique(frame->format));

1454
	if (!gs_texrender_begin(texrender, cx, cy))
1455 1456
		return false;

1457 1458
	gs_technique_begin(tech);
	gs_technique_begin_pass(tech, 0);
1459

1460
	gs_effect_set_texture(gs_effect_get_param_by_name(conv, "image"), tex);
1461 1462 1463 1464
	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);
1465
	set_eparam(conv, "width_d2",  cx * 0.5f);
1466
	set_eparam(conv, "height_d2", cy * 0.5f);
1467
	set_eparam(conv, "width_d2_i",  1.0f / (cx * 0.5f));
1468
	set_eparam(conv, "height_d2_i", 1.0f / (cy * 0.5f));
1469 1470 1471 1472 1473 1474 1475 1476 1477 1478
	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]);
1479 1480 1481 1482 1483

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

	gs_draw_sprite(tex, 0, cx, cy);

1484 1485
	gs_technique_end_pass(tech);
	gs_technique_end(tech);
1486

1487
	gs_texrender_end(texrender);
1488 1489 1490 1491

	return true;
}

1492 1493 1494
bool update_async_texture(struct obs_source *source,
		const struct obs_source_frame *frame,
		gs_texture_t *tex, gs_texrender_t *texrender)
1495
{
1496
	enum convert_type type      = get_convert_type(frame->format);
1497
	uint8_t           *ptr;
1498 1499
	uint32_t          linesize;

1500 1501
	source->async_flip       = frame->flip;
	source->async_full_range = frame->full_range;
1502 1503
	memcpy(source->async_color_matrix, frame->color_matrix,
			sizeof(frame->color_matrix));
1504 1505 1506 1507
	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);
1508

1509
	if (source->async_gpu_conversion && texrender)
1510
		return update_async_texrender(source, frame, tex, texrender);
1511

1512
	if (type == CONVERT_NONE) {
1513
		gs_texture_set_image(tex, frame->data[0], frame->linesize[0],
1514
				false);
1515 1516 1517
		return true;
	}

1518
	if (!gs_texture_map(tex, &ptr, &linesize))
1519 1520 1521
		return false;

	if (type == CONVERT_420)
J
jp9000 已提交
1522 1523 1524
		decompress_420((const uint8_t* const*)frame->data,
				frame->linesize,
				0, frame->height, ptr, linesize);
1525 1526

	else if (type == CONVERT_NV12)
J
jp9000 已提交
1527 1528 1529
		decompress_nv12((const uint8_t* const*)frame->data,
				frame->linesize,
				0, frame->height, ptr, linesize);
1530 1531

	else if (type == CONVERT_422_Y)
1532
		decompress_422(frame->data[0], frame->linesize[0],
J
jp9000 已提交
1533
				0, frame->height, ptr, linesize, true);
1534 1535

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

1539
	gs_texture_unmap(tex);
1540 1541 1542
	return true;
}

1543
static inline void obs_source_draw_texture(struct obs_source *source,
1544
		gs_effect_t *effect, float *color_matrix,
1545
		float const *color_range_min, float const *color_range_max)
1546
{
1547 1548
	gs_texture_t *tex = source->async_texture;
	gs_eparam_t  *param;
1549

1550 1551
	if (source->async_texrender)
		tex = gs_texrender_get_texture(source->async_texrender);
1552

P
Palana 已提交
1553
	if (color_range_min) {
1554
		size_t const size = sizeof(float) * 3;
1555 1556
		param = gs_effect_get_param_by_name(effect, "color_range_min");
		gs_effect_set_val(param, color_range_min, size);
P
Palana 已提交
1557
	}
1558

P
Palana 已提交
1559 1560
	if (color_range_max) {
		size_t const size = sizeof(float) * 3;
1561 1562
		param = gs_effect_get_param_by_name(effect, "color_range_max");
		gs_effect_set_val(param, color_range_max, size);
P
Palana 已提交
1563
	}
1564

P
Palana 已提交
1565
	if (color_matrix) {
1566 1567
		param = gs_effect_get_param_by_name(effect, "color_matrix");
		gs_effect_set_val(param, color_matrix, sizeof(float) * 16);
1568 1569
	}

1570 1571
	param = gs_effect_get_param_by_name(effect, "image");
	gs_effect_set_texture(param, tex);
1572

1573 1574
	gs_draw_sprite(tex, source->async_flip ? GS_FLIP_V : 0, 0, 0);
}
1575

1576 1577
static void obs_source_draw_async_texture(struct obs_source *source)
{
1578
	gs_effect_t    *effect        = gs_get_effect();
1579 1580 1581 1582
	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);
1583
	gs_technique_t *tech          = NULL;
1584 1585

	if (def_draw) {
1586
		effect = obs_get_base_effect(OBS_EFFECT_DEFAULT);
1587 1588 1589
		tech = gs_effect_get_technique(effect, type);
		gs_technique_begin(tech);
		gs_technique_begin_pass(tech, 0);
1590 1591 1592
	}

	obs_source_draw_texture(source, effect,
1593 1594 1595
			yuv ? source->async_color_matrix : NULL,
			limited_range ? source->async_color_range_min : NULL,
			limited_range ? source->async_color_range_max : NULL);
1596 1597

	if (def_draw) {
1598 1599
		gs_technique_end_pass(tech);
		gs_technique_end(tech);
1600
	}
1601 1602
}

1603
static void obs_source_update_async_video(obs_source_t *source)
1604
{
1605 1606 1607
	if (!source->async_rendered) {
		struct obs_source_frame *frame = obs_source_get_frame(source);

1608 1609 1610
		if (frame)
			frame = filter_async_video(source, frame);

1611 1612
		source->async_rendered = true;
		if (frame) {
1613 1614 1615 1616
			source->timing_adjust =
				os_gettime_ns() - frame->timestamp;
			source->timing_set = true;

1617 1618 1619
			if (set_async_texture_size(source, frame)) {
				update_async_texture(source, frame,
						source->async_texture,
1620
						source->async_texrender);
1621
			}
1622

1623 1624
			obs_source_release_frame(source, frame);
		}
1625
	}
1626
}
1627

1628 1629
static inline void obs_source_render_async_video(obs_source_t *source)
{
1630
	if (source->async_texture && source->async_active)
1631
		obs_source_draw_async_texture(source);
1632 1633
}

1634
static inline void obs_source_render_filters(obs_source_t *source)
1635 1636 1637 1638 1639 1640
{
	source->rendering_filter = true;
	obs_source_video_render(source->filters.array[0]);
	source->rendering_filter = false;
}

1641
static void obs_source_default_render(obs_source_t *source)
1642
{
1643
	gs_effect_t    *effect     = obs->video.default_effect;
1644
	gs_technique_t *tech       = gs_effect_get_technique(effect, "Draw");
1645
	size_t         passes, i;
1646

1647
	passes = gs_technique_begin(tech);
1648
	for (i = 0; i < passes; i++) {
1649
		gs_technique_begin_pass(tech, i);
1650 1651
		if (source->context.data)
			source->info.video_render(source->context.data, effect);
1652
		gs_technique_end_pass(tech);
1653
	}
1654
	gs_technique_end(tech);
1655 1656
}

1657
static inline void obs_source_main_render(obs_source_t *source)
1658
{
1659 1660
	uint32_t flags      = source->info.output_flags;
	bool custom_draw    = (flags & OBS_SOURCE_CUSTOM_DRAW) != 0;
1661 1662
	bool default_effect = !source->filter_parent &&
	                      source->filters.num == 0 &&
1663
	                      !custom_draw;
1664 1665

	if (default_effect)
1666
		obs_source_default_render(source);
1667
	else if (source->context.data)
1668
		source->info.video_render(source->context.data,
1669
				custom_draw ? NULL : gs_get_effect());
1670 1671
}

1672 1673
static bool ready_async_frame(obs_source_t *source, uint64_t sys_time);

1674
static inline void render_video(obs_source_t *source)
J
jp9000 已提交
1675
{
1676 1677
	if (source->info.type != OBS_SOURCE_TYPE_FILTER &&
	    (source->info.output_flags & OBS_SOURCE_VIDEO) == 0)
1678 1679
		return;

1680 1681
	if (source->info.type == OBS_SOURCE_TYPE_INPUT &&
	    (source->info.output_flags & OBS_SOURCE_ASYNC) != 0 &&
J
jp9000 已提交
1682 1683 1684
	    !source->rendering_filter) {
		if (deinterlacing_enabled(source))
			deinterlace_update_async_video(source);
1685
		obs_source_update_async_video(source);
J
jp9000 已提交
1686
	}
1687

1688
	if (!source->context.data || !source->enabled) {
1689 1690 1691 1692 1693
		if (source->filter_parent)
			obs_source_skip_video_filter(source);
		return;
	}

1694 1695
	if (source->filters.num && !source->rendering_filter)
		obs_source_render_filters(source);
1696

1697 1698 1699 1700
	else if (source->info.video_render)
		obs_source_main_render(source);

	else if (source->filter_target)
1701 1702
		obs_source_video_render(source->filter_target);

J
jp9000 已提交
1703 1704 1705
	else if (deinterlacing_enabled(source))
		deinterlace_render(source);

1706
	else
1707
		obs_source_render_async_video(source);
J
jp9000 已提交
1708 1709
}

1710 1711 1712 1713 1714 1715 1716 1717 1718 1719
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);
}

1720
static uint32_t get_base_width(const obs_source_t *source)
J
jp9000 已提交
1721
{
1722 1723
	bool is_filter = (source->info.type == OBS_SOURCE_TYPE_FILTER);

J
jp9000 已提交
1724 1725 1726 1727
	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)) {
1728
		return source->info.get_width(source->context.data);
1729 1730 1731 1732 1733

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

1734
	return source->async_active ? source->async_width : 0;
J
jp9000 已提交
1735 1736
}

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

J
jp9000 已提交
1741 1742 1743 1744
	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)) {
1745
		return source->info.get_height(source->context.data);
1746

1747
	} else if (is_filter) {
1748 1749 1750
		return get_base_height(source->filter_target);
	}

1751
	return source->async_active ? source->async_height : 0;
J
jp9000 已提交
1752 1753
}

1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785
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 已提交
1786 1787
	if (!data_valid(source, "obs_source_get_width"))
		return 0;
1788

1789
	return (source->info.type != OBS_SOURCE_TYPE_FILTER) ?
1790 1791 1792 1793 1794 1795
		get_recurse_width(source) :
	        get_base_width(source);
}

uint32_t obs_source_get_height(obs_source_t *source)
{
J
jp9000 已提交
1796 1797
	if (!data_valid(source, "obs_source_get_height"))
		return 0;
1798

1799
	return (source->info.type != OBS_SOURCE_TYPE_FILTER) ?
1800 1801 1802 1803
		get_recurse_height(source) :
		get_base_height(source);
}

1804 1805
uint32_t obs_source_get_base_width(obs_source_t *source)
{
J
jp9000 已提交
1806 1807
	if (!data_valid(source, "obs_source_get_base_width"))
		return 0;
1808 1809 1810 1811 1812 1813

	return get_base_width(source);
}

uint32_t obs_source_get_base_height(obs_source_t *source)
{
J
jp9000 已提交
1814 1815
	if (!data_valid(source, "obs_source_get_base_height"))
		return 0;
1816 1817 1818 1819

	return get_base_height(source);
}

1820
obs_source_t *obs_filter_get_parent(const obs_source_t *filter)
1821
{
1822 1823
	return obs_ptr_valid(filter, "obs_filter_get_parent") ?
		filter->filter_parent : NULL;
1824 1825
}

1826
obs_source_t *obs_filter_get_target(const obs_source_t *filter)
J
jp9000 已提交
1827
{
1828 1829
	return obs_ptr_valid(filter, "obs_filter_get_target") ?
		filter->filter_target : NULL;
J
jp9000 已提交
1830 1831
}

1832
void obs_source_filter_add(obs_source_t *source, obs_source_t *filter)
J
jp9000 已提交
1833
{
1834 1835
	struct calldata cd;
	uint8_t stack[128];
J
jp9000 已提交
1836

1837 1838 1839
	if (!obs_source_valid(source, "obs_source_filter_add"))
		return;
	if (!obs_ptr_valid(filter, "obs_source_filter_add"))
J
jp9000 已提交
1840 1841
		return;

1842 1843
	pthread_mutex_lock(&source->filter_mutex);

J
jp9000 已提交
1844
	if (da_find(source->filters, &filter, 0) != DARRAY_INVALID) {
J
jp9000 已提交
1845 1846
		blog(LOG_WARNING, "Tried to add a filter that was already "
		                  "present on the source");
1847
		pthread_mutex_unlock(&source->filter_mutex);
J
jp9000 已提交
1848 1849 1850
		return;
	}

1851 1852
	obs_source_addref(filter);

1853
	filter->filter_parent = source;
J
jp9000 已提交
1854 1855
	filter->filter_target = !source->filters.num ?
		source : source->filters.array[0];
1856

J
jp9000 已提交
1857
	da_insert(source->filters, 0, &filter);
1858 1859 1860

	pthread_mutex_unlock(&source->filter_mutex);

1861
	calldata_init_fixed(&cd, stack, sizeof(stack));
J
jp9000 已提交
1862 1863 1864 1865
	calldata_set_ptr(&cd, "source", source);
	calldata_set_ptr(&cd, "filter", filter);

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

	if (source && filter)
1868
		blog(LOG_DEBUG, "- filter '%s' (%s) added to source '%s'",
1869 1870
				filter->context.name, filter->info.id,
				source->context.name);
J
jp9000 已提交
1871 1872
}

1873 1874
static bool obs_source_filter_remove_refless(obs_source_t *source,
		obs_source_t *filter)
J
jp9000 已提交
1875
{
1876 1877
	struct calldata cd;
	uint8_t stack[128];
1878 1879 1880 1881 1882
	size_t idx;

	pthread_mutex_lock(&source->filter_mutex);

	idx = da_find(source->filters, &filter, 0);
1883 1884
	if (idx == DARRAY_INVALID) {
		pthread_mutex_unlock(&source->filter_mutex);
1885
		return false;
1886
	}
J
jp9000 已提交
1887 1888

	if (idx > 0) {
1889
		obs_source_t *prev = source->filters.array[idx-1];
J
jp9000 已提交
1890 1891 1892 1893
		prev->filter_target = filter->filter_target;
	}

	da_erase(source->filters, idx);
1894 1895 1896

	pthread_mutex_unlock(&source->filter_mutex);

1897
	calldata_init_fixed(&cd, stack, sizeof(stack));
J
jp9000 已提交
1898 1899 1900 1901 1902
	calldata_set_ptr(&cd, "source", source);
	calldata_set_ptr(&cd, "filter", filter);

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

1903
	if (source && filter)
1904
		blog(LOG_DEBUG, "- filter '%s' (%s) removed from source '%s'",
1905 1906 1907
				filter->context.name, filter->info.id,
				source->context.name);

1908 1909 1910 1911
	if (filter->info.filter_remove)
		filter->info.filter_remove(filter->context.data,
				filter->filter_parent);

1912
	filter->filter_parent = NULL;
J
jp9000 已提交
1913
	filter->filter_target = NULL;
1914 1915
	return true;
}
1916

1917 1918
void obs_source_filter_remove(obs_source_t *source, obs_source_t *filter)
{
1919 1920 1921 1922 1923
	if (!obs_source_valid(source, "obs_source_filter_remove"))
		return;
	if (!obs_ptr_valid(filter, "obs_source_filter_remove"))
		return;

1924 1925
	if (obs_source_filter_remove_refless(source, filter))
		obs_source_release(filter);
J
jp9000 已提交
1926 1927
}

1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 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
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 */
1967 1968
static bool move_filter_dir(obs_source_t *source,
		obs_source_t *filter, enum obs_order_movement movement)
J
jp9000 已提交
1969
{
1970
	size_t idx;
J
jp9000 已提交
1971 1972

	idx = da_find(source->filters, &filter, 0);
J
jp9000 已提交
1973
	if (idx == DARRAY_INVALID)
1974
		return false;
J
jp9000 已提交
1975

J
jp9000 已提交
1976
	if (movement == OBS_ORDER_MOVE_UP) {
1977 1978
		size_t next_id = find_next_filter(source, filter, idx);
		if (next_id == DARRAY_INVALID)
1979
			return false;
1980
		da_move_item(source->filters, idx, next_id);
J
jp9000 已提交
1981

J
jp9000 已提交
1982
	} else if (movement == OBS_ORDER_MOVE_DOWN) {
1983 1984
		size_t prev_id = find_prev_filter(source, filter, idx);
		if (prev_id == DARRAY_INVALID)
1985
			return false;
1986
		da_move_item(source->filters, idx, prev_id);
J
jp9000 已提交
1987

J
jp9000 已提交
1988
	} else if (movement == OBS_ORDER_MOVE_TOP) {
J
jp9000 已提交
1989
		if (idx == source->filters.num-1)
1990
			return false;
J
jp9000 已提交
1991 1992
		da_move_item(source->filters, idx, source->filters.num-1);

J
jp9000 已提交
1993
	} else if (movement == OBS_ORDER_MOVE_BOTTOM) {
J
jp9000 已提交
1994
		if (idx == 0)
1995
			return false;
J
jp9000 已提交
1996 1997 1998
		da_move_item(source->filters, idx, 0);
	}

1999
	/* reorder filter targets, not the nicest way of dealing with things */
2000
	for (size_t i = 0; i < source->filters.num; i++) {
2001
		obs_source_t *next_filter = (i == source->filters.num-1) ?
2002 2003
			source : source->filters.array[i + 1];

J
jp9000 已提交
2004 2005
		source->filters.array[i]->filter_target = next_filter;
	}
2006

2007 2008 2009 2010 2011 2012 2013
	return true;
}

void obs_source_filter_set_order(obs_source_t *source, obs_source_t *filter,
		enum obs_order_movement movement)
{
	bool success;
2014 2015 2016 2017

	if (!obs_source_valid(source, "obs_source_filter_set_order"))
		return;
	if (!obs_ptr_valid(filter, "obs_source_filter_set_order"))
2018 2019 2020 2021 2022 2023 2024 2025
		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 已提交
2026 2027
}

2028
obs_data_t *obs_source_get_settings(const obs_source_t *source)
J
jp9000 已提交
2029
{
2030 2031
	if (!obs_source_valid(source, "obs_source_get_settings"))
		return NULL;
J
jp9000 已提交
2032

2033 2034
	obs_data_addref(source->context.settings);
	return source->context.settings;
J
jp9000 已提交
2035 2036
}

2037
struct obs_source_frame *filter_async_video(obs_source_t *source,
2038
		struct obs_source_frame *in)
2039 2040
{
	size_t i;
2041 2042 2043

	pthread_mutex_lock(&source->filter_mutex);

2044 2045
	for (i = source->filters.num; i > 0; i--) {
		struct obs_source *filter = source->filters.array[i-1];
2046

2047 2048 2049
		if (!filter->enabled)
			continue;

2050
		if (filter->context.data && filter->info.filter_video) {
2051 2052
			in = filter->info.filter_video(filter->context.data,
					in);
2053
			if (!in)
2054
				break;
2055 2056 2057
		}
	}

2058 2059
	pthread_mutex_unlock(&source->filter_mutex);

2060 2061 2062
	return in;
}

2063 2064
static inline void copy_frame_data_line(struct obs_source_frame *dst,
		const struct obs_source_frame *src, uint32_t plane, uint32_t y)
2065
{
2066 2067 2068 2069
	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];
2070 2071 2072 2073

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

2074 2075 2076
static inline void copy_frame_data_plane(struct obs_source_frame *dst,
		const struct obs_source_frame *src,
		uint32_t plane, uint32_t lines)
2077
{
2078
	if (dst->linesize[plane] != src->linesize[plane])
2079 2080 2081 2082
		for (uint32_t y = 0; y < lines; y++)
			copy_frame_data_line(dst, src, plane, y);
	else
		memcpy(dst->data[plane], src->data[plane],
2083
				dst->linesize[plane] * lines);
2084 2085
}

2086 2087
static void copy_frame_data(struct obs_source_frame *dst,
		const struct obs_source_frame *src)
2088 2089
{
	dst->flip         = src->flip;
2090
	dst->full_range   = src->full_range;
2091 2092
	dst->timestamp    = src->timestamp;
	memcpy(dst->color_matrix, src->color_matrix, sizeof(float) * 16);
2093 2094 2095 2096 2097
	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);
	}
2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110

	switch (dst->format) {
	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 已提交
2111 2112 2113 2114 2115 2116
	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;

2117 2118 2119
	case VIDEO_FORMAT_YVYU:
	case VIDEO_FORMAT_YUY2:
	case VIDEO_FORMAT_UYVY:
J
jpk 已提交
2120
	case VIDEO_FORMAT_Y800:
2121 2122 2123 2124 2125 2126 2127 2128
	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);
	}
}

2129 2130 2131 2132
static inline bool async_texture_changed(struct obs_source *source,
		const struct obs_source_frame *frame)
{
	enum convert_type prev, cur;
J
jp9000 已提交
2133
	prev = get_convert_type(source->async_cache_format);
2134 2135
	cur  = get_convert_type(frame->format);

J
jp9000 已提交
2136 2137
	return source->async_cache_width  != frame->width ||
	       source->async_cache_height != frame->height ||
2138 2139 2140 2141 2142 2143
	       prev != cur;
}

static inline void free_async_cache(struct obs_source *source)
{
	for (size_t i = 0; i < source->async_cache.num; i++)
2144
		obs_source_frame_decref(source->async_cache.array[i].frame);
2145 2146 2147

	da_resize(source->async_cache, 0);
	da_resize(source->async_frames, 0);
2148
	source->cur_async_frame = NULL;
J
jp9000 已提交
2149
	source->prev_async_frame = NULL;
2150 2151
}

2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168
#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);
			}
		}
	}
}

2169 2170
#define MAX_ASYNC_FRAMES 30

2171
static inline struct obs_source_frame *cache_video(struct obs_source *source,
2172
		const struct obs_source_frame *frame)
2173
{
2174 2175 2176 2177
	struct obs_source_frame *new_frame = NULL;

	pthread_mutex_lock(&source->async_mutex);

2178 2179 2180 2181 2182 2183 2184
	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;
	}

2185
	if (async_texture_changed(source, frame)) {
J
jp9000 已提交
2186 2187 2188 2189
		free_async_cache(source);
		source->async_cache_width  = frame->width;
		source->async_cache_height = frame->height;
		source->async_cache_format = frame->format;
2190 2191 2192 2193 2194 2195 2196
	}

	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;
2197
			af->unused_count = 0;
2198
			break;
2199 2200 2201
		}
	}

2202 2203
	clean_cache(source);

2204 2205 2206 2207 2208 2209 2210
	if (!new_frame) {
		struct async_frame new_af;

		new_frame = obs_source_frame_create(frame->format,
				frame->width, frame->height);
		new_af.frame = new_frame;
		new_af.used = true;
2211
		new_af.unused_count = 0;
2212
		new_frame->refs = 1;
2213 2214 2215 2216

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

2217 2218
	os_atomic_inc_long(&new_frame->refs);

2219
	pthread_mutex_unlock(&source->async_mutex);
2220

2221
	copy_frame_data(new_frame, frame);
2222 2223 2224 2225 2226 2227

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

2228
	return new_frame;
2229 2230
}

2231
void obs_source_output_video(obs_source_t *source,
2232
		const struct obs_source_frame *frame)
2233
{
2234
	if (!obs_source_valid(source, "obs_source_output_video"))
J
jp9000 已提交
2235 2236
		return;

2237 2238 2239 2240 2241
	if (!frame) {
		source->async_active = false;
		return;
	}

2242 2243
	struct obs_source_frame *output = !!frame ?
		cache_video(source, frame) : NULL;
2244

2245 2246
	/* ------------------------------------------- */

2247
	if (output) {
2248 2249 2250
		pthread_mutex_lock(&source->async_mutex);
		da_push_back(source->async_frames, &output);
		pthread_mutex_unlock(&source->async_mutex);
2251
		source->async_active = true;
2252
	}
2253 2254
}

2255
static inline struct obs_audio_data *filter_async_audio(obs_source_t *source,
2256
		struct obs_audio_data *in)
2257 2258 2259 2260
{
	size_t i;
	for (i = source->filters.num; i > 0; i--) {
		struct obs_source *filter = source->filters.array[i-1];
2261

2262 2263 2264
		if (!filter->enabled)
			continue;

2265
		if (filter->context.data && filter->info.filter_audio) {
2266 2267
			in = filter->info.filter_audio(filter->context.data,
					in);
2268 2269 2270 2271 2272 2273 2274 2275
			if (!in)
				return NULL;
		}
	}

	return in;
}

2276
static inline void reset_resampler(obs_source_t *source,
2277
		const struct obs_source_audio *audio)
2278
{
J
jp9000 已提交
2279
	const struct audio_output_info *obs_info;
2280 2281
	struct resample_info output_info;

2282
	obs_info = audio_output_get_info(obs->audio.audio);
2283

2284 2285 2286 2287 2288 2289 2290 2291
	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;

2292 2293
	audio_resampler_destroy(source->resampler);
	source->resampler = NULL;
2294
	source->resample_offset = 0;
2295

2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310
	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 已提交
2311
static void copy_audio_data(obs_source_t *source,
J
jp9000 已提交
2312
		const uint8_t *const data[], uint32_t frames, uint64_t ts)
2313
{
2314 2315
	size_t planes    = audio_output_get_planes(obs->audio.audio);
	size_t blocksize = audio_output_get_block_size(obs->audio.audio);
2316 2317
	size_t size      = (size_t)frames * blocksize;
	bool   resize    = source->audio_storage_size < size;
2318

J
jp9000 已提交
2319 2320
	source->audio_data.frames    = frames;
	source->audio_data.timestamp = ts;
2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333

	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;
2334 2335
}

2336 2337 2338
/* TODO: SSE optimization */
static void downmix_to_mono_planar(struct obs_source *source, uint32_t frames)
{
J
jp9000 已提交
2339
	size_t channels = audio_output_get_channels(obs->audio.audio);
2340 2341 2342
	const float channels_i = 1.0f / (float)channels;
	float **data = (float**)source->audio_data.data;

J
jp9000 已提交
2343
	for (size_t channel = 1; channel < channels; channel++) {
2344 2345 2346 2347 2348 2349 2350
		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 已提交
2351
	for (size_t channel = 1; channel < channels; channel++) {
2352 2353 2354 2355 2356
		for (uint32_t frame = 0; frame < frames; frame++)
			data[channel][frame] = data[0][frame];
	}
}

2357
/* resamples/remixes new audio to the designated main audio output format */
2358
static void process_audio(obs_source_t *source,
2359
		const struct obs_source_audio *audio)
2360
{
2361
	uint32_t frames = audio->frames;
2362
	bool mono_output;
2363

2364 2365 2366 2367 2368 2369 2370 2371 2372
	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 已提交
2373
		uint8_t  *output[MAX_AV_PLANES];
2374

2375 2376 2377
		memset(output, 0, sizeof(output));

		audio_resampler_resample(source->resampler,
2378
				output, &frames, &source->resample_offset,
2379
				audio->data, audio->frames);
2380

J
jp9000 已提交
2381
		copy_audio_data(source, (const uint8_t *const *)output, frames,
2382
				audio->timestamp);
2383 2384 2385 2386
	} else {
		copy_audio_data(source, audio->data, audio->frames,
				audio->timestamp);
	}
2387

2388 2389 2390 2391
	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);
2392 2393
}

2394
void obs_source_output_audio(obs_source_t *source,
2395
		const struct obs_source_audio *audio)
2396
{
2397
	struct obs_audio_data *output;
2398

2399 2400 2401
	if (!obs_source_valid(source, "obs_source_output_audio"))
		return;
	if (!obs_ptr_valid(audio, "obs_source_output_audio"))
J
jp9000 已提交
2402 2403
		return;

2404
	process_audio(source, audio);
2405 2406

	pthread_mutex_lock(&source->filter_mutex);
2407
	output = filter_async_audio(source, &source->audio_data);
2408 2409

	if (output) {
2410
		struct audio_data data;
J
jp9000 已提交
2411

2412 2413
		for (int i = 0; i < MAX_AV_PLANES; i++)
			data.data[i] = output->data[i];
2414

2415 2416
		data.frames    = output->frames;
		data.timestamp = output->timestamp;
2417

2418
		pthread_mutex_lock(&source->audio_mutex);
2419
		source_output_audio_data(source, &data);
2420 2421 2422 2423 2424 2425
		pthread_mutex_unlock(&source->audio_mutex);
	}

	pthread_mutex_unlock(&source->filter_mutex);
}

2426
void remove_async_frame(obs_source_t *source, struct obs_source_frame *frame)
2427
{
J
jp9000 已提交
2428 2429 2430
	if (frame)
		frame->prev_frame = false;

2431 2432 2433 2434 2435 2436 2437 2438 2439 2440
	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 已提交
2441 2442
/* #define DEBUG_ASYNC_FRAMES 1 */

2443
static bool ready_async_frame(obs_source_t *source, uint64_t sys_time)
2444
{
2445
	struct obs_source_frame *next_frame = source->async_frames.array[0];
2446
	struct obs_source_frame *frame      = NULL;
2447 2448 2449 2450
	uint64_t sys_offset = sys_time - source->last_sys_timestamp;
	uint64_t frame_time = next_frame->timestamp;
	uint64_t frame_offset = 0;

2451
	if ((source->flags & OBS_SOURCE_FLAG_UNBUFFERED) != 0) {
2452 2453
		while (source->async_frames.num > 1) {
			da_erase(source->async_frames, 0);
2454
			remove_async_frame(source, next_frame);
2455
			next_frame = source->async_frames.array[0];
J
jp9000 已提交
2456 2457 2458 2459 2460
		}

		return true;
	}

J
jp9000 已提交
2461 2462 2463 2464 2465 2466
#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,
2467
			(unsigned long)source->async_frames.num);
J
jp9000 已提交
2468 2469
#endif

2470 2471
	/* account for timestamp invalidation */
	if (frame_out_of_bounds(source, frame_time)) {
J
jp9000 已提交
2472 2473 2474
#if DEBUG_ASYNC_FRAMES
		blog(LOG_DEBUG, "timing jump");
#endif
2475
		source->last_frame_ts = next_frame->timestamp;
J
jp9000 已提交
2476
		return true;
2477 2478
	} else {
		frame_offset = frame_time - source->last_frame_ts;
J
jp9000 已提交
2479
		source->last_frame_ts += sys_offset;
2480 2481
	}

J
jp9000 已提交
2482 2483 2484 2485 2486 2487
	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 */
2488
		if ((source->last_frame_ts - next_frame->timestamp) < 2000000)
J
jp9000 已提交
2489 2490 2491
			break;

		if (frame)
2492
			da_erase(source->async_frames, 0);
J
jp9000 已提交
2493 2494 2495 2496 2497 2498 2499 2500 2501

#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

2502
		remove_async_frame(source, frame);
2503

2504
		if (source->async_frames.num == 1)
2505 2506
			return true;

2507
		frame = next_frame;
2508
		next_frame = source->async_frames.array[1];
2509 2510

		/* more timestamp checking and compensating */
2511
		if ((next_frame->timestamp - frame_time) > MAX_TS_VAR) {
J
jp9000 已提交
2512 2513 2514
#if DEBUG_ASYNC_FRAMES
			blog(LOG_DEBUG, "timing jump");
#endif
2515 2516 2517 2518 2519 2520 2521 2522
			source->last_frame_ts =
				next_frame->timestamp - frame_offset;
		}

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

J
jp9000 已提交
2523 2524 2525 2526
#if DEBUG_ASYNC_FRAMES
	if (!frame)
		blog(LOG_DEBUG, "no frame!");
#endif
2527

2528 2529 2530
	return frame != NULL;
}

2531
static inline struct obs_source_frame *get_closest_frame(obs_source_t *source,
2532 2533
		uint64_t sys_time)
{
2534 2535 2536 2537
	if (!source->async_frames.num)
		return NULL;

	if (!source->last_frame_ts || ready_async_frame(source, sys_time)) {
2538 2539
		struct obs_source_frame *frame = source->async_frames.array[0];
		da_erase(source->async_frames, 0);
2540 2541 2542 2543

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

2544 2545 2546 2547
		return frame;
	}

	return NULL;
2548 2549
}

2550
/*
2551 2552
 * Ensures that cached frames are displayed on time.  If multiple frames
 * were cached between renders, then releases the unnecessary frames and uses
2553 2554
 * the frame with the closest timing to ensure sync.  Also ensures that timing
 * with audio is synchronized.
2555
 */
2556
struct obs_source_frame *obs_source_get_frame(obs_source_t *source)
J
jp9000 已提交
2557
{
2558
	struct obs_source_frame *frame = NULL;
2559

2560
	if (!obs_source_valid(source, "obs_source_get_frame"))
J
jp9000 已提交
2561 2562
		return NULL;

2563
	pthread_mutex_lock(&source->async_mutex);
2564

2565 2566
	frame = source->cur_async_frame;
	source->cur_async_frame = NULL;
J
jp9000 已提交
2567 2568

	if (frame) {
2569
		os_atomic_inc_long(&frame->refs);
2570 2571
	}

2572
	pthread_mutex_unlock(&source->async_mutex);
2573

2574
	return frame;
J
jp9000 已提交
2575 2576
}

2577
void obs_source_release_frame(obs_source_t *source,
2578
		struct obs_source_frame *frame)
J
jp9000 已提交
2579
{
2580
	if (!frame)
2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593
		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);
2594
	}
J
jp9000 已提交
2595
}
2596

2597
const char *obs_source_get_name(const obs_source_t *source)
2598
{
2599 2600
	return obs_source_valid(source, "obs_source_get_name") ?
		source->context.name : NULL;
2601 2602
}

2603
void obs_source_set_name(obs_source_t *source, const char *name)
2604
{
2605 2606
	if (!obs_source_valid(source, "obs_source_set_name"))
		return;
J
jp9000 已提交
2607

2608 2609
	if (!name || !*name || !source->context.name ||
			strcmp(name, source->context.name) != 0) {
J
jp9000 已提交
2610 2611 2612 2613 2614
		struct calldata data;
		char *prev_name = bstrdup(source->context.name);
		obs_context_data_setname(&source->context, name);

		calldata_init(&data);
2615 2616 2617
		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 已提交
2618 2619 2620
		if (!source->context.private)
			signal_handler_signal(obs->signals, "source_rename",
					&data);
J
jp9000 已提交
2621 2622 2623 2624
		signal_handler_signal(source->context.signals, "rename", &data);
		calldata_free(&data);
		bfree(prev_name);
	}
2625 2626
}

2627
enum obs_source_type obs_source_get_type(const obs_source_t *source)
2628
{
2629 2630
	return obs_source_valid(source, "obs_source_get_type") ?
		source->info.type : OBS_SOURCE_TYPE_INPUT;
J
jp9000 已提交
2631
}
J
jp9000 已提交
2632

2633
const char *obs_source_get_id(const obs_source_t *source)
J
jp9000 已提交
2634
{
2635 2636
	return obs_source_valid(source, "obs_source_get_id") ?
		source->info.id : NULL;
2637
}
2638

2639
static inline void render_filter_bypass(obs_source_t *target,
2640
		gs_effect_t *effect, const char *tech_name)
2641
{
2642
	gs_technique_t *tech    = gs_effect_get_technique(effect, tech_name);
2643 2644
	size_t      passes, i;

2645
	passes = gs_technique_begin(tech);
2646
	for (i = 0; i < passes; i++) {
2647
		gs_technique_begin_pass(tech, i);
2648
		obs_source_video_render(target);
2649
		gs_technique_end_pass(tech);
2650
	}
2651
	gs_technique_end(tech);
2652 2653
}

2654
static inline void render_filter_tex(gs_texture_t *tex, gs_effect_t *effect,
2655
		uint32_t width, uint32_t height, const char *tech_name)
2656
{
2657
	gs_technique_t *tech    = gs_effect_get_technique(effect, tech_name);
2658
	gs_eparam_t    *image   = gs_effect_get_param_by_name(effect, "image");
2659 2660
	size_t      passes, i;

2661
	gs_effect_set_texture(image, tex);
2662

2663
	passes = gs_technique_begin(tech);
2664
	for (i = 0; i < passes; i++) {
2665
		gs_technique_begin_pass(tech, i);
J
jp9000 已提交
2666
		gs_draw_sprite(tex, 0, width, height);
2667
		gs_technique_end_pass(tech);
2668
	}
2669
	gs_technique_end(tech);
2670 2671
}

2672 2673 2674 2675 2676 2677 2678 2679 2680 2681
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);
}

2682
bool obs_source_process_filter_begin(obs_source_t *filter,
2683
		enum gs_color_format format,
2684
		enum obs_allow_direct_render allow_direct)
2685
{
2686
	obs_source_t *target, *parent;
J
jp9000 已提交
2687
	uint32_t     parent_flags;
J
jp9000 已提交
2688 2689
	int          cx, cy;

2690
	if (!obs_ptr_valid(filter, "obs_source_process_filter_begin"))
2691
		return false;
J
jp9000 已提交
2692

2693 2694
	target       = obs_filter_get_target(filter);
	parent       = obs_filter_get_parent(filter);
J
jp9000 已提交
2695 2696 2697 2698

	if (!target) {
		blog(LOG_INFO, "filter '%s' being processed with no target!",
				filter->context.name);
2699
		return false;
J
jp9000 已提交
2700 2701 2702 2703
	}
	if (!parent) {
		blog(LOG_INFO, "filter '%s' being processed with no parent!",
				filter->context.name);
2704
		return false;
J
jp9000 已提交
2705 2706
	}

J
jp9000 已提交
2707
	parent_flags = parent->info.output_flags;
J
jp9000 已提交
2708 2709
	cx           = get_base_width(target);
	cy           = get_base_height(target);
2710

2711 2712
	filter->allow_direct = allow_direct;

2713 2714 2715 2716
	/* 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 */
2717
	if (can_bypass(target, parent, parent_flags, allow_direct)) {
2718
		return true;
2719 2720
	}

2721 2722
	if (!cx || !cy) {
		obs_source_skip_video_filter(filter);
2723
		return false;
2724 2725
	}

J
jp9000 已提交
2726
	if (!filter->filter_texrender)
2727
		filter->filter_texrender = gs_texrender_create(format,
J
jp9000 已提交
2728 2729
				GS_ZS_NONE);

2730 2731 2732
	gs_blend_state_push();
	gs_blend_function(GS_BLEND_ONE, GS_BLEND_ZERO);

2733
	if (gs_texrender_begin(filter->filter_texrender, cx, cy)) {
2734 2735
		bool custom_draw = (parent_flags & OBS_SOURCE_CUSTOM_DRAW) != 0;
		bool async = (parent_flags & OBS_SOURCE_ASYNC) != 0;
2736 2737 2738 2739
		struct vec4 clear_color;

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

2742
		if (target == parent && !custom_draw && !async)
2743
			obs_source_default_render(target);
2744 2745
		else
			obs_source_video_render(target);
2746

2747
		gs_texrender_end(filter->filter_texrender);
2748
	}
2749 2750

	gs_blend_state_pop();
2751
	return true;
2752 2753
}

2754 2755 2756 2757 2758
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 已提交
2759
	uint32_t     parent_flags;
2760 2761 2762 2763 2764

	if (!filter) return;

	target       = obs_filter_get_target(filter);
	parent       = obs_filter_get_parent(filter);
J
jp9000 已提交
2765 2766 2767 2768

	if (!target || !parent)
		return;

2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779 2780 2781
	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);
	}
}


2782 2783 2784 2785 2786
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 已提交
2787
	uint32_t     parent_flags;
2788

2789 2790
	if (!obs_ptr_valid(filter, "obs_source_process_filter_end"))
		return;
2791

2792 2793 2794
	target       = obs_filter_get_target(filter);
	parent       = obs_filter_get_parent(filter);
	parent_flags = parent->info.output_flags;
2795

2796
	if (can_bypass(target, parent, parent_flags, filter->allow_direct)) {
2797
		render_filter_bypass(target, effect, "Draw");
2798 2799
	} else {
		texture = gs_texrender_get_texture(filter->filter_texrender);
2800
		if (texture)
2801 2802
			render_filter_tex(texture, effect, width, height,
					"Draw");
2803
	}
2804
}
2805

2806 2807 2808 2809 2810 2811
void obs_source_skip_video_filter(obs_source_t *filter)
{
	obs_source_t *target, *parent;
	bool custom_draw, async;
	uint32_t parent_flags;

2812 2813
	if (!obs_ptr_valid(filter, "obs_source_skip_video_filter"))
		return;
2814 2815 2816 2817 2818 2819 2820

	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 已提交
2821 2822
	if (target == parent) {
		if (!custom_draw && !async)
2823
			obs_source_default_render(target);
J
jp9000 已提交
2824 2825
		else if (target->info.video_render)
			obs_source_main_render(target);
J
jp9000 已提交
2826 2827
		else if (deinterlacing_enabled(target))
			deinterlace_render(target);
J
jp9000 已提交
2828 2829 2830 2831 2832 2833
		else
			obs_source_render_async_video(target);

	} else {
		obs_source_video_render(target);
	}
2834 2835
}

2836
signal_handler_t *obs_source_get_signal_handler(const obs_source_t *source)
2837
{
2838 2839
	return obs_source_valid(source, "obs_source_get_signal_handler") ?
		source->context.signals : NULL;
2840 2841
}

2842
proc_handler_t *obs_source_get_proc_handler(const obs_source_t *source)
2843
{
2844 2845
	return obs_source_valid(source, "obs_source_get_proc_handler") ?
		source->context.procs : NULL;
2846
}
J
jp9000 已提交
2847

2848
void obs_source_set_volume(obs_source_t *source, float volume)
J
jp9000 已提交
2849
{
2850
	if (obs_source_valid(source, "obs_source_set_volume")) {
J
jp9000 已提交
2851 2852 2853 2854 2855 2856
		struct audio_action action = {
			.timestamp = os_gettime_ns(),
			.type      = AUDIO_ACTION_VOL,
			.vol       = volume
		};

2857 2858 2859 2860
		struct calldata data;
		uint8_t stack[128];

		calldata_init_fixed(&data, stack, sizeof(stack));
2861 2862
		calldata_set_ptr(&data, "source", source);
		calldata_set_float(&data, "volume", volume);
J
jp9000 已提交
2863

2864
		signal_handler_signal(source->context.signals, "volume", &data);
J
jp9000 已提交
2865 2866 2867
		if (!source->context.private)
			signal_handler_signal(obs->signals, "source_volume",
					&data);
J
jp9000 已提交
2868

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

J
jp9000 已提交
2871 2872 2873 2874
		pthread_mutex_lock(&source->audio_actions_mutex);
		da_push_back(source->audio_actions, &action);
		pthread_mutex_unlock(&source->audio_actions_mutex);

J
jp9000 已提交
2875
		source->user_volume = volume;
J
jp9000 已提交
2876
	}
J
jp9000 已提交
2877 2878
}

2879
float obs_source_get_volume(const obs_source_t *source)
J
jp9000 已提交
2880
{
2881 2882
	return obs_source_valid(source, "obs_source_get_volume") ?
		source->user_volume : 0.0f;
J
jp9000 已提交
2883 2884
}

2885
void obs_source_set_sync_offset(obs_source_t *source, int64_t offset)
J
jp9000 已提交
2886
{
2887
	if (obs_source_valid(source, "obs_source_set_sync_offset")) {
2888 2889
		struct calldata data;
		uint8_t stack[128];
J
jp9000 已提交
2890

2891
		calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
2892 2893 2894 2895 2896 2897 2898 2899
		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 已提交
2900 2901
}

2902
int64_t obs_source_get_sync_offset(const obs_source_t *source)
J
jp9000 已提交
2903
{
2904 2905
	return obs_source_valid(source, "obs_source_get_sync_offset") ?
		source->sync_offset : 0;
J
jp9000 已提交
2906
}
2907 2908 2909 2910 2911 2912

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

2913
static void enum_source_tree_callback(obs_source_t *parent, obs_source_t *child,
2914 2915 2916
		void *param)
{
	struct source_enum_data *data = param;
2917
	bool is_transition = child->info.type == OBS_SOURCE_TYPE_TRANSITION;
2918

2919 2920 2921
	if (is_transition)
		obs_transition_enum_sources(child,
				enum_source_tree_callback, param);
2922
	if (child->info.enum_active_sources) {
J
jp9000 已提交
2923
		if (child->context.data) {
2924
			child->info.enum_active_sources(child->context.data,
2925
					enum_source_tree_callback, data);
J
jp9000 已提交
2926
		}
2927 2928 2929 2930 2931
	}

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

2932
void obs_source_enum_active_sources(obs_source_t *source,
2933 2934 2935
		obs_source_enum_proc_t enum_callback,
		void *param)
{
J
jp9000 已提交
2936
	bool is_transition;
2937
	if (!data_valid(source, "obs_source_enum_active_sources"))
J
jp9000 已提交
2938
		return;
J
jp9000 已提交
2939 2940 2941

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

	obs_source_addref(source);

J
jp9000 已提交
2946 2947 2948 2949 2950
	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);
2951 2952 2953 2954

	obs_source_release(source);
}

2955
void obs_source_enum_active_tree(obs_source_t *source,
2956 2957 2958 2959
		obs_source_enum_proc_t enum_callback,
		void *param)
{
	struct source_enum_data data = {enum_callback, param};
J
jp9000 已提交
2960
	bool is_transition;
2961

2962
	if (!data_valid(source, "obs_source_enum_active_tree"))
J
jp9000 已提交
2963
		return;
J
jp9000 已提交
2964 2965 2966

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

	obs_source_addref(source);

J
jp9000 已提交
2971 2972 2973 2974 2975 2976
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
		obs_transition_enum_sources(source, enum_source_tree_callback,
				&data);
	if (source->info.enum_active_sources)
		source->info.enum_active_sources(source->context.data,
				enum_source_tree_callback, &data);
2977 2978 2979

	obs_source_release(source);
}
2980

J
jp9000 已提交
2981 2982 2983 2984 2985 2986 2987
struct descendant_info {
	bool exists;
	obs_source_t *target;
};

static void check_descendant(obs_source_t *parent, obs_source_t *child,
		void *param)
2988
{
J
jp9000 已提交
2989 2990 2991 2992 2993
	struct descendant_info *info = param;
	if (child == info->target || parent == info->target)
		info->exists = true;
}

2994
bool obs_source_add_active_child(obs_source_t *parent, obs_source_t *child)
J
jp9000 已提交
2995
{
2996
	struct descendant_info info = {false, parent};
2997

2998
	if (!obs_ptr_valid(parent, "obs_source_add_active_child"))
2999
		return false;
3000
	if (!obs_ptr_valid(child, "obs_source_add_active_child"))
3001 3002
		return false;
	if (parent == child) {
3003 3004
		blog(LOG_WARNING, "obs_source_add_active_child: "
				"parent == child");
3005 3006
		return false;
	}
J
jp9000 已提交
3007

3008
	obs_source_enum_active_tree(child, check_descendant, &info);
J
jp9000 已提交
3009 3010
	if (info.exists)
		return false;
3011

3012 3013 3014 3015 3016
	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 已提交
3017 3018

	return true;
3019 3020
}

3021
void obs_source_remove_active_child(obs_source_t *parent, obs_source_t *child)
3022
{
3023
	if (!obs_ptr_valid(parent, "obs_source_remove_active_child"))
3024
		return;
3025
	if (!obs_ptr_valid(child, "obs_source_remove_active_child"))
3026
		return;
3027

3028 3029 3030 3031 3032
	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);
	}
3033
}
J
jp9000 已提交
3034

3035
void obs_source_save(obs_source_t *source)
3036
{
J
jp9000 已提交
3037 3038 3039
	if (!data_valid(source, "obs_source_save"))
		return;

J
jp9000 已提交
3040 3041 3042 3043 3044
	obs_source_dosignal(source, "source_save", "save");

	if (source->info.save)
		source->info.save(source->context.data,
				source->context.settings);
3045 3046
}

3047
void obs_source_load(obs_source_t *source)
3048
{
J
jp9000 已提交
3049 3050
	if (!data_valid(source, "obs_source_load"))
		return;
J
jp9000 已提交
3051 3052 3053
	if (source->info.load)
		source->info.load(source->context.data,
				source->context.settings);
J
jp9000 已提交
3054

J
jp9000 已提交
3055
	obs_source_dosignal(source, "source_load", "load");
3056
}
J
jp9000 已提交
3057

J
jp9000 已提交
3058 3059
bool obs_source_active(const obs_source_t *source)
{
3060 3061
	return obs_source_valid(source, "obs_source_active") ?
		source->activate_refs != 0 : false;
J
jp9000 已提交
3062 3063
}

J
jp9000 已提交
3064 3065
bool obs_source_showing(const obs_source_t *source)
{
3066 3067
	return obs_source_valid(source, "obs_source_showing") ?
		source->show_refs != 0 : false;
J
jp9000 已提交
3068 3069
}

J
jp9000 已提交
3070 3071
static inline void signal_flags_updated(obs_source_t *source)
{
3072 3073
	struct calldata data;
	uint8_t stack[128];
J
jp9000 已提交
3074

3075
	calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
3076 3077 3078 3079 3080 3081 3082 3083
	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)
{
3084 3085
	if (!obs_source_valid(source, "obs_source_set_flags"))
		return;
J
jp9000 已提交
3086 3087 3088 3089 3090 3091 3092

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

3093 3094
void obs_source_set_default_flags(obs_source_t *source, uint32_t flags)
{
3095 3096
	if (!obs_source_valid(source, "obs_source_set_default_flags"))
		return;
3097 3098 3099 3100

	source->default_flags = flags;
}

J
jp9000 已提交
3101 3102
uint32_t obs_source_get_flags(const obs_source_t *source)
{
3103 3104
	return obs_source_valid(source, "obs_source_get_flags") ?
		source->flags : 0;
J
jp9000 已提交
3105
}
J
jp9000 已提交
3106

3107 3108
void obs_source_set_audio_mixers(obs_source_t *source, uint32_t mixers)
{
3109 3110
	struct calldata data;
	uint8_t stack[128];
3111

3112 3113 3114 3115
	if (!obs_source_valid(source, "obs_source_set_audio_mixers"))
		return;
	if ((source->info.output_flags & OBS_SOURCE_AUDIO) == 0)
		return;
3116

3117
	if (source->audio_mixers == mixers)
3118 3119
		return;

3120
	calldata_init_fixed(&data, stack, sizeof(stack));
3121 3122 3123 3124 3125 3126 3127
	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");

3128
	source->audio_mixers = mixers;
3129 3130 3131 3132
}

uint32_t obs_source_get_audio_mixers(const obs_source_t *source)
{
3133 3134 3135 3136
	if (!obs_source_valid(source, "obs_source_get_audio_mixers"))
		return 0;
	if ((source->info.output_flags & OBS_SOURCE_AUDIO) == 0)
		return 0;
3137

3138
	return source->audio_mixers;
3139 3140
}

J
jp9000 已提交
3141 3142 3143 3144
void obs_source_draw_set_color_matrix(const struct matrix4 *color_matrix,
		const struct vec3 *color_range_min,
		const struct vec3 *color_range_max)
{
3145 3146 3147 3148 3149 3150
	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 已提交
3151 3152 3153 3154 3155 3156
	gs_effect_t *effect = gs_get_effect();
	gs_eparam_t *matrix;
	gs_eparam_t *range_min;
	gs_eparam_t *range_max;

	if (!effect) {
3157 3158
		blog(LOG_WARNING, "obs_source_draw_set_color_matrix: no "
				"active effect!");
J
jp9000 已提交
3159 3160 3161
		return;
	}

3162
	if (!obs_ptr_valid(color_matrix, "obs_source_draw_set_color_matrix"))
J
jp9000 已提交
3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173 3174 3175 3176 3177 3178 3179 3180 3181 3182 3183 3184 3185 3186
		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) {
3187
		blog(LOG_WARNING, "obs_source_draw: no active effect!");
J
jp9000 已提交
3188 3189 3190
		return;
	}

3191
	if (!obs_ptr_valid(texture, "obs_source_draw"))
J
jp9000 已提交
3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202 3203 3204 3205 3206
		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();
}
3207

3208 3209
void obs_source_inc_showing(obs_source_t *source)
{
3210 3211
	if (obs_source_valid(source, "obs_source_inc_showing"))
		obs_source_activate(source, AUX_VIEW);
3212 3213 3214 3215
}

void obs_source_dec_showing(obs_source_t *source)
{
3216 3217
	if (obs_source_valid(source, "obs_source_dec_showing"))
		obs_source_deactivate(source, AUX_VIEW);
3218
}
3219 3220 3221 3222

void obs_source_enum_filters(obs_source_t *source,
		obs_source_enum_proc_t callback, void *param)
{
3223 3224 3225
	if (!obs_source_valid(source, "obs_source_enum_filters"))
		return;
	if (!obs_ptr_valid(callback, "obs_source_enum_filters"))
3226 3227 3228 3229
		return;

	pthread_mutex_lock(&source->filter_mutex);

3230 3231
	for (size_t i = source->filters.num; i > 0; i--) {
		struct obs_source *filter = source->filters.array[i - 1];
3232 3233 3234 3235 3236
		callback(source, filter, param);
	}

	pthread_mutex_unlock(&source->filter_mutex);
}
3237 3238 3239 3240 3241 3242

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

3243 3244 3245
	if (!obs_source_valid(source, "obs_source_get_filter_by_name"))
		return NULL;
	if (!obs_ptr_valid(name, "obs_source_get_filter_by_name"))
3246 3247 3248 3249 3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262
		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;
}
3263 3264 3265

bool obs_source_enabled(const obs_source_t *source)
{
3266 3267
	return obs_source_valid(source, "obs_source_enabled") ?
		source->enabled : false;
3268 3269 3270 3271
}

void obs_source_set_enabled(obs_source_t *source, bool enabled)
{
3272 3273
	struct calldata data;
	uint8_t stack[128];
3274

3275
	if (!obs_source_valid(source, "obs_source_set_enabled"))
3276 3277 3278 3279
		return;

	source->enabled = enabled;

3280
	calldata_init_fixed(&data, stack, sizeof(stack));
3281 3282 3283 3284 3285
	calldata_set_ptr(&data, "source", source);
	calldata_set_bool(&data, "enabled", enabled);

	signal_handler_signal(source->context.signals, "enable", &data);
}
J
jp9000 已提交
3286 3287 3288

bool obs_source_muted(const obs_source_t *source)
{
3289
	return obs_source_valid(source, "obs_source_muted") ?
J
jp9000 已提交
3290
		source->user_muted : false;
J
jp9000 已提交
3291 3292 3293 3294
}

void obs_source_set_muted(obs_source_t *source, bool muted)
{
3295 3296
	struct calldata data;
	uint8_t stack[128];
J
jp9000 已提交
3297 3298 3299 3300 3301
	struct audio_action action = {
		.timestamp = os_gettime_ns(),
		.type      = AUDIO_ACTION_MUTE,
		.set       = muted
	};
J
jp9000 已提交
3302

3303
	if (!obs_source_valid(source, "obs_source_set_muted"))
J
jp9000 已提交
3304 3305
		return;

J
jp9000 已提交
3306
	source->user_muted = muted;
J
jp9000 已提交
3307

3308
	calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
3309 3310 3311 3312 3313
	calldata_set_ptr(&data, "source", source);
	calldata_set_bool(&data, "muted", muted);

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

J
jp9000 已提交
3314 3315 3316
	pthread_mutex_lock(&source->audio_actions_mutex);
	da_push_back(source->audio_actions, &action);
	pthread_mutex_unlock(&source->audio_actions_mutex);
J
jp9000 已提交
3317
}
P
Palana 已提交
3318 3319 3320 3321

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

3325
	calldata_init_fixed(&data, stack, sizeof(stack));
P
Palana 已提交
3326 3327 3328 3329 3330 3331 3332 3333 3334
	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)
{
3335 3336
	struct calldata data;
	uint8_t stack[128];
P
Palana 已提交
3337

3338
	calldata_init_fixed(&data, stack, sizeof(stack));
P
Palana 已提交
3339 3340 3341 3342 3343 3344 3345 3346 3347
	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;
3348 3349
	if (!obs_source_valid(source, "obs_source_push_to_mute_enabled"))
		return false;
P
Palana 已提交
3350 3351 3352 3353 3354 3355 3356 3357 3358 3359

	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)
{
3360 3361
	if (!obs_source_valid(source, "obs_source_enable_push_to_mute"))
		return;
P
Palana 已提交
3362 3363 3364 3365 3366 3367 3368 3369 3370 3371 3372 3373 3374 3375 3376 3377 3378 3379 3380

	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;
3381 3382
	if (!obs_source_valid(source, "obs_source_get_push_to_mute_delay"))
		return 0;
P
Palana 已提交
3383 3384 3385 3386 3387 3388 3389 3390 3391 3392

	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)
{
3393 3394
	if (!obs_source_valid(source, "obs_source_set_push_to_mute_delay"))
		return;
P
Palana 已提交
3395 3396 3397 3398 3399 3400 3401 3402 3403 3404 3405

	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;
3406 3407
	if (!obs_source_valid(source, "obs_source_push_to_talk_enabled"))
		return false;
P
Palana 已提交
3408 3409 3410 3411 3412 3413 3414 3415 3416 3417

	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)
{
3418 3419
	if (!obs_source_valid(source, "obs_source_enable_push_to_talk"))
		return;
P
Palana 已提交
3420 3421 3422 3423 3424 3425 3426 3427 3428 3429 3430 3431 3432 3433 3434 3435 3436 3437 3438

	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;
3439 3440
	if (!obs_source_valid(source, "obs_source_get_push_to_talk_delay"))
		return 0;
P
Palana 已提交
3441 3442 3443 3444 3445 3446 3447 3448 3449 3450

	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)
{
3451 3452
	if (!obs_source_valid(source, "obs_source_set_push_to_talk_delay"))
		return;
P
Palana 已提交
3453 3454 3455 3456 3457 3458 3459

	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);
}
3460 3461 3462 3463 3464 3465

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

J
jp9000 已提交
3467 3468 3469 3470 3471 3472 3473 3474 3475 3476 3477 3478 3479 3480 3481 3482 3483 3484 3485 3486 3487 3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502 3503 3504 3505 3506 3507 3508 3509 3510 3511 3512 3513 3514 3515 3516 3517 3518 3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530 3531 3532 3533 3534 3535 3536 3537 3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551 3552 3553 3554 3555 3556 3557 3558 3559 3560 3561 3562 3563 3564 3565 3566 3567 3568 3569 3570 3571 3572 3573 3574 3575 3576 3577 3578 3579 3580 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
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);

3666 3667 3668 3669 3670 3671
	if (source->audio_input_buf[0].size < size) {
		source->audio_pending = true;
		pthread_mutex_unlock(&source->audio_buf_mutex);
		return;
	}

J
jp9000 已提交
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
	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)
{
3705
	if (!source->audio_output_buf[0][0]) {
J
jp9000 已提交
3706 3707 3708 3709 3710 3711 3712 3713 3714
		source->audio_pending = true;
		return;
	}

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

3715
	if (!source->audio_ts) {
J
jp9000 已提交
3716 3717 3718 3719 3720 3721 3722 3723 3724
		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)
{
3725 3726 3727 3728 3729
	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 已提交
3730 3731
}

3732 3733 3734 3735 3736 3737 3738 3739 3740 3741 3742 3743 3744 3745 3746 3747 3748 3749 3750 3751 3752
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];
		}
	}
}
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

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