obs-internal.h 43 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522
  1. /******************************************************************************
  2. Copyright (C) 2023 by Lain 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/deque.h"
  18. #include "util/dstr.h"
  19. #include "util/threading.h"
  20. #include "util/platform.h"
  21. #include "util/profiler.h"
  22. #include "util/task.h"
  23. #include "util/uthash.h"
  24. #include "util/array-serializer.h"
  25. #include "callback/signal.h"
  26. #include "callback/proc.h"
  27. #include "graphics/graphics.h"
  28. #include "graphics/matrix4.h"
  29. #include "media-io/audio-resampler.h"
  30. #include "media-io/video-io.h"
  31. #include "media-io/audio-io.h"
  32. #include "obs.h"
  33. #include <obsversion.h>
  34. #include <caption/caption.h>
  35. /* Custom helpers for the UUID hash table */
  36. #define HASH_FIND_UUID(head, uuid, out) HASH_FIND(hh_uuid, head, uuid, UUID_STR_LENGTH, out)
  37. #define HASH_ADD_UUID(head, uuid_field, add) HASH_ADD(hh_uuid, head, uuid_field[0], UUID_STR_LENGTH, add)
  38. #define NUM_TEXTURES 2
  39. #define NUM_CHANNELS 3
  40. #define MICROSECOND_DEN 1000000
  41. #define NUM_ENCODE_TEXTURES 10
  42. #define NUM_ENCODE_TEXTURE_FRAMES_TO_WAIT 1
  43. static inline int64_t packet_dts_usec(struct encoder_packet *packet)
  44. {
  45. return packet->dts * MICROSECOND_DEN / packet->timebase_den;
  46. }
  47. struct tick_callback {
  48. void (*tick)(void *param, float seconds);
  49. void *param;
  50. };
  51. struct draw_callback {
  52. void (*draw)(void *param, uint32_t cx, uint32_t cy);
  53. void *param;
  54. };
  55. struct rendered_callback {
  56. void (*rendered)(void *param);
  57. void *param;
  58. };
  59. struct packet_callback {
  60. void (*packet_cb)(obs_output_t *output, struct encoder_packet *pkt, struct encoder_packet_time *pkt_time,
  61. void *param);
  62. void *param;
  63. };
  64. struct reconnect_callback {
  65. bool (*reconnect_cb)(void *data, obs_output_t *output, int code);
  66. void *param;
  67. };
  68. /* ------------------------------------------------------------------------- */
  69. /* validity checks */
  70. static inline bool obs_object_valid(const void *obj, const char *f, const char *t)
  71. {
  72. if (!obj) {
  73. blog(LOG_DEBUG, "%s: Null '%s' parameter", f, t);
  74. return false;
  75. }
  76. return true;
  77. }
  78. #define obs_ptr_valid(ptr, func) obs_object_valid(ptr, func, #ptr)
  79. #define obs_source_valid obs_ptr_valid
  80. #define obs_output_valid obs_ptr_valid
  81. #define obs_encoder_valid obs_ptr_valid
  82. #define obs_service_valid obs_ptr_valid
  83. /* ------------------------------------------------------------------------- */
  84. /* modules */
  85. struct obs_module {
  86. char *mod_name;
  87. const char *file;
  88. char *bin_path;
  89. char *data_path;
  90. void *module;
  91. bool loaded;
  92. enum obs_module_load_state load_state;
  93. bool (*load)(void);
  94. void (*unload)(void);
  95. void (*post_load)(void);
  96. void (*set_locale)(const char *locale);
  97. bool (*get_string)(const char *lookup_string, const char **translated_string);
  98. void (*free_locale)(void);
  99. uint32_t (*ver)(void);
  100. void (*set_pointer)(obs_module_t *module);
  101. const char *(*name)(void);
  102. const char *(*description)(void);
  103. const char *(*author)(void);
  104. struct obs_module_metadata *metadata;
  105. struct obs_module *next;
  106. DARRAY(char *) sources;
  107. DARRAY(char *) outputs;
  108. DARRAY(char *) encoders;
  109. DARRAY(char *) services;
  110. };
  111. struct obs_disabled_module {
  112. char *mod_name;
  113. enum obs_module_load_state load_state;
  114. struct obs_module_metadata *metadata;
  115. struct obs_disabled_module *next;
  116. DARRAY(char *) sources;
  117. DARRAY(char *) outputs;
  118. DARRAY(char *) encoders;
  119. DARRAY(char *) services;
  120. };
  121. extern void free_module(struct obs_module *mod);
  122. struct obs_module_path {
  123. char *bin;
  124. char *data;
  125. };
  126. static inline void free_module_path(struct obs_module_path *omp)
  127. {
  128. if (omp) {
  129. bfree(omp->bin);
  130. bfree(omp->data);
  131. }
  132. }
  133. struct obs_module_metadata {
  134. char *display_name;
  135. char *version;
  136. char *id;
  137. char *os_arch;
  138. char *description;
  139. char *long_description;
  140. bool has_icon;
  141. bool has_banner;
  142. char *repository_url;
  143. char *support_url;
  144. char *website_url;
  145. char *name;
  146. };
  147. static inline void free_module_metadata(struct obs_module_metadata *omi)
  148. {
  149. if (omi) {
  150. bfree(omi->display_name);
  151. bfree(omi->version);
  152. bfree(omi->id);
  153. bfree(omi->os_arch);
  154. bfree(omi->description);
  155. bfree(omi->long_description);
  156. bfree(omi->repository_url);
  157. bfree(omi->support_url);
  158. bfree(omi->website_url);
  159. bfree(omi->name);
  160. }
  161. }
  162. static inline bool check_path(const char *data, const char *path, struct dstr *output)
  163. {
  164. dstr_copy(output, path);
  165. dstr_cat(output, data);
  166. return os_file_exists(output->array);
  167. }
  168. /* ------------------------------------------------------------------------- */
  169. /* hotkeys */
  170. struct obs_hotkey {
  171. obs_hotkey_id id;
  172. char *name;
  173. char *description;
  174. obs_hotkey_func func;
  175. void *data;
  176. int pressed;
  177. obs_hotkey_registerer_t registerer_type;
  178. void *registerer;
  179. obs_hotkey_id pair_partner_id;
  180. UT_hash_handle hh;
  181. };
  182. struct obs_hotkey_pair {
  183. obs_hotkey_pair_id pair_id;
  184. obs_hotkey_id id[2];
  185. obs_hotkey_active_func func[2];
  186. bool pressed0;
  187. bool pressed1;
  188. void *data[2];
  189. UT_hash_handle hh;
  190. };
  191. typedef struct obs_hotkey_pair obs_hotkey_pair_t;
  192. typedef struct obs_hotkeys_platform obs_hotkeys_platform_t;
  193. void *obs_hotkey_thread(void *param);
  194. struct obs_core_hotkeys;
  195. bool obs_hotkeys_platform_init(struct obs_core_hotkeys *hotkeys);
  196. void obs_hotkeys_platform_free(struct obs_core_hotkeys *hotkeys);
  197. bool obs_hotkeys_platform_is_pressed(obs_hotkeys_platform_t *context, obs_key_t key);
  198. const char *obs_get_hotkey_translation(obs_key_t key, const char *def);
  199. struct obs_context_data;
  200. void obs_hotkeys_context_release(struct obs_context_data *context);
  201. void obs_hotkeys_free(void);
  202. struct obs_hotkey_binding {
  203. obs_key_combination_t key;
  204. bool pressed;
  205. bool modifiers_match;
  206. obs_hotkey_id hotkey_id;
  207. obs_hotkey_t *hotkey;
  208. };
  209. struct obs_hotkey_name_map_item;
  210. void obs_hotkey_name_map_free(void);
  211. /* ------------------------------------------------------------------------- */
  212. /* views */
  213. enum view_type {
  214. INVALID_VIEW,
  215. MAIN_VIEW,
  216. AUX_VIEW,
  217. };
  218. struct obs_view {
  219. pthread_mutex_t channels_mutex;
  220. obs_source_t *channels[MAX_CHANNELS];
  221. enum view_type type;
  222. };
  223. extern bool obs_view_init(struct obs_view *view, enum view_type type);
  224. extern void obs_view_free(struct obs_view *view);
  225. /* ------------------------------------------------------------------------- */
  226. /* displays */
  227. struct obs_display {
  228. bool update_color_space;
  229. bool enabled;
  230. uint32_t cx, cy;
  231. uint32_t next_cx, next_cy;
  232. uint32_t background_color;
  233. gs_swapchain_t *swap;
  234. pthread_mutex_t draw_callbacks_mutex;
  235. pthread_mutex_t draw_info_mutex;
  236. DARRAY(struct draw_callback) draw_callbacks;
  237. bool use_clear_workaround;
  238. struct obs_display *next;
  239. struct obs_display **prev_next;
  240. };
  241. extern bool obs_display_init(struct obs_display *display, const struct gs_init_data *graphics_data);
  242. extern void obs_display_free(struct obs_display *display);
  243. /* ------------------------------------------------------------------------- */
  244. /* core */
  245. struct obs_vframe_info {
  246. uint64_t timestamp;
  247. int count;
  248. };
  249. struct obs_tex_frame {
  250. gs_texture_t *tex;
  251. gs_texture_t *tex_uv;
  252. uint32_t handle;
  253. uint64_t timestamp;
  254. uint64_t lock_key;
  255. int count;
  256. bool released;
  257. };
  258. struct obs_task_info {
  259. obs_task_t task;
  260. void *param;
  261. };
  262. struct obs_core_video_mix {
  263. struct obs_view *view;
  264. gs_stagesurf_t *active_copy_surfaces[NUM_TEXTURES][NUM_CHANNELS];
  265. gs_stagesurf_t *copy_surfaces[NUM_TEXTURES][NUM_CHANNELS];
  266. gs_texture_t *convert_textures[NUM_CHANNELS];
  267. gs_texture_t *convert_textures_encode[NUM_CHANNELS];
  268. #ifdef _WIN32
  269. gs_stagesurf_t *copy_surfaces_encode[NUM_TEXTURES];
  270. #endif
  271. gs_texture_t *render_texture;
  272. gs_texture_t *output_texture;
  273. enum gs_color_space render_space;
  274. bool texture_rendered;
  275. bool textures_copied[NUM_TEXTURES];
  276. bool texture_converted;
  277. bool using_nv12_tex;
  278. bool using_p010_tex;
  279. struct deque vframe_info_buffer;
  280. struct deque vframe_info_buffer_gpu;
  281. gs_stagesurf_t *mapped_surfaces[NUM_CHANNELS];
  282. int cur_texture;
  283. volatile long raw_active;
  284. volatile long gpu_encoder_active;
  285. bool gpu_was_active;
  286. bool raw_was_active;
  287. bool was_active;
  288. pthread_mutex_t gpu_encoder_mutex;
  289. struct deque gpu_encoder_queue;
  290. struct deque gpu_encoder_avail_queue;
  291. DARRAY(obs_encoder_t *) gpu_encoders;
  292. os_sem_t *gpu_encode_semaphore;
  293. os_event_t *gpu_encode_inactive;
  294. pthread_t gpu_encode_thread;
  295. bool gpu_encode_thread_initialized;
  296. volatile bool gpu_encode_stop;
  297. video_t *video;
  298. struct obs_video_info ovi;
  299. bool gpu_conversion;
  300. const char *conversion_techs[NUM_CHANNELS];
  301. bool conversion_needed;
  302. float conversion_width_i;
  303. float conversion_height_i;
  304. float color_matrix[16];
  305. bool encoder_only_mix;
  306. long encoder_refs;
  307. bool mix_audio;
  308. };
  309. extern struct obs_core_video_mix *obs_create_video_mix(struct obs_video_info *ovi);
  310. extern void obs_free_video_mix(struct obs_core_video_mix *video);
  311. struct obs_core_video {
  312. graphics_t *graphics;
  313. gs_effect_t *default_effect;
  314. gs_effect_t *default_rect_effect;
  315. gs_effect_t *opaque_effect;
  316. gs_effect_t *solid_effect;
  317. gs_effect_t *repeat_effect;
  318. gs_effect_t *conversion_effect;
  319. gs_effect_t *bicubic_effect;
  320. gs_effect_t *lanczos_effect;
  321. gs_effect_t *area_effect;
  322. gs_effect_t *bilinear_lowres_effect;
  323. gs_effect_t *premultiplied_alpha_effect;
  324. gs_samplerstate_t *point_sampler;
  325. uint64_t video_time;
  326. uint64_t video_frame_interval_ns;
  327. uint64_t video_half_frame_interval_ns;
  328. uint64_t video_avg_frame_time_ns;
  329. double video_fps;
  330. pthread_t video_thread;
  331. uint32_t total_frames;
  332. uint32_t lagged_frames;
  333. bool thread_initialized;
  334. gs_texture_t *transparent_texture;
  335. gs_effect_t *deinterlace_discard_effect;
  336. gs_effect_t *deinterlace_discard_2x_effect;
  337. gs_effect_t *deinterlace_linear_effect;
  338. gs_effect_t *deinterlace_linear_2x_effect;
  339. gs_effect_t *deinterlace_blend_effect;
  340. gs_effect_t *deinterlace_blend_2x_effect;
  341. gs_effect_t *deinterlace_yadif_effect;
  342. gs_effect_t *deinterlace_yadif_2x_effect;
  343. float sdr_white_level;
  344. float hdr_nominal_peak_level;
  345. pthread_mutex_t task_mutex;
  346. struct deque tasks;
  347. pthread_mutex_t encoder_group_mutex;
  348. DARRAY(obs_weak_encoder_t *) ready_encoder_groups;
  349. pthread_mutex_t mixes_mutex;
  350. DARRAY(struct obs_core_video_mix *) mixes;
  351. };
  352. extern void add_ready_encoder_group(obs_encoder_t *encoder);
  353. struct audio_monitor;
  354. struct obs_core_audio {
  355. audio_t *audio;
  356. DARRAY(struct obs_source *) render_order;
  357. DARRAY(struct obs_source *) root_nodes;
  358. uint64_t buffered_ts;
  359. struct deque buffered_timestamps;
  360. uint64_t buffering_wait_ticks;
  361. int total_buffering_ticks;
  362. int max_buffering_ticks;
  363. bool fixed_buffer;
  364. pthread_mutex_t monitoring_mutex;
  365. DARRAY(struct audio_monitor *) monitors;
  366. char *monitoring_device_name;
  367. char *monitoring_device_id;
  368. pthread_mutex_t task_mutex;
  369. struct deque tasks;
  370. struct obs_source *monitoring_duplicating_source;
  371. };
  372. /* user sources, output channels, and displays */
  373. struct obs_core_data {
  374. /* Hash tables (uthash) */
  375. struct obs_source *sources; /* Lookup by UUID (hh_uuid) */
  376. struct obs_source *public_sources; /* Lookup by name (hh) */
  377. struct obs_canvas *canvases; /* Lookup by UUID (hh_uuid) */
  378. struct obs_canvas *named_canvases; /* Lookup by name (hh) */
  379. /* Linked lists */
  380. struct obs_source *first_audio_source;
  381. struct obs_display *first_display;
  382. struct obs_output *first_output;
  383. struct obs_encoder *first_encoder;
  384. struct obs_service *first_service;
  385. pthread_mutex_t sources_mutex;
  386. pthread_mutex_t displays_mutex;
  387. pthread_mutex_t outputs_mutex;
  388. pthread_mutex_t encoders_mutex;
  389. pthread_mutex_t services_mutex;
  390. pthread_mutex_t audio_sources_mutex;
  391. pthread_mutex_t draw_callbacks_mutex;
  392. pthread_mutex_t canvases_mutex;
  393. DARRAY(struct draw_callback) draw_callbacks;
  394. DARRAY(struct rendered_callback) rendered_callbacks;
  395. DARRAY(struct tick_callback) tick_callbacks;
  396. /* Main canvas, guaranteed to exist for the lifetime of the program */
  397. struct obs_canvas *main_canvas;
  398. long long unnamed_index;
  399. obs_data_t *private_data;
  400. volatile bool valid;
  401. DARRAY(char *) protocols;
  402. DARRAY(obs_source_t *) sources_to_tick;
  403. };
  404. /* user hotkeys */
  405. struct obs_core_hotkeys {
  406. pthread_mutex_t mutex;
  407. obs_hotkey_t *hotkeys;
  408. obs_hotkey_id next_id;
  409. obs_hotkey_pair_t *hotkey_pairs;
  410. obs_hotkey_pair_id next_pair_id;
  411. pthread_t hotkey_thread;
  412. bool hotkey_thread_initialized;
  413. os_event_t *stop_event;
  414. bool thread_disable_press;
  415. bool strict_modifiers;
  416. bool reroute_hotkeys;
  417. DARRAY(obs_hotkey_binding_t) bindings;
  418. obs_hotkey_callback_router_func router_func;
  419. void *router_func_data;
  420. obs_hotkeys_platform_t *platform_context;
  421. pthread_once_t name_map_init_token;
  422. struct obs_hotkey_name_map_item *name_map;
  423. signal_handler_t *signals;
  424. char *translations[OBS_KEY_LAST_VALUE];
  425. char *mute;
  426. char *unmute;
  427. char *push_to_mute;
  428. char *push_to_talk;
  429. char *sceneitem_show;
  430. char *sceneitem_hide;
  431. };
  432. typedef DARRAY(struct obs_source_info) obs_source_info_array_t;
  433. struct obs_core {
  434. struct obs_module *first_module;
  435. struct obs_module *first_disabled_module;
  436. DARRAY(struct obs_module_path) module_paths;
  437. DARRAY(char *) safe_modules;
  438. DARRAY(char *) disabled_modules;
  439. DARRAY(char *) core_modules;
  440. obs_source_info_array_t source_types;
  441. obs_source_info_array_t input_types;
  442. obs_source_info_array_t filter_types;
  443. obs_source_info_array_t transition_types;
  444. DARRAY(struct obs_output_info) output_types;
  445. DARRAY(struct obs_encoder_info) encoder_types;
  446. DARRAY(struct obs_service_info) service_types;
  447. signal_handler_t *signals;
  448. proc_handler_t *procs;
  449. char *locale;
  450. char *module_config_path;
  451. bool name_store_owned;
  452. profiler_name_store_t *name_store;
  453. /* segmented into multiple sub-structures to keep things a bit more
  454. * clean and organized */
  455. struct obs_core_video video;
  456. struct obs_core_audio audio;
  457. struct obs_core_data data;
  458. struct obs_core_hotkeys hotkeys;
  459. os_task_queue_t *destruction_task_thread;
  460. obs_task_handler_t ui_task_handler;
  461. };
  462. extern struct obs_core *obs;
  463. struct obs_graphics_context {
  464. uint64_t last_time;
  465. uint64_t interval;
  466. uint64_t frame_time_total_ns;
  467. uint64_t fps_total_ns;
  468. uint32_t fps_total_frames;
  469. const char *video_thread_name;
  470. };
  471. extern void *obs_graphics_thread(void *param);
  472. extern bool obs_graphics_thread_loop(struct obs_graphics_context *context);
  473. #ifdef __APPLE__
  474. extern void *obs_graphics_thread_autorelease(void *param);
  475. extern bool obs_graphics_thread_loop_autorelease(struct obs_graphics_context *context);
  476. #endif
  477. extern gs_effect_t *obs_load_effect(gs_effect_t **effect, const char *file);
  478. extern bool audio_callback(void *param, uint64_t start_ts_in, uint64_t end_ts_in, uint64_t *out_ts, uint32_t mixers,
  479. struct audio_output_data *mixes);
  480. extern struct obs_core_video_mix *get_mix_for_video(video_t *video);
  481. extern void start_raw_video(video_t *video, const struct video_scale_info *conversion, uint32_t frame_rate_divisor,
  482. void (*callback)(void *param, struct video_data *frame), void *param);
  483. extern void stop_raw_video(video_t *video, void (*callback)(void *param, struct video_data *frame), void *param);
  484. /* ------------------------------------------------------------------------- */
  485. /* obs shared context data */
  486. struct obs_weak_ref {
  487. volatile long refs;
  488. volatile long weak_refs;
  489. };
  490. struct obs_weak_object {
  491. struct obs_weak_ref ref;
  492. struct obs_context_data *object;
  493. };
  494. typedef void (*obs_destroy_cb)(void *obj);
  495. struct obs_context_data {
  496. char *name;
  497. const char *uuid;
  498. void *data;
  499. obs_data_t *settings;
  500. signal_handler_t *signals;
  501. proc_handler_t *procs;
  502. enum obs_obj_type type;
  503. struct obs_weak_object *control;
  504. obs_destroy_cb destroy;
  505. DARRAY(obs_hotkey_id) hotkeys;
  506. DARRAY(obs_hotkey_pair_id) hotkey_pairs;
  507. obs_data_t *hotkey_data;
  508. DARRAY(char *) rename_cache;
  509. pthread_mutex_t rename_cache_mutex;
  510. pthread_mutex_t *mutex;
  511. struct obs_context_data *next;
  512. struct obs_context_data **prev_next;
  513. UT_hash_handle hh;
  514. UT_hash_handle hh_uuid;
  515. bool private;
  516. };
  517. extern bool obs_context_data_init(struct obs_context_data *context, enum obs_obj_type type, obs_data_t *settings,
  518. const char *name, const char *uuid, obs_data_t *hotkey_data, bool private);
  519. extern void obs_context_init_control(struct obs_context_data *context, void *object, obs_destroy_cb destroy);
  520. extern void obs_context_data_free(struct obs_context_data *context);
  521. extern void obs_context_data_insert(struct obs_context_data *context, pthread_mutex_t *mutex, void *first);
  522. extern void obs_context_data_insert_name(struct obs_context_data *context, pthread_mutex_t *mutex, void *first);
  523. extern void obs_context_data_insert_uuid(struct obs_context_data *context, pthread_mutex_t *mutex, void *first_uuid);
  524. extern void obs_context_data_remove(struct obs_context_data *context);
  525. extern void obs_context_data_remove_name(struct obs_context_data *context, pthread_mutex_t *mutex, void *phead);
  526. extern void obs_context_data_remove_uuid(struct obs_context_data *context, pthread_mutex_t *mutex, void *puuid_head);
  527. extern void obs_context_wait(struct obs_context_data *context);
  528. extern void obs_context_data_setname(struct obs_context_data *context, const char *name);
  529. extern void obs_context_data_setname_ht(struct obs_context_data *context, const char *name, void *phead);
  530. /* ------------------------------------------------------------------------- */
  531. /* ref-counting */
  532. static inline void obs_ref_addref(struct obs_weak_ref *ref)
  533. {
  534. os_atomic_inc_long(&ref->refs);
  535. }
  536. static inline bool obs_ref_release(struct obs_weak_ref *ref)
  537. {
  538. return os_atomic_dec_long(&ref->refs) == -1;
  539. }
  540. static inline void obs_weak_ref_addref(struct obs_weak_ref *ref)
  541. {
  542. os_atomic_inc_long(&ref->weak_refs);
  543. }
  544. static inline bool obs_weak_ref_release(struct obs_weak_ref *ref)
  545. {
  546. return os_atomic_dec_long(&ref->weak_refs) == -1;
  547. }
  548. static inline bool obs_weak_ref_get_ref(struct obs_weak_ref *ref)
  549. {
  550. long owners = os_atomic_load_long(&ref->refs);
  551. while (owners > -1) {
  552. if (os_atomic_compare_exchange_long(&ref->refs, &owners, owners + 1)) {
  553. return true;
  554. }
  555. }
  556. return false;
  557. }
  558. static inline bool obs_weak_ref_expired(struct obs_weak_ref *ref)
  559. {
  560. long owners = os_atomic_load_long(&ref->refs);
  561. return owners < 0;
  562. }
  563. /* ------------------------------------------------------------------------- */
  564. /* canvases */
  565. struct obs_weak_canvas {
  566. struct obs_weak_ref ref;
  567. struct obs_canvas *canvas;
  568. };
  569. struct obs_canvas {
  570. struct obs_context_data context;
  571. /* obs_canvas_flags */
  572. uint32_t flags;
  573. /* Video info for this canvas, FPS ignored */
  574. struct obs_video_info ovi;
  575. /* Hash table containing scenes (and groups) associated with this canvas */
  576. struct obs_source *sources;
  577. pthread_mutex_t sources_mutex;
  578. /* For now, canvas objects mainly act as a proxy for the existing view and video mix objects,
  579. * though this may change in the future. */
  580. struct obs_view view;
  581. struct obs_core_video_mix *mix;
  582. };
  583. extern obs_canvas_t *obs_create_main_canvas(void);
  584. extern void obs_canvas_destroy(obs_canvas_t *canvas);
  585. extern void obs_canvas_clear_mix(obs_canvas_t *canvas);
  586. extern void obs_free_canvas_mixes(void);
  587. extern bool obs_canvas_reset_video_internal(obs_canvas_t *canvas, struct obs_video_info *ovi);
  588. extern void obs_canvas_insert_source(obs_canvas_t *canvas, obs_source_t *source);
  589. extern void obs_canvas_remove_source(obs_source_t *source);
  590. extern void obs_canvas_rename_source(obs_source_t *source, const char *name);
  591. /* ------------------------------------------------------------------------- */
  592. /* sources */
  593. struct async_frame {
  594. struct obs_source_frame *frame;
  595. long unused_count;
  596. bool used;
  597. };
  598. enum audio_action_type {
  599. AUDIO_ACTION_VOL,
  600. AUDIO_ACTION_MUTE,
  601. AUDIO_ACTION_PTT,
  602. AUDIO_ACTION_PTM,
  603. };
  604. struct audio_action {
  605. uint64_t timestamp;
  606. enum audio_action_type type;
  607. union {
  608. float vol;
  609. bool set;
  610. };
  611. };
  612. struct obs_weak_source {
  613. struct obs_weak_ref ref;
  614. struct obs_source *source;
  615. };
  616. struct audio_cb_info {
  617. obs_source_audio_capture_t callback;
  618. void *param;
  619. };
  620. struct caption_cb_info {
  621. obs_source_caption_t callback;
  622. void *param;
  623. };
  624. enum media_action_type {
  625. MEDIA_ACTION_NONE,
  626. MEDIA_ACTION_PLAY_PAUSE,
  627. MEDIA_ACTION_RESTART,
  628. MEDIA_ACTION_STOP,
  629. MEDIA_ACTION_NEXT,
  630. MEDIA_ACTION_PREVIOUS,
  631. MEDIA_ACTION_SET_TIME,
  632. };
  633. struct media_action {
  634. enum media_action_type type;
  635. union {
  636. bool pause;
  637. int64_t ms;
  638. };
  639. };
  640. struct obs_source {
  641. struct obs_context_data context;
  642. struct obs_source_info info;
  643. /* general exposed flags that can be set for the source */
  644. uint32_t flags;
  645. uint32_t default_flags;
  646. uint32_t last_obs_ver;
  647. /* indicates ownership of the info.id buffer */
  648. bool owns_info_id;
  649. /* signals to call the source update in the video thread */
  650. long defer_update_count;
  651. /* ensures show/hide are only called once */
  652. volatile long show_refs;
  653. /* ensures activate/deactivate are only called once */
  654. volatile long activate_refs;
  655. /* source is in the process of being destroyed */
  656. volatile long destroying;
  657. /* used to indicate that the source has been removed and all
  658. * references to it should be released (not exactly how I would prefer
  659. * to handle things but it's the best option) */
  660. bool removed;
  661. /* used to indicate if the source should show up when queried for user ui */
  662. bool temp_removed;
  663. bool active;
  664. bool showing;
  665. /* used to temporarily disable sources if needed */
  666. bool enabled;
  667. /* hint to allow sources to render more quickly */
  668. bool texcoords_centered;
  669. /* timing (if video is present, is based upon video) */
  670. volatile bool timing_set;
  671. volatile uint64_t timing_adjust;
  672. uint64_t resample_offset;
  673. uint64_t next_audio_ts_min;
  674. uint64_t next_audio_sys_ts_min;
  675. uint64_t last_frame_ts;
  676. uint64_t last_sys_timestamp;
  677. bool async_rendered;
  678. /* audio */
  679. bool audio_failed;
  680. bool audio_pending;
  681. bool pending_stop;
  682. bool audio_active;
  683. bool user_muted;
  684. bool muted;
  685. struct obs_source *next_audio_source;
  686. struct obs_source **prev_next_audio_source;
  687. uint64_t audio_ts;
  688. struct deque audio_input_buf[MAX_AUDIO_CHANNELS];
  689. size_t last_audio_input_buf_size;
  690. DARRAY(struct audio_action) audio_actions;
  691. float *audio_output_buf[MAX_AUDIO_MIXES][MAX_AUDIO_CHANNELS];
  692. float *audio_mix_buf[MAX_AUDIO_CHANNELS];
  693. struct resample_info sample_info;
  694. audio_resampler_t *resampler;
  695. pthread_mutex_t audio_actions_mutex;
  696. pthread_mutex_t audio_buf_mutex;
  697. pthread_mutex_t audio_mutex;
  698. pthread_mutex_t audio_cb_mutex;
  699. DARRAY(struct audio_cb_info) audio_cb_list;
  700. struct obs_audio_data audio_data;
  701. size_t audio_storage_size;
  702. uint32_t audio_mixers;
  703. float user_volume;
  704. float volume;
  705. int64_t sync_offset;
  706. int64_t last_sync_offset;
  707. float balance;
  708. /* audio_is_duplicated: tracks whether a source appears multiple times in the audio tree during this tick */
  709. bool audio_is_duplicated;
  710. /* async video data */
  711. gs_texture_t *async_textures[MAX_AV_PLANES];
  712. gs_texrender_t *async_texrender;
  713. struct obs_source_frame *cur_async_frame;
  714. bool async_gpu_conversion;
  715. enum video_format async_format;
  716. bool async_full_range;
  717. uint8_t async_trc;
  718. enum video_format async_cache_format;
  719. bool async_cache_full_range;
  720. uint8_t async_cache_trc;
  721. enum gs_color_format async_texture_formats[MAX_AV_PLANES];
  722. int async_channel_count;
  723. long async_rotation;
  724. bool async_flip;
  725. bool async_linear_alpha;
  726. bool async_active;
  727. bool async_update_texture;
  728. bool async_unbuffered;
  729. bool async_decoupled;
  730. struct obs_source_frame *async_preload_frame;
  731. DARRAY(struct async_frame) async_cache;
  732. DARRAY(struct obs_source_frame *) async_frames;
  733. pthread_mutex_t async_mutex;
  734. uint32_t async_width;
  735. uint32_t async_height;
  736. uint32_t async_cache_width;
  737. uint32_t async_cache_height;
  738. uint32_t async_convert_width[MAX_AV_PLANES];
  739. uint32_t async_convert_height[MAX_AV_PLANES];
  740. uint64_t async_last_rendered_ts;
  741. pthread_mutex_t caption_cb_mutex;
  742. DARRAY(struct caption_cb_info) caption_cb_list;
  743. /* async video deinterlacing */
  744. uint64_t deinterlace_offset;
  745. uint64_t deinterlace_frame_ts;
  746. gs_effect_t *deinterlace_effect;
  747. struct obs_source_frame *prev_async_frame;
  748. gs_texture_t *async_prev_textures[MAX_AV_PLANES];
  749. gs_texrender_t *async_prev_texrender;
  750. uint32_t deinterlace_half_duration;
  751. enum obs_deinterlace_mode deinterlace_mode;
  752. bool deinterlace_top_first;
  753. bool deinterlace_rendered;
  754. /* filters */
  755. struct obs_source *filter_parent;
  756. struct obs_source *filter_target;
  757. DARRAY(struct obs_source *) filters;
  758. pthread_mutex_t filter_mutex;
  759. gs_texrender_t *filter_texrender;
  760. enum obs_allow_direct_render allow_direct;
  761. bool rendering_filter;
  762. bool filter_bypass_active;
  763. /* sources specific hotkeys */
  764. obs_hotkey_pair_id mute_unmute_key;
  765. obs_hotkey_id push_to_mute_key;
  766. obs_hotkey_id push_to_talk_key;
  767. bool push_to_mute_enabled;
  768. bool push_to_mute_pressed;
  769. bool user_push_to_mute_pressed;
  770. bool push_to_talk_enabled;
  771. bool push_to_talk_pressed;
  772. bool user_push_to_talk_pressed;
  773. uint64_t push_to_mute_delay;
  774. uint64_t push_to_mute_stop_time;
  775. uint64_t push_to_talk_delay;
  776. uint64_t push_to_talk_stop_time;
  777. /* transitions */
  778. uint64_t transition_start_time;
  779. uint64_t transition_duration;
  780. pthread_mutex_t transition_tex_mutex;
  781. gs_texrender_t *transition_texrender[2];
  782. pthread_mutex_t transition_mutex;
  783. obs_source_t *transition_sources[2];
  784. float transition_manual_clamp;
  785. float transition_manual_torque;
  786. float transition_manual_target;
  787. float transition_manual_val;
  788. bool transitioning_video;
  789. bool transitioning_audio;
  790. bool transition_source_active[2];
  791. uint32_t transition_alignment;
  792. uint32_t transition_actual_cx;
  793. uint32_t transition_actual_cy;
  794. uint32_t transition_cx;
  795. uint32_t transition_cy;
  796. uint32_t transition_fixed_duration;
  797. bool transition_use_fixed_duration;
  798. enum obs_transition_mode transition_mode;
  799. enum obs_transition_scale_type transition_scale_type;
  800. struct matrix4 transition_matrices[2];
  801. /* color space */
  802. gs_texrender_t *color_space_texrender;
  803. /* audio monitoring */
  804. struct audio_monitor *monitor;
  805. enum obs_monitoring_type monitoring_type;
  806. /* media action queue */
  807. DARRAY(struct media_action) media_actions;
  808. pthread_mutex_t media_actions_mutex;
  809. /* private data */
  810. obs_data_t *private_settings;
  811. /* canvas this source belongs to (only used for scenes) */
  812. obs_weak_canvas_t *canvas;
  813. };
  814. extern struct obs_source_info *get_source_info(const char *id);
  815. extern struct obs_source_info *get_source_info2(const char *unversioned_id, uint32_t ver);
  816. extern bool obs_source_init_context(struct obs_source *source, obs_data_t *settings, const char *name, const char *uuid,
  817. obs_data_t *hotkey_data, bool private);
  818. extern bool obs_transition_init(obs_source_t *transition);
  819. extern void obs_transition_free(obs_source_t *transition);
  820. extern void obs_transition_tick(obs_source_t *transition, float t);
  821. extern void obs_transition_enum_sources(obs_source_t *transition, obs_source_enum_proc_t enum_callback, void *param);
  822. extern void obs_transition_save(obs_source_t *source, obs_data_t *data);
  823. extern void obs_transition_load(obs_source_t *source, obs_data_t *data);
  824. struct audio_monitor *audio_monitor_create(obs_source_t *source);
  825. void audio_monitor_reset(struct audio_monitor *monitor);
  826. extern void audio_monitor_destroy(struct audio_monitor *monitor);
  827. extern obs_source_t *obs_source_create_canvas(obs_canvas_t *canvas, const char *id, const char *name,
  828. obs_data_t *settings, obs_data_t *hotkey_data);
  829. extern obs_source_t *obs_source_create_set_last_ver(obs_canvas_t *canvas, const char *id, const char *name,
  830. const char *uuid, obs_data_t *settings, obs_data_t *hotkey_data,
  831. uint32_t last_obs_ver, bool is_private);
  832. extern void obs_source_destroy(struct obs_source *source);
  833. extern void obs_source_addref(obs_source_t *source);
  834. static inline void obs_source_dosignal(struct obs_source *source, const char *signal_obs, const char *signal_source)
  835. {
  836. struct calldata data;
  837. uint8_t stack[128];
  838. calldata_init_fixed(&data, stack, sizeof(stack));
  839. calldata_set_ptr(&data, "source", source);
  840. if (signal_obs && !source->context.private)
  841. signal_handler_signal(obs->signals, signal_obs, &data);
  842. if (signal_source)
  843. signal_handler_signal(source->context.signals, signal_source, &data);
  844. }
  845. static inline void obs_source_dosignal_canvas(struct obs_source *source, struct obs_canvas *canvas,
  846. const char *signal_obs, const char *signal_source)
  847. {
  848. struct calldata data;
  849. uint8_t stack[128];
  850. calldata_init_fixed(&data, stack, sizeof(stack));
  851. calldata_set_ptr(&data, "source", source);
  852. calldata_set_ptr(&data, "canvas", canvas);
  853. if (signal_obs && !source->context.private)
  854. signal_handler_signal(obs->signals, signal_obs, &data);
  855. if (signal_source)
  856. signal_handler_signal(source->context.signals, signal_source, &data);
  857. }
  858. /* maximum timestamp variance in nanoseconds */
  859. #define MAX_TS_VAR 2000000000ULL
  860. static inline bool frame_out_of_bounds(const obs_source_t *source, uint64_t ts)
  861. {
  862. if (ts < source->last_frame_ts)
  863. return ((source->last_frame_ts - ts) > MAX_TS_VAR);
  864. else
  865. return ((ts - source->last_frame_ts) > MAX_TS_VAR);
  866. }
  867. static inline enum gs_color_format convert_video_format(enum video_format format, enum video_trc trc)
  868. {
  869. switch (trc) {
  870. case VIDEO_TRC_PQ:
  871. case VIDEO_TRC_HLG:
  872. return GS_RGBA16F;
  873. default:
  874. switch (format) {
  875. case VIDEO_FORMAT_RGBA:
  876. return GS_RGBA;
  877. case VIDEO_FORMAT_BGRA:
  878. case VIDEO_FORMAT_I40A:
  879. case VIDEO_FORMAT_I42A:
  880. case VIDEO_FORMAT_YUVA:
  881. case VIDEO_FORMAT_AYUV:
  882. return GS_BGRA;
  883. case VIDEO_FORMAT_I010:
  884. case VIDEO_FORMAT_P010:
  885. case VIDEO_FORMAT_I210:
  886. case VIDEO_FORMAT_I412:
  887. case VIDEO_FORMAT_YA2L:
  888. case VIDEO_FORMAT_P216:
  889. case VIDEO_FORMAT_P416:
  890. case VIDEO_FORMAT_V210:
  891. case VIDEO_FORMAT_R10L:
  892. return GS_RGBA16F;
  893. default:
  894. return GS_BGRX;
  895. }
  896. }
  897. }
  898. static inline enum gs_color_space convert_video_space(enum video_format format, enum video_trc trc)
  899. {
  900. enum gs_color_space space = GS_CS_SRGB;
  901. if (convert_video_format(format, trc) == GS_RGBA16F) {
  902. switch (trc) {
  903. case VIDEO_TRC_DEFAULT:
  904. case VIDEO_TRC_SRGB:
  905. space = GS_CS_SRGB_16F;
  906. break;
  907. case VIDEO_TRC_PQ:
  908. case VIDEO_TRC_HLG:
  909. space = GS_CS_709_EXTENDED;
  910. }
  911. }
  912. return space;
  913. }
  914. extern void obs_source_set_texcoords_centered(obs_source_t *source, bool centered);
  915. extern void obs_source_activate(obs_source_t *source, enum view_type type);
  916. extern void obs_source_deactivate(obs_source_t *source, enum view_type type);
  917. extern void obs_source_video_tick(obs_source_t *source, float seconds);
  918. extern float obs_source_get_target_volume(obs_source_t *source, obs_source_t *target);
  919. extern uint64_t obs_source_get_last_async_ts(const obs_source_t *source);
  920. extern void obs_source_audio_render(obs_source_t *source, uint32_t mixers, size_t channels, size_t sample_rate,
  921. size_t size);
  922. extern void add_alignment(struct vec2 *v, uint32_t align, int cx, int cy);
  923. extern struct obs_source_frame *filter_async_video(obs_source_t *source, struct obs_source_frame *in);
  924. extern bool update_async_texture(struct obs_source *source, const struct obs_source_frame *frame, gs_texture_t *tex,
  925. gs_texrender_t *texrender);
  926. extern bool update_async_textures(struct obs_source *source, const struct obs_source_frame *frame,
  927. gs_texture_t *tex[MAX_AV_PLANES], gs_texrender_t *texrender);
  928. extern bool set_async_texture_size(struct obs_source *source, const struct obs_source_frame *frame);
  929. extern void remove_async_frame(obs_source_t *source, struct obs_source_frame *frame);
  930. extern void set_deinterlace_texture_size(obs_source_t *source);
  931. extern void deinterlace_process_last_frame(obs_source_t *source, uint64_t sys_time);
  932. extern void deinterlace_update_async_video(obs_source_t *source);
  933. extern void deinterlace_render(obs_source_t *s);
  934. /* ------------------------------------------------------------------------- */
  935. /* outputs */
  936. enum delay_msg {
  937. DELAY_MSG_PACKET,
  938. DELAY_MSG_START,
  939. DELAY_MSG_STOP,
  940. };
  941. struct delay_data {
  942. enum delay_msg msg;
  943. uint64_t ts;
  944. struct encoder_packet packet;
  945. bool packet_time_valid;
  946. struct encoder_packet_time packet_time;
  947. };
  948. typedef void (*encoded_callback_t)(void *data, struct encoder_packet *packet, struct encoder_packet_time *frame_time);
  949. struct obs_weak_output {
  950. struct obs_weak_ref ref;
  951. struct obs_output *output;
  952. };
  953. #define CAPTION_LINE_CHARS (32)
  954. #define CAPTION_LINE_BYTES (4 * CAPTION_LINE_CHARS)
  955. struct caption_text {
  956. char text[CAPTION_LINE_BYTES + 1];
  957. double display_duration;
  958. struct caption_text *next;
  959. };
  960. struct caption_track_data {
  961. struct caption_text *caption_head;
  962. struct caption_text *caption_tail;
  963. pthread_mutex_t caption_mutex;
  964. double caption_timestamp;
  965. double last_caption_timestamp;
  966. struct deque caption_data;
  967. };
  968. struct pause_data {
  969. pthread_mutex_t mutex;
  970. uint64_t last_video_ts;
  971. uint64_t ts_start;
  972. uint64_t ts_end;
  973. uint64_t ts_offset;
  974. };
  975. extern bool video_pause_check(struct pause_data *pause, uint64_t timestamp);
  976. extern bool audio_pause_check(struct pause_data *pause, struct audio_data *data, size_t sample_rate);
  977. extern void pause_reset(struct pause_data *pause);
  978. enum keyframe_group_track_status {
  979. KEYFRAME_TRACK_STATUS_NOT_SEEN = 0,
  980. KEYFRAME_TRACK_STATUS_SEEN = 1,
  981. KEYFRAME_TRACK_STATUS_SKIPPED = 2,
  982. };
  983. struct keyframe_group_data {
  984. uintptr_t group_id;
  985. int64_t pts;
  986. uint32_t required_tracks;
  987. enum keyframe_group_track_status seen_on_track[MAX_OUTPUT_VIDEO_ENCODERS];
  988. };
  989. struct obs_output {
  990. struct obs_context_data context;
  991. struct obs_output_info info;
  992. /* indicates ownership of the info.id buffer */
  993. bool owns_info_id;
  994. bool received_video[MAX_OUTPUT_VIDEO_ENCODERS];
  995. DARRAY(struct keyframe_group_data) keyframe_group_tracking;
  996. bool received_audio;
  997. volatile bool data_active;
  998. volatile bool end_data_capture_thread_active;
  999. int64_t video_offsets[MAX_OUTPUT_VIDEO_ENCODERS];
  1000. int64_t audio_offsets[MAX_OUTPUT_AUDIO_ENCODERS];
  1001. int64_t highest_audio_ts;
  1002. int64_t highest_video_ts[MAX_OUTPUT_VIDEO_ENCODERS];
  1003. pthread_t end_data_capture_thread;
  1004. os_event_t *stopping_event;
  1005. pthread_mutex_t interleaved_mutex;
  1006. DARRAY(struct encoder_packet) interleaved_packets;
  1007. size_t interleaver_max_batch_size;
  1008. int stop_code;
  1009. int reconnect_retry_sec;
  1010. int reconnect_retry_max;
  1011. int reconnect_retries;
  1012. uint32_t reconnect_retry_cur_msec;
  1013. float reconnect_retry_exp;
  1014. pthread_t reconnect_thread;
  1015. os_event_t *reconnect_stop_event;
  1016. volatile bool reconnecting;
  1017. volatile bool reconnect_thread_active;
  1018. uint32_t starting_drawn_count;
  1019. uint32_t starting_lagged_count;
  1020. int total_frames;
  1021. volatile bool active;
  1022. volatile bool paused;
  1023. video_t *video;
  1024. audio_t *audio;
  1025. obs_encoder_t *video_encoders[MAX_OUTPUT_VIDEO_ENCODERS];
  1026. obs_encoder_t *audio_encoders[MAX_OUTPUT_AUDIO_ENCODERS];
  1027. obs_service_t *service;
  1028. size_t mixer_mask;
  1029. struct pause_data pause;
  1030. struct deque audio_buffer[MAX_AUDIO_MIXES][MAX_AV_PLANES];
  1031. uint64_t audio_start_ts;
  1032. uint64_t video_start_ts;
  1033. size_t audio_size;
  1034. size_t planes;
  1035. size_t sample_rate;
  1036. size_t total_audio_frames;
  1037. uint32_t scaled_width;
  1038. uint32_t scaled_height;
  1039. bool video_conversion_set;
  1040. bool audio_conversion_set;
  1041. struct video_scale_info video_conversion;
  1042. struct audio_convert_info audio_conversion;
  1043. // captions are output per track
  1044. struct caption_track_data *caption_tracks[MAX_OUTPUT_VIDEO_ENCODERS];
  1045. DARRAY(struct encoder_packet_time)
  1046. encoder_packet_times[MAX_OUTPUT_VIDEO_ENCODERS];
  1047. /* Packet callbacks */
  1048. pthread_mutex_t pkt_callbacks_mutex;
  1049. DARRAY(struct packet_callback) pkt_callbacks;
  1050. struct reconnect_callback reconnect_callback;
  1051. bool valid;
  1052. uint64_t active_delay_ns;
  1053. encoded_callback_t delay_callback;
  1054. struct deque delay_data; /* struct delay_data */
  1055. pthread_mutex_t delay_mutex;
  1056. uint32_t delay_sec;
  1057. uint32_t delay_flags;
  1058. uint32_t delay_cur_flags;
  1059. volatile long delay_restart_refs;
  1060. volatile bool delay_active;
  1061. volatile bool delay_capturing;
  1062. char *last_error_message;
  1063. float audio_data[MAX_AUDIO_CHANNELS][AUDIO_OUTPUT_FRAMES];
  1064. };
  1065. static inline void do_output_signal(struct obs_output *output, const char *signal)
  1066. {
  1067. struct calldata params = {0};
  1068. calldata_set_ptr(&params, "output", output);
  1069. signal_handler_signal(output->context.signals, signal, &params);
  1070. calldata_free(&params);
  1071. }
  1072. extern void process_delay(void *data, struct encoder_packet *packet, struct encoder_packet_time *packet_time);
  1073. extern void obs_output_cleanup_delay(obs_output_t *output);
  1074. extern bool obs_output_delay_start(obs_output_t *output);
  1075. extern void obs_output_delay_stop(obs_output_t *output);
  1076. extern bool obs_output_actual_start(obs_output_t *output);
  1077. extern void obs_output_actual_stop(obs_output_t *output, bool force, uint64_t ts);
  1078. extern const struct obs_output_info *find_output(const char *id);
  1079. extern void obs_output_remove_encoder(struct obs_output *output, struct obs_encoder *encoder);
  1080. extern void obs_encoder_packet_create_instance(struct encoder_packet *dst, const struct encoder_packet *src);
  1081. void obs_output_destroy(obs_output_t *output);
  1082. /* ------------------------------------------------------------------------- */
  1083. /* encoders */
  1084. struct obs_weak_encoder {
  1085. struct obs_weak_ref ref;
  1086. struct obs_encoder *encoder;
  1087. };
  1088. struct encoder_callback {
  1089. bool sent_first_packet;
  1090. encoded_callback_t new_packet;
  1091. void *param;
  1092. };
  1093. struct obs_encoder_group {
  1094. pthread_mutex_t mutex;
  1095. /* allows group to be destroyed even if some encoders are active */
  1096. bool destroy_on_stop;
  1097. /* holds strong references to all encoders */
  1098. DARRAY(struct obs_encoder *) encoders;
  1099. uint32_t num_encoders_started;
  1100. uint64_t start_timestamp;
  1101. };
  1102. struct obs_encoder {
  1103. struct obs_context_data context;
  1104. struct obs_encoder_info info;
  1105. /* allows re-routing to another encoder */
  1106. struct obs_encoder_info orig_info;
  1107. pthread_mutex_t init_mutex;
  1108. uint32_t samplerate;
  1109. size_t planes;
  1110. size_t blocksize;
  1111. size_t framesize;
  1112. size_t framesize_bytes;
  1113. size_t mixer_idx;
  1114. /* OBS_SCALE_DISABLE indicates GPU scaling is disabled */
  1115. enum obs_scale_type gpu_scale_type;
  1116. uint32_t scaled_width;
  1117. uint32_t scaled_height;
  1118. enum video_format preferred_format;
  1119. enum video_colorspace preferred_space;
  1120. enum video_range_type preferred_range;
  1121. volatile bool active;
  1122. volatile bool paused;
  1123. bool initialized;
  1124. /* indicates ownership of the info.id buffer */
  1125. bool owns_info_id;
  1126. uint32_t timebase_num;
  1127. uint32_t timebase_den;
  1128. // allow outputting at fractions of main composition FPS,
  1129. // e.g. 60 FPS with frame_rate_divisor = 1 turns into 30 FPS
  1130. //
  1131. // a separate counter is used in favor of using remainder calculations
  1132. // to allow "inputs" started at the same time to start on the same frame
  1133. // whereas with remainder calculation the frame alignment would depend on
  1134. // the total frame count at the time the encoder was started
  1135. uint32_t frame_rate_divisor;
  1136. uint32_t frame_rate_divisor_counter; // only used for GPU encoders
  1137. video_t *fps_override;
  1138. // Number of frames successfully encoded
  1139. uint32_t encoded_frames;
  1140. /* Regions of interest to prioritize during encoding */
  1141. pthread_mutex_t roi_mutex;
  1142. DARRAY(struct obs_encoder_roi) roi;
  1143. uint32_t roi_increment;
  1144. int64_t cur_pts;
  1145. struct deque audio_input_buffer[MAX_AV_PLANES];
  1146. uint8_t *audio_output_buffer[MAX_AV_PLANES];
  1147. /* if a video encoder is paired with an audio encoder, make it start
  1148. * up at the specific timestamp. if this is the audio encoder,
  1149. * it waits until it's ready to sync up with video */
  1150. bool first_received;
  1151. DARRAY(struct obs_weak_encoder *) paired_encoders;
  1152. int64_t offset_usec;
  1153. uint64_t first_raw_ts;
  1154. uint64_t start_ts;
  1155. /* track encoders that are part of a gop-aligned multi track group */
  1156. struct obs_encoder_group *encoder_group;
  1157. pthread_mutex_t outputs_mutex;
  1158. DARRAY(obs_output_t *) outputs;
  1159. /* stores the video/audio media output pointer. video_t *or audio_t **/
  1160. void *media;
  1161. pthread_mutex_t callbacks_mutex;
  1162. DARRAY(struct encoder_callback) callbacks;
  1163. DARRAY(struct encoder_packet_time) encoder_packet_times;
  1164. struct pause_data pause;
  1165. const char *profile_encoder_encode_name;
  1166. char *last_error_message;
  1167. /* reconfigure encoder at next possible opportunity */
  1168. bool reconfigure_requested;
  1169. };
  1170. extern struct obs_encoder_info *find_encoder(const char *id);
  1171. extern bool obs_encoder_initialize(obs_encoder_t *encoder);
  1172. extern void obs_encoder_shutdown(obs_encoder_t *encoder);
  1173. extern void obs_encoder_start(obs_encoder_t *encoder, encoded_callback_t new_packet, void *param);
  1174. extern void obs_encoder_stop(obs_encoder_t *encoder, encoded_callback_t new_packet, void *param);
  1175. extern void obs_encoder_add_output(struct obs_encoder *encoder, struct obs_output *output);
  1176. extern void obs_encoder_remove_output(struct obs_encoder *encoder, struct obs_output *output);
  1177. extern bool start_gpu_encode(obs_encoder_t *encoder);
  1178. extern void stop_gpu_encode(obs_encoder_t *encoder);
  1179. extern bool do_encode(struct obs_encoder *encoder, struct encoder_frame *frame, const uint64_t *frame_cts);
  1180. extern void send_off_encoder_packet(obs_encoder_t *encoder, bool success, bool received, struct encoder_packet *pkt);
  1181. void obs_encoder_destroy(obs_encoder_t *encoder);
  1182. /* ------------------------------------------------------------------------- */
  1183. /* services */
  1184. struct obs_weak_service {
  1185. struct obs_weak_ref ref;
  1186. struct obs_service *service;
  1187. };
  1188. struct obs_service {
  1189. struct obs_context_data context;
  1190. struct obs_service_info info;
  1191. /* indicates ownership of the info.id buffer */
  1192. bool owns_info_id;
  1193. bool active;
  1194. bool destroy;
  1195. struct obs_output *output;
  1196. };
  1197. extern const struct obs_service_info *find_service(const char *id);
  1198. extern void obs_service_activate(struct obs_service *service);
  1199. extern void obs_service_deactivate(struct obs_service *service, bool remove);
  1200. extern bool obs_service_initialize(struct obs_service *service, struct obs_output *output);
  1201. void obs_service_destroy(obs_service_t *service);
  1202. void obs_output_remove_encoder_internal(struct obs_output *output, struct obs_encoder *encoder);
  1203. /** Internal Source Profiler functions **/
  1204. /* Start of frame in graphics loop */
  1205. extern void source_profiler_frame_begin(void);
  1206. /* Process data collected during frame */
  1207. extern void source_profiler_frame_collect(void);
  1208. /* Start/end of outputs being rendered (GPU timer begin/end) */
  1209. extern void source_profiler_render_begin(void);
  1210. extern void source_profiler_render_end(void);
  1211. /* Reset settings, buffers, and GPU timers when video settings change */
  1212. extern void source_profiler_reset_video(struct obs_video_info *ovi);
  1213. /* Signal that source received an async frame */
  1214. extern void source_profiler_async_frame_received(obs_source_t *source);
  1215. /* Get timestamp for start of tick */
  1216. extern uint64_t source_profiler_source_tick_start(void);
  1217. /* Submit start timestamp for source */
  1218. extern void source_profiler_source_tick_end(obs_source_t *source, uint64_t start);
  1219. /* Obtain GPU timer and start timestamp for render start of a source. */
  1220. extern uint64_t source_profiler_source_render_begin(gs_timer_t **timer);
  1221. /* Submit start timestamp and GPU timer after rendering source */
  1222. extern void source_profiler_source_render_end(obs_source_t *source, uint64_t start, gs_timer_t *timer);
  1223. /* Remove source from profiler hashmaps */
  1224. extern void source_profiler_remove_source(obs_source_t *source);