obs-internal.h 31 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160
  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 NUM_CHANNELS 3
  32. #define MICROSECOND_DEN 1000000
  33. #define NUM_ENCODE_TEXTURES 3
  34. #define NUM_ENCODE_TEXTURE_FRAMES_TO_WAIT 1
  35. static inline int64_t packet_dts_usec(struct encoder_packet *packet)
  36. {
  37. return packet->dts * MICROSECOND_DEN / packet->timebase_den;
  38. }
  39. struct tick_callback {
  40. void (*tick)(void *param, float seconds);
  41. void *param;
  42. };
  43. struct draw_callback {
  44. void (*draw)(void *param, uint32_t cx, uint32_t cy);
  45. void *param;
  46. };
  47. /* ------------------------------------------------------------------------- */
  48. /* validity checks */
  49. static inline bool obs_object_valid(const void *obj, const char *f,
  50. const char *t)
  51. {
  52. if (!obj) {
  53. blog(LOG_DEBUG, "%s: Null '%s' parameter", f, t);
  54. return false;
  55. }
  56. return true;
  57. }
  58. #define obs_ptr_valid(ptr, func) obs_object_valid(ptr, func, #ptr)
  59. #define obs_source_valid obs_ptr_valid
  60. #define obs_output_valid obs_ptr_valid
  61. #define obs_encoder_valid obs_ptr_valid
  62. #define obs_service_valid obs_ptr_valid
  63. /* ------------------------------------------------------------------------- */
  64. /* modules */
  65. struct obs_module {
  66. char *mod_name;
  67. const char *file;
  68. char *bin_path;
  69. char *data_path;
  70. void *module;
  71. bool loaded;
  72. bool (*load)(void);
  73. void (*unload)(void);
  74. void (*post_load)(void);
  75. void (*set_locale)(const char *locale);
  76. bool (*get_string)(const char *lookup_string,
  77. const char **translated_string);
  78. void (*free_locale)(void);
  79. uint32_t (*ver)(void);
  80. void (*set_pointer)(obs_module_t *module);
  81. const char *(*name)(void);
  82. const char *(*description)(void);
  83. const char *(*author)(void);
  84. struct obs_module *next;
  85. };
  86. extern void free_module(struct obs_module *mod);
  87. struct obs_module_path {
  88. char *bin;
  89. char *data;
  90. };
  91. static inline void free_module_path(struct obs_module_path *omp)
  92. {
  93. if (omp) {
  94. bfree(omp->bin);
  95. bfree(omp->data);
  96. }
  97. }
  98. static inline bool check_path(const char *data, const char *path,
  99. struct dstr *output)
  100. {
  101. dstr_copy(output, path);
  102. dstr_cat(output, data);
  103. return os_file_exists(output->array);
  104. }
  105. /* ------------------------------------------------------------------------- */
  106. /* hotkeys */
  107. struct obs_hotkey {
  108. obs_hotkey_id id;
  109. char *name;
  110. char *description;
  111. obs_hotkey_func func;
  112. void *data;
  113. int pressed;
  114. obs_hotkey_registerer_t registerer_type;
  115. void *registerer;
  116. obs_hotkey_id pair_partner_id;
  117. };
  118. struct obs_hotkey_pair {
  119. obs_hotkey_pair_id pair_id;
  120. obs_hotkey_id id[2];
  121. obs_hotkey_active_func func[2];
  122. bool pressed0;
  123. bool pressed1;
  124. void *data[2];
  125. };
  126. typedef struct obs_hotkey_pair obs_hotkey_pair_t;
  127. typedef struct obs_hotkeys_platform obs_hotkeys_platform_t;
  128. void *obs_hotkey_thread(void *param);
  129. struct obs_core_hotkeys;
  130. bool obs_hotkeys_platform_init(struct obs_core_hotkeys *hotkeys);
  131. void obs_hotkeys_platform_free(struct obs_core_hotkeys *hotkeys);
  132. bool obs_hotkeys_platform_is_pressed(obs_hotkeys_platform_t *context,
  133. obs_key_t key);
  134. const char *obs_get_hotkey_translation(obs_key_t key, const char *def);
  135. struct obs_context_data;
  136. void obs_hotkeys_context_release(struct obs_context_data *context);
  137. void obs_hotkeys_free(void);
  138. struct obs_hotkey_binding {
  139. obs_key_combination_t key;
  140. bool pressed;
  141. bool modifiers_match;
  142. obs_hotkey_id hotkey_id;
  143. obs_hotkey_t *hotkey;
  144. };
  145. struct obs_hotkey_name_map;
  146. void obs_hotkey_name_map_free(void);
  147. /* ------------------------------------------------------------------------- */
  148. /* views */
  149. struct obs_view {
  150. pthread_mutex_t channels_mutex;
  151. obs_source_t *channels[MAX_CHANNELS];
  152. };
  153. extern bool obs_view_init(struct obs_view *view);
  154. extern void obs_view_free(struct obs_view *view);
  155. /* ------------------------------------------------------------------------- */
  156. /* displays */
  157. struct obs_display {
  158. bool size_changed;
  159. bool enabled;
  160. uint32_t cx, cy;
  161. uint32_t background_color;
  162. gs_swapchain_t *swap;
  163. pthread_mutex_t draw_callbacks_mutex;
  164. pthread_mutex_t draw_info_mutex;
  165. DARRAY(struct draw_callback) draw_callbacks;
  166. struct obs_display *next;
  167. struct obs_display **prev_next;
  168. };
  169. extern bool obs_display_init(struct obs_display *display,
  170. const struct gs_init_data *graphics_data);
  171. extern void obs_display_free(struct obs_display *display);
  172. /* ------------------------------------------------------------------------- */
  173. /* core */
  174. struct obs_vframe_info {
  175. uint64_t timestamp;
  176. int count;
  177. };
  178. struct obs_tex_frame {
  179. gs_texture_t *tex;
  180. gs_texture_t *tex_uv;
  181. uint32_t handle;
  182. uint64_t timestamp;
  183. uint64_t lock_key;
  184. int count;
  185. bool released;
  186. };
  187. struct obs_task_info {
  188. obs_task_t task;
  189. void *param;
  190. };
  191. struct obs_core_video {
  192. graphics_t *graphics;
  193. gs_stagesurf_t *copy_surfaces[NUM_TEXTURES][NUM_CHANNELS];
  194. gs_texture_t *render_texture;
  195. gs_texture_t *output_texture;
  196. gs_texture_t *convert_textures[NUM_CHANNELS];
  197. bool texture_rendered;
  198. bool textures_copied[NUM_TEXTURES];
  199. bool texture_converted;
  200. bool using_nv12_tex;
  201. struct circlebuf vframe_info_buffer;
  202. struct circlebuf vframe_info_buffer_gpu;
  203. gs_effect_t *default_effect;
  204. gs_effect_t *default_rect_effect;
  205. gs_effect_t *opaque_effect;
  206. gs_effect_t *solid_effect;
  207. gs_effect_t *repeat_effect;
  208. gs_effect_t *conversion_effect;
  209. gs_effect_t *bicubic_effect;
  210. gs_effect_t *lanczos_effect;
  211. gs_effect_t *area_effect;
  212. gs_effect_t *bilinear_lowres_effect;
  213. gs_effect_t *premultiplied_alpha_effect;
  214. gs_samplerstate_t *point_sampler;
  215. gs_stagesurf_t *mapped_surfaces[NUM_CHANNELS];
  216. int cur_texture;
  217. long raw_active;
  218. long gpu_encoder_active;
  219. pthread_mutex_t gpu_encoder_mutex;
  220. struct circlebuf gpu_encoder_queue;
  221. struct circlebuf gpu_encoder_avail_queue;
  222. DARRAY(obs_encoder_t *) gpu_encoders;
  223. os_sem_t *gpu_encode_semaphore;
  224. os_event_t *gpu_encode_inactive;
  225. pthread_t gpu_encode_thread;
  226. bool gpu_encode_thread_initialized;
  227. volatile bool gpu_encode_stop;
  228. uint64_t video_time;
  229. uint64_t video_frame_interval_ns;
  230. uint64_t video_avg_frame_time_ns;
  231. double video_fps;
  232. video_t *video;
  233. pthread_t video_thread;
  234. uint32_t total_frames;
  235. uint32_t lagged_frames;
  236. bool thread_initialized;
  237. bool gpu_conversion;
  238. const char *conversion_techs[NUM_CHANNELS];
  239. bool conversion_needed;
  240. float conversion_width_i;
  241. uint32_t output_width;
  242. uint32_t output_height;
  243. uint32_t base_width;
  244. uint32_t base_height;
  245. float color_matrix[16];
  246. enum obs_scale_type scale_type;
  247. gs_texture_t *transparent_texture;
  248. gs_effect_t *deinterlace_discard_effect;
  249. gs_effect_t *deinterlace_discard_2x_effect;
  250. gs_effect_t *deinterlace_linear_effect;
  251. gs_effect_t *deinterlace_linear_2x_effect;
  252. gs_effect_t *deinterlace_blend_effect;
  253. gs_effect_t *deinterlace_blend_2x_effect;
  254. gs_effect_t *deinterlace_yadif_effect;
  255. gs_effect_t *deinterlace_yadif_2x_effect;
  256. struct obs_video_info ovi;
  257. pthread_mutex_t task_mutex;
  258. struct circlebuf tasks;
  259. };
  260. struct audio_monitor;
  261. struct obs_core_audio {
  262. audio_t *audio;
  263. DARRAY(struct obs_source *) render_order;
  264. DARRAY(struct obs_source *) root_nodes;
  265. uint64_t buffered_ts;
  266. struct circlebuf buffered_timestamps;
  267. int buffering_wait_ticks;
  268. int total_buffering_ticks;
  269. float user_volume;
  270. pthread_mutex_t monitoring_mutex;
  271. DARRAY(struct audio_monitor *) monitors;
  272. char *monitoring_device_name;
  273. char *monitoring_device_id;
  274. };
  275. /* user sources, output channels, and displays */
  276. struct obs_core_data {
  277. struct obs_source *first_source;
  278. struct obs_source *first_audio_source;
  279. struct obs_display *first_display;
  280. struct obs_output *first_output;
  281. struct obs_encoder *first_encoder;
  282. struct obs_service *first_service;
  283. pthread_mutex_t sources_mutex;
  284. pthread_mutex_t displays_mutex;
  285. pthread_mutex_t outputs_mutex;
  286. pthread_mutex_t encoders_mutex;
  287. pthread_mutex_t services_mutex;
  288. pthread_mutex_t audio_sources_mutex;
  289. pthread_mutex_t draw_callbacks_mutex;
  290. DARRAY(struct draw_callback) draw_callbacks;
  291. DARRAY(struct tick_callback) tick_callbacks;
  292. struct obs_view main_view;
  293. long long unnamed_index;
  294. obs_data_t *private_data;
  295. volatile bool valid;
  296. };
  297. /* user hotkeys */
  298. struct obs_core_hotkeys {
  299. pthread_mutex_t mutex;
  300. DARRAY(obs_hotkey_t) hotkeys;
  301. obs_hotkey_id next_id;
  302. DARRAY(obs_hotkey_pair_t) hotkey_pairs;
  303. obs_hotkey_pair_id next_pair_id;
  304. pthread_t hotkey_thread;
  305. bool hotkey_thread_initialized;
  306. os_event_t *stop_event;
  307. bool thread_disable_press;
  308. bool strict_modifiers;
  309. bool reroute_hotkeys;
  310. DARRAY(obs_hotkey_binding_t) bindings;
  311. obs_hotkey_callback_router_func router_func;
  312. void *router_func_data;
  313. obs_hotkeys_platform_t *platform_context;
  314. pthread_once_t name_map_init_token;
  315. struct obs_hotkey_name_map *name_map;
  316. signal_handler_t *signals;
  317. char *translations[OBS_KEY_LAST_VALUE];
  318. char *mute;
  319. char *unmute;
  320. char *push_to_mute;
  321. char *push_to_talk;
  322. char *sceneitem_show;
  323. char *sceneitem_hide;
  324. };
  325. struct obs_core {
  326. struct obs_module *first_module;
  327. DARRAY(struct obs_module_path) module_paths;
  328. DARRAY(struct obs_source_info) source_types;
  329. DARRAY(struct obs_source_info) input_types;
  330. DARRAY(struct obs_source_info) filter_types;
  331. DARRAY(struct obs_source_info) transition_types;
  332. DARRAY(struct obs_output_info) output_types;
  333. DARRAY(struct obs_encoder_info) encoder_types;
  334. DARRAY(struct obs_service_info) service_types;
  335. DARRAY(struct obs_modal_ui) modal_ui_callbacks;
  336. DARRAY(struct obs_modeless_ui) modeless_ui_callbacks;
  337. signal_handler_t *signals;
  338. proc_handler_t *procs;
  339. char *locale;
  340. char *module_config_path;
  341. bool name_store_owned;
  342. profiler_name_store_t *name_store;
  343. /* segmented into multiple sub-structures to keep things a bit more
  344. * clean and organized */
  345. struct obs_core_video video;
  346. struct obs_core_audio audio;
  347. struct obs_core_data data;
  348. struct obs_core_hotkeys hotkeys;
  349. obs_task_handler_t ui_task_handler;
  350. };
  351. extern struct obs_core *obs;
  352. struct obs_graphics_context {
  353. uint64_t last_time;
  354. uint64_t interval;
  355. uint64_t frame_time_total_ns;
  356. uint64_t fps_total_ns;
  357. uint32_t fps_total_frames;
  358. #ifdef _WIN32
  359. bool gpu_was_active;
  360. #endif
  361. bool raw_was_active;
  362. bool was_active;
  363. const char *video_thread_name;
  364. };
  365. extern void *obs_graphics_thread(void *param);
  366. extern bool obs_graphics_thread_loop(struct obs_graphics_context *context);
  367. #ifdef __APPLE__
  368. extern void *obs_graphics_thread_autorelease(void *param);
  369. extern bool
  370. obs_graphics_thread_loop_autorelease(struct obs_graphics_context *context);
  371. #endif
  372. extern gs_effect_t *obs_load_effect(gs_effect_t **effect, const char *file);
  373. extern bool audio_callback(void *param, uint64_t start_ts_in,
  374. uint64_t end_ts_in, uint64_t *out_ts,
  375. uint32_t mixers, struct audio_output_data *mixes);
  376. extern void
  377. start_raw_video(video_t *video, const struct video_scale_info *conversion,
  378. void (*callback)(void *param, struct video_data *frame),
  379. void *param);
  380. extern void stop_raw_video(video_t *video,
  381. void (*callback)(void *param,
  382. struct video_data *frame),
  383. void *param);
  384. /* ------------------------------------------------------------------------- */
  385. /* obs shared context data */
  386. struct obs_context_data {
  387. char *name;
  388. void *data;
  389. obs_data_t *settings;
  390. signal_handler_t *signals;
  391. proc_handler_t *procs;
  392. enum obs_obj_type type;
  393. DARRAY(obs_hotkey_id) hotkeys;
  394. DARRAY(obs_hotkey_pair_id) hotkey_pairs;
  395. obs_data_t *hotkey_data;
  396. DARRAY(char *) rename_cache;
  397. pthread_mutex_t rename_cache_mutex;
  398. pthread_mutex_t *mutex;
  399. struct obs_context_data *next;
  400. struct obs_context_data **prev_next;
  401. bool private;
  402. };
  403. extern bool obs_context_data_init(struct obs_context_data *context,
  404. enum obs_obj_type type, obs_data_t *settings,
  405. const char *name, obs_data_t *hotkey_data,
  406. bool private);
  407. extern void obs_context_data_free(struct obs_context_data *context);
  408. extern void obs_context_data_insert(struct obs_context_data *context,
  409. pthread_mutex_t *mutex, void *first);
  410. extern void obs_context_data_remove(struct obs_context_data *context);
  411. extern void obs_context_data_setname(struct obs_context_data *context,
  412. const char *name);
  413. /* ------------------------------------------------------------------------- */
  414. /* ref-counting */
  415. struct obs_weak_ref {
  416. volatile long refs;
  417. volatile long weak_refs;
  418. };
  419. static inline void obs_ref_addref(struct obs_weak_ref *ref)
  420. {
  421. os_atomic_inc_long(&ref->refs);
  422. }
  423. static inline bool obs_ref_release(struct obs_weak_ref *ref)
  424. {
  425. return os_atomic_dec_long(&ref->refs) == -1;
  426. }
  427. static inline void obs_weak_ref_addref(struct obs_weak_ref *ref)
  428. {
  429. os_atomic_inc_long(&ref->weak_refs);
  430. }
  431. static inline bool obs_weak_ref_release(struct obs_weak_ref *ref)
  432. {
  433. return os_atomic_dec_long(&ref->weak_refs) == -1;
  434. }
  435. static inline bool obs_weak_ref_get_ref(struct obs_weak_ref *ref)
  436. {
  437. long owners = ref->refs;
  438. while (owners > -1) {
  439. if (os_atomic_compare_swap_long(&ref->refs, owners, owners + 1))
  440. return true;
  441. owners = ref->refs;
  442. }
  443. return false;
  444. }
  445. /* ------------------------------------------------------------------------- */
  446. /* sources */
  447. struct async_frame {
  448. struct obs_source_frame *frame;
  449. long unused_count;
  450. bool used;
  451. };
  452. enum audio_action_type {
  453. AUDIO_ACTION_VOL,
  454. AUDIO_ACTION_MUTE,
  455. AUDIO_ACTION_PTT,
  456. AUDIO_ACTION_PTM,
  457. };
  458. struct audio_action {
  459. uint64_t timestamp;
  460. enum audio_action_type type;
  461. union {
  462. float vol;
  463. bool set;
  464. };
  465. };
  466. struct obs_weak_source {
  467. struct obs_weak_ref ref;
  468. struct obs_source *source;
  469. };
  470. struct audio_cb_info {
  471. obs_source_audio_capture_t callback;
  472. void *param;
  473. };
  474. struct obs_source {
  475. struct obs_context_data context;
  476. struct obs_source_info info;
  477. struct obs_weak_source *control;
  478. /* general exposed flags that can be set for the source */
  479. uint32_t flags;
  480. uint32_t default_flags;
  481. uint32_t last_obs_ver;
  482. /* indicates ownership of the info.id buffer */
  483. bool owns_info_id;
  484. /* signals to call the source update in the video thread */
  485. bool defer_update;
  486. /* ensures show/hide are only called once */
  487. volatile long show_refs;
  488. /* ensures activate/deactivate are only called once */
  489. volatile long activate_refs;
  490. /* used to indicate that the source has been removed and all
  491. * references to it should be released (not exactly how I would prefer
  492. * to handle things but it's the best option) */
  493. bool removed;
  494. bool active;
  495. bool showing;
  496. /* used to temporarily disable sources if needed */
  497. bool enabled;
  498. /* timing (if video is present, is based upon video) */
  499. volatile bool timing_set;
  500. volatile uint64_t timing_adjust;
  501. uint64_t resample_offset;
  502. uint64_t last_audio_ts;
  503. uint64_t next_audio_ts_min;
  504. uint64_t next_audio_sys_ts_min;
  505. uint64_t last_frame_ts;
  506. uint64_t last_sys_timestamp;
  507. bool async_rendered;
  508. /* audio */
  509. bool audio_failed;
  510. bool audio_pending;
  511. bool pending_stop;
  512. bool audio_active;
  513. bool user_muted;
  514. bool muted;
  515. struct obs_source *next_audio_source;
  516. struct obs_source **prev_next_audio_source;
  517. uint64_t audio_ts;
  518. struct circlebuf audio_input_buf[MAX_AUDIO_CHANNELS];
  519. size_t last_audio_input_buf_size;
  520. DARRAY(struct audio_action) audio_actions;
  521. float *audio_output_buf[MAX_AUDIO_MIXES][MAX_AUDIO_CHANNELS];
  522. float *audio_mix_buf[MAX_AUDIO_CHANNELS];
  523. struct resample_info sample_info;
  524. audio_resampler_t *resampler;
  525. pthread_mutex_t audio_actions_mutex;
  526. pthread_mutex_t audio_buf_mutex;
  527. pthread_mutex_t audio_mutex;
  528. pthread_mutex_t audio_cb_mutex;
  529. DARRAY(struct audio_cb_info) audio_cb_list;
  530. struct obs_audio_data audio_data;
  531. size_t audio_storage_size;
  532. uint32_t audio_mixers;
  533. float user_volume;
  534. float volume;
  535. int64_t sync_offset;
  536. int64_t last_sync_offset;
  537. float balance;
  538. /* async video data */
  539. gs_texture_t *async_textures[MAX_AV_PLANES];
  540. gs_texrender_t *async_texrender;
  541. struct obs_source_frame *cur_async_frame;
  542. bool async_gpu_conversion;
  543. enum video_format async_format;
  544. bool async_full_range;
  545. enum video_format async_cache_format;
  546. bool async_cache_full_range;
  547. enum gs_color_format async_texture_formats[MAX_AV_PLANES];
  548. int async_channel_count;
  549. long async_rotation;
  550. bool async_flip;
  551. bool async_active;
  552. bool async_update_texture;
  553. bool async_unbuffered;
  554. bool async_decoupled;
  555. struct obs_source_frame *async_preload_frame;
  556. DARRAY(struct async_frame) async_cache;
  557. DARRAY(struct obs_source_frame *) async_frames;
  558. pthread_mutex_t async_mutex;
  559. uint32_t async_width;
  560. uint32_t async_height;
  561. uint32_t async_cache_width;
  562. uint32_t async_cache_height;
  563. uint32_t async_convert_width[MAX_AV_PLANES];
  564. uint32_t async_convert_height[MAX_AV_PLANES];
  565. /* async video deinterlacing */
  566. uint64_t deinterlace_offset;
  567. uint64_t deinterlace_frame_ts;
  568. gs_effect_t *deinterlace_effect;
  569. struct obs_source_frame *prev_async_frame;
  570. gs_texture_t *async_prev_textures[MAX_AV_PLANES];
  571. gs_texrender_t *async_prev_texrender;
  572. uint32_t deinterlace_half_duration;
  573. enum obs_deinterlace_mode deinterlace_mode;
  574. bool deinterlace_top_first;
  575. bool deinterlace_rendered;
  576. /* filters */
  577. struct obs_source *filter_parent;
  578. struct obs_source *filter_target;
  579. DARRAY(struct obs_source *) filters;
  580. pthread_mutex_t filter_mutex;
  581. gs_texrender_t *filter_texrender;
  582. enum obs_allow_direct_render allow_direct;
  583. bool rendering_filter;
  584. /* sources specific hotkeys */
  585. obs_hotkey_pair_id mute_unmute_key;
  586. obs_hotkey_id push_to_mute_key;
  587. obs_hotkey_id push_to_talk_key;
  588. bool push_to_mute_enabled;
  589. bool push_to_mute_pressed;
  590. bool user_push_to_mute_pressed;
  591. bool push_to_talk_enabled;
  592. bool push_to_talk_pressed;
  593. bool user_push_to_talk_pressed;
  594. uint64_t push_to_mute_delay;
  595. uint64_t push_to_mute_stop_time;
  596. uint64_t push_to_talk_delay;
  597. uint64_t push_to_talk_stop_time;
  598. /* transitions */
  599. uint64_t transition_start_time;
  600. uint64_t transition_duration;
  601. pthread_mutex_t transition_tex_mutex;
  602. gs_texrender_t *transition_texrender[2];
  603. pthread_mutex_t transition_mutex;
  604. obs_source_t *transition_sources[2];
  605. float transition_manual_clamp;
  606. float transition_manual_torque;
  607. float transition_manual_target;
  608. float transition_manual_val;
  609. bool transitioning_video;
  610. bool transitioning_audio;
  611. bool transition_source_active[2];
  612. uint32_t transition_alignment;
  613. uint32_t transition_actual_cx;
  614. uint32_t transition_actual_cy;
  615. uint32_t transition_cx;
  616. uint32_t transition_cy;
  617. uint32_t transition_fixed_duration;
  618. bool transition_use_fixed_duration;
  619. enum obs_transition_mode transition_mode;
  620. enum obs_transition_scale_type transition_scale_type;
  621. struct matrix4 transition_matrices[2];
  622. struct audio_monitor *monitor;
  623. enum obs_monitoring_type monitoring_type;
  624. obs_data_t *private_settings;
  625. };
  626. extern struct obs_source_info *get_source_info(const char *id);
  627. extern struct obs_source_info *get_source_info2(const char *unversioned_id,
  628. uint32_t ver);
  629. extern bool obs_source_init_context(struct obs_source *source,
  630. obs_data_t *settings, const char *name,
  631. obs_data_t *hotkey_data, bool private);
  632. extern bool obs_transition_init(obs_source_t *transition);
  633. extern void obs_transition_free(obs_source_t *transition);
  634. extern void obs_transition_tick(obs_source_t *transition, float t);
  635. extern void obs_transition_enum_sources(obs_source_t *transition,
  636. obs_source_enum_proc_t enum_callback,
  637. void *param);
  638. extern void obs_transition_save(obs_source_t *source, obs_data_t *data);
  639. extern void obs_transition_load(obs_source_t *source, obs_data_t *data);
  640. struct audio_monitor *audio_monitor_create(obs_source_t *source);
  641. void audio_monitor_reset(struct audio_monitor *monitor);
  642. extern void audio_monitor_destroy(struct audio_monitor *monitor);
  643. extern obs_source_t *obs_source_create_set_last_ver(const char *id,
  644. const char *name,
  645. obs_data_t *settings,
  646. obs_data_t *hotkey_data,
  647. uint32_t last_obs_ver);
  648. extern void obs_source_destroy(struct obs_source *source);
  649. enum view_type {
  650. MAIN_VIEW,
  651. AUX_VIEW,
  652. };
  653. static inline void obs_source_dosignal(struct obs_source *source,
  654. const char *signal_obs,
  655. const char *signal_source)
  656. {
  657. struct calldata data;
  658. uint8_t stack[128];
  659. calldata_init_fixed(&data, stack, sizeof(stack));
  660. calldata_set_ptr(&data, "source", source);
  661. if (signal_obs && !source->context.private)
  662. signal_handler_signal(obs->signals, signal_obs, &data);
  663. if (signal_source)
  664. signal_handler_signal(source->context.signals, signal_source,
  665. &data);
  666. }
  667. /* maximum timestamp variance in nanoseconds */
  668. #define MAX_TS_VAR 2000000000ULL
  669. static inline bool frame_out_of_bounds(const obs_source_t *source, uint64_t ts)
  670. {
  671. if (ts < source->last_frame_ts)
  672. return ((source->last_frame_ts - ts) > MAX_TS_VAR);
  673. else
  674. return ((ts - source->last_frame_ts) > MAX_TS_VAR);
  675. }
  676. static inline enum gs_color_format
  677. convert_video_format(enum video_format format)
  678. {
  679. switch (format) {
  680. case VIDEO_FORMAT_RGBA:
  681. return GS_RGBA;
  682. case VIDEO_FORMAT_BGRA:
  683. case VIDEO_FORMAT_I40A:
  684. case VIDEO_FORMAT_I42A:
  685. case VIDEO_FORMAT_YUVA:
  686. case VIDEO_FORMAT_AYUV:
  687. return GS_BGRA;
  688. default:
  689. return GS_BGRX;
  690. }
  691. }
  692. extern void obs_source_activate(obs_source_t *source, enum view_type type);
  693. extern void obs_source_deactivate(obs_source_t *source, enum view_type type);
  694. extern void obs_source_video_tick(obs_source_t *source, float seconds);
  695. extern float obs_source_get_target_volume(obs_source_t *source,
  696. obs_source_t *target);
  697. extern void obs_source_audio_render(obs_source_t *source, uint32_t mixers,
  698. size_t channels, size_t sample_rate,
  699. size_t size);
  700. extern void add_alignment(struct vec2 *v, uint32_t align, int cx, int cy);
  701. extern struct obs_source_frame *filter_async_video(obs_source_t *source,
  702. struct obs_source_frame *in);
  703. extern bool update_async_texture(struct obs_source *source,
  704. const struct obs_source_frame *frame,
  705. gs_texture_t *tex, gs_texrender_t *texrender);
  706. extern bool update_async_textures(struct obs_source *source,
  707. const struct obs_source_frame *frame,
  708. gs_texture_t *tex[MAX_AV_PLANES],
  709. gs_texrender_t *texrender);
  710. extern bool set_async_texture_size(struct obs_source *source,
  711. const struct obs_source_frame *frame);
  712. extern void remove_async_frame(obs_source_t *source,
  713. struct obs_source_frame *frame);
  714. extern void set_deinterlace_texture_size(obs_source_t *source);
  715. extern void deinterlace_process_last_frame(obs_source_t *source,
  716. uint64_t sys_time);
  717. extern void deinterlace_update_async_video(obs_source_t *source);
  718. extern void deinterlace_render(obs_source_t *s);
  719. /* ------------------------------------------------------------------------- */
  720. /* outputs */
  721. enum delay_msg {
  722. DELAY_MSG_PACKET,
  723. DELAY_MSG_START,
  724. DELAY_MSG_STOP,
  725. };
  726. struct delay_data {
  727. enum delay_msg msg;
  728. uint64_t ts;
  729. struct encoder_packet packet;
  730. };
  731. typedef void (*encoded_callback_t)(void *data, struct encoder_packet *packet);
  732. struct obs_weak_output {
  733. struct obs_weak_ref ref;
  734. struct obs_output *output;
  735. };
  736. #define CAPTION_LINE_CHARS (32)
  737. #define CAPTION_LINE_BYTES (4 * CAPTION_LINE_CHARS)
  738. struct caption_text {
  739. char text[CAPTION_LINE_BYTES + 1];
  740. double display_duration;
  741. struct caption_text *next;
  742. };
  743. struct pause_data {
  744. pthread_mutex_t mutex;
  745. uint64_t last_video_ts;
  746. uint64_t ts_start;
  747. uint64_t ts_end;
  748. uint64_t ts_offset;
  749. };
  750. extern bool video_pause_check(struct pause_data *pause, uint64_t timestamp);
  751. extern bool audio_pause_check(struct pause_data *pause, struct audio_data *data,
  752. size_t sample_rate);
  753. extern void pause_reset(struct pause_data *pause);
  754. struct obs_output {
  755. struct obs_context_data context;
  756. struct obs_output_info info;
  757. struct obs_weak_output *control;
  758. /* indicates ownership of the info.id buffer */
  759. bool owns_info_id;
  760. bool received_video;
  761. bool received_audio;
  762. volatile bool data_active;
  763. volatile bool end_data_capture_thread_active;
  764. int64_t video_offset;
  765. int64_t audio_offsets[MAX_AUDIO_MIXES];
  766. int64_t highest_audio_ts;
  767. int64_t highest_video_ts;
  768. pthread_t end_data_capture_thread;
  769. os_event_t *stopping_event;
  770. pthread_mutex_t interleaved_mutex;
  771. DARRAY(struct encoder_packet) interleaved_packets;
  772. int stop_code;
  773. int reconnect_retry_sec;
  774. int reconnect_retry_max;
  775. int reconnect_retries;
  776. int reconnect_retry_cur_sec;
  777. pthread_t reconnect_thread;
  778. os_event_t *reconnect_stop_event;
  779. volatile bool reconnecting;
  780. volatile bool reconnect_thread_active;
  781. uint32_t starting_drawn_count;
  782. uint32_t starting_lagged_count;
  783. uint32_t starting_frame_count;
  784. int total_frames;
  785. volatile bool active;
  786. volatile bool paused;
  787. video_t *video;
  788. audio_t *audio;
  789. obs_encoder_t *video_encoder;
  790. obs_encoder_t *audio_encoders[MAX_AUDIO_MIXES];
  791. obs_service_t *service;
  792. size_t mixer_mask;
  793. struct pause_data pause;
  794. struct circlebuf audio_buffer[MAX_AUDIO_MIXES][MAX_AV_PLANES];
  795. uint64_t audio_start_ts;
  796. uint64_t video_start_ts;
  797. size_t audio_size;
  798. size_t planes;
  799. size_t sample_rate;
  800. size_t total_audio_frames;
  801. uint32_t scaled_width;
  802. uint32_t scaled_height;
  803. bool video_conversion_set;
  804. bool audio_conversion_set;
  805. struct video_scale_info video_conversion;
  806. struct audio_convert_info audio_conversion;
  807. pthread_mutex_t caption_mutex;
  808. double caption_timestamp;
  809. struct caption_text *caption_head;
  810. struct caption_text *caption_tail;
  811. bool valid;
  812. uint64_t active_delay_ns;
  813. encoded_callback_t delay_callback;
  814. struct circlebuf delay_data; /* struct delay_data */
  815. pthread_mutex_t delay_mutex;
  816. uint32_t delay_sec;
  817. uint32_t delay_flags;
  818. uint32_t delay_cur_flags;
  819. volatile long delay_restart_refs;
  820. volatile bool delay_active;
  821. volatile bool delay_capturing;
  822. char *last_error_message;
  823. float audio_data[MAX_AUDIO_CHANNELS][AUDIO_OUTPUT_FRAMES];
  824. };
  825. static inline void do_output_signal(struct obs_output *output,
  826. const char *signal)
  827. {
  828. struct calldata params = {0};
  829. calldata_set_ptr(&params, "output", output);
  830. signal_handler_signal(output->context.signals, signal, &params);
  831. calldata_free(&params);
  832. }
  833. extern void process_delay(void *data, struct encoder_packet *packet);
  834. extern void obs_output_cleanup_delay(obs_output_t *output);
  835. extern bool obs_output_delay_start(obs_output_t *output);
  836. extern void obs_output_delay_stop(obs_output_t *output);
  837. extern bool obs_output_actual_start(obs_output_t *output);
  838. extern void obs_output_actual_stop(obs_output_t *output, bool force,
  839. uint64_t ts);
  840. extern const struct obs_output_info *find_output(const char *id);
  841. extern void obs_output_remove_encoder(struct obs_output *output,
  842. struct obs_encoder *encoder);
  843. extern void
  844. obs_encoder_packet_create_instance(struct encoder_packet *dst,
  845. const struct encoder_packet *src);
  846. void obs_output_destroy(obs_output_t *output);
  847. /* ------------------------------------------------------------------------- */
  848. /* encoders */
  849. struct obs_weak_encoder {
  850. struct obs_weak_ref ref;
  851. struct obs_encoder *encoder;
  852. };
  853. struct encoder_callback {
  854. bool sent_first_packet;
  855. void (*new_packet)(void *param, struct encoder_packet *packet);
  856. void *param;
  857. };
  858. struct obs_encoder {
  859. struct obs_context_data context;
  860. struct obs_encoder_info info;
  861. struct obs_weak_encoder *control;
  862. /* allows re-routing to another encoder */
  863. struct obs_encoder_info orig_info;
  864. pthread_mutex_t init_mutex;
  865. uint32_t samplerate;
  866. size_t planes;
  867. size_t blocksize;
  868. size_t framesize;
  869. size_t framesize_bytes;
  870. size_t mixer_idx;
  871. uint32_t scaled_width;
  872. uint32_t scaled_height;
  873. enum video_format preferred_format;
  874. volatile bool active;
  875. volatile bool paused;
  876. bool initialized;
  877. /* indicates ownership of the info.id buffer */
  878. bool owns_info_id;
  879. uint32_t timebase_num;
  880. uint32_t timebase_den;
  881. int64_t cur_pts;
  882. struct circlebuf audio_input_buffer[MAX_AV_PLANES];
  883. uint8_t *audio_output_buffer[MAX_AV_PLANES];
  884. /* if a video encoder is paired with an audio encoder, make it start
  885. * up at the specific timestamp. if this is the audio encoder,
  886. * wait_for_video makes it wait until it's ready to sync up with
  887. * video */
  888. bool wait_for_video;
  889. bool first_received;
  890. struct obs_encoder *paired_encoder;
  891. int64_t offset_usec;
  892. uint64_t first_raw_ts;
  893. uint64_t start_ts;
  894. pthread_mutex_t outputs_mutex;
  895. DARRAY(obs_output_t *) outputs;
  896. bool destroy_on_stop;
  897. /* stores the video/audio media output pointer. video_t *or audio_t **/
  898. void *media;
  899. pthread_mutex_t callbacks_mutex;
  900. DARRAY(struct encoder_callback) callbacks;
  901. struct pause_data pause;
  902. const char *profile_encoder_encode_name;
  903. char *last_error_message;
  904. };
  905. extern struct obs_encoder_info *find_encoder(const char *id);
  906. extern bool obs_encoder_initialize(obs_encoder_t *encoder);
  907. extern void obs_encoder_shutdown(obs_encoder_t *encoder);
  908. extern void obs_encoder_start(obs_encoder_t *encoder,
  909. void (*new_packet)(void *param,
  910. struct encoder_packet *packet),
  911. void *param);
  912. extern void obs_encoder_stop(obs_encoder_t *encoder,
  913. void (*new_packet)(void *param,
  914. struct encoder_packet *packet),
  915. void *param);
  916. extern void obs_encoder_add_output(struct obs_encoder *encoder,
  917. struct obs_output *output);
  918. extern void obs_encoder_remove_output(struct obs_encoder *encoder,
  919. struct obs_output *output);
  920. extern bool start_gpu_encode(obs_encoder_t *encoder);
  921. extern void stop_gpu_encode(obs_encoder_t *encoder);
  922. extern bool do_encode(struct obs_encoder *encoder, struct encoder_frame *frame);
  923. extern void send_off_encoder_packet(obs_encoder_t *encoder, bool success,
  924. bool received, struct encoder_packet *pkt);
  925. void obs_encoder_destroy(obs_encoder_t *encoder);
  926. /* ------------------------------------------------------------------------- */
  927. /* services */
  928. struct obs_weak_service {
  929. struct obs_weak_ref ref;
  930. struct obs_service *service;
  931. };
  932. struct obs_service {
  933. struct obs_context_data context;
  934. struct obs_service_info info;
  935. struct obs_weak_service *control;
  936. /* indicates ownership of the info.id buffer */
  937. bool owns_info_id;
  938. bool active;
  939. bool destroy;
  940. struct obs_output *output;
  941. };
  942. extern const struct obs_service_info *find_service(const char *id);
  943. extern void obs_service_activate(struct obs_service *service);
  944. extern void obs_service_deactivate(struct obs_service *service, bool remove);
  945. extern bool obs_service_initialize(struct obs_service *service,
  946. struct obs_output *output);
  947. void obs_service_destroy(obs_service_t *service);