diff --git a/gem5-stat b/gem5-stat index 0a42619..cf1adff 100755 --- a/gem5-stat +++ b/gem5-stat @@ -1,14 +1,26 @@ #!/usr/bin/env python3 + import common -parser = self.get_argparse( - argparse_args={'description':'Get the value of a gem5 stat from the stats.txt file.'} -) -parser.add_argument( - 'stat', - default=None, - help='Python regexp matching the full stat name of interest', - nargs='?', -) -args = self.setup(parser) -stats = self.get_stats(kwargs['stat']) -print('\n'.join(stats)) + +class Main(common.LkmcCliFunction): + def __init__(self): + super().__init__( + defaults={ + 'print_time': False, + }, + description='''\ +Get the value of a gem5 stat from the stats.txt file. +''', + ) + self.add_argument( + 'stat', + help='Python regexp matching the full stat name of interest', + nargs='?', + ) + + def timed_main(self): + stats = self.get_stats(self.env['stat']) + print('\n'.join(stats)) + +if __name__ == '__main__': + Main().cli()