80 lines
2.4 KiB
C++
80 lines
2.4 KiB
C++
// test063.cpp - custom conversion per call
|
|
|
|
#include <iomanip>
|
|
#include <math.h>
|
|
#include <rapidcsv.h>
|
|
#include "unittest.h"
|
|
|
|
// Conversion function for int data to be converted to fixed-point two decimal numbers
|
|
void ToVal(const std::string& pStr, int& pVal)
|
|
{
|
|
pVal = static_cast<int>(roundf(100.0f * std::stof(pStr)));
|
|
}
|
|
|
|
struct Struct
|
|
{
|
|
int val = 0;
|
|
};
|
|
|
|
void ToStruct(const std::string& pStr, Struct& pVal)
|
|
{
|
|
pVal.val = static_cast<int>(roundf(100.0f * std::stof(pStr)));
|
|
}
|
|
|
|
int main()
|
|
{
|
|
int rv = 0;
|
|
|
|
std::string csv =
|
|
"-,A,B,C\n"
|
|
"1,1,10,100,1000\n"
|
|
"2,0.1,0.01,0.001,0.006\n"
|
|
;
|
|
|
|
std::string path = unittest::TempPath();
|
|
unittest::WriteFile(path, csv);
|
|
|
|
try
|
|
{
|
|
rapidcsv::Document doc(path, rapidcsv::LabelParams(0, 0));
|
|
|
|
// ToVal
|
|
unittest::ExpectEqual(int, doc.GetCell<int>(0, 0, ToVal), 100);
|
|
unittest::ExpectEqual(int, doc.GetCell<int>(1, "1", ToVal), 1000);
|
|
unittest::ExpectEqual(int, doc.GetCell<int>("A", 0, ToVal), 100);
|
|
unittest::ExpectEqual(int, doc.GetCell<int>("B", "1", ToVal), 1000);
|
|
|
|
unittest::ExpectEqual(int, doc.GetRow<int>(0, ToVal).at(0), 100);
|
|
unittest::ExpectEqual(int, doc.GetRow<int>("2", ToVal).at(0), 10);
|
|
|
|
unittest::ExpectEqual(int, doc.GetColumn<int>(0, ToVal).at(0), 100);
|
|
unittest::ExpectEqual(int, doc.GetColumn<int>("B", ToVal).at(0), 1000);
|
|
|
|
// ToStruct
|
|
unittest::ExpectEqual(int, doc.GetCell<Struct>(0, 0, ToStruct).val, 100);
|
|
unittest::ExpectEqual(int, doc.GetCell<Struct>(1, "1", ToStruct).val, 1000);
|
|
unittest::ExpectEqual(int, doc.GetCell<Struct>("A", 0, ToStruct).val, 100);
|
|
unittest::ExpectEqual(int, doc.GetCell<Struct>("B", "1", ToStruct).val, 1000);
|
|
|
|
unittest::ExpectEqual(int, doc.GetRow<Struct>(0, ToStruct).at(0).val, 100);
|
|
unittest::ExpectEqual(int, doc.GetRow<Struct>("2", ToStruct).at(0).val, 10);
|
|
|
|
unittest::ExpectEqual(int, doc.GetColumn<Struct>(0, ToStruct).at(0).val, 100);
|
|
unittest::ExpectEqual(int, doc.GetColumn<Struct>("B", ToStruct).at(0).val, 1000);
|
|
|
|
// Missing custom conversion function
|
|
ExpectException(doc.GetColumn<bool>(0), rapidcsv::no_converter);
|
|
ExpectExceptionMsg(doc.GetColumn<bool>(0), rapidcsv::no_converter,
|
|
"unsupported conversion datatype");
|
|
}
|
|
catch (const std::exception& ex)
|
|
{
|
|
std::cout << ex.what() << std::endl;
|
|
rv = 1;
|
|
}
|
|
|
|
unittest::DeleteFile(path);
|
|
|
|
return rv;
|
|
}
|