Skip to content
Snippets Groups Projects
Commit cdfd5b09 authored by Himanen, Lauri (himanel1)'s avatar Himanen, Lauri (himanel1)
Browse files

Added profiling script.

parent 2091f384
Branches
Tags
No related merge requests found
...@@ -393,7 +393,9 @@ class MainParser(HierarchicalParser): ...@@ -393,7 +393,9 @@ class MainParser(HierarchicalParser):
process just override this method. process just override this method.
""" """
# Initialize the parser builder # Initialize the parser builder
parserBuilder = SimpleParserBuilder(self.root_matcher, self.backend.metaInfoEnv(), self.metainfo_to_keep, units=self.parser_context.units) default_units = self.parser_context.default_units
metainfo_units = self.parser_context.metainfo_units
parserBuilder = SimpleParserBuilder(self.root_matcher, self.backend.metaInfoEnv(), self.metainfo_to_keep, default_units=default_units, metainfo_units=metainfo_units)
# Verify the metainfo # Verify the metainfo
if not parserBuilder.verifyMetaInfo(sys.stderr): if not parserBuilder.verifyMetaInfo(sys.stderr):
...@@ -415,8 +417,9 @@ class MainParser(HierarchicalParser): ...@@ -415,8 +417,9 @@ class MainParser(HierarchicalParser):
defaultDataCachingLevel=self.default_data_caching_level, defaultDataCachingLevel=self.default_data_caching_level,
defaultSectionCachingLevel=self.default_section_caching_level, defaultSectionCachingLevel=self.default_section_caching_level,
onClose=onClose, onClose=onClose,
superBackend=self.backend) superBackend=self.backend,
self.caching_backend.units = self.parser_context.units default_units=default_units,
metainfo_units=metainfo_units)
# Compile the SimpleMatcher tree # Compile the SimpleMatcher tree
parserBuilder.compile() parserBuilder.compile()
...@@ -452,10 +455,11 @@ class MainParser(HierarchicalParser): ...@@ -452,10 +455,11 @@ class MainParser(HierarchicalParser):
class ParserContext(object): class ParserContext(object):
"""Contains everything needed to instantiate a parser implementation. """Contains everything needed to instantiate a parser implementation.
""" """
def __init__(self, files=None, metainfo_to_keep=None, backend=None, version_id=None, main_file=None, units=None): def __init__(self, files=None, metainfo_to_keep=None, backend=None, version_id=None, main_file=None, default_units=None, metainfo_units=None):
self.files = files self.files = files
self.version_id = version_id self.version_id = version_id
self.metainfo_to_keep = metainfo_to_keep self.metainfo_to_keep = metainfo_to_keep
self.backend = backend self.backend = backend
self.main_file = main_file self.main_file = main_file
self.units = units self.default_units = default_units
self.metainfo_units = metainfo_units
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
import cProfile
import pstats
from run_tests import get_results
def profile_energy_force():
"""Used to profile the CPU usage in parsing RUN_TYPE ENERGY_FORCE.
"""
profile = cProfile.Profile()
profile.run('get_results("energy_force", "section_run")')
stats = pstats.Stats(profile)
# stats.strip_dirs()
stats.sort_stats("cumulative")
stats.print_stats(30)
if __name__ == "__main__":
profile_energy_force()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment