obs-internal.h 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979
  1. /******************************************************************************
  2. Copyright (C) 2013-2014 by Hugh Bailey <[email protected]>
  3. This program is free software: you can redistribute it and/or modify
  4. it under the terms of the GNU General Public License as published by
  5. the Free Software Foundation, either version 2 of the License, or
  6. (at your option) any later version.
  7. This program is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. GNU General Public License for more details.
  11. You should have received a copy of the GNU General Public License
  12. along with this program. If not, see <http://www.gnu.org/licenses/>.
  13. ******************************************************************************/
  14. #pragma once
  15. #include "util/c99defs.h"
  16. #include "util/darray.h"
  17. #include "util/circlebuf.h"
  18. #include "util/dstr.h"
  19. #include "util/threading.h"
  20. #include "util/platform.h"
  21. #include "util/profiler.h"
  22. #include "callback/signal.h"
  23. #include "callback/proc.h"
  24. #include "graphics/graphics.h"
  25. #include "graphics/matrix4.h"
  26. #include "media-io/audio-resampler.h"
  27. #include "media-io/video-io.h"
  28. #include "media-io/audio-io.h"
  29. #include "obs.h"
  30. #define NUM_TEXTURES 2
  31. #define MICROSECOND_DEN 1000000
  32. static inline int64_t packet_dts_usec(struct encoder_packet *packet)
  33. {
  34. return packet->dts * MICROSECOND_DEN / packet->timebase_den;
  35. }
  36. struct draw_callback {
  37. void (*draw)(void *param, uint32_t cx, uint32_t cy);
  38. void *param;
  39. };
  40. /* ------------------------------------------------------------------------- */
  41. /* validity checks */
  42. static inline bool obs_object_valid(const void *obj, const char *f,
  43. const char *t)
  44. {
  45. if (!obj) {
  46. blog(LOG_DEBUG, "%s: Null '%s' parameter", f, t);
  47. return false;
  48. }
  49. return true;
  50. }
  51. #define obs_ptr_valid(ptr, func) obs_object_valid(ptr, func, #ptr)
  52. #define obs_source_valid obs_ptr_valid
  53. #define obs_output_valid obs_ptr_valid
  54. #define obs_encoder_valid obs_ptr_valid
  55. #define obs_service_valid obs_ptr_valid
  56. /* ------------------------------------------------------------------------- */
  57. /* modules */
  58. struct obs_module {
  59. char *mod_name;
  60. const char *file;
  61. char *bin_path;
  62. char *data_path;
  63. void *module;
  64. bool loaded;
  65. bool (*load)(void);
  66. void (*unload)(void);
  67. void (*set_locale)(const char *locale);
  68. void (*free_locale)(void);
  69. uint32_t (*ver)(void);
  70. void (*set_pointer)(obs_module_t *module);
  71. const char *(*name)(void);
  72. const char *(*description)(void);
  73. const char *(*author)(void);
  74. struct obs_module *next;
  75. };
  76. extern void free_module(struct obs_module *mod);
  77. struct obs_module_path {
  78. char *bin;
  79. char *data;
  80. };
  81. static inline void free_module_path(struct obs_module_path *omp)
  82. {
  83. if (omp) {
  84. bfree(omp->bin);
  85. bfree(omp->data);
  86. }
  87. }
  88. static inline bool check_path(const char *data, const char *path,
  89. struct dstr *output)
  90. {
  91. dstr_copy(output, path);
  92. dstr_cat(output, data);
  93. return os_file_exists(output->array);
  94. }
  95. /* ------------------------------------------------------------------------- */
  96. /* hotkeys */
  97. struct obs_hotkey {
  98. obs_hotkey_id id;
  99. char *name;
  100. char *description;
  101. obs_hotkey_func func;
  102. void *data;
  103. int pressed;
  104. obs_hotkey_registerer_t registerer_type;
  105. void *registerer;
  106. obs_hotkey_id pair_partner_id;
  107. };
  108. struct obs_hotkey_pair {
  109. obs_hotkey_pair_id pair_id;
  110. obs_hotkey_id id[2];
  111. obs_hotkey_active_func func[2];
  112. bool pressed0 : 1;
  113. bool pressed1 : 1;
  114. void *data[2];
  115. };
  116. typedef struct obs_hotkey_pair obs_hotkey_pair_t;
  117. typedef struct obs_hotkeys_platform obs_hotkeys_platform_t;
  118. void *obs_hotkey_thread(void *param);
  119. struct obs_core_hotkeys;
  120. bool obs_hotkeys_platform_init(struct obs_core_hotkeys *hotkeys);
  121. void obs_hotkeys_platform_free(struct obs_core_hotkeys *hotkeys);
  122. bool obs_hotkeys_platform_is_pressed(obs_hotkeys_platform_t *context,
  123. obs_key_t key);
  124. const char *obs_get_hotkey_translation(obs_key_t key, const char *def);
  125. struct obs_context_data;
  126. void obs_hotkeys_context_release(struct obs_context_data *context);
  127. void obs_hotkeys_free(void);
  128. struct obs_hotkey_binding {
  129. obs_key_combination_t key;
  130. bool pressed : 1;
  131. bool modifiers_match : 1;
  132. obs_hotkey_id hotkey_id;
  133. obs_hotkey_t *hotkey;
  134. };
  135. struct obs_hotkey_name_map;
  136. void obs_hotkey_name_map_free(void);
  137. /* ------------------------------------------------------------------------- */
  138. /* views */
  139. struct obs_view {
  140. pthread_mutex_t channels_mutex;
  141. obs_source_t *channels[MAX_CHANNELS];
  142. };
  143. extern bool obs_view_init(struct obs_view *view);
  144. extern void obs_view_free(struct obs_view *view);
  145. /* ------------------------------------------------------------------------- */
  146. /* displays */
  147. struct obs_display {
  148. bool size_changed;
  149. bool enabled;
  150. uint32_t cx, cy;
  151. uint32_t background_color;
  152. gs_swapchain_t *swap;
  153. pthread_mutex_t draw_callbacks_mutex;
  154. DARRAY(struct draw_callback) draw_callbacks;
  155. struct obs_display *next;
  156. struct obs_display **prev_next;
  157. };
  158. extern bool obs_display_init(struct obs_display *display,
  159. const struct gs_init_data *graphics_data);
  160. extern void obs_display_free(struct obs_display *display);
  161. /* ------------------------------------------------------------------------- */
  162. /* core */
  163. struct obs_vframe_info {
  164. uint64_t timestamp;
  165. int count;
  166. };
  167. struct obs_core_video {
  168. graphics_t *graphics;
  169. gs_stagesurf_t *copy_surfaces[NUM_TEXTURES];
  170. gs_texture_t *render_textures[NUM_TEXTURES];
  171. gs_texture_t *output_textures[NUM_TEXTURES];
  172. gs_texture_t *convert_textures[NUM_TEXTURES];
  173. bool textures_rendered[NUM_TEXTURES];
  174. bool textures_output[NUM_TEXTURES];
  175. bool textures_copied[NUM_TEXTURES];
  176. bool textures_converted[NUM_TEXTURES];
  177. struct circlebuf vframe_info_buffer;
  178. gs_effect_t *default_effect;
  179. gs_effect_t *default_rect_effect;
  180. gs_effect_t *opaque_effect;
  181. gs_effect_t *solid_effect;
  182. gs_effect_t *conversion_effect;
  183. gs_effect_t *bicubic_effect;
  184. gs_effect_t *lanczos_effect;
  185. gs_effect_t *bilinear_lowres_effect;
  186. gs_stagesurf_t *mapped_surface;
  187. int cur_texture;
  188. uint64_t video_time;
  189. video_t *video;
  190. pthread_t video_thread;
  191. uint32_t total_frames;
  192. uint32_t lagged_frames;
  193. bool thread_initialized;
  194. bool gpu_conversion;
  195. const char *conversion_tech;
  196. uint32_t conversion_height;
  197. uint32_t plane_offsets[3];
  198. uint32_t plane_sizes[3];
  199. uint32_t plane_linewidth[3];
  200. uint32_t output_width;
  201. uint32_t output_height;
  202. uint32_t base_width;
  203. uint32_t base_height;
  204. float color_matrix[16];
  205. enum obs_scale_type scale_type;
  206. gs_texture_t *transparent_texture;
  207. gs_effect_t *deinterlace_discard_effect;
  208. gs_effect_t *deinterlace_discard_2x_effect;
  209. gs_effect_t *deinterlace_linear_effect;
  210. gs_effect_t *deinterlace_linear_2x_effect;
  211. gs_effect_t *deinterlace_blend_effect;
  212. gs_effect_t *deinterlace_blend_2x_effect;
  213. gs_effect_t *deinterlace_yadif_effect;
  214. gs_effect_t *deinterlace_yadif_2x_effect;
  215. };
  216. struct obs_core_audio {
  217. /* TODO: sound output subsystem */
  218. audio_t *audio;
  219. DARRAY(struct obs_source*) render_order;
  220. DARRAY(struct obs_source*) root_nodes;
  221. uint64_t buffered_ts;
  222. struct circlebuf buffered_timestamps;
  223. int buffering_wait_ticks;
  224. int total_buffering_ticks;
  225. float user_volume;
  226. };
  227. /* user sources, output channels, and displays */
  228. struct obs_core_data {
  229. struct obs_source *first_source;
  230. struct obs_source *first_audio_source;
  231. struct obs_display *first_display;
  232. struct obs_output *first_output;
  233. struct obs_encoder *first_encoder;
  234. struct obs_service *first_service;
  235. pthread_mutex_t sources_mutex;
  236. pthread_mutex_t displays_mutex;
  237. pthread_mutex_t outputs_mutex;
  238. pthread_mutex_t encoders_mutex;
  239. pthread_mutex_t services_mutex;
  240. pthread_mutex_t audio_sources_mutex;
  241. struct obs_view main_view;
  242. long long unnamed_index;
  243. volatile bool valid;
  244. };
  245. /* user hotkeys */
  246. struct obs_core_hotkeys {
  247. pthread_mutex_t mutex;
  248. DARRAY(obs_hotkey_t) hotkeys;
  249. obs_hotkey_id next_id;
  250. DARRAY(obs_hotkey_pair_t) hotkey_pairs;
  251. obs_hotkey_pair_id next_pair_id;
  252. pthread_t hotkey_thread;
  253. bool hotkey_thread_initialized;
  254. os_event_t *stop_event;
  255. bool thread_disable_press : 1;
  256. bool strict_modifiers : 1;
  257. bool reroute_hotkeys : 1;
  258. DARRAY(obs_hotkey_binding_t) bindings;
  259. obs_hotkey_callback_router_func router_func;
  260. void *router_func_data;
  261. obs_hotkeys_platform_t *platform_context;
  262. pthread_once_t name_map_init_token;
  263. struct obs_hotkey_name_map *name_map;
  264. signal_handler_t *signals;
  265. char *translations[OBS_KEY_LAST_VALUE];
  266. char *mute;
  267. char *unmute;
  268. char *push_to_mute;
  269. char *push_to_talk;
  270. char *sceneitem_show;
  271. char *sceneitem_hide;
  272. };
  273. struct obs_core {
  274. struct obs_module *first_module;
  275. DARRAY(struct obs_module_path) module_paths;
  276. DARRAY(struct obs_source_info) source_types;
  277. DARRAY(struct obs_source_info) input_types;
  278. DARRAY(struct obs_source_info) filter_types;
  279. DARRAY(struct obs_source_info) transition_types;
  280. DARRAY(struct obs_output_info) output_types;
  281. DARRAY(struct obs_encoder_info) encoder_types;
  282. DARRAY(struct obs_service_info) service_types;
  283. DARRAY(struct obs_modal_ui) modal_ui_callbacks;
  284. DARRAY(struct obs_modeless_ui) modeless_ui_callbacks;
  285. signal_handler_t *signals;
  286. proc_handler_t *procs;
  287. char *locale;
  288. char *module_config_path;
  289. bool name_store_owned;
  290. profiler_name_store_t *name_store;
  291. /* segmented into multiple sub-structures to keep things a bit more
  292. * clean and organized */
  293. struct obs_core_video video;
  294. struct obs_core_audio audio;
  295. struct obs_core_data data;
  296. struct obs_core_hotkeys hotkeys;
  297. };
  298. extern struct obs_core *obs;
  299. extern void *obs_video_thread(void *param);
  300. extern gs_effect_t *obs_load_effect(gs_effect_t **effect, const char *file);
  301. extern bool audio_callback(void *param,
  302. uint64_t start_ts_in, uint64_t end_ts_in, uint64_t *out_ts,
  303. uint32_t mixers, struct audio_output_data *mixes);
  304. /* ------------------------------------------------------------------------- */
  305. /* obs shared context data */
  306. struct obs_context_data {
  307. char *name;
  308. void *data;
  309. obs_data_t *settings;
  310. signal_handler_t *signals;
  311. proc_handler_t *procs;
  312. enum obs_obj_type type;
  313. DARRAY(obs_hotkey_id) hotkeys;
  314. DARRAY(obs_hotkey_pair_id) hotkey_pairs;
  315. obs_data_t *hotkey_data;
  316. DARRAY(char*) rename_cache;
  317. pthread_mutex_t rename_cache_mutex;
  318. pthread_mutex_t *mutex;
  319. struct obs_context_data *next;
  320. struct obs_context_data **prev_next;
  321. bool private;
  322. };
  323. extern bool obs_context_data_init(
  324. struct obs_context_data *context,
  325. enum obs_obj_type type,
  326. obs_data_t *settings,
  327. const char *name,
  328. obs_data_t *hotkey_data,
  329. bool private);
  330. extern void obs_context_data_free(struct obs_context_data *context);
  331. extern void obs_context_data_insert(struct obs_context_data *context,
  332. pthread_mutex_t *mutex, void *first);
  333. extern void obs_context_data_remove(struct obs_context_data *context);
  334. extern void obs_context_data_setname(struct obs_context_data *context,
  335. const char *name);
  336. /* ------------------------------------------------------------------------- */
  337. /* ref-counting */
  338. struct obs_weak_ref {
  339. volatile long refs;
  340. volatile long weak_refs;
  341. };
  342. static inline void obs_ref_addref(struct obs_weak_ref *ref)
  343. {
  344. os_atomic_inc_long(&ref->refs);
  345. }
  346. static inline bool obs_ref_release(struct obs_weak_ref *ref)
  347. {
  348. return os_atomic_dec_long(&ref->refs) == -1;
  349. }
  350. static inline void obs_weak_ref_addref(struct obs_weak_ref *ref)
  351. {
  352. os_atomic_inc_long(&ref->weak_refs);
  353. }
  354. static inline bool obs_weak_ref_release(struct obs_weak_ref *ref)
  355. {
  356. return os_atomic_dec_long(&ref->weak_refs) == -1;
  357. }
  358. static inline bool obs_weak_ref_get_ref(struct obs_weak_ref *ref)
  359. {
  360. long owners = ref->refs;
  361. while (owners > -1) {
  362. if (os_atomic_compare_swap_long(&ref->refs, owners, owners + 1))
  363. return true;
  364. owners = ref->refs;
  365. }
  366. return false;
  367. }
  368. /* ------------------------------------------------------------------------- */
  369. /* sources */
  370. struct async_frame {
  371. struct obs_source_frame *frame;
  372. long unused_count;
  373. bool used;
  374. };
  375. enum audio_action_type {
  376. AUDIO_ACTION_VOL,
  377. AUDIO_ACTION_MUTE,
  378. AUDIO_ACTION_PTT,
  379. AUDIO_ACTION_PTM,
  380. };
  381. struct audio_action {
  382. uint64_t timestamp;
  383. enum audio_action_type type;
  384. union {
  385. float vol;
  386. bool set;
  387. };
  388. };
  389. struct obs_weak_source {
  390. struct obs_weak_ref ref;
  391. struct obs_source *source;
  392. };
  393. struct audio_cb_info {
  394. obs_source_audio_capture_t callback;
  395. void *param;
  396. };
  397. struct obs_source {
  398. struct obs_context_data context;
  399. struct obs_source_info info;
  400. struct obs_weak_source *control;
  401. /* general exposed flags that can be set for the source */
  402. uint32_t flags;
  403. uint32_t default_flags;
  404. /* indicates ownership of the info.id buffer */
  405. bool owns_info_id;
  406. /* signals to call the source update in the video thread */
  407. bool defer_update;
  408. /* ensures show/hide are only called once */
  409. volatile long show_refs;
  410. /* ensures activate/deactivate are only called once */
  411. volatile long activate_refs;
  412. /* used to indicate that the source has been removed and all
  413. * references to it should be released (not exactly how I would prefer
  414. * to handle things but it's the best option) */
  415. bool removed;
  416. bool active;
  417. bool showing;
  418. /* used to temporarily disable sources if needed */
  419. bool enabled;
  420. /* timing (if video is present, is based upon video) */
  421. volatile bool timing_set;
  422. volatile uint64_t timing_adjust;
  423. uint64_t resample_offset;
  424. uint64_t next_audio_ts_min;
  425. uint64_t next_audio_sys_ts_min;
  426. uint64_t last_frame_ts;
  427. uint64_t last_sys_timestamp;
  428. bool async_rendered;
  429. /* audio */
  430. bool audio_failed;
  431. bool audio_pending;
  432. bool user_muted;
  433. bool muted;
  434. struct obs_source *next_audio_source;
  435. struct obs_source **prev_next_audio_source;
  436. uint64_t audio_ts;
  437. struct circlebuf audio_input_buf[MAX_AUDIO_CHANNELS];
  438. size_t last_audio_input_buf_size;
  439. DARRAY(struct audio_action) audio_actions;
  440. float *audio_output_buf[MAX_AUDIO_MIXES][MAX_AUDIO_CHANNELS];
  441. struct resample_info sample_info;
  442. audio_resampler_t *resampler;
  443. pthread_mutex_t audio_actions_mutex;
  444. pthread_mutex_t audio_buf_mutex;
  445. pthread_mutex_t audio_mutex;
  446. pthread_mutex_t audio_cb_mutex;
  447. DARRAY(struct audio_cb_info) audio_cb_list;
  448. struct obs_audio_data audio_data;
  449. size_t audio_storage_size;
  450. uint32_t audio_mixers;
  451. float user_volume;
  452. float volume;
  453. int64_t sync_offset;
  454. int64_t last_sync_offset;
  455. /* async video data */
  456. gs_texture_t *async_texture;
  457. gs_texrender_t *async_texrender;
  458. struct obs_source_frame *cur_async_frame;
  459. bool async_gpu_conversion;
  460. enum video_format async_format;
  461. enum video_format async_cache_format;
  462. enum gs_color_format async_texture_format;
  463. float async_color_matrix[16];
  464. bool async_full_range;
  465. float async_color_range_min[3];
  466. float async_color_range_max[3];
  467. int async_plane_offset[2];
  468. bool async_flip;
  469. bool async_active;
  470. DARRAY(struct async_frame) async_cache;
  471. DARRAY(struct obs_source_frame*)async_frames;
  472. pthread_mutex_t async_mutex;
  473. uint32_t async_width;
  474. uint32_t async_height;
  475. uint32_t async_cache_width;
  476. uint32_t async_cache_height;
  477. uint32_t async_convert_width;
  478. uint32_t async_convert_height;
  479. /* async video deinterlacing */
  480. uint64_t deinterlace_offset;
  481. uint64_t deinterlace_frame_ts;
  482. gs_effect_t *deinterlace_effect;
  483. struct obs_source_frame *prev_async_frame;
  484. gs_texture_t *async_prev_texture;
  485. gs_texrender_t *async_prev_texrender;
  486. uint32_t deinterlace_half_duration;
  487. enum obs_deinterlace_mode deinterlace_mode;
  488. bool deinterlace_top_first;
  489. bool deinterlace_rendered;
  490. /* filters */
  491. struct obs_source *filter_parent;
  492. struct obs_source *filter_target;
  493. DARRAY(struct obs_source*) filters;
  494. pthread_mutex_t filter_mutex;
  495. gs_texrender_t *filter_texrender;
  496. enum obs_allow_direct_render allow_direct;
  497. bool rendering_filter;
  498. /* sources specific hotkeys */
  499. obs_hotkey_pair_id mute_unmute_key;
  500. obs_hotkey_id push_to_mute_key;
  501. obs_hotkey_id push_to_talk_key;
  502. bool push_to_mute_enabled : 1;
  503. bool push_to_mute_pressed : 1;
  504. bool user_push_to_mute_pressed : 1;
  505. bool push_to_talk_enabled : 1;
  506. bool push_to_talk_pressed : 1;
  507. bool user_push_to_talk_pressed : 1;
  508. uint64_t push_to_mute_delay;
  509. uint64_t push_to_mute_stop_time;
  510. uint64_t push_to_talk_delay;
  511. uint64_t push_to_talk_stop_time;
  512. /* transitions */
  513. uint64_t transition_start_time;
  514. uint64_t transition_duration;
  515. pthread_mutex_t transition_tex_mutex;
  516. gs_texrender_t *transition_texrender[2];
  517. pthread_mutex_t transition_mutex;
  518. obs_source_t *transition_sources[2];
  519. bool transitioning_video;
  520. bool transitioning_audio;
  521. bool transition_source_active[2];
  522. uint32_t transition_alignment;
  523. uint32_t transition_actual_cx;
  524. uint32_t transition_actual_cy;
  525. uint32_t transition_cx;
  526. uint32_t transition_cy;
  527. uint32_t transition_fixed_duration;
  528. bool transition_use_fixed_duration : 1;
  529. enum obs_transition_mode transition_mode;
  530. enum obs_transition_scale_type transition_scale_type;
  531. struct matrix4 transition_matrices[2];
  532. };
  533. extern const struct obs_source_info *get_source_info(const char *id);
  534. extern bool obs_source_init_context(struct obs_source *source,
  535. obs_data_t *settings, const char *name,
  536. obs_data_t *hotkey_data, bool private);
  537. extern void obs_source_save(obs_source_t *source);
  538. extern void obs_source_load(obs_source_t *source);
  539. extern bool obs_transition_init(obs_source_t *transition);
  540. extern void obs_transition_free(obs_source_t *transition);
  541. extern void obs_transition_tick(obs_source_t *transition);
  542. extern void obs_transition_enum_sources(obs_source_t *transition,
  543. obs_source_enum_proc_t enum_callback, void *param);
  544. extern void obs_transition_save(obs_source_t *source, obs_data_t *data);
  545. extern void obs_transition_load(obs_source_t *source, obs_data_t *data);
  546. extern void obs_source_destroy(struct obs_source *source);
  547. enum view_type {
  548. MAIN_VIEW,
  549. AUX_VIEW
  550. };
  551. static inline void obs_source_dosignal(struct obs_source *source,
  552. const char *signal_obs, const char *signal_source)
  553. {
  554. struct calldata data;
  555. uint8_t stack[128];
  556. calldata_init_fixed(&data, stack, sizeof(stack));
  557. calldata_set_ptr(&data, "source", source);
  558. if (signal_obs && !source->context.private)
  559. signal_handler_signal(obs->signals, signal_obs, &data);
  560. if (signal_source)
  561. signal_handler_signal(source->context.signals, signal_source,
  562. &data);
  563. }
  564. /* maximum timestamp variance in nanoseconds */
  565. #define MAX_TS_VAR 2000000000ULL
  566. static inline bool frame_out_of_bounds(const obs_source_t *source, uint64_t ts)
  567. {
  568. if (ts < source->last_frame_ts)
  569. return ((source->last_frame_ts - ts) > MAX_TS_VAR);
  570. else
  571. return ((ts - source->last_frame_ts) > MAX_TS_VAR);
  572. }
  573. static inline enum gs_color_format convert_video_format(
  574. enum video_format format)
  575. {
  576. if (format == VIDEO_FORMAT_RGBA)
  577. return GS_RGBA;
  578. else if (format == VIDEO_FORMAT_BGRA)
  579. return GS_BGRA;
  580. return GS_BGRX;
  581. }
  582. extern void obs_source_activate(obs_source_t *source, enum view_type type);
  583. extern void obs_source_deactivate(obs_source_t *source, enum view_type type);
  584. extern void obs_source_video_tick(obs_source_t *source, float seconds);
  585. extern float obs_source_get_target_volume(obs_source_t *source,
  586. obs_source_t *target);
  587. extern void obs_source_audio_render(obs_source_t *source, uint32_t mixers,
  588. size_t channels, size_t sample_rate, size_t size);
  589. extern void add_alignment(struct vec2 *v, uint32_t align, int cx, int cy);
  590. extern struct obs_source_frame *filter_async_video(obs_source_t *source,
  591. struct obs_source_frame *in);
  592. extern bool update_async_texture(struct obs_source *source,
  593. const struct obs_source_frame *frame,
  594. gs_texture_t *tex, gs_texrender_t *texrender);
  595. extern bool set_async_texture_size(struct obs_source *source,
  596. const struct obs_source_frame *frame);
  597. extern void remove_async_frame(obs_source_t *source,
  598. struct obs_source_frame *frame);
  599. extern void set_deinterlace_texture_size(obs_source_t *source);
  600. extern void deinterlace_process_last_frame(obs_source_t *source,
  601. uint64_t sys_time);
  602. extern void deinterlace_update_async_video(obs_source_t *source);
  603. extern void deinterlace_render(obs_source_t *s);
  604. /* ------------------------------------------------------------------------- */
  605. /* outputs */
  606. enum delay_msg {
  607. DELAY_MSG_PACKET,
  608. DELAY_MSG_START,
  609. DELAY_MSG_STOP,
  610. };
  611. struct delay_data {
  612. enum delay_msg msg;
  613. uint64_t ts;
  614. struct encoder_packet packet;
  615. };
  616. typedef void (*encoded_callback_t)(void *data, struct encoder_packet *packet);
  617. struct obs_weak_output {
  618. struct obs_weak_ref ref;
  619. struct obs_output *output;
  620. };
  621. struct obs_output {
  622. struct obs_context_data context;
  623. struct obs_output_info info;
  624. struct obs_weak_output *control;
  625. /* indicates ownership of the info.id buffer */
  626. bool owns_info_id;
  627. bool received_video;
  628. bool received_audio;
  629. int64_t video_offset;
  630. int64_t audio_offsets[MAX_AUDIO_MIXES];
  631. int64_t highest_audio_ts;
  632. int64_t highest_video_ts;
  633. pthread_mutex_t interleaved_mutex;
  634. DARRAY(struct encoder_packet) interleaved_packets;
  635. int reconnect_retry_sec;
  636. int reconnect_retry_max;
  637. int reconnect_retries;
  638. int reconnect_retry_cur_sec;
  639. bool reconnecting;
  640. pthread_t reconnect_thread;
  641. os_event_t *reconnect_stop_event;
  642. volatile bool reconnect_thread_active;
  643. uint32_t starting_drawn_count;
  644. uint32_t starting_lagged_count;
  645. uint32_t starting_frame_count;
  646. uint32_t starting_skipped_frame_count;
  647. int total_frames;
  648. bool active;
  649. volatile bool stopped;
  650. video_t *video;
  651. audio_t *audio;
  652. obs_encoder_t *video_encoder;
  653. obs_encoder_t *audio_encoders[MAX_AUDIO_MIXES];
  654. obs_service_t *service;
  655. size_t mixer_idx;
  656. uint32_t scaled_width;
  657. uint32_t scaled_height;
  658. bool video_conversion_set;
  659. bool audio_conversion_set;
  660. struct video_scale_info video_conversion;
  661. struct audio_convert_info audio_conversion;
  662. bool valid;
  663. uint64_t active_delay_ns;
  664. encoded_callback_t delay_callback;
  665. struct circlebuf delay_data; /* struct delay_data */
  666. pthread_mutex_t delay_mutex;
  667. uint32_t delay_sec;
  668. uint32_t delay_flags;
  669. uint32_t delay_cur_flags;
  670. volatile long delay_restart_refs;
  671. bool delay_active;
  672. bool delay_capturing;
  673. };
  674. static inline void do_output_signal(struct obs_output *output,
  675. const char *signal)
  676. {
  677. struct calldata params = {0};
  678. calldata_set_ptr(&params, "output", output);
  679. signal_handler_signal(output->context.signals, signal, &params);
  680. calldata_free(&params);
  681. }
  682. extern void process_delay(void *data, struct encoder_packet *packet);
  683. extern void obs_output_cleanup_delay(obs_output_t *output);
  684. extern bool obs_output_delay_start(obs_output_t *output);
  685. extern void obs_output_delay_stop(obs_output_t *output);
  686. extern bool obs_output_actual_start(obs_output_t *output);
  687. extern void obs_output_actual_stop(obs_output_t *output, bool force);
  688. extern const struct obs_output_info *find_output(const char *id);
  689. extern void obs_output_remove_encoder(struct obs_output *output,
  690. struct obs_encoder *encoder);
  691. void obs_output_destroy(obs_output_t *output);
  692. /* ------------------------------------------------------------------------- */
  693. /* encoders */
  694. struct obs_weak_encoder {
  695. struct obs_weak_ref ref;
  696. struct obs_encoder *encoder;
  697. };
  698. struct encoder_callback {
  699. bool sent_first_packet;
  700. void (*new_packet)(void *param, struct encoder_packet *packet);
  701. void *param;
  702. };
  703. struct obs_encoder {
  704. struct obs_context_data context;
  705. struct obs_encoder_info info;
  706. struct obs_weak_encoder *control;
  707. pthread_mutex_t init_mutex;
  708. uint32_t samplerate;
  709. size_t planes;
  710. size_t blocksize;
  711. size_t framesize;
  712. size_t framesize_bytes;
  713. size_t mixer_idx;
  714. uint32_t scaled_width;
  715. uint32_t scaled_height;
  716. enum video_format preferred_format;
  717. volatile bool active;
  718. bool initialized;
  719. /* indicates ownership of the info.id buffer */
  720. bool owns_info_id;
  721. uint32_t timebase_num;
  722. uint32_t timebase_den;
  723. int64_t cur_pts;
  724. struct circlebuf audio_input_buffer[MAX_AV_PLANES];
  725. uint8_t *audio_output_buffer[MAX_AV_PLANES];
  726. /* if a video encoder is paired with an audio encoder, make it start
  727. * up at the specific timestamp. if this is the audio encoder,
  728. * wait_for_video makes it wait until it's ready to sync up with
  729. * video */
  730. bool wait_for_video;
  731. bool first_received;
  732. struct obs_encoder *paired_encoder;
  733. int64_t offset_usec;
  734. uint64_t first_raw_ts;
  735. uint64_t start_ts;
  736. pthread_mutex_t outputs_mutex;
  737. DARRAY(obs_output_t*) outputs;
  738. bool destroy_on_stop;
  739. /* stores the video/audio media output pointer. video_t *or audio_t **/
  740. void *media;
  741. pthread_mutex_t callbacks_mutex;
  742. DARRAY(struct encoder_callback) callbacks;
  743. const char *profile_encoder_encode_name;
  744. };
  745. extern struct obs_encoder_info *find_encoder(const char *id);
  746. extern bool obs_encoder_initialize(obs_encoder_t *encoder);
  747. extern void obs_encoder_shutdown(obs_encoder_t *encoder);
  748. extern void obs_encoder_start(obs_encoder_t *encoder,
  749. void (*new_packet)(void *param, struct encoder_packet *packet),
  750. void *param);
  751. extern void obs_encoder_stop(obs_encoder_t *encoder,
  752. void (*new_packet)(void *param, struct encoder_packet *packet),
  753. void *param);
  754. extern void obs_encoder_add_output(struct obs_encoder *encoder,
  755. struct obs_output *output);
  756. extern void obs_encoder_remove_output(struct obs_encoder *encoder,
  757. struct obs_output *output);
  758. void obs_encoder_destroy(obs_encoder_t *encoder);
  759. /* ------------------------------------------------------------------------- */
  760. /* services */
  761. struct obs_weak_service {
  762. struct obs_weak_ref ref;
  763. struct obs_service *service;
  764. };
  765. struct obs_service {
  766. struct obs_context_data context;
  767. struct obs_service_info info;
  768. struct obs_weak_service *control;
  769. /* indicates ownership of the info.id buffer */
  770. bool owns_info_id;
  771. bool active;
  772. bool destroy;
  773. struct obs_output *output;
  774. };
  775. extern const struct obs_service_info *find_service(const char *id);
  776. extern void obs_service_activate(struct obs_service *service);
  777. extern void obs_service_deactivate(struct obs_service *service, bool remove);
  778. extern bool obs_service_initialize(struct obs_service *service,
  779. struct obs_output *output);
  780. void obs_service_destroy(obs_service_t *service);