EigenFileIO.tpp

Go to the documentation of this file.

template<typename Scalar, int RowsAtCompileTime, int ColsAtCompileTime>
inline bool loadMatrix(std::string filename, Eigen::Matrix<Scalar,RowsAtCompileTime,ColsAtCompileTime>& m)
{
  
  // General structure
  // 1. Read file contents into vector<double> and count number of lines
  // 2. Initialize matrix
  // 3. Put data in vector<double> into matrix
  
  std::ifstream input(filename.c_str());
  if (input.fail())
  {
    std::cerr << "ERROR. Cannot find file '" << filename << "'." << std::endl;
    m = Eigen::Matrix<Scalar,RowsAtCompileTime,ColsAtCompileTime>(0,0);
    return false;
  }
  std::string line;
  Scalar d;
  
  std::vector<Scalar> v;
  int n_rows = 0;
  while (getline(input, line))
  {
    ++n_rows;
    std::stringstream input_line(line);
    while (!input_line.eof())
    {
      input_line >> d;
      v.push_back(d);
    }
  }
  input.close();
  
  int n_cols = v.size()/n_rows;
  m = Eigen::Matrix<Scalar,RowsAtCompileTime,ColsAtCompileTime>(n_rows,n_cols);
  
  for (int i=0; i<n_rows; i++)
    for (int j=0; j<n_cols; j++)
      m(i,j) = v[i*n_cols + j];
    
  return true;
}

template<typename Scalar, int RowsAtCompileTime, int ColsAtCompileTime>
inline bool saveMatrix(std::string directory, std::string filename, Eigen::Matrix<Scalar,RowsAtCompileTime,ColsAtCompileTime> matrix, bool overwrite)
{
  if (directory.empty())
    return false;

  
  if (!boost::filesystem::exists(directory))
  {
    // Directory doesn't exist. Try to create it.
    if (!boost::filesystem::create_directories(directory))
    {
      std::cerr << "Couldn't make directory '" << directory << "'. Not saving data." << std::endl;
      return false;
    }
  }
  
  filename = directory+"/"+filename;
  saveMatrix(filename,matrix,overwrite);
  
  return true;
}

template<typename Scalar, int RowsAtCompileTime, int ColsAtCompileTime>
inline bool saveMatrix(std::string filename, Eigen::Matrix<Scalar,RowsAtCompileTime,ColsAtCompileTime> matrix, bool overwrite)
{
  if (boost::filesystem::exists(filename))
  {
    if (!overwrite)
    {
      // File exists, but overwriting is not allowed. Abort.
      std::cerr << "File '" << filename << "' already exists. Not saving data." << std::endl;
      return false;
    }
  }
  
  
  std::ofstream file;
  file.open(filename.c_str());
  if (!file.is_open())
  {
    std::cerr << "Couldn't open file '" << filename << "' for writing." << std::endl;
    return false;
  }
  
  file << std::fixed;
  file << matrix;
  file.close();
  
  return true;
  
}