From 918738c98fb775394677c17476888239df73aae9 Mon Sep 17 00:00:00 2001 From: Thanassis Tsiodras 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