#include "write.h"
#include "write_common.h"
#include "osx_write.cmdline.h"
+#include "ipc.h"
#include "error.h"
#include <CoreServices/CoreServices.h>
#include <AudioUnit/AudioUnit.h>
#include <AudioToolbox/AudioToolbox.h>
-/** describes one input buffer for the osx writer */
-struct osx_buffer {
- /** pointer to the beginning of the buffer */
- float *buffer;
- /** the size of this buffer */
- long size;
- /** current position in the buffer */
- float *ptr;
- /** number of floats not yet consuned */
- long remaining;
- /** pointer to the next audio buffer */
- struct osx_buffer *next;
-};
-
-/** data specific to the osx writer */
+/** Data specific to the osx writer. */
struct private_osx_write_data {
- /** the main control structure for audio data manipulation */
+ /** The main CoreAudio handle. */
AudioUnit audio_unit;
- /** non-zero if playback has started */
- char play;
- /** callback reads audio data from this buffer */
- struct osx_buffer *from;
- /** the post_select writes audio data here */
- struct osx_buffer *to;
- /** sample rate of the current audio stream */
+ /** True if we wrote some audio data. */
+ bool playing;
+ /** Sample rate of the current audio stream. */
unsigned sample_rate;
/** Sample format of the current audio stream */
unsigned sample_format;
- /** number of channels of the current audio stream */
+ /** Number of channels of the current audio stream. */
unsigned channels;
+ /** Serializes access to buffer tree nodes. */
+ int mutex;
+ /** The btr node of the callback. */
+ struct btr_node *callback_btrn;
};
-static void destroy_buffers(struct private_osx_write_data *powd)
-{
- struct osx_buffer *ptr;
- struct osx_buffer *ptr2;
- ptr = powd->to->next;
- powd->to->next = NULL;
- while (ptr) {
- ptr2 = ptr->next;
- free(ptr->buffer);
- free(ptr);
- ptr = ptr2;
- }
-}
-
-static void init_buffers(struct writer_node *wn)
+/* This function writes the address and the number of bytes to one end of the socket.
+ * The post_select() function then fills the buffer and notifies the callback also
+ * through the socket.
+ */
+static OSStatus osx_callback(void *cb_arg, __a_unused AudioUnitRenderActionFlags *af,
+ __a_unused const AudioTimeStamp *ts, __a_unused UInt32 bus_number,
+ __a_unused UInt32 num_frames, AudioBufferList *abl)
{
- struct private_osx_write_data *powd = wn->private_data;
- struct osx_write_args_info *conf = wn->conf;
- struct osx_buffer **ptrptr;
int i;
+ struct writer_node *wn = cb_arg;
+ struct private_osx_write_data *powd = wn->private_data;
+ size_t samples_have, samples_want = 0;
- ptrptr = &powd->to;
- for (i = 0; i < conf->numbuffers_arg; i++) {
- *ptrptr = para_malloc(sizeof(struct osx_buffer));
- (*ptrptr)->size = 0;
- (*ptrptr)->remaining = 0;
- (*ptrptr)->buffer = NULL;
- ptrptr = &(*ptrptr)->next;
- }
- *ptrptr = powd->from = powd->to;
-}
-
-static void fill_buffer(struct private_osx_write_data *powd, char *data, long bytes)
-{
- struct osx_buffer *b = powd->to;
- float *dest;
- long samples;
- enum sample_format sf = powd->sample_format;
-
- samples = (sf == SF_S8 || sf == SF_U8)? bytes : bytes / 2;
- assert(b->remaining == 0 || samples > 0);
- if (b->size != samples) {
- b->buffer = para_realloc(b->buffer, samples * sizeof(float));
- b->size = samples;
- }
- dest = b->buffer;
- switch (powd->sample_format) {
- case SF_U8: {
- uint8_t *src = (uint8_t *)data;
- while (samples--) {
- *dest++ = (*src++) / 256.0;
+ mutex_lock(powd->mutex);
+ /*
+ * We fill with zeros if no data was yet written and we do not have
+ * enough to fill all buffers.
+ */
+ if (!powd->playing) {
+ size_t want = 0, have =
+ btr_get_input_queue_size(powd->callback_btrn);
+ for (i = 0; i < abl->mNumberBuffers; i++)
+ want += abl->mBuffers[i].mDataByteSize;
+ if (have < want) {
+ PARA_DEBUG_LOG("deferring playback (have = %zu < %zu = want)\n",
+ have, want);
+ for (i = 0; i < abl->mNumberBuffers; i++)
+ memset(abl->mBuffers[i].mData, 0,
+ abl->mBuffers[i].mDataByteSize);
+ goto out;
}
- break;
+ powd->playing = true;
}
- case SF_S8: {
- int8_t *src = (int8_t *)data;
- while (samples--) {
- *dest++ = ((*src++) + 128) / 256.0;
- }
- break;
- }
- default: {
- short *src = (short *)data;
- while (samples--)
- *dest++ = (*src++) / 32768.0;
- }
- }
- b->ptr = b->buffer;
- b->remaining = b->size;
-}
-
-static OSStatus osx_callback(void * inClientData,
- __a_unused AudioUnitRenderActionFlags *inActionFlags,
- __a_unused const AudioTimeStamp *inTimeStamp,
- __a_unused UInt32 inBusNumber,
- __a_unused UInt32 inNumFrames,
- AudioBufferList *outOutputData)
-{
- long m, n;
- float *dest;
- int i;
- struct private_osx_write_data *powd = inClientData;
-// PARA_INFO_LOG("%p\n", powd);
- for (i = 0; i < outOutputData->mNumberBuffers; ++i) {
+ for (i = 0; i < abl->mNumberBuffers; i++) {
/* what we have to fill */
- m = outOutputData->mBuffers[i].mDataByteSize / sizeof(float);
- dest = outOutputData->mBuffers[i].mData;
- while (m > 0) {
- n = powd->from->remaining;
- if (n <= 0) {
- n = powd->from->next->remaining;
- if (n <= 0) {
- PARA_INFO_LOG("buffer underrun\n");
- return 0;
- }
- powd->from = powd->from->next;
- }
-// PARA_INFO_LOG("buf %p: n = %ld, m= %ld\n", powd->from->buffer, n, m);
- /*
- * we dump what we can. In fact, just the necessary
- * should be sufficient
- */
- if (n > m)
- n = m;
- memcpy(dest, powd->from->ptr, n * sizeof(float));
- dest += n;
- /* remember all done work */
- m -= n;
- powd->from->ptr += n;
- if ((powd->from->remaining -= n) <= 0)
- powd->from = powd->from->next;
+ void *dest = abl->mBuffers[i].mData;
+ size_t sz = abl->mBuffers[i].mDataByteSize, samples, bytes;
+
+ samples_want = sz / wn->min_iqs;
+ while (samples_want > 0) {
+ char *buf;
+ btr_merge(powd->callback_btrn, wn->min_iqs);
+ samples_have = btr_next_buffer(powd->callback_btrn, &buf) / wn->min_iqs;
+ //PARA_INFO_LOG("i: %d want %zu samples to addr %p, have: %zu\n", i, samples_want,
+ // dest, samples_have);
+ samples = PARA_MIN(samples_have, samples_want);
+ if (samples == 0)
+ break;
+ bytes = samples * wn->min_iqs;
+ memcpy(dest, buf, bytes);
+ btr_consume(powd->callback_btrn, bytes);
+ samples_want -= samples;
+ dest += bytes;
}
+ if (samples_want == 0)
+ continue;
+ if (btr_node_status(wn->btrn, wn->min_iqs, BTR_NT_LEAF) >= 0)
+ PARA_INFO_LOG("zero-padding (%zu samples)\n",
+ samples_want);
+ memset(dest, 0, samples_want * wn->min_iqs);
+ break;
}
- return 0;
+out:
+ mutex_unlock(powd->mutex);
+ return noErr;
}
-#ifdef WORDS_BIGENDIAN /* ppc */
-#define ENDIAN_FLAGS kLinearPCMFormatFlagIsBigEndian
-#else
-#define ENDIAN_FLAGS 0
-#endif
-
static int core_audio_init(struct writer_node *wn)
{
struct private_osx_write_data *powd = para_calloc(sizeof(*powd));
- ComponentDescription desc;
Component comp;
- AURenderCallbackStruct inputCallback = {osx_callback, powd};
- AudioStreamBasicDescription format;
int ret;
- struct btr_node *btrn = wn->btrn;
int32_t val;
+ AURenderCallbackStruct input_callback;
+ ComponentDescription desc = {
+ .componentType = kAudioUnitType_Output,
+ .componentSubType = kAudioUnitSubType_DefaultOutput,
+ .componentManufacturer = kAudioUnitManufacturer_Apple,
+ };
+ AudioStreamBasicDescription format = {
+ .mFormatID = kAudioFormatLinearPCM,
+ .mFramesPerPacket = 1,
+ };
+ struct btr_node *btrn = wn->btrn;
+ struct btr_node_description bnd;
- wn->private_data = powd;
- init_buffers(wn);
- /* where did that default audio output go? */
- desc.componentType = kAudioUnitType_Output;
- desc.componentSubType = kAudioUnitSubType_DefaultOutput;
- /* NOTE: and if default output isn't Apple? */
- desc.componentManufacturer = kAudioUnitManufacturer_Apple;
- desc.componentFlags = 0;
- desc.componentFlagsMask = 0;
+ PARA_INFO_LOG("wn: %p\n", wn);
ret = -E_DEFAULT_COMP;
comp = FindNextComponent(NULL, &desc);
if (!comp)
ret = -E_UNIT_INIT;
if (AudioUnitInitialize(powd->audio_unit))
goto e1;
- powd->play = 0;
get_btr_sample_rate(btrn, &val);
powd->sample_rate = val;
get_btr_channels(btrn, &val);
* any format conversions necessary from your format to the device's
* format.
*/
- format.mFormatID = kAudioFormatLinearPCM;
- format.mFramesPerPacket = 1;
+
format.mSampleRate = powd->sample_rate;
- /* flags specific to each format */
- format.mFormatFlags = kLinearPCMFormatFlagIsFloat
- | kLinearPCMFormatFlagIsPacked
- | ENDIAN_FLAGS;
+ format.mChannelsPerFrame = powd->channels;
+
switch (powd->sample_format) {
case SF_S8:
case SF_U8:
wn->min_iqs = powd->channels;
+ format.mBitsPerChannel = 8;
+ format.mBytesPerPacket = powd->channels;
+ format.mFormatFlags |= kLinearPCMFormatFlagIsPacked;
break;
default:
wn->min_iqs = powd->channels * 2;
+ format.mBytesPerPacket = powd->channels * 2;
+ format.mBitsPerChannel = 16;
+ format.mFormatFlags |= kLinearPCMFormatFlagIsSignedInteger;
}
- format.mBitsPerChannel = 8 * sizeof(float);
- format.mBytesPerPacket = powd->channels * sizeof(float);
format.mBytesPerFrame = format.mBytesPerPacket;
- format.mChannelsPerFrame = powd->channels;
+ if (powd->sample_format == SF_S16_BE || powd->sample_format == SF_U16_BE)
+ format.mFormatFlags |= kLinearPCMFormatFlagIsBigEndian;
+
+ input_callback = (AURenderCallbackStruct){osx_callback, wn};
ret = -E_STREAM_FORMAT;
if (AudioUnitSetProperty(powd->audio_unit, kAudioUnitProperty_StreamFormat,
- kAudioUnitScope_Input, 0, &format,
- sizeof(AudioStreamBasicDescription)))
+ kAudioUnitScope_Input, 0, &format, sizeof(format)))
goto e2;
ret = -E_ADD_CALLBACK;
if (AudioUnitSetProperty(powd->audio_unit, kAudioUnitProperty_SetRenderCallback,
- kAudioUnitScope_Input, 0, &inputCallback,
- sizeof(inputCallback)) < 0)
- goto e3;
+ kAudioUnitScope_Input, 0, &input_callback,
+ sizeof(input_callback)) < 0)
+ goto e2;
+
+ ret = mutex_new();
+ if (ret < 0)
+ goto e2;
+ powd->mutex = ret;
+ /* set up callback btr node */
+ bnd.name = "cb_node";
+ bnd.parent = btrn;
+ bnd.child = NULL;
+ bnd.handler = NULL;
+ bnd.context = powd;
+ powd->callback_btrn = btr_new_node(&bnd);
+ wn->private_data = powd;
return 1;
-e3:
- destroy_buffers(powd);
e2:
AudioUnitUninitialize(powd->audio_unit);
e1:
if (!powd)
return;
PARA_INFO_LOG("closing writer node %p\n", wn);
- AudioOutputUnitStop(powd->audio_unit);
- AudioUnitUninitialize(powd->audio_unit);
- CloseComponent(powd->audio_unit);
- destroy_buffers(powd);
+ mutex_destroy(powd->mutex);
free(powd);
+ wn->private_data = NULL;
}
-static void osx_write_post_select(__a_unused struct sched *s, struct task *t)
+/* must be called with the mutex held */
+static inline bool need_drain_delay(struct private_osx_write_data *powd)
+{
+ if (!powd->playing)
+ return false;
+ return btr_get_input_queue_size(powd->callback_btrn) != 0;
+}
+
+static void osx_write_pre_select(struct sched *s, struct task *t)
{
struct writer_node *wn = container_of(t, struct writer_node, task);
struct private_osx_write_data *powd = wn->private_data;
- struct btr_node *btrn = wn->btrn;
- char *data;
- size_t bytes;
- int ret = 0;
+ int ret;
+ bool drain_delay_nec = false;
- while (!powd || powd->to->remaining <= 0) {
+ if (!powd) {
ret = btr_node_status(wn->btrn, wn->min_iqs, BTR_NT_LEAF);
- if (ret <= 0)
- break;
- if (!powd) {
- ret = core_audio_init(wn);
- if (ret < 0)
- break;
- powd = wn->private_data;
- }
- btr_merge(btrn, 8192);
- bytes = btr_next_buffer(btrn, &data);
- //PARA_CRIT_LOG("have: %zu\n", bytes);
- fill_buffer(powd, data, bytes);
- btr_consume(btrn, bytes);
- if (!powd->play) {
- ret = -E_UNIT_START;
- if (AudioOutputUnitStart(powd->audio_unit))
- break;
- powd->play = 1;
- }
- powd->to = powd->to->next;
- }
- if (ret < 0 && (!powd || powd->from->remaining <= 0)) {
- btr_remove_node(btrn);
- t->error = ret;
+ if (ret != 0)
+ sched_min_delay(s);
+ return;
}
+
+ mutex_lock(powd->mutex);
+ ret = btr_node_status(wn->btrn, wn->min_iqs, BTR_NT_LEAF);
+ if (ret < 0)
+ drain_delay_nec = need_drain_delay(powd);
+ mutex_unlock(powd->mutex);
+
+ if (drain_delay_nec)
+ return sched_request_timeout_ms(50, s);
+ if (ret != 0)
+ return sched_min_delay(s);
+ sched_request_timeout_ms(50, s);
}
-static void osx_write_pre_select(struct sched *s, struct task *t)
+static void osx_write_post_select(__a_unused struct sched *s, struct task *t)
{
struct writer_node *wn = container_of(t, struct writer_node, task);
struct private_osx_write_data *powd = wn->private_data;
- struct timeval tmp = {.tv_sec = 1, .tv_usec = 0}, delay = tmp;
- unsigned long factor;
- size_t numbytes;
- int ret = btr_node_status(wn->btrn, wn->min_iqs, BTR_NT_LEAF);
+ struct btr_node *btrn = wn->btrn;
+ int ret;
- if (ret == 0)
- return;
- if (ret < 0 || !powd)
- return sched_min_delay(s);
- assert(powd->sample_rate > 0);
- assert(wn->min_iqs > 0);
- numbytes = powd->to->remaining * sizeof(short);
- factor = numbytes / powd->sample_rate / wn->min_iqs;
- tv_scale(factor, &tmp, &delay);
- sched_request_timeout(&delay, s);
+ if (!powd) {
+ ret = btr_node_status(btrn, wn->min_iqs, BTR_NT_LEAF);
+ if (ret == 0)
+ return;
+ if (ret < 0)
+ goto remove_btrn;
+ ret = core_audio_init(wn);
+ if (ret < 0)
+ goto remove_btrn;
+ powd = wn->private_data;
+ AudioOutputUnitStart(powd->audio_unit);
+ }
+ mutex_lock(powd->mutex);
+ btr_pushdown(btrn);
+ ret = btr_node_status(btrn, wn->min_iqs, BTR_NT_LEAF);
+ if (ret < 0 && need_drain_delay(powd))
+ ret = 0;
+ mutex_unlock(powd->mutex);
+
+ if (ret >= 0)
+ goto out;
+ AudioOutputUnitStop(powd->audio_unit);
+ AudioUnitUninitialize(powd->audio_unit);
+ CloseComponent(powd->audio_unit);
+ btr_remove_node(powd->callback_btrn);
+ btr_free_node(powd->callback_btrn);
+remove_btrn:
+ btr_remove_node(btrn);
+ PARA_NOTICE_LOG("%s\n", para_strerror(-ret));
+out:
+ t->error = ret;
}
-/** the init function of the osx writer */
+/**
+ * The init function of the osx writer.
+ *
+ * \param w Filled in by the function.
+ */
void osx_write_init(struct writer *w)
{
struct osx_write_args_info dummy;