run/
run.rs

1use std::path::{Path, PathBuf};
2
3use seq_data_file::{SeqDataFormat, SeqDataReader, SeqDataReaderSeek, SeqDataWriter};
4
5pub struct H;
6pub struct H2;
7
8impl SeqDataFormat for H {
9    const MAGIC: &'static [u8] = &[];
10    const HEADER_SIZE: usize = 0;
11}
12
13impl SeqDataFormat for H2 {
14    const MAGIC: &'static [u8] = &[0xde, 0xad, 0xbe, 0xef];
15    const HEADER_SIZE: usize = 10;
16}
17
18const DATA1: &[u8] = &[1, 2, 3, 4, 5, 6, 7];
19const DATA2: &[u8] = &[125, 33, 6, 35, 6, 235, 46, 43, 25, 37];
20const DATA3: &[u8] = &[
21    10, 20, 30, 40, 50, 60, 70, 80, 90, 10, 20, 30, 40, 50, 60, 70, 80, 90, 10, 20, 30, 40, 50, 60,
22    70, 80, 90,
23];
24
25fn main() {
26    let sdf_file = PathBuf::from("a.sdf");
27
28    if sdf_file.exists() {
29        panic!("file {} already exists", &sdf_file.display())
30    }
31
32    run_writer_reader::<H>(&sdf_file);
33    std::fs::remove_file(&sdf_file).unwrap();
34    run_writer_reader::<H2>(&sdf_file);
35    std::fs::remove_file(&sdf_file).unwrap();
36}
37
38fn run_writer_reader<H: SeqDataFormat>(sdf_file: &Path) {
39    {
40        let header = vec![0x90; H::HEADER_SIZE];
41        let mut sdf = SeqDataWriter::<H>::create(&sdf_file, &header).unwrap();
42        sdf.append(DATA1).unwrap();
43        sdf.append(DATA2).unwrap();
44        sdf.append(DATA3).unwrap();
45    }
46
47    let mut pos = Vec::new();
48    {
49        let (mut sdf, _header) = SeqDataReader::<H>::open(&sdf_file).unwrap();
50        let (p1, r1) = sdf.next().unwrap().unwrap();
51        let (p2, r2) = sdf.next().unwrap().unwrap();
52        let (p3, r3) = sdf.next().unwrap().unwrap();
53
54        pos.push(p1);
55        pos.push(p2);
56        pos.push(p3);
57
58        assert_eq!(p1, 0);
59        assert_eq!(r1, DATA1);
60
61        assert_eq!(p2, 4 + DATA1.len() as u64);
62        assert_eq!(r2, DATA2);
63
64        assert_eq!(p3, 4 * 2 + DATA1.len() as u64 + DATA2.len() as u64);
65        assert_eq!(r3, DATA3);
66
67        if let Some(x) = sdf.next() {
68            panic!("more data than expected {:?}", x)
69        }
70    }
71
72    {
73        let (mut sdf, _header) = SeqDataReaderSeek::<H>::open(sdf_file).unwrap();
74        let r2 = sdf.next_at(pos[1]).unwrap();
75        assert_eq!(r2, DATA2);
76        let r1 = sdf.next_at(pos[0]).unwrap();
77        assert_eq!(r1, DATA1);
78        let r3 = sdf.next_at(pos[2]).unwrap();
79        assert_eq!(r3, DATA3);
80    }
81}
run/run.rs

run/
run.rs
