mac-audio.c 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844
  1. #include <AudioUnit/AudioUnit.h>
  2. #include <CoreFoundation/CFString.h>
  3. #include <CoreAudio/CoreAudio.h>
  4. #include <unistd.h>
  5. #include <errno.h>
  6. #include <obs-module.h>
  7. #include <mach/mach_time.h>
  8. #include <util/threading.h>
  9. #include <util/c99defs.h>
  10. #include <util/apple/cfstring-utils.h>
  11. #include "audio-device-enum.h"
  12. #define PROPERTY_DEFAULT_DEVICE kAudioHardwarePropertyDefaultInputDevice
  13. #define PROPERTY_FORMATS kAudioStreamPropertyAvailablePhysicalFormats
  14. #define SCOPE_OUTPUT kAudioUnitScope_Output
  15. #define SCOPE_INPUT kAudioUnitScope_Input
  16. #define SCOPE_GLOBAL kAudioUnitScope_Global
  17. #define BUS_OUTPUT 0
  18. #define BUS_INPUT 1
  19. #define set_property AudioUnitSetProperty
  20. #define get_property AudioUnitGetProperty
  21. #define TEXT_AUDIO_INPUT obs_module_text("CoreAudio.InputCapture");
  22. #define TEXT_AUDIO_OUTPUT obs_module_text("CoreAudio.OutputCapture");
  23. #define TEXT_DEVICE obs_module_text("CoreAudio.Device")
  24. #define TEXT_DEVICE_DEFAULT obs_module_text("CoreAudio.Device.Default")
  25. struct coreaudio_data {
  26. char *device_name;
  27. char *device_uid;
  28. AudioUnit unit;
  29. AudioDeviceID device_id;
  30. AudioBufferList *buf_list;
  31. bool au_initialized;
  32. bool active;
  33. bool default_device;
  34. bool input;
  35. bool no_devices;
  36. uint32_t sample_rate;
  37. enum audio_format format;
  38. enum speaker_layout speakers;
  39. pthread_t reconnect_thread;
  40. os_event_t *exit_event;
  41. volatile bool reconnecting;
  42. unsigned long retry_time;
  43. obs_source_t *source;
  44. };
  45. static bool get_default_output_device(struct coreaudio_data *ca)
  46. {
  47. struct device_list list;
  48. memset(&list, 0, sizeof(struct device_list));
  49. coreaudio_enum_devices(&list, false);
  50. if (!list.items.num)
  51. return false;
  52. bfree(ca->device_uid);
  53. ca->device_uid = bstrdup(list.items.array[0].value.array);
  54. device_list_free(&list);
  55. return true;
  56. }
  57. static bool find_device_id_by_uid(struct coreaudio_data *ca)
  58. {
  59. UInt32 size = sizeof(AudioDeviceID);
  60. CFStringRef cf_uid = NULL;
  61. CFStringRef qual = NULL;
  62. UInt32 qual_size = 0;
  63. OSStatus stat;
  64. bool success;
  65. AudioObjectPropertyAddress addr = {
  66. .mScope = kAudioObjectPropertyScopeGlobal,
  67. .mElement = kAudioObjectPropertyElementMaster};
  68. if (!ca->device_uid)
  69. ca->device_uid = bstrdup("default");
  70. ca->default_device = false;
  71. ca->no_devices = false;
  72. /* have to do this because mac output devices don't actually exist */
  73. if (astrcmpi(ca->device_uid, "default") == 0) {
  74. if (ca->input) {
  75. ca->default_device = true;
  76. } else {
  77. if (!get_default_output_device(ca)) {
  78. ca->no_devices = true;
  79. return false;
  80. }
  81. }
  82. }
  83. cf_uid = CFStringCreateWithCString(NULL, ca->device_uid,
  84. kCFStringEncodingUTF8);
  85. if (ca->default_device) {
  86. addr.mSelector = PROPERTY_DEFAULT_DEVICE;
  87. stat = AudioObjectGetPropertyData(kAudioObjectSystemObject,
  88. &addr, qual_size, &qual,
  89. &size, &ca->device_id);
  90. success = (stat == noErr);
  91. } else {
  92. success = coreaudio_get_device_id(cf_uid, &ca->device_id);
  93. }
  94. if (cf_uid)
  95. CFRelease(cf_uid);
  96. return success;
  97. }
  98. static inline void ca_warn(struct coreaudio_data *ca, const char *func,
  99. const char *format, ...)
  100. {
  101. va_list args;
  102. struct dstr str = {0};
  103. va_start(args, format);
  104. dstr_printf(&str, "[%s]:[device '%s'] ", func, ca->device_name);
  105. dstr_vcatf(&str, format, args);
  106. blog(LOG_WARNING, "%s", str.array);
  107. dstr_free(&str);
  108. va_end(args);
  109. }
  110. static inline bool ca_success(OSStatus stat, struct coreaudio_data *ca,
  111. const char *func, const char *action)
  112. {
  113. if (stat != noErr) {
  114. blog(LOG_WARNING, "[%s]:[device '%s'] %s failed: %d", func,
  115. ca->device_name, action, (int)stat);
  116. return false;
  117. }
  118. return true;
  119. }
  120. enum coreaudio_io_type {
  121. IO_TYPE_INPUT,
  122. IO_TYPE_OUTPUT,
  123. };
  124. static inline bool enable_io(struct coreaudio_data *ca,
  125. enum coreaudio_io_type type, bool enable)
  126. {
  127. UInt32 enable_int = enable;
  128. return set_property(ca->unit, kAudioOutputUnitProperty_EnableIO,
  129. (type == IO_TYPE_INPUT) ? SCOPE_INPUT
  130. : SCOPE_OUTPUT,
  131. (type == IO_TYPE_INPUT) ? BUS_INPUT : BUS_OUTPUT,
  132. &enable_int, sizeof(enable_int));
  133. }
  134. static inline enum audio_format convert_ca_format(UInt32 format_flags,
  135. UInt32 bits)
  136. {
  137. bool planar = (format_flags & kAudioFormatFlagIsNonInterleaved) != 0;
  138. if (format_flags & kAudioFormatFlagIsFloat)
  139. return planar ? AUDIO_FORMAT_FLOAT_PLANAR : AUDIO_FORMAT_FLOAT;
  140. if (!(format_flags & kAudioFormatFlagIsSignedInteger) && bits == 8)
  141. return planar ? AUDIO_FORMAT_U8BIT_PLANAR : AUDIO_FORMAT_U8BIT;
  142. /* not float? not signed int? no clue, fail */
  143. if ((format_flags & kAudioFormatFlagIsSignedInteger) == 0)
  144. return AUDIO_FORMAT_UNKNOWN;
  145. if (bits == 16)
  146. return planar ? AUDIO_FORMAT_16BIT_PLANAR : AUDIO_FORMAT_16BIT;
  147. else if (bits == 32)
  148. return planar ? AUDIO_FORMAT_32BIT_PLANAR : AUDIO_FORMAT_32BIT;
  149. return AUDIO_FORMAT_UNKNOWN;
  150. }
  151. static inline enum speaker_layout convert_ca_speaker_layout(UInt32 channels)
  152. {
  153. switch (channels) {
  154. case 1:
  155. return SPEAKERS_MONO;
  156. case 2:
  157. return SPEAKERS_STEREO;
  158. case 3:
  159. return SPEAKERS_2POINT1;
  160. case 4:
  161. return SPEAKERS_4POINT0;
  162. case 5:
  163. return SPEAKERS_4POINT1;
  164. case 6:
  165. return SPEAKERS_5POINT1;
  166. case 8:
  167. return SPEAKERS_7POINT1;
  168. }
  169. return SPEAKERS_UNKNOWN;
  170. }
  171. static bool coreaudio_init_format(struct coreaudio_data *ca)
  172. {
  173. AudioStreamBasicDescription desc;
  174. OSStatus stat;
  175. UInt32 size = sizeof(desc);
  176. struct obs_audio_info aoi;
  177. int channels;
  178. if (!obs_get_audio_info(&aoi)) {
  179. blog(LOG_WARNING, "No active audio");
  180. return false;
  181. }
  182. channels = get_audio_channels(aoi.speakers);
  183. stat = get_property(ca->unit, kAudioUnitProperty_StreamFormat,
  184. SCOPE_INPUT, BUS_INPUT, &desc, &size);
  185. if (!ca_success(stat, ca, "coreaudio_init_format", "get input format"))
  186. return false;
  187. /* Certain types of devices have no limit on channel count, and
  188. * there's no way to know the actual number of channels it's using,
  189. * so if we encounter this situation just force to what is defined in output */
  190. if (desc.mChannelsPerFrame > 8) {
  191. desc.mChannelsPerFrame = channels;
  192. desc.mBytesPerFrame = channels * desc.mBitsPerChannel / 8;
  193. desc.mBytesPerPacket =
  194. desc.mFramesPerPacket * desc.mBytesPerFrame;
  195. }
  196. stat = set_property(ca->unit, kAudioUnitProperty_StreamFormat,
  197. SCOPE_OUTPUT, BUS_INPUT, &desc, size);
  198. if (!ca_success(stat, ca, "coreaudio_init_format", "set output format"))
  199. return false;
  200. if (desc.mFormatID != kAudioFormatLinearPCM) {
  201. ca_warn(ca, "coreaudio_init_format", "format is not PCM");
  202. return false;
  203. }
  204. ca->format = convert_ca_format(desc.mFormatFlags, desc.mBitsPerChannel);
  205. if (ca->format == AUDIO_FORMAT_UNKNOWN) {
  206. ca_warn(ca, "coreaudio_init_format",
  207. "unknown format flags: "
  208. "%u, bits: %u",
  209. (unsigned int)desc.mFormatFlags,
  210. (unsigned int)desc.mBitsPerChannel);
  211. return false;
  212. }
  213. ca->sample_rate = (uint32_t)desc.mSampleRate;
  214. ca->speakers = convert_ca_speaker_layout(desc.mChannelsPerFrame);
  215. if (ca->speakers == SPEAKERS_UNKNOWN) {
  216. ca_warn(ca, "coreaudio_init_format",
  217. "unknown speaker layout: "
  218. "%u channels",
  219. (unsigned int)desc.mChannelsPerFrame);
  220. return false;
  221. }
  222. return true;
  223. }
  224. static bool coreaudio_init_buffer(struct coreaudio_data *ca)
  225. {
  226. UInt32 buf_size = 0;
  227. UInt32 size = 0;
  228. UInt32 frames = 0;
  229. OSStatus stat;
  230. AudioObjectPropertyAddress addr = {
  231. kAudioDevicePropertyStreamConfiguration,
  232. kAudioDevicePropertyScopeInput,
  233. kAudioObjectPropertyElementMaster};
  234. stat = AudioObjectGetPropertyDataSize(ca->device_id, &addr, 0, NULL,
  235. &buf_size);
  236. if (!ca_success(stat, ca, "coreaudio_init_buffer", "get list size"))
  237. return false;
  238. size = sizeof(frames);
  239. stat = get_property(ca->unit, kAudioDevicePropertyBufferFrameSize,
  240. SCOPE_GLOBAL, 0, &frames, &size);
  241. if (!ca_success(stat, ca, "coreaudio_init_buffer", "get frame size"))
  242. return false;
  243. /* ---------------------- */
  244. ca->buf_list = bmalloc(buf_size);
  245. stat = AudioObjectGetPropertyData(ca->device_id, &addr, 0, NULL,
  246. &buf_size, ca->buf_list);
  247. if (!ca_success(stat, ca, "coreaudio_init_buffer", "allocate")) {
  248. bfree(ca->buf_list);
  249. ca->buf_list = NULL;
  250. return false;
  251. }
  252. for (UInt32 i = 0; i < ca->buf_list->mNumberBuffers; i++) {
  253. size = ca->buf_list->mBuffers[i].mDataByteSize;
  254. ca->buf_list->mBuffers[i].mData = bmalloc(size);
  255. }
  256. return true;
  257. }
  258. static void buf_list_free(AudioBufferList *buf_list)
  259. {
  260. if (buf_list) {
  261. for (UInt32 i = 0; i < buf_list->mNumberBuffers; i++)
  262. bfree(buf_list->mBuffers[i].mData);
  263. bfree(buf_list);
  264. }
  265. }
  266. static OSStatus input_callback(void *data,
  267. AudioUnitRenderActionFlags *action_flags,
  268. const AudioTimeStamp *ts_data, UInt32 bus_num,
  269. UInt32 frames, AudioBufferList *ignored_buffers)
  270. {
  271. struct coreaudio_data *ca = data;
  272. OSStatus stat;
  273. struct obs_source_audio audio;
  274. stat = AudioUnitRender(ca->unit, action_flags, ts_data, bus_num, frames,
  275. ca->buf_list);
  276. if (!ca_success(stat, ca, "input_callback", "audio retrieval"))
  277. return noErr;
  278. for (UInt32 i = 0; i < ca->buf_list->mNumberBuffers; i++)
  279. audio.data[i] = ca->buf_list->mBuffers[i].mData;
  280. audio.frames = frames;
  281. audio.speakers = ca->speakers;
  282. audio.format = ca->format;
  283. audio.samples_per_sec = ca->sample_rate;
  284. static double factor = 0.;
  285. static mach_timebase_info_data_t info = {0, 0};
  286. if (info.numer == 0 && info.denom == 0) {
  287. mach_timebase_info(&info);
  288. factor = ((double)info.numer) / info.denom;
  289. }
  290. if (info.numer != info.denom)
  291. audio.timestamp = (uint64_t)(factor * ts_data->mHostTime);
  292. else
  293. audio.timestamp = ts_data->mHostTime;
  294. obs_source_output_audio(ca->source, &audio);
  295. UNUSED_PARAMETER(ignored_buffers);
  296. return noErr;
  297. }
  298. static void coreaudio_stop(struct coreaudio_data *ca);
  299. static bool coreaudio_init(struct coreaudio_data *ca);
  300. static void coreaudio_uninit(struct coreaudio_data *ca);
  301. static void *reconnect_thread(void *param)
  302. {
  303. struct coreaudio_data *ca = param;
  304. ca->reconnecting = true;
  305. while (os_event_timedwait(ca->exit_event, ca->retry_time) ==
  306. ETIMEDOUT) {
  307. if (coreaudio_init(ca))
  308. break;
  309. }
  310. blog(LOG_DEBUG, "coreaudio: exit the reconnect thread");
  311. ca->reconnecting = false;
  312. return NULL;
  313. }
  314. static void coreaudio_begin_reconnect(struct coreaudio_data *ca)
  315. {
  316. int ret;
  317. if (ca->reconnecting)
  318. return;
  319. ret = pthread_create(&ca->reconnect_thread, NULL, reconnect_thread, ca);
  320. if (ret != 0)
  321. blog(LOG_WARNING,
  322. "[coreaudio_begin_reconnect] failed to "
  323. "create thread, error code: %d",
  324. ret);
  325. }
  326. static OSStatus
  327. notification_callback(AudioObjectID id, UInt32 num_addresses,
  328. const AudioObjectPropertyAddress addresses[], void *data)
  329. {
  330. struct coreaudio_data *ca = data;
  331. coreaudio_stop(ca);
  332. coreaudio_uninit(ca);
  333. if (addresses[0].mSelector == PROPERTY_DEFAULT_DEVICE)
  334. ca->retry_time = 300;
  335. else
  336. ca->retry_time = 2000;
  337. blog(LOG_INFO,
  338. "coreaudio: device '%s' disconnected or changed. "
  339. "attempting to reconnect",
  340. ca->device_name);
  341. coreaudio_begin_reconnect(ca);
  342. UNUSED_PARAMETER(id);
  343. UNUSED_PARAMETER(num_addresses);
  344. return noErr;
  345. }
  346. static OSStatus add_listener(struct coreaudio_data *ca, UInt32 property)
  347. {
  348. AudioObjectPropertyAddress addr = {property,
  349. kAudioObjectPropertyScopeGlobal,
  350. kAudioObjectPropertyElementMaster};
  351. return AudioObjectAddPropertyListener(ca->device_id, &addr,
  352. notification_callback, ca);
  353. }
  354. static bool coreaudio_init_hooks(struct coreaudio_data *ca)
  355. {
  356. OSStatus stat;
  357. AURenderCallbackStruct callback_info = {.inputProc = input_callback,
  358. .inputProcRefCon = ca};
  359. stat = add_listener(ca, kAudioDevicePropertyDeviceIsAlive);
  360. if (!ca_success(stat, ca, "coreaudio_init_hooks",
  361. "set disconnect callback"))
  362. return false;
  363. stat = add_listener(ca, PROPERTY_FORMATS);
  364. if (!ca_success(stat, ca, "coreaudio_init_hooks",
  365. "set format change callback"))
  366. return false;
  367. if (ca->default_device) {
  368. AudioObjectPropertyAddress addr = {
  369. PROPERTY_DEFAULT_DEVICE,
  370. kAudioObjectPropertyScopeGlobal,
  371. kAudioObjectPropertyElementMaster};
  372. stat = AudioObjectAddPropertyListener(kAudioObjectSystemObject,
  373. &addr,
  374. notification_callback,
  375. ca);
  376. if (!ca_success(stat, ca, "coreaudio_init_hooks",
  377. "set device change callback"))
  378. return false;
  379. }
  380. stat = set_property(ca->unit, kAudioOutputUnitProperty_SetInputCallback,
  381. SCOPE_GLOBAL, 0, &callback_info,
  382. sizeof(callback_info));
  383. if (!ca_success(stat, ca, "coreaudio_init_hooks", "set input callback"))
  384. return false;
  385. return true;
  386. }
  387. static void coreaudio_remove_hooks(struct coreaudio_data *ca)
  388. {
  389. AURenderCallbackStruct callback_info = {.inputProc = NULL,
  390. .inputProcRefCon = NULL};
  391. AudioObjectPropertyAddress addr = {kAudioDevicePropertyDeviceIsAlive,
  392. kAudioObjectPropertyScopeGlobal,
  393. kAudioObjectPropertyElementMaster};
  394. AudioObjectRemovePropertyListener(ca->device_id, &addr,
  395. notification_callback, ca);
  396. addr.mSelector = PROPERTY_FORMATS;
  397. AudioObjectRemovePropertyListener(ca->device_id, &addr,
  398. notification_callback, ca);
  399. if (ca->default_device) {
  400. addr.mSelector = PROPERTY_DEFAULT_DEVICE;
  401. AudioObjectRemovePropertyListener(kAudioObjectSystemObject,
  402. &addr, notification_callback,
  403. ca);
  404. }
  405. set_property(ca->unit, kAudioOutputUnitProperty_SetInputCallback,
  406. SCOPE_GLOBAL, 0, &callback_info, sizeof(callback_info));
  407. }
  408. static bool coreaudio_get_device_name(struct coreaudio_data *ca)
  409. {
  410. CFStringRef cf_name = NULL;
  411. UInt32 size = sizeof(CFStringRef);
  412. char *name = NULL;
  413. const AudioObjectPropertyAddress addr = {
  414. kAudioDevicePropertyDeviceNameCFString,
  415. kAudioObjectPropertyScopeInput,
  416. kAudioObjectPropertyElementMaster};
  417. OSStatus stat = AudioObjectGetPropertyData(ca->device_id, &addr, 0,
  418. NULL, &size, &cf_name);
  419. if (stat != noErr) {
  420. blog(LOG_WARNING,
  421. "[coreaudio_get_device_name] failed to "
  422. "get name: %d",
  423. (int)stat);
  424. return false;
  425. }
  426. name = cfstr_copy_cstr(cf_name, kCFStringEncodingUTF8);
  427. if (!name) {
  428. blog(LOG_WARNING, "[coreaudio_get_device_name] failed to "
  429. "convert name to cstr for some reason");
  430. return false;
  431. }
  432. bfree(ca->device_name);
  433. ca->device_name = name;
  434. if (cf_name)
  435. CFRelease(cf_name);
  436. return true;
  437. }
  438. static bool coreaudio_start(struct coreaudio_data *ca)
  439. {
  440. OSStatus stat;
  441. if (ca->active)
  442. return true;
  443. stat = AudioOutputUnitStart(ca->unit);
  444. return ca_success(stat, ca, "coreaudio_start", "start audio");
  445. }
  446. static void coreaudio_stop(struct coreaudio_data *ca)
  447. {
  448. OSStatus stat;
  449. if (!ca->active)
  450. return;
  451. ca->active = false;
  452. stat = AudioOutputUnitStop(ca->unit);
  453. ca_success(stat, ca, "coreaudio_stop", "stop audio");
  454. }
  455. static bool coreaudio_init_unit(struct coreaudio_data *ca)
  456. {
  457. AudioComponentDescription desc = {
  458. .componentType = kAudioUnitType_Output,
  459. .componentSubType = kAudioUnitSubType_HALOutput};
  460. AudioComponent component = AudioComponentFindNext(NULL, &desc);
  461. if (!component) {
  462. ca_warn(ca, "coreaudio_init_unit", "find component failed");
  463. return false;
  464. }
  465. OSStatus stat = AudioComponentInstanceNew(component, &ca->unit);
  466. if (!ca_success(stat, ca, "coreaudio_init_unit", "instance unit"))
  467. return false;
  468. ca->au_initialized = true;
  469. return true;
  470. }
  471. static bool coreaudio_init(struct coreaudio_data *ca)
  472. {
  473. OSStatus stat;
  474. if (ca->au_initialized)
  475. return true;
  476. if (!find_device_id_by_uid(ca))
  477. return false;
  478. if (!coreaudio_get_device_name(ca))
  479. return false;
  480. if (!coreaudio_init_unit(ca))
  481. return false;
  482. stat = enable_io(ca, IO_TYPE_INPUT, true);
  483. if (!ca_success(stat, ca, "coreaudio_init", "enable input io"))
  484. goto fail;
  485. stat = enable_io(ca, IO_TYPE_OUTPUT, false);
  486. if (!ca_success(stat, ca, "coreaudio_init", "disable output io"))
  487. goto fail;
  488. stat = set_property(ca->unit, kAudioOutputUnitProperty_CurrentDevice,
  489. SCOPE_GLOBAL, 0, &ca->device_id,
  490. sizeof(ca->device_id));
  491. if (!ca_success(stat, ca, "coreaudio_init", "set current device"))
  492. goto fail;
  493. if (!coreaudio_init_format(ca))
  494. goto fail;
  495. if (!coreaudio_init_buffer(ca))
  496. goto fail;
  497. if (!coreaudio_init_hooks(ca))
  498. goto fail;
  499. stat = AudioUnitInitialize(ca->unit);
  500. if (!ca_success(stat, ca, "coreaudio_initialize", "initialize"))
  501. goto fail;
  502. if (!coreaudio_start(ca))
  503. goto fail;
  504. blog(LOG_INFO, "coreaudio: Device '%s' [%" PRIu32 " Hz] initialized",
  505. ca->device_name, ca->sample_rate);
  506. return ca->au_initialized;
  507. fail:
  508. coreaudio_uninit(ca);
  509. return false;
  510. }
  511. static void coreaudio_try_init(struct coreaudio_data *ca)
  512. {
  513. if (!coreaudio_init(ca)) {
  514. blog(LOG_INFO,
  515. "coreaudio: failed to find device "
  516. "uid: %s, waiting for connection",
  517. ca->device_uid);
  518. ca->retry_time = 2000;
  519. if (ca->no_devices)
  520. blog(LOG_INFO, "coreaudio: no device found");
  521. else
  522. coreaudio_begin_reconnect(ca);
  523. }
  524. }
  525. static void coreaudio_uninit(struct coreaudio_data *ca)
  526. {
  527. if (!ca->au_initialized)
  528. return;
  529. if (ca->unit) {
  530. coreaudio_stop(ca);
  531. OSStatus stat = AudioUnitUninitialize(ca->unit);
  532. ca_success(stat, ca, "coreaudio_uninit", "uninitialize");
  533. coreaudio_remove_hooks(ca);
  534. stat = AudioComponentInstanceDispose(ca->unit);
  535. ca_success(stat, ca, "coreaudio_uninit", "dispose");
  536. ca->unit = NULL;
  537. }
  538. ca->au_initialized = false;
  539. buf_list_free(ca->buf_list);
  540. ca->buf_list = NULL;
  541. }
  542. /* ------------------------------------------------------------------------- */
  543. static const char *coreaudio_input_getname(void *unused)
  544. {
  545. UNUSED_PARAMETER(unused);
  546. return TEXT_AUDIO_INPUT;
  547. }
  548. static const char *coreaudio_output_getname(void *unused)
  549. {
  550. UNUSED_PARAMETER(unused);
  551. return TEXT_AUDIO_OUTPUT;
  552. }
  553. static void coreaudio_shutdown(struct coreaudio_data *ca)
  554. {
  555. if (ca->reconnecting) {
  556. os_event_signal(ca->exit_event);
  557. pthread_join(ca->reconnect_thread, NULL);
  558. os_event_reset(ca->exit_event);
  559. }
  560. coreaudio_uninit(ca);
  561. if (ca->unit)
  562. AudioComponentInstanceDispose(ca->unit);
  563. }
  564. static void coreaudio_destroy(void *data)
  565. {
  566. struct coreaudio_data *ca = data;
  567. if (ca) {
  568. coreaudio_shutdown(ca);
  569. os_event_destroy(ca->exit_event);
  570. bfree(ca->device_name);
  571. bfree(ca->device_uid);
  572. bfree(ca);
  573. }
  574. }
  575. static void coreaudio_update(void *data, obs_data_t *settings)
  576. {
  577. struct coreaudio_data *ca = data;
  578. coreaudio_shutdown(ca);
  579. bfree(ca->device_uid);
  580. ca->device_uid = bstrdup(obs_data_get_string(settings, "device_id"));
  581. coreaudio_try_init(ca);
  582. }
  583. static void coreaudio_defaults(obs_data_t *settings)
  584. {
  585. obs_data_set_default_string(settings, "device_id", "default");
  586. }
  587. static void *coreaudio_create(obs_data_t *settings, obs_source_t *source,
  588. bool input)
  589. {
  590. struct coreaudio_data *ca = bzalloc(sizeof(struct coreaudio_data));
  591. if (os_event_init(&ca->exit_event, OS_EVENT_TYPE_MANUAL) != 0) {
  592. blog(LOG_ERROR,
  593. "[coreaudio_create] failed to create "
  594. "semephore: %d",
  595. errno);
  596. bfree(ca);
  597. return NULL;
  598. }
  599. ca->device_uid = bstrdup(obs_data_get_string(settings, "device_id"));
  600. ca->source = source;
  601. ca->input = input;
  602. if (!ca->device_uid)
  603. ca->device_uid = bstrdup("default");
  604. coreaudio_try_init(ca);
  605. return ca;
  606. }
  607. static void *coreaudio_create_input_capture(obs_data_t *settings,
  608. obs_source_t *source)
  609. {
  610. return coreaudio_create(settings, source, true);
  611. }
  612. static void *coreaudio_create_output_capture(obs_data_t *settings,
  613. obs_source_t *source)
  614. {
  615. return coreaudio_create(settings, source, false);
  616. }
  617. static obs_properties_t *coreaudio_properties(bool input)
  618. {
  619. obs_properties_t *props = obs_properties_create();
  620. obs_property_t *property;
  621. struct device_list devices;
  622. memset(&devices, 0, sizeof(struct device_list));
  623. property = obs_properties_add_list(props, "device_id", TEXT_DEVICE,
  624. OBS_COMBO_TYPE_LIST,
  625. OBS_COMBO_FORMAT_STRING);
  626. coreaudio_enum_devices(&devices, input);
  627. if (devices.items.num)
  628. obs_property_list_add_string(property, TEXT_DEVICE_DEFAULT,
  629. "default");
  630. for (size_t i = 0; i < devices.items.num; i++) {
  631. struct device_item *item = devices.items.array + i;
  632. obs_property_list_add_string(property, item->name.array,
  633. item->value.array);
  634. }
  635. device_list_free(&devices);
  636. return props;
  637. }
  638. static obs_properties_t *coreaudio_input_properties(void *unused)
  639. {
  640. UNUSED_PARAMETER(unused);
  641. return coreaudio_properties(true);
  642. }
  643. static obs_properties_t *coreaudio_output_properties(void *unused)
  644. {
  645. UNUSED_PARAMETER(unused);
  646. return coreaudio_properties(false);
  647. }
  648. struct obs_source_info coreaudio_input_capture_info = {
  649. .id = "coreaudio_input_capture",
  650. .type = OBS_SOURCE_TYPE_INPUT,
  651. .output_flags = OBS_SOURCE_AUDIO | OBS_SOURCE_DO_NOT_DUPLICATE,
  652. .get_name = coreaudio_input_getname,
  653. .create = coreaudio_create_input_capture,
  654. .destroy = coreaudio_destroy,
  655. .update = coreaudio_update,
  656. .get_defaults = coreaudio_defaults,
  657. .get_properties = coreaudio_input_properties,
  658. .icon_type = OBS_ICON_TYPE_AUDIO_INPUT,
  659. };
  660. struct obs_source_info coreaudio_output_capture_info = {
  661. .id = "coreaudio_output_capture",
  662. .type = OBS_SOURCE_TYPE_INPUT,
  663. .output_flags = OBS_SOURCE_AUDIO | OBS_SOURCE_DO_NOT_DUPLICATE |
  664. OBS_SOURCE_DO_NOT_SELF_MONITOR,
  665. .get_name = coreaudio_output_getname,
  666. .create = coreaudio_create_output_capture,
  667. .destroy = coreaudio_destroy,
  668. .update = coreaudio_update,
  669. .get_defaults = coreaudio_defaults,
  670. .get_properties = coreaudio_output_properties,
  671. .icon_type = OBS_ICON_TYPE_AUDIO_OUTPUT,
  672. };