From 20afb205e36b50e85b53fa6db298892317706350 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Tue, 22 Feb 2022 15:02:48 +0100
Subject: [PATCH] Added perf-testing

---
 caliGraph.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/caliGraph.py b/caliGraph.py
index 703874d..dd2b282 100755
--- a/caliGraph.py
+++ b/caliGraph.py
@@ -1331,6 +1331,7 @@ def cliInterface():
     parser.add_argument('--dark', action="store_true")
     parser.add_argument('--v3d', action="store_true")
     parser.add_argument('--imgs', action="store_true")
+    parser.add_argument('--perf-test', action="store_true")
     cmds = parser.add_subparsers(required=True, dest='cmd')
 
     p_rec = cmds.add_parser('recommend', description="TODO", aliases=['rec'])
@@ -1370,6 +1371,24 @@ def cliInterface():
 
     args = parser.parse_args()
 
+    if args.perfTest:
+        perfTestCLI(args)
+    else:
+        mainCLI(args)
+
+def perfTestCLI(args):
+    from pycallgraph import PyCallGraph
+    from pycallgraph import Config
+    from pycallgraph import GlobbingFilter
+    from pycallgraph.output import GraphvizOutput
+    config = Config()
+    config.trace_filter = GlobbingFilter(exclude=[
+        "pycallgraph.*",
+    ])
+    with PyCallGraph(output=GraphvizOutput(output_file='perfTests/serve_httpd_' + str(int(time.time())) + '.png'), config=config):
+        mainCLI(args)
+
+def mainCLI(args):
     if args.cmd=="train":
         train(args.g, args.full)
         exit()