obs-scene.c 66.4 KB
Newer Older
J
jp9000 已提交
1
/******************************************************************************
S
Socapex 已提交
2 3
    Copyright (C) 2013-2015 by Hugh Bailey <obs.jim@gmail.com>
                               Philippe Groarke <philippe.groarke@gmail.com>
J
jp9000 已提交
4 5 6

    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
7
    the Free Software Foundation, either version 2 of the License, or
J
jp9000 已提交
8 9 10 11 12 13 14 15 16 17 18
    (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/>.
******************************************************************************/

J
jp9000 已提交
19
#include "util/threading.h"
J
jp9000 已提交
20 21 22
#include "graphics/math-defs.h"
#include "obs-scene.h"

J
jp9000 已提交
23
static void resize_group(obs_sceneitem_t *group);
24 25
static void signal_parent(obs_scene_t *parent, const char *name,
		calldata_t *params);
J
jp9000 已提交
26 27 28 29
static void get_ungrouped_transform(obs_sceneitem_t *group,
		struct vec2 *pos,
		struct vec2 *scale,
		float *rot);
30

31 32 33 34 35 36 37 38 39 40 41
/* NOTE: For proper mutex lock order (preventing mutual cross-locks), never
 * lock the graphics mutex inside either of the scene mutexes.
 *
 * Another thing that must be done to prevent that cross-lock (and improve
 * performance), is to not create/release/update sources within the scene
 * mutexes.
 *
 * It's okay to lock the graphics mutex before locking either of the scene
 * mutexes, but not after.
 */

42 43 44
static const char *obs_scene_signals[] = {
	"void item_add(ptr scene, ptr item)",
	"void item_remove(ptr scene, ptr item)",
S
Socapex 已提交
45
	"void reorder(ptr scene)",
46
	"void item_visible(ptr scene, ptr item, bool visible)",
47 48 49
	"void item_select(ptr scene, ptr item)",
	"void item_deselect(ptr scene, ptr item)",
	"void item_transform(ptr scene, ptr item)",
50 51 52
	NULL
};

J
jp9000 已提交
53
static inline void signal_item_remove(struct obs_scene_item *item)
J
jp9000 已提交
54
{
55 56 57 58
	struct calldata params;
	uint8_t stack[128];

	calldata_init_fixed(&params, stack, sizeof(stack));
59
	calldata_set_ptr(&params, "item", item);
J
jp9000 已提交
60

61
	signal_parent(item->parent, "item_remove", &params);
J
jp9000 已提交
62 63
}

64
static const char *scene_getname(void *unused)
65
{
66
	UNUSED_PARAMETER(unused);
67
	return "Scene";
68 69
}

70
static void *scene_create(obs_data_t *settings, struct obs_source *source)
J
jp9000 已提交
71
{
P
Palana 已提交
72
	pthread_mutexattr_t attr;
73 74
	struct obs_scene *scene = bzalloc(sizeof(struct obs_scene));
	scene->source = source;
75

76
	signal_handler_add_array(obs_source_get_signal_handler(source),
77 78
			obs_scene_signals);

P
Palana 已提交
79 80 81 82
	if (pthread_mutexattr_init(&attr) != 0)
		goto fail;
	if (pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE) != 0)
		goto fail;
83 84 85 86 87 88 89 90
	if (pthread_mutex_init(&scene->audio_mutex, &attr) != 0) {
		blog(LOG_ERROR, "scene_create: Couldn't initialize audio "
				"mutex");
		goto fail;
	}
	if (pthread_mutex_init(&scene->video_mutex, &attr) != 0) {
		blog(LOG_ERROR, "scene_create: Couldn't initialize video "
				"mutex");
P
Palana 已提交
91
		goto fail;
92
	}
J
jp9000 已提交
93

J
jp9000 已提交
94
	UNUSED_PARAMETER(settings);
J
jp9000 已提交
95
	return scene;
P
Palana 已提交
96 97 98 99 100

fail:
	pthread_mutexattr_destroy(&attr);
	bfree(scene);
	return NULL;
J
jp9000 已提交
101 102
}

103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
#define audio_lock(scene) pthread_mutex_lock(&scene->audio_mutex)
#define video_lock(scene) pthread_mutex_lock(&scene->video_mutex)
#define audio_unlock(scene) pthread_mutex_unlock(&scene->audio_mutex)
#define video_unlock(scene) pthread_mutex_unlock(&scene->video_mutex)

static inline void full_lock(struct obs_scene *scene)
{
	video_lock(scene);
	audio_lock(scene);
}

static inline void full_unlock(struct obs_scene *scene)
{
	audio_unlock(scene);
	video_unlock(scene);
}

120 121 122 123 124 125 126 127 128 129 130
static void set_visibility(struct obs_scene_item *item, bool vis);
static inline void detach_sceneitem(struct obs_scene_item *item);

static inline void remove_without_release(struct obs_scene_item *item)
{
	item->removed = true;
	set_visibility(item, false);
	signal_item_remove(item);
	detach_sceneitem(item);
}

131
static void remove_all_items(struct obs_scene *scene)
J
jp9000 已提交
132
{
J
jp9000 已提交
133
	struct obs_scene_item *item;
134 135 136
	DARRAY(struct obs_scene_item*) items;

	da_init(items);
J
jp9000 已提交
137

138
	full_lock(scene);
J
jp9000 已提交
139

J
jp9000 已提交
140 141
	item = scene->first_item;

142 143 144 145
	while (item) {
		struct obs_scene_item *del_item = item;
		item = item->next;

146 147
		remove_without_release(del_item);
		da_push_back(items, &del_item);
148
	}
J
jp9000 已提交
149

150
	full_unlock(scene);
151 152 153 154

	for (size_t i = 0; i < items.num; i++)
		obs_sceneitem_release(items.array[i]);
	da_free(items);
155 156 157 158 159
}

static void scene_destroy(void *data)
{
	struct obs_scene *scene = data;
J
jp9000 已提交
160

161
	remove_all_items(scene);
162

163 164
	pthread_mutex_destroy(&scene->video_mutex);
	pthread_mutex_destroy(&scene->audio_mutex);
J
jp9000 已提交
165 166 167
	bfree(scene);
}

168 169
static void scene_enum_sources(void *data,
		obs_source_enum_proc_t enum_callback,
170
		void *param, bool active)
171 172 173
{
	struct obs_scene *scene = data;
	struct obs_scene_item *item;
174
	struct obs_scene_item *next;
175

176
	full_lock(scene);
177
	item = scene->first_item;
178

179
	while (item) {
180
		next = item->next;
181 182

		obs_sceneitem_addref(item);
183
		if (!active || os_atomic_load_long(&item->active_refs) > 0)
184
			enum_callback(scene->source, item->source, param);
185 186 187 188 189
		obs_sceneitem_release(item);

		item = next;
	}

190
	full_unlock(scene);
191 192
}

193 194 195 196 197 198 199 200 201 202 203 204 205 206
static void scene_enum_active_sources(void *data,
		obs_source_enum_proc_t enum_callback,
		void *param)
{
	scene_enum_sources(data, enum_callback, param, true);
}

static void scene_enum_all_sources(void *data,
		obs_source_enum_proc_t enum_callback,
		void *param)
{
	scene_enum_sources(data, enum_callback, param, false);
}

207 208 209 210 211 212 213 214 215
static inline void detach_sceneitem(struct obs_scene_item *item)
{
	if (item->prev)
		item->prev->next = item->next;
	else
		item->parent->first_item = item->next;

	if (item->next)
		item->next->prev = item->prev;
J
jp9000 已提交
216 217

	item->parent = NULL;
218 219
}

J
jp9000 已提交
220 221
static inline void attach_sceneitem(struct obs_scene *parent,
		struct obs_scene_item *item, struct obs_scene_item *prev)
222
{
J
jp9000 已提交
223 224
	item->prev   = prev;
	item->parent = parent;
225 226 227 228 229 230 231

	if (prev) {
		item->next = prev->next;
		if (prev->next)
			prev->next->prev = item;
		prev->next = item;
	} else {
232 233 234 235
		item->next = parent->first_item;
		if (parent->first_item)
			parent->first_item->prev = item;
		parent->first_item = item;
236 237 238
	}
}

239
void add_alignment(struct vec2 *v, uint32_t align, int cx, int cy)
240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255
{
	if (align & OBS_ALIGN_RIGHT)
		v->x += (float)cx;
	else if ((align & OBS_ALIGN_LEFT) == 0)
		v->x += (float)(cx / 2);

	if (align & OBS_ALIGN_BOTTOM)
		v->y += (float)cy;
	else if ((align & OBS_ALIGN_TOP) == 0)
		v->y += (float)(cy / 2);
}

static void calculate_bounds_data(struct obs_scene_item *item,
		struct vec2 *origin, struct vec2 *scale,
		uint32_t *cx, uint32_t *cy)
{
J
jp9000 已提交
256 257 258 259 260 261 262 263 264 265 266 267 268
	float    width         = (float)(*cx) * fabsf(scale->x);
	float    height        = (float)(*cy) * fabsf(scale->y);
	float    item_aspect   = width / height;
	float    bounds_aspect = item->bounds.x / item->bounds.y;
	uint32_t bounds_type   = item->bounds_type;
	float    width_diff, height_diff;

	if (item->bounds_type == OBS_BOUNDS_MAX_ONLY)
		if (width > item->bounds.x || height > item->bounds.y)
			bounds_type = OBS_BOUNDS_SCALE_INNER;

	if (bounds_type == OBS_BOUNDS_SCALE_INNER ||
	    bounds_type == OBS_BOUNDS_SCALE_OUTER) {
269 270 271 272 273 274 275 276 277 278 279 280
		bool  use_width = (bounds_aspect < item_aspect);
		float mul;

		if (item->bounds_type == OBS_BOUNDS_SCALE_OUTER)
			use_width = !use_width;

		mul = use_width ?
			item->bounds.x / width :
			item->bounds.y / height;

		vec2_mulf(scale, scale, mul);

J
jp9000 已提交
281
	} else if (bounds_type == OBS_BOUNDS_SCALE_TO_WIDTH) {
282 283
		vec2_mulf(scale, scale, item->bounds.x / width);

J
jp9000 已提交
284
	} else if (bounds_type == OBS_BOUNDS_SCALE_TO_HEIGHT) {
285 286
		vec2_mulf(scale, scale, item->bounds.y / height);

J
jp9000 已提交
287
	} else if (bounds_type == OBS_BOUNDS_STRETCH) {
288 289 290 291 292 293 294 295 296 297 298 299 300 301 302
		scale->x = item->bounds.x / (float)(*cx);
		scale->y = item->bounds.y / (float)(*cy);
	}

	width       = (float)(*cx) * scale->x;
	height      = (float)(*cy) * scale->y;
	width_diff  = item->bounds.x - width;
	height_diff = item->bounds.y - height;
	*cx         = (uint32_t)item->bounds.x;
	*cy         = (uint32_t)item->bounds.y;

	add_alignment(origin, item->bounds_align,
			(int)-width_diff, (int)-height_diff);
}

303 304 305 306 307 308 309 310 311 312 313 314 315 316
static inline uint32_t calc_cx(const struct obs_scene_item *item,
		uint32_t width)
{
	uint32_t crop_cx = item->crop.left + item->crop.right;
	return (crop_cx > width) ? 2 : (width - crop_cx);
}

static inline uint32_t calc_cy(const struct obs_scene_item *item,
		uint32_t height)
{
	uint32_t crop_cy = item->crop.top + item->crop.bottom;
	return (crop_cy > height) ? 2 : (height - crop_cy);
}

317 318
static void update_item_transform(struct obs_scene_item *item)
{
319 320
	uint32_t        width         = obs_source_get_width(item->source);
	uint32_t        height        = obs_source_get_height(item->source);
321 322
	uint32_t        cx            = calc_cx(item, width);
	uint32_t        cy            = calc_cy(item, height);
J
jp9000 已提交
323 324
	struct vec2     base_origin;
	struct vec2     origin;
325
	struct vec2     scale         = item->scale;
326 327
	struct calldata params;
	uint8_t         stack[128];
328

329 330 331
	if (os_atomic_load_long(&item->defer_update) > 0)
		return;

332 333 334
	item->last_width  = width;
	item->last_height = height;

335 336 337
	width = cx;
	height = cy;

J
jp9000 已提交
338 339 340
	vec2_zero(&base_origin);
	vec2_zero(&origin);

341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361
	/* ----------------------- */

	if (item->bounds_type != OBS_BOUNDS_NONE) {
		calculate_bounds_data(item, &origin, &scale, &cx, &cy);
	} else {
		cx = (uint32_t)((float)cx * scale.x);
		cy = (uint32_t)((float)cy * scale.y);
	}

	add_alignment(&origin, item->align, (int)cx, (int)cy);

	matrix4_identity(&item->draw_transform);
	matrix4_scale3f(&item->draw_transform, &item->draw_transform,
			scale.x, scale.y, 1.0f);
	matrix4_translate3f(&item->draw_transform, &item->draw_transform,
			-origin.x, -origin.y, 0.0f);
	matrix4_rotate_aa4f(&item->draw_transform, &item->draw_transform,
			0.0f, 0.0f, 1.0f, RAD(item->rot));
	matrix4_translate3f(&item->draw_transform, &item->draw_transform,
			item->pos.x, item->pos.y, 0.0f);

362 363
	item->output_scale = scale;

364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386
	/* ----------------------- */

	if (item->bounds_type != OBS_BOUNDS_NONE) {
		vec2_copy(&scale, &item->bounds);
	} else {
		scale.x = (float)width  * item->scale.x;
		scale.y = (float)height * item->scale.y;
	}

	add_alignment(&base_origin, item->align, (int)scale.x, (int)scale.y);

	matrix4_identity(&item->box_transform);
	matrix4_scale3f(&item->box_transform, &item->box_transform,
			scale.x, scale.y, 1.0f);
	matrix4_translate3f(&item->box_transform, &item->box_transform,
			-base_origin.x, -base_origin.y, 0.0f);
	matrix4_rotate_aa4f(&item->box_transform, &item->box_transform,
			0.0f, 0.0f, 1.0f, RAD(item->rot));
	matrix4_translate3f(&item->box_transform, &item->box_transform,
			item->pos.x, item->pos.y, 0.0f);

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

387
	calldata_init_fixed(&params, stack, sizeof(stack));
388
	calldata_set_ptr(&params, "item", item);
389
	signal_parent(item->parent, "item_transform", &params);
390 391

	os_atomic_set_bool(&item->update_transform, false);
392 393 394 395
}

static inline bool source_size_changed(struct obs_scene_item *item)
{
396 397
	uint32_t width  = obs_source_get_width(item->source);
	uint32_t height = obs_source_get_height(item->source);
398 399 400 401

	return item->last_width != width || item->last_height != height;
}

402 403 404 405 406
static inline bool crop_enabled(const struct obs_sceneitem_crop *crop)
{
	return crop->left || crop->right || crop->top || crop->bottom;
}

407 408 409 410 411
static inline bool scale_filter_enabled(const struct obs_scene_item *item)
{
	return item->scale_filter != OBS_SCALE_DISABLE;
}

412 413 414 415 416
static inline bool item_is_scene(const struct obs_scene_item *item)
{
	return item->source && item->source->info.type == OBS_SOURCE_TYPE_SCENE;
}

417 418
static inline bool item_texture_enabled(const struct obs_scene_item *item)
{
419
	return crop_enabled(&item->crop) || scale_filter_enabled(item) ||
J
jp9000 已提交
420
		(item_is_scene(item) && !item->is_group);
421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467
}

static void render_item_texture(struct obs_scene_item *item)
{
	gs_texture_t *tex = gs_texrender_get_texture(item->item_render);
	gs_effect_t *effect = obs->video.default_effect;
	enum obs_scale_type type = item->scale_filter;
	uint32_t cx = gs_texture_get_width(tex);
	uint32_t cy = gs_texture_get_height(tex);

	if (type != OBS_SCALE_DISABLE) {
		if (type == OBS_SCALE_POINT) {
			gs_eparam_t *image = gs_effect_get_param_by_name(
					effect, "image");
			gs_effect_set_next_sampler(image,
					obs->video.point_sampler);

		} else if (!close_float(item->output_scale.x, 1.0f, EPSILON) ||
		           !close_float(item->output_scale.y, 1.0f, EPSILON)) {
			gs_eparam_t *scale_param;

			if (item->output_scale.x < 0.5f ||
			    item->output_scale.y < 0.5f) {
				effect = obs->video.bilinear_lowres_effect;
			} else if (type == OBS_SCALE_BICUBIC) {
				effect = obs->video.bicubic_effect;
			} else if (type == OBS_SCALE_LANCZOS) {
				effect = obs->video.lanczos_effect;
			}

			scale_param = gs_effect_get_param_by_name(effect,
					"base_dimension_i");
			if (scale_param) {
				struct vec2 base_res_i = {
					1.0f / (float)cx,
					1.0f / (float)cy
				};

				gs_effect_set_vec2(scale_param, &base_res_i);
			}
		}
	}

	while (gs_effect_loop(effect, "Draw"))
		obs_source_draw(tex, 0, 0, 0, 0, 0);
}

468 469
static inline void render_item(struct obs_scene_item *item)
{
470
	if (item->item_render) {
471 472
		uint32_t width  = obs_source_get_width(item->source);
		uint32_t height = obs_source_get_height(item->source);
473 474 475 476

		if (!width || !height)
			return;

477 478 479
		uint32_t cx = calc_cx(item, width);
		uint32_t cy = calc_cy(item, height);

480
		if (cx && cy && gs_texrender_begin(item->item_render, cx, cy)) {
481 482
			float cx_scale = (float)width  / (float)cx;
			float cy_scale = (float)height / (float)cy;
483 484 485 486 487 488 489
			struct vec4 clear_color;

			vec4_zero(&clear_color);
			gs_clear(GS_CLEAR_COLOR, &clear_color, 0.0f, 0);
			gs_ortho(0.0f, (float)width, 0.0f, (float)height,
					-100.0f, 100.0f);

490 491 492 493 494 495
			gs_matrix_scale3f(cx_scale, cy_scale, 1.0f);
			gs_matrix_translate3f(
					-(float)item->crop.left,
					-(float)item->crop.top,
					0.0f);

496 497
			gs_blend_state_push();
			gs_blend_function(GS_BLEND_ONE, GS_BLEND_ZERO);
498
			obs_source_video_render(item->source);
499
			gs_blend_state_pop();
500
			gs_texrender_end(item->item_render);
501 502 503 504 505
		}
	}

	gs_matrix_push();
	gs_matrix_mul(&item->draw_transform);
506 507
	if (item->item_render) {
		render_item_texture(item);
508 509 510 511 512 513 514 515 516 517 518 519 520 521
	} else {
		obs_source_video_render(item->source);
	}
	gs_matrix_pop();
}

static void scene_video_tick(void *data, float seconds)
{
	struct obs_scene *scene = data;
	struct obs_scene_item *item;

	video_lock(scene);
	item = scene->first_item;
	while (item) {
522 523
		if (item->item_render)
			gs_texrender_reset(item->item_render);
524 525 526 527 528 529 530
		item = item->next;
	}
	video_unlock(scene);

	UNUSED_PARAMETER(seconds);
}

531 532
/* assumes video lock */
static void update_transforms_and_prune_sources(obs_scene_t *scene,
533
		struct darray *remove_items, obs_sceneitem_t *group_sceneitem)
534 535
{
	struct obs_scene_item *item = scene->first_item;
J
jp9000 已提交
536 537
	bool rebuild_group = group_sceneitem &&
		os_atomic_load_bool(&group_sceneitem->update_group_resize);
538 539 540 541 542 543 544 545 546

	while (item) {
		if (obs_source_removed(item->source)) {
			struct obs_scene_item *del_item = item;
			item = item->next;

			remove_without_release(del_item);
			darray_push_back(sizeof(struct obs_scene_item*),
					remove_items, &del_item);
J
jp9000 已提交
547
			rebuild_group = true;
548 549 550
			continue;
		}

J
jp9000 已提交
551 552 553 554 555
		if (item->is_group) {
			obs_scene_t *group_scene = item->source->context.data;

			video_lock(group_scene);
			update_transforms_and_prune_sources(group_scene,
556
					remove_items, item);
J
jp9000 已提交
557 558 559
			video_unlock(group_scene);
		}

560 561 562 563
		if (os_atomic_load_bool(&item->update_transform) ||
		    source_size_changed(item)) {

			update_item_transform(item);
J
jp9000 已提交
564
			rebuild_group = true;
565 566 567 568
		}

		item = item->next;
	}
J
jp9000 已提交
569

570 571
	if (rebuild_group && group_sceneitem)
		resize_group(group_sceneitem);
572 573
}

574
static void scene_video_render(void *data, gs_effect_t *effect)
J
jp9000 已提交
575
{
576
	DARRAY(struct obs_scene_item*) remove_items;
577 578 579
	struct obs_scene *scene = data;
	struct obs_scene_item *item;

580 581
	da_init(remove_items);

582
	video_lock(scene);
J
jp9000 已提交
583

584 585 586
	if (!scene->is_group) {
		update_transforms_and_prune_sources(scene, &remove_items.da,
				NULL);
J
jp9000 已提交
587 588
	}

J
jp9000 已提交
589
	gs_blend_state_push();
590
	gs_reset_blend_state();
J
jp9000 已提交
591

592
	item = scene->first_item;
593
	while (item) {
594 595
		if (item->user_visible)
			render_item(item);
596 597

		item = item->next;
J
jp9000 已提交
598
	}
599

J
jp9000 已提交
600 601
	gs_blend_state_pop();

602
	video_unlock(scene);
J
jp9000 已提交
603

604 605 606 607
	for (size_t i = 0; i < remove_items.num; i++)
		obs_sceneitem_release(remove_items.array[i]);
	da_free(remove_items);

J
jp9000 已提交
608 609 610
	UNUSED_PARAMETER(effect);
}

611
static void set_visibility(struct obs_scene_item *item, bool vis)
612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631
{
	pthread_mutex_lock(&item->actions_mutex);

	da_resize(item->audio_actions, 0);

	if (os_atomic_load_long(&item->active_refs) > 0) {
		if (!vis)
			obs_source_remove_active_child(item->parent->source,
					item->source);
	} else if (vis) {
		obs_source_add_active_child(item->parent->source, item->source);
	}

	os_atomic_set_long(&item->active_refs, vis ? 1 : 0);
	item->visible = vis;
	item->user_visible = vis;

	pthread_mutex_unlock(&item->actions_mutex);
}

J
jp9000 已提交
632 633
static void scene_load(void *data, obs_data_t *settings);

634
static void scene_load_item(struct obs_scene *scene, obs_data_t *item_data)
635
{
J
jp9000 已提交
636
	const char            *name = obs_data_get_string(item_data, "name");
J
jp9000 已提交
637
	obs_source_t          *source;
638
	const char            *scale_filter_str;
639
	struct obs_scene_item *item;
J
jp9000 已提交
640
	bool is_group;
641
	bool visible;
642
	bool lock;
643

J
jp9000 已提交
644
	if (obs_data_get_bool(item_data, "group_item_backup"))
645
		return;
J
jp9000 已提交
646 647 648 649 650 651 652 653 654 655 656 657 658 659

	is_group = obs_data_get_bool(item_data, "is_group");

	if (is_group) {
		obs_scene_t *sub_scene = obs_scene_create_private(name);
		source = sub_scene->source;

	} else {
		source = obs_get_source_by_name(name);
		if (!source) {
			blog(LOG_WARNING, "[scene_load_item] Source %s not "
					"found!", name);
			return;
		}
660 661 662
	}

	item = obs_scene_add(scene, source);
P
Palana 已提交
663 664 665 666 667 668 669 670
	if (!item) {
		blog(LOG_WARNING, "[scene_load_item] Could not add source '%s' "
		                  "to scene '%s'!",
		                  name, obs_source_get_name(scene->source));
		
		obs_source_release(source);
		return;
	}
671

J
jp9000 已提交
672 673
	if (is_group) {
		item->is_group = true;
674
		((obs_scene_t *)source->context.data)->is_group = true;
J
jp9000 已提交
675 676
	}

677 678 679
	obs_data_set_default_int(item_data, "align",
			OBS_ALIGN_TOP | OBS_ALIGN_LEFT);

J
jp9000 已提交
680 681 682
	if (obs_data_has_user_value(item_data, "id"))
		item->id = obs_data_get_int(item_data, "id");

J
jp9000 已提交
683 684
	item->rot     = (float)obs_data_get_double(item_data, "rot");
	item->align   = (uint32_t)obs_data_get_int(item_data, "align");
685
	visible = obs_data_get_bool(item_data, "visible");
686
	lock = obs_data_get_bool(item_data, "locked");
687 688
	obs_data_get_vec2(item_data, "pos",    &item->pos);
	obs_data_get_vec2(item_data, "scale",  &item->scale);
689

690 691 692 693 694 695
	obs_data_release(item->private_settings);
	item->private_settings =
		obs_data_get_obj(item_data, "private_settings");
	if (!item->private_settings)
		item->private_settings = obs_data_create();

696
	set_visibility(item, visible);
697
	obs_sceneitem_set_locked(item, lock);
698

699
	item->bounds_type =
J
jp9000 已提交
700 701
		(enum obs_bounds_type)obs_data_get_int(item_data,
				"bounds_type");
702
	item->bounds_align =
J
jp9000 已提交
703
		(uint32_t)obs_data_get_int(item_data, "bounds_align");
704 705
	obs_data_get_vec2(item_data, "bounds", &item->bounds);

706 707 708 709 710
	item->crop.left   = (uint32_t)obs_data_get_int(item_data, "crop_left");
	item->crop.top    = (uint32_t)obs_data_get_int(item_data, "crop_top");
	item->crop.right  = (uint32_t)obs_data_get_int(item_data, "crop_right");
	item->crop.bottom = (uint32_t)obs_data_get_int(item_data, "crop_bottom");

J
jp9000 已提交
711 712 713 714 715 716 717
	if (item->is_group) {
		obs_data_t *group_data = obs_data_get_obj(item_data,
				"group_source");
		scene_load(source->context.data, group_data);
		obs_data_release(group_data);
	}

718 719 720 721 722 723 724 725 726 727 728 729 730
	scale_filter_str = obs_data_get_string(item_data, "scale_filter");
	item->scale_filter = OBS_SCALE_DISABLE;

	if (scale_filter_str) {
		if (astrcmpi(scale_filter_str, "point") == 0)
			item->scale_filter = OBS_SCALE_POINT;
		else if (astrcmpi(scale_filter_str, "bilinear") == 0)
			item->scale_filter = OBS_SCALE_BILINEAR;
		else if (astrcmpi(scale_filter_str, "bicubic") == 0)
			item->scale_filter = OBS_SCALE_BICUBIC;
		else if (astrcmpi(scale_filter_str, "lanczos") == 0)
			item->scale_filter = OBS_SCALE_LANCZOS;
	}
731 732

	if (item->item_render && !item_texture_enabled(item)) {
733
		obs_enter_graphics();
734 735
		gs_texrender_destroy(item->item_render);
		item->item_render = NULL;
736 737
		obs_leave_graphics();

738
	} else if (!item->item_render && item_texture_enabled(item)) {
739
		obs_enter_graphics();
740
		item->item_render = gs_texrender_create(GS_RGBA, GS_ZS_NONE);
741 742 743
		obs_leave_graphics();
	}

744
	obs_source_release(source);
745 746

	update_item_transform(item);
747 748
}

J
jp9000 已提交
749
static void scene_load(void *data, obs_data_t *settings)
750
{
J
jp9000 已提交
751
	struct obs_scene *scene = data;
752
	obs_data_array_t *items = obs_data_get_array(settings, "items");
753 754 755 756 757 758 759 760 761
	size_t           count, i;

	remove_all_items(scene);

	if (!items) return;

	count = obs_data_array_count(items);

	for (i = 0; i < count; i++) {
762
		obs_data_t *item_data = obs_data_array_item(items, i);
763 764 765 766
		scene_load_item(scene, item_data);
		obs_data_release(item_data);
	}

J
jp9000 已提交
767 768 769
	if (obs_data_has_user_value(settings, "id_counter"))
		scene->id_counter = obs_data_get_int(settings, "id_counter");

770 771 772 773 774 775
	if (obs_data_get_bool(settings, "custom_size")) {
		scene->cx = (uint32_t)obs_data_get_int(settings, "cx");
		scene->cy = (uint32_t)obs_data_get_int(settings, "cy");
		scene->custom_size = true;
	}

776 777 778
	obs_data_array_release(items);
}

J
jp9000 已提交
779 780
static void scene_save(void *data, obs_data_t *settings);

781
static void scene_save_item(obs_data_array_t *array,
782 783
		struct obs_scene_item *item,
		struct obs_scene_item *backup_group)
784
{
785
	obs_data_t *item_data = obs_data_create();
786
	const char *name     = obs_source_get_name(item->source);
787
	const char *scale_filter;
J
jp9000 已提交
788 789 790 791
	struct vec2 pos = item->pos;
	struct vec2 scale = item->scale;
	float rot = item->rot;

792 793
	if (backup_group) {
		get_ungrouped_transform(backup_group, &pos, &scale, &rot);
J
jp9000 已提交
794
	}
795

J
jp9000 已提交
796
	obs_data_set_string(item_data, "name",         name);
797
	obs_data_set_bool  (item_data, "visible",      item->user_visible);
798
	obs_data_set_bool  (item_data, "locked",       item->locked);
J
jp9000 已提交
799 800 801
	obs_data_set_double(item_data, "rot",          rot);
	obs_data_set_vec2  (item_data, "pos",          &pos);
	obs_data_set_vec2  (item_data, "scale",        &scale);
J
jp9000 已提交
802 803 804
	obs_data_set_int   (item_data, "align",        (int)item->align);
	obs_data_set_int   (item_data, "bounds_type",  (int)item->bounds_type);
	obs_data_set_int   (item_data, "bounds_align", (int)item->bounds_align);
805
	obs_data_set_vec2 (item_data, "bounds",       &item->bounds);
806 807 808 809
	obs_data_set_int  (item_data, "crop_left",    (int)item->crop.left);
	obs_data_set_int  (item_data, "crop_top",     (int)item->crop.top);
	obs_data_set_int  (item_data, "crop_right",   (int)item->crop.right);
	obs_data_set_int  (item_data, "crop_bottom",  (int)item->crop.bottom);
J
jp9000 已提交
810
	obs_data_set_int  (item_data, "id",           item->id);
J
jp9000 已提交
811
	obs_data_set_bool (item_data, "is_group",     item->is_group);
812
	obs_data_set_bool (item_data, "group_item_backup", !!backup_group);
J
jp9000 已提交
813 814 815 816 817 818 819 820 821 822 823 824 825 826 827

	if (item->is_group) {
		obs_data_t *group_data = obs_data_create();
		obs_scene_t *group_scene = item->source->context.data;
		obs_sceneitem_t *group_item;

		scene_save(group_scene, group_data);

		/* save group items as part of main scene, but ignored.
		 * causes an automatic ungroup if scene collection file
		 * is loaded in previous versions. */
		full_lock(group_scene);

		group_item = group_scene->first_item;
		while (group_item) {
828
			scene_save_item(array, group_item, item);
J
jp9000 已提交
829 830 831 832 833 834 835 836
			group_item = group_item->next;
		}

		full_unlock(group_scene);

		obs_data_set_obj(item_data, "group_source", group_data);
		obs_data_release(group_data);
	}
837 838 839 840 841 842 843 844 845 846 847 848 849

	if (item->scale_filter == OBS_SCALE_POINT)
		scale_filter = "point";
	else if (item->scale_filter == OBS_SCALE_BILINEAR)
		scale_filter = "bilinear";
	else if (item->scale_filter == OBS_SCALE_BICUBIC)
		scale_filter = "bicubic";
	else if (item->scale_filter == OBS_SCALE_LANCZOS)
		scale_filter = "lanczos";
	else
		scale_filter = "disable";

	obs_data_set_string(item_data, "scale_filter", scale_filter);
850

851 852 853
	obs_data_set_obj(item_data, "private_settings",
			item->private_settings);

854 855 856 857
	obs_data_array_push_back(array, item_data);
	obs_data_release(item_data);
}

858
static void scene_save(void *data, obs_data_t *settings)
859 860
{
	struct obs_scene      *scene = data;
861
	obs_data_array_t      *array  = obs_data_array_create();
862 863
	struct obs_scene_item *item;

864
	full_lock(scene);
865 866 867

	item = scene->first_item;
	while (item) {
868
		scene_save_item(array, item, NULL);
869 870 871
		item = item->next;
	}

J
jp9000 已提交
872
	obs_data_set_int(settings, "id_counter", scene->id_counter);
873 874 875 876 877
	obs_data_set_bool(settings, "custom_size", scene->custom_size);
	if (scene->custom_size) {
		obs_data_set_int(settings, "cx", scene->cx);
		obs_data_set_int(settings, "cy", scene->cy);
	}
J
jp9000 已提交
878

879
	full_unlock(scene);
880

J
jp9000 已提交
881
	obs_data_set_array(settings, "items", array);
882 883 884
	obs_data_array_release(array);
}

J
jp9000 已提交
885 886
static uint32_t scene_getwidth(void *data)
{
887 888
	obs_scene_t *scene = data;
	return scene->custom_size ? scene->cx : obs->video.base_width;
J
jp9000 已提交
889 890
}

J
jp9000 已提交
891
static uint32_t scene_getheight(void *data)
J
jp9000 已提交
892
{
893 894
	obs_scene_t *scene = data;
	return scene->custom_size ? scene->cy : obs->video.base_height;
J
jp9000 已提交
895 896
}

897 898 899 900 901 902
static void apply_scene_item_audio_actions(struct obs_scene_item *item,
		float **p_buf, uint64_t ts, size_t sample_rate)
{
	bool cur_visible = item->visible;
	uint64_t frame_num = 0;
	size_t deref_count = 0;
903
	float *buf = NULL;
904

905 906 907 908 909
	if (p_buf) {
		if (!*p_buf)
			*p_buf = malloc(AUDIO_OUTPUT_FRAMES * sizeof(float));
		buf = *p_buf;
	}
910 911 912 913 914 915 916 917 918 919 920 921 922 923

	pthread_mutex_lock(&item->actions_mutex);

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

		if (timestamp < ts)
			timestamp = ts;

		new_frame_num = (timestamp - ts) * (uint64_t)sample_rate /
			1000000000ULL;

924
		if (ts && new_frame_num >= AUDIO_OUTPUT_FRAMES)
925 926 927 928 929 930 931 932
			break;

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

		item->visible = action.visible;
		if (!item->visible)
			deref_count++;

933
		if (buf && new_frame_num > frame_num) {
934 935 936 937 938 939 940
			for (; frame_num < new_frame_num; frame_num++)
				buf[frame_num] = cur_visible ? 1.0f : 0.0f;
		}

		cur_visible = item->visible;
	}

941 942 943 944
	if (buf) {
		for (; frame_num < AUDIO_OUTPUT_FRAMES; frame_num++)
			buf[frame_num] = cur_visible ? 1.0f : 0.0f;
	}
945 946 947 948 949 950 951 952 953 954 955

	pthread_mutex_unlock(&item->actions_mutex);

	while (deref_count--) {
		if (os_atomic_dec_long(&item->active_refs) == 0) {
			obs_source_remove_active_child(item->parent->source,
					item->source);
		}
	}
}

956
static bool apply_scene_item_volume(struct obs_scene_item *item,
957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973
		float **buf, uint64_t ts, size_t sample_rate)
{
	bool actions_pending;
	struct item_action action;

	pthread_mutex_lock(&item->actions_mutex);

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

	pthread_mutex_unlock(&item->actions_mutex);

	if (actions_pending) {
		uint64_t duration = (uint64_t)AUDIO_OUTPUT_FRAMES *
			1000000000ULL / (uint64_t)sample_rate;

974
		if (!ts || action.timestamp < (ts + duration)) {
975 976 977 978 979 980 981 982 983
			apply_scene_item_audio_actions(item, buf, ts,
					sample_rate);
			return true;
		}
	}

	return false;
}

984 985 986 987 988 989
static void process_all_audio_actions(struct obs_scene_item *item,
		size_t sample_rate)
{
	while (apply_scene_item_volume(item, NULL, 0, sample_rate));
}

990 991 992 993 994 995 996 997 998 999 1000 1001
static void mix_audio_with_buf(float *p_out, float *p_in, float *buf_in,
		size_t pos, size_t count)
{
	register float *out = p_out;
	register float *buf = buf_in + pos;
	register float *in = p_in + pos;
	register float *end = in + count;

	while (in < end)
		*(out++) += *(in++) * *(buf++);
}

1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017
static inline void mix_audio(float *p_out, float *p_in,
		size_t pos, size_t count)
{
	register float *out = p_out;
	register float *in = p_in + pos;
	register float *end = in + count;

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

static bool scene_audio_render(void *data, uint64_t *ts_out,
		struct obs_source_audio_mix *audio_output, uint32_t mixers,
		size_t channels, size_t sample_rate)
{
	uint64_t timestamp = 0;
1018
	float *buf = NULL;
1019 1020 1021 1022 1023 1024 1025 1026
	struct obs_source_audio_mix child_audio;
	struct obs_scene *scene = data;
	struct obs_scene_item *item;

	audio_lock(scene);

	item = scene->first_item;
	while (item) {
1027
		if (!obs_source_audio_pending(item->source) && item->visible) {
1028 1029 1030
			uint64_t source_ts =
				obs_source_get_audio_timestamp(item->source);

1031
			if (source_ts && (!timestamp || source_ts < timestamp))
1032 1033 1034 1035 1036 1037 1038
				timestamp = source_ts;
		}

		item = item->next;
	}

	if (!timestamp) {
1039 1040 1041 1042 1043 1044 1045 1046
		/* just process all pending audio actions if no audio playing,
		 * otherwise audio actions will just never be processed */
		item = scene->first_item;
		while (item) {
			process_all_audio_actions(item, sample_rate);
			item = item->next;
		}

1047 1048 1049 1050 1051 1052 1053 1054
		audio_unlock(scene);
		return false;
	}

	item = scene->first_item;
	while (item) {
		uint64_t source_ts;
		size_t pos, count;
1055 1056 1057 1058
		bool apply_buf;

		apply_buf = apply_scene_item_volume(item, &buf, timestamp,
				sample_rate);
1059 1060 1061 1062 1063 1064 1065

		if (obs_source_audio_pending(item->source)) {
			item = item->next;
			continue;
		}

		source_ts = obs_source_get_audio_timestamp(item->source);
1066 1067 1068 1069 1070
		if (!source_ts) {
			item = item->next;
			continue;
		}

1071 1072 1073 1074
		pos = (size_t)ns_to_audio_frames(sample_rate,
				source_ts - timestamp);
		count = AUDIO_OUTPUT_FRAMES - pos;

1075 1076 1077 1078 1079
		if (!apply_buf && !item->visible) {
			item = item->next;
			continue;
		}

1080 1081
		obs_source_get_audio_mix(item->source, &child_audio);
		for (size_t mix = 0; mix < MAX_AUDIO_MIXES; mix++) {
1082 1083 1084
			if ((mixers & (1 << mix)) == 0)
				continue;

1085 1086 1087 1088
			for (size_t ch = 0; ch < channels; ch++) {
				float *out = audio_output->output[mix].data[ch];
				float *in = child_audio.output[mix].data[ch];

1089 1090 1091 1092 1093
				if (apply_buf)
					mix_audio_with_buf(out, in, buf, pos,
							count);
				else
					mix_audio(out, in, pos, count);
1094 1095 1096 1097 1098 1099 1100 1101
			}
		}

		item = item->next;
	}

	*ts_out = timestamp;
	audio_unlock(scene);
1102 1103

	free(buf);
1104 1105 1106
	return true;
}

1107
const struct obs_source_info scene_info =
J
jp9000 已提交
1108
{
1109
	.id            = "scene",
1110
	.type          = OBS_SOURCE_TYPE_SCENE,
1111 1112 1113
	.output_flags  = OBS_SOURCE_VIDEO |
	                 OBS_SOURCE_CUSTOM_DRAW |
	                 OBS_SOURCE_COMPOSITE,
1114 1115 1116
	.get_name      = scene_getname,
	.create        = scene_create,
	.destroy       = scene_destroy,
1117
	.video_tick    = scene_video_tick,
1118
	.video_render  = scene_video_render,
1119
	.audio_render  = scene_audio_render,
1120 1121 1122 1123
	.get_width     = scene_getwidth,
	.get_height    = scene_getheight,
	.load          = scene_load,
	.save          = scene_save,
1124 1125
	.enum_active_sources = scene_enum_active_sources,
	.enum_all_sources = scene_enum_all_sources
J
jp9000 已提交
1126 1127
};

1128
obs_scene_t *obs_scene_create(const char *name)
J
jp9000 已提交
1129
{
1130 1131
	struct obs_source *source = obs_source_create("scene", name, NULL,
			NULL);
1132
	return source->context.data;
J
jp9000 已提交
1133 1134
}

1135 1136 1137 1138 1139 1140 1141
obs_scene_t *obs_scene_create_private(const char *name)
{
	struct obs_source *source = obs_source_create_private("scene", name,
			NULL);
	return source->context.data;
}

1142
static obs_source_t *get_child_at_idx(obs_scene_t *scene, size_t idx)
1143 1144 1145
{
	struct obs_scene_item *item = scene->first_item;

1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198
	while (item && idx--)
		item = item->next;
	return item ? item->source : NULL;
}

static inline obs_source_t *dup_child(struct darray *array, size_t idx,
		obs_scene_t *new_scene, bool private)
{
	DARRAY(struct obs_scene_item*) old_items;
	obs_source_t *source;

	old_items.da = *array;

	source = old_items.array[idx]->source;

	/* if the old item is referenced more than once in the old scene,
	 * make sure they're referenced similarly in the new scene to reduce
	 * load times */
	for (size_t i = 0; i < idx; i++) {
		struct obs_scene_item *item = old_items.array[i];
		if (item->source == source) {
			source = get_child_at_idx(new_scene, i);
			obs_source_addref(source);
			return source;
		}
	}

	return obs_source_duplicate(source, NULL, private);
}

static inline obs_source_t *new_ref(obs_source_t *source)
{
	obs_source_addref(source);
	return source;
}

obs_scene_t *obs_scene_duplicate(obs_scene_t *scene, const char *name,
		enum obs_scene_duplicate_type type)
{
	bool make_unique  = ((int)type & (1<<0)) != 0;
	bool make_private = ((int)type & (1<<1)) != 0;
	DARRAY(struct obs_scene_item*) items;
	struct obs_scene *new_scene;
	struct obs_scene_item *item;
	struct obs_source *source;

	da_init(items);

	if (!obs_ptr_valid(scene, "obs_scene_duplicate"))
		return NULL;

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

1199
	full_lock(scene);
1200

1201
	item = scene->first_item;
1202
	while (item) {
1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214
		da_push_back(items, &item);
		obs_sceneitem_addref(item);
		item = item->next;
	}

	full_unlock(scene);

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

	new_scene = make_private ?
		obs_scene_create_private(name) : obs_scene_create(name);

S
SuslikV 已提交
1215 1216
	obs_source_copy_filters(new_scene->source, scene->source);

1217 1218 1219
	obs_data_apply(new_scene->source->private_settings,
			scene->source->private_settings);

1220 1221 1222 1223 1224
	for (size_t i = 0; i < items.num; i++) {
		item = items.array[i];
		source = make_unique ?
			dup_child(&items.da, i, new_scene, make_private) :
			new_ref(item->source);
1225 1226 1227 1228 1229

		if (source) {
			struct obs_scene_item *new_item =
				obs_scene_add(new_scene, source);

1230
			if (!new_item) {
1231
				obs_source_release(source);
1232 1233 1234 1235 1236
				continue;
			}

			if (!item->user_visible)
				set_visibility(new_item, false);
1237

1238 1239
			new_item->selected = item->selected;
			new_item->pos = item->pos;
1240
			new_item->rot = item->rot;
1241 1242 1243 1244
			new_item->scale = item->scale;
			new_item->align = item->align;
			new_item->last_width = item->last_width;
			new_item->last_height = item->last_height;
1245 1246
			new_item->output_scale = item->output_scale;
			new_item->scale_filter = item->scale_filter;
1247 1248 1249 1250 1251 1252
			new_item->box_transform = item->box_transform;
			new_item->draw_transform = item->draw_transform;
			new_item->bounds_type = item->bounds_type;
			new_item->bounds_align = item->bounds_align;
			new_item->bounds = item->bounds;

1253 1254 1255
			new_item->toggle_visibility =
					OBS_INVALID_HOTKEY_PAIR_ID;

1256 1257
			obs_sceneitem_set_crop(new_item, &item->crop);

1258 1259 1260 1261 1262 1263 1264 1265
			if (!new_item->item_render &&
			    item_texture_enabled(new_item)) {
				obs_enter_graphics();
				new_item->item_render = gs_texrender_create(
						GS_RGBA, GS_ZS_NONE);
				obs_leave_graphics();
			}

1266 1267
			obs_source_release(source);
		}
1268 1269
	}

1270 1271
	for (size_t i = 0; i < items.num; i++)
		obs_sceneitem_release(items.array[i]);
1272

1273
	da_free(items);
1274 1275 1276
	return new_scene;
}

1277
void obs_scene_addref(obs_scene_t *scene)
1278
{
P
Palana 已提交
1279 1280
	if (scene)
		obs_source_addref(scene->source);
1281 1282
}

1283
void obs_scene_release(obs_scene_t *scene)
J
jp9000 已提交
1284
{
P
Palana 已提交
1285 1286
	if (scene)
		obs_source_release(scene->source);
J
jp9000 已提交
1287 1288
}

1289
obs_source_t *obs_scene_get_source(const obs_scene_t *scene)
J
jp9000 已提交
1290
{
J
jp9000 已提交
1291
	return scene ? scene->source : NULL;
J
jp9000 已提交
1292 1293
}

1294
obs_scene_t *obs_scene_from_source(const obs_source_t *source)
1295
{
1296
	if (!source || source->info.id != scene_info.id)
1297 1298
		return NULL;

1299
	return source->context.data;
1300 1301
}

1302
obs_sceneitem_t *obs_scene_find_source(obs_scene_t *scene, const char *name)
1303 1304 1305
{
	struct obs_scene_item *item;

J
jp9000 已提交
1306 1307 1308
	if (!scene)
		return NULL;

1309
	full_lock(scene);
1310 1311 1312

	item = scene->first_item;
	while (item) {
1313
		if (strcmp(item->source->context.name, name) == 0)
1314 1315 1316 1317 1318
			break;

		item = item->next;
	}

1319
	full_unlock(scene);
1320 1321 1322 1323

	return item;
}

J
jp9000 已提交
1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345
obs_sceneitem_t *obs_scene_find_sceneitem_by_id(obs_scene_t *scene, int64_t id)
{
	struct obs_scene_item *item;

	if (!scene)
		return NULL;

	full_lock(scene);

	item = scene->first_item;
	while (item) {
		if (item->id == id)
			break;

		item = item->next;
	}

	full_unlock(scene);

	return item;
}

1346 1347
void obs_scene_enum_items(obs_scene_t *scene,
		bool (*callback)(obs_scene_t*, obs_sceneitem_t*, void*),
1348 1349 1350 1351
		void *param)
{
	struct obs_scene_item *item;

J
jp9000 已提交
1352 1353 1354
	if (!scene || !callback)
		return;

1355
	full_lock(scene);
1356 1357 1358

	item = scene->first_item;
	while (item) {
J
jp9000 已提交
1359 1360 1361 1362
		struct obs_scene_item *next = item->next;

		obs_sceneitem_addref(item);

1363 1364
		if (!callback(scene, item, param)) {
			obs_sceneitem_release(item);
1365
			break;
1366
		}
1367

J
jp9000 已提交
1368 1369 1370
		obs_sceneitem_release(item);

		item = next;
1371 1372
	}

1373
	full_unlock(scene);
1374 1375
}

P
Palana 已提交
1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394
static obs_sceneitem_t *sceneitem_get_ref(obs_sceneitem_t *si)
{
	long owners = si->ref;
	while (owners > 0) {
		if (os_atomic_compare_swap_long(&si->ref, owners, owners + 1))
			return si;

		owners = si->ref;
	}
	return NULL;
}

static bool hotkey_show_sceneitem(void *data, obs_hotkey_pair_id id,
		obs_hotkey_t *hotkey, bool pressed)
{
	UNUSED_PARAMETER(id);
	UNUSED_PARAMETER(hotkey);

	obs_sceneitem_t *si = sceneitem_get_ref(data);
1395
	if (pressed && si && !si->user_visible) {
P
Palana 已提交
1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411
		obs_sceneitem_set_visible(si, true);
		obs_sceneitem_release(si);
		return true;
	}

	obs_sceneitem_release(si);
	return false;
}

static bool hotkey_hide_sceneitem(void *data, obs_hotkey_pair_id id,
		obs_hotkey_t *hotkey, bool pressed)
{
	UNUSED_PARAMETER(id);
	UNUSED_PARAMETER(hotkey);

	obs_sceneitem_t *si = sceneitem_get_ref(data);
1412
	if (pressed && si && si->user_visible) {
P
Palana 已提交
1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451
		obs_sceneitem_set_visible(si, false);
		obs_sceneitem_release(si);
		return true;
	}

	obs_sceneitem_release(si);
	return false;
}

static void init_hotkeys(obs_scene_t *scene, obs_sceneitem_t *item,
		const char *name)
{
	struct dstr show = {0};
	struct dstr hide = {0};
	struct dstr show_desc = {0};
	struct dstr hide_desc = {0};

	dstr_copy(&show, "libobs.show_scene_item.%1");
	dstr_replace(&show, "%1", name);
	dstr_copy(&hide, "libobs.hide_scene_item.%1");
	dstr_replace(&hide, "%1", name);

	dstr_copy(&show_desc, obs->hotkeys.sceneitem_show);
	dstr_replace(&show_desc, "%1", name);
	dstr_copy(&hide_desc, obs->hotkeys.sceneitem_hide);
	dstr_replace(&hide_desc, "%1", name);

	item->toggle_visibility = obs_hotkey_pair_register_source(scene->source,
			show.array, show_desc.array,
			hide.array, hide_desc.array,
			hotkey_show_sceneitem, hotkey_hide_sceneitem,
			item, item);

	dstr_free(&show);
	dstr_free(&hide);
	dstr_free(&show_desc);
	dstr_free(&hide_desc);
}

1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489
static void sceneitem_rename_hotkey(const obs_sceneitem_t *scene_item,
		const char *new_name)
{
	struct dstr show = { 0 };
	struct dstr hide = { 0 };
	struct dstr show_desc = { 0 };
	struct dstr hide_desc = { 0 };

	dstr_copy(&show, "libobs.show_scene_item.%1");
	dstr_replace(&show, "%1", new_name);
	dstr_copy(&hide, "libobs.hide_scene_item.%1");
	dstr_replace(&hide, "%1", new_name);

	obs_hotkey_pair_set_names(scene_item->toggle_visibility,
			show.array, hide.array);

	dstr_copy(&show_desc, obs->hotkeys.sceneitem_show);
	dstr_replace(&show_desc, "%1", new_name);
	dstr_copy(&hide_desc, obs->hotkeys.sceneitem_hide);
	dstr_replace(&hide_desc, "%1", new_name);

	obs_hotkey_pair_set_descriptions(scene_item->toggle_visibility,
			show_desc.array, hide_desc.array);

	dstr_free(&show);
	dstr_free(&hide);
	dstr_free(&show_desc);
	dstr_free(&hide_desc);
}

static void sceneitem_renamed(void *param, calldata_t *data)
{
	obs_sceneitem_t *scene_item = param;
	const char *name = calldata_string(data, "new_name");

	sceneitem_rename_hotkey(scene_item, name);
}

1490 1491 1492 1493 1494 1495
static inline bool source_has_audio(obs_source_t *source)
{
	return (source->info.output_flags &
		(OBS_SOURCE_AUDIO | OBS_SOURCE_COMPOSITE)) != 0;
}

1496
static obs_sceneitem_t *obs_scene_add_internal(obs_scene_t *scene,
J
jp9000 已提交
1497 1498
		obs_source_t *source, obs_sceneitem_t *insert_after,
		bool is_group)
J
jp9000 已提交
1499
{
1500
	struct obs_scene_item *last;
1501
	struct obs_scene_item *item;
1502 1503 1504 1505 1506 1507
	pthread_mutex_t mutex;

	struct item_action action = {
		.visible = true,
		.timestamp = os_gettime_ns()
	};
1508

1509 1510 1511 1512
	if (!scene)
		return NULL;

	if (!source) {
J
jp9000 已提交
1513
		blog(LOG_ERROR, "Tried to add a NULL source to a scene");
1514 1515 1516
		return NULL;
	}

1517 1518 1519 1520 1521
	if (pthread_mutex_init(&mutex, NULL) != 0) {
		blog(LOG_WARNING, "Failed to create scene item mutex");
		return NULL;
	}

1522
	if (!obs_source_add_active_child(scene->source, source)) {
J
jp9000 已提交
1523 1524
		blog(LOG_WARNING, "Failed to add source to scene due to "
		                  "infinite source recursion");
1525
		pthread_mutex_destroy(&mutex);
J
jp9000 已提交
1526 1527 1528
		return NULL;
	}

1529
	item = bzalloc(sizeof(struct obs_scene_item));
J
jp9000 已提交
1530
	item->source  = source;
J
jp9000 已提交
1531
	item->id      = ++scene->id_counter;
J
jp9000 已提交
1532
	item->parent  = scene;
J
jp9000 已提交
1533
	item->ref     = 1;
1534
	item->align   = OBS_ALIGN_TOP | OBS_ALIGN_LEFT;
1535 1536
	item->actions_mutex = mutex;
	item->user_visible = true;
1537
	item->locked = false;
J
jp9000 已提交
1538
	item->is_group = is_group;
1539
	item->private_settings = obs_data_create();
1540
	os_atomic_set_long(&item->active_refs, 1);
J
jp9000 已提交
1541
	vec2_set(&item->scale, 1.0f, 1.0f);
1542 1543
	matrix4_identity(&item->draw_transform);
	matrix4_identity(&item->box_transform);
J
jp9000 已提交
1544

1545
	obs_source_addref(source);
1546

1547 1548 1549 1550 1551 1552 1553
	if (source_has_audio(source)) {
		item->visible = false;
		da_push_back(item->audio_actions, &action);
	} else {
		item->visible = true;
	}

1554 1555 1556 1557 1558 1559
	if (item_texture_enabled(item)) {
		obs_enter_graphics();
		item->item_render = gs_texrender_create(GS_RGBA, GS_ZS_NONE);
		obs_leave_graphics();
	}

1560
	full_lock(scene);
1561

1562 1563 1564 1565 1566 1567
	if (insert_after) {
		obs_sceneitem_t *next = insert_after->next;
		if (next) next->prev = item;
		item->next = insert_after->next;
		item->prev = insert_after;
		insert_after->next = item;
1568
	} else {
1569 1570 1571 1572 1573 1574 1575 1576 1577 1578
		last = scene->first_item;
		if (!last) {
			scene->first_item = item;
		} else {
			while (last->next)
				last = last->next;

			last->next = item;
			item->prev = last;
		}
1579
	}
J
jp9000 已提交
1580

1581
	full_unlock(scene);
1582

1583 1584
	if (!scene->source->context.private)
		init_hotkeys(scene, item, obs_source_get_name(source));
P
Palana 已提交
1585

1586 1587 1588
	signal_handler_connect(obs_source_get_signal_handler(source), "rename",
			sceneitem_renamed, item);

1589 1590 1591 1592 1593
	return item;
}

obs_sceneitem_t *obs_scene_add(obs_scene_t *scene, obs_source_t *source)
{
J
jp9000 已提交
1594 1595
	obs_sceneitem_t *item = obs_scene_add_internal(scene, source, NULL,
			false);
1596 1597 1598
	struct calldata params;
	uint8_t stack[128];

1599
	calldata_init_fixed(&params, stack, sizeof(stack));
1600 1601
	calldata_set_ptr(&params, "scene", scene);
	calldata_set_ptr(&params, "item", item);
1602 1603
	signal_handler_signal(scene->source->context.signals, "item_add",
			&params);
J
jp9000 已提交
1604 1605 1606
	return item;
}

1607
static void obs_sceneitem_destroy(obs_sceneitem_t *item)
J
jp9000 已提交
1608
{
1609
	if (item) {
1610
		if (item->item_render) {
1611
			obs_enter_graphics();
1612
			gs_texrender_destroy(item->item_render);
1613 1614
			obs_leave_graphics();
		}
1615
		obs_data_release(item->private_settings);
P
Palana 已提交
1616
		obs_hotkey_pair_unregister(item->toggle_visibility);
1617
		pthread_mutex_destroy(&item->actions_mutex);
1618 1619 1620
		signal_handler_disconnect(
				obs_source_get_signal_handler(item->source),
				"rename", sceneitem_renamed, item);
1621
		if (item->source)
J
jp9000 已提交
1622
			obs_source_release(item->source);
1623
		da_free(item->audio_actions);
J
jp9000 已提交
1624
		bfree(item);
J
jp9000 已提交
1625 1626 1627
	}
}

1628
void obs_sceneitem_addref(obs_sceneitem_t *item)
J
jp9000 已提交
1629
{
P
Palana 已提交
1630
	if (item)
J
jp9000 已提交
1631
		os_atomic_inc_long(&item->ref);
J
jp9000 已提交
1632
}
1633

1634
void obs_sceneitem_release(obs_sceneitem_t *item)
J
jp9000 已提交
1635
{
P
Palana 已提交
1636 1637
	if (!item)
		return;
P
Palana 已提交
1638

J
jp9000 已提交
1639
	if (os_atomic_dec_long(&item->ref) == 0)
P
Palana 已提交
1640
		obs_sceneitem_destroy(item);
1641 1642
}

1643
void obs_sceneitem_remove(obs_sceneitem_t *item)
J
jp9000 已提交
1644
{
1645
	obs_scene_t *scene;
J
jp9000 已提交
1646

1647
	if (!item)
J
jp9000 已提交
1648 1649
		return;

1650
	scene = item->parent;
J
jp9000 已提交
1651

1652
	full_lock(scene);
J
jp9000 已提交
1653

J
jp9000 已提交
1654
	if (item->removed) {
1655
		if (scene)
1656
			full_unlock(scene);
1657 1658 1659
		return;
	}

J
jp9000 已提交
1660 1661
	item->removed = true;

1662 1663
	assert(scene != NULL);
	assert(scene->source != NULL);
1664 1665

	set_visibility(item, false);
1666

1667
	signal_item_remove(item);
J
jp9000 已提交
1668
	detach_sceneitem(item);
1669

1670
	full_unlock(scene);
J
jp9000 已提交
1671 1672 1673 1674

	obs_sceneitem_release(item);
}

1675
obs_scene_t *obs_sceneitem_get_scene(const obs_sceneitem_t *item)
J
jp9000 已提交
1676
{
J
jp9000 已提交
1677
	return item ? item->parent : NULL;
J
jp9000 已提交
1678 1679
}

1680
obs_source_t *obs_sceneitem_get_source(const obs_sceneitem_t *item)
1681
{
J
jp9000 已提交
1682
	return item ? item->source : NULL;
J
jp9000 已提交
1683 1684
}

1685 1686 1687 1688 1689 1690 1691
static void signal_parent(obs_scene_t *parent, const char *command,
		calldata_t *params)
{
	calldata_set_ptr(params, "scene", parent);
	signal_handler_signal(parent->source->context.signals, command, params);
}

1692
void obs_sceneitem_select(obs_sceneitem_t *item, bool select)
1693
{
1694 1695
	struct calldata params;
	uint8_t stack[128];
1696 1697
	const char *command = select ? "item_select" : "item_deselect";

1698
	if (!item || item->selected == select || !item->parent)
1699 1700 1701 1702
		return;

	item->selected = select;

1703
	calldata_init_fixed(&params, stack, sizeof(stack));
1704
	calldata_set_ptr(&params, "item",  item);
1705 1706

	signal_parent(item->parent, command, &params);
1707 1708
}

1709
bool obs_sceneitem_selected(const obs_sceneitem_t *item)
1710 1711 1712 1713
{
	return item ? item->selected : false;
}

1714
void obs_sceneitem_set_pos(obs_sceneitem_t *item, const struct vec2 *pos)
J
jp9000 已提交
1715
{
1716
	if (item) {
J
jp9000 已提交
1717
		vec2_copy(&item->pos, pos);
1718
		os_atomic_set_bool(&item->update_transform, true);
1719
	}
J
jp9000 已提交
1720 1721
}

1722
void obs_sceneitem_set_rot(obs_sceneitem_t *item, float rot)
J
jp9000 已提交
1723
{
1724
	if (item) {
J
jp9000 已提交
1725
		item->rot = rot;
1726
		os_atomic_set_bool(&item->update_transform, true);
1727
	}
J
jp9000 已提交
1728 1729
}

1730
void obs_sceneitem_set_scale(obs_sceneitem_t *item, const struct vec2 *scale)
J
jp9000 已提交
1731
{
1732 1733
	if (item) {
		vec2_copy(&item->scale, scale);
1734
		os_atomic_set_bool(&item->update_transform, true);
1735
	}
J
jp9000 已提交
1736 1737
}

1738
void obs_sceneitem_set_alignment(obs_sceneitem_t *item, uint32_t alignment)
J
jp9000 已提交
1739
{
1740 1741
	if (item) {
		item->align = alignment;
1742
		os_atomic_set_bool(&item->update_transform, true);
1743
	}
J
jp9000 已提交
1744 1745
}

S
Socapex 已提交
1746
static inline void signal_reorder(struct obs_scene_item *item)
J
jp9000 已提交
1747
{
1748
	const char *command = NULL;
1749 1750
	struct calldata params;
	uint8_t stack[128];
J
jp9000 已提交
1751

S
Socapex 已提交
1752
	command = "reorder";
J
jp9000 已提交
1753

1754
	calldata_init_fixed(&params, stack, sizeof(stack));
1755
	signal_parent(item->parent, command, &params);
J
jp9000 已提交
1756 1757
}

1758
void obs_sceneitem_set_order(obs_sceneitem_t *item,
J
jp9000 已提交
1759
		enum obs_order_movement movement)
J
jp9000 已提交
1760
{
J
jp9000 已提交
1761 1762
	if (!item) return;

J
jp9000 已提交
1763
	struct obs_scene_item *next, *prev;
J
jp9000 已提交
1764 1765
	struct obs_scene *scene = item->parent;

J
jp9000 已提交
1766
	obs_scene_addref(scene);
1767
	full_lock(scene);
1768

J
jp9000 已提交
1769 1770 1771
	next = item->next;
	prev = item->prev;

1772 1773
	detach_sceneitem(item);

J
jp9000 已提交
1774
	if (movement == OBS_ORDER_MOVE_DOWN) {
J
jp9000 已提交
1775
		attach_sceneitem(scene, item, prev ? prev->prev : NULL);
J
jp9000 已提交
1776

J
jp9000 已提交
1777
	} else if (movement == OBS_ORDER_MOVE_UP) {
J
jp9000 已提交
1778
		attach_sceneitem(scene, item, next ? next : prev);
J
jp9000 已提交
1779

J
jp9000 已提交
1780
	} else if (movement == OBS_ORDER_MOVE_TOP) {
J
jp9000 已提交
1781
		struct obs_scene_item *last = next;
1782
		if (!last) {
J
jp9000 已提交
1783
			last = prev;
1784 1785 1786 1787
		} else {
			while (last->next)
				last = last->next;
		}
J
jp9000 已提交
1788

J
jp9000 已提交
1789
		attach_sceneitem(scene, item, last);
1790

J
jp9000 已提交
1791
	} else if (movement == OBS_ORDER_MOVE_BOTTOM) {
J
jp9000 已提交
1792
		attach_sceneitem(scene, item, NULL);
J
jp9000 已提交
1793
	}
1794

1795
	full_unlock(scene);
1796 1797

	signal_reorder(item);
1798
	obs_scene_release(scene);
J
jp9000 已提交
1799 1800
}

1801 1802 1803 1804 1805 1806 1807 1808 1809
void obs_sceneitem_set_order_position(obs_sceneitem_t *item,
		int position)
{
	if (!item) return;

	struct obs_scene *scene = item->parent;
	struct obs_scene_item *next;

	obs_scene_addref(scene);
1810
	full_lock(scene);
1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826

	detach_sceneitem(item);
	next = scene->first_item;

	if (position == 0) {
		attach_sceneitem(scene, item, NULL);
	} else {
		for (int i = position; i > 1; --i) {
			if (next->next == NULL)
				break;
			next = next->next;
		}

		attach_sceneitem(scene, item, next);
	}

1827
	full_unlock(scene);
1828 1829

	signal_reorder(item);
1830 1831 1832
	obs_scene_release(scene);
}

1833
void obs_sceneitem_set_bounds_type(obs_sceneitem_t *item,
1834 1835 1836 1837
		enum obs_bounds_type type)
{
	if (item) {
		item->bounds_type = type;
1838
		os_atomic_set_bool(&item->update_transform, true);
1839 1840 1841
	}
}

1842
void obs_sceneitem_set_bounds_alignment(obs_sceneitem_t *item,
1843 1844 1845 1846
		uint32_t alignment)
{
	if (item) {
		item->bounds_align = alignment;
1847
		os_atomic_set_bool(&item->update_transform, true);
1848 1849 1850
	}
}

1851
void obs_sceneitem_set_bounds(obs_sceneitem_t *item, const struct vec2 *bounds)
1852 1853 1854
{
	if (item) {
		item->bounds = *bounds;
1855
		os_atomic_set_bool(&item->update_transform, true);
1856 1857 1858
	}
}

1859
void obs_sceneitem_get_pos(const obs_sceneitem_t *item, struct vec2 *pos)
J
jp9000 已提交
1860
{
J
jp9000 已提交
1861 1862
	if (item)
		vec2_copy(pos, &item->pos);
J
jp9000 已提交
1863 1864
}

1865
float obs_sceneitem_get_rot(const obs_sceneitem_t *item)
J
jp9000 已提交
1866
{
J
jp9000 已提交
1867
	return item ? item->rot : 0.0f;
J
jp9000 已提交
1868 1869
}

1870
void obs_sceneitem_get_scale(const obs_sceneitem_t *item, struct vec2 *scale)
J
jp9000 已提交
1871
{
J
jp9000 已提交
1872
	if (item)
1873
		vec2_copy(scale, &item->scale);
J
jp9000 已提交
1874 1875
}

1876
uint32_t obs_sceneitem_get_alignment(const obs_sceneitem_t *item)
1877 1878 1879 1880
{
	return item ? item->align : 0;
}

1881
enum obs_bounds_type obs_sceneitem_get_bounds_type(const obs_sceneitem_t *item)
1882 1883 1884 1885
{
	return item ? item->bounds_type : OBS_BOUNDS_NONE;
}

1886
uint32_t obs_sceneitem_get_bounds_alignment(const obs_sceneitem_t *item)
1887 1888 1889 1890
{
	return item ? item->bounds_align : 0;
}

1891
void obs_sceneitem_get_bounds(const obs_sceneitem_t *item, struct vec2 *bounds)
J
jp9000 已提交
1892
{
J
jp9000 已提交
1893
	if (item)
1894 1895 1896
		*bounds = item->bounds;
}

1897
void obs_sceneitem_get_info(const obs_sceneitem_t *item,
1898
		struct obs_transform_info *info)
1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910
{
	if (item && info) {
		info->pos              = item->pos;
		info->rot              = item->rot;
		info->scale            = item->scale;
		info->alignment        = item->align;
		info->bounds_type      = item->bounds_type;
		info->bounds_alignment = item->bounds_align;
		info->bounds           = item->bounds;
	}
}

1911
void obs_sceneitem_set_info(obs_sceneitem_t *item,
1912
		const struct obs_transform_info *info)
1913 1914 1915 1916 1917 1918 1919 1920 1921
{
	if (item && info) {
		item->pos          = info->pos;
		item->rot          = info->rot;
		item->scale        = info->scale;
		item->align        = info->alignment;
		item->bounds_type  = info->bounds_type;
		item->bounds_align = info->bounds_alignment;
		item->bounds       = info->bounds;
1922
		os_atomic_set_bool(&item->update_transform, true);
1923 1924 1925
	}
}

1926
void obs_sceneitem_get_draw_transform(const obs_sceneitem_t *item,
1927 1928 1929 1930 1931 1932
		struct matrix4 *transform)
{
	if (item)
		matrix4_copy(transform, &item->draw_transform);
}

1933
void obs_sceneitem_get_box_transform(const obs_sceneitem_t *item,
1934 1935 1936 1937
		struct matrix4 *transform)
{
	if (item)
		matrix4_copy(transform, &item->box_transform);
J
jp9000 已提交
1938
}
1939 1940 1941

bool obs_sceneitem_visible(const obs_sceneitem_t *item)
{
1942
	return item ? item->user_visible : false;
1943 1944
}

1945
bool obs_sceneitem_set_visible(obs_sceneitem_t *item, bool visible)
1946
{
1947 1948
	struct calldata cd;
	uint8_t stack[256];
1949 1950 1951 1952
	struct item_action action = {
		.visible = visible,
		.timestamp = os_gettime_ns()
	};
1953 1954

	if (!item)
1955
		return false;
1956

1957
	if (item->user_visible == visible)
1958
		return false;
1959 1960

	if (!item->parent)
1961 1962 1963
		return false;

	if (visible) {
1964 1965 1966 1967 1968 1969 1970
		if (os_atomic_inc_long(&item->active_refs) == 1) {
			if (!obs_source_add_active_child(item->parent->source,
						item->source)) {
				os_atomic_dec_long(&item->active_refs);
				return false;
			}
		}
1971 1972
	}

1973
	item->user_visible = visible;
1974

1975
	calldata_init_fixed(&cd, stack, sizeof(stack));
1976 1977 1978
	calldata_set_ptr(&cd, "item", item);
	calldata_set_bool(&cd, "visible", visible);

1979
	signal_parent(item->parent, "item_visible", &cd);
1980 1981 1982 1983 1984 1985 1986 1987

	if (source_has_audio(item->source)) {
		pthread_mutex_lock(&item->actions_mutex);
		da_push_back(item->audio_actions, &action);
		pthread_mutex_unlock(&item->actions_mutex);
	} else {
		set_visibility(item, visible);
	}
1988
	return true;
1989
}
P
Palana 已提交
1990

1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011
bool obs_sceneitem_locked(const obs_sceneitem_t *item)
{
	return item ? item->locked : false;
}

bool obs_sceneitem_set_locked(obs_sceneitem_t *item, bool lock)
{
	if (!item)
		return false;

	if (item->locked == lock)
		return false;

	if (!item->parent)
		return false;

	item->locked = lock;

	return true;
}

P
Palana 已提交
2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047
static bool sceneitems_match(obs_scene_t *scene, obs_sceneitem_t * const *items,
		size_t size, bool *order_matches)
{
	obs_sceneitem_t *item = scene->first_item;

	size_t count = 0;
	while (item) {
		bool found = false;
		for (size_t i = 0; i < size; i++) {
			if (items[i] != item)
				continue;

			if (count != i)
				*order_matches = false;

			found = true;
			break;
		}

		if (!found)
			return false;

		item = item->next;
		count += 1;
	}

	return count == size;
}

bool obs_scene_reorder_items(obs_scene_t *scene,
		obs_sceneitem_t * const *item_order, size_t item_order_size)
{
	if (!scene || !item_order_size)
		return false;

	obs_scene_addref(scene);
2048
	full_lock(scene);
P
Palana 已提交
2049 2050 2051 2052

	bool order_matches = true;
	if (!sceneitems_match(scene, item_order, item_order_size,
				&order_matches) || order_matches) {
2053
		full_unlock(scene);
P
Palana 已提交
2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070
		obs_scene_release(scene);
		return false;
	}

	scene->first_item = item_order[0];

	obs_sceneitem_t *prev = NULL;
	for (size_t i = 0; i < item_order_size; i++) {
		item_order[i]->prev = prev;
		item_order[i]->next = NULL;

		if (prev)
			prev->next = item_order[i];

		prev = item_order[i];
	}

2071
	full_unlock(scene);
2072 2073

	signal_reorder(scene->first_item);
P
Palana 已提交
2074 2075 2076 2077
	obs_scene_release(scene);
	return true;
}

P
Palana 已提交
2078 2079 2080 2081 2082 2083 2084
void obs_scene_atomic_update(obs_scene_t *scene,
		obs_scene_atomic_update_func func, void *data)
{
	if (!scene)
		return;

	obs_scene_addref(scene);
2085
	full_lock(scene);
P
Palana 已提交
2086
	func(data, scene);
2087
	full_unlock(scene);
P
Palana 已提交
2088 2089
	obs_scene_release(scene);
}
2090

2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102
static inline bool crop_equal(const struct obs_sceneitem_crop *crop1,
		const struct obs_sceneitem_crop *crop2)
{
	return crop1->left   == crop2->left  &&
	       crop1->right  == crop2->right &&
	       crop1->top    == crop2->top   &&
	       crop1->bottom == crop2->bottom;
}

void obs_sceneitem_set_crop(obs_sceneitem_t *item,
		const struct obs_sceneitem_crop *crop)
{
2103
	bool item_tex_now_enabled;
2104 2105 2106 2107 2108 2109 2110 2111

	if (!obs_ptr_valid(item, "obs_sceneitem_set_crop"))
		return;
	if (!obs_ptr_valid(crop, "obs_sceneitem_set_crop"))
		return;
	if (crop_equal(crop, &item->crop))
		return;

2112 2113
	item_tex_now_enabled = crop_enabled(crop) ||
		scale_filter_enabled(item) || item_is_scene(item);
2114 2115 2116

	obs_enter_graphics();

2117 2118 2119
	if (!item_tex_now_enabled) {
		gs_texrender_destroy(item->item_render);
		item->item_render = NULL;
2120

2121 2122
	} else if (!item->item_render) {
		item->item_render = gs_texrender_create(GS_RGBA, GS_ZS_NONE);
2123 2124 2125 2126 2127 2128 2129 2130 2131 2132
	}

	memcpy(&item->crop, crop, sizeof(*crop));

	if (item->crop.left < 0) item->crop.left = 0;
	if (item->crop.right < 0) item->crop.right = 0;
	if (item->crop.top < 0) item->crop.top = 0;
	if (item->crop.bottom < 0) item->crop.bottom = 0;
	obs_leave_graphics();

2133
	os_atomic_set_bool(&item->update_transform, true);
2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146
}

void obs_sceneitem_get_crop(const obs_sceneitem_t *item,
		struct obs_sceneitem_crop *crop)
{
	if (!obs_ptr_valid(item, "obs_sceneitem_get_crop"))
		return;
	if (!obs_ptr_valid(crop, "obs_sceneitem_get_crop"))
		return;

	memcpy(crop, &item->crop, sizeof(*crop));
}

2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166
void obs_sceneitem_set_scale_filter(obs_sceneitem_t *item,
		enum obs_scale_type filter)
{
	if (!obs_ptr_valid(item, "obs_sceneitem_set_scale_filter"))
		return;

	item->scale_filter = filter;

	obs_enter_graphics();

	if (!item_texture_enabled(item)) {
		gs_texrender_destroy(item->item_render);
		item->item_render = NULL;

	} else if (!item->item_render) {
		item->item_render = gs_texrender_create(GS_RGBA, GS_ZS_NONE);
	}

	obs_leave_graphics();

2167
	os_atomic_set_bool(&item->update_transform, true);
2168 2169 2170 2171 2172 2173 2174 2175 2176
}

enum obs_scale_type obs_sceneitem_get_scale_filter(
		obs_sceneitem_t *item)
{
	return obs_ptr_valid(item, "obs_sceneitem_get_scale_filter") ?
		item->scale_filter : OBS_SCALE_DISABLE;
}

2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190
void obs_sceneitem_defer_update_begin(obs_sceneitem_t *item)
{
	if (!obs_ptr_valid(item, "obs_sceneitem_defer_update_begin"))
		return;

	os_atomic_inc_long(&item->defer_update);
}

void obs_sceneitem_defer_update_end(obs_sceneitem_t *item)
{
	if (!obs_ptr_valid(item, "obs_sceneitem_defer_update_end"))
		return;

	if (os_atomic_dec_long(&item->defer_update) == 0)
2191
		os_atomic_set_bool(&item->update_transform, true);
2192
}
J
jp9000 已提交
2193

J
jp9000 已提交
2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210
void obs_sceneitem_defer_group_resize_begin(obs_sceneitem_t *item)
{
	if (!obs_ptr_valid(item, "obs_sceneitem_defer_group_resize_begin"))
		return;

	os_atomic_inc_long(&item->defer_group_resize);
}

void obs_sceneitem_defer_group_resize_end(obs_sceneitem_t *item)
{
	if (!obs_ptr_valid(item, "obs_sceneitem_defer_group_resize_end"))
		return;

	if (os_atomic_dec_long(&item->defer_group_resize) == 0)
		os_atomic_set_bool(&item->update_group_resize, true);
}

J
jp9000 已提交
2211 2212 2213 2214 2215 2216 2217
int64_t obs_sceneitem_get_id(const obs_sceneitem_t *item)
{
	if (!obs_ptr_valid(item, "obs_sceneitem_get_id"))
		return 0;

	return item->id;
}
2218 2219 2220 2221 2222 2223 2224 2225 2226

obs_data_t *obs_sceneitem_get_private_settings(obs_sceneitem_t *item)
{
	if (!obs_ptr_valid(item, "obs_sceneitem_get_private_settings"))
		return NULL;

	obs_data_addref(item->private_settings);
	return item->private_settings;
}
J
jp9000 已提交
2227

2228
static inline void transform_val(struct vec2 *v2, struct matrix4 *transform)
J
jp9000 已提交
2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263
{
	struct vec3 v;
	vec3_set(&v, v2->x, v2->y, 0.0f);
	vec3_transform(&v, &v, transform);
	v2->x = v.x;
	v2->y = v.y;
}

static void get_ungrouped_transform(obs_sceneitem_t *group,
		struct vec2 *pos,
		struct vec2 *scale,
		float *rot)
{
	struct matrix4 transform;
	struct matrix4 mat;
	struct vec4 x_base;

	vec4_set(&x_base, 1.0f, 0.0f, 0.0f, 0.0f);

	matrix4_copy(&transform, &group->draw_transform);

	transform_val(pos, &transform);
	vec4_set(&transform.t, 0.0f, 0.0f, 0.0f, 1.0f);

	vec4_set(&mat.x, scale->x, 0.0f, 0.0f, 0.0f);
	vec4_set(&mat.y, 0.0f, scale->y, 0.0f, 0.0f);
	vec4_set(&mat.z, 0.0f, 0.0f, 1.0f, 0.0f);
	vec4_set(&mat.t, 0.0f, 0.0f, 0.0f, 1.0f);
	matrix4_mul(&mat, &mat, &transform);

	scale->x = vec4_len(&mat.x);
	scale->y = vec4_len(&mat.y);
	*rot += group->rot;
}

2264 2265
static void remove_group_transform(obs_sceneitem_t *group,
		obs_sceneitem_t *item)
J
jp9000 已提交
2266 2267
{
	obs_scene_t *parent = item->parent;
2268
	if (!parent || !group)
J
jp9000 已提交
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 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 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 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
		return;

	get_ungrouped_transform(group, &item->pos, &item->scale, &item->rot);

	update_item_transform(item);
}

static void apply_group_transform(obs_sceneitem_t *item, obs_sceneitem_t *group)
{
	struct matrix4 transform;
	struct matrix4 mat;
	struct vec4 x_base;

	vec4_set(&x_base, 1.0f, 0.0f, 0.0f, 0.0f);

	matrix4_inv(&transform, &group->draw_transform);

	transform_val(&item->pos, &transform);
	vec4_set(&transform.t, 0.0f, 0.0f, 0.0f, 1.0f);

	vec4_set(&mat.x, item->scale.x, 0.0f, 0.0f, 0.0f);
	vec4_set(&mat.y, 0.0f, item->scale.y, 0.0f, 0.0f);
	vec4_set(&mat.z, 0.0f, 0.0f, 1.0f, 0.0f);
	vec4_set(&mat.t, 0.0f, 0.0f, 0.0f, 1.0f);
	matrix4_mul(&mat, &mat, &transform);

	item->scale.x = vec4_len(&mat.x);
	item->scale.y = vec4_len(&mat.y);
	item->rot -= group->rot;

	update_item_transform(item);
}

/* assumes group scene and parent scene is locked */
static void resize_group(obs_sceneitem_t *group)
{
	obs_scene_t *scene = group->source->context.data;
	struct vec2 minv;
	struct vec2 maxv;
	struct vec2 scale;

	if (os_atomic_load_long(&group->defer_group_resize) > 0)
		return;

	vec2_set(&minv, M_INFINITE, M_INFINITE);
	vec2_set(&maxv, -M_INFINITE, -M_INFINITE);

	obs_sceneitem_t *item = scene->first_item;
	if (!item) {
		scene->cx = 0;
		scene->cy = 0;
		return;
	}

	while (item) {
#define get_min_max(x_val, y_val) \
		do { \
			struct vec3 v; \
			vec3_set(&v, x_val, y_val, 0.0f); \
			vec3_transform(&v, &v, &item->box_transform); \
			if (v.x < minv.x) minv.x = v.x; \
			if (v.y < minv.y) minv.y = v.y; \
			if (v.x > maxv.x) maxv.x = v.x; \
			if (v.y > maxv.y) maxv.y = v.y; \
		} while (false)

		get_min_max(0.0f, 0.0f);
		get_min_max(1.0f, 0.0f);
		get_min_max(0.0f, 1.0f);
		get_min_max(1.0f, 1.0f);
#undef get_min_max

		item = item->next;
	}

	item = scene->first_item;
	while (item) {
		vec2_sub(&item->pos, &item->pos, &minv);
		update_item_transform(item);
		item = item->next;
	}

	vec2_sub(&scale, &maxv, &minv);
	scene->cx = (uint32_t)ceilf(scale.x);
	scene->cy = (uint32_t)ceilf(scale.y);

	if (group->bounds_type == OBS_BOUNDS_NONE) {
		struct vec2 new_pos;

		if ((group->align & OBS_ALIGN_LEFT) != 0)
			new_pos.x = minv.x;
		else if ((group->align & OBS_ALIGN_RIGHT) != 0)
			new_pos.x = maxv.x;
		else
			new_pos.x = (maxv.x - minv.x) * 0.5f + minv.x;

		if ((group->align & OBS_ALIGN_TOP) != 0)
			new_pos.y = minv.y;
		else if ((group->align & OBS_ALIGN_BOTTOM) != 0)
			new_pos.y = maxv.y;
		else
			new_pos.y = (maxv.y - minv.y) * 0.5f + minv.y;

		transform_val(&new_pos, &group->draw_transform);
		vec2_copy(&group->pos, &new_pos);
	}

	os_atomic_set_bool(&group->update_group_resize, false);

	update_item_transform(group);
}

obs_sceneitem_t *obs_scene_add_group(obs_scene_t *scene, const char *name)
{
	return obs_scene_insert_group(scene, name, NULL, 0);
}

obs_sceneitem_t *obs_scene_insert_group(obs_scene_t *scene,
		const char *name, obs_sceneitem_t **items, size_t count)
{
	if (!scene)
		return NULL;

	/* don't allow groups or sub-items of other groups */
	for (size_t i = count; i > 0; i--) {
		obs_sceneitem_t *item = items[i - 1];
		if (item->parent != scene || item->is_group)
			return NULL;
	}

	obs_scene_t *sub_scene = obs_scene_create_private(name);
	obs_sceneitem_t *last_item = items ? items[count - 1] : NULL;

	obs_sceneitem_t *item = obs_scene_add_internal(
			scene, sub_scene->source, last_item, true);
	sub_scene->custom_size = true;

	obs_scene_release(sub_scene);

	if (!items || !count)
		return item;

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

	full_lock(scene);
	full_lock(sub_scene);
	sub_scene->first_item = items ? items[0] : NULL;

	for (size_t i = count; i > 0; i--) {
		size_t idx = i - 1;
2419
		remove_group_transform(item, items[idx]);
J
jp9000 已提交
2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497
		detach_sceneitem(items[idx]);
	}
	for (size_t i = 0; i < count; i++) {
		size_t idx = i;
		if (idx != (count - 1)) {
			size_t next_idx = idx + 1;
			items[idx]->next = items[next_idx];
			items[next_idx]->prev = items[idx];
		} else {
			items[idx]->next = NULL;
		}
		items[idx]->parent = sub_scene;
		apply_group_transform(items[idx], item);
	}
	items[0]->prev = NULL;
	resize_group(item);
	full_unlock(sub_scene);
	full_unlock(scene);

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

	return item;
}

obs_sceneitem_t *obs_scene_get_group(obs_scene_t *scene, const char *name)
{
	if (!scene || !name || !*name) {
		return NULL;
	}

	obs_sceneitem_t *group = NULL;
	obs_sceneitem_t *item;

	full_lock(scene);

	item = scene->first_item;
	while (item) {
		if (item->is_group && item->source->context.name) {
			if (strcmp(item->source->context.name, name) == 0) {
				group = item;
				break;
			}
		}

		item = item->next;
	}

	full_unlock(scene);

	return group;
}

bool obs_sceneitem_is_group(obs_sceneitem_t *item)
{
	return item && item->is_group;
}

obs_scene_t *obs_sceneitem_group_get_scene(const obs_sceneitem_t *item)
{
	return (item && item->is_group) ? item->source->context.data : NULL;
}

void obs_sceneitem_group_ungroup(obs_sceneitem_t *item)
{
	if (!item || !item->is_group)
		return;

	obs_scene_t *scene = item->parent;
	obs_scene_t *subscene = item->source->context.data;
	obs_sceneitem_t *first;
	obs_sceneitem_t *last;

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

	full_lock(subscene);
	first = subscene->first_item;
	last = first;
	while (last) {
2498
		remove_group_transform(item, last);
J
jp9000 已提交
2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547
		last->parent = scene;
		if (!last->next)
			break;
		last = last->next;
	}
	subscene->first_item = NULL;
	full_unlock(subscene);

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

	full_lock(scene);
	if (last) {
		if (item->prev) {
			first->prev = item->prev;
			item->prev->next = first;
		} else {
			scene->first_item = first;
			first->prev = NULL;
		}
		last->next = item->next;
		if (last->next)
			last->next->prev = last;
		item->next = item->prev = NULL;
		item->parent = NULL;
	} else {
		detach_sceneitem(item);
	}
	full_unlock(scene);

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

	obs_sceneitem_release(item);
}

void obs_sceneitem_group_add_item(obs_sceneitem_t *group, obs_sceneitem_t *item)
{
	if (!group || !group->is_group || !item)
		return;

	obs_scene_t *scene = group->parent;
	obs_scene_t *groupscene = group->source->context.data;
	obs_sceneitem_t *last;

	if (item->parent != scene)
		return;

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

	full_lock(scene);
2548
	remove_group_transform(group, item);
J
jp9000 已提交
2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576
	detach_sceneitem(item);

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

	full_lock(groupscene);
	last = groupscene->first_item;
	if (last) {
		for (;;) {
			if (!last->next)
				break;
			last = last->next;
		}
		last->next = item;
		item->prev = last;
	} else {
		groupscene->first_item = item;
	}
	item->parent = groupscene;
	item->next = NULL;
	apply_group_transform(item, group);
	resize_group(group);
	full_unlock(groupscene);

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

	full_unlock(scene);
}

2577 2578
void obs_sceneitem_group_remove_item(obs_sceneitem_t *group,
		obs_sceneitem_t *item)
J
jp9000 已提交
2579
{
2580
	if (!item || !group || !group->is_group)
J
jp9000 已提交
2581 2582 2583
		return;

	obs_scene_t *groupscene = item->parent;
2584
	obs_scene_t *scene = group->parent;
J
jp9000 已提交
2585 2586 2587 2588 2589

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

	full_lock(scene);
	full_lock(groupscene);
2590
	remove_group_transform(group, item);
J
jp9000 已提交
2591 2592 2593 2594
	detach_sceneitem(item);

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

2595 2596 2597
	if (group->prev) {
		group->prev->next = item;
		item->prev = group->prev;
J
jp9000 已提交
2598 2599 2600 2601
	} else {
		scene->first_item = item;
		item->prev = NULL;
	}
2602 2603
	group->prev = item;
	item->next = group;
J
jp9000 已提交
2604 2605 2606 2607
	item->parent = scene;

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

2608
	resize_group(group);
J
jp9000 已提交
2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635 2636 2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662
	full_unlock(groupscene);
	full_unlock(scene);
}

static void build_current_order_info(obs_scene_t *scene,
		struct obs_sceneitem_order_info **items_out,
		size_t *size_out)
{
	DARRAY(struct obs_sceneitem_order_info) items;
	da_init(items);

	obs_sceneitem_t *item = scene->first_item;
	while (item) {
		da_push_back(items, &item);

		if (item->is_group) {
			obs_scene_t *sub_scene = item->source->context.data;

			full_lock(sub_scene);

			obs_sceneitem_t *sub_item = sub_scene->first_item;

			while (sub_item) {
				da_push_back(items, &item);

				sub_item = sub_item->next;
			}

			full_unlock(sub_scene);
		}

		item = item->next;
	}

	*items_out = items.array;
	*size_out = items.num;
}

static bool sceneitems_match2(obs_scene_t *scene,
		struct obs_sceneitem_order_info *items, size_t size)
{
	struct obs_sceneitem_order_info *cur_items;
	size_t cur_size;

	build_current_order_info(scene, &cur_items, &cur_size);
	if (cur_size != size) {
		bfree(cur_items);
		return false;
	}

	for (size_t i = 0; i < size; i++) {
		struct obs_sceneitem_order_info *new = &items[i];
		struct obs_sceneitem_order_info *old = &cur_items[i];

H
Hunter L. Allen 已提交
2663
		if (new->group != old->group || new->item != old->item) {
J
jp9000 已提交
2664 2665 2666 2667 2668 2669 2670 2671 2672
			bfree(cur_items);
			return false;
		}
	}

	bfree(cur_items);
	return true;
}

2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686 2687 2688 2689
static obs_sceneitem_t *get_sceneitem_parent_group(obs_scene_t *scene,
		obs_sceneitem_t *group_subitem)
{
	if (group_subitem->is_group)
		return NULL;

	obs_sceneitem_t *item = scene->first_item;
	while (item) {
		if (item->is_group &&
		    item->source->context.data == group_subitem->parent)
			return item;
		item = item->next;
	}

	return NULL;
}

J
jp9000 已提交
2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708
bool obs_scene_reorder_items2(obs_scene_t *scene,
		struct obs_sceneitem_order_info *item_order,
		size_t item_order_size)
{
	if (!scene || !item_order_size || !item_order)
		return false;

	obs_scene_addref(scene);
	full_lock(scene);

	if (sceneitems_match2(scene, item_order, item_order_size)) {
		full_unlock(scene);
		obs_scene_release(scene);
		return false;
	}

	for (size_t i = 0; i < item_order_size; i++) {
		struct obs_sceneitem_order_info *info = &item_order[i];
		if (!info->item->is_group) {
2709 2710 2711
			obs_sceneitem_t *group =
				get_sceneitem_parent_group(scene, info->item);
			remove_group_transform(group, info->item);
J
jp9000 已提交
2712 2713 2714 2715 2716 2717 2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779
		}
	}

	scene->first_item = item_order[0].item;

	obs_sceneitem_t *prev = NULL;
	for (size_t i = 0; i < item_order_size; i++) {
		struct obs_sceneitem_order_info *info = &item_order[i];
		obs_sceneitem_t *item = info->item;

		if (info->item->is_group) {
			obs_sceneitem_t *sub_prev = NULL;
			obs_scene_t *sub_scene =
				info->item->source->context.data;

			sub_scene->first_item = NULL;

			obs_scene_addref(sub_scene);
			full_lock(sub_scene);

			for (i++; i < item_order_size; i++) {
				struct obs_sceneitem_order_info *sub_info =
					&item_order[i];
				obs_sceneitem_t *sub_item = sub_info->item;

				if (sub_info->group != info->item) {
					i--;
					break;
				}

				if (!sub_scene->first_item)
					sub_scene->first_item = sub_item;

				sub_item->prev = sub_prev;
				sub_item->next = NULL;
				sub_item->parent = sub_scene;

				if (sub_prev)
					sub_prev->next = sub_item;

				apply_group_transform(sub_info->item,
						sub_info->group);

				sub_prev = sub_item;
			}

			resize_group(info->item);
			full_unlock(sub_scene);
			obs_scene_release(sub_scene);
		}

		item->prev = prev;
		item->next = NULL;
		item->parent = scene;

		if (prev)
			prev->next = item;

		prev = item;
	}

	full_unlock(scene);

	signal_reorder(scene->first_item);
	obs_scene_release(scene);
	return true;
}

2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802
obs_sceneitem_t *obs_sceneitem_get_group(obs_scene_t *scene,
		obs_sceneitem_t *group_subitem)
{
	if (!scene || !group_subitem || group_subitem->is_group)
		return NULL;

	full_lock(scene);
	obs_sceneitem_t *group = get_sceneitem_parent_group(scene,
			group_subitem);
	full_unlock(scene);

	return group;
}

bool obs_source_is_group(const obs_source_t *source)
{
	if (!source || source->info.id != scene_info.id)
		return false;

	return ((obs_scene_t *)source->context.data)->is_group;
}

bool obs_scene_is_group(const obs_scene_t *scene)
J
jp9000 已提交
2803
{
2804
	return scene ? scene->is_group : false;
J
jp9000 已提交
2805 2806 2807 2808 2809 2810
}

void obs_sceneitem_group_enum_items(obs_sceneitem_t *group,
		bool (*callback)(obs_scene_t*, obs_sceneitem_t*, void*),
		void *param)
{
2811
	if (!group || !group->is_group)
J
jp9000 已提交
2812 2813 2814 2815 2816 2817
		return;

	obs_scene_t *scene = obs_scene_from_source(group->source);
	if (scene)
		obs_scene_enum_items(scene, callback, param);
}