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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
extern crate nihav;
use std::io::prelude::*;
use std::io::SeekFrom;
use std::fs::File;
use std::path::Path;
use nihav::io::byteio::{FileReader, ByteReader};
use nihav::frame::*;
use nihav::codecs::*;
use nihav::demuxers::*;
use nihav::detect;
use std::env;
fn write_pgmyuv(pfx: &str, strno: usize, num: u64, frmref: NAFrameRef) {
let frm = frmref.borrow();
if let NABufferType::None = frm.get_buffer() { return; }
let name = format!("{}out{:02}_{:04}.pgm", pfx, strno, num);
let mut ofile = File::create(name).unwrap();
let buf = frm.get_buffer().get_vbuf().unwrap();
let (w, h) = buf.get_dimensions(0);
let (w2, h2) = buf.get_dimensions(1);
let tot_h = h + h2;
let hdr = format!("P5\n{} {}\n255\n", w, tot_h);
ofile.write_all(hdr.as_bytes()).unwrap();
let dta = buf.get_data();
let ls = buf.get_stride(0);
let mut idx = 0;
let mut idx2 = w;
let mut pad: Vec<u8> = Vec::with_capacity((w - w2 * 2) / 2);
pad.resize((w - w2 * 2) / 2, 0xFF);
for _ in 0..h {
let line = &dta[idx..idx2];
ofile.write_all(line).unwrap();
idx += ls;
idx2 += ls;
}
let mut base1 = buf.get_offset(1);
let stride1 = buf.get_stride(1);
let mut base2 = buf.get_offset(2);
let stride2 = buf.get_stride(2);
for _ in 0..h2 {
let bend1 = base1 + w2;
let line = &dta[base1..bend1];
ofile.write_all(line).unwrap();
ofile.write_all(pad.as_slice()).unwrap();
let bend2 = base2 + w2;
let line = &dta[base2..bend2];
ofile.write_all(line).unwrap();
ofile.write_all(pad.as_slice()).unwrap();
base1 += stride1;
base2 += stride2;
}
}
fn write_palppm(pfx: &str, strno: usize, num: u64, frmref: NAFrameRef) {
let frm = frmref.borrow();
let name = format!("{}out{:02}_{:04}.ppm", pfx, strno, num);
let mut ofile = File::create(name).unwrap();
let buf = frm.get_buffer().get_vbuf().unwrap();
let (w, h) = buf.get_dimensions(0);
let paloff = buf.get_offset(1);
let hdr = format!("P6\n{} {}\n255\n", w, h);
ofile.write_all(hdr.as_bytes()).unwrap();
let dta = buf.get_data();
let ls = buf.get_stride(0);
let mut idx = 0;
let mut line: Vec<u8> = Vec::with_capacity(w * 3);
line.resize(w * 3, 0);
for _ in 0..h {
let src = &dta[idx..(idx+w)];
for x in 0..w {
let pix = src[x] as usize;
line[x * 3 + 0] = dta[paloff + pix * 3 + 2];
line[x * 3 + 1] = dta[paloff + pix * 3 + 1];
line[x * 3 + 2] = dta[paloff + pix * 3 + 0];
}
ofile.write_all(line.as_slice()).unwrap();
idx += ls;
}
}
fn main() {
let args: Vec<_> = env::args().collect();
if args.len() == 1 { return; }
let lastpts = if args.len() > 2 {
Some(u64::from_str_radix(args[2].as_str(), 10).unwrap())
} else { None };
let name = args[1].as_str();
let path = Path::new(name);
let mut file = File::open(path).unwrap();
let dmx_fact;
let mut fr = FileReader::new_read(&mut file);
let mut br = ByteReader::new(&mut fr);
let res = detect::detect_format(name, &mut br);
if res.is_none() {
println!("cannot detect format for {}", name);
return;
}
let (dmx_name, _) = res.unwrap();
println!("trying demuxer {}", dmx_name);
dmx_fact = find_demuxer(dmx_name).unwrap();
br.seek(SeekFrom::Start(0)).unwrap();
let mut dmx = dmx_fact.new_demuxer(&mut br);
dmx.open().unwrap();
let mut decs: Vec<Option<Box<NADecoder>>> = Vec::new();
for i in 0..dmx.get_num_streams() {
let s = dmx.get_stream(i).unwrap();
let info = s.get_info();
let decfunc = find_decoder(info.get_name());
if !info.is_video() {
decs.push(None);
} else if let Some(df) = decfunc {
let mut dec = (df)();
dec.init(info).unwrap();
decs.push(Some(dec));
} else {
panic!("decoder {} not found", info.get_name());
}
}
loop {
let pktres = dmx.get_frame();
if let Err(e) = pktres {
if e == DemuxerError::EOF { break; }
}
let pkt = pktres.unwrap();
let streamno = pkt.get_stream().get_id() as usize;
if let Some(ref mut dec) = decs[streamno] {
let frm = dec.decode(&pkt).unwrap();
if pkt.get_stream().get_info().is_video() {
if frm.borrow().get_frame_type() != FrameType::Skip {
if frm.borrow().get_buffer().get_vbuf().unwrap().get_info().get_format().is_paletted() {
write_palppm("", streamno, pkt.get_pts().unwrap(), frm);
} else {
write_pgmyuv("", streamno, pkt.get_pts().unwrap(), frm);
}
}
}
}
if pkt.get_pts() != None && pkt.get_pts() == lastpts { break; }
}
//panic!("end");
}
|
