device-driver-core 2.0.0

Internal compiler crate for the device-driver toolkit
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

use std::time::Duration;

use annotate_snippets::{Group, Level};
use device_driver_common::instant::Instant;
use device_driver_diagnostics::Diagnostic;
use device_driver_mir::PassTiming;

use crate::TimingsMode;

#[derive(Debug)]
pub struct Timings {
    mode: TimingsMode,
    lexer: Duration,
    parser: Duration,
    mir: Duration,
    mir_passes: Vec<PassTiming>,
    lir: Duration,
    codegen: Duration,
}

impl Timings {
    pub const fn new(mode: TimingsMode) -> Self {
        Self {
            mode,
            lexer: Duration::ZERO,
            parser: Duration::ZERO,
            mir: Duration::ZERO,
            mir_passes: Vec::new(),
            lir: Duration::ZERO,
            codegen: Duration::ZERO,
        }
    }

    pub fn start_lexer(&mut self) -> Timer<'_> {
        Timer::new(&mut self.lexer)
    }
    pub fn start_parser(&mut self) -> Timer<'_> {
        Timer::new(&mut self.parser)
    }
    pub fn start_mir(&mut self) -> Timer<'_> {
        Timer::new(&mut self.mir)
    }
    pub fn start_lir(&mut self) -> Timer<'_> {
        Timer::new(&mut self.lir)
    }
    pub fn start_codegen(&mut self) -> Timer<'_> {
        Timer::new(&mut self.codegen)
    }

    pub fn set_mir_timings(&mut self, mir_timings: Vec<PassTiming>) {
        if matches!(self.mode, TimingsMode::Verbose) {
            self.mir_passes = mir_timings;
        }
    }
}

impl Diagnostic for Timings {
    fn is_error(&self) -> bool {
        false
    }

    fn as_report<'a>(&'a self, _source: &'a str, _path: &'a str) -> Vec<Group<'a>> {
        [Level::INFO
            .primary_title("timings")
            .element(
                Level::INFO
                    .with_name(Some("lexer"))
                    .message(format!("  {:>8.3}ms", self.lexer.as_secs_f64() * 1000.0)),
            )
            .element(
                Level::INFO
                    .with_name(Some("parser"))
                    .message(format!(" {:>8.3}ms", self.parser.as_secs_f64() * 1000.0)),
            )
            .element(
                Level::INFO
                    .with_name(Some("mir"))
                    .message(format!("    {:>8.3}ms", self.mir.as_secs_f64() * 1000.0)),
            )
            .elements(self.mir_passes.iter().map(|pass| {
                Level::INFO.no_name().message(format!(
                    "- {}: {:0.3}ms",
                    pass.name.split("::").last().unwrap(),
                    pass.duration.as_secs_f64() * 1000.0
                ))
            }))
            .element(
                Level::INFO
                    .with_name(Some("lir"))
                    .message(format!("    {:>8.3}ms", self.lir.as_secs_f64() * 1000.0)),
            )
            .element(
                Level::INFO
                    .with_name(Some("codegen"))
                    .message(format!("{:>8.3}ms", self.codegen.as_secs_f64() * 1000.0)),
            )]
        .to_vec()
    }
}

pub struct Timer<'a> {
    start: Instant,
    result: &'a mut Duration,
}

impl<'a> Timer<'a> {
    pub fn new(result: &'a mut Duration) -> Self {
        Self {
            start: Instant::now(),
            result,
        }
    }
}

impl Drop for Timer<'_> {
    fn drop(&mut self) {
        *self.result = self.start.elapsed();
    }
}