sqlitebrowser/src/csvparser.h

#ifndef CSVPARSER_H
#define CSVPARSER_H

#include <QChar>
#include <QVector>
#include <QStringList>

class QTextStream;

/*!
 * \brief The CSVProgress class
 *
 * This is an abstract class you can provide overriden provde
 * to the CSVParser to get progress updates.
 */
class CSVProgress
{
public:
    virtual ~CSVProgress() { }

    virtual void start() = 0;
    virtual bool update(size_t pos) = 0;
    virtual void end() = 0;
};

class CSVParser
{
public:
    typedef QVector<QStringList> TCSVResult;

    CSVParser(bool trimfields = true, const QChar& fieldseparator = ',', const QChar& quotechar = '"');
    ~CSVParser();

    /*!
     * \brief parse the given stream
     * \param stream Stream with the CSV parser
     * \param nMaxRecords Max records too read, -1 if unlimited
     * \return True if parsing worked.
     */
    bool parse(QTextStream& stream, int64_t nMaxRecords = -1);

    /*!
     * \brief csv
     * \return The parse result
     */
    const TCSVResult& csv() const { return m_vCSVData; }

    /*!
     * \brief columns
     * \return Number of columns parsed
     */
    size_t columns() const { return m_nColumns; }

    void setCSVProgress(CSVProgress* csvp) { m_pCSVProgress = csvp; }

private:
    enum ParseStates
    {
        StateNormal,
        StateInQuote,
        StateEndQuote
    };

    inline void addRow(QStringList& r)
    {
        m_vCSVData.append(r);
        m_nColumns = std::max<size_t>(r.size(), m_nColumns);
        r.clear();
    }

private:
    bool m_bTrimFields;
    QChar m_cFieldSeparator;
    QChar m_cQuoteChar;
    CSVProgress* m_pCSVProgress;

    TCSVResult m_vCSVData;
    size_t m_nColumns;

    size_t m_nBufferSize; //! internal buffer read size
};

#endif // CSVPARSER_H