/
test_axisym_vs_directional.c
128 lines (106 loc) · 4.03 KB
/
test_axisym_vs_directional.c
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#include <setjmp.h>
#include <stdarg.h>
#include <stddef.h>
#include "s2let/s2let.h"
#include <cmocka.h>
const int seed = 1;
_Bool min(int a, int b) { return a > b ? b : a; }
void test_transform_axisym_vs_directional_mw(void **state) {
s2let_parameters_t parameters = *(s2let_parameters_t *)*state;
parameters.upsample = 1;
int J = s2let_j_max(¶meters);
double wav_error, scal_error;
complex double *f, *flm;
s2let_allocate_lm(&flm, parameters.L);
s2let_allocate_mw(&f, parameters.L);
s2let_lm_random_flm(flm, parameters.L, parameters.spin, seed);
ssht_core_mw_inverse_sov_sym(
f,
flm,
parameters.L,
parameters.spin,
parameters.dl_method,
parameters.verbosity);
complex double *f_wav_axisym, *f_scal_axisym, *f_wav_dir, *f_scal_dir;
s2let_transform_axisym_allocate_mw_f_wav(&f_wav_axisym, &f_scal_axisym, ¶meters);
s2let_allocate_f_wav(&f_wav_dir, &f_scal_dir, ¶meters);
s2let_transform_axisym_wav_analysis_mw(f_wav_axisym, f_scal_axisym, f, ¶meters);
s2let_analysis_px2wav(f_wav_dir, f_scal_dir, f, ¶meters);
// Compute the maximum absolute error in the computed wavelet transform
for (int i = 0;
i < (J - parameters.J_min + 1) * parameters.L * (2 * parameters.L - 1);
i += 1) {
assert_float_equal(creal(f_wav_axisym[i]), creal(f_wav_dir[i]), 1e-12);
assert_float_equal(cimag(f_wav_axisym[i]), cimag(f_wav_dir[i]), 1e-12);
}
for (int i = 0; i < parameters.L * (2 * parameters.L - 1); i += 1) {
assert_float_equal(creal(f_scal_axisym[i]), creal(f_scal_dir[i]), 1e-12);
assert_float_equal(cimag(f_scal_axisym[i]), cimag(f_scal_dir[i]), 1e-12);
}
}
void test_transform_axisym_vs_directional_mw_multires(void **state) {
const s2let_parameters_t parameters = *(s2let_parameters_t *)*state;
const int J = s2let_j_max(¶meters);
double wav_error, scal_error;
complex double *f, *flm;
s2let_allocate_lm(&flm, parameters.L);
s2let_allocate_mw(&f, parameters.L);
s2let_lm_random_flm(flm, parameters.L, parameters.spin, seed);
ssht_core_mw_inverse_sov_sym(
f,
flm,
parameters.L,
parameters.spin,
parameters.dl_method,
parameters.verbosity);
complex double *f_wav_axisym, *f_scal_axisym, *f_wav_dir, *f_scal_dir;
s2let_transform_axisym_allocate_mw_f_wav_multires(
&f_wav_axisym, &f_scal_axisym, ¶meters);
s2let_allocate_f_wav(&f_wav_dir, &f_scal_dir, ¶meters);
s2let_transform_axisym_wav_analysis_mw_multires(
f_wav_axisym, f_scal_axisym, f, ¶meters);
s2let_analysis_px2wav(f_wav_dir, f_scal_dir, f, ¶meters);
int samples = 0;
for (int j = parameters.J_min; j <= J; ++j) {
const int bandlim = min(s2let_bandlimit(j, ¶meters), parameters.L);
samples += bandlim * (2 * bandlim - 1);
}
// Compute the maximum absolute error in the computed wavelet transform
for (int i = 0; i < samples; i += 1) {
assert_float_equal(creal(f_wav_axisym[i]), creal(f_wav_dir[i]), 1e-12);
assert_float_equal(cimag(f_wav_axisym[i]), cimag(f_wav_dir[i]), 1e-12);
}
const int bandlimit =
min(s2let_bandlimit(parameters.J_min - 1, ¶meters), parameters.L);
for (int i = 0; i < bandlimit * (2 * bandlimit - 1); i += 1) {
assert_float_equal(creal(f_scal_axisym[i]), creal(f_scal_dir[i]), 1e-12);
assert_float_equal(cimag(f_scal_axisym[i]), cimag(f_scal_dir[i]), 1e-12);
}
free(flm);
free(f);
free(f_wav_axisym);
free(f_scal_axisym);
free(f_wav_dir);
free(f_scal_dir);
}
int main(void) {
const s2let_parameters_t state = {
.L = 16,
.N = 1,
.B = 2,
.J_min = 2,
.spin = 0,
.original_spin = 0,
.upsample = 0,
.dl_method = SSHT_DL_RISBO,
.reality = 0,
.sampling_scheme = S2LET_SAMPLING_MW,
};
const struct CMUnitTest tests[] = {
cmocka_unit_test_prestate(
test_transform_axisym_vs_directional_mw, (void **)&state),
cmocka_unit_test_prestate(
test_transform_axisym_vs_directional_mw_multires, (void **)&state),
};
return cmocka_run_group_tests(tests, NULL, NULL);
}