fb05b591d79b1eb2db1429f79ff405f808a4fff3
[disorder] / driver / disorder.c
1 /*
2 * This file is part of DisOrder.
3 * Copyright (C) 2005, 2007 Richard Kettlewell
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
18 * USA
19 */
20 /** @file driver/disorder.c
21 * @brief libao driver used by DisOrder
22 *
23 * The output from this driver is expected to be fed to @c
24 * disorder-normalize to convert to the confnigured target format.
25 */
26
27 #include <config.h>
28 #include "types.h"
29
30 #include <string.h>
31 #include <stdlib.h>
32 #include <errno.h>
33 #include <unistd.h>
34 #include <poll.h>
35 #include <ao/ao.h>
36 #include <ao/plugin.h>
37
38 #include "speaker-protocol.h"
39
40 /* extra declarations to help out lazy <ao/plugin.h> */
41 int ao_plugin_test(void);
42 ao_info *ao_plugin_driver_info(void);
43 char *ao_plugin_file_extension(void);
44
45 /** @brief Private data structure for this driver */
46 struct internal {
47 int fd; /* output file descriptor */
48 int exit_on_error; /* exit on write error */
49
50 /** @brief Record of sample format */
51 struct stream_header header;
52
53 };
54
55 /* like write() but never returns EINTR/EAGAIN or short */
56 static int do_write(int fd, const void *ptr, size_t n) {
57 size_t written = 0;
58 int ret;
59 struct pollfd ufd;
60
61 memset(&ufd, 0, sizeof ufd);
62 ufd.fd = fd;
63 ufd.events = POLLOUT;
64 while(written < n) {
65 ret = write(fd, (const char *)ptr + written, n - written);
66 if(ret < 0) {
67 switch(errno) {
68 case EINTR: break;
69 case EAGAIN:
70 /* Someone sneakily gave us a nonblocking file descriptor, wait until
71 * we can write again */
72 ret = poll(&ufd, 1, -1);
73 if(ret < 0 && errno != EINTR) return -1;
74 break;
75 default:
76 return -1;
77 }
78 } else
79 written += ret;
80 }
81 return written;
82 }
83
84 /* return 1 if this driver can be opened */
85 int ao_plugin_test(void) {
86 return 1;
87 }
88
89 /* return info about this driver */
90 ao_info *ao_plugin_driver_info(void) {
91 static const char *options[] = { "fd" };
92 static const ao_info info = {
93 AO_TYPE_LIVE, /* type */
94 (char *)"DisOrder format driver", /* name */
95 (char *)"disorder", /* short_name */
96 (char *)"http://www.greenend.org.uk/rjk/disorder/", /* comment */
97 (char *)"Richard Kettlewell", /* author */
98 AO_FMT_NATIVE, /* preferred_byte_format */
99 0, /* priority */
100 (char **)options, /* options */
101 1, /* option_count */
102 };
103 return (ao_info *)&info;
104 }
105
106 /* initialize the private data structure */
107 int ao_plugin_device_init(ao_device *device) {
108 struct internal *i = malloc(sizeof (struct internal));
109 const char *e;
110
111 if(!i) return 0;
112 memset(i, 0, sizeof *i);
113 if((e = getenv("DISORDER_RAW_FD")))
114 i->fd = atoi(e);
115 else
116 i->fd = 1;
117 device->internal = i;
118 return 1;
119 }
120
121 /* set an option */
122 int ao_plugin_set_option(ao_device *device,
123 const char *key,
124 const char *value) {
125 struct internal *i = device->internal;
126
127 if(!strcmp(key, "fd"))
128 i->fd = atoi(value);
129 else if(!strcmp(key, "fragile"))
130 i->exit_on_error = atoi(value);
131 /* unknown options are required to be ignored */
132 return 1;
133 }
134
135 /* open the device */
136 int ao_plugin_open(ao_device *device, ao_sample_format *format) {
137 struct internal *i = device->internal;
138
139 /* we would like native-order samples */
140 device->driver_byte_format = AO_FMT_NATIVE;
141 i->header.rate = format->rate;
142 i->header.channels = format->channels;
143 i->header.bits = format->bits;
144 i->header.endian = ENDIAN_NATIVE;
145 return 1;
146 }
147
148 /* play some samples */
149 int ao_plugin_play(ao_device *device, const char *output_samples,
150 uint_32 num_bytes) {
151 struct internal *i = device->internal;
152
153 /* Fill in and write the header */
154 i->header.nbytes = num_bytes;
155 if(do_write(i->fd, &i->header, sizeof i->header) < 0) {
156 if(i->exit_on_error) _exit(-1);
157 return 0;
158 }
159
160 /* Write the sample data */
161 if(do_write(i->fd, output_samples, num_bytes) < 0) {
162 if(i->exit_on_error) _exit(-1);
163 return 0;
164 }
165 return 1;
166 }
167
168 /* close the device */
169 int ao_plugin_close(ao_device attribute((unused)) *device) {
170 return 1;
171 }
172
173 /* delete private data structures */
174 void ao_plugin_device_clear(ao_device *device) {
175 free(device->internal);
176 device->internal = 0;
177 }
178
179 /* report preferred filename extension */
180 char *ao_plugin_file_extension(void) {
181 return 0;
182 }
183
184 /*
185 Local Variables:
186 c-basic-offset:2
187 comment-column:40
188 End:
189 */