encoder.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808
  1. #include <util/darray.h>
  2. #include <util/dstr.h>
  3. #include <obs-module.h>
  4. #ifndef _WIN32
  5. #include <AudioToolbox/AudioToolbox.h>
  6. #endif
  7. #define CA_LOG(level, format, ...) \
  8. blog(level, "[CoreAudio encoder]: " format, ##__VA_ARGS__)
  9. #define CA_LOG_ENCODER(format_name, encoder, level, format, ...) \
  10. blog(level, "[CoreAudio %s: '%s']: " format, \
  11. format_name, obs_encoder_get_name(encoder), \
  12. ##__VA_ARGS__)
  13. #define CA_BLOG(level, format, ...) \
  14. CA_LOG_ENCODER(ca->format_name, ca->encoder, level, format, \
  15. ##__VA_ARGS__)
  16. #ifdef _WIN32
  17. #include "windows-imports.h"
  18. #endif
  19. struct ca_encoder {
  20. obs_encoder_t *encoder;
  21. const char *format_name;
  22. AudioConverterRef converter;
  23. size_t output_buffer_size;
  24. uint8_t *output_buffer;
  25. size_t out_frames_per_packet;
  26. size_t in_packets;
  27. size_t in_frame_size;
  28. size_t in_bytes_required;
  29. DARRAY(uint8_t) input_buffer;
  30. size_t bytes_read;
  31. uint64_t total_samples;
  32. uint64_t samples_per_second;
  33. uint8_t *extra_data;
  34. uint32_t extra_data_size;
  35. size_t channels;
  36. };
  37. typedef struct ca_encoder ca_encoder;
  38. static const char *aac_get_name(void)
  39. {
  40. return obs_module_text("CoreAudioAAC");
  41. }
  42. static const char *code_to_str(OSStatus code)
  43. {
  44. switch (code) {
  45. #define HANDLE_CODE(c) case c: return #c
  46. HANDLE_CODE(kAudio_UnimplementedError);
  47. HANDLE_CODE(kAudio_FileNotFoundError);
  48. HANDLE_CODE(kAudio_FilePermissionError);
  49. HANDLE_CODE(kAudio_TooManyFilesOpenError);
  50. HANDLE_CODE(kAudio_BadFilePathError);
  51. HANDLE_CODE(kAudio_ParamError);
  52. HANDLE_CODE(kAudio_MemFullError);
  53. HANDLE_CODE(kAudioConverterErr_FormatNotSupported);
  54. HANDLE_CODE(kAudioConverterErr_OperationNotSupported);
  55. HANDLE_CODE(kAudioConverterErr_PropertyNotSupported);
  56. HANDLE_CODE(kAudioConverterErr_InvalidInputSize);
  57. HANDLE_CODE(kAudioConverterErr_InvalidOutputSize);
  58. HANDLE_CODE(kAudioConverterErr_UnspecifiedError);
  59. HANDLE_CODE(kAudioConverterErr_BadPropertySizeError);
  60. HANDLE_CODE(kAudioConverterErr_RequiresPacketDescriptionsError);
  61. HANDLE_CODE(kAudioConverterErr_InputSampleRateOutOfRange);
  62. HANDLE_CODE(kAudioConverterErr_OutputSampleRateOutOfRange);
  63. #undef HANDLE_CODE
  64. default: break;
  65. }
  66. return NULL;
  67. }
  68. static void log_osstatus(ca_encoder *ca, const char *context, OSStatus code)
  69. {
  70. #ifndef _WIN32
  71. CFErrorRef err = CFErrorCreate(kCFAllocatorDefault,
  72. kCFErrorDomainOSStatus, code, NULL);
  73. CFStringRef str = CFErrorCopyDescription(err);
  74. CFIndex length = CFStringGetLength(str);
  75. CFIndex max_size = CFStringGetMaximumSizeForEncoding(length,
  76. kCFStringEncodingUTF8);
  77. char *c_str = malloc(max_size);
  78. if (CFStringGetCString(str, c_str, max_size, kCFStringEncodingUTF8)) {
  79. if (ca)
  80. CA_BLOG(LOG_ERROR, "Error in %s: %s", context, c_str);
  81. else
  82. CA_LOG(LOG_ERROR, "Error in %s: %s", context, c_str);
  83. } else {
  84. #endif
  85. const char *code_str = code_to_str(code);
  86. if (ca)
  87. CA_BLOG(LOG_ERROR, "Error in %s: %s%s%d%s", context,
  88. code_str ? code_str : "",
  89. code_str ? " (" : "",
  90. (int)code,
  91. code_str ? ")" : "");
  92. else
  93. CA_LOG(LOG_ERROR, "Error in %s: %s%s%d%s", context,
  94. code_str ? code_str : "",
  95. code_str ? " (" : "",
  96. (int)code,
  97. code_str ? ")" : "");
  98. #ifndef _WIN32
  99. }
  100. free(c_str);
  101. CFRelease(str);
  102. CFRelease(err);
  103. #endif
  104. }
  105. static void aac_destroy(void *data)
  106. {
  107. ca_encoder *ca = data;
  108. if (ca->converter)
  109. AudioConverterDispose(ca->converter);
  110. da_free(ca->input_buffer);
  111. bfree(ca->extra_data);
  112. bfree(ca->output_buffer);
  113. bfree(ca);
  114. }
  115. typedef void (*bitrate_enumeration_func)(void *data, UInt32 min, UInt32 max);
  116. static bool enumerate_bitrates(ca_encoder *ca, AudioConverterRef converter,
  117. bitrate_enumeration_func enum_func, void *data)
  118. {
  119. if (!converter && ca)
  120. converter = ca->converter;
  121. UInt32 size;
  122. OSStatus code = AudioConverterGetPropertyInfo(converter,
  123. kAudioConverterApplicableEncodeBitRates,
  124. &size, NULL);
  125. if (code) {
  126. log_osstatus(ca, "AudioConverterGetPropertyInfo(bitrates)",
  127. code);
  128. return false;
  129. }
  130. if (!size) {
  131. if (ca)
  132. CA_BLOG(LOG_ERROR, "Query for applicable bitrates "
  133. "returned 0 size");
  134. else
  135. CA_LOG(LOG_ERROR, "Query for applicable bitrates "
  136. "returned 0 size");
  137. return false;
  138. }
  139. AudioValueRange *bitrates = malloc(size);
  140. code = AudioConverterGetProperty(converter,
  141. kAudioConverterApplicableEncodeBitRates,
  142. &size, bitrates);
  143. if (code) {
  144. log_osstatus(ca, "AudioConverterGetProperty(bitrates)", code);
  145. return false;
  146. }
  147. size_t num_bitrates = size / sizeof(AudioValueRange);
  148. for (size_t i = 0; i < num_bitrates; i++)
  149. enum_func(data, (UInt32)bitrates[i].mMinimum,
  150. (UInt32)bitrates[i].mMaximum);
  151. free(bitrates);
  152. return num_bitrates > 0;
  153. }
  154. struct validate_bitrate_helper {
  155. UInt32 bitrate;
  156. bool valid;
  157. };
  158. typedef struct validate_bitrate_helper validate_bitrate_helper;
  159. static void validate_bitrate_func(void *data, UInt32 min, UInt32 max)
  160. {
  161. validate_bitrate_helper *valid = data;
  162. if (valid->bitrate >= min && valid->bitrate <= max)
  163. valid->valid = true;
  164. }
  165. static bool bitrate_valid(ca_encoder *ca, AudioConverterRef converter,
  166. UInt32 bitrate)
  167. {
  168. validate_bitrate_helper helper = {
  169. .bitrate = bitrate,
  170. .valid = false,
  171. };
  172. enumerate_bitrates(ca, converter, validate_bitrate_func, &helper);
  173. return helper.valid;
  174. }
  175. static bool create_encoder(ca_encoder *ca, AudioStreamBasicDescription *in,
  176. AudioStreamBasicDescription *out,
  177. UInt32 format_id, UInt32 bitrate, UInt32 rate_control)
  178. {
  179. #define STATUS_CHECK(c) \
  180. code = c; \
  181. if (code) { \
  182. log_osstatus(ca, #c, code); \
  183. return false; \
  184. }
  185. AudioStreamBasicDescription out_ = {
  186. .mSampleRate = (Float64)ca->samples_per_second,
  187. .mChannelsPerFrame = (UInt32)ca->channels,
  188. .mBytesPerFrame = 0,
  189. .mFramesPerPacket = 0,
  190. .mBitsPerChannel = 0,
  191. .mFormatID = format_id,
  192. .mFormatFlags = 0
  193. };
  194. UInt32 size = sizeof(*out);
  195. OSStatus code;
  196. STATUS_CHECK(AudioFormatGetProperty(kAudioFormatProperty_FormatInfo,
  197. 0, NULL, &size, &out_));
  198. *out = out_;
  199. STATUS_CHECK(AudioConverterNew(in, out, &ca->converter))
  200. STATUS_CHECK(AudioConverterSetProperty(ca->converter,
  201. kAudioCodecPropertyBitRateControlMode,
  202. sizeof(rate_control), &rate_control));
  203. if (!bitrate_valid(ca, NULL, bitrate)) {
  204. CA_BLOG(LOG_ERROR, "Encoder does not support bitrate %u",
  205. (uint32_t)bitrate);
  206. return false;
  207. }
  208. return true;
  209. #undef STATUS_CHECK
  210. }
  211. static void *aac_create(obs_data_t *settings, obs_encoder_t *encoder)
  212. {
  213. #define STATUS_CHECK(c) \
  214. code = c; \
  215. if (code) { \
  216. log_osstatus(ca, #c, code); \
  217. goto free; \
  218. }
  219. UInt32 bitrate = (UInt32)obs_data_get_int(settings, "bitrate") * 1000;
  220. if (!bitrate) {
  221. CA_LOG_ENCODER("AAC", encoder, LOG_ERROR,
  222. "Invalid bitrate specified");
  223. return NULL;
  224. }
  225. const enum audio_format format = AUDIO_FORMAT_FLOAT;
  226. if (is_audio_planar(format)) {
  227. CA_LOG_ENCODER("AAC", encoder, LOG_ERROR,
  228. "Got non-interleaved audio format %d", format);
  229. return NULL;
  230. }
  231. ca_encoder *ca = bzalloc(sizeof(ca_encoder));
  232. ca->encoder = encoder;
  233. ca->format_name = "AAC";
  234. audio_t *audio = obs_encoder_audio(encoder);
  235. const struct audio_output_info *aoi = audio_output_get_info(audio);
  236. ca->channels = audio_output_get_channels(audio);
  237. ca->samples_per_second = audio_output_get_sample_rate(audio);
  238. size_t bytes_per_frame = get_audio_size(format, aoi->speakers, 1);
  239. size_t bits_per_channel = get_audio_bytes_per_channel(format) * 8;
  240. AudioStreamBasicDescription in = {
  241. .mSampleRate = (Float64)ca->samples_per_second,
  242. .mChannelsPerFrame = (UInt32)ca->channels,
  243. .mBytesPerFrame = (UInt32)bytes_per_frame,
  244. .mFramesPerPacket = 1,
  245. .mBytesPerPacket = (UInt32)(1 * bytes_per_frame),
  246. .mBitsPerChannel = (UInt32)bits_per_channel,
  247. .mFormatID = kAudioFormatLinearPCM,
  248. .mFormatFlags = kAudioFormatFlagsNativeEndian |
  249. kAudioFormatFlagIsPacked |
  250. kAudioFormatFlagIsFloat |
  251. 0
  252. };
  253. AudioStreamBasicDescription out;
  254. UInt32 rate_control = kAudioCodecBitRateControlMode_Constant;
  255. if (!create_encoder(ca, &in, &out, kAudioFormatMPEG4AAC, bitrate,
  256. rate_control))
  257. goto free;
  258. OSStatus code;
  259. UInt32 converter_quality = kAudioConverterQuality_Max;
  260. STATUS_CHECK(AudioConverterSetProperty(ca->converter,
  261. kAudioConverterCodecQuality,
  262. sizeof(converter_quality), &converter_quality));
  263. STATUS_CHECK(AudioConverterSetProperty(ca->converter,
  264. kAudioConverterEncodeBitRate,
  265. sizeof(bitrate), &bitrate));
  266. UInt32 size = sizeof(in);
  267. STATUS_CHECK(AudioConverterGetProperty(ca->converter,
  268. kAudioConverterCurrentInputStreamDescription,
  269. &size, &in));
  270. size = sizeof(out);
  271. STATUS_CHECK(AudioConverterGetProperty(ca->converter,
  272. kAudioConverterCurrentOutputStreamDescription,
  273. &size, &out));
  274. ca->in_frame_size = in.mBytesPerFrame;
  275. ca->in_packets = out.mFramesPerPacket / in.mFramesPerPacket;
  276. ca->in_bytes_required = ca->in_packets * ca->in_frame_size;
  277. ca->out_frames_per_packet = out.mFramesPerPacket;
  278. da_init(ca->input_buffer);
  279. ca->output_buffer_size = out.mBytesPerPacket;
  280. if (out.mBytesPerPacket == 0) {
  281. UInt32 max_packet_size = 0;
  282. size = sizeof(max_packet_size);
  283. code = AudioConverterGetProperty(ca->converter,
  284. kAudioConverterPropertyMaximumOutputPacketSize,
  285. &size, &max_packet_size);
  286. if (code) {
  287. log_osstatus(ca, "AudioConverterGetProperty(PacketSz)",
  288. code);
  289. ca->output_buffer_size = 32768;
  290. } else {
  291. ca->output_buffer_size = max_packet_size;
  292. }
  293. }
  294. ca->output_buffer = bmalloc(ca->output_buffer_size);
  295. CA_BLOG(LOG_INFO, "settings:\n"
  296. "\tbitrate: %u\n"
  297. "\tsample rate: %llu\n"
  298. "\tcbr: %s\n"
  299. "\toutput buffer: %lu",
  300. bitrate / 1000, ca->samples_per_second,
  301. rate_control == kAudioCodecBitRateControlMode_Constant ?
  302. "on" : "off",
  303. (unsigned long)ca->output_buffer_size);
  304. return ca;
  305. free:
  306. aac_destroy(ca);
  307. return NULL;
  308. }
  309. static OSStatus complex_input_data_proc(AudioConverterRef inAudioConverter,
  310. UInt32 *ioNumberDataPackets, AudioBufferList *ioData,
  311. AudioStreamPacketDescription **outDataPacketDescription,
  312. void *inUserData)
  313. {
  314. UNUSED_PARAMETER(inAudioConverter);
  315. UNUSED_PARAMETER(outDataPacketDescription);
  316. ca_encoder *ca = inUserData;
  317. if (ca->bytes_read) {
  318. da_erase_range(ca->input_buffer, 0, ca->bytes_read);
  319. ca->bytes_read = 0;
  320. }
  321. if (ca->input_buffer.num < ca->in_bytes_required) {
  322. *ioNumberDataPackets = 0;
  323. ioData->mBuffers[0].mData = NULL;
  324. return 1;
  325. }
  326. *ioNumberDataPackets =
  327. (UInt32)(ca->in_bytes_required / ca->in_frame_size);
  328. ioData->mNumberBuffers = 1;
  329. ioData->mBuffers[0].mData = ca->input_buffer.array;
  330. ioData->mBuffers[0].mNumberChannels = (UInt32)ca->channels;
  331. ioData->mBuffers[0].mDataByteSize = (UInt32)ca->in_bytes_required;
  332. ca->bytes_read += ca->in_packets * ca->in_frame_size;
  333. return 0;
  334. }
  335. static bool aac_encode(void *data, struct encoder_frame *frame,
  336. struct encoder_packet *packet, bool *received_packet)
  337. {
  338. ca_encoder *ca = data;
  339. da_push_back_array(ca->input_buffer, frame->data[0],
  340. frame->linesize[0]);
  341. if ((ca->input_buffer.num - ca->bytes_read) < ca->in_bytes_required)
  342. return true;
  343. UInt32 packets = 1;
  344. AudioBufferList buffer_list = {
  345. .mNumberBuffers = 1,
  346. .mBuffers = { {
  347. .mNumberChannels = (UInt32)ca->channels,
  348. .mDataByteSize = (UInt32)ca->output_buffer_size,
  349. .mData = ca->output_buffer,
  350. } },
  351. };
  352. AudioStreamPacketDescription out_desc = { 0 };
  353. OSStatus code = AudioConverterFillComplexBuffer(ca->converter,
  354. complex_input_data_proc, ca, &packets,
  355. &buffer_list, &out_desc);
  356. if (code && code != 1) {
  357. log_osstatus(ca, "AudioConverterFillComplexBuffer", code);
  358. return false;
  359. }
  360. if (!(*received_packet = packets > 0))
  361. return true;
  362. packet->pts = ca->total_samples;
  363. packet->dts = ca->total_samples;
  364. packet->timebase_num = 1;
  365. packet->timebase_den = (uint32_t)ca->samples_per_second;
  366. packet->type = OBS_ENCODER_AUDIO;
  367. packet->size = out_desc.mDataByteSize;
  368. packet->data =
  369. (uint8_t*)buffer_list.mBuffers[0].mData + out_desc.mStartOffset;
  370. ca->total_samples += ca->bytes_read / ca->in_frame_size;
  371. return true;
  372. }
  373. static void aac_audio_info(void *data, struct audio_convert_info *info)
  374. {
  375. UNUSED_PARAMETER(data);
  376. info->format = AUDIO_FORMAT_FLOAT;
  377. }
  378. static size_t aac_frame_size(void *data)
  379. {
  380. ca_encoder *ca = data;
  381. return ca->out_frames_per_packet;
  382. }
  383. /* The following code was extracted from encca_aac.c in HandBrake's libhb */
  384. #define MP4ESDescrTag 0x03
  385. #define MP4DecConfigDescrTag 0x04
  386. #define MP4DecSpecificDescrTag 0x05
  387. // based off of mov_mp4_read_descr_len from mov.c in ffmpeg's libavformat
  388. static int read_descr_len(uint8_t **buffer)
  389. {
  390. int len = 0;
  391. int count = 4;
  392. while (count--)
  393. {
  394. int c = *(*buffer)++;
  395. len = (len << 7) | (c & 0x7f);
  396. if (!(c & 0x80))
  397. break;
  398. }
  399. return len;
  400. }
  401. // based off of mov_mp4_read_descr from mov.c in ffmpeg's libavformat
  402. static int read_descr(uint8_t **buffer, int *tag)
  403. {
  404. *tag = *(*buffer)++;
  405. return read_descr_len(buffer);
  406. }
  407. // based off of mov_read_esds from mov.c in ffmpeg's libavformat
  408. static void read_esds_desc_ext(uint8_t* desc_ext, uint8_t **buffer,
  409. uint32_t *size, bool version_flags)
  410. {
  411. uint8_t *esds = desc_ext;
  412. int tag, len;
  413. *size = 0;
  414. if (version_flags)
  415. esds += 4; // version + flags
  416. read_descr(&esds, &tag);
  417. esds += 2; // ID
  418. if (tag == MP4ESDescrTag)
  419. esds++; // priority
  420. read_descr(&esds, &tag);
  421. if (tag == MP4DecConfigDescrTag) {
  422. esds++; // object type id
  423. esds++; // stream type
  424. esds += 3; // buffer size db
  425. esds += 4; // max bitrate
  426. esds += 4; // average bitrate
  427. len = read_descr(&esds, &tag);
  428. if (tag == MP4DecSpecificDescrTag) {
  429. *buffer = bzalloc(len + 8);
  430. if (*buffer) {
  431. memcpy(*buffer, esds, len);
  432. *size = len;
  433. }
  434. }
  435. }
  436. }
  437. /* extracted code ends here */
  438. static void query_extra_data(ca_encoder *ca)
  439. {
  440. UInt32 size = 0;
  441. OSStatus code;
  442. code = AudioConverterGetPropertyInfo(ca->converter,
  443. kAudioConverterCompressionMagicCookie,
  444. &size, NULL);
  445. if (code) {
  446. log_osstatus(ca, "AudioConverterGetPropertyInfo(magic_cookie)",
  447. code);
  448. return;
  449. }
  450. if (!size) {
  451. CA_BLOG(LOG_WARNING, "Got 0 data size info for magic_cookie");
  452. return;
  453. }
  454. uint8_t *extra_data = malloc(size);
  455. code = AudioConverterGetProperty(ca->converter,
  456. kAudioConverterCompressionMagicCookie,
  457. &size, extra_data);
  458. if (code) {
  459. log_osstatus(ca, "AudioConverterGetProperty(magic_cookie)",
  460. code);
  461. goto free;
  462. }
  463. if (!size) {
  464. CA_BLOG(LOG_WARNING, "Got 0 data size for magic_cookie");
  465. goto free;
  466. }
  467. read_esds_desc_ext(extra_data, &ca->extra_data, &ca->extra_data_size,
  468. false);
  469. free:
  470. free(extra_data);
  471. }
  472. static bool aac_extra_data(void *data, uint8_t **extra_data, size_t *size)
  473. {
  474. ca_encoder *ca = data;
  475. if (!ca->extra_data)
  476. query_extra_data(ca);
  477. if (!ca->extra_data_size)
  478. return false;
  479. *extra_data = ca->extra_data;
  480. *size = ca->extra_data_size;
  481. return true;
  482. }
  483. static AudioConverterRef aac_default_converter(void)
  484. {
  485. UInt32 bytes_per_frame = 8;
  486. UInt32 channels = 2;
  487. UInt32 bits_per_channel = bytes_per_frame / channels * 8;
  488. AudioStreamBasicDescription in = {
  489. .mSampleRate = 44100,
  490. .mChannelsPerFrame = channels,
  491. .mBytesPerFrame = bytes_per_frame,
  492. .mFramesPerPacket = 1,
  493. .mBytesPerPacket = 1 * bytes_per_frame,
  494. .mBitsPerChannel = bits_per_channel,
  495. .mFormatID = kAudioFormatLinearPCM,
  496. .mFormatFlags = kAudioFormatFlagsNativeEndian |
  497. kAudioFormatFlagIsPacked |
  498. kAudioFormatFlagIsFloat |
  499. 0
  500. };
  501. AudioStreamBasicDescription out = {
  502. .mSampleRate = 44100,
  503. .mChannelsPerFrame = channels,
  504. .mBytesPerFrame = 0,
  505. .mFramesPerPacket = 0,
  506. .mBitsPerChannel = 0,
  507. .mFormatID = kAudioFormatMPEG4AAC,
  508. .mFormatFlags = 0
  509. };
  510. UInt32 size = sizeof(out);
  511. OSStatus code = AudioFormatGetProperty(kAudioFormatProperty_FormatInfo,
  512. 0, NULL, &size, &out);
  513. if (code) {
  514. log_osstatus(NULL, "AudioFormatGetProperty(format_info)", code);
  515. return NULL;
  516. }
  517. AudioConverterRef converter;
  518. code = AudioConverterNew(&in, &out, &converter);
  519. if (code) {
  520. log_osstatus(NULL, "AudioConverterNew", code);
  521. return NULL;
  522. }
  523. return converter;
  524. }
  525. struct find_matching_bitrate_helper {
  526. UInt32 bitrate;
  527. UInt32 best_match;
  528. int diff;
  529. };
  530. typedef struct find_matching_bitrate_helper find_matching_bitrate_helper;
  531. static void find_matching_bitrate_func(void *data, UInt32 min, UInt32 max)
  532. {
  533. find_matching_bitrate_helper *helper = data;
  534. int min_diff = abs((int)helper->bitrate - (int)min);
  535. int max_diff = abs((int)helper->bitrate - (int)max);
  536. if (min_diff < helper->diff) {
  537. helper->best_match = min;
  538. helper->diff = min_diff;
  539. }
  540. if (max_diff < helper->diff) {
  541. helper->best_match = max;
  542. helper->diff = max_diff;
  543. }
  544. }
  545. static UInt32 find_matching_bitrate(UInt32 bitrate)
  546. {
  547. find_matching_bitrate_helper helper = {
  548. .bitrate = bitrate * 1000,
  549. .best_match = 0,
  550. .diff = INT_MAX,
  551. };
  552. AudioConverterRef converter = aac_default_converter();
  553. if (!converter) {
  554. CA_LOG(LOG_ERROR, "Could not get converter to match "
  555. "default bitrate");
  556. return bitrate;
  557. }
  558. bool has_bitrates = enumerate_bitrates(NULL, converter,
  559. find_matching_bitrate_func, &helper);
  560. AudioConverterDispose(converter);
  561. if (!has_bitrates) {
  562. CA_LOG(LOG_ERROR, "No bitrates found while matching "
  563. "default bitrate");
  564. AudioConverterDispose(converter);
  565. return bitrate;
  566. }
  567. if (helper.best_match != helper.bitrate)
  568. CA_LOG(LOG_INFO, "Returning closest matching bitrate %u "
  569. "instead of requested bitrate %u",
  570. (uint32_t)helper.best_match / 1000,
  571. (uint32_t)bitrate);
  572. return helper.best_match / 1000;
  573. }
  574. static void aac_defaults(obs_data_t *settings)
  575. {
  576. obs_data_set_default_int(settings, "bitrate",
  577. find_matching_bitrate(128));
  578. }
  579. struct add_bitrates_helper {
  580. DARRAY(UInt32) bitrates;
  581. };
  582. typedef struct add_bitrates_helper add_bitrates_helper;
  583. static void add_bitrates_func(void *data, UInt32 min, UInt32 max)
  584. {
  585. add_bitrates_helper *helper = data;
  586. if (da_find(helper->bitrates, &min, 0) == DARRAY_INVALID)
  587. da_push_back(helper->bitrates, &min);
  588. if (da_find(helper->bitrates, &max, 0) == DARRAY_INVALID)
  589. da_push_back(helper->bitrates, &max);
  590. }
  591. static int bitrate_compare(const void *data1, const void *data2)
  592. {
  593. const UInt32 *bitrate1 = data1;
  594. const UInt32 *bitrate2 = data2;
  595. return (int)*bitrate1 - (int)*bitrate2;
  596. }
  597. static void add_bitrates(obs_property_t *prop, ca_encoder *ca)
  598. {
  599. add_bitrates_helper helper = { 0 };
  600. if (!enumerate_bitrates(ca, ca ? NULL : aac_default_converter(),
  601. add_bitrates_func, &helper))
  602. return;
  603. qsort(helper.bitrates.array, helper.bitrates.num, sizeof(UInt32),
  604. bitrate_compare);
  605. struct dstr str = { 0 };
  606. for (size_t i = 0; i < helper.bitrates.num; i++) {
  607. dstr_printf(&str, "%u",
  608. (uint32_t)helper.bitrates.array[i]/1000);
  609. obs_property_list_add_int(prop, str.array,
  610. helper.bitrates.array[i]/1000);
  611. }
  612. dstr_free(&str);
  613. }
  614. static obs_properties_t *aac_properties(void *data)
  615. {
  616. ca_encoder *ca = data;
  617. obs_properties_t *props = obs_properties_create();
  618. obs_property_t *p = obs_properties_add_list(props, "bitrate",
  619. obs_module_text("Bitrate"),
  620. OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_INT);
  621. add_bitrates(p, ca);
  622. return props;
  623. }
  624. static struct obs_encoder_info aac_info = {
  625. .id = "CoreAudio_AAC",
  626. .type = OBS_ENCODER_AUDIO,
  627. .codec = "AAC",
  628. .get_name = aac_get_name,
  629. .destroy = aac_destroy,
  630. .create = aac_create,
  631. .encode = aac_encode,
  632. .get_frame_size = aac_frame_size,
  633. .get_audio_info = aac_audio_info,
  634. .get_extra_data = aac_extra_data,
  635. .get_defaults = aac_defaults,
  636. };
  637. OBS_DECLARE_MODULE()
  638. OBS_MODULE_USE_DEFAULT_LOCALE("coreaudio-encoder", "en-US")
  639. bool obs_module_load(void)
  640. {
  641. #ifdef _WIN32
  642. if (!load_core_audio()) {
  643. CA_LOG(LOG_WARNING, "Couldn't load CoreAudio AAC encoder");
  644. return true;
  645. }
  646. CA_LOG(LOG_INFO, "Adding CoreAudio AAC encoder");
  647. #endif
  648. obs_register_encoder(&aac_info);
  649. return true;
  650. }
  651. #ifdef _WIN32
  652. void obs_module_unload(void)
  653. {
  654. unload_core_audio();
  655. }
  656. #endif