2015-03-03 14:11:55 -05:00
|
|
|
from PyQt4 import QtCore, QtGui, uic
|
|
|
|
from PyQt4.QtCore import pyqtSignal, pyqtSlot
|
|
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
|
|
from PIL.ImageQt import ImageQt
|
|
|
|
import core
|
|
|
|
import numpy
|
|
|
|
import subprocess as sp
|
|
|
|
import sys
|
|
|
|
|
|
|
|
class Worker(QtCore.QObject):
|
|
|
|
|
|
|
|
videoCreated = pyqtSignal()
|
|
|
|
progressBarUpdate = pyqtSignal(int)
|
|
|
|
|
|
|
|
def __init__(self, parent=None):
|
|
|
|
QtCore.QObject.__init__(self)
|
|
|
|
parent.videoTask.connect(self.createVideo)
|
|
|
|
self.core = core.Core()
|
|
|
|
|
|
|
|
|
2017-04-20 19:00:17 -04:00
|
|
|
@pyqtSlot(str, str, QtGui.QFont, int, int, int, int, str, str)
|
2017-04-20 13:25:34 -04:00
|
|
|
def createVideo(self, backgroundImage, titleText, titleFont, fontSize, alignment, xOffset, yOffset, inputFile, outputFile):
|
2015-03-03 14:11:55 -05:00
|
|
|
# print('worker thread id: {}'.format(QtCore.QThread.currentThreadId()))
|
|
|
|
|
|
|
|
imBackground = self.core.drawBaseImage(
|
|
|
|
backgroundImage,
|
|
|
|
titleText,
|
2017-04-18 07:46:24 -04:00
|
|
|
titleFont,
|
2017-04-20 13:25:34 -04:00
|
|
|
fontSize,
|
|
|
|
alignment,
|
|
|
|
xOffset,
|
|
|
|
yOffset)
|
2015-03-03 14:11:55 -05:00
|
|
|
|
|
|
|
self.progressBarUpdate.emit(0)
|
|
|
|
|
|
|
|
completeAudioArray = self.core.readAudioFile(inputFile)
|
|
|
|
|
2016-09-10 09:09:58 -04:00
|
|
|
# test if user has libfdk_aac
|
|
|
|
encoders = sp.check_output(self.core.FFMPEG_BIN + " -encoders -hide_banner", shell=True)
|
|
|
|
if b'libfdk_aac' in encoders:
|
|
|
|
acodec = 'libfdk_aac'
|
|
|
|
else:
|
|
|
|
acodec = 'aac'
|
|
|
|
|
|
|
|
ffmpegCommand = [ self.core.FFMPEG_BIN,
|
2015-03-03 14:11:55 -05:00
|
|
|
'-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', inputFile,
|
2016-09-10 09:09:58 -04:00
|
|
|
'-acodec', acodec, # output audio codec
|
2016-09-10 01:13:34 -04:00
|
|
|
'-b:a', "192k",
|
2015-03-04 05:54:29 -05:00
|
|
|
'-vcodec', "libx264",
|
2016-09-10 01:13:34 -04:00
|
|
|
'-pix_fmt', "yuv420p",
|
2015-03-04 05:54:29 -05:00
|
|
|
'-preset', "medium",
|
2016-09-10 09:09:58 -04:00
|
|
|
'-f', "mp4"]
|
|
|
|
|
|
|
|
if acodec == 'aac':
|
|
|
|
ffmpegCommand.append('-strict')
|
|
|
|
ffmpegCommand.append('-2')
|
|
|
|
|
|
|
|
ffmpegCommand.append(outputFile)
|
|
|
|
|
|
|
|
out_pipe = sp.Popen(ffmpegCommand,
|
2015-03-03 14:11:55 -05:00
|
|
|
stdin=sp.PIPE,stdout=sys.stdout, stderr=sys.stdout)
|
|
|
|
|
|
|
|
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:
|
2016-08-17 17:34:05 -04:00
|
|
|
out_pipe.stdin.write(im.tobytes())
|
2015-03-03 14:11:55 -05:00
|
|
|
finally:
|
|
|
|
True
|
|
|
|
|
|
|
|
# increase progress bar value
|
|
|
|
if progressBarValue + 1 <= (i / len(completeAudioArray)) * 100:
|
|
|
|
progressBarValue = numpy.floor((i / len(completeAudioArray)) * 100)
|
|
|
|
self.progressBarUpdate.emit(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()
|
2015-03-04 05:54:29 -05:00
|
|
|
# out_pipe.terminate() # don't terminate ffmpeg too early
|
2015-03-03 14:11:55 -05:00
|
|
|
out_pipe.wait()
|
|
|
|
print("Video file created")
|
|
|
|
self.progressBarUpdate.emit(100)
|
2016-08-17 17:34:05 -04:00
|
|
|
self.videoCreated.emit()
|