python/lib/aubio/midiconv.py

   1 # -*- coding: utf-8 -*-
   2
   3 import sys
   4 PY3 = sys.version_info[0] == 3
   5 if PY3:
   6     string_types = [str]
   7 else:
   8     string_types = [str, unicode]
   9
  10 def note2midi(note):
  11     " convert note name to midi note number, e.g. [C-1, G9] -> [0, 127] "
  12     _valid_notenames = {'C': 0, 'D': 2, 'E': 4, 'F': 5, 'G': 7, 'A': 9, 'B': 11}
  13     _valid_modifiers = {None: 0, u'♮': 0, '#': +1, u'♯': +1, u'\udd2a': +2, 'b': -1, u'♭': -1, u'\ufffd': -2}
  14     _valid_octaves = range(-1, 10)
  15     if type(note) not in string_types:
  16         raise TypeError("a string is required, got %s (%s)" % (note, str(type(note)) ))
  17     if not (1 < len(note) < 5):
  18         raise ValueError(
  19                 "string of 2 to 4 characters expected, got %d (%s)" %
  20                 (len(note), note))
  21     notename, modifier, octave = [None]*3
  22
  23     if len(note) == 4:
  24         notename, modifier, octave_sign, octave = note
  25         octave = octave_sign + octave
  26     elif len(note) == 3:
  27         notename, modifier, octave = note
  28         if modifier == '-':
  29             octave = modifier + octave
  30             modifier = None
  31     else:
  32         notename, octave = note
  33
  34     notename = notename.upper()
  35     octave = int(octave)
  36
  37     if notename not in _valid_notenames:
  38         raise ValueError("%s is not a valid note name" % notename)
  39     if modifier not in _valid_modifiers:
  40         raise ValueError("%s is not a valid modifier" % modifier)
  41     if octave not in _valid_octaves:
  42         raise ValueError("%s is not a valid octave" % octave)
  43
  44     midi = 12 + octave * 12 + _valid_notenames[notename] + _valid_modifiers[modifier]
  45     if midi > 127:
  46         raise ValueError("%s is outside of the range C-2 to G8" % note)
  47     return midi
  48
  49 def midi2note(midi):
  50     " convert midi note number to note name, e.g. [0, 127] -> [C-1, G9] "
  51     if type(midi) != int:
  52         raise TypeError("an integer is required, got %s" % midi)
  53     if not (-1 < midi < 128):
  54         raise ValueError(
  55                 "an integer between 0 and 127 is excepted, got %d" % midi)
  56     midi = int(midi)
  57     _valid_notenames = ['C', 'C#', 'D', 'D#', 'E', 'F', 'F#', 'G', 'G#', 'A', 'A#', 'B']
  58     return _valid_notenames[midi % 12] + str( int(midi / 12) - 1 )
  59
  60 def freq2note(freq):
  61     from aubio import freqtomidi
  62     return midi2note(int(freqtomidi(freq)))