def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
self.setLayout(QtGui.QGridLayout())
self.setMinimumSize(320, 200)
tabWidget = QtGui.QTabWidget()
frameBox = QtGui.QWidget()
viewerBox = QtGui.QGroupBox('')
viewerBox.setMaximumSize(500, 150)
vLayout = QtGui.QVBoxLayout()
self.session = ftrackUtils.startASession()
basedir, inputfile = ftrackUtils.getInputFilePath(
os.environ['FTRACK_SHOTID'])
outfile = ''
infile = ''
if not inputfile == '':
outfile = ftrackUtils.getOutputFilePath(
os.environ['FTRACK_SHOTID'], os.environ['FTRACK_TASKID'],
basedir)
infile = os.path.join(basedir, inputfile)
self.inputWidget = FileBrowseWidget("Input Image File ", infile,
outfile)
self.inputWidget.addOpenFileDialogEvent()
self.outputWidget = FileBrowseWidget("Output Movie File", outfile,
outfile)
self.outputWidget.addSaveFileDialogEvent()
# Set trigger to change output path when input file is selected.
self.inputWidget.fileEdit.textChanged.connect(
self.outputWidget.setFilePath)
# Set trigger to change label when input file is selected.
self.inputWidget.fileEdit.textChanged.connect(self.setSlugLabel)
vLayout.addWidget(self.inputWidget)
vLayout.addWidget(self.outputWidget)
viewerBox.setLayout(vLayout)
frameLayout = QtGui.QVBoxLayout()
frameBox.setLayout(frameLayout)
frameLayout.addWidget(viewerBox)
self.movieWidget = MovieUploadWidget(
taskid=os.environ['FTRACK_TASKID'], session=self.session)
self.movieWidget.setMoviePath(str(self.outputWidget.getFilePath()))
self.outputWidget.fileEdit.textChanged.connect(
self.movieWidget.setMoviePath)
self.movieWidget.uploadComplete.connect(self.showUploadCompleteDialog)
# Setup the slug checkbox
hLayout = QtGui.QHBoxLayout()
self.slugBox = QtGui.QCheckBox('Slug')
hLayout.addWidget(self.slugBox)
self.slugBox.stateChanged.connect(self.showSlugOptions)
hLayout.addItem(
QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Minimum))
hLayout.addWidget(QtGui.QLabel('Frame Rate'))
self.frameDrop = QtGui.QComboBox()
self.frameDrop.addItems(['24', '25', '30'])
hLayout.addWidget(self.frameDrop)
hLayout.addItem(
QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Minimum))
vLayout.addLayout(hLayout)
vLayout.addItem(
QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Minimum,
QtGui.QSizePolicy.Expanding))
# Setup the slug options and set visibility to False.
self.slugFrameBox = QtGui.QGroupBox('Slug Options')
frameLayout.addWidget(self.slugFrameBox)
hslugLayout = QtGui.QGridLayout()
self.slugFrameBox.setLayout(hslugLayout)
hslugLayout.addWidget(QtGui.QLabel('Slug Label'), 0, 0)
self.slugTextBox = QtGui.QLineEdit('Customize Slug Label')
hslugLayout.addWidget(self.slugTextBox, 0, 1)
self.slugFrameBox.setVisible(False)
self.slugFrameBox.setMaximumSize(500, 150)
hLayout2 = QtGui.QHBoxLayout()
self.createButton = QtGui.QPushButton('Create Movie')
self.createButton.clicked.connect(self.createMovie)
hLayout2.addWidget(self.createButton)
self.openVideoButton = QtGui.QPushButton('Open Movie')
hLayout2.addWidget(self.openVideoButton)
self.openVideoButton.clicked.connect(self.openMovieFile)
hLayout2.addItem(
QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Minimum))
frameLayout.addLayout(hLayout2)
frameLayout.addItem(
QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Minimum,
QtGui.QSizePolicy.Expanding))
tabWidget.addTab(frameBox, 'ProRes Options')
tabWidget.addTab(self.movieWidget, 'Ftrack Upload Options')
self.layout().addWidget(tabWidget)
self.getFrameCount()
class NukeProResWindow(QtGui.QWidget):
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
self.setLayout(QtGui.QGridLayout())
self.setMinimumSize(320, 200)
tabWidget = QtGui.QTabWidget()
frameBox = QtGui.QWidget()
viewerBox = QtGui.QGroupBox('')
viewerBox.setMaximumSize(500, 150)
vLayout = QtGui.QVBoxLayout()
self.session = ftrackUtils.startASession()
basedir, inputfile = ftrackUtils.getInputFilePath(
os.environ['FTRACK_SHOTID'])
outfile = ''
infile = ''
if not inputfile == '':
outfile = ftrackUtils.getOutputFilePath(
os.environ['FTRACK_SHOTID'], os.environ['FTRACK_TASKID'],
basedir)
infile = os.path.join(basedir, inputfile)
self.inputWidget = FileBrowseWidget("Input Image File ", infile,
outfile)
self.inputWidget.addOpenFileDialogEvent()
self.outputWidget = FileBrowseWidget("Output Movie File", outfile,
outfile)
self.outputWidget.addSaveFileDialogEvent()
# Set trigger to change output path when input file is selected.
self.inputWidget.fileEdit.textChanged.connect(
self.outputWidget.setFilePath)
# Set trigger to change label when input file is selected.
self.inputWidget.fileEdit.textChanged.connect(self.setSlugLabel)
vLayout.addWidget(self.inputWidget)
vLayout.addWidget(self.outputWidget)
viewerBox.setLayout(vLayout)
frameLayout = QtGui.QVBoxLayout()
frameBox.setLayout(frameLayout)
frameLayout.addWidget(viewerBox)
self.movieWidget = MovieUploadWidget(
taskid=os.environ['FTRACK_TASKID'], session=self.session)
self.movieWidget.setMoviePath(str(self.outputWidget.getFilePath()))
self.outputWidget.fileEdit.textChanged.connect(
self.movieWidget.setMoviePath)
self.movieWidget.uploadComplete.connect(self.showUploadCompleteDialog)
# Setup the slug checkbox
hLayout = QtGui.QHBoxLayout()
self.slugBox = QtGui.QCheckBox('Slug')
hLayout.addWidget(self.slugBox)
self.slugBox.stateChanged.connect(self.showSlugOptions)
hLayout.addItem(
QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Minimum))
hLayout.addWidget(QtGui.QLabel('Frame Rate'))
self.frameDrop = QtGui.QComboBox()
self.frameDrop.addItems(['24', '25', '30'])
hLayout.addWidget(self.frameDrop)
hLayout.addItem(
QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Minimum))
vLayout.addLayout(hLayout)
vLayout.addItem(
QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Minimum,
QtGui.QSizePolicy.Expanding))
# Setup the slug options and set visibility to False.
self.slugFrameBox = QtGui.QGroupBox('Slug Options')
frameLayout.addWidget(self.slugFrameBox)
hslugLayout = QtGui.QGridLayout()
self.slugFrameBox.setLayout(hslugLayout)
hslugLayout.addWidget(QtGui.QLabel('Slug Label'), 0, 0)
self.slugTextBox = QtGui.QLineEdit('Customize Slug Label')
hslugLayout.addWidget(self.slugTextBox, 0, 1)
self.slugFrameBox.setVisible(False)
self.slugFrameBox.setMaximumSize(500, 150)
hLayout2 = QtGui.QHBoxLayout()
self.createButton = QtGui.QPushButton('Create Movie')
self.createButton.clicked.connect(self.createMovie)
hLayout2.addWidget(self.createButton)
self.openVideoButton = QtGui.QPushButton('Open Movie')
hLayout2.addWidget(self.openVideoButton)
self.openVideoButton.clicked.connect(self.openMovieFile)
hLayout2.addItem(
QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Minimum))
frameLayout.addLayout(hLayout2)
frameLayout.addItem(
QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Minimum,
QtGui.QSizePolicy.Expanding))
tabWidget.addTab(frameBox, 'ProRes Options')
tabWidget.addTab(self.movieWidget, 'Ftrack Upload Options')
self.layout().addWidget(tabWidget)
self.getFrameCount()
def getFrameCount(self):
infile = str(self.inputWidget.getFilePath())
if infile:
inputFolder = os.path.dirname(str(infile))
imageExt = str(infile).split('.')[-1]
shotName, firstFrame, lastFrame, date, firstFrameStr = utils.getShotInfo(
str(inputFolder), str(imageExt))
self.movieWidget.setFrameCount(firstFrame, lastFrame)
def showSlugOptions(self, state):
'''
Sets visibilty of slug options based on state of slug check box.
Resizes the window appropriately.
:param state: State of the slug check box.
'''
if state == 2:
self.slugFrameBox.setVisible(True)
infile = str(self.inputWidget.getFilePath())
if infile:
self.setSlugLabel(infile)
else:
self.slugFrameBox.setVisible(False)
def setSlugLabel(self, filename):
'''
Sets the slug label based on input file name.
:param filename: Name of the input file
'''
inputFolder = os.path.dirname(str(filename))
imageExt = str(filename).split('.')[-1]
if inputFolder:
shotName, firstFrame, lastFrame, date, firstFrameStr = utils.getShotInfo(
str(inputFolder), str(imageExt))
project = ftrackUtils.getProjectFromShot(
self.session, os.environ['FTRACK_SHOTID'])
label = '%s %s %s Frame#' % (project, date, shotName)
else:
label = 'Customize Slug Label'
self.slugTextBox.setText(label)
def createMovie(self):
frameRate = self.frameDrop.currentText()
self.movieWidget.setFrameRate(frameRate)
self.createButton.setDisabled(True)
inputFile = self.inputWidget.getFilePath()
outputFile = str(self.outputWidget.getFilePath())
slugChoice = self.slugBox.checkState()
if 'Select' in inputFile or 'Select' in outputFile or inputFile == '' or outputFile == '':
nuke.message("Please select input and output folder")
return
inputFolder = os.path.dirname(str(inputFile))
imageExt = str(inputFile).split('.')[-1]
if not outputFile.endswith('.mov'):
outputFile = '%s.mov' % outputFile
shotName, firstFrame, lastFrame, date, firstFrameStr = utils.getShotInfo(
inputFolder, imageExt)
self.movieWidget.setFrameCount(firstFrame, lastFrame)
if slugChoice == 2:
tmpDir = '%s/tmp' % os.environ['TEMP']
if not os.path.exists(tmpDir):
os.mkdir(tmpDir)
task = nuke.ProgressTask("Slug Creation")
task.setMessage("Creating slug files")
slugResult = utils.generateSlugImages(tmpDir,
self.slugTextBox.text(),
firstFrame, lastFrame, date,
firstFrameStr, task)
if slugResult != 0:
nuke.message("Error while creating slug images!")
self.createButton.setEnabled(True)
return
slugMovResult = utils.generateSlugMovie(tmpDir, firstFrame,
firstFrameStr, frameRate)
if slugMovResult != 0:
nuke.message("Error while creating slug movie!")
self.createButton.setEnabled(True)
return
finalMovCmd = utils.generateFileMovie(inputFolder, tmpDir,
outputFile, firstFrame,
shotName, imageExt,
lastFrame, firstFrameStr,
frameRate)
else:
finalMovCmd = utils.generateFileMovieNoSlug(
inputFolder, outputFile, firstFrame, shotName, imageExt,
lastFrame, firstFrameStr, frameRate)
threading.Thread(None, self.movieProgress, args=[finalMovCmd]).start()
#if os.path.exists('%s/imageSeq' % os.environ['TEMP']):
# shutil.rmtree('%s/imageSeq' % os.environ['TEMP'])
def movieProgress(self, finalMovCmd):
task = nuke.ProgressTask("Movie Conversion")
task.setMessage("Encoding files")
task.setProgress(0)
p = subprocess.Popen(finalMovCmd,
shell=True,
bufsize=64,
stderr=subprocess.PIPE)
self.updateProgressBar(p, task)
def updateProgressBar(self, process, task):
while True:
chatter = process.stderr.read(1024)
durationRes = re.search(r"Duration:\s(?P<duration>\S+)", chatter)
if durationRes:
durationList = durationRes.groupdict()['duration'][:-1].split(
':')
duration = int(durationList[0]) * 3600 + int(
durationList[1]) * 60 + float(durationList[2])
result = re.search(r'\stime=(?P<time>\S+)', chatter)
if result:
elapsed_time = result.groupdict()['time'].split(':')
secs = int(elapsed_time[0]) * 3600 + int(
elapsed_time[1]) * 60 + float(elapsed_time[2])
curValue = 10
outOf = 100 - curValue
progress = secs / duration * outOf
QtGui.QApplication.processEvents()
task.setProgress(int(progress + curValue))
if not chatter:
self.createButton.setEnabled(True)
break
def openMovieFile(self):
outfile = str(self.outputWidget.getFilePath())
if os.path.exists(outfile):
videoPlayerDir = utils.getVideoPlayer()
if not videoPlayerDir == '':
self.openVideoButton.setText('Opening Movie ...')
self.openVideoButton.setDisabled(True)
threading.Thread(None,
self.playMovie,
args=[outfile, videoPlayerDir]).start()
else:
nuke.message(
'Video player error: QuickTime or VLC not installed.')
else:
nuke.message('Movie does not exist. Cannot play the video.')
def showUploadCompleteDialog(self, txt):
nuke.message(txt)
def playMovie(self, movFile, videoPlayerDir):
cmd = '"%s" "%s"' % (videoPlayerDir, movFile)
args = shlex.split(cmd)
result = subprocess.call(args)
self.openVideoButton.setText('Open Movie')
self.openVideoButton.setEnabled(True)
return result
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
self.setLayout(QtGui.QGridLayout())
self.setMinimumSize(320,200)
tabWidget = QtGui.QTabWidget()
frameBox = QtGui.QWidget()
viewerBox = QtGui.QGroupBox('')
viewerBox.setMaximumSize(500, 150)
vLayout = QtGui.QVBoxLayout()
basedir, inputfile = ftrackUtils.getInputFilePath(os.environ['FTRACK_SHOTID'])
outfile = ''
infile = ''
if not inputfile == '':
outfile = ftrackUtils.getOutputFilePath(os.environ['FTRACK_SHOTID'], os.environ['FTRACK_TASKID'], basedir)
infile = os.path.join(basedir, inputfile)
self.inputWidget = FileBrowseWidget("Input Image File ", infile, outfile)
self.inputWidget.addOpenFileDialogEvent()
self.outputWidget = FileBrowseWidget("Output Movie File", outfile, outfile)
self.outputWidget.addSaveFileDialogEvent()
# Set trigger to change output path when input file is selected.
self.inputWidget.fileEdit.textChanged.connect(self.outputWidget.setFilePath)
# Set trigger to change label when input file is selected.
self.inputWidget.fileEdit.textChanged.connect(self.setSlugLabel)
vLayout.addWidget(self.inputWidget)
vLayout.addWidget(self.outputWidget)
viewerBox.setLayout(vLayout)
frameLayout = QtGui.QVBoxLayout()
frameBox.setLayout(frameLayout)
frameLayout.addWidget(viewerBox)
self.movieWidget = MovieUploadWidget(taskid=os.environ['FTRACK_TASKID'])
self.movieWidget.setMoviePath(str(self.outputWidget.getFilePath()))
self.outputWidget.fileEdit.textChanged.connect(self.movieWidget.setMoviePath)
self.movieWidget.uploadComplete.connect(self.showUploadCompleteDialog)
# Setup the slug checkbox
hLayout = QtGui.QHBoxLayout()
self.slugBox = QtGui.QCheckBox('Slug')
hLayout.addWidget(self.slugBox)
self.slugBox.stateChanged.connect(self.showSlugOptions)
hLayout.addItem(QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum))
hLayout.addWidget(QtGui.QLabel('Frame Rate'))
self.frameDrop = QtGui.QComboBox()
self.frameDrop.addItems(['24', '25', '30'])
hLayout.addWidget(self.frameDrop)
hLayout.addItem(QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum))
vLayout.addLayout(hLayout)
vLayout.addItem(QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Expanding))
# Setup the slug options and set visibility to False.
self.slugFrameBox = QtGui.QGroupBox('Slug Options')
frameLayout.addWidget(self.slugFrameBox)
hslugLayout = QtGui.QGridLayout()
self.slugFrameBox.setLayout(hslugLayout)
hslugLayout.addWidget(QtGui.QLabel('Slug Label'),0,0)
self.slugTextBox = QtGui.QLineEdit('Customize Slug Label')
hslugLayout.addWidget(self.slugTextBox,0,1)
self.slugFrameBox.setVisible(False)
self.slugFrameBox.setMaximumSize(500, 150)
hLayout2 = QtGui.QHBoxLayout()
self.createButton = QtGui.QPushButton('Create Movie')
self.createButton.clicked.connect(self.createMovie)
hLayout2.addWidget(self.createButton)
self.openVideoButton = QtGui.QPushButton('Open Movie')
hLayout2.addWidget(self.openVideoButton)
self.openVideoButton.clicked.connect(self.openMovieFile)
hLayout2.addItem(QtGui.QSpacerItem(10,10, QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum))
frameLayout.addLayout(hLayout2)
frameLayout.addItem(QtGui.QSpacerItem(10,10, QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Expanding))
tabWidget.addTab(frameBox, 'ProRes Options')
tabWidget.addTab(self.movieWidget, 'Ftrack Upload Options')
self.layout().addWidget(tabWidget)
self.getFrameCount()
class NukeProResWindow(QtGui.QWidget):
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
self.setLayout(QtGui.QGridLayout())
self.setMinimumSize(320,200)
tabWidget = QtGui.QTabWidget()
frameBox = QtGui.QWidget()
viewerBox = QtGui.QGroupBox('')
viewerBox.setMaximumSize(500, 150)
vLayout = QtGui.QVBoxLayout()
basedir, inputfile = ftrackUtils.getInputFilePath(os.environ['FTRACK_SHOTID'])
outfile = ''
infile = ''
if not inputfile == '':
outfile = ftrackUtils.getOutputFilePath(os.environ['FTRACK_SHOTID'], os.environ['FTRACK_TASKID'], basedir)
infile = os.path.join(basedir, inputfile)
self.inputWidget = FileBrowseWidget("Input Image File ", infile, outfile)
self.inputWidget.addOpenFileDialogEvent()
self.outputWidget = FileBrowseWidget("Output Movie File", outfile, outfile)
self.outputWidget.addSaveFileDialogEvent()
# Set trigger to change output path when input file is selected.
self.inputWidget.fileEdit.textChanged.connect(self.outputWidget.setFilePath)
# Set trigger to change label when input file is selected.
self.inputWidget.fileEdit.textChanged.connect(self.setSlugLabel)
vLayout.addWidget(self.inputWidget)
vLayout.addWidget(self.outputWidget)
viewerBox.setLayout(vLayout)
frameLayout = QtGui.QVBoxLayout()
frameBox.setLayout(frameLayout)
frameLayout.addWidget(viewerBox)
self.movieWidget = MovieUploadWidget(taskid=os.environ['FTRACK_TASKID'])
self.movieWidget.setMoviePath(str(self.outputWidget.getFilePath()))
self.outputWidget.fileEdit.textChanged.connect(self.movieWidget.setMoviePath)
self.movieWidget.uploadComplete.connect(self.showUploadCompleteDialog)
# Setup the slug checkbox
hLayout = QtGui.QHBoxLayout()
self.slugBox = QtGui.QCheckBox('Slug')
hLayout.addWidget(self.slugBox)
self.slugBox.stateChanged.connect(self.showSlugOptions)
hLayout.addItem(QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum))
hLayout.addWidget(QtGui.QLabel('Frame Rate'))
self.frameDrop = QtGui.QComboBox()
self.frameDrop.addItems(['24', '25', '30'])
hLayout.addWidget(self.frameDrop)
hLayout.addItem(QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum))
vLayout.addLayout(hLayout)
vLayout.addItem(QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Expanding))
# Setup the slug options and set visibility to False.
self.slugFrameBox = QtGui.QGroupBox('Slug Options')
frameLayout.addWidget(self.slugFrameBox)
hslugLayout = QtGui.QGridLayout()
self.slugFrameBox.setLayout(hslugLayout)
hslugLayout.addWidget(QtGui.QLabel('Slug Label'),0,0)
self.slugTextBox = QtGui.QLineEdit('Customize Slug Label')
hslugLayout.addWidget(self.slugTextBox,0,1)
self.slugFrameBox.setVisible(False)
self.slugFrameBox.setMaximumSize(500, 150)
hLayout2 = QtGui.QHBoxLayout()
self.createButton = QtGui.QPushButton('Create Movie')
self.createButton.clicked.connect(self.createMovie)
hLayout2.addWidget(self.createButton)
self.openVideoButton = QtGui.QPushButton('Open Movie')
hLayout2.addWidget(self.openVideoButton)
self.openVideoButton.clicked.connect(self.openMovieFile)
hLayout2.addItem(QtGui.QSpacerItem(10,10, QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum))
frameLayout.addLayout(hLayout2)
frameLayout.addItem(QtGui.QSpacerItem(10,10, QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Expanding))
tabWidget.addTab(frameBox, 'ProRes Options')
tabWidget.addTab(self.movieWidget, 'Ftrack Upload Options')
self.layout().addWidget(tabWidget)
self.getFrameCount()
def getFrameCount(self):
infile = str(self.inputWidget.getFilePath())
if infile:
inputFolder = os.path.dirname(str(infile))
imageExt = str(infile).split('.')[-1]
shotName, firstFrame, lastFrame, date, firstFrameStr = utils.getShotInfo(str(inputFolder), str(imageExt))
self.movieWidget.setFrameCount(firstFrame, lastFrame)
def showSlugOptions(self, state):
'''
Sets visibilty of slug options based on state of slug check box.
Resizes the window appropriately.
:param state: State of the slug check box.
'''
if state == 2:
self.slugFrameBox.setVisible(True)
infile = str(self.inputWidget.getFilePath())
if infile:
self.setSlugLabel(infile)
else:
self.slugFrameBox.setVisible(False)
def setSlugLabel(self, filename):
'''
Sets the slug label based on input file name.
:param filename: Name of the input file
'''
inputFolder = os.path.dirname(str(filename))
imageExt = str(filename).split('.')[-1]
if inputFolder:
shotName, firstFrame,lastFrame, date, firstFrameStr = utils.getShotInfo(str(inputFolder), str(imageExt))
project = ftrackUtils.getProjectFromShot(os.environ['FTRACK_SHOTID'])
label = '%s %s %s Frame#' % (project, date, shotName)
else:
label = 'Customize Slug Label'
self.slugTextBox.setText(label)
def createMovie(self):
frameRate = self.frameDrop.currentText()
self.movieWidget.setFrameRate(frameRate)
self.createButton.setDisabled(True)
inputFile = self.inputWidget.getFilePath()
outputFile = str(self.outputWidget.getFilePath())
slugChoice = self.slugBox.checkState()
if 'Select' in inputFile or 'Select' in outputFile or inputFile == '' or outputFile == '':
nuke.message("Please select input and output folder")
return
inputFolder = os.path.dirname(str(inputFile))
imageExt = str(inputFile).split('.')[-1]
if not outputFile.endswith('.mov'):
outputFile = '%s.mov' % outputFile
shotName, firstFrame, lastFrame, date, firstFrameStr = utils.getShotInfo(inputFolder, imageExt)
self.movieWidget.setFrameCount(firstFrame, lastFrame)
if slugChoice == 2:
tmpDir = '%s\\tmp' % os.environ['TEMP']
if not os.path.exists(tmpDir):
os.mkdir(tmpDir)
task = nuke.ProgressTask("Slug Creation")
task.setMessage("Creating slug files")
slugResult = utils.generateSlugImages(tmpDir, self.slugTextBox.text(), firstFrame,
lastFrame, date, firstFrameStr, task)
if slugResult != 0:
nuke.message("Error while creating slug images!")
self.createButton.setEnabled(True)
return
slugMovResult = utils.generateSlugMovie(tmpDir, firstFrame, firstFrameStr, frameRate)
if slugMovResult != 0:
nuke.message("Error while creating slug movie!")
self.createButton.setEnabled(True)
return
finalMovCmd = utils.generateFileMovie(inputFolder, tmpDir, outputFile, firstFrame,
shotName, imageExt, lastFrame, firstFrameStr, frameRate)
else:
finalMovCmd = utils.generateFileMovieNoSlug(inputFolder, outputFile, firstFrame,
shotName, imageExt, lastFrame, firstFrameStr, frameRate)
threading.Thread( None, self.movieProgress, args=[finalMovCmd]).start()
#if os.path.exists('%s/imageSeq' % os.environ['TEMP']):
# shutil.rmtree('%s/imageSeq' % os.environ['TEMP'])
def movieProgress(self, finalMovCmd):
task = nuke.ProgressTask("Movie Conversion")
task.setMessage("Encoding files")
task.setProgress(0)
p = subprocess.Popen(finalMovCmd, shell=True, bufsize=64, stderr=subprocess.PIPE)
self.updateProgressBar(p, task)
def updateProgressBar(self, process, task):
while True:
chatter = process.stderr.read(1024)
durationRes = re.search(r"Duration:\s(?P<duration>\S+)", chatter)
if durationRes:
durationList = durationRes.groupdict()['duration'][:-1].split(':')
duration = int(durationList[0])*3600 + int(durationList[1])*60 + float(durationList[2])
result = re.search(r'\stime=(?P<time>\S+)', chatter)
if result:
elapsed_time = result.groupdict()['time'].split(':')
secs = int(elapsed_time[0])*3600 + int(elapsed_time[1])*60 + float(elapsed_time[2])
curValue = 10
outOf = 100-curValue
progress = secs/duration * outOf
QtGui.QApplication.processEvents()
task.setProgress(int(progress+curValue))
if not chatter:
self.createButton.setEnabled(True)
break
def openMovieFile(self):
outfile = str(self.outputWidget.getFilePath())
outfile = outfile.replace('/', '\\')
if os.path.exists(outfile):
videoPlayerDir = utils.getVideoPlayer()
if not videoPlayerDir == '':
self.openVideoButton.setText('Opening Movie ...')
self.openVideoButton.setDisabled(True)
threading.Thread(None, self.playMovie, args=[outfile, videoPlayerDir]).start()
else:
nuke.message('Video player error: QuickTime or VLC not installed.')
else:
nuke.message('Movie does not exist. Cannot play the video.')
def showUploadCompleteDialog(self, txt):
nuke.message(txt)
def playMovie(self, movFile, videoPlayerDir):
cmd = '"%s" "%s"' % (videoPlayerDir, movFile)
args = shlex.split(cmd)
result = subprocess.call(args, shell=True)
self.openVideoButton.setText('Open Movie')
self.openVideoButton.setEnabled(True)
return result
def setupUI(self):
# Setup the input and output file widgets
viewerBox = QtGui.QGroupBox('File Options')
vLayout = QtGui.QVBoxLayout()
self.inputWidget = FileBrowseWidget("Input Image File ")
self.inputWidget.addOpenFileDialogEvent()
self.outputWidget = FileBrowseWidget("Output Movie File")
self.outputWidget.addSaveFileDialogEvent()
# Set trigger to change output path when input file is selected.
self.inputWidget.fileEdit.textChanged.connect(self.outputWidget.setFilePath)
# Set trigger to change label when input file is selected.
self.inputWidget.fileEdit.textChanged.connect(self.setSlugLabel)
vLayout.addWidget(self.inputWidget)
vLayout.addWidget(self.outputWidget)
viewerBox.setLayout(vLayout)
self.centralLayout.addWidget(viewerBox)
hLayout = QtGui.QHBoxLayout()
# Setup the slug checkbox
self.slugBox = QtGui.QCheckBox()
hLayout.addItem(QtGui.QSpacerItem(10,10, QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Minimum))
hLayout.addWidget(QtGui.QLabel('Slug'))
hLayout.addWidget(self.slugBox)
self.slugBox.stateChanged.connect(self.showSlugOptions)
hLayout.addItem(QtGui.QSpacerItem(10,10, QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Fixed))
# Setup the movie checkbox
hLayout.addWidget(QtGui.QLabel('Open Output Movie'))
self.movBox = QtGui.QCheckBox()
hLayout.addWidget(self.movBox)
hLayout.addItem(QtGui.QSpacerItem(20,10, QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Fixed))
# Setup batch options
hLayout.addWidget(QtGui.QLabel("Batch"))
self.batchBox = QtGui.QCheckBox()
hLayout.addWidget(self.batchBox)
self.batchBox.stateChanged.connect(self.showBatchOptions)
vLayout.addLayout(hLayout)
vLayout.addItem(QtGui.QSpacerItem(10,1, QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Expanding))
# Setup the slug options and set visibility to False.
self.slugFrameBox = QtGui.QGroupBox('Slug Options')
self.centralLayout.addWidget(self.slugFrameBox,1,0)
hslugLayout = QtGui.QGridLayout()
self.slugFrameBox.setLayout(hslugLayout)
hslugLayout.addWidget(QtGui.QLabel('Slug Label'),0,0)
self.globalLabelName = 'Customize Slug Label'
self.slugTextBox = QtGui.QLineEdit(self.globalLabelName)
hslugLayout.addWidget(self.slugTextBox,0,1)
self.slugFrameBox.setVisible(False)
# Setup Batch Widget
self.batchWidget = BatchRunWidget()
self.centralLayout.addWidget(self.batchWidget, 0,1)
self.batchWidget.setVisible(False)
createButton = QtGui.QPushButton('Create Movie')
createButton.clicked.connect(self.createMovie)
self.centralLayout.addWidget(createButton,2,0)
# Setup the progress bar and set visible to False
self.pLabel = QtGui.QLabel('')
self.pBar = QtGui.QProgressBar()
self.pBar.setVisible(False)
self.pLabel.setVisible(False)
self.centralLayout.addWidget(self.pLabel,3,0)
self.centralLayout.addWidget(self.pBar,4,0)
class Q_ProresGui(QtGui.QMainWindow):
'''
Main Application Class.
'''
def __init__(self):
'''
Basic UI setup.
'''
super(Q_ProresGui, self).__init__()
self.setWindowTitle('Loco VFX - QX Tools 2015 v2.5')
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Minimum)
self.setSizePolicy(sizePolicy)
self.setMinimumSize(320,200)
window = QtGui.QWidget()
from style import pyqt_style_rc # Import the Qt Style Sheet
f = QtCore.QFile('style/style.qss')
f.open(QtCore.QFile.ReadOnly | QtCore.QFile.Text)
ts = QtCore.QTextStream(f)
self.stylesheet = ts.readAll()
window.setStyleSheet(self.stylesheet)
self.centralLayout = QtGui.QGridLayout()
window.setLayout(self.centralLayout)
self.setCentralWidget(window)
self.setupUI()
def setupUI(self):
# Setup the input and output file widgets
viewerBox = QtGui.QGroupBox('File Options')
vLayout = QtGui.QVBoxLayout()
self.inputWidget = FileBrowseWidget("Input Image File ")
self.inputWidget.addOpenFileDialogEvent()
self.outputWidget = FileBrowseWidget("Output Movie File")
self.outputWidget.addSaveFileDialogEvent()
# Set trigger to change output path when input file is selected.
self.inputWidget.fileEdit.textChanged.connect(self.outputWidget.setFilePath)
# Set trigger to change label when input file is selected.
self.inputWidget.fileEdit.textChanged.connect(self.setSlugLabel)
vLayout.addWidget(self.inputWidget)
vLayout.addWidget(self.outputWidget)
viewerBox.setLayout(vLayout)
self.centralLayout.addWidget(viewerBox)
hLayout = QtGui.QHBoxLayout()
# Setup the slug checkbox
self.slugBox = QtGui.QCheckBox()
hLayout.addItem(QtGui.QSpacerItem(10,10, QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Minimum))
hLayout.addWidget(QtGui.QLabel('Slug'))
hLayout.addWidget(self.slugBox)
self.slugBox.stateChanged.connect(self.showSlugOptions)
hLayout.addItem(QtGui.QSpacerItem(10,10, QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Fixed))
# Setup the movie checkbox
hLayout.addWidget(QtGui.QLabel('Open Output Movie'))
self.movBox = QtGui.QCheckBox()
hLayout.addWidget(self.movBox)
hLayout.addItem(QtGui.QSpacerItem(20,10, QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Fixed))
# Setup batch options
hLayout.addWidget(QtGui.QLabel("Batch"))
self.batchBox = QtGui.QCheckBox()
hLayout.addWidget(self.batchBox)
self.batchBox.stateChanged.connect(self.showBatchOptions)
vLayout.addLayout(hLayout)
vLayout.addItem(QtGui.QSpacerItem(10,1, QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Expanding))
# Setup the slug options and set visibility to False.
self.slugFrameBox = QtGui.QGroupBox('Slug Options')
self.centralLayout.addWidget(self.slugFrameBox,1,0)
hslugLayout = QtGui.QGridLayout()
self.slugFrameBox.setLayout(hslugLayout)
hslugLayout.addWidget(QtGui.QLabel('Slug Label'),0,0)
self.globalLabelName = 'Customize Slug Label'
self.slugTextBox = QtGui.QLineEdit(self.globalLabelName)
hslugLayout.addWidget(self.slugTextBox,0,1)
self.slugFrameBox.setVisible(False)
# Setup Batch Widget
self.batchWidget = BatchRunWidget()
self.centralLayout.addWidget(self.batchWidget, 0,1)
self.batchWidget.setVisible(False)
createButton = QtGui.QPushButton('Create Movie')
createButton.clicked.connect(self.createMovie)
self.centralLayout.addWidget(createButton,2,0)
# Setup the progress bar and set visible to False
self.pLabel = QtGui.QLabel('')
self.pBar = QtGui.QProgressBar()
self.pBar.setVisible(False)
self.pLabel.setVisible(False)
self.centralLayout.addWidget(self.pLabel,3,0)
self.centralLayout.addWidget(self.pBar,4,0)
def setSlugLabel(self, filename):
'''
Sets the slug label based on input file name.
:param filename: Name of the input file
'''
inputFolder = os.path.dirname(str(filename))
imageExt = str(filename).split('.')[-1]
if imageExt.lower() == 'avi' or imageExt.lower() == 'mov':
inputFolder = '%s/imageSeq' % os.environ['TEMP']
result = self.extractJpegToTmp(filename, inputFolder)
if result != 0:
self.setStyleSheet(self.stylesheet)
QtGui.QMessageBox.warning(self, "Error", "Error while extracting images from file!")
shotName = filename.split('/')[-1].split('.')[0]
d = datetime.now()
date = '%s/%s/%s' % (d.day, d.month, d.year)
else:
shotName, firstFrame,lastFrame, date, firstFrameStr = self.getShotInfo(str(inputFolder), str(imageExt))
label = 'Quarks %s %s Frame#' % (date, shotName)
self.slugTextBox.setText(label)
self.globalLabelName = label
def extractJpegToTmp(self, filename, tmpFolder):
if not os.path.exists(tmpFolder):
os.mkdir(tmpFolder)
cmd = 'ffmpeg.exe -i %s -f image2 "%s/temp.1%%3d.jpeg"' % (filename, tmpFolder)
args = shlex.split(cmd)
result = subprocess.call(args, shell=True)
return result
def showSlugOptions(self, state):
'''
Sets visibilty of slug options based on state of slug check box.
Resizes the window appropriately.
:param state: State of the slug check box.
'''
if state == 2:
self.slugFrameBox.setVisible(True)
self.resize(self.sizeHint())
if self.batchBox.checkState() == 2:
d = datetime.now()
date = '%s/%s/%s' % (d.day, d.month, d.year)
self.slugTextBox.setText('Quarks %s ShotName Frame#' % date)
else:
self.slugFrameBox.setVisible(False)
self.resize(self.sizeHint())
self.slugTextBox.setText(self.globalLabelName)
def showBatchOptions(self, state):
'''
Sets visibilty of the batch options based on state of checkbox
:param state: State of batch checkbox
'''
if state == 2:
self.batchWidget.setVisible(True)
self.resize(self.sizeHint())
if self.slugBox.checkState() == 2:
d = datetime.now()
date = '%s/%s/%s' % (d.day, d.month, d.year)
self.slugTextBox.setText('Quarks %s ShotName Frame#' % date)
else:
self.batchWidget.setVisible(False)
self.resize(self.sizeHint())
self.slugTextBox.setText(self.globalLabelName)
def createMovie(self, event):
'''
:param event: This event is triggered when the "Create Movie" button is pressed.
:return:
'''
self.pBar.setVisible(True)
self.pLabel.setVisible(True)
self.resize(self.sizeHint())
self.pBar.setValue(0)
self.pBar.setMinimum(0)
self.pBar.setMaximum(100)
if self.batchBox.checkState() == 2:
self.createMovieBatch()
else:
self.createMovieNoBatch()
def progressBarAnimationComplete(self):
self.pLabel.setText('Encoding complete.')
self.setStyleSheet(self.stylesheet)
QtGui.QMessageBox.about(self, "Complete", "Conversion Complete")
if self.movBox.checkState() == 2:
self.openOutputMovie(str(self.outputWidget.getFilePath()))
def createMovieBatch(self):
inFolder, outputFolder = self.batchWidget.getFilePath()
slugChoice = self.slugBox.checkState()
if 'Select' in inFolder or inFolder is '' \
or 'Select' in outputFolder or outputFolder is '':
self.setStyleSheet(self.stylesheet)
QtGui.QMessageBox.warning(self, "Warning", "Please select input and output folder")
return
dirList = self.batchWidget.getAllCheckedItems()
self.resultDict = {}
total = len(dirList)
count = 0
for dir in dirList:
count = count + 1
if dir == './':
inputFolder = inFolder
else:
inputFolder = '%s/%s' % (inFolder, dir)
imageExt = self.batchWidget.getImageExt(inputFolder)
shotName, firstFrame,lastFrame, date, firstFrameStr = self.getShotInfo(inputFolder, imageExt)
outputFile = '%s/%s.mov' % (outputFolder, shotName)
self.pLabel.setText('Making movie %s.mov' % shotName )
QtGui.QApplication.processEvents()
if slugChoice == 2:
self.batchSlugMovie(inputFolder, outputFile, firstFrame,
shotName, imageExt, lastFrame, date, firstFrameStr)
else:
self.batchNoSlugMovie(inputFolder, outputFile, firstFrame,
shotName, imageExt, lastFrame, firstFrameStr)
percentage = count * 100 / total
self.pBar.setValue(percentage)
QtGui.QApplication.processEvents()
self.processResult()
def processResult(self):
successList = []
failList = []
for key in self.resultDict.keys():
shot, ext = key
if self.resultDict[key] == 0:
successList.append(shot)
else:
failList.append(shot)
if len(failList) == 0:
self.setStyleSheet(self.stylesheet)
QtGui.QMessageBox.about(self, "Complete", "Conversion Complete!")
else:
self.setStyleSheet(self.stylesheet)
listStr = ''
for file in failList:
listStr = '%s %s \n' % (listStr, file)
QtGui.QMessageBox.warning(self, "Error", "Conversion complete for all shots except: %s" % listStr)
def batchSlugMovie(self, inputFolder, outputFile, firstFrame, shotName, imageExt, lastFrame, date, firstFrameStr):
tmpDir = '%s\\tmp_%s' % (os.environ['TEMP'], shotName)
if not os.path.exists(tmpDir):
os.mkdir(tmpDir)
slugResult = self.generateSlugImages(tmpDir, shotName, firstFrame,lastFrame, date, firstFrameStr)
if slugResult == 0:
slugMovResult = self.generateSlugMovie(tmpDir, firstFrame, firstFrameStr)
if slugMovResult == 0:
result = self.generateFileMovie(inputFolder, tmpDir, outputFile, firstFrame,
shotName, imageExt, lastFrame, firstFrameStr)
self.resultDict[shotName, imageExt] = result
else:
self.resultDict[shotName, imageExt] = slugMovResult
else:
self.resultDict[shotName, imageExt] = slugResult
def batchNoSlugMovie(self, inputFolder, outputFile, firstFrame, shotName, imageExt, lastFrame, firstFrameStr):
self.pLabel.setText('Making movie %s.mov' % shotName)
result = self.generateFileMovieNoSlug(inputFolder, outputFile, firstFrame,
shotName, imageExt, lastFrame, firstFrameStr)
self.resultDict[shotName, imageExt] = result
def createMovieNoBatch(self):
inputFile = self.inputWidget.getFilePath()
outputFile = str(self.outputWidget.getFilePath())
slugChoice = self.slugBox.checkState()
# Check if file names are valid.
if 'Select' in inputFile or 'Select' in outputFile:
self.setStyleSheet(self.stylesheet)
QtGui.QMessageBox.warning(self, "Warning", "Please select input and output folder")
return
inputFolder = os.path.dirname(str(inputFile))
imageExt = str(inputFile).split('.')[-1]
if not outputFile.endswith('.mov'):
outputFile = '%s.mov' % outputFile
if imageExt.lower() == 'avi' or imageExt.lower() == 'mov':
inputFolder = '%s/imageSeq' % os.environ['TEMP']
imageExt = 'jpeg'
shotName, firstFrame, lastFrame, date, firstFrameStr = self.getShotInfo(inputFolder, imageExt)
if slugChoice == 2:
self.pLabel.setText('Creating Slug...')
tmpDir = '%s\\tmp' % os.environ['TEMP']
if not os.path.exists(tmpDir):
os.mkdir(tmpDir)
slugResult = self.generateSlugImages(tmpDir, shotName, firstFrame,lastFrame, date, firstFrameStr)
if slugResult != 0:
self.setStyleSheet(self.stylesheet)
QtGui.QMessageBox.warning(self, "Error", "Error while creating slug images!")
return
slugMovResult = self.generateSlugMovie(tmpDir, firstFrame, firstFrameStr)
if slugMovResult != 0:
self.setStyleSheet(self.stylesheet)
QtGui.QMessageBox.warning(self, "Error", "Error while creating slug movie!")
return
self.pLabel.setText('Encoding movie...')
result = self.generateFileMovie(inputFolder, tmpDir, outputFile, firstFrame,
shotName, imageExt, lastFrame, firstFrameStr)
if result != 0:
self.setStyleSheet(self.stylesheet)
QtGui.QMessageBox.warning(self, "Error", "Error during final movie conversion!")
return
self.pBar.setValue(100)
self.progressBarAnimationComplete()
shutil.rmtree(tmpDir)
else:
self.pLabel.setText('Encoding Movie...')
result = self.generateFileMovieNoSlug(inputFolder, outputFile, firstFrame,
shotName, imageExt, lastFrame, firstFrameStr)
if result == 0:
self.pLabel.setText('Encoding Complete')
self.pBar.setValue(100)
self.setStyleSheet(self.stylesheet)
QtGui.QMessageBox.about(self, "Complete", "Conversion Complete")
if self.movBox.checkState() == 2:
self.openOutputMovie(outputFile)
else:
self.setStyleSheet(self.stylesheet)
QtGui.QMessageBox.warning(self, "Error", "Conversion Error!")
if os.path.exists('%s/imageSeq' % os.environ['TEMP']):
shutil.rmtree('%s/imageSeq' % os.environ['TEMP'])
def openOutputMovie(self, outputFile):
'''
Opens the output movie file with the installed player using autoit
:param outputFile: Output movie file
'''
videoPlayerDir = self.getVideoPlayer()
if videoPlayerDir == '':
return
self.pLabel.setText('Opening Movie %s' % outputFile.split('/')[-1])
title = outputFile.split('/')[-1].split('.')[0]
outputFile = outputFile.replace('/','\\')
autoit.run('%s %s' % (videoPlayerDir, outputFile))
if 'QuickTime' in videoPlayerDir:
autoit.win_wait(title, 100)
import time
# This is a hack. We wait 3 sec for the movie to load completely before sending the next signal.
# Otherwise the signal is not registered.
time.sleep(1)
autoit.control_send(title, '', '{CTRLDOWN}0{CTRLUP}')
def getVideoPlayer(self):
'''
Checks if QuickTimePlayer exists. If not checks for VLC player.
:return: videoPlayerDir: Path of the video player
'''
videoPlayerDir = ''
videoPlayerDirList = glob.glob('C:\\Program*\\QuickTime*')
if videoPlayerDirList:
videoPlayerDir = '%s\\QuickTimePlayer.exe' % videoPlayerDirList[0]
else:
videoPlayerDirList = glob.glob('C:\Program*\\VideoLan*')
if videoPlayerDirList:
videoPlayerDir = '%s\\VLC\\vlc.exe' % videoPlayerDirList[0]
else:
self.setStyleSheet(self.stylesheet)
QtGui.QMessageBox.warning(self, "Video Player Error", "QuickTime or VLC not installed.")
return videoPlayerDir
def generateSlugImages(self, tmpDir, shotName, firstFrame, lastFrame, date, firstFrameStr):
'''
Slug Images are generated and stored in tmpDir
:param tmpDir: Temporary Directory in the Users local temp
:param shotName: Name of the shot type:str
:param firstFrame: First frame type:int
:param lastFrame: Last frame type: int
:param date: Date mm/dd/yyyy type:str
:return:
'''
slugCommand = 'convert.exe -size 450x40 -background black -fill white -pointsize 20 ' \
'label:"quarks %s ball frames:10" %s\\slug.jpg' % (date,tmpDir)
args = shlex.split(slugCommand)
result = []
label = str(self.slugTextBox.text())
label = label.replace('Frame#', '')
if self.batchBox.checkState() == 2:
label = 'Quarks %s %s' % (date, shotName)
totalFrames = lastFrame - firstFrame
incrValue = 40.0/totalFrames
count = self.pBar.value()
for i in range(firstFrame, lastFrame+1):
frameStr = firstFrameStr[:-(len(str(i)))] + str(i)
args[-1] = '%s\\slug.%s.jpg' % (tmpDir, frameStr)
args[-2] = 'label:%s %s' % (label, str(frameStr))
result.append(subprocess.call(args, shell=True))
count = count + incrValue
if self.batchBox.checkState() != 2:
self.pBar.setValue(count)
for i in result:
if i != 0:
return 1
return 0
def getShotInfo(self, inputFolder, imageExt):
'''
Returns shot information
:param inputFolder: Input Folder
:param imageExt: Image extension
:return: shotName, first frame, last frame and date
'''
date = datetime.now()
dateStr = '%s/%s/%s' % (date.day, date.month, date.year)
files = [file for file in os.listdir(inputFolder) if file.endswith(imageExt)]
files.sort()
if files:
shotName = files[0].split('.')[0]
firstFrame = files[0].split('.')[1]
lastFrame = files[-1].split('.')[1]
return shotName, int(firstFrame), int(lastFrame), dateStr, firstFrame
else:
return '',0,0,dateStr
def generateSlugMovie(self, tmpDir, firstFrame, firstFrameStr):
'''
Generates a movie of the slug images. Stores it in the same temp folder
:param tmpDir: Temp Folder in the users local temp.
:param firstFrame: first frame
:return:
'''
if len(str(firstFrame)) == 1:
frameLen = len(str(firstFrameStr))
slugMovCmd = 'ffmpeg.exe -y -an -i "%s\\slug.%%0%sd.jpg" ' \
'-vcodec prores -profile:v 2 "%s\\slug.mov"' % (tmpDir, frameLen, tmpDir)
else:
frameLen = len(str(firstFrameStr))-1
slugMovCmd = 'ffmpeg.exe -y -an -i "%s\\slug.1%%0%sd.jpg" ' \
'-vcodec prores -profile:v 2 "%s\\slug.mov"' % (tmpDir, frameLen, tmpDir)
args = shlex.split(slugMovCmd)
result = subprocess.call(args, shell=True)
return result
def generateFileMovie(self, inputFolder, tmpDir, outputFile, firstFrame, fileName, imageExt, lastFrame, firstFrameStr):
'''
Composites the slug movie with the input images to generate the final movie.
'''
if imageExt == 'exr':
# Convert exr to exr using imagemagik to get the exr format correct.
self.convertExr(inputFolder, fileName, firstFrame, lastFrame, firstFrameStr)
filePath = '%s\\exrTmp\\%s' % (os.environ['TEMP'], fileName)
else:
filePath = '%s\\%s' % (inputFolder, fileName)
inputFile = '%s.%s.%s' % (fileName, firstFrame, imageExt)
if len(str(firstFrame)) == 1:
frameLen = len(str(firstFrameStr))
finalMovCmd = 'ffmpeg.exe -y -an -i "%s.%%0%sd.%s" ' \
'-i "%s\\slug.mov" -metadata comment="Source Image:%s" -filter_complex "overlay=1:1" ' \
'-vcodec prores -profile:v 2 "%s" ' % (filePath, frameLen, imageExt,
tmpDir, inputFile, outputFile)
else:
frameLen = len(str(firstFrameStr))-1
finalMovCmd = 'ffmpeg.exe -y -an -i "%s.1%%0%sd.%s" ' \
'-i "%s\\slug.mov" -metadata comment="Source Image:%s" -filter_complex "overlay=1:1" ' \
'-vcodec prores -profile:v 2 "%s" ' % (filePath, frameLen, imageExt,
tmpDir, inputFile, outputFile)
args = shlex.split(finalMovCmd)
try:
p = subprocess.Popen(finalMovCmd, shell=True, bufsize=64, stderr=subprocess.PIPE)
self.updateProgressBar(p)
result = 0
except:
result = 1
if imageExt == '.exr':
shutil.rmtree('%s/exrTmp' % inputFolder)
return result
def generateFileMovieNoSlug(self, inputFolder, outputFile, firstFrame, fileName, imageExt, lastFrame, firstFrameStr):
'''
Generate the movie without the slug, only from the input image sequence.
'''
if imageExt == 'exr':
# Convert exr to correct format using imagemagik
self.convertExr(inputFolder, fileName, firstFrame, lastFrame, firstFrameStr)
filePath = '%s\\exrTmp\\%s' % (os.environ['TEMP'], fileName)
else:
filePath = '%s\\%s' % (inputFolder, fileName)
if len(str(firstFrame)) == 1:
frameLen = len(str(firstFrameStr))
finalMovCmd = 'ffmpeg.exe -y -an -i "%s.%%0%sd.%s" ' \
'-metadata comment="Source Image:%s.%s.%s" -vcodec prores ' \
'-profile:v 2 "%s" ' % (filePath, frameLen, imageExt, fileName,
firstFrame,imageExt, outputFile)
else:
frameLen = len(str(firstFrameStr))-1
finalMovCmd = 'ffmpeg.exe -y -an -i "%s.1%%0%sd.%s" ' \
'-metadata comment="Source Image:%s.%s.%s" -vcodec prores ' \
'-profile:v 2 "%s" ' % (filePath, frameLen, imageExt, fileName,
firstFrame,imageExt, outputFile)
args = shlex.split(finalMovCmd)
try:
p = subprocess.Popen(finalMovCmd, shell=True, bufsize=64, stderr=subprocess.PIPE)
self.updateProgressBar(p)
result = 0
except:
result = 1
if imageExt == '.exr':
shutil.rmtree('%s/exrTmp' % inputFolder)
return result
def convertExr(self, inputFolder, fileName, firstFrame, lastFrame, firstFrameStr):
'''
Generate new exr from input exr images using ImageMagik.
This was required as the compression type of the input exr images was not supported.
'''
if not os.path.exists('%s/exrTmp' % os.environ['TEMP']):
os.mkdir('%s/exrTmp' % os.environ['TEMP'])
slugCommand = 'convert.exe %s\\%s.exr "%s\\exrTmp\\%s.exr"' % (inputFolder,fileName,os.environ['TEMP'],fileName)
args = shlex.split(slugCommand)
totalFrames = lastFrame-firstFrame
incrValue = 50.0/totalFrames
count = self.pBar.value()
for i in range(firstFrame, lastFrame+1):
frameStr = firstFrameStr[:-(len(str(i)))] + str(i)
args[1] = '%s/%s.%s.exr' % (inputFolder, fileName, frameStr)
args[2] = '%s/exrTmp/%s.%s.exr' % (os.environ['TEMP'], fileName, frameStr)
subprocess.call(args, shell=True)
count = count + incrValue
if self.batchBox.checkState() != 2:
self.pBar.setValue(count)
def updateProgressBar(self, process):
while True:
chatter = process.stderr.read(1024)
durationRes = re.search(r"Duration:\s(?P<duration>\S+)", chatter)
if durationRes:
durationList = durationRes.groupdict()['duration'][:-1].split(':')
duration = int(durationList[0])*3600 + int(durationList[1])*60 + float(durationList[2])
result = re.search(r'\stime=(?P<time>\S+)', chatter)
if result:
elapsed_time = result.groupdict()['time'].split(':')
secs = int(elapsed_time[0])*3600 + int(elapsed_time[1])*60 + float(elapsed_time[2])
curValue = self.pBar.value()
outOf = 100-curValue
progress = secs/duration * outOf
QtGui.QApplication.processEvents()
self.pBar.setValue(int(progress+curValue))
if not chatter:
break
