#include "CSVOperator.h"
//////////////////////////////////////////////////////////////////////////
//CSV operator
// Download by http://www.codefans.net
CCSVOperator::CCSVOperator(const char* path)
{
LoadCSV(path);
}
bool CCSVOperator::LoadCSV(const char* path)
{
FILE* pfile = fopen(path, "r");
if (pfile)
{
fseek(pfile,0,SEEK_END);
u32 dwsize = ftell(pfile);
rewind(pfile);
char* filebuffer = new char[dwsize];
fread(filebuffer, 1, dwsize, pfile);
std::map<u32, std::string> StringMap;
char* pBegin = filebuffer;
char* pEnd = strchr(filebuffer, '\n');
u32 uiIndex = 1;
while (pEnd != NULL)
{
std::string strbuff;
strbuff.insert(0, pBegin, pEnd-pBegin);
if (!strbuff.empty())
{
StringMap[uiIndex] = strbuff;
}
pBegin = pEnd + 1;
pEnd = strchr(pEnd + 1, '\n');
++uiIndex;
}
delete[] filebuffer;
std::map<u32, std::string>::iterator iter = StringMap.begin();
for (; iter != StringMap.end(); ++iter)
{
std::vector<std::string> StringVec;
std::map<u32, std::string> l_StringMap;
StringParser::GetParamFromString(iter->second, StringVec);
for (int i = 0; i < StringVec.size(); ++i)
{
l_StringMap[i+1] = StringVec.at(i);
}
m_StringKeyMap[iter->first] = l_StringMap;
}
fclose(pfile);
m_CSVName = path;
return true;
}
return false;
}
bool CCSVOperator::GetInt(u32 uiLine, u32 uiRow, int& iValue)
{
std::string* pKey = GetString(uiLine, uiRow);
if (pKey)
{
iValue = atoi(pKey->c_str());
return true;
}
else
{
return false;
}
}
bool CCSVOperator::GetFloat(u32 uiLine, u32 uiRow, float& fValue)
{
std::string* pKey = GetString(uiLine, uiRow);
if (pKey)
{
fValue = atof(pKey->c_str());
return true;
}
else
{
return false;
}
}
std::string* CCSVOperator::GetString(u32 uiLine, u32 uiRow)
{
std::map<u32, std::map<u32, std::string> >::iterator iterLine = m_StringKeyMap.find(uiLine);
if (iterLine != m_StringKeyMap.end())
{
std::map<u32, std::string>& rStringMap = iterLine->second;
std::map<u32, std::string>::iterator iterRow = rStringMap.find(uiRow);
if (iterRow != rStringMap.end())
{
return &iterRow->second;
}
else
{
return NULL;
}
}
else
{
return NULL;
}
}
bool CCSVOperator::SetNumber(u32 uiLine, u32 uiRow, int iValue)
{
std::string* pKey = GetString(uiLine, uiRow);
if (pKey)
{
char buffer[100];
memset(buffer, 0, sizeof(buffer));
sprintf(buffer, "%d", iValue);
//pKey->clear();
*pKey = buffer;
return true;
}
else
{
return false;
}
}
bool CCSVOperator::SetNumber(u32 uiLine, u32 uiRow, float fValue)
{
std::string* pKey = GetString(uiLine, uiRow);
if (pKey)
{
char buffer[100];
memset(buffer, 0, sizeof(buffer));
sprintf(buffer, "%d", fValue);
//pKey->clear();
*pKey = buffer;
return true;
}
else
{
return false;
}
}
bool CCSVOperator::SetString(u32 uiLine, u32 uiRow, const char* pStr)
{
std::string* pKey = GetString(uiLine, uiRow);
if (pKey)
{
*pKey = pStr;
return true;
}
else
{
return false;
}
}
bool CCSVOperator::SaveCSV(const char* path)
{
if (path != NULL)
{
m_CSVName = path;
}
FILE* pfile = fopen(m_CSVName.c_str(), "w");
if (pfile)
{
std::map<u32, std::map<u32, std::string> >::iterator iter = m_StringKeyMap.begin();
for (; iter != m_StringKeyMap.end(); ++iter)
{
std::map<u32, std::string>& rStringMap = iter->second;
std::map<u32, std::string>::iterator it = rStringMap.begin();
for (; it != rStringMap.end(); ++it)
{
std::string key = it->second;
key += ',';
fwrite(key.c_str(), 1, key.size(), pfile);
}
char Delim = '\n';
fwrite(&Delim, 1, 1, pfile);
}
fclose(pfile);
}
else
{
return false;
}
return true;
}
VC++6.0编译通过的读写CSV文件的代码及实例
5星 · 超过95%的资源 需积分: 31 43 浏览量
2017-09-08
10:06:54
上传
评论 1
收藏 2.72MB ZIP 举报 
csvdx.zip （18个子文件） 
csvdx 
CSVOperator.cpp 5KB csvdx.dsw 518B csvdx.ncb 49KB CSVOperator.h 789B StringParser.h 2KB csvdx.opt 48KB csvdx.dsp 5KB csvdx.plg 116KB Debug 
csvdx.exe 600KB CSVOperator.obj 328KB csvdx.pch 7.21MB csvdx.pdb 1.24MB vc60.idb 217KB vc60.pdb 196KB csvdx.ilk 910KB CSV_OP.obj 291KB CSV_OP.cpp 808B config.csv 125B
评论2
最新资源