tally.py

from collections import OrderedDict
from omv.common.inout import trim_path


class Tallyman(object):
    def __init__(self, mepomt):
        self.omt = mepomt.omt_path
        self.mep = mepomt.mep_path
        self.engine = mepomt.engine
        self.modelpath = mepomt.modelpath
        self.experiments = {}
        self.report_passing_if_no_exps = False

    def all_passed(self):
        alltrue = True if any(self.experiments) else self.report_passing_if_no_exps
        for res in self.experiments.values():
            alltrue = alltrue and all(res.values())
        return alltrue

    def add_experiment(self, exp, results):
        self.experiments[exp.name] = results

    def __lt__(self, other):
        if other.mep is None:
            return True
        return self.omt < other.omt

    def serialize(self):
        s = OrderedDict({"engine": self.engine})
        s["MEP file"] = self.mep
        s["OMT file"] = self.omt
        s["Model path"] = self.modelpath
        s["Experiments"] = self.experiments
        s["All Passed"] = self.all_passed()
        # return dump(s, default_flow_style=False)
        return s

    def __repr__(self):
        return str(self.serialize())


class TallyHolder(object):
    tallies = {}
    all_engines = []

    def add(self, tally):
        if tally.engine not in self.all_engines:
            self.all_engines.append(tally.engine)

        mp = trim_path(tally.modelpath)

        if mp not in self.tallies:
            self.tallies[mp] = {}

        mptallies = self.tallies[mp]

        if tally.engine not in mptallies:
            mptallies[tally.engine] = []

        mptallies[tally.engine].append(tally)

    def summary(self):
        width1 = 60
        border = "+" + "-" * width1 + "+"
        header = "|" + " " * width1 + "| "

        totals = {}
        for engine in self.all_engines:
            border += "-" * (len(engine) + 2) + "+"
            header += engine + " | "
            totals[engine] = 0

        summary = "%s\n%s\n%s\n" % (border, header, border)

        for mp in self.tallies.keys():
            mp_ = mp
            max = width1 - 2
            if len(mp_) > max:
                pre = 8
                mp_ = "%s(...)%s" % (mp_[:pre], mp_[-1 * (max - pre - 6) :])

            summary += "| " + mp_ + " " * (width1 - len(mp_) - 2) + " |   "

            mptallies = self.tallies[mp]

            for engine in self.all_engines:
                if engine not in mptallies:
                    info = "%s" % (" ")
                    summary += " " * (len(engine) - len(info) - 2) + info + " |   "
                else:
                    tals = mptallies[engine]
                    for t in tals:
                        if t.engine == engine:
                            info = "%s" % (len(t.experiments))
                            if not t.mep:
                                info += "d"
                            if not t.all_passed():
                                info += "X"
                            totals[engine] += len(t.experiments)
                            summary += (
                                " " * (len(engine) - len(info) - 2) + info + " |   "
                            )
            summary += "\n"

        summary += border + "\n" + "| Totals: " + " " * (width1 - 9) + "|   "

        for engine in self.all_engines:
            info = "%s" % (totals[engine])
            summary += " " * (len(engine) - len(info) - 2) + info + " |   "

        summary += "\n" + border
        return summary