Merge pull request #249 from mikmart/compilation-time

b: Track and report compilation times

Author: rexim

Makefile

@@ -24,6 +24,7 @@ RSS=\
 	$(SRC)/lexer.rs \
 	$(SRC)/nob.rs \
 	$(SRC)/targets.rs \
+	$(SRC)/time.rs \
 	$(SRC)/jim.rs \
 	$(SRC)/jimp.rs \
 	$(SRC)/codegen/gas_aarch64.rs \
@@ -38,6 +39,7 @@ POSIX_OBJS=\
 	$(BUILD)/glob.posix.o \
 	$(BUILD)/libc.posix.o \
 	$(BUILD)/arena.posix.o \
+	$(BUILD)/time.posix.o \
 	$(BUILD)/jim.posix.o \
 	$(BUILD)/jimp.posix.o \
 	$(BUILD)/shlex.posix.o \
@@ -48,6 +50,7 @@ MINGW32_OBJS=\
 	$(BUILD)/glob.mingw32.o \
 	$(BUILD)/libc.mingw32.o \
 	$(BUILD)/arena.mingw32.o \
+	$(BUILD)/time.mingw32.o \
 	$(BUILD)/jim.mingw32.o \
 	$(BUILD)/jimp.mingw32.o \
 	$(BUILD)/shlex.mingw32.o \

src/b.rs

@@ -35,6 +35,7 @@ pub mod codegen;
 pub mod lexer;
 pub mod targets;
 pub mod ir;
+pub mod time;
 pub mod shlex;
 
 use core::ffi::*;
@@ -50,6 +51,7 @@ use arena::Arena;
 use targets::*;
 use lexer::{Lexer, Loc, Token};
 use ir::*;
+use time::Instant;
 use codegen::*;
 use shlex::*;
 
@@ -1302,7 +1304,9 @@ pub unsafe fn main(mut argc: i32, mut argv: *mut*mut c_char) -> Option<()> {
             sb_appendf(&mut sb, c!("%s"), input_path);
         }
         da_append(&mut sb, 0);
-        log(Log_Level::INFO, c!("compiling files %s"), sb.items);
+        log(Log_Level::INFO, c!("compiling %zu files: %s"), input_paths.count, sb.items);
+
+        let compilation_start = Instant::now();
 
         let mut input: String_Builder = zeroed();
 
@@ -1324,15 +1328,17 @@ pub unsafe fn main(mut argc: i32, mut argv: *mut*mut c_char) -> Option<()> {
 
             if find_var_deep(&mut c.vars, used_global.name).is_null() {
                 diagf!(used_global.loc, c!("ERROR: could not find name `%s`\n"), used_global.name);
-                bump_error_count(&mut c);
+                bump_error_count(&mut c)?;
             }
         }
 
         scope_pop(&mut c.vars);          // end global scope
 
         if c.error_count > 0 {
-            return None
+            return None;
         }
+
+        log(Log_Level::INFO, c!("compilation took %.3fs"), compilation_start.elapsed().as_secs_f64());
     }
 
     let mut output: String_Builder = zeroed();

src/time.rs

@@ -0,0 +1,29 @@
+use core::time::Duration;
+
+#[derive(Clone, Copy, PartialEq, PartialOrd)]
+pub struct Instant {
+    nanos: u64,
+}
+
+impl Instant {
+    pub unsafe fn now() -> Instant {
+        extern "C" {
+            fn nanos_since_unspecified_epoch() -> u64;
+        }
+        Instant {
+            nanos: nanos_since_unspecified_epoch(),
+        }
+    }
+
+    pub unsafe fn elapsed(self) -> Duration {
+        Instant::now().duration_since(self)
+    }
+
+    pub unsafe fn duration_since(self, earlier: Instant) -> Duration {
+        if self > earlier {
+            Duration::from_nanos(self.nanos - earlier.nanos)
+        } else {
+            Duration::ZERO
+        }
+    }
+}

thirdparty/time.c

@@ -0,0 +1,32 @@
+#include <time.h>
+#include <stdint.h>
+
+#ifdef _WIN32
+#    define WIN32_LEAN_AND_MEAN
+#    include <windows.h>
+#endif // _WIN32
+
+#define NANOS_PER_SEC 1000000000
+
+// The maximum time span representable is 584 years.
+uint64_t nanos_since_unspecified_epoch() {
+#ifdef _WIN32
+    LARGE_INTEGER Time;
+    QueryPerformanceCounter(&Time);
+
+    static LARGE_INTEGER Frequency = {0};
+    if (Frequency.QuadPart == 0) {
+        QueryPerformanceFrequency(&Frequency);
+    }
+
+    uint64_t Secs  = Time.QuadPart / Frequency.QuadPart;
+    uint64_t Nanos = Time.QuadPart % Frequency.QuadPart * NANOS_PER_SEC / Frequency.QuadPart;
+    return NANOS_PER_SEC * Secs + Nanos;
+#else
+    struct timespec ts;
+    clock_gettime(CLOCK_MONOTONIC, &ts);
+
+    return NANOS_PER_SEC * ts.tv_sec + ts.tv_nsec;
+#endif // _WIN32
+}
+// TODO: Consider making this a part of https://github.com/tsoding/nob.h