obs-encoder.c 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866
  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 "obs.h"
  15. #include "obs-internal.h"
  16. struct obs_encoder_info *find_encoder(const char *id)
  17. {
  18. for (size_t i = 0; i < obs->encoder_types.num; i++) {
  19. struct obs_encoder_info *info = obs->encoder_types.array+i;
  20. if (strcmp(info->id, id) == 0)
  21. return info;
  22. }
  23. return NULL;
  24. }
  25. const char *obs_encoder_get_display_name(const char *id)
  26. {
  27. struct obs_encoder_info *ei = find_encoder(id);
  28. return ei ? ei->get_name() : NULL;
  29. }
  30. static bool init_encoder(struct obs_encoder *encoder, const char *name,
  31. obs_data_t *settings, obs_data_t *hotkey_data)
  32. {
  33. pthread_mutex_init_value(&encoder->callbacks_mutex);
  34. pthread_mutex_init_value(&encoder->outputs_mutex);
  35. if (!obs_context_data_init(&encoder->context, settings, name,
  36. hotkey_data))
  37. return false;
  38. if (pthread_mutex_init(&encoder->callbacks_mutex, NULL) != 0)
  39. return false;
  40. if (pthread_mutex_init(&encoder->outputs_mutex, NULL) != 0)
  41. return false;
  42. if (encoder->info.get_defaults)
  43. encoder->info.get_defaults(encoder->context.settings);
  44. return true;
  45. }
  46. static struct obs_encoder *create_encoder(const char *id,
  47. enum obs_encoder_type type, const char *name,
  48. obs_data_t *settings, size_t mixer_idx, obs_data_t *hotkey_data)
  49. {
  50. struct obs_encoder *encoder;
  51. struct obs_encoder_info *ei = find_encoder(id);
  52. bool success;
  53. if (!ei || ei->type != type)
  54. return NULL;
  55. encoder = bzalloc(sizeof(struct obs_encoder));
  56. encoder->info = *ei;
  57. encoder->mixer_idx = mixer_idx;
  58. success = init_encoder(encoder, name, settings, hotkey_data);
  59. if (!success) {
  60. obs_encoder_destroy(encoder);
  61. encoder = NULL;
  62. }
  63. encoder->control = bzalloc(sizeof(obs_weak_encoder_t));
  64. encoder->control->encoder = encoder;
  65. obs_context_data_insert(&encoder->context,
  66. &obs->data.encoders_mutex,
  67. &obs->data.first_encoder);
  68. blog(LOG_INFO, "encoder '%s' (%s) created", name, id);
  69. return encoder;
  70. }
  71. obs_encoder_t *obs_video_encoder_create(const char *id, const char *name,
  72. obs_data_t *settings, obs_data_t *hotkey_data)
  73. {
  74. if (!name || !id) return NULL;
  75. return create_encoder(id, OBS_ENCODER_VIDEO, name, settings, 0,
  76. hotkey_data);
  77. }
  78. obs_encoder_t *obs_audio_encoder_create(const char *id, const char *name,
  79. obs_data_t *settings, size_t mixer_idx, obs_data_t *hotkey_data)
  80. {
  81. if (!name || !id) return NULL;
  82. return create_encoder(id, OBS_ENCODER_AUDIO, name, settings, mixer_idx,
  83. hotkey_data);
  84. }
  85. static void receive_video(void *param, struct video_data *frame);
  86. static void receive_audio(void *param, size_t mix_idx, struct audio_data *data);
  87. static inline void get_audio_info(const struct obs_encoder *encoder,
  88. struct audio_convert_info *info)
  89. {
  90. const struct audio_output_info *aoi;
  91. aoi = audio_output_get_info(encoder->media);
  92. if (info->format == AUDIO_FORMAT_UNKNOWN)
  93. info->format = aoi->format;
  94. if (!info->samples_per_sec)
  95. info->samples_per_sec = aoi->samples_per_sec;
  96. if (info->speakers == SPEAKERS_UNKNOWN)
  97. info->speakers = aoi->speakers;
  98. if (encoder->info.get_audio_info)
  99. encoder->info.get_audio_info(encoder->context.data, info);
  100. }
  101. static inline void get_video_info(struct obs_encoder *encoder,
  102. struct video_scale_info *info)
  103. {
  104. const struct video_output_info *voi;
  105. voi = video_output_get_info(encoder->media);
  106. info->format = voi->format;
  107. info->colorspace = voi->colorspace;
  108. info->range = voi->range;
  109. info->width = obs_encoder_get_width(encoder);
  110. info->height = obs_encoder_get_height(encoder);
  111. if (encoder->info.get_video_info)
  112. encoder->info.get_video_info(encoder->context.data, info);
  113. if (info->width != voi->width || info->height != voi->height)
  114. obs_encoder_set_scaled_size(encoder, info->width, info->height);
  115. }
  116. static inline bool has_scaling(const struct obs_encoder *encoder)
  117. {
  118. uint32_t video_width = video_output_get_width(encoder->media);
  119. uint32_t video_height = video_output_get_height(encoder->media);
  120. return encoder->scaled_width && encoder->scaled_height &&
  121. (video_width != encoder->scaled_width ||
  122. video_height != encoder->scaled_height);
  123. }
  124. static void add_connection(struct obs_encoder *encoder)
  125. {
  126. if (encoder->info.type == OBS_ENCODER_AUDIO) {
  127. struct audio_convert_info audio_info = {0};
  128. get_audio_info(encoder, &audio_info);
  129. audio_output_connect(encoder->media, encoder->mixer_idx,
  130. &audio_info, receive_audio, encoder);
  131. } else {
  132. struct video_scale_info info = {0};
  133. get_video_info(encoder, &info);
  134. video_output_connect(encoder->media, &info, receive_video,
  135. encoder);
  136. }
  137. encoder->active = true;
  138. }
  139. static void remove_connection(struct obs_encoder *encoder)
  140. {
  141. if (encoder->info.type == OBS_ENCODER_AUDIO)
  142. audio_output_disconnect(encoder->media, encoder->mixer_idx,
  143. receive_audio, encoder);
  144. else
  145. video_output_disconnect(encoder->media, receive_video,
  146. encoder);
  147. encoder->active = false;
  148. }
  149. static inline void free_audio_buffers(struct obs_encoder *encoder)
  150. {
  151. for (size_t i = 0; i < MAX_AV_PLANES; i++) {
  152. circlebuf_free(&encoder->audio_input_buffer[i]);
  153. bfree(encoder->audio_output_buffer[i]);
  154. encoder->audio_output_buffer[i] = NULL;
  155. }
  156. }
  157. static void obs_encoder_actually_destroy(obs_encoder_t *encoder)
  158. {
  159. if (encoder) {
  160. pthread_mutex_lock(&encoder->outputs_mutex);
  161. for (size_t i = 0; i < encoder->outputs.num; i++) {
  162. struct obs_output *output = encoder->outputs.array[i];
  163. obs_output_remove_encoder(output, encoder);
  164. }
  165. da_free(encoder->outputs);
  166. pthread_mutex_unlock(&encoder->outputs_mutex);
  167. blog(LOG_INFO, "encoder '%s' destroyed", encoder->context.name);
  168. free_audio_buffers(encoder);
  169. if (encoder->context.data)
  170. encoder->info.destroy(encoder->context.data);
  171. da_free(encoder->callbacks);
  172. pthread_mutex_destroy(&encoder->callbacks_mutex);
  173. pthread_mutex_destroy(&encoder->outputs_mutex);
  174. obs_context_data_free(&encoder->context);
  175. bfree(encoder);
  176. }
  177. }
  178. /* does not actually destroy the encoder until all connections to it have been
  179. * removed. (full reference counting really would have been superfluous) */
  180. void obs_encoder_destroy(obs_encoder_t *encoder)
  181. {
  182. if (encoder) {
  183. bool destroy;
  184. obs_context_data_remove(&encoder->context);
  185. pthread_mutex_lock(&encoder->callbacks_mutex);
  186. destroy = encoder->callbacks.num == 0;
  187. if (!destroy)
  188. encoder->destroy_on_stop = true;
  189. pthread_mutex_unlock(&encoder->callbacks_mutex);
  190. if (destroy)
  191. obs_encoder_actually_destroy(encoder);
  192. }
  193. }
  194. const char *obs_encoder_get_name(const obs_encoder_t *encoder)
  195. {
  196. return encoder ? encoder->context.name : NULL;
  197. }
  198. void obs_encoder_set_name(obs_encoder_t *encoder, const char *name)
  199. {
  200. if (!encoder) return;
  201. if (name && *name && strcmp(name, encoder->context.name) != 0)
  202. obs_context_data_setname(&encoder->context, name);
  203. }
  204. static inline obs_data_t *get_defaults(const struct obs_encoder_info *info)
  205. {
  206. obs_data_t *settings = obs_data_create();
  207. if (info->get_defaults)
  208. info->get_defaults(settings);
  209. return settings;
  210. }
  211. obs_data_t *obs_encoder_defaults(const char *id)
  212. {
  213. const struct obs_encoder_info *info = find_encoder(id);
  214. return (info) ? get_defaults(info) : NULL;
  215. }
  216. obs_properties_t *obs_get_encoder_properties(const char *id)
  217. {
  218. const struct obs_encoder_info *ei = find_encoder(id);
  219. if (ei && ei->get_properties) {
  220. obs_data_t *defaults = get_defaults(ei);
  221. obs_properties_t *properties;
  222. properties = ei->get_properties(NULL);
  223. obs_properties_apply_settings(properties, defaults);
  224. obs_data_release(defaults);
  225. return properties;
  226. }
  227. return NULL;
  228. }
  229. obs_properties_t *obs_encoder_properties(const obs_encoder_t *encoder)
  230. {
  231. if (encoder && encoder->info.get_properties) {
  232. obs_properties_t *props;
  233. props = encoder->info.get_properties(encoder->context.data);
  234. obs_properties_apply_settings(props, encoder->context.settings);
  235. return props;
  236. }
  237. return NULL;
  238. }
  239. void obs_encoder_update(obs_encoder_t *encoder, obs_data_t *settings)
  240. {
  241. if (!encoder) return;
  242. obs_data_apply(encoder->context.settings, settings);
  243. if (encoder->info.update && encoder->context.data)
  244. encoder->info.update(encoder->context.data,
  245. encoder->context.settings);
  246. }
  247. bool obs_encoder_get_extra_data(const obs_encoder_t *encoder,
  248. uint8_t **extra_data, size_t *size)
  249. {
  250. if (encoder && encoder->info.get_extra_data && encoder->context.data)
  251. return encoder->info.get_extra_data(encoder->context.data,
  252. extra_data, size);
  253. return false;
  254. }
  255. obs_data_t *obs_encoder_get_settings(const obs_encoder_t *encoder)
  256. {
  257. if (!encoder) return NULL;
  258. obs_data_addref(encoder->context.settings);
  259. return encoder->context.settings;
  260. }
  261. static inline void reset_audio_buffers(struct obs_encoder *encoder)
  262. {
  263. free_audio_buffers(encoder);
  264. for (size_t i = 0; i < encoder->planes; i++)
  265. encoder->audio_output_buffer[i] =
  266. bmalloc(encoder->framesize_bytes);
  267. }
  268. static void intitialize_audio_encoder(struct obs_encoder *encoder)
  269. {
  270. struct audio_convert_info info = {0};
  271. get_audio_info(encoder, &info);
  272. encoder->samplerate = info.samples_per_sec;
  273. encoder->planes = get_audio_planes(info.format, info.speakers);
  274. encoder->blocksize = get_audio_size(info.format, info.speakers, 1);
  275. encoder->framesize = encoder->info.get_frame_size(
  276. encoder->context.data);
  277. encoder->framesize_bytes = encoder->blocksize * encoder->framesize;
  278. reset_audio_buffers(encoder);
  279. }
  280. bool obs_encoder_initialize(obs_encoder_t *encoder)
  281. {
  282. if (!encoder) return false;
  283. if (encoder->active)
  284. return true;
  285. if (encoder->context.data)
  286. encoder->info.destroy(encoder->context.data);
  287. encoder->context.data = encoder->info.create(encoder->context.settings,
  288. encoder);
  289. if (!encoder->context.data)
  290. return false;
  291. encoder->paired_encoder = NULL;
  292. encoder->start_ts = 0;
  293. if (encoder->info.type == OBS_ENCODER_AUDIO)
  294. intitialize_audio_encoder(encoder);
  295. return true;
  296. }
  297. static inline size_t get_callback_idx(
  298. const struct obs_encoder *encoder,
  299. void (*new_packet)(void *param, struct encoder_packet *packet),
  300. void *param)
  301. {
  302. for (size_t i = 0; i < encoder->callbacks.num; i++) {
  303. struct encoder_callback *cb = encoder->callbacks.array+i;
  304. if (cb->new_packet == new_packet && cb->param == param)
  305. return i;
  306. }
  307. return DARRAY_INVALID;
  308. }
  309. void obs_encoder_start(obs_encoder_t *encoder,
  310. void (*new_packet)(void *param, struct encoder_packet *packet),
  311. void *param)
  312. {
  313. struct encoder_callback cb = {false, new_packet, param};
  314. bool first = false;
  315. if (!encoder || !new_packet || !encoder->context.data) return;
  316. pthread_mutex_lock(&encoder->callbacks_mutex);
  317. first = (encoder->callbacks.num == 0);
  318. size_t idx = get_callback_idx(encoder, new_packet, param);
  319. if (idx == DARRAY_INVALID)
  320. da_push_back(encoder->callbacks, &cb);
  321. pthread_mutex_unlock(&encoder->callbacks_mutex);
  322. if (first) {
  323. encoder->cur_pts = 0;
  324. add_connection(encoder);
  325. }
  326. }
  327. void obs_encoder_stop(obs_encoder_t *encoder,
  328. void (*new_packet)(void *param, struct encoder_packet *packet),
  329. void *param)
  330. {
  331. bool last = false;
  332. size_t idx;
  333. if (!encoder) return;
  334. pthread_mutex_lock(&encoder->callbacks_mutex);
  335. idx = get_callback_idx(encoder, new_packet, param);
  336. if (idx != DARRAY_INVALID) {
  337. da_erase(encoder->callbacks, idx);
  338. last = (encoder->callbacks.num == 0);
  339. }
  340. pthread_mutex_unlock(&encoder->callbacks_mutex);
  341. if (last) {
  342. remove_connection(encoder);
  343. if (encoder->destroy_on_stop)
  344. obs_encoder_actually_destroy(encoder);
  345. }
  346. }
  347. const char *obs_encoder_get_codec(const obs_encoder_t *encoder)
  348. {
  349. return encoder ? encoder->info.codec : NULL;
  350. }
  351. const char *obs_get_encoder_codec(const char *id)
  352. {
  353. struct obs_encoder_info *info = find_encoder(id);
  354. return info ? info->codec : NULL;
  355. }
  356. enum obs_encoder_type obs_encoder_get_type(const obs_encoder_t *encoder)
  357. {
  358. return encoder ? encoder->info.type : OBS_ENCODER_AUDIO;
  359. }
  360. enum obs_encoder_type obs_get_encoder_type(const char *id)
  361. {
  362. struct obs_encoder_info *info = find_encoder(id);
  363. return info ? info->type : OBS_ENCODER_AUDIO;
  364. }
  365. void obs_encoder_set_scaled_size(obs_encoder_t *encoder, uint32_t width,
  366. uint32_t height)
  367. {
  368. if (!encoder || encoder->info.type != OBS_ENCODER_VIDEO)
  369. return;
  370. if (encoder->active) {
  371. blog(LOG_WARNING, "encoder '%s': Cannot set the scaled "
  372. "resolution while the encoder is active",
  373. obs_encoder_get_name(encoder));
  374. return;
  375. }
  376. encoder->scaled_width = width;
  377. encoder->scaled_height = height;
  378. }
  379. uint32_t obs_encoder_get_width(const obs_encoder_t *encoder)
  380. {
  381. if (!encoder || !encoder->media ||
  382. encoder->info.type != OBS_ENCODER_VIDEO)
  383. return 0;
  384. return encoder->scaled_width != 0 ?
  385. encoder->scaled_width :
  386. video_output_get_width(encoder->media);
  387. }
  388. uint32_t obs_encoder_get_height(const obs_encoder_t *encoder)
  389. {
  390. if (!encoder || !encoder->media ||
  391. encoder->info.type != OBS_ENCODER_VIDEO)
  392. return 0;
  393. return encoder->scaled_width != 0 ?
  394. encoder->scaled_height :
  395. video_output_get_height(encoder->media);
  396. }
  397. void obs_encoder_set_video(obs_encoder_t *encoder, video_t *video)
  398. {
  399. const struct video_output_info *voi;
  400. if (!video || !encoder || encoder->info.type != OBS_ENCODER_VIDEO)
  401. return;
  402. voi = video_output_get_info(video);
  403. encoder->media = video;
  404. encoder->timebase_num = voi->fps_den;
  405. encoder->timebase_den = voi->fps_num;
  406. }
  407. void obs_encoder_set_audio(obs_encoder_t *encoder, audio_t *audio)
  408. {
  409. if (!audio || !encoder || encoder->info.type != OBS_ENCODER_AUDIO)
  410. return;
  411. encoder->media = audio;
  412. encoder->timebase_num = 1;
  413. encoder->timebase_den = audio_output_get_sample_rate(audio);
  414. }
  415. video_t *obs_encoder_video(const obs_encoder_t *encoder)
  416. {
  417. return (encoder && encoder->info.type == OBS_ENCODER_VIDEO) ?
  418. encoder->media : NULL;
  419. }
  420. audio_t *obs_encoder_audio(const obs_encoder_t *encoder)
  421. {
  422. return (encoder && encoder->info.type == OBS_ENCODER_AUDIO) ?
  423. encoder->media : NULL;
  424. }
  425. bool obs_encoder_active(const obs_encoder_t *encoder)
  426. {
  427. return encoder ? encoder->active : false;
  428. }
  429. static inline bool get_sei(const struct obs_encoder *encoder,
  430. uint8_t **sei, size_t *size)
  431. {
  432. if (encoder->info.get_sei_data)
  433. return encoder->info.get_sei_data(encoder->context.data, sei,
  434. size);
  435. return false;
  436. }
  437. static void send_first_video_packet(struct obs_encoder *encoder,
  438. struct encoder_callback *cb, struct encoder_packet *packet)
  439. {
  440. struct encoder_packet first_packet;
  441. DARRAY(uint8_t) data;
  442. uint8_t *sei;
  443. size_t size;
  444. /* always wait for first keyframe */
  445. if (!packet->keyframe)
  446. return;
  447. da_init(data);
  448. if (!get_sei(encoder, &sei, &size)) {
  449. cb->new_packet(cb->param, packet);
  450. cb->sent_first_packet = true;
  451. return;
  452. }
  453. da_push_back_array(data, sei, size);
  454. da_push_back_array(data, packet->data, packet->size);
  455. first_packet = *packet;
  456. first_packet.data = data.array;
  457. first_packet.size = data.num;
  458. cb->new_packet(cb->param, &first_packet);
  459. cb->sent_first_packet = true;
  460. da_free(data);
  461. }
  462. static inline void send_packet(struct obs_encoder *encoder,
  463. struct encoder_callback *cb, struct encoder_packet *packet)
  464. {
  465. /* include SEI in first video packet */
  466. if (encoder->info.type == OBS_ENCODER_VIDEO && !cb->sent_first_packet)
  467. send_first_video_packet(encoder, cb, packet);
  468. else
  469. cb->new_packet(cb->param, packet);
  470. }
  471. static void full_stop(struct obs_encoder *encoder)
  472. {
  473. if (encoder) {
  474. pthread_mutex_lock(&encoder->callbacks_mutex);
  475. da_free(encoder->callbacks);
  476. remove_connection(encoder);
  477. pthread_mutex_unlock(&encoder->callbacks_mutex);
  478. }
  479. }
  480. static inline void do_encode(struct obs_encoder *encoder,
  481. struct encoder_frame *frame)
  482. {
  483. struct encoder_packet pkt = {0};
  484. bool received = false;
  485. bool success;
  486. pkt.timebase_num = encoder->timebase_num;
  487. pkt.timebase_den = encoder->timebase_den;
  488. pkt.encoder = encoder;
  489. success = encoder->info.encode(encoder->context.data, frame, &pkt,
  490. &received);
  491. if (!success) {
  492. full_stop(encoder);
  493. blog(LOG_ERROR, "Error encoding with encoder '%s'",
  494. encoder->context.name);
  495. return;
  496. }
  497. if (received) {
  498. /* we use system time here to ensure sync with other encoders,
  499. * you do not want to use relative timestamps here */
  500. pkt.dts_usec = encoder->start_ts / 1000 + packet_dts_usec(&pkt);
  501. pthread_mutex_lock(&encoder->callbacks_mutex);
  502. for (size_t i = 0; i < encoder->callbacks.num; i++) {
  503. struct encoder_callback *cb;
  504. cb = encoder->callbacks.array+i;
  505. send_packet(encoder, cb, &pkt);
  506. }
  507. pthread_mutex_unlock(&encoder->callbacks_mutex);
  508. }
  509. }
  510. static void receive_video(void *param, struct video_data *frame)
  511. {
  512. struct obs_encoder *encoder = param;
  513. struct encoder_frame enc_frame;
  514. memset(&enc_frame, 0, sizeof(struct encoder_frame));
  515. for (size_t i = 0; i < MAX_AV_PLANES; i++) {
  516. enc_frame.data[i] = frame->data[i];
  517. enc_frame.linesize[i] = frame->linesize[i];
  518. }
  519. if (!encoder->start_ts)
  520. encoder->start_ts = frame->timestamp;
  521. enc_frame.frames = 1;
  522. enc_frame.pts = encoder->cur_pts;
  523. do_encode(encoder, &enc_frame);
  524. encoder->cur_pts += encoder->timebase_num;
  525. }
  526. static bool buffer_audio(struct obs_encoder *encoder, struct audio_data *data)
  527. {
  528. size_t samplerate = encoder->samplerate;
  529. size_t size = data->frames * encoder->blocksize;
  530. size_t offset_size = 0;
  531. if (!encoder->start_ts && encoder->paired_encoder) {
  532. uint64_t end_ts = data->timestamp;
  533. uint64_t v_start_ts = encoder->paired_encoder->start_ts;
  534. /* no video yet, so don't start audio */
  535. if (!v_start_ts)
  536. return false;
  537. /* audio starting point still not synced with video starting
  538. * point, so don't start audio */
  539. end_ts += (uint64_t)data->frames * 1000000000ULL / samplerate;
  540. if (end_ts <= v_start_ts)
  541. return false;
  542. /* ready to start audio, truncate if necessary */
  543. if (data->timestamp < v_start_ts) {
  544. uint64_t offset = v_start_ts - data->timestamp;
  545. offset = (int)(offset * samplerate / 1000000000);
  546. offset_size = (size_t)offset * encoder->blocksize;
  547. }
  548. encoder->start_ts = v_start_ts;
  549. } else if (!encoder->start_ts && !encoder->paired_encoder) {
  550. encoder->start_ts = data->timestamp;
  551. }
  552. size -= offset_size;
  553. /* push in to the circular buffer */
  554. if (size)
  555. for (size_t i = 0; i < encoder->planes; i++)
  556. circlebuf_push_back(&encoder->audio_input_buffer[i],
  557. data->data[i] + offset_size, size);
  558. return true;
  559. }
  560. static void send_audio_data(struct obs_encoder *encoder)
  561. {
  562. struct encoder_frame enc_frame;
  563. memset(&enc_frame, 0, sizeof(struct encoder_frame));
  564. for (size_t i = 0; i < encoder->planes; i++) {
  565. circlebuf_pop_front(&encoder->audio_input_buffer[i],
  566. encoder->audio_output_buffer[i],
  567. encoder->framesize_bytes);
  568. enc_frame.data[i] = encoder->audio_output_buffer[i];
  569. enc_frame.linesize[i] = (uint32_t)encoder->framesize_bytes;
  570. }
  571. enc_frame.frames = (uint32_t)encoder->framesize;
  572. enc_frame.pts = encoder->cur_pts;
  573. do_encode(encoder, &enc_frame);
  574. encoder->cur_pts += encoder->framesize;
  575. }
  576. static void receive_audio(void *param, size_t mix_idx, struct audio_data *data)
  577. {
  578. struct obs_encoder *encoder = param;
  579. if (!buffer_audio(encoder, data))
  580. return;
  581. while (encoder->audio_input_buffer[0].size >= encoder->framesize_bytes)
  582. send_audio_data(encoder);
  583. UNUSED_PARAMETER(mix_idx);
  584. }
  585. void obs_encoder_add_output(struct obs_encoder *encoder,
  586. struct obs_output *output)
  587. {
  588. if (!encoder) return;
  589. pthread_mutex_lock(&encoder->outputs_mutex);
  590. da_push_back(encoder->outputs, &output);
  591. pthread_mutex_unlock(&encoder->outputs_mutex);
  592. }
  593. void obs_encoder_remove_output(struct obs_encoder *encoder,
  594. struct obs_output *output)
  595. {
  596. if (!encoder) return;
  597. pthread_mutex_lock(&encoder->outputs_mutex);
  598. da_erase_item(encoder->outputs, &output);
  599. pthread_mutex_unlock(&encoder->outputs_mutex);
  600. }
  601. void obs_duplicate_encoder_packet(struct encoder_packet *dst,
  602. const struct encoder_packet *src)
  603. {
  604. *dst = *src;
  605. dst->data = bmemdup(src->data, src->size);
  606. }
  607. void obs_free_encoder_packet(struct encoder_packet *packet)
  608. {
  609. bfree(packet->data);
  610. memset(packet, 0, sizeof(struct encoder_packet));
  611. }
  612. void obs_encoder_set_preferred_video_format(obs_encoder_t *encoder,
  613. enum video_format format)
  614. {
  615. if (!encoder || encoder->info.type != OBS_ENCODER_VIDEO)
  616. return;
  617. encoder->preferred_format = format;
  618. }
  619. enum video_format obs_encoder_get_preferred_video_format(
  620. const obs_encoder_t *encoder)
  621. {
  622. if (!encoder || encoder->info.type != OBS_ENCODER_VIDEO)
  623. return VIDEO_FORMAT_NONE;
  624. return encoder->preferred_format;
  625. }
  626. void obs_encoder_addref(obs_encoder_t *encoder)
  627. {
  628. if (!encoder)
  629. return;
  630. obs_ref_addref(&encoder->control->ref);
  631. }
  632. void obs_encoder_release(obs_encoder_t *encoder)
  633. {
  634. if (!encoder)
  635. return;
  636. obs_weak_encoder_t *control = encoder->control;
  637. if (obs_ref_release(&control->ref)) {
  638. // The order of operations is important here since
  639. // get_context_by_name in obs.c relies on weak refs
  640. // being alive while the context is listed
  641. obs_encoder_destroy(encoder);
  642. obs_weak_encoder_release(control);
  643. }
  644. }
  645. void obs_weak_encoder_addref(obs_weak_encoder_t *weak)
  646. {
  647. if (!weak)
  648. return;
  649. obs_weak_ref_addref(&weak->ref);
  650. }
  651. void obs_weak_encoder_release(obs_weak_encoder_t *weak)
  652. {
  653. if (!weak)
  654. return;
  655. if (obs_weak_ref_release(&weak->ref))
  656. bfree(weak);
  657. }
  658. obs_encoder_t *obs_encoder_get_ref(obs_encoder_t *encoder)
  659. {
  660. if (!encoder)
  661. return NULL;
  662. return obs_weak_encoder_get_encoder(encoder->control);
  663. }
  664. obs_weak_encoder_t *obs_encoder_get_weak_encoder(obs_encoder_t *encoder)
  665. {
  666. if (!encoder)
  667. return NULL;
  668. obs_weak_encoder_t *weak = encoder->control;
  669. obs_weak_encoder_addref(weak);
  670. return weak;
  671. }
  672. obs_encoder_t *obs_weak_encoder_get_encoder(obs_weak_encoder_t *weak)
  673. {
  674. if (!weak)
  675. return NULL;
  676. if (obs_weak_ref_get_ref(&weak->ref))
  677. return weak->encoder;
  678. return NULL;
  679. }
  680. bool obs_weak_encoder_references_encoder(obs_weak_encoder_t *weak,
  681. obs_encoder_t *encoder)
  682. {
  683. return weak && encoder && weak->encoder == encoder;
  684. }