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

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
6
    the Free Software Foundation, either version 2 of the License, or
J
jp9000 已提交
7 8 9 10 11 12 13 14 15 16 17
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
******************************************************************************/

18 19
#include <inttypes.h>

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

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

32
static inline bool source_valid(const struct obs_source *source)
33 34 35 36
{
	return source && source->context.data;
}

37
const struct obs_source_info *find_source(struct darray *list, const char *id)
J
jp9000 已提交
38 39
{
	size_t i;
J
jp9000 已提交
40
	struct obs_source_info *array = list->array;
J
jp9000 已提交
41 42

	for (i = 0; i < list->num; i++) {
J
jp9000 已提交
43
		struct obs_source_info *info = array+i;
44
		if (strcmp(info->id, id) == 0)
J
jp9000 已提交
45 46 47 48 49 50
			return info;
	}

	return NULL;
}

J
jp9000 已提交
51
static const struct obs_source_info *get_source_info(enum obs_source_type type,
52 53 54 55 56
		const char *id)
{
	struct darray *list = NULL;

	switch (type) {
J
jp9000 已提交
57 58 59 60 61 62 63 64 65 66 67
	case OBS_SOURCE_TYPE_INPUT:
		list = &obs->input_types.da;
		break;

	case OBS_SOURCE_TYPE_FILTER:
		list = &obs->filter_types.da;
		break;

	case OBS_SOURCE_TYPE_TRANSITION:
		list = &obs->transition_types.da;
		break;
68 69 70 71 72
	}

	return find_source(list, id);
}

73 74 75 76 77 78 79 80
static const char *source_signals[] = {
	"void destroy(ptr source)",
	"void add(ptr source)",
	"void remove(ptr source)",
	"void activate(ptr source)",
	"void deactivate(ptr source)",
	"void show(ptr source)",
	"void hide(ptr source)",
J
jp9000 已提交
81
	"void rename(ptr source, string new_name, string prev_name)",
82
	"void volume(ptr source, in out float volume)",
83
	"void update_properties(ptr source)",
J
jp9000 已提交
84
	"void update_flags(ptr source, int flags)",
J
jp9000 已提交
85
	"void audio_sync(ptr source, int out int offset)",
86
	"void audio_data(ptr source, ptr data)",
87
	"void audio_mixers(ptr source, in out int mixers)",
J
jp9000 已提交
88 89
	"void filter_add(ptr source, ptr filter)",
	"void filter_remove(ptr source, ptr filter)",
90 91 92
	NULL
};

93
bool obs_source_init_context(struct obs_source *source,
94
		obs_data_t *settings, const char *name)
95
{
96
	if (!obs_context_data_init(&source->context, settings, name))
97 98
		return false;

99 100
	return signal_handler_add_array(source->context.signals,
			source_signals);
101 102
}

103 104
const char *obs_source_get_display_name(enum obs_source_type type,
		const char *id)
105
{
J
jp9000 已提交
106
	const struct obs_source_info *info = get_source_info(type, id);
107
	return (info != NULL) ? info->get_name() : NULL;
108 109
}

110
/* internal initialization */
J
jp9000 已提交
111 112
bool obs_source_init(struct obs_source *source,
		const struct obs_source_info *info)
J
jp9000 已提交
113
{
114
	source->refs = 1;
J
jp9000 已提交
115
	source->user_volume = 1.0f;
116 117
	source->present_volume = 1.0f;
	source->base_volume = 0.0f;
J
jp9000 已提交
118
	source->sync_offset = 0;
119
	pthread_mutex_init_value(&source->filter_mutex);
120
	pthread_mutex_init_value(&source->async_mutex);
121
	pthread_mutex_init_value(&source->audio_mutex);
122

123 124 125 126
	if (pthread_mutex_init(&source->filter_mutex, NULL) != 0)
		return false;
	if (pthread_mutex_init(&source->audio_mutex, NULL) != 0)
		return false;
127
	if (pthread_mutex_init(&source->async_mutex, NULL) != 0)
128
		return false;
J
jp9000 已提交
129

130
	if (info && info->output_flags & OBS_SOURCE_AUDIO) {
131
		source->audio_line = audio_output_create_line(obs->audio.audio,
132
				source->context.name, 0xF);
133 134
		if (!source->audio_line) {
			blog(LOG_ERROR, "Failed to create audio line for "
135
			                "source '%s'", source->context.name);
136 137 138
			return false;
		}
	}
139

140 141 142
	obs_context_data_insert(&source->context,
			&obs->data.sources_mutex,
			&obs->data.first_source);
143
	return true;
J
jp9000 已提交
144 145
}

146
static inline void obs_source_dosignal(struct obs_source *source,
147
		const char *signal_obs, const char *signal_source)
148 149 150 151
{
	struct calldata data;

	calldata_init(&data);
152
	calldata_set_ptr(&data, "source", source);
153 154 155
	if (signal_obs)
		signal_handler_signal(obs->signals, signal_obs, &data);
	if (signal_source)
156 157
		signal_handler_signal(source->context.signals, signal_source,
				&data);
158 159 160
	calldata_free(&data);
}

161 162
obs_source_t *obs_source_create(enum obs_source_type type, const char *id,
		const char *name, obs_data_t *settings)
J
jp9000 已提交
163
{
164
	struct obs_source *source = bzalloc(sizeof(struct obs_source));
J
jp9000 已提交
165

J
jp9000 已提交
166
	const struct obs_source_info *info = get_source_info(type, id);
J
jp9000 已提交
167
	if (!info) {
P
Palana 已提交
168
		blog(LOG_ERROR, "Source ID '%s' not found", id);
J
jp9000 已提交
169

170 171 172 173 174 175
		source->info.id      = bstrdup(id);
		source->info.type    = type;
		source->owns_info_id = true;
	} else {
		source->info = *info;
	}
176

177
	if (!obs_source_init_context(source, settings, name))
178 179
		goto fail;

180 181
	if (info && info->get_defaults)
		info->get_defaults(source->context.settings);
J
jp9000 已提交
182

183 184
	/* allow the source to be created even if creation fails so that the
	 * user's data doesn't become lost */
185 186 187
	if (info)
		source->context.data = info->create(source->context.settings,
				source);
188
	if (!source->context.data)
189
		blog(LOG_ERROR, "Failed to create source '%s'!", name);
190

J
jp9000 已提交
191
	if (!obs_source_init(source, info))
192
		goto fail;
J
jp9000 已提交
193

194
	blog(LOG_INFO, "source '%s' (%s) created", name, id);
195
	obs_source_dosignal(source, "source_create", NULL);
J
jp9000 已提交
196

197 198
	source->flags = source->default_flags;

199 200 201
	/* prevents the source from clearing the cache on the first frame */
	source->async_reset_texture = true;

202
	if (info && info->type == OBS_SOURCE_TYPE_TRANSITION)
J
jp9000 已提交
203
		os_atomic_inc_long(&obs->data.active_transitions);
J
jp9000 已提交
204
	return source;
205 206 207 208 209

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

212 213
void obs_source_frame_init(struct obs_source_frame *frame,
		enum video_format format, uint32_t width, uint32_t height)
214
{
215
	struct video_frame vid_frame;
J
jp9000 已提交
216 217 218 219

	if (!frame)
		return;

220
	video_frame_init(&vid_frame, format, width, height);
221 222 223
	frame->format = format;
	frame->width  = width;
	frame->height = height;
224

225 226 227
	for (size_t i = 0; i < MAX_AV_PLANES; i++) {
		frame->data[i]     = vid_frame.data[i];
		frame->linesize[i] = vid_frame.linesize[i];
228 229 230
	}
}

231 232 233 234 235 236
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);
}

237
void obs_source_destroy(struct obs_source *source)
J
jp9000 已提交
238
{
239
	size_t i;
240

J
jp9000 已提交
241 242 243
	if (!source)
		return;

J
jp9000 已提交
244 245 246
	if (source->info.type == OBS_SOURCE_TYPE_TRANSITION)
		os_atomic_dec_long(&obs->data.active_transitions);

247 248
	obs_context_data_remove(&source->context);

249 250
	blog(LOG_INFO, "source '%s' destroyed", source->context.name);

251
	obs_source_dosignal(source, "source_destroy", "destroy");
252

253
	if (source->context.data) {
254
		source->info.destroy(source->context.data);
255 256
		source->context.data = NULL;
	}
257

258 259
	if (source->filter_parent)
		obs_source_filter_remove(source->filter_parent, source);
260

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

264
	for (i = 0; i < source->async_cache.num; i++)
265
		obs_source_frame_decref(source->async_cache.array[i].frame);
266

267 268 269
	gs_enter_context(obs->video.graphics);
	gs_texrender_destroy(source->async_convert_texrender);
	gs_texture_destroy(source->async_texture);
J
jp9000 已提交
270
	gs_texrender_destroy(source->filter_texrender);
271
	gs_leave_context();
J
jp9000 已提交
272

J
jp9000 已提交
273
	for (i = 0; i < MAX_AV_PLANES; i++)
274 275
		bfree(source->audio_data.data[i]);

276 277 278
	audio_line_destroy(source->audio_line);
	audio_resampler_destroy(source->resampler);

279
	da_free(source->async_cache);
280
	da_free(source->async_frames);
281 282 283
	da_free(source->filters);
	pthread_mutex_destroy(&source->filter_mutex);
	pthread_mutex_destroy(&source->audio_mutex);
284
	pthread_mutex_destroy(&source->async_mutex);
285
	obs_context_data_free(&source->context);
286

287 288 289
	if (source->owns_info_id)
		bfree((void*)source->info.id);

290 291 292
	bfree(source);
}

293
void obs_source_addref(obs_source_t *source)
294
{
P
Palana 已提交
295
	if (source)
J
jp9000 已提交
296
		os_atomic_inc_long(&source->refs);
297 298
}

299
void obs_source_release(obs_source_t *source)
300
{
P
Palana 已提交
301 302
	if (!source)
		return;
303

J
jp9000 已提交
304
	if (os_atomic_dec_long(&source->refs) == 0)
P
Palana 已提交
305
		obs_source_destroy(source);
306 307
}

308
void obs_source_remove(obs_source_t *source)
309
{
310
	struct obs_core_data *data = &obs->data;
311
	size_t id;
312
	bool   exists;
313 314 315

	pthread_mutex_lock(&data->sources_mutex);

J
jp9000 已提交
316 317
	if (!source || source->removed) {
		pthread_mutex_unlock(&data->sources_mutex);
J
jp9000 已提交
318
		return;
J
jp9000 已提交
319
	}
J
jp9000 已提交
320

J
jp9000 已提交
321
	source->removed = true;
J
jp9000 已提交
322

J
jp9000 已提交
323 324
	obs_source_addref(source);

325 326 327 328
	id = da_find(data->user_sources, &source, 0);
	exists = (id != DARRAY_INVALID);
	if (exists) {
		da_erase(data->user_sources, id);
J
jp9000 已提交
329
		obs_source_release(source);
330 331 332
	}

	pthread_mutex_unlock(&data->sources_mutex);
J
jp9000 已提交
333

334 335 336
	if (exists)
		obs_source_dosignal(source, "source_remove", "remove");

J
jp9000 已提交
337
	obs_source_release(source);
338 339
}

340
bool obs_source_removed(const obs_source_t *source)
341
{
J
jp9000 已提交
342
	return source ? source->removed : true;
J
jp9000 已提交
343 344
}

345
static inline obs_data_t *get_defaults(const struct obs_source_info *info)
J
jp9000 已提交
346
{
347
	obs_data_t *settings = obs_data_create();
348 349
	if (info->get_defaults)
		info->get_defaults(settings);
J
jp9000 已提交
350 351 352
	return settings;
}

353
obs_data_t *obs_source_settings(enum obs_source_type type, const char *id)
J
jp9000 已提交
354 355
{
	const struct obs_source_info *info = get_source_info(type, id);
J
jp9000 已提交
356
	return (info) ? get_defaults(info) : NULL;
J
jp9000 已提交
357 358
}

359
obs_properties_t *obs_get_source_properties(enum obs_source_type type,
360
		const char *id)
J
jp9000 已提交
361
{
J
jp9000 已提交
362
	const struct obs_source_info *info = get_source_info(type, id);
363
	if (info && info->get_properties) {
364 365
		obs_data_t       *defaults = get_defaults(info);
		obs_properties_t *properties;
J
jp9000 已提交
366

367
		properties = info->get_properties(NULL);
J
jp9000 已提交
368 369 370 371
		obs_properties_apply_settings(properties, defaults);
		obs_data_release(defaults);
		return properties;
	}
J
jp9000 已提交
372 373 374
	return NULL;
}

375
obs_properties_t *obs_source_properties(const obs_source_t *source)
376
{
377
	if (source_valid(source) && source->info.get_properties) {
378
		obs_properties_t *props;
379
		props = source->info.get_properties(source->context.data);
380
		obs_properties_apply_settings(props, source->context.settings);
J
jp9000 已提交
381 382 383
		return props;
	}

384 385 386
	return NULL;
}

387
uint32_t obs_source_get_output_flags(const obs_source_t *source)
J
jp9000 已提交
388
{
J
jp9000 已提交
389
	return source ? source->info.output_flags : 0;
J
jp9000 已提交
390 391
}

392
static void obs_source_deferred_update(obs_source_t *source)
393
{
394 395 396 397
	if (source->context.data && source->info.update)
		source->info.update(source->context.data,
				source->context.settings);

398 399 400
	source->defer_update = false;
}

401
void obs_source_update(obs_source_t *source, obs_data_t *settings)
J
jp9000 已提交
402
{
J
jp9000 已提交
403 404
	if (!source) return;

405 406 407 408 409 410 411 412
	if (settings)
		obs_data_apply(source->context.settings, settings);

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

416 417 418 419 420 421 422 423 424 425 426 427 428 429 430
void obs_source_update_properties(obs_source_t *source)
{
	calldata_t calldata;

	if (!source) return;

	calldata_init(&calldata);
	calldata_set_ptr(&calldata, "source", source);

	signal_handler_signal(obs_source_get_signal_handler(source),
			"update_properties", &calldata);

	calldata_free(&calldata);
}

431
void obs_source_send_mouse_click(obs_source_t *source,
K
kc5nra 已提交
432 433 434 435 436 437 438 439 440 441 442 443 444 445 446
		const struct obs_mouse_event *event,
		int32_t type, bool mouse_up,
		uint32_t click_count)
{
	if (!source)
		return;

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

447
void obs_source_send_mouse_move(obs_source_t *source,
K
kc5nra 已提交
448 449 450 451 452 453 454 455 456 457 458 459 460
		const struct obs_mouse_event *event, bool mouse_leave)
{
	if (!source)
		return;

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

461
void obs_source_send_mouse_wheel(obs_source_t *source,
K
kc5nra 已提交
462 463 464 465 466 467 468 469 470 471 472 473 474
		const struct obs_mouse_event *event, int x_delta, int y_delta)
{
	if (!source)
		return;

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

475
void obs_source_send_focus(obs_source_t *source, bool focus)
K
kc5nra 已提交
476 477 478 479 480 481 482 483 484 485 486
{
	if (!source)
		return;

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

487
void obs_source_send_key_click(obs_source_t *source,
K
kc5nra 已提交
488 489 490 491 492 493 494 495 496 497 498 499 500
		const struct obs_key_event *event, bool key_up)
{
	if (!source)
		return;

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

501
static void activate_source(obs_source_t *source)
J
jp9000 已提交
502
{
503
	if (source->context.data && source->info.activate)
504
		source->info.activate(source->context.data);
505
	obs_source_dosignal(source, "source_activate", "activate");
J
jp9000 已提交
506 507
}

508
static void deactivate_source(obs_source_t *source)
J
jp9000 已提交
509
{
510
	if (source->context.data && source->info.deactivate)
511
		source->info.deactivate(source->context.data);
512
	obs_source_dosignal(source, "source_deactivate", "deactivate");
513
}
514

515
static void show_source(obs_source_t *source)
516
{
517
	if (source->context.data && source->info.show)
518
		source->info.show(source->context.data);
519
	obs_source_dosignal(source, "source_show", "show");
520 521
}

522
static void hide_source(obs_source_t *source)
523
{
524
	if (source->context.data && source->info.hide)
525
		source->info.hide(source->context.data);
526
	obs_source_dosignal(source, "source_hide", "hide");
527 528
}

529 530
static void activate_tree(obs_source_t *parent, obs_source_t *child,
		void *param)
531
{
532
	os_atomic_inc_long(&child->activate_refs);
J
jp9000 已提交
533 534 535

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
536 537
}

538
static void deactivate_tree(obs_source_t *parent, obs_source_t *child,
539 540
		void *param)
{
541
	os_atomic_dec_long(&child->activate_refs);
J
jp9000 已提交
542 543 544

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
545 546
}

547
static void show_tree(obs_source_t *parent, obs_source_t *child, void *param)
548
{
549
	os_atomic_inc_long(&child->show_refs);
550 551 552 553 554

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
}

555
static void hide_tree(obs_source_t *parent, obs_source_t *child, void *param)
556
{
557
	os_atomic_dec_long(&child->show_refs);
558 559 560 561 562

	UNUSED_PARAMETER(parent);
	UNUSED_PARAMETER(param);
}

563
void obs_source_activate(obs_source_t *source, enum view_type type)
564 565 566
{
	if (!source) return;

J
jp9000 已提交
567
	if (os_atomic_inc_long(&source->show_refs) == 1) {
568 569 570 571
		obs_source_enum_tree(source, show_tree, NULL);
	}

	if (type == MAIN_VIEW) {
J
jp9000 已提交
572
		if (os_atomic_inc_long(&source->activate_refs) == 1) {
573 574
			obs_source_enum_tree(source, activate_tree, NULL);
		}
575 576 577
	}
}

578
void obs_source_deactivate(obs_source_t *source, enum view_type type)
579 580 581
{
	if (!source) return;

J
jp9000 已提交
582
	if (os_atomic_dec_long(&source->show_refs) == 0) {
583 584 585 586
		obs_source_enum_tree(source, hide_tree, NULL);
	}

	if (type == MAIN_VIEW) {
J
jp9000 已提交
587
		if (os_atomic_dec_long(&source->activate_refs) == 0) {
588 589
			obs_source_enum_tree(source, deactivate_tree, NULL);
		}
590
	}
J
jp9000 已提交
591 592
}

593
void obs_source_video_tick(obs_source_t *source, float seconds)
J
jp9000 已提交
594
{
595 596
	bool now_showing, now_active;

J
jp9000 已提交
597 598
	if (!source) return;

599 600 601
	if (source->defer_update)
		obs_source_deferred_update(source);

J
jp9000 已提交
602 603
	/* reset the filter render texture information once every frame */
	if (source->filter_texrender)
604
		gs_texrender_reset(source->filter_texrender);
J
jp9000 已提交
605

606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629
	/* 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;
	}

630
	if (source->context.data && source->info.video_tick)
631
		source->info.video_tick(source->context.data, seconds);
632 633

	source->async_rendered = false;
J
jp9000 已提交
634 635
}

636
/* unless the value is 3+ hours worth of frames, this won't overflow */
J
jp9000 已提交
637
static inline uint64_t conv_frames_to_time(size_t frames)
638
{
J
jp9000 已提交
639
	const struct audio_output_info *info;
640
	info = audio_output_get_info(obs->audio.audio);
641 642 643

	return (uint64_t)frames * 1000000000ULL /
		(uint64_t)info->samples_per_sec;
644 645
}

646 647
/* maximum timestamp variance in nanoseconds */
#define MAX_TS_VAR          2000000000ULL
648

649 650
static inline void reset_audio_timing(obs_source_t *source, uint64_t timestamp,
		uint64_t os_time)
651 652
{
	source->timing_set    = true;
653
	source->timing_adjust = os_time - timestamp;
654
}
655

656
static inline void handle_ts_jump(obs_source_t *source, uint64_t expected,
657
		uint64_t ts, uint64_t diff, uint64_t os_time)
658
{
J
jp9000 已提交
659
	blog(LOG_DEBUG, "Timestamp for source '%s' jumped by '%"PRIu64"', "
660
	                "expected value %"PRIu64", input value %"PRIu64,
661
	                source->context.name, diff, expected, ts);
662 663

	/* if has video, ignore audio data until reset */
664
	if (!(source->info.output_flags & OBS_SOURCE_ASYNC))
665
		reset_audio_timing(source, ts, os_time);
666 667
}

668
static void source_signal_audio_data(obs_source_t *source,
669 670
		struct audio_data *in)
{
671
	struct calldata data;
672

673
	calldata_init(&data);
674

675 676
	calldata_set_ptr(&data, "source", source);
	calldata_set_ptr(&data, "data",   in);
677

678
	signal_handler_signal(source->context.signals, "audio_data", &data);
679

680
	calldata_free(&data);
681 682
}

683 684 685 686 687
static inline uint64_t uint64_diff(uint64_t ts1, uint64_t ts2)
{
	return (ts1 < ts2) ?  (ts2 - ts1) : (ts1 - ts2);
}

688
static void source_output_audio_line(obs_source_t *source,
689 690 691
		const struct audio_data *data)
{
	struct audio_data in = *data;
692
	uint64_t diff;
693
	uint64_t os_time = os_gettime_ns();
694

695 696 697 698 699
	/* 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;
700

701 702
	} else if (!source->timing_set) {
		reset_audio_timing(source, in.timestamp, os_time);
703

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

707
		/* smooth audio if within threshold */
708
		if (diff > MAX_TS_VAR)
709
			handle_ts_jump(source, source->next_audio_ts_min,
710
					in.timestamp, diff, os_time);
711
		else if (diff < TS_SMOOTHING_THRESHOLD)
712
			in.timestamp = source->next_audio_ts_min;
713 714
	}

715
	source->next_audio_ts_min = in.timestamp +
J
jp9000 已提交
716
		conv_frames_to_time(in.frames);
717

J
jp9000 已提交
718
	in.timestamp += source->timing_adjust + source->sync_offset;
719 720 721
	in.volume = source->base_volume * source->user_volume *
		source->present_volume * obs->audio.user_volume *
		obs->audio.present_volume;
722

723
	audio_line_output(source->audio_line, &in);
724
	source_signal_audio_data(source, &in);
725 726
}

727 728 729 730 731 732 733 734
enum convert_type {
	CONVERT_NONE,
	CONVERT_NV12,
	CONVERT_420,
	CONVERT_422_U,
	CONVERT_422_Y,
};

735
static inline enum convert_type get_convert_type(enum video_format format)
736
{
737
	switch (format) {
738 739 740 741 742 743 744 745 746 747 748
	case VIDEO_FORMAT_I420:
		return CONVERT_420;
	case VIDEO_FORMAT_NV12:
		return CONVERT_NV12;

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

749
	case VIDEO_FORMAT_NONE:
750 751 752 753 754 755 756 757 758
	case VIDEO_FORMAT_RGBA:
	case VIDEO_FORMAT_BGRA:
	case VIDEO_FORMAT_BGRX:
		return CONVERT_NONE;
	}

	return CONVERT_NONE;
}

759
static inline bool set_packed422_sizes(struct obs_source *source,
760
		const struct obs_source_frame *frame)
761 762
{
	source->async_convert_height = frame->height;
763 764 765 766 767 768
	source->async_convert_width  = frame->width / 2;
	source->async_texture_format = GS_BGRA;
	return true;
}

static inline bool set_planar420_sizes(struct obs_source *source,
769
		const struct obs_source_frame *frame)
770 771 772 773 774 775 776
{
	uint32_t size = frame->width * frame->height;
	size += size/2;

	source->async_convert_width   = frame->width;
	source->async_convert_height  = (size / frame->width + 1) & 0xFFFFFFFE;
	source->async_texture_format  = GS_R8;
777 778
	source->async_plane_offset[0] = (int)(frame->data[1] - frame->data[0]);
	source->async_plane_offset[1] = (int)(frame->data[2] - frame->data[0]);
779 780 781
	return true;
}

J
jp9000 已提交
782
static inline bool set_nv12_sizes(struct obs_source *source,
783
		const struct obs_source_frame *frame)
J
jp9000 已提交
784 785 786 787 788 789 790
{
	uint32_t size = frame->width * frame->height;
	size += size/2;

	source->async_convert_width   = frame->width;
	source->async_convert_height  = (size / frame->width + 1) & 0xFFFFFFFE;
	source->async_texture_format  = GS_R8;
791
	source->async_plane_offset[0] = (int)(frame->data[1] - frame->data[0]);
J
jp9000 已提交
792 793 794
	return true;
}

795
static inline bool init_gpu_conversion(struct obs_source *source,
796
		const struct obs_source_frame *frame)
797 798 799 800 801 802 803
{
	switch (get_convert_type(frame->format)) {
		case CONVERT_422_Y:
		case CONVERT_422_U:
			return set_packed422_sizes(source, frame);

		case CONVERT_420:
804 805 806
			return set_planar420_sizes(source, frame);

		case CONVERT_NV12:
J
jp9000 已提交
807
			return set_nv12_sizes(source, frame);
808 809 810 811 812 813 814 815 816 817
			break;

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

	}
	return false;
}

818 819 820 821 822 823 824 825 826 827 828
static inline enum gs_color_format convert_video_format(
		enum video_format format)
{
	if (format == VIDEO_FORMAT_RGBA)
		return GS_RGBA;
	else if (format == VIDEO_FORMAT_BGRA)
		return GS_BGRA;

	return GS_BGRX;
}

829
static inline bool set_async_texture_size(struct obs_source *source,
830
		const struct obs_source_frame *frame)
831
{
832 833 834 835 836 837
	enum convert_type cur = get_convert_type(frame->format);

	if (!source->async_reset_texture)
		return true;

	source->async_reset_texture = false;
838

839 840
	gs_texture_destroy(source->async_texture);
	gs_texrender_destroy(source->async_convert_texrender);
841 842 843 844 845 846
	source->async_convert_texrender = NULL;

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

		source->async_convert_texrender =
847
			gs_texrender_create(GS_BGRX, GS_ZS_NONE);
848

849
		source->async_texture = gs_texture_create(
850 851
				source->async_convert_width,
				source->async_convert_height,
852 853
				source->async_texture_format,
				1, NULL, GS_DYNAMIC);
854 855

	} else {
856 857
		enum gs_color_format format = convert_video_format(
				frame->format);
858 859
		source->async_gpu_conversion = false;

860
		source->async_texture = gs_texture_create(
861
				frame->width, frame->height,
862
				format, 1, NULL, GS_DYNAMIC);
863 864
	}

865
	return !!source->async_texture;
866 867
}

868
static void upload_raw_frame(gs_texture_t *tex,
869
		const struct obs_source_frame *frame)
870 871 872 873
{
	switch (get_convert_type(frame->format)) {
		case CONVERT_422_U:
		case CONVERT_422_Y:
874
			gs_texture_set_image(tex, frame->data[0],
875 876 877 878
					frame->linesize[0], false);
			break;

		case CONVERT_420:
879
			gs_texture_set_image(tex, frame->data[0],
880 881 882 883
					frame->width, false);
			break;

		case CONVERT_NV12:
J
jp9000 已提交
884 885
			gs_texture_set_image(tex, frame->data[0],
					frame->width, false);
886 887 888 889 890 891 892 893 894 895 896 897
			break;

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

static const char *select_conversion_technique(enum video_format format)
{
	switch (format) {
		case VIDEO_FORMAT_UYVY:
898
			return "UYVY_Reverse";
899 900 901 902 903 904 905 906

		case VIDEO_FORMAT_YUY2:
			return "YUY2_Reverse";

		case VIDEO_FORMAT_YVYU:
			return "YVYU_Reverse";

		case VIDEO_FORMAT_I420:
907 908 909
			return "I420_Reverse";

		case VIDEO_FORMAT_NV12:
J
jp9000 已提交
910
			return "NV12_Reverse";
911 912 913 914 915 916 917 918 919 920 921 922
			break;

		case VIDEO_FORMAT_BGRA:
		case VIDEO_FORMAT_BGRX:
		case VIDEO_FORMAT_RGBA:
		case VIDEO_FORMAT_NONE:
			assert(false && "No conversion requested");
			break;
	}
	return NULL;
}

923
static inline void set_eparam(gs_effect_t *effect, const char *name, float val)
924
{
925
	gs_eparam_t *param = gs_effect_get_param_by_name(effect, name);
926
	gs_effect_set_float(param, val);
927 928 929
}

static bool update_async_texrender(struct obs_source *source,
930
		const struct obs_source_frame *frame)
931
{
932 933
	gs_texture_t   *tex       = source->async_texture;
	gs_texrender_t *texrender = source->async_convert_texrender;
934

935
	gs_texrender_reset(texrender);
936 937 938 939 940 941

	upload_raw_frame(tex, frame);

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

942 943 944
	float convert_width  = (float)source->async_convert_width;
	float convert_height = (float)source->async_convert_height;

945 946
	gs_effect_t *conv = obs->video.conversion_effect;
	gs_technique_t *tech = gs_effect_get_technique(conv,
947 948
			select_conversion_technique(frame->format));

949
	if (!gs_texrender_begin(texrender, cx, cy))
950 951
		return false;

952 953
	gs_technique_begin(tech);
	gs_technique_begin_pass(tech, 0);
954

955
	gs_effect_set_texture(gs_effect_get_param_by_name(conv, "image"), tex);
956 957 958 959
	set_eparam(conv, "width",  (float)cx);
	set_eparam(conv, "height", (float)cy);
	set_eparam(conv, "width_i",  1.0f / cx);
	set_eparam(conv, "height_i", 1.0f / cy);
960
	set_eparam(conv, "width_d2",  cx * 0.5f);
961
	set_eparam(conv, "height_d2", cy * 0.5f);
962
	set_eparam(conv, "width_d2_i",  1.0f / (cx * 0.5f));
963
	set_eparam(conv, "height_d2_i", 1.0f / (cy * 0.5f));
964 965 966 967 968 969 970 971 972 973
	set_eparam(conv, "input_width",  convert_width);
	set_eparam(conv, "input_height", convert_height);
	set_eparam(conv, "input_width_i",  1.0f / convert_width);
	set_eparam(conv, "input_height_i", 1.0f / convert_height);
	set_eparam(conv, "input_width_i_d2",  (1.0f / convert_width)  * 0.5f);
	set_eparam(conv, "input_height_i_d2", (1.0f / convert_height) * 0.5f);
	set_eparam(conv, "u_plane_offset",
			(float)source->async_plane_offset[0]);
	set_eparam(conv, "v_plane_offset",
			(float)source->async_plane_offset[1]);
974 975 976 977 978

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

	gs_draw_sprite(tex, 0, cx, cy);

979 980
	gs_technique_end_pass(tech);
	gs_technique_end(tech);
981

982
	gs_texrender_end(texrender);
983 984 985 986

	return true;
}

987
static bool update_async_texture(struct obs_source *source,
988
		const struct obs_source_frame *frame)
989
{
990 991
	gs_texture_t      *tex       = source->async_texture;
	gs_texrender_t    *texrender = source->async_convert_texrender;
992
	enum convert_type type      = get_convert_type(frame->format);
993
	uint8_t           *ptr;
994 995
	uint32_t          linesize;

996 997
	source->async_flip       = frame->flip;
	source->async_full_range = frame->full_range;
998 999
	memcpy(source->async_color_matrix, frame->color_matrix,
			sizeof(frame->color_matrix));
1000 1001 1002 1003
	memcpy(source->async_color_range_min, frame->color_range_min,
			sizeof frame->color_range_min);
	memcpy(source->async_color_range_max, frame->color_range_max,
			sizeof frame->color_range_max);
1004

1005 1006 1007
	if (source->async_gpu_conversion && texrender)
		return update_async_texrender(source, frame);

1008
	if (type == CONVERT_NONE) {
1009
		gs_texture_set_image(tex, frame->data[0], frame->linesize[0],
1010
				false);
1011 1012 1013
		return true;
	}

1014
	if (!gs_texture_map(tex, &ptr, &linesize))
1015 1016 1017
		return false;

	if (type == CONVERT_420)
J
jp9000 已提交
1018 1019 1020
		decompress_420((const uint8_t* const*)frame->data,
				frame->linesize,
				0, frame->height, ptr, linesize);
1021 1022

	else if (type == CONVERT_NV12)
J
jp9000 已提交
1023 1024 1025
		decompress_nv12((const uint8_t* const*)frame->data,
				frame->linesize,
				0, frame->height, ptr, linesize);
1026 1027

	else if (type == CONVERT_422_Y)
1028
		decompress_422(frame->data[0], frame->linesize[0],
J
jp9000 已提交
1029
				0, frame->height, ptr, linesize, true);
1030 1031

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

1035
	gs_texture_unmap(tex);
1036 1037 1038
	return true;
}

1039
static inline void obs_source_draw_texture(struct obs_source *source,
1040
		gs_effect_t *effect, float *color_matrix,
1041
		float const *color_range_min, float const *color_range_max)
1042
{
1043 1044
	gs_texture_t *tex = source->async_texture;
	gs_eparam_t  *param;
1045

1046
	if (source->async_convert_texrender)
1047
		tex = gs_texrender_get_texture(source->async_convert_texrender);
1048

P
Palana 已提交
1049
	if (color_range_min) {
1050
		size_t const size = sizeof(float) * 3;
1051 1052
		param = gs_effect_get_param_by_name(effect, "color_range_min");
		gs_effect_set_val(param, color_range_min, size);
P
Palana 已提交
1053
	}
1054

P
Palana 已提交
1055 1056
	if (color_range_max) {
		size_t const size = sizeof(float) * 3;
1057 1058
		param = gs_effect_get_param_by_name(effect, "color_range_max");
		gs_effect_set_val(param, color_range_max, size);
P
Palana 已提交
1059
	}
1060

P
Palana 已提交
1061
	if (color_matrix) {
1062 1063
		param = gs_effect_get_param_by_name(effect, "color_matrix");
		gs_effect_set_val(param, color_matrix, sizeof(float) * 16);
1064 1065
	}

1066 1067
	param = gs_effect_get_param_by_name(effect, "image");
	gs_effect_set_texture(param, tex);
1068

1069 1070
	gs_draw_sprite(tex, source->async_flip ? GS_FLIP_V : 0, 0, 0);
}
1071

1072 1073
static void obs_source_draw_async_texture(struct obs_source *source)
{
1074
	gs_effect_t    *effect        = gs_get_effect();
1075 1076 1077 1078
	bool           yuv           = format_is_yuv(source->async_format);
	bool           limited_range = yuv && !source->async_full_range;
	const char     *type         = yuv ? "DrawMatrix" : "Draw";
	bool           def_draw      = (!effect);
1079
	gs_technique_t *tech          = NULL;
1080 1081 1082

	if (def_draw) {
		effect = obs_get_default_effect();
1083 1084 1085
		tech = gs_effect_get_technique(effect, type);
		gs_technique_begin(tech);
		gs_technique_begin_pass(tech, 0);
1086 1087 1088
	}

	obs_source_draw_texture(source, effect,
1089 1090 1091
			yuv ? source->async_color_matrix : NULL,
			limited_range ? source->async_color_range_min : NULL,
			limited_range ? source->async_color_range_max : NULL);
1092 1093

	if (def_draw) {
1094 1095
		gs_technique_end_pass(tech);
		gs_technique_end(tech);
1096
	}
1097 1098
}

1099
static void obs_source_render_async_video(obs_source_t *source)
1100
{
1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112
	if (!source->async_rendered) {
		struct obs_source_frame *frame = obs_source_get_frame(source);

		source->async_rendered = true;
		if (frame) {
			if (!set_async_texture_size(source, frame))
				return;
			if (!update_async_texture(source, frame))
				return;
		}

		obs_source_release_frame(source, frame);
1113
	}
1114

1115
	if (source->async_texture && source->async_active)
1116
		obs_source_draw_async_texture(source);
1117 1118
}

1119
static inline void obs_source_render_filters(obs_source_t *source)
1120 1121 1122 1123 1124 1125
{
	source->rendering_filter = true;
	obs_source_video_render(source->filters.array[0]);
	source->rendering_filter = false;
}

J
jp9000 已提交
1126
static void obs_source_default_render(obs_source_t *source, bool color_matrix)
1127
{
1128
	gs_effect_t    *effect     = obs->video.default_effect;
1129
	const char     *tech_name = color_matrix ? "DrawMatrix" : "Draw";
1130
	gs_technique_t *tech       = gs_effect_get_technique(effect, tech_name);
1131
	size_t         passes, i;
1132

1133
	passes = gs_technique_begin(tech);
1134
	for (i = 0; i < passes; i++) {
1135
		gs_technique_begin_pass(tech, i);
1136 1137
		if (source->context.data)
			source->info.video_render(source->context.data, effect);
1138
		gs_technique_end_pass(tech);
1139
	}
1140
	gs_technique_end(tech);
1141 1142
}

1143
static inline void obs_source_main_render(obs_source_t *source)
1144
{
1145 1146 1147
	uint32_t flags      = source->info.output_flags;
	bool color_matrix   = (flags & OBS_SOURCE_COLOR_MATRIX) != 0;
	bool custom_draw    = (flags & OBS_SOURCE_CUSTOM_DRAW) != 0;
1148 1149
	bool default_effect = !source->filter_parent &&
	                      source->filters.num == 0 &&
1150
	                      !custom_draw;
1151 1152

	if (default_effect)
J
jp9000 已提交
1153
		obs_source_default_render(source, color_matrix);
1154
	else if (source->context.data)
1155
		source->info.video_render(source->context.data,
1156
				custom_draw ? NULL : gs_get_effect());
1157 1158
}

1159
void obs_source_video_render(obs_source_t *source)
J
jp9000 已提交
1160
{
1161
	if (!source_valid(source)) return;
J
jp9000 已提交
1162

1163 1164
	if (source->filters.num && !source->rendering_filter)
		obs_source_render_filters(source);
1165

1166 1167 1168 1169
	else if (source->info.video_render)
		obs_source_main_render(source);

	else if (source->filter_target)
1170 1171
		obs_source_video_render(source->filter_target);

1172
	else
1173
		obs_source_render_async_video(source);
J
jp9000 已提交
1174 1175
}

1176
static uint32_t get_base_width(const obs_source_t *source)
J
jp9000 已提交
1177
{
1178
	if (source->info.get_width) {
1179
		return source->info.get_width(source->context.data);
1180 1181 1182 1183 1184

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

1185
	return source->async_width;
J
jp9000 已提交
1186 1187
}

1188
static uint32_t get_base_height(const obs_source_t *source)
J
jp9000 已提交
1189
{
1190
	if (source->info.get_height) {
1191
		return source->info.get_height(source->context.data);
1192 1193 1194 1195 1196

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

1197
	return source->async_height;
J
jp9000 已提交
1198 1199
}

1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247
static uint32_t get_recurse_width(obs_source_t *source)
{
	uint32_t width;

	pthread_mutex_lock(&source->filter_mutex);

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

	pthread_mutex_unlock(&source->filter_mutex);

	return width;
}

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

	pthread_mutex_lock(&source->filter_mutex);

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

	pthread_mutex_unlock(&source->filter_mutex);

	return height;
}

uint32_t obs_source_get_width(obs_source_t *source)
{
	if (!source_valid(source)) return 0;

	return (source->info.type == OBS_SOURCE_TYPE_INPUT) ?
		get_recurse_width(source) :
	        get_base_width(source);
}

uint32_t obs_source_get_height(obs_source_t *source)
{
	if (!source_valid(source)) return 0;

	return (source->info.type == OBS_SOURCE_TYPE_INPUT) ?
		get_recurse_height(source) :
		get_base_height(source);
}

1248
obs_source_t *obs_filter_get_parent(const obs_source_t *filter)
1249
{
J
jp9000 已提交
1250
	return filter ? filter->filter_parent : NULL;
1251 1252
}

1253
obs_source_t *obs_filter_get_target(const obs_source_t *filter)
J
jp9000 已提交
1254
{
J
jp9000 已提交
1255
	return filter ? filter->filter_target : NULL;
J
jp9000 已提交
1256 1257
}

1258
void obs_source_filter_add(obs_source_t *source, obs_source_t *filter)
J
jp9000 已提交
1259
{
J
jp9000 已提交
1260 1261
	struct calldata cd = {0};

J
jp9000 已提交
1262 1263 1264
	if (!source || !filter)
		return;

1265 1266
	pthread_mutex_lock(&source->filter_mutex);

J
jp9000 已提交
1267
	if (da_find(source->filters, &filter, 0) != DARRAY_INVALID) {
J
jp9000 已提交
1268 1269 1270 1271 1272
		blog(LOG_WARNING, "Tried to add a filter that was already "
		                  "present on the source");
		return;
	}

1273 1274
	obs_source_addref(filter);

J
jp9000 已提交
1275
	if (source->filters.num) {
1276
		obs_source_t **back = da_end(source->filters);
J
jp9000 已提交
1277 1278 1279
		(*back)->filter_target = filter;
	}

1280 1281 1282
	filter->filter_parent = source;
	filter->filter_target = source;

J
jp9000 已提交
1283
	da_push_back(source->filters, &filter);
1284 1285 1286

	pthread_mutex_unlock(&source->filter_mutex);

J
jp9000 已提交
1287 1288 1289 1290 1291 1292
	calldata_set_ptr(&cd, "source", source);
	calldata_set_ptr(&cd, "filter", filter);

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

	calldata_free(&cd);
J
jp9000 已提交
1293 1294
}

1295
void obs_source_filter_remove(obs_source_t *source, obs_source_t *filter)
J
jp9000 已提交
1296
{
J
jp9000 已提交
1297
	struct calldata cd = {0};
1298 1299
	size_t idx;

J
jp9000 已提交
1300 1301 1302
	if (!source || !filter)
		return;

1303 1304 1305
	pthread_mutex_lock(&source->filter_mutex);

	idx = da_find(source->filters, &filter, 0);
J
jp9000 已提交
1306
	if (idx == DARRAY_INVALID)
J
jp9000 已提交
1307 1308 1309
		return;

	if (idx > 0) {
1310
		obs_source_t *prev = source->filters.array[idx-1];
J
jp9000 已提交
1311 1312 1313 1314
		prev->filter_target = filter->filter_target;
	}

	da_erase(source->filters, idx);
1315 1316 1317

	pthread_mutex_unlock(&source->filter_mutex);

J
jp9000 已提交
1318 1319 1320 1321 1322 1323 1324
	calldata_set_ptr(&cd, "source", source);
	calldata_set_ptr(&cd, "filter", filter);

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

	calldata_free(&cd);

1325
	filter->filter_parent = NULL;
J
jp9000 已提交
1326
	filter->filter_target = NULL;
1327 1328

	obs_source_release(filter);
J
jp9000 已提交
1329 1330
}

1331
void obs_source_filter_set_order(obs_source_t *source, obs_source_t *filter,
J
jp9000 已提交
1332
		enum obs_order_movement movement)
J
jp9000 已提交
1333
{
J
jp9000 已提交
1334 1335 1336 1337 1338 1339
	size_t idx, i;

	if (!source || !filter)
		return;

	idx = da_find(source->filters, &filter, 0);
J
jp9000 已提交
1340
	if (idx == DARRAY_INVALID)
J
jp9000 已提交
1341 1342
		return;

J
jp9000 已提交
1343
	if (movement == OBS_ORDER_MOVE_UP) {
J
jp9000 已提交
1344 1345 1346 1347
		if (idx == source->filters.num-1)
			return;
		da_move_item(source->filters, idx, idx+1);

J
jp9000 已提交
1348
	} else if (movement == OBS_ORDER_MOVE_DOWN) {
J
jp9000 已提交
1349 1350 1351 1352
		if (idx == 0)
			return;
		da_move_item(source->filters, idx, idx-1);

J
jp9000 已提交
1353
	} else if (movement == OBS_ORDER_MOVE_TOP) {
J
jp9000 已提交
1354 1355 1356 1357
		if (idx == source->filters.num-1)
			return;
		da_move_item(source->filters, idx, source->filters.num-1);

J
jp9000 已提交
1358
	} else if (movement == OBS_ORDER_MOVE_BOTTOM) {
J
jp9000 已提交
1359 1360 1361 1362 1363
		if (idx == 0)
			return;
		da_move_item(source->filters, idx, 0);
	}

1364
	/* reorder filter targets, not the nicest way of dealing with things */
J
jp9000 已提交
1365
	for (i = 0; i < source->filters.num; i++) {
1366
		obs_source_t *next_filter = (i == source->filters.num-1) ?
J
jp9000 已提交
1367 1368 1369 1370 1371
			source : source->filters.array[idx+1];
		source->filters.array[i]->filter_target = next_filter;
	}
}

1372
obs_data_t *obs_source_get_settings(const obs_source_t *source)
J
jp9000 已提交
1373
{
J
jp9000 已提交
1374 1375
	if (!source) return NULL;

1376 1377
	obs_data_addref(source->context.settings);
	return source->context.settings;
J
jp9000 已提交
1378 1379
}

1380
static inline struct obs_source_frame *filter_async_video(obs_source_t *source,
1381
		struct obs_source_frame *in)
1382 1383 1384 1385
{
	size_t i;
	for (i = source->filters.num; i > 0; i--) {
		struct obs_source *filter = source->filters.array[i-1];
1386 1387

		if (filter->context.data && filter->info.filter_video) {
1388 1389
			in = filter->info.filter_video(filter->context.data,
					in);
1390 1391 1392 1393 1394 1395 1396 1397
			if (!in)
				return NULL;
		}
	}

	return in;
}

1398 1399
static inline void copy_frame_data_line(struct obs_source_frame *dst,
		const struct obs_source_frame *src, uint32_t plane, uint32_t y)
1400
{
1401 1402 1403 1404
	uint32_t pos_src = y * src->linesize[plane];
	uint32_t pos_dst = y * dst->linesize[plane];
	uint32_t bytes = dst->linesize[plane] < src->linesize[plane] ?
		dst->linesize[plane] : src->linesize[plane];
1405 1406 1407 1408

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

1409 1410 1411
static inline void copy_frame_data_plane(struct obs_source_frame *dst,
		const struct obs_source_frame *src,
		uint32_t plane, uint32_t lines)
1412
{
1413
	if (dst->linesize[plane] != src->linesize[plane])
1414 1415 1416 1417
		for (uint32_t y = 0; y < lines; y++)
			copy_frame_data_line(dst, src, plane, y);
	else
		memcpy(dst->data[plane], src->data[plane],
1418
				dst->linesize[plane] * lines);
1419 1420
}

1421 1422
static void copy_frame_data(struct obs_source_frame *dst,
		const struct obs_source_frame *src)
1423 1424
{
	dst->flip         = src->flip;
1425
	dst->full_range   = src->full_range;
1426 1427
	dst->timestamp    = src->timestamp;
	memcpy(dst->color_matrix, src->color_matrix, sizeof(float) * 16);
1428 1429 1430 1431 1432
	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);
	}
1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456

	switch (dst->format) {
	case VIDEO_FORMAT_I420:
		copy_frame_data_plane(dst, src, 0, dst->height);
		copy_frame_data_plane(dst, src, 1, dst->height/2);
		copy_frame_data_plane(dst, src, 2, dst->height/2);
		break;

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

	case VIDEO_FORMAT_YVYU:
	case VIDEO_FORMAT_YUY2:
	case VIDEO_FORMAT_UYVY:
	case VIDEO_FORMAT_NONE:
	case VIDEO_FORMAT_RGBA:
	case VIDEO_FORMAT_BGRA:
	case VIDEO_FORMAT_BGRX:
		copy_frame_data_plane(dst, src, 0, dst->height);
	}
}

1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471
static inline bool async_texture_changed(struct obs_source *source,
		const struct obs_source_frame *frame)
{
	enum convert_type prev, cur;
	prev = get_convert_type(source->async_format);
	cur  = get_convert_type(frame->format);

	return source->async_width  != frame->width ||
	       source->async_height != frame->height ||
	       prev != cur;
}

static inline void free_async_cache(struct obs_source *source)
{
	for (size_t i = 0; i < source->async_cache.num; i++)
1472
		obs_source_frame_decref(source->async_cache.array[i].frame);
1473 1474 1475 1476 1477 1478

	da_resize(source->async_cache, 0);
	da_resize(source->async_frames, 0);
}

static inline struct obs_source_frame *cache_video(struct obs_source *source,
1479
		const struct obs_source_frame *frame)
1480
{
1481 1482 1483 1484 1485
	struct obs_source_frame *new_frame = NULL;

	pthread_mutex_lock(&source->async_mutex);

	if (async_texture_changed(source, frame)) {
1486 1487 1488 1489
		/* prevents the cache from being freed on the first frame */
		if (!source->async_reset_texture)
			free_async_cache(source);

1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500
		source->async_width         = frame->width;
		source->async_height        = frame->height;
		source->async_format        = frame->format;
		source->async_reset_texture = true;
	}

	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;
1501
			break;
1502 1503 1504 1505 1506 1507 1508 1509 1510 1511
		}
	}

	if (!new_frame) {
		struct async_frame new_af;

		new_frame = obs_source_frame_create(frame->format,
				frame->width, frame->height);
		new_af.frame = new_frame;
		new_af.used = true;
1512
		new_frame->refs = 1;
1513 1514 1515 1516 1517

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

	pthread_mutex_unlock(&source->async_mutex);
1518

1519
	copy_frame_data(new_frame, frame);
1520
	return new_frame;
1521 1522
}

1523
static bool ready_async_frame(obs_source_t *source, uint64_t sys_time);
1524 1525 1526

static inline void cycle_frames(struct obs_source *source)
{
1527
	if (source->async_frames.num && !source->activate_refs)
1528
		ready_async_frame(source, os_gettime_ns());
1529 1530
}

1531
void obs_source_output_video(obs_source_t *source,
1532
		const struct obs_source_frame *frame)
1533
{
1534
	if (!source)
J
jp9000 已提交
1535 1536
		return;

1537 1538
	struct obs_source_frame *output = !!frame ?
		cache_video(source, frame) : NULL;
1539 1540

	pthread_mutex_lock(&source->filter_mutex);
1541 1542 1543 1544

	if (output)
		os_atomic_inc_long(&output->refs);

1545
	output = filter_async_video(source, output);
1546 1547 1548 1549 1550 1551

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

1552 1553
	pthread_mutex_unlock(&source->filter_mutex);

1554 1555
	/* ------------------------------------------- */

1556
	if (output) {
1557
		pthread_mutex_lock(&source->async_mutex);
1558
		cycle_frames(source);
1559 1560
		da_push_back(source->async_frames, &output);
		pthread_mutex_unlock(&source->async_mutex);
1561 1562 1563 1564 1565
		source->async_active = true;
	} else {
		source->async_active = false;
		source->async_width = 0;
		source->async_height = 0;
1566
	}
1567 1568
}

1569
static inline struct obs_audio_data *filter_async_audio(obs_source_t *source,
1570
		struct obs_audio_data *in)
1571 1572 1573 1574
{
	size_t i;
	for (i = source->filters.num; i > 0; i--) {
		struct obs_source *filter = source->filters.array[i-1];
1575 1576

		if (filter->context.data && filter->info.filter_audio) {
1577 1578
			in = filter->info.filter_audio(filter->context.data,
					in);
1579 1580 1581 1582 1583 1584 1585 1586
			if (!in)
				return NULL;
		}
	}

	return in;
}

1587
static inline void reset_resampler(obs_source_t *source,
1588
		const struct obs_source_audio *audio)
1589
{
J
jp9000 已提交
1590
	const struct audio_output_info *obs_info;
1591 1592
	struct resample_info output_info;

1593
	obs_info = audio_output_get_info(obs->audio.audio);
1594

1595 1596 1597 1598 1599 1600 1601 1602
	output_info.format           = obs_info->format;
	output_info.samples_per_sec  = obs_info->samples_per_sec;
	output_info.speakers         = obs_info->speakers;

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

1603 1604 1605
	audio_resampler_destroy(source->resampler);
	source->resampler = NULL;

1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620
	if (source->sample_info.samples_per_sec == obs_info->samples_per_sec &&
	    source->sample_info.format          == obs_info->format          &&
	    source->sample_info.speakers        == obs_info->speakers) {
		source->audio_failed = false;
		return;
	}

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

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

J
jp9000 已提交
1621
static void copy_audio_data(obs_source_t *source,
J
jp9000 已提交
1622
		const uint8_t *const data[], uint32_t frames, uint64_t ts)
1623
{
1624 1625
	size_t planes    = audio_output_get_planes(obs->audio.audio);
	size_t blocksize = audio_output_get_block_size(obs->audio.audio);
1626 1627
	size_t size      = (size_t)frames * blocksize;
	bool   resize    = source->audio_storage_size < size;
1628

J
jp9000 已提交
1629 1630
	source->audio_data.frames    = frames;
	source->audio_data.timestamp = ts;
1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643

	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;
1644 1645
}

1646 1647 1648
/* TODO: SSE optimization */
static void downmix_to_mono_planar(struct obs_source *source, uint32_t frames)
{
J
jp9000 已提交
1649
	size_t channels = audio_output_get_channels(obs->audio.audio);
1650 1651 1652
	const float channels_i = 1.0f / (float)channels;
	float **data = (float**)source->audio_data.data;

J
jp9000 已提交
1653
	for (size_t channel = 1; channel < channels; channel++) {
1654 1655 1656 1657 1658 1659 1660
		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 已提交
1661
	for (size_t channel = 1; channel < channels; channel++) {
1662 1663 1664 1665 1666
		for (uint32_t frame = 0; frame < frames; frame++)
			data[channel][frame] = data[0][frame];
	}
}

1667
/* resamples/remixes new audio to the designated main audio output format */
1668
static void process_audio(obs_source_t *source,
1669
		const struct obs_source_audio *audio)
1670
{
1671
	uint32_t frames = audio->frames;
1672
	bool mono_output;
1673

1674 1675 1676 1677 1678 1679 1680 1681 1682
	if (source->sample_info.samples_per_sec != audio->samples_per_sec ||
	    source->sample_info.format          != audio->format          ||
	    source->sample_info.speakers        != audio->speakers)
		reset_resampler(source, audio);

	if (source->audio_failed)
		return;

	if (source->resampler) {
J
jp9000 已提交
1683
		uint8_t  *output[MAX_AV_PLANES];
1684 1685
		uint64_t offset;

1686 1687 1688 1689 1690
		memset(output, 0, sizeof(output));

		audio_resampler_resample(source->resampler,
				output, &frames, &offset,
				audio->data, audio->frames);
1691

J
jp9000 已提交
1692
		copy_audio_data(source, (const uint8_t *const *)output, frames,
1693 1694 1695 1696 1697
				audio->timestamp - offset);
	} else {
		copy_audio_data(source, audio->data, audio->frames,
				audio->timestamp);
	}
1698

1699 1700 1701 1702
	mono_output = audio_output_get_channels(obs->audio.audio) == 1;

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

1705
void obs_source_output_audio(obs_source_t *source,
1706
		const struct obs_source_audio *audio)
1707
{
J
jp9000 已提交
1708
	uint32_t flags;
1709
	struct obs_audio_data *output;
1710

J
jp9000 已提交
1711 1712 1713 1714
	if (!source || !audio)
		return;

	flags = source->info.output_flags;
1715
	process_audio(source, audio);
1716 1717

	pthread_mutex_lock(&source->filter_mutex);
1718
	output = filter_async_audio(source, &source->audio_data);
1719 1720

	if (output) {
1721
		struct audio_data data;
J
jp9000 已提交
1722

1723 1724
		for (int i = 0; i < MAX_AV_PLANES; i++)
			data.data[i] = output->data[i];
1725

1726 1727
		data.frames    = output->frames;
		data.timestamp = output->timestamp;
1728

1729 1730
		pthread_mutex_lock(&source->audio_mutex);
		source_output_audio_line(source, &data);
1731 1732 1733 1734 1735 1736
		pthread_mutex_unlock(&source->audio_mutex);
	}

	pthread_mutex_unlock(&source->filter_mutex);
}

1737
static inline bool frame_out_of_bounds(const obs_source_t *source, uint64_t ts)
1738
{
J
jp9000 已提交
1739
	if (ts < source->last_frame_ts)
1740
		return ((source->last_frame_ts - ts) > MAX_TS_VAR);
J
jp9000 已提交
1741
	else
1742
		return ((ts - source->last_frame_ts) > MAX_TS_VAR);
1743 1744
}

1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757
static void remove_async_frame(obs_source_t *source,
		struct obs_source_frame *frame)
{
	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 已提交
1758 1759
/* #define DEBUG_ASYNC_FRAMES 1 */

1760
static bool ready_async_frame(obs_source_t *source, uint64_t sys_time)
1761
{
1762
	struct obs_source_frame *next_frame = source->async_frames.array[0];
1763
	struct obs_source_frame *frame      = NULL;
1764 1765 1766 1767
	uint64_t sys_offset = sys_time - source->last_sys_timestamp;
	uint64_t frame_time = next_frame->timestamp;
	uint64_t frame_offset = 0;

1768
	if ((source->flags & OBS_SOURCE_FLAG_UNBUFFERED) != 0) {
1769 1770
		while (source->async_frames.num > 1) {
			da_erase(source->async_frames, 0);
1771
			remove_async_frame(source, next_frame);
1772
			next_frame = source->async_frames.array[0];
J
jp9000 已提交
1773 1774 1775 1776 1777
		}

		return true;
	}

J
jp9000 已提交
1778 1779 1780 1781 1782 1783
#if DEBUG_ASYNC_FRAMES
	blog(LOG_DEBUG, "source->last_frame_ts: %llu, frame_time: %llu, "
			"sys_offset: %llu, frame_offset: %llu, "
			"number of frames: %lu",
			source->last_frame_ts, frame_time, sys_offset,
			frame_time - source->last_frame_ts,
1784
			(unsigned long)source->async_frames.num);
J
jp9000 已提交
1785 1786
#endif

1787 1788
	/* account for timestamp invalidation */
	if (frame_out_of_bounds(source, frame_time)) {
J
jp9000 已提交
1789 1790 1791
#if DEBUG_ASYNC_FRAMES
		blog(LOG_DEBUG, "timing jump");
#endif
1792
		source->last_frame_ts = next_frame->timestamp;
J
jp9000 已提交
1793
		return true;
1794 1795
	} else {
		frame_offset = frame_time - source->last_frame_ts;
J
jp9000 已提交
1796
		source->last_frame_ts += sys_offset;
1797 1798
	}

J
jp9000 已提交
1799 1800 1801 1802 1803 1804 1805 1806 1807 1808
	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 */
		if ((source->last_frame_ts - next_frame->timestamp) < 1000000)
			break;

		if (frame)
1809
			da_erase(source->async_frames, 0);
J
jp9000 已提交
1810 1811 1812 1813 1814 1815 1816 1817 1818

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

1819
		remove_async_frame(source, frame);
1820

1821
		if (source->async_frames.num == 1)
1822 1823
			return true;

1824
		frame = next_frame;
1825
		next_frame = source->async_frames.array[1];
1826 1827

		/* more timestamp checking and compensating */
1828
		if ((next_frame->timestamp - frame_time) > MAX_TS_VAR) {
J
jp9000 已提交
1829 1830 1831
#if DEBUG_ASYNC_FRAMES
			blog(LOG_DEBUG, "timing jump");
#endif
1832 1833 1834 1835 1836 1837 1838 1839
			source->last_frame_ts =
				next_frame->timestamp - frame_offset;
		}

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

J
jp9000 已提交
1840 1841 1842 1843
#if DEBUG_ASYNC_FRAMES
	if (!frame)
		blog(LOG_DEBUG, "no frame!");
#endif
1844

1845 1846 1847
	return frame != NULL;
}

1848
static inline struct obs_source_frame *get_closest_frame(obs_source_t *source,
1849 1850
		uint64_t sys_time)
{
1851
	if (ready_async_frame(source, sys_time)) {
1852 1853
		struct obs_source_frame *frame = source->async_frames.array[0];
		da_erase(source->async_frames, 0);
1854 1855 1856 1857
		return frame;
	}

	return NULL;
1858 1859
}

1860
/*
1861 1862
 * Ensures that cached frames are displayed on time.  If multiple frames
 * were cached between renders, then releases the unnecessary frames and uses
1863 1864
 * the frame with the closest timing to ensure sync.  Also ensures that timing
 * with audio is synchronized.
1865
 */
1866
struct obs_source_frame *obs_source_get_frame(obs_source_t *source)
J
jp9000 已提交
1867
{
1868
	struct obs_source_frame *frame = NULL;
1869
	uint64_t sys_time;
1870

J
jp9000 已提交
1871 1872 1873
	if (!source)
		return NULL;

1874
	pthread_mutex_lock(&source->async_mutex);
1875

J
jp9000 已提交
1876 1877
	sys_time = os_gettime_ns();

1878
	if (!source->async_frames.num)
1879 1880
		goto unlock;

1881
	if (!source->last_frame_ts) {
1882 1883
		frame = source->async_frames.array[0];
		da_erase(source->async_frames, 0);
1884

1885
		source->last_frame_ts = frame->timestamp;
1886
	} else {
1887
		frame = get_closest_frame(source, sys_time);
J
jp9000 已提交
1888 1889 1890 1891 1892 1893
	}

	/* reset timing to current system time */
	if (frame) {
		source->timing_adjust = sys_time - frame->timestamp;
		source->timing_set = true;
1894 1895

		os_atomic_inc_long(&frame->refs);
1896 1897
	}

J
jp9000 已提交
1898
unlock:
1899 1900
	source->last_sys_timestamp = sys_time;

1901
	pthread_mutex_unlock(&source->async_mutex);
1902

1903
	return frame;
J
jp9000 已提交
1904 1905
}

1906
void obs_source_release_frame(obs_source_t *source,
1907
		struct obs_source_frame *frame)
J
jp9000 已提交
1908
{
1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922
	if (!frame)
		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);
1923
	}
J
jp9000 已提交
1924
}
1925

1926
const char *obs_source_get_name(const obs_source_t *source)
1927
{
1928
	return source ? source->context.name : NULL;
1929 1930
}

1931
void obs_source_set_name(obs_source_t *source, const char *name)
1932
{
J
jp9000 已提交
1933
	if (!source) return;
J
jp9000 已提交
1934 1935 1936 1937 1938 1939 1940

	if (!name || !*name || strcmp(name, source->context.name) != 0) {
		struct calldata data;
		char *prev_name = bstrdup(source->context.name);
		obs_context_data_setname(&source->context, name);

		calldata_init(&data);
1941 1942 1943
		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 已提交
1944 1945 1946 1947 1948
		signal_handler_signal(obs->signals, "source_rename", &data);
		signal_handler_signal(source->context.signals, "rename", &data);
		calldata_free(&data);
		bfree(prev_name);
	}
1949 1950
}

1951
enum obs_source_type obs_source_get_type(const obs_source_t *source)
1952
{
J
jp9000 已提交
1953 1954
	return source ? source->info.type : OBS_SOURCE_TYPE_INPUT;
}
J
jp9000 已提交
1955

1956
const char *obs_source_get_id(const obs_source_t *source)
J
jp9000 已提交
1957 1958
{
	return source ? source->info.id : NULL;
1959
}
1960

1961 1962
static inline void render_filter_bypass(obs_source_t *target,
		gs_effect_t *effect, bool use_matrix)
1963
{
J
jp9000 已提交
1964
	const char  *tech_name = use_matrix ? "DrawMatrix" : "Draw";
1965
	gs_technique_t *tech    = gs_effect_get_technique(effect, tech_name);
1966 1967
	size_t      passes, i;

1968
	passes = gs_technique_begin(tech);
1969
	for (i = 0; i < passes; i++) {
1970
		gs_technique_begin_pass(tech, i);
1971
		obs_source_video_render(target);
1972
		gs_technique_end_pass(tech);
1973
	}
1974
	gs_technique_end(tech);
1975 1976
}

1977
static inline void render_filter_tex(gs_texture_t *tex, gs_effect_t *effect,
J
jp9000 已提交
1978
		uint32_t width, uint32_t height, bool use_matrix)
1979
{
J
jp9000 已提交
1980
	const char  *tech_name = use_matrix ? "DrawMatrix" : "Draw";
1981 1982
	gs_technique_t *tech    = gs_effect_get_technique(effect, tech_name);
	gs_eparam_t    *image   = gs_effect_get_param_by_name(effect, "image");
1983 1984
	size_t      passes, i;

1985
	gs_effect_set_texture(image, tex);
1986

1987
	passes = gs_technique_begin(tech);
1988
	for (i = 0; i < passes; i++) {
1989
		gs_technique_begin_pass(tech, i);
J
jp9000 已提交
1990
		gs_draw_sprite(tex, 0, width, height);
1991
		gs_technique_end_pass(tech);
1992
	}
1993
	gs_technique_end(tech);
1994 1995
}

1996
void obs_source_process_filter(obs_source_t *filter, gs_effect_t *effect,
J
jp9000 已提交
1997
		uint32_t width, uint32_t height, enum gs_color_format format,
1998
		enum obs_allow_direct_render allow_direct)
1999
{
2000
	obs_source_t *target, *parent;
J
jp9000 已提交
2001 2002 2003 2004 2005 2006
	uint32_t     target_flags, parent_flags;
	int          cx, cy;
	bool         use_matrix, expects_def, can_directly;

	if (!filter) return;

2007 2008
	target       = obs_filter_get_target(filter);
	parent       = obs_filter_get_parent(filter);
J
jp9000 已提交
2009 2010
	target_flags = target->info.output_flags;
	parent_flags = parent->info.output_flags;
J
jp9000 已提交
2011 2012
	cx           = get_base_width(target);
	cy           = get_base_height(target);
J
jp9000 已提交
2013 2014
	use_matrix   = !!(target_flags & OBS_SOURCE_COLOR_MATRIX);
	expects_def  = !(parent_flags & OBS_SOURCE_CUSTOM_DRAW);
2015
	can_directly = allow_direct == OBS_ALLOW_DIRECT_RENDERING;
2016 2017 2018 2019 2020

	/* 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 */
2021
	if (can_directly && expects_def && target == parent) {
J
jp9000 已提交
2022
		render_filter_bypass(target, effect, use_matrix);
2023 2024 2025
		return;
	}

J
jp9000 已提交
2026
	if (!filter->filter_texrender)
2027
		filter->filter_texrender = gs_texrender_create(format,
J
jp9000 已提交
2028 2029
				GS_ZS_NONE);

2030
	if (gs_texrender_begin(filter->filter_texrender, cx, cy)) {
2031
		gs_ortho(0.0f, (float)cx, 0.0f, (float)cy, -100.0f, 100.0f);
2032
		obs_source_video_render(target);
2033
		gs_texrender_end(filter->filter_texrender);
2034 2035 2036 2037
	}

	/* --------------------------- */

2038
	render_filter_tex(gs_texrender_get_texture(filter->filter_texrender),
J
jp9000 已提交
2039
			effect, width, height, use_matrix);
2040
}
2041

2042
signal_handler_t *obs_source_get_signal_handler(const obs_source_t *source)
2043
{
2044
	return source ? source->context.signals : NULL;
2045 2046
}

2047
proc_handler_t *obs_source_get_proc_handler(const obs_source_t *source)
2048
{
2049
	return source ? source->context.procs : NULL;
2050
}
J
jp9000 已提交
2051

2052
void obs_source_set_volume(obs_source_t *source, float volume)
J
jp9000 已提交
2053
{
J
jp9000 已提交
2054 2055
	if (source) {
		struct calldata data = {0};
2056 2057
		calldata_set_ptr(&data, "source", source);
		calldata_set_float(&data, "volume", volume);
J
jp9000 已提交
2058

2059
		signal_handler_signal(source->context.signals, "volume", &data);
2060
		signal_handler_signal(obs->signals, "source_volume", &data);
J
jp9000 已提交
2061

2062
		volume = (float)calldata_float(&data, "volume");
J
jp9000 已提交
2063 2064
		calldata_free(&data);

J
jp9000 已提交
2065
		source->user_volume = volume;
J
jp9000 已提交
2066
	}
J
jp9000 已提交
2067 2068
}

2069
static void set_tree_preset_vol(obs_source_t *parent, obs_source_t *child,
J
jp9000 已提交
2070 2071 2072 2073 2074 2075 2076 2077
		void *param)
{
	float *vol = param;
	child->present_volume = *vol;

	UNUSED_PARAMETER(parent);
}

2078
void obs_source_set_present_volume(obs_source_t *source, float volume)
J
jp9000 已提交
2079
{
2080
	if (source)
J
jp9000 已提交
2081
		source->present_volume = volume;
J
jp9000 已提交
2082 2083
}

2084
float obs_source_get_volume(const obs_source_t *source)
J
jp9000 已提交
2085
{
J
jp9000 已提交
2086
	return source ? source->user_volume : 0.0f;
J
jp9000 已提交
2087 2088
}

2089
float obs_source_get_present_volume(const obs_source_t *source)
J
jp9000 已提交
2090
{
J
jp9000 已提交
2091 2092 2093
	return source ? source->present_volume : 0.0f;
}

2094
void obs_source_set_sync_offset(obs_source_t *source, int64_t offset)
J
jp9000 已提交
2095
{
J
jp9000 已提交
2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107
	if (source) {
		struct calldata data = {0};

		calldata_set_ptr(&data, "source", source);
		calldata_set_int(&data, "offset", offset);

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

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

2110
int64_t obs_source_get_sync_offset(const obs_source_t *source)
J
jp9000 已提交
2111 2112
{
	return source ? source->sync_offset : 0;
J
jp9000 已提交
2113
}
2114 2115 2116 2117 2118 2119

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

2120
static void enum_source_tree_callback(obs_source_t *parent, obs_source_t *child,
2121 2122 2123 2124
		void *param)
{
	struct source_enum_data *data = param;

J
jp9000 已提交
2125 2126
	if (child->info.enum_sources) {
		if (child->context.data) {
2127 2128
			child->info.enum_sources(child->context.data,
					enum_source_tree_callback, data);
J
jp9000 已提交
2129
		}
2130 2131 2132 2133 2134
	}

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

2135
void obs_source_enum_sources(obs_source_t *source,
2136 2137 2138
		obs_source_enum_proc_t enum_callback,
		void *param)
{
J
jp9000 已提交
2139
	if (!source_valid(source) || !source->info.enum_sources)
2140 2141 2142 2143
		return;

	obs_source_addref(source);

2144
	source->info.enum_sources(source->context.data, enum_callback, param);
2145 2146 2147 2148

	obs_source_release(source);
}

2149
void obs_source_enum_tree(obs_source_t *source,
2150 2151 2152 2153 2154
		obs_source_enum_proc_t enum_callback,
		void *param)
{
	struct source_enum_data data = {enum_callback, param};

J
jp9000 已提交
2155
	if (!source_valid(source) || !source->info.enum_sources)
2156 2157 2158 2159
		return;

	obs_source_addref(source);

2160 2161
	source->info.enum_sources(source->context.data,
			enum_source_tree_callback,
2162 2163 2164 2165
			&data);

	obs_source_release(source);
}
2166

J
jp9000 已提交
2167 2168 2169 2170 2171 2172 2173
struct descendant_info {
	bool exists;
	obs_source_t *target;
};

static void check_descendant(obs_source_t *parent, obs_source_t *child,
		void *param)
2174
{
J
jp9000 已提交
2175 2176 2177 2178 2179 2180 2181
	struct descendant_info *info = param;
	if (child == info->target || parent == info->target)
		info->exists = true;
}

bool obs_source_add_child(obs_source_t *parent, obs_source_t *child)
{
2182 2183
	struct descendant_info info = {false, parent};
	if (!parent || !child || parent == child) return false;
J
jp9000 已提交
2184

2185
	obs_source_enum_tree(child, check_descendant, &info);
J
jp9000 已提交
2186 2187
	if (info.exists)
		return false;
2188

2189 2190 2191 2192 2193
	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 已提交
2194 2195

	return true;
2196 2197
}

2198
void obs_source_remove_child(obs_source_t *parent, obs_source_t *child)
2199 2200 2201
{
	if (!parent || !child) return;

2202 2203 2204 2205 2206
	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);
	}
2207
}
J
jp9000 已提交
2208

2209
void obs_source_save(obs_source_t *source)
2210
{
2211
	if (!source_valid(source) || !source->info.save) return;
2212 2213 2214
	source->info.save(source->context.data, source->context.settings);
}

2215
void obs_source_load(obs_source_t *source)
2216
{
2217
	if (!source_valid(source) || !source->info.load) return;
2218 2219
	source->info.load(source->context.data, source->context.settings);
}
J
jp9000 已提交
2220

J
jp9000 已提交
2221 2222
bool obs_source_active(const obs_source_t *source)
{
2223
	return source ? source->activate_refs != 0 : false;
J
jp9000 已提交
2224 2225
}

J
jp9000 已提交
2226 2227 2228 2229 2230
bool obs_source_showing(const obs_source_t *source)
{
	return source ? source->show_refs != 0 : false;
}

J
jp9000 已提交
2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252
static inline void signal_flags_updated(obs_source_t *source)
{
	struct calldata data = {0};

	calldata_set_ptr(&data, "source", source);
	calldata_set_int(&data, "flags", source->flags);

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

	calldata_free(&data);
}

void obs_source_set_flags(obs_source_t *source, uint32_t flags)
{
	if (!source) return;

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

2253 2254 2255 2256 2257 2258 2259
void obs_source_set_default_flags(obs_source_t *source, uint32_t flags)
{
	if (!source) return;

	source->default_flags = flags;
}

J
jp9000 已提交
2260 2261 2262 2263
uint32_t obs_source_get_flags(const obs_source_t *source)
{
	return source ? source->flags : 0;
}
J
jp9000 已提交
2264

2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295
void obs_source_set_audio_mixers(obs_source_t *source, uint32_t mixers)
{
	struct calldata data = {0};
	uint32_t cur_mixers;

	if (!source) return;
	if ((source->info.output_flags & OBS_SOURCE_AUDIO) == 0) return;

	cur_mixers = audio_line_get_mixers(source->audio_line);
	if (cur_mixers == mixers)
		return;

	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");
	calldata_free(&data);

	audio_line_set_mixers(source->audio_line, mixers);
}

uint32_t obs_source_get_audio_mixers(const obs_source_t *source)
{
	if (!source) return 0;
	if ((source->info.output_flags & OBS_SOURCE_AUDIO) == 0) return 0;

	return audio_line_get_mixers(source->audio_line);
}

J
jp9000 已提交
2296 2297 2298 2299
void obs_source_draw_set_color_matrix(const struct matrix4 *color_matrix,
		const struct vec3 *color_range_min,
		const struct vec3 *color_range_max)
{
2300 2301 2302 2303 2304 2305
	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 已提交
2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366
	gs_effect_t *effect = gs_get_effect();
	gs_eparam_t *matrix;
	gs_eparam_t *range_min;
	gs_eparam_t *range_max;

	if (!effect) {
		blog(LOG_WARNING, "obs_source_draw_set_color_matrix: NULL "
				"effect");
		return;
	}

	if (!color_matrix) {
		blog(LOG_WARNING, "obs_source_draw_set_color_matrix: NULL "
				"color_matrix");
		return;
	}

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

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

	gs_effect_set_matrix4(matrix, color_matrix);
	gs_effect_set_val(range_min, color_range_min, sizeof(float)*3);
	gs_effect_set_val(range_max, color_range_max, sizeof(float)*3);
}

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

	if (!effect) {
		blog(LOG_WARNING, "obs_source_draw: NULL effect");
		return;
	}

	if (!texture) {
		blog(LOG_WARNING, "obs_source_draw: NULL texture");
		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();
}
2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430

static inline float get_transition_volume(obs_source_t *source,
		obs_source_t *child)
{
	if (source && child && source->info.get_transition_volume)
		return source->info.get_transition_volume(source->context.data,
				child);
	return 0.0f;
}

static float obs_source_get_target_volume_refs(obs_source_t *source,
		obs_source_t *target, int refs);

struct base_vol_enum_info {
	obs_source_t *target;
	float vol;
};

static void get_transition_child_vol(obs_source_t *parent, obs_source_t *child,
		void *param)
{
	struct base_vol_enum_info *info = param;
	float vol = obs_source_get_target_volume(child, info->target);

	info->vol += vol * get_transition_volume(parent, child);
}

static void get_source_base_vol(obs_source_t *parent, obs_source_t *child,
		void *param)
{
	struct base_vol_enum_info *info = param;
	float vol = obs_source_get_target_volume(child, info->target);

	if (vol > info->vol)
		info->vol = vol;

	UNUSED_PARAMETER(parent);
}

/*
 * This traverses a source tree for any references to a particular source.
 * If the source is found, it'll just return 1.0.  However, if the source
 * exists within some transition somewhere, the transition source will be able
 * to control what the volume of the source will be.  If the source is also
 * active outside the transition, then it'll just use 1.0.
 */
float obs_source_get_target_volume(obs_source_t *source, obs_source_t *target)
{
	struct base_vol_enum_info info = {target, 0.0f};
	bool transition = source->info.type == OBS_SOURCE_TYPE_TRANSITION;

	if (source == target)
		return 1.0f;

	if (source->info.enum_sources) {
		source->info.enum_sources(source->context.data,
				transition ?
					get_transition_child_vol :
					get_source_base_vol,
				&info);
	}

	return info.vol;
}
2431 2432 2433 2434 2435 2436 2437 2438 2439 2440

void obs_source_inc_showing(obs_source_t *source)
{
	obs_source_activate(source, AUX_VIEW);
}

void obs_source_dec_showing(obs_source_t *source)
{
	obs_source_deactivate(source, AUX_VIEW);
}
2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456

void obs_source_enum_filters(obs_source_t *source,
		obs_source_enum_proc_t callback, void *param)
{
	if (!source || !callback)
		return;

	pthread_mutex_lock(&source->filter_mutex);

	for (size_t i = 0; i < source->filters.num; i++) {
		struct obs_source *filter = source->filters.array[i];
		callback(source, filter, param);
	}

	pthread_mutex_unlock(&source->filter_mutex);
}
2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480

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

	if (!source || !name)
		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;
}