X-Git-Url: http://git.mutantstargoat.com/user/nuclear/?p=bootcensus;a=blobdiff_plain;f=src%2Fcensus%2Futil.h;fp=src%2Fcensus%2Futil.h;h=76dd044eae32bfe11d9419d1a3cb25e42adf1530;hp=0000000000000000000000000000000000000000;hb=10843571c724084c68d33d0438167d400cc8de2e;hpb=f9194c22d8073b69b3e64fc26b9e245b6b54d009

diff --git a/src/census/util.h b/src/census/util.h
new file mode 100644
index 0000000..76dd044
--- /dev/null
+++ b/src/census/util.h
@@ -0,0 +1,102 @@
+#ifndef UTIL_H_
+#define UTIL_H_
+
+#include "inttypes.h"
+
+#ifdef __GNUC__
+#define INLINE __inline
+
+#elif defined(__WATCOMC__)
+#define INLINE __inline
+
+#else
+#define INLINE
+#endif
+
+/* fast conversion of double -> 32bit int
+ * for details see:
+ *  - http://chrishecker.com/images/f/fb/Gdmfp.pdf
+ *  - http://stereopsis.com/FPU.html#convert
+ */
+static INLINE int32_t cround64(double val)
+{
+	val += 6755399441055744.0;
+	return *(int32_t*)&val;
+}
+
+extern uint32_t perf_start_count, perf_interval_count;
+
+#ifdef __WATCOMC__
+void perf_start(void);
+#pragma aux perf_start = \
+	"xor eax, eax" \
+	"cpuid" \
+	"rdtsc" \
+	"mov [perf_start_count], eax" \
+	modify[eax ebx ecx edx];
+
+void perf_end(void);
+#pragma aux perf_end = \
+	"xor eax, eax" \
+	"cpuid" \
+	"rdtsc" \
+	"sub eax, [perf_start_count]" \
+	"mov [perf_interval_count], eax" \
+	modify [eax ebx ecx edx];
+
+void debug_break(void);
+#pragma aux debug_break = "int 3";
+#endif
+
+#ifdef __GNUC__
+#define perf_start()  asm volatile ( \
+	"xor %%eax, %%eax\n" \
+	"cpuid\n" \
+	"rdtsc\n" \
+	"mov %%eax, %0\n" \
+	: "=m"(perf_start_count) \
+	:: "%eax", "%ebx", "%ecx", "%edx")
+
+#define perf_end() asm volatile ( \
+	"xor %%eax, %%eax\n" \
+	"cpuid\n" \
+	"rdtsc\n" \
+	"sub %1, %%eax\n" \
+	"mov %%eax, %0\n" \
+	: "=m"(perf_interval_count) \
+	: "m"(perf_start_count) \
+	: "%eax", "%ebx", "%ecx", "%edx")
+
+#define debug_break() \
+	asm volatile ("int $3")
+#endif
+
+#ifdef _MSC_VER
+#define perf_start() \
+	do { \
+		__asm { \
+			xor eax, eax \
+			cpuid \
+			rdtsc \
+			mov [perf_start_count], eax \
+		} \
+	} while(0)
+
+#define perf_end() \
+	do { \
+		__asm { \
+			xor eax, eax \
+			cpuid \
+			rdtsc \
+			sub eax, [perf_start_count] \
+			mov [perf_interval_count], eax \
+		} \
+	} while(0)
+
+#define debug_break() \
+	do { \
+		__asm { int 3 } \
+	} while(0)
+#endif
+
+#endif	/* UTIL_H_ */