def __setRecordParams(self): ##设置音频输入参数
selectedFile = self.ui.editOutputFile.text().strip()
if (selectedFile == ""):
QMessageBox.critical(self, "错误", "请先设置录音输出文件")
return False
if os.path.exists(selectedFile):
os.remove(selectedFile) # 删除已有文件
## QMessageBox.critical(self,"错误","录音输出文件被占用，无法删除")
## return False
recordFile = QUrl.fromLocalFile(selectedFile)
self.recorder.setOutputLocation(recordFile) # 设置输出文件
recordDevice = self.ui.comboDevices.currentText()
self.recorder.setAudioInput(recordDevice) # 设置录入设备
settings = QAudioEncoderSettings() # 音频编码设置
settings.setCodec(self.ui.comboCodec.currentText()) # 编码
sampRate = int(self.ui.comboSampleRate.currentText())
settings.setSampleRate(sampRate) # 采样率
channelCount = int(self.ui.comboChannels.currentText())
settings.setChannelCount(channelCount) # 通道数
settings.setEncodingMode(QMultimedia.ConstantBitRateEncoding) # 固定比特率
self.recorder.setAudioSettings(settings) # 音频设置
return True
class SimpleAudioRecorder(QAudioRecorder):
def __init__(self, data: ConverterData, transcription: Transcription,
app_settings: AppSettings) -> None:
super().__init__()
self.app_settings = app_settings
self.temp = data.get_temp_file()
self.transcription = transcription
self.file_path = os.path.join(self.temp,
f'{self.transcription.id}.wav')
self.settings = QAudioEncoderSettings()
def start_recording(self) -> None:
LOG_RECORDER.info("Audio recording started.")
self.settings.setCodec('audio/pcm')
self.settings.setChannelCount(1)
self.settings.setBitRate(96000)
self.settings.setSampleRate(44100)
self.settings.setQuality(self.app_settings.audio_quality)
self.settings.setEncodingMode(QMultimedia.ConstantQualityEncoding)
container = 'audio/x-wav'
self.setEncodingSettings(self.settings, QVideoEncoderSettings(),
container)
self.setOutputLocation(QUrl.fromLocalFile(self.file_path))
self.record()
def stop_recording(self) -> Union[str, None]:
LOG_RECORDER.info("Audio recording finished.")
LOG_RECORDER.info(f"Audio file: {self.file_path}")
self.stop()
return self.file_path
def __setRecordParams(self): ##设置音频输入参数
selectedFile = self.ui.editOutputFile.text().strip()
if (selectedFile == ""):
QMessageBox.critical(self, "错误", "请先设置录音输出文件")
return False
if os.path.exists(selectedFile):
os.remove(selectedFile) #删除已有文件
## QMessageBox.critical(self,"错误","录音输出文件被占用，无法删除")
## return False
recordFile = QUrl.fromLocalFile(selectedFile)
self.recorder.setOutputLocation(recordFile) #设置输出文件
recordDevice = self.ui.comboDevices.currentText()
self.recorder.setAudioInput(recordDevice) #设置录入设备
settings = QAudioEncoderSettings() #音频编码设置
settings.setCodec(self.ui.comboCodec.currentText()) #编码
sampRate = int(self.ui.comboSampleRate.currentText())
settings.setSampleRate(sampRate) #采样率
bitRate = int(self.ui.comboBitrate.currentText())
settings.setBitRate(bitRate) #比特率
channelCount = int(self.ui.comboChannels.currentText())
settings.setChannelCount(channelCount) #通道数
## quality=self.ui.sliderQuality.value() #也可以
quality = QMultimedia.EncodingQuality(self.ui.sliderQuality.value())
settings.setQuality(quality) #品质
if self.ui.radioQuality.isChecked(): #编码模式为固定品质,自动决定采样率，采样点大小
settings.setEncodingMode(QMultimedia.ConstantQualityEncoding)
else:
settings.setEncodingMode(
QMultimedia.ConstantBitRateEncoding) #固定比特率
self.recorder.setAudioSettings(settings) #音频设置
return True
def record(self, filename):
url = QtCore.QUrl.fromLocalFile(
QtCore.QFileInfo(filename).absoluteFilePath())
#content = QMediaObject(url)
#self.recorder = QAudioRecorder()
#source = QAudioInput()
#source = QMediaObject()
self.recorder = QAudioRecorder()
settings = QAudioEncoderSettings()
settings.setChannelCount(1)
settings.setSampleRate(44100)
settings.setBitRate(32)
settings.setCodec("audio/vorbis")
#settings.setEncodingMode(QtMultimedia.ConstantQualityEnconding)
self.recorder.setContainerFormat("ogg")
self.recorder.setEncodingSettings(settings)
self.recorder.setOutputLocation(url)
def __init__(self):
super(App, self).__init__()
self.loginButton.clicked.connect(self.showLoginDialog)
self.client_thread.text_signal.connect(self.showText) # 显示文本消息
self.client_thread.usr_signal.connect(self.showUserList) # 更新在线用户
self.client_thread.file_signal.connect(self.showFile) # 显示文件消息
self.emojis.emoji_signal.connect(self.addEmoji)
# 通过QListWidget的当前item变化来切换QStackedWidget中的序号
self.userListWidget.currentRowChanged.connect(self.dialogChanged)
self.usrList = [] # 保存上一次的在线用户列表
self.groupList = [] # 群组列表
self.md5 = hashlib.md5() # 用于加密密码
# 录音机
self.recorder = QAudioRecorder(self)
settings = QAudioEncoderSettings()
settings.setChannelCount(2)
settings.setSampleRate(16000)
self.recorder.setEncodingSettings(settings)
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)