obs-source.c 116.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
#include <inttypes.h>
C
cg2121 已提交
19
#include <math.h>
20

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

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

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

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

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

	return NULL;
}

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

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

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

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

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

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

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

J
jp9000 已提交
118 119 120
static inline bool is_async_video_source(const struct obs_source *source)
{
	return (source->info.output_flags & OBS_SOURCE_ASYNC_VIDEO) ==
J
jp9000 已提交
121
	       OBS_SOURCE_ASYNC_VIDEO;
J
jp9000 已提交
122 123
}

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

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

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

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

168
	if (is_audio_source(source) || is_composite_source(source))
169
		allocate_audio_output_buffer(source);
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;
179
	source->audio_mixers = 0xFF;
180

181
	if (is_audio_source(source)) {
182 183 184
		pthread_mutex_lock(&obs->data.audio_sources_mutex);

		source->next_audio_source = obs->data.first_audio_source;
J
jp9000 已提交
185
		source->prev_next_audio_source = &obs->data.first_audio_source;
186 187 188 189 190 191
		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);
192
	}
193

194 195
	source->private_settings = obs_data_create();

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

J
jp9000 已提交
201 202
static bool obs_source_hotkey_mute(void *data, obs_hotkey_pair_id id,
				   obs_hotkey_t *key, bool pressed)
P
Palana 已提交
203 204 205 206 207 208
{
	UNUSED_PARAMETER(id);
	UNUSED_PARAMETER(key);

	struct obs_source *source = data;

J
jp9000 已提交
209 210
	if (!pressed || obs_source_muted(source))
		return false;
P
Palana 已提交
211 212 213 214 215

	obs_source_set_muted(source, true);
	return true;
}

J
jp9000 已提交
216 217
static bool obs_source_hotkey_unmute(void *data, obs_hotkey_pair_id id,
				     obs_hotkey_t *key, bool pressed)
P
Palana 已提交
218 219 220 221 222 223
{
	UNUSED_PARAMETER(id);
	UNUSED_PARAMETER(key);

	struct obs_source *source = data;

J
jp9000 已提交
224 225
	if (!pressed || !obs_source_muted(source))
		return false;
P
Palana 已提交
226 227 228 229 230

	obs_source_set_muted(source, false);
	return true;
}

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

P
Palana 已提交
238 239 240 241 242
	UNUSED_PARAMETER(id);
	UNUSED_PARAMETER(key);

	struct obs_source *source = data;

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

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

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) {
J
jp9000 已提交
273
		source->mute_unmute_key = OBS_INVALID_HOTKEY_ID;
P
Palana 已提交
274 275 276 277
		source->push_to_talk_key = OBS_INVALID_HOTKEY_ID;
		return;
	}

J
jp9000 已提交
278 279 280 281
	source->mute_unmute_key = obs_hotkey_pair_register_source(
		source, "libobs.mute", obs->hotkeys.mute, "libobs.unmute",
		obs->hotkeys.unmute, obs_source_hotkey_mute,
		obs_source_hotkey_unmute, source, source);
P
Palana 已提交
282

J
jp9000 已提交
283 284 285
	source->push_to_mute_key = obs_hotkey_register_source(
		source, "libobs.push-to-mute", obs->hotkeys.push_to_mute,
		obs_source_hotkey_push_to_mute, source);
P
Palana 已提交
286

J
jp9000 已提交
287 288 289
	source->push_to_talk_key = obs_hotkey_register_source(
		source, "libobs.push-to-talk", obs->hotkeys.push_to_talk,
		obs_source_hotkey_push_to_talk, source);
P
Palana 已提交
290 291
}

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

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

J
jp9000 已提交
304
		source->info.id = bstrdup(id);
305 306 307
		source->owns_info_id = true;
	} else {
		source->info = *info;
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)
J
jp9000 已提交
314 315
		private
		= true;
316
	}
317

J
jp9000 已提交
318
	source->mute_unmute_key = OBS_INVALID_HOTKEY_PAIR_ID;
P
Palana 已提交
319 320 321
	source->push_to_mute_key = OBS_INVALID_HOTKEY_ID;
	source->push_to_talk_key = OBS_INVALID_HOTKEY_ID;

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

326 327 328
	if (info) {
		if (info->get_defaults2)
			info->get_defaults2(info->type_data,
J
jp9000 已提交
329
					    source->context.settings);
330 331 332
		else if (info->get_defaults)
			info->get_defaults(source->context.settings);
	}
J
jp9000 已提交
333

334
	if (!obs_source_init(source))
335 336
		goto fail;

J
jp9000 已提交
337 338
	if (!private)
		obs_source_init_audio_hotkeys(source);
P
Palana 已提交
339

340 341
	/* allow the source to be created even if creation fails so that the
	 * user's data doesn't become lost */
342
	if (info)
J
jp9000 已提交
343 344
		source->context.data =
			info->create(source->context.settings, source);
345
	if (!source->context.data)
346
		blog(LOG_ERROR, "Failed to create source '%s'!", name);
347

J
jp9000 已提交
348 349
	blog(LOG_DEBUG, "%ssource '%s' (%s) created", private ? "private " : "",
	     name, id);
J
jp9000 已提交
350

351
	source->flags = source->default_flags;
352
	source->enabled = true;
353 354 355 356 357

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

J
jp9000 已提交
358
	return source;
359 360 361 362 363

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

J
jp9000 已提交
366
obs_source_t *obs_source_create(const char *id, const char *name,
J
jp9000 已提交
367
				obs_data_t *settings, obs_data_t *hotkey_data)
J
jp9000 已提交
368 369
{
	return obs_source_create_internal(id, name, settings, hotkey_data,
J
jp9000 已提交
370
					  false);
J
jp9000 已提交
371 372 373
}

obs_source_t *obs_source_create_private(const char *id, const char *name,
J
jp9000 已提交
374
					obs_data_t *settings)
J
jp9000 已提交
375 376 377 378
{
	return obs_source_create_internal(id, name, settings, NULL, true);
}

379 380 381 382 383 384 385 386
static char *get_new_filter_name(obs_source_t *dst, const char *name)
{
	struct dstr new_name = {0};
	int inc = 0;

	dstr_copy(&new_name, name);

	for (;;) {
J
jp9000 已提交
387 388
		obs_source_t *existing_filter =
			obs_source_get_filter_by_name(dst, new_name.array);
389 390 391 392 393 394 395 396 397 398 399
		if (!existing_filter)
			break;

		obs_source_release(existing_filter);

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

	return new_name.array;
}

400
static void duplicate_filters(obs_source_t *dst, obs_source_t *src,
J
jp9000 已提交
401
			      bool private)
402
{
J
jp9000 已提交
403
	DARRAY(obs_source_t *) filters;
404 405 406 407 408 409 410 411 412 413 414

	da_init(filters);

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

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

J
jp9000 已提交
419 420
		obs_source_t *dst_filter =
			obs_source_duplicate(src_filter, new_name, private);
421
		obs_source_set_enabled(dst_filter, enabled);
422

423
		bfree(new_name);
424 425 426 427 428 429 430 431
		obs_source_filter_add(dst, dst_filter);
		obs_source_release(dst_filter);
		obs_source_release(src_filter);
	}

	da_free(filters);
}

432 433
void obs_source_copy_filters(obs_source_t *dst, obs_source_t *src)
{
434 435 436 437 438
	if (!obs_source_valid(dst, "obs_source_copy_filters"))
		return;
	if (!obs_source_valid(src, "obs_source_copy_filters"))
		return;

439
	duplicate_filters(dst, src, dst->context.private);
440 441
}

J
jp9000 已提交
442 443
extern obs_scene_t *obs_group_from_source(const obs_source_t *source);

J
jp9000 已提交
444 445
obs_source_t *obs_source_duplicate(obs_source_t *source, const char *new_name,
				   bool create_private)
446 447 448 449 450 451 452 453 454 455 456 457
{
	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;
	}

458 459
	if (source->info.type == OBS_SOURCE_TYPE_SCENE) {
		obs_scene_t *scene = obs_scene_from_source(source);
J
jp9000 已提交
460 461 462 463 464
		if (!scene)
			scene = obs_group_from_source(source);
		if (!scene)
			return NULL;

J
jp9000 已提交
465 466 467 468
		obs_scene_t *new_scene = obs_scene_duplicate(
			scene, new_name,
			create_private ? OBS_SCENE_DUP_PRIVATE_COPY
				       : OBS_SCENE_DUP_COPY);
469 470
		obs_source_t *new_source = obs_scene_get_source(new_scene);
		return new_source;
471 472
	}

473 474 475
	settings = obs_data_create();
	obs_data_apply(settings, source->context.settings);

J
jp9000 已提交
476 477 478 479 480
	new_source = create_private
			     ? obs_source_create_private(source->info.id,
							 new_name, settings)
			     : obs_source_create(source->info.id, new_name,
						 settings, NULL);
481 482 483 484 485 486 487 488 489

	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;

490 491
	obs_data_apply(new_source->private_settings, source->private_settings);

492 493 494 495 496 497 498
	if (source->info.type != OBS_SOURCE_TYPE_FILTER)
		duplicate_filters(new_source, source, create_private);

	obs_data_release(settings);
	return new_source;
}

499
void obs_source_frame_init(struct obs_source_frame *frame,
J
jp9000 已提交
500 501
			   enum video_format format, uint32_t width,
			   uint32_t height)
502
{
503
	struct video_frame vid_frame;
J
jp9000 已提交
504

505
	if (!obs_ptr_valid(frame, "obs_source_frame_init"))
J
jp9000 已提交
506 507
		return;

508
	video_frame_init(&vid_frame, format, width, height);
509
	frame->format = format;
J
jp9000 已提交
510
	frame->width = width;
511
	frame->height = height;
512

513
	for (size_t i = 0; i < MAX_AV_PLANES; i++) {
J
jp9000 已提交
514
		frame->data[i] = vid_frame.data[i];
515
		frame->linesize[i] = vid_frame.linesize[i];
516 517 518
	}
}

519 520 521 522 523 524
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);
}

525
static bool obs_source_filter_remove_refless(obs_source_t *source,
J
jp9000 已提交
526
					     obs_source_t *filter);
527

528
void obs_source_destroy(struct obs_source *source)
J
jp9000 已提交
529
{
530
	size_t i;
531

532
	if (!obs_source_valid(source, "obs_source_destroy"))
J
jp9000 已提交
533 534
		return;

J
jp9000 已提交
535 536 537
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
		obs_transition_clear(source);

538 539 540 541 542 543 544 545 546
	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);

547 548 549 550 551 552
	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]);

553 554
	obs_context_data_remove(&source->context);

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

558
	obs_source_dosignal(source, "source_destroy", "destroy");
559

560
	if (source->context.data) {
561
		source->info.destroy(source->context.data);
562 563
		source->context.data = NULL;
	}
564

J
jp9000 已提交
565 566
	audio_monitor_destroy(source->monitor);

P
Palana 已提交
567 568 569 570
	obs_hotkey_unregister(source->push_to_talk_key);
	obs_hotkey_unregister(source->push_to_mute_key);
	obs_hotkey_pair_unregister(source->mute_unmute_key);

571
	for (i = 0; i < source->async_cache.num; i++)
572
		obs_source_frame_decref(source->async_cache.array[i].frame);
573

574
	gs_enter_context(obs->video.graphics);
575 576
	if (source->async_texrender)
		gs_texrender_destroy(source->async_texrender);
J
jp9000 已提交
577 578
	if (source->async_prev_texrender)
		gs_texrender_destroy(source->async_prev_texrender);
579 580
	if (source->async_texture)
		gs_texture_destroy(source->async_texture);
J
jp9000 已提交
581 582
	if (source->async_prev_texture)
		gs_texture_destroy(source->async_prev_texture);
583 584
	if (source->filter_texrender)
		gs_texrender_destroy(source->filter_texrender);
585
	gs_leave_context();
J
jp9000 已提交
586

J
jp9000 已提交
587
	for (i = 0; i < MAX_AV_PLANES; i++)
588
		bfree(source->audio_data.data[i]);
589 590
	for (i = 0; i < MAX_AUDIO_CHANNELS; i++)
		circlebuf_free(&source->audio_input_buf[i]);
591
	audio_resampler_destroy(source->resampler);
592
	bfree(source->audio_output_buf[0][0]);
593

594 595
	obs_source_frame_destroy(source->async_preload_frame);

J
jp9000 已提交
596 597 598
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
		obs_transition_free(source);

J
jp9000 已提交
599
	da_free(source->audio_actions);
600
	da_free(source->audio_cb_list);
601
	da_free(source->async_cache);
602
	da_free(source->async_frames);
603 604
	da_free(source->filters);
	pthread_mutex_destroy(&source->filter_mutex);
J
jp9000 已提交
605
	pthread_mutex_destroy(&source->audio_actions_mutex);
606
	pthread_mutex_destroy(&source->audio_buf_mutex);
607
	pthread_mutex_destroy(&source->audio_cb_mutex);
608
	pthread_mutex_destroy(&source->audio_mutex);
609
	pthread_mutex_destroy(&source->async_mutex);
610
	obs_data_release(source->private_settings);
611
	obs_context_data_free(&source->context);
612

613
	if (source->owns_info_id)
J
jp9000 已提交
614
		bfree((void *)source->info.id);
615

616 617 618
	bfree(source);
}

619
void obs_source_addref(obs_source_t *source)
620
{
621 622 623 624
	if (!source)
		return;

	obs_ref_addref(&source->control->ref);
625 626
}

627
void obs_source_release(obs_source_t *source)
628
{
629 630
	if (!obs) {
		blog(LOG_WARNING, "Tried to release a source when the OBS "
J
jp9000 已提交
631
				  "core is shut down!");
632 633 634
		return;
	}

P
Palana 已提交
635 636
	if (!source)
		return;
637

638 639
	obs_weak_source_t *control = source->control;
	if (obs_ref_release(&control->ref)) {
P
Palana 已提交
640
		obs_source_destroy(source);
641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691
		obs_weak_source_release(control);
	}
}

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

	obs_weak_ref_addref(&weak->ref);
}

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

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

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

	return obs_weak_source_get_source(source->control);
}

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

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

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

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

	return NULL;
}

bool obs_weak_source_references_source(obs_weak_source_t *weak,
J
jp9000 已提交
692
				       obs_source_t *source)
693 694
{
	return weak && source && weak->source == source;
695 696
}

697
void obs_source_remove(obs_source_t *source)
698
{
699 700 701
	if (!obs_source_valid(source, "obs_source_remove"))
		return;

702 703
	if (!source->removed) {
		source->removed = true;
704
		obs_source_dosignal(source, "source_remove", "remove");
705
	}
706 707
}

708
bool obs_source_removed(const obs_source_t *source)
709
{
J
jp9000 已提交
710 711
	return obs_source_valid(source, "obs_source_removed") ? source->removed
							      : true;
J
jp9000 已提交
712 713
}

714
static inline obs_data_t *get_defaults(const struct obs_source_info *info)
J
jp9000 已提交
715
{
716
	obs_data_t *settings = obs_data_create();
717 718 719
	if (info->get_defaults2)
		info->get_defaults2(info->type_data, settings);
	else if (info->get_defaults)
720
		info->get_defaults(settings);
J
jp9000 已提交
721 722 723
	return settings;
}

724
obs_data_t *obs_source_settings(const char *id)
J
jp9000 已提交
725
{
726
	const struct obs_source_info *info = get_source_info(id);
J
jp9000 已提交
727
	return (info) ? get_defaults(info) : NULL;
J
jp9000 已提交
728 729
}

730
obs_data_t *obs_get_source_defaults(const char *id)
J
jp9000 已提交
731
{
732
	const struct obs_source_info *info = get_source_info(id);
J
jp9000 已提交
733 734 735
	return info ? get_defaults(info) : NULL;
}

736
obs_properties_t *obs_get_source_properties(const char *id)
J
jp9000 已提交
737
{
738
	const struct obs_source_info *info = get_source_info(id);
739
	if (info && (info->get_properties || info->get_properties2)) {
J
jp9000 已提交
740
		obs_data_t *defaults = get_defaults(info);
741 742 743 744 745 746
		obs_properties_t *props;

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

748
		obs_properties_apply_settings(props, defaults);
J
jp9000 已提交
749
		obs_data_release(defaults);
750
		return props;
J
jp9000 已提交
751
	}
J
jp9000 已提交
752 753 754
	return NULL;
}

755 756 757
bool obs_is_source_configurable(const char *id)
{
	const struct obs_source_info *info = get_source_info(id);
758
	return info && (info->get_properties || info->get_properties2);
759 760 761 762 763
}

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

767
obs_properties_t *obs_source_properties(const obs_source_t *source)
768
{
J
jp9000 已提交
769 770 771
	if (!data_valid(source, "obs_source_properties"))
		return NULL;

772 773 774
	if (source->info.get_properties2) {
		obs_properties_t *props;
		props = source->info.get_properties2(source->context.data,
J
jp9000 已提交
775
						     source->info.type_data);
776 777 778 779
		obs_properties_apply_settings(props, source->context.settings);
		return props;

	} else if (source->info.get_properties) {
780
		obs_properties_t *props;
781
		props = source->info.get_properties(source->context.data);
782
		obs_properties_apply_settings(props, source->context.settings);
J
jp9000 已提交
783 784 785
		return props;
	}

786 787 788
	return NULL;
}

789
uint32_t obs_source_get_output_flags(const obs_source_t *source)
J
jp9000 已提交
790
{
J
jp9000 已提交
791 792 793
	return obs_source_valid(source, "obs_source_get_output_flags")
		       ? source->info.output_flags
		       : 0;
J
jp9000 已提交
794 795
}

796
uint32_t obs_get_source_output_flags(const char *id)
797
{
798
	const struct obs_source_info *info = get_source_info(id);
799 800 801
	return info ? info->output_flags : 0;
}

802
static void obs_source_deferred_update(obs_source_t *source)
803
{
804 805
	if (source->context.data && source->info.update)
		source->info.update(source->context.data,
J
jp9000 已提交
806
				    source->context.settings);
807

808 809 810
	source->defer_update = false;
}

811
void obs_source_update(obs_source_t *source, obs_data_t *settings)
J
jp9000 已提交
812
{
813 814
	if (!obs_source_valid(source, "obs_source_update"))
		return;
J
jp9000 已提交
815

816 817 818 819 820 821 822
	if (settings)
		obs_data_apply(source->context.settings, settings);

	if (source->info.output_flags & OBS_SOURCE_VIDEO) {
		source->defer_update = true;
	} else if (source->context.data && source->info.update) {
		source->info.update(source->context.data,
J
jp9000 已提交
823
				    source->context.settings);
824
	}
J
jp9000 已提交
825 826
}

827 828
void obs_source_update_properties(obs_source_t *source)
{
829 830
	if (!obs_source_valid(source, "obs_source_update_properties"))
		return;
831

832
	obs_source_dosignal(source, NULL, "update_properties");
833 834
}

835
void obs_source_send_mouse_click(obs_source_t *source,
J
jp9000 已提交
836 837 838
				 const struct obs_mouse_event *event,
				 int32_t type, bool mouse_up,
				 uint32_t click_count)
K
kc5nra 已提交
839
{
840
	if (!obs_source_valid(source, "obs_source_send_mouse_click"))
K
kc5nra 已提交
841 842 843 844
		return;

	if (source->info.output_flags & OBS_SOURCE_INTERACTION) {
		if (source->info.mouse_click) {
J
jp9000 已提交
845 846
			source->info.mouse_click(source->context.data, event,
						 type, mouse_up, click_count);
K
kc5nra 已提交
847 848 849 850
		}
	}
}

851
void obs_source_send_mouse_move(obs_source_t *source,
J
jp9000 已提交
852 853
				const struct obs_mouse_event *event,
				bool mouse_leave)
K
kc5nra 已提交
854
{
855
	if (!obs_source_valid(source, "obs_source_send_mouse_move"))
K
kc5nra 已提交
856 857 858 859
		return;

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

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

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

881
void obs_source_send_focus(obs_source_t *source, bool focus)
K
kc5nra 已提交
882
{
883
	if (!obs_source_valid(source, "obs_source_send_focus"))
K
kc5nra 已提交
884 885 886 887 888 889 890 891 892
		return;

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

893
void obs_source_send_key_click(obs_source_t *source,
J
jp9000 已提交
894
			       const struct obs_key_event *event, bool key_up)
K
kc5nra 已提交
895
{
896
	if (!obs_source_valid(source, "obs_source_send_key_click"))
K
kc5nra 已提交
897 898 899 900 901
		return;

	if (source->info.output_flags & OBS_SOURCE_INTERACTION) {
		if (source->info.key_click) {
			source->info.key_click(source->context.data, event,
J
jp9000 已提交
902
					       key_up);
K
kc5nra 已提交
903 904 905 906
		}
	}
}

907
static void activate_source(obs_source_t *source)
J
jp9000 已提交
908
{
909
	if (source->context.data && source->info.activate)
910
		source->info.activate(source->context.data);
911
	obs_source_dosignal(source, "source_activate", "activate");
J
jp9000 已提交
912 913
}

914
static void deactivate_source(obs_source_t *source)
J
jp9000 已提交
915
{
916
	if (source->context.data && source->info.deactivate)
917
		source->info.deactivate(source->context.data);
918
	obs_source_dosignal(source, "source_deactivate", "deactivate");
919
}
920

921
static void show_source(obs_source_t *source)
922
{
923
	if (source->context.data && source->info.show)
924
		source->info.show(source->context.data);
925
	obs_source_dosignal(source, "source_show", "show");
926 927
}

928
static void hide_source(obs_source_t *source)
929
{
930
	if (source->context.data && source->info.hide)
931
		source->info.hide(source->context.data);
932
	obs_source_dosignal(source, "source_hide", "hide");
933 934
}

935
static void activate_tree(obs_source_t *parent, obs_source_t *child,
J
jp9000 已提交
936
			  void *param)
937
{
938
	os_atomic_inc_long(&child->activate_refs);
J
jp9000 已提交
939 940 941

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
942 943
}

944
static void deactivate_tree(obs_source_t *parent, obs_source_t *child,
J
jp9000 已提交
945
			    void *param)
946
{
947
	os_atomic_dec_long(&child->activate_refs);
J
jp9000 已提交
948 949 950

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
951 952
}

953
static void show_tree(obs_source_t *parent, obs_source_t *child, void *param)
954
{
955
	os_atomic_inc_long(&child->show_refs);
956 957 958 959 960

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
}

961
static void hide_tree(obs_source_t *parent, obs_source_t *child, void *param)
962
{
963
	os_atomic_dec_long(&child->show_refs);
964 965 966 967 968

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
}

969
void obs_source_activate(obs_source_t *source, enum view_type type)
970
{
971 972
	if (!obs_source_valid(source, "obs_source_activate"))
		return;
973

974 975
	os_atomic_inc_long(&source->show_refs);
	obs_source_enum_active_tree(source, show_tree, NULL);
976 977

	if (type == MAIN_VIEW) {
978 979
		os_atomic_inc_long(&source->activate_refs);
		obs_source_enum_active_tree(source, activate_tree, NULL);
980 981 982
	}
}

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

988 989
	if (os_atomic_load_long(&source->show_refs) > 0) {
		os_atomic_dec_long(&source->show_refs);
990
		obs_source_enum_active_tree(source, hide_tree, NULL);
991 992 993
	}

	if (type == MAIN_VIEW) {
994 995
		if (os_atomic_load_long(&source->activate_refs) > 0) {
			os_atomic_dec_long(&source->activate_refs);
996
			obs_source_enum_active_tree(source, deactivate_tree,
J
jp9000 已提交
997
						    NULL);
998
		}
999
	}
J
jp9000 已提交
1000 1001
}

1002
static inline struct obs_source_frame *get_closest_frame(obs_source_t *source,
J
jp9000 已提交
1003
							 uint64_t sys_time);
1004
bool set_async_texture_size(struct obs_source *source,
J
jp9000 已提交
1005
			    const struct obs_source_frame *frame);
1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016

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

	pthread_mutex_lock(&source->async_mutex);

	if (deinterlacing_enabled(source)) {
		deinterlace_process_last_frame(source, sys_time);
	} else {
		if (source->cur_async_frame) {
J
jp9000 已提交
1017
			remove_async_frame(source, source->cur_async_frame);
1018 1019 1020
			source->cur_async_frame = NULL;
		}

J
jp9000 已提交
1021
		source->cur_async_frame = get_closest_frame(source, sys_time);
1022 1023 1024 1025 1026 1027
	}

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

	if (source->cur_async_frame)
J
jp9000 已提交
1028 1029
		source->async_update_texture =
			set_async_texture_size(source, source->cur_async_frame);
1030
}
1031

1032
void obs_source_video_tick(obs_source_t *source, float seconds)
J
jp9000 已提交
1033
{
1034 1035
	bool now_showing, now_active;

1036 1037
	if (!obs_source_valid(source, "obs_source_video_tick"))
		return;
J
jp9000 已提交
1038

J
jp9000 已提交
1039 1040 1041
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
		obs_transition_tick(source);

1042 1043
	if ((source->info.output_flags & OBS_SOURCE_ASYNC) != 0)
		async_tick(source);
1044

1045 1046 1047
	if (source->defer_update)
		obs_source_deferred_update(source);

J
jp9000 已提交
1048 1049
	/* reset the filter render texture information once every frame */
	if (source->filter_texrender)
1050
		gs_texrender_reset(source->filter_texrender);
J
jp9000 已提交
1051

1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075
	/* 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;
	}

1076
	if (source->context.data && source->info.video_tick)
1077
		source->info.video_tick(source->context.data, seconds);
1078 1079

	source->async_rendered = false;
J
jp9000 已提交
1080
	source->deinterlace_rendered = false;
J
jp9000 已提交
1081 1082
}

1083
/* unless the value is 3+ hours worth of frames, this won't overflow */
1084
static inline uint64_t conv_frames_to_time(const size_t sample_rate,
J
jp9000 已提交
1085
					   const size_t frames)
1086
{
1087 1088
	if (!sample_rate)
		return 0;
J
jp9000 已提交
1089

1090
	return (uint64_t)frames * 1000000000ULL / (uint64_t)sample_rate;
1091 1092
}

J
jp9000 已提交
1093
static inline size_t conv_time_to_frames(const size_t sample_rate,
J
jp9000 已提交
1094
					 const uint64_t duration)
J
jp9000 已提交
1095 1096 1097 1098 1099
{
	return (size_t)(duration * (uint64_t)sample_rate / 1000000000ULL);
}

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

J
jp9000 已提交
1102 1103 1104 1105
/* time threshold in nanoseconds to ensure audio timing is as seamless as
 * possible */
#define TS_SMOOTHING_THRESHOLD 70000000ULL

1106
static inline void reset_audio_timing(obs_source_t *source, uint64_t timestamp,
J
jp9000 已提交
1107
				      uint64_t os_time)
1108
{
J
jp9000 已提交
1109
	source->timing_set = true;
1110
	source->timing_adjust = os_time - timestamp;
1111
}
1112

1113 1114 1115 1116 1117
static void reset_audio_data(obs_source_t *source, uint64_t os_time)
{
	for (size_t i = 0; i < MAX_AUDIO_CHANNELS; i++) {
		if (source->audio_input_buf[i].size)
			circlebuf_pop_front(&source->audio_input_buf[i], NULL,
J
jp9000 已提交
1118
					    source->audio_input_buf[i].size);
1119 1120
	}

1121
	source->last_audio_input_buf_size = 0;
1122
	source->audio_ts = os_time;
1123
	source->next_audio_sys_ts_min = os_time;
1124 1125
}

J
jp9000 已提交
1126 1127
static void handle_ts_jump(obs_source_t *source, uint64_t expected, uint64_t ts,
			   uint64_t diff, uint64_t os_time)
1128
{
J
jp9000 已提交
1129 1130 1131 1132
	blog(LOG_DEBUG,
	     "Timestamp for source '%s' jumped by '%" PRIu64 "', "
	     "expected value %" PRIu64 ", input value %" PRIu64,
	     source->context.name, diff, expected, ts);
1133

1134
	pthread_mutex_lock(&source->audio_buf_mutex);
1135
	reset_audio_timing(source, ts, os_time);
1136
	pthread_mutex_unlock(&source->audio_buf_mutex);
1137 1138
}

1139
static void source_signal_audio_data(obs_source_t *source,
J
jp9000 已提交
1140
				     const struct audio_data *in, bool muted)
1141
{
1142
	pthread_mutex_lock(&source->audio_cb_mutex);
1143

1144 1145 1146 1147
	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);
	}
1148

1149
	pthread_mutex_unlock(&source->audio_cb_mutex);
1150 1151
}

1152 1153
static inline uint64_t uint64_diff(uint64_t ts1, uint64_t ts2)
{
J
jp9000 已提交
1154
	return (ts1 < ts2) ? (ts2 - ts1) : (ts1 - ts2);
1155 1156
}

1157 1158 1159 1160 1161 1162 1163
static inline size_t get_buf_placement(audio_t *audio, uint64_t offset)
{
	uint32_t sample_rate = audio_output_get_sample_rate(audio);
	return (size_t)(offset * (uint64_t)sample_rate / 1000000000ULL);
}

static void source_output_audio_place(obs_source_t *source,
J
jp9000 已提交
1164
				      const struct audio_data *in)
1165 1166 1167 1168 1169 1170 1171 1172 1173
{
	audio_t *audio = obs->audio.audio;
	size_t buf_placement;
	size_t channels = audio_output_get_channels(audio);
	size_t size = in->frames * sizeof(float);

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

J
jp9000 已提交
1174 1175 1176
	buf_placement =
		get_buf_placement(audio, in->timestamp - source->audio_ts) *
		sizeof(float);
1177 1178

#if DEBUG_AUDIO == 1
J
jp9000 已提交
1179 1180 1181 1182 1183
	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);
1184 1185
#endif

J
jp9000 已提交
1186 1187 1188 1189 1190
	/* 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++) {
1191 1192
		circlebuf_place(&source->audio_input_buf[i], buf_placement,
				in->data[i], size);
J
jp9000 已提交
1193
		circlebuf_pop_back(&source->audio_input_buf[i], NULL,
J
jp9000 已提交
1194 1195
				   source->audio_input_buf[i].size -
					   (buf_placement + size));
J
jp9000 已提交
1196
	}
1197 1198

	source->last_audio_input_buf_size = 0;
1199 1200 1201
}

static inline void source_output_audio_push_back(obs_source_t *source,
J
jp9000 已提交
1202
						 const struct audio_data *in)
1203 1204 1205
{
	audio_t *audio = obs->audio.audio;
	size_t channels = audio_output_get_channels(audio);
J
jp9000 已提交
1206 1207 1208 1209 1210
	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;
1211 1212

	for (size_t i = 0; i < channels; i++)
J
jp9000 已提交
1213 1214
		circlebuf_push_back(&source->audio_input_buf[i], in->data[i],
				    size);
1215 1216 1217 1218

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

1221 1222
static inline bool source_muted(obs_source_t *source, uint64_t os_time)
{
J
jp9000 已提交
1223
	if (source->push_to_mute_enabled && source->user_push_to_mute_pressed)
J
jp9000 已提交
1224 1225
		source->push_to_mute_stop_time =
			os_time + source->push_to_mute_delay * 1000000;
1226

J
jp9000 已提交
1227
	if (source->push_to_talk_enabled && source->user_push_to_talk_pressed)
J
jp9000 已提交
1228 1229
		source->push_to_talk_stop_time =
			os_time + source->push_to_talk_delay * 1000000;
1230

J
jp9000 已提交
1231
	bool push_to_mute_active = source->user_push_to_mute_pressed ||
J
jp9000 已提交
1232
				   os_time < source->push_to_mute_stop_time;
J
jp9000 已提交
1233
	bool push_to_talk_active = source->user_push_to_talk_pressed ||
J
jp9000 已提交
1234
				   os_time < source->push_to_talk_stop_time;
1235

J
jp9000 已提交
1236
	return !source->enabled || source->user_muted ||
J
jp9000 已提交
1237 1238
	       (source->push_to_mute_enabled && push_to_mute_active) ||
	       (source->push_to_talk_enabled && !push_to_talk_active);
1239 1240
}

1241
static void source_output_audio_data(obs_source_t *source,
J
jp9000 已提交
1242
				     const struct audio_data *data)
1243
{
1244
	size_t sample_rate = audio_output_get_sample_rate(obs->audio.audio);
1245
	struct audio_data in = *data;
1246
	uint64_t diff;
1247
	uint64_t os_time = os_gettime_ns();
1248
	int64_t sync_offset;
1249 1250
	bool using_direct_ts = false;
	bool push_back = false;
1251

1252 1253 1254 1255 1256
	/* 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;
1257 1258
		using_direct_ts = true;
	}
1259

1260
	if (!source->timing_set) {
1261
		reset_audio_timing(source, in.timestamp, os_time);
1262

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

1266
		/* smooth audio if within threshold */
1267
		if (diff > MAX_TS_VAR && !using_direct_ts)
1268
			handle_ts_jump(source, source->next_audio_ts_min,
J
jp9000 已提交
1269
				       in.timestamp, diff, os_time);
1270
		else if (diff < TS_SMOOTHING_THRESHOLD)
1271
			in.timestamp = source->next_audio_ts_min;
1272 1273
	}

J
jp9000 已提交
1274
	source->last_audio_ts = in.timestamp;
J
jp9000 已提交
1275 1276
	source->next_audio_ts_min =
		in.timestamp + conv_frames_to_time(sample_rate, in.frames);
1277

1278
	in.timestamp += source->timing_adjust;
1279

1280 1281
	pthread_mutex_lock(&source->audio_buf_mutex);

J
jp9000 已提交
1282
	if (source->next_audio_sys_ts_min == in.timestamp) {
1283
		push_back = true;
1284 1285

	} else if (source->next_audio_sys_ts_min) {
J
jp9000 已提交
1286
		diff = uint64_diff(source->next_audio_sys_ts_min, in.timestamp);
1287 1288

		if (diff < TS_SMOOTHING_THRESHOLD) {
1289 1290
			push_back = true;

J
jp9000 已提交
1291
			/* This typically only happens if used with async video when
1292 1293 1294 1295 1296
		 * 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. */
1297
		} else if (diff > MAX_TS_VAR) {
J
jp9000 已提交
1298
			reset_audio_timing(source, data->timestamp, os_time);
1299 1300
			in.timestamp = data->timestamp + source->timing_adjust;
		}
J
jp9000 已提交
1301 1302
	}

1303 1304
	sync_offset = source->sync_offset;
	in.timestamp += sync_offset;
1305 1306
	in.timestamp -= source->resample_offset;

J
jp9000 已提交
1307 1308
	source->next_audio_sys_ts_min =
		source->next_audio_ts_min + source->timing_adjust;
1309 1310 1311 1312 1313 1314

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

J
jp9000 已提交
1316 1317 1318 1319 1320 1321
	if (source->monitoring_type != OBS_MONITORING_TYPE_MONITOR_ONLY) {
		if (push_back && source->audio_ts)
			source_output_audio_push_back(source, &in);
		else
			source_output_audio_place(source, &in);
	}
1322 1323 1324

	pthread_mutex_unlock(&source->audio_buf_mutex);

1325
	source_signal_audio_data(source, data, source_muted(source, os_time));
1326 1327
}

1328 1329 1330 1331 1332 1333
enum convert_type {
	CONVERT_NONE,
	CONVERT_NV12,
	CONVERT_420,
	CONVERT_422_U,
	CONVERT_422_Y,
J
James Park 已提交
1334
	CONVERT_444,
1335 1336
	CONVERT_800,
	CONVERT_RGB_LIMITED,
1337
	CONVERT_BGR3,
1338 1339
};

1340
static inline enum convert_type get_convert_type(enum video_format format,
J
jp9000 已提交
1341
						 bool full_range)
1342
{
1343
	switch (format) {
1344 1345 1346 1347
	case VIDEO_FORMAT_I420:
		return CONVERT_420;
	case VIDEO_FORMAT_NV12:
		return CONVERT_NV12;
J
James Park 已提交
1348 1349
	case VIDEO_FORMAT_I444:
		return CONVERT_444;
1350 1351 1352 1353 1354 1355 1356

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

J
jpk 已提交
1357
	case VIDEO_FORMAT_Y800:
1358 1359
		return CONVERT_800;

1360
	case VIDEO_FORMAT_NONE:
1361 1362 1363
	case VIDEO_FORMAT_RGBA:
	case VIDEO_FORMAT_BGRA:
	case VIDEO_FORMAT_BGRX:
1364
		return full_range ? CONVERT_NONE : CONVERT_RGB_LIMITED;
1365 1366 1367

	case VIDEO_FORMAT_BGR3:
		return CONVERT_BGR3;
1368 1369 1370 1371 1372
	}

	return CONVERT_NONE;
}

1373
static inline bool set_packed422_sizes(struct obs_source *source,
J
jp9000 已提交
1374
				       const struct obs_source_frame *frame)
1375
{
J
jp9000 已提交
1376
	source->async_convert_width = frame->width / 2;
J
James Park 已提交
1377
	source->async_convert_height = frame->height;
1378 1379 1380 1381
	source->async_texture_format = GS_BGRA;
	return true;
}

J
James Park 已提交
1382
static inline bool set_planar444_sizes(struct obs_source *source,
J
jp9000 已提交
1383
				       const struct obs_source_frame *frame)
J
James Park 已提交
1384
{
J
jp9000 已提交
1385 1386 1387
	source->async_convert_width = frame->width;
	source->async_convert_height = frame->height * 3;
	source->async_texture_format = GS_R8;
J
James Park 已提交
1388 1389 1390 1391 1392
	source->async_plane_offset[0] = (int)(frame->data[1] - frame->data[0]);
	source->async_plane_offset[1] = (int)(frame->data[2] - frame->data[0]);
	return true;
}

1393
static inline bool set_planar420_sizes(struct obs_source *source,
J
jp9000 已提交
1394
				       const struct obs_source_frame *frame)
1395 1396
{
	uint32_t size = frame->width * frame->height;
J
jp9000 已提交
1397
	size += size / 2;
1398

J
jp9000 已提交
1399 1400 1401
	source->async_convert_width = frame->width;
	source->async_convert_height = size / frame->width;
	source->async_texture_format = GS_R8;
1402 1403
	source->async_plane_offset[0] = (int)(frame->data[1] - frame->data[0]);
	source->async_plane_offset[1] = (int)(frame->data[2] - frame->data[0]);
1404 1405 1406
	return true;
}

J
jp9000 已提交
1407
static inline bool set_nv12_sizes(struct obs_source *source,
J
jp9000 已提交
1408
				  const struct obs_source_frame *frame)
J
jp9000 已提交
1409 1410
{
	uint32_t size = frame->width * frame->height;
J
jp9000 已提交
1411
	size += size / 2;
J
jp9000 已提交
1412

J
jp9000 已提交
1413 1414 1415
	source->async_convert_width = frame->width;
	source->async_convert_height = size / frame->width;
	source->async_texture_format = GS_R8;
1416
	source->async_plane_offset[0] = (int)(frame->data[1] - frame->data[0]);
J
jp9000 已提交
1417 1418 1419
	return true;
}

1420
static inline bool set_y800_sizes(struct obs_source *source,
J
jp9000 已提交
1421
				  const struct obs_source_frame *frame)
1422
{
J
jp9000 已提交
1423 1424 1425
	source->async_convert_width = frame->width;
	source->async_convert_height = frame->height;
	source->async_texture_format = GS_R8;
1426 1427 1428 1429
	return true;
}

static inline bool set_rgb_limited_sizes(struct obs_source *source,
J
jp9000 已提交
1430
					 const struct obs_source_frame *frame)
1431
{
J
jp9000 已提交
1432 1433 1434
	source->async_convert_width = frame->width;
	source->async_convert_height = frame->height;
	source->async_texture_format = convert_video_format(frame->format);
1435 1436 1437
	return true;
}

1438
static inline bool set_bgr3_sizes(struct obs_source *source,
J
jp9000 已提交
1439
				  const struct obs_source_frame *frame)
1440
{
J
jp9000 已提交
1441 1442 1443
	source->async_convert_width = frame->width * 3;
	source->async_convert_height = frame->height;
	source->async_texture_format = GS_R8;
1444 1445 1446
	return true;
}

1447
static inline bool init_gpu_conversion(struct obs_source *source,
J
jp9000 已提交
1448
				       const struct obs_source_frame *frame)
1449
{
1450
	switch (get_convert_type(frame->format, frame->full_range)) {
J
jp9000 已提交
1451 1452 1453
	case CONVERT_422_Y:
	case CONVERT_422_U:
		return set_packed422_sizes(source, frame);
1454

J
jp9000 已提交
1455 1456
	case CONVERT_420:
		return set_planar420_sizes(source, frame);
J
James Park 已提交
1457

J
jp9000 已提交
1458 1459
	case CONVERT_NV12:
		return set_nv12_sizes(source, frame);
1460

J
jp9000 已提交
1461 1462
	case CONVERT_444:
		return set_planar444_sizes(source, frame);
1463

J
jp9000 已提交
1464 1465
	case CONVERT_800:
		return set_y800_sizes(source, frame);
1466

J
jp9000 已提交
1467 1468
	case CONVERT_RGB_LIMITED:
		return set_rgb_limited_sizes(source, frame);
1469

J
jp9000 已提交
1470 1471
	case CONVERT_BGR3:
		return set_bgr3_sizes(source, frame);
1472

J
jp9000 已提交
1473 1474 1475
	case CONVERT_NONE:
		assert(false && "No conversion requested");
		break;
1476 1477 1478 1479
	}
	return false;
}

1480
bool set_async_texture_size(struct obs_source *source,
J
jp9000 已提交
1481
			    const struct obs_source_frame *frame)
1482
{
J
jp9000 已提交
1483 1484
	enum convert_type cur =
		get_convert_type(frame->format, frame->full_range);
1485

J
jp9000 已提交
1486 1487 1488
	if (source->async_width == frame->width &&
	    source->async_height == frame->height &&
	    source->async_format == frame->format &&
1489
	    source->async_full_range == frame->full_range)
1490 1491
		return true;

J
jp9000 已提交
1492 1493 1494
	source->async_width = frame->width;
	source->async_height = frame->height;
	source->async_format = frame->format;
1495
	source->async_full_range = frame->full_range;
1496

1497 1498
	gs_enter_context(obs->video.graphics);

1499
	gs_texture_destroy(source->async_texture);
J
jp9000 已提交
1500
	gs_texture_destroy(source->async_prev_texture);
1501
	gs_texrender_destroy(source->async_texrender);
J
jp9000 已提交
1502 1503 1504
	gs_texrender_destroy(source->async_prev_texrender);
	source->async_texture = NULL;
	source->async_prev_texture = NULL;
1505
	source->async_texrender = NULL;
J
jp9000 已提交
1506
	source->async_prev_texrender = NULL;
1507 1508 1509 1510

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

J
jp9000 已提交
1511
		enum gs_color_format format =
J
jpark37 已提交
1512
			(cur == CONVERT_RGB_LIMITED)
J
jp9000 已提交
1513 1514
				? convert_video_format(frame->format)
				: GS_BGRX;
1515
		source->async_texrender =
1516
			gs_texrender_create(format, GS_ZS_NONE);
1517

1518
		source->async_texture = gs_texture_create(
J
jp9000 已提交
1519 1520 1521
			source->async_convert_width,
			source->async_convert_height,
			source->async_texture_format, 1, NULL, GS_DYNAMIC);
1522 1523

	} else {
J
jp9000 已提交
1524 1525
		enum gs_color_format format =
			convert_video_format(frame->format);
1526 1527
		source->async_gpu_conversion = false;

J
jp9000 已提交
1528 1529 1530
		source->async_texture = gs_texture_create(frame->width,
							  frame->height, format,
							  1, NULL, GS_DYNAMIC);
1531 1532
	}

J
jp9000 已提交
1533 1534 1535
	if (deinterlacing_enabled(source))
		set_deinterlace_texture_size(source);

1536 1537
	gs_leave_context();

1538
	return !!source->async_texture;
1539 1540
}

1541
static void upload_raw_frame(gs_texture_t *tex,
J
jp9000 已提交
1542
			     const struct obs_source_frame *frame)
1543
{
1544
	switch (get_convert_type(frame->format, frame->full_range)) {
J
jp9000 已提交
1545 1546 1547 1548 1549 1550 1551 1552
	case CONVERT_422_U:
	case CONVERT_422_Y:
	case CONVERT_800:
	case CONVERT_RGB_LIMITED:
	case CONVERT_BGR3:
		gs_texture_set_image(tex, frame->data[0], frame->linesize[0],
				     false);
		break;
1553

J
jp9000 已提交
1554 1555 1556 1557 1558
	case CONVERT_420:
	case CONVERT_NV12:
	case CONVERT_444:
		gs_texture_set_image(tex, frame->data[0], frame->width, false);
		break;
J
James Park 已提交
1559

J
jp9000 已提交
1560 1561 1562
	case CONVERT_NONE:
		assert(false && "No conversion requested");
		break;
1563 1564 1565
	}
}

1566
static const char *select_conversion_technique(enum video_format format,
J
jp9000 已提交
1567
					       bool full_range)
1568 1569
{
	switch (format) {
J
jp9000 已提交
1570 1571
	case VIDEO_FORMAT_UYVY:
		return "UYVY_Reverse";
1572

J
jp9000 已提交
1573 1574
	case VIDEO_FORMAT_YUY2:
		return "YUY2_Reverse";
1575

J
jp9000 已提交
1576 1577
	case VIDEO_FORMAT_YVYU:
		return "YVYU_Reverse";
1578

J
jp9000 已提交
1579 1580
	case VIDEO_FORMAT_I420:
		return "I420_Reverse";
1581

J
jp9000 已提交
1582 1583
	case VIDEO_FORMAT_NV12:
		return "NV12_Reverse";
J
James Park 已提交
1584

J
jp9000 已提交
1585 1586
	case VIDEO_FORMAT_I444:
		return "I444_Reverse";
1587

J
jp9000 已提交
1588 1589
	case VIDEO_FORMAT_Y800:
		return full_range ? "Y800_Full" : "Y800_Limited";
1590

J
jp9000 已提交
1591 1592
	case VIDEO_FORMAT_BGR3:
		return full_range ? "BGR3_Full" : "BGR3_Limited";
1593

J
jp9000 已提交
1594 1595 1596 1597 1598 1599 1600 1601 1602
	case VIDEO_FORMAT_BGRA:
	case VIDEO_FORMAT_BGRX:
	case VIDEO_FORMAT_RGBA:
	case VIDEO_FORMAT_NONE:
		if (full_range)
			assert(false && "No conversion requested");
		else
			return "RGB_Limited";
		break;
1603 1604 1605 1606
	}
	return NULL;
}

1607
static inline void set_eparam(gs_effect_t *effect, const char *name, float val)
1608
{
1609
	gs_eparam_t *param = gs_effect_get_param_by_name(effect, name);
1610
	gs_effect_set_float(param, val);
1611 1612
}

1613 1614 1615 1616 1617 1618
static inline void set_eparami(gs_effect_t *effect, const char *name, int val)
{
	gs_eparam_t *param = gs_effect_get_param_by_name(effect, name);
	gs_effect_set_int(param, val);
}

1619
static bool update_async_texrender(struct obs_source *source,
J
jp9000 已提交
1620 1621
				   const struct obs_source_frame *frame,
				   gs_texture_t *tex, gs_texrender_t *texrender)
1622
{
1623 1624
	GS_DEBUG_MARKER_BEGIN(GS_DEBUG_COLOR_CONVERT_FORMAT, "Convert Format");

1625
	gs_texrender_reset(texrender);
1626 1627 1628 1629 1630 1631

	upload_raw_frame(tex, frame);

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

J
jp9000 已提交
1632
	float convert_width = (float)source->async_convert_width;
1633

1634
	gs_effect_t *conv = obs->video.conversion_effect;
J
jp9000 已提交
1635 1636
	const char *tech_name =
		select_conversion_technique(frame->format, frame->full_range);
1637
	gs_technique_t *tech = gs_effect_get_technique(conv, tech_name);
1638

1639 1640
	if (!gs_texrender_begin(texrender, cx, cy)) {
		GS_DEBUG_MARKER_END();
1641
		return false;
1642
	}
1643

1644 1645
	gs_enable_blending(false);

1646 1647
	gs_technique_begin(tech);
	gs_technique_begin_pass(tech, 0);
1648

1649
	gs_effect_set_texture(gs_effect_get_param_by_name(conv, "image"), tex);
J
jp9000 已提交
1650
	set_eparam(conv, "width", (float)cx);
1651
	set_eparam(conv, "height", (float)cy);
J
jp9000 已提交
1652 1653 1654
	set_eparam(conv, "width_d2", cx * 0.5f);
	set_eparam(conv, "width_d2_i", 1.0f / (cx * 0.5f));
	set_eparam(conv, "input_width_i_d2", (1.0f / convert_width) * 0.5f);
1655 1656 1657 1658

	set_eparami(conv, "int_width", (int)cx);
	set_eparami(conv, "int_input_width", (int)source->async_convert_width);
	set_eparami(conv, "int_u_plane_offset",
J
jp9000 已提交
1659
		    (int)source->async_plane_offset[0]);
1660
	set_eparami(conv, "int_v_plane_offset",
J
jp9000 已提交
1661
		    (int)source->async_plane_offset[1]);
1662

J
James Park 已提交
1663
	gs_effect_set_val(gs_effect_get_param_by_name(conv, "color_matrix"),
J
jp9000 已提交
1664
			  frame->color_matrix, sizeof(float) * 16);
J
James Park 已提交
1665
	if (!frame->full_range) {
J
jp9000 已提交
1666 1667
		gs_eparam_t *min_param =
			gs_effect_get_param_by_name(conv, "color_range_min");
J
James Park 已提交
1668
		gs_effect_set_val(min_param, frame->color_range_min,
J
jp9000 已提交
1669 1670 1671
				  sizeof(float) * 3);
		gs_eparam_t *max_param =
			gs_effect_get_param_by_name(conv, "color_range_max");
J
James Park 已提交
1672
		gs_effect_set_val(max_param, frame->color_range_max,
J
jp9000 已提交
1673
				  sizeof(float) * 3);
J
James Park 已提交
1674 1675
	}

1676
	gs_draw(GS_TRIS, 0, 3);
1677

1678 1679
	gs_technique_end_pass(tech);
	gs_technique_end(tech);
1680

1681 1682
	gs_enable_blending(true);

1683
	gs_texrender_end(texrender);
1684

1685
	GS_DEBUG_MARKER_END();
1686 1687 1688
	return true;
}

1689
bool update_async_texture(struct obs_source *source,
J
jp9000 已提交
1690 1691
			  const struct obs_source_frame *frame,
			  gs_texture_t *tex, gs_texrender_t *texrender)
1692
{
1693
	enum convert_type type;
1694

J
jp9000 已提交
1695
	source->async_flip = frame->flip;
1696

1697
	if (source->async_gpu_conversion && texrender)
1698
		return update_async_texrender(source, frame, tex, texrender);
1699

1700
	type = get_convert_type(frame->format, frame->full_range);
1701
	if (type == CONVERT_NONE) {
1702
		gs_texture_set_image(tex, frame->data[0], frame->linesize[0],
J
jp9000 已提交
1703
				     false);
1704 1705 1706
		return true;
	}

1707
	return false;
1708 1709
}

1710
static inline void obs_source_draw_texture(struct obs_source *source,
J
jp9000 已提交
1711
					   gs_effect_t *effect)
1712
{
1713
	gs_texture_t *tex = source->async_texture;
J
jp9000 已提交
1714
	gs_eparam_t *param;
1715

1716 1717
	if (source->async_texrender)
		tex = gs_texrender_get_texture(source->async_texrender);
1718

1719 1720
	param = gs_effect_get_param_by_name(effect, "image");
	gs_effect_set_texture(param, tex);
1721

1722 1723
	gs_draw_sprite(tex, source->async_flip ? GS_FLIP_V : 0, 0, 0);
}
1724

1725 1726
static void obs_source_draw_async_texture(struct obs_source *source)
{
J
jp9000 已提交
1727 1728 1729
	gs_effect_t *effect = gs_get_effect();
	bool def_draw = (!effect);
	gs_technique_t *tech = NULL;
1730 1731

	if (def_draw) {
1732
		effect = obs_get_base_effect(OBS_EFFECT_DEFAULT);
J
James Park 已提交
1733
		tech = gs_effect_get_technique(effect, "Draw");
1734 1735
		gs_technique_begin(tech);
		gs_technique_begin_pass(tech, 0);
1736 1737
	}

J
James Park 已提交
1738
	obs_source_draw_texture(source, effect);
1739 1740

	if (def_draw) {
1741 1742
		gs_technique_end_pass(tech);
		gs_technique_end(tech);
1743
	}
1744 1745
}

1746
static void obs_source_update_async_video(obs_source_t *source)
1747
{
1748 1749 1750
	if (!source->async_rendered) {
		struct obs_source_frame *frame = obs_source_get_frame(source);

1751 1752 1753
		if (frame)
			frame = filter_async_video(source, frame);

1754 1755
		source->async_rendered = true;
		if (frame) {
1756 1757
			if (!source->async_decoupled ||
			    !source->async_unbuffered) {
J
jp9000 已提交
1758 1759
				source->timing_adjust = obs->video.video_time -
							frame->timestamp;
1760 1761
				source->timing_set = true;
			}
1762

1763
			if (source->async_update_texture) {
1764
				update_async_texture(source, frame,
J
jp9000 已提交
1765 1766
						     source->async_texture,
						     source->async_texrender);
1767
				source->async_update_texture = false;
1768
			}
1769

1770 1771
			obs_source_release_frame(source, frame);
		}
1772
	}
1773
}
1774

1775 1776
static inline void obs_source_render_async_video(obs_source_t *source)
{
1777
	if (source->async_texture && source->async_active)
1778
		obs_source_draw_async_texture(source);
1779 1780
}

1781
static inline void obs_source_render_filters(obs_source_t *source)
1782
{
1783 1784 1785 1786 1787 1788 1789
	obs_source_t *first_filter;

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

1790
	source->rendering_filter = true;
1791
	obs_source_video_render(first_filter);
1792
	source->rendering_filter = false;
1793 1794

	obs_source_release(first_filter);
1795 1796
}

1797
void obs_source_default_render(obs_source_t *source)
1798
{
J
jp9000 已提交
1799 1800 1801
	gs_effect_t *effect = obs->video.default_effect;
	gs_technique_t *tech = gs_effect_get_technique(effect, "Draw");
	size_t passes, i;
1802

1803
	passes = gs_technique_begin(tech);
1804
	for (i = 0; i < passes; i++) {
1805
		gs_technique_begin_pass(tech, i);
1806 1807
		if (source->context.data)
			source->info.video_render(source->context.data, effect);
1808
		gs_technique_end_pass(tech);
1809
	}
1810
	gs_technique_end(tech);
1811 1812
}

1813
static inline void obs_source_main_render(obs_source_t *source)
1814
{
J
jp9000 已提交
1815 1816
	uint32_t flags = source->info.output_flags;
	bool custom_draw = (flags & OBS_SOURCE_CUSTOM_DRAW) != 0;
1817
	bool default_effect = !source->filter_parent &&
J
jp9000 已提交
1818
			      source->filters.num == 0 && !custom_draw;
1819 1820

	if (default_effect)
1821
		obs_source_default_render(source);
1822
	else if (source->context.data)
1823
		source->info.video_render(source->context.data,
J
jp9000 已提交
1824
					  custom_draw ? NULL : gs_get_effect());
1825 1826
}

1827 1828
static bool ready_async_frame(obs_source_t *source, uint64_t sys_time);

1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846
#if GS_USE_DEBUG_MARKERS
static const char *get_type_format(enum obs_source_type type)
{
	switch (type) {
	case OBS_SOURCE_TYPE_INPUT:
		return "Input: %s";
	case OBS_SOURCE_TYPE_FILTER:
		return "Filter: %s";
	case OBS_SOURCE_TYPE_TRANSITION:
		return "Transition: %s";
	case OBS_SOURCE_TYPE_SCENE:
		return "Scene: %s";
	default:
		return "[Unknown]: %s";
	}
}
#endif

1847
static inline void render_video(obs_source_t *source)
J
jp9000 已提交
1848
{
1849
	if (source->info.type != OBS_SOURCE_TYPE_FILTER &&
1850 1851 1852
	    (source->info.output_flags & OBS_SOURCE_VIDEO) == 0) {
		if (source->filter_parent)
			obs_source_skip_video_filter(source);
1853
		return;
1854
	}
1855

1856 1857
	if (source->info.type == OBS_SOURCE_TYPE_INPUT &&
	    (source->info.output_flags & OBS_SOURCE_ASYNC) != 0 &&
J
jp9000 已提交
1858 1859 1860
	    !source->rendering_filter) {
		if (deinterlacing_enabled(source))
			deinterlace_update_async_video(source);
1861
		obs_source_update_async_video(source);
J
jp9000 已提交
1862
	}
1863

1864
	if (!source->context.data || !source->enabled) {
1865 1866 1867 1868 1869
		if (source->filter_parent)
			obs_source_skip_video_filter(source);
		return;
	}

1870
	GS_DEBUG_MARKER_BEGIN_FORMAT(GS_DEBUG_COLOR_SOURCE,
J
jp9000 已提交
1871 1872
				     get_type_format(source->info.type),
				     obs_source_get_name(source));
1873

1874 1875
	if (source->filters.num && !source->rendering_filter)
		obs_source_render_filters(source);
1876

1877 1878 1879 1880
	else if (source->info.video_render)
		obs_source_main_render(source);

	else if (source->filter_target)
1881 1882
		obs_source_video_render(source->filter_target);

J
jp9000 已提交
1883 1884 1885
	else if (deinterlacing_enabled(source))
		deinterlace_render(source);

1886
	else
1887
		obs_source_render_async_video(source);
1888 1889

	GS_DEBUG_MARKER_END();
J
jp9000 已提交
1890 1891
}

1892 1893 1894 1895 1896 1897 1898 1899 1900 1901
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);
}

1902
static uint32_t get_base_width(const obs_source_t *source)
J
jp9000 已提交
1903
{
1904
	bool is_filter = !!source->filter_parent;
1905
	bool func_valid = source->context.data && source->info.get_width;
1906

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

1910
	} else if (func_valid && (!is_filter || source->enabled)) {
1911
		return source->info.get_width(source->context.data);
1912

1913
	} else if (is_filter) {
1914 1915 1916
		return get_base_width(source->filter_target);
	}

1917
	return source->async_active ? source->async_width : 0;
J
jp9000 已提交
1918 1919
}

1920
static uint32_t get_base_height(const obs_source_t *source)
J
jp9000 已提交
1921
{
1922
	bool is_filter = !!source->filter_parent;
1923
	bool func_valid = source->context.data && source->info.get_height;
1924

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

1928
	} else if (func_valid && (!is_filter || source->enabled)) {
1929
		return source->info.get_height(source->context.data);
1930

1931
	} else if (is_filter) {
1932 1933 1934
		return get_base_height(source->filter_target);
	}

1935
	return source->async_active ? source->async_height : 0;
J
jp9000 已提交
1936 1937
}

1938 1939 1940 1941 1942 1943
static uint32_t get_recurse_width(obs_source_t *source)
{
	uint32_t width;

	pthread_mutex_lock(&source->filter_mutex);

J
jp9000 已提交
1944 1945
	width = (source->filters.num) ? get_base_width(source->filters.array[0])
				      : get_base_width(source);
1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957

	pthread_mutex_unlock(&source->filter_mutex);

	return width;
}

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

	pthread_mutex_lock(&source->filter_mutex);

J
jp9000 已提交
1958 1959 1960
	height = (source->filters.num)
			 ? get_base_height(source->filters.array[0])
			 : get_base_height(source);
1961 1962 1963 1964 1965 1966 1967 1968

	pthread_mutex_unlock(&source->filter_mutex);

	return height;
}

uint32_t obs_source_get_width(obs_source_t *source)
{
J
jp9000 已提交
1969 1970
	if (!data_valid(source, "obs_source_get_width"))
		return 0;
1971

J
jp9000 已提交
1972 1973 1974
	return (source->info.type != OBS_SOURCE_TYPE_FILTER)
		       ? get_recurse_width(source)
		       : get_base_width(source);
1975 1976 1977 1978
}

uint32_t obs_source_get_height(obs_source_t *source)
{
J
jp9000 已提交
1979 1980
	if (!data_valid(source, "obs_source_get_height"))
		return 0;
1981

J
jp9000 已提交
1982 1983 1984
	return (source->info.type != OBS_SOURCE_TYPE_FILTER)
		       ? get_recurse_height(source)
		       : get_base_height(source);
1985 1986
}

1987 1988
uint32_t obs_source_get_base_width(obs_source_t *source)
{
J
jp9000 已提交
1989 1990
	if (!data_valid(source, "obs_source_get_base_width"))
		return 0;
1991 1992 1993 1994 1995 1996

	return get_base_width(source);
}

uint32_t obs_source_get_base_height(obs_source_t *source)
{
J
jp9000 已提交
1997 1998
	if (!data_valid(source, "obs_source_get_base_height"))
		return 0;
1999 2000 2001 2002

	return get_base_height(source);
}

2003
obs_source_t *obs_filter_get_parent(const obs_source_t *filter)
2004
{
J
jp9000 已提交
2005 2006 2007
	return obs_ptr_valid(filter, "obs_filter_get_parent")
		       ? filter->filter_parent
		       : NULL;
2008 2009
}

2010
obs_source_t *obs_filter_get_target(const obs_source_t *filter)
J
jp9000 已提交
2011
{
J
jp9000 已提交
2012 2013 2014
	return obs_ptr_valid(filter, "obs_filter_get_target")
		       ? filter->filter_target
		       : NULL;
J
jp9000 已提交
2015 2016
}

2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028
static bool filter_compatible(obs_source_t *source, obs_source_t *filter)
{
	uint32_t s_caps = source->info.output_flags;
	uint32_t f_caps = filter->info.output_flags;

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

	return (s_caps & f_caps) == f_caps;
}

2029
void obs_source_filter_add(obs_source_t *source, obs_source_t *filter)
J
jp9000 已提交
2030
{
2031 2032
	struct calldata cd;
	uint8_t stack[128];
J
jp9000 已提交
2033

2034 2035 2036
	if (!obs_source_valid(source, "obs_source_filter_add"))
		return;
	if (!obs_ptr_valid(filter, "obs_source_filter_add"))
J
jp9000 已提交
2037 2038
		return;

2039 2040
	pthread_mutex_lock(&source->filter_mutex);

J
jp9000 已提交
2041
	if (da_find(source->filters, &filter, 0) != DARRAY_INVALID) {
J
jp9000 已提交
2042
		blog(LOG_WARNING, "Tried to add a filter that was already "
J
jp9000 已提交
2043
				  "present on the source");
2044
		pthread_mutex_unlock(&source->filter_mutex);
J
jp9000 已提交
2045 2046 2047
		return;
	}

2048 2049 2050 2051 2052
	if (!filter_compatible(source, filter)) {
		pthread_mutex_unlock(&source->filter_mutex);
		return;
	}

2053 2054
	obs_source_addref(filter);

2055
	filter->filter_parent = source;
J
jp9000 已提交
2056 2057
	filter->filter_target = !source->filters.num ? source
						     : source->filters.array[0];
2058

J
jp9000 已提交
2059
	da_insert(source->filters, 0, &filter);
2060 2061 2062

	pthread_mutex_unlock(&source->filter_mutex);

2063
	calldata_init_fixed(&cd, stack, sizeof(stack));
J
jp9000 已提交
2064 2065 2066 2067
	calldata_set_ptr(&cd, "source", source);
	calldata_set_ptr(&cd, "filter", filter);

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

2069
	blog(LOG_DEBUG, "- filter '%s' (%s) added to source '%s'",
J
jp9000 已提交
2070
	     filter->context.name, filter->info.id, source->context.name);
J
jp9000 已提交
2071 2072
}

2073
static bool obs_source_filter_remove_refless(obs_source_t *source,
J
jp9000 已提交
2074
					     obs_source_t *filter)
J
jp9000 已提交
2075
{
2076 2077
	struct calldata cd;
	uint8_t stack[128];
2078 2079 2080 2081 2082
	size_t idx;

	pthread_mutex_lock(&source->filter_mutex);

	idx = da_find(source->filters, &filter, 0);
2083 2084
	if (idx == DARRAY_INVALID) {
		pthread_mutex_unlock(&source->filter_mutex);
2085
		return false;
2086
	}
J
jp9000 已提交
2087 2088

	if (idx > 0) {
J
jp9000 已提交
2089
		obs_source_t *prev = source->filters.array[idx - 1];
J
jp9000 已提交
2090 2091 2092 2093
		prev->filter_target = filter->filter_target;
	}

	da_erase(source->filters, idx);
2094 2095 2096

	pthread_mutex_unlock(&source->filter_mutex);

2097
	calldata_init_fixed(&cd, stack, sizeof(stack));
J
jp9000 已提交
2098 2099 2100 2101 2102
	calldata_set_ptr(&cd, "source", source);
	calldata_set_ptr(&cd, "filter", filter);

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

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

2106 2107
	if (filter->info.filter_remove)
		filter->info.filter_remove(filter->context.data,
J
jp9000 已提交
2108
					   filter->filter_parent);
2109

2110
	filter->filter_parent = NULL;
J
jp9000 已提交
2111
	filter->filter_target = NULL;
2112 2113
	return true;
}
2114

2115 2116
void obs_source_filter_remove(obs_source_t *source, obs_source_t *filter)
{
2117 2118 2119 2120 2121
	if (!obs_source_valid(source, "obs_source_filter_remove"))
		return;
	if (!obs_ptr_valid(filter, "obs_source_filter_remove"))
		return;

2122 2123
	if (obs_source_filter_remove_refless(source, filter))
		obs_source_release(filter);
J
jp9000 已提交
2124 2125
}

2126
static size_t find_next_filter(obs_source_t *source, obs_source_t *filter,
J
jp9000 已提交
2127
			       size_t cur_idx)
2128 2129 2130 2131 2132
{
	bool curAsync = (filter->info.output_flags & OBS_SOURCE_ASYNC) != 0;
	bool nextAsync;
	obs_source_t *next;

J
jp9000 已提交
2133
	if (cur_idx == source->filters.num - 1)
2134 2135
		return DARRAY_INVALID;

J
jp9000 已提交
2136
	next = source->filters.array[cur_idx + 1];
2137 2138 2139
	nextAsync = (next->info.output_flags & OBS_SOURCE_ASYNC);

	if (nextAsync == curAsync)
J
jp9000 已提交
2140
		return cur_idx + 1;
2141
	else
J
jp9000 已提交
2142
		return find_next_filter(source, filter, cur_idx + 1);
2143 2144 2145
}

static size_t find_prev_filter(obs_source_t *source, obs_source_t *filter,
J
jp9000 已提交
2146
			       size_t cur_idx)
2147 2148 2149 2150 2151 2152 2153 2154
{
	bool curAsync = (filter->info.output_flags & OBS_SOURCE_ASYNC) != 0;
	bool prevAsync;
	obs_source_t *prev;

	if (cur_idx == 0)
		return DARRAY_INVALID;

J
jp9000 已提交
2155
	prev = source->filters.array[cur_idx - 1];
2156 2157 2158
	prevAsync = (prev->info.output_flags & OBS_SOURCE_ASYNC);

	if (prevAsync == curAsync)
J
jp9000 已提交
2159
		return cur_idx - 1;
2160
	else
J
jp9000 已提交
2161
		return find_prev_filter(source, filter, cur_idx - 1);
2162 2163 2164
}

/* moves filters above/below matching filter types */
J
jp9000 已提交
2165 2166
static bool move_filter_dir(obs_source_t *source, obs_source_t *filter,
			    enum obs_order_movement movement)
J
jp9000 已提交
2167
{
2168
	size_t idx;
J
jp9000 已提交
2169 2170

	idx = da_find(source->filters, &filter, 0);
J
jp9000 已提交
2171
	if (idx == DARRAY_INVALID)
2172
		return false;
J
jp9000 已提交
2173

J
jp9000 已提交
2174
	if (movement == OBS_ORDER_MOVE_UP) {
2175 2176
		size_t next_id = find_next_filter(source, filter, idx);
		if (next_id == DARRAY_INVALID)
2177
			return false;
2178
		da_move_item(source->filters, idx, next_id);
J
jp9000 已提交
2179

J
jp9000 已提交
2180
	} else if (movement == OBS_ORDER_MOVE_DOWN) {
2181 2182
		size_t prev_id = find_prev_filter(source, filter, idx);
		if (prev_id == DARRAY_INVALID)
2183
			return false;
2184
		da_move_item(source->filters, idx, prev_id);
J
jp9000 已提交
2185

J
jp9000 已提交
2186
	} else if (movement == OBS_ORDER_MOVE_TOP) {
J
jp9000 已提交
2187
		if (idx == source->filters.num - 1)
2188
			return false;
J
jp9000 已提交
2189
		da_move_item(source->filters, idx, source->filters.num - 1);
J
jp9000 已提交
2190

J
jp9000 已提交
2191
	} else if (movement == OBS_ORDER_MOVE_BOTTOM) {
J
jp9000 已提交
2192
		if (idx == 0)
2193
			return false;
J
jp9000 已提交
2194 2195 2196
		da_move_item(source->filters, idx, 0);
	}

2197
	/* reorder filter targets, not the nicest way of dealing with things */
2198
	for (size_t i = 0; i < source->filters.num; i++) {
J
jp9000 已提交
2199 2200 2201 2202
		obs_source_t *next_filter =
			(i == source->filters.num - 1)
				? source
				: source->filters.array[i + 1];
2203

J
jp9000 已提交
2204 2205
		source->filters.array[i]->filter_target = next_filter;
	}
2206

2207 2208 2209 2210
	return true;
}

void obs_source_filter_set_order(obs_source_t *source, obs_source_t *filter,
J
jp9000 已提交
2211
				 enum obs_order_movement movement)
2212 2213
{
	bool success;
2214 2215 2216 2217

	if (!obs_source_valid(source, "obs_source_filter_set_order"))
		return;
	if (!obs_ptr_valid(filter, "obs_source_filter_set_order"))
2218 2219 2220 2221 2222 2223 2224 2225
		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 已提交
2226 2227
}

2228
obs_data_t *obs_source_get_settings(const obs_source_t *source)
J
jp9000 已提交
2229
{
2230 2231
	if (!obs_source_valid(source, "obs_source_get_settings"))
		return NULL;
J
jp9000 已提交
2232

2233 2234
	obs_data_addref(source->context.settings);
	return source->context.settings;
J
jp9000 已提交
2235 2236
}

2237
struct obs_source_frame *filter_async_video(obs_source_t *source,
J
jp9000 已提交
2238
					    struct obs_source_frame *in)
2239 2240
{
	size_t i;
2241 2242 2243

	pthread_mutex_lock(&source->filter_mutex);

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

2247 2248 2249
		if (!filter->enabled)
			continue;

2250
		if (filter->context.data && filter->info.filter_video) {
2251
			in = filter->info.filter_video(filter->context.data,
J
jp9000 已提交
2252
						       in);
2253
			if (!in)
2254
				break;
2255 2256 2257
		}
	}

2258 2259
	pthread_mutex_unlock(&source->filter_mutex);

2260 2261 2262
	return in;
}

2263
static inline void copy_frame_data_line(struct obs_source_frame *dst,
J
jp9000 已提交
2264 2265
					const struct obs_source_frame *src,
					uint32_t plane, uint32_t y)
2266
{
2267 2268
	uint32_t pos_src = y * src->linesize[plane];
	uint32_t pos_dst = y * dst->linesize[plane];
J
jp9000 已提交
2269 2270 2271
	uint32_t bytes = dst->linesize[plane] < src->linesize[plane]
				 ? dst->linesize[plane]
				 : src->linesize[plane];
2272 2273 2274 2275

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

2276
static inline void copy_frame_data_plane(struct obs_source_frame *dst,
J
jp9000 已提交
2277 2278
					 const struct obs_source_frame *src,
					 uint32_t plane, uint32_t lines)
2279
{
2280
	if (dst->linesize[plane] != src->linesize[plane])
2281 2282 2283 2284
		for (uint32_t y = 0; y < lines; y++)
			copy_frame_data_line(dst, src, plane, y);
	else
		memcpy(dst->data[plane], src->data[plane],
J
jp9000 已提交
2285
		       dst->linesize[plane] * lines);
2286 2287
}

2288
static void copy_frame_data(struct obs_source_frame *dst,
J
jp9000 已提交
2289
			    const struct obs_source_frame *src)
2290
{
J
jp9000 已提交
2291 2292 2293
	dst->flip = src->flip;
	dst->full_range = src->full_range;
	dst->timestamp = src->timestamp;
2294
	memcpy(dst->color_matrix, src->color_matrix, sizeof(float) * 16);
2295 2296 2297 2298 2299
	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);
	}
2300

J
jp9000 已提交
2301
	switch (src->format) {
2302 2303
	case VIDEO_FORMAT_I420:
		copy_frame_data_plane(dst, src, 0, dst->height);
J
jp9000 已提交
2304 2305
		copy_frame_data_plane(dst, src, 1, dst->height / 2);
		copy_frame_data_plane(dst, src, 2, dst->height / 2);
2306 2307 2308 2309
		break;

	case VIDEO_FORMAT_NV12:
		copy_frame_data_plane(dst, src, 0, dst->height);
J
jp9000 已提交
2310
		copy_frame_data_plane(dst, src, 1, dst->height / 2);
2311 2312
		break;

J
jp9000 已提交
2313 2314 2315 2316 2317 2318
	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;

2319 2320 2321 2322 2323 2324 2325
	case VIDEO_FORMAT_YVYU:
	case VIDEO_FORMAT_YUY2:
	case VIDEO_FORMAT_UYVY:
	case VIDEO_FORMAT_NONE:
	case VIDEO_FORMAT_RGBA:
	case VIDEO_FORMAT_BGRA:
	case VIDEO_FORMAT_BGRX:
2326
	case VIDEO_FORMAT_Y800:
2327
	case VIDEO_FORMAT_BGR3:
2328
		copy_frame_data_plane(dst, src, 0, dst->height);
2329
		break;
2330 2331 2332
	}
}

B
Bas van Meel 已提交
2333
void obs_source_frame_copy(struct obs_source_frame *dst,
J
jp9000 已提交
2334
			   const struct obs_source_frame *src)
B
Bas van Meel 已提交
2335 2336 2337 2338
{
	copy_frame_data(dst, src);
}

2339
static inline bool async_texture_changed(struct obs_source *source,
J
jp9000 已提交
2340
					 const struct obs_source_frame *frame)
2341 2342
{
	enum convert_type prev, cur;
2343
	prev = get_convert_type(source->async_cache_format,
J
jp9000 已提交
2344 2345
				source->async_cache_full_range);
	cur = get_convert_type(frame->format, frame->full_range);
2346

J
jp9000 已提交
2347 2348
	return source->async_cache_width != frame->width ||
	       source->async_cache_height != frame->height || prev != cur;
2349 2350 2351 2352 2353
}

static inline void free_async_cache(struct obs_source *source)
{
	for (size_t i = 0; i < source->async_cache.num; i++)
2354
		obs_source_frame_decref(source->async_cache.array[i].frame);
2355 2356 2357

	da_resize(source->async_cache, 0);
	da_resize(source->async_frames, 0);
2358
	source->cur_async_frame = NULL;
J
jp9000 已提交
2359
	source->prev_async_frame = NULL;
2360 2361
}

2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378
#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);
			}
		}
	}
}

2379
#define MAX_ASYNC_FRAMES 30
2380
//if return value is not null then do (os_atomic_dec_long(&output->refs) == 0) && obs_source_frame_destroy(output)
J
jp9000 已提交
2381 2382
static inline struct obs_source_frame *
cache_video(struct obs_source *source, const struct obs_source_frame *frame)
2383
{
2384 2385 2386 2387
	struct obs_source_frame *new_frame = NULL;

	pthread_mutex_lock(&source->async_mutex);

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

2395
	if (async_texture_changed(source, frame)) {
J
jp9000 已提交
2396
		free_async_cache(source);
J
jp9000 已提交
2397 2398 2399
		source->async_cache_width = frame->width;
		source->async_cache_height = frame->height;
		source->async_cache_format = frame->format;
2400
		source->async_cache_full_range = frame->full_range;
2401 2402 2403 2404 2405 2406 2407
	}

	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;
2408
			af->unused_count = 0;
2409
			break;
2410 2411 2412
		}
	}

2413 2414
	clean_cache(source);

2415 2416
	if (!new_frame) {
		struct async_frame new_af;
2417
		enum video_format format = frame->format;
J
jp9000 已提交
2418

J
jp9000 已提交
2419 2420
		new_frame = obs_source_frame_create(format, frame->width,
						    frame->height);
2421 2422
		new_af.frame = new_frame;
		new_af.used = true;
2423
		new_af.unused_count = 0;
2424
		new_frame->refs = 1;
2425 2426 2427 2428

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

2429 2430
	os_atomic_inc_long(&new_frame->refs);

2431
	pthread_mutex_unlock(&source->async_mutex);
2432

2433
	copy_frame_data(new_frame, frame);
2434

2435
	return new_frame;
2436 2437
}

J
jp9000 已提交
2438 2439 2440
static void
obs_source_output_video_internal(obs_source_t *source,
				 const struct obs_source_frame *frame)
2441
{
2442
	if (!obs_source_valid(source, "obs_source_output_video"))
J
jp9000 已提交
2443 2444
		return;

2445 2446 2447 2448 2449
	if (!frame) {
		source->async_active = false;
		return;
	}

J
jp9000 已提交
2450 2451
	struct obs_source_frame *output = !!frame ? cache_video(source, frame)
						  : NULL;
2452

2453
	/* ------------------------------------------- */
2454
	pthread_mutex_lock(&source->async_mutex);
2455
	if (output) {
2456 2457 2458 2459 2460 2461 2462
		if (os_atomic_dec_long(&output->refs) == 0) {
			obs_source_frame_destroy(output);
			output = NULL;
		} else {
			da_push_back(source->async_frames, &output);
			source->async_active = true;
		}
2463
	}
2464
	pthread_mutex_unlock(&source->async_mutex);
2465 2466
}

2467
void obs_source_output_video(obs_source_t *source,
J
jp9000 已提交
2468
			     const struct obs_source_frame *frame)
2469 2470 2471 2472 2473 2474 2475
{
	if (!frame) {
		obs_source_output_video_internal(source, NULL);
		return;
	}

	struct obs_source_frame new_frame = *frame;
J
jp9000 已提交
2476 2477
	new_frame.full_range =
		format_is_yuv(frame->format) ? new_frame.full_range : true;
2478 2479 2480 2481 2482

	obs_source_output_video_internal(source, &new_frame);
}

void obs_source_output_video2(obs_source_t *source,
J
jp9000 已提交
2483
			      const struct obs_source_frame2 *frame)
2484 2485 2486 2487 2488 2489 2490
{
	if (!frame) {
		obs_source_output_video_internal(source, NULL);
		return;
	}

	struct obs_source_frame new_frame;
J
jp9000 已提交
2491 2492
	enum video_range_type range =
		resolve_video_range(frame->format, frame->range);
2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506

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

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

	memcpy(&new_frame.color_matrix, &frame->color_matrix,
J
jp9000 已提交
2507
	       sizeof(frame->color_matrix));
2508
	memcpy(&new_frame.color_range_min, &frame->color_range_min,
J
jp9000 已提交
2509
	       sizeof(frame->color_range_min));
2510
	memcpy(&new_frame.color_range_max, &frame->color_range_max,
J
jp9000 已提交
2511
	       sizeof(frame->color_range_max));
2512 2513 2514 2515

	obs_source_output_video_internal(source, &new_frame);
}

2516
static inline bool preload_frame_changed(obs_source_t *source,
J
jp9000 已提交
2517
					 const struct obs_source_frame *in)
2518 2519 2520 2521
{
	if (!source->async_preload_frame)
		return true;

J
jp9000 已提交
2522
	return in->width != source->async_preload_frame->width ||
2523 2524 2525 2526
	       in->height != source->async_preload_frame->height ||
	       in->format != source->async_preload_frame->format;
}

J
jp9000 已提交
2527 2528 2529
static void
obs_source_preload_video_internal(obs_source_t *source,
				  const struct obs_source_frame *frame)
2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540
{
	if (!obs_source_valid(source, "obs_source_preload_video"))
		return;
	if (!frame)
		return;

	obs_enter_graphics();

	if (preload_frame_changed(source, frame)) {
		obs_source_frame_destroy(source->async_preload_frame);
		source->async_preload_frame = obs_source_frame_create(
J
jp9000 已提交
2541
			frame->format, frame->width, frame->height);
2542 2543 2544 2545 2546
	}

	copy_frame_data(source->async_preload_frame, frame);
	set_async_texture_size(source, source->async_preload_frame);
	update_async_texture(source, source->async_preload_frame,
J
jp9000 已提交
2547
			     source->async_texture, source->async_texrender);
2548 2549 2550 2551 2552 2553

	source->last_frame_ts = frame->timestamp;

	obs_leave_graphics();
}

2554
void obs_source_preload_video(obs_source_t *source,
J
jp9000 已提交
2555
			      const struct obs_source_frame *frame)
2556 2557 2558 2559 2560 2561 2562
{
	if (!frame) {
		obs_source_preload_video_internal(source, NULL);
		return;
	}

	struct obs_source_frame new_frame = *frame;
J
jp9000 已提交
2563 2564
	new_frame.full_range =
		format_is_yuv(frame->format) ? new_frame.full_range : true;
2565 2566 2567 2568 2569

	obs_source_preload_video_internal(source, &new_frame);
}

void obs_source_preload_video2(obs_source_t *source,
J
jp9000 已提交
2570
			       const struct obs_source_frame2 *frame)
2571 2572 2573 2574 2575 2576 2577
{
	if (!frame) {
		obs_source_preload_video_internal(source, NULL);
		return;
	}

	struct obs_source_frame new_frame;
J
jp9000 已提交
2578 2579
	enum video_range_type range =
		resolve_video_range(frame->format, frame->range);
2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593

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

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

	memcpy(&new_frame.color_matrix, &frame->color_matrix,
J
jp9000 已提交
2594
	       sizeof(frame->color_matrix));
2595
	memcpy(&new_frame.color_range_min, &frame->color_range_min,
J
jp9000 已提交
2596
	       sizeof(frame->color_range_min));
2597
	memcpy(&new_frame.color_range_max, &frame->color_range_max,
J
jp9000 已提交
2598
	       sizeof(frame->color_range_max));
2599 2600 2601 2602

	obs_source_preload_video_internal(source, &new_frame);
}

2603 2604 2605 2606 2607 2608 2609 2610 2611 2612
void obs_source_show_preloaded_video(obs_source_t *source)
{
	uint64_t sys_ts;

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

	source->async_active = true;

	pthread_mutex_lock(&source->audio_buf_mutex);
2613
	sys_ts = (source->monitoring_type != OBS_MONITORING_TYPE_MONITOR_ONLY)
J
jp9000 已提交
2614 2615
			 ? os_gettime_ns()
			 : 0;
2616 2617 2618 2619 2620
	reset_audio_timing(source, source->last_frame_ts, sys_ts);
	reset_audio_data(source, sys_ts);
	pthread_mutex_unlock(&source->audio_buf_mutex);
}

J
jp9000 已提交
2621 2622
static inline struct obs_audio_data *
filter_async_audio(obs_source_t *source, struct obs_audio_data *in)
2623 2624 2625
{
	size_t i;
	for (i = source->filters.num; i > 0; i--) {
J
jp9000 已提交
2626
		struct obs_source *filter = source->filters.array[i - 1];
2627

2628 2629 2630
		if (!filter->enabled)
			continue;

2631
		if (filter->context.data && filter->info.filter_audio) {
2632
			in = filter->info.filter_audio(filter->context.data,
J
jp9000 已提交
2633
						       in);
2634 2635 2636 2637 2638 2639 2640 2641
			if (!in)
				return NULL;
		}
	}

	return in;
}

2642
static inline void reset_resampler(obs_source_t *source,
J
jp9000 已提交
2643
				   const struct obs_source_audio *audio)
2644
{
J
jp9000 已提交
2645
	const struct audio_output_info *obs_info;
2646 2647
	struct resample_info output_info;

2648
	obs_info = audio_output_get_info(obs->audio.audio);
2649

J
jp9000 已提交
2650 2651 2652
	output_info.format = obs_info->format;
	output_info.samples_per_sec = obs_info->samples_per_sec;
	output_info.speakers = obs_info->speakers;
2653

J
jp9000 已提交
2654
	source->sample_info.format = audio->format;
2655
	source->sample_info.samples_per_sec = audio->samples_per_sec;
J
jp9000 已提交
2656
	source->sample_info.speakers = audio->speakers;
2657

2658 2659
	audio_resampler_destroy(source->resampler);
	source->resampler = NULL;
2660
	source->resample_offset = 0;
2661

2662
	if (source->sample_info.samples_per_sec == obs_info->samples_per_sec &&
J
jp9000 已提交
2663 2664
	    source->sample_info.format == obs_info->format &&
	    source->sample_info.speakers == obs_info->speakers) {
2665 2666 2667 2668
		source->audio_failed = false;
		return;
	}

J
jp9000 已提交
2669 2670
	source->resampler =
		audio_resampler_create(&output_info, &source->sample_info);
2671 2672 2673 2674 2675 2676

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

J
jp9000 已提交
2677 2678
static void copy_audio_data(obs_source_t *source, const uint8_t *const data[],
			    uint32_t frames, uint64_t ts)
2679
{
J
jp9000 已提交
2680
	size_t planes = audio_output_get_planes(obs->audio.audio);
2681
	size_t blocksize = audio_output_get_block_size(obs->audio.audio);
J
jp9000 已提交
2682 2683
	size_t size = (size_t)frames * blocksize;
	bool resize = source->audio_storage_size < size;
2684

J
jp9000 已提交
2685
	source->audio_data.frames = frames;
J
jp9000 已提交
2686
	source->audio_data.timestamp = ts;
2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699

	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;
2700 2701
}

2702 2703 2704
/* TODO: SSE optimization */
static void downmix_to_mono_planar(struct obs_source *source, uint32_t frames)
{
J
jp9000 已提交
2705
	size_t channels = audio_output_get_channels(obs->audio.audio);
2706
	const float channels_i = 1.0f / (float)channels;
J
jp9000 已提交
2707
	float **data = (float **)source->audio_data.data;
2708

J
jp9000 已提交
2709
	for (size_t channel = 1; channel < channels; channel++) {
2710 2711 2712 2713 2714 2715 2716
		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 已提交
2717
	for (size_t channel = 1; channel < channels; channel++) {
2718 2719 2720 2721 2722
		for (uint32_t frame = 0; frame < frames; frame++)
			data[channel][frame] = data[0][frame];
	}
}

C
cg2121 已提交
2723
static void process_audio_balancing(struct obs_source *source, uint32_t frames,
J
jp9000 已提交
2724
				    float balance, enum obs_balance_type type)
C
cg2121 已提交
2725
{
J
jp9000 已提交
2726
	float **data = (float **)source->audio_data.data;
C
cg2121 已提交
2727

J
jp9000 已提交
2728
	switch (type) {
C
cg2121 已提交
2729 2730 2731
	case OBS_BALANCE_TYPE_SINE_LAW:
		for (uint32_t frame = 0; frame < frames; frame++) {
			data[0][frame] = data[0][frame] *
J
jp9000 已提交
2732 2733 2734
					 sinf((1.0f - balance) * (M_PI / 2.0f));
			data[1][frame] =
				data[1][frame] * sinf(balance * (M_PI / 2.0f));
C
cg2121 已提交
2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753
		}
		break;
	case OBS_BALANCE_TYPE_SQUARE_LAW:
		for (uint32_t frame = 0; frame < frames; frame++) {
			data[0][frame] = data[0][frame] * sqrtf(1.0f - balance);
			data[1][frame] = data[1][frame] * sqrtf(balance);
		}
		break;
	case OBS_BALANCE_TYPE_LINEAR:
		for (uint32_t frame = 0; frame < frames; frame++) {
			data[0][frame] = data[0][frame] * (1.0f - balance);
			data[1][frame] = data[1][frame] * balance;
		}
		break;
	default:
		break;
	}
}

2754
/* resamples/remixes new audio to the designated main audio output format */
2755
static void process_audio(obs_source_t *source,
J
jp9000 已提交
2756
			  const struct obs_source_audio *audio)
2757
{
2758
	uint32_t frames = audio->frames;
2759
	bool mono_output;
2760

2761
	if (source->sample_info.samples_per_sec != audio->samples_per_sec ||
J
jp9000 已提交
2762 2763
	    source->sample_info.format != audio->format ||
	    source->sample_info.speakers != audio->speakers)
2764 2765 2766 2767 2768 2769
		reset_resampler(source, audio);

	if (source->audio_failed)
		return;

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

2772 2773
		memset(output, 0, sizeof(output));

J
jp9000 已提交
2774 2775 2776
		audio_resampler_resample(source->resampler, output, &frames,
					 &source->resample_offset, audio->data,
					 audio->frames);
2777

J
jp9000 已提交
2778
		copy_audio_data(source, (const uint8_t *const *)output, frames,
2779
				audio->timestamp);
2780 2781 2782 2783
	} else {
		copy_audio_data(source, audio->data, audio->frames,
				audio->timestamp);
	}
2784

2785 2786
	mono_output = audio_output_get_channels(obs->audio.audio) == 1;

2787 2788
	if (!mono_output && source->sample_info.speakers == SPEAKERS_STEREO &&
	    (source->balance > 0.51f || source->balance < 0.49f)) {
C
cg2121 已提交
2789
		process_audio_balancing(source, frames, source->balance,
J
jp9000 已提交
2790
					OBS_BALANCE_TYPE_SINE_LAW);
C
cg2121 已提交
2791 2792
	}

2793 2794
	if (!mono_output && (source->flags & OBS_SOURCE_FLAG_FORCE_MONO) != 0)
		downmix_to_mono_planar(source, frames);
2795 2796
}

2797
void obs_source_output_audio(obs_source_t *source,
J
jp9000 已提交
2798
			     const struct obs_source_audio *audio)
2799
{
2800
	struct obs_audio_data *output;
2801

2802 2803 2804
	if (!obs_source_valid(source, "obs_source_output_audio"))
		return;
	if (!obs_ptr_valid(audio, "obs_source_output_audio"))
J
jp9000 已提交
2805 2806
		return;

2807
	process_audio(source, audio);
2808 2809

	pthread_mutex_lock(&source->filter_mutex);
2810
	output = filter_async_audio(source, &source->audio_data);
2811 2812

	if (output) {
2813
		struct audio_data data;
J
jp9000 已提交
2814

2815 2816
		for (int i = 0; i < MAX_AV_PLANES; i++)
			data.data[i] = output->data[i];
2817

J
jp9000 已提交
2818
		data.frames = output->frames;
2819
		data.timestamp = output->timestamp;
2820

2821
		pthread_mutex_lock(&source->audio_mutex);
2822
		source_output_audio_data(source, &data);
2823 2824 2825 2826 2827 2828
		pthread_mutex_unlock(&source->audio_mutex);
	}

	pthread_mutex_unlock(&source->filter_mutex);
}

2829
void remove_async_frame(obs_source_t *source, struct obs_source_frame *frame)
2830
{
J
jp9000 已提交
2831 2832 2833
	if (frame)
		frame->prev_frame = false;

2834 2835 2836 2837 2838 2839 2840 2841 2842 2843
	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 已提交
2844 2845
/* #define DEBUG_ASYNC_FRAMES 1 */

2846
static bool ready_async_frame(obs_source_t *source, uint64_t sys_time)
2847
{
2848
	struct obs_source_frame *next_frame = source->async_frames.array[0];
J
jp9000 已提交
2849
	struct obs_source_frame *frame = NULL;
2850 2851 2852 2853
	uint64_t sys_offset = sys_time - source->last_sys_timestamp;
	uint64_t frame_time = next_frame->timestamp;
	uint64_t frame_offset = 0;

2854
	if (source->async_unbuffered) {
2855 2856
		while (source->async_frames.num > 1) {
			da_erase(source->async_frames, 0);
2857
			remove_async_frame(source, next_frame);
2858
			next_frame = source->async_frames.array[0];
J
jp9000 已提交
2859 2860
		}

2861
		source->last_frame_ts = next_frame->timestamp;
J
jp9000 已提交
2862 2863 2864
		return true;
	}

J
jp9000 已提交
2865
#if DEBUG_ASYNC_FRAMES
J
jp9000 已提交
2866 2867 2868 2869 2870 2871 2872
	blog(LOG_DEBUG,
	     "source->last_frame_ts: %llu, frame_time: %llu, "
	     "sys_offset: %llu, frame_offset: %llu, "
	     "number of frames: %lu",
	     source->last_frame_ts, frame_time, sys_offset,
	     frame_time - source->last_frame_ts,
	     (unsigned long)source->async_frames.num);
J
jp9000 已提交
2873 2874
#endif

2875 2876
	/* account for timestamp invalidation */
	if (frame_out_of_bounds(source, frame_time)) {
J
jp9000 已提交
2877 2878 2879
#if DEBUG_ASYNC_FRAMES
		blog(LOG_DEBUG, "timing jump");
#endif
2880
		source->last_frame_ts = next_frame->timestamp;
J
jp9000 已提交
2881
		return true;
2882 2883
	} else {
		frame_offset = frame_time - source->last_frame_ts;
J
jp9000 已提交
2884
		source->last_frame_ts += sys_offset;
2885 2886
	}

J
jp9000 已提交
2887 2888 2889 2890 2891 2892
	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 */
2893
		if ((source->last_frame_ts - next_frame->timestamp) < 2000000)
J
jp9000 已提交
2894 2895 2896
			break;

		if (frame)
2897
			da_erase(source->async_frames, 0);
J
jp9000 已提交
2898 2899

#if DEBUG_ASYNC_FRAMES
J
jp9000 已提交
2900 2901 2902 2903 2904
		blog(LOG_DEBUG,
		     "new frame, "
		     "source->last_frame_ts: %llu, "
		     "next_frame->timestamp: %llu",
		     source->last_frame_ts, next_frame->timestamp);
J
jp9000 已提交
2905 2906
#endif

2907
		remove_async_frame(source, frame);
2908

2909
		if (source->async_frames.num == 1)
2910 2911
			return true;

2912
		frame = next_frame;
2913
		next_frame = source->async_frames.array[1];
2914 2915

		/* more timestamp checking and compensating */
2916
		if ((next_frame->timestamp - frame_time) > MAX_TS_VAR) {
J
jp9000 已提交
2917 2918 2919
#if DEBUG_ASYNC_FRAMES
			blog(LOG_DEBUG, "timing jump");
#endif
2920 2921 2922 2923
			source->last_frame_ts =
				next_frame->timestamp - frame_offset;
		}

J
jp9000 已提交
2924
		frame_time = next_frame->timestamp;
2925 2926 2927
		frame_offset = frame_time - source->last_frame_ts;
	}

J
jp9000 已提交
2928 2929 2930 2931
#if DEBUG_ASYNC_FRAMES
	if (!frame)
		blog(LOG_DEBUG, "no frame!");
#endif
2932

2933 2934 2935
	return frame != NULL;
}

2936
static inline struct obs_source_frame *get_closest_frame(obs_source_t *source,
J
jp9000 已提交
2937
							 uint64_t sys_time)
2938
{
2939 2940 2941 2942
	if (!source->async_frames.num)
		return NULL;

	if (!source->last_frame_ts || ready_async_frame(source, sys_time)) {
2943 2944
		struct obs_source_frame *frame = source->async_frames.array[0];
		da_erase(source->async_frames, 0);
2945 2946 2947 2948

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

2949 2950 2951 2952
		return frame;
	}

	return NULL;
2953 2954
}

2955
/*
2956 2957
 * Ensures that cached frames are displayed on time.  If multiple frames
 * were cached between renders, then releases the unnecessary frames and uses
2958 2959
 * the frame with the closest timing to ensure sync.  Also ensures that timing
 * with audio is synchronized.
2960
 */
2961
struct obs_source_frame *obs_source_get_frame(obs_source_t *source)
J
jp9000 已提交
2962
{
2963
	struct obs_source_frame *frame = NULL;
2964

2965
	if (!obs_source_valid(source, "obs_source_get_frame"))
J
jp9000 已提交
2966 2967
		return NULL;

2968
	pthread_mutex_lock(&source->async_mutex);
2969

2970 2971
	frame = source->cur_async_frame;
	source->cur_async_frame = NULL;
J
jp9000 已提交
2972 2973

	if (frame) {
2974
		os_atomic_inc_long(&frame->refs);
2975 2976
	}

2977
	pthread_mutex_unlock(&source->async_mutex);
2978

2979
	return frame;
J
jp9000 已提交
2980 2981
}

2982
void obs_source_release_frame(obs_source_t *source,
J
jp9000 已提交
2983
			      struct obs_source_frame *frame)
J
jp9000 已提交
2984
{
2985
	if (!frame)
2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998
		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);
2999
	}
J
jp9000 已提交
3000
}
3001

3002
const char *obs_source_get_name(const obs_source_t *source)
3003
{
J
jp9000 已提交
3004 3005 3006
	return obs_source_valid(source, "obs_source_get_name")
		       ? source->context.name
		       : NULL;
3007 3008
}

3009
void obs_source_set_name(obs_source_t *source, const char *name)
3010
{
3011 3012
	if (!obs_source_valid(source, "obs_source_set_name"))
		return;
J
jp9000 已提交
3013

3014
	if (!name || !*name || !source->context.name ||
J
jp9000 已提交
3015
	    strcmp(name, source->context.name) != 0) {
J
jp9000 已提交
3016 3017 3018 3019 3020
		struct calldata data;
		char *prev_name = bstrdup(source->context.name);
		obs_context_data_setname(&source->context, name);

		calldata_init(&data);
3021 3022 3023
		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 已提交
3024 3025
		if (!source->context.private)
			signal_handler_signal(obs->signals, "source_rename",
J
jp9000 已提交
3026
					      &data);
J
jp9000 已提交
3027 3028 3029 3030
		signal_handler_signal(source->context.signals, "rename", &data);
		calldata_free(&data);
		bfree(prev_name);
	}
3031 3032
}

3033
enum obs_source_type obs_source_get_type(const obs_source_t *source)
3034
{
J
jp9000 已提交
3035 3036 3037
	return obs_source_valid(source, "obs_source_get_type")
		       ? source->info.type
		       : OBS_SOURCE_TYPE_INPUT;
J
jp9000 已提交
3038
}
J
jp9000 已提交
3039

3040
const char *obs_source_get_id(const obs_source_t *source)
J
jp9000 已提交
3041
{
J
jp9000 已提交
3042 3043
	return obs_source_valid(source, "obs_source_get_id") ? source->info.id
							     : NULL;
3044
}
3045

3046
static inline void render_filter_bypass(obs_source_t *target,
J
jp9000 已提交
3047 3048
					gs_effect_t *effect,
					const char *tech_name)
3049
{
J
jp9000 已提交
3050 3051
	gs_technique_t *tech = gs_effect_get_technique(effect, tech_name);
	size_t passes, i;
3052

3053
	passes = gs_technique_begin(tech);
3054
	for (i = 0; i < passes; i++) {
3055
		gs_technique_begin_pass(tech, i);
3056
		obs_source_video_render(target);
3057
		gs_technique_end_pass(tech);
3058
	}
3059
	gs_technique_end(tech);
3060 3061
}

3062
static inline void render_filter_tex(gs_texture_t *tex, gs_effect_t *effect,
J
jp9000 已提交
3063 3064
				     uint32_t width, uint32_t height,
				     const char *tech_name)
3065
{
J
jp9000 已提交
3066 3067 3068
	gs_technique_t *tech = gs_effect_get_technique(effect, tech_name);
	gs_eparam_t *image = gs_effect_get_param_by_name(effect, "image");
	size_t passes, i;
3069

3070
	gs_effect_set_texture(image, tex);
3071

3072
	passes = gs_technique_begin(tech);
3073
	for (i = 0; i < passes; i++) {
3074
		gs_technique_begin_pass(tech, i);
J
jp9000 已提交
3075
		gs_draw_sprite(tex, 0, width, height);
3076
		gs_technique_end_pass(tech);
3077
	}
3078
	gs_technique_end(tech);
3079 3080
}

3081
static inline bool can_bypass(obs_source_t *target, obs_source_t *parent,
J
jp9000 已提交
3082 3083
			      uint32_t parent_flags,
			      enum obs_allow_direct_render allow_direct)
3084 3085
{
	return (target == parent) &&
J
jp9000 已提交
3086 3087 3088
	       (allow_direct == OBS_ALLOW_DIRECT_RENDERING) &&
	       ((parent_flags & OBS_SOURCE_CUSTOM_DRAW) == 0) &&
	       ((parent_flags & OBS_SOURCE_ASYNC) == 0);
3089 3090
}

3091
bool obs_source_process_filter_begin(obs_source_t *filter,
J
jp9000 已提交
3092 3093
				     enum gs_color_format format,
				     enum obs_allow_direct_render allow_direct)
3094
{
3095
	obs_source_t *target, *parent;
J
jp9000 已提交
3096 3097
	uint32_t parent_flags;
	int cx, cy;
J
jp9000 已提交
3098

3099
	if (!obs_ptr_valid(filter, "obs_source_process_filter_begin"))
3100
		return false;
J
jp9000 已提交
3101

J
jp9000 已提交
3102 3103
	target = obs_filter_get_target(filter);
	parent = obs_filter_get_parent(filter);
J
jp9000 已提交
3104 3105 3106

	if (!target) {
		blog(LOG_INFO, "filter '%s' being processed with no target!",
J
jp9000 已提交
3107
		     filter->context.name);
3108
		return false;
J
jp9000 已提交
3109 3110 3111
	}
	if (!parent) {
		blog(LOG_INFO, "filter '%s' being processed with no parent!",
J
jp9000 已提交
3112
		     filter->context.name);
3113
		return false;
J
jp9000 已提交
3114 3115
	}

J
jp9000 已提交
3116
	parent_flags = parent->info.output_flags;
J
jp9000 已提交
3117 3118
	cx = get_base_width(target);
	cy = get_base_height(target);
3119

3120 3121
	filter->allow_direct = allow_direct;

3122 3123 3124 3125
	/* 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 */
3126
	if (can_bypass(target, parent, parent_flags, allow_direct)) {
3127
		return true;
3128 3129
	}

3130 3131
	if (!cx || !cy) {
		obs_source_skip_video_filter(filter);
3132
		return false;
3133 3134
	}

J
jp9000 已提交
3135
	if (!filter->filter_texrender)
J
jp9000 已提交
3136 3137
		filter->filter_texrender =
			gs_texrender_create(format, GS_ZS_NONE);
J
jp9000 已提交
3138

3139 3140 3141
	gs_blend_state_push();
	gs_blend_function(GS_BLEND_ONE, GS_BLEND_ZERO);

3142
	if (gs_texrender_begin(filter->filter_texrender, cx, cy)) {
3143 3144
		bool custom_draw = (parent_flags & OBS_SOURCE_CUSTOM_DRAW) != 0;
		bool async = (parent_flags & OBS_SOURCE_ASYNC) != 0;
3145 3146
		struct vec4 clear_color;

3147
		vec4_zero(&clear_color);
3148
		gs_clear(GS_CLEAR_COLOR, &clear_color, 0.0f, 0);
3149
		gs_ortho(0.0f, (float)cx, 0.0f, (float)cy, -100.0f, 100.0f);
3150

3151
		if (target == parent && !custom_draw && !async)
3152
			obs_source_default_render(target);
3153 3154
		else
			obs_source_video_render(target);
3155

3156
		gs_texrender_end(filter->filter_texrender);
3157
	}
3158 3159

	gs_blend_state_pop();
3160
	return true;
3161 3162
}

J
jp9000 已提交
3163 3164 3165
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)
3166 3167 3168
{
	obs_source_t *target, *parent;
	gs_texture_t *texture;
J
jp9000 已提交
3169
	uint32_t parent_flags;
3170

J
jp9000 已提交
3171 3172
	if (!filter)
		return;
3173

J
jp9000 已提交
3174 3175
	target = obs_filter_get_target(filter);
	parent = obs_filter_get_parent(filter);
J
jp9000 已提交
3176 3177 3178 3179

	if (!target || !parent)
		return;

3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191
	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);
	}
}

3192
void obs_source_process_filter_end(obs_source_t *filter, gs_effect_t *effect,
J
jp9000 已提交
3193
				   uint32_t width, uint32_t height)
3194 3195 3196
{
	obs_source_t *target, *parent;
	gs_texture_t *texture;
J
jp9000 已提交
3197
	uint32_t parent_flags;
3198

3199 3200
	if (!obs_ptr_valid(filter, "obs_source_process_filter_end"))
		return;
3201

J
jp9000 已提交
3202 3203
	target = obs_filter_get_target(filter);
	parent = obs_filter_get_parent(filter);
3204
	parent_flags = parent->info.output_flags;
3205

3206
	if (can_bypass(target, parent, parent_flags, filter->allow_direct)) {
3207
		render_filter_bypass(target, effect, "Draw");
3208 3209
	} else {
		texture = gs_texrender_get_texture(filter->filter_texrender);
3210
		if (texture)
3211
			render_filter_tex(texture, effect, width, height,
J
jp9000 已提交
3212
					  "Draw");
3213
	}
3214
}
3215

3216 3217 3218 3219 3220 3221
void obs_source_skip_video_filter(obs_source_t *filter)
{
	obs_source_t *target, *parent;
	bool custom_draw, async;
	uint32_t parent_flags;

3222 3223
	if (!obs_ptr_valid(filter, "obs_source_skip_video_filter"))
		return;
3224 3225 3226 3227 3228 3229 3230

	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 已提交
3231 3232
	if (target == parent) {
		if (!custom_draw && !async)
3233
			obs_source_default_render(target);
J
jp9000 已提交
3234 3235
		else if (target->info.video_render)
			obs_source_main_render(target);
J
jp9000 已提交
3236 3237
		else if (deinterlacing_enabled(target))
			deinterlace_render(target);
J
jp9000 已提交
3238 3239 3240 3241 3242 3243
		else
			obs_source_render_async_video(target);

	} else {
		obs_source_video_render(target);
	}
3244 3245
}

3246
signal_handler_t *obs_source_get_signal_handler(const obs_source_t *source)
3247
{
J
jp9000 已提交
3248 3249 3250
	return obs_source_valid(source, "obs_source_get_signal_handler")
		       ? source->context.signals
		       : NULL;
3251 3252
}

3253
proc_handler_t *obs_source_get_proc_handler(const obs_source_t *source)
3254
{
J
jp9000 已提交
3255 3256 3257
	return obs_source_valid(source, "obs_source_get_proc_handler")
		       ? source->context.procs
		       : NULL;
3258
}
J
jp9000 已提交
3259

3260
void obs_source_set_volume(obs_source_t *source, float volume)
J
jp9000 已提交
3261
{
3262
	if (obs_source_valid(source, "obs_source_set_volume")) {
J
jp9000 已提交
3263 3264 3265
		struct audio_action action = {.timestamp = os_gettime_ns(),
					      .type = AUDIO_ACTION_VOL,
					      .vol = volume};
J
jp9000 已提交
3266

3267 3268 3269 3270
		struct calldata data;
		uint8_t stack[128];

		calldata_init_fixed(&data, stack, sizeof(stack));
3271 3272
		calldata_set_ptr(&data, "source", source);
		calldata_set_float(&data, "volume", volume);
J
jp9000 已提交
3273

3274
		signal_handler_signal(source->context.signals, "volume", &data);
J
jp9000 已提交
3275 3276
		if (!source->context.private)
			signal_handler_signal(obs->signals, "source_volume",
J
jp9000 已提交
3277
					      &data);
J
jp9000 已提交
3278

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

J
jp9000 已提交
3281 3282 3283 3284
		pthread_mutex_lock(&source->audio_actions_mutex);
		da_push_back(source->audio_actions, &action);
		pthread_mutex_unlock(&source->audio_actions_mutex);

J
jp9000 已提交
3285
		source->user_volume = volume;
J
jp9000 已提交
3286
	}
J
jp9000 已提交
3287 3288
}

3289
float obs_source_get_volume(const obs_source_t *source)
J
jp9000 已提交
3290
{
J
jp9000 已提交
3291 3292 3293
	return obs_source_valid(source, "obs_source_get_volume")
		       ? source->user_volume
		       : 0.0f;
J
jp9000 已提交
3294 3295
}

3296
void obs_source_set_sync_offset(obs_source_t *source, int64_t offset)
J
jp9000 已提交
3297
{
3298
	if (obs_source_valid(source, "obs_source_set_sync_offset")) {
3299 3300
		struct calldata data;
		uint8_t stack[128];
J
jp9000 已提交
3301

3302
		calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
3303 3304 3305 3306
		calldata_set_ptr(&data, "source", source);
		calldata_set_int(&data, "offset", offset);

		signal_handler_signal(source->context.signals, "audio_sync",
J
jp9000 已提交
3307
				      &data);
J
jp9000 已提交
3308 3309 3310

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

3313
int64_t obs_source_get_sync_offset(const obs_source_t *source)
J
jp9000 已提交
3314
{
J
jp9000 已提交
3315 3316 3317
	return obs_source_valid(source, "obs_source_get_sync_offset")
		       ? source->sync_offset
		       : 0;
J
jp9000 已提交
3318
}
3319 3320 3321 3322 3323 3324

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

3325
static void enum_source_active_tree_callback(obs_source_t *parent,
J
jp9000 已提交
3326
					     obs_source_t *child, void *param)
3327 3328
{
	struct source_enum_data *data = param;
3329
	bool is_transition = child->info.type == OBS_SOURCE_TYPE_TRANSITION;
3330

3331
	if (is_transition)
J
jp9000 已提交
3332 3333
		obs_transition_enum_sources(
			child, enum_source_active_tree_callback, param);
3334
	if (child->info.enum_active_sources) {
J
jp9000 已提交
3335
		if (child->context.data) {
J
jp9000 已提交
3336 3337 3338
			child->info.enum_active_sources(
				child->context.data,
				enum_source_active_tree_callback, data);
J
jp9000 已提交
3339
		}
3340 3341 3342 3343 3344
	}

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

3345
void obs_source_enum_active_sources(obs_source_t *source,
J
jp9000 已提交
3346 3347
				    obs_source_enum_proc_t enum_callback,
				    void *param)
3348
{
J
jp9000 已提交
3349
	bool is_transition;
3350
	if (!data_valid(source, "obs_source_enum_active_sources"))
J
jp9000 已提交
3351
		return;
J
jp9000 已提交
3352 3353 3354

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

	obs_source_addref(source);

J
jp9000 已提交
3359 3360 3361 3362
	if (is_transition)
		obs_transition_enum_sources(source, enum_callback, param);
	if (source->info.enum_active_sources)
		source->info.enum_active_sources(source->context.data,
J
jp9000 已提交
3363
						 enum_callback, param);
3364 3365 3366 3367

	obs_source_release(source);
}

3368
void obs_source_enum_active_tree(obs_source_t *source,
J
jp9000 已提交
3369 3370
				 obs_source_enum_proc_t enum_callback,
				 void *param)
3371 3372
{
	struct source_enum_data data = {enum_callback, param};
J
jp9000 已提交
3373
	bool is_transition;
3374

3375
	if (!data_valid(source, "obs_source_enum_active_tree"))
J
jp9000 已提交
3376
		return;
J
jp9000 已提交
3377 3378 3379

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

	obs_source_addref(source);

J
jp9000 已提交
3384
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
J
jp9000 已提交
3385 3386
		obs_transition_enum_sources(
			source, enum_source_active_tree_callback, &data);
J
jp9000 已提交
3387
	if (source->info.enum_active_sources)
J
jp9000 已提交
3388 3389 3390
		source->info.enum_active_sources(
			source->context.data, enum_source_active_tree_callback,
			&data);
3391 3392 3393 3394 3395

	obs_source_release(source);
}

static void enum_source_full_tree_callback(obs_source_t *parent,
J
jp9000 已提交
3396
					   obs_source_t *child, void *param)
3397 3398 3399 3400 3401
{
	struct source_enum_data *data = param;
	bool is_transition = child->info.type == OBS_SOURCE_TYPE_TRANSITION;

	if (is_transition)
J
jp9000 已提交
3402 3403
		obs_transition_enum_sources(
			child, enum_source_full_tree_callback, param);
3404 3405
	if (child->info.enum_all_sources) {
		if (child->context.data) {
J
jp9000 已提交
3406 3407 3408
			child->info.enum_active_sources(
				child->context.data,
				enum_source_full_tree_callback, data);
3409 3410 3411
		}
	} else if (child->info.enum_active_sources) {
		if (child->context.data) {
J
jp9000 已提交
3412 3413 3414
			child->info.enum_active_sources(
				child->context.data,
				enum_source_full_tree_callback, data);
3415 3416 3417 3418 3419 3420 3421
		}
	}

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

static void obs_source_enum_full_tree(obs_source_t *source,
J
jp9000 已提交
3422 3423
				      obs_source_enum_proc_t enum_callback,
				      void *param)
3424 3425 3426 3427 3428 3429 3430 3431 3432 3433 3434 3435 3436 3437
{
	struct source_enum_data data = {enum_callback, param};
	bool is_transition;

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

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

	obs_source_addref(source);

	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
J
jp9000 已提交
3438 3439
		obs_transition_enum_sources(
			source, enum_source_full_tree_callback, &data);
3440 3441 3442

	if (source->info.enum_all_sources) {
		source->info.enum_all_sources(source->context.data,
J
jp9000 已提交
3443 3444
					      enum_source_full_tree_callback,
					      &data);
3445 3446 3447

	} else if (source->info.enum_active_sources) {
		source->info.enum_active_sources(source->context.data,
J
jp9000 已提交
3448 3449
						 enum_source_full_tree_callback,
						 &data);
3450
	}
3451 3452 3453

	obs_source_release(source);
}
3454

J
jp9000 已提交
3455 3456 3457 3458 3459 3460
struct descendant_info {
	bool exists;
	obs_source_t *target;
};

static void check_descendant(obs_source_t *parent, obs_source_t *child,
J
jp9000 已提交
3461
			     void *param)
3462
{
J
jp9000 已提交
3463 3464 3465 3466 3467
	struct descendant_info *info = param;
	if (child == info->target || parent == info->target)
		info->exists = true;
}

3468
bool obs_source_add_active_child(obs_source_t *parent, obs_source_t *child)
J
jp9000 已提交
3469
{
3470
	struct descendant_info info = {false, parent};
3471

3472
	if (!obs_ptr_valid(parent, "obs_source_add_active_child"))
3473
		return false;
3474
	if (!obs_ptr_valid(child, "obs_source_add_active_child"))
3475 3476
		return false;
	if (parent == child) {
3477
		blog(LOG_WARNING, "obs_source_add_active_child: "
J
jp9000 已提交
3478
				  "parent == child");
3479 3480
		return false;
	}
J
jp9000 已提交
3481

3482
	obs_source_enum_full_tree(child, check_descendant, &info);
J
jp9000 已提交
3483 3484
	if (info.exists)
		return false;
3485

3486 3487 3488 3489 3490
	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 已提交
3491 3492

	return true;
3493 3494
}

3495
void obs_source_remove_active_child(obs_source_t *parent, obs_source_t *child)
3496
{
3497
	if (!obs_ptr_valid(parent, "obs_source_remove_active_child"))
3498
		return;
3499
	if (!obs_ptr_valid(child, "obs_source_remove_active_child"))
3500
		return;
3501

3502 3503 3504 3505 3506
	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);
	}
3507
}
J
jp9000 已提交
3508

3509
void obs_source_save(obs_source_t *source)
3510
{
J
jp9000 已提交
3511 3512 3513
	if (!data_valid(source, "obs_source_save"))
		return;

J
jp9000 已提交
3514 3515 3516 3517
	obs_source_dosignal(source, "source_save", "save");

	if (source->info.save)
		source->info.save(source->context.data,
J
jp9000 已提交
3518
				  source->context.settings);
3519 3520
}

3521
void obs_source_load(obs_source_t *source)
3522
{
J
jp9000 已提交
3523 3524
	if (!data_valid(source, "obs_source_load"))
		return;
J
jp9000 已提交
3525 3526
	if (source->info.load)
		source->info.load(source->context.data,
J
jp9000 已提交
3527
				  source->context.settings);
J
jp9000 已提交
3528

J
jp9000 已提交
3529
	obs_source_dosignal(source, "source_load", "load");
3530
}
J
jp9000 已提交
3531

J
jp9000 已提交
3532 3533
bool obs_source_active(const obs_source_t *source)
{
J
jp9000 已提交
3534 3535 3536
	return obs_source_valid(source, "obs_source_active")
		       ? source->activate_refs != 0
		       : false;
J
jp9000 已提交
3537 3538
}

J
jp9000 已提交
3539 3540
bool obs_source_showing(const obs_source_t *source)
{
J
jp9000 已提交
3541 3542 3543
	return obs_source_valid(source, "obs_source_showing")
		       ? source->show_refs != 0
		       : false;
J
jp9000 已提交
3544 3545
}

J
jp9000 已提交
3546 3547
static inline void signal_flags_updated(obs_source_t *source)
{
3548 3549
	struct calldata data;
	uint8_t stack[128];
J
jp9000 已提交
3550

3551
	calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
3552 3553 3554 3555 3556 3557 3558 3559
	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)
{
3560 3561
	if (!obs_source_valid(source, "obs_source_set_flags"))
		return;
J
jp9000 已提交
3562 3563 3564 3565 3566 3567 3568

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

3569 3570
void obs_source_set_default_flags(obs_source_t *source, uint32_t flags)
{
3571 3572
	if (!obs_source_valid(source, "obs_source_set_default_flags"))
		return;
3573 3574 3575 3576

	source->default_flags = flags;
}

J
jp9000 已提交
3577 3578
uint32_t obs_source_get_flags(const obs_source_t *source)
{
J
jp9000 已提交
3579 3580
	return obs_source_valid(source, "obs_source_get_flags") ? source->flags
								: 0;
J
jp9000 已提交
3581
}
J
jp9000 已提交
3582

3583 3584
void obs_source_set_audio_mixers(obs_source_t *source, uint32_t mixers)
{
3585 3586
	struct calldata data;
	uint8_t stack[128];
3587

3588 3589 3590 3591
	if (!obs_source_valid(source, "obs_source_set_audio_mixers"))
		return;
	if ((source->info.output_flags & OBS_SOURCE_AUDIO) == 0)
		return;
3592

3593
	if (source->audio_mixers == mixers)
3594 3595
		return;

3596
	calldata_init_fixed(&data, stack, sizeof(stack));
3597 3598 3599 3600 3601 3602 3603
	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");

3604
	source->audio_mixers = mixers;
3605 3606 3607 3608
}

uint32_t obs_source_get_audio_mixers(const obs_source_t *source)
{
3609 3610 3611 3612
	if (!obs_source_valid(source, "obs_source_get_audio_mixers"))
		return 0;
	if ((source->info.output_flags & OBS_SOURCE_AUDIO) == 0)
		return 0;
3613

3614
	return source->audio_mixers;
3615 3616
}

J
jp9000 已提交
3617
void obs_source_draw_set_color_matrix(const struct matrix4 *color_matrix,
J
jp9000 已提交
3618 3619
				      const struct vec3 *color_range_min,
				      const struct vec3 *color_range_max)
J
jp9000 已提交
3620
{
3621 3622 3623 3624 3625 3626
	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 已提交
3627 3628 3629 3630 3631 3632
	gs_effect_t *effect = gs_get_effect();
	gs_eparam_t *matrix;
	gs_eparam_t *range_min;
	gs_eparam_t *range_max;

	if (!effect) {
3633
		blog(LOG_WARNING, "obs_source_draw_set_color_matrix: no "
J
jp9000 已提交
3634
				  "active effect!");
J
jp9000 已提交
3635 3636 3637
		return;
	}

3638
	if (!obs_ptr_valid(color_matrix, "obs_source_draw_set_color_matrix"))
J
jp9000 已提交
3639 3640 3641 3642 3643 3644 3645 3646 3647 3648 3649 3650
		return;

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

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

	gs_effect_set_matrix4(matrix, color_matrix);
J
jp9000 已提交
3651 3652
	gs_effect_set_val(range_min, color_range_min, sizeof(float) * 3);
	gs_effect_set_val(range_max, color_range_max, sizeof(float) * 3);
J
jp9000 已提交
3653 3654 3655
}

void obs_source_draw(gs_texture_t *texture, int x, int y, uint32_t cx,
J
jp9000 已提交
3656
		     uint32_t cy, bool flip)
J
jp9000 已提交
3657 3658 3659 3660 3661 3662
{
	gs_effect_t *effect = gs_get_effect();
	bool change_pos = (x != 0 || y != 0);
	gs_eparam_t *image;

	if (!effect) {
3663
		blog(LOG_WARNING, "obs_source_draw: no active effect!");
J
jp9000 已提交
3664 3665 3666
		return;
	}

3667
	if (!obs_ptr_valid(texture, "obs_source_draw"))
J
jp9000 已提交
3668 3669 3670 3671 3672 3673 3674 3675 3676 3677 3678 3679 3680 3681 3682
		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();
}
3683

3684 3685
void obs_source_inc_showing(obs_source_t *source)
{
3686 3687
	if (obs_source_valid(source, "obs_source_inc_showing"))
		obs_source_activate(source, AUX_VIEW);
3688 3689
}

3690 3691 3692 3693 3694 3695
void obs_source_inc_active(obs_source_t *source)
{
	if (obs_source_valid(source, "obs_source_inc_active"))
		obs_source_activate(source, MAIN_VIEW);
}

3696 3697
void obs_source_dec_showing(obs_source_t *source)
{
3698 3699
	if (obs_source_valid(source, "obs_source_dec_showing"))
		obs_source_deactivate(source, AUX_VIEW);
3700
}
3701

3702 3703 3704 3705 3706 3707
void obs_source_dec_active(obs_source_t *source)
{
	if (obs_source_valid(source, "obs_source_dec_active"))
		obs_source_deactivate(source, MAIN_VIEW);
}

3708
void obs_source_enum_filters(obs_source_t *source,
J
jp9000 已提交
3709
			     obs_source_enum_proc_t callback, void *param)
3710
{
3711 3712 3713
	if (!obs_source_valid(source, "obs_source_enum_filters"))
		return;
	if (!obs_ptr_valid(callback, "obs_source_enum_filters"))
3714 3715 3716 3717
		return;

	pthread_mutex_lock(&source->filter_mutex);

3718 3719
	for (size_t i = source->filters.num; i > 0; i--) {
		struct obs_source *filter = source->filters.array[i - 1];
3720 3721 3722 3723 3724
		callback(source, filter, param);
	}

	pthread_mutex_unlock(&source->filter_mutex);
}
3725 3726

obs_source_t *obs_source_get_filter_by_name(obs_source_t *source,
J
jp9000 已提交
3727
					    const char *name)
3728 3729 3730
{
	obs_source_t *filter = NULL;

3731 3732 3733
	if (!obs_source_valid(source, "obs_source_get_filter_by_name"))
		return NULL;
	if (!obs_ptr_valid(name, "obs_source_get_filter_by_name"))
3734 3735 3736 3737 3738 3739 3740 3741 3742 3743 3744 3745 3746 3747 3748 3749 3750
		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;
}
3751 3752 3753

bool obs_source_enabled(const obs_source_t *source)
{
J
jp9000 已提交
3754 3755
	return obs_source_valid(source, "obs_source_enabled") ? source->enabled
							      : false;
3756 3757 3758 3759
}

void obs_source_set_enabled(obs_source_t *source, bool enabled)
{
3760 3761
	struct calldata data;
	uint8_t stack[128];
3762

3763
	if (!obs_source_valid(source, "obs_source_set_enabled"))
3764 3765 3766 3767
		return;

	source->enabled = enabled;

3768
	calldata_init_fixed(&data, stack, sizeof(stack));
3769 3770 3771 3772 3773
	calldata_set_ptr(&data, "source", source);
	calldata_set_bool(&data, "enabled", enabled);

	signal_handler_signal(source->context.signals, "enable", &data);
}
J
jp9000 已提交
3774 3775 3776

bool obs_source_muted(const obs_source_t *source)
{
J
jp9000 已提交
3777 3778
	return obs_source_valid(source, "obs_source_muted") ? source->user_muted
							    : false;
J
jp9000 已提交
3779 3780 3781 3782
}

void obs_source_set_muted(obs_source_t *source, bool muted)
{
3783 3784
	struct calldata data;
	uint8_t stack[128];
J
jp9000 已提交
3785 3786 3787
	struct audio_action action = {.timestamp = os_gettime_ns(),
				      .type = AUDIO_ACTION_MUTE,
				      .set = muted};
J
jp9000 已提交
3788

3789
	if (!obs_source_valid(source, "obs_source_set_muted"))
J
jp9000 已提交
3790 3791
		return;

J
jp9000 已提交
3792
	source->user_muted = muted;
J
jp9000 已提交
3793

3794
	calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
3795 3796 3797 3798 3799
	calldata_set_ptr(&data, "source", source);
	calldata_set_bool(&data, "muted", muted);

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

J
jp9000 已提交
3800 3801 3802
	pthread_mutex_lock(&source->audio_actions_mutex);
	da_push_back(source->audio_actions, &action);
	pthread_mutex_unlock(&source->audio_actions_mutex);
J
jp9000 已提交
3803
}
P
Palana 已提交
3804 3805

static void source_signal_push_to_changed(obs_source_t *source,
J
jp9000 已提交
3806
					  const char *signal, bool enabled)
P
Palana 已提交
3807
{
3808 3809
	struct calldata data;
	uint8_t stack[128];
P
Palana 已提交
3810

3811
	calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
3812
	calldata_set_ptr(&data, "source", source);
P
Palana 已提交
3813 3814 3815 3816 3817 3818
	calldata_set_bool(&data, "enabled", enabled);

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

static void source_signal_push_to_delay(obs_source_t *source,
J
jp9000 已提交
3819
					const char *signal, uint64_t delay)
P
Palana 已提交
3820
{
3821 3822
	struct calldata data;
	uint8_t stack[128];
P
Palana 已提交
3823

3824
	calldata_init_fixed(&data, stack, sizeof(stack));
J
jp9000 已提交
3825 3826
	calldata_set_ptr(&data, "source", source);
	calldata_set_bool(&data, "delay", delay);
P
Palana 已提交
3827 3828 3829 3830 3831 3832 3833

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

bool obs_source_push_to_mute_enabled(obs_source_t *source)
{
	bool enabled;
3834 3835
	if (!obs_source_valid(source, "obs_source_push_to_mute_enabled"))
		return false;
P
Palana 已提交
3836 3837 3838 3839 3840 3841 3842 3843 3844 3845

	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)
{
3846 3847
	if (!obs_source_valid(source, "obs_source_enable_push_to_mute"))
		return;
P
Palana 已提交
3848 3849 3850 3851 3852

	pthread_mutex_lock(&source->audio_mutex);
	bool changed = source->push_to_mute_enabled != enabled;
	if (obs_source_get_output_flags(source) & OBS_SOURCE_AUDIO && changed)
		blog(LOG_INFO, "source '%s' %s push-to-mute",
J
jp9000 已提交
3853 3854
		     obs_source_get_name(source),
		     enabled ? "enabled" : "disabled");
P
Palana 已提交
3855 3856 3857 3858 3859

	source->push_to_mute_enabled = enabled;

	if (changed)
		source_signal_push_to_changed(source, "push_to_mute_changed",
J
jp9000 已提交
3860
					      enabled);
P
Palana 已提交
3861 3862 3863 3864 3865 3866
	pthread_mutex_unlock(&source->audio_mutex);
}

uint64_t obs_source_get_push_to_mute_delay(obs_source_t *source)
{
	uint64_t delay;
3867 3868
	if (!obs_source_valid(source, "obs_source_get_push_to_mute_delay"))
		return 0;
P
Palana 已提交
3869 3870 3871 3872 3873 3874 3875 3876 3877 3878

	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)
{
3879 3880
	if (!obs_source_valid(source, "obs_source_set_push_to_mute_delay"))
		return;
P
Palana 已提交
3881 3882 3883 3884 3885 3886 3887 3888 3889 3890 3891

	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;
3892 3893
	if (!obs_source_valid(source, "obs_source_push_to_talk_enabled"))
		return false;
P
Palana 已提交
3894 3895 3896 3897 3898 3899 3900 3901 3902 3903

	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)
{
3904 3905
	if (!obs_source_valid(source, "obs_source_enable_push_to_talk"))
		return;
P
Palana 已提交
3906 3907 3908 3909 3910

	pthread_mutex_lock(&source->audio_mutex);
	bool changed = source->push_to_talk_enabled != enabled;
	if (obs_source_get_output_flags(source) & OBS_SOURCE_AUDIO && changed)
		blog(LOG_INFO, "source '%s' %s push-to-talk",
J
jp9000 已提交
3911 3912
		     obs_source_get_name(source),
		     enabled ? "enabled" : "disabled");
P
Palana 已提交
3913 3914 3915 3916 3917

	source->push_to_talk_enabled = enabled;

	if (changed)
		source_signal_push_to_changed(source, "push_to_talk_changed",
J
jp9000 已提交
3918
					      enabled);
P
Palana 已提交
3919 3920 3921 3922 3923 3924
	pthread_mutex_unlock(&source->audio_mutex);
}

uint64_t obs_source_get_push_to_talk_delay(obs_source_t *source)
{
	uint64_t delay;
3925 3926
	if (!obs_source_valid(source, "obs_source_get_push_to_talk_delay"))
		return 0;
P
Palana 已提交
3927 3928 3929 3930 3931 3932 3933 3934 3935 3936

	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)
{
3937 3938
	if (!obs_source_valid(source, "obs_source_set_push_to_talk_delay"))
		return;
P
Palana 已提交
3939 3940 3941 3942 3943 3944 3945

	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);
}
3946 3947 3948 3949

void *obs_source_get_type_data(obs_source_t *source)
{
	return obs_source_valid(source, "obs_source_get_type_data")
J
jp9000 已提交
3950 3951
		       ? source->info.type_data
		       : NULL;
3952
}
3953

J
jp9000 已提交
3954 3955 3956
static float get_source_volume(obs_source_t *source, uint64_t os_time)
{
	if (source->push_to_mute_enabled && source->push_to_mute_pressed)
J
jp9000 已提交
3957 3958
		source->push_to_mute_stop_time =
			os_time + source->push_to_mute_delay * 1000000;
J
jp9000 已提交
3959 3960

	if (source->push_to_talk_enabled && source->push_to_talk_pressed)
J
jp9000 已提交
3961 3962
		source->push_to_talk_stop_time =
			os_time + source->push_to_talk_delay * 1000000;
J
jp9000 已提交
3963 3964

	bool push_to_mute_active = source->push_to_mute_pressed ||
J
jp9000 已提交
3965
				   os_time < source->push_to_mute_stop_time;
J
jp9000 已提交
3966
	bool push_to_talk_active = source->push_to_talk_pressed ||
J
jp9000 已提交
3967
				   os_time < source->push_to_talk_stop_time;
J
jp9000 已提交
3968 3969

	bool muted = !source->enabled || source->muted ||
J
jp9000 已提交
3970 3971
		     (source->push_to_mute_enabled && push_to_mute_active) ||
		     (source->push_to_talk_enabled && !push_to_talk_active);
J
jp9000 已提交
3972 3973 3974 3975 3976 3977 3978 3979 3980 3981

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

	return source->volume;
}

static inline void multiply_output_audio(obs_source_t *source, size_t mix,
J
jp9000 已提交
3982
					 size_t channels, float vol)
J
jp9000 已提交
3983 3984 3985 3986 3987 3988 3989 3990 3991
{
	register float *out = source->audio_output_buf[mix][0];
	register float *end = out + AUDIO_OUTPUT_FRAMES * channels;

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

static inline void multiply_vol_data(obs_source_t *source, size_t mix,
J
jp9000 已提交
3992
				     size_t channels, float *vol_data)
J
jp9000 已提交
3993 3994 3995 3996 3997 3998 3999 4000 4001 4002 4003 4004
{
	for (size_t ch = 0; ch < channels; ch++) {
		register float *out = source->audio_output_buf[mix][ch];
		register float *end = out + AUDIO_OUTPUT_FRAMES;
		register float *vol = vol_data;

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

static inline void apply_audio_action(obs_source_t *source,
J
jp9000 已提交
4005
				      const struct audio_action *action)
J
jp9000 已提交
4006 4007 4008
{
	switch (action->type) {
	case AUDIO_ACTION_VOL:
J
jp9000 已提交
4009 4010
		source->volume = action->vol;
		break;
J
jp9000 已提交
4011
	case AUDIO_ACTION_MUTE:
J
jp9000 已提交
4012 4013
		source->muted = action->set;
		break;
J
jp9000 已提交
4014
	case AUDIO_ACTION_PTT:
J
jp9000 已提交
4015 4016
		source->push_to_talk_pressed = action->set;
		break;
J
jp9000 已提交
4017
	case AUDIO_ACTION_PTM:
J
jp9000 已提交
4018 4019
		source->push_to_mute_pressed = action->set;
		break;
J
jp9000 已提交
4020 4021 4022 4023
	}
}

static void apply_audio_actions(obs_source_t *source, size_t channels,
J
jp9000 已提交
4024
				size_t sample_rate)
J
jp9000 已提交
4025 4026 4027 4028 4029 4030 4031 4032 4033 4034 4035 4036 4037 4038 4039
{
	float *vol_data = malloc(sizeof(float) * AUDIO_OUTPUT_FRAMES);
	float cur_vol = get_source_volume(source, source->audio_ts);
	size_t frame_num = 0;

	pthread_mutex_lock(&source->audio_actions_mutex);

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

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

J
jp9000 已提交
4040 4041
		new_frame_num = conv_time_to_frames(
			sample_rate, timestamp - source->audio_ts);
J
jp9000 已提交
4042 4043 4044 4045 4046 4047 4048 4049 4050 4051 4052 4053 4054 4055 4056 4057 4058 4059 4060 4061 4062 4063 4064 4065 4066 4067 4068 4069 4070 4071

		if (new_frame_num >= AUDIO_OUTPUT_FRAMES)
			break;

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

		apply_audio_action(source, &action);

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

		cur_vol = get_source_volume(source, timestamp);
	}

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

	pthread_mutex_unlock(&source->audio_actions_mutex);

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

	free(vol_data);
}

static void apply_audio_volume(obs_source_t *source, uint32_t mixers,
J
jp9000 已提交
4072
			       size_t channels, size_t sample_rate)
J
jp9000 已提交
4073 4074 4075 4076 4077 4078 4079 4080 4081 4082 4083 4084 4085 4086
{
	struct audio_action action;
	bool actions_pending;
	float vol;

	pthread_mutex_lock(&source->audio_actions_mutex);

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

	pthread_mutex_unlock(&source->audio_actions_mutex);

	if (actions_pending) {
J
jp9000 已提交
4087 4088
		uint64_t duration =
			conv_frames_to_time(sample_rate, AUDIO_OUTPUT_FRAMES);
J
jp9000 已提交
4089 4090 4091 4092 4093 4094 4095 4096 4097 4098 4099 4100 4101

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

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

	if (vol == 0.0f || mixers == 0) {
		memset(source->audio_output_buf[0][0], 0,
J
jp9000 已提交
4102 4103
		       AUDIO_OUTPUT_FRAMES * sizeof(float) *
			       MAX_AUDIO_CHANNELS * MAX_AUDIO_MIXES);
J
jp9000 已提交
4104 4105 4106 4107 4108 4109 4110 4111 4112 4113 4114 4115
		return;
	}

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

static void custom_audio_render(obs_source_t *source, uint32_t mixers,
J
jp9000 已提交
4116
				size_t channels, size_t sample_rate)
J
jp9000 已提交
4117 4118 4119 4120 4121 4122
{
	struct obs_source_audio_mix audio_data;
	bool success;
	uint64_t ts;

	for (size_t mix = 0; mix < MAX_AUDIO_MIXES; mix++) {
4123
		for (size_t ch = 0; ch < channels; ch++) {
J
jp9000 已提交
4124 4125
			audio_data.output[mix].data[ch] =
				source->audio_output_buf[mix][ch];
4126
		}
J
jp9000 已提交
4127

4128 4129
		if ((source->audio_mixers & mixers & (1 << mix)) != 0) {
			memset(source->audio_output_buf[mix][0], 0,
J
jp9000 已提交
4130
			       sizeof(float) * AUDIO_OUTPUT_FRAMES * channels);
4131 4132
		}
	}
J
jp9000 已提交
4133 4134

	success = source->info.audio_render(source->context.data, &ts,
J
jp9000 已提交
4135 4136
					    &audio_data, mixers, channels,
					    sample_rate);
J
jp9000 已提交
4137 4138 4139 4140 4141 4142 4143
	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++) {
4144 4145 4146 4147 4148 4149
		uint32_t mix_bit = 1 << mix;

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

		if ((source->audio_mixers & mix_bit) == 0) {
J
jp9000 已提交
4150
			memset(source->audio_output_buf[mix][0], 0,
J
jp9000 已提交
4151
			       sizeof(float) * AUDIO_OUTPUT_FRAMES * channels);
J
jp9000 已提交
4152 4153 4154 4155 4156 4157 4158
		}
	}

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

static inline void process_audio_source_tick(obs_source_t *source,
J
jp9000 已提交
4159 4160
					     uint32_t mixers, size_t channels,
					     size_t sample_rate, size_t size)
J
jp9000 已提交
4161 4162 4163
{
	pthread_mutex_lock(&source->audio_buf_mutex);

4164 4165 4166 4167 4168 4169
	if (source->audio_input_buf[0].size < size) {
		source->audio_pending = true;
		pthread_mutex_unlock(&source->audio_buf_mutex);
		return;
	}

J
jp9000 已提交
4170 4171
	for (size_t ch = 0; ch < channels; ch++)
		circlebuf_peek_front(&source->audio_input_buf[ch],
J
jp9000 已提交
4172
				     source->audio_output_buf[0][ch], size);
J
jp9000 已提交
4173 4174 4175 4176 4177 4178 4179 4180

	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) {
J
jp9000 已提交
4181 4182
			memset(source->audio_output_buf[mix][0], 0,
			       size * channels);
J
jp9000 已提交
4183 4184 4185 4186 4187
			continue;
		}

		for (size_t ch = 0; ch < channels; ch++)
			memcpy(source->audio_output_buf[mix][ch],
J
jp9000 已提交
4188
			       source->audio_output_buf[0][ch], size);
J
jp9000 已提交
4189 4190 4191
	}

	if ((source->audio_mixers & 1) == 0 || (mixers & 1) == 0)
J
jp9000 已提交
4192
		memset(source->audio_output_buf[0][0], 0, size * channels);
J
jp9000 已提交
4193 4194 4195 4196 4197 4198

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

void obs_source_audio_render(obs_source_t *source, uint32_t mixers,
J
jp9000 已提交
4199
			     size_t channels, size_t sample_rate, size_t size)
J
jp9000 已提交
4200
{
4201
	if (!source->audio_output_buf[0][0]) {
J
jp9000 已提交
4202 4203 4204 4205 4206 4207 4208 4209 4210
		source->audio_pending = true;
		return;
	}

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

4211
	if (!source->audio_ts) {
J
jp9000 已提交
4212 4213 4214 4215 4216 4217 4218 4219 4220
		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)
{
4221 4222 4223
	if (!obs_source_valid(source, "obs_source_audio_pending"))
		return true;

J
jp9000 已提交
4224 4225 4226
	return (is_composite_source(source) || is_audio_source(source))
		       ? source->audio_pending
		       : true;
J
jp9000 已提交
4227 4228
}

4229 4230
uint64_t obs_source_get_audio_timestamp(const obs_source_t *source)
{
J
jp9000 已提交
4231 4232 4233
	return obs_source_valid(source, "obs_source_get_audio_timestamp")
		       ? source->audio_ts
		       : 0;
4234 4235 4236
}

void obs_source_get_audio_mix(const obs_source_t *source,
J
jp9000 已提交
4237
			      struct obs_source_audio_mix *audio)
4238 4239 4240 4241 4242 4243 4244 4245 4246 4247 4248 4249 4250
{
	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];
		}
	}
}
4251 4252

void obs_source_add_audio_capture_callback(obs_source_t *source,
J
jp9000 已提交
4253 4254
					   obs_source_audio_capture_t callback,
					   void *param)
4255 4256 4257 4258 4259 4260 4261 4262 4263 4264 4265
{
	struct audio_cb_info info = {callback, param};

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

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

J
jp9000 已提交
4266 4267
void obs_source_remove_audio_capture_callback(
	obs_source_t *source, obs_source_audio_capture_t callback, void *param)
4268 4269 4270
{
	struct audio_cb_info info = {callback, param};

J
jp9000 已提交
4271 4272
	if (!obs_source_valid(source,
			      "obs_source_remove_audio_capture_callback"))
4273 4274 4275 4276 4277 4278
		return;

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

void obs_source_set_monitoring_type(obs_source_t *source,
J
jp9000 已提交
4281
				    enum obs_monitoring_type type)
J
jp9000 已提交
4282 4283 4284 4285 4286 4287 4288 4289 4290 4291 4292 4293 4294 4295 4296 4297 4298 4299 4300 4301 4302 4303 4304 4305
{
	bool was_on;
	bool now_on;

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

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

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

	source->monitoring_type = type;
}

J
jp9000 已提交
4306 4307
enum obs_monitoring_type
obs_source_get_monitoring_type(const obs_source_t *source)
J
jp9000 已提交
4308
{
J
jp9000 已提交
4309 4310 4311
	return obs_source_valid(source, "obs_source_get_monitoring_type")
		       ? source->monitoring_type
		       : OBS_MONITORING_TYPE_NONE;
J
jp9000 已提交
4312
}
4313 4314 4315 4316 4317 4318 4319 4320 4321 4322 4323

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

	source->async_unbuffered = unbuffered;
}

bool obs_source_async_unbuffered(const obs_source_t *source)
{
J
jp9000 已提交
4324 4325 4326
	return obs_source_valid(source, "obs_source_async_unbuffered")
		       ? source->async_unbuffered
		       : false;
4327
}
4328 4329 4330 4331 4332 4333 4334 4335 4336

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

	obs_data_addref(source->private_settings);
	return source->private_settings;
}
4337 4338 4339 4340 4341 4342 4343 4344 4345 4346 4347 4348 4349 4350 4351 4352 4353

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

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

bool obs_source_async_decoupled(const obs_source_t *source)
{
J
jp9000 已提交
4354 4355 4356
	return obs_source_valid(source, "obs_source_async_decoupled")
		       ? source->async_decoupled
		       : false;
4357
}
4358 4359 4360 4361 4362 4363 4364 4365 4366 4367 4368 4369 4370

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

	if (enable)
		info->output_flags &= ~OBS_SOURCE_CAP_DISABLED;
	else
		info->output_flags |= OBS_SOURCE_CAP_DISABLED;
}
C
cg2121 已提交
4371 4372 4373 4374 4375 4376 4377 4378 4379 4380 4381 4382 4383 4384 4385 4386 4387 4388 4389

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

	return source->sample_info.speakers;
}

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

	source->balance = balance;
}

float obs_source_get_balance_value(const obs_source_t *source)
{
J
jp9000 已提交
4390 4391 4392
	return obs_source_valid(source, "obs_source_get_balance_value")
		       ? source->balance
		       : 0.5f;
C
cg2121 已提交
4393
}