2 files changed, 99 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
diff --git a/native/src/cpu_time.cpp b/native/src/cpu_time.cpp
index d003f22..8dde662 100644
--- a/native/src/cpu_time.cpp
+++ b/native/src/cpu_time.cpp
@@ -3,6 +3,8 @@
 #include <sys/time.h>
 #include <sys/resource.h>
 
+#include <cstring>
+
 #include "cpu_time.h"
 
 
@@ -47,3 +49,31 @@ double get_cpu_usage(void)
 }
 
 #endif
+
+
+std::ostream& operator<<(std::ostream &os, const CPUClock &clock)
+{
+        if (clock.get_function())
+                os << clock.get_function() << "::";
+        os << clock.get_name()
+           << ": total=" << clock.get_total() * 1000 << "ms "
+           << "last="    << clock.get_last()  * 1000 << "ms "
+           << "average=" << clock.get_average() * 1000 << "ms "
+           << "count="   << clock.get_count();
+        return os;
+}
+
+char *strip_types(const char* pretty_func)
+{
+        char *copy = strdup(pretty_func);
+
+        char *start = strchr(copy, ' ');
+        char *end = strchr(copy, '(');
+
+        if (start)
+                copy = start + 1;
+        if (end)
+                *end = '\0';
+
+        return copy;
+}