Spaces:
Sleeping
Sleeping
File size: 3,703 Bytes
22890fa |
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 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 |
"""Midi creator module"""
import math
from collections import Counter
import librosa
import numpy as np
import pretty_midi
from modules.Pitcher.pitcher import get_frequencies_with_high_confidence
from modules.Ultrastar.ultrastar_converter import (
get_end_time_from_ultrastar,
get_start_time_from_ultrastar,
ultrastar_note_to_midi_note,
)
from modules.console_colors import (
ULTRASINGER_HEAD,
red_highlighted,
)
from modules.Ultrastar.ultrastar_txt import UltrastarTxtValue
from modules.Pitcher.pitched_data import PitchedData
def convert_ultrastar_to_midi_instrument(ultrastar_class: UltrastarTxtValue) -> object:
"""Converts an Ultrastar data to a midi instrument"""
print(f"{ULTRASINGER_HEAD} Creating midi instrument from Ultrastar txt")
instrument = pretty_midi.Instrument(program=0)
velocity = 100
for i in enumerate(ultrastar_class.words):
pos = i[0]
start_time = get_start_time_from_ultrastar(ultrastar_class, pos)
end_time = get_end_time_from_ultrastar(ultrastar_class, pos)
pitch = ultrastar_note_to_midi_note(int(ultrastar_class.pitches[pos]))
note = pretty_midi.Note(velocity, pitch, start_time, end_time)
instrument.notes.append(note)
return instrument
def instruments_to_midi(instruments: list[object], bpm: float, midi_output: str) -> None:
"""Write instruments to midi file"""
print(f"{ULTRASINGER_HEAD} Creating midi file -> {midi_output}")
midi_data = pretty_midi.PrettyMIDI(initial_tempo=bpm)
for instrument in instruments:
midi_data.instruments.append(instrument)
midi_data.write(midi_output)
class MidiCreator:
"""Docstring"""
def convert_frequencies_to_notes(frequency: [str]) -> list[list[str]]:
"""Converts frequencies to notes"""
notes = []
for freq in frequency:
notes.append(librosa.hz_to_note(float(freq)))
return notes
def most_frequent(array: [str]) -> list[tuple[str, int]]:
"""Get most frequent item in array"""
return Counter(array).most_common(1)
def find_nearest_index(array: list[float], value: float) -> int:
"""Nearest index in array"""
idx = np.searchsorted(array, value, side="left")
if idx > 0 and (
idx == len(array)
or math.fabs(value - array[idx - 1]) < math.fabs(value - array[idx])
):
return idx - 1
return idx
def create_midi_notes_from_pitched_data(start_times: list[float], end_times: list[float], pitched_data: PitchedData) -> list[str]:
"""Create midi notes from pitched data"""
print(f"{ULTRASINGER_HEAD} Creating midi notes from pitched data")
midi_notes = []
for i in enumerate(start_times):
pos = i[0]
start_time = start_times[pos]
end_time = end_times[pos]
note = create_midi_note_from_pitched_data(
start_time, end_time, pitched_data
)
midi_notes.append(note)
# todo: Progress?
# print(filename + " f: " + str(mean))
return midi_notes
def create_midi_note_from_pitched_data(start_time: float, end_time: float, pitched_data: PitchedData) -> str:
"""Create midi note from pitched data"""
start = find_nearest_index(pitched_data.times, start_time)
end = find_nearest_index(pitched_data.times, end_time)
if start == end:
freqs = [pitched_data.frequencies[start]]
confs = [pitched_data.confidence[start]]
else:
freqs = pitched_data.frequencies[start:end]
confs = pitched_data.confidence[start:end]
conf_f = get_frequencies_with_high_confidence(freqs, confs)
notes = convert_frequencies_to_notes(conf_f)
note = most_frequent(notes)[0][0]
return note
|