Merge pull request thuva4#383 from syam3526/patch-4

thuva4 · web-flow · commit eb9081e61a06 · 2017-10-26T20:00:19.000+05:30
fast fourier transform using java
diff --git a/FastFourierTransform/Java/fft_java.java b/FastFourierTransform/Java/fft_java.java
@@ -0,0 +1,95 @@
+import static java.lang.Math.*;
+ 
+public class FastFourierTransform {
+ 
+    public static int bitReverse(int n, int bits) {
+        int reversedN = n;
+        int count = bits - 1;
+ 
+        n >>= 1;
+        while (n > 0) {
+            reversedN = (reversedN << 1) | (n & 1);
+            count--;
+            n >>= 1;
+        }
+ 
+        return ((reversedN << count) & ((1 << bits) - 1));
+    }
+ 
+    static void fft(Complex[] buffer) {
+ 
+        int bits = (int) (log(buffer.length) / log(2));
+        for (int j = 1; j < buffer.length / 2; j++) {
+ 
+            int swapPos = bitReverse(j, bits);
+            Complex temp = buffer[j];
+            buffer[j] = buffer[swapPos];
+            buffer[swapPos] = temp;
+        }
+ 
+        for (int N = 2; N <= buffer.length; N <<= 1) {
+            for (int i = 0; i < buffer.length; i += N) {
+                for (int k = 0; k < N / 2; k++) {
+ 
+                    int evenIndex = i + k;
+                    int oddIndex = i + k + (N / 2);
+                    Complex even = buffer[evenIndex];
+                    Complex odd = buffer[oddIndex];
+ 
+                    double term = (-2 * PI * k) / (double) N;
+                    Complex exp = (new Complex(cos(term), sin(term)).mult(odd));
+ 
+                    buffer[evenIndex] = even.add(exp);
+                    buffer[oddIndex] = even.sub(exp);
+                }
+            }
+        }
+    }
+ 
+    public static void main(String[] args) {
+        double[] input = {1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0};
+ 
+        Complex[] cinput = new Complex[input.length];
+        for (int i = 0; i < input.length; i++)
+            cinput[i] = new Complex(input[i], 0.0);
+ 
+        fft(cinput);
+ 
+        System.out.println("Results:");
+        for (Complex c : cinput) {
+            System.out.println(c);
+        }
+    }
+}
+ 
+class Complex {
+    public final double re;
+    public final double im;
+ 
+    public Complex() {
+        this(0, 0);
+    }
+ 
+    public Complex(double r, double i) {
+        re = r;
+        im = i;
+    }
+ 
+    public Complex add(Complex b) {
+        return new Complex(this.re + b.re, this.im + b.im);
+    }
+ 
+    public Complex sub(Complex b) {
+        return new Complex(this.re - b.re, this.im - b.im);
+    }
+ 
+    public Complex mult(Complex b) {
+        return new Complex(this.re * b.re - this.im * b.im,
+                this.re * b.im + this.im * b.re);
+    }
+ 
+    @Override
+    public String toString() {
+        return String.format("(%f,%f)", re, im);
+    }
+}
