Added runtime measurement

README.md

@@ -5,7 +5,6 @@
  - $\alpha$-equivalence check
  - Prettyprint functions (rename bound variables)
  - Write a nice README
- - Handle or avoid recursion errors
  - Fix colors
  - Print macro content if only a macro is typed
 

lamb/runner.py

@@ -4,6 +4,7 @@ from prompt_toolkit import print_formatted_text as printf
 from prompt_toolkit.shortcuts import clear as clear_screen
 import enum
 import math
+import time
 
 import lamb
 
@@ -105,6 +106,7 @@ class Runner:
 		macro_expansions = 0
 
 		stop_reason = StopReason.MAX_EXCEEDED
+		start_time = time.time()
 
 		while (self.reduction_limit is None) or (i < self.reduction_limit):
 
@@ -138,13 +140,10 @@ class Runner:
 			# Clear reduction counter
 			print(" " * round(14 + math.log10(i)), end = "\r")
 
-		if (
+		out_text = [
-				stop_reason == StopReason.BETA_NORMAL or
+			("class:result_header", f"\nRuntime: "),
-				stop_reason == StopReason.LOOP_DETECTED
+			("class:text", f"{time.time() - start_time:.03f} seconds"),
-			):
-			out_str = str(new_node) # type: ignore
 
-			printf(FormattedText([
 			("class:result_header", f"\nExit reason: "),
 			stop_reason.value,
 
@@ -152,23 +151,19 @@ class Runner:
 			("class:text", str(macro_expansions)),
 
 			("class:result_header", f"\nReductions: "),
-				("class:text", str(i)),
+			("class:text", str(i))
-
+		]
 
+		if (stop_reason == StopReason.BETA_NORMAL or stop_reason == StopReason.LOOP_DETECTED):
+			out_text += [
 				("class:result_header", "\n\n    => "),
-				("class:text", out_str),
+				("class:text", str(new_node)), # type: ignore
-			]), style = lamb.utils.style)
+			]
-		else:
-			printf(FormattedText([
-				("class:result_header", f"\nExit reason: "),
-				stop_reason.value,
 
-				("class:result_header", f"\nMacro expansions: "),
+		printf(
-				("class:text", str(macro_expansions)),
+			FormattedText(out_text),
-
+			style = lamb.utils.style
-				("class:result_header", f"\nReductions: "),
+		)
-				("class:text", str(i)),
-			]), style = lamb.utils.style)
 
 	def save_macro(
 			self,