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 package ffx.numerics.fft;
39
40 import static org.junit.Assert.assertEquals;
41 import static org.junit.Assert.assertTrue;
42
43 import java.util.Arrays;
44 import java.util.Collection;
45 import java.util.Random;
46
47 import ffx.utilities.FFXTest;
48 import org.junit.Test;
49 import org.junit.runner.RunWith;
50 import org.junit.runners.Parameterized;
51 import org.junit.runners.Parameterized.Parameters;
52
53
54 @RunWith(Parameterized.class)
55 public class ComplexTest extends FFXTest {
56
57 private final int n;
58 private final String info;
59 private final boolean preferred;
60 private final double[] data;
61 private final double[] orig;
62 private final double[] dft;
63
64 public ComplexTest(String info, int n, boolean preferred) {
65 this.info = info;
66 this.n = n;
67 this.preferred = preferred;
68 data = new double[n * 2];
69 orig = new double[n * 2];
70 dft = new double[n * 2];
71 Random r = new Random();
72 for (int i = 0; i < n; i++) {
73 double d = r.nextDouble();
74 data[i * 2] = d;
75 orig[i * 2] = d;
76 }
77 }
78
79 @Parameters
80 public static Collection<Object[]> data() {
81 return Arrays.asList(
82 new Object[][] {
83 {"Test n = 162 with factors [6, 3, 3, 3]", 162, true},
84 {"Test n = 160 with factors [5, 4, 4, 2]", 160, true},
85 {"Test n = 120 with factors [6, 5, 4]", 120, true},
86 {"Test n = 64 with factors [4, 4, 4]", 64, true},
87 {"Test n = 48 with factors [6, 4, 2]", 48, true},
88 {"Test n = 21 with factors [7, 3]", 21, true},
89 {"Test n = 38 with factors [2, 19]", 38, false},
90 {"Test n = 22 with factors [2, 11]", 22, false},
91 });
92 }
93
94
95 @Test
96 public void testFft() {
97 double tolerance = 1.0e-11;
98
99 int offset = 0;
100 int stride = 2;
101 Complex complex = new Complex(n);
102
103
104
105 long dftTime = System.nanoTime();
106 Complex.dft(data, dft);
107 dftTime = System.nanoTime() - dftTime;
108 String dftString = " DFT Time: " + dftTime * 1.0e-9 + " s\n";
109
110 long fftTime = System.nanoTime();
111 complex.fft(data, offset, stride);
112 fftTime = System.nanoTime() - fftTime;
113 String fftString = " FFT Time: " + fftTime * 1.0e-9 + " s";
114
115
116 for (int i = 0; i < 2 * n; i++) {
117 assertEquals(" Forward " + info + " at position: " + i, dft[i], data[i], tolerance);
118 }
119
120
121 String message = fftString + dftString;
122
123
124
125 complex.inverse(data, offset, stride);
126 for (int i = 0; i < n; i++) {
127 double orig = this.orig[i * 2];
128 double actual = data[i * 2];
129 assertEquals(" IFFT(FFT(X)) " + info + " at position: " + i, orig, actual, tolerance);
130 }
131 }
132
133
134 @Test
135 public void testPreferredDimension() {
136 boolean result = Complex.preferredDimension(n);
137 assertEquals(info, preferred, result);
138 }
139 }