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', epilog='EXAMPLE COMMAND: main.py myvideotemplate.avp ' '-i ~/Music/song.mp3 -o ~/video.mp4 ' '-c 0 image ~/Pictures/thisWeeksPicture.jpg ' '-c 1 vis classic') self.parser.add_argument( '-i', '--input', metavar='SOUND', help='input audio file', required=True) self.parser.add_argument( '-o', '--output', metavar='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( '-c', '--comp', metavar=('LAYER', 'NAME', 'ARG'), help='create/edit component NAME at LAYER.' '"help" for information about possible args', nargs=3, action='append') ''' 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) if self.args.comp: for comp in self.args.comp: pos, name, arg = comp realName = self.parseCompName(name) if not realName: print(name, 'is not a valid component name.') quit() modI = self.core.moduleIndexFor(realName) self.core.insertComponent(int(pos), modI, self) 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 def parseCompName(self, name): '''Deduces a proper component name out of a commandline arg''' compFileNames = [ \ os.path.splitext(os.path.basename( mod.__file__))[0] \ for mod in self.core.modules \ ] if name.title() in self.core.compNames: return name.title() for compName in self.core.compNames: if name.capitalize() in compName: return compName for i, compFileName in enumerate(compFileNames): if name.lower() in compFileName: return self.core.compNames[i] return return None