LemVisualizer/visualizer.rb at master · clemcke/LemVisualizer · GitHub

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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
# Visualizer
# run with command: rp5 run visulizer.rb

require 'ruby-processing'
class Visualizer < Processing::App
  load_library "minim"
  import "ddf.minim"
  import "ddf.minim.analysis"

  def setup
    @vars = {}
    @effects = []
    setup_music
    smooth
    size(@width,@height)
    background 10
  end

  def draw
    update_sound
    @effects.each(&:draw)
  end

  def setup_music
    @minim = Minim.new(self)
    @input = @minim.get_line_in
    @fft = FFT.new(@input.left.size, 44100)
    @vars[:beat] = BeatDetect.new

    @freqs = [60, 170, 310, 600, 1000, 3000, 6000, 12000, 14000, 16000]

    @current_ffts = Array.new(@freqs.size, 0.001)
    @previous_ffts= Array.new(@freqs.size, 0.001)
    @max_ffts = Array.new(@freqs.size, 0.001)
    @vars[:scaled_ffts] = Array.new(@freqs.size, 0.001)

    @fft_smoothing = 0.8
  end

  def update_sound
    @fft.forward(@input.left)
    @previous_ffts = @current_ffts
    @freqs.each_with_index do |freq, i|
      new_fft = @fft.get_freq(freq)
      @max_ffts[i] = new_fft if new_fft > @max_ffts[i]
      @current_ffts[i] = ((1 - @fft_smoothing) * new_fft) + (@fft_smoothing * @previous_ffts[i])
      @vars[:scaled_ffts][i] = (@current_ffts[i]/@max_ffts[i])
    end
    @vars[:beat].detect(@input.left)
  end

  def effects
    @effects
  end

  def vars
    @vars
  end
end