converters.h

Go to the documentation of this file.

#ifndef CONVERTERS_H
#define CONVERTERS_H

#include <itpp/base/itcompat.h>
#include <itpp/base/help_functions.h>
#include <itpp/base/math/misc.h>


namespace itpp
{



// ----------------------------------------------------------------------
// Converters for vectors
// ----------------------------------------------------------------------

template <class T>
bvec to_bvec(const Vec<T> &v)
{
  bvec temp(v.length());
  for (int i = 0; i < v.length(); ++i) {
    temp(i) = static_cast<bin>(v(i));
  }
  return temp;
}

template <class T>
svec to_svec(const Vec<T> &v)
{
  svec temp(v.length());
  for (int i = 0; i < v.length(); ++i) {
    temp(i) = static_cast<short>(v(i));
  }
  return temp;
}

template <class T>
ivec to_ivec(const Vec<T> &v)
{
  ivec temp(v.length());
  for (int i = 0; i < v.length(); ++i) {
    temp(i) = static_cast<int>(v(i));
  }
  return temp;
}

template <class T>
vec to_vec(const Vec<T> &v)
{
  vec temp(v.length());
  for (int i = 0; i < v.length(); ++i) {
    temp(i) = static_cast<double>(v(i));
  }
  return temp;
}

template <class T>
cvec to_cvec(const Vec<T> &v)
{
  cvec temp(v.length());
  for (int i = 0; i < v.length(); ++i) {
    temp(i) = std::complex<double>(static_cast<double>(v(i)), 0.0);
  }
  return temp;
}

template<> inline
cvec to_cvec(const cvec& v)
{
  return v;
}

template <class T>
cvec to_cvec(const Vec<T> &real, const Vec<T> &imag)
{
  it_assert(real.length() == imag.length(),
            "to_cvec(): real and imaginary parts must have the same length");
  cvec temp(real.length());
  for (int i = 0; i < real.length(); ++i) {
    temp(i) = std::complex<double>(static_cast<double>(real(i)),
                                   static_cast<double>(imag(i)));
  }
  return temp;
}

ivec to_ivec(int s);

vec to_vec(double s);

cvec to_cvec(double real, double imag);

// ----------------------------------------------------------------------
// Converters for matrices
// ----------------------------------------------------------------------

template <class T>
bmat to_bmat(const Mat<T> &m)
{
  bmat temp(m.rows(), m.cols());
  for (int i = 0; i < temp.rows(); ++i) {
    for (int j = 0; j < temp.cols(); ++j) {
      temp(i, j) = static_cast<bin>(m(i, j));
    }
  }
  return temp;
}

template <class T>
smat to_smat(const Mat<T> &m)
{
  smat temp(m.rows(), m.cols());
  for (int i = 0; i < temp.rows(); ++i) {
    for (int j = 0; j < temp.cols(); ++j) {
      temp(i, j) = static_cast<short>(m(i, j));
    }
  }
  return temp;
}

template <class T>
imat to_imat(const Mat<T> &m)
{
  imat temp(m.rows(), m.cols());
  for (int i = 0; i < temp.rows(); ++i) {
    for (int j = 0; j < temp.cols(); ++j) {
      temp(i, j) = static_cast<int>(m(i, j));
    }
  }
  return temp;
}

template <class T>
mat to_mat(const Mat<T> &m)
{
  mat temp(m.rows(), m.cols());
  for (int i = 0; i < temp.rows(); ++i) {
    for (int j = 0; j < temp.cols(); ++j) {
      temp(i, j) = static_cast<double>(m(i, j));
    }
  }
  return temp;
}

template <class T>
cmat to_cmat(const Mat<T> &m)
{
  cmat temp(m.rows(), m.cols());
  for (int i = 0; i < temp.rows(); ++i) {
    for (int j = 0; j < temp.cols(); ++j) {
      temp(i, j) = std::complex<double>(static_cast<double>(m(i, j)), 0.0);
    }
  }
  return temp;
}

template<> inline
cmat to_cmat(const cmat& m)
{
  return m;
}

template <class T>
cmat to_cmat(const Mat<T> &real, const Mat<T> &imag)
{
  it_assert_debug((real.rows() == imag.rows())
                  && (real.cols() == imag.cols()),
                  "to_cmat(): real and imag part sizes does not match");
  cmat temp(real.rows(), real.cols());
  for (int i = 0; i < temp.rows(); ++i) {
    for (int j = 0; j < temp.cols(); ++j) {
      temp(i, j) = std::complex<double>(static_cast<double>(real(i, j)),
                                        static_cast<double>(imag(i, j)));
    }
  }
  return temp;
}


bvec dec2bin(int length, int index);

void dec2bin(int index, bvec &v);

bvec dec2bin(int index, bool msb_first = true);

int bin2dec(const bvec &inbvec, bool msb_first = true);

bvec oct2bin(const ivec &octalindex, short keepzeros = 0);

ivec bin2oct(const bvec &inbits);

ivec bin2pol(const bvec &inbvec);

bvec pol2bin(const ivec &inpol);

inline double rad_to_deg(double x) { return (180.0 / itpp::pi * x); }
inline double deg_to_rad(double x) { return (itpp::pi / 180.0 * x); }

inline double round(double x) { return ::rint(x); }
inline vec round(const vec &x) { return apply_function<double>(::rint, x); }
inline mat round(const mat &x) { return apply_function<double>(::rint, x); }
inline int round_i(double x) { return static_cast<int>(::rint(x)); }
ivec round_i(const vec &x);
imat round_i(const mat &x);

inline vec ceil(const vec &x) { return apply_function<double>(std::ceil, x); }
inline mat ceil(const mat &x) { return apply_function<double>(std::ceil, x); }
inline int ceil_i(double x) { return static_cast<int>(std::ceil(x)); }
ivec ceil_i(const vec &x);
imat ceil_i(const mat &x);

inline vec floor(const vec &x) { return apply_function<double>(std::floor, x); }
inline mat floor(const mat &x) { return apply_function<double>(std::floor, x); }
inline int floor_i(double x) { return static_cast<int>(std::floor(x)); }
ivec floor_i(const vec &x);
imat floor_i(const mat &x);


inline double round_to_zero(double x, double threshold = 1e-14)
{
  return ((std::fabs(x) < threshold) ? 0.0 : x);
}

inline std::complex<double> round_to_zero(const std::complex<double>& x,
    double threshold = 1e-14)
{
  return std::complex<double>(round_to_zero(x.real(), threshold),
                              round_to_zero(x.imag(), threshold));
}

inline vec round_to_zero(const vec &x, double threshold = 1e-14)
{
  return apply_function<double>(round_to_zero, x, threshold);
}

inline mat round_to_zero(const mat &x, double threshold = 1e-14)
{
  return apply_function<double>(round_to_zero, x, threshold);
}

cvec round_to_zero(const cvec &x, double threshold = 1e-14);

cmat round_to_zero(const cmat &x, double threshold = 1e-14);


inline int gray_code(int x) { return x ^(x >> 1); }


template <typename T>
std::string to_str(const T &i);

std::string to_str(const double &i, const int precision);


template <typename T>
std::string to_str(const T &i)
{
  std::ostringstream ss;
  ss.precision(8);
  ss.setf(std::ostringstream::scientific, std::ostringstream::floatfield);
  ss << i;
  return ss.str();
}


// ---------------------------------------------------------------------
// Instantiations
// ---------------------------------------------------------------------

#ifdef HAVE_EXTERN_TEMPLATE

extern template bvec to_bvec(const svec &v);
extern template bvec to_bvec(const ivec &v);

extern template svec to_svec(const bvec &v);
extern template svec to_svec(const ivec &v);
extern template svec to_svec(const vec &v);

// Workaround for GCC 3.3.x error when using -finine-functions or -O3 flag
#if (GCC_VERSION >= 30400)
extern template ivec to_ivec(const bvec &v);
#endif
extern template ivec to_ivec(const svec &v);
extern template ivec to_ivec(const vec &v);

extern template vec to_vec(const bvec &v);
extern template vec to_vec(const svec &v);
extern template vec to_vec(const ivec &v);

extern template cvec to_cvec(const bvec &v);
extern template cvec to_cvec(const svec &v);
extern template cvec to_cvec(const ivec &v);
extern template cvec to_cvec(const vec &v);

extern template cvec to_cvec(const bvec &real, const bvec &imag);
extern template cvec to_cvec(const svec &real, const svec &imag);
extern template cvec to_cvec(const ivec &real, const ivec &imag);
extern template cvec to_cvec(const vec &real, const vec &imag);

extern template bmat to_bmat(const smat &m);
extern template bmat to_bmat(const imat &m);

extern template smat to_smat(const bmat &m);
extern template smat to_smat(const imat &m);
extern template smat to_smat(const mat &m);

extern template imat to_imat(const bmat &m);
extern template imat to_imat(const smat &m);
extern template imat to_imat(const mat &m);

extern template mat to_mat(const bmat &m);
#if (GCC_VERSION >= 30400)
extern template mat to_mat(const smat &m);
extern template mat to_mat(const imat &m);
#endif

extern template cmat to_cmat(const bmat &m);
extern template cmat to_cmat(const smat &m);
extern template cmat to_cmat(const imat &m);
extern template cmat to_cmat(const mat &m);

extern template cmat to_cmat(const bmat &real, const bmat &imag);
extern template cmat to_cmat(const smat &real, const smat &imag);
extern template cmat to_cmat(const imat &real, const imat &imag);
extern template cmat to_cmat(const mat &real, const mat &imag);

#endif // HAVE_EXTERN_TEMPLATE


} // namespace itpp

#endif // CONVERTERS_H

---