pybeatcoder/pybeatcoder.py at main · Victor-Chavanne/pybeatcoder · 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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
import sys
from os.path import exists
import pygments
from pygments.lexers import PythonLexer
from mido import Message, MidiFile, MidiTrack

def generate_scale(file_name):
    # https://random-music-generators.herokuapp.com/melody
    scale = [79, 74, 76, 74, 72, 67, 71, 64, 65, 59, 64, 71]
    return scale

def generate_note(token):
    token_name = str(token[0]).split('.')[1]
    token_value = str(token[1])
    if (token_name == "Text" and token_value == " ") or token_name in ["Whitespace", "Escape", "Error", "Other"]:
        return 0
    elif token_name == "Text":
        return 1
    elif token_name == "Keyword":
        return 2
    elif token_name == "Name":
        return 3
    elif token_name == "Literal":
        return 4
    elif token_name == "String":
        return 5
    elif token_name == "Number":
        return 6
    elif token_name == "Punctuation":
        return 7
    elif token_name == "Operator":
        return 8
    elif token_name == "Comment":
        return 9
    elif token_name == "Generic":
        return 10
    else:
        return 11

def generate_bass_track(token_list, scale):
    track = MidiTrack()

    for line in token_list:
        for token in line:
            note = generate_note(token)
            track.append(Message('note_on', channel=0, note=scale[note], velocity=64, time=0))
            track.append(Message('note_off', note=scale[note], velocity=127, time=64))
    return track

def generate_token_notes(token_line):
    token_notes_list = []
    for token in token_line:
        pass

def main(input_file, output_file):
    # Get lines of code in a list
    code = []
    with open(input_file, "r") as inp:
        for line in inp.read().splitlines():
            code.append(line)

    # Iterate over lines to get the tokens
    pl = PythonLexer()
    tokens_list = [[x for x in pygments.lex(line, pl)] for line in code]

    # generate scale
    scale = generate_scale(input_file)

    # Create midi file
    mid = MidiFile()
    mid.tracks.append(generate_bass_track(tokens_list, scale))
    mid.save(output_file)

def print_help():
    print("""Usage: pybeatcoder [INPUT FILE] [OUTPUT FILE]
Converts a source code file into a midi file

    -h, --help      display this help and exit
    -p, --print     print the midi messages of a .mid file""")

def print_midi(midi_file):
    mid = MidiFile(midi_file)
    for i, track in enumerate(mid.tracks):
        print('Track {}: {}'.format(i, track.name))
        for msg in track:
            print(msg)

if __name__ == '__main__':
    if len(sys.argv) == 3 :
        if sys.argv[1] in ["-h", "--help"]:
            print_help()
        elif sys.argv[1] in ["-p", "--print"]:
            print_midi(sys.argv[2])
        elif exists(sys.argv[1]):
            main(sys.argv[1], sys.argv[2])
        else:
            print_help()
    else:
        print_help()