1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
#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
|
