Utils.cpp

Go to the documentation of this file.

#include "utils.h"
#include <armadillo>
#include <boost/log/trivial.hpp>
#include <boost/program_options.hpp>
#include <fstream>

using namespace std;
using namespace arma;

arma::sp_mat Utils::resize_patch(const arma::sp_mat &Mat, const unsigned int nR,
                                 const unsigned int nC) {
  arma::sp_mat MMat(nR, nC);
  MMat.zeros();
  if (nR >= Mat.n_rows && nC >= Mat.n_cols) {
    if (Mat.n_rows >= 1 && Mat.n_cols >= 1)
      MMat.submat(0, 0, Mat.n_rows - 1, Mat.n_cols - 1) = Mat;
  } else {
    if (nR <= Mat.n_rows && nC <= Mat.n_cols)
      MMat = Mat.submat(0, 0, nR, nC);
    else
      throw string(
          "Error in resize() - the patch for arma::resize. Either both "
          "dimension should be larger or both should be smaller!");
  }
  return MMat;
}

// For arma::mat
arma::mat Utils::resize_patch(const arma::mat &Mat, const unsigned int nR,
                              const unsigned int nC) {
  arma::mat MMat(nR, nC);
  MMat.zeros();
  if (nR >= Mat.n_rows && nC >= Mat.n_cols) {
    if (Mat.n_rows >= 1 && Mat.n_cols >= 1)
      MMat.submat(0, 0, Mat.n_rows - 1, Mat.n_cols - 1) = Mat;
  } else {
    if (nR <= Mat.n_rows && nC <= Mat.n_cols)
      MMat = Mat.submat(0, 0, nR, nC);
    else
      throw string(
          "Error in resize() - the patch for arma::resize. Either both "
          "dimension should be larger or both should be smaller!");
  }
  return MMat;
}

// For arma::vec
arma::vec Utils::resize_patch(const arma::vec &Mat, const unsigned int nR) {
  arma::vec MMat(nR);
  MMat.zeros();
  if (nR > Mat.n_rows)
    MMat.subvec(0, Mat.n_rows - 1) = Mat;
  else
    MMat = Mat.subvec(0, nR);
  return MMat;
}

void Utils::appendSave(const sp_mat &matrix, 
                       const string out,     
                       const string header,  
                       bool erase 
                       )
{
  // Using C++ file operations to copy the data into the target given by @out
  unsigned int nR{0}, nC{0}, nnz{0};

  ofstream outfile(out, erase ? ios::out : ios::app);

  nR = matrix.n_rows;
  nC = matrix.n_cols;
  nnz = matrix.n_nonzero;

  outfile << header << "\n";
  outfile << nR << "\t" << nC << "\t" << nnz << "\n";
  for (auto it = matrix.begin(); it != matrix.end(); ++it)
    outfile << it.row() << "\t" << it.col() << "\t" << (*it)
            << "\n"; // Write the required information of sp_mat
  outfile << "\n";
  outfile.close(); // and close it
}

long int Utils::appendRead(
    sp_mat &matrix,  
    const string in, 
    long int pos,    
    const string header 
    )
{
  unsigned int nR, nC, nnz;

  ifstream infile(in, ios::in);
  infile.seekg(pos);

  string header_checkwith;
  infile >> header_checkwith;

  if (header != "" && header != header_checkwith)
    throw string(
        "Error in Utils::appendRead<sp_mat>. Wrong header. Expected: " +
        header + " Found: " + header_checkwith);

  infile >> nR >> nC >> nnz;
  if (nR == 0 || nC == 0)
    matrix.set_size(nR, nC);
  else {
    arma::umat locations(2, nnz);
    arma::vec values(nnz);

    unsigned int r, c;
    double val;

    for (unsigned int i = 0; i < nnz; ++i) {
      infile >> r >> c >> val;
      locations(0, i) = r;
      locations(1, i) = c;
      values(i) = val;
    }
    matrix = arma::sp_mat(locations, values, nR, nC);
  }

  pos = infile.tellg();
  infile.close();

  return pos;
}

void appendSave(const vector<double> v, const string out, const string header,
                bool erase) {
  ofstream outfile(out, erase ? ios::out : ios::app);
  outfile << header << "\n" << v.size() << "\n";
  for (const double x : v)
    outfile << x << "\n";
  outfile.close();
}

long int appendRead(vector<double> &v, const string in, long int pos,
                    const string header) {
  unsigned long int size;
  ifstream infile(in, ios::in);
  infile.seekg(pos);
  string header_checkwith;
  infile >> header_checkwith;

  if (header != "" && header != header_checkwith)
    throw string(
        "Error in Utils::appendRead<sp_mat>. Wrong header. Expected: " +
        header + " Found: " + header_checkwith);

  infile >> size;

  v.resize(size);
  for (unsigned int i = 0; i < size; ++i)
    infile >> v[i];
  pos = infile.tellg();
  infile.close();
  return pos;
}

void Utils::appendSave(const vec &matrix,   
                       const string out,    
                       const string header, 
                       bool erase 
) {
  // Using C++ file operations to copy the data into the target given by @out
  unsigned int nR{0};

  ofstream outfile(out, erase ? ios::out : ios::app);

  nR = matrix.n_rows;

  outfile << header << "\n";

  outfile << nR << "\n";
  for (auto it = matrix.begin(); it != matrix.end(); ++it)
    outfile << (*it) << "\n"; // Write the required information of sp_mat
  outfile << "\n";
  outfile.close(); // and close it
}

long int Utils::appendRead(
    vec &matrix,     
    const string in, 
    long int pos,    
    const string header 
) {
  unsigned int nR;
  string buffers;
  string checkwith;
  ifstream in_file(in, ios::in);
  in_file.seekg(pos);

  in_file >> checkwith;
  if (header != "" && checkwith != header)
    throw string("Error in Utils::appendRead<vec>. Wrong header. Expected: " +
                 header + " Found: " + checkwith);
  in_file >> nR;
  matrix.zeros(nR);
  for (unsigned int i = 0; i < nR; ++i) {
    double val;
    in_file >> val;
    matrix.at(i) = val;
  }

  pos = in_file.tellg();
  in_file.close();

  return pos;
}

void Utils::appendSave(const long int v, const string out, const string header,
                       bool erase)
{
  ofstream outfile(out, erase ? ios::out : ios::app);
  outfile << header << "\n";
  outfile << v << "\n";
  outfile.close();
}

long int Utils::appendRead(long int &v, const string in, long int pos,
                           const string header) {
  ifstream infile(in, ios::in);
  infile.seekg(pos);

  string header_checkwith;
  infile >> header_checkwith;

  if (header != "" && header != header_checkwith)
    throw string(
        "Error in Utils::appendRead<long int>. Wrong header. Expected: " +
        header + " Found: " + header_checkwith);

  long int val;
  infile >> val;
  v = val;

  pos = infile.tellg();
  infile.close();

  return pos;
}

void Utils::appendSave(const unsigned int v, const string out,
                       const string header, bool erase)
{
  ofstream outfile(out, erase ? ios::out : ios::app);
  outfile << header << "\n";
  outfile << v << "\n";
  outfile.close();
}

long int Utils::appendRead(unsigned int &v, const string in, long int pos,
                           const string header) {
  ifstream infile(in, ios::in);
  infile.seekg(pos);

  string header_checkwith;
  infile >> header_checkwith;

  if (header != "" && header != header_checkwith)
    throw string(
        "Error in Utils::appendRead<unsigned int>. Wrong header. Expected: " +
        header + " Found: " + header_checkwith);

  unsigned int val;
  infile >> val;
  v = val;

  pos = infile.tellg();
  infile.close();

  return pos;
}

void Utils::appendSave(const string v, const string out, bool erase)
{
  ofstream outfile(out, erase ? ios::out : ios::app);
  outfile << v << "\n";
  outfile.close();
}

long int Utils::appendRead(string &v, const string in, long int pos) {
  ifstream infile(in, ios::in);
  infile.seekg(pos);

  string val;
  infile >> val;
  v = val;

  pos = infile.tellg();
  infile.close();

  return pos;
}
unsigned long int Utils::vec_to_num(std::vector<short int> binary) {
  unsigned long int number = 0;
  unsigned int posn = 1;
  while (!binary.empty()) {
    short int bit = (binary.back() + 1) / 2; // The least significant bit
    number += (bit * posn);
    posn *= 2;         // Update place value
    binary.pop_back(); // Remove that bit
  }
  return number;
}

std::vector<short int> Utils::num_to_vec(unsigned long int number,
                                         const unsigned int &nCompl) {
  std::vector<short int> binary{};
  for (unsigned int vv = 0; vv < nCompl; vv++) {
    binary.push_back(number % 2);
    number /= 2;
  }
  std::for_each(binary.begin(), binary.end(),
                [](short int &vv) { vv = (vv == 0 ? -1 : 1); });
  std::reverse(binary.begin(), binary.end());
  return binary;
}