1#[cfg(feature = "profile")]
 2use colored::Colorize;
 3#[cfg(feature = "profile")]
 4use core::sync::atomic::AtomicUsize;
 5#[cfg(feature = "profile")]
 6use core::sync::atomic::Ordering;
 7#[cfg(feature = "profile")]
 8use std::time::Instant;
 9
10#[cfg(feature = "profile")]
11pub static CALL_DEPTH: AtomicUsize = AtomicUsize::new(0);
12
13#[cfg(feature = "profile")]
14pub struct Timer {
15    label: String,
16    timer: Instant,
17}
18
19#[cfg(feature = "profile")]
20impl Timer {
21    #[inline(always)]
22    pub fn new(label: &str) -> Self {
23        let timer = Instant::now();
24        CALL_DEPTH.fetch_add(1, Ordering::Relaxed);
25        let star = "* ";
26        println!(
27            "{:indent$}{}{}",
28            "",
29            star,
30            label.yellow().bold(),
31            indent = 2 * CALL_DEPTH.fetch_add(0, Ordering::Relaxed)
32        );
33        Self {
34            label: label.to_string(),
35            timer,
36        }
37    }
38
39    #[inline(always)]
40    pub fn stop(&self) {
41        let duration = self.timer.elapsed();
42        let star = "* ";
43        println!(
44            "{:indent$}{}{} {:?}",
45            "",
46            star,
47            self.label.blue().bold(),
48            duration,
49            indent = 2 * CALL_DEPTH.fetch_add(0, Ordering::Relaxed)
50        );
51        CALL_DEPTH.fetch_sub(1, Ordering::Relaxed);
52    }
53
54    #[inline(always)]
55    pub fn print(msg: &str) {
56        CALL_DEPTH.fetch_add(1, Ordering::Relaxed);
57        let star = "* ";
58        println!(
59            "{:indent$}{}{}",
60            "",
61            star,
62            msg.to_string().green().bold(),
63            indent = 2 * CALL_DEPTH.fetch_add(0, Ordering::Relaxed)
64        );
65        CALL_DEPTH.fetch_sub(1, Ordering::Relaxed);
66    }
67}
68
69#[cfg(not(feature = "profile"))]
70pub struct Timer {
71    _label: String,
72}
73
74#[cfg(not(feature = "profile"))]
75impl Timer {
76    #[inline(always)]
77    pub fn new(label: &str) -> Self {
78        Self {
79            _label: label.to_string(),
80        }
81    }
82
83    #[inline(always)]
84    pub fn stop(&self) {}
85
86    #[inline(always)]
87    pub fn print(_msg: &str) {}
88}