Contents

module SPCore
# Perform DFT transforms, forward and inverse.
# @author James Tunnell
class DFT
  # @param [Array] input  array of real values, representing the time domain
  #                       signal to be passed into the forward DFT.
  def self.forward input
    input_size = input.size
    raise ArgumentError, "input.size is not even" unless (input_size % 2 == 0)
    
    output_size = input_size
    output = Array.new(output_size)
    
    output.each_index do |k|
      sum = Complex(0.0)
      input.each_index do |n|
        a = TWO_PI * n * k / input_size
        sum += Complex(input[n] * Math::cos(a), -input[n] * Math::sin(a))
      end
      output[k] = sum
    end
    
    return output
  end
  
  # @param [Array] input  array of complex values, representing the frequency domain
  #                       signal obtained from the forward DFT.
  def self.inverse input
    input_size = input.size
    raise ArgumentError, "input.size is not even" unless (input_size % 2 == 0)
    
    output = Array.new(input_size)
    output_size = output.size
    
    output.each_index do |k|
      sum = Complex(0.0)
      input.each_index do |n|
        a = TWO_PI * n * k / input_size
        sum += Complex(input[n] * Math::cos(a), input[n] * Math::sin(a))
      end
      output[k] = sum / output_size
    end
    
    return output    
  end
end
end

Version data entries

9 entries across 9 versions & 1 rubygems

Version	Path
spcore-0.2.1	lib/spcore/transforms/dft.rb
spcore-0.2.0	lib/spcore/transforms/dft.rb
spcore-0.1.9	lib/spcore/transforms/dft.rb
spcore-0.1.8	lib/spcore/transforms/dft.rb
spcore-0.1.7	lib/spcore/transforms/dft.rb
spcore-0.1.6	lib/spcore/transforms/dft.rb
spcore-0.1.5	lib/spcore/transforms/dft.rb
spcore-0.1.4	lib/spcore/transforms/dft.rb
spcore-0.1.3	lib/spcore/transforms/dft.rb