godot-bindings 0.5.2

Internal crate used by godot-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
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

/*
 * Copyright (c) godot-rust; Bromeon and contributors.
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

use std::fs::File;
use std::io::{BufWriter, Write};
use std::path::Path;
use std::time::{Duration, Instant};

pub struct StopWatch {
    last_instant: Instant,
    metrics: Vec<Metric>,
    lwidth: usize,
}

impl StopWatch {
    pub fn start() -> Self {
        Self {
            last_instant: Instant::now(),
            metrics: vec![],
            lwidth: 0,
        }
    }

    pub fn record(&mut self, what: impl Into<String>) {
        let now = Instant::now();
        let duration = now - self.last_instant;
        let name = what.into();

        self.last_instant = now;
        self.lwidth = usize::max(self.lwidth, name.len());
        self.metrics.push(Metric { name, duration });
    }

    pub fn write_stats_to(self, to_file: &Path) {
        let file = File::create(to_file).expect("failed to create stats file");
        let mut writer = BufWriter::new(file);

        // Accumulate total
        let mut total = Duration::ZERO;
        for metric in self.metrics.iter() {
            total += metric.duration;
        }
        let rwidth = log10(total.as_millis());
        let total_metric = Metric {
            name: "total".to_string(),
            duration: total,
        };

        // Write to file
        for metric in self.metrics.iter() {
            Self::write_metric(&mut writer, metric, self.lwidth, rwidth);
        }
        writeln!(&mut writer, "{}", "-".repeat(self.lwidth + rwidth + 5))
            .expect("failed to write to stats file");
        Self::write_metric(&mut writer, &total_metric, self.lwidth, rwidth);
    }

    fn write_metric(writer: &mut BufWriter<File>, metric: &Metric, lwidth: usize, rwidth: usize) {
        writeln!(
            writer,
            "{: >l$}: {: >r$} ms",
            metric.name,
            metric.duration.as_millis(),
            l = lwidth,
            r = rwidth,
        )
        .expect("failed to write to stats file");
    }
}

fn log10(n: u128) -> usize {
    std::iter::successors(Some(n), |&n| (n >= 10).then_some(n / 10)).count()
}

struct Metric {
    name: String,
    duration: Duration,
}