自定函数实现C++读取CSV文件

自定函数实现C++读取CSV文件
http://developer.51cto.com  2010-02-01 13:26  佚名  博客园  我要评论(0)
C++读取CSV文件需要一个合适的函数来进行具体的操作。如果没有现成的函数供我们使用，我们可以自己编写一个来方便我们的使用。

C++是一个应用方式非常灵活的编程语言。我们可以通过自己的需求，自己编写一个适合自己开发的函数来实现一些特定的功能。比如今天为大家介绍的就是自定义函数来实现C++读取CSV文件这一功能。
C++静态构造函数相关概念解析
C++ Traits应用技巧分享
C++虚函数示例解读
C++函数指针与C#委托之间有何联系
C++ explicit关键字应用于构造函数

今天恰好做一个东西需要一个C++读取CSV文件的函数。

没有找到现成的函数，所以就自己写了一个。

目的就是实现C++读取CSV文件，每一行读到一个数组中。

有点生疏了，字符串处理函数都不知道有什么现成的，所以也写了两个辅助的字符串处理函数。

第一个函数：计算一个字符串中的某个子串的个数
int strstr_cnt(const char *string, const char
*substring) { 
int i,j,k,count = 0; 
for (i = 0; string[i]; i++){ 
for (j = i, k = 0; (string[j] ==
substring[k] && (j < strlen(string))); j++,k++) { 
if (! substring[k + 1]) { 
count++; 
} 
} 
} 
return count; 
}

第二个函数：计算一个子串在字符串中的位置
int substring_index(const char *s1,const char *s2, int pos){ 
int i,j,k; 
for( i = pos ; s1[i] ; i++ ) { 
for( j = i, k = 0 ; s1[j] == s2[k]; j++,k++ ){ 
if (! s2[k + 1]) { 
return i; 
} 
} 
} 
return -1; 
}

第三个函数：读取已经打开的一个CSV文件的一行，将这一行处理到一个数组中。
char *fgetcsvline(vector<string> &csv_databuf, FILE *fhead) { 
char *ret_stat; 
char data_buf[1024]; 
string stringbuf; 
ret_stat = fgets( data_buf, 1024, fhead ); 
if (ret_stat != NULL) { 
int len = strstr_cnt(data_buf,"/",/""); 
if (len > 0){ 
int pos = substring_index(data_buf,"/",/"",0); 
int startpos = 1; 
string csv_buf; 
while (pos > 0) { 
stringbuf = (string)data_buf; 
csv_buf = stringbuf.substr(startpos,pos - startpos); 
csv_databuf.push_back(csv_buf); 
startpos = pos + 3; 
pos = substring_index(data_buf,"/",/"",pos + 2); 
} 
if ((substring_index(data_buf,"/n",0)) > 0){ 
csv_buf = stringbuf.substr(startpos, stringbuf.length()
- startpos - 2); 
} else { 
csv_buf = stringbuf.substr(startpos, stringbuf.length()
- startpos - 1); 
} 
csv_databuf.push_back(csv_buf); 
} 
} 
return ret_stat;  
}

这个函数使用上面的两个函数来处理字符串。

另外这个函数用来处理的CSV文件是带双引号格式的：
"sss","ddd","444" 
"ttt","www","ooo" 
"sss","qqq","000"

使用方法如下：
int main(int argc, char* argv[]) { 
FILE *fp_head; 
string csvFileName = "test.csv"; 
char *ret_stat; 
vector<string> csv_data; 
fp_head = fopen( csvFileName, "rt" ); 
ret_stat = fgetcsvline(csv_data, fp_head); 
while (ret_stat != NULL) { 
//get csv data use csv_data[n] 
ret_stat = fgetcsvline(csv_data, fp_head); 
} 
return 0; 
}

以上代码可能需要稍加调试。也可稍加改动用来读取其它格式的csv文件。

以上就是C++读取CSV文件的相关实现方法。