2 * Copyright (C) 2005-2006 Andre Noll <maan@systemlinux.org>
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.
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.
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.
21 #include "write.cmdline.h"
26 #include "write_common.h"
32 struct check_wav_task
{
40 struct initial_delay_task
{
41 struct timeval start_time
;
45 struct gengetopt_args_info conf
;
46 struct stdin_task sit
;
47 struct check_wav_task cwt
;
48 struct initial_delay_task idt
;
49 static struct writer_node_group
*wng
;
51 #define WAV_HEADER_LEN 44
54 * test if audio buffer contains a valid wave header
56 * \return If not, return 0, otherwise, store number of channels and sample rate
57 * in struct conf and return WAV_HEADER_LEN.
59 static void check_wav_pre_select(struct sched
*s
, struct task
*t
)
61 struct check_wav_task
*cwt
= t
->private_data
;
64 if (*cwt
->loaded
< WAV_HEADER_LEN
) {
68 a
= (unsigned char*)cwt
->buf
;
69 t
->ret
= -E_NO_WAV_HEADER
;
70 if (a
[0] != 'R' || a
[1] != 'I' || a
[2] != 'F' || a
[3] != 'F')
72 cwt
->channels
= (unsigned) a
[22];
73 cwt
->sample_rate
= a
[24] + (a
[25] << 8) + (a
[26] << 16) + (a
[27] << 24);
74 *cwt
->loaded
-= WAV_HEADER_LEN
;
75 memmove(cwt
->buf
, cwt
->buf
+ WAV_HEADER_LEN
, *cwt
->loaded
);
77 PARA_INFO_LOG("channels: %d, sample_rate: %d\n", cwt
->channels
, cwt
->sample_rate
);
80 static void initial_delay_pre_select(struct sched
*s
, struct task
*t
)
82 struct initial_delay_task
*idt
= t
->private_data
;
85 PARA_ERROR_LOG("task %p, ret: %d\n", t
, t
->ret
);
87 if (!idt
->start_time
.tv_sec
&& !idt
->start_time
.tv_usec
)
89 t
->ret
= 0; /* timeout */
90 if (tv_diff(&s
->now
, &idt
->start_time
, &diff
) > 0)
93 if (tv_diff(&s
->timeout
, &diff
, NULL
) > 0)
97 void para_log(int ll
, const char* fmt
,...)
101 if (ll
< conf
.loglevel_arg
)
104 vfprintf(stderr
, fmt
, argp
);
108 static struct writer_node_group
*check_args(void)
110 int i
, ret
= -E_WRITE_SYNTAX
;
111 struct writer_node_group
*wng
= NULL
;
113 if (conf
.list_writers_given
) {
116 char *tmp
= make_message("%s%s%s",
123 fprintf(stderr
, "%s\n", msg
);
127 if (conf
.prebuffer_arg
< 0 || conf
.prebuffer_arg
> 100)
129 if (conf
.start_time_given
) {
130 long unsigned sec
, usec
;
131 if (sscanf(conf
.start_time_arg
, "%lu:%lu",
134 idt
.start_time
.tv_sec
= sec
;
135 idt
.start_time
.tv_usec
= usec
;
137 if (!conf
.writer_given
) {
138 wng
= setup_default_wng();
142 wng
= wng_new(conf
.writer_given
);
143 for (i
= 0; i
< conf
.writer_given
; i
++) {
144 ret
= check_writer_arg(conf
.writer_arg
[i
]);
147 wng
->writer_nodes
[i
].writer
= &writers
[ret
];
159 static void idt_error_handler(struct task
*t
)
161 PARA_ERROR_LOG("task %p, ret: %d\n", t
, t
->ret
);
165 wng
->loaded
= &sit
.loaded
;
168 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret
));
173 static void cwt_error_handler(struct task
*t
)
175 PARA_ERROR_LOG("task %p, ret: %d\n", t
, t
->ret
);
177 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-t
->ret
));
178 if (t
->ret
!= -E_NO_WAV_HEADER
&& t
->ret
!= -E_PRE_EOF
)
180 if (t
->ret
== -E_PRE_EOF
) {
181 conf
.channels_arg
= cwt
.channels
;
182 conf
.sample_rate_arg
= cwt
.sample_rate
;
186 idt
.task
.pre_select
= initial_delay_pre_select
;
187 idt
.task
.private_data
= &idt
;
188 idt
.task
.error_handler
= idt_error_handler
;
189 idt
.task
.flags
= PRE_EOF_IS_ERROR
;
190 register_task(&idt
.task
);
193 static void stdin_error_handler(struct task
*t
)
196 PARA_INFO_LOG("task %p, ret: %d\n", t
, t
->ret
);
198 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-t
->ret
));
201 int main(int argc
, char *argv
[])
203 int ret
= -E_WRITE_SYNTAX
;
206 cmdline_parser(argc
, argv
, &conf
);
210 init_supported_writers();
213 sit
.bufsize
= 16 * 1024,
214 sit
.buf
= para_malloc(16 * 1024),
216 sit
.task
.pre_select
= stdin_pre_select
;
217 sit
.task
.post_select
= stdin_post_select
;
218 sit
.task
.error_handler
= stdin_error_handler
;
219 sit
.task
.flags
= POST_EOF_IS_ERROR
;
220 sit
.task
.private_data
= &sit
;
221 register_task(&sit
.task
);
223 cwt
.task
.pre_select
= check_wav_pre_select
;
224 cwt
.task
.private_data
= &cwt
;
225 cwt
.task
.error_handler
= cwt_error_handler
;
227 cwt
.loaded
= &sit
.loaded
;
228 cwt
.task
.flags
= PRE_EOF_IS_ERROR
;
229 register_task(&cwt
.task
);
231 s
.default_timeout
.tv_sec
= 1;
232 s
.default_timeout
.tv_usec
= 0;
237 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret
));