fortran fft代码1

下面是一个简单的Fortran FFT代码，用于计算一维FFT：

program fft1d
  implicit none
  integer :: i, j, n, m, k
  complex :: w, arg, tmp
  real :: pi = acos(-1.0)
  complex, dimension(:), allocatable :: x, y, wsave

  ! 读入数据的长度
  write(*,*) "请输入数据长度："
  read(*,*) n

  ! 分配存储空间
  allocate(x(n), y(n), wsave(n/2))

  ! 读入数据
  do i = 1, n
    write(*,*) "请输入第", i, "个数据："
    read(*,*) x(i)
  end do

  ! 计算FFT
  m = int(log(n) / log(2))
  do j = 1, m
    do i = 0, n-1, 2**j
      do k = 0, 2**(j-1)-1
        arg = -2.0_complex * pi * k / 2.0**j
        w = cmplx(cos(arg), sin(arg))
        tmp = w * x(i+k+2**(j-1))
        y(i+k+1) = x(i+k+1) + tmp
        y(i+k+1+2**(j-1)) = x(i+k+1) - tmp
      end do
    end do
    x(:) = y(:)
  end do

  ! 输出结果
  write(*,*) "FFT结果为："
  do i = 1, n
    write(*,*) y(i)
  end do

  ! 释放存储空间
  deallocate(x, y, wsave)

end program fft1d

注：这是一个基于蝴蝶算法的FFT实现，可能不是最优的实现方式。