xref: /freebsd/contrib/cortex-strings/scripts/libplot.py (revision 8c4282b370bd66908b45b6a223226a9fc2b69d57)

"""Shared routines for the plotters."""

import fileinput
import collections

Record = collections.namedtuple('Record', 'variant function bytes loops src_alignment dst_alignment run_id elapsed rest')


def make_colours():
    return iter('m b g r c y k pink orange brown grey'.split())

def parse_value(v):
    """Turn text into a primitive"""
    try:
        if '.' in v:
            return float(v)
        else:
            return int(v)
    except ValueError:
        return v

def create_column_tuple(record, names):
    cols = [getattr(record, name) for name in names]
    return tuple(cols)

def unique(records, name, prefer=''):
    """Return the unique values of a column in the records"""
    if type(name) == tuple:
        values = list(set(create_column_tuple(x, name) for x in records))
    else:
        values = list(set(getattr(x, name) for x in records))

    if not values:
        return values
    elif type(values[0]) == str:
        return sorted(values, key=lambda x: '%-06d|%s' % (-prefer.find(x), x))
    else:
        return sorted(values)

def alignments_equal(alignments):
    for alignment in alignments:
        if alignment[0] != alignment[1]:
            return False
    return True

def parse_row(line):
    return Record(*[parse_value(y) for y in line.split(':')])

def parse():
    """Parse a record file into named tuples, correcting for loop
    overhead along the way.
    """
    records = [parse_row(x) for x in fileinput.input()]

    # Pull out any bounce values
    costs = {}

    for record in [x for x in records if x.function=='bounce']:
        costs[(record.bytes, record.loops)] = record.elapsed

    # Fix up all of the records for cost
    out = []

    for record in records:
        if record.function == 'bounce':
            continue

        cost = costs.get((record.bytes, record.loops), None)

        if not cost:
            out.append(record)
        else:
            # Unfortunately you can't update a namedtuple...
            values = list(record)
            values[-2] -= cost
            out.append(Record(*values))

    return out