diff options
Diffstat (limited to 'main.py')
| -rw-r--r-- | main.py | 196 |
1 files changed, 196 insertions, 0 deletions
@@ -0,0 +1,196 @@ +import sys, io, os +from PyQt4 import QtCore, QtGui, uic +from PyQt4.QtGui import QPainter, QColor, QFont +from os.path import expanduser +import subprocess as sp +import numpy +from PIL import Image, ImageDraw, ImageFont +from PIL.ImageQt import ImageQt +import atexit +from queue import Queue +from PyQt4.QtCore import QSettings + +import preview_thread, core + +class Main(QtCore.QObject): + + newTask = QtCore.pyqtSignal(str, str, QFont) + processTask = QtCore.pyqtSignal() + + def __init__(self, window): + + QtCore.QObject.__init__(self) + + # print('main thread id: {}'.format(QtCore.QThread.currentThreadId())) + self.window = window + self.core = core.Core() + + self.settings = QSettings('settings.ini', QSettings.IniFormat) + + self.previewQueue = Queue() + + self.previewThread = QtCore.QThread(self) + self.previewWorker = preview_thread.Worker(self, self.previewQueue) + + self.previewWorker.moveToThread(self.previewThread) + self.previewWorker.imageCreated.connect(self.showPreviewImage) + + self.previewThread.start() + + self.timer = QtCore.QTimer(self) + self.timer.timeout.connect(self.processTask.emit) + self.timer.start(500) + + window.pushButton_selectInput.clicked.connect(self.openInputFileDialog) + window.pushButton_selectOutput.clicked.connect(self.openOutputFileDialog) + window.pushButton_createVideo.clicked.connect(self.createAudioVisualisation) + window.pushButton_selectBackground.clicked.connect(self.openBackgroundFileDialog) + + window.fontComboBox.currentFontChanged.connect(self.drawPreview) + window.lineEdit_title.textChanged.connect(self.drawPreview) + + window.progressBar_create.setValue(0) + window.setWindowTitle("Audio Visualizer") + window.pushButton_selectInput.setText("Select Input Music File") + window.pushButton_selectOutput.setText("Select Output Video File") + window.pushButton_selectBackground.setText("Select Background Image") + window.label_font.setText("Title Font") + window.label_title.setText("Title Text") + window.pushButton_createVideo.setText("Create Video") + window.groupBox_create.setTitle("Create") + window.groupBox_settings.setTitle("Settings") + window.groupBox_preview.setTitle("Preview") + + titleFont = self.settings.value("titleFont") + if not titleFont == None: + window.fontComboBox.setCurrentFont(QFont(titleFont)) + + self.drawPreview() + + window.show() + + def cleanUp(self): + self.timer.stop() + self.previewThread.quit() + self.previewThread.wait() + + self.settings.setValue("titleFont", self.window.fontComboBox.currentFont().toString()) + + def openInputFileDialog(self): + inputDir = self.settings.value("inputDir", expanduser("~")) + + fileName = QtGui.QFileDialog.getOpenFileName(self.window, + "Open Music File", inputDir, "Music Files (*.mp3 *.wav *.ogg *.flac)"); + + if not fileName == "": + self.settings.setValue("inputDir", os.path.dirname(fileName)) + self.window.label_input.setText(fileName) + + def openOutputFileDialog(self): + outputDir = self.settings.value("outputDir", expanduser("~")) + + fileName = QtGui.QFileDialog.getSaveFileName(self.window, + "Set Output Video File", outputDir, "Video Files (*.mp4)"); + + if not fileName == "": + self.settings.setValue("outputDir", os.path.dirname(fileName)) + self.window.label_output.setText(fileName) + + def openBackgroundFileDialog(self): + backgroundDir = self.settings.value("backgroundDir", expanduser("~")) + + fileName = QtGui.QFileDialog.getOpenFileName(self.window, + "Open Background Image", backgroundDir, "Image Files (*.jpg *.png)"); + + if not fileName == "": + self.settings.setValue("backgroundDir", os.path.dirname(fileName)) + self.window.label_background.setText(fileName) + self.drawPreview() + + def createAudioVisualisation(self): + + imBackground = self.core.drawBaseImage( + self.window.label_background.text(), + self.window.lineEdit_title.text(), + self.window.fontComboBox.currentFont()) + + self.window.progressBar_create.setValue(0) + + completeAudioArray = self.core.readAudioFile(self.window.label_input.text()) + + out_pipe = sp.Popen([ self.core.FFMPEG_BIN, + '-y', # (optional) means overwrite the output file if it already exists. + '-f', 'rawvideo', + '-vcodec', 'rawvideo', + '-s', '1280x720', # size of one frame + '-pix_fmt', 'rgb24', + '-r', '30', # frames per second + '-i', '-', # The input comes from a pipe + '-an', + '-i', self.window.label_input.text(), + '-acodec', "libmp3lame", # output audio codec + self.window.label_output.text()], + stdin=sp.PIPE,stdout=sp.DEVNULL, stderr=sp.DEVNULL) + + smoothConstantDown = 0.08 + smoothConstantUp = 0.8 + lastSpectrum = None + progressBarValue = 0 + sampleSize = 1470 + + numpy.seterr(divide='ignore') + + for i in range(0, len(completeAudioArray), sampleSize): + # create video for output + lastSpectrum = self.core.transformData( + i, + completeAudioArray, + sampleSize, + smoothConstantDown, + smoothConstantUp, + lastSpectrum) + im = self.core.drawBars(lastSpectrum, imBackground) + + # write to out_pipe + try: + out_pipe.stdin.write(im.tostring()) + finally: + True + + # increase progress bar value + if progressBarValue + 1 <= (i / len(completeAudioArray)) * 100: + progressBarValue = numpy.floor((i / len(completeAudioArray)) * 100) + self.window.progressBar_create.setValue(progressBarValue) + + numpy.seterr(all='print') + + out_pipe.stdin.close() + if out_pipe.stderr is not None: + print(out_pipe.stderr.read()) + out_pipe.stderr.close() + out_pipe.terminate() + out_pipe.wait() + print("Video file created") + self.window.progressBar_create.setValue(100) + + def drawPreview(self): + self.newTask.emit(self.window.label_background.text(), + self.window.lineEdit_title.text(), + self.window.fontComboBox.currentFont()) + # self.processTask.emit() + + def showPreviewImage(self, image): + self._scaledPreviewImage = image + self._previewPixmap = QtGui.QPixmap.fromImage(self._scaledPreviewImage) + + self.window.label_preview.setPixmap(self._previewPixmap) + +if __name__ == "__main__": + app = QtGui.QApplication(sys.argv) + window = uic.loadUi("main.ui") + + main = Main(window) + + atexit.register(main.cleanUp) + + sys.exit(app.exec_()) |