98 lines
3.2 KiB
Python
98 lines
3.2 KiB
Python
from PyQt4 import QtCore
|
|
from PyQt4.QtCore import QSettings
|
|
import argparse
|
|
import os
|
|
|
|
import core
|
|
import video_thread
|
|
from main import LoadDefaultSettings
|
|
|
|
|
|
class Command(QtCore.QObject):
|
|
|
|
videoTask = QtCore.pyqtSignal(str, str, list)
|
|
|
|
def __init__(self):
|
|
QtCore.QObject.__init__(self)
|
|
self.core = core.Core()
|
|
self.dataDir = self.core.dataDir
|
|
|
|
self.parser = argparse.ArgumentParser(
|
|
description='Create a visualization for an audio file')
|
|
self.parser.add_argument(
|
|
'-i', '--input', help='input audio file', required=True)
|
|
self.parser.add_argument(
|
|
'-o', '--output', help='output video file', required=True)
|
|
|
|
# optional arguments
|
|
self.parser.add_argument(
|
|
'projpath', metavar='path-to-project',
|
|
help='open a project file (.avp)', nargs='?')
|
|
|
|
'''
|
|
self.parser.add_argument(
|
|
'-b', '--background', dest='bgimage',
|
|
help='background image file', required=True)
|
|
self.parser.add_argument(
|
|
'-t', '--text', dest='text', help='title text', required=True)
|
|
self.parser.add_argument(
|
|
'-f', '--font', dest='font', help='title font', required=False)
|
|
self.parser.add_argument(
|
|
'-s', '--fontsize', dest='fontsize',
|
|
help='title font size', required=False)
|
|
self.parser.add_argument(
|
|
'-c', '--textcolor', dest='textcolor',
|
|
help='title text color in r,g,b format', required=False)
|
|
self.parser.add_argument(
|
|
'-C', '--viscolor', dest='viscolor',
|
|
help='visualization color in r,g,b format', required=False)
|
|
self.parser.add_argument(
|
|
'-x', '--xposition', dest='xposition',
|
|
help='x position', required=False)
|
|
self.parser.add_argument(
|
|
'-y', '--yposition', dest='yposition',
|
|
help='y position', required=False)
|
|
self.parser.add_argument(
|
|
'-a', '--alignment', dest='alignment',
|
|
help='title alignment', required=False,
|
|
type=int, choices=[0, 1, 2])
|
|
'''
|
|
|
|
self.args = self.parser.parse_args()
|
|
self.settings = QSettings(
|
|
os.path.join(self.dataDir, 'settings.ini'), QSettings.IniFormat)
|
|
LoadDefaultSettings(self)
|
|
|
|
if self.args.projpath:
|
|
self.core.openProject(self, self.args.projpath)
|
|
|
|
self.createAudioVisualisation()
|
|
|
|
def createAudioVisualisation(self):
|
|
self.videoThread = QtCore.QThread(self)
|
|
self.videoWorker = video_thread.Worker(self)
|
|
self.videoWorker.moveToThread(self.videoThread)
|
|
self.videoWorker.videoCreated.connect(self.videoCreated)
|
|
|
|
self.videoThread.start()
|
|
self.videoTask.emit(
|
|
self.args.input,
|
|
self.args.output,
|
|
self.core.selectedComponents)
|
|
|
|
def videoCreated(self):
|
|
self.videoThread.quit()
|
|
self.videoThread.wait()
|
|
self.cleanUp()
|
|
|
|
def showMessage(self, **kwargs):
|
|
print(kwargs['msg'])
|
|
if 'detail' in kwargs:
|
|
print(kwargs['detail'])
|
|
|
|
def drawPreview(self, *args):
|
|
pass
|
|
|
|
def cleanUp(self, *args):
|
|
pass
|