OpenAssetTools/src/ObjCommon/Csv/CsvStream.cpp

#include "CsvStream.h"

#include "Utils/StringUtils.h"

#include <cstdlib>
#include <sstream>

constexpr char CSV_SEPARATOR = ',';

CsvCell::CsvCell(std::string value)
    : m_value(std::move(value))
{
}

bool CsvCell::AsFloat(float& out) const
{
    const char* startPtr = m_value.c_str();
    char* endPtr;
    out = std::strtof(startPtr, &endPtr);

    if (endPtr == startPtr)
        return false;

    for (const auto* c = endPtr; *c; c++)
    {
        if (!isspace(*c))
            return false;
    }

    return true;
}

bool CsvCell::AsInt32(int32_t& out) const
{
    const char* startPtr = m_value.c_str();
    char* endPtr;
    out = std::strtol(startPtr, &endPtr, 0);

    if (endPtr == startPtr)
        return false;

    for (const auto* c = endPtr; *c; c++)
    {
        if (!isspace(*c))
            return false;
    }

    return true;
}

bool CsvCell::AsUInt32(uint32_t& out) const
{
    const char* startPtr = m_value.c_str();
    char* endPtr;
    out = std::strtoul(startPtr, &endPtr, 0);

    if (endPtr == startPtr)
        return false;

    for (const auto* c = endPtr; *c; c++)
    {
        if (!isspace(*c))
            return false;
    }

    return true;
}

CsvInputStream::CsvInputStream(std::istream& stream)
    : m_stream(stream)
{
}

bool CsvInputStream::NextRow(std::vector<CsvCell>& out) const
{
    if (!out.empty())
        out.clear();

    return EmitNextRow(
        [&out](std::string value)
        {
            out.emplace_back(std::move(value));
        });
}

bool CsvInputStream::NextRow(std::vector<std::string>& out) const
{
    if (!out.empty())
        out.clear();

    return EmitNextRow(
        [&out](std::string value)
        {
            out.emplace_back(std::move(value));
        });
}

bool CsvInputStream::NextRow(std::vector<const char*>& out, MemoryManager& memory) const
{
    if (!out.empty())
        out.clear();

    return EmitNextRow(
        [&out, &memory](const std::string& value)
        {
            out.emplace_back(memory.Dup(value.c_str()));
        });
}

bool CsvInputStream::EmitNextRow(const std::function<void(std::string)>& cb) const
{
    auto c = m_stream.get();
    const auto isEof = c == EOF;
    std::ostringstream col;
    auto content = false;
    while (c != EOF)
    {
        if (c == CSV_SEPARATOR)
        {
            auto value = col.str();
            utils::StringTrimR(value);
            cb(std::move(value));
            col.clear();
            col.str(std::string());
            content = false;
        }
        else if (c == '\r')
        {
            c = m_stream.get();
            if (c == '\n')
                break;
            col << '\r';
        }
        else if (c == '\n')
        {
            break;
        }
        else if (isspace(c))
        {
            if (content)
                col << static_cast<char>(c);
        }
        else
        {
            content = true;
            col << static_cast<char>(c);
        }

        c = m_stream.get();
    }

    if (!isEof)
    {
        auto value = col.str();
        utils::StringTrimR(value);
        cb(std::move(value));
    }

    return !isEof;
}

CsvOutputStream::CsvOutputStream(std::ostream& stream)
    : m_stream(stream),
      m_column_count(0),
      m_current_column(0),
      m_first_row(true)
{
}

void CsvOutputStream::WriteColumn(const std::string& value)
{
    if (m_current_column++ > 0)
        m_stream << CSV_SEPARATOR;

    auto containsSeparator = false;
    auto containsQuote = false;
    for (const auto& c : value)
    {
        if (c == '"')
        {
            containsQuote = true;
            break;
        }

        if (c == CSV_SEPARATOR)
            containsSeparator = true;
    }

    if (containsQuote)
    {
        m_stream << "\"";
        for (const auto& c : value)
        {
            if (c == '"')
                m_stream << "\"\"";
            else
                m_stream << c;
        }

        m_stream << "\"";
    }
    else if (containsSeparator)
    {
        m_stream << "\"" << value << "\"";
    }
    else
    {
        m_stream << value;
    }
}

void CsvOutputStream::NextRow()
{
    if (m_first_row)
    {
        m_first_row = false;
        m_column_count = m_current_column;
    }
    else
    {
        while (m_current_column < m_column_count)
        {
            m_stream << CSV_SEPARATOR;
            m_current_column++;
        }
    }

    m_stream << "\n";
    m_current_column = 0;
}