writers: Kill ->open.
[paraslash.git] / osx_write.c
1 /*
2 * Copyright (C) 2006-2010 Andre Noll <maan@systemlinux.org>
3 *
4 * Licensed under the GPL v2. For licencing details see COPYING.
5 */
6
7 /** \file osx_write.c paraslash's output plugin for MacOs */
8
9 /*
10 * based on mosx-mpg123, by Guillaume Outters and Steven A. Kortze
11 * <skortze@sourceforge.net>
12 */
13
14 #include <regex.h>
15 #include <sys/types.h>
16 #include <dirent.h>
17 #include <stdbool.h>
18
19 #include "para.h"
20 #include "fd.h"
21 #include "string.h"
22 #include "list.h"
23 #include "sched.h"
24 #include "ggo.h"
25 #include "buffer_tree.h"
26 #include "write.h"
27 #include "write_common.h"
28 #include "osx_write.cmdline.h"
29 #include "error.h"
30
31 #include <CoreServices/CoreServices.h>
32 #include <AudioUnit/AudioUnit.h>
33 #include <AudioToolbox/AudioToolbox.h>
34
35 /** describes one input buffer for the osx writer */
36 struct osx_buffer {
37 /** pointer to the beginning of the buffer */
38 float *buffer;
39 /** the size of this buffer */
40 long size;
41 /** current position in the buffer */
42 float *ptr;
43 /** number of floats not yet consuned */
44 long remaining;
45 /** pointer to the next audio buffer */
46 struct osx_buffer *next;
47 };
48
49 /** data specific to the osx writer */
50 struct private_osx_write_data {
51 /** the main control structure for audio data manipulation */
52 AudioUnit audio_unit;
53 /** non-zero if playback has started */
54 char play;
55 /** callback reads audio data from this buffer */
56 struct osx_buffer *from;
57 /** the post_select writes audio data here */
58 struct osx_buffer *to;
59 /** sample rate of the current audio stream */
60 unsigned sample_rate;
61 /** Sample format of the current audio stream */
62 unsigned sample_format;
63 /** number of channels of the current audio stream */
64 unsigned channels;
65 };
66
67 static void destroy_buffers(struct private_osx_write_data *powd)
68 {
69 struct osx_buffer *ptr;
70 struct osx_buffer *ptr2;
71 ptr = powd->to->next;
72 powd->to->next = NULL;
73 while (ptr) {
74 ptr2 = ptr->next;
75 free(ptr->buffer);
76 free(ptr);
77 ptr = ptr2;
78 }
79 }
80
81 static void init_buffers(struct writer_node *wn)
82 {
83 struct private_osx_write_data *powd = wn->private_data;
84 struct osx_write_args_info *conf = wn->conf;
85 struct osx_buffer **ptrptr;
86 int i;
87
88 ptrptr = &powd->to;
89 for (i = 0; i < conf->numbuffers_arg; i++) {
90 *ptrptr = para_malloc(sizeof(struct osx_buffer));
91 (*ptrptr)->size = 0;
92 (*ptrptr)->remaining = 0;
93 (*ptrptr)->buffer = NULL;
94 ptrptr = &(*ptrptr)->next;
95 }
96 *ptrptr = powd->from = powd->to;
97 }
98
99 static void fill_buffer(struct private_osx_write_data *powd, char *data, long bytes)
100 {
101 struct osx_buffer *b = powd->to;
102 float *dest;
103 long samples;
104 enum sample_format sf = powd->sample_format;
105
106 samples = (sf == SF_S8 || sf == SF_U8)? bytes : bytes / 2;
107 assert(b->remaining == 0 || samples > 0);
108 if (b->size != samples) {
109 b->buffer = para_realloc(b->buffer, samples * sizeof(float));
110 b->size = samples;
111 }
112 dest = b->buffer;
113 switch (powd->sample_format) {
114 case SF_U8: {
115 uint8_t *src = (uint8_t *)data;
116 while (samples--) {
117 *dest++ = (*src++) / 256.0;
118 }
119 break;
120 }
121 case SF_S8: {
122 int8_t *src = (int8_t *)data;
123 while (samples--) {
124 *dest++ = ((*src++) + 128) / 256.0;
125 }
126 break;
127 }
128 default: {
129 short *src = (short *)data;
130 while (samples--)
131 *dest++ = (*src++) / 32768.0;
132 }
133 }
134 b->ptr = b->buffer;
135 b->remaining = b->size;
136 }
137
138 static OSStatus osx_callback(void * inClientData,
139 __a_unused AudioUnitRenderActionFlags *inActionFlags,
140 __a_unused const AudioTimeStamp *inTimeStamp,
141 __a_unused UInt32 inBusNumber,
142 __a_unused UInt32 inNumFrames,
143 AudioBufferList *outOutputData)
144 {
145 long m, n;
146 float *dest;
147 int i;
148 struct private_osx_write_data *powd = inClientData;
149
150 // PARA_INFO_LOG("%p\n", powd);
151 for (i = 0; i < outOutputData->mNumberBuffers; ++i) {
152 /* what we have to fill */
153 m = outOutputData->mBuffers[i].mDataByteSize / sizeof(float);
154 dest = outOutputData->mBuffers[i].mData;
155 while (m > 0) {
156 n = powd->from->remaining;
157 if (n <= 0) {
158 n = powd->from->next->remaining;
159 if (n <= 0) {
160 PARA_INFO_LOG("buffer underrun\n");
161 return 0;
162 }
163 powd->from = powd->from->next;
164 }
165 // PARA_INFO_LOG("buf %p: n = %ld, m= %ld\n", powd->from->buffer, n, m);
166 /*
167 * we dump what we can. In fact, just the necessary
168 * should be sufficient
169 */
170 if (n > m)
171 n = m;
172 memcpy(dest, powd->from->ptr, n * sizeof(float));
173 dest += n;
174 /* remember all done work */
175 m -= n;
176 powd->from->ptr += n;
177 if ((powd->from->remaining -= n) <= 0)
178 powd->from = powd->from->next;
179 }
180 }
181 return 0;
182 }
183
184 #ifdef WORDS_BIGENDIAN /* ppc */
185 #define ENDIAN_FLAGS kLinearPCMFormatFlagIsBigEndian
186 #else
187 #define ENDIAN_FLAGS 0
188 #endif
189
190 static int core_audio_init(struct writer_node *wn)
191 {
192 struct private_osx_write_data *powd = para_calloc(sizeof(*powd));
193 ComponentDescription desc;
194 Component comp;
195 AURenderCallbackStruct inputCallback = {osx_callback, powd};
196 AudioStreamBasicDescription format;
197 int ret;
198 struct btr_node *btrn = wn->btrn;
199 int32_t val;
200
201 wn->private_data = powd;
202 init_buffers(wn);
203 /* where did that default audio output go? */
204 desc.componentType = kAudioUnitType_Output;
205 desc.componentSubType = kAudioUnitSubType_DefaultOutput;
206 /* NOTE: and if default output isn't Apple? */
207 desc.componentManufacturer = kAudioUnitManufacturer_Apple;
208 desc.componentFlags = 0;
209 desc.componentFlagsMask = 0;
210 ret = -E_DEFAULT_COMP;
211 comp = FindNextComponent(NULL, &desc);
212 if (!comp)
213 goto e0;
214 ret = -E_OPEN_COMP;
215 if (OpenAComponent(comp, &powd->audio_unit))
216 goto e0;
217 ret = -E_UNIT_INIT;
218 if (AudioUnitInitialize(powd->audio_unit))
219 goto e1;
220 powd->play = 0;
221 get_btr_sample_rate(btrn, &val);
222 powd->sample_rate = val;
223 get_btr_channels(btrn, &val);
224 powd->channels = val;
225 get_btr_sample_format(btrn, &val);
226 powd->sample_format = val;
227 /*
228 * Choose PCM format. We tell the Output Unit what format we're going
229 * to supply data to it. This is necessary if you're providing data
230 * through an input callback AND you want the DefaultOutputUnit to do
231 * any format conversions necessary from your format to the device's
232 * format.
233 */
234 format.mFormatID = kAudioFormatLinearPCM;
235 format.mFramesPerPacket = 1;
236 format.mSampleRate = powd->sample_rate;
237 /* flags specific to each format */
238 format.mFormatFlags = kLinearPCMFormatFlagIsFloat
239 | kLinearPCMFormatFlagIsPacked
240 | ENDIAN_FLAGS;
241 switch (powd->sample_format) {
242 case SF_S8:
243 case SF_U8:
244 wn->min_iqs = powd->channels;
245 break;
246 default:
247 wn->min_iqs = powd->channels * 2;
248 }
249 format.mBitsPerChannel = 8 * sizeof(float);
250 format.mBytesPerPacket = powd->channels * sizeof(float);
251 format.mBytesPerFrame = format.mBytesPerPacket;
252 format.mChannelsPerFrame = powd->channels;
253
254 ret = -E_STREAM_FORMAT;
255 if (AudioUnitSetProperty(powd->audio_unit, kAudioUnitProperty_StreamFormat,
256 kAudioUnitScope_Input, 0, &format,
257 sizeof(AudioStreamBasicDescription)))
258 goto e2;
259 ret = -E_ADD_CALLBACK;
260 if (AudioUnitSetProperty(powd->audio_unit, kAudioUnitProperty_SetRenderCallback,
261 kAudioUnitScope_Input, 0, &inputCallback,
262 sizeof(inputCallback)) < 0)
263 goto e3;
264 return 1;
265 e3:
266 destroy_buffers(powd);
267 e2:
268 AudioUnitUninitialize(powd->audio_unit);
269 e1:
270 CloseComponent(powd->audio_unit);
271 e0:
272 free(powd);
273 wn->private_data = NULL;
274 return ret;
275 }
276
277 __malloc static void *osx_write_parse_config_or_die(const char *options)
278 {
279 struct osx_write_args_info *conf = para_calloc(sizeof(*conf));
280
281 /* exits on errors */
282 osx_cmdline_parser_string(options, conf, "osx_write");
283 return conf;
284 }
285
286 static void osx_free_config(void *conf)
287 {
288 osx_cmdline_parser_free(conf);
289 }
290
291 static void osx_write_close(struct writer_node *wn)
292 {
293 struct private_osx_write_data *powd = wn->private_data;
294
295 if (!powd)
296 return;
297 PARA_INFO_LOG("closing writer node %p\n", wn);
298 AudioOutputUnitStop(powd->audio_unit);
299 AudioUnitUninitialize(powd->audio_unit);
300 CloseComponent(powd->audio_unit);
301 destroy_buffers(powd);
302 free(powd);
303 }
304
305 static void osx_write_post_select(__a_unused struct sched *s, struct task *t)
306 {
307 struct writer_node *wn = container_of(t, struct writer_node, task);
308 struct private_osx_write_data *powd = wn->private_data;
309 struct btr_node *btrn = wn->btrn;
310 char *data;
311 size_t bytes;
312 int ret = 0;
313
314 while (!powd || powd->to->remaining <= 0) {
315 ret = btr_node_status(wn->btrn, wn->min_iqs, BTR_NT_LEAF);
316 if (ret <= 0)
317 break;
318 if (!powd) {
319 ret = core_audio_init(wn);
320 if (ret < 0)
321 break;
322 powd = wn->private_data;
323 }
324 btr_merge(btrn, 8192);
325 bytes = btr_next_buffer(btrn, &data);
326 //PARA_CRIT_LOG("have: %zu\n", bytes);
327 fill_buffer(powd, data, bytes);
328 btr_consume(btrn, bytes);
329 if (!powd->play) {
330 ret = -E_UNIT_START;
331 if (AudioOutputUnitStart(powd->audio_unit))
332 break;
333 powd->play = 1;
334 }
335 powd->to = powd->to->next;
336 }
337 if (ret < 0 && (!powd || powd->from->remaining <= 0)) {
338 btr_remove_node(btrn);
339 t->error = ret;
340 }
341 }
342
343 static void osx_write_pre_select(struct sched *s, struct task *t)
344 {
345 struct writer_node *wn = container_of(t, struct writer_node, task);
346 struct private_osx_write_data *powd = wn->private_data;
347 struct timeval tmp = {.tv_sec = 1, .tv_usec = 0}, delay = tmp;
348 unsigned long factor;
349 size_t numbytes;
350 int ret = btr_node_status(wn->btrn, wn->min_iqs, BTR_NT_LEAF);
351
352 if (ret == 0)
353 return;
354 if (ret < 0 || !powd)
355 return sched_min_delay(s);
356 assert(powd->sample_rate > 0);
357 assert(wn->min_iqs > 0);
358 numbytes = powd->to->remaining * sizeof(short);
359 factor = numbytes / powd->sample_rate / wn->min_iqs;
360 tv_scale(factor, &tmp, &delay);
361 sched_request_timeout(&delay, s);
362 }
363
364 /** the init function of the osx writer */
365 void osx_write_init(struct writer *w)
366 {
367 struct osx_write_args_info dummy;
368
369 osx_cmdline_parser_init(&dummy);
370 w->close = osx_write_close;
371 w->pre_select = osx_write_pre_select;
372 w->post_select = osx_write_post_select;
373 w->parse_config_or_die = osx_write_parse_config_or_die;
374 w->free_config = osx_free_config;
375 w->shutdown = NULL; /* nothing to do */
376 w->help = (struct ggo_help) {
377 .short_help = osx_write_args_info_help,
378 .detailed_help = osx_write_args_info_detailed_help
379 };
380 osx_cmdline_parser_free(&dummy);
381 }