summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--source/kit_test/CMakeLists.txt3
-rw-r--r--source/kit_test/bench.c346
-rw-r--r--source/kit_test/bench.h91
-rw-r--r--source/kit_test/test.c2
-rw-r--r--source/kit_test/test.h6
-rw-r--r--source/test/unittests/CMakeLists.txt9
-rw-r--r--source/test/unittests/foo.bench.c13
-rw-r--r--source/test/unittests/main.test.c6
8 files changed, 463 insertions, 13 deletions
diff --git a/source/kit_test/CMakeLists.txt b/source/kit_test/CMakeLists.txt
index e590437..0e61fe3 100644
--- a/source/kit_test/CMakeLists.txt
+++ b/source/kit_test/CMakeLists.txt
@@ -1,6 +1,7 @@
target_sources(
kit_test
PRIVATE
- test.c
+ test.c bench.c
PUBLIC
+ $<BUILD_INTERFACE:${CMAKE_CURRENT_LIST_DIR}/bench.h>
$<BUILD_INTERFACE:${CMAKE_CURRENT_LIST_DIR}/test.h>)
diff --git a/source/kit_test/bench.c b/source/kit_test/bench.c
new file mode 100644
index 0000000..13dfb78
--- /dev/null
+++ b/source/kit_test/bench.c
@@ -0,0 +1,346 @@
+#include "bench.h"
+
+#define _GNU_SOURCE
+#include <setjmp.h>
+#include <signal.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <time.h>
+
+kit_benchs_list_t kit_benchs_list = { 0 };
+
+static void bench_begin(int i) {
+ int const n = kit_benchs_list.benchs[i].repeats++;
+
+ if (n >= KIT_BENCH_REPEATS)
+ return;
+
+ struct timespec tv;
+ timespec_get(&tv, TIME_UTC);
+
+ kit_benchs_list.benchs[i].sec[n] = (int64_t) tv.tv_sec;
+ kit_benchs_list.benchs[i].nsec[n] = (int64_t) tv.tv_nsec;
+}
+
+static void bench_end(int i) {
+ int const n = kit_benchs_list.benchs[i].repeats - 1;
+
+ if (n < 0 || n >= KIT_BENCH_REPEATS)
+ return;
+
+ struct timespec tv;
+ timespec_get(&tv, TIME_UTC);
+
+ int64_t const sec = ((int64_t) tv.tv_sec) -
+ kit_benchs_list.benchs[i].sec[n];
+ int64_t const nsec = ((int64_t) tv.tv_nsec) -
+ kit_benchs_list.benchs[i].nsec[n];
+
+ kit_benchs_list.benchs[i].duration_nsec[n] = sec * 1000000000 +
+ nsec;
+}
+
+enum { white, blue, light, yellow, red, green };
+
+static char const *const color_codes[] = {
+ [white] = "\x1b[38m", [blue] = "\x1b[34m", [light] = "\x1b[37m",
+ [yellow] = "\x1b[33m", [red] = "\x1b[31m", [green] = "\x1b[32m"
+};
+
+static int print_color(int c) {
+ return printf("%s", color_codes[c]);
+}
+
+void kit_bench_register(char const *name, char const *file,
+ kit_bench_run_fn fn) {
+ int n = kit_benchs_list.size++;
+ if (n < KIT_BENCHS_SIZE_LIMIT) {
+ kit_benchs_list.benchs[n].bench_fn = fn;
+ kit_benchs_list.benchs[n].bench_name = name;
+ kit_benchs_list.benchs[n].bench_file = file;
+ kit_benchs_list.benchs[n].repeats = 0;
+ }
+}
+
+static jmp_buf kit_bench_restore_execution;
+
+static int const signums[] = { SIGINT, SIGILL, SIGABRT,
+ SIGFPE, SIGSEGV, SIGTERM };
+
+static char const *const signames[] = {
+ [SIGINT] = "Interactive attention signal",
+ [SIGILL] = "Illegal instruction",
+ [SIGABRT] = "Abnormal termination",
+ [SIGFPE] = "Erroneous arithmetic operation",
+ [SIGSEGV] = "Invalid access to storage",
+ [SIGTERM] = "Termination request"
+};
+
+static void handle_signal(int signum) {
+ longjmp(kit_bench_restore_execution, signum);
+}
+
+static void setup_signals() {
+ int i;
+
+ for (i = 0; i < sizeof signums / sizeof *signums; i++) {
+#if defined(_WIN32) && !(defined __CYGWIN__)
+ signal(signums[i], handle_signal);
+#else
+ struct sigaction action;
+ memset(&action, 0, sizeof action);
+ action.sa_handler = handle_signal;
+
+ sigaction(signums[i], &action, NULL);
+#endif
+ }
+}
+
+static int run_bench(volatile int i) {
+ int signum = setjmp(kit_bench_restore_execution);
+
+ if (signum != 0) {
+ kit_benchs_list.benchs[i].signal = signum;
+ return 0;
+ }
+
+ kit_benchs_list.benchs[i].bench_fn(i, bench_begin, bench_end);
+ return 1;
+}
+
+int compare_64_(void const *x_, void const *y_) {
+ int64_t const *x = (int64_t const *) x_;
+ int64_t const *y = (int64_t const *) y_;
+ return *x - *y;
+}
+
+int kit_run_benchmarks(int argc, char **argv) {
+ int success_count = 0;
+ int status = 0;
+ int no_color = 0;
+ int line_width = 20;
+ int carriage_return = 1;
+
+ int i, j;
+
+ char const *specific_bench = NULL;
+
+ setup_signals();
+
+ for (i = 0; i < argc; i++)
+ if (strcmp("--no-term-color", argv[i]) == 0)
+ no_color = 1;
+ else if (strcmp("--no-carriage-return", argv[i]) == 0)
+ carriage_return = 0;
+ else if (strcmp("--match", argv[i]) == 0)
+ specific_bench = argv[++i];
+
+ if (specific_bench != NULL) {
+ no_color || print_color(light);
+ printf("Run benchmarks matching ");
+ no_color || print_color(white);
+ printf("*%s*", specific_bench);
+ no_color || print_color(light);
+ printf("\n\n");
+ }
+
+ char const *file = NULL;
+ ptrdiff_t file_root = -1;
+ int benchs_total = 0;
+
+ for (i = 0; i < kit_benchs_list.size && i < KIT_BENCHS_SIZE_LIMIT;
+ i++) {
+ if (specific_bench != NULL &&
+ strstr(kit_benchs_list.benchs[i].bench_name,
+ specific_bench) == NULL)
+ continue;
+ benchs_total++;
+ int const l = 2 +
+ (int) strlen(kit_benchs_list.benchs[i].bench_name);
+ if (line_width < l)
+ line_width = l;
+ }
+
+ if (benchs_total > 0) {
+ char const *const s = kit_benchs_list.benchs[0].bench_file;
+
+ for (j = 1; j < kit_benchs_list.size && j < KIT_BENCHS_SIZE_LIMIT;
+ j++) {
+ if (specific_bench != NULL &&
+ strstr(kit_benchs_list.benchs[j].bench_name,
+ specific_bench) == NULL)
+ continue;
+ if (strcmp(s, kit_benchs_list.benchs[j].bench_file) == 0)
+ continue;
+ int k = 0;
+ for (; s[k] != '\0' &&
+ kit_benchs_list.benchs[j].bench_file[k] != '\0' &&
+ s[k] == kit_benchs_list.benchs[j].bench_file[k];
+ k++) { }
+ if (file_root == -1 || file_root > k)
+ file_root = k;
+ }
+
+ if (file_root == -1) {
+ for (i = 0; s[i] != '\0'; i++)
+ if (s[i] == '/' || s[i] == '\\')
+ file_root = i + 1;
+ }
+ }
+
+ no_color || print_color(blue);
+ printf("# ");
+ no_color || print_color(light);
+ printf("BENCHMARK");
+ printf("%*c", line_width - 9, ' ');
+ no_color || print_color(green);
+ printf(" LOW ");
+ no_color || print_color(light);
+ printf("|");
+ no_color || print_color(blue);
+ printf(" MEDIAN ");
+ no_color || print_color(light);
+ printf("|");
+ no_color || print_color(yellow);
+ printf(" HIGH ");
+ no_color || print_color(light);
+ printf(" in nanoseconds\n\n");
+
+ for (i = 0; i < kit_benchs_list.size && i < KIT_BENCHS_SIZE_LIMIT;
+ i++) {
+ if (specific_bench != NULL &&
+ strstr(kit_benchs_list.benchs[i].bench_name,
+ specific_bench) == NULL)
+ continue;
+ if (file == NULL ||
+ strcmp(file, kit_benchs_list.benchs[i].bench_file) != 0) {
+ if (file != NULL)
+ printf("\n");
+ file = kit_benchs_list.benchs[i].bench_file;
+ no_color || print_color(blue);
+ printf("* ");
+ no_color || print_color(white);
+ printf("%s\n", file + file_root);
+ }
+
+ !carriage_return || no_color || print_color(yellow);
+ carriage_return || no_color || print_color(light);
+ printf("` %s ", kit_benchs_list.benchs[i].bench_name);
+ !carriage_return || printf("\r");
+ fflush(stdout);
+
+ int bench_status = run_bench(i);
+
+ if (kit_benchs_list.benchs[i].repeats > KIT_BENCH_REPEATS)
+ bench_status = 0;
+
+ !carriage_return || no_color || print_color(light);
+ !carriage_return ||
+ printf("` %s ", kit_benchs_list.benchs[i].bench_name);
+
+ int const l = (int) strlen(kit_benchs_list.benchs[i].bench_name);
+ printf("%*c", line_width - l, ' ');
+
+ if (kit_benchs_list.benchs[i].repeats <= 0) {
+ no_color || print_color(yellow);
+ printf(" 0 runs\n");
+ success_count++;
+ } else if (bench_status == 0) {
+ no_color || print_color(red);
+ printf(" FAIL\n");
+ status = 1;
+ } else {
+ int const repeats = kit_benchs_list.benchs[i].repeats;
+
+ qsort(kit_benchs_list.benchs[i].duration_nsec, repeats,
+ sizeof *kit_benchs_list.benchs[i].duration_nsec,
+ compare_64_);
+
+ int64_t const average =
+ kit_benchs_list.benchs[i].duration_nsec[repeats / 2];
+ int64_t const floor =
+ kit_benchs_list.benchs[i].duration_nsec[repeats / 10];
+ int64_t const roof =
+ kit_benchs_list.benchs[i]
+ .duration_nsec[repeats - repeats / 10 - 1];
+
+ no_color || print_color(white);
+ printf("%-8lld", (long long) floor);
+ no_color || print_color(light);
+ printf("| ");
+ no_color || print_color(white);
+ printf("%-8lld", (long long) average);
+ no_color || print_color(light);
+ printf("| ");
+ no_color || print_color(white);
+ printf("%-8lld", (long long) roof);
+ no_color || print_color(light);
+ printf(" %d runs\n", repeats);
+ success_count++;
+ }
+ }
+
+ printf("\n");
+
+ if (status != 0) {
+ for (i = 0; i < kit_benchs_list.size && i < KIT_BENCHS_SIZE_LIMIT;
+ i++) {
+ if (specific_bench != NULL &&
+ strstr(kit_benchs_list.benchs[i].bench_name,
+ specific_bench) == NULL)
+ continue;
+ if (kit_benchs_list.benchs[i].signal != 0) {
+ int signum = kit_benchs_list.benchs[i].signal;
+ if (signum >= 0 &&
+ signum < sizeof signames / sizeof *signames &&
+ signames[signum] != NULL) {
+ no_color || print_color(light);
+ printf("Signal \"%s\" (%d) for \"", signames[signum],
+ signum);
+ no_color || print_color(white);
+ printf("%s", kit_benchs_list.benchs[i].bench_name);
+ no_color || print_color(light);
+ printf("\" in \"");
+ no_color || print_color(white);
+ printf("%s",
+ kit_benchs_list.benchs[i].bench_file + file_root);
+ no_color || print_color(light);
+ printf("\"!.\n");
+ } else {
+ no_color || print_color(light);
+ printf("Unknown signal (%d) for \"", signum);
+ no_color || print_color(white);
+ printf("%s", kit_benchs_list.benchs[i].bench_name);
+ no_color || print_color(light);
+ printf("\" in \"");
+ no_color || print_color(white);
+ printf("%s",
+ kit_benchs_list.benchs[i].bench_file + file_root);
+ no_color || print_color(light);
+ printf("\"!.\n");
+ }
+ }
+ }
+
+ printf("\n");
+ }
+
+ if (kit_benchs_list.size > KIT_BENCHS_SIZE_LIMIT) {
+ no_color || print_color(light);
+ printf("Too many benchmarks!\n\n");
+ status = 1;
+ }
+
+ if (status == 0) {
+ no_color || print_color(green);
+ printf("DONE\n");
+ } else {
+ no_color || print_color(red);
+ printf("DONE WITH ERRORS\n");
+ }
+
+ no_color || print_color(white);
+ printf("\n");
+ return status;
+}
diff --git a/source/kit_test/bench.h b/source/kit_test/bench.h
new file mode 100644
index 0000000..131d89d
--- /dev/null
+++ b/source/kit_test/bench.h
@@ -0,0 +1,91 @@
+#ifndef KIT_BENCH_BENCH_H
+#define KIT_BENCH_BENCH_H
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#include "test.h"
+#include <stdint.h>
+
+#ifndef KIT_BENCH_FILE
+# define KIT_BENCH_FILE kit_bench
+#endif
+
+#ifndef KIT_BENCHS_SIZE_LIMIT
+# define KIT_BENCHS_SIZE_LIMIT 0x1000
+#endif
+
+#ifndef KIT_BENCH_REPEATS
+# define KIT_BENCH_REPEATS 400
+#endif
+
+typedef void (*kit_bench_begin_fn)(int bench_index);
+typedef void (*kit_bench_end_fn)(int bench_index);
+typedef void (*kit_bench_run_fn)(
+ int kit_bench_index_, kit_bench_begin_fn kit_bench_begin_fn_,
+ kit_bench_end_fn kit_bench_end_fn_);
+
+typedef struct {
+ char const *bench_name;
+ char const *bench_file;
+ kit_bench_run_fn bench_fn;
+ int64_t sec[KIT_BENCH_REPEATS];
+ int32_t nsec[KIT_BENCH_REPEATS];
+ int64_t duration_nsec[KIT_BENCH_REPEATS];
+ int repeats;
+ int signal;
+} kit_benchmark_t;
+
+typedef struct {
+ int size;
+ kit_benchmark_t benchs[KIT_BENCHS_SIZE_LIMIT];
+} kit_benchs_list_t;
+
+extern kit_benchs_list_t kit_benchs_list;
+
+void kit_bench_register(char const *name, char const *file,
+ kit_bench_run_fn fn);
+
+#define KIT_BENCHMARK(name) \
+ static void KIT_TEST_CONCAT3_(kit_bench_run_, __LINE__, \
+ KIT_BENCH_FILE)( \
+ int, kit_bench_begin_fn, kit_bench_end_fn); \
+ KIT_TEST_ON_START_( \
+ KIT_TEST_CONCAT3_(kit_benchmark_, __LINE__, KIT_BENCH_FILE)) { \
+ kit_bench_register(name, __FILE__, \
+ KIT_TEST_CONCAT3_(kit_bench_run_, __LINE__, \
+ KIT_BENCH_FILE)); \
+ } \
+ static void KIT_TEST_CONCAT3_(kit_bench_run_, __LINE__, \
+ KIT_BENCH_FILE)( \
+ int kit_bench_index_, kit_bench_begin_fn kit_bench_begin_fn_, \
+ kit_bench_end_fn kit_bench_end_fn_)
+
+#define KIT_BENCHMARK_BEGIN \
+ for (int kit_bench_repeat_ = 0; \
+ kit_bench_repeat_ < KIT_BENCH_REPEATS; kit_bench_repeat_++) { \
+ kit_bench_begin_fn_(kit_bench_index_); \
+ {
+
+#define KIT_BENCHMARK_END \
+ } \
+ kit_bench_end_fn_(kit_bench_index_); \
+ }
+
+int kit_run_benchmarks(int argc, char **argv);
+
+#ifndef KIT_DISABLE_SHORT_NAMES
+# define BENCHMARK KIT_BENCHMARK
+# define BENCHMARK_BEGIN KIT_BENCHMARK_BEGIN
+# define BENCHMARK_END KIT_BENCHMARK_END
+
+# define bench_register kit_bench_register
+# define run_benchmarks kit_run_benchmarks
+#endif
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif
diff --git a/source/kit_test/test.c b/source/kit_test/test.c
index ed7e7a7..3d3e180 100644
--- a/source/kit_test/test.c
+++ b/source/kit_test/test.c
@@ -29,7 +29,7 @@ static long long sec_to_ms(long long sec) {
enum { white, blue, light, yellow, red, green };
-char const *const color_codes[] = {
+static char const *const color_codes[] = {
[white] = "\x1b[38m", [blue] = "\x1b[34m", [light] = "\x1b[37m",
[yellow] = "\x1b[33m", [red] = "\x1b[31m", [green] = "\x1b[32m"
};
diff --git a/source/kit_test/test.h b/source/kit_test/test.h
index afc1009..39c28e7 100644
--- a/source/kit_test/test.h
+++ b/source/kit_test/test.h
@@ -51,12 +51,6 @@ extern kit_tests_list_t kit_tests_list;
static void f(void)
#else
# ifdef _MSC_VER
-# ifdef __cplusplus
-# define KIT_EXTERN_C_ extern "C"
-# else
-# define KIT_EXTERN_C_
-# endif
-
# pragma section(".CRT$XCU", read)
# define KIT_TEST_ON_START_2_(f, p) \
static void f(void); \
diff --git a/source/test/unittests/CMakeLists.txt b/source/test/unittests/CMakeLists.txt
index a73ca6e..b8a6dca 100644
--- a/source/test/unittests/CMakeLists.txt
+++ b/source/test/unittests/CMakeLists.txt
@@ -2,7 +2,8 @@ target_sources(
kit_test_suite
PRIVATE
async_function.test.c bigint.test.c mutex.test.c
- test_duration.test.c main.test.c string_ref.test.c atomic.test.c thread.test.c
- array_ref.test.c input_stream.test.c sha256.test.c lower_bound.test.c
- secure_random.test.c condition_variable.test.c mersenne_twister_64.test.c
- input_buffer.test.c move_back.test.c dynamic_array.test.c file.test.c)
+ test_duration.test.c main.test.c string_ref.test.c atomic.test.c foo.bench.c
+ thread.test.c array_ref.test.c input_stream.test.c sha256.test.c
+ lower_bound.test.c secure_random.test.c condition_variable.test.c
+ mersenne_twister_64.test.c input_buffer.test.c move_back.test.c dynamic_array.test.c
+ file.test.c)
diff --git a/source/test/unittests/foo.bench.c b/source/test/unittests/foo.bench.c
new file mode 100644
index 0000000..531f1d8
--- /dev/null
+++ b/source/test/unittests/foo.bench.c
@@ -0,0 +1,13 @@
+#define KIT_BENCH_FILE foo
+#include "../../kit_test/bench.h"
+
+BENCHMARK("foo") {
+ BENCHMARK_BEGIN;
+ {
+ volatile int x = 0;
+ for (int i = 0; i < 200000; i++) x++;
+ }
+ BENCHMARK_END;
+}
+
+BENCHMARK("bar") { }
diff --git a/source/test/unittests/main.test.c b/source/test/unittests/main.test.c
index 3ced74c..60c2e36 100644
--- a/source/test/unittests/main.test.c
+++ b/source/test/unittests/main.test.c
@@ -1,5 +1,9 @@
+#include "../../kit_test/bench.h"
#include "../../kit_test/test.h"
int main(int argc, char **argv) {
- return run_tests(argc, argv);
+ int status = run_tests(argc, argv);
+ if (status == 0)
+ status = run_benchmarks(argc, argv);
+ return status;
}