import os
import csv
import re
from collections import defaultdict
import gen_level

class TLData:
    """Read a toplev output CSV file.

   Exported:
    times[n] All time stamps
    vals[n]  All values, as dicts mapping (name, cpu)->float
    levels{name} All levels (includes metrics), name->list of fields
    units{name}  All units, name->unit
    headers(set) All headers (including metrics)
    metrics(set) All metrics
    helptxt[col] All help texts.
    cpus(set)    All CPUs
    """

    def __init__(self, fn, verbose=False):
        self.times = []
        self.vals = []
        self.fn = fn
        self.levels = defaultdict(set)
        self.metrics = set()
        self.headers = set()
        self.mtime = None
        self.helptxt = {}
        self.cpus = set()
        self.verbose = verbose
        self.units = {}

    def update(self):
        mtime = os.path.getmtime(self.fn)
        if self.mtime == mtime:
            return
        self.mtime = mtime
        csvf = csv.reader(open(self.fn, 'r'))
        prevts = None
        val = {}
        for r in csvf:
            if r[0].strip().startswith("#"):
                continue
            if r[0] == "Timestamp" or r[0] == "CPUs":
                continue
            # 1.001088024,C1,Frontend_Bound,42.9,% Slots,,frontend_retired.latency_ge_4:pp,0.0,100.0,<==,Y
            if re.match(r'[CS]?\d+.*', r[1]):
                ts, cpu, name, pct, unit, helptxt = r[0], r[1], r[2], r[3], r[4], r[5]
            else:
                ts, name, pct, unit, helptxt = r[0], r[1], r[2], r[3], r[4]
                cpu = None
            key = (name, cpu)
            ts, pct = float(ts), float(pct.replace("%", ""))
            if name not in self.helptxt or self.helptxt[name] == "":
                self.helptxt[name] = helptxt
            if unit.endswith("<"):
                unit = unit[:-2]
                if not self.verbose:
                    continue
            self.units[name] = unit
            if prevts and ts != prevts:
                self.times.append(prevts)
                self.vals.append(val)
                val = {}
            val[key] = pct
            n = gen_level.level_name(name)
            if cpu:
                self.cpus.add(cpu)
            self.headers.add(name)
            if gen_level.is_metric(name):
                self.metrics.add(n)
            self.levels[n].add(name)
            prevts = ts
        if len(val.keys()) > 0:
            self.times.append(prevts)
            self.vals.append(val)

early_plots = ["TopLevel", "CPU utilization", "Power", "Frequency", "CPU-METRIC"]

def sort_key(i, data):
    if i in early_plots:
        return early_plots.index(i)
    if i in data.metrics:
        return 30
    return list(data.levels.keys()).index(i)

def level_order(data):
    """Return plot order of all levels."""
    return sorted(data.levels.keys(), key=lambda a: sort_key(a, data))