obs.c 32 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358
  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. #include <inttypes.h>
  15. #include "callback/calldata.h"
  16. #include "obs.h"
  17. #include "obs-internal.h"
  18. struct obs_core *obs = NULL;
  19. extern void add_default_module_paths(void);
  20. extern char *find_libobs_data_file(const char *file);
  21. static inline void make_gs_init_data(struct gs_init_data *gid,
  22. struct obs_video_info *ovi)
  23. {
  24. memcpy(&gid->window, &ovi->window, sizeof(struct gs_window));
  25. gid->cx = ovi->window_width;
  26. gid->cy = ovi->window_height;
  27. gid->num_backbuffers = 2;
  28. gid->format = GS_RGBA;
  29. gid->zsformat = GS_ZS_NONE;
  30. gid->adapter = ovi->adapter;
  31. }
  32. static inline void make_video_info(struct video_output_info *vi,
  33. struct obs_video_info *ovi)
  34. {
  35. vi->name = "video";
  36. vi->format = ovi->output_format;
  37. vi->fps_num = ovi->fps_num;
  38. vi->fps_den = ovi->fps_den;
  39. vi->width = ovi->output_width;
  40. vi->height = ovi->output_height;
  41. }
  42. #define PIXEL_SIZE 4
  43. #define GET_ALIGN(val, align) \
  44. (((val) + (align-1)) & ~(align-1))
  45. static inline void set_420p_sizes(const struct obs_video_info *ovi)
  46. {
  47. struct obs_core_video *video = &obs->video;
  48. uint32_t chroma_pixels;
  49. uint32_t total_bytes;
  50. chroma_pixels = (ovi->output_width * ovi->output_height / 4);
  51. chroma_pixels = GET_ALIGN(chroma_pixels, PIXEL_SIZE);
  52. video->plane_offsets[0] = 0;
  53. video->plane_offsets[1] = ovi->output_width * ovi->output_height;
  54. video->plane_offsets[2] = video->plane_offsets[1] + chroma_pixels;
  55. video->plane_linewidth[0] = ovi->output_width;
  56. video->plane_linewidth[1] = ovi->output_width/2;
  57. video->plane_linewidth[2] = ovi->output_width/2;
  58. video->plane_sizes[0] = video->plane_offsets[1];
  59. video->plane_sizes[1] = video->plane_sizes[0]/4;
  60. video->plane_sizes[2] = video->plane_sizes[1];
  61. total_bytes = video->plane_offsets[2] + chroma_pixels;
  62. video->conversion_height =
  63. (total_bytes/PIXEL_SIZE + ovi->output_width-1) /
  64. ovi->output_width;
  65. video->conversion_height = GET_ALIGN(video->conversion_height, 2);
  66. video->conversion_tech = "Planar420";
  67. }
  68. static inline void set_nv12_sizes(const struct obs_video_info *ovi)
  69. {
  70. struct obs_core_video *video = &obs->video;
  71. uint32_t chroma_pixels;
  72. uint32_t total_bytes;
  73. chroma_pixels = (ovi->output_width * ovi->output_height / 2);
  74. chroma_pixels = GET_ALIGN(chroma_pixels, PIXEL_SIZE);
  75. video->plane_offsets[0] = 0;
  76. video->plane_offsets[1] = ovi->output_width * ovi->output_height;
  77. video->plane_linewidth[0] = ovi->output_width;
  78. video->plane_linewidth[1] = ovi->output_width;
  79. video->plane_sizes[0] = video->plane_offsets[1];
  80. video->plane_sizes[1] = video->plane_sizes[0]/2;
  81. total_bytes = video->plane_offsets[1] + chroma_pixels;
  82. video->conversion_height =
  83. (total_bytes/PIXEL_SIZE + ovi->output_width-1) /
  84. ovi->output_width;
  85. video->conversion_height = GET_ALIGN(video->conversion_height, 2);
  86. video->conversion_tech = "NV12";
  87. }
  88. static inline void calc_gpu_conversion_sizes(const struct obs_video_info *ovi)
  89. {
  90. obs->video.conversion_height = 0;
  91. memset(obs->video.plane_offsets, 0, sizeof(obs->video.plane_offsets));
  92. memset(obs->video.plane_sizes, 0, sizeof(obs->video.plane_sizes));
  93. memset(obs->video.plane_linewidth, 0,
  94. sizeof(obs->video.plane_linewidth));
  95. switch ((uint32_t)ovi->output_format) {
  96. case VIDEO_FORMAT_I420:
  97. set_420p_sizes(ovi);
  98. break;
  99. case VIDEO_FORMAT_NV12:
  100. set_nv12_sizes(ovi);
  101. break;
  102. }
  103. }
  104. static bool obs_init_gpu_conversion(struct obs_video_info *ovi)
  105. {
  106. struct obs_core_video *video = &obs->video;
  107. calc_gpu_conversion_sizes(ovi);
  108. if (!video->conversion_height) {
  109. blog(LOG_INFO, "GPU conversion not available for format: %u",
  110. (unsigned int)ovi->output_format);
  111. video->gpu_conversion = false;
  112. return true;
  113. }
  114. for (size_t i = 0; i < NUM_TEXTURES; i++) {
  115. video->convert_textures[i] = gs_create_texture(
  116. ovi->output_width, video->conversion_height,
  117. GS_RGBA, 1, NULL, GS_RENDERTARGET);
  118. if (!video->convert_textures[i])
  119. return false;
  120. }
  121. return true;
  122. }
  123. static bool obs_init_textures(struct obs_video_info *ovi)
  124. {
  125. struct obs_core_video *video = &obs->video;
  126. bool yuv = format_is_yuv(ovi->output_format);
  127. uint32_t output_height = video->gpu_conversion ?
  128. video->conversion_height : ovi->output_height;
  129. size_t i;
  130. for (i = 0; i < NUM_TEXTURES; i++) {
  131. video->copy_surfaces[i] = gs_create_stagesurface(
  132. ovi->output_width, output_height, GS_RGBA);
  133. if (!video->copy_surfaces[i])
  134. return false;
  135. video->render_textures[i] = gs_create_texture(
  136. ovi->base_width, ovi->base_height,
  137. GS_RGBA, 1, NULL, GS_RENDERTARGET);
  138. if (!video->render_textures[i])
  139. return false;
  140. video->output_textures[i] = gs_create_texture(
  141. ovi->output_width, ovi->output_height,
  142. GS_RGBA, 1, NULL, GS_RENDERTARGET);
  143. if (!video->output_textures[i])
  144. return false;
  145. if (yuv)
  146. source_frame_init(&video->convert_frames[i],
  147. ovi->output_format,
  148. ovi->output_width, ovi->output_height);
  149. }
  150. return true;
  151. }
  152. static int obs_init_graphics(struct obs_video_info *ovi)
  153. {
  154. struct obs_core_video *video = &obs->video;
  155. struct gs_init_data graphics_data;
  156. bool success = true;
  157. int errorcode;
  158. make_gs_init_data(&graphics_data, ovi);
  159. errorcode = gs_create(&video->graphics, ovi->graphics_module,
  160. &graphics_data);
  161. if (errorcode != GS_SUCCESS) {
  162. switch (errorcode) {
  163. case GS_ERROR_MODULE_NOT_FOUND:
  164. return OBS_VIDEO_MODULE_NOT_FOUND;
  165. case GS_ERROR_NOT_SUPPORTED:
  166. return OBS_VIDEO_NOT_SUPPORTED;
  167. default:
  168. return OBS_VIDEO_FAIL;
  169. }
  170. }
  171. gs_entercontext(video->graphics);
  172. char *filename = find_libobs_data_file("default.effect");
  173. video->default_effect = gs_create_effect_from_file(filename,
  174. NULL);
  175. bfree(filename);
  176. filename = find_libobs_data_file("solid.effect");
  177. video->solid_effect = gs_create_effect_from_file(filename,
  178. NULL);
  179. bfree(filename);
  180. filename = find_libobs_data_file("format_conversion.effect");
  181. video->conversion_effect = gs_create_effect_from_file(filename,
  182. NULL);
  183. bfree(filename);
  184. if (!video->default_effect)
  185. success = false;
  186. if (!video->solid_effect)
  187. success = false;
  188. if (!video->conversion_effect)
  189. success = false;
  190. gs_leavecontext();
  191. return success ? OBS_VIDEO_SUCCESS : OBS_VIDEO_FAIL;
  192. }
  193. static int obs_init_video(struct obs_video_info *ovi)
  194. {
  195. struct obs_core_video *video = &obs->video;
  196. struct video_output_info vi;
  197. int errorcode;
  198. make_video_info(&vi, ovi);
  199. video->base_width = ovi->base_width;
  200. video->base_height = ovi->base_height;
  201. video->output_width = ovi->output_width;
  202. video->output_height = ovi->output_height;
  203. video->gpu_conversion = ovi->gpu_conversion;
  204. errorcode = video_output_open(&video->video, &vi);
  205. if (errorcode != VIDEO_OUTPUT_SUCCESS) {
  206. if (errorcode == VIDEO_OUTPUT_INVALIDPARAM) {
  207. blog(LOG_ERROR, "Invalid video parameters specified");
  208. return OBS_VIDEO_INVALID_PARAM;
  209. } else {
  210. blog(LOG_ERROR, "Could not open video output");
  211. }
  212. return OBS_VIDEO_FAIL;
  213. }
  214. if (!obs_display_init(&video->main_display, NULL))
  215. return OBS_VIDEO_FAIL;
  216. video->main_display.cx = ovi->window_width;
  217. video->main_display.cy = ovi->window_height;
  218. gs_entercontext(video->graphics);
  219. if (ovi->gpu_conversion && !obs_init_gpu_conversion(ovi))
  220. return OBS_VIDEO_FAIL;
  221. if (!obs_init_textures(ovi))
  222. return OBS_VIDEO_FAIL;
  223. gs_leavecontext();
  224. errorcode = pthread_create(&video->video_thread, NULL,
  225. obs_video_thread, obs);
  226. if (errorcode != 0)
  227. return OBS_VIDEO_FAIL;
  228. video->thread_initialized = true;
  229. return OBS_VIDEO_SUCCESS;
  230. }
  231. static void stop_video(void)
  232. {
  233. struct obs_core_video *video = &obs->video;
  234. void *thread_retval;
  235. if (video->video) {
  236. video_output_stop(video->video);
  237. if (video->thread_initialized) {
  238. pthread_join(video->video_thread, &thread_retval);
  239. video->thread_initialized = false;
  240. }
  241. }
  242. }
  243. static void obs_free_video(void)
  244. {
  245. struct obs_core_video *video = &obs->video;
  246. if (video->video) {
  247. uint32_t total_frames = video_output_total_frames(video->video);
  248. uint32_t skipped_frames =
  249. video_output_num_skipped_frames(video->video);
  250. double percentage_skipped =
  251. (double)skipped_frames / (double)total_frames * 100.0;
  252. obs_display_free(&video->main_display);
  253. blog(LOG_INFO, "Video session ending");
  254. blog(LOG_INFO, "Total frames: %"PRIu32, total_frames);
  255. if (total_frames) {
  256. blog(LOG_INFO, "Number of skipped frames: "
  257. "%"PRIu32" (%g%%)",
  258. skipped_frames, percentage_skipped);
  259. }
  260. blog(LOG_INFO, "-------------------------------------------");
  261. video_output_close(video->video);
  262. video->video = NULL;
  263. if (!video->graphics)
  264. return;
  265. gs_entercontext(video->graphics);
  266. if (video->mapped_surface) {
  267. stagesurface_unmap(video->mapped_surface);
  268. video->mapped_surface = NULL;
  269. }
  270. for (size_t i = 0; i < NUM_TEXTURES; i++) {
  271. stagesurface_destroy(video->copy_surfaces[i]);
  272. texture_destroy(video->render_textures[i]);
  273. texture_destroy(video->convert_textures[i]);
  274. texture_destroy(video->output_textures[i]);
  275. source_frame_free(&video->convert_frames[i]);
  276. video->copy_surfaces[i] = NULL;
  277. video->render_textures[i] = NULL;
  278. video->convert_textures[i] = NULL;
  279. video->output_textures[i] = NULL;
  280. }
  281. gs_leavecontext();
  282. video->cur_texture = 0;
  283. }
  284. }
  285. static void obs_free_graphics(void)
  286. {
  287. struct obs_core_video *video = &obs->video;
  288. if (video->graphics) {
  289. gs_entercontext(video->graphics);
  290. effect_destroy(video->default_effect);
  291. effect_destroy(video->solid_effect);
  292. effect_destroy(video->conversion_effect);
  293. video->default_effect = NULL;
  294. gs_leavecontext();
  295. gs_destroy(video->graphics);
  296. video->graphics = NULL;
  297. }
  298. }
  299. static bool obs_init_audio(struct audio_output_info *ai)
  300. {
  301. struct obs_core_audio *audio = &obs->audio;
  302. int errorcode;
  303. /* TODO: sound subsystem */
  304. audio->user_volume = 1.0f;
  305. audio->present_volume = 1.0f;
  306. errorcode = audio_output_open(&audio->audio, ai);
  307. if (errorcode == AUDIO_OUTPUT_SUCCESS)
  308. return true;
  309. else if (errorcode == AUDIO_OUTPUT_INVALIDPARAM)
  310. blog(LOG_ERROR, "Invalid audio parameters specified");
  311. else
  312. blog(LOG_ERROR, "Could not open audio output");
  313. return false;
  314. }
  315. static void obs_free_audio(void)
  316. {
  317. struct obs_core_audio *audio = &obs->audio;
  318. if (audio->audio)
  319. audio_output_close(audio->audio);
  320. memset(audio, 0, sizeof(struct obs_core_audio));
  321. }
  322. static bool obs_init_data(void)
  323. {
  324. struct obs_core_data *data = &obs->data;
  325. pthread_mutexattr_t attr;
  326. assert(data != NULL);
  327. pthread_mutex_init_value(&obs->data.displays_mutex);
  328. if (pthread_mutexattr_init(&attr) != 0)
  329. return false;
  330. if (pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE) != 0)
  331. goto fail;
  332. if (pthread_mutex_init(&data->user_sources_mutex, &attr) != 0)
  333. goto fail;
  334. if (pthread_mutex_init(&data->sources_mutex, &attr) != 0)
  335. goto fail;
  336. if (pthread_mutex_init(&data->displays_mutex, &attr) != 0)
  337. goto fail;
  338. if (pthread_mutex_init(&data->outputs_mutex, &attr) != 0)
  339. goto fail;
  340. if (pthread_mutex_init(&data->encoders_mutex, &attr) != 0)
  341. goto fail;
  342. if (pthread_mutex_init(&data->services_mutex, &attr) != 0)
  343. goto fail;
  344. if (!obs_view_init(&data->main_view))
  345. goto fail;
  346. data->valid = true;
  347. fail:
  348. pthread_mutexattr_destroy(&attr);
  349. return data->valid;
  350. }
  351. #define FREE_OBS_LINKED_LIST(type) \
  352. do { \
  353. int unfreed = 0; \
  354. while (data->first_ ## type ) { \
  355. obs_ ## type ## _destroy(data->first_ ## type ); \
  356. unfreed++; \
  357. } \
  358. if (unfreed) \
  359. blog(LOG_INFO, "\t%d " #type "(s) were remaining", \
  360. unfreed); \
  361. } while (false)
  362. static void obs_free_data(void)
  363. {
  364. struct obs_core_data *data = &obs->data;
  365. data->valid = false;
  366. obs_view_free(&data->main_view);
  367. blog(LOG_INFO, "Freeing OBS context data");
  368. if (data->user_sources.num)
  369. blog(LOG_INFO, "\t%d user source(s) were remaining",
  370. (int)data->user_sources.num);
  371. while (data->user_sources.num)
  372. obs_source_remove(data->user_sources.array[0]);
  373. da_free(data->user_sources);
  374. FREE_OBS_LINKED_LIST(source);
  375. FREE_OBS_LINKED_LIST(output);
  376. FREE_OBS_LINKED_LIST(encoder);
  377. FREE_OBS_LINKED_LIST(display);
  378. FREE_OBS_LINKED_LIST(service);
  379. pthread_mutex_destroy(&data->user_sources_mutex);
  380. pthread_mutex_destroy(&data->sources_mutex);
  381. pthread_mutex_destroy(&data->displays_mutex);
  382. pthread_mutex_destroy(&data->outputs_mutex);
  383. pthread_mutex_destroy(&data->encoders_mutex);
  384. pthread_mutex_destroy(&data->services_mutex);
  385. }
  386. static const char *obs_signals[] = {
  387. "void source_create(ptr source)",
  388. "void source_destroy(ptr source)",
  389. "void source_add(ptr source)",
  390. "void source_remove(ptr source)",
  391. "void source_activate(ptr source)",
  392. "void source_deactivate(ptr source)",
  393. "void source_show(ptr source)",
  394. "void source_hide(ptr source)",
  395. "void source_rename(ptr source, string new_name, string prev_name)",
  396. "void source_volume(ptr source, in out float volume)",
  397. "void source_volume_level(ptr source, float level, float magnitude, "
  398. "float peak)",
  399. "void channel_change(int channel, in out ptr source, ptr prev_source)",
  400. "void master_volume(in out float volume)",
  401. NULL
  402. };
  403. static inline bool obs_init_handlers(void)
  404. {
  405. obs->signals = signal_handler_create();
  406. if (!obs->signals)
  407. return false;
  408. obs->procs = proc_handler_create();
  409. if (!obs->procs)
  410. return false;
  411. return signal_handler_add_array(obs->signals, obs_signals);
  412. }
  413. extern const struct obs_source_info scene_info;
  414. extern void log_system_info(void);
  415. static bool obs_init(const char *locale)
  416. {
  417. obs = bzalloc(sizeof(struct obs_core));
  418. log_system_info();
  419. if (!obs_init_data())
  420. return false;
  421. if (!obs_init_handlers())
  422. return false;
  423. obs->locale = bstrdup(locale);
  424. obs_register_source(&scene_info);
  425. add_default_module_paths();
  426. return true;
  427. }
  428. bool obs_startup(const char *locale)
  429. {
  430. bool success;
  431. if (obs) {
  432. blog(LOG_WARNING, "Tried to call obs_startup more than once");
  433. return false;
  434. }
  435. success = obs_init(locale);
  436. if (!success)
  437. obs_shutdown();
  438. return success;
  439. }
  440. void obs_shutdown(void)
  441. {
  442. struct obs_module *module;
  443. if (!obs)
  444. return;
  445. da_free(obs->input_types);
  446. da_free(obs->filter_types);
  447. da_free(obs->encoder_types);
  448. da_free(obs->transition_types);
  449. da_free(obs->output_types);
  450. da_free(obs->service_types);
  451. da_free(obs->modal_ui_callbacks);
  452. da_free(obs->modeless_ui_callbacks);
  453. stop_video();
  454. obs_free_data();
  455. obs_free_video();
  456. obs_free_graphics();
  457. obs_free_audio();
  458. proc_handler_destroy(obs->procs);
  459. signal_handler_destroy(obs->signals);
  460. module = obs->first_module;
  461. while (module) {
  462. struct obs_module *next = module->next;
  463. free_module(module);
  464. module = next;
  465. }
  466. obs->first_module = NULL;
  467. for (size_t i = 0; i < obs->module_paths.num; i++)
  468. free_module_path(obs->module_paths.array+i);
  469. da_free(obs->module_paths);
  470. bfree(obs->locale);
  471. bfree(obs);
  472. obs = NULL;
  473. }
  474. bool obs_initialized(void)
  475. {
  476. return obs != NULL;
  477. }
  478. uint32_t obs_get_version(void)
  479. {
  480. return LIBOBS_API_VER;
  481. }
  482. void obs_set_locale(const char *locale)
  483. {
  484. struct obs_module *module;
  485. if (!obs)
  486. return;
  487. if (obs->locale)
  488. bfree(obs->locale);
  489. obs->locale = bstrdup(locale);
  490. module = obs->first_module;
  491. while (module) {
  492. if (module->set_locale)
  493. module->set_locale(locale);
  494. module = module->next;
  495. }
  496. }
  497. const char *obs_get_locale(void)
  498. {
  499. return obs ? obs->locale : NULL;
  500. }
  501. #define OBS_SIZE_MIN 2
  502. #define OBS_SIZE_MAX (32 * 1024)
  503. static inline bool size_valid(uint32_t width, uint32_t height)
  504. {
  505. return (width >= OBS_SIZE_MIN && height >= OBS_SIZE_MIN &&
  506. width <= OBS_SIZE_MAX && height <= OBS_SIZE_MAX);
  507. }
  508. int obs_reset_video(struct obs_video_info *ovi)
  509. {
  510. if (!obs) return OBS_VIDEO_FAIL;
  511. /* don't allow changing of video settings if active. */
  512. if (obs->video.video && video_output_active(obs->video.video))
  513. return OBS_VIDEO_CURRENTLY_ACTIVE;
  514. if (!size_valid(ovi->output_width, ovi->output_height) ||
  515. !size_valid(ovi->base_width, ovi->base_height))
  516. return OBS_VIDEO_INVALID_PARAM;
  517. struct obs_core_video *video = &obs->video;
  518. stop_video();
  519. obs_free_video();
  520. if (!ovi) {
  521. obs_free_graphics();
  522. return OBS_VIDEO_SUCCESS;
  523. }
  524. /* align to multiple-of-two and SSE alignment sizes */
  525. ovi->output_width &= 0xFFFFFFFC;
  526. ovi->output_height &= 0xFFFFFFFE;
  527. if (!video->graphics) {
  528. int errorcode = obs_init_graphics(ovi);
  529. if (errorcode != OBS_VIDEO_SUCCESS)
  530. return errorcode;
  531. }
  532. blog(LOG_INFO, "video settings reset:\n"
  533. "\tbase resolution: %dx%d\n"
  534. "\toutput resolution: %dx%d\n"
  535. "\tfps: %d/%d",
  536. ovi->base_width, ovi->base_height,
  537. ovi->output_width, ovi->output_height,
  538. ovi->fps_num, ovi->fps_den);
  539. return obs_init_video(ovi);
  540. }
  541. bool obs_reset_audio(struct audio_output_info *ai)
  542. {
  543. if (!obs) return false;
  544. /* don't allow changing of audio settings if active. */
  545. if (obs->audio.audio && audio_output_active(obs->audio.audio))
  546. return false;
  547. obs_free_audio();
  548. if(!ai)
  549. return true;
  550. blog(LOG_INFO, "audio settings reset:\n"
  551. "\tsamples per sec: %d\n"
  552. "\tspeakers: %d\n"
  553. "\tbuffering (ms): %d\n",
  554. (int)ai->samples_per_sec,
  555. (int)ai->speakers,
  556. (int)ai->buffer_ms);
  557. return obs_init_audio(ai);
  558. }
  559. bool obs_get_video_info(struct obs_video_info *ovi)
  560. {
  561. struct obs_core_video *video = &obs->video;
  562. const struct video_output_info *info;
  563. if (!obs || !video->graphics)
  564. return false;
  565. info = video_output_getinfo(video->video);
  566. memset(ovi, 0, sizeof(struct obs_video_info));
  567. ovi->base_width = video->base_width;
  568. ovi->base_height = video->base_height;
  569. ovi->output_width = info->width;
  570. ovi->output_height = info->height;
  571. ovi->output_format = info->format;
  572. ovi->fps_num = info->fps_num;
  573. ovi->fps_den = info->fps_den;
  574. return true;
  575. }
  576. bool obs_get_audio_info(struct audio_output_info *aoi)
  577. {
  578. struct obs_core_audio *audio = &obs->audio;
  579. const struct audio_output_info *info;
  580. if (!obs || !audio->audio)
  581. return false;
  582. info = audio_output_getinfo(audio->audio);
  583. memcpy(aoi, info, sizeof(struct audio_output_info));
  584. return true;
  585. }
  586. bool obs_enum_input_types(size_t idx, const char **id)
  587. {
  588. if (!obs) return false;
  589. if (idx >= obs->input_types.num)
  590. return false;
  591. *id = obs->input_types.array[idx].id;
  592. return true;
  593. }
  594. bool obs_enum_filter_types(size_t idx, const char **id)
  595. {
  596. if (!obs) return false;
  597. if (idx >= obs->filter_types.num)
  598. return false;
  599. *id = obs->filter_types.array[idx].id;
  600. return true;
  601. }
  602. bool obs_enum_transition_types(size_t idx, const char **id)
  603. {
  604. if (!obs) return false;
  605. if (idx >= obs->transition_types.num)
  606. return false;
  607. *id = obs->transition_types.array[idx].id;
  608. return true;
  609. }
  610. bool obs_enum_output_types(size_t idx, const char **id)
  611. {
  612. if (!obs) return false;
  613. if (idx >= obs->output_types.num)
  614. return false;
  615. *id = obs->output_types.array[idx].id;
  616. return true;
  617. }
  618. bool obs_enum_encoder_types(size_t idx, const char **id)
  619. {
  620. if (!obs) return false;
  621. if (idx >= obs->encoder_types.num)
  622. return false;
  623. *id = obs->encoder_types.array[idx].id;
  624. return true;
  625. }
  626. bool obs_enum_service_types(size_t idx, const char **id)
  627. {
  628. if (!obs) return false;
  629. if (idx >= obs->service_types.num)
  630. return false;
  631. *id = obs->service_types.array[idx].id;
  632. return true;
  633. }
  634. graphics_t obs_graphics(void)
  635. {
  636. return (obs != NULL) ? obs->video.graphics : NULL;
  637. }
  638. audio_t obs_audio(void)
  639. {
  640. return (obs != NULL) ? obs->audio.audio : NULL;
  641. }
  642. video_t obs_video(void)
  643. {
  644. return (obs != NULL) ? obs->video.video : NULL;
  645. }
  646. /* TODO: optimize this later so it's not just O(N) string lookups */
  647. static inline struct obs_modal_ui *get_modal_ui_callback(const char *id,
  648. const char *task, const char *target)
  649. {
  650. for (size_t i = 0; i < obs->modal_ui_callbacks.num; i++) {
  651. struct obs_modal_ui *callback = obs->modal_ui_callbacks.array+i;
  652. if (strcmp(callback->id, id) == 0 &&
  653. strcmp(callback->task, task) == 0 &&
  654. strcmp(callback->target, target) == 0)
  655. return callback;
  656. }
  657. return NULL;
  658. }
  659. static inline struct obs_modeless_ui *get_modeless_ui_callback(const char *id,
  660. const char *task, const char *target)
  661. {
  662. for (size_t i = 0; i < obs->modeless_ui_callbacks.num; i++) {
  663. struct obs_modeless_ui *callback;
  664. callback = obs->modeless_ui_callbacks.array+i;
  665. if (strcmp(callback->id, id) == 0 &&
  666. strcmp(callback->task, task) == 0 &&
  667. strcmp(callback->target, target) == 0)
  668. return callback;
  669. }
  670. return NULL;
  671. }
  672. int obs_exec_ui(const char *name, const char *task, const char *target,
  673. void *data, void *ui_data)
  674. {
  675. struct obs_modal_ui *callback;
  676. int errorcode = OBS_UI_NOTFOUND;
  677. if (!obs) return errorcode;
  678. callback = get_modal_ui_callback(name, task, target);
  679. if (callback) {
  680. bool success = callback->exec(data, ui_data);
  681. errorcode = success ? OBS_UI_SUCCESS : OBS_UI_CANCEL;
  682. }
  683. return errorcode;
  684. }
  685. void *obs_create_ui(const char *name, const char *task, const char *target,
  686. void *data, void *ui_data)
  687. {
  688. struct obs_modeless_ui *callback;
  689. if (!obs) return NULL;
  690. callback = get_modeless_ui_callback(name, task, target);
  691. return callback ? callback->create(data, ui_data) : NULL;
  692. }
  693. bool obs_add_source(obs_source_t source)
  694. {
  695. struct calldata params = {0};
  696. if (!obs) return false;
  697. if (!source) return false;
  698. pthread_mutex_lock(&obs->data.sources_mutex);
  699. da_push_back(obs->data.user_sources, &source);
  700. obs_source_addref(source);
  701. pthread_mutex_unlock(&obs->data.sources_mutex);
  702. calldata_setptr(&params, "source", source);
  703. signal_handler_signal(obs->signals, "source_add", &params);
  704. calldata_free(&params);
  705. return true;
  706. }
  707. obs_source_t obs_get_output_source(uint32_t channel)
  708. {
  709. if (!obs) return NULL;
  710. return obs_view_getsource(&obs->data.main_view, channel);
  711. }
  712. void obs_set_output_source(uint32_t channel, obs_source_t source)
  713. {
  714. assert(channel < MAX_CHANNELS);
  715. if (!obs) return;
  716. if (channel >= MAX_CHANNELS) return;
  717. struct obs_source *prev_source;
  718. struct obs_view *view = &obs->data.main_view;
  719. struct calldata params = {0};
  720. pthread_mutex_lock(&view->channels_mutex);
  721. obs_source_addref(source);
  722. prev_source = view->channels[channel];
  723. calldata_setint(&params, "channel", channel);
  724. calldata_setptr(&params, "prev_source", prev_source);
  725. calldata_setptr(&params, "source", source);
  726. signal_handler_signal(obs->signals, "channel_change", &params);
  727. calldata_getptr(&params, "source", &source);
  728. calldata_free(&params);
  729. view->channels[channel] = source;
  730. pthread_mutex_unlock(&view->channels_mutex);
  731. if (source)
  732. obs_source_activate(source, MAIN_VIEW);
  733. if (prev_source) {
  734. obs_source_deactivate(prev_source, MAIN_VIEW);
  735. obs_source_release(prev_source);
  736. }
  737. }
  738. void obs_enum_sources(bool (*enum_proc)(void*, obs_source_t), void *param)
  739. {
  740. if (!obs) return;
  741. pthread_mutex_lock(&obs->data.user_sources_mutex);
  742. for (size_t i = 0; i < obs->data.user_sources.num; i++) {
  743. struct obs_source *source = obs->data.user_sources.array[i];
  744. if (!enum_proc(param, source))
  745. break;
  746. }
  747. pthread_mutex_unlock(&obs->data.user_sources_mutex);
  748. }
  749. static inline void obs_enum(void *pstart, pthread_mutex_t *mutex, void *proc,
  750. void *param)
  751. {
  752. struct obs_context_data **start = pstart, *context;
  753. bool (*enum_proc)(void*, void*) = proc;
  754. assert(start);
  755. assert(mutex);
  756. assert(enum_proc);
  757. pthread_mutex_lock(mutex);
  758. context = *start;
  759. while (context) {
  760. if (!enum_proc(param, context))
  761. break;
  762. context = context->next;
  763. }
  764. pthread_mutex_unlock(mutex);
  765. }
  766. void obs_enum_outputs(bool (*enum_proc)(void*, obs_output_t), void *param)
  767. {
  768. if (!obs) return;
  769. obs_enum(&obs->data.first_output, &obs->data.outputs_mutex,
  770. enum_proc, param);
  771. }
  772. void obs_enum_encoders(bool (*enum_proc)(void*, obs_encoder_t), void *param)
  773. {
  774. if (!obs) return;
  775. obs_enum(&obs->data.first_encoder, &obs->data.encoders_mutex,
  776. enum_proc, param);
  777. }
  778. void obs_enum_services(bool (*enum_proc)(void*, obs_service_t), void *param)
  779. {
  780. if (!obs) return;
  781. obs_enum(&obs->data.first_service, &obs->data.services_mutex,
  782. enum_proc, param);
  783. }
  784. obs_source_t obs_get_source_by_name(const char *name)
  785. {
  786. struct obs_core_data *data = &obs->data;
  787. struct obs_source *source = NULL;
  788. size_t i;
  789. if (!obs) return NULL;
  790. pthread_mutex_lock(&data->user_sources_mutex);
  791. for (i = 0; i < data->user_sources.num; i++) {
  792. struct obs_source *cur_source = data->user_sources.array[i];
  793. if (strcmp(cur_source->context.name, name) == 0) {
  794. source = cur_source;
  795. obs_source_addref(source);
  796. break;
  797. }
  798. }
  799. pthread_mutex_unlock(&data->user_sources_mutex);
  800. return source;
  801. }
  802. static inline void *get_context_by_name(void *vfirst, const char *name,
  803. pthread_mutex_t *mutex)
  804. {
  805. struct obs_context_data **first = vfirst;
  806. struct obs_context_data *context;
  807. pthread_mutex_lock(mutex);
  808. context = *first;
  809. while (context) {
  810. if (strcmp(context->name, name) == 0)
  811. break;
  812. context = context->next;
  813. }
  814. pthread_mutex_unlock(mutex);
  815. return context;
  816. }
  817. obs_output_t obs_get_output_by_name(const char *name)
  818. {
  819. if (!obs) return NULL;
  820. return get_context_by_name(&obs->data.first_output, name,
  821. &obs->data.outputs_mutex);
  822. }
  823. obs_encoder_t obs_get_encoder_by_name(const char *name)
  824. {
  825. if (!obs) return NULL;
  826. return get_context_by_name(&obs->data.first_encoder, name,
  827. &obs->data.encoders_mutex);
  828. }
  829. obs_service_t obs_get_service_by_name(const char *name)
  830. {
  831. if (!obs) return NULL;
  832. return get_context_by_name(&obs->data.first_service, name,
  833. &obs->data.services_mutex);
  834. }
  835. effect_t obs_get_default_effect(void)
  836. {
  837. if (!obs) return NULL;
  838. return obs->video.default_effect;
  839. }
  840. effect_t obs_get_solid_effect(void)
  841. {
  842. if (!obs) return NULL;
  843. return obs->video.solid_effect;
  844. }
  845. signal_handler_t obs_signalhandler(void)
  846. {
  847. if (!obs) return NULL;
  848. return obs->signals;
  849. }
  850. proc_handler_t obs_prochandler(void)
  851. {
  852. if (!obs) return NULL;
  853. return obs->procs;
  854. }
  855. void obs_add_draw_callback(
  856. void (*draw)(void *param, uint32_t cx, uint32_t cy),
  857. void *param)
  858. {
  859. if (!obs) return;
  860. obs_display_add_draw_callback(&obs->video.main_display, draw, param);
  861. }
  862. void obs_remove_draw_callback(
  863. void (*draw)(void *param, uint32_t cx, uint32_t cy),
  864. void *param)
  865. {
  866. if (!obs) return;
  867. obs_display_remove_draw_callback(&obs->video.main_display, draw, param);
  868. }
  869. void obs_resize(uint32_t cx, uint32_t cy)
  870. {
  871. if (!obs || !obs->video.video || !obs->video.graphics) return;
  872. obs_display_resize(&obs->video.main_display, cx, cy);
  873. }
  874. void obs_render_main_view(void)
  875. {
  876. if (!obs) return;
  877. obs_view_render(&obs->data.main_view);
  878. }
  879. void obs_set_master_volume(float volume)
  880. {
  881. struct calldata data = {0};
  882. if (!obs) return;
  883. calldata_setfloat(&data, "volume", volume);
  884. signal_handler_signal(obs->signals, "master_volume", &data);
  885. volume = (float)calldata_float(&data, "volume");
  886. calldata_free(&data);
  887. obs->audio.user_volume = volume;
  888. }
  889. void obs_set_present_volume(float volume)
  890. {
  891. if (!obs) return;
  892. obs->audio.present_volume = volume;
  893. }
  894. float obs_get_master_volume(void)
  895. {
  896. return obs ? obs->audio.user_volume : 0.0f;
  897. }
  898. float obs_get_present_volume(void)
  899. {
  900. return obs ? obs->audio.present_volume : 0.0f;
  901. }
  902. obs_source_t obs_load_source(obs_data_t source_data)
  903. {
  904. obs_source_t source;
  905. const char *name = obs_data_getstring(source_data, "name");
  906. const char *id = obs_data_getstring(source_data, "id");
  907. obs_data_t settings = obs_data_getobj(source_data, "settings");
  908. double volume;
  909. source = obs_source_create(OBS_SOURCE_TYPE_INPUT, id, name, settings);
  910. obs_data_set_default_double(source_data, "volume", 1.0);
  911. volume = obs_data_getdouble(source_data, "volume");
  912. obs_source_setvolume(source, (float)volume);
  913. obs_data_release(settings);
  914. return source;
  915. }
  916. void obs_load_sources(obs_data_array_t array)
  917. {
  918. size_t count;
  919. size_t i;
  920. if (!obs) return;
  921. count = obs_data_array_count(array);
  922. pthread_mutex_lock(&obs->data.user_sources_mutex);
  923. for (i = 0; i < count; i++) {
  924. obs_data_t source_data = obs_data_array_item(array, i);
  925. obs_source_t source = obs_load_source(source_data);
  926. obs_add_source(source);
  927. obs_source_release(source);
  928. obs_data_release(source_data);
  929. }
  930. /* tell sources that we want to load */
  931. for (i = 0; i < obs->data.user_sources.num; i++)
  932. obs_source_load(obs->data.user_sources.array[i]);
  933. pthread_mutex_unlock(&obs->data.user_sources_mutex);
  934. }
  935. obs_data_t obs_save_source(obs_source_t source)
  936. {
  937. obs_data_t source_data = obs_data_create();
  938. obs_data_t settings = obs_source_getsettings(source);
  939. float volume = obs_source_getvolume(source);
  940. const char *name = obs_source_getname(source);
  941. const char *id;
  942. obs_source_save(source);
  943. obs_source_gettype(source, NULL, &id);
  944. obs_data_setstring(source_data, "name", name);
  945. obs_data_setstring(source_data, "id", id);
  946. obs_data_setobj (source_data, "settings", settings);
  947. obs_data_setdouble(source_data, "volume", volume);
  948. obs_data_release(settings);
  949. return source_data;
  950. }
  951. obs_data_array_t obs_save_sources(void)
  952. {
  953. obs_data_array_t array;
  954. size_t i;
  955. if (!obs) return NULL;
  956. array = obs_data_array_create();
  957. pthread_mutex_lock(&obs->data.user_sources_mutex);
  958. for (i = 0; i < obs->data.user_sources.num; i++) {
  959. obs_source_t source = obs->data.user_sources.array[i];
  960. obs_data_t source_data = obs_save_source(source);
  961. obs_data_array_push_back(array, source_data);
  962. obs_data_release(source_data);
  963. }
  964. pthread_mutex_unlock(&obs->data.user_sources_mutex);
  965. return array;
  966. }
  967. /* ensures that names are never blank */
  968. static inline char *dup_name(const char *name)
  969. {
  970. if (!name || !*name) {
  971. struct dstr unnamed = {0};
  972. dstr_printf(&unnamed, "__unnamed%004lld",
  973. obs->data.unnamed_index++);
  974. return unnamed.array;
  975. } else {
  976. return bstrdup(name);
  977. }
  978. }
  979. static inline bool obs_context_data_init_wrap(
  980. struct obs_context_data *context,
  981. obs_data_t settings,
  982. const char *name)
  983. {
  984. assert(context);
  985. memset(context, 0, sizeof(*context));
  986. pthread_mutex_init_value(&context->rename_cache_mutex);
  987. if (pthread_mutex_init(&context->rename_cache_mutex, NULL) < 0)
  988. return false;
  989. context->signals = signal_handler_create();
  990. if (!context)
  991. return false;
  992. context->procs = proc_handler_create();
  993. if (!context->procs)
  994. return false;
  995. context->name = dup_name(name);
  996. context->settings = obs_data_newref(settings);
  997. return true;
  998. }
  999. bool obs_context_data_init(
  1000. struct obs_context_data *context,
  1001. obs_data_t settings,
  1002. const char *name)
  1003. {
  1004. if (obs_context_data_init_wrap(context, settings, name)) {
  1005. return true;
  1006. } else {
  1007. obs_context_data_free(context);
  1008. return false;
  1009. }
  1010. }
  1011. void obs_context_data_free(struct obs_context_data *context)
  1012. {
  1013. signal_handler_destroy(context->signals);
  1014. proc_handler_destroy(context->procs);
  1015. obs_data_release(context->settings);
  1016. obs_context_data_remove(context);
  1017. pthread_mutex_destroy(&context->rename_cache_mutex);
  1018. bfree(context->name);
  1019. for (size_t i = 0; i < context->rename_cache.num; i++)
  1020. bfree(context->rename_cache.array[i]);
  1021. da_free(context->rename_cache);
  1022. memset(context, 0, sizeof(*context));
  1023. }
  1024. void obs_context_data_insert(struct obs_context_data *context,
  1025. pthread_mutex_t *mutex, void *pfirst)
  1026. {
  1027. struct obs_context_data **first = pfirst;
  1028. assert(context);
  1029. assert(mutex);
  1030. assert(first);
  1031. context->mutex = mutex;
  1032. pthread_mutex_lock(mutex);
  1033. context->prev_next = first;
  1034. context->next = *first;
  1035. *first = context;
  1036. if (context->next)
  1037. context->next->prev_next = &context->next;
  1038. pthread_mutex_unlock(mutex);
  1039. }
  1040. void obs_context_data_remove(struct obs_context_data *context)
  1041. {
  1042. if (context && context->mutex) {
  1043. pthread_mutex_lock(context->mutex);
  1044. *context->prev_next = context->next;
  1045. if (context->next)
  1046. context->next->prev_next = context->prev_next;
  1047. pthread_mutex_unlock(context->mutex);
  1048. context->mutex = NULL;
  1049. }
  1050. }
  1051. void obs_context_data_setname(struct obs_context_data *context,
  1052. const char *name)
  1053. {
  1054. pthread_mutex_lock(&context->rename_cache_mutex);
  1055. if (context->name)
  1056. da_push_back(context->rename_cache, &context->name);
  1057. context->name = dup_name(name);
  1058. pthread_mutex_unlock(&context->rename_cache_mutex);
  1059. }