iw4x-client/src/Utils/CSV.cpp

147 lines
2.4 KiB
C++
Raw Normal View History

2017-01-19 16:23:59 -05:00
#include "STDInclude.hpp"
namespace Utils
{
CSV::CSV(std::string file, bool isFile, bool allowComments)
{
2017-04-04 16:35:30 -04:00
this->valid = false;
2017-01-19 16:23:59 -05:00
this->parse(file, isFile, allowComments);
}
CSV::~CSV()
{
for (auto row : this->dataMap)
{
for (auto entry : row)
{
entry.clear();
}
row.clear();
}
this->dataMap.clear();
}
int CSV::getRows()
{
return this->dataMap.size();
}
int CSV::getColumns(size_t row)
{
if (this->dataMap.size() > row)
{
return this->dataMap[row].size();
}
return 0;
}
int CSV::getColumns()
{
int count = 0;
for (int i = 0; i < this->getRows(); ++i)
{
count = std::max(this->getColumns(i), count);
}
return count;
}
std::string CSV::getElementAt(size_t row, size_t column)
{
if (this->dataMap.size() > row)
{
auto _row = this->dataMap[row];
if (_row.size() > column)
{
return _row[column];
}
}
return "";
}
void CSV::parse(std::string file, bool isFile, bool allowComments)
{
std::string buffer;
if (isFile)
{
if (!Utils::IO::FileExists(file)) return;
buffer = Utils::IO::ReadFile(file);
2017-04-04 16:35:30 -04:00
this->valid = true;
2017-01-19 16:23:59 -05:00
}
else
{
buffer = file;
}
if (!buffer.empty())
{
auto rows = Utils::String::Explode(buffer, '\n');
for (auto row : rows)
{
this->parseRow(row, allowComments);
}
}
}
void CSV::parseRow(std::string row, bool allowComments)
{
bool isString = false;
std::string element;
std::vector<std::string> _row;
char tempStr = 0;
for (unsigned int i = 0; i < row.size(); ++i)
{
2017-03-20 16:07:55 -04:00
if (row[i] == ',' && !isString) // Flush entry
2017-01-19 16:23:59 -05:00
{
_row.push_back(element);
element.clear();
continue;
}
else if (row[i] == '"') // Start/Terminate string
{
isString = !isString;
continue;
}
else if (i < (row.size() - 1) && row[i] == '\\' &&row[i + 1] == '"' && isString) // Handle quotes in strings as \"
{
tempStr = '"';
++i;
}
else if (!isString && (row[i] == '\n' || row[i] == '\x0D' || row[i] == '\x0A' || row[i] == '\t'))
{
//++i;
continue;
}
else if (!isString && row[i] == '#' && allowComments) // Skip comments. I know CSVs usually don't have comments, but in this case it's useful
{
return;
}
else
{
tempStr = row[i];
}
element.append(&tempStr, 1);
}
// Push last element
_row.push_back(element);
if (_row.size() == 0 || (_row.size() == 1 && !_row[0].size())) // Skip empty rows
{
return;
}
this->dataMap.push_back(_row);
}
}