module RBMusic

  class Note
    attr_accessor :coord

    def initialize(coord)
      self.coord = coord
    end

    def self.from_latin(name)
      raise ArgumentError unless name.is_a?(String)

      note_parts = name.split(/(\d+)/)
      note_name = note_parts.first
      octave = note_parts.last.to_i

      unless NOTES.has_key?(note_name) && note_parts.size < 3
        raise ArgumentError
      end

      coordinate = [NOTES[note_name][0] + octave, NOTES[note_name][1]]

      coordinate[0] -= BASE_OFFSET[0]
      coordinate[1] -= BASE_OFFSET[1]

      Note.new(coordinate)
    end

    def frequency
      BASE_FREQ * (2.0 ** ((coord[0] * 1200 + coord[1] * 700.0) / 1200.0))
    end

    def accidental
      @accidental ||= ((coord[1] + BASE_OFFSET[1]) / 7.0).round
    end

    def octave
      # calculate octave of base note without accidentals
      @octave ||= coord[0] + BASE_OFFSET[0] + 4 * accidental + ((coord[1] + BASE_OFFSET[1] - 7 * accidental) / 2).floor
    end

    def latin
      return @latin if @latin
      accidentalName = ACCIDENTALS[accidental + 2]
      @latin ||= base_note_name + accidentalName
    end

    def ==(other)
      other.is_a?(Note) && other.latin == latin && other.octave == octave
    end

    def enharmonic?(other)
      raise ArgumentError unless other.is_a?(Note)

      other.frequency == frequency
    end
    alias_method :enharmonically_equivalent_to?, :enharmonic?

    def midi_note_number
      # see http://www.phys.unsw.edu.au/jw/notes.html
      @note_number ||= (12 * Math.log2(frequency / 440) + 69).round
    end

    def scale(name, octaves = 1)
      NoteSet.from_scale(Scale.new(latin, name), octave, octaves)
    end

    def add(that)
      # if input is an array return an array
      if that.is_a?(Array)
        notes = that.map { |thing| add(thing) }
        return NoteSet.new(notes)
      end

      # if input is string/symbol try to parse it as interval
      if that.is_a?(String) || that.is_a?(Symbol)
        that = Interval.from_name(that)
      end

      Note.new([coord[0] + that.coord[0], coord[1] + that.coord[1]])
    end

    def subtract(that)
      if that.is_a?(Array)
        notes = that.map { |thing| subtract(thing) }
        return NoteSet.new(notes)
      end

      # if input is string try to parse it as interval
      if that.is_a?(String) || that.is_a?(Symbol)
        that = Interval.from_name(that)
      end

      coordinate = [coord[0] - that.coord[0], coord[1] - that.coord[1]]

      # if input is another note return the difference as an Interval
      that.is_a?(Note) ? Interval.new(coordinate) : Note.new(coordinate)
    end

    private

    def base_note_name
      @base_note_name ||= NOTE_NAMES[coord[1] + BASE_OFFSET[1] - accidental * 7 + 3]
    end
  end

end