Program evaluation benchmarks

Author: WalkerRout

examples/test_fn.lang

@@ -2,13 +2,27 @@ let add = fn(x, y) {
 	x + y
 };
 
+let sub = fn(x, y) {
+	x - y
+};
+
+let mul = fn(x, y) {
+	x * y
+};
+
+let div = fn(x, y) {
+	x / y
+};
+
 let t = true;
 let f = false;
 
-let res = if(add(1, 2) == 3) {
+let res = if(div(20, 2) == 10) {
+	let f = if(div(500, 10) == 50) { f };
 	!f
 } else {
+	let t = if(mul(5, 5) == 25) { t };
 	!t
 };
 
-res
+res

src/interpreter.nim

@@ -6,6 +6,7 @@ import lexer
 import parser
 import obj
 import eval
+import util
 
 proc evaluate(path: string): string =
   result = "Unable to evaluate program!"
@@ -21,7 +22,8 @@ proc evaluate(path: string): string =
   parser.check_parser_errors()
 
   var env = obj.environment()
-  let obj = eval.eval(program, env)
+  let obj = util.benchmark "program evaluation":
+    eval.eval(program, env)
   if obj != nil:
     result = obj.inspect()
 

src/repl.nim

@@ -2,6 +2,7 @@ import lexer
 import parser
 import eval
 import obj
+import util
 
 type Status {.pure.} = enum
   stBreak
@@ -33,7 +34,8 @@ proc repl*() =
     let program = parser.parse_program()
     parser.check_parser_errors()
 
-    let obj = eval.eval(program, env)
+    let obj = util.benchmark "eval repl program":
+      eval.eval(program, env)
     if obj != nil:
       echo obj.inspect()
 

src/util.nim

@@ -1,6 +1,22 @@
+import std/times
+import std/os
+import std/strutils
 
 template dynamic_cast*[T](root: untyped): untyped =
   try:
     T(root)
   except:
-    nil
+    nil
+
+template benchmark*(bench_name: string, statements: untyped): untyped =
+  block:
+    let t0 = epochTime()
+
+    let result = statements
+
+    let t1 = epochTime()
+    let dt = t1 - t0
+    let elapsed = strutils.formatFloat(dt, ffDecimal, 5)
+    echo "CPU Time [", bench_name, "] ", elapsed, "s"
+
+    result

tests/test_obj.nim

@@ -1,6 +1,6 @@
 import std/unittest
-import std/tables
-import std/strutils
+#import std/tables
+#import std/strutils
 
 import ../src/obj
 

tests/test_parser.nim

@@ -1,6 +1,6 @@
 import std/unittest
 import std/tables
-import std/strutils
+#import std/strutils
 import std/typetraits
 
 import ../src/parser