Size benchmarks are working pretty well.
diff --git a/benchmarks/gen_synthetic_protos.py b/benchmarks/gen_synthetic_protos.py
new file mode 100644
index 0000000..36d1f7d
--- /dev/null
+++ b/benchmarks/gen_synthetic_protos.py
@@ -0,0 +1,85 @@
+
+import sys
+import random
+
+base = sys.argv[1]
+
+field_freqs = [
+ (('bool', 'optional'), 8.321),
+ (('bool', 'repeated'), 0.033),
+ (('bytes', 'optional'), 0.809),
+ (('bytes', 'repeated'), 0.065),
+ (('double', 'optional'), 2.845),
+ (('double', 'repeated'), 0.143),
+ (('fixed32', 'optional'), 0.084),
+ (('fixed32', 'repeated'), 0.012),
+ (('fixed64', 'optional'), 0.204),
+ (('fixed64', 'repeated'), 0.027),
+ (('float', 'optional'), 2.355),
+ (('float', 'repeated'), 0.132),
+ (('int32', 'optional'), 6.717),
+ (('int32', 'repeated'), 0.366),
+ (('int64', 'optional'), 9.678),
+ (('int64', 'repeated'), 0.425),
+ (('sfixed32', 'optional'), 0.018),
+ (('sfixed32', 'repeated'), 0.005),
+ (('sfixed64', 'optional'), 0.022),
+ (('sfixed64', 'repeated'), 0.005),
+ (('sint32', 'optional'), 0.026),
+ (('sint32', 'repeated'), 0.009),
+ (('sint64', 'optional'), 0.018),
+ (('sint64', 'repeated'), 0.006),
+ (('string', 'optional'), 25.461),
+ (('string', 'repeated'), 2.606),
+ (('Enum', 'optional'), 6.16),
+ (('Enum', 'repeated'), 0.576),
+ (('Message', 'optional'), 22.472),
+ (('Message', 'repeated'), 7.766),
+ (('uint32', 'optional'), 1.289),
+ (('uint32', 'repeated'), 0.051),
+ (('uint64', 'optional'), 1.044),
+ (('uint64', 'repeated'), 0.079),
+]
+
+population = [item[0] for item in field_freqs]
+weights = [item[1] for item in field_freqs]
+
+with open(base + "/benchmarks/100_msgs.proto", "w") as f:
+ f.write('syntax = "proto3";\n')
+ f.write('package upb_benchmark;\n')
+ f.write('message Message {}\n')
+ for i in range(2, 101):
+ f.write(f'message Message{i} {{}}\n')
+
+with open(base + "/benchmarks/200_msgs.proto", "w") as f:
+ f.write('syntax = "proto3";\n')
+ f.write('package upb_benchmark;\n')
+ f.write('message Message {}\n')
+ for i in range(2, 501):
+ f.write(f'message Message{i} {{}}\n')
+
+with open(base + "/benchmarks/100_fields.proto", "w") as f:
+ f.write('syntax = "proto2";\n')
+ f.write('package upb_benchmark;\n')
+ f.write('enum Enum { ZERO = 0; }\n')
+ f.write('message Message {\n')
+ i = 1
+ random.seed(a=0, version=2)
+ for field in random.choices(population=population, weights=weights, k=100):
+ field_type, label = field
+ f.write(f' {label} {field_type} field{i} = {i};\n')
+ i += 1
+ f.write('}\n')
+
+with open(base + "/benchmarks/200_fields.proto", "w") as f:
+ f.write('syntax = "proto2";\n')
+ f.write('package upb_benchmark;\n')
+ f.write('enum Enum { ZERO = 0; }\n')
+ f.write('message Message {\n')
+ i = 1
+ random.seed(a=0, version=2)
+ for field in random.choices(population=population, weights=weights, k=200):
+ field_type, label = field
+ f.write(f' {label} {field_type} field{i} = {i};\n')
+ i += 1
+ f.write('}\n')
