merge LCOV file counters

lcov_merge.py

#!/usr/bin/python
# merge LCOV file counters

import sys

filedict = {}
currentitem = {}
currentfile = ''
for line in sys.stdin:
   subline = line[3:][:-1]
   if line.startswith('SF:'):
      executed = 0
      try:
         currentitem = filedict[subline]
      except KeyError:
         currentitem = {}
         currentitem['FN'] = {}
         currentitem['DA'] = {}
         filedict[subline] = currentitem
      currentfile = subline
   elif line.startswith('FN:'):
      currentitem['FN'][subline] = 1
   elif line.startswith('DA:'):
      elems = subline.split(',')
      try:
         try:
            currentitem['DA'][elems[0]] += int(elems[1])
         except KeyError:
            currentitem['DA'][elems[0]] = int(elems[1])
      except:
         continue
      if currentitem['DA'][elems[0]]:
         executed += 1
   elif line.startswith('LF:'):
      currentitem['LF'] = int(subline)
   elif line.startswith('LH:'):
      try:
         currentitem['LH'] = max(currentitem['LH'],int(subline))
      except KeyError:
         currentitem['LH'] = int(subline)
      except ValueError:
         currentitem['LH'] = executed
   elif line.startswith('end_of_record'):
      # sys.stderr.write('updated ' + currentfile + '\n')
      pass

for (sourcefile,item) in filedict.items():
   print 'SF:'+sourcefile
   for function in item['FN'].keys():
      print 'FN:'+function
   for (linenum, cov) in item['DA'].items():
      print 'DA:'+linenum+','+str(cov)
   print 'LF:'+str(item['LF'])
   print 'LH:'+str(item['LH'])
   print 'end_of_record'