cdata.h

Go to the documentation of this file.

#ifndef SOPT_TOOLS_FOR_TESTS_CDATA
#define SOPT_TOOLS_FOR_TESTS_CDATA
 
#include "sopt/config.h"
#include <Eigen/Core>
#include "sopt/linear_transform.h"
 
namespace sopt {
// Wraps calls to sampling and wavelets to C style
template <typename T>
struct CData {
  using t_Vector = Eigen::Matrix<T, Eigen::Dynamic, 1>;
  typename t_Vector::Index nin, nout;
  sopt::LinearTransform<t_Vector> const &transform;
  t_uint direct_calls, adjoint_calls;
};
 
template <typename T>
void direct_transform(void *out, void *in, void **data) {
  CData<T> const &cdata = *(CData<T> *)data;
  using t_Vector = Eigen::Matrix<T, Eigen::Dynamic, 1>;
  t_Vector const eval = cdata.transform * t_Vector::Map((T *)in, cdata.nin);
  ++(((CData<T> *)data)->direct_calls);
  t_Vector::Map((T *)out, cdata.nout) = eval;
}
template <typename T>
void adjoint_transform(void *out, void *in, void **data) {
  CData<T> const &cdata = *(CData<T> *)data;
  using t_Vector = Eigen::Matrix<T, Eigen::Dynamic, 1>;
  t_Vector const eval = cdata.transform.adjoint() * t_Vector::Map((T *)in, cdata.nout);
  ++(((CData<T> *)data)->adjoint_calls);
  t_Vector::Map((T *)out, cdata.nin) = eval;
}
 
}  // namespace sopt
#endif