telemetry-parser 0.2.6

Library to parse real-time metadata embedded in video files or telemetry from other sources.
Documentation 
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

// SPDX-License-Identifier: MIT OR Apache-2.0
// Copyright © 2021 Adrian <adrian.eddy at gmail>

use quick_xml::events::Event;
use quick_xml::Reader;

#[allow(dead_code)]
pub fn parse_from_file(stream: &mut std::fs::File) -> std::io::Result<Metadata> {
    let ctx = mp4parse::read_mp4(stream)?;

    if let Some(Ok(md)) = ctx.metadata {
        if let Some(mp4parse::XmlBox::StringXmlBox(x)) = md.xml {
            let parsed_md = parse(&x[..]);
            log::info!("Model: {} {}, frames: {}", parsed_md.manufacturer, parsed_md.model, parsed_md.frame_count);
            return Ok(parsed_md);
        }
    }
    Err(std::io::ErrorKind::NotFound.into())
}

pub struct Metadata {
    pub manufacturer: String,
    pub model: String,
    pub frame_count: usize,
}

pub fn parse(data: &[u8]) -> Metadata {
    let mut reader = Reader::from_reader(data);
    let mut buf = Vec::new();

    let mut frame_count = 0usize;
    let mut model = String::new();
    let mut manufacturer = String::new();

    loop {
        match reader.read_event(&mut buf) {
            Ok(Event::Empty(ref e)) => {
                if e.name() == b"Duration" || e.name() == b"Device" {
                    for ox in e.attributes() {
                        if let Ok(x) = ox {
                            if x.key == b"value"        { frame_count = String::from_utf8_lossy(&x.value).parse::<usize>().unwrap(); }
                            if x.key == b"modelName"    { model = String::from_utf8_lossy(&x.value).into(); }
                            if x.key == b"manufacturer" { manufacturer = String::from_utf8_lossy(&x.value).into(); }
                        }
                    }
                }
            },
            Ok(Event::Eof) => break, // exits the loop when reaching end of file
            Err(e) => panic!("Error at position {}: {:?}", reader.buffer_position(), e),
            _ => (), // There are several other `Event`s we do not consider here
        }
        buf.clear();
    }
    Metadata { manufacturer, model, frame_count }
}