Improve Harmonize performance

scottmacphersonmusic · May 15, 2015 · 51c5792 · 51c5792
1 parent 0015c5c
commit 51c5792
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/music_theory/harmonize.rb b/lib/music_theory/harmonize.rb
@@ -6,11 +6,12 @@ class Harmonize
  attr_accessor :samples
 
  def initialize(*things_to_flatten)
- @samples = []
+ max_sample_array_length = things_to_flatten.map(&:length).max
+ @samples = Array.new(max_sample_array_length, 0)
+
  things_to_flatten.each do |group|
  group.each_with_index do |value, i|
- @samples[i] ||= 0
- @samples[i] += value
+ @samples[i] += value
  end
  end