doxygen_filter.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
import sys
import os
import re
 
image_path = 'ioplots'
files = os.listdir(image_path)
 
# list of modules with I/O plots, _without_ "Module" suffix
modules_with_plots = [f[:-4] for f in files if re.match(r'.*.png', f)]
 
# corresponding header files, in lower case
headers = [m.lower() + 'module.h' for m in modules_with_plots]
 
filename = sys.argv[1]
subdirs = filename[len(os.path.abspath(os.curdir)) + 1:].split(os.path.sep)
group = subdirs[0]
if 'modules' in subdirs:
    group += '_modules'
elif 'dataobjects' in subdirs:
    group += '_dataobjects'
 
found_idx = None
try:
    found_idx = headers.index(os.path.basename(filename).lower())
    # ok, we've got a header with corresponding plot
    module = modules_with_plots[found_idx]
    classname = module + 'Module'
except:
    pass
 
belle2ns = False
for line in open(filename):
    try:
        line = line.replace('## ', '## ')
    except IOError:
        pass  # doxygen closes pipe for whatever reason
    if found_idx is not None:
        # TODO: also allow other whitespace after class?
        if re.match(r'.*class ' + classname + '.*', line):
            # found class declaration, add comment before it
            print('''/**
* \image html ''' + module + '''.png
*/''')
    try:
        if re.match(r'^}.*', line) and belle2ns:
            belle2ns = False
            print('  /*! @} */')
    except IOError:
        pass  # doxygen closes pipe for whatever reason
    try:
            print('/**')
            sys.stdout.write(line)  # print raw, without added \n
    except IOError:
        pass  # doxygen closes pipe for whatever reason
    try:
        if re.match(r'namespace Belle2 {.*', line):
            belle2ns = True
            print("""  /**
   * @addtogroup %s
   * @{
   */""" % group)
    except IOError:
        pass  # doxygen closes pipe for whatever reason
 
# avoid some further errors with lost stdout/stderr
try:
    sys.stdout.close()
except:
    pass
try:
    sys.stderr.close()
except:
    pass