GET FREQUENCY AND ENERGY SEQUENCE AND PARSONS CODE
@author: albertovaldez
"""
import onset
import parsons
import easygram as ez
import getBPM_Bass as bpmb
import mp3
songName = "spectre.mp3"
plotPath, csvPath = "PCP/plots/", "PCP/csv/"
wavFile = mp3.toWAV(songName, "mp3/")
song = onset.Song(wavFile)
tr = onset.CalculateThreshold_RMS(song.data)
#bpms = bpmb.GetBPM(song, tr)
bpm_user = int(input("Enter selected BPM: \n"))
#SONG THRESHOLD FOR ALPHA PEAK, 0.0 if cut, 0.1 if instant, 0.5 if soft fade in, 0.8 if loud fade in
songTR = 0.8
noteTR = 0.7
alphaPeak = song.FindAlphaPeak(0, songTR)
bpm, measure, unitSize = bpm_user, 4, 0.25
#freqBands = [i*60 for i in range(5,26)] # Melody/Lead use continuous
#freqBands = [i*60 for i in range(2,6)] # Body/Snare use step
#freqBands = [9000, 16000] # High/Hats need custom de-peak process
freqBands = [i * 60 for i in range(0, 3)] # Bass/Kick use continuous
maxBars = int((song.length_seconds * (1 / (60 / bpm)) / measure))
def Song(file):
p = Path(file)
directory = str(p.parent) + "/"
if ".mp3" in file:
wavFile = toWAV(file)
song = onset.Song(wavFile)
else:
song = onset.Song(file)
songName = file.split("/")[-1:][0]
print(songName)
u1 = int(input(("Press 1 if your song has no fade-in, Press 2 if the fade-in is quiet. Press 3 if the fade-in is loud.\n")))
if u1 == 1:
songTR = 0.1
elif u1 == 2:
songTR = 0.5
elif u1 == 3:
songTR = 0.8
else:
songTR = 0.8
alphaPeak = song.FindAlphaPeak(0,songTR)
u2 = str(input("Press 1 to get a list of suggested BPMs. Press any other key to skip:\n"))
if u2 == "1":
bpmb.GetBPM(song, onset.CalculateThreshold_RMS(song.data))
bpm_user = float(input("Enter selected BPM: \n"))
measure_user = int(input("Enter how many beats are in a bar: \n"))
bpm, measure = bpm_user, measure_user
maxBars = int((song.length_seconds * (1/(60/bpm)) / measure ))
print("Song duration in bars: " + str(maxBars))
def Melody():
noteThreshold = 0.6
unitSize = 0.25
print("\nObtaining Melody...")
# Continuous, Peaks, 4 by 4, unit 0.25
x_all, y_energy, z_freq = [],[],[]
freqBands = [300,1800]
barBlock = 4
for i in range(maxBars//barBlock):
barNum = i*barBlock
x,y,z = ez.GetNotesPeaks3D_Continuous(freqBands, song, bpm, barNum, barNum+barBlock, measure, unitSize, noteThreshold)
offset = i*barBlock*measure*(1/unitSize)
for j in range(len(y)):
x_all.append(x[j] + offset)
y_energy.append(y[j])
z_freq.append(z[j])
peaks = []
for i in x_all:
peaks.append(i * (60/bpm)*unitSize * song.sampfreq)
onset.SavePeaks(peaks, song.sampfreq, 1, song.peakAlphaIndex, directory + "peaksMelody" + "_" + songName.split(".")[0] + ".csv")
pc_e,pc_f = parsons.GetPCode(x_all, y_energy),parsons.GetPCode(x_all, z_freq)
x = [i*unitSize for i in x_all]
ez.PlotComplete(x, pc_e, pc_f, bpm, maxBars, measure, unitSize, freqBands, directory, "plotMelody_" + songName, noteThreshold, song.GetRMS(), alphaPeak)
parsons.SaveCSV3D(x, pc_e,pc_f, directory, "dataMelody_" + songName)
print("Done.")
def Snare():
noteThreshold = 0.6
unitSize = 0.5
print("\nObtaining Snare...")
# Step, Peaks, 4 by 4, unit 0.5
x_all, y_energy, z_freq = [],[],[]
freqBands = [120,300]
barBlock = 4
for i in range(maxBars//barBlock):
barNum = i*barBlock
x,y,z = ez.GetNotesPeaks3D_Step(freqBands, song, bpm, barNum, barNum+barBlock, measure, unitSize, noteThreshold)
offset = i*barBlock*measure*(1/unitSize)
for j in range(len(y)):
x_all.append(x[j] + offset)
y_energy.append(y[j])
z_freq.append(z[j])
peaks = []
for i in x_all:
peaks.append(i * (60/bpm)*unitSize * song.sampfreq)
onset.SavePeaks(peaks, song.sampfreq, 1, song.peakAlphaIndex, directory + "peaksSnare" + "_" + songName.split(".")[0] + ".csv")
pc_e,pc_f = parsons.GetPCode(x_all, y_energy),parsons.GetPCode(x_all, z_freq)
x = [i*unitSize for i in x_all]
ez.PlotComplete(x, pc_e, pc_f, bpm, maxBars, measure, unitSize, freqBands, directory, "plotSnare_" + songName, noteThreshold, song.GetRMS(), alphaPeak)
parsons.SaveCSV3D(x, pc_e,pc_f, directory, "dataSnare_" + songName)
print("Done.")
def Bass():
noteThreshold = 0.7
unitSize = 0.25
print("\nObtaining Bass...")
# Continuous, Peaks, all bars, unit 0.25
x_all, y_energy, z_freq = [],[],[]
freqBands = [0,120]
barBlock = maxBars
for i in range(maxBars//barBlock):
barNum = i*barBlock
x,y,z = ez.GetNotesPeaks3D_Continuous(freqBands, song, bpm, barNum, barNum+barBlock, measure, unitSize, noteThreshold)
offset = i*barBlock*measure*(1/unitSize)
for j in range(len(y)):
x_all.append(x[j] + offset)
y_energy.append(y[j])
z_freq.append(z[j])
peaks = []
for i in x_all:
peaks.append(i * (60/bpm)*unitSize * song.sampfreq)
onset.SavePeaks(peaks, song.sampfreq, 1, song.peakAlphaIndex, directory + "peaksBass" + "_" + songName.split(".")[0] + ".csv")
pc_e,pc_f = parsons.GetPCode(x_all, y_energy),parsons.GetPCode(x_all, z_freq)
x = [i*unitSize for i in x_all]
ez.PlotComplete(x, pc_e, pc_f, bpm, maxBars, measure, unitSize, freqBands, directory, "plotBass_" + songName, noteThreshold, song.GetRMS(), alphaPeak)
parsons.SaveCSV3D(x, pc_e,pc_f, directory, "dataBass_" + songName)
print("Done.")
def Hats():
noteThreshold = 0.7
unitSize = 0.25
print("\nObtaining HiHats...")
# Continuous, Peaks, all bars, unit 0.25
x_all, y_energy, z_freq = [],[],[]
freqBands = [9000,16000]
barBlock = maxBars
for i in range(maxBars//barBlock):
barNum = i*barBlock
x,y,z = ez.GetNotesPeaks3D_Continuous_Sum(freqBands, song, bpm, barNum, barNum+barBlock, measure, unitSize, noteThreshold)
offset = i*barBlock*measure*(1/unitSize)
for j in range(len(y)):
x_all.append(x[j] + offset)
y_energy.append(y[j])
z_freq.append(z[j])
peaks = []
for i in x_all:
peaks.append(i * (60/bpm)*unitSize * song.sampfreq)
onset.SavePeaks(peaks, song.sampfreq, 1, song.peakAlphaIndex, directory + "peaksHats" + "_" + songName.split(".")[0] + ".csv")
pc_e,pc_f = parsons.GetPCode(x_all, y_energy),parsons.GetPCode(x_all, z_freq)
x = [i*unitSize for i in x_all]
ez.PlotComplete(x, pc_e, pc_f, bpm, maxBars, measure, unitSize, freqBands, directory, "plotHats_" + songName, noteThreshold, song.GetRMS(), alphaPeak)
parsons.SaveCSV3D(x, pc_e,pc_f, directory, "dataHats_" + songName)
print("Done.")
Melody()
Snare()
Bass()
Hats()
os.remove(wavFile)