1 files changed, 66 insertions, 0 deletions
diff --git a/flatcc/test/benchmark/benchmain/benchmain.h b/flatcc/test/benchmark/benchmain/benchmain.h
new file mode 100644
index 0000000..f29c548
--- /dev/null
+++ b/flatcc/test/benchmark/benchmain/benchmain.h
@@ -0,0 +1,66 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <assert.h>
+#include "flatcc/support/elapsed.h"
+
+#ifdef NDEBUG
+#define COMPILE_TYPE "(optimized)"
+#else
+#define COMPILE_TYPE "(debug)"
+#endif
+
+int main(int argc, char *argv[])
+{
+    /*
+     * The size must be large enough to hold different representations,
+     * including printed json, but we know the printed json is close to
+     * 700 bytes.
+     */
+    const int bufsize = BENCHMARK_BUFSIZ, rep = 1000000;
+    void *buf;
+    size_t size, old_size;
+    double t1, t2, t3;
+    /* Use volatie to prevent over optimization. */
+    volatile int64_t total = 0;
+    int i, ret = 0;
+    DECLARE_BENCHMARK(BM);
+
+    buf = malloc(bufsize);
+
+    /* Warmup to preallocate internal buffers. */
+    size = bufsize;
+    old_size = size;
+    encode(BM, buf, &size);
+    t1 = elapsed_realtime();
+    for (i = 0; i < rep; ++i) {
+        size = bufsize;
+        ret |= encode(BM, buf, &size);
+        assert(ret == 0);
+        if (i > 0 && size != old_size) {
+            printf("abort on inconsistent encoding size\n");
+            goto done;
+        }
+        old_size = size;
+    }
+    t2 = elapsed_realtime();
+    for (i = 0; i < rep; ++i) {
+        total = decode(BM, buf, size, total);
+    }
+    t3 = elapsed_realtime();
+    if (total != -8725036910085654784LL) {
+        printf("ABORT ON CHECKSUM FAILURE\n");
+        goto done;
+    }
+    printf("----\n");
+    show_benchmark(BENCH_TITLE " encode " COMPILE_TYPE, t1, t2, size, rep, "1M");
+    printf("\n");
+    show_benchmark(BENCH_TITLE " decode/traverse " COMPILE_TYPE, t2, t3, size, rep, "1M");
+    printf("----\n");
+    ret = 0;
+done:
+    if (buf) {
+        free(buf);
+    }
+    CLEAR_BENCHMARK(BM);
+    return 0;
+}