xlnt/benchmarks/memory.cpp

66 lines
1.2 KiB
C++
Raw Normal View History

2016-02-06 23:04:41 +08:00
#include <cassert>
#ifdef __APPLE__
#include<mach/mach.h>
#endif
#include <xlnt/xlnt.hpp>
#include "../tests/helpers/path_helper.hpp"
int calc_memory_usage()
{
#ifdef __APPLE__
struct task_basic_info t_info;
mach_msg_type_number_t t_info_count = TASK_BASIC_INFO_COUNT;
if (KERN_SUCCESS != task_info(mach_task_self(),
TASK_BASIC_INFO, (task_info_t)&t_info,
&t_info_count))
{
return 0;
}
return t_info.virtual_size;
#endif
return 0;
}
void test_memory_use()
{
// Naive test that assumes memory use will never be more than 120 % of
// that for first 50 rows
auto current_folder = PathHelper::GetExecutableDirectory();
auto src = current_folder + "rks/files/very_large.xlsx";
xlnt::workbook wb;
wb.load(src);
auto ws = wb.get_active_sheet();
int initial_use = 0;
int n = 0;
for (auto line : ws.rows())
{
if (n % 50 == 0)
{
auto use = calc_memory_usage();
if (initial_use == 0)
{
initial_use = use;
}
assert(use / initial_use < 1.2);
std::cout << n << " " << use << std::endl;
}
n++;
}
}
int main()
{
test_memory_use();
}