tools/scripts/scov_merge.py

#!/usr/bin/env python
#
# Merge mce serverity coverage file from several test cases. This can
# be used to check the coverage of several test cases.
#
# Copyright (C) 2009, Intel Corp.
#   Author: Huang Ying <ying.huang@intel.com>
#
# This file is released under the GPLv2.

import sys

def die(str):
    print str
    sys.exit(-1)

def die_on(cond, str):
    if cond:
        die(str)

class SCovLine(object):
    def __init__(self, l):
        object.__init__(self)
        self.parse(l)
    def parse(self, l):
        sep = l.find('\t')
        die_on(sep <= 0, 'input error format error')
        remain = l[sep:]
        scnt = l[:sep]
        count = int(scnt)
        self.count = count
        self.remain = remain
    def merge(self, gcl):
        self.count = self.count + gcl.count
    def write(self, of):
        if self.count == 0:
            of.write("0")
        else:
            of.write("1")
        of.write(self.remain)

def parse(f):
    return [SCovLine(l) for l in f]

def merge(gcls1, gcls2):
    for gcl1, gcl2 in zip(gcls1, gcls2):
        gcl1.merge(gcl2)

def scov_merge(fns, of):
    f = file(fns[0])
    gcls_base = parse(f)

    for fn in fns[1:]:
        f = file(fn)
        gcls = parse(f)
        merge(gcls_base, gcls)

    for gcl in gcls_base:
        gcl.write(of)

def usage():
    print 'Usage: %s <severities coverage files>' % (sys.argv[0])

if __name__ == '__main__':
    if len(sys.argv) <= 1:
        usage()
        exit -1
    scov_merge(sys.argv[1:], sys.stdout)