CsvReader/src/include/CSV/Reader.h

Go to the documentation of this file.

/****
 * Reader.h
 *
 * Copyright 2021 mikee47 <mike@sillyhouse.net>
 *
 * This file is part of the CsvReader Library
 *
 * This library 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, version 3 or later.
 *
 * This library 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 library.
 * If not, see <https://www.gnu.org/licenses/>.
 *
 ****/
 
#pragma once
 
#include "Parser.h"
#include <memory>
 
namespace CSV
{
class Reader : private Parser
{
public:
    Reader(IDataSourceStream* source, const Options& options, const CStringArray& headings = nullptr);
 
    Reader(IDataSourceStream* source, char fieldSeparator = ',', const CStringArray& headings = nullptr,
           uint16_t maxLineLength = 2048)
        : Reader(source,
                 Options{
                     .lineLength = maxLineLength,
                     .fieldSeparator = fieldSeparator,
                 },
                 headings)
    {
    }
 
    void reset()
    {
        seek(BOF);
    }
 
    bool next()
    {
        return source ? readRow(*source) : false;
    }
 
    unsigned count() const
    {
        return headings.count();
    }
 
    using Parser::getRow;
 
    const char* getValue(unsigned index) const
    {
        return getRow()[index];
    }
 
    const char* getValue(const char* name) const
    {
        return getValue(getColumn(name));
    }
 
    int getColumn(const char* name) const
    {
        return headings.indexOf(name);
    }
 
    explicit operator bool() const
    {
        return bool(source);
    }
 
    const CStringArray& getHeadings() const
    {
        return headings;
    }
 
    using Parser::getCursor;
 
    using Parser::tell;
 
    bool seek(int offset);
 
    bool seek(const Cursor& cursor)
    {
        return seek(cursor.start);
    }
 
private:
    std::unique_ptr<IDataSourceStream> source;
    CStringArray headings;
    unsigned start{0}; 
};
 
} // namespace CSV