性能テスト——(2)windowsの下のテストマクロに基づく
#ifndef STAT_H
#define STAT_H
#include<map>
#include<string>
using namespace std;
/***
wx, ,
**/
struct st_stat{
public:
st_stat():call_count(0),time(0){}
LARGE_INTEGER start;
LARGE_INTEGER end;
int call_count;
__int64 time;
};
/**
*/
class Runnable
{
public:
explicit Runnable(const std::string& name): _name(name) {}
~Runnable() {}
const std::string& name() const { return _name; }
inline void begin() ;
inline void end() ;
void diff_time(LONGLONG diff)
{
stat.time -= (stat.end.QuadPart-diff);
}
int getCallCount(){return stat.call_count;}
__int64 getCallTime(){return stat.time;}
st_stat stat;
private:
std::string _name;
};
class test_suite
{
public:
static test_suite& get_instance()
{
static test_suite me;
return me;
}
~test_suite(){if(fp) fclose(fp);fp=NULL;
}
void setWorkingDir(const std::string& val) { _working_dir = val; }
const std::string& getWorkingDir() const { return _working_dir; }
void setTestFilter(const std::string& val) { _test_filter = val; }
const std::string& getTestFilter() const { return _test_filter; }
void addTest(Runnable* test) { test_vec.insert(pair<string,Runnable*>(test->name(),test)); }
void func_begin(const std::string &name);
void func_end(const std::string &name);
void init(const std::string &dir);
void finish();
private:
test_suite():fp(NULL),_working_dir(),_test_filter(){}
void printHeading();
void printFooter();
void printError(const std::string&msg);
std::string _working_dir;
std::string _test_filter;
typedef std::map<std::string,Runnable*> MAP;
typedef MAP::iterator IT;
typedef MAP::const_iterator const_it;
MAP test_vec;//
std::vector<Runnable*> none_init_vec;// , new ,
FILE *fp;
};
void call_stat_begin(st_stat*pstat);
void call_stat_end(st_stat*pstat);
#if defined ST_TEST_PERFORMANCE
#define STAT_TEST_INIT(name) \
struct st_##name##_test: Runnable { \
st_##name##_test(): Runnable(#name) { \
test_suite::get_instance().addTest(this); \
} \
} name##_test_instance
#define STAT_TEST_BEGIN(name) \
test_suite::get_instance().func_begin(#name)
#define STAT_TEST_END(name) \
test_suite::get_instance().func_end(#name)
#define STAT_GLOBAL_INIT(fileDir) test_suite::get_instance().init(fileDir);
#define STAT_GLOBAL_FINISH() test_suite::get_instance().finish();
#else
#define STAT_TEST_INIT(name)
#define STAT_TEST_BEGIN(name) (void*)(0)
#define STAT_TEST_END(name) (void*)(0)
#define STAT_GLOBAL_INIT(fileDir)
#define STAT_GLOBAL_FINISH() (void*)(0)
#endif
#endif
st_stat.cppファイル
#include"stdafx.h"
#include"st_stat.h"
#ifdef _WIN32
#include<windows.h>
void call_stat_begin(st_stat*pstat)
{
pstat->call_count++;
LARGE_INTEGER large_integer;
DWORD mask = 1;
//DWORD_PTR oldmask = ::SetThreadAffinityMask(::GetCurrentThread(),mask);// , cpu , ,
BOOL bRet = ::QueryPerformanceCounter(&large_integer);
//::SetThreadAffinityMask(::GetCurrentThread(),oldmask);
_ASSERT(bRet);
pstat->start = large_integer;
}
void call_stat_end(st_stat*pstat)
{
DWORD mask = 1;
//DWORD_PTR oldmask = ::SetThreadAffinityMask(::GetCurrentThread(),mask);
LARGE_INTEGER large_integer;
BOOL bRet = ::QueryPerformanceCounter(&large_integer);
_ASSERT(bRet);
//::SetThreadAffinityMask(::GetCurrentThread(),oldmask);
pstat->end = large_integer;
pstat->time += pstat->end.QuadPart-pstat->start.QuadPart;
}
void Runnable::begin()
{
::call_stat_begin(&stat);
}
void Runnable::end()
{
call_stat_end(&stat);
}
void test_suite::func_begin(const std::string & name)
{
// Run test initializers
const_it it = test_vec.find(name);
if(it!=test_vec.end())
{
LARGE_INTEGER large_integer;
BOOL bRet = ::QueryPerformanceCounter(&large_integer);
_ASSERT(bRet);
(*it).second->begin();
return;
}
Runnable *pTest = new Runnable(name);
test_vec.insert(pair<string,Runnable*>(name,pTest));
none_init_vec.push_back(pTest);
pTest->begin();
}
void test_suite::func_end(const std::string & name)
{
// Run test initializers
const_it it = test_vec.find(name);
if(it!=test_vec.end())
{
LARGE_INTEGER large_integer;
BOOL bRet = ::QueryPerformanceCounter(&large_integer);
_ASSERT(bRet);
(*it).second->end();
(*it).second->diff_time(large_integer.QuadPart);
}
//printError(name);
/*vector<Runnable*>::iterator it = test_vec.begin();
for (; it != test_vec.end(); ++it)
{
if ((*it)->name()==name)
{
LARGE_INTEGER large_integer;
BOOL bRet = ::QueryPerformanceCounter(&large_integer);
_ASSERT(bRet);
(*it)->end();
(*it)->diff_time(large_integer.QuadPart);
}
}*/
}
void test_suite::printHeading()
{
if(fp)
{
fprintf(fp," , , PerformanceCounter
");
fprintf(fp,"start Tick Count=%d ms\r ",::GetTickCount());
}
}
void test_suite::printFooter()
{
if(fp)
{
fprintf(fp,"end Tick Count=%d ms\r",::GetTickCount());
}
}
void test_suite:: init(const std::string &dir){
setWorkingDir(dir);
if(fp)
{
fclose(fp);
fp = NULL;
}
if(getWorkingDir().size()>0)
{
std::string filePath(dir);
filePath.append("test_performance.csv");
fp=fopen(filePath.c_str(),"a+");
}
else
{
fp=fopen("test_performance.csv","a+");
}
printHeading();
}
void test_suite::finish()
{
if(fp)
{
for(IT it = test_vec.begin();it!=test_vec.end();it++)
{
fprintf(fp,"%s,%d,%I64d\r",it->second->name().c_str(),it->second->getCallCount(),it->second->getCallTime());
}
}
if(fp)
{
fclose(fp);
fp =NULL;
}
std::vector<Runnable*>::iterator it = none_init_vec.begin();
for(;it !=none_init_vec.end();it++)
{
if(*it!=NULL)
{
delete *it;
*it=NULL;
}
}
}
void test_suite::printError(const std::string&msg)
{
if(fp)
{
fprintf(fp,"%s",msg.c_str());
}
}
#endif
これは前回の改良で実現されたもので,C++実装に基づいて変数を定義する必要はない.
使用方法:
(1)テストマクロを開く必要があり、C++エンジニアリングマクロ定義でST_を定義できるTEST_PERFORMANCE
(2)テスト関数fは、以下のようにすればよい
bool CCompile::OutPutReulstFile()
{
STAT_TEST_BEGIN(OutPutReulstFile);// , C ,
// ;
STAT_TEST_END(OutPutReulstFile);// begin
return 0;
}
(3)プログラム起動時にSTAT_を呼び出すGLOBAL_INIT(".")
プロセス中に終了する前、または結果を印刷する必要がある場所でSTAT_を呼び出すGLOBAL_FINISH();テスト結果ファイルは生成されません
これにより追加+aでファイルが生成され、test_performance.csv;ファイル生成パスを指定できます.プロセスが正常に終了すると、デフォルトでファイルが生成されます.
マルチスレッドは現在サポートされていません!!!これは、テストしたモジュールが1つのスレッド呼び出ししかないことを確認するために十分注意してください.