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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
#define TYPEDEPARGS 0, 1, 2, 8
#define SINGLEARGS
#define REALARGS
#define OCTFILENAME comp_maskedheapintreal
#define OCTFILEHELP "Computes masked heapint.\n\
Usage: c = comp_maskedheapintreal(s, itime, ifreq, mask, a, M, tol, do_timeinv, usephase);\n\
Yeah."
#include "ltfat_oct_template_helper.h"
static inline void
fwd_maskedheapintreal(const double *s, const double *tgrad, const double *fgrad,
const int* mask, const octave_idx_type a, const octave_idx_type M,
const octave_idx_type L, const octave_idx_type W,
const double tol, int phasetype, double *phase)
{
if (phasetype == 2)
ltfat_maskedheapintreal_d( s, tgrad, fgrad, mask, a, M, L, W, tol, phase);
else
ltfat_maskedheapintreal_relgrad_d( s, tgrad, fgrad, mask, a, M, L, W, tol,
static_cast<ltfat_phaseconvention>(phasetype), phase);
}
static inline void
fwd_maskedheapintreal(const float *s, const float *tgrad, const float *fgrad,
const int* mask, const octave_idx_type a, const octave_idx_type M,
const octave_idx_type L, const octave_idx_type W,
const float tol, int phasetype, float *phase)
{
if (phasetype == 2)
ltfat_maskedheapintreal_s( s, tgrad, fgrad, mask, a, M, L, W, tol, phase);
else
ltfat_maskedheapintreal_relgrad_s( s, tgrad, fgrad, mask, a, M, L, W, tol,
static_cast<ltfat_phaseconvention>(phasetype), phase);
}
template <class LTFAT_TYPE, class LTFAT_REAL, class LTFAT_COMPLEX>
octave_value_list octFunction(const octave_value_list& args, int nargout)
{
MArray<LTFAT_REAL> s = ltfatOctArray<LTFAT_REAL>(args(0));
MArray<LTFAT_REAL> tgrad = ltfatOctArray<LTFAT_REAL>(args(1));
MArray<LTFAT_REAL> fgrad = ltfatOctArray<LTFAT_REAL>(args(2));
MArray<double> maskDouble = ltfatOctArray<double>(args(3));
const octave_idx_type a = args(4).int_value();
const octave_idx_type M = args(5).int_value();
const double tol = args(6).double_value();
const int phasetype = args(7).int_value()== 1? LTFAT_TIMEINV: LTFAT_FREQINV;
MArray<LTFAT_REAL> usephase = ltfatOctArray<LTFAT_REAL>(args(8));
const octave_idx_type M2 = args(0).rows();
const octave_idx_type N = args(0).columns();
const octave_idx_type L = N * a;
const octave_idx_type W = s.numel() / (M2 * N);
MArray<LTFAT_REAL> phase(dim_vector(M2, N, W));
memcpy(phase.fortran_vec(), usephase.data(), M2 * N * W * sizeof(LTFAT_REAL));
int* mask = new int[M2 * N * W];
for (octave_idx_type w = 0; w < M2 * N * W; w++ )
mask[w] = (int) maskDouble.data()[w];
fwd_maskedheapintreal(s.data(), tgrad.data(), fgrad.data(),
mask, a, M, L, W, static_cast<LTFAT_REAL>(tol),
phasetype, phase.fortran_vec());
delete [] mask;
return octave_value(phase);
}
|
