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

    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 "media-io/format-conversion.h"
19
#include "util/platform.h"
20
#include "callback/calldata.h"
21 22
#include "graphics/matrix3.h"
#include "graphics/vec3.h"
23

J
jp9000 已提交
24 25 26
#include "obs.h"
#include "obs-data.h"

27 28
static void obs_source_destroy(obs_source_t source);

29
bool load_source_info(void *module, const char *module_name,
30
		const char *id, struct source_info *info)
J
jp9000 已提交
31
{
32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
	LOAD_MODULE_SUBFUNC(getname, true);
	LOAD_MODULE_SUBFUNC(create, true);
	LOAD_MODULE_SUBFUNC(destroy, true);
	LOAD_MODULE_SUBFUNC(get_output_flags, true);

	LOAD_MODULE_SUBFUNC(update, false);
	LOAD_MODULE_SUBFUNC(activate, false);
	LOAD_MODULE_SUBFUNC(deactivate, false);
	LOAD_MODULE_SUBFUNC(video_tick, false);
	LOAD_MODULE_SUBFUNC(video_render, false);
	LOAD_MODULE_SUBFUNC(getwidth, false);
	LOAD_MODULE_SUBFUNC(getheight, false);
	LOAD_MODULE_SUBFUNC(getparam, false);
	LOAD_MODULE_SUBFUNC(setparam, false);
	LOAD_MODULE_SUBFUNC(filter_video, false);
	LOAD_MODULE_SUBFUNC(filter_audio, false);

	info->id = id;
J
jp9000 已提交
50 51 52
	return true;
}

53
static inline const struct source_info *find_source(struct darray *list,
54
		const char *id)
J
jp9000 已提交
55 56 57 58 59 60
{
	size_t i;
	struct source_info *array = list->array;

	for (i = 0; i < list->num; i++) {
		struct source_info *info = array+i;
61
		if (strcmp(info->id, id) == 0)
J
jp9000 已提交
62 63 64 65 66 67
			return info;
	}

	return NULL;
}

68
static const struct source_info *get_source_info(enum obs_source_type type,
69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85
		const char *id)
{
	struct darray *list = NULL;

	switch (type) {
	case SOURCE_INPUT:      list = &obs->input_types.da; break;
	case SOURCE_FILTER:     list = &obs->filter_types.da; break;
	case SOURCE_TRANSITION: list = &obs->transition_types.da; break;
	case SOURCE_SCENE:
	default:
		blog(LOG_WARNING, "get_source_info: invalid source type");
		return NULL;
	}

	return find_source(list, id);
}

J
jp9000 已提交
86
bool obs_source_init_handlers(struct obs_source *source)
87 88 89 90 91 92 93 94 95
{
	source->signals = signal_handler_create();
	if (!source->signals)
		return false;

	source->procs = proc_handler_create();
	return (source->procs != NULL);
}

96 97 98 99 100 101 102
const char *obs_source_getdisplayname(enum obs_source_type type,
		const char *id, const char *locale)
{
	const struct source_info *info = get_source_info(type, id);
	return (info != NULL) ? info->getname(locale) : NULL;
}

103
/* internal initialization */
104 105
bool obs_source_init(struct obs_source *source, const char *settings,
		const struct source_info *info)
J
jp9000 已提交
106
{
107 108
	uint32_t flags = info->get_output_flags(source->data);

109
	source->refs = 1;
J
jp9000 已提交
110
	source->volume = 1.0f;
111 112 113 114 115 116 117 118 119 120 121 122
	pthread_mutex_init_value(&source->filter_mutex);
	pthread_mutex_init_value(&source->video_mutex);
	pthread_mutex_init_value(&source->audio_mutex);
	dstr_copy(&source->settings, settings);
	memcpy(&source->callbacks, info, sizeof(struct source_info));

	if (pthread_mutex_init(&source->filter_mutex, NULL) != 0)
		return false;
	if (pthread_mutex_init(&source->audio_mutex, NULL) != 0)
		return false;
	if (pthread_mutex_init(&source->video_mutex, NULL) != 0)
		return false;
J
jp9000 已提交
123

124
	if (flags & SOURCE_AUDIO) {
J
jp9000 已提交
125 126
		source->audio_line = audio_output_createline(obs->audio.audio,
				source->name);
127 128
		if (!source->audio_line) {
			blog(LOG_ERROR, "Failed to create audio line for "
J
jp9000 已提交
129
			                "source '%s'", source->name);
130 131 132
			return false;
		}
	}
133

134
	return true;
J
jp9000 已提交
135 136
}

137 138 139 140 141 142 143 144 145 146 147
static inline void obs_source_dosignal(struct obs_source *source,
		const char *signal)
{
	struct calldata data;

	calldata_init(&data);
	calldata_setptr(&data, "source", source);
	signal_handler_signal(obs->signals, signal, &data);
	calldata_free(&data);
}

148 149
obs_source_t obs_source_create(enum obs_source_type type, const char *id,
		const char *name, const char *settings)
J
jp9000 已提交
150 151 152
{
	struct obs_source *source;

153
	const struct source_info *info = get_source_info(type, id);
J
jp9000 已提交
154
	if (!info) {
155
		blog(LOG_WARNING, "Source '%s' not found", id);
J
jp9000 已提交
156 157 158 159
		return NULL;
	}

	source = bmalloc(sizeof(struct obs_source));
160 161
	memset(source, 0, sizeof(struct obs_source));

162 163 164
	if (!obs_source_init_handlers(source))
		goto fail;

165
	source->name = bstrdup(name);
166
	source->type = type;
J
jp9000 已提交
167
	source->data = info->create(settings, source);
168 169 170 171 172
	if (!source->data)
		goto fail;

	if (!obs_source_init(source, settings, info))
		goto fail;
J
jp9000 已提交
173

J
jp9000 已提交
174
	obs_source_dosignal(source, "source-create");
J
jp9000 已提交
175
	return source;
176 177 178 179 180

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

183
static void obs_source_destroy(obs_source_t source)
J
jp9000 已提交
184
{
185
	size_t i;
186

187 188
	obs_source_dosignal(source, "source-destroy");

189 190
	if (source->filter_parent)
		obs_source_filter_remove(source->filter_parent, source);
191

192 193
	for (i = 0; i < source->filters.num; i++)
		obs_source_release(source->filters.array[i]);
194

195 196
	for (i = 0; i < source->video_frames.num; i++)
		source_frame_destroy(source->video_frames.array[i]);
197

198 199 200
	gs_entercontext(obs->video.graphics);
	texture_destroy(source->output_texture);
	gs_leavecontext();
J
jp9000 已提交
201

202 203
	if (source->data)
		source->callbacks.destroy(source->data);
204

205 206 207 208
	bfree(source->audio_data.data);
	audio_line_destroy(source->audio_line);
	audio_resampler_destroy(source->resampler);

209 210 211
	proc_handler_destroy(source->procs);
	signal_handler_destroy(source->signals);

212 213 214 215 216 217
	da_free(source->video_frames);
	da_free(source->filters);
	pthread_mutex_destroy(&source->filter_mutex);
	pthread_mutex_destroy(&source->audio_mutex);
	pthread_mutex_destroy(&source->video_mutex);
	dstr_free(&source->settings);
218
	bfree(source->name);
219 220 221
	bfree(source);
}

222
int obs_source_addref(obs_source_t source)
223 224 225
{
	assert(source != NULL);
	if (!source)
226
		return 0;
227

228
	return ++source->refs;
229 230
}

231
int obs_source_release(obs_source_t source)
232
{
233 234
	int refs;

235 236
	assert(source != NULL);
	if (!source)
237
		return 0;
238

239 240
	refs = --source->refs;
	if (refs == 0)
241
		obs_source_destroy(source);
242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261

	return refs;
}

void obs_source_remove(obs_source_t source)
{
	struct obs_data *data = &obs->data;
	size_t id;

	source->removed = true;

	pthread_mutex_lock(&data->sources_mutex);

	id = da_find(data->sources, &source, 0);
	if (id != DARRAY_INVALID) {
		da_erase_item(data->sources, &source);
		obs_source_release(source);
	}

	pthread_mutex_unlock(&data->sources_mutex);
262 263 264 265 266
}

bool obs_source_removed(obs_source_t source)
{
	return source->removed;
J
jp9000 已提交
267 268
}

269
uint32_t obs_source_get_output_flags(obs_source_t source)
J
jp9000 已提交
270 271 272 273
{
	return source->callbacks.get_output_flags(source->data);
}

274
void obs_source_update(obs_source_t source, const char *settings)
J
jp9000 已提交
275
{
276 277
	if (source->callbacks.update)
		source->callbacks.update(source->data, settings);
J
jp9000 已提交
278 279
}

280
void obs_source_activate(obs_source_t source)
J
jp9000 已提交
281 282 283 284 285
{
	if (source->callbacks.activate)
		source->callbacks.activate(source->data);
}

286
void obs_source_deactivate(obs_source_t source)
J
jp9000 已提交
287 288 289 290 291
{
	if (source->callbacks.deactivate)
		source->callbacks.deactivate(source->data);
}

292
void obs_source_video_tick(obs_source_t source, float seconds)
J
jp9000 已提交
293 294 295 296 297
{
	if (source->callbacks.video_tick)
		source->callbacks.video_tick(source->data, seconds);
}

298 299
static inline uint64_t conv_frames_to_time(obs_source_t source, size_t frames)
{
J
jp9000 已提交
300 301 302 303 304
	const struct audio_output_info *info;
	double sps_to_ns;

	info = audio_output_getinfo(obs->audio.audio);
	sps_to_ns = 1000000000.0 / (double)info->samples_per_sec;
305 306 307
	return (uint64_t)((double)frames * sps_to_ns);
}

308
/* maximum "direct" timestamp variance in nanoseconds */
309
#define MAX_TS_VAR         5000000000ULL
310
/* maximum time that timestamp can jump in nanoseconds */
311
#define MAX_TIMESTAMP_JUMP 2000000000ULL
312 313 314 315 316 317

static inline void reset_audio_timing(obs_source_t source, uint64_t timetamp)
{
	source->timing_set    = true;
	source->timing_adjust = os_gettime_ns() - timetamp;
}
318

319 320 321 322 323 324 325 326 327 328 329 330 331 332 333
static inline void handle_ts_jump(obs_source_t source, uint64_t ts,
		uint64_t diff)
{
	uint32_t flags = source->callbacks.get_output_flags(source->data);

	blog(LOG_DEBUG, "Timestamp for source '%s' jumped by '%lld', "
	                "resetting audio timing", source->name, diff);

	/* if has video, ignore audio data until reset */
	if (flags & SOURCE_ASYNC_VIDEO)
		source->audio_reset_ref--;
	else 
		reset_audio_timing(source, ts);
}

334 335 336 337
static void source_output_audio_line(obs_source_t source,
		const struct audio_data *data)
{
	struct audio_data in = *data;
338
	uint64_t diff;
339 340

	if (!source->timing_set) {
341
		reset_audio_timing(source, in.timestamp);
342 343

		/* detects 'directly' set timestamps as long as they're within
344
		 * a certain threshold */
345
		if ((source->timing_adjust + MAX_TS_VAR) < MAX_TS_VAR * 2)
346
			source->timing_adjust = 0;
347

348
	} else {
349
		diff = in.timestamp - source->next_audio_ts_min;
350 351 352

		/* don't need signed because negative will trigger it
		 * regardless, which is what we want */
353 354
		if (diff > MAX_TIMESTAMP_JUMP)
			handle_ts_jump(source, in.timestamp, diff);
355 356
	}

357
	source->next_audio_ts_min = in.timestamp +
358 359
		conv_frames_to_time(source, in.frames);

360 361 362
	if (source->audio_reset_ref != 0)
		return;

363
	in.timestamp += source->timing_adjust;
J
jp9000 已提交
364
	in.volume = source->volume;
365 366 367
	audio_line_output(source->audio_line, &in);
}

368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384
static bool set_texture_size(obs_source_t source, struct source_frame *frame)
{
	if (source->output_texture) {
		uint32_t width  = texture_getwidth(source->output_texture);
		uint32_t height = texture_getheight(source->output_texture);

		if (width == frame->width && height == frame->height)
			return true;
	}

	texture_destroy(source->output_texture);
	source->output_texture = gs_create_texture(frame->width, frame->height,
			GS_RGBA, 1, NULL, GS_DYNAMIC);

	return source->output_texture != NULL;
}

385 386 387 388 389 390 391 392
enum convert_type {
	CONVERT_NONE,
	CONVERT_NV12,
	CONVERT_420,
	CONVERT_422_U,
	CONVERT_422_Y,
};

393
static inline enum convert_type get_convert_type(enum video_format format)
394
{
395
	switch (format) {
396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418
	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;

	case VIDEO_FORMAT_UNKNOWN:
	case VIDEO_FORMAT_YUVX:
	case VIDEO_FORMAT_UYVX:
	case VIDEO_FORMAT_RGBA:
	case VIDEO_FORMAT_BGRA:
	case VIDEO_FORMAT_BGRX:
		return CONVERT_NONE;
	}

	return CONVERT_NONE;
}

419
static inline bool is_yuv(enum video_format format)
420
{
421
	switch (format) {
422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443
	case VIDEO_FORMAT_I420:
	case VIDEO_FORMAT_NV12:
	case VIDEO_FORMAT_YVYU:
	case VIDEO_FORMAT_YUY2:
	case VIDEO_FORMAT_UYVY:
	case VIDEO_FORMAT_YUVX:
	case VIDEO_FORMAT_UYVX:
		return true;
	case VIDEO_FORMAT_UNKNOWN:
	case VIDEO_FORMAT_RGBA:
	case VIDEO_FORMAT_BGRA:
	case VIDEO_FORMAT_BGRX:
		return false;
	}

	return false;
}

static bool upload_frame(texture_t tex, const struct source_frame *frame)
{
	void *ptr;
	uint32_t row_bytes;
444
	enum convert_type type = get_convert_type(frame->format);
445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473

	if (type == CONVERT_NONE) {
		texture_setimage(tex, frame->data, frame->row_bytes, false);
		return true;
	}

	if (!texture_map(tex, &ptr, &row_bytes))
		return false;

	if (type == CONVERT_420)
		decompress_420(frame->data, frame->width, frame->height,
				frame->row_bytes, 0, frame->height, ptr);

	else if (type == CONVERT_NV12)
		decompress_nv12(frame->data, frame->width, frame->height,
				frame->row_bytes, 0, frame->height, ptr);

	else if (type == CONVERT_422_Y)
		decompress_422(frame->data, frame->width, frame->height,
				frame->row_bytes, 0, frame->height, ptr, true);

	else if (type == CONVERT_422_U)
		decompress_422(frame->data, frame->width, frame->height,
				frame->row_bytes, 0, frame->height, ptr, false);

	texture_unmap(tex);
	return true;
}

474 475
static void obs_source_draw_texture(texture_t tex, struct source_frame *frame)
{
476
	effect_t    effect = obs->video.default_effect;
477
	bool        yuv    = is_yuv(frame->format);
478
	const char  *type  = yuv ? "DrawYUV" : "DrawRGB";
479
	technique_t tech;
480
	eparam_t    param;
481

482 483
	if (!upload_frame(tex, frame))
		return;
484 485 486 487 488

	tech = effect_gettechnique(effect, type);
	technique_begin(tech);
	technique_beginpass(tech, 0);

489
	if (yuv) {
490 491 492 493 494 495 496 497
		param = effect_getparambyname(effect, "yuv_matrix");
		effect_setval(effect, param, frame->yuv_matrix,
				sizeof(float) * 16);
	}

	param = effect_getparambyname(effect, "diffuse");
	effect_settexture(effect, param, tex);

498
	gs_draw_sprite(tex, frame->flip ? GS_FLIP_V : 0, 0, 0);
499 500 501 502 503

	technique_endpass(tech);
	technique_end(tech);
}

504 505 506 507 508 509
static void obs_source_render_async_video(obs_source_t source)
{
	struct source_frame *frame = obs_source_getframe(source);
	if (!frame)
		return;

J
jp9000 已提交
510
	if (set_texture_size(source, frame))
511
		obs_source_draw_texture(source->output_texture, frame);
512 513 514 515

	obs_source_releaseframe(source, frame);
}

516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551
static inline void obs_source_render_filters(obs_source_t source)
{
	source->rendering_filter = true;
	obs_source_video_render(source->filters.array[0]);
	source->rendering_filter = false;
}

static inline void obs_source_default_render(obs_source_t source, bool yuv)
{
	effect_t    effect     = obs->video.default_effect;
	const char  *tech_name = yuv ? "DrawYUV" : "DrawRGB";
	technique_t tech       = effect_gettechnique(effect, tech_name);
	size_t      passes, i;

	passes = technique_begin(tech);
	for (i = 0; i < passes; i++) {
		technique_beginpass(tech, i);
		source->callbacks.video_render(source->data);
		technique_endpass(tech);
	}
	technique_end(tech);
}

static inline void obs_source_main_render(obs_source_t source)
{
	uint32_t flags = source->callbacks.get_output_flags(source->data);
	bool default_effect = !source->filter_parent &&
	                      source->filters.num == 0 &&
	                      (flags & SOURCE_DEFAULT_EFFECT) != 0;

	if (default_effect)
		obs_source_default_render(source, (flags & SOURCE_YUV) != 0);
	else
		source->callbacks.video_render(source->data);
}

552
void obs_source_video_render(obs_source_t source)
J
jp9000 已提交
553 554
{
	if (source->callbacks.video_render) {
555 556 557 558
		if (source->filters.num && !source->rendering_filter)
			obs_source_render_filters(source);
		else
			obs_source_main_render(source);
559 560 561 562 563 564

	} else if (source->filter_target) {
		obs_source_video_render(source->filter_target);

	} else {
		obs_source_render_async_video(source);
J
jp9000 已提交
565 566 567
	}
}

J
jp9000 已提交
568
uint32_t obs_source_getwidth(obs_source_t source)
J
jp9000 已提交
569 570 571 572 573 574
{
	if (source->callbacks.getwidth)
		return source->callbacks.getwidth(source->data);
	return 0;
}

J
jp9000 已提交
575
uint32_t obs_source_getheight(obs_source_t source)
J
jp9000 已提交
576 577 578 579 580 581
{
	if (source->callbacks.getheight)
		return source->callbacks.getheight(source->data);
	return 0;
}

582
size_t obs_source_getparam(obs_source_t source, const char *param, void *buf,
J
jp9000 已提交
583 584 585 586 587 588 589 590
		size_t buf_size)
{
	if (source->callbacks.getparam)
		return source->callbacks.getparam(source->data, param, buf,
				buf_size);
	return 0;
}

591 592
void obs_source_setparam(obs_source_t source, const char *param,
		const void *data, size_t size)
J
jp9000 已提交
593 594 595 596 597
{
	if (source->callbacks.setparam)
		source->callbacks.setparam(source->data, param, data, size);
}

598 599 600 601 602
obs_source_t obs_filter_getparent(obs_source_t filter)
{
	return filter->filter_parent;
}

603
obs_source_t obs_filter_gettarget(obs_source_t filter)
J
jp9000 已提交
604 605 606 607
{
	return filter->filter_target;
}

608
void obs_source_filter_add(obs_source_t source, obs_source_t filter)
J
jp9000 已提交
609
{
610 611
	pthread_mutex_lock(&source->filter_mutex);

J
jp9000 已提交
612
	if (da_find(source->filters, &filter, 0) != DARRAY_INVALID) {
J
jp9000 已提交
613 614 615 616 617 618
		blog(LOG_WARNING, "Tried to add a filter that was already "
		                  "present on the source");
		return;
	}

	if (source->filters.num) {
619
		obs_source_t *back = da_end(source->filters);
J
jp9000 已提交
620 621 622 623
		(*back)->filter_target = filter;
	}

	da_push_back(source->filters, &filter);
624 625 626 627

	pthread_mutex_unlock(&source->filter_mutex);

	filter->filter_parent = source;
J
jp9000 已提交
628 629 630
	filter->filter_target = source;
}

631
void obs_source_filter_remove(obs_source_t source, obs_source_t filter)
J
jp9000 已提交
632
{
633 634 635 636 637
	size_t idx;

	pthread_mutex_lock(&source->filter_mutex);

	idx = da_find(source->filters, &filter, 0);
J
jp9000 已提交
638
	if (idx == DARRAY_INVALID)
J
jp9000 已提交
639 640 641
		return;

	if (idx > 0) {
642
		obs_source_t prev = source->filters.array[idx-1];
J
jp9000 已提交
643 644 645 646
		prev->filter_target = filter->filter_target;
	}

	da_erase(source->filters, idx);
647 648 649 650

	pthread_mutex_unlock(&source->filter_mutex);

	filter->filter_parent = NULL;
J
jp9000 已提交
651 652 653
	filter->filter_target = NULL;
}

654
void obs_source_filter_setorder(obs_source_t source, obs_source_t filter,
J
jp9000 已提交
655 656 657 658
		enum order_movement movement)
{
	size_t idx = da_find(source->filters, &filter, 0);
	size_t i;
J
jp9000 已提交
659
	if (idx == DARRAY_INVALID)
J
jp9000 已提交
660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682
		return;

	if (movement == ORDER_MOVE_UP) {
		if (idx == source->filters.num-1)
			return;
		da_move_item(source->filters, idx, idx+1);

	} else if (movement == ORDER_MOVE_DOWN) {
		if (idx == 0)
			return;
		da_move_item(source->filters, idx, idx-1);

	} else if (movement == ORDER_MOVE_TOP) {
		if (idx == source->filters.num-1)
			return;
		da_move_item(source->filters, idx, source->filters.num-1);

	} else if (movement == ORDER_MOVE_BOTTOM) {
		if (idx == 0)
			return;
		da_move_item(source->filters, idx, 0);
	}

683
	/* reorder filter targets, not the nicest way of dealing with things */
J
jp9000 已提交
684
	for (i = 0; i < source->filters.num; i++) {
685
		obs_source_t next_filter = (i == source->filters.num-1) ?
J
jp9000 已提交
686 687 688 689 690
			source : source->filters.array[idx+1];
		source->filters.array[i]->filter_target = next_filter;
	}
}

691
const char *obs_source_getsettings(obs_source_t source)
J
jp9000 已提交
692 693 694 695
{
	return source->settings.array;
}

696
void obs_source_savesettings(obs_source_t source, const char *settings)
J
jp9000 已提交
697 698 699 700
{
	dstr_copy(&source->settings, settings);
}

701 702
static inline struct source_frame *filter_async_video(obs_source_t source,
		struct source_frame *in)
703 704 705 706 707 708 709 710 711 712 713 714 715 716
{
	size_t i;
	for (i = source->filters.num; i > 0; i--) {
		struct obs_source *filter = source->filters.array[i-1];
		if (filter->callbacks.filter_video) {
			in = filter->callbacks.filter_video(filter->data, in);
			if (!in)
				return NULL;
		}
	}

	return in;
}

717 718
static inline struct source_frame *cache_video(obs_source_t source,
		const struct source_frame *frame)
719
{
720 721 722 723 724 725
	/* TODO: use an actual cache */
	struct source_frame *new_frame = bmalloc(sizeof(struct source_frame));
	memcpy(new_frame, frame, sizeof(struct source_frame));
	new_frame->data = bmalloc(frame->row_bytes * frame->height);

	return new_frame;
726 727 728
}

void obs_source_output_video(obs_source_t source,
729
		const struct source_frame *frame)
730
{
731
	struct source_frame *output = cache_video(source, frame);
732 733 734 735 736

	pthread_mutex_lock(&source->filter_mutex);
	output = filter_async_video(source, output);
	pthread_mutex_unlock(&source->filter_mutex);

737 738 739 740 741
	if (output) {
		pthread_mutex_lock(&source->video_mutex);
		da_push_back(source->video_frames, &output);
		pthread_mutex_unlock(&source->video_mutex);
	}
742 743
}

744 745
static inline struct filtered_audio *filter_async_audio(obs_source_t source,
		struct filtered_audio *in)
746 747 748 749 750 751 752 753 754 755 756 757 758 759
{
	size_t i;
	for (i = source->filters.num; i > 0; i--) {
		struct obs_source *filter = source->filters.array[i-1];
		if (filter->callbacks.filter_audio) {
			in = filter->callbacks.filter_audio(filter->data, in);
			if (!in)
				return NULL;
		}
	}

	return in;
}

760
static inline void reset_resampler(obs_source_t source,
761 762
		const struct source_audio *audio)
{
J
jp9000 已提交
763
	const struct audio_output_info *obs_info;
764 765
	struct resample_info output_info;

766 767
	obs_info = audio_output_getinfo(obs->audio.audio);

768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794
	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;

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

	audio_resampler_destroy(source->resampler);
	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");
}

static inline void copy_audio_data(obs_source_t source,
		const void *data, uint32_t frames, uint64_t timestamp)
{
795
	size_t blocksize = audio_output_blocksize(obs->audio.audio);
796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834
	size_t size = (size_t)frames * blocksize;

	/* ensure audio storage capacity */
	if (source->audio_storage_size < size) {
		bfree(source->audio_data.data);
		source->audio_data.data = bmalloc(size);
		source->audio_storage_size = size;
	}

	source->audio_data.frames = frames;
	source->audio_data.timestamp = timestamp;
	memcpy(source->audio_data.data, data, size);
}

/* resamples/remixes new audio to the designated main audio output format */
static void process_audio(obs_source_t source, const struct source_audio *audio)
{
	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) {
		void *output;
		uint32_t frames;
		uint64_t offset;

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

		copy_audio_data(source, output, frames,
				audio->timestamp - offset);
	} else {
		copy_audio_data(source, audio->data, audio->frames,
				audio->timestamp);
	}
835 836 837 838 839 840
}

void obs_source_output_audio(obs_source_t source,
		const struct source_audio *audio)
{
	uint32_t flags = obs_source_get_output_flags(source);
841
	size_t blocksize = audio_output_blocksize(obs->audio.audio);
842
	struct filtered_audio *output;
843

844
	process_audio(source, audio);
845 846

	pthread_mutex_lock(&source->filter_mutex);
847
	output = filter_async_audio(source, &source->audio_data);
848 849 850 851

	if (output) {
		pthread_mutex_lock(&source->audio_mutex);

852 853
		/* wait for video to start before outputting any audio so we
		 * have a base for sync */
854
		if (source->timing_set || (flags & SOURCE_ASYNC_VIDEO) == 0) {
855 856 857 858 859
			struct audio_data data;
			data.data      = output->data;
			data.frames    = output->frames;
			data.timestamp = output->timestamp;
			source_output_audio_line(source, &data);
860 861 862 863 864 865 866 867
		}

		pthread_mutex_unlock(&source->audio_mutex);
	}

	pthread_mutex_unlock(&source->filter_mutex);
}

868 869 870 871 872 873
static inline bool frame_out_of_bounds(obs_source_t source, uint64_t ts)
{
	return ((ts - source->last_frame_ts) > MAX_TIMESTAMP_JUMP);
}

static inline struct source_frame *get_closest_frame(obs_source_t source,
J
jp9000 已提交
874
		uint64_t sys_time, int *audio_time_refs)
875 876 877 878 879 880 881 882 883 884
{
	struct source_frame *next_frame = source->video_frames.array[0];
	struct source_frame *frame      = NULL;
	uint64_t sys_offset = sys_time - source->last_sys_timestamp;
	uint64_t frame_time = next_frame->timestamp;
	uint64_t frame_offset = 0;

	/* account for timestamp invalidation */
	if (frame_out_of_bounds(source, frame_time)) {
		source->last_frame_ts = next_frame->timestamp;
J
jp9000 已提交
885
		(*audio_time_refs)++;
886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905
	} else {
		frame_offset = frame_time - source->last_frame_ts;
		source->last_frame_ts += sys_offset;
	}

	while (frame_offset <= sys_offset) {
		source_frame_destroy(frame);

		frame = next_frame;
		da_erase(source->video_frames, 0);

		if (!source->video_frames.num)
			break;

		next_frame = source->video_frames.array[0];

		/* more timestamp checking and compensating */
		if ((next_frame->timestamp - frame_time) > MAX_TIMESTAMP_JUMP) {
			source->last_frame_ts =
				next_frame->timestamp - frame_offset;
J
jp9000 已提交
906
			(*audio_time_refs)++;
907 908 909 910 911 912 913 914 915
		}

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

	return frame;
}

916
/*
917 918
 * Ensures that cached frames are displayed on time.  If multiple frames
 * were cached between renders, then releases the unnecessary frames and uses
919 920
 * the frame with the closest timing to ensure sync.  Also ensures that timing
 * with audio is synchronized.
921
 */
922
struct source_frame *obs_source_getframe(obs_source_t source)
J
jp9000 已提交
923
{
924
	struct source_frame *frame = NULL;
J
jp9000 已提交
925 926
	uint64_t last_frame_time = source->last_frame_ts;
	int      audio_time_refs = 0;
927
	uint64_t sys_time;
928 929 930 931 932 933

	pthread_mutex_lock(&source->video_mutex);

	if (!source->video_frames.num)
		goto unlock;

934
	sys_time = os_gettime_ns();
935

936 937
	if (!source->last_frame_ts) {
		frame = source->video_frames.array[0];
938 939
		da_erase(source->video_frames, 0);

940
		source->last_frame_ts = frame->timestamp;
941
	} else {
J
jp9000 已提交
942 943 944 945 946 947 948 949
		frame = get_closest_frame(source, sys_time, &audio_time_refs);
	}

	/* reset timing to current system time */
	if (frame) {
		source->audio_reset_ref += audio_time_refs;
		source->timing_adjust = sys_time - frame->timestamp;
		source->timing_set = true;
950 951 952 953 954 955
	}

	source->last_sys_timestamp = sys_time;

unlock:
	pthread_mutex_unlock(&source->video_mutex);
956

957
	if (frame)
958 959
		obs_source_addref(source);

960
	return frame;
J
jp9000 已提交
961 962
}

963
void obs_source_releaseframe(obs_source_t source, struct source_frame *frame)
J
jp9000 已提交
964
{
965 966 967 968
	if (frame) {
		source_frame_destroy(frame);
		obs_source_release(source);
	}
J
jp9000 已提交
969
}
970 971 972 973 974 975 976 977 978 979 980 981

const char *obs_source_getname(obs_source_t source)
{
	return source->name;
}

void obs_source_setname(obs_source_t source, const char *name)
{
	bfree(source->name);
	source->name = bstrdup(name);
}

982
void obs_source_gettype(obs_source_t source, enum obs_source_type *type,
983 984
		const char **id)
{
985 986
	if (type) *type = source->type;
	if (id)   *id   = source->callbacks.id;
987
}
988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025

static inline void render_filter_bypass(obs_source_t target, effect_t effect,
		uint32_t width, uint32_t height, bool yuv)
{
	const char  *tech_name = yuv ? "DrawYUV" : "DrawRGB";
	technique_t tech       = effect_gettechnique(effect, tech_name);
	eparam_t    diffuse    = effect_getparambyname(effect, "diffuse");
	size_t      passes, i;

	passes = technique_begin(tech);
	for (i = 0; i < passes; i++) {
		technique_beginpass(tech, i);
		obs_source_video_render(target);
		technique_endpass(tech);
	}
	technique_end(tech);
}

static inline void render_filter_tex(texture_t tex, effect_t effect,
		uint32_t width, uint32_t height, bool yuv)
{
	const char  *tech_name = yuv ? "DrawYUV" : "DrawRGB";
	technique_t tech       = effect_gettechnique(effect, tech_name);
	eparam_t    diffuse    = effect_getparambyname(effect, "diffuse");
	size_t      passes, i;

	effect_settexture(effect, diffuse, tex);

	passes = technique_begin(tech);
	for (i = 0; i < passes; i++) {
		technique_beginpass(tech, i);
		gs_draw_sprite(tex, width, height, 0);
		technique_endpass(tech);
	}
	technique_end(tech);
}

void obs_source_process_filter(obs_source_t filter, texrender_t texrender,
1026 1027
		effect_t effect, uint32_t width, uint32_t height,
		enum allow_direct_render allow_direct)
1028 1029 1030 1031 1032 1033 1034 1035
{
	obs_source_t target       = obs_filter_gettarget(filter);
	obs_source_t parent       = obs_filter_getparent(filter);
	uint32_t     target_flags = obs_source_get_output_flags(target);
	uint32_t     parent_flags = obs_source_get_output_flags(parent);
	int          cx           = obs_source_getwidth(target);
	int          cy           = obs_source_getheight(target);
	bool         yuv          = (target_flags & SOURCE_YUV) != 0;
1036 1037
	bool         expects_def  = (parent_flags & SOURCE_DEFAULT_EFFECT) != 0;
	bool         can_directly = allow_direct == ALLOW_DIRECT_RENDERING;
1038 1039 1040 1041 1042

	/* 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 */
1043
	if (can_directly && expects_def && target == parent) {
1044 1045 1046 1047 1048 1049
		render_filter_bypass(target, effect, width, height, yuv);
		return;
	}

	if (texrender_begin(texrender, cx, cy)) {
		gs_ortho(0.0f, (float)cx, 0.0f, (float)cy, -100.0f, 100.0f);
1050 1051 1052 1053
		if (expects_def && parent == target)
			obs_source_default_render(parent, yuv);
		else
			obs_source_video_render(target);
1054 1055 1056 1057 1058 1059 1060 1061
		texrender_end(texrender);
	}

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

	render_filter_tex(texrender_gettexture(texrender), effect,
			width, height, yuv);
}
1062 1063 1064 1065 1066 1067 1068 1069 1070 1071

signal_handler_t obs_source_signalhandler(obs_source_t source)
{
	return source->signals;
}

proc_handler_t obs_source_prochandler(obs_source_t source)
{
	return source->procs;
}
J
jp9000 已提交
1072 1073 1074 1075 1076 1077 1078 1079 1080 1081

void obs_source_setvolume(obs_source_t source, float volume)
{
	source->volume = volume;
}

float obs_source_getvolume(obs_source_t source)
{
	return source->volume;
}