#!/usr/bin/env python3 import argparse def main(): parser = argparse.ArgumentParser() parser.add_argument('--max_args', type=int, help='Max number of arguments to generate case statements for', required=True) parser.add_argument('--output', help='Output header file to include', required=True) args = parser.parse_args() output='' for i in range(args.max_args+1): output += 'case %d:\n'%(i) output += '((void (*)(kmp_int32 *, kmp_int32 *\n' for j in range(i): output += ', void *' if (j+1)%4 == 0: output += '\n' output += '))fn)(&global_tid, &bound_tid\n' for j in range(i): output += ', args[%d]'%(j) if (j+1)%4 == 0: output += '\n' output += ');\n' output += 'break;\n' with open(args.output, 'w') as f: print(output, file=f) if __name__ == "__main__": main()