From 918738c98fb775394677c17476888239df73aae9 Mon Sep 17 00:00:00 2001
From: Thanassis Tsiodras <ttsiodras@gmail.com>
Date: Mon, 29 Jul 2019 14:41:30 +0200
Subject: [PATCH] Add profiling option, to track where the time is spent.

---
 dmt/aadl2glueC.py | 18 ++++++++++++++++++
 setup.py          |  2 +-
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/dmt/aadl2glueC.py b/dmt/aadl2glueC.py
index 638b146..9f89b07 100755
--- a/dmt/aadl2glueC.py
+++ b/dmt/aadl2glueC.py
@@ -452,6 +452,24 @@ def main() -> None:
         import pdb  # pragma: no cover pylint: disable=wrong-import-position,wrong-import-order
         pdb.set_trace()  # pragma: no cover
 
+    if "-profile" in sys.argv:
+        sys.argv.remove("-profile")
+        import cProfile
+        import pstats
+        import io
+        pr = cProfile.Profile()
+        pr.enable()
+        import atexit
+
+        def dumpSpeedData():
+            pr.disable()
+            s = io.StringIO()
+            sortby = 'cumulative'
+            ps = pstats.Stats(pr, stream=s).sort_stats(sortby)
+            ps.print_stats()
+            print(s.getvalue())
+        atexit.register(dumpSpeedData)
+
     if "-v" in sys.argv:
         import pkg_resources  # pragma: no cover
         version = pkg_resources.require("dmt")[0].version  # pragma: no cover
diff --git a/setup.py b/setup.py
index 4743506..263aff6 100755
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='dmt',
-    version="2.1.31",
+    version="2.1.32",
     packages=find_packages(),
     author='Thanassis Tsiodras',
     author_email='Thanassis.Tsiodras@esa.int',
-- 
GitLab