#!/usr/bin/env python

import CppHeaderParser
import sys

def generate_enum(fname):
    hdr = CppHeaderParser.CppHeader(fname)
    with open(fname[:-2] + '_enum.h', 'w') as f:
        f.write('#include "' + fname + '"\n')
        for enum in hdr.enums:
            f.write('const char* ToString(' + enum['name'] + ' val) {\n')
            f.write('   switch(val) {\n')
            for value in enum['values']:
                f.write('       case ' + value['name'] + ': return "' + value['name'] + '";\n')
            f.write('   }\n')
            f.write('}\n')

if __name__ == '__main__':
    if len(sys.argv) != 2:
        print "Usage:", sys.argv[0], "header.h"
        exit(-1)
    generate_enum(sys.argv[1])