3e5a97e8076e9a3d70bbb0d5386ee9b605a4cae7
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
{
41 struct initial_delay_task
{
42 struct timeval start_time
;
46 struct gengetopt_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
;
52 #define WAV_HEADER_LEN 44
55 * test if audio buffer contains a valid wave header
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.
60 static void check_wav_pre_select(struct sched
*s
, struct task
*t
)
62 struct check_wav_task
*cwt
= t
->private_data
;
65 if (*cwt
->loaded
< WAV_HEADER_LEN
) {
66 t
->ret
= *cwt
->eof
? -E_PREMATURE_END
: 1;
69 a
= (unsigned char*)cwt
->buf
;
70 t
->ret
= -E_NO_WAV_HEADER
;
71 if (a
[0] != 'R' || a
[1] != 'I' || a
[2] != 'F' || a
[3] != 'F')
73 cwt
->channels
= (unsigned) a
[22];
74 cwt
->sample_rate
= a
[24] + (a
[25] << 8) + (a
[26] << 16) + (a
[27] << 24);
75 *cwt
->loaded
-= WAV_HEADER_LEN
;
76 memmove(cwt
->buf
, cwt
->buf
+ WAV_HEADER_LEN
, *cwt
->loaded
);
77 t
->ret
= -E_WAV_HEADER_SUCCESS
;
78 PARA_INFO_LOG("channels: %d, sample_rate: %d\n", cwt
->channels
, cwt
->sample_rate
);
81 static void initial_delay_pre_select(struct sched
*s
, struct task
*t
)
83 struct initial_delay_task
*idt
= t
->private_data
;
87 if (!idt
->start_time
.tv_sec
&& !idt
->start_time
.tv_usec
)
89 t
->ret
= -E_DELAY_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
];
148 sprintf(wng
->writer_nodes
[i
].task
.status
, "%s",
160 static void idt_error_handler(struct task
*t
)
164 PARA_INFO_LOG("%s\n", PARA_STRERROR(-t
->ret
));
167 wng
->loaded
= &sit
.loaded
;
169 sprintf(wng
->task
.status
, "%s", "writer node group");
172 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret
));
177 static void cwt_error_handler(struct task
*t
)
179 if (t
->ret
!= -E_NO_WAV_HEADER
&& t
->ret
!= -E_WAV_HEADER_SUCCESS
) {
180 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-t
->ret
));
183 PARA_INFO_LOG("%s\n", PARA_STRERROR(-t
->ret
));
185 if (t
->ret
== -E_WAV_HEADER_SUCCESS
) {
186 conf
.channels_arg
= cwt
.channels
;
187 conf
.sample_rate_arg
= cwt
.sample_rate
;
189 idt
.task
.pre_select
= initial_delay_pre_select
;
190 idt
.task
.private_data
= &idt
;
191 idt
.task
.error_handler
= idt_error_handler
;
192 sprintf(idt
.task
.status
, "initial_delay");
193 register_task(&idt
.task
);
196 static void stdin_error_handler(struct task
*t
)
199 if (t
->ret
!= -E_STDIN_EOF
)
200 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-t
->ret
));
202 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-t
->ret
));
205 int main(int argc
, char *argv
[])
207 int ret
= -E_WRITE_SYNTAX
;
210 cmdline_parser(argc
, argv
, &conf
);
214 init_supported_writers();
217 sit
.bufsize
= 16 * 1024,
218 sit
.buf
= para_malloc(16 * 1024),
220 sit
.task
.pre_select
= stdin_pre_select
;
221 sit
.task
.post_select
= stdin_post_select
;
222 sit
.task
.error_handler
= stdin_error_handler
;
223 sit
.task
.private_data
= &sit
;
224 sprintf(sit
.task
.status
, "stdin reader");
225 register_task(&sit
.task
);
227 cwt
.task
.pre_select
= check_wav_pre_select
;
228 cwt
.task
.private_data
= &cwt
;
229 cwt
.task
.error_handler
= cwt_error_handler
;
231 cwt
.loaded
= &sit
.loaded
;
233 sprintf(cwt
.task
.status
, "check wav");
234 register_task(&cwt
.task
);
236 s
.default_timeout
.tv_sec
= 1;
237 s
.default_timeout
.tv_usec
= 0;
242 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret
));