66 lines
1.5 KiB
C++
66 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include <functional>
|
|
#include <chrono>
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <iomanip>
|
|
#include <locale>
|
|
|
|
namespace utils
|
|
{
|
|
|
|
template<typename T>
|
|
std::string format(const T& value)
|
|
{
|
|
static std::locale loc("");
|
|
std::stringstream ss;
|
|
ss.imbue(loc);
|
|
ss << value;
|
|
return ss.str();
|
|
}
|
|
|
|
template<>
|
|
std::string format(const double & value)
|
|
{
|
|
static std::locale loc("");
|
|
std::stringstream ss;
|
|
ss.imbue(loc);
|
|
ss << std::fixed << std::setprecision(1) << value;
|
|
return ss.str();
|
|
}
|
|
|
|
inline void bench(const std::string& fn_name, const std::chrono::milliseconds &duration, const std::function<void() >& fn)
|
|
{
|
|
using namespace std::chrono;
|
|
typedef steady_clock the_clock;
|
|
size_t counter = 0;
|
|
seconds print_interval(1);
|
|
auto start_time = the_clock::now();
|
|
auto lastPrintTime = start_time;
|
|
while (true) {
|
|
fn();
|
|
++counter;
|
|
auto now = the_clock::now();
|
|
if (now - start_time >= duration)
|
|
break;
|
|
if (now - lastPrintTime >= print_interval) {
|
|
std::cout << fn_name << ": " << format(counter) << " per sec" << std::endl;
|
|
counter = 0;
|
|
lastPrintTime = the_clock::now();
|
|
}
|
|
}
|
|
}
|
|
|
|
inline void bench(const std::string& fn_name, const std::function<void() >& fn)
|
|
{
|
|
|
|
using namespace std::chrono;
|
|
auto start = steady_clock::now();
|
|
fn();
|
|
auto delta = steady_clock::now() - start;
|
|
|
|
std::cout << fn_name << ": " << duration_cast<milliseconds>(delta).count() << " ms" << std::endl;
|
|
}
|
|
}
|