Add benchmarking helper class

CPUClock is useful for repeatedly measuring how long
some piece of code executes.

1 files changed, 69 insertions, 0 deletions

diff --git a/native/include/cpu_time.h b/native/include/cpu_time.h
index 2484159..0efea8b 100644
--- a/native/include/cpu_time.h
+++ b/native/include/cpu_time.h
@@ -1,7 +1,76 @@
 #ifndef CPU_TIME_H
 #define CPU_TIME_H
 
+#include <iostream>
+
 // How much CPU time used (in seconds)?
 double get_cpu_usage(void);
 
+class CPUClock
+{
+private:
+        const char *name;
+        const char *func;
+
+        unsigned int count;
+
+        double start_time;
+        double last;
+        double total;
+
+public:
+        CPUClock(const char *_name = 0, const char *_func = 0)
+                : name(_name), func(_func),
+                  count(0), start_time(0), last(0), total(0)
+        {}
+
+        void start()
+        {
+                start_time = get_cpu_usage();
+        }
+
+        void stop()
+        {
+                last = get_cpu_usage() - start_time;
+                total += last;
+                count++;
+        }
+
+        double get_total() const
+        {
+                return total;
+        }
+
+        double get_last() const
+        {
+                return last;
+        }
+
+        double get_count() const
+        {
+                return count;
+        }
+
+        double get_average() const
+        {
+                return total / ( count ? count : 1);
+        }
+
+        const char *get_name() const
+        {
+                return name;
+        }
+
+        const char *get_function() const
+        {
+                return func;
+        }
+};
+
+std::ostream& operator<<(std::ostream &os, const CPUClock &clock);
+
+char* strip_types(const char* pretty_func);
+
+#define DEFINE_CPU_CLOCK(var) CPUClock var = CPUClock(#var, strip_types(__PRETTY_FUNCTION__))
+
 #endif