# Copyright (c) 2005# The Regents of The University of Michigan# All Rights Reserved## This code is part of the M5 simulator, developed by Nathan Binkert,# Erik Hallnor, Steve Raasch, and Steve Reinhardt, with contributions# from Ron Dreslinski, Dave Greene, Lisa Hsu, Kevin Lim, Ali Saidi,# and Andrew Schultz.## Permission is granted to use, copy, create derivative works and# redistribute this software and such derivative works for any purpose,# so long as the copyright notice above, this grant of permission, and# the disclaimer below appear in all copies made; and so long as the# name of The University of Michigan is not used in any advertising or# publicity pertaining to the use or distribution of this software# without specific, written prior authorization.## THIS SOFTWARE IS PROVIDED AS IS, WITHOUT REPRESENTATION FROM THE# UNIVERSITY OF MICHIGAN AS TO ITS FITNESS FOR ANY PURPOSE, AND WITHOUT# WARRANTY BY THE UNIVERSITY OF MICHIGAN OF ANY KIND, EITHER EXPRESS OR# IMPLIED, INCLUDING WITHOUT LIMITATION THE IMPLIED WARRANTIES OF# MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE REGENTS OF# THE UNIVERSITY OF MICHIGAN SHALL NOT BE LIABLE FOR ANY DAMAGES,# INCLUDING DIRECT, SPECIAL, INDIRECT, INCIDENTAL, OR CONSEQUENTIAL# DAMAGES, WITH RESPECT TO ANY CLAIM ARISING OUT OF OR IN CONNECTION# WITH THE USE OF THE SOFTWARE, EVEN IF IT HAS BEEN OR IS HEREAFTER# ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.from orderdict import orderdictimport outputclass ProfileData(object):    def __init__(self):        self.data = {}        self.total = {}        self.runs = orderdict()        self.runlist = []    def addvalue(self, run, cpu, symbol, value):        value = float(value)        self.data[run, cpu, symbol] = self.getvalue(run, cpu, symbol) + value        self.total[run, cpu] = self.gettotal(run, cpu) + value        if run not in self.runs:            self.runs[run] = orderdict()        if cpu not in self.runs[run]:            self.runs[run][cpu] = {}        if symbol not in self.runs[run][cpu]:            self.runs[run][cpu][symbol] = 0        self.runs[run][cpu][symbol] += value    def getvalue(self, run, cpu, symbol):        return self.data.get((run, cpu, symbol), 0)    def gettotal(self, run, cpu):        return self.total.get((run, cpu), 0)class Profile(object):    default_order = ['ste', 'hte', 'htd', 'ocm', 'occ', 'ocp']    # This list controls the order of values in stacked bar data output    default_categories = [ 'interrupt',                           'driver',                           'stack',                           'bufmgt',                           'copy',                           'user',                           'other',                           'idle']    def __init__(self, run_order=[], categories=[], stacknames=[]):        if not run_order:            run_order = Profile.default_order        if not categories:            categories = Profile.default_categories                    self.run_order = run_order        self.categories = categories        self.rcategories = []        self.rcategories.extend(categories)        self.rcategories.reverse()        self.stacknames = stacknames        self.prof = ProfileData()        self.categorize = True        self.showidle = True        self.maxsymlen = 0    def category(self, symbol):        from categories import categories, categories_re        if categories.has_key(symbol):            return categories[symbol]        for regexp, cat in categories_re:            if regexp.match(symbol):                return cat        return 'other'    # Parse input file and put the results in the given run and cpu    def parsefile(self, run, cpu, filename):        fd = file(filename)        for line in fd:            (symbol, count) = line.split()            if symbol == "0x0":                continue            count = int(count)            if self.categorize:                symbol = self.category(symbol)                if symbol == 'idle' and not self.showidle:                    continue                if symbol not in self.categories:                    symbol = 'other'            self.maxsymlen = max(self.maxsymlen, len(symbol))            self.prof.addvalue(run, cpu, symbol, count)        fd.close()    # Read in files    def inputdir(self, directory):        import os, os.path, re        from os.path import expanduser, join as joinpath        directory = expanduser(directory)        label_ex = re.compile(r'm5prof\.(.*)')        for root,dirs,files in os.walk(directory):            for name in files:                match = label_ex.match(name)                if not match:                    continue                filename = joinpath(root, name)                prefix = os.path.commonprefix([root, directory])                dirname = root[len(prefix)+1:]                self.parsefile(dirname, match.group(1), filename)    def get(self, job, stat):        if job.system is None:            raise AttributeError, 'The job must have a system set'                cpu =  '%s.full0' % job.system        values = []        for cat in self.categories:            values.append(self.prof.getvalue(job.name, cpu, cat))        return values