125 lines
3.3 KiB
C++
125 lines
3.3 KiB
C++
#include "bench_framework.hpp"
|
|
#include <cstring>
|
|
#include <cstdlib>
|
|
#include "../plugins/input/csv/csv_utils.hpp"
|
|
|
|
class test : public benchmark::test_case
|
|
{
|
|
public:
|
|
std::string line_data_;
|
|
test(mapnik::parameters const& params)
|
|
: test_case(params),
|
|
line_data_("this is one line\nand this is a second line\nand a third line")
|
|
{
|
|
boost::optional<std::string> line_data = params.get<std::string>("line");
|
|
if (line_data)
|
|
{
|
|
line_data_ = *line_data;
|
|
}
|
|
}
|
|
|
|
bool validate() const
|
|
{
|
|
std::string first = line_data_.substr(line_data_.find_first_not_of('\n'));
|
|
char newline = '\n';
|
|
std::string csv_line;
|
|
std::stringstream s;
|
|
s << line_data_;
|
|
std::getline(s,csv_line,newline);
|
|
if (csv_line != first)
|
|
{
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
std::clog << "Error: the parsed line (" << csv_line << ") should be a subset of the original line (" << line_data_ << ") (ensure you pass a line with a \\n)\n";
|
|
}
|
|
return true;
|
|
}
|
|
bool operator()() const
|
|
{
|
|
char newline = '\n';
|
|
std::string csv_line;
|
|
std::stringstream s;
|
|
s << line_data_;
|
|
for (unsigned i=0;i<iterations_;++i)
|
|
{
|
|
std::getline(s,csv_line,newline);
|
|
}
|
|
return true;
|
|
}
|
|
};
|
|
|
|
|
|
class test2 : public benchmark::test_case
|
|
{
|
|
public:
|
|
std::string line_data_;
|
|
test2(mapnik::parameters const& params)
|
|
: test_case(params),
|
|
line_data_("this is one line\nand this is a second line\nand a third line")
|
|
{
|
|
boost::optional<std::string> line_data = params.get<std::string>("line");
|
|
if (line_data)
|
|
{
|
|
line_data_ = *line_data;
|
|
}
|
|
}
|
|
|
|
bool validate() const
|
|
{
|
|
std::string first = line_data_.substr(line_data_.find_first_not_of('\n'));
|
|
char newline = '\n';
|
|
char quote = '"';
|
|
std::string csv_line;
|
|
std::stringstream s;
|
|
s << line_data_;
|
|
csv_utils::getline_csv(s,csv_line,newline,quote);
|
|
if (csv_line != first)
|
|
{
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
std::clog << "Error: the parsed line (" << csv_line << ") should be a subset of the original line (" << line_data_ << ") (ensure you pass a line with a \\n)\n";
|
|
}
|
|
return true;
|
|
}
|
|
bool operator()() const
|
|
{
|
|
char newline = '\n';
|
|
char quote = '"';
|
|
std::string csv_line;
|
|
std::stringstream s;
|
|
s << line_data_;
|
|
for (unsigned i=0;i<iterations_;++i)
|
|
{
|
|
csv_utils::getline_csv(s,csv_line,newline,quote);
|
|
}
|
|
return true;
|
|
}
|
|
};
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
int return_value = 0;
|
|
try
|
|
{
|
|
mapnik::parameters params;
|
|
benchmark::handle_args(argc,argv,params);
|
|
{
|
|
test test_runner(params);
|
|
return_value = return_value | run(test_runner,"std::getline");
|
|
}
|
|
{
|
|
test2 test_runner2(params);
|
|
return_value = return_value | run(test_runner2,"csv_utils::getline_csv");
|
|
}
|
|
}
|
|
catch (std::exception const& ex)
|
|
{
|
|
std::clog << ex.what() << "\n";
|
|
return -1;
|
|
}
|
|
return return_value;
|
|
}
|