make osx_write work for any bitrate
[paraslash.git] / osx_writer.c
1 /*
2 * Copyright (C) 2006 Andre Noll <maan@systemlinux.org>
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
17 */
18
19 /** \file osx_writer.c paraslash's output plugin for MacOs */
20
21 #include <CoreAudio/CoreAudio.h>
22 #include "para.h"
23 #include "fd.h"
24 #include "string.h"
25 #include "list.h"
26 #include "sched.h"
27 #include "write.h"
28 #include "osx_write.cmdline.h"
29 #include "error.h"
30
31
32 #include <CoreAudio/CoreAudio.h>
33 #include <AudioUnit/AudioUnit.h>
34 #include <AudioToolbox/DefaultAudioOutput.h>
35 #include <semaphore.h>
36 struct osx_buffer {
37 float *buffer;
38 long size;
39 float *ptr; /* Where in the buffer are we? */
40 long remaining;
41 struct osx_buffer *next;
42 };
43 typedef struct osx_buffer osx_buffer;
44
45 struct private_osx_writer_data {
46 long size;
47 short *ptr;
48 AudioUnit output;
49 char play;
50 sem_t *semaphore;
51 osx_buffer *from; /* Current buffers */
52 osx_buffer *to;
53 unsigned samplerate;
54 };
55
56
57 /*
58 * Tried with 3 buffers, but then any little window move is sufficient to
59 * stop the sound (OK, on a G3 400 with a Public Beta. Perhaps now we can
60 * go down to 2 buffers). With 16 buffers we have 1.5 seconds music
61 * buffered (or, if you're pessimistic, 1.5 seconds latency). Note 0
62 * buffers don't work much further than the Bus error.
63 */
64 #define NUMBER_BUFFERS 2
65
66 static void destroy_buffers(struct private_osx_writer_data *powd)
67 {
68 osx_buffer *ptr;
69 osx_buffer *ptr2;
70 ptr = powd->to->next;
71 powd->to->next = NULL;
72 while (ptr) {
73 ptr2 = ptr->next;
74 if (ptr->buffer)
75 free(ptr->buffer);
76 free(ptr);
77 ptr = ptr2;
78 }
79 }
80
81 static void init_buffers(struct private_osx_writer_data *powd)
82 {
83 int i;
84
85 osx_buffer ** ptrptr;
86 ptrptr = &powd->to;
87 for (i = 0; i < NUMBER_BUFFERS; i++) {
88 *ptrptr = malloc(sizeof(osx_buffer));
89 (*ptrptr)->size = 0;
90 (*ptrptr)->remaining = 0;
91 (*ptrptr)->buffer = NULL;
92 ptrptr = &(*ptrptr)->next;
93 /* This buffer is ready for filling (of course, it is empty!) */
94 sem_post(powd->semaphore);
95 }
96 *ptrptr = powd->from = powd->to;
97 }
98
99 static void fill_buffer(osx_buffer *b, short *source, long size)
100 {
101 float *dest;
102
103 if (b->remaining) /* Non empty buffer, must still be playing */
104 return;
105 PARA_INFO_LOG("%ld\n", size);
106 if (b->size != size) {
107 /*
108 * Hey! What's that? Coudn't this buffer size be fixed
109 * once (well, perhaps we just didn't allocate it yet)
110 */
111 if (b->buffer)
112 free(b->buffer);
113 b->buffer = malloc(size * sizeof(float));
114 b->size = size;
115 }
116 dest = b->buffer;
117 while (size--) {
118 char *tmp = (char *)source;
119 char c = *tmp;
120 *tmp = *(tmp + 1);
121 *(tmp + 1) = c;
122 /* *dest++ = ((*source++) + 32768) / 65536.0; */
123 *dest++ = (*source++) / 32768.0;
124 }
125 b->ptr = b->buffer;
126 b->remaining = b->size;
127 }
128
129 static OSStatus osx_callback(void * inClientData,
130 __a_unused AudioUnitRenderActionFlags *inActionFlags,
131 __a_unused const AudioTimeStamp *inTimeStamp,
132 __a_unused UInt32 inBusNumber,
133 __a_unused UInt32 inNumFrames,
134 AudioBufferList *outOutputData)
135 {
136 long m, n;
137 float *dest;
138 int i;
139 struct private_osx_writer_data *powd = inClientData;
140
141 // PARA_INFO_LOG("%p\n", powd);
142 for (i = 0; i < outOutputData->mNumberBuffers; ++i) {
143 /* what we have to fill */
144 m = outOutputData->mBuffers[i].mDataByteSize / sizeof(float);
145 dest = outOutputData->mBuffers[i].mData;
146 while (m > 0) {
147 if ((n = powd->from->remaining) <= 0) {
148 /* no more bytes in the current read buffer! */
149 while ((n = powd->from->remaining) <= 0)
150 /* wait for the results */
151 usleep(2000);
152 }
153 PARA_INFO_LOG("buf %p: n = %ld, m= %ld\n", powd->from->buffer, n, m);
154 /*
155 * we dump what we can. In fact, just the necessary
156 * should be sufficient
157 */
158 if (n > m)
159 n = m;
160 memcpy(dest, powd->from->ptr, n * sizeof(float));
161 dest += n;
162 /* remember all done work */
163 m -= n;
164 powd->from->ptr += n;
165 if ((powd->from->remaining -= n) <= 0) {
166 /* tell that there's a buffer to fill */
167 sem_post(powd->semaphore);
168 powd->from = powd->from->next;
169 }
170 }
171 }
172 return 0;
173 }
174
175 static int osx_writer_open(struct writer_node *wn)
176 {
177 struct private_osx_writer_data *powd = para_calloc(
178 sizeof(struct private_osx_writer_data));
179 ComponentDescription desc;
180 Component comp;
181 AURenderCallbackStruct inputCallback = {osx_callback, powd};
182 AudioStreamBasicDescription format;
183 char s[10];
184 int m, ret;
185 struct writer_node_group *wng = wn->wng;
186 struct osx_write_args_info *conf = wn->conf;
187
188 wn->private_data = powd;
189 /* where did that default audio output go? */
190 desc.componentType = kAudioUnitType_Output;
191 desc.componentSubType = kAudioUnitSubType_DefaultOutput;
192 /* NOTE: and if default output isn't Apple? */
193 desc.componentManufacturer = kAudioUnitManufacturer_Apple;
194 desc.componentFlags = 0;
195 desc.componentFlagsMask = 0;
196 ret = -E_DEFAULT_COMP;
197 comp = FindNextComponent(NULL, &desc);
198 if (!comp)
199 goto e0;
200 ret = -E_OPEN_COMP;
201 if (OpenAComponent(comp, &powd->output))
202 goto e0;
203 ret = -E_UNIT_INIT;
204 if (AudioUnitInitialize(powd->output))
205 goto e1;
206 powd->size = 0;
207 powd->ptr = NULL;
208 powd->play = 0;
209 /* Hmmm, let's choose PCM format */
210 /* We tell the Output Unit what format we're going to supply data to it.
211 * This is necessary if you're providing data through an input callback
212 * AND you want the DefaultOutputUnit to do any format conversions
213 * necessary from your format to the device's format.
214 */
215 if (!conf->samplerate_given && wng->samplerate)
216 powd->samplerate = *wng->samplerate;
217 else
218 powd->samplerate = conf->samplerate_arg;
219 format.mSampleRate = powd->samplerate;
220 /* The specific encoding type of audio stream*/
221 format.mFormatID = kAudioFormatLinearPCM;
222 /* flags specific to each format */
223 format.mFormatFlags = kLinearPCMFormatFlagIsFloat
224 | kLinearPCMFormatFlagIsPacked
225 | kLinearPCMFormatFlagIsBigEndian;
226 /*
227 * We produce 2-channel audio. Now if we have a mega-super-hyper card for our
228 * audio, it is its problem to convert it to 8-, 16-, 32- or 1024-channel data.
229 */
230 format.mBytesPerFrame = (format.mFramesPerPacket = 1)
231 * (format.mBytesPerPacket = (format.mChannelsPerFrame = 2) * sizeof(float));
232 /* one of the most constant constants of the whole computer history */
233 format.mBitsPerChannel = sizeof(float) * 8;
234 ret = -E_STREAM_FORMAT;
235 if (AudioUnitSetProperty(powd->output, kAudioUnitProperty_StreamFormat,
236 kAudioUnitScope_Input, 0, &format,
237 sizeof(AudioStreamBasicDescription)))
238 goto e2;
239 /* init the semaphore */
240 strcpy(s, "/mpg123-0000");
241 do {
242 for (m = 10;; m--)
243 if( (s[m]++) <= '9')
244 break;
245 else
246 s[m] = '0';
247 } while ((powd->semaphore = sem_open(s, O_CREAT | O_EXCL, 0644, 0))
248 == (sem_t *)SEM_FAILED);
249 init_buffers(powd);
250 ret = -E_ADD_CALLBACK;
251 if (AudioUnitSetProperty(powd->output, kAudioUnitProperty_SetRenderCallback,
252 kAudioUnitScope_Input, 0, &inputCallback,
253 sizeof(inputCallback)) < 0)
254 goto e3;
255 return 1;
256 e3:
257 destroy_buffers(powd);
258 e2:
259 AudioUnitUninitialize(powd->output);
260 e1:
261 CloseComponent(powd->output);
262 e0:
263 return ret;
264 }
265
266 __malloc void *osx_write_parse_config(char *options)
267 {
268 struct osx_write_args_info *conf
269 = para_calloc(sizeof(struct osx_write_args_info));
270 PARA_INFO_LOG("options: %s\n", options);
271 int ret = osx_cmdline_parser_string(options, conf, "osx_write");
272 if (ret)
273 goto err_out;
274 return conf;
275 err_out:
276 free(conf);
277 return NULL;
278
279 }
280
281 static void osx_writer_close(struct writer_node *wn)
282 {
283 struct private_osx_writer_data *powd = wn->private_data;
284
285 PARA_INFO_LOG("closing writer node %p\n", wn);
286 AudioOutputUnitStop(powd->output);
287 AudioUnitUninitialize(powd->output);
288 CloseComponent(powd->output);
289 destroy_buffers(powd);
290 sem_close(powd->semaphore);
291 free(powd);
292 }
293
294 static int need_new_buffer(struct writer_node *wn)
295 {
296 struct writer_node_group *wng = wn->wng;
297 struct private_osx_writer_data *powd = wn->private_data;
298
299 if (*wng->loaded < sizeof(short))
300 return 0;
301 if (powd->to->remaining) /* Non empty buffer, must still be playing */
302 return 0;
303 return 1;
304 }
305
306 static int osx_write_post_select(__a_unused struct sched *s,
307 struct writer_node *wn)
308 {
309 struct private_osx_writer_data *powd = wn->private_data;
310 struct writer_node_group *wng = wn->wng;
311 short *data = (short*)wng->buf;
312
313 if (!need_new_buffer(wn))
314 return 1;
315 fill_buffer(powd->to, data, *wng->loaded / sizeof(short));
316 powd->to = powd->to->next;
317 wn->written = *wng->loaded;
318 if (!powd->play) {
319 if (AudioOutputUnitStart(powd->output))
320 return -1;
321 powd->play = 1;
322 }
323 return 1;
324 }
325
326 static int osx_write_pre_select(struct sched *s, __a_unused struct writer_node *wn)
327 {
328 s->timeout.tv_sec = 0;
329 s->timeout.tv_usec = 20;
330 return 1;
331 }
332
333 void osx_writer_init(struct writer *w)
334 {
335 w->open = osx_writer_open;
336 w->close = osx_writer_close;
337 w->pre_select = osx_write_pre_select;
338 w->post_select = osx_write_post_select;
339 w->parse_config = osx_write_parse_config;
340 w->shutdown = NULL; /* nothing to do */
341 }