icinga2/test_drivers/performance/iconvtest.cpp

133 lines
3.7 KiB
C++

#include <iconv.h>
#include <string.h>
#include "../../source/utf8.h"
#include "timer.h"
#include <fstream>
#include <algorithm>
#include <vector>
using namespace std;
using namespace utf8;
int main(int argc, char** argv)
{
if (argc != 2) {
cout << "\nUsage: iconvtest filename\n";
return 0;
}
const char* test_file_path = argv[1];
// Open the test file (UTF-8 encoded text)
ifstream fs8(test_file_path, ios::binary);
if (!fs8.is_open()) {
cout << "Could not open " << test_file_path << endl;
return 0;
}
// get length
fs8.seekg(0, ios::end);
int length = fs8.tellg();
fs8.seekg(0, ios::beg);
// allocate the buffer (no vector - we are benchmarking conversions, not STL
char* buf = new char[length];
char* end_buf = buf + length;
// fill the data
fs8.read(buf, length);
fs8.close();
// the UTF-16 result will not be larger than this (I hope :) )
vector<unsigned char> temputf16;
utf8::utf8to16(buf, end_buf, back_inserter(temputf16));
int wlength = temputf16.size();
unsigned short* utf16buf = new unsigned short[wlength];
cout << "UTF8 to UTF-16\n";
{
memset (utf16buf, 0 , wlength * sizeof(unsigned short));
// utf-8 cpp:
cout << "utf8::utf8to16: ";
timer t(cout);
utf8::utf8to16(buf, buf + length, utf16buf);
t.print_time();
}
{
memset (utf16buf, 0 , wlength * sizeof(unsigned short));
// utf-8 cpp:
cout << "unchecked::utf8to16: ";
timer t(cout);
utf8::unchecked::utf8to16(buf, buf + length, utf16buf);
t.print_time();
}
// the UTF-16 result will not be larger than this (I hope :) )
unsigned short* utf16iconvbuf = new unsigned short[wlength];
{
memset (utf16iconvbuf, 0 , wlength * sizeof(unsigned short));
// iconv
cout << "iconv: ";
iconv_t cd = iconv_open("UTF-16LE", "UTF-8");
if (cd == iconv_t(-1)) {
cout << "Error openning the iconv stream";
return 0;
}
char* inbuf = buf;
size_t in_bytes_left = length;
char* outbuf = (char*)utf16iconvbuf;
size_t out_bytes_left = wlength * sizeof (unsigned short);
{
timer t(cout);
iconv(cd, &inbuf, &in_bytes_left, &outbuf, &out_bytes_left);
t.print_time();
}
iconv_close(cd);
}
// just check the correctness while we are here:
if (!equal(utf16buf, utf16buf + wlength, utf16iconvbuf))
cout << "Different result!!!\n";
// the other way around
cout << "UTF16 to UTF-8\n";
{
//iconv
memset(buf, 0, length);
cout<< "iconv: ";
iconv_t cd = iconv_open("UTF-8", "UTF-16LE");
if (cd == iconv_t(-1)) {
cout << "Error openning the iconv stream";
return 0;
}
char* inbuf = (char*)utf16buf;
size_t in_bytes_left = wlength * sizeof(unsigned short);
char* outbuf =buf;
size_t out_bytes_left = length;
{
timer t(cout);
iconv(cd, &inbuf, &in_bytes_left, &outbuf, &out_bytes_left);
t.print_time();
}
iconv_close(cd);
}
{
memset (buf, 0 , length);
// utf-8 cpp:
cout << "unchecked::utf16to8: ";
timer t(cout);
utf8::unchecked::utf16to8(utf16buf, utf16buf + wlength, buf);
t.print_time();
}
{
memset (buf, 0 , length);
cout << "utf16to8: ";
timer t(cout);
utf8::utf16to8(utf16buf, utf16buf + wlength, buf);
t.print_time();
}
delete [] buf;
delete [] utf16buf;
}