class HeadMusic::Scale
  def self.get(root_pitch, scale_type_name = nil)
    root_pitch = HeadMusic::Pitch.get(root_pitch)
    scale_type_name ||= :major
    scale_type ||= HeadMusic::ScaleType.get(scale_type_name)
    @scales ||= {}
    @scales[root_pitch.to_s] ||= {}
    @scales[root_pitch.to_s][scale_type.name] ||= new(root_pitch, scale_type)
  end

  attr_reader :root_pitch, :scale_type

  def initialize(root_pitch, scale_type)
    @root_pitch = HeadMusic::Pitch.get(root_pitch)
    @scale_type = HeadMusic::ScaleType.get(scale_type)
  end

  def pitches(direction: :ascending, octaves: 1)
    @pitches ||= {}
    @pitches[direction] ||= {}
    @pitches[direction][octaves] ||= begin
      letter_cycle = root_pitch.letter_cycle
      semitones_from_root = 0
      [root_pitch].tap do |pitches|
        if [:ascending, :both].include?(direction)
          (1..octaves).each do |i|
            scale_type.ascending_intervals.each_with_index do |semitones, i|
              semitones_from_root += semitones
              pitches << pitch_for_step(i+1, semitones_from_root, :ascending)
            end
          end
        end
        if [:descending, :both].include?(direction)
          (1..octaves).each do |i|
            scale_type.descending_intervals.each_with_index do |semitones, i|
              semitones_from_root -= semitones
              pitches << pitch_for_step(i+1, semitones_from_root, :descending)
            end
          end
        end
      end
    end
  end

  def spellings(direction: :ascending, octaves: 1)
    pitches(direction: direction, octaves: octaves).map(&:spelling).map(&:to_s)
  end

  def pitch_names(direction: :ascending, octaves: 1)
    pitches(direction: direction, octaves: octaves).map(&:name)
  end

  def letter_cycle
    @letter_cycle ||= root_pitch.letter_cycle
  end

  def root_pitch_number
    @root_pitch_number ||= root_pitch.number
  end

  def degree(degree_number)
    pitches[degree_number - 1]
  end

  private

  def parent_scale_pitches
    HeadMusic::Scale.get(root_pitch, scale_type.parent_name).pitches if scale_type.parent
  end

  def parent_scale_pitch_for(semitones_from_root)
    parent_scale_pitches.detect { |parent_scale_pitch|
      parent_scale_pitch.pitch_class == (root_pitch + semitones_from_root).to_i % 12
    }
  end

  def letter_for_step(step, semitones_from_root, direction)
    pitch_class_number = (root_pitch.pitch_class.to_i + semitones_from_root) % 12
    if scale_type.intervals.length == 7
      direction == :ascending ? letter_cycle[step % 7] : letter_cycle[-step % 7]
    elsif scale_type.intervals.length < 7 && parent_scale_pitches
      parent_scale_pitch_for(semitones_from_root).letter
    elsif root_pitch.flat?
      HeadMusic::PitchClass::FLAT_SPELLINGS[pitch_class_number]
    else
      HeadMusic::PitchClass::SHARP_SPELLINGS[pitch_class_number]
    end
  end

  def pitch_for_step(step, semitones_from_root, direction)
    pitch_number = root_pitch_number + semitones_from_root
    letter = letter_for_step(step, semitones_from_root, direction)
    HeadMusic::Pitch.from_number_and_letter(pitch_number, letter)
  end
end