class MainWindow_EXEC():
def __init__(self):
#-------------------Init QT Setup---------------------------
app = QtWidgets.QApplication(sys.argv)
self.MainWindow = QtWidgets.QMainWindow()
self.ui = Ui_MainWindow()
self.ui.setupUi(self.MainWindow)
app.setStyleSheet(qdarkstyle.load_stylesheet_pyqt5())
#------------------Exporting Setup------------------------------
self.ui.export_midi.clicked.connect(self.openDirectory_midi)
self.ui.export_midi.setFocusPolicy(QtCore.Qt.NoFocus)
self.ui.export_audio.clicked.connect(self.openDirectory_audio)
self.ui.export_audio.setFocusPolicy(QtCore.Qt.NoFocus)
#------------------Metronome Setup------------------------------
self.ui.metronome_button.clicked.connect(self.metro_thread)
#------------------Recording Setup------------------------------
self.ui.start_stop_rec.clicked.connect(self.start_stop_recording)
self.ui.play_gui.clicked.connect(self.play)
# QAudio setup
self.settings = QAudioEncoderSettings()
self.settings.setBitRate(16)
self.settings.setChannelCount(1)
self.audioRecorder = QAudioRecorder()
self.audioRecorder.setEncodingSettings(self.settings)
self.file_path = os.path.abspath(
os.path.join(os.path.dirname(__file__),
resource_path("resources/output.wav")))
self.url = QUrl.fromLocalFile(self.file_path)
self.audioRecorder.setOutputLocation(self.url)
#------------------Audio Terrain Gui Setup------------------------------
self.terrain = Terrain()
self.terrain.update()
self.terrain_widget = self.terrain.getwidget()
self.layout = QtGui.QGridLayout()
self.layout.setContentsMargins(0, 0, 0, 0)
self.ui.t_widget.setLayout(self.layout)
self.layout.addWidget(self.terrain_widget, 0, 0, 1, 1)
#------------------Audio Trimmer Setup------------------------------
self.ui.audio_trimmer.clicked.connect(self.trim_audio)
if os.path.isfile("resources/output.wav"):
self.y, self.sr = librosa.load(
resource_path("resources/output.wav"), sr=44100)
else:
new_wav = AudioSegment.empty()
new_wav.export("resources/output.wav", format="wav")
self.y, self.sr = librosa.load(
resource_path("resources/output.wav"), sr=44100)
self.duration = round(
librosa.core.get_duration(y=self.y, sr=self.sr) * self.sr)
self.maxv = np.iinfo(np.int16).max
self.win = pg.GraphicsLayoutWidget()
self.p = self.win.addPlot()
#removes X & Y Axis and disables mouse movement
self.p.showAxis('bottom', show=False)
self.p.showAxis('left', show=False)
self.p.setMouseEnabled(x=False, y=False)
self.region = pg.LinearRegionItem(brush=(100, 100, 100, 60),
bounds=(0, self.duration))
self.region.setRegion([0, self.duration])
self.p.addItem(self.region, ignoreBounds=True)
self.p.plot(self.y, pen="w")
self.layout.addWidget(self.win)
self.win.hide()
#------------------Midi Setup------------------------------
self.ui.convert_midi.clicked.connect(self.convertMidi)
self.ui.midi_play.clicked.connect(self.midiplayer_thread)
self.ui.tempo_slider.valueChanged[int].connect(self.tempo_value)
self.ui.midi_loop.toggle()
# default bpm is 120
self.current_tempo = 120
self.detected_tempo = 120
#------------------Drum Kit Selector Setup----------------------
self.ui.drum_kits.clicked.connect(self.select_drumkit)
self.drum_number = 0
self.drum_folders = [
'Drum_Kit_1', 'Drum_Kit_2', 'Drum_Kit_3', 'Drum_Kit_4'
]
self.drum_current = self.drum_folders[self.drum_number]
#------------------EXEC Window---------------------------------
self.MainWindow.show()
sys.exit(app.exec_())
#---------------------------------------------------------------
#------------------Functions----------------------------------
#------------------Drum Kit Selector------------------------------
def select_drumkit(self):
if self.drum_number < 3:
self.drum_number += 1
self.drum_current = self.drum_folders[self.drum_number]
self.ui.drum_kits.setText(self.drum_current.replace("_", " "))
else:
self.drum_number = 0
self.drum_current = self.drum_folders[self.drum_number]
self.ui.drum_kits.setText(self.drum_current.replace("_", " "))
#------------------Audio Trimmer------------------------------
def trim_audio(self):
# Switch to Trimmer widget
self.layout.removeWidget(self.terrain_widget)
self.terrain_widget.hide()
self.win.show()
self.trim_values = self.region.getRegion()
self.updateaudio()
# Trims signal array with region values
self.y = self.y[round(self.trim_values[0]):round(self.trim_values[1])]
# save the new signal values to wav
librosa.output.write_wav(resource_path("resources/output.wav"),
(self.y * self.maxv).astype(np.int16),
self.sr)
self.updateplot()
def updateplot(self):
# Replot the trimmed wav and update region bounds
self.duration = round(
librosa.core.get_duration(y=self.y, sr=self.sr) * self.sr)
self.p.plot(clear=True)
self.p.plot(self.y, pen="w")
self.region = pg.LinearRegionItem(brush=(100, 100, 100, 50),
bounds=(0, self.duration))
self.p.addItem(self.region, ignoreBounds=True)
self.region.setRegion([0, self.duration])
def updateaudio(self):
self.y, self.sr = librosa.load(resource_path("resources/output.wav"),
sr=44100)
#------------------Metronome Threading------------------------------
def metro_thread(self):
if self.ui.metronome_button.isChecked():
print('metronome is On')
self.thread = QThread(
) # a new thread to run our background tasks in
self.worker = Worker(
self.current_tempo) # a new worker to perform those tasks
self.worker.moveToThread(
self.thread
) # move the worker into the thread, do this first before connecting the signals
self.thread.started.connect(
self.worker.work
) # begin our worker object's loop when the thread starts running
self.thread.start()
else:
print('metronome is Off')
self.stop_loop()
self.worker.finished.connect(
self.loop_finished
) # do something in the gui when the worker loop ends
self.worker.finished.connect(
self.thread.quit) # tell the thread it's time to stop running
self.worker.finished.connect(self.thread.wait)
self.worker.finished.connect(
self.worker.deleteLater
) # have worker mark itself for deletion
self.thread.finished.connect(self.thread.deleteLater)
def stop_loop(self):
self.worker.working = False
def loop_finished(self):
# print('Worker Finished')
pass
#---------------------------------------------------------
#------------------ MIDI ------------------------------
def tempo_value(self, value):
self.current_tempo = value
def convertMidi(self):
self.ui.convert_midi.setEnabled(False)
self.thread2 = QThread()
self.worker2 = ConvertMidi_Worker()
self.worker2.moveToThread(self.thread2)
self.thread2.started.connect(self.worker2.work)
self.thread2.start()
self.worker2.finished.connect(self.convert_finished)
self.worker2.finished.connect(self.thread2.quit)
self.worker2.finished.connect(self.thread2.wait)
self.worker2.finished.connect(self.worker2.deleteLater)
self.thread2.finished.connect(self.thread2.deleteLater)
def convert_finished(self, tempo):
self.detected_tempo = tempo
self.ui.tempo_slider.setValue(self.detected_tempo)
self.ui.convert_midi.clearFocus()
self.ui.convert_midi.setEnabled(True)
print('Midi Conversion finished')
def midiplayer_thread(self):
if self.ui.midi_play.isChecked() and self.ui.midi_loop.isChecked(
) == False:
self.ui.midi_play.setEnabled(False)
self.win.hide()
self.terrain_widget.show()
self.terrain.animate()
self.thread3 = QThread()
self.worker3 = MidiPlayer_Worker(self.current_tempo,
self.drum_current)
self.worker3.moveToThread(self.thread3)
self.thread3.started.connect(self.worker3.workonce)
self.thread3.start()
self.worker3.finished2.connect(self.midi_loop_finished2)
self.worker3.finished2.connect(self.thread3.quit)
self.worker3.finished2.connect(self.thread3.wait)
self.worker3.finished2.connect(self.worker3.deleteLater)
self.thread3.finished.connect(self.thread3.deleteLater)
elif self.ui.midi_play.isChecked() and self.ui.midi_loop.isChecked(
) == True:
self.win.hide()
self.terrain_widget.show()
self.start_Midi_Thread()
self.terrain.animate()
elif self.ui.midi_play.isChecked() == False:
self.terrain.stop_animate()
self.stop_Midi_Thread()
def start_Midi_Thread(self):
self.thread3 = QThread()
self.worker3 = MidiPlayer_Worker(self.current_tempo, self.drum_current)
self.worker3.moveToThread(self.thread3)
self.thread3.started.connect(self.worker3.work)
self.thread3.start()
def stop_Midi_Thread(self):
self.worker3.working = False
self.worker3.stop()
self.worker3.finished.connect(self.midi_loop_finished)
self.worker3.finished.connect(self.thread3.quit)
self.worker3.finished.connect(self.thread3.wait)
self.worker3.finished.connect(self.worker3.deleteLater)
self.thread3.finished.connect(self.thread3.deleteLater)
print('done')
def midi_loop_finished(self):
print('Midi loop Finished')
def midi_loop_finished2(self):
print('Midi Player Finished')
self.ui.midi_play.toggle()
self.ui.midi_play.setEnabled(True)
self.terrain.stop_animate()
#---------------------------------------------------------
#------------------ Recorder & Player ------------------------------
def start_stop_recording(self):
if self.ui.start_stop_rec.isChecked():
self.ui.play_gui.setEnabled(False)
self.ui.audio_trimmer.setEnabled(False)
self.win.hide()
self.terrain_widget.show()
self.layout.addWidget(self.terrain_widget)
self.audioRecorder.record()
self.terrain.update()
self.terrain.animate()
print('Recording...')
else:
self.ui.play_gui.setEnabled(True)
self.ui.audio_trimmer.setEnabled(True)
self.terrain.stop_animate()
self.audioRecorder.stop()
self.layout.removeWidget(self.terrain_widget)
self.terrain_widget.hide()
self.updateaudio()
self.win.show()
self.updateplot()
print('Stop Recording')
def play(self):
if self.ui.play_gui.isChecked():
self.win.hide()
self.terrain_widget.show()
self.player = QSound(resource_path("resources/output.wav"))
self.terrain.animate()
self.player.play()
# if self.player.isFinished():
# self.ui.play_gui.toggle()
# print('done')
else:
self.terrain.stop_animate()
self.player.stop()
self.player.deleteLater()
#------------------ Exporting ------------------------------
def openDirectory_midi(self):
self.openDirectoryDialog = QtGui.QFileDialog.getExistingDirectory(
self.MainWindow, "Save Midi File")
if self.openDirectoryDialog:
self.saveMidi(self.openDirectoryDialog)
else:
pass
def openDirectory_audio(self):
self.openDirectoryDialog = QtGui.QFileDialog.getExistingDirectory(
self.MainWindow, "Save Audio File")
if self.openDirectoryDialog:
self.saveAudio(self.openDirectoryDialog)
else:
pass
def saveMidi(self, directory):
shutil.copy("resources/beatbox.mid", directory)
def saveAudio(self, directory):
shutil.copy("resources/output.wav", directory)