devel/libmdw/trunk/libmdw/src/CsvParser.cpp

/***************************************************************************
 *   Copyright (C) 2010 by Jean-Baptiste Wons   *
 *   wonsjb@gmail.com   *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
#include "CsvParser.h"
#include <iostream>
#include <cstdlib>
#include <fstream>

using namespace std;

namespace mdw
{

CsvParser::CsvParser (const string &fileName) : _file (fileName.c_str())
{
    getline (_file, _currentLine);
    _nPos = 0;
    string header;
    while (true)
    {
        *this >> header;
        if (header.empty())
        {
            break;
        }
        _header.push_back (header);
    }
}

bool CsvParser::nextLine()
{
    if (_file.eof())
    {
        return false;
    }
    getline (_file, _currentLine);
    if (_currentLine.empty())
    {
        return false;
    }
}

void CsvParser::skipSpaces (void)
{
    while (_nPos < _currentLine.length() && _currentLine[_nPos] == ' ')
        _nPos++;
}

CsvParser & CsvParser::operator >> (int & nOut)
{
    string sTmp = "";
    skipSpaces();
    while (_nPos < _currentLine.length() && _currentLine[_nPos] != ',')
        sTmp += _currentLine[_nPos++];

    _nPos++; // skip past comma
    nOut = atoi (sTmp.c_str());
    return *this;
}

CsvParser & CsvParser::operator >> (long & nOut)
{
    string sTmp = "";
    skipSpaces();
    while (_nPos < _currentLine.length() && _currentLine[_nPos] != ',')
        sTmp += _currentLine[_nPos++];

    _nPos++; // skip past comma
    nOut = atoi (sTmp.c_str());
    return *this;
}

CsvParser & CsvParser::operator >> (bool & out)
{
    string sTmp = "";
    skipSpaces();
    while (_nPos < _currentLine.length() && _currentLine[_nPos] != ',')
        sTmp += _currentLine[_nPos++];

    _nPos++; // skip past comma
    out = (sTmp == "true");
    return *this;
}

CsvParser & CsvParser::operator >> (double & nOut)
{
    string sTmp = "";
    skipSpaces();
    while (_nPos < _currentLine.length() && _currentLine[_nPos] != ',')
        sTmp += _currentLine[_nPos++];

    _nPos++; // skip past comma
    nOut = atof (sTmp.c_str());
    return *this;
}

CsvParser & CsvParser::operator >> (string & sOut)
{
    bool bQuotes = false;
    sOut = "";
    skipSpaces();

    // Jump past first " if necessary
    if (_nPos < _currentLine.length() && _currentLine[_nPos] == '"')
    {
        bQuotes = true;
        _nPos++;
    }

    while (_nPos < _currentLine.length())
    {
        if (!bQuotes && _currentLine[_nPos] == ',')
            break;
        if (bQuotes && _currentLine[_nPos] == '"')
        {
            if (_nPos + 1 >= _currentLine.length() - 1)
                break;
            if (_currentLine[_nPos+1] == ',')
                break;
        }
        sOut += _currentLine[_nPos++];
    }

    // Jump past last " if necessary
    if (bQuotes && _nPos < _currentLine.length() && _currentLine[_nPos] == '"')
        _nPos++;

    // Jump past , if necessary
    if (_nPos < _currentLine.length() && _currentLine[_nPos] == ',')
        _nPos++;

    return *this;
}
}

---