def get_frame_per_file_render_arguments(self):
"""
Generates the render arguments for when FramePerFile mode is set to True.
This means that each output frame has an associated scene file with the same frame number.
This is mostly used in Octane 1 before .orbx was a thing.
:return: a list of commandline arguments, and the scene file
"""
render_args = ['-e', '-q']
sceneFile = self.GetPluginInfoEntry("SceneFile")
sceneFile = PathUtils.ToPlatformIndependentPath(
RepositoryUtils.CheckPathMapping(sceneFile))
outputFile = self.CreateOutputFile()
paddingSize = 0
if not self.GetBooleanPluginInfoEntryWithDefault("SingleFile", True):
currPadding = FrameUtils.GetFrameStringFromFilename(sceneFile)
paddingSize = len(currPadding)
if paddingSize > 0:
newPadding = StringUtils.ToZeroPaddedString(
self.GetStartFrame(), paddingSize, False)
sceneFile = FrameUtils.SubstituteFrameNumber(
sceneFile, newPadding)
# Build the new output file name.
if outputFile:
outputFile = PathUtils.ToPlatformIndependentPath(
RepositoryUtils.CheckPathMapping(outputFile))
# Add padding to output file if necessary.
if paddingSize > 0:
outputFile = FrameUtils.SubstituteFrameNumber(
outputFile, newPadding)
outputPath = os.path.dirname(outputFile)
outputFileName, outputExtension = os.path.splitext(outputFile)
outputFile = os.path.join(
outputPath, outputFileName + newPadding + outputExtension)
render_args.extend(['-o', outputFile])
sample = self.GetIntegerPluginInfoEntryWithDefault(
"OverrideSampling", 0)
if sample > 0:
render_args.extend(['-s', str(sample)])
render_target = self.GetPluginInfoEntryWithDefault(
"RenderTargetOCS", "")
if render_target:
render_args.extend(['-t', render_target])
return render_args, sceneFile
def isValidSubmission():
"""
Runs sanity checks on the current submission params to see if we know the submission is bad, so we can fail
fast. If there's warnings and/or errors, we present them to the user so they can be fixed.
:return: a boolean indicating if we believe this submission is valid
"""
global scriptDialog
errors = ""
warnings = ""
# Check if SU file exists
sceneFile = scriptDialog.GetValue( "SceneBox" )
if not os.path.isfile( sceneFile ):
errors += 'SketchUp file "%s" does not exist.\n\n' % sceneFile
elif PathUtils.IsPathLocal( sceneFile ) and not scriptDialog.GetValue( "SubmitSceneBox" ):
warnings += 'SketchUp file "%s" is local.\n\n' % sceneFile
# Check Output
exportDirectory = scriptDialog.GetValue( "ExportDirectoryBox" ).strip()
if not exportDirectory:
errors += "An output directory was not specific.\n\n"
elif not os.path.isdir( exportDirectory ):
errors += 'The directory of the output file does not exist: "%s"\n\n' % exportDirectory
isVray = scriptDialog.GetEnabled( "VrayBox" ) and scriptDialog.GetValue( "VrayBox" )
vrayVersion = int( scriptDialog.GetValue( "VrayVersionBox" ) )
vrayFrames = scriptDialog.GetValue( "VrayFramesBox" ).strip()
# Check if a valid frame range has been specified for V-Ray 3 or later
if isVray and vrayVersion >= 3 and is2dAnimation() and not FrameUtils.FrameRangeValid( vrayFrames ):
errors += 'Frame range "%s" is not valid.\n\n' % vrayFrames
if errors:
scriptDialog.ShowMessageBox( "The following errors occurred, you must fix these before continuing.\n\n%s" % errors.strip(), "Error" )
return False
elif warnings:
result = scriptDialog.ShowMessageBox( "The following warnings occurred, are you sure you want to continue?\n\n%s" % warnings.strip(), "Warning", ( "Yes", "No" ) )
if result == "No":
return False
return True
def SubmitButtonPressed(*args):
global scriptDialog
global integration_dialog
try:
submitScene = scriptDialog.GetValue("SubmitSceneBox")
# Check if Integration options are valid
if not integration_dialog.CheckIntegrationSanity():
return
warnings = []
errors = []
# Check if max files exist.
sceneFiles = StringUtils.FromSemicolonSeparatedString(
scriptDialog.GetValue("SceneBox"), False)
if not sceneFiles:
errors.append("No 3dsmax file specified")
for sceneFile in sceneFiles:
if not File.Exists(sceneFile):
errors.append("3dsmax file %s does not exist" % sceneFile)
return
elif not submitScene and PathUtils.IsPathLocal(sceneFile):
warnings.append(
"The scene file " + sceneFile +
" is local and is not being submitted with the job, are you sure you want to continue?"
)
# Check if path config file exists.
pathConfigFile = scriptDialog.GetValue("PathConfigBox").strip()
if pathConfigFile:
if not File.Exists(pathConfigFile):
errors.append("Path configuration file %s does not exist" %
pathConfigFile)
# Check if PreLoad script file exists.
preLoadFile = scriptDialog.GetValue("PreLoadBox").strip()
if preLoadFile:
if not File.Exists(preLoadFile):
errors.append("PreLoad MAXScript file %s does not exist" %
preLoadFile)
# Check if PostLoad script file exists.
postLoadFile = scriptDialog.GetValue("PostLoadBox").strip()
if postLoadFile:
if not File.Exists(postLoadFile):
errors.append("PostLoad MAXScript file %s does not exist" %
postLoadFile)
# Check if PreFrame script file exists.
preFrameFile = scriptDialog.GetValue("PreFrameBox").strip()
if preFrameFile:
if not File.Exists(preFrameFile):
errors.append("PreFrame MAXScript file %s does not exist" %
preFrameFile)
# Check if PostFrame script file exists.
postFrameFile = scriptDialog.GetValue("PostFrameBox").strip()
if postFrameFile:
if not File.Exists(postFrameFile):
errors.append("PostFrame MAXScript file %s does not exist" %
postFrameFile)
# Check if a valid frame range has been specified.
frames = scriptDialog.GetValue("FramesBox")
if scriptDialog.GetValue("DBRModeBox") != "Disabled":
frameArray = FrameUtils.Parse(frames)
if len(frameArray) != 1:
errors.append(
"Please specify a single frame for distributed rendering in the Frame List."
)
elif not FrameUtils.FrameRangeValid(frames):
errors.append(
"Please specify a valid single frame for distributed rendering in the Frame List."
)
frames = "0-" + str(scriptDialog.GetValue("DBRServersBox") - 1)
else:
if not FrameUtils.FrameRangeValid(frames):
errors.append("Frame range %s is not valid" % frames)
# If using 'select GPU device Ids' then check device Id syntax is valid
if scriptDialog.GetValue(
"GPUsPerTaskBox") == 0 and scriptDialog.GetValue(
"GPUsSelectDevicesBox"):
regex = re.compile("^(\d{1,2}(,\d{1,2})*)?$")
validSyntax = regex.match(
scriptDialog.GetValue("GPUsSelectDevicesBox"))
if not validSyntax:
errors.append(
"'Select GPU Devices' syntax is invalid!\nTrailing 'commas' if present, should be removed.\nValid Examples: 0 or 2 or 0,1,2 or 0,3,4 etc"
)
# Check if concurrent threads > 1
if scriptDialog.GetValue("ConcurrentTasksBox") > 1:
errors.append(
"If using 'Select GPU Devices', then 'Concurrent Tasks' must be set to 1"
)
if errors:
scriptDialog.ShowMessageBox(
"The following errors were encountered:\n\n%s\n\nPlease resolve these issues and submit again.\n"
% ("\n\n".join(errors)), "Errors")
return
if warnings:
result = scriptDialog.ShowMessageBox(
"Warnings:\n\n%s\n\nDo you still want to continue?" %
("\n\n".join(warnings)), "Warnings", ("Yes", "No"))
if result == "No":
return
successes = 0
failures = 0
# Submit each scene file separately.
for sceneFile in sceneFiles:
jobName = scriptDialog.GetValue("NameBox")
if len(sceneFiles) > 1:
jobName = jobName + " [" + Path.GetFileName(sceneFile) + "]"
# Create job info file.
jobInfoFilename = Path.Combine(ClientUtils.GetDeadlineTempPath(),
"max_job_info.job")
writer = StreamWriter(jobInfoFilename, False, Encoding.Unicode)
writer.WriteLine("Plugin=3dsmax")
writer.WriteLine("Name=%s" % jobName)
writer.WriteLine("Comment=%s" %
scriptDialog.GetValue("CommentBox"))
writer.WriteLine("Department=%s" %
scriptDialog.GetValue("DepartmentBox"))
writer.WriteLine("Pool=%s" % scriptDialog.GetValue("PoolBox"))
writer.WriteLine("SecondaryPool=%s" %
scriptDialog.GetValue("SecondaryPoolBox"))
writer.WriteLine("Group=%s" % scriptDialog.GetValue("GroupBox"))
writer.WriteLine("Priority=%s" %
scriptDialog.GetValue("PriorityBox"))
writer.WriteLine("TaskTimeoutMinutes=%s" %
scriptDialog.GetValue("TaskTimeoutBox"))
writer.WriteLine("EnableAutoTimeout=%s" %
scriptDialog.GetValue("AutoTimeoutBox"))
writer.WriteLine("ConcurrentTasks=%s" %
scriptDialog.GetValue("ConcurrentTasksBox"))
writer.WriteLine("LimitConcurrentTasksToNumberOfCpus=%s" %
scriptDialog.GetValue("LimitConcurrentTasksBox"))
writer.WriteLine("MachineLimit=%s" %
scriptDialog.GetValue("MachineLimitBox"))
if scriptDialog.GetValue("IsBlacklistBox"):
writer.WriteLine("Blacklist=%s" %
scriptDialog.GetValue("MachineListBox"))
else:
writer.WriteLine("Whitelist=%s" %
scriptDialog.GetValue("MachineListBox"))
writer.WriteLine("LimitGroups=%s" %
scriptDialog.GetValue("LimitGroupBox"))
writer.WriteLine("JobDependencies=%s" %
scriptDialog.GetValue("DependencyBox"))
writer.WriteLine("OnJobComplete=%s" %
scriptDialog.GetValue("OnJobCompleteBox"))
if scriptDialog.GetValue("SubmitSuspendedBox"):
writer.WriteLine("InitialStatus=Suspended")
writer.WriteLine("Frames=%s" % frames)
if scriptDialog.GetValue("DBRModeBox") == "Disabled":
writer.WriteLine("ChunkSize=%s" %
scriptDialog.GetValue("ChunkSizeBox"))
# Integration
extraKVPIndex = 0
groupBatch = False
if integration_dialog.IntegrationProcessingRequested():
extraKVPIndex = integration_dialog.WriteIntegrationInfo(
writer, extraKVPIndex)
groupBatch = groupBatch or integration_dialog.IntegrationGroupBatchRequested(
)
if groupBatch:
writer.WriteLine("BatchName=%s\n" % jobName)
writer.Close()
# Create plugin info file.
pluginInfoFilename = Path.Combine(
ClientUtils.GetDeadlineTempPath(), "max_plugin_info.job")
writer = StreamWriter(pluginInfoFilename, False, Encoding.Unicode)
writer.WriteLine("Version=%s" %
scriptDialog.GetValue("VersionBox"))
writer.WriteLine("IsMaxDesign=%s" %
scriptDialog.GetValue("IsMaxDesignBox"))
if int(scriptDialog.GetValue("VersionBox")) < 2014:
writer.WriteLine("MaxVersionToForce=%s" %
scriptDialog.GetValue("BuildBox"))
writer.WriteLine("MaxVersionToForce0=None")
writer.WriteLine("MaxVersionToForce1=32bit")
writer.WriteLine("MaxVersionToForce2=64bit")
writer.WriteLine("UseSlaveMode=%d" %
(not scriptDialog.GetValue("WorkstationModeBox")))
if scriptDialog.GetValue("WorkstationModeBox"):
writer.WriteLine("UseSilentMode=%s" %
(scriptDialog.GetValue("SilentModeBox")))
else:
writer.WriteLine("UseSilentMode=False")
writer.WriteLine("ShowFrameBuffer=%s" %
scriptDialog.GetValue("ShowVfbBox"))
writer.WriteLine("RemovePadding=%s" %
scriptDialog.GetValue("RemovePaddingBox"))
writer.WriteLine("RestartRendererMode=%s" %
scriptDialog.GetValue("RestartRendererBox"))
writer.WriteLine(
"IgnoreMissingExternalFiles=%s" %
scriptDialog.GetValue("IgnoreMissingExternalFilesBox"))
writer.WriteLine("IgnoreMissingUVWs=%s" %
scriptDialog.GetValue("IgnoreMissingUVWsBox"))
writer.WriteLine("IgnoreMissingXREFs=%s" %
scriptDialog.GetValue("IgnoreMissingXREFsBox"))
writer.WriteLine("IgnoreMissingDLLs=%s" %
scriptDialog.GetValue("IgnoreMissingDLLsBox"))
writer.WriteLine("LocalRendering=%s" %
scriptDialog.GetValue("LocalRenderingBox"))
writer.WriteLine("DisableMultipass=%s" %
scriptDialog.GetValue("DisableMultipassBox"))
writer.WriteLine("OneCpuPerTask=%s" %
scriptDialog.GetValue("OneCpuPerTaskBox"))
if pathConfigFile:
writer.WriteLine("PathConfigFile=%s" %
Path.GetFileName(pathConfigFile))
writer.WriteLine("MergePathConfigFile=%s" %
scriptDialog.GetValue("MergePathConfigBox"))
if preLoadFile:
writer.WriteLine("PreLoadScript=%s" %
Path.GetFileName(preLoadFile))
if postLoadFile:
writer.WriteLine("PostLoadScript=%s" %
Path.GetFileName(postLoadFile))
if preFrameFile:
writer.WriteLine("PreFrameScript=%s" %
Path.GetFileName(preFrameFile))
if postFrameFile:
writer.WriteLine("PostFrameScript=%s" %
Path.GetFileName(postFrameFile))
pluginIniOverride = scriptDialog.GetValue("PluginIniBox").strip()
if pluginIniOverride:
writer.WriteLine("OverridePluginIni=%s" % pluginIniOverride)
writer.WriteLine("GammaCorrection=%s" %
scriptDialog.GetValue("GammaCorrectionBox"))
writer.WriteLine("GammaInput=%s" %
scriptDialog.GetValue("GammaInputBox"))
writer.WriteLine("GammaOutput=%s" %
scriptDialog.GetValue("GammaOutputBox"))
if int(scriptDialog.GetValue("VersionBox")) >= 2013:
if scriptDialog.GetValue("OverrideLanguageBox"):
writer.WriteLine("Language=%s" %
scriptDialog.GetValue("LanguageBox"))
else:
writer.WriteLine("Language=Default")
camera = scriptDialog.GetValue("CameraBox")
if camera:
writer.WriteLine("Camera=%s" % camera)
writer.WriteLine("Camera0=")
writer.WriteLine("Camera1=%s" % camera)
if not submitScene:
writer.WriteLine("SceneFile=%s" % sceneFile)
if scriptDialog.GetValue("DBRModeBox") != "Disabled":
if scriptDialog.GetValue("DBRModeBox") == "VRay DBR":
writer.WriteLine("VRayDBRJob=True")
elif scriptDialog.GetValue(
"DBRModeBox") == "Mental Ray Satellite":
writer.WriteLine("MentalRayDBRJob=True")
elif scriptDialog.GetValue("DBRModeBox") == "VRay RT DBR":
writer.WriteLine("VRayRtDBRJob=True")
writer.WriteLine("DBRJobFrame=%s" %
scriptDialog.GetValue("FramesBox"))
if scriptDialog.GetValue("DBRModeBox") == "Disabled":
writer.WriteLine("GPUsPerTask=%s" %
scriptDialog.GetValue("GPUsPerTaskBox"))
writer.WriteLine("GPUsSelectDevices=%s" %
scriptDialog.GetValue("GPUsSelectDevicesBox"))
writer.Close()
# Setup the command line arguments.
arguments = [jobInfoFilename, pluginInfoFilename]
if scriptDialog.GetValue("SubmitSceneBox"):
arguments.append(sceneFile)
if pathConfigFile:
arguments.append(pathConfigFile)
if preLoadFile:
arguments.append(preLoadFile)
if postLoadFile:
arguments.append(postLoadFile)
if preFrameFile:
arguments.append(preFrameFile)
if postFrameFile:
arguments.append(postFrameFile)
if len(sceneFiles) == 1:
results = ClientUtils.ExecuteCommandAndGetOutput(arguments)
scriptDialog.ShowMessageBox(results, "Submission Results")
else:
# Now submit the job.
exitCode = ClientUtils.ExecuteCommand(arguments)
if exitCode == 0:
successes += 1
else:
failures += 1
if len(sceneFiles) > 1:
scriptDialog.ShowMessageBox(
"Jobs submitted successfully: %d\nJobs not submitted: %d" %
(successes, failures), "Submission Results")
except:
scriptDialog.ShowMessageBox(traceback.format_exc(), "")
def PreRenderTasks(self):
self.LogInfo("Starting V-Ray Task")
# Get the scene file to render.
sceneFilename = self.GetPluginInfoEntry("InputFilename")
sceneFilename = RepositoryUtils.CheckPathMapping(sceneFilename)
sceneFilename = PathUtils.ToPlatformIndependentPath(sceneFilename)
# If we're rendering separate files per frame, need to get the vrscene file for the current frame.
separateFilesPerFrame = self.GetBooleanPluginInfoEntryWithDefault(
"SeparateFilesPerFrame", False)
if separateFilesPerFrame:
currPadding = FrameUtils.GetFrameStringFromFilename(sceneFilename)
paddingSize = len(currPadding)
newPadding = StringUtils.ToZeroPaddedString(
self.GetStartFrame(), paddingSize, False)
sceneFilename = sceneFilename.replace(currPadding, newPadding)
# Check if we should be doing path mapping.
slaveInfo = RepositoryUtils.GetSlaveInfo(self.GetSlaveName(), True)
if self.GetBooleanConfigEntryWithDefault(
"EnableVrscenePathMapping",
True) or slaveInfo.IsAWSPortalInstance:
self.LogInfo("Performing path mapping on vrscene file")
tempSceneDirectory = self.CreateTempDirectory(
"thread" + str(self.GetThreadNumber()))
tempSceneFileName = Path.GetFileName(sceneFilename)
self.tempSceneFilename = os.path.join(tempSceneDirectory,
tempSceneFileName)
mappedFiles = set()
filesToMap = [(sceneFilename, os.path.dirname(sceneFilename))]
while filesToMap:
curFile, originalSceneDirectory = filesToMap.pop()
#Include directives normally contain relative paths but just to be safe we must run pathmapping again.
curFile = RepositoryUtils.CheckPathMapping(curFile)
if not os.path.isabs(curFile):
curFile = os.path.join(originalSceneDirectory, curFile)
if not os.path.exists(curFile):
self.LogInfo("The include file %s does not exist." %
curFile)
continue
if os.path.basename(curFile) in mappedFiles:
self.LogInfo(
"The include file %s has already been mapped." %
curFile)
continue
self.LogInfo("Starting to map %s" % curFile)
mappedFiles.add(os.path.basename(curFile))
tempFileName = os.path.join(tempSceneDirectory,
os.path.basename(curFile))
foundFiles = self.map_vrscene_includes_cancelable(
curFile, tempFileName)
newOriginDirectory = os.path.dirname(curFile)
for foundFile in foundFiles:
filesToMap.append((foundFile, newOriginDirectory))
map_files_and_replace_slashes(tempFileName, tempFileName)
if self.IsCanceled():
self.FailRender(
"Received cancel task command from Deadline.")
else:
self.tempSceneFilename = sceneFilename
self.tempSceneFilename = PathUtils.ToPlatformIndependentPath(
self.tempSceneFilename)
def _submit_button_pressed( self, *args ):
"""
Callback for the "submit" button. Submits a job to Deadline
:param args:
:return:
"""
warnings = []
errors = []
# If using 'select GPU device Ids' then check device Id syntax is valid.
if self.GetValue( "GPUsPerTaskBox" ) == 0 and self.GetValue( "GPUsSelectDevicesBox" ) != "":
# regex = re.compile( "^(\d{1,2}(,\d{1,2})*)?$" ) # Multiple gpu check
regex = re.compile( "^([0-9]|1[0-6])$" ) # Single gpu check (Altus 1.8 currently only supports 1 gpu)
valid_syntax = regex.match( self.GetValue( "GPUsSelectDevicesBox" ) )
if not valid_syntax:
# errors.append( "'Select GPU Devices' syntax is invalid!\nTrailing 'commas' if present, should be removed.\nValid Examples: 0 or 2 or 0,1,2 or 0,3,4 etc" ) # Multiple gpu check
errors.append( "'Select GPU Device' syntax is invalid!\nSingle GPU ID 0-16 supported only.\nValid Examples: 0 or 1 or 2 etc" ) # Single gpu check (Altus 1.8 currently only supports 1 gpu per thread)
# Check if concurrent threads > 1.
if self.GetValue( "ConcurrentTasksBox" ) > 1:
errors.append( "If using 'Select GPU Devices', then 'Concurrent Tasks' must be set to 1" )
stereo_enabled = self.GetValue( "StereoBox" )
version = float( self.GetValue( "VersionBox" ) )
full_output_path = self.GetValue( "OutputBox" )
use_config = self.GetValue( "configCheck" )
# Check if a valid frame range has been specified.
frames = self.GetValue( "FramesBox" )
if not FrameUtils.FrameRangeValid( frames ):
errors.append( "Frame range: %s is not valid" % frames )
if not full_output_path:
errors.append( "The output image name is required." )
self._check_output_destination( errors, warnings )
if use_config:
self._check_config_input( errors, warnings )
else:
self._check_file_inputs( errors, warnings )
# Check if Integration options are valid.
if not self.integration_dialog.CheckIntegrationSanity( full_output_path ):
return
if errors:
self.ShowMessageBox( "The following errors were encountered:\n\n%s\n\nPlease resolve these issues and submit again.\n" % ( "\n".join( errors ) ), "Errors" )
return
if warnings:
result = self.ShowMessageBox( "Warnings:\n\n%s\n\nDo you still want to continue?" % ( "\n".join( warnings ) ), "Warnings", ( "Yes","No" ) )
if result == "No":
return
# Create job info file.
job_info_file = os.path.join( ClientUtils.GetDeadlineTempPath(), "altus_job_info.job" )
writer = StreamWriter( job_info_file, False, Encoding.Unicode )
writer.WriteLine( "Plugin=Altus" )
writer.WriteLine( "Name=%s" % self.GetValue( "NameBox" ) )
writer.WriteLine( "Comment=%s" % self.GetValue( "CommentBox" ) )
writer.WriteLine( "Department=%s" % self.GetValue( "DepartmentBox" ) )
writer.WriteLine( "Pool=%s" % self.GetValue( "PoolBox" ) )
writer.WriteLine( "SecondaryPool=%s" % self.GetValue( "SecondaryPoolBox" ) )
writer.WriteLine( "Group=%s" % self.GetValue( "GroupBox" ) )
writer.WriteLine( "Priority=%s" % self.GetValue( "PriorityBox" ) )
writer.WriteLine( "TaskTimeoutMinutes=%s" % self.GetValue( "TaskTimeoutBox" ) )
writer.WriteLine( "EnableAutoTimeout=%s" % self.GetValue( "AutoTimeoutBox" ) )
writer.WriteLine( "ConcurrentTasks=%s" % self.GetValue( "ConcurrentTasksBox" ) )
writer.WriteLine( "LimitConcurrentTasksToNumberOfCpus=%s" % self.GetValue( "LimitConcurrentTasksBox" ) )
writer.WriteLine( "MachineLimit=%s" % self.GetValue( "MachineLimitBox" ) )
if self.GetValue( "IsBlacklistBox" ):
writer.WriteLine( "Blacklist=%s" % self.GetValue( "MachineListBox" ) )
else:
writer.WriteLine( "Whitelist=%s" % self.GetValue( "MachineListBox" ) )
writer.WriteLine( "LimitGroups=%s" % self.GetValue( "LimitGroupBox" ) )
writer.WriteLine( "JobDependencies=%s" % self.GetValue( "DependencyBox" ) )
writer.WriteLine( "OnJobComplete=%s" % self.GetValue( "OnJobCompleteBox" ) )
full_output_path = full_output_path.replace( "\\", "/" )
writer.WriteLine( "OutputFilename0=%s" % full_output_path )
if self.GetValue( "SubmitSuspendedBox" ):
writer.WriteLine( "InitialStatus=Suspended" )
writer.WriteLine( "Frames=%s" % frames )
writer.WriteLine( "ChunkSize=%s" % self.GetValue( "ChunkSizeBox" ) )
extra_kvp_index = 0
group_batch = False
if self.integration_dialog.IntegrationProcessingRequested():
extra_kvp_index = self.integration_dialog.WriteIntegrationInfo( writer, extra_kvp_index )
group_batch = group_batch or self.integration_dialog.IntegrationGroupBatchRequested()
if group_batch:
writer.WriteLine( "BatchName=%s" % ( self.GetValue( "NameBox" ) ) )
writer.Close()
# Create plugin info file.
plugin_info_file = os.path.join( ClientUtils.GetDeadlineTempPath(), "altus_plugin_info.job" )
writer = StreamWriter( plugin_info_file, False, Encoding.Unicode )
exe_type = self.GetValue( "ExecutableTypeBox" )
writer.WriteLine( "ExecutableType=%s" % exe_type )
writer.WriteLine( "Version=%s" % version )
writer.WriteLine( "OutputFile=%s" % full_output_path )
if exe_type in [ "OpenCl", "GPU", "Auto Select" ]:
writer.WriteLine( "GPUsPerTask=%s" % self.GetValue( "GPUsPerTaskBox" ) )
writer.WriteLine( "GPUsSelectDevices=%s" % self.GetValue( "GPUsSelectDevicesBox" ) )
# If config file defined, then skip all other settings.
if use_config:
writer.WriteLine( "Config=%s" % self.GetValue( "ConfigBox" ) )
else:
for passType in AltusSubmissionDialog.PASS_TYPES:
if self._pass_enabled( passType ):
writer.WriteLine( "%sFiles=%s;" % ( passType, self.GetValue( "%sBox" % passType ) ) )
writer.WriteLine( "Renderer=%s" % self.GetValue( "RendererBox" ) )
writer.WriteLine( "PreserveLayers=%s" % self.GetValue( "LayerPreservationBox" ) )
writer.WriteLine( "OutputQuality=%s" % self.GetValue( "QualityBox" ) )
writer.WriteLine( "AOVQuality=%s" % self.GetValue( "AOVQualityBox" ) )
writer.WriteLine( "FrameRadius=%s" % self.GetValue( "FrameRadiusBox" ) )
writer.WriteLine( "FilterRadius=%s" % self.GetValue( "FilterRadiusBox" ) )
writer.WriteLine( "Kc_1=%s" % self.GetValue( "Kc1Box" ) )
writer.WriteLine( "Kc_2=%s" % self.GetValue( "Kc2Box" ) )
writer.WriteLine( "Kc_4=%s" % self.GetValue( "Kc4Box" ) )
writer.WriteLine( "Kf=%s" % self.GetValue( "KfBox" ) )
writer.WriteLine( "Stereo=%s" % stereo_enabled )
if version == 1.8:
# 1.8 is either False (Level 0) or True (Level 3)
writer.WriteLine( "ForceContinue=%s" % bool( self.GetValue( "ForceContinueIntBox" )[ 0 ] ) )
else:
writer.WriteLine( "ForceContinueInt=%s" % self.GetValue( "ForceContinueIntBox" )[ 0 ] ) # Just need first character
# Tile options
writer.WriteLine( "Tile=%s" % self.GetValue( "TileBox" ) )
writer.WriteLine( "TileSize=%s" % self.GetValue( "TileSizeBox" ) )
writer.WriteLine( "FireFly=%s" % self.GetValue( "FireFlyBox" ) )
if version >= 1.9:
writer.WriteLine( "SkipFrameRadius=%s" % self.GetValue( "SkipFrameRadiusBox" ) )
writer.WriteLine( "IgnoreAlpha=%s" % self.GetValue( "IgnoreAlphaBox" ) )
if version >= 2.1:
writer.WriteLine( "SinglePass=%s" % self.GetValue( "SinglePassBox" ) )
if version >= 2.3:
writer.WriteLine( "Force16bitOutput=%s" % self.GetValue( "Force16bitOutputBox" ) )
writer.WriteLine( "EXRCompression=%s" % self.GetValue( "EXRCompressionBox" ) )
writer.Close()
# Setup the command line arguments.
arguments = [ job_info_file, plugin_info_file ]
# Now submit the job.
results = ClientUtils.ExecuteCommandAndGetOutput( arguments )
self.ShowMessageBox( results, "Submission Results" )
def PostRenderTasks(self):
failures = 0
if self.FailWithoutFinishedMessage:
if not self.RenderSuccess: # fail here if Succeed didn't get tripped.
self.FailAERender(
"The composition was not rendered completely. Found NO \"Finished Composition\" notice. To disable this failure if applicable, ensure 'Fail Without Finished Message' is set to False in AE plugin configuration."
)
myRePath = self.getAERenderOnlyFileName()
self.CleanupRenderEngineFile(myRePath)
# Clean up the temp file if we did path mapping on the aepx project file.
if (os.path.splitext(self.TempSceneFilename)[1]
).lower() == ".aepx" and self.GetBooleanConfigEntryWithDefault(
"EnableAepxPathMapping", True):
os.remove(self.TempSceneFilename)
if self.DelayedFailure:
self.FailRender(
"There was an error but the message could not properly be extracted\n"
+ self.FailureMessage)
else:
# Since the output option is only valid when rendering a specific comp, we will only check
# the output size in this case.
comp = self.GetPluginInfoEntryWithDefault("Comp", "")
if comp and not self.MultiMachine:
if self.LocalRendering:
self.LogInfo("Moving output files and folders from " +
self.LocalFilePath + " to " +
self.NetworkFilePath)
self.VerifyAndMoveDirectory(self.LocalFilePath,
self.NetworkFilePath, True, 0)
outputDirectories = self.GetJob().JobOutputDirectories
outputFilenames = self.GetJob().JobOutputFileNames
minSizeInKB = self.GetIntegerPluginInfoEntryWithDefault(
"MinFileSize", 0)
deleteFilesUnderMinSize = self.GetBooleanPluginInfoEntryWithDefault(
"DeleteFilesUnderMinSize", False)
for outputDirectory, outputFilename in zip(
outputDirectories, outputFilenames):
outputFile = os.path.join(outputDirectory, outputFilename)
outputFile = RepositoryUtils.CheckPathMapping(outputFile)
outputFile = PathUtils.ToPlatformIndependentPath(
outputFile)
if outputFile and minSizeInKB > 0:
if outputFile.find("#") >= 0:
frames = range(self.GetStartFrame(),
self.GetEndFrame() + 1)
for frame in frames:
currFile = FrameUtils.ReplacePaddingWithFrameNumber(
outputFile, frame)
failures += self.CheckFileSize(
currFile, minSizeInKB,
deleteFilesUnderMinSize)
else:
failures += self.CheckFileSize(
outputFile, minSizeInKB,
deleteFilesUnderMinSize)
if failures != 0:
self.FailRender(
"There were 1 or more errors with the output")
else:
self.SetProgress(100)
self.SetStatusMessage("Finished Rendering " +
str(self.GetEndFrame() -
self.GetStartFrame() + 1) +
" frames")
else:
self.LogInfo(
"Skipping output file size check because a specific comp isn't being rendered or Multi-Machine rendering is enabled"
)
def _scene_box_changed(self, *args):
"""Event handler fired when the input file UI control's value is changed"""
if not self.updating_output_file:
success = False
try:
filename = self.GetValue("SceneBox")
is_vrscene = False
if filename != "":
init_frame = FrameUtils.GetFrameNumberFromFilename(
filename)
padding_size = FrameUtils.GetPaddingSizeFromFilename(
filename)
extension = os.path.splitext(filename)[1].lower()
is_vrscene = extension == ".vrscene"
start_frame = 0
end_frame = 0
if padding_size > 0:
filename = FrameUtils.GetLowerFrameFilename(
filename, init_frame, padding_size)
self.updating_output_file = True
self.SetValue(
"SceneBox",
FrameUtils.ReplaceFrameNumberWithPadding(
filename, "#"))
self.updating_output_file = False
start_frame = FrameUtils.GetLowerFrameRange(
filename, init_frame, padding_size)
end_frame = FrameUtils.GetUpperFrameRange(
filename, init_frame, padding_size)
output_filename = FrameUtils.GetFilenameWithoutPadding(
filename)
else:
output_filename = filename
self.SetValue("FrameListBox",
"%s-%s" % (start_frame, end_frame))
self.SetValue(
"NameBox",
os.path.splitext(os.path.basename(output_filename))[0])
self.SetEnabled("VrsceneNodeNameBox", is_vrscene)
self.SetEnabled("VrsceneNodeNameLabel", is_vrscene)
self.SetEnabled("VrsceneApplyTransformBox", is_vrscene)
self.SetEnabled("VrsceneVelocityBox", is_vrscene)
self.SetEnabled("VrsceneAnimationBox", is_vrscene)
animation_checked = self.GetValue("VrsceneAnimationBox")
self.SetEnabled("VrsceneStartFrameBox", is_vrscene
and animation_checked)
self.SetEnabled("VrsceneEndFrameBox", is_vrscene
and animation_checked)
except Exception as e:
self.ShowMessageBox(e.Message, "Error Parsing Input Images")
else:
success = True
if not success:
self.SetValue("SceneBox", "")
self.SetValue("NameBox", "Untitled")
def SubmitButtonPressed(*args):
global scriptDialog
global integration_dialog
submitScene = bool(scriptDialog.GetValue("SubmitSceneBox"))
multiMachine = bool(scriptDialog.GetValue("MultiMachineBox"))
# Check if scene file exists
sceneFile = scriptDialog.GetValue( "SceneBox" )
if( not File.Exists( sceneFile ) ):
scriptDialog.ShowMessageBox("Project file %s does not exist." % sceneFile, "Error" )
return
elif(not submitScene and PathUtils.IsPathLocal(sceneFile)):
result = scriptDialog.ShowMessageBox("The project file " + sceneFile + " is local, are you sure you want to continue?","Warning", ("Yes","No") )
if( result == "No" ):
return
concurrentTasks = scriptDialog.GetValue( "ConcurrentTasksBox" )
if concurrentTasks > 1:
result = scriptDialog.ShowMessageBox("The concurrent tasks is set to a value greater than 1. This can cause Jobs to hang when rendering, are you sure you want to continue?","Warning", ("Yes","No") )
if( result == "No" ):
return
outputFile = ""
frames = ""
jobName = scriptDialog.GetValue( "NameBox" )
# Get the comp
comp = scriptDialog.GetValue("CompBox")
if comp != "":
# Check that the output is valid
outputFile = scriptDialog.GetValue( "OutputBox" ).strip()
if len(outputFile) > 0:
if not Directory.Exists( Path.GetDirectoryName(outputFile) ):
scriptDialog.ShowMessageBox( "The directory of the output file does not exist:\n" + Path.GetDirectoryName(outputFile), "Error" )
return
elif(PathUtils.IsPathLocal(outputFile)):
result = scriptDialog.ShowMessageBox("The output file " + outputFile + " is local, are you sure you want to continue?","Warning", ("Yes","No") )
if( result == "No" ):
return
extension = Path.GetExtension( outputFile )
if not IsMovieFormat( extension ):
if outputFile.find( "[#" ) < 0 and outputFile.find( "#]" ) < 0:
directory = Path.GetDirectoryName( outputFile )
filename = Path.GetFileNameWithoutExtension( outputFile )
outputFile = Path.Combine( directory, filename + "[#####]" + extension )
#Since we don't specify ranges for multi-machine rendering, don't check frame range when Multi-Machine Rendering = True
if not multiMachine:
# Check if a valid frame range has been specified.
frames = scriptDialog.GetValue( "FramesBox" )
if( not FrameUtils.FrameRangeValid( frames ) ):
scriptDialog.ShowMessageBox( "Frame range %s is not valid" % frames, "Error" )
return
else:
jobName = jobName + " - Entire Render Queue"
if multiMachine:
jobName = jobName + " (multi-machine rendering)"
# Check if Integration options are valid
if not integration_dialog.CheckIntegrationSanity( outputFile ):
return
# Create job info file.
jobInfoFilename = Path.Combine( ClientUtils.GetDeadlineTempPath(), "ae_job_info.job" )
writer = StreamWriter( jobInfoFilename, False, Encoding.Unicode )
writer.WriteLine( "Plugin=AfterEffects" )
writer.WriteLine( "Name=%s" % jobName )
writer.WriteLine( "Comment=%s" % scriptDialog.GetValue( "CommentBox" ) )
writer.WriteLine( "Department=%s" % scriptDialog.GetValue( "DepartmentBox" ) )
writer.WriteLine( "Pool=%s" % scriptDialog.GetValue( "PoolBox" ) )
writer.WriteLine( "SecondaryPool=%s" % scriptDialog.GetValue( "SecondaryPoolBox" ) )
writer.WriteLine( "Group=%s" % scriptDialog.GetValue( "GroupBox" ) )
writer.WriteLine( "Priority=%s" % scriptDialog.GetValue( "PriorityBox" ) )
writer.WriteLine( "TaskTimeoutMinutes=%s" % scriptDialog.GetValue( "TaskTimeoutBox" ) )
writer.WriteLine( "EnableAutoTimeout=%s" % scriptDialog.GetValue( "AutoTimeoutBox" ) )
writer.WriteLine( "ConcurrentTasks=%s" % concurrentTasks )
writer.WriteLine( "LimitConcurrentTasksToNumberOfCpus=%s" % scriptDialog.GetValue( "LimitConcurrentTasksBox" ) )
if( bool(scriptDialog.GetValue( "IsBlacklistBox" )) ):
writer.WriteLine( "Blacklist=%s" % scriptDialog.GetValue( "MachineListBox" ) )
else:
writer.WriteLine( "Whitelist=%s" % scriptDialog.GetValue( "MachineListBox" ) )
writer.WriteLine( "LimitGroups=%s" % scriptDialog.GetValue( "LimitGroupBox" ) )
writer.WriteLine( "JobDependencies=%s" % scriptDialog.GetValue( "DependencyBox" ) )
writer.WriteLine( "OnJobComplete=%s" % scriptDialog.GetValue( "OnJobCompleteBox" ) )
if( bool(scriptDialog.GetValue( "SubmitSuspendedBox" )) ):
writer.WriteLine( "InitialStatus=Suspended" )
if multiMachine:
writer.WriteLine( "MachineLimit=0" )
writer.WriteLine( "Frames=1-%s" % scriptDialog.GetValue( "MultiMachineTasksBox" ) )
writer.WriteLine( "ChunkSize=1" )
else:
if comp != "":
writer.WriteLine( "MachineLimit=%s" % scriptDialog.GetValue( "MachineLimitBox" ) )
writer.WriteLine( "Frames=%s" % frames )
writer.WriteLine( "ChunkSize=%s" % scriptDialog.GetValue( "ChunkSizeBox" ) )
else:
writer.WriteLine( "MachineLimit=1" )
writer.WriteLine( "Frames=0" )
writer.WriteLine( "ChunkSize=1" )
if len(outputFile) > 0:
writer.WriteLine( "OutputFilename0=%s" % outputFile.replace( "[#####]", "#####" ) )
extraKVPIndex = 0
groupBatch = False
if integration_dialog.IntegrationProcessingRequested():
extraKVPIndex = integration_dialog.WriteIntegrationInfo( writer, extraKVPIndex )
groupBatch = groupBatch or integration_dialog.IntegrationGroupBatchRequested()
if groupBatch:
writer.WriteLine( "BatchName=%s\n" % ( jobName ) )
writer.Close()
# Create plugin info file.
version = GetVersionNumber()
pluginInfoFilename = Path.Combine( ClientUtils.GetDeadlineTempPath(), "ae_plugin_info.job" )
writer = StreamWriter( pluginInfoFilename, False, Encoding.Unicode )
if(not bool(scriptDialog.GetValue("SubmitSceneBox"))):
writer.WriteLine("SceneFile=%s" % scriptDialog.GetValue("SceneBox").replace("\\","/").strip())
writer.WriteLine("Comp=%s" % scriptDialog.GetValue("CompBox"))
writer.WriteLine("Version=%s" % str(version) )
writer.WriteLine("IgnoreMissingLayerDependenciesErrors=%s" % scriptDialog.GetValue("MissingLayers"))
writer.WriteLine("IgnoreMissingEffectReferencesErrors=%s" % scriptDialog.GetValue("MissingEffects"))
writer.WriteLine("FailOnWarnings=%s" % scriptDialog.GetValue("FailOnWarnings"))
if multiMachine:
writer.WriteLine("MultiMachineMode=True")
else:
writer.WriteLine("LocalRendering=%s" % scriptDialog.GetValue("LocalRendering"))
writer.WriteLine("OverrideFailOnExistingAEProcess=%s" % scriptDialog.GetValue("OverrideFailOnExistingAEProcess"))
writer.WriteLine("FailOnExistingAEProcess=%s" % scriptDialog.GetValue("OverrideFailOnExistingAEProcess"))
writer.WriteLine("MemoryManagement=%s" % scriptDialog.GetValue("MemoryManagement"))
writer.WriteLine("ImageCachePercentage=%s" % scriptDialog.GetValue("ImageCachePercentage"))
writer.WriteLine("MaxMemoryPercentage=%s" % scriptDialog.GetValue("MaxMemoryPercentage"))
writer.WriteLine("MultiProcess=%s" % scriptDialog.GetValue("MultiProcess"))
if version > 8:
writer.WriteLine("ContinueOnMissingFootage=%s" % scriptDialog.GetValue("MissingFootage"))
if len(outputFile) > 0:
writer.WriteLine("Output=%s" % outputFile)
writer.Close()
# Setup the command line arguments.
arguments = [ jobInfoFilename, pluginInfoFilename ]
if scriptDialog.GetValue( "SubmitSceneBox" ):
arguments.append( sceneFile )
# Now submit the job.
results = ClientUtils.ExecuteCommandAndGetOutput( arguments )
scriptDialog.ShowMessageBox( results, "Submission Results" )