fix gcc warnings on shadowed declarations
[paraslash.git] / write.c
1 /*
2 * Copyright (C) 2005-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 #include "para.h"
20 #include "string.h"
21 #include "write.cmdline.h"
22 #include "list.h"
23 #include "sched.h"
24 #include "stdin.h"
25 #include "write.h"
26 #include "write_common.h"
27 #include "fd.h"
28 #include "error.h"
29
30 INIT_WRITE_ERRLISTS;
31
32 struct check_wav_task {
33 char *buf;
34 size_t *loaded;
35 int *eof;
36 unsigned channels;
37 unsigned samplerate;
38 struct task task;
39 };
40
41 struct initial_delay_task {
42 struct timeval start_time;
43 struct task task;
44 };
45
46 static struct write_args_info conf;
47 struct stdin_task sit;
48 struct check_wav_task cwt;
49 struct initial_delay_task idt;
50 static struct writer_node_group *wng;
51
52 #define WAV_HEADER_LEN 44
53
54 /**
55 * test if audio buffer contains a valid wave header
56 *
57 * \return If not, return -E_NO_WAV_HEADER, otherwise, return zero. If
58 * there is less than WAV_HEADER_LEN bytes awailable, return one.
59 */
60 static void check_wav_pre_select(__a_unused struct sched *s, struct task *t)
61 {
62 struct check_wav_task *wt = t->private_data;
63 unsigned char *a;
64
65 if (*wt->loaded < WAV_HEADER_LEN) {
66 t->ret = *wt->eof? -E_PREMATURE_END : 1;
67 return;
68 }
69 wt->channels = 2;
70 wt->samplerate = 44100;
71 a = (unsigned char*)wt->buf;
72 t->ret = -E_NO_WAV_HEADER;
73 if (a[0] != 'R' || a[1] != 'I' || a[2] != 'F' || a[3] != 'F')
74 return;
75 wt->channels = (unsigned) a[22];
76 wt->samplerate = a[24] + (a[25] << 8) + (a[26] << 16) + (a[27] << 24);
77 *wt->loaded -= WAV_HEADER_LEN;
78 memmove(wt->buf, wt->buf + WAV_HEADER_LEN, *wt->loaded);
79 t->ret = -E_WAV_HEADER_SUCCESS;
80 PARA_INFO_LOG("channels: %d, sample rate: %d\n", wt->channels, wt->samplerate);
81 }
82
83 static void initial_delay_pre_select(struct sched *s, struct task *t)
84 {
85 struct initial_delay_task *dt = t->private_data;
86 struct timeval diff;
87
88 t->ret = -E_NO_DELAY;
89 if (!dt->start_time.tv_sec && !dt->start_time.tv_usec)
90 return;
91 t->ret = -E_DELAY_TIMEOUT;
92 if (tv_diff(now, &dt->start_time, &diff) > 0)
93 return;
94 t->ret = 1;
95 if (tv_diff(&s->timeout , &diff, NULL) > 0)
96 s->timeout = diff;
97 }
98
99 void para_log(int ll, const char* fmt,...)
100 {
101 va_list argp;
102
103 if (ll < conf.loglevel_arg)
104 return;
105 va_start(argp, fmt);
106 vfprintf(stderr, fmt, argp);
107 va_end(argp);
108 }
109
110 static struct writer_node_group *check_args(void)
111 {
112 int i, ret = -E_WRITE_SYNTAX;
113 struct writer_node_group *g = NULL;
114
115 if (conf.list_writers_given) {
116 char *msg = NULL;
117 FOR_EACH_WRITER(i) {
118 char *tmp = make_message("%s%s%s",
119 i? msg : "",
120 i? " " : "",
121 writer_names[i]);
122 free(msg);
123 msg = tmp;
124 }
125 fprintf(stderr, "%s\n", msg);
126 free(msg);
127 exit(EXIT_SUCCESS);
128 }
129 if (conf.start_time_given) {
130 long unsigned sec, usec;
131 if (sscanf(conf.start_time_arg, "%lu:%lu",
132 &sec, &usec) != 2)
133 goto out;
134 idt.start_time.tv_sec = sec;
135 idt.start_time.tv_usec = usec;
136 }
137 if (!conf.writer_given) {
138 g = setup_default_wng();
139 ret = 1;
140 goto out;
141 }
142 g = wng_new(conf.writer_given);
143 ret = -E_WRITE_SYNTAX;
144 for (i = 0; i < conf.writer_given; i++) {
145 int writer_num;
146 g->writer_nodes[i].conf = check_writer_arg(
147 conf.writer_arg[i], &writer_num);
148 if (!g->writer_nodes[i].conf)
149 goto out;
150 g->writer_nodes[i].writer = &writers[writer_num];
151 }
152 ret = 1;
153 out:
154 if (ret > 0)
155 return g;
156 free(g);
157 return NULL;
158 }
159
160 static void wng_event_handler(struct task *t)
161 {
162 struct writer_node_group *g = t->private_data;
163
164 PARA_INFO_LOG("%s\n", PARA_STRERROR(-t->ret));
165 unregister_task(t);
166 wng_close(g);
167 }
168
169
170 static void idt_event_handler(struct task *t)
171 {
172 int ret;
173
174 PARA_INFO_LOG("%s\n", PARA_STRERROR(-t->ret));
175 unregister_task(t);
176 wng->buf = sit.buf;
177 wng->loaded = &sit.loaded;
178 wng->input_eof = &sit.eof;
179 wng->task.event_handler = wng_event_handler;
180 wng->channels = &cwt.channels;
181 wng->samplerate = &cwt.samplerate;
182 ret = wng_open(wng);
183 if (ret < 0) {
184 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
185 exit(EXIT_FAILURE);
186 }
187 }
188
189 static void cwt_event_handler(struct task *t)
190 {
191 if (t->ret != -E_NO_WAV_HEADER && t->ret != -E_WAV_HEADER_SUCCESS) {
192 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-t->ret));
193 exit(EXIT_FAILURE);
194 }
195 PARA_INFO_LOG("%s\n", PARA_STRERROR(-t->ret));
196 unregister_task(t);
197 // if (t->ret == -E_WAV_HEADER_SUCCESS) {
198 // conf.channels_arg = cwt.channels;
199 // conf.sample_rate_arg = cwt.sample_rate;
200 // }
201 idt.task.pre_select = initial_delay_pre_select;
202 idt.task.private_data = &idt;
203 idt.task.event_handler = idt_event_handler;
204 sprintf(idt.task.status, "initial_delay");
205 register_task(&idt.task);
206 }
207
208 int main(int argc, char *argv[])
209 {
210 int ret = -E_WRITE_SYNTAX;
211 struct sched s;
212
213 write_cmdline_parser(argc, argv, &conf);
214 init_supported_writers();
215
216 wng = check_args();
217 if (!wng)
218 goto out;
219 stdin_set_defaults(&sit);
220 if (conf.bufsize_given)
221 sit.bufsize = conf.bufsize_arg;
222 sit.buf = para_malloc(sit.bufsize),
223 register_task(&sit.task);
224
225 cwt.task.pre_select = check_wav_pre_select;
226 cwt.task.private_data = &cwt;
227 cwt.task.event_handler = cwt_event_handler;
228 cwt.buf = sit.buf;
229 cwt.loaded = &sit.loaded;
230 cwt.eof = &sit.eof;
231 sprintf(cwt.task.status, "check wav");
232 register_task(&cwt.task);
233
234 s.default_timeout.tv_sec = 1;
235 s.default_timeout.tv_usec = 0;
236 ret = sched(&s);
237
238 out:
239 if (ret < 0) {
240 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
241 ret = EXIT_FAILURE;
242 } else
243 ret = EXIT_SUCCESS;
244 return ret;
245 }