grpc/third_party/upb/benchmarks/gen_upb_binary_c.py

import sys
import re

include = sys.argv[1]
msg_basename = sys.argv[2]
count = 1

m = re.search(r'(.*\D)(\d+)$', sys.argv[2])
if m:
  msg_basename = m.group(1)
  count = int(m.group(2))

print('''
#include "{include}"

char buf[1];

int main() {{
  upb_arena *arena = upb_arena_new();
  size_t size;
'''.format(include=include))

def RefMessage(name):
  print('''
  {{
    {name} *proto = {name}_parse(buf, 1, arena);
    {name}_serialize(proto, arena, &size);
  }}
  '''.format(name=name))

RefMessage(msg_basename)

for i in range(2, count + 1):
  RefMessage(msg_basename + str(i))

print('''
  return 0;
}''')