obs-output.c 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940
  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 "util/platform.h"
  16. #include "obs.h"
  17. #include "obs-internal.h"
  18. static inline void signal_stop(struct obs_output *output, int code);
  19. const struct obs_output_info *find_output(const char *id)
  20. {
  21. size_t i;
  22. for (i = 0; i < obs->output_types.num; i++)
  23. if (strcmp(obs->output_types.array[i].id, id) == 0)
  24. return obs->output_types.array+i;
  25. return NULL;
  26. }
  27. const char *obs_output_get_display_name(const char *id)
  28. {
  29. const struct obs_output_info *info = find_output(id);
  30. return (info != NULL) ? info->get_name() : NULL;
  31. }
  32. static const char *output_signals[] = {
  33. "void start(ptr output)",
  34. "void stop(ptr output, int code)",
  35. "void reconnect(ptr output)",
  36. "void reconnect_success(ptr output)",
  37. NULL
  38. };
  39. static bool init_output_handlers(struct obs_output *output, const char *name,
  40. obs_data_t *settings)
  41. {
  42. if (!obs_context_data_init(&output->context, settings, name))
  43. return false;
  44. signal_handler_add_array(output->context.signals, output_signals);
  45. return true;
  46. }
  47. obs_output_t *obs_output_create(const char *id, const char *name,
  48. obs_data_t *settings)
  49. {
  50. const struct obs_output_info *info = find_output(id);
  51. struct obs_output *output;
  52. int ret;
  53. if (!info) {
  54. blog(LOG_ERROR, "Output '%s' not found", id);
  55. return NULL;
  56. }
  57. output = bzalloc(sizeof(struct obs_output));
  58. pthread_mutex_init_value(&output->interleaved_mutex);
  59. if (pthread_mutex_init(&output->interleaved_mutex, NULL) != 0)
  60. goto fail;
  61. if (!init_output_handlers(output, name, settings))
  62. goto fail;
  63. output->info = *info;
  64. output->video = obs_get_video();
  65. output->audio = obs_get_audio();
  66. if (output->info.get_defaults)
  67. output->info.get_defaults(output->context.settings);
  68. ret = os_event_init(&output->reconnect_stop_event,
  69. OS_EVENT_TYPE_MANUAL);
  70. if (ret < 0)
  71. goto fail;
  72. output->context.data = info->create(output->context.settings, output);
  73. if (!output->context.data)
  74. goto fail;
  75. output->reconnect_retry_sec = 2;
  76. output->reconnect_retry_max = 20;
  77. output->valid = true;
  78. obs_context_data_insert(&output->context,
  79. &obs->data.outputs_mutex,
  80. &obs->data.first_output);
  81. blog(LOG_INFO, "output '%s' (%s) created", name, id);
  82. return output;
  83. fail:
  84. obs_output_destroy(output);
  85. return NULL;
  86. }
  87. static inline void free_packets(struct obs_output *output)
  88. {
  89. for (size_t i = 0; i < output->interleaved_packets.num; i++)
  90. obs_free_encoder_packet(output->interleaved_packets.array+i);
  91. da_free(output->interleaved_packets);
  92. }
  93. void obs_output_destroy(obs_output_t *output)
  94. {
  95. if (output) {
  96. obs_context_data_remove(&output->context);
  97. blog(LOG_INFO, "output '%s' destroyed", output->context.name);
  98. if (output->valid && output->active)
  99. obs_output_stop(output);
  100. if (output->service)
  101. output->service->output = NULL;
  102. free_packets(output);
  103. if (output->context.data)
  104. output->info.destroy(output->context.data);
  105. if (output->video_encoder) {
  106. obs_encoder_remove_output(output->video_encoder,
  107. output);
  108. }
  109. if (output->audio_encoder) {
  110. obs_encoder_remove_output(output->audio_encoder,
  111. output);
  112. }
  113. pthread_mutex_destroy(&output->interleaved_mutex);
  114. os_event_destroy(output->reconnect_stop_event);
  115. obs_context_data_free(&output->context);
  116. bfree(output);
  117. }
  118. }
  119. const char *obs_output_get_name(const obs_output_t *output)
  120. {
  121. return output ? output->context.name : NULL;
  122. }
  123. bool obs_output_start(obs_output_t *output)
  124. {
  125. bool success;
  126. if (!output)
  127. return false;
  128. success = output->info.start(output->context.data);
  129. if (success && output->video) {
  130. output->starting_frame_count =
  131. video_output_get_total_frames(output->video);
  132. output->starting_skipped_frame_count =
  133. video_output_get_skipped_frames(output->video);
  134. }
  135. return success;
  136. }
  137. static void log_frame_info(struct obs_output *output)
  138. {
  139. uint32_t video_frames = video_output_get_total_frames(output->video);
  140. uint32_t video_skipped = video_output_get_skipped_frames(output->video);
  141. uint32_t total = video_frames - output->starting_frame_count;
  142. uint32_t skipped = video_skipped - output->starting_skipped_frame_count;
  143. double percentage_skipped = (double)skipped / (double)total * 100.0;
  144. blog(LOG_INFO, "Output '%s': stopping", output->context.name);
  145. blog(LOG_INFO, "Output '%s': Total frames: %"PRIu32,
  146. output->context.name, total);
  147. if (total)
  148. blog(LOG_INFO, "Output '%s': Number of skipped frames: "
  149. "%"PRIu32" (%g%%)",
  150. output->context.name,
  151. skipped, percentage_skipped);
  152. }
  153. void obs_output_stop(obs_output_t *output)
  154. {
  155. if (output) {
  156. os_event_signal(output->reconnect_stop_event);
  157. if (output->reconnect_thread_active)
  158. pthread_join(output->reconnect_thread, NULL);
  159. output->info.stop(output->context.data);
  160. signal_stop(output, OBS_OUTPUT_SUCCESS);
  161. if (output->video)
  162. log_frame_info(output);
  163. }
  164. }
  165. bool obs_output_active(const obs_output_t *output)
  166. {
  167. return (output != NULL) ?
  168. (output->active || output->reconnecting) : false;
  169. }
  170. static inline obs_data_t *get_defaults(const struct obs_output_info *info)
  171. {
  172. obs_data_t *settings = obs_data_create();
  173. if (info->get_defaults)
  174. info->get_defaults(settings);
  175. return settings;
  176. }
  177. obs_data_t *obs_output_defaults(const char *id)
  178. {
  179. const struct obs_output_info *info = find_output(id);
  180. return (info) ? get_defaults(info) : NULL;
  181. }
  182. obs_properties_t *obs_get_output_properties(const char *id)
  183. {
  184. const struct obs_output_info *info = find_output(id);
  185. if (info && info->get_properties) {
  186. obs_data_t *defaults = get_defaults(info);
  187. obs_properties_t *properties;
  188. properties = info->get_properties();
  189. obs_properties_apply_settings(properties, defaults);
  190. obs_data_release(defaults);
  191. return properties;
  192. }
  193. return NULL;
  194. }
  195. obs_properties_t *obs_output_properties(const obs_output_t *output)
  196. {
  197. if (output && output->info.get_properties) {
  198. obs_properties_t *props;
  199. props = output->info.get_properties();
  200. obs_properties_apply_settings(props, output->context.settings);
  201. return props;
  202. }
  203. return NULL;
  204. }
  205. void obs_output_update(obs_output_t *output, obs_data_t *settings)
  206. {
  207. if (!output) return;
  208. obs_data_apply(output->context.settings, settings);
  209. if (output->info.update)
  210. output->info.update(output->context.data,
  211. output->context.settings);
  212. }
  213. obs_data_t *obs_output_get_settings(const obs_output_t *output)
  214. {
  215. if (!output)
  216. return NULL;
  217. obs_data_addref(output->context.settings);
  218. return output->context.settings;
  219. }
  220. bool obs_output_canpause(const obs_output_t *output)
  221. {
  222. return output ? (output->info.pause != NULL) : false;
  223. }
  224. void obs_output_pause(obs_output_t *output)
  225. {
  226. if (output && output->info.pause)
  227. output->info.pause(output->context.data);
  228. }
  229. signal_handler_t *obs_output_get_signal_handler(const obs_output_t *output)
  230. {
  231. return output ? output->context.signals : NULL;
  232. }
  233. proc_handler_t *obs_output_get_proc_handler(const obs_output_t *output)
  234. {
  235. return output ? output->context.procs : NULL;
  236. }
  237. void obs_output_set_media(obs_output_t *output, video_t *video, audio_t *audio)
  238. {
  239. if (!output)
  240. return;
  241. output->video = video;
  242. output->audio = audio;
  243. }
  244. video_t *obs_output_video(const obs_output_t *output)
  245. {
  246. return output ? output->video : NULL;
  247. }
  248. audio_t *obs_output_audio(const obs_output_t *output)
  249. {
  250. return output ? output->audio : NULL;
  251. }
  252. void obs_output_remove_encoder(struct obs_output *output,
  253. struct obs_encoder *encoder)
  254. {
  255. if (!output) return;
  256. if (output->video_encoder == encoder)
  257. output->video_encoder = NULL;
  258. else if (output->audio_encoder == encoder)
  259. output->audio_encoder = NULL;
  260. }
  261. void obs_output_set_video_encoder(obs_output_t *output, obs_encoder_t *encoder)
  262. {
  263. if (!output) return;
  264. if (output->video_encoder == encoder) return;
  265. if (encoder && encoder->info.type != OBS_ENCODER_VIDEO) return;
  266. obs_encoder_remove_output(encoder, output);
  267. obs_encoder_add_output(encoder, output);
  268. output->video_encoder = encoder;
  269. /* set the preferred resolution on the encoder */
  270. if (output->scaled_width && output->scaled_height)
  271. obs_encoder_set_scaled_size(output->video_encoder,
  272. output->scaled_width, output->scaled_height);
  273. }
  274. void obs_output_set_audio_encoder(obs_output_t *output, obs_encoder_t *encoder)
  275. {
  276. if (!output) return;
  277. if (output->audio_encoder == encoder) return;
  278. if (encoder && encoder->info.type != OBS_ENCODER_AUDIO) return;
  279. obs_encoder_remove_output(encoder, output);
  280. obs_encoder_add_output(encoder, output);
  281. output->audio_encoder = encoder;
  282. }
  283. obs_encoder_t *obs_output_get_video_encoder(const obs_output_t *output)
  284. {
  285. return output ? output->video_encoder : NULL;
  286. }
  287. obs_encoder_t *obs_output_get_audio_encoder(const obs_output_t *output)
  288. {
  289. return output ? output->audio_encoder : NULL;
  290. }
  291. void obs_output_set_service(obs_output_t *output, obs_service_t *service)
  292. {
  293. if (!output || output->active || !service || service->active) return;
  294. if (service->output)
  295. service->output->service = NULL;
  296. output->service = service;
  297. service->output = output;
  298. }
  299. obs_service_t *obs_output_get_service(const obs_output_t *output)
  300. {
  301. return output ? output->service : NULL;
  302. }
  303. void obs_output_set_reconnect_settings(obs_output_t *output,
  304. int retry_count, int retry_sec)
  305. {
  306. if (!output) return;
  307. output->reconnect_retry_max = retry_count;
  308. output->reconnect_retry_sec = retry_sec;
  309. }
  310. uint64_t obs_output_get_total_bytes(const obs_output_t *output)
  311. {
  312. if (!output || !output->info.get_total_bytes)
  313. return 0;
  314. return output->info.get_total_bytes(output->context.data);
  315. }
  316. int obs_output_get_frames_dropped(const obs_output_t *output)
  317. {
  318. if (!output || !output->info.get_dropped_frames)
  319. return 0;
  320. return output->info.get_dropped_frames(output->context.data);
  321. }
  322. int obs_output_get_total_frames(const obs_output_t *output)
  323. {
  324. return output ? output->total_frames : 0;
  325. }
  326. void obs_output_set_preferred_size(obs_output_t *output, uint32_t width,
  327. uint32_t height)
  328. {
  329. if (!output || (output->info.flags & OBS_OUTPUT_VIDEO) == 0)
  330. return;
  331. if (output->active) {
  332. blog(LOG_WARNING, "output '%s': Cannot set the preferred "
  333. "resolution while the output is active",
  334. obs_output_get_name(output));
  335. return;
  336. }
  337. output->scaled_width = width;
  338. output->scaled_height = height;
  339. if (output->info.flags & OBS_OUTPUT_ENCODED) {
  340. if (output->video_encoder)
  341. obs_encoder_set_scaled_size(output->video_encoder,
  342. width, height);
  343. }
  344. }
  345. uint32_t obs_output_get_width(const obs_output_t *output)
  346. {
  347. if (!output || (output->info.flags & OBS_OUTPUT_VIDEO) == 0)
  348. return 0;
  349. if (output->info.flags & OBS_OUTPUT_ENCODED)
  350. return obs_encoder_get_width(output->video_encoder);
  351. else
  352. return output->scaled_width != 0 ?
  353. output->scaled_width :
  354. video_output_get_width(output->video);
  355. }
  356. uint32_t obs_output_get_height(const obs_output_t *output)
  357. {
  358. if (!output || (output->info.flags & OBS_OUTPUT_VIDEO) == 0)
  359. return 0;
  360. if (output->info.flags & OBS_OUTPUT_ENCODED)
  361. return obs_encoder_get_height(output->video_encoder);
  362. else
  363. return output->scaled_height != 0 ?
  364. output->scaled_height :
  365. video_output_get_height(output->video);
  366. }
  367. void obs_output_set_video_conversion(obs_output_t *output,
  368. const struct video_scale_info *conversion)
  369. {
  370. if (!output || !conversion) return;
  371. output->video_conversion = *conversion;
  372. output->video_conversion_set = true;
  373. }
  374. void obs_output_set_audio_conversion(obs_output_t *output,
  375. const struct audio_convert_info *conversion)
  376. {
  377. if (!output || !conversion) return;
  378. output->audio_conversion = *conversion;
  379. output->audio_conversion_set = true;
  380. }
  381. static bool can_begin_data_capture(const struct obs_output *output,
  382. bool encoded, bool has_video, bool has_audio, bool has_service)
  383. {
  384. if (has_video) {
  385. if (encoded) {
  386. if (!output->video_encoder)
  387. return false;
  388. } else {
  389. if (!output->video)
  390. return false;
  391. }
  392. }
  393. if (has_audio) {
  394. if (encoded) {
  395. if (!output->audio_encoder)
  396. return false;
  397. } else {
  398. if (!output->audio)
  399. return false;
  400. }
  401. }
  402. if (has_service && !output->service)
  403. return false;
  404. return true;
  405. }
  406. static inline bool has_scaling(const struct obs_output *output)
  407. {
  408. uint32_t video_width = video_output_get_width(output->video);
  409. uint32_t video_height = video_output_get_height(output->video);
  410. return output->scaled_width && output->scaled_height &&
  411. (video_width != output->scaled_width ||
  412. video_height != output->scaled_height);
  413. }
  414. static inline struct video_scale_info *get_video_conversion(
  415. struct obs_output *output)
  416. {
  417. if (output->video_conversion_set) {
  418. if (!output->video_conversion.width)
  419. output->video_conversion.width =
  420. obs_output_get_width(output);
  421. if (!output->video_conversion.height)
  422. output->video_conversion.height =
  423. obs_output_get_height(output);
  424. return &output->video_conversion;
  425. } else if (has_scaling(output)) {
  426. const struct video_output_info *info =
  427. video_output_get_info(output->video);
  428. output->video_conversion.format = info->format;
  429. output->video_conversion.colorspace = VIDEO_CS_DEFAULT;
  430. output->video_conversion.range = VIDEO_RANGE_DEFAULT;
  431. output->video_conversion.width = output->scaled_width;
  432. output->video_conversion.height = output->scaled_height;
  433. return &output->video_conversion;
  434. }
  435. return NULL;
  436. }
  437. static inline struct audio_convert_info *get_audio_conversion(
  438. struct obs_output *output)
  439. {
  440. return output->audio_conversion_set ? &output->audio_conversion : NULL;
  441. }
  442. static bool prepare_interleaved_packet(struct obs_output *output,
  443. struct encoder_packet *out, struct encoder_packet *in)
  444. {
  445. int64_t offset;
  446. /* audio and video need to start at timestamp 0, and the encoders
  447. * may not currently be at 0 when we get data. so, we store the
  448. * current dts as offset and subtract that value from the dts/pts
  449. * of the output packet. */
  450. if (in->type == OBS_ENCODER_VIDEO) {
  451. if (!output->received_video) {
  452. output->first_video_ts = in->dts_usec;
  453. output->video_offset = in->dts;
  454. output->received_video = true;
  455. }
  456. offset = output->video_offset;
  457. } else{
  458. /* don't accept audio that's before the first video timestamp */
  459. if (!output->received_video ||
  460. in->dts_usec < output->first_video_ts)
  461. return false;
  462. if (!output->received_audio) {
  463. output->audio_offset = in->dts;
  464. output->received_audio = true;
  465. }
  466. offset = output->audio_offset;
  467. }
  468. obs_duplicate_encoder_packet(out, in);
  469. out->dts -= offset;
  470. out->pts -= offset;
  471. /* convert the newly adjusted dts to relative dts time to ensure proper
  472. * interleaving. if we're using an audio encoder that's already been
  473. * started on another output, then the first audio packet may not be
  474. * quite perfectly synced up in terms of system time (and there's
  475. * nothing we can really do about that), but it will always at least be
  476. * within a 23ish millisecond threshold (at least for AAC) */
  477. out->dts_usec = packet_dts_usec(out);
  478. return true;
  479. }
  480. static inline bool has_higher_opposing_ts(struct obs_output *output,
  481. struct encoder_packet *packet)
  482. {
  483. if (packet->type == OBS_ENCODER_VIDEO)
  484. return output->highest_audio_ts > packet->dts_usec;
  485. else
  486. return output->highest_video_ts > packet->dts_usec;
  487. }
  488. static inline void send_interleaved(struct obs_output *output)
  489. {
  490. struct encoder_packet out = output->interleaved_packets.array[0];
  491. /* do not send an interleaved packet if there's no packet of the
  492. * opposing type of a higher timstamp in the interleave buffer.
  493. * this ensures that the timestamps are monotonic */
  494. if (!has_higher_opposing_ts(output, &out))
  495. return;
  496. if (out.type == OBS_ENCODER_VIDEO)
  497. output->total_frames++;
  498. da_erase(output->interleaved_packets, 0);
  499. output->info.encoded_packet(output->context.data, &out);
  500. obs_free_encoder_packet(&out);
  501. }
  502. static inline void set_higher_ts(struct obs_output *output,
  503. struct encoder_packet *packet)
  504. {
  505. if (packet->type == OBS_ENCODER_VIDEO) {
  506. if (output->highest_video_ts < packet->dts_usec)
  507. output->highest_video_ts = packet->dts_usec;
  508. } else {
  509. if (output->highest_audio_ts < packet->dts_usec)
  510. output->highest_audio_ts = packet->dts_usec;
  511. }
  512. }
  513. static void interleave_packets(void *data, struct encoder_packet *packet)
  514. {
  515. struct obs_output *output = data;
  516. struct encoder_packet out;
  517. size_t idx;
  518. pthread_mutex_lock(&output->interleaved_mutex);
  519. if (prepare_interleaved_packet(output, &out, packet)) {
  520. for (idx = 0; idx < output->interleaved_packets.num; idx++) {
  521. struct encoder_packet *cur_packet;
  522. cur_packet = output->interleaved_packets.array + idx;
  523. if (out.dts_usec < cur_packet->dts_usec)
  524. break;
  525. }
  526. da_insert(output->interleaved_packets, idx, &out);
  527. set_higher_ts(output, &out);
  528. /* when both video and audio have been received, we're ready
  529. * to start sending out packets (one at a time) */
  530. if (output->received_audio && output->received_video)
  531. send_interleaved(output);
  532. }
  533. pthread_mutex_unlock(&output->interleaved_mutex);
  534. }
  535. static void default_encoded_callback(void *param, struct encoder_packet *packet)
  536. {
  537. struct obs_output *output = param;
  538. output->info.encoded_packet(output->context.data, packet);
  539. if (packet->type == OBS_ENCODER_VIDEO)
  540. output->total_frames++;
  541. }
  542. static void default_raw_video_callback(void *param, struct video_data *frame)
  543. {
  544. struct obs_output *output = param;
  545. output->info.raw_video(output->context.data, frame);
  546. output->total_frames++;
  547. }
  548. static void hook_data_capture(struct obs_output *output, bool encoded,
  549. bool has_video, bool has_audio)
  550. {
  551. void (*encoded_callback)(void *data, struct encoder_packet *packet);
  552. if (encoded) {
  553. output->received_video = false;
  554. output->received_video = false;
  555. output->highest_audio_ts = 0;
  556. output->highest_video_ts = 0;
  557. free_packets(output);
  558. encoded_callback = (has_video && has_audio) ?
  559. interleave_packets : default_encoded_callback;
  560. if (has_video)
  561. obs_encoder_start(output->video_encoder,
  562. encoded_callback, output);
  563. if (has_audio)
  564. obs_encoder_start(output->audio_encoder,
  565. encoded_callback, output);
  566. } else {
  567. if (has_video)
  568. video_output_connect(output->video,
  569. get_video_conversion(output),
  570. default_raw_video_callback, output);
  571. if (has_audio)
  572. audio_output_connect(output->audio,
  573. get_audio_conversion(output),
  574. output->info.raw_audio,
  575. output->context.data);
  576. }
  577. }
  578. static inline void do_output_signal(struct obs_output *output,
  579. const char *signal)
  580. {
  581. struct calldata params = {0};
  582. calldata_set_ptr(&params, "output", output);
  583. signal_handler_signal(output->context.signals, signal, &params);
  584. calldata_free(&params);
  585. }
  586. static inline void signal_start(struct obs_output *output)
  587. {
  588. do_output_signal(output, "start");
  589. }
  590. static inline void signal_reconnect(struct obs_output *output)
  591. {
  592. do_output_signal(output, "reconnect");
  593. }
  594. static inline void signal_reconnect_success(struct obs_output *output)
  595. {
  596. do_output_signal(output, "reconnect_success");
  597. }
  598. static inline void signal_stop(struct obs_output *output, int code)
  599. {
  600. struct calldata params = {0};
  601. calldata_set_int(&params, "code", code);
  602. calldata_set_ptr(&params, "output", output);
  603. signal_handler_signal(output->context.signals, "stop", &params);
  604. calldata_free(&params);
  605. }
  606. static inline void convert_flags(const struct obs_output *output,
  607. uint32_t flags, bool *encoded, bool *has_video, bool *has_audio,
  608. bool *has_service)
  609. {
  610. *encoded = (output->info.flags & OBS_OUTPUT_ENCODED) != 0;
  611. if (!flags)
  612. flags = output->info.flags;
  613. else
  614. flags &= output->info.flags;
  615. *has_video = (flags & OBS_OUTPUT_VIDEO) != 0;
  616. *has_audio = (flags & OBS_OUTPUT_AUDIO) != 0;
  617. *has_service = (flags & OBS_OUTPUT_SERVICE) != 0;
  618. }
  619. bool obs_output_can_begin_data_capture(const obs_output_t *output,
  620. uint32_t flags)
  621. {
  622. bool encoded, has_video, has_audio, has_service;
  623. if (!output) return false;
  624. if (output->active) return false;
  625. convert_flags(output, flags, &encoded, &has_video, &has_audio,
  626. &has_service);
  627. return can_begin_data_capture(output, encoded, has_video, has_audio,
  628. has_service);
  629. }
  630. bool obs_output_initialize_encoders(obs_output_t *output, uint32_t flags)
  631. {
  632. bool encoded, has_video, has_audio, has_service;
  633. if (!output) return false;
  634. if (output->active) return false;
  635. convert_flags(output, flags, &encoded, &has_video, &has_audio,
  636. &has_service);
  637. if (!encoded)
  638. return false;
  639. if (has_service && !obs_service_initialize(output->service, output))
  640. return false;
  641. if (has_video && !obs_encoder_initialize(output->video_encoder))
  642. return false;
  643. if (has_audio && !obs_encoder_initialize(output->audio_encoder))
  644. return false;
  645. if (has_video && has_audio && !output->audio_encoder->active &&
  646. !output->video_encoder->active) {
  647. output->audio_encoder->wait_for_video = true;
  648. output->video_encoder->paired_encoder = output->audio_encoder;
  649. output->audio_encoder->paired_encoder = output->video_encoder;
  650. }
  651. return true;
  652. }
  653. bool obs_output_begin_data_capture(obs_output_t *output, uint32_t flags)
  654. {
  655. bool encoded, has_video, has_audio, has_service;
  656. if (!output) return false;
  657. if (output->active) return false;
  658. output->total_frames = 0;
  659. convert_flags(output, flags, &encoded, &has_video, &has_audio,
  660. &has_service);
  661. if (!can_begin_data_capture(output, encoded, has_video, has_audio,
  662. has_service))
  663. return false;
  664. hook_data_capture(output, encoded, has_video, has_audio);
  665. if (has_service)
  666. obs_service_activate(output->service);
  667. output->active = true;
  668. if (output->reconnecting) {
  669. signal_reconnect_success(output);
  670. output->reconnecting = false;
  671. } else {
  672. signal_start(output);
  673. }
  674. return true;
  675. }
  676. void obs_output_end_data_capture(obs_output_t *output)
  677. {
  678. bool encoded, has_video, has_audio, has_service;
  679. void (*encoded_callback)(void *data, struct encoder_packet *packet);
  680. if (!output) return;
  681. if (!output->active) return;
  682. convert_flags(output, 0, &encoded, &has_video, &has_audio,
  683. &has_service);
  684. if (encoded) {
  685. encoded_callback = (has_video && has_audio) ?
  686. interleave_packets : default_encoded_callback;
  687. if (has_video)
  688. obs_encoder_stop(output->video_encoder,
  689. encoded_callback, output);
  690. if (has_audio)
  691. obs_encoder_stop(output->audio_encoder,
  692. encoded_callback, output);
  693. } else {
  694. if (has_video)
  695. video_output_disconnect(output->video,
  696. default_raw_video_callback, output);
  697. if (has_audio)
  698. audio_output_disconnect(output->audio,
  699. output->info.raw_audio,
  700. output->context.data);
  701. }
  702. if (has_service)
  703. obs_service_deactivate(output->service, false);
  704. output->active = false;
  705. }
  706. static void *reconnect_thread(void *param)
  707. {
  708. struct obs_output *output = param;
  709. unsigned long ms = output->reconnect_retry_sec * 1000;
  710. output->reconnect_thread_active = true;
  711. if (os_event_timedwait(output->reconnect_stop_event, ms) == ETIMEDOUT)
  712. obs_output_start(output);
  713. if (os_event_try(output->reconnect_stop_event) == EAGAIN)
  714. pthread_detach(output->reconnect_thread);
  715. output->reconnect_thread_active = false;
  716. return NULL;
  717. }
  718. static void output_reconnect(struct obs_output *output)
  719. {
  720. int ret;
  721. if (!output->reconnecting)
  722. output->reconnect_retries = 0;
  723. if (output->reconnect_retries >= output->reconnect_retry_max) {
  724. output->reconnecting = false;
  725. signal_stop(output, OBS_OUTPUT_DISCONNECTED);
  726. return;
  727. }
  728. if (!output->reconnecting) {
  729. output->reconnecting = true;
  730. os_event_reset(output->reconnect_stop_event);
  731. }
  732. output->reconnect_retries++;
  733. ret = pthread_create(&output->reconnect_thread, NULL,
  734. &reconnect_thread, output);
  735. if (ret < 0) {
  736. blog(LOG_WARNING, "Failed to create reconnect thread");
  737. output->reconnecting = false;
  738. signal_stop(output, OBS_OUTPUT_DISCONNECTED);
  739. } else {
  740. blog(LOG_INFO, "Output '%s': Reconnecting in %d seconds..",
  741. output->context.name,
  742. output->reconnect_retry_sec);
  743. signal_reconnect(output);
  744. }
  745. }
  746. void obs_output_signal_stop(obs_output_t *output, int code)
  747. {
  748. if (!output)
  749. return;
  750. obs_output_end_data_capture(output);
  751. if ((output->reconnecting && code != OBS_OUTPUT_SUCCESS) ||
  752. code == OBS_OUTPUT_DISCONNECTED)
  753. output_reconnect(output);
  754. else
  755. signal_stop(output, code);
  756. }