mcl/include/cybozu/benchmark.hpp

#pragma once
/**
	@file
	@brief measure exec time of function
	@author MITSUNARI Shigeo
*/
#if defined(_MSC_VER) && (MSC_VER <= 1500)
	#include <cybozu/inttype.hpp>
#else
	#include <stdint.h>
#endif
#include <stdio.h>

#ifdef __EMSCRIPTEN__
	#define CYBOZU_BENCH_USE_GETTIMEOFDAY
#endif

#ifdef CYBOZU_BENCH_USE_GETTIMEOFDAY
	#include <sys/time.h>
#elif !defined(CYBOZU_BENCH_DONT_USE_RDTSC)
	#if defined(_M_IX86) || defined(_M_X64) || defined(__i386__) || defined(__x86_64__)
		#define CYBOZU_BENCH_USE_RDTSC
		#define CYBOZU_BENCH_USE_CPU_TIMER
	#endif
	#if defined(__GNUC__) && defined(__ARM_ARCH_7A__)
//		#define CYBOZU_BENCH_USE_MRC
//		#define CYBOZU_BENCH_USE_CPU_TIMER
	#endif
#endif


#include <assert.h>
#include <time.h>
#ifdef _MSC_VER
	#include <intrin.h>
	#include <sys/timeb.h>
#else
#endif

#ifndef CYBOZU_UNUSED
	#ifdef __GNUC__
		#define CYBOZU_UNUSED __attribute__((unused))
	#else
		#define CYBOZU_UNUSED
	#endif
#endif

namespace cybozu {

namespace bench {

static void (*g_putCallback)(double);

static inline void setPutCallback(void (*f)(double))
{
	g_putCallback = f;
}

} // cybozu::bench

class CpuClock {
public:
	static inline uint64_t getCpuClk()
	{
#ifdef CYBOZU_BENCH_USE_RDTSC
#ifdef _MSC_VER
		return __rdtsc();
#else
		unsigned int eax, edx;
		__asm__ volatile("rdtsc" : "=a"(eax), "=d"(edx));
		return ((uint64_t)edx << 32) | eax;
#endif
#elif defined(CYBOZU_BENCH_USE_MRC)
		uint32_t clk;
		__asm__ volatile("mrc p15, 0, %0, c9, c13, 0" : "=r"(clk));
		return clk;
#else
#ifdef _MSC_VER
		struct _timeb timeb;
		_ftime_s(&timeb);
		return uint64_t(timeb.time) * 1000000000 + timeb.millitm * 1000000;
#elif defined(CYBOZU_BENCH_USE_GETTIMEOFDAY)
		struct timeval tv;
		int ret CYBOZU_UNUSED = gettimeofday(&tv, 0);
		assert(ret == 0);
		return uint64_t(tv.tv_sec) * 1000000000 + tv.tv_usec * 1000;
#else
		struct timespec tp;
		int ret CYBOZU_UNUSED = clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &tp);
		assert(ret == 0);
		return uint64_t(tp.tv_sec) * 1000000000 + tp.tv_nsec;
#endif
#endif
	}
	CpuClock()
		: clock_(0)
		, count_(0)
	{
	}
	void begin()
	{
		clock_ -= getCpuClk();
	}
	void end()
	{
		clock_ += getCpuClk();
		count_++;
	}
	int getCount() const { return count_; }
	uint64_t getClock() const { return clock_; }
	void clear() { count_ = 0; clock_ = 0; }
	void put(const char *msg = 0, int N = 1) const
	{
		double t = getClock() / double(getCount()) / N;
		if (msg && *msg) printf("%s ", msg);
		if (bench::g_putCallback) {
			bench::g_putCallback(t);
			return;
		}
#ifdef CYBOZU_BENCH_USE_CPU_TIMER
		if (t > 1e6) {
			printf("%7.3fMclk", t * 1e-6);
		} else if (t > 1e3) {
			printf("%7.3fKclk", t * 1e-3);
		} else {
			printf("%6.2f clk", t);
		}
#else
		if (t > 1e6) {
			printf("%7.3fmsec", t * 1e-6);
		} else if (t > 1e3) {
			printf("%7.3fusec", t * 1e-3);
		} else {
			printf("%6.2fnsec", t);
		}
#endif
		if (msg && *msg) printf("\n");
	}
	// adhoc constatns for CYBOZU_BENCH
#ifdef CYBOZU_BENCH_USE_CPU_TIMER
	static const int loopN1 = 1000;
	static const int loopN2 = 100;
	static const uint64_t maxClk = (uint64_t)1e8;
#else
	static const int loopN1 = 100;
	static const int loopN2 = 100;
	static const uint64_t maxClk = (uint64_t)1e8;
#endif
private:
	uint64_t clock_;
	int count_;
};

namespace bench {

static CpuClock g_clk;
static int CYBOZU_UNUSED g_loopNum;

} // cybozu::bench
/*
	loop counter is automatically determined
	CYBOZU_BENCH(<msg>, <func>, <param1>, <param2>, ...);
	if msg == "" then only set g_clk, g_loopNum
*/
#define CYBOZU_BENCH(msg, func, ...) \
{ \
	const uint64_t _cybozu_maxClk = cybozu::CpuClock::maxClk; \
	cybozu::CpuClock _cybozu_clk; \
	for (int _cybozu_i = 0; _cybozu_i < cybozu::CpuClock::loopN2; _cybozu_i++) { \
		_cybozu_clk.begin(); \
		for (int _cybozu_j = 0; _cybozu_j < cybozu::CpuClock::loopN1; _cybozu_j++) { func(__VA_ARGS__); } \
		_cybozu_clk.end(); \
		if (_cybozu_clk.getClock() > _cybozu_maxClk) break; \
	} \
	if (msg && *msg) _cybozu_clk.put(msg, cybozu::CpuClock::loopN1); \
	cybozu::bench::g_clk = _cybozu_clk; cybozu::bench::g_loopNum = cybozu::CpuClock::loopN1; \
}

/*
	double clk;
	CYBOZU_BENCH_T(clk, <func>, <param1>, <param2>, ...);
	clk is set by CYBOZU_BENCH_T
*/
#define CYBOZU_BENCH_T(clk, func, ...) \
{ \
	const uint64_t _cybozu_maxClk = cybozu::CpuClock::maxClk; \
	cybozu::CpuClock _cybozu_clk; \
	for (int _cybozu_i = 0; _cybozu_i < cybozu::CpuClock::loopN2; _cybozu_i++) { \
		_cybozu_clk.begin(); \
		for (int _cybozu_j = 0; _cybozu_j < cybozu::CpuClock::loopN1; _cybozu_j++) { func(__VA_ARGS__); } \
		_cybozu_clk.end(); \
		if (_cybozu_clk.getClock() > _cybozu_maxClk) break; \
	} \
	clk = _cybozu_clk.getClock() / (double)_cybozu_clk.getCount() / cybozu::CpuClock::loopN1; \
}

/*
	loop counter N is given
	CYBOZU_BENCH_C(<msg>, <counter>, <func>, <param1>, <param2>, ...);
	if msg == "" then only set g_clk, g_loopNum
*/
#define CYBOZU_BENCH_C(msg, _N, func, ...) \
{ \
	cybozu::CpuClock _cybozu_clk; \
	_cybozu_clk.begin(); \
	for (int _cybozu_j = 0; _cybozu_j < _N; _cybozu_j++) { func(__VA_ARGS__); } \
	_cybozu_clk.end(); \
	if (msg && *msg) _cybozu_clk.put(msg, _N); \
	cybozu::bench::g_clk = _cybozu_clk; cybozu::bench::g_loopNum = _N; \
}

} // cybozu