Advertisement
Guest User

Untitled

a guest
Dec 10th, 2016
62
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Java 2.05 KB | None | 0 0
  1. public class FFT {
  2.  
  3.     public static int bitReverse(int n, int bits) {
  4.         int reversedN = n;
  5.         int count = bits - 1;
  6.  
  7.         n >>= 1;
  8.         while (n > 0) {
  9.             reversedN = (reversedN << 1) | (n & 1);
  10.             count--;
  11.             n >>= 1;
  12.         }
  13.  
  14.         return ((reversedN << count) & ((1 << bits) - 1));
  15.     }
  16.  
  17.     static void fft(Complex[] buffer) {
  18.  
  19.         int bits = (int) (Math.log(buffer.length) / Math.log(2));
  20.         for (int j = 1; j < buffer.length / 2; j++) {
  21.  
  22.             int swapPos = bitReverse(j, bits);
  23.             Complex temp = buffer[j];
  24.             buffer[j] = buffer[swapPos];
  25.             buffer[swapPos] = temp;
  26.         }
  27.  
  28.         for (int N = 2; N <= buffer.length; N <<= 1) {
  29.             for (int i = 0; i < buffer.length; i += N) {
  30.                 for (int k = 0; k < N / 2; k++) {
  31.  
  32.                     int evenIndex = i + k;
  33.                     int oddIndex = i + k + (N / 2);
  34.                     Complex even = buffer[evenIndex];
  35.                     Complex odd = buffer[oddIndex];
  36.  
  37.                     double term = (-2 * Math.PI * k) / (double) N;
  38.                     Complex exp = (new Complex(Math.cos(term), Math.sin(term)).mult(odd));
  39.  
  40.                     buffer[evenIndex] = even.add(exp);
  41.                     buffer[oddIndex] = even.sub(exp);
  42.                 }
  43.             }
  44.         }
  45.     }
  46. }
  47.  
  48. class Complex {
  49.     public final double re;
  50.     public final double im;
  51.  
  52.     public Complex() {
  53.         this(0, 0);
  54.     }
  55.  
  56.     public Complex(double r, double i) {
  57.         re = r;
  58.         im = i;
  59.     }
  60.  
  61.     public Complex add(Complex b) {
  62.         return new Complex(this.re + b.re, this.im + b.im);
  63.     }
  64.  
  65.     public Complex sub(Complex b) {
  66.         return new Complex(this.re - b.re, this.im - b.im);
  67.     }
  68.  
  69.     public Complex mult(Complex b) {
  70.         return new Complex(this.re * b.re - this.im * b.im,
  71.                 this.re * b.im + this.im * b.re);
  72.     }
  73.  
  74.     @Override
  75.     public String toString() {
  76.         return String.format("(%f,%f)", re, im);
  77.     }
  78. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement