1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
extern mod extra;
use extra::getopts::*;
use extra::time::*;
use std::io::{stdout, stderr};
use std::os;
use std::path::Path;
use std::rt::io::{Create, io_error, Open, Write, Writer};
use std::rt::io::file::open;
use std::task::{SingleThreaded, spawn_sched};
use oss::OssDevice;
use mp3lame::LameContext;
mod oss;
mod mp3lame;
fn main() {
let args = os::args();
let opts = ~[
groups::optmulti("f", "file", "OSS device file", "/dev/dsp"),
groups::optmulti("r", "rate", "Sample rate in Hz", "44100"),
groups::optopt("s", "split",
"Number of minutes at which to split MP3 files", "60"),
groups::optflag("a", "align",
"Align splits as if the first one happened midnight Jan. 1, 1970"),
groups::optopt("b", "bitrate", "MP3 bitrate in kbps", "128"),
groups::optopt("q", "quality", "MP3 quality", "2"),
groups::optflag("d", "dump", "Dumps raw samples to stdout"),
];
let DSP_FILES = ~[~"/dev/dsp", ~"/dev/dsp1"];
let DSP_SPEEDS = ~[44100i, 48000i];
let FILE_FORMAT = "%F %H.%M.%S%z.mp3";
let matches = match groups::getopts(args.tail(), opts) {
Ok(m) => m,
Err(f) => {
println(f.to_err_msg());
print(groups::usage("Usage: per [options]", opts));
return;
}
};
let dsp_files = match matches.opt_strs("f") {
[] => DSP_FILES,
f => f
};
let lame = LameContext::new();
let mut foo = None;
for file_name in dsp_files.iter() {
match OssDevice::new(&Path(file_name.as_slice())) {
Some(x) => { foo = Some(x); break }
None => {}
}
};
let dsp = match foo {
Some(x) => x,
None => fail!("Unable to open dsp device")
};
dsp.reset();
let dsp_speeds = match matches.opt_strs("r") {
[] => DSP_SPEEDS,
r => r.map(|x| { from_str::<int>(*x).unwrap() })
};
let mut speed: int = 0;
for dsp_speed in dsp_speeds.iter() {
do io_error::cond.trap(|_| {speed = 0}).inside {
dsp.set_speed(*dsp_speed);
speed = *dsp_speed;
}
if speed != 0 {
break;
}
}
dsp.set_format();
dsp.set_stereo();
lame.set_num_channels(2);
let quality = match matches.opt_str("q") {
Some(q) => from_str::<int>(q).unwrap(),
None => 2
};
let bitrate = match matches.opt_str("b") {
Some(b) => from_str::<int>(b).unwrap(),
None => 128
};
lame.set_in_samplerate(speed);
lame.set_quality(quality);
lame.set_bitrate(bitrate);
lame.set_disable_reservoir(true);
lame.init_params();
stderr().write_str(fmt!("Recording sample rate: %d Hz\n", speed));
stderr().write_str(fmt!("Encoding sample rate: %d Hz\n",
lame.get_out_samplerate()));
let dump = matches.opt_present("d");
let split = match matches.opt_str("s") {
Some(s) => from_str::<int>(s).unwrap() * 60,
None => 3600
};
let mut next_split = if matches.opt_present("a") {
let Timespec { sec, _ } = get_time();
Timespec::new(sec - sec % split as i64, 0)
} else {
get_time()
};
let mut out_file = open(&Path("/dev/null"), Open, Write).unwrap();
let (port, chan) = stream::<~[u8]>();
do spawn_sched(SingleThreaded) {
dsp.read_all(&chan);
}
loop {
let now = get_time();
if next_split <= now {
debug!("split!");
out_file.write(lame.encode_flush_nogap());
out_file.flush();
out_file = open(&Path(at(now).strftime(FILE_FORMAT)), Create, Write)
.unwrap();
let Timespec { sec, nsec } = next_split;
next_split = Timespec::new(sec + split as i64, nsec);
}
let buffer = port.recv();
debug!("Read buffer of length %u", buffer.len());
if dump {
stdout().write(buffer);
}
let mp3buf = lame.encode_buffer_interleaved(buffer);
debug!("Encoded buffer of length %u", mp3buf.len());
out_file.write(mp3buf);
}
}
|
