Util.h

Go to the documentation of this file.

// \author Darko Veberic
// \date 2014

#ifndef _Util_h_
#define _Util_h_

#include <iostream>
#include <vector>
#include <string>
#include <sstream>


template<typename T>
struct Separator {
  Separator(const std::vector<T>& v, const char* const sep)
    : fVec(v), fSep(sep) { }
  const std::vector<T>& fVec;
  const char* const fSep;
};


template<typename T>
inline
Separator<T>
Separate(const std::vector<T>& v, const char* const sep)
{
  return Separator<T>(v, sep);
}


template<typename T>
struct Separator2D {
  Separator2D(const std::vector<T>& v, const char* const sep1, const char* const sep2)
    : fVec(v), fSep1(sep1), fSep2(sep2) { }
  const std::vector<T>& fVec;
  const char* const fSep1;
  const char* const fSep2;
};


template<typename T>
inline
Separator2D<T>
Separate(const std::vector<T>& v, const char* const sep1, const char* const sep2)
{
  return Separator2D<T>(v, sep1, sep2);
}


namespace std {

template<typename T>
inline
ostream&
operator<<(ostream& os, const Separator<T>& s)
{
  copy(s.fVec.begin(), s.fVec.end(), ostream_iterator<T>(os, s.fSep));
  return os;
}


template<typename T>
inline
ostream&
operator<<(ostream& os, const Separator2D<T>& s)
{
  for (const auto it : s.fVec)
    os << Separate(it, s.fSep1) << s.fSep2;
  return os;
}

}


inline
std::vector<double>
ParseListRange(const std::string& str)
{
  const bool isRange = (str.find(':') != std::string::npos);
  const char sep = (isRange ? ':' : ',');
  std::istringstream iss(str);
  std::string n;
  std::vector<double> ret;
  while (std::getline(iss, n, sep))
    ret.push_back(stod(n));
  if (isRange) {
    const double a = ret.at(0);
    const double b = ret.at(1);
    const unsigned int n = ret.at(2);
    ret.clear();
    for (unsigned int i = 0; i <= n; ++i)
      ret.push_back(a + (b-a)*i/n);
  }
  return ret;
}


inline
void
Update(std::vector<double>& init, const std::vector<double>& res)
{
  // overwrite whole init with beginning of res
  for (unsigned int i = 0, n = init.size(); i < n; ++i)
    init[i] = res.at(i);
}


#endif