CSVRow.cpp

Go to the documentation of this file.

/*
 *  This 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 3 of the License, or
 *  any later version.
 *  The software 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 the software. If not, see <http://www.gnu.org/licenses/>.
 *
 *  Copyright 2017, Kuylen E, Willem L, Broeckhove J
 */
#include "CSVRow.h"

#include "CSV.h"

#include <iostream>
#include <ostream>

namespace {

inline bool IsFloat(const std::string& s)
{
        float             a;
        std::stringstream ss(s);
        return (ss >> a) ? true : false;
}

} // namespace

namespace stride {
namespace util {

CSVRow::CSVRow(const CSV* parent, const std::vector<std::string>& values) : m_parent(parent), m_values()
{
        if (values.size() != parent->GetColumnCount()) {
                throw std::runtime_error("Tried adding row with " + ToString(values.size()) + " value(s) to CSV with " +
                                         ToString(parent->GetColumnCount()) + " columns.");
        }
        for (const std::string& value : values) {
                this->m_values.push_back(Trim(value, "\""));
        }
}

template <>
std::string CSVRow::GetValue<std::string>(size_t index) const
{
        if (index >= m_parent->GetColumnCount()) {
                throw std::runtime_error("Index out of range for CSV: " + ToString(index));
        }
        return m_values.at(index);
}

template <>
std::string CSVRow::GetValue<std::string>(const std::string& label) const
{
        size_t index = m_parent->GetIndexForLabel(label);
        return GetValue(index);
}

bool CSVRow::operator==(const CSVRow& other) const { return m_values == other.m_values; }

std::ostream& operator<<(std::ostream& os, const CSVRow& row)
{
        for (unsigned int i = 0; i < row.m_values.size(); ++i) {
                const std::string& value = row.m_values.at(i);
                if (!IsFloat(value)) {
                        os << "\"";
                }
                os << value;
                if (!IsFloat(value)) {
                        os << "\"";
                }
                if (i != row.m_values.size() - 1) {
                        os << ",";
                }
        }
        return os;
}

} // namespace util
} // namespace stride