adder-codec-rs 0.4.9

Encoder/transcoder/decoder for raw and compressed ADΔER (Address, Decimation, Δt Event Representation) streams. Includes a transcoder for casting either framed or event video into an ADΔER representation in a manner which preserves the temporal resolution of the source.
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
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

use adder_codec_core::codec::rate_controller::Crf;
use adder_codec_core::codec::{EncoderOptions, EncoderType};
use adder_codec_core::SourceCamera::Dvs;
use adder_codec_core::{PixelMultiMode, TimeMode};
use adder_codec_rs::transcoder::source::prophesee::Prophesee;
use adder_codec_rs::transcoder::source::video::{Source, SourceError, VideoBuilder};
use adder_codec_rs::utils::viz::ShowFeatureMode;
use clap::Parser;
use std::fs::File;
use std::io::BufWriter;

#[derive(Parser, Debug, Default, serde::Deserialize)]
#[clap(author, version, about, long_about = None)]
pub struct MyArgs {
    /// Number of ticks per input interval
    #[clap(short, long, default_value_t = 1)]
    pub ref_time: u32,

    /// Max number of ticks for first event at a new intensity
    #[clap(short, long, default_value_t = 2)]
    pub delta_t_max: u32,

    /// Path to input file
    #[clap(short, long, default_value = "./in.dat")]
    pub input: String,

    /// Path to output events file
    #[clap(long, default_value = "")]
    pub output: String,

    #[clap(long, default_value_t = 3)]
    pub crf: u8,

    /// Number of threads to use. If not provided, will default to the number of cores on the
    /// system.
    #[clap(long, default_value_t = 8)]
    pub thread_count: u8,

    #[clap(short, long, action)]
    pub features: bool,
}

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let args: MyArgs = MyArgs::parse();

    let mut prophesee_source: Prophesee<BufWriter<File>> =
        Prophesee::new(args.ref_time, args.input)?.crf(args.crf);
    let adu_interval =
        (prophesee_source.get_video_ref().state.tps as f32 / args.ref_time as f32) as usize;
    let plane = prophesee_source.get_video_ref().state.plane;

    let writer = BufWriter::new(File::create(args.output)?);
    prophesee_source = *prophesee_source.write_out(
        Dvs,
        TimeMode::AbsoluteT,
        PixelMultiMode::Collapse,
        Some(adu_interval),
        EncoderType::Compressed,
        EncoderOptions {
            event_drop: Default::default(),
            event_order: Default::default(),
            crf: Crf::new(Some(args.crf), plane),
        },
        writer,
    )?;
    prophesee_source.get_video_mut().update_detect_features(
        args.features,
        ShowFeatureMode::Off,
        true,
        true,
    );
    // prophesee_source
    //     .get_video_mut()
    //     .encoder
    //     .options
    //     .crf
    //     .override_feature_c_radius(2);

    let pool = rayon::ThreadPoolBuilder::new()
        .num_threads(args.thread_count.into())
        .build()
        .unwrap();

    loop {
        match prophesee_source.consume() {
            Ok(_) => {}
            Err(SourceError::Open) => return Ok(()),
            Err(e) => {
                eprintln!("Consume Error: {e:?}");
                prophesee_source.get_video_mut().end_write_stream()?;
                return Ok(());
            }
        };
    }

    Ok(())
}