robot-rust 0.2.1

The robot-rust project, bindings for wpilib, ctre, and rev in rust.
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

use std::collections::HashMap;

use crate::ffi;
use crate::logging::{now, str_to_wpi};
use crate::prelude::*;

pub struct Logger {
    datalog: *mut ffi::WPI_DataLog,
    // TODO: Looking up in hashmap is inefficient
    entries: HashMap<String, i32>,
}

impl Logger {
    pub fn new(filename: &str) -> Self {
        let mut filename = str_to_wpi(filename);
        let mut empty = str_to_wpi("");

        let mut error = 0;
        let datalog =
            unsafe { ffi::WPI_DataLog_CreateWriter(&mut filename, &mut error, &mut empty) };

        Self {
            datalog,
            entries: HashMap::new(),
        }
    }

    pub fn append_value<T: Loggable>(&mut self, path: &str, value: T) {
        let entry = *self.entries.entry(path.to_string()).or_insert_with(|| {
            let mut name = str_to_wpi(path);
            let mut typ = str_to_wpi(T::typ());
            let mut empty = str_to_wpi("");

            unsafe { ffi::WPI_DataLog_Start(self.datalog, &mut name, &mut typ, &mut empty, now()) }
        });

        T::append(self.datalog, entry, now(), value);
    }

    pub fn append<T: AutoLog>(&mut self, path: &str, value: T) {
        value.append_to(self, path);
    }
}

impl Drop for Logger {
    fn drop(&mut self) {
        unsafe {
            ffi::WPI_DataLog_Release(self.datalog);
        }
    }
}