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

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

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

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

18
#pragma once
J
jp9000 已提交
19

J
jp9000 已提交
20
#include "util/c99defs.h"
J
jp9000 已提交
21
#include "util/darray.h"
22
#include "util/circlebuf.h"
J
jp9000 已提交
23
#include "util/dstr.h"
J
jp9000 已提交
24
#include "util/threading.h"
J
jp9000 已提交
25
#include "util/platform.h"
P
Palana 已提交
26
#include "util/profiler.h"
J
jp9000 已提交
27 28
#include "callback/signal.h"
#include "callback/proc.h"
J
jp9000 已提交
29 30 31

#include "graphics/graphics.h"

J
jp9000 已提交
32
#include "media-io/audio-resampler.h"
J
jp9000 已提交
33 34 35 36
#include "media-io/video-io.h"
#include "media-io/audio-io.h"

#include "obs.h"
37

J
jp9000 已提交
38
#define NUM_TEXTURES 2
39
#define MICROSECOND_DEN 1000000
J
jp9000 已提交
40

41 42 43 44
static inline int64_t packet_dts_usec(struct encoder_packet *packet)
{
	return packet->dts * MICROSECOND_DEN / packet->timebase_den;
}
45

46 47 48 49 50
struct draw_callback {
	void (*draw)(void *param, uint32_t cx, uint32_t cy);
	void *param;
};

51 52 53 54 55 56 57
/* ------------------------------------------------------------------------- */
/* validity checks */

static inline bool obs_object_valid(const void *obj, const char *f,
		const char *t)
{
	if (!obj) {
58
		blog(LOG_DEBUG, "%s: Null '%s' parameter", f, t);
59 60 61 62 63 64
		return false;
	}

	return true;
}

65 66 67 68 69
#define obs_ptr_valid(ptr, func) obs_object_valid(ptr, func, #ptr)
#define obs_source_valid  obs_ptr_valid
#define obs_output_valid  obs_ptr_valid
#define obs_encoder_valid obs_ptr_valid
#define obs_service_valid obs_ptr_valid
70

J
jp9000 已提交
71
/* ------------------------------------------------------------------------- */
72
/* modules */
J
jp9000 已提交
73 74

struct obs_module {
75
	char *mod_name;
J
jp9000 已提交
76 77 78
	const char *file;
	char *bin_path;
	char *data_path;
J
jp9000 已提交
79
	void *module;
J
jp9000 已提交
80 81
	bool loaded;

82
	bool        (*load)(void);
J
jp9000 已提交
83 84 85 86
	void        (*unload)(void);
	void        (*set_locale)(const char *locale);
	void        (*free_locale)(void);
	uint32_t    (*ver)(void);
87
	void        (*set_pointer)(obs_module_t *module);
J
jp9000 已提交
88 89 90 91 92
	const char *(*name)(void);
	const char *(*description)(void);
	const char *(*author)(void);

	struct obs_module *next;
J
jp9000 已提交
93 94
};

J
jp9000 已提交
95 96
extern void free_module(struct obs_module *mod);

J
jp9000 已提交
97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118
struct obs_module_path {
	char *bin;
	char *data;
};

static inline void free_module_path(struct obs_module_path *omp)
{
	if (omp) {
		bfree(omp->bin);
		bfree(omp->data);
	}
}

static inline bool check_path(const char *data, const char *path,
		struct dstr *output)
{
	dstr_copy(output, path);
	dstr_cat(output, data);

	return os_file_exists(output->array);
}

J
jp9000 已提交
119

P
Palana 已提交
120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178
/* ------------------------------------------------------------------------- */
/* hotkeys */

struct obs_hotkey {
	obs_hotkey_id               id;
	char                        *name;
	char                        *description;

	obs_hotkey_func             func;
	void                        *data;
	int                         pressed;

	obs_hotkey_registerer_t     registerer_type;
	void                        *registerer;

	obs_hotkey_id               pair_partner_id;
};

struct obs_hotkey_pair {
	obs_hotkey_pair_id          pair_id;
	obs_hotkey_id               id[2];
	obs_hotkey_active_func      func[2];
	bool                        pressed0 : 1;
	bool                        pressed1 : 1;
	void                        *data[2];
};

typedef struct obs_hotkey_pair obs_hotkey_pair_t;

typedef struct obs_hotkeys_platform obs_hotkeys_platform_t;

void *obs_hotkey_thread(void *param);

struct obs_core_hotkeys;
bool obs_hotkeys_platform_init(struct obs_core_hotkeys *hotkeys);
void obs_hotkeys_platform_free(struct obs_core_hotkeys *hotkeys);
bool obs_hotkeys_platform_is_pressed(obs_hotkeys_platform_t *context,
		obs_key_t key);

const char *obs_get_hotkey_translation(obs_key_t key, const char *def);

struct obs_context_data;
void obs_hotkeys_context_release(struct obs_context_data *context);

void obs_hotkeys_free(void);

struct obs_hotkey_binding {
	obs_key_combination_t       key;
	bool                        pressed : 1;
	bool                        modifiers_match : 1;

	obs_hotkey_id               hotkey_id;
	obs_hotkey_t                *hotkey;
};

struct obs_hotkey_name_map;
void obs_hotkey_name_map_free(void);


179
/* ------------------------------------------------------------------------- */
J
jp9000 已提交
180
/* views */
181

J
jp9000 已提交
182
struct obs_view {
183
	pthread_mutex_t                 channels_mutex;
184
	obs_source_t                    *channels[MAX_CHANNELS];
185 186
};

J
jp9000 已提交
187 188
extern bool obs_view_init(struct obs_view *view);
extern void obs_view_free(struct obs_view *view);
189 190 191 192 193 194 195


/* ------------------------------------------------------------------------- */
/* displays */

struct obs_display {
	bool                            size_changed;
J
jp9000 已提交
196
	bool                            enabled;
197
	uint32_t                        cx, cy;
198
	uint32_t                        background_color;
199
	gs_swapchain_t                  *swap;
200 201
	pthread_mutex_t                 draw_callbacks_mutex;
	DARRAY(struct draw_callback)    draw_callbacks;
202 203 204

	struct obs_display              *next;
	struct obs_display              **prev_next;
205 206 207
};

extern bool obs_display_init(struct obs_display *display,
208
		const struct gs_init_data *graphics_data);
209 210 211
extern void obs_display_free(struct obs_display *display);


212
/* ------------------------------------------------------------------------- */
J
jp9000 已提交
213
/* core */
214

215 216 217 218 219
struct obs_vframe_info {
	uint64_t timestamp;
	int count;
};

220
struct obs_core_video {
221 222 223 224 225
	graphics_t                      *graphics;
	gs_stagesurf_t                  *copy_surfaces[NUM_TEXTURES];
	gs_texture_t                    *render_textures[NUM_TEXTURES];
	gs_texture_t                    *output_textures[NUM_TEXTURES];
	gs_texture_t                    *convert_textures[NUM_TEXTURES];
226 227 228
	bool                            textures_rendered[NUM_TEXTURES];
	bool                            textures_output[NUM_TEXTURES];
	bool                            textures_copied[NUM_TEXTURES];
J
jp9000 已提交
229
	bool                            textures_converted[NUM_TEXTURES];
230
	struct circlebuf                vframe_info_buffer;
231
	gs_effect_t                     *default_effect;
P
Palana 已提交
232
	gs_effect_t                     *default_rect_effect;
233
	gs_effect_t                     *opaque_effect;
234 235
	gs_effect_t                     *solid_effect;
	gs_effect_t                     *conversion_effect;
236 237
	gs_effect_t                     *bicubic_effect;
	gs_effect_t                     *lanczos_effect;
238
	gs_effect_t                     *bilinear_lowres_effect;
239
	gs_stagesurf_t                  *mapped_surface;
240 241
	int                             cur_texture;

J
jp9000 已提交
242
	uint64_t                        video_time;
243
	video_t                         *video;
244
	pthread_t                       video_thread;
245 246
	uint32_t                        total_frames;
	uint32_t                        lagged_frames;
247 248
	bool                            thread_initialized;

J
jp9000 已提交
249 250 251 252 253 254 255 256 257
	bool                            gpu_conversion;
	const char                      *conversion_tech;
	uint32_t                        conversion_height;
	uint32_t                        plane_offsets[3];
	uint32_t                        plane_sizes[3];
	uint32_t                        plane_linewidth[3];

	uint32_t                        output_width;
	uint32_t                        output_height;
258 259
	uint32_t                        base_width;
	uint32_t                        base_height;
260
	float                           color_matrix[16];
261
	enum obs_scale_type             scale_type;
262 263

	gs_texture_t                    *transparent_texture;
264 265
};

266
struct obs_core_audio {
J
jp9000 已提交
267
	/* TODO: sound output subsystem */
268
	audio_t                         *audio;
J
jp9000 已提交
269

J
jp9000 已提交
270 271 272 273 274 275 276 277
	DARRAY(struct obs_source*)      render_order;
	DARRAY(struct obs_source*)      root_nodes;

	uint64_t                        buffered_ts;
	struct circlebuf                buffered_timestamps;
	int                             buffering_wait_ticks;
	int                             total_buffering_ticks;

J
jp9000 已提交
278
	float                           user_volume;
279 280
};

281
/* user sources, output channels, and displays */
282
struct obs_core_data {
283
	struct obs_source               *first_source;
284
	struct obs_source               *first_audio_source;
285 286 287 288
	struct obs_display              *first_display;
	struct obs_output               *first_output;
	struct obs_encoder              *first_encoder;
	struct obs_service              *first_service;
289 290 291 292 293

	pthread_mutex_t                 sources_mutex;
	pthread_mutex_t                 displays_mutex;
	pthread_mutex_t                 outputs_mutex;
	pthread_mutex_t                 encoders_mutex;
294
	pthread_mutex_t                 services_mutex;
295
	pthread_mutex_t                 audio_sources_mutex;
296

J
jp9000 已提交
297
	struct obs_view                 main_view;
298

J
jp9000 已提交
299 300
	long long                       unnamed_index;

301
	volatile bool                   valid;
302 303
};

P
Palana 已提交
304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330
/* user hotkeys */
struct obs_core_hotkeys {
	pthread_mutex_t                 mutex;
	DARRAY(obs_hotkey_t)            hotkeys;
	obs_hotkey_id                   next_id;
	DARRAY(obs_hotkey_pair_t)       hotkey_pairs;
	obs_hotkey_pair_id              next_pair_id;

	pthread_t                       hotkey_thread;
	bool                            hotkey_thread_initialized;
	os_event_t                      *stop_event;
	bool                            thread_disable_press : 1;
	bool                            strict_modifiers : 1;
	bool                            reroute_hotkeys : 1;
	DARRAY(obs_hotkey_binding_t)    bindings;

	obs_hotkey_callback_router_func router_func;
	void                            *router_func_data;

	obs_hotkeys_platform_t          *platform_context;

	pthread_once_t                  name_map_init_token;
	struct obs_hotkey_name_map      *name_map;

	signal_handler_t                *signals;

	char                            *translations[OBS_KEY_LAST_VALUE];
P
Palana 已提交
331 332 333 334
	char                            *mute;
	char                            *unmute;
	char                            *push_to_mute;
	char                            *push_to_talk;
P
Palana 已提交
335 336
	char                            *sceneitem_show;
	char                            *sceneitem_hide;
P
Palana 已提交
337 338
};

339
struct obs_core {
J
jp9000 已提交
340 341 342
	struct obs_module               *first_module;
	DARRAY(struct obs_module_path)  module_paths;

343
	DARRAY(struct obs_source_info)  source_types;
J
jp9000 已提交
344 345 346 347 348 349 350 351 352
	DARRAY(struct obs_source_info)  input_types;
	DARRAY(struct obs_source_info)  filter_types;
	DARRAY(struct obs_source_info)  transition_types;
	DARRAY(struct obs_output_info)  output_types;
	DARRAY(struct obs_encoder_info) encoder_types;
	DARRAY(struct obs_service_info) service_types;
	DARRAY(struct obs_modal_ui)     modal_ui_callbacks;
	DARRAY(struct obs_modeless_ui)  modeless_ui_callbacks;

353 354
	signal_handler_t                *signals;
	proc_handler_t                  *procs;
355

356
	char                            *locale;
357
	char                            *module_config_path;
358 359
	bool                            name_store_owned;
	profiler_name_store_t           *name_store;
360

361 362
	/* segmented into multiple sub-structures to keep things a bit more
	 * clean and organized */
J
jp9000 已提交
363 364 365
	struct obs_core_video           video;
	struct obs_core_audio           audio;
	struct obs_core_data            data;
P
Palana 已提交
366
	struct obs_core_hotkeys         hotkeys;
J
jp9000 已提交
367 368
};

369
extern struct obs_core *obs;
370

J
jp9000 已提交
371
extern void *obs_video_thread(void *param);
J
jp9000 已提交
372

J
jp9000 已提交
373 374 375 376
extern bool audio_callback(void *param,
		uint64_t start_ts_in, uint64_t end_ts_in, uint64_t *out_ts,
		uint32_t mixers, struct audio_output_data *mixes);

J
jp9000 已提交
377 378

/* ------------------------------------------------------------------------- */
379
/* obs shared context data */
J
jp9000 已提交
380

381 382
struct obs_context_data {
	char                            *name;
383
	void                            *data;
384 385 386
	obs_data_t                      *settings;
	signal_handler_t                *signals;
	proc_handler_t                  *procs;
J
jp9000 已提交
387

P
Palana 已提交
388 389 390 391
	DARRAY(obs_hotkey_id)           hotkeys;
	DARRAY(obs_hotkey_pair_id)      hotkey_pairs;
	obs_data_t                      *hotkey_data;

J
jp9000 已提交
392 393 394
	DARRAY(char*)                   rename_cache;
	pthread_mutex_t                 rename_cache_mutex;

395 396 397 398 399 400 401
	pthread_mutex_t                 *mutex;
	struct obs_context_data         *next;
	struct obs_context_data         **prev_next;
};

extern bool obs_context_data_init(
		struct obs_context_data *context,
402
		obs_data_t              *settings,
P
Palana 已提交
403 404
		const char              *name,
		obs_data_t              *hotkey_data);
405 406 407 408 409 410 411 412 413 414
extern void obs_context_data_free(struct obs_context_data *context);

extern void obs_context_data_insert(struct obs_context_data *context,
		pthread_mutex_t *mutex, void *first);
extern void obs_context_data_remove(struct obs_context_data *context);

extern void obs_context_data_setname(struct obs_context_data *context,
		const char *name);


P
Palana 已提交
415 416 417 418 419 420 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
/* ------------------------------------------------------------------------- */
/* ref-counting  */

struct obs_weak_ref {
	volatile long refs;
	volatile long weak_refs;
};

static inline void obs_ref_addref(struct obs_weak_ref *ref)
{
	os_atomic_inc_long(&ref->refs);
}

static inline bool obs_ref_release(struct obs_weak_ref *ref)
{
	return os_atomic_dec_long(&ref->refs) == -1;
}

static inline void obs_weak_ref_addref(struct obs_weak_ref *ref)
{
	os_atomic_inc_long(&ref->weak_refs);
}

static inline bool obs_weak_ref_release(struct obs_weak_ref *ref)
{
	return os_atomic_dec_long(&ref->weak_refs) == -1;
}

static inline bool obs_weak_ref_get_ref(struct obs_weak_ref *ref)
{
	long owners = ref->refs;
	while (owners > -1) {
		if (os_atomic_compare_swap_long(&ref->refs, owners, owners + 1))
			return true;

		owners = ref->refs;
	}

	return false;
}


457 458 459
/* ------------------------------------------------------------------------- */
/* sources  */

460 461
struct async_frame {
	struct obs_source_frame *frame;
462
	long unused_count;
463 464 465
	bool used;
};

J
jp9000 已提交
466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481
enum audio_action_type {
	AUDIO_ACTION_VOL,
	AUDIO_ACTION_MUTE,
	AUDIO_ACTION_PTT,
	AUDIO_ACTION_PTM,
};

struct audio_action {
	uint64_t timestamp;
	enum audio_action_type type;
	union {
		float vol;
		bool  set;
	};
};

482 483 484 485 486
struct obs_weak_source {
	struct obs_weak_ref ref;
	struct obs_source *source;
};

487 488 489
struct obs_source {
	struct obs_context_data         context;
	struct obs_source_info          info;
490
	struct obs_weak_source          *control;
491

J
jp9000 已提交
492 493
	/* general exposed flags that can be set for the source */
	uint32_t                        flags;
494
	uint32_t                        default_flags;
J
jp9000 已提交
495

496 497 498
	/* indicates ownership of the info.id buffer */
	bool                            owns_info_id;

499
	/* signals to call the source update in the video thread */
500 501
	bool                            defer_update;

502
	/* ensures show/hide are only called once */
J
jp9000 已提交
503
	volatile long                   show_refs;
504

505
	/* ensures activate/deactivate are only called once */
J
jp9000 已提交
506
	volatile long                   activate_refs;
507

J
jp9000 已提交
508 509 510
	/* used to indicate that the source has been removed and all
	 * references to it should be released (not exactly how I would prefer
	 * to handle things but it's the best option) */
511
	bool                            removed;
J
jp9000 已提交
512

513 514 515
	bool                            active;
	bool                            showing;

516 517 518
	/* used to temporarily disable sources if needed */
	bool                            enabled;

J
jp9000 已提交
519
	/* timing (if video is present, is based upon video) */
520 521 522
	volatile bool                   timing_set;
	volatile uint64_t               timing_adjust;
	uint64_t                        next_audio_ts_min;
523
	uint64_t                        next_audio_sys_ts_min;
524 525
	uint64_t                        last_frame_ts;
	uint64_t                        last_sys_timestamp;
526
	bool                            async_rendered;
J
jp9000 已提交
527 528

	/* audio */
529
	bool                            audio_failed;
530
	bool                            audio_pending;
J
jp9000 已提交
531
	bool                            user_muted;
J
jp9000 已提交
532
	bool                            muted;
533 534
	struct obs_source               *next_audio_source;
	struct obs_source               **prev_next_audio_source;
535 536
	uint64_t                        audio_ts;
	struct circlebuf                audio_input_buf[MAX_AUDIO_CHANNELS];
J
jp9000 已提交
537
	DARRAY(struct audio_action)     audio_actions;
538
	float                           *audio_output_buf[MAX_AUDIO_MIXES][MAX_AUDIO_CHANNELS];
539
	struct resample_info            sample_info;
540
	audio_resampler_t               *resampler;
J
jp9000 已提交
541
	pthread_mutex_t                 audio_actions_mutex;
542
	pthread_mutex_t                 audio_buf_mutex;
543
	pthread_mutex_t                 audio_mutex;
544
	struct obs_audio_data           audio_data;
545
	size_t                          audio_storage_size;
546
	uint32_t                        audio_mixers;
J
jp9000 已提交
547
	float                           user_volume;
J
jp9000 已提交
548
	float                           volume;
J
jp9000 已提交
549
	int64_t                         sync_offset;
J
jp9000 已提交
550 551

	/* async video data */
552 553
	gs_texture_t                    *async_texture;
	gs_texrender_t                  *async_convert_texrender;
554
	struct obs_source_frame         *cur_async_frame;
555
	bool                            async_gpu_conversion;
556
	enum video_format               async_format;
J
jp9000 已提交
557
	enum video_format               async_cache_format;
558
	enum gs_color_format            async_texture_format;
559
	float                           async_color_matrix[16];
560 561 562
	bool                            async_full_range;
	float                           async_color_range_min[3];
	float                           async_color_range_max[3];
563
	int                             async_plane_offset[2];
564
	bool                            async_flip;
565
	bool                            async_active;
566
	DARRAY(struct async_frame)      async_cache;
567 568
	DARRAY(struct obs_source_frame*)async_frames;
	pthread_mutex_t                 async_mutex;
569 570
	uint32_t                        async_width;
	uint32_t                        async_height;
J
jp9000 已提交
571 572
	uint32_t                        async_cache_width;
	uint32_t                        async_cache_height;
573 574
	uint32_t                        async_convert_width;
	uint32_t                        async_convert_height;
J
jp9000 已提交
575 576

	/* filters */
577 578 579 580
	struct obs_source               *filter_parent;
	struct obs_source               *filter_target;
	DARRAY(struct obs_source*)      filters;
	pthread_mutex_t                 filter_mutex;
581
	gs_texrender_t                  *filter_texrender;
582
	enum obs_allow_direct_render    allow_direct;
583
	bool                            rendering_filter;
P
Palana 已提交
584 585 586 587 588 589 590

	/* sources specific hotkeys */
	obs_hotkey_pair_id              mute_unmute_key;
	obs_hotkey_id                   push_to_mute_key;
	obs_hotkey_id                   push_to_talk_key;
	bool                            push_to_mute_enabled : 1;
	bool                            push_to_mute_pressed : 1;
J
jp9000 已提交
591
	bool                            user_push_to_mute_pressed : 1;
P
Palana 已提交
592 593
	bool                            push_to_talk_enabled : 1;
	bool                            push_to_talk_pressed : 1;
J
jp9000 已提交
594
	bool                            user_push_to_talk_pressed : 1;
P
Palana 已提交
595 596 597 598
	uint64_t                        push_to_mute_delay;
	uint64_t                        push_to_mute_stop_time;
	uint64_t                        push_to_talk_delay;
	uint64_t                        push_to_talk_stop_time;
J
jp9000 已提交
599 600
};

601
extern const struct obs_source_info *get_source_info(const char *id);
602
extern bool obs_source_init_context(struct obs_source *source,
603 604
		obs_data_t *settings, const char *name,
		obs_data_t *hotkey_data);
J
jp9000 已提交
605

606 607 608
extern void obs_source_save(obs_source_t *source);
extern void obs_source_load(obs_source_t *source);

609 610
extern void obs_source_destroy(struct obs_source *source);

611 612 613 614 615
enum view_type {
	MAIN_VIEW,
	AUX_VIEW
};

616 617 618 619 620 621 622 623 624 625 626 627 628 629 630
static inline void obs_source_dosignal(struct obs_source *source,
		const char *signal_obs, const char *signal_source)
{
	struct calldata data;

	calldata_init(&data);
	calldata_set_ptr(&data, "source", source);
	if (signal_obs)
		signal_handler_signal(obs->signals, signal_obs, &data);
	if (signal_source)
		signal_handler_signal(source->context.signals, signal_source,
				&data);
	calldata_free(&data);
}

631 632 633
extern void obs_source_activate(obs_source_t *source, enum view_type type);
extern void obs_source_deactivate(obs_source_t *source, enum view_type type);
extern void obs_source_video_tick(obs_source_t *source, float seconds);
634 635
extern float obs_source_get_target_volume(obs_source_t *source,
		obs_source_t *target);
J
jp9000 已提交
636

637 638 639
extern void obs_source_audio_render(obs_source_t *source, uint32_t mixers,
		size_t channels, size_t sample_rate, size_t size);

640 641
extern void add_alignment(struct vec2 *v, uint32_t align, int cx, int cy);

J
jp9000 已提交
642 643 644 645

/* ------------------------------------------------------------------------- */
/* outputs  */

646 647 648 649 650 651 652 653 654 655 656 657
enum delay_msg {
	DELAY_MSG_PACKET,
	DELAY_MSG_START,
	DELAY_MSG_STOP,
};

struct delay_data {
	enum delay_msg msg;
	uint64_t ts;
	struct encoder_packet packet;
};

658 659
typedef void (*encoded_callback_t)(void *data, struct encoder_packet *packet);

660 661 662 663 664
struct obs_weak_output {
	struct obs_weak_ref ref;
	struct obs_output *output;
};

J
jp9000 已提交
665
struct obs_output {
666
	struct obs_context_data         context;
667
	struct obs_output_info          info;
668
	struct obs_weak_output          *control;
669

670 671 672
	/* indicates ownership of the info.id buffer */
	bool                            owns_info_id;

673 674 675
	bool                            received_video;
	bool                            received_audio;
	int64_t                         video_offset;
676
	int64_t                         audio_offsets[MAX_AUDIO_MIXES];
J
jp9000 已提交
677 678
	int64_t                         highest_audio_ts;
	int64_t                         highest_video_ts;
J
jp9000 已提交
679
	pthread_mutex_t                 interleaved_mutex;
680
	DARRAY(struct encoder_packet)   interleaved_packets;
J
jp9000 已提交
681

J
jp9000 已提交
682 683 684
	int                             reconnect_retry_sec;
	int                             reconnect_retry_max;
	int                             reconnect_retries;
685
	int                             reconnect_retry_cur_sec;
J
jp9000 已提交
686 687
	bool                            reconnecting;
	pthread_t                       reconnect_thread;
688
	os_event_t                      *reconnect_stop_event;
J
jp9000 已提交
689 690
	volatile bool                   reconnect_thread_active;

691 692
	uint32_t                        starting_drawn_count;
	uint32_t                        starting_lagged_count;
J
jp9000 已提交
693 694 695
	uint32_t                        starting_frame_count;
	uint32_t                        starting_skipped_frame_count;

696 697
	int                             total_frames;

J
jp9000 已提交
698
	bool                            active;
699
	volatile bool                   stopped;
700 701 702
	video_t                         *video;
	audio_t                         *audio;
	obs_encoder_t                   *video_encoder;
703
	obs_encoder_t                   *audio_encoders[MAX_AUDIO_MIXES];
704
	obs_service_t                   *service;
705
	size_t                          mixer_idx;
J
jp9000 已提交
706

707 708 709
	uint32_t                        scaled_width;
	uint32_t                        scaled_height;

J
jp9000 已提交
710 711 712 713 714
	bool                            video_conversion_set;
	bool                            audio_conversion_set;
	struct video_scale_info         video_conversion;
	struct audio_convert_info       audio_conversion;

715
	bool                            valid;
716 717 718 719 720 721 722 723 724 725 726

	uint64_t                        active_delay_ns;
	encoded_callback_t              delay_callback;
	struct circlebuf                delay_data; /* struct delay_data */
	pthread_mutex_t                 delay_mutex;
	uint32_t                        delay_sec;
	uint32_t                        delay_flags;
	uint32_t                        delay_cur_flags;
	volatile long                   delay_restart_refs;
	bool                            delay_active;
	bool                            delay_capturing;
J
jp9000 已提交
727 728
};

729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744
static inline void do_output_signal(struct obs_output *output,
		const char *signal)
{
	struct calldata params = {0};
	calldata_set_ptr(&params, "output", output);
	signal_handler_signal(output->context.signals, signal, &params);
	calldata_free(&params);
}

extern void process_delay(void *data, struct encoder_packet *packet);
extern void obs_output_cleanup_delay(obs_output_t *output);
extern bool obs_output_delay_start(obs_output_t *output);
extern void obs_output_delay_stop(obs_output_t *output);
extern bool obs_output_actual_start(obs_output_t *output);
extern void obs_output_actual_stop(obs_output_t *output, bool force);

745 746
extern const struct obs_output_info *find_output(const char *id);

J
jp9000 已提交
747 748 749
extern void obs_output_remove_encoder(struct obs_output *output,
		struct obs_encoder *encoder);

750 751
void obs_output_destroy(obs_output_t *output);

J
jp9000 已提交
752 753 754 755

/* ------------------------------------------------------------------------- */
/* encoders  */

756 757 758 759 760
struct obs_weak_encoder {
	struct obs_weak_ref ref;
	struct obs_encoder *encoder;
};

761
struct encoder_callback {
762
	bool sent_first_packet;
J
jp9000 已提交
763 764 765 766 767
	void (*new_packet)(void *param, struct encoder_packet *packet);
	void *param;
};

struct obs_encoder {
768
	struct obs_context_data         context;
769
	struct obs_encoder_info         info;
770
	struct obs_weak_encoder         *control;
J
jp9000 已提交
771

J
jp9000 已提交
772 773
	pthread_mutex_t                 init_mutex;

774 775 776 777 778 779
	uint32_t                        samplerate;
	size_t                          planes;
	size_t                          blocksize;
	size_t                          framesize;
	size_t                          framesize_bytes;

780 781
	size_t                          mixer_idx;

782 783
	uint32_t                        scaled_width;
	uint32_t                        scaled_height;
784
	enum video_format               preferred_format;
785

J
jp9000 已提交
786 787
	volatile bool                   active;
	bool                            initialized;
J
jp9000 已提交
788

789 790 791
	/* indicates ownership of the info.id buffer */
	bool                            owns_info_id;

792 793 794 795 796
	uint32_t                        timebase_num;
	uint32_t                        timebase_den;

	int64_t                         cur_pts;

797 798 799 800 801 802 803 804
	struct circlebuf                audio_input_buffer[MAX_AV_PLANES];
	uint8_t                         *audio_output_buffer[MAX_AV_PLANES];

	/* if a video encoder is paired with an audio encoder, make it start
	 * up at the specific timestamp.  if this is the audio encoder,
	 * wait_for_video makes it wait until it's ready to sync up with
	 * video */
	bool                            wait_for_video;
805
	bool                            first_received;
806
	struct obs_encoder              *paired_encoder;
807
	int64_t                         offset_usec;
808 809
	uint64_t                        start_ts;

J
jp9000 已提交
810
	pthread_mutex_t                 outputs_mutex;
811
	DARRAY(obs_output_t*)            outputs;
J
jp9000 已提交
812 813 814

	bool                            destroy_on_stop;

815
	/* stores the video/audio media output pointer.  video_t *or audio_t **/
J
jp9000 已提交
816
	void                            *media;
817 818 819

	pthread_mutex_t                 callbacks_mutex;
	DARRAY(struct encoder_callback) callbacks;
P
Palana 已提交
820 821

	const char                      *profile_encoder_encode_name;
J
jp9000 已提交
822
};
J
jp9000 已提交
823

824 825
extern struct obs_encoder_info *find_encoder(const char *id);

826
extern bool obs_encoder_initialize(obs_encoder_t *encoder);
827
extern void obs_encoder_shutdown(obs_encoder_t *encoder);
828

829
extern void obs_encoder_start(obs_encoder_t *encoder,
830 831
		void (*new_packet)(void *param, struct encoder_packet *packet),
		void *param);
832
extern void obs_encoder_stop(obs_encoder_t *encoder,
833 834 835
		void (*new_packet)(void *param, struct encoder_packet *packet),
		void *param);

J
jp9000 已提交
836 837 838 839
extern void obs_encoder_add_output(struct obs_encoder *encoder,
		struct obs_output *output);
extern void obs_encoder_remove_output(struct obs_encoder *encoder,
		struct obs_output *output);
840

841 842
void obs_encoder_destroy(obs_encoder_t *encoder);

843 844 845
/* ------------------------------------------------------------------------- */
/* services */

846 847 848 849 850
struct obs_weak_service {
	struct obs_weak_ref ref;
	struct obs_service *service;
};

851 852 853
struct obs_service {
	struct obs_context_data         context;
	struct obs_service_info         info;
854
	struct obs_weak_service         *control;
855

856 857 858
	/* indicates ownership of the info.id buffer */
	bool                            owns_info_id;

859 860 861
	bool                            active;
	bool                            destroy;
	struct obs_output               *output;
862
};
863

864 865
extern const struct obs_service_info *find_service(const char *id);

866 867 868 869
extern void obs_service_activate(struct obs_service *service);
extern void obs_service_deactivate(struct obs_service *service, bool remove);
extern bool obs_service_initialize(struct obs_service *service,
		struct obs_output *output);
870 871 872

void obs_service_destroy(obs_service_t *service);