C++でCSVを１行ずつ読み取るクラス

UEC（電気通信大学）アドベントカレンダーの10日目の記事です。

概要

2年ほど前、C++でCSVファイルを扱うためのクラスを自作し、Qiitaに掲載したのですが、現在も稀にストックされることがあるので、現在私が使っている、もう少し実用性を高めたバージョンも記します。

ライブラリを追加するのとかがめんどくさい時、簡易的な読み取りであれば、これで十分かと思います。
１行ずつ読み取るので、巨大なCSVファイルでも読み取ることができます。

コード

loadcsv.cpp

#include <string>
#include <vector>

using namespace std;

typedef vector<string> CellsTy;

CellsTy cells_split(char* line, char comma) {

    size_t len = strlen(line);
    line[len - 1] = comma;
    line[len] = '\0';
    int last = 0;

    CellsTy cells;
    string l;

    for (int i = 0; i < len; i++) {
        if (line[i] == '\r') continue;
        if (line[i] == comma) {
            cells.push_back(l);
            last = i;
            l.clear();
            continue;
        }
        l.push_back(line[i]);
    }
    return cells;
}

使い方

必要な標準ライブラリを読み込んだ後に、上記コードをペーストし、main関数のように1行ずつ読み取ります。
下の使用例は、data.csvというCSVを１行ずつ読み取り、各セルを出力するプログラムです。

main.cpp

#include <string>
#include <vector>
#include <iostream>
#include <cstdio>

using namespace std;

typedef vector<string> CellsTy;

CellsTy cells_split(char* line, char comma) {

    size_t len = strlen(line);
    line[len - 1] = comma;
    line[len] = '\0';
    int last = 0;

    CellsTy cells;
    string l;

    for (int i = 0; i < len; i++) {
        if (line[i] == '\r') continue;
        if (line[i] == comma) {
            cells.push_back(l);
            last = i;
            l.clear();
            continue;
        }
        l.push_back(line[i]);
    }
    return cells;
}

const int MaxLine = 1024; // 読み取るファイルの１行あたりの最大文字数

int main() {
    FILE* fp = fopen("data.csv", "r");
    if (fp == NULL) return 1;
    while (!feof(fp)) {
        char line[MaxLine];
        fgets(line, MaxLine, fp);
        CellsTy cells = cells_split(line, ',');
        if (cells.size() == 0) continue;

        // セルの出力処理
        for (int i = 0; i < cells.size(); i++) {
            cout << cells[i] << "   ";
        }
        cout << endl;
    }
}

少し汎用性を持たせる

使用する型をtypedefで別途定義することで、汎用性を持たせました。wstringを使用する場合は、typedef~の行を変更してください。

loadcsv.cpp

#include <string>
#include <vector>
#include <cstdio>

using namespace std;

typedef string cells_string;
typedef char cells_char;
typedef vector<cells_string> CellsTy;

CellsTy cells_split(cells_char* line, cells_char comma) {

    size_t len = strlen(line);
    line[len - 1] = comma;
    line[len] = '\0';
    int last = 0;

    CellsTy cells;
    cells_string l;

    for (int i = 0; i < len; i++) {
        if (line[i] == '\r') continue;
        if (line[i] == comma) {
            cells.push_back(l);
            last = i;
            l.clear();
            continue;
        }
        l.push_back(line[i]);
    }
    return cells;
}

template関数にする

さらに、引数の型をtemplate化したバージョンです。
char型にもwchar_t型にも対応することができます。

loadcsv.cpp

#include <string>
#include <vector>

using namespace std;

template <typename cell_char> vector<basic_string<cell_char>> cells_split(cell_char* line, cell_char comma) {

    size_t len = strlen(line);
    line[len - 1] = comma;
    line[len] = '\0';
    int last = 0;

    CellsTy cells;
    basic_string<cell_char> l;

    for (int i = 0; i < len; i++) {
        if (line[i] == '\r') continue;
        if (line[i] == comma) {
            cells.push_back(l);
            last = i;
            l.clear();
            continue;
        }
        l.push_back(line[i]);
    }
    return cells;
}

こうした方がよいという改善案や、挙動がおかしな点がありましたらコメントお願い致します。