csvread.h

Go to the documentation of this file.

// ========================================================================= //
// This file is part of MyraMath, copyright (c) 2014-2019 by Ryan A Chilton  //
// and distributed by MyraCore, LLC. See LICENSE.txt for license terms.      //
// ========================================================================= //

#ifndef MYRAMATH_DENSE_CSVREAD_H
#define MYRAMATH_DENSE_CSVREAD_H

#include <myramath/utility/Number.h>
#include <myramath/MYRAMATH_EXPORT.h>
#include <myramath/utility/eprintf.h>

#include <myramath/expression/vector.h>
#include <myramath/expression/transpose.h>
#include <myramath/expression/reshape.h>

#include <vector>
#include <fstream>
#include <sstream>

namespace myra {

// Forward declarations.
template<class Number> class Matrix;
template<class Number> class Vector;

template<class Number> Matrix<Number> csvread_Matrix(const char* filename)
  {
  std::ifstream in(filename);
  // Size and numeric values.
  int I = 0;
  std::vector<int> J;
  std::vector<Number> v;
  // For each row..
  std::string rowstr;
  while (std::getline(in,rowstr))
    {
    std::stringstream row(rowstr);
    int j = 0;
    // For each cell on this row..
    std::string cellstr;    
    while(std::getline(row,cellstr,','))
      {
      std::stringstream cell(cellstr);
      Number n;      
      cell >> n;      
      v.push_back(n);
      ++j;
      }
    J.push_back(j);
    ++I;      
    }
  // Verify every row had the same number of columns.
  for (int i = 1; i < I; ++i)
    if (J[i] != J[0])
      throw eprintf("csvread_Matrix(), size mismatch between rows J[%d] != J[0] [%d != %d]",i,J[i],J[0]);
  // Cast into an expression, reshape, transpose, and populate output Matrix.
  return Matrix<Number>::evaluate( transpose(reshape(expr(v),J[0],I) ) );
  }

template<class Number> Vector<Number> csvread_Vector(const char* filename)
  {
  std::ifstream in(filename);
  std::vector<Number> v;
  Number n;
  while (in >> n)
    v.push_back(n);
  in.close();
  return Vector<Number>::evaluate(expr(v));
  }

template<class T> std::vector<T> csvread_vector(const char* filename)
  {
  std::ifstream in(filename);
  std::vector<T> v;
  T t;
  while (in >> t)
    v.push_back(t);
  in.close();
  return v;
  }

} // namespace

#endif