1 #define TYPEDEPARGS 0, 1
2 #define SINGLEARGS
3 #define COMPLEXARGS
4 #define OCTFILENAME comp_nonsepdgt_shear // change to filename
5 #define OCTFILEHELP "This function calls the C-library\n\
6 c=comp_nonsepdgt_shear(f,g,a,M,s0,s1,br);\n"
7
8
9 #include "ltfat_oct_template_helper.h"
10 // octave_idx_type is 32 or 64 bit signed integer
11
12 static inline void
fwd_dgt_shear(const Complex * f,const Complex * g,const octave_idx_type L,const octave_idx_type W,const octave_idx_type a,const octave_idx_type M,const octave_idx_type s0,const octave_idx_type s1,const octave_idx_type br,Complex * cout)13 fwd_dgt_shear(const Complex *f, const Complex *g,
14 const octave_idx_type L, const octave_idx_type W,
15 const octave_idx_type a, const octave_idx_type M,
16 const octave_idx_type s0, const octave_idx_type s1,
17 const octave_idx_type br, Complex *cout)
18 {
19 ltfat_dgt_shear_d(reinterpret_cast<const ltfat_complex_d *>(f),
20 reinterpret_cast<const ltfat_complex_d *>(g),
21 L, W, a, M, s0, s1, br,
22 reinterpret_cast<ltfat_complex_d *>(cout));
23 }
24
25 static inline void
fwd_dgt_shear(const FloatComplex * f,const FloatComplex * g,const octave_idx_type L,const octave_idx_type W,const octave_idx_type a,const octave_idx_type M,const octave_idx_type s0,const octave_idx_type s1,const octave_idx_type br,FloatComplex * cout)26 fwd_dgt_shear(const FloatComplex *f, const FloatComplex *g,
27 const octave_idx_type L, const octave_idx_type W,
28 const octave_idx_type a, const octave_idx_type M,
29 const octave_idx_type s0, const octave_idx_type s1,
30 const octave_idx_type br, FloatComplex *cout)
31 {
32 ltfat_dgt_shear_s(reinterpret_cast<const ltfat_complex_s *>(f),
33 reinterpret_cast<const ltfat_complex_s *>(g),
34 L, W, a, M, s0, s1, br,
35 reinterpret_cast<ltfat_complex_s *>(cout));
36 }
37
38 template <class LTFAT_TYPE, class LTFAT_REAL, class LTFAT_COMPLEX>
octFunction(const octave_value_list & args,int nargout)39 octave_value_list octFunction(const octave_value_list& args, int nargout)
40 {
41 DEBUGINFO;
42 MArray<LTFAT_TYPE> f = ltfatOctArray<LTFAT_TYPE>(args(0));
43 MArray<LTFAT_TYPE> g = ltfatOctArray<LTFAT_TYPE>(args(1));
44 const octave_idx_type a = args(2).int_value();
45 const octave_idx_type M = args(3).int_value();
46 const octave_idx_type s0 = args(4).int_value();
47 const octave_idx_type s1 = args(5).int_value();
48 const octave_idx_type br = args(6).int_value();
49
50 const octave_idx_type L = f.rows();
51 const octave_idx_type W = f.cols();
52 const octave_idx_type N = L / a;
53
54 dim_vector dims_out(M, N, W);
55 dims_out.chop_trailing_singletons();
56
57 MArray<LTFAT_COMPLEX> cout(dims_out);
58
59 fwd_dgt_shear(f.data(), g.data(), L, W, a, M, s0, s1, br,
60 cout.fortran_vec());
61
62 return octave_value(cout);
63 }
64