first commit

1 files changed, 196 insertions, 0 deletions

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..df18222
--- /dev/null
+++ b/main.py
@@ -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_())