C++写的一个读取CSV文件的函数

毕业就没有再写过C++程序，真有点遗憾。
今天恰好做一个东西需要一个C++读取CSV文件的函数。
没有找到现成的函数，所以就自己写了一个。
目的就是将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文件。

转载于:https://www.cnblogs.com/c-delight/archive/2005/12/12/295564.html