#! /usr/bin/env python
#
# This file is part of DisOrder.
# Copyright (C) 2007 Richard Kettlewell
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
import re,sys,os,string

def fatal(msg):
  sys.stderr.write("%s\n" % msg)
  sys.exit(1)

def sgmlquotechar(c):
  if c == '&' or c == '<' or ord(c) < 32 or ord(c) > 126:
    return "&#%d;" % ord(c)
  else:
    return c

def sgmlquote(s):
  return string.join(map(sgmlquotechar, s),'')

def line_count(f):
  return len(open(f, "r").readlines())

missing = {}
percent = {}
total_lines = 0
covered_lines = 0
args = sys.argv[1:]
htmldir = None
while len(args) > 0 and re.match("^--", args[0]):
  if args[0] == "--html":
    htmldir = args[1]
    args = args[2:]
  else:
    fatal("unknown option '%s'" % args[0])
for s in args:
  missing[s] = True

name = None
for line in sys.stdin:
  line = line[:-1]
  r = re.match("File ['`](?:.*/)?([^/]+.c)'", line)
  if r:
    name = r.group(1)
    if name in missing:
      del missing[name]
  r = re.match("Lines executed:([0-9\\.]+)% of ([0-9]+)", line)
  if r:
    if name:
      this_pc = float(r.group(1))
      this_lines = int(r.group(2))
      percent[name] = this_pc
      total_lines += this_lines
      covered_lines += this_lines * this_pc / 100.0
    name = None

for m in missing:
  percent[m] = 0
  total_lines += line_count(m)

def cmp(a,b):
  if percent[a] < percent[b]: return -1 
  elif percent[a] > percent[b]: return 1
  else: return 0

keys = percent.keys()
keys.sort(cmp)

if len(keys):
  for k in keys:
    print "%20s: %d%%" % (k, percent[k])
  print "Total coverage: %d%%" % (100 * (covered_lines / total_lines))

if htmldir is not None and len(keys):
  index = open(os.path.join(htmldir, "index.html"), "w")
  index.write("<html><head><title>gcov report</title>\n")
  index.write("<style type=\"text/css\">\n");
  index.write(".missing {\n");
  index.write("  color: red;\n");
  index.write("}\n");
  index.write(".bar {\n");
  index.write("  width: 400px;\n");
  index.write("}\n");
  index.write(".bar div {\n");
  index.write("  background-color: #00ff00;\n");
  index.write("  height: 1em;\n");
  index.write("}\n");
  index.write("</style>\n");
  index.write("<body><h1>gcov report</h1>\n")
  index.write("<table><tr><th>File</th><th colspan=2>Coverage</th></tr>\n")
  for k in keys:
    index.write("<tr>\n")
    if k in missing:
      index.write("<td><span class=missing>%s</span>\n" % sgmlquote(k))
    else:
      index.write("<td><a href=\"%s.html\">%s</a>\n" %
                  (sgmlquote(k), sgmlquote(k)))
    index.write("<td>%d%%\n" % percent[k])
    index.write("<td class=bar><div style=\"width: %d%%\"></div></td>\n"
                % int(percent[k]))
  index.write("</table>\n")
  index.write("<p>Total coverage: %d%%</p>\n" % (100 * (covered_lines / total_lines)))
  for k in keys:
    if k in missing:
      continue
    html = open(os.path.join(htmldir, "%s.html" % k), "w")
    html.write("<html><head><title>%s</title>\n" % sgmlquote(k))
    html.write("<body><h1>%s</h1>\n" % sgmlquote(k))
    html.write("<pre>")
    r = re.compile("^ *#####:.*")
    for line in open("%s.gcov" % k, "r"):
      if len(line) > 0 and line[-1] == '\n':
        line = line[:-1]
      if r.match(line):
        html.write("<span style='background-color:#ffff00'>%s</span>\n" % sgmlquote(line))
      else:
        html.write("%s\n" % sgmlquote(line))
    html.write("</pre>\n")
